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(Data: 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 | std::array<uint8_t, 20> 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 | |