10b57cec5SDimitry Andric //===- DebugChecksumsSubsection.h -------------------------------*- C++ -*-===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric 90b57cec5SDimitry Andric #ifndef LLVM_DEBUGINFO_CODEVIEW_DEBUGCHECKSUMSSUBSECTION_H 100b57cec5SDimitry Andric #define LLVM_DEBUGINFO_CODEVIEW_DEBUGCHECKSUMSSUBSECTION_H 110b57cec5SDimitry Andric 120b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h" 130b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h" 140b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h" 150b57cec5SDimitry Andric #include "llvm/DebugInfo/CodeView/CodeView.h" 160b57cec5SDimitry Andric #include "llvm/DebugInfo/CodeView/DebugSubsection.h" 170b57cec5SDimitry Andric #include "llvm/Support/Allocator.h" 180b57cec5SDimitry Andric #include "llvm/Support/BinaryStreamArray.h" 190b57cec5SDimitry Andric #include "llvm/Support/BinaryStreamRef.h" 200b57cec5SDimitry Andric #include "llvm/Support/Error.h" 210b57cec5SDimitry Andric #include <cstdint> 220b57cec5SDimitry Andric #include <vector> 230b57cec5SDimitry Andric 240b57cec5SDimitry Andric namespace llvm { 250b57cec5SDimitry Andric 2681ad6265SDimitry Andric class BinaryStreamReader; 2781ad6265SDimitry Andric class BinaryStreamWriter; 2881ad6265SDimitry Andric 290b57cec5SDimitry Andric namespace codeview { 300b57cec5SDimitry Andric 310b57cec5SDimitry Andric class DebugStringTableSubsection; 320b57cec5SDimitry Andric 330b57cec5SDimitry Andric struct FileChecksumEntry { 340b57cec5SDimitry Andric uint32_t FileNameOffset; // Byte offset of filename in global stringtable. 350b57cec5SDimitry Andric FileChecksumKind Kind; // The type of checksum. 360b57cec5SDimitry Andric ArrayRef<uint8_t> Checksum; // The bytes of the checksum. 370b57cec5SDimitry Andric }; 380b57cec5SDimitry Andric 390b57cec5SDimitry Andric } // end namespace codeview 400b57cec5SDimitry Andric 410b57cec5SDimitry Andric template <> struct VarStreamArrayExtractor<codeview::FileChecksumEntry> { 420b57cec5SDimitry Andric public: 430b57cec5SDimitry Andric using ContextType = void; 440b57cec5SDimitry Andric 450b57cec5SDimitry Andric Error operator()(BinaryStreamRef Stream, uint32_t &Len, 460b57cec5SDimitry Andric codeview::FileChecksumEntry &Item); 470b57cec5SDimitry Andric }; 480b57cec5SDimitry Andric 490b57cec5SDimitry Andric namespace codeview { 500b57cec5SDimitry Andric 510b57cec5SDimitry Andric class DebugChecksumsSubsectionRef final : public DebugSubsectionRef { 520b57cec5SDimitry Andric using FileChecksumArray = VarStreamArray<codeview::FileChecksumEntry>; 530b57cec5SDimitry Andric using Iterator = FileChecksumArray::Iterator; 540b57cec5SDimitry Andric 550b57cec5SDimitry Andric public: 560b57cec5SDimitry Andric DebugChecksumsSubsectionRef() 570b57cec5SDimitry Andric : DebugSubsectionRef(DebugSubsectionKind::FileChecksums) {} 580b57cec5SDimitry Andric 590b57cec5SDimitry Andric static bool classof(const DebugSubsectionRef *S) { 600b57cec5SDimitry Andric return S->kind() == DebugSubsectionKind::FileChecksums; 610b57cec5SDimitry Andric } 620b57cec5SDimitry Andric 630b57cec5SDimitry Andric bool valid() const { return Checksums.valid(); } 640b57cec5SDimitry Andric 650b57cec5SDimitry Andric Error initialize(BinaryStreamReader Reader); 660b57cec5SDimitry Andric Error initialize(BinaryStreamRef Stream); 670b57cec5SDimitry Andric 680b57cec5SDimitry Andric Iterator begin() const { return Checksums.begin(); } 690b57cec5SDimitry Andric Iterator end() const { return Checksums.end(); } 700b57cec5SDimitry Andric 710b57cec5SDimitry Andric const FileChecksumArray &getArray() const { return Checksums; } 720b57cec5SDimitry Andric 730b57cec5SDimitry Andric private: 740b57cec5SDimitry Andric FileChecksumArray Checksums; 750b57cec5SDimitry Andric }; 760b57cec5SDimitry Andric 770b57cec5SDimitry Andric class DebugChecksumsSubsection final : public DebugSubsection { 780b57cec5SDimitry Andric public: 790b57cec5SDimitry Andric explicit DebugChecksumsSubsection(DebugStringTableSubsection &Strings); 800b57cec5SDimitry Andric 810b57cec5SDimitry Andric static bool classof(const DebugSubsection *S) { 820b57cec5SDimitry Andric return S->kind() == DebugSubsectionKind::FileChecksums; 830b57cec5SDimitry Andric } 840b57cec5SDimitry Andric 850b57cec5SDimitry Andric void addChecksum(StringRef FileName, FileChecksumKind Kind, 860b57cec5SDimitry Andric ArrayRef<uint8_t> Bytes); 870b57cec5SDimitry Andric 880b57cec5SDimitry Andric uint32_t calculateSerializedSize() const override; 890b57cec5SDimitry Andric Error commit(BinaryStreamWriter &Writer) const override; 900b57cec5SDimitry Andric uint32_t mapChecksumOffset(StringRef FileName) const; 910b57cec5SDimitry Andric 920b57cec5SDimitry Andric private: 930b57cec5SDimitry Andric DebugStringTableSubsection &Strings; 940b57cec5SDimitry Andric 950b57cec5SDimitry Andric DenseMap<uint32_t, uint32_t> OffsetMap; 960b57cec5SDimitry Andric uint32_t SerializedSize = 0; 970b57cec5SDimitry Andric BumpPtrAllocator Storage; 980b57cec5SDimitry Andric std::vector<FileChecksumEntry> Checksums; 990b57cec5SDimitry Andric }; 1000b57cec5SDimitry Andric 1010b57cec5SDimitry Andric } // end namespace codeview 1020b57cec5SDimitry Andric 1030b57cec5SDimitry Andric } // end namespace llvm 1040b57cec5SDimitry Andric 1050b57cec5SDimitry Andric #endif // LLVM_DEBUGINFO_CODEVIEW_DEBUGCHECKSUMSSUBSECTION_H 106