301 lines
		
	
	
		
			9.1 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			301 lines
		
	
	
		
			9.1 KiB
		
	
	
	
		
			C++
		
	
	
	
//===- DWARFUnitIndex.cpp -------------------------------------------------===//
 | 
						|
//
 | 
						|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | 
						|
// See https://llvm.org/LICENSE.txt for license information.
 | 
						|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
 | 
						|
#include "llvm/DebugInfo/DWARF/DWARFUnitIndex.h"
 | 
						|
#include "llvm/ADT/STLExtras.h"
 | 
						|
#include "llvm/ADT/StringRef.h"
 | 
						|
#include "llvm/Support/ErrorHandling.h"
 | 
						|
#include "llvm/Support/Format.h"
 | 
						|
#include "llvm/Support/raw_ostream.h"
 | 
						|
#include <cinttypes>
 | 
						|
#include <cstdint>
 | 
						|
 | 
						|
using namespace llvm;
 | 
						|
 | 
						|
namespace {
 | 
						|
 | 
						|
enum class DWARFSectionKindV2 {
 | 
						|
  DW_SECT_INFO = 1,
 | 
						|
  DW_SECT_TYPES = 2,
 | 
						|
  DW_SECT_ABBREV = 3,
 | 
						|
  DW_SECT_LINE = 4,
 | 
						|
  DW_SECT_LOC = 5,
 | 
						|
  DW_SECT_STR_OFFSETS = 6,
 | 
						|
  DW_SECT_MACINFO = 7,
 | 
						|
  DW_SECT_MACRO = 8,
 | 
						|
};
 | 
						|
 | 
						|
} // namespace
 | 
						|
 | 
						|
// Return true if the section identifier is defined in the DWARFv5 standard.
 | 
						|
constexpr bool isKnownV5SectionID(uint32_t ID) {
 | 
						|
  return ID >= DW_SECT_INFO && ID <= DW_SECT_RNGLISTS &&
 | 
						|
         ID != DW_SECT_EXT_TYPES;
 | 
						|
}
 | 
						|
 | 
						|
