1 //===- DebugChecksumsSubsection.cpp ---------------------------------------===//
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 #include "llvm/DebugInfo/CodeView/DebugChecksumsSubsection.h"
10 #include "llvm/ADT/ArrayRef.h"
11 #include "llvm/DebugInfo/CodeView/CodeView.h"
12 #include "llvm/DebugInfo/CodeView/DebugStringTableSubsection.h"
13 #include "llvm/Support/BinaryStreamReader.h"
14 #include "llvm/Support/BinaryStreamWriter.h"
15 #include "llvm/Support/Endian.h"
16 #include "llvm/Support/Error.h"
17 #include "llvm/Support/MathExtras.h"
18 #include <cassert>
19 #include <cstdint>
20 #include <cstring>
21 
22 using namespace llvm;
23 using namespace llvm::codeview;
24 
25 struct FileChecksumEntryHeader {
26   using ulittle32_t = support::ulittle32_t;
27 
28   ulittle32_t FileNameOffset; // Byte offset of filename in global string table.
29   uint8_t ChecksumSize;       // Number of bytes of checksum.
30   uint8_t ChecksumKind;       // FileChecksumKind
31                               // Checksum bytes follow.
32 };
33 
34 Error VarStreamArrayExtractor<FileChecksumEntry>::
operator ()(BinaryStreamRef Stream,uint32_t & Len,FileChecksumEntry & Item)35 operator()(BinaryStreamRef Stream, uint32_t &Len, FileChecksumEntry &Item) {
36   BinaryStreamReader Reader(Stream);
37 
38   const FileChecksumEntryHeader *Header;
39   if (auto EC = Reader.readObject(Header))
40     return EC;
41 
42   Item.FileNameOffset = Header->FileNameOffset;
43   Item.Kind = static_cast<FileChecksumKind>(Header->ChecksumKind);
44   if (auto EC = Reader.readBytes(Item.Checksum, Header->ChecksumSize))
45     return EC;
46 
47   Len = alignTo(Header->ChecksumSize + sizeof(FileChecksumEntryHeader), 4);
48   return Error::success();
49 }
50 
initialize(BinaryStreamReader Reader)51 Error DebugChecksumsSubsectionRef::initialize(BinaryStreamReader Reader) {
52   if (auto EC = Reader.readArray(Checksums, Reader.bytesRemaining()))
53     return EC;
54 
55   return Error::success();
56 }
57 
initialize(BinaryStreamRef Section)58 Error DebugChecksumsSubsectionRef::initialize(BinaryStreamRef Section) {
59   BinaryStreamReader Reader(Section);
60   return initialize(Reader);
61 }
62 
DebugChecksumsSubsection(DebugStringTableSubsection & Strings)63 DebugChecksumsSubsection::DebugChecksumsSubsection(
64     DebugStringTableSubsection &Strings)
65     : DebugSubsection(DebugSubsectionKind::FileChecksums), Strings(Strings) {}
66 
addChecksum(StringRef FileName,FileChecksumKind Kind,ArrayRef<uint8_t> Bytes)67 void DebugChecksumsSubsection::addChecksum(StringRef FileName,
68                                            FileChecksumKind Kind,
69                                            ArrayRef<uint8_t> Bytes) {
70   FileChecksumEntry Entry;
71   if (!Bytes.empty()) {
72     uint8_t *Copy = Storage.Allocate<uint8_t>(Bytes.size());
73     ::memcpy(Copy, Bytes.data(), Bytes.size());
74     Entry.Checksum = makeArrayRef(Copy, Bytes.size());
75   }
76 
77   Entry.FileNameOffset = Strings.insert(FileName);
78   Entry.Kind = Kind;
79   Checksums.push_back(Entry);
80 
81   // This maps the offset of this string in the string table to the offset
82   // of this checksum entry in the checksum buffer.
83   OffsetMap[Entry.FileNameOffset] = SerializedSize;
84   assert(SerializedSize % 4 == 0);
85 
86   uint32_t Len = alignTo(sizeof(FileChecksumEntryHeader) + Bytes.size(), 4);
87   SerializedSize += Len;
88 }
89 
calculateSerializedSize() const90 uint32_t DebugChecksumsSubsection::calculateSerializedSize() const {
91   return SerializedSize;
92 }
93 
commit(BinaryStreamWriter & Writer) const94 Error DebugChecksumsSubsection::commit(BinaryStreamWriter &Writer) const {
95   for (const auto &FC : Checksums) {
96     FileChecksumEntryHeader Header;
97     Header.ChecksumKind = uint8_t(FC.Kind);
98     Header.ChecksumSize = FC.Checksum.size();
99     Header.FileNameOffset = FC.FileNameOffset;
100     if (auto EC = Writer.writeObject(Header))
101       return EC;
102     if (auto EC = Writer.writeArray(makeArrayRef(FC.Checksum)))
103       return EC;
104     if (auto EC = Writer.padToAlignment(4))
105       return EC;
106   }
107   return Error::success();
108 }
109 
mapChecksumOffset(StringRef FileName) const110 uint32_t DebugChecksumsSubsection::mapChecksumOffset(StringRef FileName) const {
111   uint32_t Offset = Strings.getIdForString(FileName);
112   auto Iter = OffsetMap.find(Offset);
113   assert(Iter != OffsetMap.end());
114   return Iter->second;
115 }
116