1//===- DebugChecksumsSubsection.h -------------------------------*- 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#ifndef LLVM_DEBUGINFO_CODEVIEW_DEBUGCHECKSUMSSUBSECTION_H 10#define LLVM_DEBUGINFO_CODEVIEW_DEBUGCHECKSUMSSUBSECTION_H 11 12#include "llvm/ADT/ArrayRef.h" 13#include "llvm/ADT/DenseMap.h" 14#include "llvm/ADT/StringRef.h" 15#include "llvm/DebugInfo/CodeView/CodeView.h" 16#include "llvm/DebugInfo/CodeView/DebugSubsection.h" 17#include "llvm/Support/Allocator.h" 18#include "llvm/Support/BinaryStreamArray.h" 19#include "llvm/Support/BinaryStreamRef.h" 20#include "llvm/Support/Error.h" 21#include <cstdint> 22#include <vector> 23 24namespace llvm { 25 26class BinaryStreamReader; 27class BinaryStreamWriter; 28 29namespace codeview { 30 31class DebugStringTableSubsection; 32 33struct FileChecksumEntry { 34 uint32_t FileNameOffset; // Byte offset of filename in global stringtable. 35 FileChecksumKind Kind; // The type of checksum. 36 ArrayRef<uint8_t> Checksum; // The bytes of the checksum. 37}; 38 39} // end namespace codeview 40 41template <> struct VarStreamArrayExtractor<codeview::FileChecksumEntry> { 42public: 43 using ContextType = void; 44 45 Error operator()(BinaryStreamRef Stream, uint32_t &Len, 46 codeview::FileChecksumEntry &Item); 47}; 48 49namespace codeview { 50 51class DebugChecksumsSubsectionRef final : public DebugSubsectionRef { 52 using FileChecksumArray = VarStreamArray<codeview::FileChecksumEntry>; 53 using Iterator = FileChecksumArray::Iterator; 54 55public: 56 DebugChecksumsSubsectionRef() 57 : DebugSubsectionRef(DebugSubsectionKind::FileChecksums) {} 58 59 static bool classof(const DebugSubsectionRef *S) { 60 return S->kind() == DebugSubsectionKind::FileChecksums; 61 } 62 63 bool valid() const { return Checksums.valid(); } 64 65 Error initialize(BinaryStreamReader Reader); 66 Error initialize(BinaryStreamRef Stream); 67 68 Iterator begin() const { return Checksums.begin(); } 69 Iterator end() const { return Checksums.end(); } 70 71 const FileChecksumArray &getArray() const { return Checksums; } 72 73private: 74 FileChecksumArray Checksums; 75}; 76 77class DebugChecksumsSubsection final : public DebugSubsection { 78public: 79 explicit DebugChecksumsSubsection(DebugStringTableSubsection &Strings); 80 81 static bool classof(const DebugSubsection *S) { 82 return S->kind() == DebugSubsectionKind::FileChecksums; 83 } 84 85 void addChecksum(StringRef FileName, FileChecksumKind Kind, 86 ArrayRef<uint8_t> Bytes); 87 88 uint32_t calculateSerializedSize() const override; 89 Error commit(BinaryStreamWriter &Writer) const override; 90 uint32_t mapChecksumOffset(StringRef FileName) const; 91 92private: 93 DebugStringTableSubsection &Strings; 94 95 DenseMap<uint32_t, uint32_t> OffsetMap; 96 uint32_t SerializedSize = 0; 97 BumpPtrAllocator Storage; 98 std::vector<FileChecksumEntry> Checksums; 99}; 100 101} // end namespace codeview 102 103} // end namespace llvm 104 105#endif // LLVM_DEBUGINFO_CODEVIEW_DEBUGCHECKSUMSSUBSECTION_H 106