uint32_t llvm::serializeSectionKind(DWARFSectionKind Kind,
 | 
						|
                                    unsigned IndexVersion) {
 | 
						|
  if (IndexVersion == 5) {
 | 
						|
    assert(isKnownV5SectionID(Kind));
 | 
						|
    return static_cast<uint32_t>(Kind);
 | 
						|
  }
 | 
						|
  assert(IndexVersion == 2);
 | 
						|
  switch (Kind) {
 | 
						|
#define CASE(S,T) \
 | 
						|
  case DW_SECT_##S: \
 | 
						|
    return static_cast<uint32_t>(DWARFSectionKindV2::DW_SECT_##T)
 | 
						|
  CASE(INFO, INFO);
 | 
						|
  CASE(EXT_TYPES, TYPES);
 | 
						|
  CASE(ABBREV, ABBREV);
 | 
						|
  CASE(LINE, LINE);
 | 
						|
  CASE(EXT_LOC, LOC);
 | 
						|
  CASE(STR_OFFSETS, STR_OFFSETS);
 | 
						|
  CASE(EXT_MACINFO, MACINFO);
 | 
						|
  CASE(MACRO, MACRO);
 | 
						|
#undef CASE
 | 
						|
  default:
 | 
						|
    // All other section kinds have no corresponding values in v2 indexes.
 | 
						|
    llvm_unreachable("Invalid DWARFSectionKind");
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
DWARFSectionKind llvm::deserializeSectionKind(uint32_t Value,
 | 
						|
                                              unsigned IndexVersion) {
 | 
						|
  if (IndexVersion == 5)
 | 
						|
    return isKnownV5SectionID(Value)
 | 
						|
               ? static_cast<DWARFSectionKind>(Value)
 | 
						|
               : DW_SECT_EXT_unknown;
 | 
						|
  assert(IndexVersion == 2);
 | 
						|
  switch (static_cast<DWARFSectionKindV2>(Value)) {
 | 
						|
#define CASE(S,T) \
 | 
						|
  case DWARFSectionKindV2::DW_SECT_##S: \
 | 
						|
    return DW_SECT_##T
 | 
						|
  CASE(INFO, INFO);
 | 
						|
  CASE(TYPES, EXT_TYPES);
 | 
						|
  CASE(ABBREV, ABBREV);
 | 
						|
  CASE(LINE, LINE);
 | 
						|
  CASE(LOC, EXT_LOC);
 | 
						|
  CASE(STR_OFFSETS, STR_OFFSETS);
 | 
						|
  CASE(MACINFO, EXT_MACINFO);
 | 
						|
  CASE(MACRO, MACRO);
 | 
						|
#undef CASE
 | 
						|
  }
 | 
						|
  return DW_SECT_EXT_unknown;
 | 
						|
}
 | 
						|
 | 
						|
bool DWARFUnitIndex::Header::parse(DataExtractor IndexData,
 | 
						|
                                   uint64_t *OffsetPtr) {
 | 
						|
  const uint64_t BeginOffset = *OffsetPtr;
 | 
						|
  if (!IndexData.isValidOffsetForDataOfSize(*OffsetPtr, 16))
 | 
						|
    return false;
 | 
						|
  // GCC Debug Fission defines the version as an unsigned 32-bit field
 | 
						|
  // with value of 2, https://gcc.gnu.org/wiki/DebugFissionDWP.
 | 
						|
  // DWARFv5 defines the same space as an uhalf version field with value of 5
 | 
						|
  // and a 2 bytes long padding, see Section 7.3.5.3.
 | 
						|
  Version = IndexData.getU32(OffsetPtr);
 | 
						|
  if (Version != 2) {
 | 
						|
    *OffsetPtr = BeginOffset;
 | 
						|
    Version = IndexData.getU16(OffsetPtr);
 | 
						|
    if (Version != 5)
 | 
						|
      return false;
 | 
						|
    *OffsetPtr += 2; // Skip padding.
 | 
						|
  }
 | 
						|
  NumColumns = IndexData.getU32(OffsetPtr);
 | 
						|
  NumUnits = IndexData.getU32(OffsetPtr);
 | 
						|
  NumBuckets = IndexData.getU32(OffsetPtr);
 | 
						|
  return true;
 | 
						|
}
 | 
						|
 | 
						|
void DWARFUnitIndex::Header::dump(raw_ostream &OS) const {
 | 
						|
  OS << format("version = %u, units = %u, slots = %u\n\n", Version, NumUnits, NumBuckets);
 | 
						|
}
 | 
						|
 | 
						|
bool DWARFUnitIndex::parse(DataExtractor IndexData) {
 | 
						|
  bool b = parseImpl(IndexData);
 | 
						|
  if (!b) {
 | 
						|
    // Make sure we don't try to dump anything
 | 
						|
    Header.NumBuckets = 0;
 | 
						|
    // Release any partially initialized data.
 | 
						|
    ColumnKinds.reset();
 | 
						|
    Rows.reset();
 | 
						|
  }
 | 
						|
  return b;
 | 
						|
}
 | 
						|
 | 
						|
bool DWARFUnitIndex::parseImpl(DataExtractor IndexData) {
 | 
						|
  uint64_t Offset = 0;
 | 
						|
  if (!Header.parse(IndexData, &Offset))
 | 
						|
    return false;
 | 
						|
 | 
						|
  // Fix InfoColumnKind: in DWARFv5, type units are in .debug_info.dwo.
 | 
						|
  if (Header.Version == 5)
 | 
						|
    InfoColumnKind = DW_SECT_INFO;
 | 
						|
 | 
						|
  if (!IndexData.isValidOffsetForDataOfSize(
 | 
						|
          Offset, Header.NumBuckets * (8 + 4) +
 | 
						|
                      (2 * Header.NumUnits + 1) * 4 * Header.NumColumns))
 | 
						|
    return false;
 | 
						|
 | 
						|
  Rows = std::make_unique<Entry[]>(Header.NumBuckets);
 | 
						|
  auto Contribs =
 | 
						|
      std::make_unique<Entry::SectionContribution *[]>(Header.NumUnits);
 | 
						|
  ColumnKinds = std::make_unique<DWARFSectionKind[]>(Header.NumColumns);
 | 
						|
  RawSectionIds = std::make_unique<uint32_t[]>(Header.NumColumns);
 | 
						|
 | 
						|
  // Read Hash Table of Signatures
 | 
						|
  for (unsigned i = 0; i != Header.NumBuckets; ++i)
 | 
						|
    Rows[i].Signature = IndexData.getU64(&Offset);
 | 
						|
 | 
						|
  // Read Parallel Table of Indexes
 | 
						|
  for (unsigned i = 0; i != Header.NumBuckets; ++i) {
 | 
						|
    auto Index = IndexData.getU32(&Offset);
 | 
						|
    if (!Index)
 | 
						|
      continue;
 | 
						|
    Rows[i].Index = this;
 | 
						|
    Rows[i].Contributions =
 | 
						|
        std::make_unique<Entry::SectionContribution[]>(Header.NumColumns);
 | 
						|
    Contribs[Index - 1] = Rows[i].Contributions.get();
 | 
						|
  }
 | 
						|
 | 
						|
  // Read the Column Headers
 | 
						|
  for (unsigned i = 0; i != Header.NumColumns; ++i) {
 | 
						|
    RawSectionIds[i] = IndexData.getU32(&Offset);
 | 
						|
    ColumnKinds[i] = deserializeSectionKind(RawSectionIds[i], Header.Version);
 | 
						|
    if (ColumnKinds[i] == InfoColumnKind) {
 | 
						|
      if (InfoColumn != -1)
 | 
						|
        return false;
 | 
						|
      InfoColumn = i;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  if (InfoColumn == -1)
 | 
						|
    return false;
 | 
						|
 | 
						|
  // Read Table of Section Offsets
 | 
						|
  for (unsigned i = 0; i != Header.NumUnits; ++i) {
 | 
						|
    auto *Contrib = Contribs[i];
 | 
						|
    for (unsigned i = 0; i != Header.NumColumns; ++i)
 | 
						|
      Contrib[i].Offset = IndexData.getU32(&Offset);
 | 
						|
  }
 | 
						|
 | 
						|
  // Read Table of Section Sizes
 | 
						|
  for (unsigned i = 0; i != Header.NumUnits; ++i) {
 | 
						|
    auto *Contrib = Contribs[i];
 | 
						|
    for (unsigned i = 0; i != Header.NumColumns; ++i)
 | 
						|
      Contrib[i].Length = IndexData.getU32(&Offset);
 | 
						|
  }
 | 
						|
 | 
						|
  return true;
 | 
						|
}
 | 
						|
 | 
						|
StringRef DWARFUnitIndex::getColumnHeader(DWARFSectionKind DS) {
 | 
						|
  switch (DS) {
 | 
						|
#define HANDLE_DW_SECT(ID, NAME)                                               \
 | 
						|
  case DW_SECT_##NAME:                                                         \
 | 
						|
    return #NAME;
 | 
						|
#include "llvm/BinaryFormat/Dwarf.def"
 | 
						|
  case DW_SECT_EXT_TYPES:
 | 
						|
    return "TYPES";
 | 
						|
  case DW_SECT_EXT_LOC:
 | 
						|
    return "LOC";
 | 
						|
  case DW_SECT_EXT_MACINFO:
 | 
						|
    return "MACINFO";
 | 
						|
  case DW_SECT_EXT_unknown:
 | 
						|
    return StringRef();
 | 
						|
  }
 | 
						|
  llvm_unreachable("Unknown DWARFSectionKind");
 | 
						|
}
 | 
						|
 | 
						|
void DWARFUnitIndex::dump(raw_ostream &OS) const {
 | 
						|
  if (!*this)
 | 
						|
    return;
 | 
						|
 | 
						|
  Header.dump(OS);
 | 
						|
  OS << "Index Signature         ";
 | 
						|
  for (unsigned i = 0; i != Header.NumColumns; ++i) {
 | 
						|
    DWARFSectionKind Kind = ColumnKinds[i];
 | 
						|
    StringRef Name = getColumnHeader(Kind);
 | 
						|
    if (!Name.empty())
 | 
						|
      OS << ' ' << left_justify(Name, 24);
 | 
						|
    else
 | 
						|
      OS << format(" Unknown: %-15" PRIu32, RawSectionIds[i]);
 | 
						|
  }
 | 
						|
  OS << "\n----- ------------------";
 | 
						|
  for (unsigned i = 0; i != Header.NumColumns; ++i)
 | 
						|
    OS << " ------------------------";
 | 
						|
  OS << '\n';
 | 
						|
  for (unsigned i = 0; i != Header.NumBuckets; ++i) {
 | 
						|
    auto &Row = Rows[i];
 | 
						|
    if (auto *Contribs = Row.Contributions.get()) {
 | 
						|
      OS << format("%5u 0x%016" PRIx64 " ", i + 1, Row.Signature);
 | 
						|
      for (unsigned i = 0; i != Header.NumColumns; ++i) {
 | 
						|
        auto &Contrib = Contribs[i];
 | 
						|
        OS << format("[0x%08x, 0x%08x) ", Contrib.Offset,
 | 
						|
                     Contrib.Offset + Contrib.Length);
 | 
						|
      }
 | 
						|
      OS << '\n';
 | 
						|
    }
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
const DWARFUnitIndex::Entry::SectionContribution *
 | 
						|
DWARFUnitIndex::Entry::getContribution(DWARFSectionKind Sec) const {
 | 
						|
  uint32_t i = 0;
 | 
						|
  for (; i != Index->Header.NumColumns; ++i)
 | 
						|
    if (Index->ColumnKinds[i] == Sec)
 | 
						|
      return &Contributions[i];
 | 
						|
  return nullptr;
 | 
						|
}
 | 
						|
 | 
						|
const DWARFUnitIndex::Entry::SectionContribution *
 | 
						|
DWARFUnitIndex::Entry::getContribution() const {
 | 
						|
  return &Contributions[Index->InfoColumn];
 | 
						|
}
 | 
						|
 | 
						|
const DWARFUnitIndex::Entry *
 | 
						|
DWARFUnitIndex::getFromOffset(uint32_t Offset) const {
 | 
						|
  if (OffsetLookup.empty()) {
 | 
						|
    for (uint32_t i = 0; i != Header.NumBuckets; ++i)
 | 
						|
      if (Rows[i].Contributions)
 | 
						|
        OffsetLookup.push_back(&Rows[i]);
 | 
						|
    llvm::sort(OffsetLookup, [&](Entry *E1, Entry *E2) {
 | 
						|
      return E1->Contributions[InfoColumn].Offset <
 | 
						|
             E2->Contributions[InfoColumn].Offset;
 | 
						|
    });
 | 
						|
  }
 | 
						|
  auto I = partition_point(OffsetLookup, [&](Entry *E2) {
 | 
						|
    return E2->Contributions[InfoColumn].Offset <= Offset;
 | 
						|
  });
 | 
						|
  if (I == OffsetLookup.begin())
 | 
						|
    return nullptr;
 | 
						|
  --I;
 | 
						|
  const auto *E = *I;
 | 
						|
  const auto &InfoContrib = E->Contributions[InfoColumn];
 | 
						|
  if ((InfoContrib.Offset + InfoContrib.Length) <= Offset)
 | 
						|
    return nullptr;
 | 
						|
  return E;
 | 
						|
}
 | 
						|
 | 
						|
const DWARFUnitIndex::Entry *DWARFUnitIndex::getFromHash(uint64_t S) const {
 | 
						|
  uint64_t Mask = Header.NumBuckets - 1;
 | 
						|
 | 
						|
  auto H = S & Mask;
 | 
						|
  auto HP = ((S >> 32) & Mask) | 1;
 | 
						|
  // The spec says "while 0 is a valid hash value, the row index in a used slot
 | 
						|
  // will always be non-zero". Loop until we find a match or an empty slot.
 | 
						|
  while (Rows[H].getSignature() != S && Rows[H].Index != nullptr)
 | 
						|
    H = (H + HP) & Mask;
 | 
						|
 | 
						|
  // If the slot is empty, we don't care whether the signature matches (it could
 | 
						|
  // be zero and still match the zeros in the empty slot).
 | 
						|
  if (Rows[H].Index == nullptr)
 | 
						|
    return nullptr;
 | 
						|
 | 
						|
  return &Rows[H];
 | 
						|
}
 |