[llvm] r178994 - Implement MachOObjectFile::getHeader directly.

Rafael Espindola rafael.espindola at gmail.com
Sun Apr 7 12:26:58 PDT 2013


Author: rafael
Date: Sun Apr  7 14:26:57 2013
New Revision: 178994

URL: http://llvm.org/viewvc/llvm-project?rev=178994&view=rev
Log:
Implement MachOObjectFile::getHeader directly.

Modified:
    llvm/trunk/include/llvm/Object/MachO.h
    llvm/trunk/lib/Object/MachOObjectFile.cpp
    llvm/trunk/tools/llvm-objdump/MachODump.cpp

Modified: llvm/trunk/include/llvm/Object/MachO.h
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/Object/MachO.h?rev=178994&r1=178993&r2=178994&view=diff
==============================================================================
--- llvm/trunk/include/llvm/Object/MachO.h (original)
+++ llvm/trunk/include/llvm/Object/MachO.h Sun Apr  7 14:26:57 2013
@@ -125,6 +125,16 @@ namespace MachOFormat {
     support::ulittle32_t DataOffset;
     support::ulittle32_t DataSize;
   };
+
+  struct Header {
+    support::ulittle32_t Magic;
+    support::ulittle32_t CPUType;
+    support::ulittle32_t CPUSubtype;
+    support::ulittle32_t FileType;
+    support::ulittle32_t NumLoadCommands;
+    support::ulittle32_t SizeOfLoadCommands;
+    support::ulittle32_t Flags;
+  };
 }
 
 class MachOObjectFile : public ObjectFile {
@@ -164,7 +174,7 @@ public:
   bool is64Bit() const;
   const MachOFormat::LoadCommand *getLoadCommandInfo(unsigned Index) const;
   void ReadULEB128s(uint64_t Index, SmallVectorImpl<uint64_t> &Out) const;
-  const macho::Header &getHeader() const;
+  const MachOFormat::Header *getHeader() const;
   unsigned getHeaderSize() const;
   StringRef getData(size_t Offset, size_t Size) const;
 

Modified: llvm/trunk/lib/Object/MachOObjectFile.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Object/MachOObjectFile.cpp?rev=178994&r1=178993&r2=178994&view=diff
==============================================================================
--- llvm/trunk/lib/Object/MachOObjectFile.cpp (original)
+++ llvm/trunk/lib/Object/MachOObjectFile.cpp Sun Apr  7 14:26:57 2013
@@ -46,7 +46,7 @@ MachOObjectFile::MachOObjectFile(MemoryB
 
   DataRefImpl DRI;
   moveToNextSection(DRI);
-  uint32_t LoadCommandCount = getHeader().NumLoadCommands;
+  uint32_t LoadCommandCount = getHeader()->NumLoadCommands;
   while (DRI.d.a < LoadCommandCount) {
     Sections.push_back(DRI);
     DRI.d.b++;
@@ -80,8 +80,9 @@ void MachOObjectFile::ReadULEB128s(uint6
   return MachOObj->ReadULEB128s(Index, Out);
 }
 
-const macho::Header &MachOObjectFile::getHeader() const {
-  return MachOObj->getHeader();
+const MachOFormat::Header *MachOObjectFile::getHeader() const {
+  StringRef Data = getData(0, sizeof(MachOFormat::Header));
+  return reinterpret_cast<const MachOFormat::Header*>(Data.data());
 }
 
 unsigned MachOObjectFile::getHeaderSize() const {
@@ -103,7 +104,7 @@ ObjectFile *ObjectFile::createMachOObjec
 /*===-- Symbols -----------------------------------------------------------===*/
 
 void MachOObjectFile::moveToNextSymbol(DataRefImpl &DRI) const {
-  uint32_t LoadCommandCount = getHeader().NumLoadCommands;
+  uint32_t LoadCommandCount = getHeader()->NumLoadCommands;
   while (DRI.d.a < LoadCommandCount) {
     const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a);
     if (Command->Type == macho::LCT_Symtab) {
@@ -229,7 +230,7 @@ error_code MachOObjectFile::getSymbolAdd
 
 error_code MachOObjectFile::getSymbolSize(DataRefImpl DRI,
                                           uint64_t &Result) const {
-  uint32_t LoadCommandCount = getHeader().NumLoadCommands;
+  uint32_t LoadCommandCount = getHeader()->NumLoadCommands;
   uint64_t BeginOffset;
   uint64_t EndOffset = 0;
   uint8_t SectionIndex;
@@ -431,7 +432,7 @@ symbol_iterator MachOObjectFile::begin_s
 
 symbol_iterator MachOObjectFile::end_symbols() const {
   DataRefImpl DRI;
-  DRI.d.a = getHeader().NumLoadCommands;
+  DRI.d.a = getHeader()->NumLoadCommands;
   return symbol_iterator(SymbolRef(DRI, this));
 }
 
@@ -463,7 +464,7 @@ StringRef MachOObjectFile::getLoadName()
 /*===-- Sections ----------------------------------------------------------===*/
 
 void MachOObjectFile::moveToNextSection(DataRefImpl &DRI) const {
-  uint32_t LoadCommandCount = getHeader().NumLoadCommands;
+  uint32_t LoadCommandCount = getHeader()->NumLoadCommands;
   while (DRI.d.a < LoadCommandCount) {
     const MachOFormat::LoadCommand *Command = getLoadCommandInfo(DRI.d.a);
     if (Command->Type == macho::LCT_Segment) {
@@ -740,7 +741,7 @@ section_iterator MachOObjectFile::begin_
 
 section_iterator MachOObjectFile::end_sections() const {
   DataRefImpl DRI;
-  DRI.d.a = getHeader().NumLoadCommands;
+  DRI.d.a = getHeader()->NumLoadCommands;
   return section_iterator(SectionRef(DRI, this));
 }
 
@@ -816,7 +817,7 @@ error_code MachOObjectFile::getRelocatio
     for (unsigned i = 0; i < SymbolIdx; i++) {
       Sym.d.b++;
       moveToNextSymbol(Sym);
-      assert(Sym.d.a < getHeader().NumLoadCommands &&
+      assert(Sym.d.a < getHeader()->NumLoadCommands &&
              "Relocation symbol index out of range!");
     }
   }
@@ -1300,7 +1301,7 @@ uint8_t MachOObjectFile::getBytesInAddre
 
 StringRef MachOObjectFile::getFileFormatName() const {
   if (!is64Bit()) {
-    switch (getHeader().CPUType) {
+    switch (getHeader()->CPUType) {
     case llvm::MachO::CPUTypeI386:
       return "Mach-O 32-bit i386";
     case llvm::MachO::CPUTypeARM:
@@ -1308,18 +1309,18 @@ StringRef MachOObjectFile::getFileFormat
     case llvm::MachO::CPUTypePowerPC:
       return "Mach-O 32-bit ppc";
     default:
-      assert((getHeader().CPUType & llvm::MachO::CPUArchABI64) == 0 &&
+      assert((getHeader()->CPUType & llvm::MachO::CPUArchABI64) == 0 &&
              "64-bit object file when we're not 64-bit?");
       return "Mach-O 32-bit unknown";
     }
   }
 
   // Make sure the cpu type has the correct mask.
-  assert((getHeader().CPUType & llvm::MachO::CPUArchABI64)
+  assert((getHeader()->CPUType & llvm::MachO::CPUArchABI64)
 	 == llvm::MachO::CPUArchABI64 &&
 	 "32-bit object file when we're 64-bit?");
 
-  switch (getHeader().CPUType) {
+  switch (getHeader()->CPUType) {
   case llvm::MachO::CPUTypeX86_64:
     return "Mach-O 64-bit x86-64";
   case llvm::MachO::CPUTypePowerPC64:
@@ -1330,7 +1331,7 @@ StringRef MachOObjectFile::getFileFormat
 }
 
 unsigned MachOObjectFile::getArch() const {
-  switch (getHeader().CPUType) {
+  switch (getHeader()->CPUType) {
   case llvm::MachO::CPUTypeI386:
     return Triple::x86;
   case llvm::MachO::CPUTypeX86_64:

Modified: llvm/trunk/tools/llvm-objdump/MachODump.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/tools/llvm-objdump/MachODump.cpp?rev=178994&r1=178993&r2=178994&view=diff
==============================================================================
--- llvm/trunk/tools/llvm-objdump/MachODump.cpp (original)
+++ llvm/trunk/tools/llvm-objdump/MachODump.cpp Sun Apr  7 14:26:57 2013
@@ -56,7 +56,7 @@ static const Target *GetTarget(const Mac
   // Figure out the target triple.
   if (TripleName.empty()) {
     llvm::Triple TT("unknown-unknown-unknown");
-    switch (MachOObj->getHeader().CPUType) {
+    switch (MachOObj->getHeader()->CPUType) {
     case llvm::MachO::CPUTypeI386:
       TT.setArch(Triple::ArchType(Triple::x86));
       break;
@@ -199,7 +199,7 @@ static void emitDOTFile(const char *File
   Out << "}\n";
 }
 
-static void getSectionsAndSymbols(const macho::Header &Header,
+static void getSectionsAndSymbols(const MachOFormat::Header *Header,
                                   MachOObjectFile *MachOObj,
                                   std::vector<SectionRef> &Sections,
                                   std::vector<SymbolRef> &Symbols,
@@ -217,7 +217,7 @@ static void getSectionsAndSymbols(const
     Sections.push_back(*SI);
   }
 
-  for (unsigned i = 0; i != Header.NumLoadCommands; ++i) {
+  for (unsigned i = 0; i != Header->NumLoadCommands; ++i) {
     const MachOFormat::LoadCommand *Command = MachOObj->getLoadCommandInfo(i);
     if (Command->Type == macho::LCT_FunctionStarts) {
       // We found a function starts segment, parse the addresses for later
@@ -269,7 +269,7 @@ void llvm::DisassembleInputMachO(StringR
 
   outs() << '\n' << Filename << ":\n\n";
 
-  const macho::Header &Header = MachOOF->getHeader();
+  const MachOFormat::Header *Header = MachOOF->getHeader();
 
   std::vector<SectionRef> Sections;
   std::vector<SymbolRef> Symbols;





More information about the llvm-commits mailing list