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