119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman//===-- DWARFDebugAbbrev.cpp ----------------------------------------------===//
219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman//
319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman//                     The LLVM Compiler Infrastructure
419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman//
519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman// This file is distributed under the University of Illinois Open Source
619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman// License. See LICENSE.TXT for details.
719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman//
819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman//===----------------------------------------------------------------------===//
919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
1019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman#include "DWARFDebugAbbrev.h"
1119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman#include "llvm/Support/Format.h"
1219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman#include "llvm/Support/raw_ostream.h"
1319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Baumanusing namespace llvm;
1419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
1519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Baumanbool DWARFAbbreviationDeclarationSet::extract(DataExtractor data,
1619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman                                              uint32_t* offset_ptr) {
1719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  const uint32_t beginOffset = *offset_ptr;
1819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  Offset = beginOffset;
1919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  clear();
2019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  DWARFAbbreviationDeclaration abbrevDeclaration;
2119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  uint32_t prevAbbrAode = 0;
2219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  while (abbrevDeclaration.extract(data, offset_ptr)) {
2319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    Decls.push_back(abbrevDeclaration);
2419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    if (IdxOffset == 0) {
2519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman      IdxOffset = abbrevDeclaration.getCode();
2619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    } else {
2719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman      if (prevAbbrAode + 1 != abbrevDeclaration.getCode())
2819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman        IdxOffset = UINT32_MAX;// Out of order indexes, we can't do O(1) lookups
2919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    }
3019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    prevAbbrAode = abbrevDeclaration.getCode();
3119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  }
3219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  return beginOffset != *offset_ptr;
3319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman}
3419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
3519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Baumanvoid DWARFAbbreviationDeclarationSet::dump(raw_ostream &OS) const {
3619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  for (unsigned i = 0, e = Decls.size(); i != e; ++i)
3719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    Decls[i].dump(OS);
3819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman}
3919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
4019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Baumanconst DWARFAbbreviationDeclaration*
4119bac1e08be200c31efd26f0f5fd144c9b3eefd3John BaumanDWARFAbbreviationDeclarationSet::getAbbreviationDeclaration(uint32_t abbrCode)
4219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  const {
4319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  if (IdxOffset == UINT32_MAX) {
4419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    DWARFAbbreviationDeclarationCollConstIter pos;
4519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    DWARFAbbreviationDeclarationCollConstIter end = Decls.end();
4619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    for (pos = Decls.begin(); pos != end; ++pos) {
4719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman      if (pos->getCode() == abbrCode)
4819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman        return &(*pos);
4919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    }
5019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  } else {
5119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    uint32_t idx = abbrCode - IdxOffset;
5219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    if (idx < Decls.size())
5319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman      return &Decls[idx];
5419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  }
5519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  return NULL;
5619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman}
5719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
5819bac1e08be200c31efd26f0f5fd144c9b3eefd3John BaumanDWARFDebugAbbrev::DWARFDebugAbbrev() :
5919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  AbbrevCollMap(),
6019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  PrevAbbrOffsetPos(AbbrevCollMap.end()) {}
6119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
6219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
6319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Baumanvoid DWARFDebugAbbrev::parse(DataExtractor data) {
6419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  uint32_t offset = 0;
6519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
6619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  while (data.isValidOffset(offset)) {
6719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    uint32_t initial_cu_offset = offset;
6819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    DWARFAbbreviationDeclarationSet abbrevDeclSet;
6919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
7019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    if (abbrevDeclSet.extract(data, &offset))
7119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman      AbbrevCollMap[initial_cu_offset] = abbrevDeclSet;
7219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    else
7319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman      break;
7419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  }
7519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  PrevAbbrOffsetPos = AbbrevCollMap.end();
7619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman}
7719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
7819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Baumanvoid DWARFDebugAbbrev::dump(raw_ostream &OS) const {
7919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  if (AbbrevCollMap.empty()) {
8019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    OS << "< EMPTY >\n";
8119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    return;
8219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  }
8319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
8419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  DWARFAbbreviationDeclarationCollMapConstIter pos;
8519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  for (pos = AbbrevCollMap.begin(); pos != AbbrevCollMap.end(); ++pos) {
8619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    OS << format("Abbrev table for offset: 0x%8.8x\n", pos->first);
8719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    pos->second.dump(OS);
8819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  }
8919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman}
9019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
9119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Baumanconst DWARFAbbreviationDeclarationSet*
9219bac1e08be200c31efd26f0f5fd144c9b3eefd3John BaumanDWARFDebugAbbrev::getAbbreviationDeclarationSet(uint64_t cu_abbr_offset) const {
9319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  DWARFAbbreviationDeclarationCollMapConstIter end = AbbrevCollMap.end();
9419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  DWARFAbbreviationDeclarationCollMapConstIter pos;
9519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  if (PrevAbbrOffsetPos != end &&
9619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman      PrevAbbrOffsetPos->first == cu_abbr_offset) {
9719bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    return &(PrevAbbrOffsetPos->second);
9819bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  } else {
9919bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    pos = AbbrevCollMap.find(cu_abbr_offset);
10019bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    PrevAbbrOffsetPos = pos;
10119bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  }
10219bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman
10319bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  if (pos != AbbrevCollMap.end())
10419bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman    return &(pos->second);
10519bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman  return NULL;
10619bac1e08be200c31efd26f0f5fd144c9b3eefd3John Bauman}
107