18e90adafSMichael J. Spencer //===- COFFObjectFile.cpp - COFF object file implementation -----*- C++ -*-===//
28e90adafSMichael J. Spencer //
38e90adafSMichael J. Spencer //                     The LLVM Compiler Infrastructure
48e90adafSMichael J. Spencer //
58e90adafSMichael J. Spencer // This file is distributed under the University of Illinois Open Source
68e90adafSMichael J. Spencer // License. See LICENSE.TXT for details.
78e90adafSMichael J. Spencer //
88e90adafSMichael J. Spencer //===----------------------------------------------------------------------===//
98e90adafSMichael J. Spencer //
108e90adafSMichael J. Spencer // This file declares the COFFObjectFile class.
118e90adafSMichael J. Spencer //
128e90adafSMichael J. Spencer //===----------------------------------------------------------------------===//
138e90adafSMichael J. Spencer 
14ec29b121SMichael J. Spencer #include "llvm/Object/COFF.h"
15e5fd0047SMichael J. Spencer #include "llvm/ADT/SmallString.h"
168e90adafSMichael J. Spencer #include "llvm/ADT/StringSwitch.h"
178e90adafSMichael J. Spencer #include "llvm/ADT/Triple.h"
188e90adafSMichael J. Spencer 
198e90adafSMichael J. Spencer using namespace llvm;
208e90adafSMichael J. Spencer using namespace object;
218e90adafSMichael J. Spencer 
228e90adafSMichael J. Spencer namespace {
238e90adafSMichael J. Spencer using support::ulittle8_t;
248e90adafSMichael J. Spencer using support::ulittle16_t;
258e90adafSMichael J. Spencer using support::ulittle32_t;
268e90adafSMichael J. Spencer using support::little16_t;
278e90adafSMichael J. Spencer }
288e90adafSMichael J. Spencer 
291d6167fdSMichael J. Spencer namespace {
301d6167fdSMichael J. Spencer // Returns false if size is greater than the buffer size. And sets ec.
311d6167fdSMichael J. Spencer bool checkSize(const MemoryBuffer *m, error_code &ec, uint64_t size) {
321d6167fdSMichael J. Spencer   if (m->getBufferSize() < size) {
331d6167fdSMichael J. Spencer     ec = object_error::unexpected_eof;
341d6167fdSMichael J. Spencer     return false;
351d6167fdSMichael J. Spencer   }
361d6167fdSMichael J. Spencer   return true;
378e90adafSMichael J. Spencer }
388e90adafSMichael J. Spencer 
391d6167fdSMichael J. Spencer // Returns false if any bytes in [addr, addr + size) fall outsize of m.
401d6167fdSMichael J. Spencer bool checkAddr(const MemoryBuffer *m,
411d6167fdSMichael J. Spencer                error_code &ec,
421d6167fdSMichael J. Spencer                uintptr_t addr,
431d6167fdSMichael J. Spencer                uint64_t size) {
441d6167fdSMichael J. Spencer   if (addr + size < addr ||
451d6167fdSMichael J. Spencer       addr + size < size ||
461d6167fdSMichael J. Spencer       addr + size > uintptr_t(m->getBufferEnd())) {
471d6167fdSMichael J. Spencer     ec = object_error::unexpected_eof;
481d6167fdSMichael J. Spencer     return false;
491d6167fdSMichael J. Spencer   }
501d6167fdSMichael J. Spencer   return true;
511d6167fdSMichael J. Spencer }
521d6167fdSMichael J. Spencer }
531d6167fdSMichael J. Spencer 
541d6167fdSMichael J. Spencer const coff_symbol *COFFObjectFile::toSymb(DataRefImpl Symb) const {
551d6167fdSMichael J. Spencer   const coff_symbol *addr = reinterpret_cast<const coff_symbol*>(Symb.p);
561d6167fdSMichael J. Spencer 
571d6167fdSMichael J. Spencer # ifndef NDEBUG
581d6167fdSMichael J. Spencer   // Verify that the symbol points to a valid entry in the symbol table.
591d6167fdSMichael J. Spencer   uintptr_t offset = uintptr_t(addr) - uintptr_t(base());
601d6167fdSMichael J. Spencer   if (offset < Header->PointerToSymbolTable
611d6167fdSMichael J. Spencer       || offset >= Header->PointerToSymbolTable
621d6167fdSMichael J. Spencer          + (Header->NumberOfSymbols * sizeof(coff_symbol)))
631d6167fdSMichael J. Spencer     report_fatal_error("Symbol was outside of symbol table.");
641d6167fdSMichael J. Spencer 
651d6167fdSMichael J. Spencer   assert((offset - Header->PointerToSymbolTable) % sizeof(coff_symbol)
661d6167fdSMichael J. Spencer          == 0 && "Symbol did not point to the beginning of a symbol");
671d6167fdSMichael J. Spencer # endif
681d6167fdSMichael J. Spencer 
691d6167fdSMichael J. Spencer   return addr;
701d6167fdSMichael J. Spencer }
711d6167fdSMichael J. Spencer 
721d6167fdSMichael J. Spencer const coff_section *COFFObjectFile::toSec(DataRefImpl Sec) const {
731d6167fdSMichael J. Spencer   const coff_section *addr = reinterpret_cast<const coff_section*>(Sec.p);
741d6167fdSMichael J. Spencer 
751d6167fdSMichael J. Spencer # ifndef NDEBUG
761d6167fdSMichael J. Spencer   // Verify that the section points to a valid entry in the section table.
771d6167fdSMichael J. Spencer   if (addr < SectionTable
781d6167fdSMichael J. Spencer       || addr >= (SectionTable + Header->NumberOfSections))
791d6167fdSMichael J. Spencer     report_fatal_error("Section was outside of section table.");
801d6167fdSMichael J. Spencer 
811d6167fdSMichael J. Spencer   uintptr_t offset = uintptr_t(addr) - uintptr_t(SectionTable);
821d6167fdSMichael J. Spencer   assert(offset % sizeof(coff_section) == 0 &&
831d6167fdSMichael J. Spencer          "Section did not point to the beginning of a section");
841d6167fdSMichael J. Spencer # endif
851d6167fdSMichael J. Spencer 
861d6167fdSMichael J. Spencer   return addr;
871d6167fdSMichael J. Spencer }
881d6167fdSMichael J. Spencer 
891d6167fdSMichael J. Spencer error_code COFFObjectFile::getSymbolNext(DataRefImpl Symb,
901d6167fdSMichael J. Spencer                                          SymbolRef &Result) const {
911d6167fdSMichael J. Spencer   const coff_symbol *symb = toSymb(Symb);
921d6167fdSMichael J. Spencer   symb += 1 + symb->NumberOfAuxSymbols;
931d6167fdSMichael J. Spencer   Symb.p = reinterpret_cast<uintptr_t>(symb);
941d6167fdSMichael J. Spencer   Result = SymbolRef(Symb, this);
951d6167fdSMichael J. Spencer   return object_error::success;
961d6167fdSMichael J. Spencer }
971d6167fdSMichael J. Spencer 
981d6167fdSMichael J. Spencer  error_code COFFObjectFile::getSymbolName(DataRefImpl Symb,
991d6167fdSMichael J. Spencer                                           StringRef &Result) const {
1001d6167fdSMichael J. Spencer   const coff_symbol *symb = toSymb(Symb);
10189a7a5eaSMichael J. Spencer   return getSymbolName(symb, Result);
1028e90adafSMichael J. Spencer }
1038e90adafSMichael J. Spencer 
104cbe72fc9SDanil Malyshev error_code COFFObjectFile::getSymbolFileOffset(DataRefImpl Symb,
1051d6167fdSMichael J. Spencer                                             uint64_t &Result) const {
1061d6167fdSMichael J. Spencer   const coff_symbol *symb = toSymb(Symb);
1075ebaed24SMichael J. Spencer   const coff_section *Section = NULL;
1081d6167fdSMichael J. Spencer   if (error_code ec = getSection(symb->SectionNumber, Section))
1091d6167fdSMichael J. Spencer     return ec;
1101d6167fdSMichael J. Spencer   char Type;
1111d6167fdSMichael J. Spencer   if (error_code ec = getSymbolNMTypeChar(Symb, Type))
1121d6167fdSMichael J. Spencer     return ec;
1138e90adafSMichael J. Spencer   if (Type == 'U' || Type == 'w')
1141d6167fdSMichael J. Spencer     Result = UnknownAddressOrSize;
1151d6167fdSMichael J. Spencer   else if (Section)
116cbe72fc9SDanil Malyshev     Result = Section->PointerToRawData + symb->Value;
1171d6167fdSMichael J. Spencer   else
1181d6167fdSMichael J. Spencer     Result = symb->Value;
1191d6167fdSMichael J. Spencer   return object_error::success;
1208e90adafSMichael J. Spencer }
1218e90adafSMichael J. Spencer 
12275d1cf33SBenjamin Kramer error_code COFFObjectFile::getSymbolAddress(DataRefImpl Symb,
12375d1cf33SBenjamin Kramer                                             uint64_t &Result) const {
12475d1cf33SBenjamin Kramer   const coff_symbol *symb = toSymb(Symb);
12575d1cf33SBenjamin Kramer   const coff_section *Section = NULL;
12675d1cf33SBenjamin Kramer   if (error_code ec = getSection(symb->SectionNumber, Section))
12775d1cf33SBenjamin Kramer     return ec;
12875d1cf33SBenjamin Kramer   char Type;
12975d1cf33SBenjamin Kramer   if (error_code ec = getSymbolNMTypeChar(Symb, Type))
13075d1cf33SBenjamin Kramer     return ec;
13175d1cf33SBenjamin Kramer   if (Type == 'U' || Type == 'w')
13275d1cf33SBenjamin Kramer     Result = UnknownAddressOrSize;
13375d1cf33SBenjamin Kramer   else if (Section)
134cbe72fc9SDanil Malyshev     Result = Section->VirtualAddress + symb->Value;
13575d1cf33SBenjamin Kramer   else
136cbe72fc9SDanil Malyshev     Result = symb->Value;
13775d1cf33SBenjamin Kramer   return object_error::success;
13875d1cf33SBenjamin Kramer }
13975d1cf33SBenjamin Kramer 
14075d1cf33SBenjamin Kramer error_code COFFObjectFile::getSymbolType(DataRefImpl Symb,
141d3946676SMichael J. Spencer                                          SymbolRef::Type &Result) const {
14275d1cf33SBenjamin Kramer   const coff_symbol *symb = toSymb(Symb);
14375d1cf33SBenjamin Kramer   Result = SymbolRef::ST_Other;
14475d1cf33SBenjamin Kramer   if (symb->StorageClass == COFF::IMAGE_SYM_CLASS_EXTERNAL &&
14575d1cf33SBenjamin Kramer       symb->SectionNumber == COFF::IMAGE_SYM_UNDEFINED) {
1467e4b976cSDavid Meyer     Result = SymbolRef::ST_Unknown;
14775d1cf33SBenjamin Kramer   } else {
148097be9f6SMichael J. Spencer     if (symb->getComplexType() == COFF::IMAGE_SYM_DTYPE_FUNCTION) {
14975d1cf33SBenjamin Kramer       Result = SymbolRef::ST_Function;
15075d1cf33SBenjamin Kramer     } else {
15175d1cf33SBenjamin Kramer       char Type;
15275d1cf33SBenjamin Kramer       if (error_code ec = getSymbolNMTypeChar(Symb, Type))
15375d1cf33SBenjamin Kramer         return ec;
15475d1cf33SBenjamin Kramer       if (Type == 'r' || Type == 'R') {
15575d1cf33SBenjamin Kramer         Result = SymbolRef::ST_Data;
15675d1cf33SBenjamin Kramer       }
15775d1cf33SBenjamin Kramer     }
15875d1cf33SBenjamin Kramer   }
15975d1cf33SBenjamin Kramer   return object_error::success;
16075d1cf33SBenjamin Kramer }
16175d1cf33SBenjamin Kramer 
1621df4b84dSDavid Meyer error_code COFFObjectFile::getSymbolFlags(DataRefImpl Symb,
1631df4b84dSDavid Meyer                                           uint32_t &Result) const {
16475d1cf33SBenjamin Kramer   const coff_symbol *symb = toSymb(Symb);
1651df4b84dSDavid Meyer   Result = SymbolRef::SF_None;
16675d1cf33SBenjamin Kramer 
1677e4b976cSDavid Meyer   // TODO: Correctly set SF_FormatSpecific, SF_ThreadLocal, SF_Common
1687e4b976cSDavid Meyer 
1697e4b976cSDavid Meyer   if (symb->StorageClass == COFF::IMAGE_SYM_CLASS_EXTERNAL &&
1707e4b976cSDavid Meyer       symb->SectionNumber == COFF::IMAGE_SYM_UNDEFINED)
1717e4b976cSDavid Meyer     Result |= SymbolRef::SF_Undefined;
1721df4b84dSDavid Meyer 
1731df4b84dSDavid Meyer   // TODO: This are certainly too restrictive.
1741df4b84dSDavid Meyer   if (symb->StorageClass == COFF::IMAGE_SYM_CLASS_EXTERNAL)
1751df4b84dSDavid Meyer     Result |= SymbolRef::SF_Global;
1761df4b84dSDavid Meyer 
1771df4b84dSDavid Meyer   if (symb->StorageClass == COFF::IMAGE_SYM_CLASS_WEAK_EXTERNAL)
1781df4b84dSDavid Meyer     Result |= SymbolRef::SF_Weak;
1791df4b84dSDavid Meyer 
1801df4b84dSDavid Meyer   if (symb->SectionNumber == COFF::IMAGE_SYM_ABSOLUTE)
1811df4b84dSDavid Meyer     Result |= SymbolRef::SF_Absolute;
1821df4b84dSDavid Meyer 
18301759754SMichael J. Spencer   return object_error::success;
18401759754SMichael J. Spencer }
18501759754SMichael J. Spencer 
1861d6167fdSMichael J. Spencer error_code COFFObjectFile::getSymbolSize(DataRefImpl Symb,
1871d6167fdSMichael J. Spencer                                          uint64_t &Result) const {
1888e90adafSMichael J. Spencer   // FIXME: Return the correct size. This requires looking at all the symbols
1898e90adafSMichael J. Spencer   //        in the same section as this symbol, and looking for either the next
1908e90adafSMichael J. Spencer   //        symbol, or the end of the section.
1911d6167fdSMichael J. Spencer   const coff_symbol *symb = toSymb(Symb);
1925ebaed24SMichael J. Spencer   const coff_section *Section = NULL;
1931d6167fdSMichael J. Spencer   if (error_code ec = getSection(symb->SectionNumber, Section))
1941d6167fdSMichael J. Spencer     return ec;
1951d6167fdSMichael J. Spencer   char Type;
1961d6167fdSMichael J. Spencer   if (error_code ec = getSymbolNMTypeChar(Symb, Type))
1971d6167fdSMichael J. Spencer     return ec;
1988e90adafSMichael J. Spencer   if (Type == 'U' || Type == 'w')
1991d6167fdSMichael J. Spencer     Result = UnknownAddressOrSize;
2001d6167fdSMichael J. Spencer   else if (Section)
2011d6167fdSMichael J. Spencer     Result = Section->SizeOfRawData - symb->Value;
2021d6167fdSMichael J. Spencer   else
2031d6167fdSMichael J. Spencer     Result = 0;
2041d6167fdSMichael J. Spencer   return object_error::success;
2058e90adafSMichael J. Spencer }
2068e90adafSMichael J. Spencer 
2071d6167fdSMichael J. Spencer error_code COFFObjectFile::getSymbolNMTypeChar(DataRefImpl Symb,
2081d6167fdSMichael J. Spencer                                                char &Result) const {
2091d6167fdSMichael J. Spencer   const coff_symbol *symb = toSymb(Symb);
2101d6167fdSMichael J. Spencer   StringRef name;
2111d6167fdSMichael J. Spencer   if (error_code ec = getSymbolName(Symb, name))
2121d6167fdSMichael J. Spencer     return ec;
2131d6167fdSMichael J. Spencer   char ret = StringSwitch<char>(name)
2148e90adafSMichael J. Spencer     .StartsWith(".debug", 'N')
2158e90adafSMichael J. Spencer     .StartsWith(".sxdata", 'N')
2168e90adafSMichael J. Spencer     .Default('?');
2178e90adafSMichael J. Spencer 
2181d6167fdSMichael J. Spencer   if (ret != '?') {
2191d6167fdSMichael J. Spencer     Result = ret;
2201d6167fdSMichael J. Spencer     return object_error::success;
2211d6167fdSMichael J. Spencer   }
2228e90adafSMichael J. Spencer 
2238e90adafSMichael J. Spencer   uint32_t Characteristics = 0;
2241d6167fdSMichael J. Spencer   if (symb->SectionNumber > 0) {
2255ebaed24SMichael J. Spencer     const coff_section *Section = NULL;
2261d6167fdSMichael J. Spencer     if (error_code ec = getSection(symb->SectionNumber, Section))
2271d6167fdSMichael J. Spencer       return ec;
2288e90adafSMichael J. Spencer     Characteristics = Section->Characteristics;
2298e90adafSMichael J. Spencer   }
2308e90adafSMichael J. Spencer 
2318e90adafSMichael J. Spencer   switch (symb->SectionNumber) {
2328e90adafSMichael J. Spencer   case COFF::IMAGE_SYM_UNDEFINED:
2338e90adafSMichael J. Spencer     // Check storage classes.
2341d6167fdSMichael J. Spencer     if (symb->StorageClass == COFF::IMAGE_SYM_CLASS_WEAK_EXTERNAL) {
2351d6167fdSMichael J. Spencer       Result = 'w';
2361d6167fdSMichael J. Spencer       return object_error::success; // Don't do ::toupper.
237d27d51fbSMichael J. Spencer     } else if (symb->Value != 0) // Check for common symbols.
238d27d51fbSMichael J. Spencer       ret = 'c';
239d27d51fbSMichael J. Spencer     else
2408e90adafSMichael J. Spencer       ret = 'u';
2418e90adafSMichael J. Spencer     break;
2428e90adafSMichael J. Spencer   case COFF::IMAGE_SYM_ABSOLUTE:
2438e90adafSMichael J. Spencer     ret = 'a';
2448e90adafSMichael J. Spencer     break;
2458e90adafSMichael J. Spencer   case COFF::IMAGE_SYM_DEBUG:
2468e90adafSMichael J. Spencer     ret = 'n';
2478e90adafSMichael J. Spencer     break;
2488e90adafSMichael J. Spencer   default:
2498e90adafSMichael J. Spencer     // Check section type.
2508e90adafSMichael J. Spencer     if (Characteristics & COFF::IMAGE_SCN_CNT_CODE)
2518e90adafSMichael J. Spencer       ret = 't';
2528e90adafSMichael J. Spencer     else if (  Characteristics & COFF::IMAGE_SCN_MEM_READ
2538e90adafSMichael J. Spencer             && ~Characteristics & COFF::IMAGE_SCN_MEM_WRITE) // Read only.
2548e90adafSMichael J. Spencer       ret = 'r';
2558e90adafSMichael J. Spencer     else if (Characteristics & COFF::IMAGE_SCN_CNT_INITIALIZED_DATA)
2568e90adafSMichael J. Spencer       ret = 'd';
2578e90adafSMichael J. Spencer     else if (Characteristics & COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA)
2588e90adafSMichael J. Spencer       ret = 'b';
2598e90adafSMichael J. Spencer     else if (Characteristics & COFF::IMAGE_SCN_LNK_INFO)
2608e90adafSMichael J. Spencer       ret = 'i';
2618e90adafSMichael J. Spencer 
2628e90adafSMichael J. Spencer     // Check for section symbol.
2638e90adafSMichael J. Spencer     else if (  symb->StorageClass == COFF::IMAGE_SYM_CLASS_STATIC
2648e90adafSMichael J. Spencer             && symb->Value == 0)
2658e90adafSMichael J. Spencer        ret = 's';
2668e90adafSMichael J. Spencer   }
2678e90adafSMichael J. Spencer 
2688e90adafSMichael J. Spencer   if (symb->StorageClass == COFF::IMAGE_SYM_CLASS_EXTERNAL)
2698e90adafSMichael J. Spencer     ret = ::toupper(ret);
2708e90adafSMichael J. Spencer 
2711d6167fdSMichael J. Spencer   Result = ret;
2721d6167fdSMichael J. Spencer   return object_error::success;
2738e90adafSMichael J. Spencer }
2748e90adafSMichael J. Spencer 
27532173153SMichael J. Spencer error_code COFFObjectFile::getSymbolSection(DataRefImpl Symb,
27632173153SMichael J. Spencer                                             section_iterator &Result) const {
27732173153SMichael J. Spencer   const coff_symbol *symb = toSymb(Symb);
27832173153SMichael J. Spencer   if (symb->SectionNumber <= COFF::IMAGE_SYM_UNDEFINED)
27932173153SMichael J. Spencer     Result = end_sections();
28032173153SMichael J. Spencer   else {
2819339d455SDaniel Dunbar     const coff_section *sec = 0;
28232173153SMichael J. Spencer     if (error_code ec = getSection(symb->SectionNumber, sec)) return ec;
28332173153SMichael J. Spencer     DataRefImpl Sec;
284d0f5b0e1SMichael J. Spencer     std::memset(&Sec, 0, sizeof(Sec));
28532173153SMichael J. Spencer     Sec.p = reinterpret_cast<uintptr_t>(sec);
28632173153SMichael J. Spencer     Result = section_iterator(SectionRef(Sec, this));
28732173153SMichael J. Spencer   }
28832173153SMichael J. Spencer   return object_error::success;
28932173153SMichael J. Spencer }
29032173153SMichael J. Spencer 
2911d6167fdSMichael J. Spencer error_code COFFObjectFile::getSectionNext(DataRefImpl Sec,
2921d6167fdSMichael J. Spencer                                           SectionRef &Result) const {
2931d6167fdSMichael J. Spencer   const coff_section *sec = toSec(Sec);
2948e90adafSMichael J. Spencer   sec += 1;
2951d6167fdSMichael J. Spencer   Sec.p = reinterpret_cast<uintptr_t>(sec);
2961d6167fdSMichael J. Spencer   Result = SectionRef(Sec, this);
2971d6167fdSMichael J. Spencer   return object_error::success;
2988e90adafSMichael J. Spencer }
2998e90adafSMichael J. Spencer 
3001d6167fdSMichael J. Spencer error_code COFFObjectFile::getSectionName(DataRefImpl Sec,
3011d6167fdSMichael J. Spencer                                           StringRef &Result) const {
3021d6167fdSMichael J. Spencer   const coff_section *sec = toSec(Sec);
3038e90adafSMichael J. Spencer   StringRef name;
3048e90adafSMichael J. Spencer   if (sec->Name[7] == 0)
3058e90adafSMichael J. Spencer     // Null terminated, let ::strlen figure out the length.
3068e90adafSMichael J. Spencer     name = sec->Name;
3078e90adafSMichael J. Spencer   else
3088e90adafSMichael J. Spencer     // Not null terminated, use all 8 bytes.
3098e90adafSMichael J. Spencer     name = StringRef(sec->Name, 8);
3108e90adafSMichael J. Spencer 
3118e90adafSMichael J. Spencer   // Check for string table entry. First byte is '/'.
3128e90adafSMichael J. Spencer   if (name[0] == '/') {
3138e90adafSMichael J. Spencer     uint32_t Offset;
314ee066fc4SEric Christopher     name.substr(1).getAsInteger(10, Offset);
3151d6167fdSMichael J. Spencer     if (error_code ec = getString(Offset, name))
3161d6167fdSMichael J. Spencer       return ec;
3178e90adafSMichael J. Spencer   }
3188e90adafSMichael J. Spencer 
3191d6167fdSMichael J. Spencer   Result = name;
3201d6167fdSMichael J. Spencer   return object_error::success;
3218e90adafSMichael J. Spencer }
3228e90adafSMichael J. Spencer 
3231d6167fdSMichael J. Spencer error_code COFFObjectFile::getSectionAddress(DataRefImpl Sec,
3241d6167fdSMichael J. Spencer                                              uint64_t &Result) const {
3251d6167fdSMichael J. Spencer   const coff_section *sec = toSec(Sec);
3261d6167fdSMichael J. Spencer   Result = sec->VirtualAddress;
3271d6167fdSMichael J. Spencer   return object_error::success;
3288e90adafSMichael J. Spencer }
3298e90adafSMichael J. Spencer 
3301d6167fdSMichael J. Spencer error_code COFFObjectFile::getSectionSize(DataRefImpl Sec,
3311d6167fdSMichael J. Spencer                                           uint64_t &Result) const {
3321d6167fdSMichael J. Spencer   const coff_section *sec = toSec(Sec);
3331d6167fdSMichael J. Spencer   Result = sec->SizeOfRawData;
3341d6167fdSMichael J. Spencer   return object_error::success;
3358e90adafSMichael J. Spencer }
3368e90adafSMichael J. Spencer 
3371d6167fdSMichael J. Spencer error_code COFFObjectFile::getSectionContents(DataRefImpl Sec,
3381d6167fdSMichael J. Spencer                                               StringRef &Result) const {
3391d6167fdSMichael J. Spencer   const coff_section *sec = toSec(Sec);
3401d6167fdSMichael J. Spencer   // The only thing that we need to verify is that the contents is contained
3411d6167fdSMichael J. Spencer   // within the file bounds. We don't need to make sure it doesn't cover other
3421d6167fdSMichael J. Spencer   // data, as there's nothing that says that is not allowed.
3431d6167fdSMichael J. Spencer   uintptr_t con_start = uintptr_t(base()) + sec->PointerToRawData;
3441d6167fdSMichael J. Spencer   uintptr_t con_end = con_start + sec->SizeOfRawData;
345d5930ca9SMichael J. Spencer   if (con_end > uintptr_t(Data->getBufferEnd()))
3461d6167fdSMichael J. Spencer     return object_error::parse_failed;
3471d6167fdSMichael J. Spencer   Result = StringRef(reinterpret_cast<const char*>(con_start),
3488e90adafSMichael J. Spencer                      sec->SizeOfRawData);
3491d6167fdSMichael J. Spencer   return object_error::success;
3508e90adafSMichael J. Spencer }
3518e90adafSMichael J. Spencer 
3527989460aSMichael J. Spencer error_code COFFObjectFile::getSectionAlignment(DataRefImpl Sec,
3537989460aSMichael J. Spencer                                                uint64_t &Res) const {
3547989460aSMichael J. Spencer   const coff_section *sec = toSec(Sec);
3557989460aSMichael J. Spencer   if (!sec)
3567989460aSMichael J. Spencer     return object_error::parse_failed;
3577989460aSMichael J. Spencer   Res = uint64_t(1) << (((sec->Characteristics & 0x00F00000) >> 20) - 1);
3587989460aSMichael J. Spencer   return object_error::success;
3597989460aSMichael J. Spencer }
3607989460aSMichael J. Spencer 
3611d6167fdSMichael J. Spencer error_code COFFObjectFile::isSectionText(DataRefImpl Sec,
3621d6167fdSMichael J. Spencer                                          bool &Result) const {
3631d6167fdSMichael J. Spencer   const coff_section *sec = toSec(Sec);
3641d6167fdSMichael J. Spencer   Result = sec->Characteristics & COFF::IMAGE_SCN_CNT_CODE;
3651d6167fdSMichael J. Spencer   return object_error::success;
3668e90adafSMichael J. Spencer }
3678e90adafSMichael J. Spencer 
368800619f2SMichael J. Spencer error_code COFFObjectFile::isSectionData(DataRefImpl Sec,
369800619f2SMichael J. Spencer                                          bool &Result) const {
370800619f2SMichael J. Spencer   const coff_section *sec = toSec(Sec);
371800619f2SMichael J. Spencer   Result = sec->Characteristics & COFF::IMAGE_SCN_CNT_INITIALIZED_DATA;
372800619f2SMichael J. Spencer   return object_error::success;
373800619f2SMichael J. Spencer }
374800619f2SMichael J. Spencer 
375800619f2SMichael J. Spencer error_code COFFObjectFile::isSectionBSS(DataRefImpl Sec,
376800619f2SMichael J. Spencer                                         bool &Result) const {
377800619f2SMichael J. Spencer   const coff_section *sec = toSec(Sec);
378800619f2SMichael J. Spencer   Result = sec->Characteristics & COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA;
379800619f2SMichael J. Spencer   return object_error::success;
380800619f2SMichael J. Spencer }
381800619f2SMichael J. Spencer 
382f6f3e81cSBenjamin Kramer error_code COFFObjectFile::sectionContainsSymbol(DataRefImpl Sec,
383f6f3e81cSBenjamin Kramer                                                  DataRefImpl Symb,
384f6f3e81cSBenjamin Kramer                                                  bool &Result) const {
3859a28851eSMichael J. Spencer   const coff_section *sec = toSec(Sec);
3869a28851eSMichael J. Spencer   const coff_symbol *symb = toSymb(Symb);
3879339d455SDaniel Dunbar   const coff_section *symb_sec = 0;
3889a28851eSMichael J. Spencer   if (error_code ec = getSection(symb->SectionNumber, symb_sec)) return ec;
3899a28851eSMichael J. Spencer   if (symb_sec == sec)
3909a28851eSMichael J. Spencer     Result = true;
3919a28851eSMichael J. Spencer   else
392f6f3e81cSBenjamin Kramer     Result = false;
393f6f3e81cSBenjamin Kramer   return object_error::success;
394f6f3e81cSBenjamin Kramer }
395f6f3e81cSBenjamin Kramer 
396e5fd0047SMichael J. Spencer relocation_iterator COFFObjectFile::getSectionRelBegin(DataRefImpl Sec) const {
397e5fd0047SMichael J. Spencer   const coff_section *sec = toSec(Sec);
398e5fd0047SMichael J. Spencer   DataRefImpl ret;
399e5fd0047SMichael J. Spencer   std::memset(&ret, 0, sizeof(ret));
400e5fd0047SMichael J. Spencer   if (sec->NumberOfRelocations == 0)
401e5fd0047SMichael J. Spencer     ret.p = 0;
402e5fd0047SMichael J. Spencer   else
403e5fd0047SMichael J. Spencer     ret.p = reinterpret_cast<uintptr_t>(base() + sec->PointerToRelocations);
404e5fd0047SMichael J. Spencer 
405e5fd0047SMichael J. Spencer   return relocation_iterator(RelocationRef(ret, this));
406e5fd0047SMichael J. Spencer }
407e5fd0047SMichael J. Spencer 
408e5fd0047SMichael J. Spencer relocation_iterator COFFObjectFile::getSectionRelEnd(DataRefImpl Sec) const {
409e5fd0047SMichael J. Spencer   const coff_section *sec = toSec(Sec);
410e5fd0047SMichael J. Spencer   DataRefImpl ret;
411e5fd0047SMichael J. Spencer   std::memset(&ret, 0, sizeof(ret));
412e5fd0047SMichael J. Spencer   if (sec->NumberOfRelocations == 0)
413e5fd0047SMichael J. Spencer     ret.p = 0;
414e5fd0047SMichael J. Spencer   else
415e5fd0047SMichael J. Spencer     ret.p = reinterpret_cast<uintptr_t>(
416e5fd0047SMichael J. Spencer               reinterpret_cast<const coff_relocation*>(
417e5fd0047SMichael J. Spencer                 base() + sec->PointerToRelocations)
418e5fd0047SMichael J. Spencer               + sec->NumberOfRelocations);
419e5fd0047SMichael J. Spencer 
420e5fd0047SMichael J. Spencer   return relocation_iterator(RelocationRef(ret, this));
421e5fd0047SMichael J. Spencer }
422e5fd0047SMichael J. Spencer 
423ec29b121SMichael J. Spencer COFFObjectFile::COFFObjectFile(MemoryBuffer *Object, error_code &ec)
424d5930ca9SMichael J. Spencer   : ObjectFile(Binary::isCOFF, Object, ec)
425d5930ca9SMichael J. Spencer   , Header(0)
426d5930ca9SMichael J. Spencer   , SectionTable(0)
427d5930ca9SMichael J. Spencer   , SymbolTable(0)
428d5930ca9SMichael J. Spencer   , StringTable(0)
429d5930ca9SMichael J. Spencer   , StringTableSize(0) {
4301d6167fdSMichael J. Spencer   // Check that we at least have enough room for a header.
4311d6167fdSMichael J. Spencer   if (!checkSize(Data, ec, sizeof(coff_file_header))) return;
432ee066fc4SEric Christopher 
4331d6167fdSMichael J. Spencer   // The actual starting location of the COFF header in the file. This can be
4341d6167fdSMichael J. Spencer   // non-zero in PE/COFF files.
4351d6167fdSMichael J. Spencer   uint64_t HeaderStart = 0;
436ee066fc4SEric Christopher 
4371d6167fdSMichael J. Spencer   // Check if this is a PE/COFF file.
438ec29b121SMichael J. Spencer   if (base()[0] == 0x4d && base()[1] == 0x5a) {
439ee066fc4SEric Christopher     // PE/COFF, seek through MS-DOS compatibility stub and 4-byte
440ee066fc4SEric Christopher     // PE signature to find 'normal' COFF header.
4411d6167fdSMichael J. Spencer     if (!checkSize(Data, ec, 0x3c + 8)) return;
442d5930ca9SMichael J. Spencer     HeaderStart = *reinterpret_cast<const ulittle16_t *>(base() + 0x3c);
4431d6167fdSMichael J. Spencer     // Check the PE header. ("PE\0\0")
444e6490e03SBenjamin Kramer     if (std::memcmp(base() + HeaderStart, "PE\0\0", 4) != 0) {
4451d6167fdSMichael J. Spencer       ec = object_error::parse_failed;
4461d6167fdSMichael J. Spencer       return;
4471d6167fdSMichael J. Spencer     }
4481d6167fdSMichael J. Spencer     HeaderStart += 4; // Skip the PE Header.
449ee066fc4SEric Christopher   }
450ee066fc4SEric Christopher 
4511d6167fdSMichael J. Spencer   Header = reinterpret_cast<const coff_file_header *>(base() + HeaderStart);
4521d6167fdSMichael J. Spencer   if (!checkAddr(Data, ec, uintptr_t(Header), sizeof(coff_file_header)))
4531d6167fdSMichael J. Spencer     return;
4541d6167fdSMichael J. Spencer 
4558e90adafSMichael J. Spencer   SectionTable =
456ec29b121SMichael J. Spencer     reinterpret_cast<const coff_section *>( base()
4571d6167fdSMichael J. Spencer                                           + HeaderStart
4588e90adafSMichael J. Spencer                                           + sizeof(coff_file_header)
4598e90adafSMichael J. Spencer                                           + Header->SizeOfOptionalHeader);
4601d6167fdSMichael J. Spencer   if (!checkAddr(Data, ec, uintptr_t(SectionTable),
4611d6167fdSMichael J. Spencer                  Header->NumberOfSections * sizeof(coff_section)))
4621d6167fdSMichael J. Spencer     return;
4631d6167fdSMichael J. Spencer 
464d5930ca9SMichael J. Spencer   if (Header->PointerToSymbolTable != 0) {
4658e90adafSMichael J. Spencer     SymbolTable =
466ec29b121SMichael J. Spencer       reinterpret_cast<const coff_symbol *>(base()
467ec29b121SMichael J. Spencer                                             + Header->PointerToSymbolTable);
4681d6167fdSMichael J. Spencer     if (!checkAddr(Data, ec, uintptr_t(SymbolTable),
4691d6167fdSMichael J. Spencer                    Header->NumberOfSymbols * sizeof(coff_symbol)))
4701d6167fdSMichael J. Spencer       return;
4718e90adafSMichael J. Spencer 
4728e90adafSMichael J. Spencer     // Find string table.
473ec29b121SMichael J. Spencer     StringTable = reinterpret_cast<const char *>(base())
4748e90adafSMichael J. Spencer                   + Header->PointerToSymbolTable
4751d6167fdSMichael J. Spencer                   + Header->NumberOfSymbols * sizeof(coff_symbol);
4761d6167fdSMichael J. Spencer     if (!checkAddr(Data, ec, uintptr_t(StringTable), sizeof(ulittle32_t)))
4771d6167fdSMichael J. Spencer       return;
4781d6167fdSMichael J. Spencer 
4791d6167fdSMichael J. Spencer     StringTableSize = *reinterpret_cast<const ulittle32_t *>(StringTable);
4801d6167fdSMichael J. Spencer     if (!checkAddr(Data, ec, uintptr_t(StringTable), StringTableSize))
4811d6167fdSMichael J. Spencer       return;
4821d6167fdSMichael J. Spencer     // Check that the string table is null terminated if has any in it.
4831d6167fdSMichael J. Spencer     if (StringTableSize < 4
4841d6167fdSMichael J. Spencer         || (StringTableSize > 4 && StringTable[StringTableSize - 1] != 0)) {
4851d6167fdSMichael J. Spencer       ec = object_error::parse_failed;
4861d6167fdSMichael J. Spencer       return;
4871d6167fdSMichael J. Spencer     }
488d5930ca9SMichael J. Spencer   }
4891d6167fdSMichael J. Spencer 
4901d6167fdSMichael J. Spencer   ec = object_error::success;
4918e90adafSMichael J. Spencer }
4928e90adafSMichael J. Spencer 
493e5fd0047SMichael J. Spencer symbol_iterator COFFObjectFile::begin_symbols() const {
4940324b672SMichael J. Spencer   DataRefImpl ret;
4951d6167fdSMichael J. Spencer   std::memset(&ret, 0, sizeof(DataRefImpl));
4960324b672SMichael J. Spencer   ret.p = reinterpret_cast<intptr_t>(SymbolTable);
4970324b672SMichael J. Spencer   return symbol_iterator(SymbolRef(ret, this));
4988e90adafSMichael J. Spencer }
4998e90adafSMichael J. Spencer 
500e5fd0047SMichael J. Spencer symbol_iterator COFFObjectFile::end_symbols() const {
5018e90adafSMichael J. Spencer   // The symbol table ends where the string table begins.
5020324b672SMichael J. Spencer   DataRefImpl ret;
5031d6167fdSMichael J. Spencer   std::memset(&ret, 0, sizeof(DataRefImpl));
5040324b672SMichael J. Spencer   ret.p = reinterpret_cast<intptr_t>(StringTable);
5050324b672SMichael J. Spencer   return symbol_iterator(SymbolRef(ret, this));
5068e90adafSMichael J. Spencer }
5078e90adafSMichael J. Spencer 
5088c4729fdSMichael J. Spencer symbol_iterator COFFObjectFile::begin_dynamic_symbols() const {
5098c4729fdSMichael J. Spencer   // TODO: implement
5108c4729fdSMichael J. Spencer   report_fatal_error("Dynamic symbols unimplemented in COFFObjectFile");
5118c4729fdSMichael J. Spencer }
5128c4729fdSMichael J. Spencer 
5138c4729fdSMichael J. Spencer symbol_iterator COFFObjectFile::end_dynamic_symbols() const {
5148c4729fdSMichael J. Spencer   // TODO: implement
5158c4729fdSMichael J. Spencer   report_fatal_error("Dynamic symbols unimplemented in COFFObjectFile");
5168c4729fdSMichael J. Spencer }
5178c4729fdSMichael J. Spencer 
518*2fc34c5fSDavid Meyer library_iterator COFFObjectFile::begin_libraries_needed() const {
519*2fc34c5fSDavid Meyer   // TODO: implement
520*2fc34c5fSDavid Meyer   report_fatal_error("Libraries needed unimplemented in COFFObjectFile");
521*2fc34c5fSDavid Meyer }
522*2fc34c5fSDavid Meyer 
523*2fc34c5fSDavid Meyer library_iterator COFFObjectFile::end_libraries_needed() const {
524*2fc34c5fSDavid Meyer   // TODO: implement
525*2fc34c5fSDavid Meyer   report_fatal_error("Libraries needed unimplemented in COFFObjectFile");
526*2fc34c5fSDavid Meyer }
527*2fc34c5fSDavid Meyer 
528e5fd0047SMichael J. Spencer section_iterator COFFObjectFile::begin_sections() const {
5290324b672SMichael J. Spencer   DataRefImpl ret;
5301d6167fdSMichael J. Spencer   std::memset(&ret, 0, sizeof(DataRefImpl));
5310324b672SMichael J. Spencer   ret.p = reinterpret_cast<intptr_t>(SectionTable);
5320324b672SMichael J. Spencer   return section_iterator(SectionRef(ret, this));
5338e90adafSMichael J. Spencer }
5348e90adafSMichael J. Spencer 
535e5fd0047SMichael J. Spencer section_iterator COFFObjectFile::end_sections() const {
5360324b672SMichael J. Spencer   DataRefImpl ret;
5371d6167fdSMichael J. Spencer   std::memset(&ret, 0, sizeof(DataRefImpl));
5380324b672SMichael J. Spencer   ret.p = reinterpret_cast<intptr_t>(SectionTable + Header->NumberOfSections);
5390324b672SMichael J. Spencer   return section_iterator(SectionRef(ret, this));
5408e90adafSMichael J. Spencer }
5418e90adafSMichael J. Spencer 
5428e90adafSMichael J. Spencer uint8_t COFFObjectFile::getBytesInAddress() const {
5430324b672SMichael J. Spencer   return getArch() == Triple::x86_64 ? 8 : 4;
5448e90adafSMichael J. Spencer }
5458e90adafSMichael J. Spencer 
5468e90adafSMichael J. Spencer StringRef COFFObjectFile::getFileFormatName() const {
5478e90adafSMichael J. Spencer   switch(Header->Machine) {
5488e90adafSMichael J. Spencer   case COFF::IMAGE_FILE_MACHINE_I386:
5498e90adafSMichael J. Spencer     return "COFF-i386";
5508e90adafSMichael J. Spencer   case COFF::IMAGE_FILE_MACHINE_AMD64:
5518e90adafSMichael J. Spencer     return "COFF-x86-64";
5528e90adafSMichael J. Spencer   default:
5538e90adafSMichael J. Spencer     return "COFF-<unknown arch>";
5548e90adafSMichael J. Spencer   }
5558e90adafSMichael J. Spencer }
5568e90adafSMichael J. Spencer 
5578e90adafSMichael J. Spencer unsigned COFFObjectFile::getArch() const {
5588e90adafSMichael J. Spencer   switch(Header->Machine) {
5598e90adafSMichael J. Spencer   case COFF::IMAGE_FILE_MACHINE_I386:
5608e90adafSMichael J. Spencer     return Triple::x86;
5618e90adafSMichael J. Spencer   case COFF::IMAGE_FILE_MACHINE_AMD64:
5628e90adafSMichael J. Spencer     return Triple::x86_64;
5638e90adafSMichael J. Spencer   default:
5648e90adafSMichael J. Spencer     return Triple::UnknownArch;
5658e90adafSMichael J. Spencer   }
5668e90adafSMichael J. Spencer }
5678e90adafSMichael J. Spencer 
56889a7a5eaSMichael J. Spencer error_code COFFObjectFile::getHeader(const coff_file_header *&Res) const {
56989a7a5eaSMichael J. Spencer   Res = Header;
57089a7a5eaSMichael J. Spencer   return object_error::success;
57189a7a5eaSMichael J. Spencer }
57289a7a5eaSMichael J. Spencer 
5731d6167fdSMichael J. Spencer error_code COFFObjectFile::getSection(int32_t index,
5741d6167fdSMichael J. Spencer                                       const coff_section *&Result) const {
5751d6167fdSMichael J. Spencer   // Check for special index values.
5761d6167fdSMichael J. Spencer   if (index == COFF::IMAGE_SYM_UNDEFINED ||
5771d6167fdSMichael J. Spencer       index == COFF::IMAGE_SYM_ABSOLUTE ||
5781d6167fdSMichael J. Spencer       index == COFF::IMAGE_SYM_DEBUG)
5791d6167fdSMichael J. Spencer     Result = NULL;
5801d6167fdSMichael J. Spencer   else if (index > 0 && index <= Header->NumberOfSections)
5811d6167fdSMichael J. Spencer     // We already verified the section table data, so no need to check again.
5821d6167fdSMichael J. Spencer     Result = SectionTable + (index - 1);
5831d6167fdSMichael J. Spencer   else
5841d6167fdSMichael J. Spencer     return object_error::parse_failed;
5851d6167fdSMichael J. Spencer   return object_error::success;
5868e90adafSMichael J. Spencer }
5878e90adafSMichael J. Spencer 
5881d6167fdSMichael J. Spencer error_code COFFObjectFile::getString(uint32_t offset,
5891d6167fdSMichael J. Spencer                                      StringRef &Result) const {
5901d6167fdSMichael J. Spencer   if (StringTableSize <= 4)
5911d6167fdSMichael J. Spencer     // Tried to get a string from an empty string table.
5921d6167fdSMichael J. Spencer     return object_error::parse_failed;
5931d6167fdSMichael J. Spencer   if (offset >= StringTableSize)
5941d6167fdSMichael J. Spencer     return object_error::unexpected_eof;
5951d6167fdSMichael J. Spencer   Result = StringRef(StringTable + offset);
5961d6167fdSMichael J. Spencer   return object_error::success;
5978e90adafSMichael J. Spencer }
598022ecdf2SBenjamin Kramer 
599e5fd0047SMichael J. Spencer error_code COFFObjectFile::getSymbol(uint32_t index,
600e5fd0047SMichael J. Spencer                                      const coff_symbol *&Result) const {
601443410d7SMichael J. Spencer   if (index < Header->NumberOfSymbols)
602e5fd0047SMichael J. Spencer     Result = SymbolTable + index;
603e5fd0047SMichael J. Spencer   else
604e5fd0047SMichael J. Spencer     return object_error::parse_failed;
605e5fd0047SMichael J. Spencer   return object_error::success;
606e5fd0047SMichael J. Spencer }
607e5fd0047SMichael J. Spencer 
60889a7a5eaSMichael J. Spencer error_code COFFObjectFile::getSymbolName(const coff_symbol *symbol,
60989a7a5eaSMichael J. Spencer                                          StringRef &Res) const {
61089a7a5eaSMichael J. Spencer   // Check for string table entry. First 4 bytes are 0.
61189a7a5eaSMichael J. Spencer   if (symbol->Name.Offset.Zeroes == 0) {
61289a7a5eaSMichael J. Spencer     uint32_t Offset = symbol->Name.Offset.Offset;
61389a7a5eaSMichael J. Spencer     if (error_code ec = getString(Offset, Res))
61489a7a5eaSMichael J. Spencer       return ec;
61589a7a5eaSMichael J. Spencer     return object_error::success;
61689a7a5eaSMichael J. Spencer   }
61789a7a5eaSMichael J. Spencer 
61889a7a5eaSMichael J. Spencer   if (symbol->Name.ShortName[7] == 0)
61989a7a5eaSMichael J. Spencer     // Null terminated, let ::strlen figure out the length.
62089a7a5eaSMichael J. Spencer     Res = StringRef(symbol->Name.ShortName);
62189a7a5eaSMichael J. Spencer   else
62289a7a5eaSMichael J. Spencer     // Not null terminated, use all 8 bytes.
62389a7a5eaSMichael J. Spencer     Res = StringRef(symbol->Name.ShortName, 8);
62489a7a5eaSMichael J. Spencer   return object_error::success;
62589a7a5eaSMichael J. Spencer }
62689a7a5eaSMichael J. Spencer 
627022ecdf2SBenjamin Kramer const coff_relocation *COFFObjectFile::toRel(DataRefImpl Rel) const {
628e5fd0047SMichael J. Spencer   return reinterpret_cast<const coff_relocation*>(Rel.p);
629022ecdf2SBenjamin Kramer }
630022ecdf2SBenjamin Kramer error_code COFFObjectFile::getRelocationNext(DataRefImpl Rel,
631022ecdf2SBenjamin Kramer                                              RelocationRef &Res) const {
632e5fd0047SMichael J. Spencer   Rel.p = reinterpret_cast<uintptr_t>(
633e5fd0047SMichael J. Spencer             reinterpret_cast<const coff_relocation*>(Rel.p) + 1);
634022ecdf2SBenjamin Kramer   Res = RelocationRef(Rel, this);
635022ecdf2SBenjamin Kramer   return object_error::success;
636022ecdf2SBenjamin Kramer }
637022ecdf2SBenjamin Kramer error_code COFFObjectFile::getRelocationAddress(DataRefImpl Rel,
638022ecdf2SBenjamin Kramer                                                 uint64_t &Res) const {
639e5fd0047SMichael J. Spencer   Res = toRel(Rel)->VirtualAddress;
640022ecdf2SBenjamin Kramer   return object_error::success;
641022ecdf2SBenjamin Kramer }
642cbe72fc9SDanil Malyshev error_code COFFObjectFile::getRelocationOffset(DataRefImpl Rel,
643cbe72fc9SDanil Malyshev                                                uint64_t &Res) const {
644cbe72fc9SDanil Malyshev   Res = toRel(Rel)->VirtualAddress;
645cbe72fc9SDanil Malyshev   return object_error::success;
646cbe72fc9SDanil Malyshev }
647022ecdf2SBenjamin Kramer error_code COFFObjectFile::getRelocationSymbol(DataRefImpl Rel,
648022ecdf2SBenjamin Kramer                                                SymbolRef &Res) const {
649022ecdf2SBenjamin Kramer   const coff_relocation* R = toRel(Rel);
650022ecdf2SBenjamin Kramer   DataRefImpl Symb;
651d0f5b0e1SMichael J. Spencer   std::memset(&Symb, 0, sizeof(Symb));
652022ecdf2SBenjamin Kramer   Symb.p = reinterpret_cast<uintptr_t>(SymbolTable + R->SymbolTableIndex);
653022ecdf2SBenjamin Kramer   Res = SymbolRef(Symb, this);
654022ecdf2SBenjamin Kramer   return object_error::success;
655022ecdf2SBenjamin Kramer }
656022ecdf2SBenjamin Kramer error_code COFFObjectFile::getRelocationType(DataRefImpl Rel,
6577be76590SOwen Anderson                                              uint64_t &Res) const {
658022ecdf2SBenjamin Kramer   const coff_relocation* R = toRel(Rel);
659022ecdf2SBenjamin Kramer   Res = R->Type;
660022ecdf2SBenjamin Kramer   return object_error::success;
661022ecdf2SBenjamin Kramer }
662e5fd0047SMichael J. Spencer 
663e5fd0047SMichael J. Spencer #define LLVM_COFF_SWITCH_RELOC_TYPE_NAME(enum) \
664e5fd0047SMichael J. Spencer   case COFF::enum: res = #enum; break;
665e5fd0047SMichael J. Spencer 
666e5fd0047SMichael J. Spencer error_code COFFObjectFile::getRelocationTypeName(DataRefImpl Rel,
667e5fd0047SMichael J. Spencer                                           SmallVectorImpl<char> &Result) const {
668e5fd0047SMichael J. Spencer   const coff_relocation *reloc = toRel(Rel);
669e5fd0047SMichael J. Spencer   StringRef res;
670e5fd0047SMichael J. Spencer   switch (Header->Machine) {
671e5fd0047SMichael J. Spencer   case COFF::IMAGE_FILE_MACHINE_AMD64:
672e5fd0047SMichael J. Spencer     switch (reloc->Type) {
673e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_ABSOLUTE);
674e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_ADDR64);
675e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_ADDR32);
676e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_ADDR32NB);
677e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_REL32);
678e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_REL32_1);
679e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_REL32_2);
680e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_REL32_3);
681e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_REL32_4);
682e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_REL32_5);
683e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_SECTION);
684e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_SECREL);
685e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_SECREL7);
686e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_TOKEN);
687e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_SREL32);
688e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_PAIR);
689e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_AMD64_SSPAN32);
690e5fd0047SMichael J. Spencer     default:
691e5fd0047SMichael J. Spencer       res = "Unknown";
692e5fd0047SMichael J. Spencer     }
693e5fd0047SMichael J. Spencer     break;
694e5fd0047SMichael J. Spencer   case COFF::IMAGE_FILE_MACHINE_I386:
695e5fd0047SMichael J. Spencer     switch (reloc->Type) {
696e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_ABSOLUTE);
697e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_DIR16);
698e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_REL16);
699e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_DIR32);
700e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_DIR32NB);
701e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_SEG12);
702e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_SECTION);
703e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_SECREL);
704e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_TOKEN);
705e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_SECREL7);
706e5fd0047SMichael J. Spencer     LLVM_COFF_SWITCH_RELOC_TYPE_NAME(IMAGE_REL_I386_REL32);
707e5fd0047SMichael J. Spencer     default:
708e5fd0047SMichael J. Spencer       res = "Unknown";
709e5fd0047SMichael J. Spencer     }
710e5fd0047SMichael J. Spencer     break;
711e5fd0047SMichael J. Spencer   default:
712e5fd0047SMichael J. Spencer     res = "Unknown";
713e5fd0047SMichael J. Spencer   }
714e5fd0047SMichael J. Spencer   Result.append(res.begin(), res.end());
715e5fd0047SMichael J. Spencer   return object_error::success;
716e5fd0047SMichael J. Spencer }
717e5fd0047SMichael J. Spencer 
718e5fd0047SMichael J. Spencer #undef LLVM_COFF_SWITCH_RELOC_TYPE_NAME
719e5fd0047SMichael J. Spencer 
720022ecdf2SBenjamin Kramer error_code COFFObjectFile::getRelocationAdditionalInfo(DataRefImpl Rel,
721022ecdf2SBenjamin Kramer                                                        int64_t &Res) const {
722022ecdf2SBenjamin Kramer   Res = 0;
723022ecdf2SBenjamin Kramer   return object_error::success;
724022ecdf2SBenjamin Kramer }
725e5fd0047SMichael J. Spencer error_code COFFObjectFile::getRelocationValueString(DataRefImpl Rel,
726e5fd0047SMichael J. Spencer                                           SmallVectorImpl<char> &Result) const {
727e5fd0047SMichael J. Spencer   const coff_relocation *reloc = toRel(Rel);
728648b2fafSNAKAMURA Takumi   const coff_symbol *symb = 0;
729e5fd0047SMichael J. Spencer   if (error_code ec = getSymbol(reloc->SymbolTableIndex, symb)) return ec;
730e5fd0047SMichael J. Spencer   DataRefImpl sym;
731e5fd0047SMichael J. Spencer   ::memset(&sym, 0, sizeof(sym));
732e5fd0047SMichael J. Spencer   sym.p = reinterpret_cast<uintptr_t>(symb);
733e5fd0047SMichael J. Spencer   StringRef symname;
734e5fd0047SMichael J. Spencer   if (error_code ec = getSymbolName(sym, symname)) return ec;
735e5fd0047SMichael J. Spencer   Result.append(symname.begin(), symname.end());
736e5fd0047SMichael J. Spencer   return object_error::success;
737022ecdf2SBenjamin Kramer }
7388e90adafSMichael J. Spencer 
739*2fc34c5fSDavid Meyer error_code COFFObjectFile::getLibraryNext(DataRefImpl LibData,
740*2fc34c5fSDavid Meyer                                           LibraryRef &Result) const {
741*2fc34c5fSDavid Meyer   report_fatal_error("getLibraryNext not implemented in COFFObjectFile");
742*2fc34c5fSDavid Meyer }
743*2fc34c5fSDavid Meyer 
744*2fc34c5fSDavid Meyer error_code COFFObjectFile::getLibraryPath(DataRefImpl LibData,
745*2fc34c5fSDavid Meyer                                           StringRef &Result) const {
746*2fc34c5fSDavid Meyer   report_fatal_error("getLibraryPath not implemented in COFFObjectFile");
747*2fc34c5fSDavid Meyer }
748*2fc34c5fSDavid Meyer 
7498e90adafSMichael J. Spencer namespace llvm {
7508e90adafSMichael J. Spencer 
7518e90adafSMichael J. Spencer   ObjectFile *ObjectFile::createCOFFObjectFile(MemoryBuffer *Object) {
752ec29b121SMichael J. Spencer     error_code ec;
753ec29b121SMichael J. Spencer     return new COFFObjectFile(Object, ec);
7548e90adafSMichael J. Spencer   }
7558e90adafSMichael J. Spencer 
7568e90adafSMichael J. Spencer } // end namespace llvm
757