include/llvm/Support/raw_sha1_ostream.h

   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