include/llvm/Object/COFF.h

   1 //===- COFF.h - COFF object file implementation -----------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file declares the COFFObjectFile class.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_OBJECT_COFF_H
  15 #define LLVM_OBJECT_COFF_H
  16
  17 #include "llvm/Object/ObjectFile.h"
  18 #include "llvm/Support/COFF.h"
  19 #include "llvm/Support/Endian.h"
  20
  21 namespace llvm {
  22 namespace object {
  23
  24 struct coff_file_header {
  25   support::ulittle16_t Machine;
  26   support::ulittle16_t NumberOfSections;
  27   support::ulittle32_t TimeDateStamp;
  28   support::ulittle32_t PointerToSymbolTable;
  29   support::ulittle32_t NumberOfSymbols;
  30   support::ulittle16_t SizeOfOptionalHeader;
  31   support::ulittle16_t Characteristics;
  32 };
  33
  34 struct coff_symbol {
  35   struct StringTableOffset {
  36     support::ulittle32_t Zeroes;
  37     support::ulittle32_t Offset;
  38   };
  39
  40   union {
  41     char ShortName[8];
  42     StringTableOffset Offset;
  43   } Name;
  44
  45   support::ulittle32_t Value;
  46   support::little16_t SectionNumber;
  47
  48   struct {
  49     support::ulittle8_t BaseType;
  50     support::ulittle8_t ComplexType;
  51   } Type;
  52
  53   support::ulittle8_t  StorageClass;
  54   support::ulittle8_t  NumberOfAuxSymbols;
  55 };
  56
  57 struct coff_section {
  58   char Name[8];
  59   support::ulittle32_t VirtualSize;
  60   support::ulittle32_t VirtualAddress;
  61   support::ulittle32_t SizeOfRawData;
  62   support::ulittle32_t PointerToRawData;
  63   support::ulittle32_t PointerToRelocations;
  64   support::ulittle32_t PointerToLinenumbers;
  65   support::ulittle16_t NumberOfRelocations;
  66   support::ulittle16_t NumberOfLinenumbers;
  67   support::ulittle32_t Characteristics;
  68 };
  69
  70 class COFFObjectFile : public ObjectFile {
  71 private:
  72         uint64_t         HeaderOff;
  73   const coff_file_header *Header;
  74   const coff_section     *SectionTable;
  75   const coff_symbol      *SymbolTable;
  76   const char             *StringTable;
  77
  78   const coff_section     *getSection(std::size_t index) const;
  79   const char             *getString(std::size_t offset) const;
  80
  81 protected:
  82   virtual SymbolRef getSymbolNext(DataRefImpl Symb) const;
  83   virtual StringRef getSymbolName(DataRefImpl Symb) const;
  84   virtual uint64_t  getSymbolAddress(DataRefImpl Symb) const;
  85   virtual uint64_t  getSymbolSize(DataRefImpl Symb) const;
  86   virtual char      getSymbolNMTypeChar(DataRefImpl Symb) const;
  87   virtual bool      isSymbolInternal(DataRefImpl Symb) const;
  88
  89   virtual SectionRef getSectionNext(DataRefImpl Sec) const;
  90   virtual StringRef  getSectionName(DataRefImpl Sec) const;
  91   virtual uint64_t   getSectionAddress(DataRefImpl Sec) const;
  92   virtual uint64_t   getSectionSize(DataRefImpl Sec) const;
  93   virtual StringRef  getSectionContents(DataRefImpl Sec) const;
  94   virtual bool       isSectionText(DataRefImpl Sec) const;
  95
  96 public:
  97   COFFObjectFile(MemoryBuffer *Object, error_code &ec);
  98   virtual symbol_iterator begin_symbols() const;
  99   virtual symbol_iterator end_symbols() const;
 100   virtual section_iterator begin_sections() const;
 101   virtual section_iterator end_sections() const;
 102
 103   virtual uint8_t getBytesInAddress() const;
 104   virtual StringRef getFileFormatName() const;
 105   virtual unsigned getArch() const;
 106 };
 107
 108 }
 109 }
 110
 111 #endif