Fix use after free in BinaryStream library.
[lldb.git] / llvm / include / llvm / DebugInfo / CodeView / ModuleDebugFileChecksumFragment.h
1 //===- ModuleDebugFileChecksumFragment.h ------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #ifndef LLVM_DEBUGINFO_CODEVIEW_MODULEDEBUGFILECHECKSUMFRAGMENT_H
11 #define LLVM_DEBUGINFO_CODEVIEW_MODULEDEBUGFILECHECKSUMFRAGMENT_H
12
13 #include "llvm/ADT/ArrayRef.h"
14 #include "llvm/ADT/DenseMap.h"
15 #include "llvm/DebugInfo/CodeView/ModuleDebugFragment.h"
16 #include "llvm/Support/Allocator.h"
17 #include "llvm/Support/BinaryStreamArray.h"
18 #include "llvm/Support/BinaryStreamReader.h"
19 #include "llvm/Support/Endian.h"
20
21 namespace llvm {
22 namespace codeview {
23
24 struct FileChecksumEntry {
25   uint32_t FileNameOffset;    // Byte offset of filename in global stringtable.
26   FileChecksumKind Kind;      // The type of checksum.
27   ArrayRef<uint8_t> Checksum; // The bytes of the checksum.
28 };
29 }
30 }
31
32 namespace llvm {
33 template <> struct VarStreamArrayExtractor<codeview::FileChecksumEntry> {
34 public:
35   typedef void ContextType;
36
37   static Error extract(BinaryStreamRef Stream, uint32_t &Len,
38                        codeview::FileChecksumEntry &Item);
39 };
40 }
41
42 namespace llvm {
43 namespace codeview {
44 class ModuleDebugFileChecksumFragmentRef final : public ModuleDebugFragmentRef {
45   typedef VarStreamArray<codeview::FileChecksumEntry> FileChecksumArray;
46   typedef FileChecksumArray::Iterator Iterator;
47
48 public:
49   ModuleDebugFileChecksumFragmentRef()
50       : ModuleDebugFragmentRef(ModuleDebugFragmentKind::FileChecksums) {}
51
52   static bool classof(const ModuleDebugFragmentRef *S) {
53     return S->kind() == ModuleDebugFragmentKind::FileChecksums;
54   }
55
56   Error initialize(BinaryStreamReader Reader);
57
58   Iterator begin() { return Checksums.begin(); }
59   Iterator end() { return Checksums.end(); }
60
61   const FileChecksumArray &getArray() const { return Checksums; }
62
63 private:
64   FileChecksumArray Checksums;
65 };
66
67 class ModuleDebugFileChecksumFragment final : public ModuleDebugFragment {
68 public:
69   ModuleDebugFileChecksumFragment();
70
71   static bool classof(const ModuleDebugFragment *S) {
72     return S->kind() == ModuleDebugFragmentKind::FileChecksums;
73   }
74
75   void addChecksum(uint32_t StringTableOffset, FileChecksumKind Kind,
76                    ArrayRef<uint8_t> Bytes);
77
78   uint32_t calculateSerializedLength() override;
79   Error commit(BinaryStreamWriter &Writer) override;
80   uint32_t mapChecksumOffset(uint32_t StringTableOffset) const;
81
82 private:
83   DenseMap<uint32_t, uint32_t> OffsetMap;
84   uint32_t SerializedSize = 0;
85   llvm::BumpPtrAllocator Storage;
86   std::vector<FileChecksumEntry> Checksums;
87 };
88 }
89 }
90
91 #endif