ELFObjectWriter.cpp revision af3d38f70ebba9fec0048c2540dbdf0168d280d2
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;
146484291c27319668ad99cb87def000254357736fbRafael Espindola    SmallPtrSet<const MCSymbol *, 16> WeakrefUsedInReloc;
14788182132470527e27231f09b25a885893e528c66Rafael Espindola    DenseMap<const MCSymbol *, const MCSymbol *> Renames;
1488f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola
1493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    llvm::DenseMap<const MCSectionData*,
1503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                   std::vector<ELFRelocationEntry> > Relocations;
1513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    DenseMap<const MCSection*, uint64_t> SectionStringTableIndex;
1523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @}
1543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @name Symbol Table Data
1553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @{
1563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    SmallString<256> StringTable;
1583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    std::vector<ELFSymbolData> LocalSymbolData;
1593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    std::vector<ELFSymbolData> ExternalSymbolData;
1603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    std::vector<ELFSymbolData> UndefinedSymbolData;
1613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// @}
1633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1648f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    int NumRegularSections;
1658f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola
1665c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    bool NeedsGOT;
1675c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola
1687be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    bool NeedsSymtabShndx;
1697be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola
1703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFObjectWriter *Writer;
1713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    raw_ostream &OS;
1733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    unsigned Is64Bit : 1;
1753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1763565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    bool HasRelocationAddend;
1773565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1785baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky    Triple::OSType OSType;
1795baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky
180eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck    uint16_t EMachine;
181eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck
1823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // This holds the symbol table index of the last local symbol.
1833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    unsigned LastLocalSymbolIndex;
1843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // This holds the .strtab section index.
1853565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    unsigned StringTableIndex;
1867be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    // This holds the .symtab section index.
1877be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    unsigned SymbolTableIndex;
1883565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    unsigned ShstrtabIndex;
1903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  public:
1923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFObjectWriterImpl(ELFObjectWriter *_Writer, bool _Is64Bit,
193eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck                        uint16_t _EMachine, bool _HasRelAddend,
194eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck                        Triple::OSType _OSType)
1957be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola      : NeedsGOT(false), NeedsSymtabShndx(false), Writer(_Writer),
1967be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola        OS(Writer->getStream()),
1975baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky        Is64Bit(_Is64Bit), HasRelocationAddend(_HasRelAddend),
198eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck        OSType(_OSType), EMachine(_EMachine) {
1993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void Write8(uint8_t Value) { Writer->Write8(Value); }
2023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void Write16(uint16_t Value) { Writer->Write16(Value); }
2033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void Write32(uint32_t Value) { Writer->Write32(Value); }
204b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    //void Write64(uint64_t Value) { Writer->Write64(Value); }
2053565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteZeros(unsigned N) { Writer->WriteZeros(N); }
206b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    //void WriteBytes(StringRef Str, unsigned ZeroFillSize = 0) {
207b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    //  Writer->WriteBytes(Str, ZeroFillSize);
208b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner    //}
2093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteWord(uint64_t W) {
211b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner      if (Is64Bit)
2123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming        Writer->Write64(W);
213b188a3789e2fc64ec7088a220c4c428c41d86faaChris Lattner      else
2143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming        Writer->Write32(W);
2153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringLE16(char *buf, uint16_t Value) {
2183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[0] = char(Value >> 0);
2193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[1] = char(Value >> 8);
2203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringLE32(char *buf, uint32_t Value) {
22336c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer      StringLE16(buf, uint16_t(Value >> 0));
224c522f6e70b02d161e4bf37023edb20643ee34acfBenjamin Kramer      StringLE16(buf + 2, uint16_t(Value >> 16));
2253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringLE64(char *buf, uint64_t Value) {
22836c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer      StringLE32(buf, uint32_t(Value >> 0));
229c522f6e70b02d161e4bf37023edb20643ee34acfBenjamin Kramer      StringLE32(buf + 4, uint32_t(Value >> 32));
2303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringBE16(char *buf ,uint16_t Value) {
2333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[0] = char(Value >> 8);
2343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      buf[1] = char(Value >> 0);
2353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringBE32(char *buf, uint32_t Value) {
23836c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer      StringBE16(buf, uint16_t(Value >> 16));
239c522f6e70b02d161e4bf37023edb20643ee34acfBenjamin Kramer      StringBE16(buf + 2, uint16_t(Value >> 0));
2403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void StringBE64(char *buf, uint64_t Value) {
24336c6dc22bcaed92f03f7019a0d1cd47ea69e12daBenjamin Kramer      StringBE32(buf, uint32_t(Value >> 32));
244c522f6e70b02d161e4bf37023edb20643ee34acfBenjamin Kramer      StringBE32(buf + 4, uint32_t(Value >> 0));
2453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
247af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    void String8(MCDataFragment &F, uint8_t Value) {
248af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      char buf[1];
249af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      buf[0] = Value;
250af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      F.getContents() += StringRef(buf, 1);
251af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    }
252af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola
253af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    void String16(MCDataFragment &F, uint16_t Value) {
254af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      char buf[2];
2553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      if (Writer->isLittleEndian())
256f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringLE16(buf, Value);
2573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      else
258f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringBE16(buf, Value);
259af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      F.getContents() += StringRef(buf, 2);
2603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
262af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    void String32(MCDataFragment &F, uint32_t Value) {
263af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      char buf[4];
2643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      if (Writer->isLittleEndian())
265f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringLE32(buf, Value);
2663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      else
267f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringBE32(buf, Value);
268af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      F.getContents() += StringRef(buf, 4);
2693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
271af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    void String64(MCDataFragment &F, uint64_t Value) {
272af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      char buf[8];
2733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      if (Writer->isLittleEndian())
274f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringLE64(buf, Value);
2753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      else
276f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman        StringBE64(buf, Value);
277af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      F.getContents() += StringRef(buf, 8);
2783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
2793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteHeader(uint64_t SectionDataSize, unsigned NumberOfSections);
2813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2827be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    void WriteSymbolEntry(MCDataFragment *SymtabF, MCDataFragment *ShndxF,
2837be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                          uint64_t name, uint8_t info,
2843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint64_t value, uint64_t size,
2857be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                          uint8_t other, uint32_t shndx,
2867be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                          bool Reserved);
2873565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2887be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    void WriteSymbol(MCDataFragment *SymtabF,  MCDataFragment *ShndxF,
2897be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                     ELFSymbolData &MSD,
2903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     const MCAsmLayout &Layout);
2913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2927be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    void WriteSymbolTable(MCDataFragment *SymtabF, MCDataFragment *ShndxF,
2937be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                          const MCAssembler &Asm,
29471859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                          const MCAsmLayout &Layout,
29571859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                          unsigned NumRegularSections);
2963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
2973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void RecordRelocation(const MCAssembler &Asm, const MCAsmLayout &Layout,
2983565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          const MCFragment *Fragment, const MCFixup &Fixup,
2993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          MCValue Target, uint64_t &FixedValue);
3003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3010b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin Kramer    uint64_t getSymbolIndexInSymbolTable(const MCAssembler &Asm,
3020b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin Kramer                                         const MCSymbol *S);
3033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3043565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// ComputeSymbolTable - Compute the symbol table data
3053565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ///
3063565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// \param StringTable [out] - The string table data.
3073565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// \param StringIndexMap [out] - Map from symbol names to offsets in the
3083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    /// string table.
3093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void ComputeSymbolTable(MCAssembler &Asm);
3103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteRelocation(MCAssembler &Asm, MCAsmLayout &Layout,
3123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                         const MCSectionData &SD);
3133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteRelocations(MCAssembler &Asm, MCAsmLayout &Layout) {
3153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      for (MCAssembler::const_iterator it = Asm.begin(),
3163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming             ie = Asm.end(); it != ie; ++it) {
3173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming        WriteRelocation(Asm, Layout, *it);
3183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      }
3193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
3203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void CreateMetadataSections(MCAssembler &Asm, MCAsmLayout &Layout);
3223565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
32388182132470527e27231f09b25a885893e528c66Rafael Espindola    void ExecutePostLayoutBinding(MCAssembler &Asm);
3243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteSecHdrEntry(uint32_t Name, uint32_t Type, uint64_t Flags,
3263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint64_t Address, uint64_t Offset,
3273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint64_t Size, uint32_t Link, uint32_t Info,
3283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                          uint64_t Alignment, uint64_t EntrySize);
3293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    void WriteRelocationsFragment(const MCAssembler &Asm, MCDataFragment *F,
3313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                  const MCSectionData *SD);
3323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3337070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    bool IsFixupFullyResolved(const MCAssembler &Asm,
3347070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                              const MCValue Target,
3357070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                              bool IsPCRel,
3367070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                              const MCFragment *DF) const;
3377070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
3388f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    void WriteObject(MCAssembler &Asm, const MCAsmLayout &Layout);
3393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  };
3403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
3423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming// Emit the ELF header.
3443565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteHeader(uint64_t SectionDataSize,
3453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                      unsigned NumberOfSections) {
3463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ELF Header
3473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ----------
3483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  //
3493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Note
3503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ----
3513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // emitWord method behaves differently for ELF32 and ELF64, writing
3523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // 4 bytes in the former and 8 in the latter.
3533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(0x7f); // e_ident[EI_MAG0]
3553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8('E');  // e_ident[EI_MAG1]
3563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8('L');  // e_ident[EI_MAG2]
3573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8('F');  // e_ident[EI_MAG3]
3583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(Is64Bit ? ELF::ELFCLASS64 : ELF::ELFCLASS32); // e_ident[EI_CLASS]
3603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_ident[EI_DATA]
3623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(Writer->isLittleEndian() ? ELF::ELFDATA2LSB : ELF::ELFDATA2MSB);
3633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(ELF::EV_CURRENT);        // e_ident[EI_VERSION]
3655baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky  // e_ident[EI_OSABI]
3665baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky  switch (OSType) {
3675baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky    case Triple::FreeBSD:  Write8(ELF::ELFOSABI_FREEBSD); break;
3685baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky    case Triple::Linux:    Write8(ELF::ELFOSABI_LINUX); break;
3695baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky    default:               Write8(ELF::ELFOSABI_NONE); break;
3705baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky  }
3713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write8(0);                  // e_ident[EI_ABIVERSION]
3723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteZeros(ELF::EI_NIDENT - ELF::EI_PAD);
3743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(ELF::ET_REL);             // e_type
3763565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
377eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck  Write16(EMachine); // e_machine = target
3783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(ELF::EV_CURRENT);         // e_version
3803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(0);                    // e_entry, no entry point in .o file
3813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(0);                    // e_phoff, no program header for .o
382eb97677764beb115c658ac559d3649c6c4068eb9Benjamin Kramer  WriteWord(SectionDataSize + (Is64Bit ? sizeof(ELF::Elf64_Ehdr) :
383eb97677764beb115c658ac559d3649c6c4068eb9Benjamin Kramer            sizeof(ELF::Elf32_Ehdr)));  // e_shoff = sec hdr table off in bytes
3843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3853565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // FIXME: Make this configurable.
3863565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(0);   // e_flags = whatever the target wants
3873565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3883565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_ehsize = ELF header size
3893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(Is64Bit ? sizeof(ELF::Elf64_Ehdr) : sizeof(ELF::Elf32_Ehdr));
3903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(0);                  // e_phentsize = prog header entry size
3923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(0);                  // e_phnum = # prog header entries = 0
3933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_shentsize = Section header entry size
3953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write16(Is64Bit ? sizeof(ELF::Elf64_Shdr) : sizeof(ELF::Elf32_Shdr));
3963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
3973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_shnum     = # of section header ents
3987be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  if (NumberOfSections >= ELF::SHN_LORESERVE)
3997be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    Write16(0);
4007be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  else
4017be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    Write16(NumberOfSections);
4023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // e_shstrndx  = Section # of '.shstrtab'
4047be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  if (NumberOfSections >= ELF::SHN_LORESERVE)
4057be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    Write16(ELF::SHN_XINDEX);
4067be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  else
4077be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    Write16(ShstrtabIndex);
4083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
4093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4107be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindolavoid ELFObjectWriterImpl::WriteSymbolEntry(MCDataFragment *SymtabF,
4117be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                                           MCDataFragment *ShndxF,
4127be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                                           uint64_t name,
4133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint8_t info, uint64_t value,
4143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t size, uint8_t other,
4157be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                                           uint32_t shndx,
4167be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                                           bool Reserved) {
4177be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  if (ShndxF) {
4187be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    if (shndx >= ELF::SHN_LORESERVE && !Reserved)
419af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      String32(*ShndxF, shndx);
4207be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    else
421af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      String32(*ShndxF, 0);
4227be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  }
4237be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola
424af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola  uint16_t Index = (shndx >= ELF::SHN_LORESERVE && !Reserved) ?
425af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    uint16_t(ELF::SHN_XINDEX) : shndx;
4263565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
427af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola  if (Is64Bit) {
428af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String32(*SymtabF, name);  // st_name
429af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String8(*SymtabF, info);   // st_info
430af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String8(*SymtabF, other);  // st_other
431af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String16(*SymtabF, Index); // st_shndx
432af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String64(*SymtabF, value); // st_value
433af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String64(*SymtabF, size);  // st_size
4343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  } else {
435af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String32(*SymtabF, name);  // st_name
436af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String32(*SymtabF, value); // st_value
437af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String32(*SymtabF, size);  // st_size
438af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String8(*SymtabF, info);   // st_info
439af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String8(*SymtabF, other);  // st_other
440af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola    String16(*SymtabF, Index); // st_shndx
4413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
4423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
4433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
4442c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindolastatic uint64_t SymbolValue(MCSymbolData &Data, const MCAsmLayout &Layout) {
4452c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  if (Data.isCommon() && Data.isExternal())
4462c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola    return Data.getCommonAlignment();
4472c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola
4482c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  const MCSymbol &Symbol = Data.getSymbol();
4492c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  if (!Symbol.isInSection())
4502c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola    return 0;
4512c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola
4521973d4379250a88938a9358c12d96e96bdc8dedeRafael Espindola  if (MCFragment *FF = Data.getFragment())
4531973d4379250a88938a9358c12d96e96bdc8dedeRafael Espindola    return Layout.getSymbolAddress(&Data) -
4541973d4379250a88938a9358c12d96e96bdc8dedeRafael Espindola      Layout.getSectionAddress(FF->getParent());
4552c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola
4562c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  return 0;
4572c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola}
4582c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola
459de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindolastatic const MCSymbol &AliasedSymbol(const MCSymbol &Symbol) {
460de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  const MCSymbol *S = &Symbol;
461de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  while (S->isVariable()) {
462de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola    const MCExpr *Value = S->getVariableValue();
463a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    if (Value->getKind() != MCExpr::SymbolRef)
464a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      return *S;
465152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola    const MCSymbolRefExpr *Ref = static_cast<const MCSymbolRefExpr*>(Value);
466de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola    S = &Ref->getSymbol();
467152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  }
468de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  return *S;
469de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola}
470de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola
47188182132470527e27231f09b25a885893e528c66Rafael Espindolavoid ELFObjectWriterImpl::ExecutePostLayoutBinding(MCAssembler &Asm) {
47288182132470527e27231f09b25a885893e528c66Rafael Espindola  // The presence of symbol versions causes undefined symbols and
47388182132470527e27231f09b25a885893e528c66Rafael Espindola  // versions declared with @@@ to be renamed.
47488182132470527e27231f09b25a885893e528c66Rafael Espindola
47588182132470527e27231f09b25a885893e528c66Rafael Espindola  for (MCAssembler::symbol_iterator it = Asm.symbol_begin(),
47688182132470527e27231f09b25a885893e528c66Rafael Espindola         ie = Asm.symbol_end(); it != ie; ++it) {
47788182132470527e27231f09b25a885893e528c66Rafael Espindola    const MCSymbol &Alias = it->getSymbol();
47888182132470527e27231f09b25a885893e528c66Rafael Espindola    const MCSymbol &Symbol = AliasedSymbol(Alias);
479f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    MCSymbolData &SD = Asm.getSymbolData(Symbol);
480f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola
481f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // Undefined symbols are global, but this is the first place we
482f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // are able to set it.
483f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    if (Symbol.isUndefined() && !Symbol.isVariable()) {
484f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola      if (GetBinding(SD) == ELF::STB_LOCAL) {
485f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola        SetBinding(SD, ELF::STB_GLOBAL);
486f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola        SetBinding(*it, ELF::STB_GLOBAL);
487f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola      }
488f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    }
489f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola
490f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // Not an alias.
491f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    if (&Symbol == &Alias)
492f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola      continue;
493f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola
49407ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer    StringRef AliasName = Alias.getName();
49588182132470527e27231f09b25a885893e528c66Rafael Espindola    size_t Pos = AliasName.find('@');
49688182132470527e27231f09b25a885893e528c66Rafael Espindola    if (Pos == StringRef::npos)
49788182132470527e27231f09b25a885893e528c66Rafael Espindola      continue;
49888182132470527e27231f09b25a885893e528c66Rafael Espindola
499f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // Aliases defined with .symvar copy the binding from the symbol they alias.
500f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    // This is the first place we are able to copy this information.
501f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    it->setExternal(SD.isExternal());
502f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    SetBinding(*it, GetBinding(SD));
503f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola
50407ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer    StringRef Rest = AliasName.substr(Pos);
50588182132470527e27231f09b25a885893e528c66Rafael Espindola    if (!Symbol.isUndefined() && !Rest.startswith("@@@"))
50688182132470527e27231f09b25a885893e528c66Rafael Espindola      continue;
50788182132470527e27231f09b25a885893e528c66Rafael Espindola
50883ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola    // FIXME: produce a better error message.
50983ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola    if (Symbol.isUndefined() && Rest.startswith("@@") &&
51083ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola        !Rest.startswith("@@@"))
51183ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola      report_fatal_error("A @@ version cannot be undefined");
51283ff4d2b0d6133d48055a27fe25c5d241bca7e9bRafael Espindola
51307ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer    Renames.insert(std::make_pair(&Symbol, &Alias));
51488182132470527e27231f09b25a885893e528c66Rafael Espindola  }
51588182132470527e27231f09b25a885893e528c66Rafael Espindola}
51688182132470527e27231f09b25a885893e528c66Rafael Espindola
5177be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindolavoid ELFObjectWriterImpl::WriteSymbol(MCDataFragment *SymtabF,
5187be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                                      MCDataFragment *ShndxF,
5197be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                                      ELFSymbolData &MSD,
520de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola                                      const MCAsmLayout &Layout) {
521de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  MCSymbolData &OrigData = *MSD.SymbolData;
522de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola  MCSymbolData &Data =
523de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola    Layout.getAssembler().getSymbolData(AliasedSymbol(OrigData.getSymbol()));
524152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola
5257be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  bool IsReserved = Data.isCommon() || Data.getSymbol().isAbsolute() ||
5267be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    Data.getSymbol().isVariable();
5277be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola
528152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Binding = GetBinding(OrigData);
529152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Visibility = GetVisibility(OrigData);
530152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Type = GetType(Data);
531152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola
532152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Info = (Binding << ELF_STB_Shift) | (Type << ELF_STT_Shift);
533152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola  uint8_t Other = Visibility;
534152c1061e0b4ad379eec5fa38ee0091fc11ff936Rafael Espindola
5352c6ec31df6c6c45ad735892eedab78612ac2847bRafael Espindola  uint64_t Value = SymbolValue(Data, Layout);
5363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  uint64_t Size = 0;
5373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  const MCExpr *ESize;
5383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
539f7c10a3cff61c70efe8e405d9bdc5386e8e3fc0aRafael Espindola  assert(!(Data.isCommon() && !Data.isExternal()));
540f7c10a3cff61c70efe8e405d9bdc5386e8e3fc0aRafael Espindola
5413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  ESize = Data.getSize();
5423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  if (Data.getSize()) {
5433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCValue Res;
5443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    if (ESize->getKind() == MCExpr::Binary) {
5453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      const MCBinaryExpr *BE = static_cast<const MCBinaryExpr *>(ESize);
5463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      if (BE->EvaluateAsRelocatable(Res, &Layout)) {
54824f1206c7de0f21d309d27737071ad742d5ddf27Benjamin Kramer        assert(!Res.getSymA() || !Res.getSymA()->getSymbol().isDefined());
54924f1206c7de0f21d309d27737071ad742d5ddf27Benjamin Kramer        assert(!Res.getSymB() || !Res.getSymB()->getSymbol().isDefined());
550f230df9af4012f9510de664b6d62b128e26a5861Rafael Espindola        Size = Res.getConstant();
5513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      }
5523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    } else if (ESize->getKind() == MCExpr::Constant) {
553368ae7e4ec0cc0dfc404fa1d07cdf94a7f2a15beBenjamin Kramer      Size = static_cast<const MCConstantExpr *>(ESize)->getValue();
5543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    } else {
5553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      assert(0 && "Unsupported size expression");
5563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
5573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
5583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Write out the symbol table entry
5607be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  WriteSymbolEntry(SymtabF, ShndxF, MSD.StringIndex, Info, Value,
5617be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                   Size, Other, MSD.SectionIndex, IsReserved);
5623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
5633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5647be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindolavoid ELFObjectWriterImpl::WriteSymbolTable(MCDataFragment *SymtabF,
5657be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                                           MCDataFragment *ShndxF,
5663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           const MCAssembler &Asm,
56771859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                                           const MCAsmLayout &Layout,
56871859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                                           unsigned NumRegularSections) {
5693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // The string table must be emitted first because we need the index
5703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // into the string table for all the symbol names.
5713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  assert(StringTable.size() && "Missing string table");
5723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // FIXME: Make sure the start of the symbol table is aligned.
5743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // The first entry is the undefined symbol entry.
5767be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  WriteSymbolEntry(SymtabF, ShndxF, 0, 0, 0, 0, 0, 0, false);
5773565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
5783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Write the symbol table entries.
5793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  LastLocalSymbolIndex = LocalSymbolData.size() + 1;
5803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = LocalSymbolData.size(); i != e; ++i) {
5813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFSymbolData &MSD = LocalSymbolData[i];
5827be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    WriteSymbol(SymtabF, ShndxF, MSD, Layout);
5833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
5843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
58571859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  // Write out a symbol table entry for each regular section.
586a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman  unsigned Index = 1;
58771859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  for (MCAssembler::const_iterator it = Asm.begin();
58871859c640f6a36251aca223fd503c58dc314e296Rafael Espindola       Index <= NumRegularSections; ++it, ++Index) {
589a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman    const MCSectionELF &Section =
590368ae7e4ec0cc0dfc404fa1d07cdf94a7f2a15beBenjamin Kramer      static_cast<const MCSectionELF&>(it->getSection());
591a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman    // Leave out relocations so we don't have indexes within
592a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman    // the relocations messed up
593377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer    if (Section.getType() == ELF::SHT_RELA || Section.getType() == ELF::SHT_REL)
594a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman      continue;
5957be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    WriteSymbolEntry(SymtabF, ShndxF, 0, ELF::STT_SECTION, 0, 0,
5967be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                     ELF::STV_DEFAULT, Index, false);
597a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman    LastLocalSymbolIndex++;
598a44fa2473a5f2a13730d44e0aea8897b5656b555Eli Friedman  }
5993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
6003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = ExternalSymbolData.size(); i != e; ++i) {
6013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFSymbolData &MSD = ExternalSymbolData[i];
6023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCSymbolData &Data = *MSD.SymbolData;
6033223f19ff0920ffee686faba3bf74babf580e8a5Rafael Espindola    assert(((Data.getFlags() & ELF_STB_Global) ||
6043223f19ff0920ffee686faba3bf74babf580e8a5Rafael Espindola            (Data.getFlags() & ELF_STB_Weak)) &&
6053223f19ff0920ffee686faba3bf74babf580e8a5Rafael Espindola           "External symbol requires STB_GLOBAL or STB_WEAK flag");
6067be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    WriteSymbol(SymtabF, ShndxF, MSD, Layout);
607e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola    if (GetBinding(Data) == ELF::STB_LOCAL)
6083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      LastLocalSymbolIndex++;
6093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
6103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
6113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = UndefinedSymbolData.size(); i != e; ++i) {
6123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFSymbolData &MSD = UndefinedSymbolData[i];
6133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCSymbolData &Data = *MSD.SymbolData;
6147be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    WriteSymbol(SymtabF, ShndxF, MSD, Layout);
615e15eb4e14cd543b925a837026cbdde9f94393e1cRafael Espindola    if (GetBinding(Data) == ELF::STB_LOCAL)
6163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      LastLocalSymbolIndex++;
6173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
6183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
6193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
6207eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindolastatic bool ShouldRelocOnSymbol(const MCSymbolData &SD,
6213729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola                                const MCValue &Target,
6223729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola                                const MCFragment &F) {
6237eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  const MCSymbol &Symbol = SD.getSymbol();
6247eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  if (Symbol.isUndefined())
6257eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    return true;
6267eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola
6277eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  const MCSectionELF &Section =
6287eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    static_cast<const MCSectionELF&>(Symbol.getSection());
6297eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola
6307eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  if (SD.isExternal())
6317eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    return true;
6327eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola
6338cecf253e45f144b9a7fd0ace85eeeeb0bebfc83Rafael Espindola  MCSymbolRefExpr::VariantKind Kind = Target.getSymA()->getKind();
6343729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola  const MCSectionELF &Sec2 =
6353729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola    static_cast<const MCSectionELF&>(F.getParent()->getSection());
6363729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola
637ce2d3c57758619e99a99104e1168a9558658fab0Rafael Espindola  if (Section.getKind().isBSS())
638ce2d3c57758619e99a99104e1168a9558658fab0Rafael Espindola    return false;
639ce2d3c57758619e99a99104e1168a9558658fab0Rafael Espindola
6408cecf253e45f144b9a7fd0ace85eeeeb0bebfc83Rafael Espindola  if (&Sec2 != &Section &&
641c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola      (Kind == MCSymbolRefExpr::VK_PLT ||
642c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola       Kind == MCSymbolRefExpr::VK_GOTPCREL ||
643c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola       Kind == MCSymbolRefExpr::VK_GOTOFF))
6443729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola    return true;
6453729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola
646c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola  if (Section.getFlags() & MCSectionELF::SHF_MERGE)
647c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola    return Target.getConstant() != 0;
648c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola
6497eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola  return false;
65073ffea47d20bc9f559b4ce0c60166ee504073832Rafael Espindola}
65173ffea47d20bc9f559b4ce0c60166ee504073832Rafael Espindola
652e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer// FIXME: this is currently X86/X86_64 only
6533565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::RecordRelocation(const MCAssembler &Asm,
6543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           const MCAsmLayout &Layout,
6553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           const MCFragment *Fragment,
6563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           const MCFixup &Fixup,
6573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           MCValue Target,
6583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t &FixedValue) {
6593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  int64_t Addend = 0;
6608f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  int Index = 0;
66195c602ade3aa30de17abd53b85968a3feb8fdfbcBenjamin Kramer  int64_t Value = Target.getConstant();
6628f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  const MCSymbol *Symbol = 0;
66329129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola  const MCSymbol *Renamed = 0;
6643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
6650007489312ecb1b51ae275f1e452786480151a4cRafael Espindola  bool IsPCRel = isFixupKindX86PCRel(Fixup.getKind());
66681cfb8529b7fd1e98fefb79fb46b10301b93ab75Benjamin Kramer  if (!Target.isAbsolute()) {
667de89b010b0b347cdbd1bc9553a970880a3e721bfRafael Espindola    Symbol = &AliasedSymbol(Target.getSymA()->getSymbol());
66829129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola    Renamed = Renames.lookup(Symbol);
66929129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola    if (!Renamed)
670484291c27319668ad99cb87def000254357736fbRafael Espindola      Renamed = &Target.getSymA()->getSymbol();
6713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCSymbolData &SD = Asm.getSymbolData(*Symbol);
67263d37b932278724d51e7f8bd1f2574c7b45f85f0Benjamin Kramer    MCFragment *F = SD.getFragment();
6733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
6749d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola    if (const MCSymbolRefExpr *RefB = Target.getSymB()) {
6759d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola      const MCSymbol &SymbolB = RefB->getSymbol();
6769d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola      MCSymbolData &SDB = Asm.getSymbolData(SymbolB);
6779d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola      IsPCRel = true;
67855fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      MCSectionData *Sec = Fragment->getParent();
67955fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola
68055fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      // Offset of the symbol in the section
68155fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      int64_t a = Layout.getSymbolAddress(&SDB) - Layout.getSectionAddress(Sec);
68255fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola
68355fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      // Ofeset of the relocation in the section
68455fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      int64_t b = Layout.getFragmentOffset(Fragment) + Fixup.getOffset();
68555fb102130925ada2bbc78abbbb70394d96d945eRafael Espindola      Value += b - a;
6869d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola    }
6879d8b7555cde8b9722dd6f8dc77c627bae0718c67Rafael Espindola
6887eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    // Check that this case has already been fully resolved before we get
6897eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    // here.
6900007489312ecb1b51ae275f1e452786480151a4cRafael Espindola    if (Symbol->isDefined() && !SD.isExternal() &&
6910007489312ecb1b51ae275f1e452786480151a4cRafael Espindola        IsPCRel &&
6920007489312ecb1b51ae275f1e452786480151a4cRafael Espindola        &Fragment->getParent()->getSection() == &Symbol->getSection()) {
6937eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      llvm_unreachable("We don't need a relocation in this case.");
6940007489312ecb1b51ae275f1e452786480151a4cRafael Espindola      return;
6950007489312ecb1b51ae275f1e452786480151a4cRafael Espindola    }
6960007489312ecb1b51ae275f1e452786480151a4cRafael Espindola
6973729d0052bda365d04fee900c6f3d09460f1e108Rafael Espindola    bool RelocOnSymbol = ShouldRelocOnSymbol(SD, Target, *Fragment);
6987eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    if (!RelocOnSymbol) {
6998f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      Index = F->getParent()->getOrdinal();
7007eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola
7017eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      MCSectionData *FSD = F->getParent();
7027eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      // Offset of the symbol in the section
7037eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      Value += Layout.getSymbolAddress(&SD) - Layout.getSectionAddress(FSD);
7048f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    } else {
70529129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola      UsedInReloc.insert(Renamed);
706484291c27319668ad99cb87def000254357736fbRafael Espindola      MCSymbolData &RenamedSD = Asm.getSymbolData(*Renamed);
707484291c27319668ad99cb87def000254357736fbRafael Espindola      if (RenamedSD.getFlags() & ELF_Other_Weakref) {
708484291c27319668ad99cb87def000254357736fbRafael Espindola        WeakrefUsedInReloc.insert(Symbol);
709484291c27319668ad99cb87def000254357736fbRafael Espindola      }
7108f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      Index = -1;
7118f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    }
7127eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    Addend = Value;
7137eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    // Compensate for the addend on i386.
7147eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola    if (Is64Bit)
7157eae36b38b874f417fa191fc1cfec22c100f164dRafael Espindola      Value = 0;
7163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
7173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
71895c602ade3aa30de17abd53b85968a3feb8fdfbcBenjamin Kramer  FixedValue = Value;
71995c602ade3aa30de17abd53b85968a3feb8fdfbcBenjamin Kramer
7203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // determine the type of the relocation
72128f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola
72228f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola  MCSymbolRefExpr::VariantKind Modifier = Target.getSymA()->getKind();
72363d37b932278724d51e7f8bd1f2574c7b45f85f0Benjamin Kramer  unsigned Type;
724e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer  if (Is64Bit) {
725e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    if (IsPCRel) {
72692bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola      switch (Modifier) {
7279edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      default:
7289edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        llvm_unreachable("Unimplemented");
72992bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola      case MCSymbolRefExpr::VK_None:
73092bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola        Type = ELF::R_X86_64_PC32;
73192bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola        break;
73292bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola      case MCSymbolRefExpr::VK_PLT:
73392bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola        Type = ELF::R_X86_64_PLT32;
73492bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola        break;
735a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola      case MCSymbolRefExpr::VK_GOTPCREL:
736607d1f6d3216b647c07965bafae8d1aba6312136Rafael Espindola        Type = ELF::R_X86_64_GOTPCREL;
737607d1f6d3216b647c07965bafae8d1aba6312136Rafael Espindola        break;
738bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola      case MCSymbolRefExpr::VK_GOTTPOFF:
739bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        Type = ELF::R_X86_64_GOTTPOFF;
740bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        break;
741bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola      case MCSymbolRefExpr::VK_TLSGD:
742bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        Type = ELF::R_X86_64_TLSGD;
743bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        break;
744b4d1721eff7b43577e5f2e53f885973fb6c43683Rafael Espindola      case MCSymbolRefExpr::VK_TLSLD:
745b4d1721eff7b43577e5f2e53f885973fb6c43683Rafael Espindola        Type = ELF::R_X86_64_TLSLD;
746b4d1721eff7b43577e5f2e53f885973fb6c43683Rafael Espindola        break;
74792bf6684f62e1df48ecc0a9b3cc3a99ce7c00747Rafael Espindola      }
748e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    } else {
749e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      switch ((unsigned)Fixup.getKind()) {
750e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      default: llvm_unreachable("invalid fixup kind!");
751e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_8: Type = ELF::R_X86_64_64; break;
752a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola      case X86::reloc_signed_4byte:
753e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case X86::reloc_pcrel_4byte:
754a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola        assert(isInt<32>(Target.getConstant()));
75528f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola        switch (Modifier) {
7569edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        default:
7579edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola          llvm_unreachable("Unimplemented");
75828f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola        case MCSymbolRefExpr::VK_None:
75928f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola          Type = ELF::R_X86_64_32S;
76028f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola          break;
76128f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola        case MCSymbolRefExpr::VK_GOT:
76228f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola          Type = ELF::R_X86_64_GOT32;
76328f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola          break;
764c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        case MCSymbolRefExpr::VK_GOTPCREL:
7658cecf253e45f144b9a7fd0ace85eeeeb0bebfc83Rafael Espindola          Type = ELF::R_X86_64_GOTPCREL;
7668cecf253e45f144b9a7fd0ace85eeeeb0bebfc83Rafael Espindola          break;
767bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola        case MCSymbolRefExpr::VK_TPOFF:
768bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola          Type = ELF::R_X86_64_TPOFF32;
769bc82d8b84f6ae15985d1b01e720ed5c37d714012Rafael Espindola          break;
770aa8f1f01352bdaaabf712369f8a8a615c776b508Rafael Espindola        case MCSymbolRefExpr::VK_DTPOFF:
771aa8f1f01352bdaaabf712369f8a8a615c776b508Rafael Espindola          Type = ELF::R_X86_64_DTPOFF32;
772aa8f1f01352bdaaabf712369f8a8a615c776b508Rafael Espindola          break;
77328f9ac81012e1e278128a4148e93f6ab873f15d8Rafael Espindola        }
774a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola        break;
775e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_4:
776a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola        Type = ELF::R_X86_64_32;
777e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer        break;
778e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_2: Type = ELF::R_X86_64_16; break;
779e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case X86::reloc_pcrel_1byte:
780e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_1: Type = ELF::R_X86_64_8; break;
781e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      }
782e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    }
7833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  } else {
784e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    if (IsPCRel) {
7859edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      switch (Modifier) {
7869edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      default:
78724dc9ecdc55d732ac93fc782d6232413fe2c1160Rafael Espindola        llvm_unreachable("Unimplemented");
78824dc9ecdc55d732ac93fc782d6232413fe2c1160Rafael Espindola      case MCSymbolRefExpr::VK_None:
7899baee3b3a3924bc45d007e54c71333739d135f8eRafael Espindola        Type = ELF::R_386_PC32;
7909baee3b3a3924bc45d007e54c71333739d135f8eRafael Espindola        break;
7919edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      case MCSymbolRefExpr::VK_PLT:
7929edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        Type = ELF::R_386_PLT32;
7939edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        break;
7949edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola      }
795e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer    } else {
796e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      switch ((unsigned)Fixup.getKind()) {
797e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      default: llvm_unreachable("invalid fixup kind!");
798a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola
79924ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola      case X86::reloc_global_offset_table:
80024ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola        Type = ELF::R_386_GOTPC;
80124ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola        break;
80224ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola
803a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola      // FIXME: Should we avoid selecting reloc_signed_4byte in 32 bit mode
804a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola      // instead?
805a8c02c3bdd68e65d14fb6b0d56989663754059b0Rafael Espindola      case X86::reloc_signed_4byte:
806e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case X86::reloc_pcrel_4byte:
807bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola      case FK_Data_4:
808c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        switch (Modifier) {
8099edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola        default:
8109edab3a9e15c40c1c9bf70df81c6afdab1cd02c2Rafael Espindola          llvm_unreachable("Unimplemented");
811bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola        case MCSymbolRefExpr::VK_None:
81224ba4f7f5f27c8d64be7ad653863b33594e5f019Rafael Espindola          Type = ELF::R_386_32;
813bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola          break;
814eada30479399014c22c7b7edb008177c312eefeeRafael Espindola        case MCSymbolRefExpr::VK_GOT:
815eada30479399014c22c7b7edb008177c312eefeeRafael Espindola          Type = ELF::R_386_GOT32;
816eada30479399014c22c7b7edb008177c312eefeeRafael Espindola          break;
817c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        case MCSymbolRefExpr::VK_GOTOFF:
818c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola          Type = ELF::R_386_GOTOFF;
819c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola          break;
8203cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola        case MCSymbolRefExpr::VK_TLSGD:
8213cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          Type = ELF::R_386_TLS_GD;
8223cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          break;
8233cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola        case MCSymbolRefExpr::VK_TPOFF:
8243cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          Type = ELF::R_386_TLS_LE_32;
8253cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          break;
8263cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola        case MCSymbolRefExpr::VK_INDNTPOFF:
8273cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          Type = ELF::R_386_TLS_IE;
8283cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          break;
8293cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola        case MCSymbolRefExpr::VK_NTPOFF:
8303cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          Type = ELF::R_386_TLS_LE;
8313cede2d0b2b6cc0a06f55da7c2f8e4263ec0091eRafael Espindola          break;
832a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola        case MCSymbolRefExpr::VK_GOTNTPOFF:
833a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola          Type = ELF::R_386_TLS_GOTIE;
834a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola          break;
835a264f72d3fb9dec1427480fcf17ef3c746ea723aRafael Espindola        case MCSymbolRefExpr::VK_TLSLDM:
836a264f72d3fb9dec1427480fcf17ef3c746ea723aRafael Espindola          Type = ELF::R_386_TLS_LDM;
837a264f72d3fb9dec1427480fcf17ef3c746ea723aRafael Espindola          break;
8380cf15d61b7e3bf53f5a99f58ada37b93bc039559Rafael Espindola        case MCSymbolRefExpr::VK_DTPOFF:
8390cf15d61b7e3bf53f5a99f58ada37b93bc039559Rafael Espindola          Type = ELF::R_386_TLS_LDO_32;
8400cf15d61b7e3bf53f5a99f58ada37b93bc039559Rafael Espindola          break;
841c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        }
842c97f80efc80030c7544a9903c79d2dccd197a0ffRafael Espindola        break;
843e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_2: Type = ELF::R_386_16; break;
844e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case X86::reloc_pcrel_1byte:
845e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      case FK_Data_1: Type = ELF::R_386_8; break;
846e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer      }
8473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
8483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
8493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
850a0a2f8734cdfc19d44201b791a969bcdda96bb70Rafael Espindola  if (RelocNeedsGOT(Modifier))
8515c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    NeedsGOT = true;
8525c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola
853e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer  ELFRelocationEntry ERE;
854e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer
8558f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  ERE.Index = Index;
8568f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  ERE.Type = Type;
85729129728f1d4febe44792d1481077cc47fe83a08Rafael Espindola  ERE.Symbol = Renamed;
8583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
85963d37b932278724d51e7f8bd1f2574c7b45f85f0Benjamin Kramer  ERE.r_offset = Layout.getFragmentOffset(Fragment) + Fixup.getOffset();
860e5b57347e9485b59dd8d70f8f90d7794f473147aBenjamin Kramer
8613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  if (HasRelocationAddend)
8623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ERE.r_addend = Addend;
863172d7d6a12b313dd5a8c1588b8f6bd49b1ce16b6Benjamin Kramer  else
864172d7d6a12b313dd5a8c1588b8f6bd49b1ce16b6Benjamin Kramer    ERE.r_addend = 0; // Silence compiler warning.
8653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
8663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Relocations[Fragment->getParent()].push_back(ERE);
8673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
8683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
8690b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin Krameruint64_t
8700b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin KramerELFObjectWriterImpl::getSymbolIndexInSymbolTable(const MCAssembler &Asm,
8710b6cbfe04cb36872255aff11a41a2b976ceba462Benjamin Kramer                                                 const MCSymbol *S) {
8727b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer  MCSymbolData &SD = Asm.getSymbolData(*S);
8737b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer
8747b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer  // Local symbol.
8757b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer  if (!SD.isExternal() && !S->isUndefined())
8767b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer    return SD.getIndex() + /* empty symbol */ 1;
8777b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer
8787b83c26051b0474fb5f8b73ba6e74bd4f40324baBenjamin Kramer  // External or undefined symbol.
8798f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  return SD.getIndex() + NumRegularSections + /* empty symbol */ 1;
8803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
8813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
882737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindolastatic bool isInSymtab(const MCAssembler &Asm, const MCSymbolData &Data,
88388182132470527e27231f09b25a885893e528c66Rafael Espindola                       bool Used, bool Renamed) {
884484291c27319668ad99cb87def000254357736fbRafael Espindola  if (Data.getFlags() & ELF_Other_Weakref)
885484291c27319668ad99cb87def000254357736fbRafael Espindola    return false;
886484291c27319668ad99cb87def000254357736fbRafael Espindola
887bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola  if (Used)
888bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola    return true;
889bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola
89088182132470527e27231f09b25a885893e528c66Rafael Espindola  if (Renamed)
89188182132470527e27231f09b25a885893e528c66Rafael Espindola    return false;
89288182132470527e27231f09b25a885893e528c66Rafael Espindola
893737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  const MCSymbol &Symbol = Data.getSymbol();
894a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola
895d179886f0595eb3564a9edfdfff79def130d89ccRafael Espindola  if (Symbol.getName() == "_GLOBAL_OFFSET_TABLE_")
896d179886f0595eb3564a9edfdfff79def130d89ccRafael Espindola    return true;
897d179886f0595eb3564a9edfdfff79def130d89ccRafael Espindola
898a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola  const MCSymbol &A = AliasedSymbol(Symbol);
899d179886f0595eb3564a9edfdfff79def130d89ccRafael Espindola  if (!A.isVariable() && A.isUndefined() && !Data.isCommon())
900a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    return false;
901a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola
902737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  if (!Asm.isSymbolLinkerVisible(Symbol) && !Symbol.isUndefined())
903737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola    return false;
904737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
905bd70118f163f94a4d77eb036e6f83dd0bfcb6268Rafael Espindola  if (Symbol.isTemporary())
906737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola    return false;
907737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
908737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  return true;
909737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola}
910737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
911737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindolastatic bool isLocal(const MCSymbolData &Data) {
912737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  if (Data.isExternal())
913737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola    return false;
914737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
915737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  const MCSymbol &Symbol = Data.getSymbol();
916737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  if (Symbol.isUndefined() && !Symbol.isVariable())
917737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola    return false;
918737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
919737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola  return true;
920737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola}
921737cd213e359c2862253c3a1ee443419566e90b9Rafael Espindola
9223565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::ComputeSymbolTable(MCAssembler &Asm) {
9235c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola  // FIXME: Is this the correct place to do this?
9245c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola  if (NeedsGOT) {
9255c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    llvm::StringRef Name = "_GLOBAL_OFFSET_TABLE_";
9265c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    MCSymbol *Sym = Asm.getContext().GetOrCreateSymbol(Name);
9275c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    MCSymbolData &Data = Asm.getOrCreateSymbolData(*Sym);
9285c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola    Data.setExternal(true);
929f571f9a8fe764d5010970e45203415cb00eab739Rafael Espindola    SetBinding(Data, ELF::STB_GLOBAL);
9305c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola  }
9315c77c16f311d702a315547c0eb32b7a34a9d55c8Rafael Espindola
9323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Build section lookup table.
9338f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  NumRegularSections = Asm.size();
934f5c347dfc25e133e357c7086ad1b24edad4d2bc8Rafael Espindola  DenseMap<const MCSection*, uint32_t> SectionIndexMap;
9353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  unsigned Index = 1;
9363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::iterator it = Asm.begin(),
9373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it, ++Index)
9383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    SectionIndexMap[&it->getSection()] = Index;
9393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
9403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Index 0 is always the empty string.
9413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StringMap<uint64_t> StringIndexMap;
9423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StringTable += '\x00';
9433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
944a0949b50dcea35c08b50542091f97275f401529dRafael Espindola  // Add the data for the symbols.
9453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::symbol_iterator it = Asm.symbol_begin(),
9463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.symbol_end(); it != ie; ++it) {
9473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSymbol &Symbol = it->getSymbol();
9483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
949484291c27319668ad99cb87def000254357736fbRafael Espindola    bool Used = UsedInReloc.count(&Symbol);
950484291c27319668ad99cb87def000254357736fbRafael Espindola    bool WeakrefUsed = WeakrefUsedInReloc.count(&Symbol);
951484291c27319668ad99cb87def000254357736fbRafael Espindola    if (!isInSymtab(Asm, *it, Used || WeakrefUsed,
95288182132470527e27231f09b25a885893e528c66Rafael Espindola                    Renames.count(&Symbol)))
9533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      continue;
9543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
9553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFSymbolData MSD;
9563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MSD.SymbolData = it;
957a0949b50dcea35c08b50542091f97275f401529dRafael Espindola    bool Local = isLocal(*it);
95888182132470527e27231f09b25a885893e528c66Rafael Espindola    const MCSymbol &RefSymbol = AliasedSymbol(Symbol);
9593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
960484291c27319668ad99cb87def000254357736fbRafael Espindola    if (RefSymbol.isUndefined() && !Used && WeakrefUsed)
961484291c27319668ad99cb87def000254357736fbRafael Espindola      SetBinding(*it, ELF::STB_WEAK);
962484291c27319668ad99cb87def000254357736fbRafael Espindola
963f7c10a3cff61c70efe8e405d9bdc5386e8e3fc0aRafael Espindola    if (it->isCommon()) {
964a0949b50dcea35c08b50542091f97275f401529dRafael Espindola      assert(!Local);
965f7c10a3cff61c70efe8e405d9bdc5386e8e3fc0aRafael Espindola      MSD.SectionIndex = ELF::SHN_COMMON;
966bf052ac5d1c8f21075bc675f629709c20791c5f7Rafael Espindola    } else if (Symbol.isAbsolute() || RefSymbol.isVariable()) {
967a0949b50dcea35c08b50542091f97275f401529dRafael Espindola      MSD.SectionIndex = ELF::SHN_ABS;
96888182132470527e27231f09b25a885893e528c66Rafael Espindola    } else if (RefSymbol.isUndefined()) {
9693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      MSD.SectionIndex = ELF::SHN_UNDEF;
9703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    } else {
971bf052ac5d1c8f21075bc675f629709c20791c5f7Rafael Espindola      MSD.SectionIndex = SectionIndexMap.lookup(&RefSymbol.getSection());
9727be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola      if (MSD.SectionIndex >= ELF::SHN_LORESERVE)
9737be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola        NeedsSymtabShndx = true;
9743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      assert(MSD.SectionIndex && "Invalid section index!");
9755df0b65e8f5ef55f0b515a0b24879f5d117d6042Rafael Espindola    }
9765df0b65e8f5ef55f0b515a0b24879f5d117d6042Rafael Espindola
97788182132470527e27231f09b25a885893e528c66Rafael Espindola    // The @@@ in symbol version is replaced with @ in undefined symbols and
97888182132470527e27231f09b25a885893e528c66Rafael Espindola    // @@ in defined ones.
97988182132470527e27231f09b25a885893e528c66Rafael Espindola    StringRef Name = Symbol.getName();
98088182132470527e27231f09b25a885893e528c66Rafael Espindola    size_t Pos = Name.find("@@@");
98188182132470527e27231f09b25a885893e528c66Rafael Espindola    std::string FinalName;
98288182132470527e27231f09b25a885893e528c66Rafael Espindola    if (Pos != StringRef::npos) {
98307ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer      StringRef Prefix = Name.substr(0, Pos);
98488182132470527e27231f09b25a885893e528c66Rafael Espindola      unsigned n = MSD.SectionIndex == ELF::SHN_UNDEF ? 2 : 1;
98507ee63283cc2b9a4d6c98ecd30d1b721fe3cb29aBenjamin Kramer      StringRef Suffix = Name.substr(Pos + n);
98688182132470527e27231f09b25a885893e528c66Rafael Espindola      FinalName = Prefix.str() + Suffix.str();
98788182132470527e27231f09b25a885893e528c66Rafael Espindola    } else {
98888182132470527e27231f09b25a885893e528c66Rafael Espindola      FinalName = Name.str();
98988182132470527e27231f09b25a885893e528c66Rafael Espindola    }
99088182132470527e27231f09b25a885893e528c66Rafael Espindola
99188182132470527e27231f09b25a885893e528c66Rafael Espindola    uint64_t &Entry = StringIndexMap[FinalName];
992a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    if (!Entry) {
993a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      Entry = StringTable.size();
99488182132470527e27231f09b25a885893e528c66Rafael Espindola      StringTable += FinalName;
995a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      StringTable += '\x00';
9963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
997a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    MSD.StringIndex = Entry;
998a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    if (MSD.SectionIndex == ELF::SHN_UNDEF)
999a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      UndefinedSymbolData.push_back(MSD);
1000a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    else if (Local)
1001a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      LocalSymbolData.push_back(MSD);
1002a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola    else
1003a6866969ba3b12901b412bb53ba8c61a9991b8adRafael Espindola      ExternalSymbolData.push_back(MSD);
10043565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
10053565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10063565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Symbols are required to be in lexicographic order.
10073565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  array_pod_sort(LocalSymbolData.begin(), LocalSymbolData.end());
10083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  array_pod_sort(ExternalSymbolData.begin(), ExternalSymbolData.end());
10093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  array_pod_sort(UndefinedSymbolData.begin(), UndefinedSymbolData.end());
10103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Set the symbol indices. Local symbols must come before all other
10123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // symbols with non-local bindings.
10133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Index = 0;
10143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = LocalSymbolData.size(); i != e; ++i)
10153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    LocalSymbolData[i].SymbolData->setIndex(Index++);
10163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = ExternalSymbolData.size(); i != e; ++i)
10173565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ExternalSymbolData[i].SymbolData->setIndex(Index++);
10183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = UndefinedSymbolData.size(); i != e; ++i)
10193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    UndefinedSymbolData[i].SymbolData->setIndex(Index++);
10203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
10213565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10223565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteRelocation(MCAssembler &Asm, MCAsmLayout &Layout,
10233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                          const MCSectionData &SD) {
10243565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  if (!Relocations[&SD].empty()) {
10253565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCContext &Ctx = Asm.getContext();
10264283f4b81e8c1cbf5c7a7b51e949e109ae25ff8cRafael Espindola    const MCSectionELF *RelaSection;
10273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionELF &Section =
10283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      static_cast<const MCSectionELF&>(SD.getSection());
10293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const StringRef SectionName = Section.getSectionName();
1031377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer    std::string RelaSectionName = HasRelocationAddend ? ".rela" : ".rel";
10323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    RelaSectionName += SectionName;
1033299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer
1034299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer    unsigned EntrySize;
1035299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer    if (HasRelocationAddend)
1036299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer      EntrySize = Is64Bit ? sizeof(ELF::Elf64_Rela) : sizeof(ELF::Elf32_Rela);
1037299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer    else
1038299fbe3a5aad8f08c52c488a9b28df70c7595e0dBenjamin Kramer      EntrySize = Is64Bit ? sizeof(ELF::Elf64_Rel) : sizeof(ELF::Elf32_Rel);
10393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1040377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer    RelaSection = Ctx.getELFSection(RelaSectionName, HasRelocationAddend ?
1041377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer                                    ELF::SHT_RELA : ELF::SHT_REL, 0,
10423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                    SectionKind::getReadOnly(),
104334be396a12b00a95a1353c356d64868798ea3098Rafael Espindola                                    EntrySize);
10443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCSectionData &RelaSD = Asm.getOrCreateSectionData(*RelaSection);
1046a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    RelaSD.setAlignment(Is64Bit ? 8 : 4);
10473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    MCDataFragment *F = new MCDataFragment(&RelaSD);
10493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    WriteRelocationsFragment(Asm, F, &SD);
10513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10527070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    Asm.AddSectionToTheEnd(*Writer, RelaSD, Layout);
10533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
10543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
10553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10563565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteSecHdrEntry(uint32_t Name, uint32_t Type,
10573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t Flags, uint64_t Address,
10583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t Offset, uint64_t Size,
10593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint32_t Link, uint32_t Info,
10603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t Alignment,
10613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                           uint64_t EntrySize) {
10623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(Name);        // sh_name: index into string table
10633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(Type);        // sh_type
10643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Flags);     // sh_flags
10653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Address);   // sh_addr
10663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Offset);    // sh_offset
10673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Size);      // sh_size
10683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(Link);        // sh_link
10693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  Write32(Info);        // sh_info
10703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(Alignment); // sh_addralign
10713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  WriteWord(EntrySize); // sh_entsize
10723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
10733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10743565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::WriteRelocationsFragment(const MCAssembler &Asm,
10753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                   MCDataFragment *F,
10763565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                   const MCSectionData *SD) {
10773565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  std::vector<ELFRelocationEntry> &Relocs = Relocations[SD];
10783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // sort by the r_offset just like gnu as does
10793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  array_pod_sort(Relocs.begin(), Relocs.end());
10803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (unsigned i = 0, e = Relocs.size(); i != e; ++i) {
10823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    ELFRelocationEntry entry = Relocs[e - i - 1];
10833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
10848f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    if (entry.Index < 0)
10858f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      entry.Index = getSymbolIndexInSymbolTable(Asm, entry.Symbol);
10868f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola    else
10878f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      entry.Index += LocalSymbolData.size() + 1;
10885e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer    if (Is64Bit) {
1089af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      String64(*F, entry.r_offset);
10905e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
10918f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      struct ELF::Elf64_Rela ERE64;
10928f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      ERE64.setSymbolAndType(entry.Index, entry.Type);
1093af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      String64(*F, ERE64.r_info);
10945e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
1095af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      if (HasRelocationAddend)
1096af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola        String64(*F, entry.r_addend);
10975e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer    } else {
1098af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      String32(*F, entry.r_offset);
10995e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
11008f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      struct ELF::Elf32_Rela ERE32;
11018f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola      ERE32.setSymbolAndType(entry.Index, entry.Type);
1102af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      String32(*F, ERE32.r_info);
11035e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer
1104af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola      if (HasRelocationAddend)
1105af3d38f70ebba9fec0048c2540dbdf0168d280d2Rafael Espindola        String32(*F, entry.r_addend);
11065e492e8d42fa1f2940165a937d3d613c61b57708Benjamin Kramer    }
11073565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
11083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
11093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11103565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriterImpl::CreateMetadataSections(MCAssembler &Asm,
11113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                 MCAsmLayout &Layout) {
11123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  MCContext &Ctx = Asm.getContext();
11133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  MCDataFragment *F;
11143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11153565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  unsigned EntrySize = Is64Bit ? ELF::SYMENTRY_SIZE64 : ELF::SYMENTRY_SIZE32;
11163565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
111771859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  unsigned NumRegularSections = Asm.size();
111871859c640f6a36251aca223fd503c58dc314e296Rafael Espindola
111938738bf1a847292003a5495f17e42a75d1274bf7Rafael Espindola  // We construct .shstrtab, .symtab and .strtab in this order to match gnu as.
11204283f4b81e8c1cbf5c7a7b51e949e109ae25ff8cRafael Espindola  const MCSectionELF *ShstrtabSection =
11217be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    Ctx.getELFSection(".shstrtab", ELF::SHT_STRTAB, 0,
11227be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                      SectionKind::getReadOnly(), false);
112371859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  MCSectionData &ShstrtabSD = Asm.getOrCreateSectionData(*ShstrtabSection);
112471859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  ShstrtabSD.setAlignment(1);
112571859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  ShstrtabIndex = Asm.size();
112671859c640f6a36251aca223fd503c58dc314e296Rafael Espindola
11274283f4b81e8c1cbf5c7a7b51e949e109ae25ff8cRafael Espindola  const MCSectionELF *SymtabSection =
11287be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    Ctx.getELFSection(".symtab", ELF::SHT_SYMTAB, 0,
11297be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola                      SectionKind::getReadOnly(),
113034be396a12b00a95a1353c356d64868798ea3098Rafael Espindola                      EntrySize);
11313565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  MCSectionData &SymtabSD = Asm.getOrCreateSectionData(*SymtabSection);
11323565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  SymtabSD.setAlignment(Is64Bit ? 8 : 4);
11337be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  SymbolTableIndex = Asm.size();
11347be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola
11357be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  MCSectionData *SymtabShndxSD = NULL;
11367be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola
11377be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  if (NeedsSymtabShndx) {
11384283f4b81e8c1cbf5c7a7b51e949e109ae25ff8cRafael Espindola    const MCSectionELF *SymtabShndxSection =
11397be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola      Ctx.getELFSection(".symtab_shndx", ELF::SHT_SYMTAB_SHNDX, 0,
114034be396a12b00a95a1353c356d64868798ea3098Rafael Espindola                        SectionKind::getReadOnly(), 4);
11417be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    SymtabShndxSD = &Asm.getOrCreateSectionData(*SymtabShndxSection);
11427be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    SymtabShndxSD->setAlignment(4);
11437be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  }
11443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  const MCSection *StrtabSection;
11463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StrtabSection = Ctx.getELFSection(".strtab", ELF::SHT_STRTAB, 0,
11473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                    SectionKind::getReadOnly(), false);
11483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  MCSectionData &StrtabSD = Asm.getOrCreateSectionData(*StrtabSection);
11493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StrtabSD.setAlignment(1);
11503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  StringTableIndex = Asm.size();
11513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1152c3c413f63117896f186fc4385bdaac0578d3613fRafael Espindola  WriteRelocations(Asm, Layout);
115371859c640f6a36251aca223fd503c58dc314e296Rafael Espindola
115471859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  // Symbol table
115571859c640f6a36251aca223fd503c58dc314e296Rafael Espindola  F = new MCDataFragment(&SymtabSD);
11567be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  MCDataFragment *ShndxF = NULL;
11577be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  if (NeedsSymtabShndx) {
11587be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    ShndxF = new MCDataFragment(SymtabShndxSD);
11597be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    Asm.AddSectionToTheEnd(*Writer, *SymtabShndxSD, Layout);
11607be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  }
11617be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  WriteSymbolTable(F, ShndxF, Asm, Layout, NumRegularSections);
11627070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  Asm.AddSectionToTheEnd(*Writer, SymtabSD, Layout);
116371859c640f6a36251aca223fd503c58dc314e296Rafael Espindola
11643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  F = new MCDataFragment(&StrtabSD);
11653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  F->getContents().append(StringTable.begin(), StringTable.end());
11667070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  Asm.AddSectionToTheEnd(*Writer, StrtabSD, Layout);
11673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  F = new MCDataFragment(&ShstrtabSD);
11693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Section header string table.
11713565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  //
11723565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // The first entry of a string table holds a null character so skip
11733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // section 0.
11743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  uint64_t Index = 1;
11753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  F->getContents() += '\x00';
11763565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11773565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::const_iterator it = Asm.begin(),
11783565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it) {
11793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionELF &Section =
1180368ae7e4ec0cc0dfc404fa1d07cdf94a7f2a15beBenjamin Kramer      static_cast<const MCSectionELF&>(it->getSection());
118151efe7a253190b672519e8388ce5d45f1dcf1a24Rafael Espindola    // FIXME: We could merge suffixes like in .text and .rela.text.
11823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // Remember the index into the string table so we can write it
11843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // into the sh_name field of the section header table.
11853565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    SectionStringTableIndex[&it->getSection()] = Index;
11863565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11873565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    Index += Section.getSectionName().size() + 1;
11883565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += Section.getSectionName();
11893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    F->getContents() += '\x00';
11903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
11913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
11927070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  Asm.AddSectionToTheEnd(*Writer, ShstrtabSD, Layout);
11937070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola}
11947070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
11957070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindolabool ELFObjectWriterImpl::IsFixupFullyResolved(const MCAssembler &Asm,
11967070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                               const MCValue Target,
11977070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                               bool IsPCRel,
11987070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                               const MCFragment *DF) const {
11997070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  // If this is a PCrel relocation, find the section this fixup value is
12007070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  // relative to.
12017070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSection *BaseSection = 0;
12027070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (IsPCRel) {
12037070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    BaseSection = &DF->getParent()->getSection();
12047070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    assert(BaseSection);
12057070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  }
12067070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
12077070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSection *SectionA = 0;
12087070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSymbol *SymbolA = 0;
12097070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (const MCSymbolRefExpr *A = Target.getSymA()) {
12107070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    SymbolA = &A->getSymbol();
12117070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    SectionA = &SymbolA->getSection();
12127070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  }
12137070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
12147070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSection *SectionB = 0;
12157070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (const MCSymbolRefExpr *B = Target.getSymB()) {
12167070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    SectionB = &B->getSymbol().getSection();
12177070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  }
12187070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
12197070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (!BaseSection)
12207070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    return SectionA == SectionB;
12217070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
12227070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  const MCSymbolData &DataA = Asm.getSymbolData(*SymbolA);
12237070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  if (DataA.isExternal())
12247070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola    return false;
12257070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
12267070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  return !SectionB && BaseSection == SectionA;
12273565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
12283565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12298f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindolavoid ELFObjectWriterImpl::WriteObject(MCAssembler &Asm,
12303565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                      const MCAsmLayout &Layout) {
12318f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  // Compute symbol table information.
12328f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola  ComputeSymbolTable(Asm);
12338f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindola
12343565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  CreateMetadataSections(const_cast<MCAssembler&>(Asm),
12353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                         const_cast<MCAsmLayout&>(Layout));
12363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Add 1 for the null section.
12383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  unsigned NumSections = Asm.size() + 1;
1239a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  uint64_t NaturalAlignment = Is64Bit ? 8 : 4;
1240a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  uint64_t HeaderSize = Is64Bit ? sizeof(ELF::Elf64_Ehdr) : sizeof(ELF::Elf32_Ehdr);
1241a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  uint64_t FileOff = HeaderSize;
12423565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12433565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::const_iterator it = Asm.begin(),
12443565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it) {
12453565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionData &SD = *it;
12463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1247a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    FileOff = RoundUpToAlignment(FileOff, SD.getAlignment());
1248a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
12493565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // Get the size of the section in the output file (including padding).
12503565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    uint64_t Size = Layout.getSectionFileSize(&SD);
1251a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
1252a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    FileOff += Size;
12533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
12543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1255a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  FileOff = RoundUpToAlignment(FileOff, NaturalAlignment);
1256a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
12573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Write out the ELF header ...
1258a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  WriteHeader(FileOff - HeaderSize, NumSections);
1259a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
1260a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  FileOff = HeaderSize;
12613565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ... then all of the sections ...
12633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  DenseMap<const MCSection*, uint64_t> SectionOffsetMap;
12643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1265d8e0bfe07a7d101eeed7204d1dfcaae17ad51908Rafael Espindola  DenseMap<const MCSection*, uint32_t> SectionIndexMap;
126644cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer
126744cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer  unsigned Index = 1;
12683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::const_iterator it = Asm.begin(),
12693565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it) {
1270a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    const MCSectionData &SD = *it;
1271a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
1272a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    uint64_t Padding = OffsetToAlignment(FileOff, SD.getAlignment());
1273a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    WriteZeros(Padding);
1274a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer    FileOff += Padding;
1275a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
12763565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    // Remember the offset into the file for this section.
12773565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    SectionOffsetMap[&it->getSection()] = FileOff;
127844cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer    SectionIndexMap[&it->getSection()] = Index++;
127944cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer
12803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    FileOff += Layout.getSectionFileSize(&SD);
12813565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    Asm.WriteSectionData(it, Layout, Writer);
12833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
12843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1285a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  uint64_t Padding = OffsetToAlignment(FileOff, NaturalAlignment);
1286a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  WriteZeros(Padding);
1287a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer  FileOff += Padding;
1288a9eadca2fdf48e3228521f25b415b68219bcee31Benjamin Kramer
12893565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // ... and then the section header table.
12903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Should we align the section header table?
12913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  //
12923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  // Null section first.
12937be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  uint64_t FirstSectionSize =
12947be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    NumSections >= ELF::SHN_LORESERVE ? NumSections : 0;
12957be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  uint32_t FirstSectionLink =
12967be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    ShstrtabIndex >= ELF::SHN_LORESERVE ? ShstrtabIndex : 0;
12977be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola  WriteSecHdrEntry(0, 0, 0, 0, 0, FirstSectionSize, FirstSectionLink, 0, 0, 0);
12983565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
12993565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  for (MCAssembler::const_iterator it = Asm.begin(),
13003565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming         ie = Asm.end(); it != ie; ++it) {
13013565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionData &SD = *it;
13023565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    const MCSectionELF &Section =
13033565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      static_cast<const MCSectionELF&>(SD.getSection());
13043565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13053565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    uint64_t sh_link = 0;
13063565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    uint64_t sh_info = 0;
13073565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    switch(Section.getType()) {
13093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_DYNAMIC:
13103565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      sh_link = SectionStringTableIndex[&it->getSection()];
13113565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      sh_info = 0;
13123565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
13133565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13143565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_REL:
1315f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman    case ELF::SHT_RELA: {
13164283f4b81e8c1cbf5c7a7b51e949e109ae25ff8cRafael Espindola      const MCSectionELF *SymtabSection;
13174283f4b81e8c1cbf5c7a7b51e949e109ae25ff8cRafael Espindola      const MCSectionELF *InfoSection;
13183565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13193565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      SymtabSection = Asm.getContext().getELFSection(".symtab", ELF::SHT_SYMTAB, 0,
13203565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                     SectionKind::getReadOnly(),
1321f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman                                                     false);
132244cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer      sh_link = SectionIndexMap[SymtabSection];
13233565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
1324377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer      // Remove ".rel" and ".rela" prefixes.
1325377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer      unsigned SecNameLen = (Section.getType() == ELF::SHT_REL) ? 4 : 5;
1326377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer      StringRef SectionName = Section.getSectionName().substr(SecNameLen);
1327377a572d9b92304a0fde4371d726721586c1a5bdBenjamin Kramer
1328f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman      InfoSection = Asm.getContext().getELFSection(SectionName,
13293565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                   ELF::SHT_PROGBITS, 0,
1330f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman                                                   SectionKind::getReadOnly(),
1331f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman                                                   false);
133244cbde85badb60c7078e37e14575c15e671521b1Benjamin Kramer      sh_info = SectionIndexMap[InfoSection];
13333565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
1334f8020a3978b9a56074a3a5f9821c63165e37bff7Eli Friedman    }
13353565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13363565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_SYMTAB:
13373565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_DYNSYM:
13383565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      sh_link = StringTableIndex;
13393565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      sh_info = LastLocalSymbolIndex;
13403565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
13413565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13427be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola    case ELF::SHT_SYMTAB_SHNDX:
13437be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola      sh_link = SymbolTableIndex;
13447be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola      break;
13457be2c33193aeada8c0ff3555ea54b930f204132aRafael Espindola
13463565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_PROGBITS:
13473565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_STRTAB:
13483565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    case ELF::SHT_NOBITS:
134919dc7faf25e3a8cc37a656b74ebeef3ab976aa8eBenjamin Kramer    case ELF::SHT_NULL:
1350cecbc3d28277ff4916326311cbf87335ed05d106Rafael Espindola    case ELF::SHT_ARM_ATTRIBUTES:
13513565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      // Nothing to do.
13523565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
13533565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13543565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    default:
13553565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      assert(0 && "FIXME: sh_type value not supported!");
13563565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming      break;
13573565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    }
13583565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13593565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming    WriteSecHdrEntry(SectionStringTableIndex[&it->getSection()],
13603565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     Section.getType(), Section.getFlags(),
136171859c640f6a36251aca223fd503c58dc314e296Rafael Espindola                     0,
13623565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     SectionOffsetMap.lookup(&SD.getSection()),
13633565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     Layout.getSectionSize(&SD), sh_link,
13643565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     sh_info, SD.getAlignment(),
13653565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                     Section.getEntrySize());
13663565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  }
13673565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13683565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13693565a06ebf44a193a8b333cbeff2ee154298d450Matt FlemingELFObjectWriter::ELFObjectWriter(raw_ostream &OS,
13703565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                 bool Is64Bit,
13715baf79edc067a4b17d024cc10324ac88c17e3e43Roman Divacky                                 Triple::OSType OSType,
1372eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck                                 uint16_t EMachine,
13733565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                 bool IsLittleEndian,
13743565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                 bool HasRelocationAddend)
13753565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  : MCObjectWriter(OS, IsLittleEndian)
13763565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming{
1377eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck  Impl = new ELFObjectWriterImpl(this, Is64Bit, EMachine,
1378eecb858ca86fa949c06f819d6127e2ac68d165c8Wesley Peck                                 HasRelocationAddend, OSType);
13793565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13803565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13813565a06ebf44a193a8b333cbeff2ee154298d450Matt FlemingELFObjectWriter::~ELFObjectWriter() {
13823565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  delete (ELFObjectWriterImpl*) Impl;
13833565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13843565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13853565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriter::ExecutePostLayoutBinding(MCAssembler &Asm) {
13863565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  ((ELFObjectWriterImpl*) Impl)->ExecutePostLayoutBinding(Asm);
13873565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13883565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13893565a06ebf44a193a8b333cbeff2ee154298d450Matt Flemingvoid ELFObjectWriter::RecordRelocation(const MCAssembler &Asm,
13903565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                       const MCAsmLayout &Layout,
13913565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                       const MCFragment *Fragment,
13923565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                       const MCFixup &Fixup, MCValue Target,
13933565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                       uint64_t &FixedValue) {
13943565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  ((ELFObjectWriterImpl*) Impl)->RecordRelocation(Asm, Layout, Fragment, Fixup,
13953565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                                  Target, FixedValue);
13963565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
13973565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming
13987070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindolabool ELFObjectWriter::IsFixupFullyResolved(const MCAssembler &Asm,
13997070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                           const MCValue Target,
14007070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                           bool IsPCRel,
14017070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                           const MCFragment *DF) const {
14027070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola  return ((ELFObjectWriterImpl*) Impl)->IsFixupFullyResolved(Asm, Target,
14037070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola                                                             IsPCRel, DF);
14047070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola}
14057070387f08f7dc797b554ed8013cba9f8b74121aRafael Espindola
14068f413fa9c00dbaea000ddfe265ab5edd285ea3abRafael Espindolavoid ELFObjectWriter::WriteObject(MCAssembler &Asm,
14073565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming                                  const MCAsmLayout &Layout) {
14083565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming  ((ELFObjectWriterImpl*) Impl)->WriteObject(Asm, Layout);
14093565a06ebf44a193a8b333cbeff2ee154298d450Matt Fleming}
1410