86 lines · cpp
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-exception6//7//===----------------------------------------------------------------------===//8//9// This file defines classes for handling the YAML representation of CodeView10// 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 19using namespace llvm;20using namespace llvm::codeview;21using namespace llvm::CodeViewYAML;22using namespace llvm::yaml;23 24namespace llvm {25namespace yaml {26 27void 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 33void ScalarTraits<GlobalHash>::output(const GlobalHash &GH, void *Ctx,34 raw_ostream &OS) {35 ScalarTraits<BinaryRef>::output(GH.Hash, Ctx, OS);36}37 38StringRef ScalarTraits<GlobalHash>::input(StringRef Scalar, void *Ctx,39 GlobalHash &GH) {40 return ScalarTraits<BinaryRef>::input(Scalar, Ctx, GH.Hash);41}42 43} // end namespace yaml44} // end namespace llvm45 46DebugHSection 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 65ArrayRef<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