1 //===- CodeViewYAMLTypeHashing.cpp - CodeView YAMLIO type hashing ---------===//
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 classes for handling the YAML representation of CodeView
10 // Debug Info.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/ObjectYAML/CodeViewYAMLTypeHashing.h"
15
16 #include "llvm/Support/BinaryStreamReader.h"
17 #include "llvm/Support/BinaryStreamWriter.h"
18
19 using namespace llvm;
20 using namespace llvm::codeview;
21 using namespace llvm::CodeViewYAML;
22 using namespace llvm::yaml;
23
24 namespace llvm {
25 namespace yaml {
26
mapping(IO & io,DebugHSection & DebugH)27 void MappingTraits<DebugHSection>::mapping(IO &io, DebugHSection &DebugH) {
28 io.mapRequired("Version", DebugH.Version);
29 io.mapRequired("HashAlgorithm", DebugH.HashAlgorithm);
30 io.mapOptional("HashValues", DebugH.Hashes);
31 }
32
output(const GlobalHash & GH,void * Ctx,raw_ostream & OS)33 void ScalarTraits<GlobalHash>::output(const GlobalHash &GH, void *Ctx,
34 raw_ostream &OS) {
35 ScalarTraits<BinaryRef>::output(GH.Hash, Ctx, OS);
36 }
37
input(StringRef Scalar,void * Ctx,GlobalHash & GH)38 StringRef ScalarTraits<GlobalHash>::input(StringRef Scalar, void *Ctx,
39 GlobalHash &GH) {
40 return ScalarTraits<BinaryRef>::input(Scalar, Ctx, GH.Hash);
41 }
42
43 } // end namespace yaml
44 } // end namespace llvm
45
fromDebugH(ArrayRef<uint8_t> DebugH)46 DebugHSection llvm::CodeViewYAML::fromDebugH(ArrayRef<uint8_t> DebugH) {
47 assert(DebugH.size() >= 8);
48 assert((DebugH.size() - 8) % 8 == 0);
49
50 BinaryStreamReader Reader(DebugH, llvm::endianness::little);
51 DebugHSection DHS;
52 cantFail(Reader.readInteger(DHS.Magic));
53 cantFail(Reader.readInteger(DHS.Version));
54 cantFail(Reader.readInteger(DHS.HashAlgorithm));
55
56 while (Reader.bytesRemaining() != 0) {
57 ArrayRef<uint8_t> S;
58 cantFail(Reader.readBytes(S, 8));
59 DHS.Hashes.emplace_back(S);
60 }
61 assert(Reader.bytesRemaining() == 0);
62 return DHS;
63 }
64
toDebugH(const DebugHSection & DebugH,BumpPtrAllocator & Alloc)65 ArrayRef<uint8_t> llvm::CodeViewYAML::toDebugH(const DebugHSection &DebugH,
66 BumpPtrAllocator &Alloc) {
67 uint32_t Size = 8 + 8 * DebugH.Hashes.size();
68 uint8_t *Data = Alloc.Allocate<uint8_t>(Size);
69 MutableArrayRef<uint8_t> Buffer(Data, Size);
70 BinaryStreamWriter Writer(Buffer, llvm::endianness::little);
71
72 cantFail(Writer.writeInteger(DebugH.Magic));
73 cantFail(Writer.writeInteger(DebugH.Version));
74 cantFail(Writer.writeInteger(DebugH.HashAlgorithm));
75 SmallString<8> Hash;
76 for (const auto &H : DebugH.Hashes) {
77 Hash.clear();
78 raw_svector_ostream OS(Hash);
79 H.Hash.writeAsBinary(OS);
80 assert((Hash.size() == 8) && "Invalid hash size!");
81 cantFail(Writer.writeFixedString(Hash));
82 }
83 assert(Writer.bytesRemaining() == 0);
84 return Buffer;
85 }
86