blob: 47d14868600187cd4898cf0432ffc484d78258fa [file] [log] [blame]
Nick Lewyckyc1f96942013-03-09 09:31:44 +00001//===- lib/MC/MCELFStreamer.cpp - ELF Object Output -----------------------===//
Matt Fleming6c1ad482010-08-16 18:57:57 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file assembles .s files and emits ELF .o object files.
11//
12//===----------------------------------------------------------------------===//
13
Logan Chien59ff0702012-12-07 15:50:40 +000014#include "llvm/MC/MCELFStreamer.h"
Peter Collingbourneadac4072013-04-10 16:52:15 +000015#include "llvm/ADT/STLExtras.h"
Chandler Carruth8a8cd2b2014-01-07 11:48:04 +000016#include "llvm/ADT/SmallPtrSet.h"
David Peixotto72667312013-11-25 19:11:13 +000017#include "llvm/MC/MCAsmBackend.h"
Rafael Espindola7b61ddf2014-10-15 16:12:52 +000018#include "llvm/MC/MCAsmInfo.h"
Chandler Carruth8a8cd2b2014-01-07 11:48:04 +000019#include "llvm/MC/MCAssembler.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000020#include "llvm/MC/MCCodeEmitter.h"
Rafael Espindola81a62742012-01-07 23:18:39 +000021#include "llvm/MC/MCContext.h"
Tim Northover5cc3dc82012-12-07 16:50:23 +000022#include "llvm/MC/MCELF.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000023#include "llvm/MC/MCELFSymbolFlags.h"
24#include "llvm/MC/MCExpr.h"
25#include "llvm/MC/MCInst.h"
Rafael Espindolae308c0c2014-01-23 22:49:25 +000026#include "llvm/MC/MCObjectFileInfo.h"
Rafael Espindola81a62742012-01-07 23:18:39 +000027#include "llvm/MC/MCObjectStreamer.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000028#include "llvm/MC/MCSection.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000029#include "llvm/MC/MCSectionELF.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000030#include "llvm/MC/MCSymbol.h"
Rafael Espindola16145972010-11-01 14:28:48 +000031#include "llvm/MC/MCValue.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000032#include "llvm/Support/Debug.h"
33#include "llvm/Support/ELF.h"
34#include "llvm/Support/ErrorHandling.h"
35#include "llvm/Support/raw_ostream.h"
Matt Fleming6c1ad482010-08-16 18:57:57 +000036
37using namespace llvm;
38
Logan Chien59ff0702012-12-07 15:50:40 +000039MCELFStreamer::~MCELFStreamer() {
Rafael Espindola81a62742012-01-07 23:18:39 +000040}
41
Rafael Espindola7b61ddf2014-10-15 16:12:52 +000042void MCELFStreamer::InitSections(bool NoExecStack) {
Rafael Espindolaf667d922010-09-15 21:48:40 +000043 // This emulates the same behavior of GNU as. This makes it easier
44 // to compare the output as the major sections are in the same order.
Rafael Espindola7b61ddf2014-10-15 16:12:52 +000045 MCContext &Ctx = getContext();
46 SwitchSection(Ctx.getObjectFileInfo()->getTextSection());
Rafael Espindola7b514962014-02-04 18:34:04 +000047 EmitCodeAlignment(4);
Rafael Espindola247f9512014-01-24 02:18:40 +000048
Rafael Espindola7b61ddf2014-10-15 16:12:52 +000049 SwitchSection(Ctx.getObjectFileInfo()->getDataSection());
Rafael Espindola7b514962014-02-04 18:34:04 +000050 EmitCodeAlignment(4);
Rafael Espindola247f9512014-01-24 02:18:40 +000051
Rafael Espindola7b61ddf2014-10-15 16:12:52 +000052 SwitchSection(Ctx.getObjectFileInfo()->getBSSSection());
Rafael Espindola7b514962014-02-04 18:34:04 +000053 EmitCodeAlignment(4);
Rafael Espindola247f9512014-01-24 02:18:40 +000054
Rafael Espindola7b61ddf2014-10-15 16:12:52 +000055 SwitchSection(Ctx.getObjectFileInfo()->getTextSection());
56
57 if (NoExecStack)
58 SwitchSection(Ctx.getAsmInfo()->getNonexecutableStackSection(Ctx));
Rafael Espindolaf667d922010-09-15 21:48:40 +000059}
60
Matt Fleming6c1ad482010-08-16 18:57:57 +000061void MCELFStreamer::EmitLabel(MCSymbol *Symbol) {
Rafael Espindola16795802010-11-28 16:22:59 +000062 assert(Symbol->isUndefined() && "Cannot define a symbol twice!");
63
Rafael Espindolae5b74152010-11-28 17:18:55 +000064 MCObjectStreamer::EmitLabel(Symbol);
Rafael Espindola75d65b92010-09-25 05:42:19 +000065
Rafael Espindola84d03182010-11-11 19:04:55 +000066 const MCSectionELF &Section =
67 static_cast<const MCSectionELF&>(Symbol->getSection());
Rafael Espindolae5b74152010-11-28 17:18:55 +000068 MCSymbolData &SD = getAssembler().getSymbolData(*Symbol);
Rafael Espindola0e7e34e2011-01-23 04:43:11 +000069 if (Section.getFlags() & ELF::SHF_TLS)
Jan Sjödin30a52de2011-02-28 21:45:04 +000070 MCELF::SetType(SD, ELF::STT_TLS);
Matt Fleming6c1ad482010-08-16 18:57:57 +000071}
72
73void MCELFStreamer::EmitAssemblerFlag(MCAssemblerFlag Flag) {
David Peixotto72667312013-11-25 19:11:13 +000074 // Let the target do whatever target specific stuff it needs to do.
75 getAssembler().getBackend().handleAssemblerFlag(Flag);
76 // Do any generic stuff we need to do.
Matt Fleming6c1ad482010-08-16 18:57:57 +000077 switch (Flag) {
Jim Grosbach5a2c68d2010-11-05 22:08:08 +000078 case MCAF_SyntaxUnified: return; // no-op here.
Evan Cheng481ebb02011-07-27 00:38:12 +000079 case MCAF_Code16: return; // Change parsing mode; no-op here.
80 case MCAF_Code32: return; // Change parsing mode; no-op here.
81 case MCAF_Code64: return; // Change parsing mode; no-op here.
Matt Fleming6c1ad482010-08-16 18:57:57 +000082 case MCAF_SubsectionsViaSymbols:
83 getAssembler().setSubsectionsViaSymbols(true);
84 return;
85 }
86
Craig Toppera2886c22012-02-07 05:05:23 +000087 llvm_unreachable("invalid assembler flag!");
Matt Fleming6c1ad482010-08-16 18:57:57 +000088}
89
Peter Collingbourne2f495b92013-04-17 21:18:16 +000090void MCELFStreamer::ChangeSection(const MCSection *Section,
91 const MCExpr *Subsection) {
Eli Benderskyf483ff92012-12-20 19:05:53 +000092 MCSectionData *CurSection = getCurrentSectionData();
93 if (CurSection && CurSection->isBundleLocked())
94 report_fatal_error("Unterminated .bundle_lock when changing a section");
Rafael Espindolab6613022014-10-17 01:48:58 +000095
96 MCAssembler &Asm = getAssembler();
97 auto *SectionELF = static_cast<const MCSectionELF *>(Section);
98 const MCSymbol *Grp = SectionELF->getGroup();
Rafael Espindola7d0ba342010-11-14 04:17:37 +000099 if (Grp)
Rafael Espindolab6613022014-10-17 01:48:58 +0000100 Asm.getOrCreateSymbolData(*Grp);
101
Peter Collingbourne2f495b92013-04-17 21:18:16 +0000102 this->MCObjectStreamer::ChangeSection(Section, Subsection);
Rafael Espindolab6613022014-10-17 01:48:58 +0000103 MCSymbol *SectionSymbol = getContext().getOrCreateSectionSymbol(*SectionELF);
104 if (SectionSymbol->isUndefined()) {
105 EmitLabel(SectionSymbol);
106 MCELF::SetType(Asm.getSymbolData(*SectionSymbol), ELF::STT_SECTION);
107 }
Rafael Espindola7d0ba342010-11-14 04:17:37 +0000108}
109
Rafael Espindola16145972010-11-01 14:28:48 +0000110void MCELFStreamer::EmitWeakReference(MCSymbol *Alias, const MCSymbol *Symbol) {
111 getAssembler().getOrCreateSymbolData(*Symbol);
Rafael Espindola7fadc0e2014-03-20 02:12:01 +0000112 const MCExpr *Value = MCSymbolRefExpr::Create(
113 Symbol, MCSymbolRefExpr::VK_WEAKREF, getContext());
Rafael Espindola16145972010-11-01 14:28:48 +0000114 Alias->setVariableValue(Value);
115}
116
Peter Collingbourneadac4072013-04-10 16:52:15 +0000117// When GNU as encounters more than one .type declaration for an object it seems
118// to use a mechanism similar to the one below to decide which type is actually
119// used in the object file. The greater of T1 and T2 is selected based on the
120// following ordering:
121// STT_NOTYPE < STT_OBJECT < STT_FUNC < STT_GNU_IFUNC < STT_TLS < anything else
122// If neither T1 < T2 nor T2 < T1 according to this ordering, use T2 (the user
123// provided type).
124static unsigned CombineSymbolTypes(unsigned T1, unsigned T2) {
Benjamin Kramer867bfc52015-03-07 17:41:00 +0000125 for (unsigned Type : {ELF::STT_NOTYPE, ELF::STT_OBJECT, ELF::STT_FUNC,
126 ELF::STT_GNU_IFUNC, ELF::STT_TLS}) {
127 if (T1 == Type)
Peter Collingbourneadac4072013-04-10 16:52:15 +0000128 return T2;
Benjamin Kramer867bfc52015-03-07 17:41:00 +0000129 if (T2 == Type)
Peter Collingbourneadac4072013-04-10 16:52:15 +0000130 return T1;
131 }
132
133 return T2;
134}
135
Saleem Abdulrasool4208b612013-08-09 01:52:03 +0000136bool MCELFStreamer::EmitSymbolAttribute(MCSymbol *Symbol,
137 MCSymbolAttr Attribute) {
Matt Fleming6c1ad482010-08-16 18:57:57 +0000138 // Indirect symbols are handled differently, to match how 'as' handles
139 // them. This makes writing matching .o files easier.
140 if (Attribute == MCSA_IndirectSymbol) {
141 // Note that we intentionally cannot use the symbol data here; this is
142 // important for matching the string table that 'as' generates.
143 IndirectSymbolData ISD;
144 ISD.Symbol = Symbol;
145 ISD.SectionData = getCurrentSectionData();
146 getAssembler().getIndirectSymbols().push_back(ISD);
Saleem Abdulrasool4208b612013-08-09 01:52:03 +0000147 return true;
Matt Fleming6c1ad482010-08-16 18:57:57 +0000148 }
149
150 // Adding a symbol attribute always introduces the symbol, note that an
151 // important side effect of calling getOrCreateSymbolData here is to register
152 // the symbol with the assembler.
153 MCSymbolData &SD = getAssembler().getOrCreateSymbolData(*Symbol);
154
155 // The implementation of symbol attributes is designed to match 'as', but it
156 // leaves much to desired. It doesn't really make sense to arbitrarily add and
157 // remove flags, but 'as' allows this (in particular, see .desc).
158 //
159 // In the future it might be worth trying to make these operations more well
160 // defined.
161 switch (Attribute) {
162 case MCSA_LazyReference:
163 case MCSA_Reference:
Kevin Enderby8be14412010-11-19 18:39:33 +0000164 case MCSA_SymbolResolver:
Matt Fleming6c1ad482010-08-16 18:57:57 +0000165 case MCSA_PrivateExtern:
Matt Fleming6c1ad482010-08-16 18:57:57 +0000166 case MCSA_WeakDefinition:
Eli Friedmanb20b5242010-08-16 19:15:06 +0000167 case MCSA_WeakDefAutoPrivate:
Matt Fleming6c1ad482010-08-16 18:57:57 +0000168 case MCSA_Invalid:
Matt Fleming6c1ad482010-08-16 18:57:57 +0000169 case MCSA_IndirectSymbol:
Saleem Abdulrasool4208b612013-08-09 01:52:03 +0000170 return false;
Matt Fleming6c1ad482010-08-16 18:57:57 +0000171
Jim Grosbachb7b750d2012-09-13 23:11:31 +0000172 case MCSA_NoDeadStrip:
Rafael Espindolaeba90222010-11-14 01:34:31 +0000173 // Ignore for now.
174 break;
175
Rafael Espindola5fa925e2015-01-23 04:44:35 +0000176 case MCSA_ELF_TypeGnuUniqueObject:
177 MCELF::SetType(SD, CombineSymbolTypes(MCELF::GetType(SD), ELF::STT_OBJECT));
178 MCELF::SetBinding(SD, ELF::STB_GNU_UNIQUE);
179 SD.setExternal(true);
180 BindingExplicitlySet.insert(Symbol);
181 break;
182
Matt Fleming6c1ad482010-08-16 18:57:57 +0000183 case MCSA_Global:
Jan Sjödin30a52de2011-02-28 21:45:04 +0000184 MCELF::SetBinding(SD, ELF::STB_GLOBAL);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000185 SD.setExternal(true);
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000186 BindingExplicitlySet.insert(Symbol);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000187 break;
188
Benjamin Kramer5af02b02010-09-02 17:18:32 +0000189 case MCSA_WeakReference:
Matt Fleming6c1ad482010-08-16 18:57:57 +0000190 case MCSA_Weak:
Jan Sjödin30a52de2011-02-28 21:45:04 +0000191 MCELF::SetBinding(SD, ELF::STB_WEAK);
Rafael Espindola83b2a332010-10-06 16:47:31 +0000192 SD.setExternal(true);
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000193 BindingExplicitlySet.insert(Symbol);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000194 break;
195
196 case MCSA_Local:
Jan Sjödin30a52de2011-02-28 21:45:04 +0000197 MCELF::SetBinding(SD, ELF::STB_LOCAL);
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000198 SD.setExternal(false);
199 BindingExplicitlySet.insert(Symbol);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000200 break;
201
202 case MCSA_ELF_TypeFunction:
Peter Collingbourneadac4072013-04-10 16:52:15 +0000203 MCELF::SetType(SD, CombineSymbolTypes(MCELF::GetType(SD),
204 ELF::STT_FUNC));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000205 break;
206
Roman Divacky735cb8b2011-12-12 17:34:04 +0000207 case MCSA_ELF_TypeIndFunction:
Peter Collingbourneadac4072013-04-10 16:52:15 +0000208 MCELF::SetType(SD, CombineSymbolTypes(MCELF::GetType(SD),
209 ELF::STT_GNU_IFUNC));
Roman Divacky735cb8b2011-12-12 17:34:04 +0000210 break;
211
Matt Fleming6c1ad482010-08-16 18:57:57 +0000212 case MCSA_ELF_TypeObject:
Peter Collingbourneadac4072013-04-10 16:52:15 +0000213 MCELF::SetType(SD, CombineSymbolTypes(MCELF::GetType(SD),
214 ELF::STT_OBJECT));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000215 break;
216
217 case MCSA_ELF_TypeTLS:
Peter Collingbourneadac4072013-04-10 16:52:15 +0000218 MCELF::SetType(SD, CombineSymbolTypes(MCELF::GetType(SD),
219 ELF::STT_TLS));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000220 break;
221
222 case MCSA_ELF_TypeCommon:
Peter Collingbourneadac4072013-04-10 16:52:15 +0000223 // TODO: Emit these as a common symbol.
224 MCELF::SetType(SD, CombineSymbolTypes(MCELF::GetType(SD),
225 ELF::STT_OBJECT));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000226 break;
227
228 case MCSA_ELF_TypeNoType:
Peter Collingbourneadac4072013-04-10 16:52:15 +0000229 MCELF::SetType(SD, CombineSymbolTypes(MCELF::GetType(SD),
230 ELF::STT_NOTYPE));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000231 break;
232
233 case MCSA_Protected:
Jan Sjödin30a52de2011-02-28 21:45:04 +0000234 MCELF::SetVisibility(SD, ELF::STV_PROTECTED);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000235 break;
236
237 case MCSA_Hidden:
Jan Sjödin30a52de2011-02-28 21:45:04 +0000238 MCELF::SetVisibility(SD, ELF::STV_HIDDEN);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000239 break;
240
241 case MCSA_Internal:
Jan Sjödin30a52de2011-02-28 21:45:04 +0000242 MCELF::SetVisibility(SD, ELF::STV_INTERNAL);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000243 break;
244 }
Saleem Abdulrasool4208b612013-08-09 01:52:03 +0000245
246 return true;
Matt Fleming6c1ad482010-08-16 18:57:57 +0000247}
248
249void MCELFStreamer::EmitCommonSymbol(MCSymbol *Symbol, uint64_t Size,
250 unsigned ByteAlignment) {
251 MCSymbolData &SD = getAssembler().getOrCreateSymbolData(*Symbol);
252
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000253 if (!BindingExplicitlySet.count(Symbol)) {
Jan Sjödin30a52de2011-02-28 21:45:04 +0000254 MCELF::SetBinding(SD, ELF::STB_GLOBAL);
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000255 SD.setExternal(true);
256 }
257
Jan Sjödin30a52de2011-02-28 21:45:04 +0000258 MCELF::SetType(SD, ELF::STT_OBJECT);
Rafael Espindolaa7d0bed2010-11-14 21:11:16 +0000259
Jan Sjödin30a52de2011-02-28 21:45:04 +0000260 if (MCELF::GetBinding(SD) == ELF_STB_Local) {
Rafael Espindolaba31e272015-01-29 17:33:21 +0000261 const MCSection *Section = getAssembler().getContext().getELFSection(
262 ".bss", ELF::SHT_NOBITS, ELF::SHF_WRITE | ELF::SHF_ALLOC);
Richard Mitton21101b32013-09-19 23:21:01 +0000263
264 AssignSection(Symbol, Section);
Rafael Espindola9bac6cb2010-09-22 17:43:04 +0000265
Rafael Espindola53f0bf12010-09-29 14:52:01 +0000266 struct LocalCommon L = {&SD, Size, ByteAlignment};
267 LocalCommons.push_back(L);
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000268 } else {
269 SD.setCommon(Size, ByteAlignment);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000270 }
271
Rafael Espindolaf0591c12010-09-21 00:24:38 +0000272 SD.setSize(MCConstantExpr::Create(Size, getContext()));
Matt Fleming6c1ad482010-08-16 18:57:57 +0000273}
274
Logan Chien59ff0702012-12-07 15:50:40 +0000275void MCELFStreamer::EmitELFSize(MCSymbol *Symbol, const MCExpr *Value) {
276 MCSymbolData &SD = getAssembler().getOrCreateSymbolData(*Symbol);
277 SD.setSize(Value);
278}
279
Benjamin Kramer63970512011-09-01 23:04:27 +0000280void MCELFStreamer::EmitLocalCommonSymbol(MCSymbol *Symbol, uint64_t Size,
281 unsigned ByteAlignment) {
Jason W Kim4c1386a2010-12-16 03:12:17 +0000282 // FIXME: Should this be caught and done earlier?
283 MCSymbolData &SD = getAssembler().getOrCreateSymbolData(*Symbol);
Jan Sjödin30a52de2011-02-28 21:45:04 +0000284 MCELF::SetBinding(SD, ELF::STB_LOCAL);
Jason W Kim4c1386a2010-12-16 03:12:17 +0000285 SD.setExternal(false);
286 BindingExplicitlySet.insert(Symbol);
Benjamin Kramer63970512011-09-01 23:04:27 +0000287 EmitCommonSymbol(Symbol, Size, ByteAlignment);
Jason W Kim4c1386a2010-12-16 03:12:17 +0000288}
289
Kevin Enderby96918bc2014-04-22 17:27:29 +0000290void MCELFStreamer::EmitValueImpl(const MCExpr *Value, unsigned Size,
291 const SMLoc &Loc) {
Eli Benderskyf483ff92012-12-20 19:05:53 +0000292 if (getCurrentSectionData()->isBundleLocked())
293 report_fatal_error("Emitting values inside a locked bundle is forbidden");
David Meyer44ec69e2012-02-15 15:09:06 +0000294 fixSymbolsInTLSFixups(Value);
Kevin Enderby96918bc2014-04-22 17:27:29 +0000295 MCObjectStreamer::EmitValueImpl(Value, Size, Loc);
David Meyer44ec69e2012-02-15 15:09:06 +0000296}
297
Eli Benderskyf483ff92012-12-20 19:05:53 +0000298void MCELFStreamer::EmitValueToAlignment(unsigned ByteAlignment,
299 int64_t Value,
300 unsigned ValueSize,
301 unsigned MaxBytesToEmit) {
302 if (getCurrentSectionData()->isBundleLocked())
303 report_fatal_error("Emitting values inside a locked bundle is forbidden");
304 MCObjectStreamer::EmitValueToAlignment(ByteAlignment, Value,
305 ValueSize, MaxBytesToEmit);
306}
307
Joerg Sonnenbergerfc184732013-10-29 01:06:17 +0000308// Add a symbol for the file name of this module. They start after the
309// null symbol and don't count as normal symbol, i.e. a non-STT_FILE symbol
310// with the same name may appear.
Matt Fleming6c1ad482010-08-16 18:57:57 +0000311void MCELFStreamer::EmitFileDirective(StringRef Filename) {
Joerg Sonnenbergerfc184732013-10-29 01:06:17 +0000312 getAssembler().addFileName(Filename);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000313}
314
Rafael Espindola5645bad2013-10-16 01:05:45 +0000315void MCELFStreamer::EmitIdent(StringRef IdentString) {
316 const MCSection *Comment = getAssembler().getContext().getELFSection(
Rafael Espindolaba31e272015-01-29 17:33:21 +0000317 ".comment", ELF::SHT_PROGBITS, ELF::SHF_MERGE | ELF::SHF_STRINGS, 1, "");
Rafael Espindola5645bad2013-10-16 01:05:45 +0000318 PushSection();
319 SwitchSection(Comment);
320 if (!SeenIdent) {
321 EmitIntValue(0, 1);
322 SeenIdent = true;
323 }
324 EmitBytes(IdentString);
325 EmitIntValue(0, 1);
326 PopSection();
327}
328
329void MCELFStreamer::fixSymbolsInTLSFixups(const MCExpr *expr) {
Rafael Espindola4e70ac72010-11-24 02:19:40 +0000330 switch (expr->getKind()) {
Tim Northovere0e3aef2013-01-31 12:12:40 +0000331 case MCExpr::Target:
332 cast<MCTargetExpr>(expr)->fixELFSymbolsInTLSFixups(getAssembler());
333 break;
Rafael Espindola4e70ac72010-11-24 02:19:40 +0000334 case MCExpr::Constant:
335 break;
336
337 case MCExpr::Binary: {
338 const MCBinaryExpr *be = cast<MCBinaryExpr>(expr);
339 fixSymbolsInTLSFixups(be->getLHS());
340 fixSymbolsInTLSFixups(be->getRHS());
341 break;
342 }
343
344 case MCExpr::SymbolRef: {
345 const MCSymbolRefExpr &symRef = *cast<MCSymbolRefExpr>(expr);
Rafael Espindolae98d4832010-11-24 18:51:21 +0000346 switch (symRef.getKind()) {
347 default:
Rafael Espindola4e70ac72010-11-24 02:19:40 +0000348 return;
Joerg Sonnenberger07de07e2011-03-17 00:35:10 +0000349 case MCSymbolRefExpr::VK_GOTTPOFF:
350 case MCSymbolRefExpr::VK_INDNTPOFF:
Rafael Espindolae98d4832010-11-24 18:51:21 +0000351 case MCSymbolRefExpr::VK_NTPOFF:
352 case MCSymbolRefExpr::VK_GOTNTPOFF:
353 case MCSymbolRefExpr::VK_TLSGD:
Joerg Sonnenberger07de07e2011-03-17 00:35:10 +0000354 case MCSymbolRefExpr::VK_TLSLD:
Rafael Espindolae98d4832010-11-24 18:51:21 +0000355 case MCSymbolRefExpr::VK_TLSLDM:
356 case MCSymbolRefExpr::VK_TPOFF:
357 case MCSymbolRefExpr::VK_DTPOFF:
Bruno Cardoso Lopes82b077e2011-10-25 18:13:20 +0000358 case MCSymbolRefExpr::VK_Mips_TLSGD:
359 case MCSymbolRefExpr::VK_Mips_GOTTPREL:
360 case MCSymbolRefExpr::VK_Mips_TPREL_HI:
361 case MCSymbolRefExpr::VK_Mips_TPREL_LO:
Ulrich Weigandf11efe72013-07-01 23:33:29 +0000362 case MCSymbolRefExpr::VK_PPC_DTPMOD:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000363 case MCSymbolRefExpr::VK_PPC_TPREL:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000364 case MCSymbolRefExpr::VK_PPC_TPREL_LO:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000365 case MCSymbolRefExpr::VK_PPC_TPREL_HI:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000366 case MCSymbolRefExpr::VK_PPC_TPREL_HA:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000367 case MCSymbolRefExpr::VK_PPC_TPREL_HIGHER:
368 case MCSymbolRefExpr::VK_PPC_TPREL_HIGHERA:
369 case MCSymbolRefExpr::VK_PPC_TPREL_HIGHEST:
370 case MCSymbolRefExpr::VK_PPC_TPREL_HIGHESTA:
371 case MCSymbolRefExpr::VK_PPC_DTPREL:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000372 case MCSymbolRefExpr::VK_PPC_DTPREL_LO:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000373 case MCSymbolRefExpr::VK_PPC_DTPREL_HI:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000374 case MCSymbolRefExpr::VK_PPC_DTPREL_HA:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000375 case MCSymbolRefExpr::VK_PPC_DTPREL_HIGHER:
376 case MCSymbolRefExpr::VK_PPC_DTPREL_HIGHERA:
377 case MCSymbolRefExpr::VK_PPC_DTPREL_HIGHEST:
378 case MCSymbolRefExpr::VK_PPC_DTPREL_HIGHESTA:
379 case MCSymbolRefExpr::VK_PPC_GOT_TPREL:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000380 case MCSymbolRefExpr::VK_PPC_GOT_TPREL_LO:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000381 case MCSymbolRefExpr::VK_PPC_GOT_TPREL_HI:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000382 case MCSymbolRefExpr::VK_PPC_GOT_TPREL_HA:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000383 case MCSymbolRefExpr::VK_PPC_GOT_DTPREL:
384 case MCSymbolRefExpr::VK_PPC_GOT_DTPREL_LO:
385 case MCSymbolRefExpr::VK_PPC_GOT_DTPREL_HI:
386 case MCSymbolRefExpr::VK_PPC_GOT_DTPREL_HA:
Bill Schmidt441907d2013-02-26 16:41:03 +0000387 case MCSymbolRefExpr::VK_PPC_TLS:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000388 case MCSymbolRefExpr::VK_PPC_GOT_TLSGD:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000389 case MCSymbolRefExpr::VK_PPC_GOT_TLSGD_LO:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000390 case MCSymbolRefExpr::VK_PPC_GOT_TLSGD_HI:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000391 case MCSymbolRefExpr::VK_PPC_GOT_TLSGD_HA:
Ulrich Weigand52cf8e42013-07-09 16:41:09 +0000392 case MCSymbolRefExpr::VK_PPC_TLSGD:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000393 case MCSymbolRefExpr::VK_PPC_GOT_TLSLD:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000394 case MCSymbolRefExpr::VK_PPC_GOT_TLSLD_LO:
Ulrich Weigand876a0d02013-06-21 14:44:15 +0000395 case MCSymbolRefExpr::VK_PPC_GOT_TLSLD_HI:
Ulrich Weigandd51c09f2013-06-21 14:42:20 +0000396 case MCSymbolRefExpr::VK_PPC_GOT_TLSLD_HA:
Ulrich Weigand52cf8e42013-07-09 16:41:09 +0000397 case MCSymbolRefExpr::VK_PPC_TLSLD:
Rafael Espindolae98d4832010-11-24 18:51:21 +0000398 break;
399 }
Rafael Espindola4e70ac72010-11-24 02:19:40 +0000400 MCSymbolData &SD = getAssembler().getOrCreateSymbolData(symRef.getSymbol());
Jan Sjödin30a52de2011-02-28 21:45:04 +0000401 MCELF::SetType(SD, ELF::STT_TLS);
Rafael Espindola4e70ac72010-11-24 02:19:40 +0000402 break;
403 }
404
405 case MCExpr::Unary:
406 fixSymbolsInTLSFixups(cast<MCUnaryExpr>(expr)->getSubExpr());
407 break;
408 }
409}
410
David Woodhouse6f3c73f2014-01-28 23:12:49 +0000411void MCELFStreamer::EmitInstToFragment(const MCInst &Inst,
412 const MCSubtargetInfo &STI) {
413 this->MCObjectStreamer::EmitInstToFragment(Inst, STI);
Eli Bendersky4d9ada02013-01-08 00:22:56 +0000414 MCRelaxableFragment &F = *cast<MCRelaxableFragment>(getCurrentFragment());
Benjamin Kramer1f601242010-08-27 10:40:51 +0000415
Rafael Espindola3fa6fc12010-12-02 05:44:06 +0000416 for (unsigned i = 0, e = F.getFixups().size(); i != e; ++i)
417 fixSymbolsInTLSFixups(F.getFixups()[i].getValue());
Benjamin Kramer1f601242010-08-27 10:40:51 +0000418}
419
David Woodhouse6f3c73f2014-01-28 23:12:49 +0000420void MCELFStreamer::EmitInstToData(const MCInst &Inst,
421 const MCSubtargetInfo &STI) {
Eli Benderskyf483ff92012-12-20 19:05:53 +0000422 MCAssembler &Assembler = getAssembler();
Benjamin Kramer1f601242010-08-27 10:40:51 +0000423 SmallVector<MCFixup, 4> Fixups;
424 SmallString<256> Code;
425 raw_svector_ostream VecOS(Code);
David Woodhouse9784cef2014-01-28 23:13:07 +0000426 Assembler.getEmitter().EncodeInstruction(Inst, VecOS, Fixups, STI);
Benjamin Kramer1f601242010-08-27 10:40:51 +0000427 VecOS.flush();
428
Rafael Espindola4e70ac72010-11-24 02:19:40 +0000429 for (unsigned i = 0, e = Fixups.size(); i != e; ++i)
430 fixSymbolsInTLSFixups(Fixups[i].getValue());
431
Eli Benderskyf483ff92012-12-20 19:05:53 +0000432 // There are several possibilities here:
433 //
434 // If bundling is disabled, append the encoded instruction to the current data
435 // fragment (or create a new such fragment if the current fragment is not a
436 // data fragment).
437 //
438 // If bundling is enabled:
Eli Benderskycf6009b2013-01-15 23:22:09 +0000439 // - If we're not in a bundle-locked group, emit the instruction into a
440 // fragment of its own. If there are no fixups registered for the
441 // instruction, emit a MCCompactEncodedInstFragment. Otherwise, emit a
442 // MCDataFragment.
Eli Benderskyf483ff92012-12-20 19:05:53 +0000443 // - If we're in a bundle-locked group, append the instruction to the current
444 // data fragment because we want all the instructions in a group to get into
445 // the same fragment. Be careful not to do that for the first instruction in
446 // the group, though.
447 MCDataFragment *DF;
448
449 if (Assembler.isBundlingEnabled()) {
450 MCSectionData *SD = getCurrentSectionData();
451 if (SD->isBundleLocked() && !SD->isBundleGroupBeforeFirstInst())
Derek Schuff8878bcc92013-02-15 22:50:52 +0000452 // If we are bundle-locked, we re-use the current fragment.
453 // The bundle-locking directive ensures this is a new data fragment.
454 DF = cast<MCDataFragment>(getCurrentFragment());
Eli Benderskycf6009b2013-01-15 23:22:09 +0000455 else if (!SD->isBundleLocked() && Fixups.size() == 0) {
456 // Optimize memory usage by emitting the instruction to a
457 // MCCompactEncodedInstFragment when not in a bundle-locked group and
458 // there are no fixups registered.
Peter Collingbourne2f495b92013-04-17 21:18:16 +0000459 MCCompactEncodedInstFragment *CEIF = new MCCompactEncodedInstFragment();
460 insert(CEIF);
Eli Benderskycf6009b2013-01-15 23:22:09 +0000461 CEIF->getContents().append(Code.begin(), Code.end());
462 return;
Derek Schuff8878bcc92013-02-15 22:50:52 +0000463 } else {
Peter Collingbourne2f495b92013-04-17 21:18:16 +0000464 DF = new MCDataFragment();
465 insert(DF);
Derek Schuff05fb7352014-10-15 17:10:04 +0000466 }
467 if (SD->getBundleLockState() == MCSectionData::BundleLockedAlignToEnd) {
468 // If this fragment is for a group marked "align_to_end", set a flag
469 // in the fragment. This can happen after the fragment has already been
470 // created if there are nested bundle_align groups and an inner one
471 // is the one marked align_to_end.
472 DF->setAlignToBundleEnd(true);
Eli Bendersky802b6282013-01-07 21:51:08 +0000473 }
Eli Benderskyf483ff92012-12-20 19:05:53 +0000474
475 // We're now emitting an instruction in a bundle group, so this flag has
476 // to be turned off.
477 SD->setBundleGroupBeforeFirstInst(false);
478 } else {
479 DF = getOrCreateDataFragment();
480 }
481
Benjamin Kramer1f601242010-08-27 10:40:51 +0000482 // Add the fixups and data.
483 for (unsigned i = 0, e = Fixups.size(); i != e; ++i) {
484 Fixups[i].setOffset(Fixups[i].getOffset() + DF->getContents().size());
Eli Benderskya31a8942012-12-07 19:13:57 +0000485 DF->getFixups().push_back(Fixups[i]);
Benjamin Kramer1f601242010-08-27 10:40:51 +0000486 }
Eli Benderskyf483ff92012-12-20 19:05:53 +0000487 DF->setHasInstructions(true);
Benjamin Kramer1f601242010-08-27 10:40:51 +0000488 DF->getContents().append(Code.begin(), Code.end());
489}
490
Eli Benderskyf483ff92012-12-20 19:05:53 +0000491void MCELFStreamer::EmitBundleAlignMode(unsigned AlignPow2) {
492 assert(AlignPow2 <= 30 && "Invalid bundle alignment");
493 MCAssembler &Assembler = getAssembler();
Derek Schuff05fb7352014-10-15 17:10:04 +0000494 if (AlignPow2 > 0 && (Assembler.getBundleAlignSize() == 0 ||
495 Assembler.getBundleAlignSize() == 1U << AlignPow2))
496 Assembler.setBundleAlignSize(1U << AlignPow2);
Eli Benderskyf483ff92012-12-20 19:05:53 +0000497 else
Derek Schuff05fb7352014-10-15 17:10:04 +0000498 report_fatal_error(".bundle_align_mode cannot be changed once set");
Eli Benderskyf483ff92012-12-20 19:05:53 +0000499}
500
Eli Bendersky802b6282013-01-07 21:51:08 +0000501void MCELFStreamer::EmitBundleLock(bool AlignToEnd) {
Eli Benderskyf483ff92012-12-20 19:05:53 +0000502 MCSectionData *SD = getCurrentSectionData();
503
504 // Sanity checks
505 //
506 if (!getAssembler().isBundlingEnabled())
507 report_fatal_error(".bundle_lock forbidden when bundling is disabled");
Derek Schuff05fb7352014-10-15 17:10:04 +0000508
509 if (!SD->isBundleLocked())
510 SD->setBundleGroupBeforeFirstInst(true);
Eli Benderskyf483ff92012-12-20 19:05:53 +0000511
Eli Bendersky802b6282013-01-07 21:51:08 +0000512 SD->setBundleLockState(AlignToEnd ? MCSectionData::BundleLockedAlignToEnd :
513 MCSectionData::BundleLocked);
Eli Benderskyf483ff92012-12-20 19:05:53 +0000514}
515
516void MCELFStreamer::EmitBundleUnlock() {
517 MCSectionData *SD = getCurrentSectionData();
518
519 // Sanity checks
520 if (!getAssembler().isBundlingEnabled())
521 report_fatal_error(".bundle_unlock forbidden when bundling is disabled");
522 else if (!SD->isBundleLocked())
523 report_fatal_error(".bundle_unlock without matching lock");
524 else if (SD->isBundleGroupBeforeFirstInst())
525 report_fatal_error("Empty bundle-locked group is forbidden");
526
Eli Bendersky802b6282013-01-07 21:51:08 +0000527 SD->setBundleLockState(MCSectionData::NotBundleLocked);
Eli Benderskyf483ff92012-12-20 19:05:53 +0000528}
529
Richard Mitton21101b32013-09-19 23:21:01 +0000530void MCELFStreamer::Flush() {
Rafael Espindola53f0bf12010-09-29 14:52:01 +0000531 for (std::vector<LocalCommon>::const_iterator i = LocalCommons.begin(),
532 e = LocalCommons.end();
533 i != e; ++i) {
534 MCSymbolData *SD = i->SD;
535 uint64_t Size = i->Size;
536 unsigned ByteAlignment = i->ByteAlignment;
537 const MCSymbol &Symbol = SD->getSymbol();
538 const MCSection &Section = Symbol.getSection();
539
540 MCSectionData &SectData = getAssembler().getOrCreateSectionData(Section);
541 new MCAlignFragment(ByteAlignment, 0, 1, ByteAlignment, &SectData);
542
543 MCFragment *F = new MCFillFragment(0, 0, Size, &SectData);
544 SD->setFragment(F);
545
546 // Update the maximum alignment of the section if necessary.
547 if (ByteAlignment > SectData.getAlignment())
548 SectData.setAlignment(ByteAlignment);
549 }
550
Richard Mitton21101b32013-09-19 23:21:01 +0000551 LocalCommons.clear();
552}
553
554void MCELFStreamer::FinishImpl() {
Rafael Espindola7f4ccce2014-05-12 13:30:10 +0000555 EmitFrames(nullptr);
Richard Mitton21101b32013-09-19 23:21:01 +0000556
557 Flush();
558
Rafael Espindola07082092012-01-07 03:13:18 +0000559 this->MCObjectStreamer::FinishImpl();
Matt Fleming6c1ad482010-08-16 18:57:57 +0000560}
Richard Mitton21101b32013-09-19 23:21:01 +0000561
Rafael Espindola24ea09e2014-01-26 06:06:37 +0000562MCStreamer *llvm::createELFStreamer(MCContext &Context, MCAsmBackend &MAB,
563 raw_ostream &OS, MCCodeEmitter *CE,
Rafael Espindola7b61ddf2014-10-15 16:12:52 +0000564 bool RelaxAll) {
Rafael Espindola24ea09e2014-01-26 06:06:37 +0000565 MCELFStreamer *S = new MCELFStreamer(Context, MAB, OS, CE);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000566 if (RelaxAll)
567 S->getAssembler().setRelaxAll(true);
Matt Fleming6c1ad482010-08-16 18:57:57 +0000568 return S;
569}
Logan Chien59ff0702012-12-07 15:50:40 +0000570
Tim Northover5cc3dc82012-12-07 16:50:23 +0000571void MCELFStreamer::EmitThumbFunc(MCSymbol *Func) {
572 llvm_unreachable("Generic ELF doesn't support this directive");
573}
574
Logan Chien59ff0702012-12-07 15:50:40 +0000575void MCELFStreamer::EmitSymbolDesc(MCSymbol *Symbol, unsigned DescValue) {
576 llvm_unreachable("ELF doesn't support this directive");
577}
578
579void MCELFStreamer::BeginCOFFSymbolDef(const MCSymbol *Symbol) {
580 llvm_unreachable("ELF doesn't support this directive");
581}
582
583void MCELFStreamer::EmitCOFFSymbolStorageClass(int StorageClass) {
584 llvm_unreachable("ELF doesn't support this directive");
585}
586
587void MCELFStreamer::EmitCOFFSymbolType(int Type) {
588 llvm_unreachable("ELF doesn't support this directive");
589}
590
591void MCELFStreamer::EndCOFFSymbolDef() {
592 llvm_unreachable("ELF doesn't support this directive");
593}
594
595void MCELFStreamer::EmitZerofill(const MCSection *Section, MCSymbol *Symbol,
596 uint64_t Size, unsigned ByteAlignment) {
597 llvm_unreachable("ELF doesn't support this directive");
598}
599
600void MCELFStreamer::EmitTBSSSymbol(const MCSection *Section, MCSymbol *Symbol,
601 uint64_t Size, unsigned ByteAlignment) {
602 llvm_unreachable("ELF doesn't support this directive");
603}