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