blob: c0e7b93cf118b37b522b626e330f02c16f452107 [file] [log] [blame]
Chris Lattner35f0a4f2005-06-27 06:29:00 +00001//===-- ELFWriter.cpp - Target-independent ELF Writer code ----------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattner35f0a4f2005-06-27 06:29:00 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the target-independent ELF writer. This file writes out
11// the ELF file in the following order:
12//
13// #1. ELF Header
Chris Lattner80ed8fa2005-07-07 07:02:20 +000014// #2. '.text' section
15// #3. '.data' section
16// #4. '.bss' section (conceptual position in file)
Chris Lattner35f0a4f2005-06-27 06:29:00 +000017// ...
18// #X. '.shstrtab' section
19// #Y. Section Table
20//
21// The entries in the section table are laid out as:
22// #0. Null entry [required]
Chris Lattner80ed8fa2005-07-07 07:02:20 +000023// #1. ".text" entry - the program code
24// #2. ".data" entry - global variables with initializers. [ if needed ]
25// #3. ".bss" entry - global variables without initializers. [ if needed ]
Chris Lattner35f0a4f2005-06-27 06:29:00 +000026// ...
27// #N. ".shstrtab" entry - String table for the section names.
Chris Lattner35f0a4f2005-06-27 06:29:00 +000028//
Chris Lattner35f0a4f2005-06-27 06:29:00 +000029//===----------------------------------------------------------------------===//
30
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +000031#define DEBUG_TYPE "elfwriter"
Bruno Cardoso Lopes45f5d642009-07-02 18:29:24 +000032#include "ELF.h"
Bill Wendling8f84f1f2007-02-08 01:35:27 +000033#include "ELFWriter.h"
Bruno Cardoso Lopes4cb31432009-06-03 17:47:27 +000034#include "ELFCodeEmitter.h"
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +000035#include "llvm/Constants.h"
Chris Lattner35f0a4f2005-06-27 06:29:00 +000036#include "llvm/Module.h"
Bill Wendling8f84f1f2007-02-08 01:35:27 +000037#include "llvm/PassManager.h"
Chris Lattner02b73ab2009-01-04 20:19:20 +000038#include "llvm/DerivedTypes.h"
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +000039#include "llvm/CodeGen/BinaryObject.h"
Bill Wendling8f84f1f2007-02-08 01:35:27 +000040#include "llvm/CodeGen/FileWriters.h"
Chris Lattneraa507db2005-07-11 05:17:18 +000041#include "llvm/CodeGen/MachineCodeEmitter.h"
Bruno Cardoso Lopesac57e6e2009-07-06 05:09:34 +000042#include "llvm/CodeGen/ObjectCodeEmitter.h"
43#include "llvm/CodeGen/MachineCodeEmitter.h"
Chris Lattneraa507db2005-07-11 05:17:18 +000044#include "llvm/CodeGen/MachineConstantPool.h"
Bruno Cardoso Lopes45f5d642009-07-02 18:29:24 +000045#include "llvm/Target/TargetAsmInfo.h"
Owen Anderson07000c62006-05-12 06:33:49 +000046#include "llvm/Target/TargetData.h"
Bruno Cardoso Lopes45f5d642009-07-02 18:29:24 +000047#include "llvm/Target/TargetELFWriterInfo.h"
Chris Lattner35f0a4f2005-06-27 06:29:00 +000048#include "llvm/Target/TargetMachine.h"
Chris Lattner5acd1202005-07-11 03:11:47 +000049#include "llvm/Support/Mangler.h"
Bill Wendlingbdc679d2006-11-29 00:39:47 +000050#include "llvm/Support/Streams.h"
Owen Andersoncb371882008-08-21 00:14:44 +000051#include "llvm/Support/raw_ostream.h"
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +000052#include "llvm/Support/Debug.h"
Torok Edwinc25e7582009-07-11 20:10:48 +000053#include "llvm/Support/ErrorHandling.h"
Bruno Cardoso Lopes6933d3e2009-07-06 09:26:48 +000054
Chris Lattner35f0a4f2005-06-27 06:29:00 +000055using namespace llvm;
56
Devang Patel19974732007-05-03 01:11:54 +000057char ELFWriter::ID = 0;
Bruno Cardoso Lopes6933d3e2009-07-06 09:26:48 +000058
59/// AddELFWriter - Add the ELF writer to the function pass manager
Bruno Cardoso Lopesac57e6e2009-07-06 05:09:34 +000060ObjectCodeEmitter *llvm::AddELFWriter(PassManagerBase &PM,
Bruno Cardoso Lopes6933d3e2009-07-06 09:26:48 +000061 raw_ostream &O,
62 TargetMachine &TM) {
Bill Wendling8f84f1f2007-02-08 01:35:27 +000063 ELFWriter *EW = new ELFWriter(O, TM);
Dan Gohmanbfae8312008-03-11 22:29:46 +000064 PM.add(EW);
Bruno Cardoso Lopes6933d3e2009-07-06 09:26:48 +000065 return EW->getObjectCodeEmitter();
Bill Wendling8f84f1f2007-02-08 01:35:27 +000066}
67
Chris Lattner0e180502005-07-11 06:34:30 +000068//===----------------------------------------------------------------------===//
Chris Lattner0e180502005-07-11 06:34:30 +000069// ELFWriter Implementation
70//===----------------------------------------------------------------------===//
Chris Lattneraa507db2005-07-11 05:17:18 +000071
Bruno Cardoso Lopes4cb31432009-06-03 17:47:27 +000072ELFWriter::ELFWriter(raw_ostream &o, TargetMachine &tm)
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +000073 : MachineFunctionPass(&ID), O(o), TM(tm),
74 is64Bit(TM.getTargetData()->getPointerSizeInBits() == 64),
75 isLittleEndian(TM.getTargetData()->isLittleEndian()),
76 ElfHdr(isLittleEndian, is64Bit) {
Chris Lattneraa507db2005-07-11 05:17:18 +000077
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +000078 TAI = TM.getTargetAsmInfo();
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +000079 TEW = TM.getELFWriterInfo();
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +000080
Bruno Cardoso Lopes6933d3e2009-07-06 09:26:48 +000081 // Create the object code emitter object for this target.
82 ElfCE = new ELFCodeEmitter(*this);
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +000083
84 // Inital number of sections
Chris Lattner5f48ff72005-07-16 08:01:13 +000085 NumSections = 0;
Chris Lattneraa507db2005-07-11 05:17:18 +000086}
87
88ELFWriter::~ELFWriter() {
Bruno Cardoso Lopes6933d3e2009-07-06 09:26:48 +000089 delete ElfCE;
Chris Lattner35f0a4f2005-06-27 06:29:00 +000090}
91
92// doInitialization - Emit the file header and all of the global variables for
93// the module to the ELF file.
94bool ELFWriter::doInitialization(Module &M) {
Chris Lattner5acd1202005-07-11 03:11:47 +000095 Mang = new Mangler(M);
96
Bruno Cardoso Lopesf5b0c5a2009-06-06 03:56:29 +000097 // ELF Header
98 // ----------
99 // Fields e_shnum e_shstrndx are only known after all section have
100 // been emitted. They locations in the ouput buffer are recorded so
101 // to be patched up later.
102 //
103 // Note
104 // ----
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000105 // emitWord method behaves differently for ELF32 and ELF64, writing
Bruno Cardoso Lopesf5b0c5a2009-06-06 03:56:29 +0000106 // 4 bytes in the former and 8 in the last for *_off and *_addr elf types
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000107
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000108 ElfHdr.emitByte(0x7f); // e_ident[EI_MAG0]
109 ElfHdr.emitByte('E'); // e_ident[EI_MAG1]
110 ElfHdr.emitByte('L'); // e_ident[EI_MAG2]
111 ElfHdr.emitByte('F'); // e_ident[EI_MAG3]
Bruno Cardoso Lopesf5b0c5a2009-06-06 03:56:29 +0000112
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000113 ElfHdr.emitByte(TEW->getEIClass()); // e_ident[EI_CLASS]
114 ElfHdr.emitByte(TEW->getEIData()); // e_ident[EI_DATA]
115 ElfHdr.emitByte(EV_CURRENT); // e_ident[EI_VERSION]
116 ElfHdr.emitAlignment(16); // e_ident[EI_NIDENT-EI_PAD]
Bruno Cardoso Lopesf5b0c5a2009-06-06 03:56:29 +0000117
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000118 ElfHdr.emitWord16(ET_REL); // e_type
119 ElfHdr.emitWord16(TEW->getEMachine()); // e_machine = target
120 ElfHdr.emitWord32(EV_CURRENT); // e_version
121 ElfHdr.emitWord(0); // e_entry, no entry point in .o file
122 ElfHdr.emitWord(0); // e_phoff, no program header for .o
123 ELFHdr_e_shoff_Offset = ElfHdr.size();
124 ElfHdr.emitWord(0); // e_shoff = sec hdr table off in bytes
125 ElfHdr.emitWord32(TEW->getEFlags()); // e_flags = whatever the target wants
126 ElfHdr.emitWord16(TEW->getHdrSize()); // e_ehsize = ELF header size
127 ElfHdr.emitWord16(0); // e_phentsize = prog header entry size
128 ElfHdr.emitWord16(0); // e_phnum = # prog header entries = 0
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000129
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +0000130 // e_shentsize = Section header entry size
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000131 ElfHdr.emitWord16(TEW->getSHdrSize());
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000132
Bruno Cardoso Lopesf5b0c5a2009-06-06 03:56:29 +0000133 // e_shnum = # of section header ents
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000134 ELFHdr_e_shnum_Offset = ElfHdr.size();
135 ElfHdr.emitWord16(0); // Placeholder
Jeff Cohen00b168892005-07-27 06:12:32 +0000136
Bruno Cardoso Lopesf5b0c5a2009-06-06 03:56:29 +0000137 // e_shstrndx = Section # of '.shstrtab'
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000138 ELFHdr_e_shstrndx_Offset = ElfHdr.size();
139 ElfHdr.emitWord16(0); // Placeholder
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000140
Chris Lattner5f48ff72005-07-16 08:01:13 +0000141 // Add the null section, which is required to be first in the file.
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000142 getNullSection();
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000143
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000144 return false;
145}
146
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000147// Get jump table section on the section name returned by TAI
Bruno Cardoso Lopese2b0ecd2009-07-18 23:24:01 +0000148ELFSection &ELFWriter::getJumpTableSection() {
149 unsigned Align = TM.getTargetData()->getPointerABIAlignment();
150 return getSection(TAI->getJumpTableDataSection(),
151 ELFSection::SHT_PROGBITS,
152 ELFSection::SHF_ALLOC, Align);
153}
154
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000155// Get a constant pool section based on the section name returned by TAI
Bruno Cardoso Lopese2b0ecd2009-07-18 23:24:01 +0000156ELFSection &ELFWriter::getConstantPoolSection(MachineConstantPoolEntry &CPE) {
Chris Lattner5c2f7892009-07-26 06:26:55 +0000157 SectionKind Kind;
158 switch (CPE.getRelocationInfo()) {
159 default: llvm_unreachable("Unknown section kind");
Chris Lattner4c509222009-07-26 07:00:12 +0000160 case 2: Kind = SectionKind::get(SectionKind::ReadOnlyWithRel,false); break;
161 case 1:
162 Kind = SectionKind::get(SectionKind::ReadOnlyWithRelLocal,false);
163 break;
Chris Lattner5c2f7892009-07-26 06:26:55 +0000164 case 0:
165 switch (TM.getTargetData()->getTypeAllocSize(CPE.getType())) {
Chris Lattner4c509222009-07-26 07:00:12 +0000166 case 4: Kind = SectionKind::get(SectionKind::MergeableConst4,false); break;
167 case 8: Kind = SectionKind::get(SectionKind::MergeableConst8,false); break;
168 case 16: Kind = SectionKind::get(SectionKind::MergeableConst16,false);break;
169 default: Kind = SectionKind::get(SectionKind::MergeableConst,false); break;
Chris Lattner5c2f7892009-07-26 06:26:55 +0000170 }
171 }
Chris Lattner298414e2009-07-22 00:28:43 +0000172
Chris Lattner2ceb60a2009-07-26 06:48:26 +0000173 return getSection(TAI->getSectionForMergeableConstant(Kind)->getName(),
Bruno Cardoso Lopese2b0ecd2009-07-18 23:24:01 +0000174 ELFSection::SHT_PROGBITS,
175 ELFSection::SHF_MERGE | ELFSection::SHF_ALLOC,
176 CPE.getAlignment());
177}
178
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000179// Return the relocation section of section 'S'. 'RelA' is true
180// if the relocation section contains entries with addends.
181ELFSection &ELFWriter::getRelocSection(ELFSection &S) {
182 unsigned SectionHeaderTy = TEW->hasRelocationAddend() ?
183 ELFSection::SHT_RELA : ELFSection::SHT_REL;
184 std::string RelSName(".rel");
185 if (TEW->hasRelocationAddend())
186 RelSName.append("a");
187 RelSName.append(S.getName());
188
189 return getSection(RelSName, SectionHeaderTy, 0, TEW->getPrefELFAlignment());
190}
191
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000192// getGlobalELFVisibility - Returns the ELF specific visibility type
Bruno Cardoso Lopes45f5d642009-07-02 18:29:24 +0000193unsigned ELFWriter::getGlobalELFVisibility(const GlobalValue *GV) {
194 switch (GV->getVisibility()) {
195 default:
Torok Edwinc23197a2009-07-14 16:55:14 +0000196 llvm_unreachable("unknown visibility type");
Bruno Cardoso Lopes45f5d642009-07-02 18:29:24 +0000197 case GlobalValue::DefaultVisibility:
198 return ELFSym::STV_DEFAULT;
199 case GlobalValue::HiddenVisibility:
200 return ELFSym::STV_HIDDEN;
201 case GlobalValue::ProtectedVisibility:
202 return ELFSym::STV_PROTECTED;
203 }
Bruno Cardoso Lopes45f5d642009-07-02 18:29:24 +0000204 return 0;
205}
206
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000207// getGlobalELFBinding - Returns the ELF specific binding type
208unsigned ELFWriter::getGlobalELFBinding(const GlobalValue *GV) {
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000209 if (GV->hasInternalLinkage())
210 return ELFSym::STB_LOCAL;
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000211
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000212 if (GV->hasWeakLinkage())
213 return ELFSym::STB_WEAK;
214
215 return ELFSym::STB_GLOBAL;
216}
217
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000218// getGlobalELFType - Returns the ELF specific type for a global
219unsigned ELFWriter::getGlobalELFType(const GlobalValue *GV) {
220 if (GV->isDeclaration())
221 return ELFSym::STT_NOTYPE;
222
223 if (isa<Function>(GV))
224 return ELFSym::STT_FUNC;
225
226 return ELFSym::STT_OBJECT;
227}
228
229// getElfSectionFlags - Get the ELF Section Header flags based
230// on the flags defined in ELFTargetAsmInfo.
Bruno Cardoso Lopes0b1308f2009-07-03 04:36:26 +0000231unsigned ELFWriter::getElfSectionFlags(unsigned Flags) {
232 unsigned ElfSectionFlags = ELFSection::SHF_ALLOC;
233
234 if (Flags & SectionFlags::Code)
235 ElfSectionFlags |= ELFSection::SHF_EXECINSTR;
Chris Lattnerb4fc4192009-07-25 18:57:34 +0000236 if (Flags & SectionFlags::Writable)
Bruno Cardoso Lopes0b1308f2009-07-03 04:36:26 +0000237 ElfSectionFlags |= ELFSection::SHF_WRITE;
238 if (Flags & SectionFlags::Mergeable)
239 ElfSectionFlags |= ELFSection::SHF_MERGE;
240 if (Flags & SectionFlags::TLS)
241 ElfSectionFlags |= ELFSection::SHF_TLS;
242 if (Flags & SectionFlags::Strings)
243 ElfSectionFlags |= ELFSection::SHF_STRINGS;
244
245 return ElfSectionFlags;
246}
247
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000248// isELFUndefSym - the symbol has no section and must be placed in
249// the symbol table with a reference to the null section.
250static bool isELFUndefSym(const GlobalValue *GV) {
251 return GV->isDeclaration();
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000252}
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000253
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000254// isELFBssSym - for an undef or null value, the symbol must go to a bss
255// section if it's not weak for linker, otherwise it's a common sym.
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000256static bool isELFBssSym(const GlobalVariable *GV) {
257 const Constant *CV = GV->getInitializer();
258 return ((CV->isNullValue() || isa<UndefValue>(CV)) && !GV->isWeakForLinker());
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000259}
260
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000261// isELFCommonSym - for an undef or null value, the symbol must go to a
262// common section if it's weak for linker, otherwise bss.
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000263static bool isELFCommonSym(const GlobalVariable *GV) {
264 const Constant *CV = GV->getInitializer();
265 return ((CV->isNullValue() || isa<UndefValue>(CV)) && GV->isWeakForLinker());
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000266}
267
268// isELFDataSym - if the symbol is an initialized but no null constant
269// it must go to some kind of data section gathered from TAI
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000270static bool isELFDataSym(const Constant *CV) {
271 return (!(CV->isNullValue() || isa<UndefValue>(CV)));
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000272}
273
274// EmitGlobal - Choose the right section for global and emit it
275void ELFWriter::EmitGlobal(const GlobalValue *GV) {
276
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000277 // Check if the referenced symbol is already emitted
278 if (GblSymLookup.find(GV) != GblSymLookup.end())
279 return;
280
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000281 // Handle ELF Bind, Visibility and Type for the current symbol
282 unsigned SymBind = getGlobalELFBinding(GV);
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000283 ELFSym *GblSym = new ELFSym(GV);
284 GblSym->setBind(SymBind);
285 GblSym->setVisibility(getGlobalELFVisibility(GV));
286 GblSym->setType(getGlobalELFType(GV));
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000287
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000288 if (isELFUndefSym(GV)) {
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000289 GblSym->SectionIdx = ELFSection::SHN_UNDEF;
Bruno Cardoso Lopese39493e2009-06-23 04:39:27 +0000290 } else {
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000291 assert(isa<GlobalVariable>(GV) && "GV not a global variable!");
292 const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GV);
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000293
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000294 // Get ELF section from TAI
295 const Section *S = TAI->SectionForGlobal(GV);
296 unsigned SectionFlags = getElfSectionFlags(S->getFlags());
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000297
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000298 // The symbol align should update the section alignment if needed
299 const TargetData *TD = TM.getTargetData();
300 unsigned Align = TD->getPreferredAlignment(GVar);
301 unsigned Size = TD->getTypeAllocSize(GVar->getInitializer()->getType());
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000302 GblSym->Size = Size;
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000303
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000304 if (isELFCommonSym(GVar)) {
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000305 GblSym->SectionIdx = ELFSection::SHN_COMMON;
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000306 getSection(S->getName(), ELFSection::SHT_NOBITS, SectionFlags, 1);
307
308 // A new linkonce section is created for each global in the
309 // common section, the default alignment is 1 and the symbol
310 // value contains its alignment.
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000311 GblSym->Value = Align;
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000312
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000313 } else if (isELFBssSym(GVar)) {
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000314 ELFSection &ES =
315 getSection(S->getName(), ELFSection::SHT_NOBITS, SectionFlags);
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000316 GblSym->SectionIdx = ES.SectionIdx;
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000317
318 // Update the size with alignment and the next object can
319 // start in the right offset in the section
320 if (Align) ES.Size = (ES.Size + Align-1) & ~(Align-1);
321 ES.Align = std::max(ES.Align, Align);
322
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000323 // GblSym->Value should contain the virtual offset inside the section.
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000324 // Virtual because the BSS space is not allocated on ELF objects
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000325 GblSym->Value = ES.Size;
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000326 ES.Size += Size;
327
328 } else if (isELFDataSym(GV)) {
329 ELFSection &ES =
330 getSection(S->getName(), ELFSection::SHT_PROGBITS, SectionFlags);
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000331 GblSym->SectionIdx = ES.SectionIdx;
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000332
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000333 // GblSym->Value should contain the symbol offset inside the section,
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000334 // and all symbols should start on their required alignment boundary
335 ES.Align = std::max(ES.Align, Align);
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000336 GblSym->Value = (ES.size() + (Align-1)) & (-Align);
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000337 ES.emitAlignment(ES.Align);
338
339 // Emit the global to the data section 'ES'
340 EmitGlobalConstant(GVar->getInitializer(), ES);
341 }
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000342 }
343
Bruno Cardoso Lopes171375f2009-07-18 19:30:09 +0000344 // Private symbols must never go to the symbol table.
345 unsigned SymIdx = 0;
346 if (GV->hasPrivateLinkage()) {
347 PrivateSyms.push_back(GblSym);
348 SymIdx = PrivateSyms.size()-1;
349 } else {
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000350 SymbolList.push_back(GblSym);
Bruno Cardoso Lopes171375f2009-07-18 19:30:09 +0000351 }
352
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000353 setGlobalSymLookup(GV, SymIdx);
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000354}
355
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000356void ELFWriter::EmitGlobalConstantStruct(const ConstantStruct *CVS,
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000357 ELFSection &GblS) {
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000358
359 // Print the fields in successive locations. Pad to align if needed!
360 const TargetData *TD = TM.getTargetData();
361 unsigned Size = TD->getTypeAllocSize(CVS->getType());
362 const StructLayout *cvsLayout = TD->getStructLayout(CVS->getType());
363 uint64_t sizeSoFar = 0;
364 for (unsigned i = 0, e = CVS->getNumOperands(); i != e; ++i) {
365 const Constant* field = CVS->getOperand(i);
366
367 // Check if padding is needed and insert one or more 0s.
368 uint64_t fieldSize = TD->getTypeAllocSize(field->getType());
369 uint64_t padSize = ((i == e-1 ? Size : cvsLayout->getElementOffset(i+1))
370 - cvsLayout->getElementOffset(i)) - fieldSize;
371 sizeSoFar += fieldSize + padSize;
372
373 // Now print the actual field value.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000374 EmitGlobalConstant(field, GblS);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000375
376 // Insert padding - this may include padding to increase the size of the
377 // current field up to the ABI size (if the struct is not packed) as well
378 // as padding to ensure that the next field starts at the right offset.
379 for (unsigned p=0; p < padSize; p++)
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000380 GblS.emitByte(0);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000381 }
382 assert(sizeSoFar == cvsLayout->getSizeInBytes() &&
383 "Layout of constant struct may be incorrect!");
384}
385
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000386void ELFWriter::EmitGlobalConstant(const Constant *CV, ELFSection &GblS) {
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000387 const TargetData *TD = TM.getTargetData();
388 unsigned Size = TD->getTypeAllocSize(CV->getType());
389
390 if (const ConstantArray *CVA = dyn_cast<ConstantArray>(CV)) {
391 if (CVA->isString()) {
392 std::string GblStr = CVA->getAsString();
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000393 GblStr.resize(GblStr.size()-1);
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000394 GblS.emitString(GblStr);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000395 } else { // Not a string. Print the values in successive locations
396 for (unsigned i = 0, e = CVA->getNumOperands(); i != e; ++i)
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000397 EmitGlobalConstant(CVA->getOperand(i), GblS);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000398 }
399 return;
400 } else if (const ConstantStruct *CVS = dyn_cast<ConstantStruct>(CV)) {
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000401 EmitGlobalConstantStruct(CVS, GblS);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000402 return;
403 } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(CV)) {
404 uint64_t Val = CFP->getValueAPF().bitcastToAPInt().getZExtValue();
405 if (CFP->getType() == Type::DoubleTy)
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000406 GblS.emitWord64(Val);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000407 else if (CFP->getType() == Type::FloatTy)
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000408 GblS.emitWord32(Val);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000409 else if (CFP->getType() == Type::X86_FP80Ty) {
Torok Edwinc23197a2009-07-14 16:55:14 +0000410 llvm_unreachable("X86_FP80Ty global emission not implemented");
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000411 } else if (CFP->getType() == Type::PPC_FP128Ty)
Torok Edwinc23197a2009-07-14 16:55:14 +0000412 llvm_unreachable("PPC_FP128Ty global emission not implemented");
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000413 return;
414 } else if (const ConstantInt *CI = dyn_cast<ConstantInt>(CV)) {
415 if (Size == 4)
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000416 GblS.emitWord32(CI->getZExtValue());
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000417 else if (Size == 8)
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000418 GblS.emitWord64(CI->getZExtValue());
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000419 else
Torok Edwinc23197a2009-07-14 16:55:14 +0000420 llvm_unreachable("LargeInt global emission not implemented");
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000421 return;
422 } else if (const ConstantVector *CP = dyn_cast<ConstantVector>(CV)) {
423 const VectorType *PTy = CP->getType();
424 for (unsigned I = 0, E = PTy->getNumElements(); I < E; ++I)
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000425 EmitGlobalConstant(CP->getOperand(I), GblS);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000426 return;
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000427 } else if (const GlobalValue *GV = dyn_cast<GlobalValue>(CV)) {
428 // This is a constant address for a global variable or function and
429 // therefore must be referenced using a relocation entry.
430
431 // Check if the referenced symbol is already emitted
432 if (GblSymLookup.find(GV) == GblSymLookup.end())
433 EmitGlobal(GV);
434
435 // Create the relocation entry for the global value
436 MachineRelocation MR =
437 MachineRelocation::getGV(GblS.getCurrentPCOffset(),
438 TEW->getAbsoluteLabelMachineRelTy(),
439 const_cast<GlobalValue*>(GV));
440
441 // Fill the data entry with zeros
442 for (unsigned i=0; i < Size; ++i)
443 GblS.emitByte(0);
444
445 // Add the relocation entry for the current data section
446 GblS.addRelocation(MR);
447 return;
448 } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(CV)) {
449 if (CE->getOpcode() == Instruction::BitCast) {
450 EmitGlobalConstant(CE->getOperand(0), GblS);
451 return;
452 }
453 // See AsmPrinter::EmitConstantValueOnly for other ConstantExpr types
454 llvm_unreachable("Unsupported ConstantExpr type");
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000455 }
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000456
457 llvm_unreachable("Unknown global constant type");
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000458}
459
460
461bool ELFWriter::runOnMachineFunction(MachineFunction &MF) {
Bruno Cardoso Lopes6933d3e2009-07-06 09:26:48 +0000462 // Nothing to do here, this is all done through the ElfCE object above.
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000463 return false;
464}
465
466/// doFinalization - Now that the module has been completely processed, emit
467/// the ELF file to 'O'.
468bool ELFWriter::doFinalization(Module &M) {
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000469 // Emit .data section placeholder
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000470 getDataSection();
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000471
472 // Emit .bss section placeholder
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000473 getBSSSection();
474
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000475 // Build and emit data, bss and "common" sections.
Chris Lattner5f48ff72005-07-16 08:01:13 +0000476 for (Module::global_iterator I = M.global_begin(), E = M.global_end();
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000477 I != E; ++I)
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000478 EmitGlobal(I);
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000479
480 // Emit all pending globals
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000481 for (SetVector<GlobalValue*>::const_iterator I = PendingGlobals.begin(),
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000482 E = PendingGlobals.end(); I != E; ++I)
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000483 EmitGlobal(*I);
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000484
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000485 // Emit non-executable stack note
486 if (TAI->getNonexecutableStackDirective())
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000487 getNonExecStackSection();
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000488
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000489 // Emit a symbol for each section created until now, skip null section
490 for (unsigned i = 1, e = SectionList.size(); i < e; ++i) {
491 ELFSection &ES = *SectionList[i];
492 ELFSym *SectionSym = new ELFSym(0);
493 SectionSym->SectionIdx = ES.SectionIdx;
494 SectionSym->Size = 0;
495 SectionSym->setBind(ELFSym::STB_LOCAL);
496 SectionSym->setType(ELFSym::STT_SECTION);
497 SectionSym->setVisibility(ELFSym::STV_DEFAULT);
498 SymbolList.push_back(SectionSym);
499 ES.Sym = SymbolList.back();
Bruno Cardoso Lopesa5e0abd2009-06-25 07:36:24 +0000500 }
501
Bruno Cardoso Lopesc236a342009-06-22 19:29:56 +0000502 // Emit string table
503 EmitStringTable();
504
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000505 // Emit the symbol table now, if non-empty.
506 EmitSymbolTable();
507
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +0000508 // Emit the relocation sections.
509 EmitRelocations();
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000510
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000511 // Emit the sections string table.
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000512 EmitSectionTableStringTable();
513
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000514 // Dump the sections and section table to the .o file.
Chris Lattner5f48ff72005-07-16 08:01:13 +0000515 OutputSectionsAndSectionTable();
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000516
Chris Lattner5f48ff72005-07-16 08:01:13 +0000517 // We are done with the abstract symbols.
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000518 SymbolList.clear();
Chris Lattner5f48ff72005-07-16 08:01:13 +0000519 SectionList.clear();
520 NumSections = 0;
Chris Lattner5acd1202005-07-11 03:11:47 +0000521
522 // Release the name mangler object.
523 delete Mang; Mang = 0;
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000524 return false;
525}
526
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000527// RelocateField - Patch relocatable field with 'Offset' in 'BO'
528// using a 'Value' of known 'Size'
529void ELFWriter::RelocateField(BinaryObject &BO, uint32_t Offset,
530 int64_t Value, unsigned Size) {
531 if (Size == 32)
532 BO.fixWord32(Value, Offset);
533 else if (Size == 64)
534 BO.fixWord64(Value, Offset);
535 else
536 llvm_unreachable("don't know howto patch relocatable field");
537}
538
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +0000539/// EmitRelocations - Emit relocations
540void ELFWriter::EmitRelocations() {
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000541
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000542 // True if the target uses the relocation entry to hold the addend,
543 // otherwise the addend is written directly to the relocatable field.
544 bool HasRelA = TEW->hasRelocationAddend();
545
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000546 // Create Relocation sections for each section which needs it.
Bruno Cardoso Lopesb674dbe2009-07-17 18:02:30 +0000547 for (unsigned i=0, e=SectionList.size(); i != e; ++i) {
548 ELFSection &S = *SectionList[i];
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000549
550 // This section does not have relocations
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000551 if (!S.hasRelocations()) continue;
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000552 ELFSection &RelSec = getRelocSection(S);
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000553
554 // 'Link' - Section hdr idx of the associated symbol table
555 // 'Info' - Section hdr idx of the section to which the relocation applies
556 ELFSection &SymTab = getSymbolTableSection();
557 RelSec.Link = SymTab.SectionIdx;
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000558 RelSec.Info = S.SectionIdx;
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000559 RelSec.EntSize = TEW->getRelocationEntrySize();
560
561 // Get the relocations from Section
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000562 std::vector<MachineRelocation> Relos = S.getRelocations();
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000563 for (std::vector<MachineRelocation>::iterator MRI = Relos.begin(),
564 MRE = Relos.end(); MRI != MRE; ++MRI) {
565 MachineRelocation &MR = *MRI;
566
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000567 // Relocatable field offset from the section start
568 unsigned RelOffset = MR.getMachineCodeOffset();
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000569
570 // Symbol index in the symbol table
571 unsigned SymIdx = 0;
572
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000573 // Target specific relocation field type and size
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000574 unsigned RelType = TEW->getRelocationType(MR.getRelocationType());
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000575 unsigned RelTySize = TEW->getRelocationTySize(RelType);
Bruno Cardoso Lopesa5e0abd2009-06-25 07:36:24 +0000576 int64_t Addend = 0;
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000577
578 // There are several machine relocations types, and each one of
579 // them needs a different approach to retrieve the symbol table index.
580 if (MR.isGlobalValue()) {
581 const GlobalValue *G = MR.getGlobalValue();
582 SymIdx = GblSymLookup[G];
Bruno Cardoso Lopes171375f2009-07-18 19:30:09 +0000583 if (G->hasPrivateLinkage()) {
584 // If the target uses a section offset in the relocation:
585 // SymIdx + Addend = section sym for global + section offset
586 unsigned SectionIdx = PrivateSyms[SymIdx]->SectionIdx;
587 Addend = PrivateSyms[SymIdx]->Value;
588 SymIdx = SectionList[SectionIdx]->getSymbolTableIndex();
589 } else {
590 Addend = TEW->getDefaultAddendForRelTy(RelType);
591 }
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000592 } else {
Bruno Cardoso Lopese2b0ecd2009-07-18 23:24:01 +0000593 // Get the symbol index for the section symbol
Bruno Cardoso Lopesa5e0abd2009-06-25 07:36:24 +0000594 unsigned SectionIdx = MR.getConstantVal();
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000595 SymIdx = SectionList[SectionIdx]->getSymbolTableIndex();
596 Addend = (uint64_t)MR.getResultPointer();
597
598 // For pc relative relocations where symbols are defined in the same
599 // section they are referenced, ignore the relocation entry and patch
600 // the relocatable field with the symbol offset directly.
601 if (S.SectionIdx == SectionIdx && TEW->isPCRelativeRel(RelType)) {
602 int64_t Value = TEW->computeRelocation(Addend, RelOffset, RelType);
603 RelocateField(S, RelOffset, Value, RelTySize);
604 continue;
605 }
Bruno Cardoso Lopese2b0ecd2009-07-18 23:24:01 +0000606
607 // Handle Jump Table Index relocation
608 if ((SectionIdx == getJumpTableSection().SectionIdx) &&
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000609 TEW->hasCustomJumpTableIndexRelTy()) {
Bruno Cardoso Lopese2b0ecd2009-07-18 23:24:01 +0000610 RelType = TEW->getJumpTableIndexRelTy();
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000611 RelTySize = TEW->getRelocationTySize(RelType);
612 }
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000613 }
614
Bruno Cardoso Lopes171375f2009-07-18 19:30:09 +0000615 // The target without addend on the relocation symbol must be
616 // patched in the relocation place itself to contain the addend
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000617 if (!HasRelA)
618 RelocateField(S, RelOffset, Addend, RelTySize);
Bruno Cardoso Lopes171375f2009-07-18 19:30:09 +0000619
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000620 // Get the relocation entry and emit to the relocation section
Bruno Cardoso Lopes0a38dc52009-07-20 08:52:02 +0000621 ELFRelocation Rel(RelOffset, SymIdx, RelType, HasRelA, Addend);
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000622 EmitRelocation(RelSec, Rel, HasRelA);
623 }
624 }
625}
626
627/// EmitRelocation - Write relocation 'Rel' to the relocation section 'Rel'
628void ELFWriter::EmitRelocation(BinaryObject &RelSec, ELFRelocation &Rel,
629 bool HasRelA) {
630 RelSec.emitWord(Rel.getOffset());
631 RelSec.emitWord(Rel.getInfo(is64Bit));
632 if (HasRelA)
633 RelSec.emitWord(Rel.getAddend());
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +0000634}
635
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000636/// EmitSymbol - Write symbol 'Sym' to the symbol table 'SymbolTable'
637void ELFWriter::EmitSymbol(BinaryObject &SymbolTable, ELFSym &Sym) {
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000638 if (is64Bit) {
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000639 SymbolTable.emitWord32(Sym.NameIdx);
640 SymbolTable.emitByte(Sym.Info);
641 SymbolTable.emitByte(Sym.Other);
642 SymbolTable.emitWord16(Sym.SectionIdx);
643 SymbolTable.emitWord64(Sym.Value);
644 SymbolTable.emitWord64(Sym.Size);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000645 } else {
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000646 SymbolTable.emitWord32(Sym.NameIdx);
647 SymbolTable.emitWord32(Sym.Value);
648 SymbolTable.emitWord32(Sym.Size);
649 SymbolTable.emitByte(Sym.Info);
650 SymbolTable.emitByte(Sym.Other);
651 SymbolTable.emitWord16(Sym.SectionIdx);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000652 }
653}
654
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000655/// EmitSectionHeader - Write section 'Section' header in 'SHdrTab'
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000656/// Section Header Table
Bruno Cardoso Lopesd2910662009-07-13 22:40:39 +0000657void ELFWriter::EmitSectionHeader(BinaryObject &SHdrTab,
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000658 const ELFSection &SHdr) {
659 SHdrTab.emitWord32(SHdr.NameIdx);
660 SHdrTab.emitWord32(SHdr.Type);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000661 if (is64Bit) {
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000662 SHdrTab.emitWord64(SHdr.Flags);
663 SHdrTab.emitWord(SHdr.Addr);
664 SHdrTab.emitWord(SHdr.Offset);
665 SHdrTab.emitWord64(SHdr.Size);
666 SHdrTab.emitWord32(SHdr.Link);
667 SHdrTab.emitWord32(SHdr.Info);
668 SHdrTab.emitWord64(SHdr.Align);
669 SHdrTab.emitWord64(SHdr.EntSize);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000670 } else {
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000671 SHdrTab.emitWord32(SHdr.Flags);
672 SHdrTab.emitWord(SHdr.Addr);
673 SHdrTab.emitWord(SHdr.Offset);
674 SHdrTab.emitWord32(SHdr.Size);
675 SHdrTab.emitWord32(SHdr.Link);
676 SHdrTab.emitWord32(SHdr.Info);
677 SHdrTab.emitWord32(SHdr.Align);
678 SHdrTab.emitWord32(SHdr.EntSize);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000679 }
680}
681
Bruno Cardoso Lopesc236a342009-06-22 19:29:56 +0000682/// EmitStringTable - If the current symbol table is non-empty, emit the string
683/// table for it
684void ELFWriter::EmitStringTable() {
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000685 if (!SymbolList.size()) return; // Empty symbol table.
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000686 ELFSection &StrTab = getStringTableSection();
Chris Lattner5f48ff72005-07-16 08:01:13 +0000687
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000688 // Set the zero'th symbol to a null byte, as required.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000689 StrTab.emitByte(0);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000690
Bruno Cardoso Lopesb674dbe2009-07-17 18:02:30 +0000691 // Walk on the symbol list and write symbol names into the string table.
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000692 unsigned Index = 1;
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000693 for (ELFSymIter I=SymbolList.begin(), E=SymbolList.end(); I != E; ++I) {
694 ELFSym &Sym = *(*I);
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000695
Chris Lattner5acd1202005-07-11 03:11:47 +0000696 // Use the name mangler to uniquify the LLVM symbol.
Bruno Cardoso Lopesa5e0abd2009-06-25 07:36:24 +0000697 std::string Name;
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000698 if (Sym.GV) Name.append(Mang->getMangledName(Sym.GV));
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000699
700 if (Name.empty()) {
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000701 Sym.NameIdx = 0;
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000702 } else {
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000703 Sym.NameIdx = Index;
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000704 StrTab.emitString(Name);
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000705
706 // Keep track of the number of bytes emitted to this section.
707 Index += Name.size()+1;
708 }
709 }
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000710 assert(Index == StrTab.size());
Chris Lattner5f48ff72005-07-16 08:01:13 +0000711 StrTab.Size = Index;
Bruno Cardoso Lopesc236a342009-06-22 19:29:56 +0000712}
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000713
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000714// SortSymbols - On the symbol table local symbols must come before
715// all other symbols with non-local bindings. The return value is
716// the position of the first non local symbol.
717unsigned ELFWriter::SortSymbols() {
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000718 unsigned FirstNonLocalSymbol;
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000719 std::vector<ELFSym*> LocalSyms, OtherSyms;
720
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000721 for (ELFSymIter I=SymbolList.begin(), E=SymbolList.end(); I != E; ++I) {
722 if ((*I)->isLocalBind())
723 LocalSyms.push_back(*I);
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000724 else
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000725 OtherSyms.push_back(*I);
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000726 }
727 SymbolList.clear();
728 FirstNonLocalSymbol = LocalSyms.size();
729
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000730 for (unsigned i = 0; i < FirstNonLocalSymbol; ++i)
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000731 SymbolList.push_back(LocalSyms[i]);
732
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000733 for (ELFSymIter I=OtherSyms.begin(), E=OtherSyms.end(); I != E; ++I)
734 SymbolList.push_back(*I);
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000735
736 LocalSyms.clear();
737 OtherSyms.clear();
738
739 return FirstNonLocalSymbol;
740}
741
Bruno Cardoso Lopesc236a342009-06-22 19:29:56 +0000742/// EmitSymbolTable - Emit the symbol table itself.
743void ELFWriter::EmitSymbolTable() {
744 if (!SymbolList.size()) return; // Empty symbol table.
745
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000746 // Now that we have emitted the string table and know the offset into the
747 // string table of each symbol, emit the symbol table itself.
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000748 ELFSection &SymTab = getSymbolTableSection();
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000749 SymTab.Align = TEW->getPrefELFAlignment();
Bruno Cardoso Lopesc236a342009-06-22 19:29:56 +0000750
751 // Section Index of .strtab.
752 SymTab.Link = getStringTableSection().SectionIdx;
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000753
754 // Size of each symtab entry.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000755 SymTab.EntSize = TEW->getSymTabEntrySize();
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000756
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000757 // The first entry in the symtab is the null symbol
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000758 SymbolList.insert(SymbolList.begin(), new ELFSym(0));
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000759
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000760 // Reorder the symbol table with local symbols first!
761 unsigned FirstNonLocalSymbol = SortSymbols();
762
763 // Emit all the symbols to the symbol table.
764 for (unsigned i = 0, e = SymbolList.size(); i < e; ++i) {
765 ELFSym &Sym = *SymbolList[i];
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000766
767 // Emit symbol to the symbol table
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000768 EmitSymbol(SymTab, Sym);
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000769
770 // Record the symbol table index for each global value
Bruno Cardoso Lopes68491c12009-07-21 06:51:32 +0000771 if (Sym.GV) setGlobalSymLookup(Sym.GV, i);
Bruno Cardoso Lopesa5e0abd2009-06-25 07:36:24 +0000772
773 // Keep track on the symbol index into the symbol table
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000774 Sym.SymTabIdx = i;
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000775 }
776
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000777 // One greater than the symbol table index of the last local symbol
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000778 SymTab.Info = FirstNonLocalSymbol;
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000779 SymTab.Size = SymTab.size();
Chris Lattner80ed8fa2005-07-07 07:02:20 +0000780}
781
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000782/// EmitSectionTableStringTable - This method adds and emits a section for the
783/// ELF Section Table string table: the string table that holds all of the
784/// section names.
785void ELFWriter::EmitSectionTableStringTable() {
786 // First step: add the section for the string table to the list of sections:
Bruno Cardoso Lopese39493e2009-06-23 04:39:27 +0000787 ELFSection &SHStrTab = getSectionHeaderStringTableSection();
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000788
789 // Now that we know which section number is the .shstrtab section, update the
790 // e_shstrndx entry in the ELF header.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000791 ElfHdr.fixWord16(SHStrTab.SectionIdx, ELFHdr_e_shstrndx_Offset);
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000792
793 // Set the NameIdx of each section in the string table and emit the bytes for
794 // the string table.
795 unsigned Index = 0;
796
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000797 for (ELFSectionIter I=SectionList.begin(), E=SectionList.end(); I != E; ++I) {
798 ELFSection &S = *(*I);
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000799 // Set the index into the table. Note if we have lots of entries with
800 // common suffixes, we could memoize them here if we cared.
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000801 S.NameIdx = Index;
802 SHStrTab.emitString(S.getName());
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000803
804 // Keep track of the number of bytes emitted to this section.
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000805 Index += S.getName().size()+1;
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000806 }
807
808 // Set the size of .shstrtab now that we know what it is.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000809 assert(Index == SHStrTab.size());
Chris Lattner5f48ff72005-07-16 08:01:13 +0000810 SHStrTab.Size = Index;
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000811}
812
Chris Lattner5f48ff72005-07-16 08:01:13 +0000813/// OutputSectionsAndSectionTable - Now that we have constructed the file header
814/// and all of the sections, emit these to the ostream destination and emit the
815/// SectionTable.
816void ELFWriter::OutputSectionsAndSectionTable() {
817 // Pass #1: Compute the file offset for each section.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000818 size_t FileOff = ElfHdr.size(); // File header first.
Chris Lattner5f48ff72005-07-16 08:01:13 +0000819
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000820 // Adjust alignment of all section if needed, skip the null section.
821 for (unsigned i=1, e=SectionList.size(); i < e; ++i) {
822 ELFSection &ES = *SectionList[i];
823 if (!ES.size()) {
824 ES.Offset = FileOff;
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000825 continue;
826 }
827
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +0000828 // Update Section size
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000829 if (!ES.Size)
830 ES.Size = ES.size();
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +0000831
Chris Lattner5f48ff72005-07-16 08:01:13 +0000832 // Align FileOff to whatever the alignment restrictions of the section are.
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000833 if (ES.Align)
834 FileOff = (FileOff+ES.Align-1) & ~(ES.Align-1);
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +0000835
Bruno Cardoso Lopes4b70fab2009-07-15 20:49:10 +0000836 ES.Offset = FileOff;
837 FileOff += ES.Size;
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000838 }
839
Chris Lattner5f48ff72005-07-16 08:01:13 +0000840 // Align Section Header.
Bruno Cardoso Lopes0d3193e2009-06-22 19:16:16 +0000841 unsigned TableAlign = TEW->getPrefELFAlignment();
Chris Lattner5f48ff72005-07-16 08:01:13 +0000842 FileOff = (FileOff+TableAlign-1) & ~(TableAlign-1);
843
844 // Now that we know where all of the sections will be emitted, set the e_shnum
845 // entry in the ELF header.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000846 ElfHdr.fixWord16(NumSections, ELFHdr_e_shnum_Offset);
Jeff Cohen00b168892005-07-27 06:12:32 +0000847
Chris Lattner5f48ff72005-07-16 08:01:13 +0000848 // Now that we know the offset in the file of the section table, update the
849 // e_shoff address in the ELF header.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000850 ElfHdr.fixWord(FileOff, ELFHdr_e_shoff_Offset);
Jeff Cohen00b168892005-07-27 06:12:32 +0000851
Chris Lattner5f48ff72005-07-16 08:01:13 +0000852 // Now that we know all of the data in the file header, emit it and all of the
853 // sections!
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000854 O.write((char *)&ElfHdr.getData()[0], ElfHdr.size());
855 FileOff = ElfHdr.size();
Chris Lattner5f48ff72005-07-16 08:01:13 +0000856
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000857 // Section Header Table blob
858 BinaryObject SHdrTable(isLittleEndian, is64Bit);
Chris Lattner5f48ff72005-07-16 08:01:13 +0000859
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000860 // Emit all of sections to the file and build the section header table.
Bruno Cardoso Lopes115934e2009-07-16 06:26:41 +0000861 for (ELFSectionIter I=SectionList.begin(), E=SectionList.end(); I != E; ++I) {
862 ELFSection &S = *(*I);
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000863 DOUT << "SectionIdx: " << S.SectionIdx << ", Name: " << S.getName()
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000864 << ", Size: " << S.Size << ", Offset: " << S.Offset
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000865 << ", SectionData Size: " << S.size() << "\n";
Chris Lattner5f48ff72005-07-16 08:01:13 +0000866
867 // Align FileOff to whatever the alignment restrictions of the section are.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000868 if (S.size()) {
Bruno Cardoso Lopese39493e2009-06-23 04:39:27 +0000869 if (S.Align) {
870 for (size_t NewFileOff = (FileOff+S.Align-1) & ~(S.Align-1);
871 FileOff != NewFileOff; ++FileOff)
872 O << (char)0xAB;
873 }
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000874 O.write((char *)&S.getData()[0], S.Size);
Bruno Cardoso Lopesc997d452009-06-11 19:16:03 +0000875 FileOff += S.Size;
876 }
Bruno Cardoso Lopesa029a272009-06-07 21:22:38 +0000877
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000878 EmitSectionHeader(SHdrTable, S);
Chris Lattner5f48ff72005-07-16 08:01:13 +0000879 }
880
881 // Align output for the section table.
882 for (size_t NewFileOff = (FileOff+TableAlign-1) & ~(TableAlign-1);
883 FileOff != NewFileOff; ++FileOff)
Owen Andersoncb371882008-08-21 00:14:44 +0000884 O << (char)0xAB;
Jeff Cohen00b168892005-07-27 06:12:32 +0000885
Chris Lattner5f48ff72005-07-16 08:01:13 +0000886 // Emit the section table itself.
Bruno Cardoso Lopesae9163f2009-06-14 07:53:21 +0000887 O.write((char *)&SHdrTable.getData()[0], SHdrTable.size());
Chris Lattner35f0a4f2005-06-27 06:29:00 +0000888}