X86PLT.cpp revision 5460a1f25d9ddecb5c70667267d66d51af177a99
15460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//===- X86PLT.cpp -----------------------------------------------------------===// 25460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 35460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// The MCLinker Project 45460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 55460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// This file is distributed under the University of Illinois Open Source 65460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// License. See LICENSE.TXT for details. 75460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// 85460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//===----------------------------------------------------------------------===// 95460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include "X86GOT.h" 105460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include "X86PLT.h" 115460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <llvm/Support/raw_ostream.h> 125460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <llvm/Support/ErrorHandling.h> 135460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <llvm/Support/ELF.h> 145460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <mcld/MC/MCLDOutput.h> 155460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao#include <new> 165460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 175460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaonamespace { 185460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 195460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoconst uint8_t x86_dyn_plt0[] = { 205460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xff, 0xb3, 0x04, 0, 0, 0, // pushl 0x4(%ebx) 215460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xff, 0xa3, 0x08, 0, 0, 0, // jmp *0x8(%ebx) 225460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xf, 0x1f, 0x4, 0 // nopl 0(%eax) 235460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao}; 245460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 255460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoconst uint8_t x86_dyn_plt1[] = { 265460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xff, 0xa3, 0, 0, 0, 0, // jmp *sym@GOT(%ebx) 275460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0x68, 0, 0, 0, 0, // pushl $offset 285460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xe9, 0, 0, 0, 0 // jmp plt0 295460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao}; 305460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 315460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoconst uint8_t x86_exec_plt0[] = { 325460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xff, 0x35, 0, 0, 0, 0, // pushl .got + 4 335460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xff, 0x25, 0, 0, 0, 0, // jmp *(.got + 8) 345460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xf, 0x1f, 0x4, 0 // nopl 0(%eax) 355460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao}; 365460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 375460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaoconst uint8_t x86_exec_plt1[] = { 385460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xff, 0x25, 0, 0, 0, 0, // jmp *(sym in .got) 395460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0x68, 0, 0, 0, 0, // pushl $offset 405460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 0xe9, 0, 0, 0, 0 // jmp plt0 415460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao}; 425460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 435460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 445460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 455460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaonamespace mcld { 465460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 475460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei LiaoX86PLT0::X86PLT0(llvm::MCSectionData* pParent, unsigned int pSize) 485460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao : PLTEntry(pSize, pParent) { } 495460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 505460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei LiaoX86PLT1::X86PLT1(llvm::MCSectionData* pParent, unsigned int pSize) 515460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao : PLTEntry(pSize, pParent) { } 525460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 535460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao//===----------------------------------------------------------------------===// 545460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// X86PLT 555460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 565460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei LiaoX86PLT::X86PLT(LDSection& pSection, 575460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao llvm::MCSectionData& pSectionData, 585460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86GOT &pGOTPLT, 595460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao const Output& pOutput) 605460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao : PLT(pSection, pSectionData), m_GOT(pGOTPLT), m_PLTEntryIterator() 615460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 625460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert (Output::DynObj == pOutput.type() || Output::Exec == pOutput.type()); 635460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (Output::DynObj == pOutput.type()) { 645460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLT0 = x86_dyn_plt0; 655460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLT1 = x86_dyn_plt1; 665460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLT0Size = sizeof (x86_dyn_plt0); 675460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLT1Size = sizeof (x86_dyn_plt1); 685460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 695460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao else { 705460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLT0 = x86_exec_plt0; 715460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLT1 = x86_exec_plt1; 725460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLT0Size = sizeof (x86_exec_plt0); 735460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLT1Size = sizeof (x86_exec_plt1); 745460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 755460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT0* plt0_entry = new X86PLT0(&m_SectionData, m_PLT0Size); 765460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 775460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_Section.setSize(m_Section.size() + plt0_entry->getEntrySize()); 785460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 795460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_PLTEntryIterator = pSectionData.begin(); 805460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 815460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 825460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei LiaoX86PLT::~X86PLT() 835460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 845460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 855460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 865460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaovoid X86PLT::reserveEntry(size_t pNum) 875460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 885460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT1* plt1_entry = 0; 895460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao GOTEntry* got_entry = 0; 905460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 915460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao for (size_t i = 0; i < pNum; ++i) { 925460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao plt1_entry = new (std::nothrow) X86PLT1(&m_SectionData, m_PLT1Size); 935460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 945460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (!plt1_entry) 955460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao llvm::report_fatal_error("Allocating new memory for X86PLT1 failed!"); 965460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 975460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_Section.setSize(m_Section.size() + plt1_entry->getEntrySize()); 985460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 995460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao got_entry= new (std::nothrow) GOTEntry(0, m_GOT.getEntrySize(), 1005460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao &(m_GOT.m_SectionData)); 1015460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1025460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (!got_entry) 1035460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao llvm::report_fatal_error("Allocating new memory for GOT failed!"); 1045460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1055460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao m_GOT.m_Section.setSize(m_GOT.m_Section.size() + m_GOT.f_EntrySize); 1065460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1075460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++(m_GOT.m_GOTPLTNum); 1085460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++(m_GOT.m_GeneralGOTIterator); 1095460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 1105460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1115460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1125460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei LiaoPLTEntry* X86PLT::getPLTEntry(const ResolveInfo& pSymbol, bool& pExist) 1135460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1145460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT1 *&PLTEntry = m_PLTEntryMap[&pSymbol]; 1155460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1165460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = 1; 1175460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1185460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (!PLTEntry) { 1195460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao GOTEntry *&GOTPLTEntry = m_GOT.m_GOTPLTMap[&pSymbol]; 1205460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(!GOTPLTEntry && "PLT entry and got.plt entry doesn't match!"); 1215460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1225460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = 0; 1235460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1245460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // This will skip PLT0. 1255460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++m_PLTEntryIterator; 1265460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(m_PLTEntryIterator != m_SectionData.end() && 1275460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao "The number of PLT Entries and ResolveInfo doesn't match"); 1285460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++(m_GOT.m_GOTPLTIterator); 1295460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1305460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao PLTEntry = llvm::cast<X86PLT1>(&(*m_PLTEntryIterator)); 1315460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao GOTPLTEntry = llvm::cast<GOTEntry>(&(*(m_GOT.m_GOTPLTIterator))); 1325460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 1335460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1345460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return PLTEntry; 1355460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1365460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1375460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei LiaoGOTEntry* X86PLT::getGOTPLTEntry(const ResolveInfo& pSymbol, bool& pExist) 1385460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao{ 1395460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao GOTEntry *&GOTPLTEntry = m_GOT.m_GOTPLTMap[&pSymbol]; 1405460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1415460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = 1; 1425460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1435460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (!GOTPLTEntry) { 1445460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT1 *&PLTEntry = m_PLTEntryMap[&pSymbol]; 1455460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(!PLTEntry && "PLT entry and got.plt entry doesn't match!"); 1465460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1475460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao pExist = 0; 1485460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1495460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // This will skip PLT0. 1505460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++m_PLTEntryIterator; 1515460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(m_PLTEntryIterator != m_SectionData.end() && 1525460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao "The number of PLT Entries and ResolveInfo doesn't match"); 1535460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++(m_GOT.m_GOTPLTIterator); 1545460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1555460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao PLTEntry = llvm::cast<X86PLT1>(&(*m_PLTEntryIterator)); 1565460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao GOTPLTEntry = llvm::cast<GOTEntry>(&(*(m_GOT.m_GOTPLTIterator))); 1575460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 1585460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1595460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return GOTPLTEntry; 1605460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1615460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1625460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei LiaoX86PLT0* X86PLT::getPLT0() const { 1635460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1645460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao iterator first = m_SectionData.getFragmentList().begin(); 1655460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao iterator end = m_SectionData.getFragmentList().end(); 1665460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1675460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(first!=end && "FragmentList is empty, getPLT0 failed!"); 1685460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1695460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT0* plt0 = &(llvm::cast<X86PLT0>(*first)); 1705460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1715460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao return plt0; 1725460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 1735460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1745460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// FIXME: It only works on little endian machine. 1755460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaovoid X86PLT::applyPLT0() { 1765460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1775460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao iterator first = m_SectionData.getFragmentList().begin(); 1785460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao iterator end = m_SectionData.getFragmentList().end(); 1795460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1805460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(first!=end && "FragmentList is empty, applyPLT0 failed!"); 1815460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1825460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT0* plt0 = &(llvm::cast<X86PLT0>(*first)); 1835460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1845460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao unsigned char* data = 0; 1855460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao data = static_cast<unsigned char*>(malloc(plt0->getEntrySize())); 1865460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1875460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (!data) 1885460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao llvm::report_fatal_error("Allocating new memory for plt0 failed!"); 1895460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1905460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao memcpy(data, m_PLT0, plt0->getEntrySize()); 1915460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 1925460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (m_PLT0 == x86_exec_plt0) { 1935460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint64_t got_base = m_GOT.getSection().addr(); 1945460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(got_base && ".got base address is NULL!"); 1955460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint32_t *offset = reinterpret_cast<uint32_t*>(data + 2); 1965460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao *offset = got_base + 4; 1975460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao offset = reinterpret_cast<uint32_t*>(data + 8); 1985460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao *offset = got_base + 8; 1995460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 2005460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2015460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao plt0->setContent(data); 2025460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 2035460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2045460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao// FIXME: It only works on little endian machine. 2055460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liaovoid X86PLT::applyPLT1() { 2065460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2075460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint64_t plt_base = m_Section.addr(); 2085460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(plt_base && ".plt base address is NULL!"); 2095460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2105460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint64_t got_base = m_GOT.getSection().addr(); 2115460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(got_base && ".got base address is NULL!"); 2125460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2135460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT::iterator it = m_SectionData.begin(); 2145460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT::iterator ie = m_SectionData.end(); 2155460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao assert(it!=ie && "FragmentList is empty, applyPLT1 failed!"); 2165460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2175460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint64_t GOTEntrySize = m_GOT.getEntrySize(); 2185460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2195460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao // Skip GOT0 2205460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint64_t GOTEntryOffset = GOTEntrySize * X86GOT0Num; 2215460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2225460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao //skip PLT0 2235460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint64_t PLTEntryOffset = m_PLT0Size; 2245460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++it; 2255460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2265460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86PLT1* plt1 = 0; 2275460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2285460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint64_t PLTRelOffset = 0; 2295460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2305460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao while (it != ie) { 2315460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao plt1 = &(llvm::cast<X86PLT1>(*it)); 2325460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao unsigned char *data; 2335460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao data = static_cast<unsigned char*>(malloc(plt1->getEntrySize())); 2345460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2355460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (!data) 2365460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao llvm::report_fatal_error("Allocating new memory for plt1 failed!"); 2375460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2385460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao memcpy(data, m_PLT1, plt1->getEntrySize()); 2395460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2405460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint32_t* offset; 2415460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2425460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao offset = reinterpret_cast<uint32_t*>(data + 2); 2435460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao *offset = GOTEntryOffset; 2445460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao GOTEntryOffset += GOTEntrySize; 2455460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2465460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao offset = reinterpret_cast<uint32_t*>(data + 7); 2475460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao *offset = PLTRelOffset; 2485460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao PLTRelOffset += sizeof (llvm::ELF::Elf32_Rel); 2495460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2505460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao offset = reinterpret_cast<uint32_t*>(data + 12); 2515460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao *offset = -(PLTEntryOffset + 12 + 4); 2525460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao PLTEntryOffset += m_PLT1Size; 2535460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2545460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao plt1->setContent(data); 2555460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++it; 2565460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 2575460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2585460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao unsigned int GOTPLTNum = m_GOT.getGOTPLTNum(); 2595460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2605460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (GOTPLTNum != 0) { 2615460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86GOT::iterator gotplt_it = m_GOT.getLastGOT0(); 2625460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao X86GOT::iterator list_ie = m_GOT.getSectionData().getFragmentList().end(); 2635460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2645460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++gotplt_it; 2655460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao uint64_t PLTEntryAddress = plt_base + m_PLT0Size; 2665460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao for (unsigned int i = 0; i < GOTPLTNum; ++i) { 2675460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao if (gotplt_it == list_ie) 2685460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao llvm::report_fatal_error( 2695460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao "The number of got.plt entries is inconsistent!"); 2705460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2715460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao llvm::cast<GOTEntry>(*gotplt_it).setContent(PLTEntryAddress + 6); 2725460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao PLTEntryAddress += m_PLT1Size; 2735460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao ++gotplt_it; 2745460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 2755460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao } 2765460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} 2775460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 2785460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao} // end namespace mcld 2795460a1f25d9ddecb5c70667267d66d51af177a99Shih-wei Liao 280