1 //==- raw_sha1_ostream.h - raw_ostream that compute SHA1 --*- C++ -*-==// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This file defines the raw_sha1_ostream class. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef LLVM_SUPPORT_RAW_SHA1_OSTREAM_H 15 #define LLVM_SUPPORT_RAW_SHA1_OSTREAM_H 16 17 #include "llvm/Support/raw_ostream.h" 18 #include "llvm/Support/SHA1.h" 19 #include "llvm/ADT/ArrayRef.h" 20 21 namespace llvm { 22 23 /// A raw_ostream that hash the content using the sha1 algorithm. 24 class raw_sha1_ostream : public raw_ostream { 25 SHA1 State; 26 27 /// See raw_ostream::write_impl. 28 void write_impl(const char *Ptr, size_t Size) override { 29 State.update(ArrayRef<uint8_t>((const uint8_t *)Ptr, Size)); 30 } 31 32 public: 33 /// Return the current SHA1 hash for the content of the stream 34 StringRef sha1() { 35 flush(); 36 return State.result(); 37 } 38 39 /// Reset the internal state to start over from scratch. 40 void resetHash() { State.init(); } 41 42 uint64_t current_pos() const override { return 0; } 43 }; 44 45 } // end llvm namespace 46 47 #endif 48
