ELFObjectWriter.cpp revision d179886f0595eb3564a9edfdfff79def130d89cc
13565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//===- lib/MC/ELFObjectWriter.cpp - ELF File Writer -------------------===//
23565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//
33565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//                     The LLVM Compiler Infrastructure
43565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//
53565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming// This file is distributed under the University of Illinois Open Source
63565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming// License. See LICENSE.TXT for details.
73565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//
83565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//===----------------------------------------------------------------------===//
93565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//
103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming// This file implements ELF object file writer information.
113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//
123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming//===----------------------------------------------------------------------===//
133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/ELFObjectWriter.h"
158f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola#include "llvm/ADT/SmallPtrSet.h"
163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/ADT/STLExtras.h"
173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/ADT/StringMap.h"
183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/ADT/Twine.h"
193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCAssembler.h"
203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCAsmLayout.h"
213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCContext.h"
223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCELFSymbolFlags.h"
233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCExpr.h"
243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCObjectWriter.h"
253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCSectionELF.h"
263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCSymbol.h"
273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/MC/MCValue.h"
283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/Support/Debug.h"
293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/Support/ErrorHandling.h"
303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/Support/ELF.h"
313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "llvm/Target/TargetAsmBackend.h"
323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include "../Target/X86/X86FixupKinds.h"
343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming#include <vector>
363565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingusing namespace llvm;
373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
38ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindolastatic unsigned GetType(const MCSymbolData &SD) {
39ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola  uint32_t Type = (SD.getFlags() & (0xf << ELF_STT_Shift)) >> ELF_STT_Shift;
40ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola  assert(Type == ELF::STT_NOTYPE || Type == ELF::STT_OBJECT ||
41ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola         Type == ELF::STT_FUNC || Type == ELF::STT_SECTION ||
42ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola         Type == ELF::STT_FILE || Type == ELF::STT_COMMON ||
43ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola         Type == ELF::STT_TLS);
44ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola  return Type;
45ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola}
46ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola
47e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindolastatic unsigned GetBinding(const MCSymbolData &SD) {
48e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola  uint32_t Binding = (SD.getFlags() & (0xf << ELF_STB_Shift)) >> ELF_STB_Shift;
49e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola  assert(Binding == ELF::STB_LOCAL || Binding == ELF::STB_GLOBAL ||
50e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola         Binding == ELF::STB_WEAK);
51e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola  return Binding;
52e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola}
53e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola
54e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindolastatic void SetBinding(MCSymbolData &SD, unsigned Binding) {
55e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola  assert(Binding == ELF::STB_LOCAL || Binding == ELF::STB_GLOBAL ||
56e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola         Binding == ELF::STB_WEAK);
57e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola  uint32_t OtherFlags = SD.getFlags() & ~(0xf << ELF_STB_Shift);
58e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola  SD.setFlags(OtherFlags | (Binding << ELF_STB_Shift));
59e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola}
60e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola
61152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindolastatic unsigned GetVisibility(MCSymbolData &SD) {
62152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  unsigned Visibility =
63152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola    (SD.getFlags() & (0xf << ELF_STV_Shift)) >> ELF_STV_Shift;
64152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  assert(Visibility == ELF::STV_DEFAULT || Visibility == ELF::STV_INTERNAL ||
65152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola         Visibility == ELF::STV_HIDDEN || Visibility == ELF::STV_PROTECTED);
66152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  return Visibility;
67152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola}
68152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola
69cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindolastatic bool isFixupKindX86PCRel(unsigned Kind) {
70cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola  switch (Kind) {
71cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola  default:
72cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola    return false;
73cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola  case X86::reloc_pcrel_1byte:
74cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola  case X86::reloc_pcrel_4byte:
75cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola  case X86::reloc_riprel_4byte:
76cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola  case X86::reloc_riprel_4byte_movq_load:
77cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola    return true;
78cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola  }
79cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola}
80cebdc01d66c55a4e8c260ad182d6816f2bcd0c44Rafael Espindola
81a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindolastatic bool RelocNeedsGOT(MCSymbolRefExpr::VariantKind Variant) {
82a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  switch (Variant) {
835c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola  default:
845c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    return false;
85a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_GOT:
86a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_PLT:
87a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_GOTPCREL:
88a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_TPOFF:
89a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_TLSGD:
90a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_GOTTPOFF:
91a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_INDNTPOFF:
92a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_NTPOFF:
93a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  case MCSymbolRefExpr::VK_GOTNTPOFF:
94a264f72d3fb9dec1427480fcf17ef3c746ea723aRafael Espindola  case MCSymbolRefExpr::VK_TLSLDM:
950cf15d61b7e3bf53f5a99f58ada37b93bc039559Rafael Espindola  case MCSymbolRefExpr::VK_DTPOFF:
96b4d1721eff7b43577e5f2e53f885973fb6c43683Rafael Espindola  case MCSymbolRefExpr::VK_TLSLD:
975c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    return true;
985c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola  }
995c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola}
1005c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola
1013565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingnamespace {
1023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  class ELFObjectWriterImpl {
104b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    /*static bool isFixupKindX86RIPRel(unsigned Kind) {
1053565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      return Kind == X86::reloc_riprel_4byte ||
1063565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming        Kind == X86::reloc_riprel_4byte_movq_load;
107b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    }*/
1083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// ELFSymbolData - Helper struct for containing some precomputed information
1113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// on symbols.
1123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    struct ELFSymbolData {
1133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      MCSymbolData *SymbolData;
1143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      uint64_t StringIndex;
1153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      uint32_t SectionIndex;
1163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      // Support lexicographic sorting.
1183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      bool operator<(const ELFSymbolData &RHS) const {
119ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola        if (GetType(*SymbolData) == ELF::STT_FILE)
120ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola          return true;
121ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola        if (GetType(*RHS.SymbolData) == ELF::STT_FILE)
122ad49cf586624c400302d17ecc2c2e41ea4346f1aRafael Espindola          return false;
12336c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer        return SymbolData->getSymbol().getName() <
12436c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer               RHS.SymbolData->getSymbol().getName();
1253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      }
1263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    };
1273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @name Relocation Data
1293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @{
1303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    struct ELFRelocationEntry {
1323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      // Make these big enough for both 32-bit and 64-bit
1333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      uint64_t r_offset;
1348f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      int Index;
1358f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      unsigned Type;
1368f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      const MCSymbol *Symbol;
1373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      uint64_t r_addend;
1383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      // Support lexicographic sorting.
1403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      bool operator<(const ELFRelocationEntry &RE) const {
1413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming        return RE.r_offset < r_offset;
1423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      }
1433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    };
1443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1458f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    SmallPtrSet<const MCSymbol *, 16> UsedInReloc;
14688182132470527e27231f09b25a885893e528c66Rafael Espindola    DenseMap<const MCSymbol *, const MCSymbol *> Renames;
1478f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola
1483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    llvm::DenseMap<const MCSectionData*,
1493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                   std::vector<ELFRelocationEntry> > Relocations;
1503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    DenseMap<const MCSection*, uint64_t> SectionStringTableIndex;
1513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @}
1533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @name Symbol Table Data
1543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @{
1553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    SmallString<256> StringTable;
1573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    std::vector<ELFSymbolData> LocalSymbolData;
1583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    std::vector<ELFSymbolData> ExternalSymbolData;
1593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    std::vector<ELFSymbolData> UndefinedSymbolData;
1603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @}
1623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1638f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    int NumRegularSections;
1648f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola
1655c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    bool NeedsGOT;
1665c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola
1673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFObjectWriter *Writer;
1683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    raw_ostream &OS;
1703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    unsigned Is64Bit : 1;
1723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    bool HasRelocationAddend;
1743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1755baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky    Triple::OSType OSType;
1765baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky
177eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck    uint16_t EMachine;
178eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck
1793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // This holds the symbol table index of the last local symbol.
1803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    unsigned LastLocalSymbolIndex;
1813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // This holds the .strtab section index.
1823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    unsigned StringTableIndex;
1833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    unsigned ShstrtabIndex;
1853565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1863565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  public:
1873565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFObjectWriterImpl(ELFObjectWriter *_Writer, bool _Is64Bit,
188eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck                        uint16_t _EMachine, bool _HasRelAddend,
189eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck                        Triple::OSType _OSType)
1905c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola      : NeedsGOT(false), Writer(_Writer), OS(Writer->getStream()),
1915baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky        Is64Bit(_Is64Bit), HasRelocationAddend(_HasRelAddend),
192eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck        OSType(_OSType), EMachine(_EMachine) {
1933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
1943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void Write8(uint8_t Value) { Writer->Write8(Value); }
1963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void Write16(uint16_t Value) { Writer->Write16(Value); }
1973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void Write32(uint32_t Value) { Writer->Write32(Value); }
198b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    //void Write64(uint64_t Value) { Writer->Write64(Value); }
1993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteZeros(unsigned N) { Writer->WriteZeros(N); }
200b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    //void WriteBytes(StringRef Str, unsigned ZeroFillSize = 0) {
201b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    //  Writer->WriteBytes(Str, ZeroFillSize);
202b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    //}
2033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2043565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteWord(uint64_t W) {
205b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner      if (Is64Bit)
2063565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming        Writer->Write64(W);
207b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner      else
2083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming        Writer->Write32(W);
2093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void String8(char *buf, uint8_t Value) {
2123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[0] = Value;
2133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringLE16(char *buf, uint16_t Value) {
2163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[0] = char(Value >> 0);
2173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[1] = char(Value >> 8);
2183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringLE32(char *buf, uint32_t Value) {
22136c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer      StringLE16(buf, uint16_t(Value >> 0));
222c522f6e70b02d161e4bf37023edb20643ee34acfBenjamin Kramer      StringLE16(buf + 2, uint16_t(Value >> 16));
2233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringLE64(char *buf, uint64_t Value) {
22636c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer      StringLE32(buf, uint32_t(Value >> 0));
227c522f6e70b02d161e4bf37023edb20643ee34acfBenjamin Kramer      StringLE32(buf + 4, uint32_t(Value >> 32));
2283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringBE16(char *buf ,uint16_t Value) {
2313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[0] = char(Value >> 8);
2323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[1] = char(Value >> 0);
2333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringBE32(char *buf, uint32_t Value) {
23636c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer      StringBE16(buf, uint16_t(Value >> 16));
237c522f6e70b02d161e4bf37023edb20643ee34acfBenjamin Kramer      StringBE16(buf + 2, uint16_t(Value >> 0));
2383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringBE64(char *buf, uint64_t Value) {
24136c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer      StringBE32(buf, uint32_t(Value >> 32));
242c522f6e70b02d161e4bf37023edb20643ee34acfBenjamin Kramer      StringBE32(buf + 4, uint32_t(Value >> 0));
2433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void String16(char *buf, uint16_t Value) {
2463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      if (Writer->isLittleEndian())
247f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringLE16(buf, Value);
2483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      else
249f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringBE16(buf, Value);
2503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void String32(char *buf, uint32_t Value) {
2533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      if (Writer->isLittleEndian())
254f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringLE32(buf, Value);
2553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      else
256f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringBE32(buf, Value);
2573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void String64(char *buf, uint64_t Value) {
2603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      if (Writer->isLittleEndian())
261f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringLE64(buf, Value);
2623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      else
263f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringBE64(buf, Value);
2643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteHeader(uint64_t SectionDataSize, unsigned NumberOfSections);
2673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteSymbolEntry(MCDataFragment *F, uint64_t name, uint8_t info,
2693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint64_t value, uint64_t size,
2703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint8_t other, uint16_t shndx);
2713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteSymbol(MCDataFragment *F, ELFSymbolData &MSD,
2733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     const MCAsmLayout &Layout);
2743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteSymbolTable(MCDataFragment *F, const MCAssembler &Asm,
27671859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                          const MCAsmLayout &Layout,
27771859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                          unsigned NumRegularSections);
2783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void RecordRelocation(const MCAssembler &Asm, const MCAsmLayout &Layout,
2803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          const MCFragment *Fragment, const MCFixup &Fixup,
2813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          MCValue Target, uint64_t &FixedValue);
2823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2830b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin Kramer    uint64_t getSymbolIndexInSymbolTable(const MCAssembler &Asm,
2840b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin Kramer                                         const MCSymbol *S);
2853565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2863565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// ComputeSymbolTable - Compute the symbol table data
2873565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ///
2883565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// \param StringTable [out] - The string table data.
2893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// \param StringIndexMap [out] - Map from symbol names to offsets in the
2903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// string table.
2913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void ComputeSymbolTable(MCAssembler &Asm);
2923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteRelocation(MCAssembler &Asm, MCAsmLayout &Layout,
2943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                         const MCSectionData &SD);
2953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteRelocations(MCAssembler &Asm, MCAsmLayout &Layout) {
2973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      for (MCAssembler::const_iterator it = Asm.begin(),
2983565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming             ie = Asm.end(); it != ie; ++it) {
2993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming        WriteRelocation(Asm, Layout, *it);
3003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      }
3013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
3023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void CreateMetadataSections(MCAssembler &Asm, MCAsmLayout &Layout);
3043565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
30588182132470527e27231f09b25a885893e528c66Rafael Espindola    void ExecutePostLayoutBinding(MCAssembler &Asm);
3063565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3073565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteSecHdrEntry(uint32_t Name, uint32_t Type, uint64_t Flags,
3083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint64_t Address, uint64_t Offset,
3093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint64_t Size, uint32_t Link, uint32_t Info,
3103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint64_t Alignment, uint64_t EntrySize);
3113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteRelocationsFragment(const MCAssembler &Asm, MCDataFragment *F,
3133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                  const MCSectionData *SD);
3143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3157070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    bool IsFixupFullyResolved(const MCAssembler &Asm,
3167070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                              const MCValue Target,
3177070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                              bool IsPCRel,
3187070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                              const MCFragment *DF) const;
3197070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
3208f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    void WriteObject(MCAssembler &Asm, const MCAsmLayout &Layout);
3213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  };
3223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
3243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming// Emit the ELF header.
3263565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteHeader(uint64_t SectionDataSize,
3273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                      unsigned NumberOfSections) {
3283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ELF Header
3293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ----------
3303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  //
3313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Note
3323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ----
3333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // emitWord method behaves differently for ELF32 and ELF64, writing
3343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // 4 bytes in the former and 8 in the latter.
3353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(0x7f); // e_ident[EI_MAG0]
3373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8('E');  // e_ident[EI_MAG1]
3383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8('L');  // e_ident[EI_MAG2]
3393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8('F');  // e_ident[EI_MAG3]
3403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(Is64Bit ? ELF::ELFCLASS64 : ELF::ELFCLASS32); // e_ident[EI_CLASS]
3423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_ident[EI_DATA]
3443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(Writer->isLittleEndian() ? ELF::ELFDATA2LSB : ELF::ELFDATA2MSB);
3453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(ELF::EV_CURRENT);        // e_ident[EI_VERSION]
3475baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky  // e_ident[EI_OSABI]
3485baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky  switch (OSType) {
3495baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky    case Triple::FreeBSD:  Write8(ELF::ELFOSABI_FREEBSD); break;
3505baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky    case Triple::Linux:    Write8(ELF::ELFOSABI_LINUX); break;
3515baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky    default:               Write8(ELF::ELFOSABI_NONE); break;
3525baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky  }
3533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(0);                  // e_ident[EI_ABIVERSION]
3543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteZeros(ELF::EI_NIDENT - ELF::EI_PAD);
3563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(ELF::ET_REL);             // e_type
3583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
359eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck  Write16(EMachine); // e_machine = target
3603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(ELF::EV_CURRENT);         // e_version
3623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(0);                    // e_entry, no entry point in .o file
3633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(0);                    // e_phoff, no program header for .o
364eb97677764beb115c658ac559d3649c6c4068eb9Benjamin Kramer  WriteWord(SectionDataSize + (Is64Bit ? sizeof(ELF::Elf64_Ehdr) :
365eb97677764beb115c658ac559d3649c6c4068eb9Benjamin Kramer            sizeof(ELF::Elf32_Ehdr)));  // e_shoff = sec hdr table off in bytes
3663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // FIXME: Make this configurable.
3683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(0);   // e_flags = whatever the target wants
3693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_ehsize = ELF header size
3713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(Is64Bit ? sizeof(ELF::Elf64_Ehdr) : sizeof(ELF::Elf32_Ehdr));
3723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(0);                  // e_phentsize = prog header entry size
3743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(0);                  // e_phnum = # prog header entries = 0
3753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3763565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_shentsize = Section header entry size
3773565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(Is64Bit ? sizeof(ELF::Elf64_Shdr) : sizeof(ELF::Elf32_Shdr));
3783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_shnum     = # of section header ents
3803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(NumberOfSections);
3813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_shstrndx  = Section # of '.shstrtab'
3833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(ShstrtabIndex);
3843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
3853565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3863565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteSymbolEntry(MCDataFragment *F, uint64_t name,
3873565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint8_t info, uint64_t value,
3883565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t size, uint8_t other,
3893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint16_t shndx) {
3903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  if (Is64Bit) {
3913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    char buf[8];
3923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String32(buf, name);
3943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 4); // st_name
3953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String8(buf, info);
3973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 1);  // st_info
398368ae7e4ec0cc0dfc404fa1d07cdf94a7f2a15beBenjamin Kramer
3993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String8(buf, other);
4003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 1); // st_other
4013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String16(buf, shndx);
4033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 2); // st_shndx
4043565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4053565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String64(buf, value);
4063565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 8); // st_value
4073565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String64(buf, size);
4093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 8);  // st_size
4103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  } else {
4113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    char buf[4];
4123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String32(buf, name);
4143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 4);  // st_name
4153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String32(buf, value);
4173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 4); // st_value
4183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String32(buf, size);
4203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 4);  // st_size
4213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String8(buf, info);
4233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 1);  // st_info
4243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String8(buf, other);
4263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 1); // st_other
4273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    String16(buf, shndx);
4293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += StringRef(buf, 2); // st_shndx
4303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
4313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
4323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4332c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindolastatic uint64_t SymbolValue(MCSymbolData &Data, const MCAsmLayout &Layout) {
4342c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  if (Data.isCommon() && Data.isExternal())
4352c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola    return Data.getCommonAlignment();
4362c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola
4372c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  const MCSymbol &Symbol = Data.getSymbol();
4382c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  if (!Symbol.isInSection())
4392c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola    return 0;
4402c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola
4411973d4379250a88938a9358c12d96e96bdc8dedeRafael Espindola  if (MCFragment *FF = Data.getFragment())
4421973d4379250a88938a9358c12d96e96bdc8dedeRafael Espindola    return Layout.getSymbolAddress(&Data) -
4431973d4379250a88938a9358c12d96e96bdc8dedeRafael Espindola      Layout.getSectionAddress(FF->getParent());
4442c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola
4452c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  return 0;
4462c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola}
4472c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola
448de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindolastatic const MCSymbol &AliasedSymbol(const MCSymbol &Symbol) {
449de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  const MCSymbol *S = &Symbol;
450de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  while (S->isVariable()) {
451de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola    const MCExpr *Value = S->getVariableValue();
452a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    if (Value->getKind() != MCExpr::SymbolRef)
453a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      return *S;
454152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola    const MCSymbolRefExpr *Ref = static_cast<const MCSymbolRefExpr*>(Value);
455de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola    S = &Ref->getSymbol();
456152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  }
457de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  return *S;
458de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola}
459de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola
46088182132470527e27231f09b25a885893e528c66Rafael Espindolavoid ELFObjectWriterImpl::ExecutePostLayoutBinding(MCAssembler &Asm) {
46188182132470527e27231f09b25a885893e528c66Rafael Espindola  // The presence of symbol versions causes undefined symbols and
46288182132470527e27231f09b25a885893e528c66Rafael Espindola  // versions declared with @@@ to be renamed.
46388182132470527e27231f09b25a885893e528c66Rafael Espindola
46488182132470527e27231f09b25a885893e528c66Rafael Espindola  for (MCAssembler::symbol_iterator it = Asm.symbol_begin(),
46588182132470527e27231f09b25a885893e528c66Rafael Espindola         ie = Asm.symbol_end(); it != ie; ++it) {
46688182132470527e27231f09b25a885893e528c66Rafael Espindola    const MCSymbol &Alias = it->getSymbol();
46788182132470527e27231f09b25a885893e528c66Rafael Espindola    const MCSymbol &Symbol = AliasedSymbol(Alias);
468f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    MCSymbolData &SD = Asm.getSymbolData(Symbol);
469f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola
470f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // Undefined symbols are global, but this is the first place we
471f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // are able to set it.
472f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    if (Symbol.isUndefined() && !Symbol.isVariable()) {
473f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola      if (GetBinding(SD) == ELF::STB_LOCAL) {
474f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola        SetBinding(SD, ELF::STB_GLOBAL);
475f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola        SetBinding(*it, ELF::STB_GLOBAL);
476f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola      }
477f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    }
478f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola
479f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // Not an alias.
480f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    if (&Symbol == &Alias)
481f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola      continue;
482f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola
48307ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer    StringRef AliasName = Alias.getName();
48488182132470527e27231f09b25a885893e528c66Rafael Espindola    size_t Pos = AliasName.find('@');
48588182132470527e27231f09b25a885893e528c66Rafael Espindola    if (Pos == StringRef::npos)
48688182132470527e27231f09b25a885893e528c66Rafael Espindola      continue;
48788182132470527e27231f09b25a885893e528c66Rafael Espindola
488f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // Aliases defined with .symvar copy the binding from the symbol they alias.
489f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // This is the first place we are able to copy this information.
490f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    it->setExternal(SD.isExternal());
491f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    SetBinding(*it, GetBinding(SD));
492f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola
49307ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer    StringRef Rest = AliasName.substr(Pos);
49488182132470527e27231f09b25a885893e528c66Rafael Espindola    if (!Symbol.isUndefined() && !Rest.startswith("@@@"))
49588182132470527e27231f09b25a885893e528c66Rafael Espindola      continue;
49688182132470527e27231f09b25a885893e528c66Rafael Espindola
49783ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola    // FIXME: produce a better error message.
49883ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola    if (Symbol.isUndefined() && Rest.startswith("@@") &&
49983ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola        !Rest.startswith("@@@"))
50083ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola      report_fatal_error("A @@ version cannot be undefined");
50183ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola
50207ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer    Renames.insert(std::make_pair(&Symbol, &Alias));
50388182132470527e27231f09b25a885893e528c66Rafael Espindola  }
50488182132470527e27231f09b25a885893e528c66Rafael Espindola}
50588182132470527e27231f09b25a885893e528c66Rafael Espindola
506de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindolavoid ELFObjectWriterImpl::WriteSymbol(MCDataFragment *F, ELFSymbolData &MSD,
507de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola                                      const MCAsmLayout &Layout) {
508de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  MCSymbolData &OrigData = *MSD.SymbolData;
509de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  MCSymbolData &Data =
510de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola    Layout.getAssembler().getSymbolData(AliasedSymbol(OrigData.getSymbol()));
511152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola
512152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Binding = GetBinding(OrigData);
513152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Visibility = GetVisibility(OrigData);
514152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Type = GetType(Data);
515152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola
516152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Info = (Binding << ELF_STB_Shift) | (Type << ELF_STT_Shift);
517152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Other = Visibility;
518152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola
5192c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  uint64_t Value = SymbolValue(Data, Layout);
5203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  uint64_t Size = 0;
5213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  const MCExpr *ESize;
5223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
523f7c10a3cff61c70efe8e405d9bdc5386e8e3fc0aRafael Espindola  assert(!(Data.isCommon() && !Data.isExternal()));
524f7c10a3cff61c70efe8e405d9bdc5386e8e3fc0aRafael Espindola
5253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  ESize = Data.getSize();
5263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  if (Data.getSize()) {
5273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCValue Res;
5283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    if (ESize->getKind() == MCExpr::Binary) {
5293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      const MCBinaryExpr *BE = static_cast<const MCBinaryExpr *>(ESize);
5303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      if (BE->EvaluateAsRelocatable(Res, &Layout)) {
53224f1206c7de0f21d309d27737071ad742d5ddf27Benjamin Kramer        assert(!Res.getSymA() || !Res.getSymA()->getSymbol().isDefined());
53324f1206c7de0f21d309d27737071ad742d5ddf27Benjamin Kramer        assert(!Res.getSymB() || !Res.getSymB()->getSymbol().isDefined());
534f230df9af4012f9510de664b6d62b128e26a5861Rafael Espindola        Size = Res.getConstant();
5353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      }
5363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    } else if (ESize->getKind() == MCExpr::Constant) {
537368ae7e4ec0cc0dfc404fa1d07cdf94a7f2a15beBenjamin Kramer      Size = static_cast<const MCConstantExpr *>(ESize)->getValue();
5383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    } else {
5393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      assert(0 && "Unsupported size expression");
5403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
5413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
5423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Write out the symbol table entry
5443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteSymbolEntry(F, MSD.StringIndex, Info, Value,
5453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                   Size, Other, MSD.SectionIndex);
5463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
5473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5483565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteSymbolTable(MCDataFragment *F,
5493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           const MCAssembler &Asm,
55071859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                                           const MCAsmLayout &Layout,
55171859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                                           unsigned NumRegularSections) {
5523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // The string table must be emitted first because we need the index
5533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // into the string table for all the symbol names.
5543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  assert(StringTable.size() && "Missing string table");
5553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // FIXME: Make sure the start of the symbol table is aligned.
5573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // The first entry is the undefined symbol entry.
5593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  unsigned EntrySize = Is64Bit ? ELF::SYMENTRY_SIZE64 : ELF::SYMENTRY_SIZE32;
560368ae7e4ec0cc0dfc404fa1d07cdf94a7f2a15beBenjamin Kramer  F->getContents().append(EntrySize, '\x00');
5613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Write the symbol table entries.
5633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  LastLocalSymbolIndex = LocalSymbolData.size() + 1;
5643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = LocalSymbolData.size(); i != e; ++i) {
5653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFSymbolData &MSD = LocalSymbolData[i];
5663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    WriteSymbol(F, MSD, Layout);
5673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
5683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
56971859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  // Write out a symbol table entry for each regular section.
570a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman  unsigned Index = 1;
57171859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  for (MCAssembler::const_iterator it = Asm.begin();
57271859c640f6a36251aca223fd503c58dc314e296Rafael Espindola       Index <= NumRegularSections; ++it, ++Index) {
573a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman    const MCSectionELF &Section =
574368ae7e4ec0cc0dfc404fa1d07cdf94a7f2a15beBenjamin Kramer      static_cast<const MCSectionELF&>(it->getSection());
575a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman    // Leave out relocations so we don't have indexes within
576a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman    // the relocations messed up
577377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer    if (Section.getType() == ELF::SHT_RELA || Section.getType() == ELF::SHT_REL)
578a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman      continue;
5793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    WriteSymbolEntry(F, 0, ELF::STT_SECTION, 0, 0, ELF::STV_DEFAULT, Index);
580a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman    LastLocalSymbolIndex++;
581a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman  }
5823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = ExternalSymbolData.size(); i != e; ++i) {
5843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFSymbolData &MSD = ExternalSymbolData[i];
5853565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCSymbolData &Data = *MSD.SymbolData;
5863223f19ff0920ffee686faba3bf74babf580e8a5Rafael Espindola    assert(((Data.getFlags() & ELF_STB_Global) ||
5873223f19ff0920ffee686faba3bf74babf580e8a5Rafael Espindola            (Data.getFlags() & ELF_STB_Weak)) &&
5883223f19ff0920ffee686faba3bf74babf580e8a5Rafael Espindola           "External symbol requires STB_GLOBAL or STB_WEAK flag");
5893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    WriteSymbol(F, MSD, Layout);
590e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola    if (GetBinding(Data) == ELF::STB_LOCAL)
5913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      LastLocalSymbolIndex++;
5923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
5933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = UndefinedSymbolData.size(); i != e; ++i) {
5953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFSymbolData &MSD = UndefinedSymbolData[i];
5963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCSymbolData &Data = *MSD.SymbolData;
5973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    WriteSymbol(F, MSD, Layout);
598e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola    if (GetBinding(Data) == ELF::STB_LOCAL)
5993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      LastLocalSymbolIndex++;
6003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
6013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
6023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
6037eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindolastatic bool ShouldRelocOnSymbol(const MCSymbolData &SD,
6043729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola                                const MCValue &Target,
6053729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola                                const MCFragment &F) {
6067eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  const MCSymbol &Symbol = SD.getSymbol();
6077eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  if (Symbol.isUndefined())
6087eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    return true;
6097eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola
6107eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  const MCSectionELF &Section =
6117eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    static_cast<const MCSectionELF&>(Symbol.getSection());
6127eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola
6137eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  if (SD.isExternal())
6147eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    return true;
6157eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola
6168cecf253e45f144b9a7fd0ace85eeeeb0bebfc83Rafael Espindola  MCSymbolRefExpr::VariantKind Kind = Target.getSymA()->getKind();
6173729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola  const MCSectionELF &Sec2 =
6183729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola    static_cast<const MCSectionELF&>(F.getParent()->getSection());
6193729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola
620ce2d3c57758619e99a99104e1168a9558658fab0Rafael Espindola  if (Section.getKind().isBSS())
621ce2d3c57758619e99a99104e1168a9558658fab0Rafael Espindola    return false;
622ce2d3c57758619e99a99104e1168a9558658fab0Rafael Espindola
6238cecf253e45f144b9a7fd0ace85eeeeb0bebfc83Rafael Espindola  if (&Sec2 != &Section &&
624c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola      (Kind == MCSymbolRefExpr::VK_PLT ||
625c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola       Kind == MCSymbolRefExpr::VK_GOTPCREL ||
626c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola       Kind == MCSymbolRefExpr::VK_GOTOFF))
6273729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola    return true;
6283729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola
629c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola  if (Section.getFlags() & MCSectionELF::SHF_MERGE)
630c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola    return Target.getConstant() != 0;
631c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola
6327eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  return false;
63373ffea47d20bc9f559b4ce0c60166ee504073832Rafael Espindola}
63473ffea47d20bc9f559b4ce0c60166ee504073832Rafael Espindola
635e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer// FIXME: this is currently X86/X86_64 only
6363565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::RecordRelocation(const MCAssembler &Asm,
6373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           const MCAsmLayout &Layout,
6383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           const MCFragment *Fragment,
6393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           const MCFixup &Fixup,
6403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           MCValue Target,
6413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t &FixedValue) {
6423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  int64_t Addend = 0;
6438f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  int Index = 0;
64495c602ade3aa30de17abd53b85968a3feb8fdfbcBenjamin Kramer  int64_t Value = Target.getConstant();
6458f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  const MCSymbol *Symbol = 0;
64629129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola  const MCSymbol *Renamed = 0;
6473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
6480007489312ecb1b51ae275f1e452786480151a4cRafael Espindola  bool IsPCRel = isFixupKindX86PCRel(Fixup.getKind());
64981cfb8529b7fd1e98fefb79fb46b10301b93ab75Benjamin Kramer  if (!Target.isAbsolute()) {
650de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola    Symbol = &AliasedSymbol(Target.getSymA()->getSymbol());
65129129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola    Renamed = Renames.lookup(Symbol);
65229129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola    if (!Renamed)
65329129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola      Renamed = Symbol;
6543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCSymbolData &SD = Asm.getSymbolData(*Symbol);
65563d37b932278724d51e7f8bd1f2574c7b45f85f0Benjamin Kramer    MCFragment *F = SD.getFragment();
6563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
6579d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola    if (const MCSymbolRefExpr *RefB = Target.getSymB()) {
6589d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola      const MCSymbol &SymbolB = RefB->getSymbol();
6599d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola      MCSymbolData &SDB = Asm.getSymbolData(SymbolB);
6609d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola      IsPCRel = true;
66155fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      MCSectionData *Sec = Fragment->getParent();
66255fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola
66355fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      // Offset of the symbol in the section
66455fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      int64_t a = Layout.getSymbolAddress(&SDB) - Layout.getSectionAddress(Sec);
66555fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola
66655fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      // Ofeset of the relocation in the section
66755fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      int64_t b = Layout.getFragmentOffset(Fragment) + Fixup.getOffset();
66855fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      Value += b - a;
6699d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola    }
6709d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola
6717eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    // Check that this case has already been fully resolved before we get
6727eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    // here.
6730007489312ecb1b51ae275f1e452786480151a4cRafael Espindola    if (Symbol->isDefined() && !SD.isExternal() &&
6740007489312ecb1b51ae275f1e452786480151a4cRafael Espindola        IsPCRel &&
6750007489312ecb1b51ae275f1e452786480151a4cRafael Espindola        &Fragment->getParent()->getSection() == &Symbol->getSection()) {
6767eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      llvm_unreachable("We don't need a relocation in this case.");
6770007489312ecb1b51ae275f1e452786480151a4cRafael Espindola      return;
6780007489312ecb1b51ae275f1e452786480151a4cRafael Espindola    }
6790007489312ecb1b51ae275f1e452786480151a4cRafael Espindola
6803729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola    bool RelocOnSymbol = ShouldRelocOnSymbol(SD, Target, *Fragment);
6817eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    if (!RelocOnSymbol) {
6828f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      Index = F->getParent()->getOrdinal();
6837eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola
6847eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      MCSectionData *FSD = F->getParent();
6857eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      // Offset of the symbol in the section
6867eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      Value += Layout.getSymbolAddress(&SD) - Layout.getSectionAddress(FSD);
6878f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    } else {
68829129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola      UsedInReloc.insert(Renamed);
6898f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      Index = -1;
6908f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    }
6917eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    Addend = Value;
6927eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    // Compensate for the addend on i386.
6937eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    if (Is64Bit)
6947eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      Value = 0;
6953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
6963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
69795c602ade3aa30de17abd53b85968a3feb8fdfbcBenjamin Kramer  FixedValue = Value;
69895c602ade3aa30de17abd53b85968a3feb8fdfbcBenjamin Kramer
6993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // determine the type of the relocation
70028f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola
70128f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola  MCSymbolRefExpr::VariantKind Modifier = Target.getSymA()->getKind();
70263d37b932278724d51e7f8bd1f2574c7b45f85f0Benjamin Kramer  unsigned Type;
703e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer  if (Is64Bit) {
704e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    if (IsPCRel) {
70592bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola      switch (Modifier) {
7069edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      default:
7079edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        llvm_unreachable("Unimplemented");
70892bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola      case MCSymbolRefExpr::VK_None:
70992bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola        Type = ELF::R_X86_64_PC32;
71092bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola        break;
71192bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola      case MCSymbolRefExpr::VK_PLT:
71292bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola        Type = ELF::R_X86_64_PLT32;
71392bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola        break;
714a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola      case MCSymbolRefExpr::VK_GOTPCREL:
715607d1f6d3216b647c07965bafae8d1aba6312136Rafael Espindola        Type = ELF::R_X86_64_GOTPCREL;
716607d1f6d3216b647c07965bafae8d1aba6312136Rafael Espindola        break;
717bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola      case MCSymbolRefExpr::VK_GOTTPOFF:
718bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        Type = ELF::R_X86_64_GOTTPOFF;
719bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        break;
720bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola      case MCSymbolRefExpr::VK_TLSGD:
721bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        Type = ELF::R_X86_64_TLSGD;
722bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        break;
723b4d1721eff7b43577e5f2e53f885973fb6c43683Rafael Espindola      case MCSymbolRefExpr::VK_TLSLD:
724b4d1721eff7b43577e5f2e53f885973fb6c43683Rafael Espindola        Type = ELF::R_X86_64_TLSLD;
725b4d1721eff7b43577e5f2e53f885973fb6c43683Rafael Espindola        break;
72692bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola      }
727e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    } else {
728e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      switch ((unsigned)Fixup.getKind()) {
729e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      default: llvm_unreachable("invalid fixup kind!");
730e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_8: Type = ELF::R_X86_64_64; break;
731a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola      case X86::reloc_signed_4byte:
732e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case X86::reloc_pcrel_4byte:
733a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola        assert(isInt<32>(Target.getConstant()));
73428f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola        switch (Modifier) {
7359edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        default:
7369edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola          llvm_unreachable("Unimplemented");
73728f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola        case MCSymbolRefExpr::VK_None:
73828f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola          Type = ELF::R_X86_64_32S;
73928f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola          break;
74028f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola        case MCSymbolRefExpr::VK_GOT:
74128f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola          Type = ELF::R_X86_64_GOT32;
74228f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola          break;
743c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        case MCSymbolRefExpr::VK_GOTPCREL:
7448cecf253e45f144b9a7fd0ace85eeeeb0bebfc83Rafael Espindola          Type = ELF::R_X86_64_GOTPCREL;
7458cecf253e45f144b9a7fd0ace85eeeeb0bebfc83Rafael Espindola          break;
746bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        case MCSymbolRefExpr::VK_TPOFF:
747bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola          Type = ELF::R_X86_64_TPOFF32;
748bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola          break;
749aa8f1f01352bdaaabf712369f8a8a615c776b508Rafael Espindola        case MCSymbolRefExpr::VK_DTPOFF:
750aa8f1f01352bdaaabf712369f8a8a615c776b508Rafael Espindola          Type = ELF::R_X86_64_DTPOFF32;
751aa8f1f01352bdaaabf712369f8a8a615c776b508Rafael Espindola          break;
75228f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola        }
753a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola        break;
754e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_4:
755a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola        Type = ELF::R_X86_64_32;
756e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer        break;
757e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_2: Type = ELF::R_X86_64_16; break;
758e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case X86::reloc_pcrel_1byte:
759e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_1: Type = ELF::R_X86_64_8; break;
760e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      }
761e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    }
7623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  } else {
763e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    if (IsPCRel) {
7649edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      switch (Modifier) {
7659edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      default:
76624dc9ecdc55d732ac93fc782d6232413fe2c1160Rafael Espindola        llvm_unreachable("Unimplemented");
76724dc9ecdc55d732ac93fc782d6232413fe2c1160Rafael Espindola      case MCSymbolRefExpr::VK_None:
7689baee3b3a3924bc45d007e54c71333739d135f8eRafael Espindola        Type = ELF::R_386_PC32;
7699baee3b3a3924bc45d007e54c71333739d135f8eRafael Espindola        break;
7709edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      case MCSymbolRefExpr::VK_PLT:
7719edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        Type = ELF::R_386_PLT32;
7729edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        break;
7739edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      }
774e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    } else {
775e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      switch ((unsigned)Fixup.getKind()) {
776e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      default: llvm_unreachable("invalid fixup kind!");
777a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola
77824ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola      case X86::reloc_global_offset_table:
77924ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola        Type = ELF::R_386_GOTPC;
78024ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola        break;
78124ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola
782a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola      // FIXME: Should we avoid selecting reloc_signed_4byte in 32 bit mode
783a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola      // instead?
784a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola      case X86::reloc_signed_4byte:
785e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case X86::reloc_pcrel_4byte:
786bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola      case FK_Data_4:
787c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        switch (Modifier) {
7889edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        default:
7899edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola          llvm_unreachable("Unimplemented");
790bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola        case MCSymbolRefExpr::VK_None:
79124ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola          Type = ELF::R_386_32;
792bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola          break;
793eada30479399014c22c7b7edb008177c312eefeeRafael Espindola        case MCSymbolRefExpr::VK_GOT:
794eada30479399014c22c7b7edb008177c312eefeeRafael Espindola          Type = ELF::R_386_GOT32;
795eada30479399014c22c7b7edb008177c312eefeeRafael Espindola          break;
796c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        case MCSymbolRefExpr::VK_GOTOFF:
797c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola          Type = ELF::R_386_GOTOFF;
798c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola          break;
7993cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola        case MCSymbolRefExpr::VK_TLSGD:
8003cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          Type = ELF::R_386_TLS_GD;
8013cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          break;
8023cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola        case MCSymbolRefExpr::VK_TPOFF:
8033cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          Type = ELF::R_386_TLS_LE_32;
8043cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          break;
8053cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola        case MCSymbolRefExpr::VK_INDNTPOFF:
8063cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          Type = ELF::R_386_TLS_IE;
8073cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          break;
8083cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola        case MCSymbolRefExpr::VK_NTPOFF:
8093cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          Type = ELF::R_386_TLS_LE;
8103cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          break;
811a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola        case MCSymbolRefExpr::VK_GOTNTPOFF:
812a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola          Type = ELF::R_386_TLS_GOTIE;
813a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola          break;
814a264f72d3fb9dec1427480fcf17ef3c746ea723aRafael Espindola        case MCSymbolRefExpr::VK_TLSLDM:
815a264f72d3fb9dec1427480fcf17ef3c746ea723aRafael Espindola          Type = ELF::R_386_TLS_LDM;
816a264f72d3fb9dec1427480fcf17ef3c746ea723aRafael Espindola          break;
8170cf15d61b7e3bf53f5a99f58ada37b93bc039559Rafael Espindola        case MCSymbolRefExpr::VK_DTPOFF:
8180cf15d61b7e3bf53f5a99f58ada37b93bc039559Rafael Espindola          Type = ELF::R_386_TLS_LDO_32;
8190cf15d61b7e3bf53f5a99f58ada37b93bc039559Rafael Espindola          break;
820c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        }
821c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        break;
822e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_2: Type = ELF::R_386_16; break;
823e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case X86::reloc_pcrel_1byte:
824e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_1: Type = ELF::R_386_8; break;
825e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      }
8263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
8273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
8283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
829a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  if (RelocNeedsGOT(Modifier))
8305c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    NeedsGOT = true;
8315c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola
832e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer  ELFRelocationEntry ERE;
833e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer
8348f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  ERE.Index = Index;
8358f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  ERE.Type = Type;
83629129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola  ERE.Symbol = Renamed;
8373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
83863d37b932278724d51e7f8bd1f2574c7b45f85f0Benjamin Kramer  ERE.r_offset = Layout.getFragmentOffset(Fragment) + Fixup.getOffset();
839e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer
8403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  if (HasRelocationAddend)
8413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ERE.r_addend = Addend;
842172d7d6a12b313dd5a8c1588b8f6bd49b1ce16b6Benjamin Kramer  else
843172d7d6a12b313dd5a8c1588b8f6bd49b1ce16b6Benjamin Kramer    ERE.r_addend = 0; // Silence compiler warning.
8443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
8453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Relocations[Fragment->getParent()].push_back(ERE);
8463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
8473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
8480b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin Krameruint64_t
8490b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin KramerELFObjectWriterImpl::getSymbolIndexInSymbolTable(const MCAssembler &Asm,
8500b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin Kramer                                                 const MCSymbol *S) {
8517b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer  MCSymbolData &SD = Asm.getSymbolData(*S);
8527b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer
8537b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer  // Local symbol.
8547b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer  if (!SD.isExternal() && !S->isUndefined())
8557b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer    return SD.getIndex() + /* empty symbol */ 1;
8567b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer
8577b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer  // External or undefined symbol.
8588f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  return SD.getIndex() + NumRegularSections + /* empty symbol */ 1;
8593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
8603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
861737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindolastatic bool isInSymtab(const MCAssembler &Asm, const MCSymbolData &Data,
86288182132470527e27231f09b25a885893e528c66Rafael Espindola                       bool Used, bool Renamed) {
863bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola  if (Used)
864bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola    return true;
865bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola
86688182132470527e27231f09b25a885893e528c66Rafael Espindola  if (Renamed)
86788182132470527e27231f09b25a885893e528c66Rafael Espindola    return false;
86888182132470527e27231f09b25a885893e528c66Rafael Espindola
869737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  const MCSymbol &Symbol = Data.getSymbol();
870a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola
871d179886f0595eb3564a9edfdfff79def130d89ccRafael Espindola  if (Symbol.getName() == "_GLOBAL_OFFSET_TABLE_")
872d179886f0595eb3564a9edfdfff79def130d89ccRafael Espindola    return true;
873d179886f0595eb3564a9edfdfff79def130d89ccRafael Espindola
874a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola  const MCSymbol &A = AliasedSymbol(Symbol);
875d179886f0595eb3564a9edfdfff79def130d89ccRafael Espindola  if (!A.isVariable() && A.isUndefined() && !Data.isCommon())
876a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    return false;
877a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola
878737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  if (!Asm.isSymbolLinkerVisible(Symbol) && !Symbol.isUndefined())
879737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola    return false;
880737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
881bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola  if (Symbol.isTemporary())
882737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola    return false;
883737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
884737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  return true;
885737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola}
886737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
887737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindolastatic bool isLocal(const MCSymbolData &Data) {
888737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  if (Data.isExternal())
889737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola    return false;
890737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
891737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  const MCSymbol &Symbol = Data.getSymbol();
892737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  if (Symbol.isUndefined() && !Symbol.isVariable())
893737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola    return false;
894737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
895737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  return true;
896737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola}
897737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
8983565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::ComputeSymbolTable(MCAssembler &Asm) {
8995c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola  // FIXME: Is this the correct place to do this?
9005c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola  if (NeedsGOT) {
9015c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    llvm::StringRef Name = "_GLOBAL_OFFSET_TABLE_";
9025c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    MCSymbol *Sym = Asm.getContext().GetOrCreateSymbol(Name);
9035c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    MCSymbolData &Data = Asm.getOrCreateSymbolData(*Sym);
9045c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    Data.setExternal(true);
905f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    SetBinding(Data, ELF::STB_GLOBAL);
9065c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola  }
9075c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola
9083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Build section lookup table.
9098f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  NumRegularSections = Asm.size();
910f5c347dfc25e133e357c7086ad1b24edad4d2bc8Rafael Espindola  DenseMap<const MCSection*, uint32_t> SectionIndexMap;
9113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  unsigned Index = 1;
9123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::iterator it = Asm.begin(),
9133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it, ++Index)
9143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    SectionIndexMap[&it->getSection()] = Index;
9153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
9163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Index 0 is always the empty string.
9173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StringMap<uint64_t> StringIndexMap;
9183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StringTable += '\x00';
9193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
920a0949b50dcea35c08b50542091f97275f401529dRafael Espindola  // Add the data for the symbols.
9213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::symbol_iterator it = Asm.symbol_begin(),
9223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.symbol_end(); it != ie; ++it) {
9233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSymbol &Symbol = it->getSymbol();
9243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
92588182132470527e27231f09b25a885893e528c66Rafael Espindola    if (!isInSymtab(Asm, *it, UsedInReloc.count(&Symbol),
92688182132470527e27231f09b25a885893e528c66Rafael Espindola                    Renames.count(&Symbol)))
9273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      continue;
9283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
9293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFSymbolData MSD;
9303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MSD.SymbolData = it;
931a0949b50dcea35c08b50542091f97275f401529dRafael Espindola    bool Local = isLocal(*it);
93288182132470527e27231f09b25a885893e528c66Rafael Espindola    const MCSymbol &RefSymbol = AliasedSymbol(Symbol);
9333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
934f7c10a3cff61c70efe8e405d9bdc5386e8e3fc0aRafael Espindola    if (it->isCommon()) {
935a0949b50dcea35c08b50542091f97275f401529dRafael Espindola      assert(!Local);
936f7c10a3cff61c70efe8e405d9bdc5386e8e3fc0aRafael Espindola      MSD.SectionIndex = ELF::SHN_COMMON;
937bf052ac5d1c8f21075bc675f629709c20791c5f7Rafael Espindola    } else if (Symbol.isAbsolute() || RefSymbol.isVariable()) {
938a0949b50dcea35c08b50542091f97275f401529dRafael Espindola      MSD.SectionIndex = ELF::SHN_ABS;
93988182132470527e27231f09b25a885893e528c66Rafael Espindola    } else if (RefSymbol.isUndefined()) {
9403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      MSD.SectionIndex = ELF::SHN_UNDEF;
9413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    } else {
942bf052ac5d1c8f21075bc675f629709c20791c5f7Rafael Espindola      MSD.SectionIndex = SectionIndexMap.lookup(&RefSymbol.getSection());
9433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      assert(MSD.SectionIndex && "Invalid section index!");
9445df0b65e8f5ef55f0b515a0b24879f5d117d6042Rafael Espindola    }
9455df0b65e8f5ef55f0b515a0b24879f5d117d6042Rafael Espindola
94688182132470527e27231f09b25a885893e528c66Rafael Espindola    // The @@@ in symbol version is replaced with @ in undefined symbols and
94788182132470527e27231f09b25a885893e528c66Rafael Espindola    // @@ in defined ones.
94888182132470527e27231f09b25a885893e528c66Rafael Espindola    StringRef Name = Symbol.getName();
94988182132470527e27231f09b25a885893e528c66Rafael Espindola    size_t Pos = Name.find("@@@");
95088182132470527e27231f09b25a885893e528c66Rafael Espindola    std::string FinalName;
95188182132470527e27231f09b25a885893e528c66Rafael Espindola    if (Pos != StringRef::npos) {
95207ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer      StringRef Prefix = Name.substr(0, Pos);
95388182132470527e27231f09b25a885893e528c66Rafael Espindola      unsigned n = MSD.SectionIndex == ELF::SHN_UNDEF ? 2 : 1;
95407ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer      StringRef Suffix = Name.substr(Pos + n);
95588182132470527e27231f09b25a885893e528c66Rafael Espindola      FinalName = Prefix.str() + Suffix.str();
95688182132470527e27231f09b25a885893e528c66Rafael Espindola    } else {
95788182132470527e27231f09b25a885893e528c66Rafael Espindola      FinalName = Name.str();
95888182132470527e27231f09b25a885893e528c66Rafael Espindola    }
95988182132470527e27231f09b25a885893e528c66Rafael Espindola
96088182132470527e27231f09b25a885893e528c66Rafael Espindola    uint64_t &Entry = StringIndexMap[FinalName];
961a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    if (!Entry) {
962a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      Entry = StringTable.size();
96388182132470527e27231f09b25a885893e528c66Rafael Espindola      StringTable += FinalName;
964a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      StringTable += '\x00';
9653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
966a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    MSD.StringIndex = Entry;
967a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    if (MSD.SectionIndex == ELF::SHN_UNDEF)
968a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      UndefinedSymbolData.push_back(MSD);
969a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    else if (Local)
970a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      LocalSymbolData.push_back(MSD);
971a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    else
972a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      ExternalSymbolData.push_back(MSD);
9733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
9743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
9753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Symbols are required to be in lexicographic order.
9763565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  array_pod_sort(LocalSymbolData.begin(), LocalSymbolData.end());
9773565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  array_pod_sort(ExternalSymbolData.begin(), ExternalSymbolData.end());
9783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  array_pod_sort(UndefinedSymbolData.begin(), UndefinedSymbolData.end());
9793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
9803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Set the symbol indices. Local symbols must come before all other
9813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // symbols with non-local bindings.
9823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Index = 0;
9833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = LocalSymbolData.size(); i != e; ++i)
9843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    LocalSymbolData[i].SymbolData->setIndex(Index++);
9853565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = ExternalSymbolData.size(); i != e; ++i)
9863565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ExternalSymbolData[i].SymbolData->setIndex(Index++);
9873565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = UndefinedSymbolData.size(); i != e; ++i)
9883565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    UndefinedSymbolData[i].SymbolData->setIndex(Index++);
9893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
9903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
9913565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteRelocation(MCAssembler &Asm, MCAsmLayout &Layout,
9923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                          const MCSectionData &SD) {
9933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  if (!Relocations[&SD].empty()) {
9943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCContext &Ctx = Asm.getContext();
9953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSection *RelaSection;
9963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionELF &Section =
9973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      static_cast<const MCSectionELF&>(SD.getSection());
9983565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
9993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const StringRef SectionName = Section.getSectionName();
1000377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer    std::string RelaSectionName = HasRelocationAddend ? ".rela" : ".rel";
10013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    RelaSectionName += SectionName;
1002299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer
1003299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer    unsigned EntrySize;
1004299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer    if (HasRelocationAddend)
1005299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer      EntrySize = Is64Bit ? sizeof(ELF::Elf64_Rela) : sizeof(ELF::Elf32_Rela);
1006299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer    else
1007299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer      EntrySize = Is64Bit ? sizeof(ELF::Elf64_Rel) : sizeof(ELF::Elf32_Rel);
10083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1009377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer    RelaSection = Ctx.getELFSection(RelaSectionName, HasRelocationAddend ?
1010377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer                                    ELF::SHT_RELA : ELF::SHT_REL, 0,
10113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                    SectionKind::getReadOnly(),
10123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                    false, EntrySize);
10133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCSectionData &RelaSD = Asm.getOrCreateSectionData(*RelaSection);
1015a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    RelaSD.setAlignment(Is64Bit ? 8 : 4);
10163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCDataFragment *F = new MCDataFragment(&RelaSD);
10183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    WriteRelocationsFragment(Asm, F, &SD);
10203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10217070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    Asm.AddSectionToTheEnd(*Writer, RelaSD, Layout);
10223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
10233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
10243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10253565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteSecHdrEntry(uint32_t Name, uint32_t Type,
10263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t Flags, uint64_t Address,
10273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t Offset, uint64_t Size,
10283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint32_t Link, uint32_t Info,
10293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t Alignment,
10303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t EntrySize) {
10313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(Name);        // sh_name: index into string table
10323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(Type);        // sh_type
10333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Flags);     // sh_flags
10343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Address);   // sh_addr
10353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Offset);    // sh_offset
10363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Size);      // sh_size
10373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(Link);        // sh_link
10383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(Info);        // sh_info
10393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Alignment); // sh_addralign
10403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(EntrySize); // sh_entsize
10413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
10423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10433565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteRelocationsFragment(const MCAssembler &Asm,
10443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                   MCDataFragment *F,
10453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                   const MCSectionData *SD) {
10463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  std::vector<ELFRelocationEntry> &Relocs = Relocations[SD];
10473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // sort by the r_offset just like gnu as does
10483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  array_pod_sort(Relocs.begin(), Relocs.end());
10493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = Relocs.size(); i != e; ++i) {
10513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFRelocationEntry entry = Relocs[e - i - 1];
10523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10538f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    if (entry.Index < 0)
10548f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      entry.Index = getSymbolIndexInSymbolTable(Asm, entry.Symbol);
10558f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    else
10568f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      entry.Index += LocalSymbolData.size() + 1;
10575e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer    if (Is64Bit) {
10585e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      char buf[8];
10593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10605e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      String64(buf, entry.r_offset);
10615e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      F->getContents() += StringRef(buf, 8);
10625e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
10638f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      struct ELF::Elf64_Rela ERE64;
10648f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      ERE64.setSymbolAndType(entry.Index, entry.Type);
10658f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      String64(buf, ERE64.r_info);
10665e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      F->getContents() += StringRef(buf, 8);
10675e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
10685e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      if (HasRelocationAddend) {
10695e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer        String64(buf, entry.r_addend);
10705e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer        F->getContents() += StringRef(buf, 8);
10715e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      }
10725e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer    } else {
10735e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      char buf[4];
10745e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
10755e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      String32(buf, entry.r_offset);
10765e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      F->getContents() += StringRef(buf, 4);
10775e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
10788f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      struct ELF::Elf32_Rela ERE32;
10798f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      ERE32.setSymbolAndType(entry.Index, entry.Type);
10808f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      String32(buf, ERE32.r_info);
10815e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      F->getContents() += StringRef(buf, 4);
10825e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
10835e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      if (HasRelocationAddend) {
10845e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer        String32(buf, entry.r_addend);
10855e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer        F->getContents() += StringRef(buf, 4);
10865e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer      }
10875e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer    }
10883565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
10893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
10903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10913565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::CreateMetadataSections(MCAssembler &Asm,
10923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                 MCAsmLayout &Layout) {
10933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  MCContext &Ctx = Asm.getContext();
10943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  MCDataFragment *F;
10953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  const MCSection *SymtabSection;
10973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  unsigned EntrySize = Is64Bit ? ELF::SYMENTRY_SIZE64 : ELF::SYMENTRY_SIZE32;
10983565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
109971859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  unsigned NumRegularSections = Asm.size();
110071859c640f6a36251aca223fd503c58dc314e296Rafael Espindola
110138738bf1a847292003a5495f17e42a75d1274bf7Rafael Espindola  // We construct .shstrtab, .symtab and .strtab in this order to match gnu as.
110271859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  const MCSection *ShstrtabSection;
110371859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  ShstrtabSection = Ctx.getELFSection(".shstrtab", ELF::SHT_STRTAB, 0,
110471859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                                      SectionKind::getReadOnly(), false);
110571859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  MCSectionData &ShstrtabSD = Asm.getOrCreateSectionData(*ShstrtabSection);
110671859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  ShstrtabSD.setAlignment(1);
110771859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  ShstrtabIndex = Asm.size();
110871859c640f6a36251aca223fd503c58dc314e296Rafael Espindola
11093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  SymtabSection = Ctx.getELFSection(".symtab", ELF::SHT_SYMTAB, 0,
11103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                    SectionKind::getReadOnly(),
11113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                    false, EntrySize);
11123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  MCSectionData &SymtabSD = Asm.getOrCreateSectionData(*SymtabSection);
11133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  SymtabSD.setAlignment(Is64Bit ? 8 : 4);
11143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  const MCSection *StrtabSection;
11163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StrtabSection = Ctx.getELFSection(".strtab", ELF::SHT_STRTAB, 0,
11173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                    SectionKind::getReadOnly(), false);
11183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  MCSectionData &StrtabSD = Asm.getOrCreateSectionData(*StrtabSection);
11193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StrtabSD.setAlignment(1);
11203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StringTableIndex = Asm.size();
11213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1122c3c413f63117896f186fc4385bdaac0578d3613fRafael Espindola  WriteRelocations(Asm, Layout);
112371859c640f6a36251aca223fd503c58dc314e296Rafael Espindola
112471859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  // Symbol table
112571859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  F = new MCDataFragment(&SymtabSD);
112671859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  WriteSymbolTable(F, Asm, Layout, NumRegularSections);
11277070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  Asm.AddSectionToTheEnd(*Writer, SymtabSD, Layout);
112871859c640f6a36251aca223fd503c58dc314e296Rafael Espindola
11293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  F = new MCDataFragment(&StrtabSD);
11303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  F->getContents().append(StringTable.begin(), StringTable.end());
11317070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  Asm.AddSectionToTheEnd(*Writer, StrtabSD, Layout);
11323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  F = new MCDataFragment(&ShstrtabSD);
11343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Section header string table.
11363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  //
11373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // The first entry of a string table holds a null character so skip
11383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // section 0.
11393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  uint64_t Index = 1;
11403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  F->getContents() += '\x00';
11413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::const_iterator it = Asm.begin(),
11433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it) {
11443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionELF &Section =
1145368ae7e4ec0cc0dfc404fa1d07cdf94a7f2a15beBenjamin Kramer      static_cast<const MCSectionELF&>(it->getSection());
114651efe7a253190b672519e8388ce5d45f1dcf1a24Rafael Espindola    // FIXME: We could merge suffixes like in .text and .rela.text.
11473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // Remember the index into the string table so we can write it
11493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // into the sh_name field of the section header table.
11503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    SectionStringTableIndex[&it->getSection()] = Index;
11513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    Index += Section.getSectionName().size() + 1;
11533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += Section.getSectionName();
11543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += '\x00';
11553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
11563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11577070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  Asm.AddSectionToTheEnd(*Writer, ShstrtabSD, Layout);
11587070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola}
11597070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
11607070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindolabool ELFObjectWriterImpl::IsFixupFullyResolved(const MCAssembler &Asm,
11617070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                               const MCValue Target,
11627070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                               bool IsPCRel,
11637070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                               const MCFragment *DF) const {
11647070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  // If this is a PCrel relocation, find the section this fixup value is
11657070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  // relative to.
11667070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSection *BaseSection = 0;
11677070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (IsPCRel) {
11687070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    BaseSection = &DF->getParent()->getSection();
11697070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    assert(BaseSection);
11707070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  }
11717070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
11727070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSection *SectionA = 0;
11737070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSymbol *SymbolA = 0;
11747070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (const MCSymbolRefExpr *A = Target.getSymA()) {
11757070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    SymbolA = &A->getSymbol();
11767070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    SectionA = &SymbolA->getSection();
11777070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  }
11787070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
11797070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSection *SectionB = 0;
11807070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (const MCSymbolRefExpr *B = Target.getSymB()) {
11817070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    SectionB = &B->getSymbol().getSection();
11827070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  }
11837070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
11847070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (!BaseSection)
11857070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    return SectionA == SectionB;
11867070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
11877070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSymbolData &DataA = Asm.getSymbolData(*SymbolA);
11887070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (DataA.isExternal())
11897070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    return false;
11907070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
11917070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  return !SectionB && BaseSection == SectionA;
11923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
11933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11948f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindolavoid ELFObjectWriterImpl::WriteObject(MCAssembler &Asm,
11953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                      const MCAsmLayout &Layout) {
11968f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  // Compute symbol table information.
11978f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  ComputeSymbolTable(Asm);
11988f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola
11993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  CreateMetadataSections(const_cast<MCAssembler&>(Asm),
12003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                         const_cast<MCAsmLayout&>(Layout));
12013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Add 1 for the null section.
12033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  unsigned NumSections = Asm.size() + 1;
1204a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  uint64_t NaturalAlignment = Is64Bit ? 8 : 4;
1205a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  uint64_t HeaderSize = Is64Bit ? sizeof(ELF::Elf64_Ehdr) : sizeof(ELF::Elf32_Ehdr);
1206a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  uint64_t FileOff = HeaderSize;
12073565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::const_iterator it = Asm.begin(),
12093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it) {
12103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionData &SD = *it;
12113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1212a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    FileOff = RoundUpToAlignment(FileOff, SD.getAlignment());
1213a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
12143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // Get the size of the section in the output file (including padding).
12153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    uint64_t Size = Layout.getSectionFileSize(&SD);
1216a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
1217a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    FileOff += Size;
12183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
12193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1220a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  FileOff = RoundUpToAlignment(FileOff, NaturalAlignment);
1221a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
12223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Write out the ELF header ...
1223a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  WriteHeader(FileOff - HeaderSize, NumSections);
1224a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
1225a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  FileOff = HeaderSize;
12263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ... then all of the sections ...
12283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  DenseMap<const MCSection*, uint64_t> SectionOffsetMap;
12293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1230d8e0bfe07a7d101eeed7204d1dfcaae17ad51908Rafael Espindola  DenseMap<const MCSection*, uint32_t> SectionIndexMap;
123144cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer
123244cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer  unsigned Index = 1;
12333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::const_iterator it = Asm.begin(),
12343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it) {
1235a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    const MCSectionData &SD = *it;
1236a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
1237a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    uint64_t Padding = OffsetToAlignment(FileOff, SD.getAlignment());
1238a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    WriteZeros(Padding);
1239a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    FileOff += Padding;
1240a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
12413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // Remember the offset into the file for this section.
12423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    SectionOffsetMap[&it->getSection()] = FileOff;
124344cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer    SectionIndexMap[&it->getSection()] = Index++;
124444cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer
12453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    FileOff += Layout.getSectionFileSize(&SD);
12463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    Asm.WriteSectionData(it, Layout, Writer);
12483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
12493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1250a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  uint64_t Padding = OffsetToAlignment(FileOff, NaturalAlignment);
1251a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  WriteZeros(Padding);
1252a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  FileOff += Padding;
1253a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
12543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ... and then the section header table.
12553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Should we align the section header table?
12563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  //
12573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Null section first.
12583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteSecHdrEntry(0, 0, 0, 0, 0, 0, 0, 0, 0, 0);
12593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::const_iterator it = Asm.begin(),
12613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it) {
12623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionData &SD = *it;
12633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionELF &Section =
12643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      static_cast<const MCSectionELF&>(SD.getSection());
12653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    uint64_t sh_link = 0;
12673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    uint64_t sh_info = 0;
12683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    switch(Section.getType()) {
12703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_DYNAMIC:
12713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      sh_link = SectionStringTableIndex[&it->getSection()];
12723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      sh_info = 0;
12733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
12743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_REL:
1276f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman    case ELF::SHT_RELA: {
12773565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      const MCSection *SymtabSection;
12783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      const MCSection *InfoSection;
12793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      SymtabSection = Asm.getContext().getELFSection(".symtab", ELF::SHT_SYMTAB, 0,
12813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                     SectionKind::getReadOnly(),
1282f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman                                                     false);
128344cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer      sh_link = SectionIndexMap[SymtabSection];
12843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1285377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer      // Remove ".rel" and ".rela" prefixes.
1286377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer      unsigned SecNameLen = (Section.getType() == ELF::SHT_REL) ? 4 : 5;
1287377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer      StringRef SectionName = Section.getSectionName().substr(SecNameLen);
1288377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer
1289f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman      InfoSection = Asm.getContext().getELFSection(SectionName,
12903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                   ELF::SHT_PROGBITS, 0,
1291f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman                                                   SectionKind::getReadOnly(),
1292f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman                                                   false);
129344cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer      sh_info = SectionIndexMap[InfoSection];
12943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
1295f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman    }
12963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_SYMTAB:
12983565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_DYNSYM:
12993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      sh_link = StringTableIndex;
13003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      sh_info = LastLocalSymbolIndex;
13013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
13023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_PROGBITS:
13043565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_STRTAB:
13053565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_NOBITS:
130619dc7faf25e3a8cc37a656b74ebeef3ab976aa8eBenjamin Kramer    case ELF::SHT_NULL:
1307cecbc3d28277ff4916326311cbf87335ed05d106Rafael Espindola    case ELF::SHT_ARM_ATTRIBUTES:
13083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      // Nothing to do.
13093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
13103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    default:
13123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      assert(0 && "FIXME: sh_type value not supported!");
13133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
13143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
13153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    WriteSecHdrEntry(SectionStringTableIndex[&it->getSection()],
13173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     Section.getType(), Section.getFlags(),
131871859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                     0,
13193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     SectionOffsetMap.lookup(&SD.getSection()),
13203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     Layout.getSectionSize(&SD), sh_link,
13213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     sh_info, SD.getAlignment(),
13223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     Section.getEntrySize());
13233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
13243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13263565a06ebf44a193a8b333cbeff2ee154298d450Matt FlemingELFObjectWriter::ELFObjectWriter(raw_ostream &OS,
13273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                 bool Is64Bit,
13285baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky                                 Triple::OSType OSType,
1329eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck                                 uint16_t EMachine,
13303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                 bool IsLittleEndian,
13313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                 bool HasRelocationAddend)
13323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  : MCObjectWriter(OS, IsLittleEndian)
13333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming{
1334eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck  Impl = new ELFObjectWriterImpl(this, Is64Bit, EMachine,
1335eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck                                 HasRelocationAddend, OSType);
13363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13383565a06ebf44a193a8b333cbeff2ee154298d450Matt FlemingELFObjectWriter::~ELFObjectWriter() {
13393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  delete (ELFObjectWriterImpl*) Impl;
13403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13423565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriter::ExecutePostLayoutBinding(MCAssembler &Asm) {
13433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  ((ELFObjectWriterImpl*) Impl)->ExecutePostLayoutBinding(Asm);
13443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13463565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriter::RecordRelocation(const MCAssembler &Asm,
13473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                       const MCAsmLayout &Layout,
13483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                       const MCFragment *Fragment,
13493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                       const MCFixup &Fixup, MCValue Target,
13503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                       uint64_t &FixedValue) {
13513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  ((ELFObjectWriterImpl*) Impl)->RecordRelocation(Asm, Layout, Fragment, Fixup,
13523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                  Target, FixedValue);
13533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13557070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindolabool ELFObjectWriter::IsFixupFullyResolved(const MCAssembler &Asm,
13567070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                           const MCValue Target,
13577070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                           bool IsPCRel,
13587070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                           const MCFragment *DF) const {
13597070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  return ((ELFObjectWriterImpl*) Impl)->IsFixupFullyResolved(Asm, Target,
13607070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                                             IsPCRel, DF);
13617070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola}
13627070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
13638f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindolavoid ELFObjectWriter::WriteObject(MCAssembler &Asm,
13643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                  const MCAsmLayout &Layout) {
13653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  ((ELFObjectWriterImpl*) Impl)->WriteObject(Asm, Layout);
13663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
1367