Template the MachO types over the word size.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@179051 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Rafael Espindola 2013-04-08 20:45:01 +00:00
parent a9408bafcc
commit 335f1d46d8
3 changed files with 119 additions and 82 deletions

View File

@ -27,7 +27,11 @@ namespace llvm {
namespace object { namespace object {
namespace MachOFormat { namespace MachOFormat {
struct Section { template<bool is64Bits>
struct Section;
template<>
struct Section<false> {
char Name[16]; char Name[16];
char SegmentName[16]; char SegmentName[16];
support::ulittle32_t Address; support::ulittle32_t Address;
@ -41,7 +45,8 @@ namespace MachOFormat {
support::ulittle32_t Reserved2; support::ulittle32_t Reserved2;
}; };
struct Section64 { template<>
struct Section<true> {
char Name[16]; char Name[16];
char SegmentName[16]; char SegmentName[16];
support::ulittle64_t Address; support::ulittle64_t Address;
@ -61,7 +66,11 @@ namespace MachOFormat {
support::ulittle32_t Word1; support::ulittle32_t Word1;
}; };
struct SymbolTableEntry { template<bool is64Bits>
struct SymbolTableEntry;
template<>
struct SymbolTableEntry<false> {
support::ulittle32_t StringIndex; support::ulittle32_t StringIndex;
uint8_t Type; uint8_t Type;
uint8_t SectionIndex; uint8_t SectionIndex;
@ -69,7 +78,8 @@ namespace MachOFormat {
support::ulittle32_t Value; support::ulittle32_t Value;
}; };
struct Symbol64TableEntry { template<>
struct SymbolTableEntry<true> {
support::ulittle32_t StringIndex; support::ulittle32_t StringIndex;
uint8_t Type; uint8_t Type;
uint8_t SectionIndex; uint8_t SectionIndex;
@ -91,7 +101,11 @@ namespace MachOFormat {
support::ulittle32_t StringTableSize; support::ulittle32_t StringTableSize;
}; };
struct SegmentLoadCommand { template<bool is64Bits>
struct SegmentLoadCommand;
template<>
struct SegmentLoadCommand<false> {
support::ulittle32_t Type; support::ulittle32_t Type;
support::ulittle32_t Size; support::ulittle32_t Size;
char Name[16]; char Name[16];
@ -105,7 +119,8 @@ namespace MachOFormat {
support::ulittle32_t Flags; support::ulittle32_t Flags;
}; };
struct Segment64LoadCommand { template<>
struct SegmentLoadCommand<true> {
support::ulittle32_t Type; support::ulittle32_t Type;
support::ulittle32_t Size; support::ulittle32_t Size;
char Name[16]; char Name[16];
@ -165,11 +180,11 @@ public:
ArrayRef<char> getSectionRawName(DataRefImpl Sec) const; ArrayRef<char> getSectionRawName(DataRefImpl Sec) const;
ArrayRef<char>getSectionRawFinalSegmentName(DataRefImpl Sec) const; ArrayRef<char>getSectionRawFinalSegmentName(DataRefImpl Sec) const;
const MachOFormat::Section64 *getSection64(DataRefImpl DRI) const; const MachOFormat::Section<true> *getSection64(DataRefImpl DRI) const;
const MachOFormat::Section *getSection(DataRefImpl DRI) const; const MachOFormat::Section<false> *getSection(DataRefImpl DRI) const;
const MachOFormat::Symbol64TableEntry * const MachOFormat::SymbolTableEntry<true> *
getSymbol64TableEntry(DataRefImpl DRI) const; getSymbol64TableEntry(DataRefImpl DRI) const;
const MachOFormat::SymbolTableEntry * const MachOFormat::SymbolTableEntry<false> *
getSymbolTableEntry(DataRefImpl DRI) const; getSymbolTableEntry(DataRefImpl DRI) const;
bool is64Bit() const; bool is64Bit() const;
const MachOFormat::LoadCommand *getLoadCommandInfo(unsigned Index) const; const MachOFormat::LoadCommand *getLoadCommandInfo(unsigned Index) const;
@ -242,11 +257,11 @@ private:
void moveToNextSection(DataRefImpl &DRI) const; void moveToNextSection(DataRefImpl &DRI) const;
const MachOFormat::SymbolTableEntry * const MachOFormat::SymbolTableEntry<false> *
getSymbolTableEntry(DataRefImpl DRI, getSymbolTableEntry(DataRefImpl DRI,
const MachOFormat::SymtabLoadCommand *SymtabLoadCmd) const; const MachOFormat::SymtabLoadCommand *SymtabLoadCmd) const;
const MachOFormat::Symbol64TableEntry * const MachOFormat::SymbolTableEntry<true> *
getSymbol64TableEntry(DataRefImpl DRI, getSymbol64TableEntry(DataRefImpl DRI,
const MachOFormat::SymtabLoadCommand *SymtabLoadCmd) const; const MachOFormat::SymtabLoadCommand *SymtabLoadCmd) const;

View File

@ -116,7 +116,7 @@ void MachOObjectFile::moveToNextSymbol(DataRefImpl &DRI) const {
} }
} }
const MachOFormat::SymbolTableEntry * const MachOFormat::SymbolTableEntry<false> *
MachOObjectFile::getSymbolTableEntry(DataRefImpl DRI) const { MachOObjectFile::getSymbolTableEntry(DataRefImpl DRI) const {
const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a); const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a);
const MachOFormat::SymtabLoadCommand *SymtabLoadCmd = const MachOFormat::SymtabLoadCommand *SymtabLoadCmd =
@ -125,18 +125,20 @@ MachOObjectFile::getSymbolTableEntry(DataRefImpl DRI) const {
return getSymbolTableEntry(DRI, SymtabLoadCmd); return getSymbolTableEntry(DRI, SymtabLoadCmd);
} }
const MachOFormat::SymbolTableEntry * const MachOFormat::SymbolTableEntry<false> *
MachOObjectFile::getSymbolTableEntry(DataRefImpl DRI, MachOObjectFile::getSymbolTableEntry(DataRefImpl DRI,
const MachOFormat::SymtabLoadCommand *SymtabLoadCmd) const { const MachOFormat::SymtabLoadCommand *SymtabLoadCmd) const {
uint64_t SymbolTableOffset = SymtabLoadCmd->SymbolTableOffset; uint64_t SymbolTableOffset = SymtabLoadCmd->SymbolTableOffset;
unsigned Index = DRI.d.b; unsigned Index = DRI.d.b;
uint64_t Offset = (SymbolTableOffset + uint64_t Offset = (SymbolTableOffset +
Index * sizeof(macho::SymbolTableEntry)); Index * sizeof(MachOFormat::SymbolTableEntry<false>));
StringRef Data = getData(Offset, sizeof(MachOFormat::SymbolTableEntry)); StringRef Data =
return reinterpret_cast<const MachOFormat::SymbolTableEntry*>(Data.data()); getData(Offset, sizeof(MachOFormat::SymbolTableEntry<false>));
return
reinterpret_cast<const MachOFormat::SymbolTableEntry<false>*>(Data.data());
} }
const MachOFormat::Symbol64TableEntry* const MachOFormat::SymbolTableEntry<true>*
MachOObjectFile::getSymbol64TableEntry(DataRefImpl DRI) const { MachOObjectFile::getSymbol64TableEntry(DataRefImpl DRI) const {
const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a); const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a);
const MachOFormat::SymtabLoadCommand *SymtabLoadCmd = const MachOFormat::SymtabLoadCommand *SymtabLoadCmd =
@ -145,15 +147,16 @@ MachOObjectFile::getSymbol64TableEntry(DataRefImpl DRI) const {
return getSymbol64TableEntry(DRI, SymtabLoadCmd); return getSymbol64TableEntry(DRI, SymtabLoadCmd);
} }
const MachOFormat::Symbol64TableEntry* const MachOFormat::SymbolTableEntry<true>*
MachOObjectFile::getSymbol64TableEntry(DataRefImpl DRI, MachOObjectFile::getSymbol64TableEntry(DataRefImpl DRI,
const MachOFormat::SymtabLoadCommand *SymtabLoadCmd) const { const MachOFormat::SymtabLoadCommand *SymtabLoadCmd) const {
uint64_t SymbolTableOffset = SymtabLoadCmd->SymbolTableOffset; uint64_t SymbolTableOffset = SymtabLoadCmd->SymbolTableOffset;
unsigned Index = DRI.d.b; unsigned Index = DRI.d.b;
uint64_t Offset = (SymbolTableOffset + uint64_t Offset = (SymbolTableOffset +
Index * sizeof(macho::Symbol64TableEntry)); Index * sizeof(MachOFormat::SymbolTableEntry<true>));
StringRef Data = getData(Offset, sizeof(MachOFormat::Symbol64TableEntry)); StringRef Data = getData(Offset, sizeof(MachOFormat::SymbolTableEntry<true>));
return reinterpret_cast<const MachOFormat::Symbol64TableEntry*>(Data.data()); return
reinterpret_cast<const MachOFormat::SymbolTableEntry<true>*>(Data.data());
} }
error_code MachOObjectFile::getSymbolNext(DataRefImpl DRI, error_code MachOObjectFile::getSymbolNext(DataRefImpl DRI,
@ -175,11 +178,11 @@ error_code MachOObjectFile::getSymbolName(DataRefImpl DRI,
uint32_t StringIndex; uint32_t StringIndex;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(DRI, SymtabLoadCmd); getSymbol64TableEntry(DRI, SymtabLoadCmd);
StringIndex = Entry->StringIndex; StringIndex = Entry->StringIndex;
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(DRI, SymtabLoadCmd); getSymbolTableEntry(DRI, SymtabLoadCmd);
StringIndex = Entry->StringIndex; StringIndex = Entry->StringIndex;
} }
@ -193,18 +196,20 @@ error_code MachOObjectFile::getSymbolName(DataRefImpl DRI,
error_code MachOObjectFile::getSymbolFileOffset(DataRefImpl DRI, error_code MachOObjectFile::getSymbolFileOffset(DataRefImpl DRI,
uint64_t &Result) const { uint64_t &Result) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = getSymbol64TableEntry(DRI); const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(DRI);
Result = Entry->Value; Result = Entry->Value;
if (Entry->SectionIndex) { if (Entry->SectionIndex) {
const MachOFormat::Section64 *Section = const MachOFormat::Section<true> *Section =
getSection64(Sections[Entry->SectionIndex-1]); getSection64(Sections[Entry->SectionIndex-1]);
Result += Section->Offset - Section->Address; Result += Section->Offset - Section->Address;
} }
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = getSymbolTableEntry(DRI); const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(DRI);
Result = Entry->Value; Result = Entry->Value;
if (Entry->SectionIndex) { if (Entry->SectionIndex) {
const MachOFormat::Section *Section = const MachOFormat::Section<false> *Section =
getSection(Sections[Entry->SectionIndex-1]); getSection(Sections[Entry->SectionIndex-1]);
Result += Section->Offset - Section->Address; Result += Section->Offset - Section->Address;
} }
@ -216,10 +221,12 @@ error_code MachOObjectFile::getSymbolFileOffset(DataRefImpl DRI,
error_code MachOObjectFile::getSymbolAddress(DataRefImpl DRI, error_code MachOObjectFile::getSymbolAddress(DataRefImpl DRI,
uint64_t &Result) const { uint64_t &Result) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = getSymbol64TableEntry(DRI); const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(DRI);
Result = Entry->Value; Result = Entry->Value;
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = getSymbolTableEntry(DRI); const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(DRI);
Result = Entry->Value; Result = Entry->Value;
} }
return object_error::success; return object_error::success;
@ -232,7 +239,8 @@ error_code MachOObjectFile::getSymbolSize(DataRefImpl DRI,
uint64_t EndOffset = 0; uint64_t EndOffset = 0;
uint8_t SectionIndex; uint8_t SectionIndex;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = getSymbol64TableEntry(DRI); const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(DRI);
BeginOffset = Entry->Value; BeginOffset = Entry->Value;
SectionIndex = Entry->SectionIndex; SectionIndex = Entry->SectionIndex;
if (!SectionIndex) { if (!SectionIndex) {
@ -259,7 +267,8 @@ error_code MachOObjectFile::getSymbolSize(DataRefImpl DRI,
DRI.d.b++; DRI.d.b++;
} }
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = getSymbolTableEntry(DRI); const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(DRI);
BeginOffset = Entry->Value; BeginOffset = Entry->Value;
SectionIndex = Entry->SectionIndex; SectionIndex = Entry->SectionIndex;
if (!SectionIndex) { if (!SectionIndex) {
@ -300,11 +309,13 @@ error_code MachOObjectFile::getSymbolNMTypeChar(DataRefImpl DRI,
char &Result) const { char &Result) const {
uint8_t Type, Flags; uint8_t Type, Flags;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = getSymbol64TableEntry(DRI); const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(DRI);
Type = Entry->Type; Type = Entry->Type;
Flags = Entry->Flags; Flags = Entry->Flags;
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = getSymbolTableEntry(DRI); const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(DRI);
Type = Entry->Type; Type = Entry->Type;
Flags = Entry->Flags; Flags = Entry->Flags;
} }
@ -334,11 +345,13 @@ error_code MachOObjectFile::getSymbolFlags(DataRefImpl DRI,
uint16_t MachOFlags; uint16_t MachOFlags;
uint8_t MachOType; uint8_t MachOType;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = getSymbol64TableEntry(DRI); const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(DRI);
MachOFlags = Entry->Flags; MachOFlags = Entry->Flags;
MachOType = Entry->Type; MachOType = Entry->Type;
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = getSymbolTableEntry(DRI); const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(DRI);
MachOFlags = Entry->Flags; MachOFlags = Entry->Flags;
MachOType = Entry->Type; MachOType = Entry->Type;
} }
@ -371,10 +384,12 @@ error_code MachOObjectFile::getSymbolSection(DataRefImpl Symb,
section_iterator &Res) const { section_iterator &Res) const {
uint8_t index; uint8_t index;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = getSymbol64TableEntry(Symb); const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(Symb);
index = Entry->SectionIndex; index = Entry->SectionIndex;
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = getSymbolTableEntry(Symb); const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(Symb);
index = Entry->SectionIndex; index = Entry->SectionIndex;
} }
@ -390,10 +405,12 @@ error_code MachOObjectFile::getSymbolType(DataRefImpl Symb,
SymbolRef::Type &Res) const { SymbolRef::Type &Res) const {
uint8_t n_type; uint8_t n_type;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = getSymbol64TableEntry(Symb); const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(Symb);
n_type = Entry->Type; n_type = Entry->Type;
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = getSymbolTableEntry(Symb); const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(Symb);
n_type = Entry->Type; n_type = Entry->Type;
} }
Res = SymbolRef::ST_Other; Res = SymbolRef::ST_Other;
@ -465,14 +482,14 @@ void MachOObjectFile::moveToNextSection(DataRefImpl &DRI) const {
while (DRI.d.a < LoadCommandCount) { while (DRI.d.a < LoadCommandCount) {
const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a); const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a);
if (Command->Type == macho::LCT_Segment) { if (Command->Type == macho::LCT_Segment) {
const MachOFormat::SegmentLoadCommand *SegmentLoadCmd = const MachOFormat::SegmentLoadCommand<false> *SegmentLoadCmd =
reinterpret_cast<const MachOFormat::SegmentLoadCommand*>(Command); reinterpret_cast<const MachOFormat::SegmentLoadCommand<false>*>(Command);
if (DRI.d.b < SegmentLoadCmd->NumSections) if (DRI.d.b < SegmentLoadCmd->NumSections)
return; return;
} else if (Command->Type == macho::LCT_Segment64) { } else if (Command->Type == macho::LCT_Segment64) {
const MachOFormat::Segment64LoadCommand *Segment64LoadCmd = const MachOFormat::SegmentLoadCommand<true> *SegmentLoadCmd =
reinterpret_cast<const MachOFormat::Segment64LoadCommand*>(Command); reinterpret_cast<const MachOFormat::SegmentLoadCommand<true>*>(Command);
if (DRI.d.b < Segment64LoadCmd->NumSections) if (DRI.d.b < SegmentLoadCmd->NumSections)
return; return;
} }
@ -489,13 +506,14 @@ error_code MachOObjectFile::getSectionNext(DataRefImpl DRI,
return object_error::success; return object_error::success;
} }
const MachOFormat::Section *MachOObjectFile::getSection(DataRefImpl DRI) const { const MachOFormat::Section<false> *
MachOObjectFile::getSection(DataRefImpl DRI) const {
assert(!is64Bit()); assert(!is64Bit());
const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a); const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a);
uintptr_t CommandAddr = reinterpret_cast<uintptr_t>(Command); uintptr_t CommandAddr = reinterpret_cast<uintptr_t>(Command);
uintptr_t SectionAddr = CommandAddr + sizeof(macho::SegmentLoadCommand) + uintptr_t SectionAddr = CommandAddr + sizeof(macho::SegmentLoadCommand) +
DRI.d.b * sizeof(MachOFormat::Section); DRI.d.b * sizeof(MachOFormat::Section<false>);
return reinterpret_cast<const MachOFormat::Section*>(SectionAddr); return reinterpret_cast<const MachOFormat::Section<false>*>(SectionAddr);
} }
std::size_t MachOObjectFile::getSectionIndex(DataRefImpl Sec) const { std::size_t MachOObjectFile::getSectionIndex(DataRefImpl Sec) const {
@ -505,14 +523,14 @@ std::size_t MachOObjectFile::getSectionIndex(DataRefImpl Sec) const {
return std::distance(Sections.begin(), loc); return std::distance(Sections.begin(), loc);
} }
const MachOFormat::Section64 * const MachOFormat::Section<true> *
MachOObjectFile::getSection64(DataRefImpl DRI) const { MachOObjectFile::getSection64(DataRefImpl DRI) const {
assert(is64Bit()); assert(is64Bit());
const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a); const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a);
uintptr_t CommandAddr = reinterpret_cast<uintptr_t>(Command); uintptr_t CommandAddr = reinterpret_cast<uintptr_t>(Command);
uintptr_t SectionAddr = CommandAddr + sizeof(macho::Segment64LoadCommand) + uintptr_t SectionAddr = CommandAddr + sizeof(macho::Segment64LoadCommand) +
DRI.d.b * sizeof(MachOFormat::Section64); DRI.d.b * sizeof(MachOFormat::Section<true>);
return reinterpret_cast<const MachOFormat::Section64*>(SectionAddr); return reinterpret_cast<const MachOFormat::Section<true>*>(SectionAddr);
} }
static StringRef parseSegmentOrSectionName(const char *P) { static StringRef parseSegmentOrSectionName(const char *P) {
@ -525,10 +543,10 @@ static StringRef parseSegmentOrSectionName(const char *P) {
ArrayRef<char> MachOObjectFile::getSectionRawName(DataRefImpl DRI) const { ArrayRef<char> MachOObjectFile::getSectionRawName(DataRefImpl DRI) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *sec = getSection64(DRI); const MachOFormat::Section<true> *sec = getSection64(DRI);
return ArrayRef<char>(sec->Name); return ArrayRef<char>(sec->Name);
} else { } else {
const MachOFormat::Section *sec = getSection(DRI); const MachOFormat::Section<false> *sec = getSection(DRI);
return ArrayRef<char>(sec->Name); return ArrayRef<char>(sec->Name);
} }
} }
@ -543,10 +561,10 @@ error_code MachOObjectFile::getSectionName(DataRefImpl DRI,
ArrayRef<char> ArrayRef<char>
MachOObjectFile::getSectionRawFinalSegmentName(DataRefImpl Sec) const { MachOObjectFile::getSectionRawFinalSegmentName(DataRefImpl Sec) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *sec = getSection64(Sec); const MachOFormat::Section<true> *sec = getSection64(Sec);
return ArrayRef<char>(sec->SegmentName, 16); return ArrayRef<char>(sec->SegmentName, 16);
} else { } else {
const MachOFormat::Section *sec = getSection(Sec); const MachOFormat::Section<false> *sec = getSection(Sec);
return ArrayRef<char>(sec->SegmentName); return ArrayRef<char>(sec->SegmentName);
} }
} }
@ -559,10 +577,10 @@ StringRef MachOObjectFile::getSectionFinalSegmentName(DataRefImpl DRI) const {
error_code MachOObjectFile::getSectionAddress(DataRefImpl DRI, error_code MachOObjectFile::getSectionAddress(DataRefImpl DRI,
uint64_t &Result) const { uint64_t &Result) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(DRI); const MachOFormat::Section<true> *Sect = getSection64(DRI);
Result = Sect->Address; Result = Sect->Address;
} else { } else {
const MachOFormat::Section *Sect = getSection(DRI); const MachOFormat::Section<false> *Sect = getSection(DRI);
Result = Sect->Address; Result = Sect->Address;
} }
return object_error::success; return object_error::success;
@ -571,10 +589,10 @@ error_code MachOObjectFile::getSectionAddress(DataRefImpl DRI,
error_code MachOObjectFile::getSectionSize(DataRefImpl DRI, error_code MachOObjectFile::getSectionSize(DataRefImpl DRI,
uint64_t &Result) const { uint64_t &Result) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(DRI); const MachOFormat::Section<true> *Sect = getSection64(DRI);
Result = Sect->Size; Result = Sect->Size;
} else { } else {
const MachOFormat::Section *Sect = getSection(DRI); const MachOFormat::Section<false> *Sect = getSection(DRI);
Result = Sect->Size; Result = Sect->Size;
} }
return object_error::success; return object_error::success;
@ -583,10 +601,10 @@ error_code MachOObjectFile::getSectionSize(DataRefImpl DRI,
error_code MachOObjectFile::getSectionContents(DataRefImpl DRI, error_code MachOObjectFile::getSectionContents(DataRefImpl DRI,
StringRef &Result) const { StringRef &Result) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(DRI); const MachOFormat::Section<true> *Sect = getSection64(DRI);
Result = getData(Sect->Offset, Sect->Size); Result = getData(Sect->Offset, Sect->Size);
} else { } else {
const MachOFormat::Section *Sect = getSection(DRI); const MachOFormat::Section<false> *Sect = getSection(DRI);
Result = getData(Sect->Offset, Sect->Size); Result = getData(Sect->Offset, Sect->Size);
} }
return object_error::success; return object_error::success;
@ -595,10 +613,10 @@ error_code MachOObjectFile::getSectionContents(DataRefImpl DRI,
error_code MachOObjectFile::getSectionAlignment(DataRefImpl DRI, error_code MachOObjectFile::getSectionAlignment(DataRefImpl DRI,
uint64_t &Result) const { uint64_t &Result) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(DRI); const MachOFormat::Section<true> *Sect = getSection64(DRI);
Result = uint64_t(1) << Sect->Align; Result = uint64_t(1) << Sect->Align;
} else { } else {
const MachOFormat::Section *Sect = getSection(DRI); const MachOFormat::Section<false> *Sect = getSection(DRI);
Result = uint64_t(1) << Sect->Align; Result = uint64_t(1) << Sect->Align;
} }
return object_error::success; return object_error::success;
@ -607,10 +625,10 @@ error_code MachOObjectFile::getSectionAlignment(DataRefImpl DRI,
error_code MachOObjectFile::isSectionText(DataRefImpl DRI, error_code MachOObjectFile::isSectionText(DataRefImpl DRI,
bool &Result) const { bool &Result) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(DRI); const MachOFormat::Section<true> *Sect = getSection64(DRI);
Result = Sect->Flags & macho::SF_PureInstructions; Result = Sect->Flags & macho::SF_PureInstructions;
} else { } else {
const MachOFormat::Section *Sect = getSection(DRI); const MachOFormat::Section<false> *Sect = getSection(DRI);
Result = Sect->Flags & macho::SF_PureInstructions; Result = Sect->Flags & macho::SF_PureInstructions;
} }
return object_error::success; return object_error::success;
@ -647,12 +665,12 @@ error_code MachOObjectFile::isSectionVirtual(DataRefImpl Sec,
error_code MachOObjectFile::isSectionZeroInit(DataRefImpl DRI, error_code MachOObjectFile::isSectionZeroInit(DataRefImpl DRI,
bool &Result) const { bool &Result) const {
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(DRI); const MachOFormat::Section<true> *Sect = getSection64(DRI);
unsigned SectionType = Sect->Flags & MachO::SectionFlagMaskSectionType; unsigned SectionType = Sect->Flags & MachO::SectionFlagMaskSectionType;
Result = (SectionType == MachO::SectionTypeZeroFill || Result = (SectionType == MachO::SectionTypeZeroFill ||
SectionType == MachO::SectionTypeZeroFillLarge); SectionType == MachO::SectionTypeZeroFillLarge);
} else { } else {
const MachOFormat::Section *Sect = getSection(DRI); const MachOFormat::Section<false> *Sect = getSection(DRI);
unsigned SectionType = Sect->Flags & MachO::SectionFlagMaskSectionType; unsigned SectionType = Sect->Flags & MachO::SectionFlagMaskSectionType;
Result = (SectionType == MachO::SectionTypeZeroFill || Result = (SectionType == MachO::SectionTypeZeroFill ||
SectionType == MachO::SectionTypeZeroFillLarge); SectionType == MachO::SectionTypeZeroFillLarge);
@ -688,11 +706,13 @@ error_code MachOObjectFile::sectionContainsSymbol(DataRefImpl Sec,
SectEnd += SectBegin; SectEnd += SectBegin;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = getSymbol64TableEntry(Symb); const MachOFormat::SymbolTableEntry<true> *Entry =
getSymbol64TableEntry(Symb);
uint64_t SymAddr= Entry->Value; uint64_t SymAddr= Entry->Value;
Result = (SymAddr >= SectBegin) && (SymAddr < SectEnd); Result = (SymAddr >= SectBegin) && (SymAddr < SectEnd);
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = getSymbolTableEntry(Symb); const MachOFormat::SymbolTableEntry<false> *Entry =
getSymbolTableEntry(Symb);
uint64_t SymAddr= Entry->Value; uint64_t SymAddr= Entry->Value;
Result = (SymAddr >= SectBegin) && (SymAddr < SectEnd); Result = (SymAddr >= SectBegin) && (SymAddr < SectEnd);
} }
@ -705,13 +725,14 @@ relocation_iterator MachOObjectFile::getSectionRelBegin(DataRefImpl Sec) const {
ret.d.b = getSectionIndex(Sec); ret.d.b = getSectionIndex(Sec);
return relocation_iterator(RelocationRef(ret, this)); return relocation_iterator(RelocationRef(ret, this));
} }
relocation_iterator MachOObjectFile::getSectionRelEnd(DataRefImpl Sec) const { relocation_iterator MachOObjectFile::getSectionRelEnd(DataRefImpl Sec) const {
uint32_t last_reloc; uint32_t last_reloc;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(Sec); const MachOFormat::Section<true> *Sect = getSection64(Sec);
last_reloc = Sect->NumRelocationTableEntries; last_reloc = Sect->NumRelocationTableEntries;
} else { } else {
const MachOFormat::Section *Sect = getSection(Sec); const MachOFormat::Section<false> *Sect = getSection(Sec);
last_reloc = Sect->NumRelocationTableEntries; last_reloc = Sect->NumRelocationTableEntries;
} }
DataRefImpl ret; DataRefImpl ret;
@ -738,10 +759,10 @@ const MachOFormat::RelocationEntry *
MachOObjectFile::getRelocation(DataRefImpl Rel) const { MachOObjectFile::getRelocation(DataRefImpl Rel) const {
uint32_t relOffset; uint32_t relOffset;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(Sections[Rel.d.b]); const MachOFormat::Section<true> *Sect = getSection64(Sections[Rel.d.b]);
relOffset = Sect->RelocationTableOffset; relOffset = Sect->RelocationTableOffset;
} else { } else {
const MachOFormat::Section *Sect = getSection(Sections[Rel.d.b]); const MachOFormat::Section<false> *Sect = getSection(Sections[Rel.d.b]);
relOffset = Sect->RelocationTableOffset; relOffset = Sect->RelocationTableOffset;
} }
uint64_t Offset = relOffset + Rel.d.a * sizeof(MachOFormat::RelocationEntry); uint64_t Offset = relOffset + Rel.d.a * sizeof(MachOFormat::RelocationEntry);
@ -759,10 +780,10 @@ error_code MachOObjectFile::getRelocationAddress(DataRefImpl Rel,
uint64_t &Res) const { uint64_t &Res) const {
const uint8_t* sectAddress = 0; const uint8_t* sectAddress = 0;
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(Sections[Rel.d.b]); const MachOFormat::Section<true> *Sect = getSection64(Sections[Rel.d.b]);
sectAddress += Sect->Address; sectAddress += Sect->Address;
} else { } else {
const MachOFormat::Section *Sect = getSection(Sections[Rel.d.b]); const MachOFormat::Section<false> *Sect = getSection(Sections[Rel.d.b]);
sectAddress += Sect->Address; sectAddress += Sect->Address;
} }
const MachOFormat::RelocationEntry *RE = getRelocation(Rel); const MachOFormat::RelocationEntry *RE = getRelocation(Rel);
@ -926,10 +947,10 @@ error_code MachOObjectFile::getRelocationAdditionalInfo(DataRefImpl Rel,
if (!isExtern) { if (!isExtern) {
const uint8_t* sectAddress = base(); const uint8_t* sectAddress = base();
if (is64Bit()) { if (is64Bit()) {
const MachOFormat::Section64 *Sect = getSection64(Sections[Rel.d.b]); const MachOFormat::Section<true> *Sect = getSection64(Sections[Rel.d.b]);
sectAddress += Sect->Offset; sectAddress += Sect->Offset;
} else { } else {
const MachOFormat::Section *Sect = getSection(Sections[Rel.d.b]); const MachOFormat::Section<false> *Sect = getSection(Sections[Rel.d.b]);
sectAddress += Sect->Offset; sectAddress += Sect->Offset;
} }
Res = reinterpret_cast<uintptr_t>(sectAddress); Res = reinterpret_cast<uintptr_t>(sectAddress);

View File

@ -161,7 +161,7 @@ static void getSection(const MachOObjectFile *Obj,
DataRefImpl DRI, DataRefImpl DRI,
MachOSection &Section) { MachOSection &Section) {
if (Obj->is64Bit()) { if (Obj->is64Bit()) {
const MachOFormat::Section64 *Sect = Obj->getSection64(DRI); const MachOFormat::Section<true> *Sect = Obj->getSection64(DRI);
Section.Address = Sect->Address; Section.Address = Sect->Address;
Section.Size = Sect->Size; Section.Size = Sect->Size;
@ -173,7 +173,7 @@ static void getSection(const MachOObjectFile *Obj,
Section.Reserved1 = Sect->Reserved1; Section.Reserved1 = Sect->Reserved1;
Section.Reserved2 = Sect->Reserved2; Section.Reserved2 = Sect->Reserved2;
} else { } else {
const MachOFormat::Section *Sect = Obj->getSection(DRI); const MachOFormat::Section<false> *Sect = Obj->getSection(DRI);
Section.Address = Sect->Address; Section.Address = Sect->Address;
Section.Size = Sect->Size; Section.Size = Sect->Size;
@ -191,15 +191,16 @@ static void getSymbol(const MachOObjectFile *Obj,
DataRefImpl DRI, DataRefImpl DRI,
MachOSymbol &Symbol) { MachOSymbol &Symbol) {
if (Obj->is64Bit()) { if (Obj->is64Bit()) {
const MachOFormat::Symbol64TableEntry *Entry = const MachOFormat::SymbolTableEntry<true> *Entry =
Obj->getSymbol64TableEntry( DRI); Obj->getSymbol64TableEntry(DRI);
Symbol.StringIndex = Entry->StringIndex; Symbol.StringIndex = Entry->StringIndex;
Symbol.Type = Entry->Type; Symbol.Type = Entry->Type;
Symbol.SectionIndex = Entry->SectionIndex; Symbol.SectionIndex = Entry->SectionIndex;
Symbol.Flags = Entry->Flags; Symbol.Flags = Entry->Flags;
Symbol.Value = Entry->Value; Symbol.Value = Entry->Value;
} else { } else {
const MachOFormat::SymbolTableEntry *Entry = Obj->getSymbolTableEntry(DRI); const MachOFormat::SymbolTableEntry<false> *Entry =
Obj->getSymbolTableEntry(DRI);
Symbol.StringIndex = Entry->StringIndex; Symbol.StringIndex = Entry->StringIndex;
Symbol.Type = Entry->Type; Symbol.Type = Entry->Type;
Symbol.SectionIndex = Entry->SectionIndex; Symbol.SectionIndex = Entry->SectionIndex;