mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-08 13:56:06 +00:00
cf468d86f3
The raw CodeView format references strings by "offsets", but it's confusing what table the offset refers to. In the case of line number information, it's an offset into a buffer of records, and an indirection is required to get another offset into a different table to find the final string. And in the case of checksum information, there is no indirection, and the offset refers directly to the location of the string in another buffer. This would be less confusing if we always just referred to the strings by their value, and have the library be smart enough to correctly resolve the offsets on its own from the right location. This patch makes that possible. When either reading or writing, all the user deals with are strings, and the library does the appropriate translations behind the scenes. llvm-svn: 302053
72 lines
2.0 KiB
C++
72 lines
2.0 KiB
C++
//===- StringTable.cpp - CodeView String Table Reader/Writer ----*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/DebugInfo/CodeView/StringTable.h"
|
|
|
|
#include "llvm/Support/BinaryStream.h"
|
|
#include "llvm/Support/BinaryStreamReader.h"
|
|
#include "llvm/Support/BinaryStreamWriter.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::codeview;
|
|
|
|
StringTableRef::StringTableRef() {}
|
|
|
|
Error StringTableRef::initialize(BinaryStreamRef Contents) {
|
|
Stream = Contents;
|
|
return Error::success();
|
|
}
|
|
|
|
Expected<StringRef> StringTableRef::getString(uint32_t Offset) const {
|
|
BinaryStreamReader Reader(Stream);
|
|
Reader.setOffset(Offset);
|
|
StringRef Result;
|
|
if (auto EC = Reader.readCString(Result))
|
|
return std::move(EC);
|
|
return Result;
|
|
}
|
|
|
|
uint32_t StringTable::insert(StringRef S) {
|
|
auto P = Strings.insert({S, StringSize});
|
|
|
|
// If a given string didn't exist in the string table, we want to increment
|
|
// the string table size.
|
|
if (P.second)
|
|
StringSize += S.size() + 1; // +1 for '\0'
|
|
return P.first->second;
|
|
}
|
|
|
|
uint32_t StringTable::calculateSerializedSize() const { return StringSize; }
|
|
|
|
Error StringTable::commit(BinaryStreamWriter &Writer) const {
|
|
assert(Writer.bytesRemaining() == StringSize);
|
|
uint32_t MaxOffset = 1;
|
|
|
|
for (auto &Pair : Strings) {
|
|
StringRef S = Pair.getKey();
|
|
uint32_t Offset = Pair.getValue();
|
|
Writer.setOffset(Offset);
|
|
if (auto EC = Writer.writeCString(S))
|
|
return EC;
|
|
MaxOffset = std::max<uint32_t>(MaxOffset, Offset + S.size() + 1);
|
|
}
|
|
|
|
Writer.setOffset(MaxOffset);
|
|
assert(Writer.bytesRemaining() == 0);
|
|
return Error::success();
|
|
}
|
|
|
|
uint32_t StringTable::size() const { return Strings.size(); }
|
|
|
|
uint32_t StringTable::getStringId(StringRef S) const {
|
|
auto P = Strings.find(S);
|
|
assert(P != Strings.end());
|
|
return P->second;
|
|
}
|