blob: 3c9487bffec90ea1cb2ea8f428b5d9ec5f6790d7 [file] [log] [blame]
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +00001//===- lib/MC/MCAssembler.cpp - Assembler Backend Implementation ----------===//
2//
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
Daniel Dunbar0adcd352009-08-25 21:10:45 +000010#define DEBUG_TYPE "assembler"
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +000011#include "llvm/MC/MCAssembler.h"
Daniel Dunbar18ff2cc2010-03-11 05:53:33 +000012#include "llvm/MC/MCAsmLayout.h"
Daniel Dunbarb36052f2010-03-19 10:43:23 +000013#include "llvm/MC/MCCodeEmitter.h"
Rafael Espindolafea753b2010-12-24 21:22:02 +000014#include "llvm/MC/MCContext.h"
Daniel Dunbar1253a6f2009-10-16 01:58:03 +000015#include "llvm/MC/MCExpr.h"
Daniel Dunbar53b23382010-03-19 09:28:59 +000016#include "llvm/MC/MCObjectWriter.h"
Kevin Enderbyc0957932010-09-30 16:52:03 +000017#include "llvm/MC/MCSection.h"
Daniel Dunbar1253a6f2009-10-16 01:58:03 +000018#include "llvm/MC/MCSymbol.h"
19#include "llvm/MC/MCValue.h"
Kevin Enderbyc0957932010-09-30 16:52:03 +000020#include "llvm/MC/MCDwarf.h"
Daniel Dunbar1a9158c2010-03-19 10:43:26 +000021#include "llvm/ADT/OwningPtr.h"
Daniel Dunbar0adcd352009-08-25 21:10:45 +000022#include "llvm/ADT/Statistic.h"
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +000023#include "llvm/ADT/StringExtras.h"
Daniel Dunbard6f761e2009-08-21 23:07:38 +000024#include "llvm/ADT/Twine.h"
Daniel Dunbarac2884a2010-03-25 22:49:09 +000025#include "llvm/Support/Debug.h"
Daniel Dunbar0705fbf2009-08-21 18:29:01 +000026#include "llvm/Support/ErrorHandling.h"
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +000027#include "llvm/Support/raw_ostream.h"
Daniel Dunbaree0d8922010-03-13 22:10:17 +000028#include "llvm/Target/TargetRegistry.h"
Daniel Dunbardf3c8f22010-03-12 21:00:49 +000029#include "llvm/Target/TargetAsmBackend.h"
Daniel Dunbarf6346762010-02-13 09:29:02 +000030
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +000031using namespace llvm;
32
Daniel Dunbarff547842010-03-23 23:47:14 +000033namespace {
34namespace stats {
Daniel Dunbar0adcd352009-08-25 21:10:45 +000035STATISTIC(EmittedFragments, "Number of emitted assembler fragments");
Daniel Dunbarff547842010-03-23 23:47:14 +000036STATISTIC(EvaluateFixup, "Number of evaluated fixups");
Daniel Dunbarac2884a2010-03-25 22:49:09 +000037STATISTIC(FragmentLayouts, "Number of fragment layouts");
Daniel Dunbarff547842010-03-23 23:47:14 +000038STATISTIC(ObjectBytes, "Number of emitted object file bytes");
Daniel Dunbarac2884a2010-03-25 22:49:09 +000039STATISTIC(RelaxationSteps, "Number of assembler layout and relaxation steps");
40STATISTIC(RelaxedInstructions, "Number of relaxed instructions");
Daniel Dunbarff547842010-03-23 23:47:14 +000041}
42}
Daniel Dunbar0adcd352009-08-25 21:10:45 +000043
Daniel Dunbar8f4d1462009-08-28 07:08:35 +000044// FIXME FIXME FIXME: There are number of places in this file where we convert
45// what is a 64-bit assembler value used for computation into a value in the
46// object file, which may truncate it. We should detect that truncation where
47// invalid and report errors back.
48
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +000049/* *** */
50
Daniel Dunbar9005d452010-05-14 00:37:21 +000051MCAsmLayout::MCAsmLayout(MCAssembler &Asm)
Rafael Espindola4f4363a2010-12-07 23:32:26 +000052 : Assembler(Asm), LastValidFragment()
Daniel Dunbar9005d452010-05-14 00:37:21 +000053 {
Daniel Dunbarbc1a0cf2010-05-12 15:42:59 +000054 // Compute the section layout order. Virtual sections must go last.
55 for (MCAssembler::iterator it = Asm.begin(), ie = Asm.end(); it != ie; ++it)
Rafael Espindolaf2dc4aa2010-11-17 20:03:54 +000056 if (!it->getSection().isVirtualSection())
Daniel Dunbarbc1a0cf2010-05-12 15:42:59 +000057 SectionOrder.push_back(&*it);
58 for (MCAssembler::iterator it = Asm.begin(), ie = Asm.end(); it != ie; ++it)
Rafael Espindolaf2dc4aa2010-11-17 20:03:54 +000059 if (it->getSection().isVirtualSection())
Daniel Dunbarbc1a0cf2010-05-12 15:42:59 +000060 SectionOrder.push_back(&*it);
61}
62
Daniel Dunbar9005d452010-05-14 00:37:21 +000063bool MCAsmLayout::isFragmentUpToDate(const MCFragment *F) const {
Rafael Espindola4f4363a2010-12-07 23:32:26 +000064 const MCSectionData &SD = *F->getParent();
65 const MCFragment *LastValid = LastValidFragment.lookup(&SD);
66 if (!LastValid)
67 return false;
68 assert(LastValid->getParent() == F->getParent());
69 return F->getLayoutOrder() <= LastValid->getLayoutOrder();
Daniel Dunbar9005d452010-05-14 00:37:21 +000070}
71
Rafael Espindolaa9d42812010-11-23 08:08:33 +000072void MCAsmLayout::Invalidate(MCFragment *F) {
Daniel Dunbar47b3ec42010-05-14 00:51:14 +000073 // If this fragment wasn't already up-to-date, we don't need to do anything.
74 if (!isFragmentUpToDate(F))
75 return;
Daniel Dunbar0cc8bd42010-03-25 19:35:56 +000076
Rafael Espindola2bf6afc2010-12-15 08:45:53 +000077 // Otherwise, reset the last valid fragment to this fragment.
Rafael Espindola4f4363a2010-12-07 23:32:26 +000078 const MCSectionData &SD = *F->getParent();
Rafael Espindola2bf6afc2010-12-15 08:45:53 +000079 LastValidFragment[&SD] = F;
Daniel Dunbar47b3ec42010-05-14 00:51:14 +000080}
Daniel Dunbar0cc8bd42010-03-25 19:35:56 +000081
Daniel Dunbar47b3ec42010-05-14 00:51:14 +000082void MCAsmLayout::EnsureValid(const MCFragment *F) const {
Rafael Espindola4f4363a2010-12-07 23:32:26 +000083 MCSectionData &SD = *F->getParent();
84
85 MCFragment *Cur = LastValidFragment[&SD];
86 if (!Cur)
87 Cur = &*SD.begin();
88 else
89 Cur = Cur->getNextNode();
90
Daniel Dunbar47b3ec42010-05-14 00:51:14 +000091 // Advance the layout position until the fragment is up-to-date.
92 while (!isFragmentUpToDate(F)) {
Daniel Dunbar47b3ec42010-05-14 00:51:14 +000093 const_cast<MCAsmLayout*>(this)->LayoutFragment(Cur);
Rafael Espindola4f4363a2010-12-07 23:32:26 +000094 Cur = Cur->getNextNode();
Daniel Dunbar47b3ec42010-05-14 00:51:14 +000095 }
Daniel Dunbar0cc8bd42010-03-25 19:35:56 +000096}
97
Daniel Dunbar432cd5f2010-03-25 02:00:02 +000098uint64_t MCAsmLayout::getFragmentOffset(const MCFragment *F) const {
Daniel Dunbar47b3ec42010-05-14 00:51:14 +000099 EnsureValid(F);
Daniel Dunbar432cd5f2010-03-25 02:00:02 +0000100 assert(F->Offset != ~UINT64_C(0) && "Address not set!");
101 return F->Offset;
102}
103
Rafael Espindolaffd902b2010-12-06 02:57:26 +0000104uint64_t MCAsmLayout::getSymbolOffset(const MCSymbolData *SD) const {
105 assert(SD->getFragment() && "Invalid getOffset() on undefined symbol!");
106 return getFragmentOffset(SD->getFragment()) + SD->getOffset();
107}
108
Daniel Dunbar2661f112010-05-13 03:19:50 +0000109uint64_t MCAsmLayout::getSectionAddressSize(const MCSectionData *SD) const {
Daniel Dunbarafc6acd2010-05-14 00:37:11 +0000110 // The size is the last fragment's end offset.
Daniel Dunbar2661f112010-05-13 03:19:50 +0000111 const MCFragment &F = SD->getFragmentList().back();
Rafael Espindola7a459032010-12-21 20:35:18 +0000112 return getFragmentOffset(&F) + getAssembler().ComputeFragmentSize(*this, F);
Daniel Dunbar5d428512010-03-25 02:00:07 +0000113}
114
115uint64_t MCAsmLayout::getSectionFileSize(const MCSectionData *SD) const {
Daniel Dunbar2661f112010-05-13 03:19:50 +0000116 // Virtual sections have no file size.
Rafael Espindolaf2dc4aa2010-11-17 20:03:54 +0000117 if (SD->getSection().isVirtualSection())
Daniel Dunbar2661f112010-05-13 03:19:50 +0000118 return 0;
119
120 // Otherwise, the file size is the same as the address space size.
121 return getSectionAddressSize(SD);
Daniel Dunbar5d428512010-03-25 02:00:07 +0000122}
123
Daniel Dunbar207e06e2010-03-24 03:43:40 +0000124/* *** */
125
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000126MCFragment::MCFragment() : Kind(FragmentType(~0)) {
127}
128
Daniel Dunbar36880e72010-07-28 20:28:45 +0000129MCFragment::~MCFragment() {
130}
131
Daniel Dunbar5e835962009-08-26 02:48:04 +0000132MCFragment::MCFragment(FragmentType _Kind, MCSectionData *_Parent)
Rafael Espindola2bf6afc2010-12-15 08:45:53 +0000133 : Kind(_Kind), Parent(_Parent), Atom(0), Offset(~UINT64_C(0))
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +0000134{
Daniel Dunbar5e835962009-08-26 02:48:04 +0000135 if (Parent)
136 Parent->getFragmentList().push_back(this);
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +0000137}
138
139/* *** */
140
Daniel Dunbar81e40002009-08-27 00:38:04 +0000141MCSectionData::MCSectionData() : Section(0) {}
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +0000142
143MCSectionData::MCSectionData(const MCSection &_Section, MCAssembler *A)
Daniel Dunbar81e40002009-08-27 00:38:04 +0000144 : Section(&_Section),
Rafael Espindolaf8803fe2010-12-06 03:48:09 +0000145 Ordinal(~UINT32_C(0)),
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +0000146 Alignment(1),
Daniel Dunbare1ec6172010-02-02 21:44:01 +0000147 HasInstructions(false)
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +0000148{
149 if (A)
150 A->getSectionList().push_back(this);
151}
152
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +0000153/* *** */
154
Daniel Dunbarefbb5332009-09-01 04:09:03 +0000155MCSymbolData::MCSymbolData() : Symbol(0) {}
Daniel Dunbarf3d2ef02009-08-22 10:13:24 +0000156
Daniel Dunbarcb579b32009-08-31 08:08:06 +0000157MCSymbolData::MCSymbolData(const MCSymbol &_Symbol, MCFragment *_Fragment,
Daniel Dunbarf3d2ef02009-08-22 10:13:24 +0000158 uint64_t _Offset, MCAssembler *A)
Daniel Dunbarefbb5332009-09-01 04:09:03 +0000159 : Symbol(&_Symbol), Fragment(_Fragment), Offset(_Offset),
Daniel Dunbar8f4d1462009-08-28 07:08:35 +0000160 IsExternal(false), IsPrivateExtern(false),
Matt Fleming6c8b3d22010-08-16 18:34:31 +0000161 CommonSize(0), SymbolSize(0), CommonAlign(0),
162 Flags(0), Index(0)
Daniel Dunbarf3d2ef02009-08-22 10:13:24 +0000163{
164 if (A)
165 A->getSymbolList().push_back(this);
166}
167
168/* *** */
169
Daniel Dunbarfeb7ba32010-12-17 02:45:41 +0000170MCAssembler::MCAssembler(MCContext &Context_, TargetAsmBackend &Backend_,
171 MCCodeEmitter &Emitter_, MCObjectWriter &Writer_,
172 raw_ostream &OS_)
173 : Context(Context_), Backend(Backend_), Emitter(Emitter_), Writer(Writer_),
Rafael Espindola96aa78c2011-01-23 17:55:27 +0000174 OS(OS_), RelaxAll(false), NoExecStack(false), SubsectionsViaSymbols(false)
Daniel Dunbar6009db42009-08-26 21:22:22 +0000175{
176}
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +0000177
178MCAssembler::~MCAssembler() {
179}
180
Daniel Dunbar843aa1f2010-06-16 20:04:29 +0000181bool MCAssembler::isSymbolLinkerVisible(const MCSymbol &Symbol) const {
Daniel Dunbar23869852010-03-19 03:18:09 +0000182 // Non-temporary labels should always be visible to the linker.
Daniel Dunbar843aa1f2010-06-16 20:04:29 +0000183 if (!Symbol.isTemporary())
Daniel Dunbar23869852010-03-19 03:18:09 +0000184 return true;
185
186 // Absolute temporary labels are never visible.
Daniel Dunbar843aa1f2010-06-16 20:04:29 +0000187 if (!Symbol.isInSection())
Daniel Dunbar23869852010-03-19 03:18:09 +0000188 return false;
189
190 // Otherwise, check if the section requires symbols even for temporary labels.
Daniel Dunbar843aa1f2010-06-16 20:04:29 +0000191 return getBackend().doesSectionRequireSymbols(Symbol.getSection());
Daniel Dunbar23869852010-03-19 03:18:09 +0000192}
193
Rafael Espindolab8141102010-09-27 18:13:03 +0000194const MCSymbolData *MCAssembler::getAtom(const MCSymbolData *SD) const {
Daniel Dunbar8ad0dcc2010-03-19 03:18:15 +0000195 // Linker visible symbols define atoms.
Daniel Dunbar843aa1f2010-06-16 20:04:29 +0000196 if (isSymbolLinkerVisible(SD->getSymbol()))
Daniel Dunbar8ad0dcc2010-03-19 03:18:15 +0000197 return SD;
198
199 // Absolute and undefined symbols have no defining atom.
200 if (!SD->getFragment())
201 return 0;
202
Daniel Dunbara5f1d572010-05-12 00:38:17 +0000203 // Non-linker visible symbols in sections which can't be atomized have no
204 // defining atom.
205 if (!getBackend().isSectionAtomizable(
206 SD->getFragment()->getParent()->getSection()))
207 return 0;
208
Daniel Dunbar651804c2010-05-11 17:22:50 +0000209 // Otherwise, return the atom for the containing fragment.
210 return SD->getFragment()->getAtom();
Daniel Dunbar8ad0dcc2010-03-19 03:18:15 +0000211}
212
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000213bool MCAssembler::EvaluateFixup(const MCAsmLayout &Layout,
Daniel Dunbarc90e30a2010-05-26 15:18:56 +0000214 const MCFixup &Fixup, const MCFragment *DF,
Daniel Dunbardf3c8f22010-03-12 21:00:49 +0000215 MCValue &Target, uint64_t &Value) const {
Daniel Dunbarff547842010-03-23 23:47:14 +0000216 ++stats::EvaluateFixup;
217
Rafael Espindola33a38a12010-12-22 16:11:57 +0000218 if (!Fixup.getValue()->EvaluateAsRelocatable(Target, Layout))
Chris Lattner75361b62010-04-07 22:58:41 +0000219 report_fatal_error("expected relocatable expression");
Daniel Dunbardf3c8f22010-03-12 21:00:49 +0000220
Rafael Espindolafea753b2010-12-24 21:22:02 +0000221 bool IsPCRel = Backend.getFixupKindInfo(
222 Fixup.getKind()).Flags & MCFixupKindInfo::FKF_IsPCRel;
223
224 bool IsResolved;
225 if (IsPCRel) {
226 if (Target.getSymB()) {
227 IsResolved = false;
228 } else if (!Target.getSymA()) {
229 IsResolved = false;
230 } else {
Rafael Espindola908159b2011-02-16 03:25:55 +0000231 const MCSymbolRefExpr *A = Target.getSymA();
232 const MCSymbol &SA = A->getSymbol();
233 if (A->getKind() != MCSymbolRefExpr::VK_None ||
234 SA.AliasedSymbol().isUndefined()) {
Rafael Espindolafea753b2010-12-24 21:22:02 +0000235 IsResolved = false;
236 } else {
237 const MCSymbolData &DataA = getSymbolData(SA);
238 IsResolved =
239 getWriter().IsSymbolRefDifferenceFullyResolvedImpl(*this, DataA,
240 *DF, false, true);
241 }
242 }
243 } else {
244 IsResolved = Target.isAbsolute();
245 }
Daniel Dunbardf3c8f22010-03-12 21:00:49 +0000246
247 Value = Target.getConstant();
248
Jim Grosbach47500202010-12-14 18:46:57 +0000249 bool IsThumb = false;
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000250 if (const MCSymbolRefExpr *A = Target.getSymA()) {
Rafael Espindola94ed5fc2010-11-15 16:33:49 +0000251 const MCSymbol &Sym = A->getSymbol().AliasedSymbol();
252 if (Sym.isDefined())
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000253 Value += Layout.getSymbolOffset(&getSymbolData(Sym));
Jim Grosbach47500202010-12-14 18:46:57 +0000254 if (isThumbFunc(&Sym))
255 IsThumb = true;
Daniel Dunbardf3c8f22010-03-12 21:00:49 +0000256 }
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000257 if (const MCSymbolRefExpr *B = Target.getSymB()) {
Rafael Espindola94ed5fc2010-11-15 16:33:49 +0000258 const MCSymbol &Sym = B->getSymbol().AliasedSymbol();
259 if (Sym.isDefined())
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000260 Value -= Layout.getSymbolOffset(&getSymbolData(Sym));
Daniel Dunbar939f8d72010-03-19 03:18:12 +0000261 }
Daniel Dunbardf3c8f22010-03-12 21:00:49 +0000262
Daniel Dunbardf3c8f22010-03-12 21:00:49 +0000263
Daniel Dunbar2761fc42010-12-16 03:20:06 +0000264 bool ShouldAlignPC = Backend.getFixupKindInfo(Fixup.getKind()).Flags &
Owen Anderson47dbd422010-12-15 18:48:27 +0000265 MCFixupKindInfo::FKF_IsAlignedDownTo32Bits;
266 assert((ShouldAlignPC ? IsPCRel : true) &&
267 "FKF_IsAlignedDownTo32Bits is only allowed on PC-relative fixups!");
268
Owen Anderson05018c22010-12-09 20:27:52 +0000269 if (IsPCRel) {
Owen Anderson175fb362010-12-17 21:49:48 +0000270 uint32_t Offset = Layout.getFragmentOffset(DF) + Fixup.getOffset();
Owen Anderson47dbd422010-12-15 18:48:27 +0000271
272 // A number of ARM fixups in Thumb mode require that the effective PC
273 // address be determined as the 32-bit aligned version of the actual offset.
Owen Andersond18e0112010-12-15 19:24:24 +0000274 if (ShouldAlignPC) Offset &= ~0x3;
Owen Anderson175fb362010-12-17 21:49:48 +0000275 Value -= Offset;
Owen Anderson05018c22010-12-09 20:27:52 +0000276 }
Daniel Dunbardf3c8f22010-03-12 21:00:49 +0000277
Jim Grosbach47500202010-12-14 18:46:57 +0000278 // ARM fixups based from a thumb function address need to have the low
279 // bit set. The actual value is always at least 16-bit aligned, so the
280 // low bit is normally clear and available for use as an ISA flag for
281 // interworking.
282 if (IsThumb)
283 Value |= 1;
284
Daniel Dunbardf3c8f22010-03-12 21:00:49 +0000285 return IsResolved;
286}
287
Rafael Espindola7a459032010-12-21 20:35:18 +0000288uint64_t MCAssembler::ComputeFragmentSize(const MCAsmLayout &Layout,
289 const MCFragment &F) const {
Daniel Dunbar2c18d3b2010-05-13 18:35:06 +0000290 switch (F.getKind()) {
291 case MCFragment::FT_Data:
292 return cast<MCDataFragment>(F).getContents().size();
293 case MCFragment::FT_Fill:
294 return cast<MCFillFragment>(F).getSize();
295 case MCFragment::FT_Inst:
296 return cast<MCInstFragment>(F).getInstSize();
297
Rafael Espindola3ff57092010-11-02 17:22:24 +0000298 case MCFragment::FT_LEB:
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000299 return cast<MCLEBFragment>(F).getContents().size();
Rafael Espindola3ff57092010-11-02 17:22:24 +0000300
Rafael Espindola7a459032010-12-21 20:35:18 +0000301 case MCFragment::FT_Align: {
302 const MCAlignFragment &AF = cast<MCAlignFragment>(F);
303 unsigned Offset = Layout.getFragmentOffset(&AF);
304 unsigned Size = OffsetToAlignment(Offset, AF.getAlignment());
305 if (Size > AF.getMaxBytesToEmit())
306 return 0;
307 return Size;
308 }
Daniel Dunbar2c18d3b2010-05-13 18:35:06 +0000309
Rafael Espindola7a459032010-12-21 20:35:18 +0000310 case MCFragment::FT_Org: {
311 MCOrgFragment &OF = cast<MCOrgFragment>(F);
312 int64_t TargetLocation;
313 if (!OF.getOffset().EvaluateAsAbsolute(TargetLocation, Layout))
314 report_fatal_error("expected assembly-time absolute expression");
315
316 // FIXME: We need a way to communicate this error.
317 uint64_t FragmentOffset = Layout.getFragmentOffset(&OF);
318 int64_t Size = TargetLocation - FragmentOffset;
319 if (Size < 0 || Size >= 0x40000000)
320 report_fatal_error("invalid .org offset '" + Twine(TargetLocation) +
321 "' (at offset '" + Twine(FragmentOffset) + "')");
322 return Size;
323 }
Kevin Enderbyc0957932010-09-30 16:52:03 +0000324
Rafael Espindola187d8332010-11-07 02:07:12 +0000325 case MCFragment::FT_Dwarf:
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000326 return cast<MCDwarfLineAddrFragment>(F).getContents().size();
Rafael Espindola245a1e22010-12-28 05:39:27 +0000327 case MCFragment::FT_DwarfFrame:
328 return cast<MCDwarfCallFrameFragment>(F).getContents().size();
Daniel Dunbar2c18d3b2010-05-13 18:35:06 +0000329 }
330
331 assert(0 && "invalid fragment kind");
332 return 0;
333}
334
Daniel Dunbarb69fc042010-05-13 20:40:12 +0000335void MCAsmLayout::LayoutFragment(MCFragment *F) {
Daniel Dunbar9005d452010-05-14 00:37:21 +0000336 MCFragment *Prev = F->getPrevNode();
Daniel Dunbarf0d17d22010-05-12 21:35:25 +0000337
Daniel Dunbar9005d452010-05-14 00:37:21 +0000338 // We should never try to recompute something which is up-to-date.
339 assert(!isFragmentUpToDate(F) && "Attempt to recompute up-to-date fragment!");
Daniel Dunbar9005d452010-05-14 00:37:21 +0000340 // We should never try to compute the fragment layout if it's predecessor
341 // isn't up-to-date.
342 assert((!Prev || isFragmentUpToDate(Prev)) &&
343 "Attempt to compute fragment before it's predecessor!");
Daniel Dunbarf0d17d22010-05-12 21:35:25 +0000344
345 ++stats::FragmentLayouts;
346
Daniel Dunbarb69fc042010-05-13 20:40:12 +0000347 // Compute fragment offset and size.
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000348 uint64_t Offset = 0;
349 if (Prev)
Rafael Espindola7a459032010-12-21 20:35:18 +0000350 Offset += Prev->Offset + getAssembler().ComputeFragmentSize(*this, *Prev);
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000351
352 F->Offset = Offset;
Rafael Espindola4f4363a2010-12-07 23:32:26 +0000353 LastValidFragment[F->getParent()] = F;
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000354}
355
Daniel Dunbar53b23382010-03-19 09:28:59 +0000356/// WriteFragmentData - Write the \arg F data to the output file.
Daniel Dunbar432cd5f2010-03-25 02:00:02 +0000357static void WriteFragmentData(const MCAssembler &Asm, const MCAsmLayout &Layout,
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000358 const MCFragment &F) {
359 MCObjectWriter *OW = &Asm.getWriter();
Daniel Dunbar53b23382010-03-19 09:28:59 +0000360 uint64_t Start = OW->getStream().tell();
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000361 (void) Start;
Daniel Dunbar7eb85192009-10-16 01:58:15 +0000362
Daniel Dunbarff547842010-03-23 23:47:14 +0000363 ++stats::EmittedFragments;
Daniel Dunbar0adcd352009-08-25 21:10:45 +0000364
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000365 // FIXME: Embed in fragments instead?
Rafael Espindola7a459032010-12-21 20:35:18 +0000366 uint64_t FragmentSize = Asm.ComputeFragmentSize(Layout, F);
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000367 switch (F.getKind()) {
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000368 case MCFragment::FT_Align: {
369 MCAlignFragment &AF = cast<MCAlignFragment>(F);
Daniel Dunbar432cd5f2010-03-25 02:00:02 +0000370 uint64_t Count = FragmentSize / AF.getValueSize();
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000371
Daniel Dunbare73d49e2010-05-12 22:51:27 +0000372 assert(AF.getValueSize() && "Invalid virtual align in concrete fragment!");
373
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000374 // FIXME: This error shouldn't actually occur (the front end should emit
375 // multiple .align directives to enforce the semantics it wants), but is
376 // severe enough that we want to report it. How to handle this?
Daniel Dunbar432cd5f2010-03-25 02:00:02 +0000377 if (Count * AF.getValueSize() != FragmentSize)
Chris Lattner75361b62010-04-07 22:58:41 +0000378 report_fatal_error("undefined .align directive, value size '" +
Daniel Dunbar7eb85192009-10-16 01:58:15 +0000379 Twine(AF.getValueSize()) +
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000380 "' is not a divisor of padding size '" +
Daniel Dunbar432cd5f2010-03-25 02:00:02 +0000381 Twine(FragmentSize) + "'");
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000382
Kevin Enderby6e720482010-02-23 18:26:34 +0000383 // See if we are aligning with nops, and if so do that first to try to fill
384 // the Count bytes. Then if that did not fill any bytes or there are any
385 // bytes left to fill use the the Value and ValueSize to fill the rest.
Daniel Dunbar8f9b80e2010-03-23 02:36:58 +0000386 // If we are aligning with nops, ask that target to emit the right data.
Daniel Dunbar1c154132010-05-12 22:56:23 +0000387 if (AF.hasEmitNops()) {
Daniel Dunbar8f9b80e2010-03-23 02:36:58 +0000388 if (!Asm.getBackend().WriteNopData(Count, OW))
Chris Lattner75361b62010-04-07 22:58:41 +0000389 report_fatal_error("unable to write nop sequence of " +
Daniel Dunbar8f9b80e2010-03-23 02:36:58 +0000390 Twine(Count) + " bytes");
391 break;
Kevin Enderby6e720482010-02-23 18:26:34 +0000392 }
393
Daniel Dunbar8f9b80e2010-03-23 02:36:58 +0000394 // Otherwise, write out in multiples of the value size.
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000395 for (uint64_t i = 0; i != Count; ++i) {
396 switch (AF.getValueSize()) {
397 default:
398 assert(0 && "Invalid size!");
Daniel Dunbarbdd92812010-03-19 09:28:55 +0000399 case 1: OW->Write8 (uint8_t (AF.getValue())); break;
400 case 2: OW->Write16(uint16_t(AF.getValue())); break;
401 case 4: OW->Write32(uint32_t(AF.getValue())); break;
402 case 8: OW->Write64(uint64_t(AF.getValue())); break;
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000403 }
404 }
405 break;
406 }
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000407
Daniel Dunbar3a30b822010-02-13 09:28:15 +0000408 case MCFragment::FT_Data: {
Daniel Dunbar3f4dcd92010-03-22 23:16:48 +0000409 MCDataFragment &DF = cast<MCDataFragment>(F);
Daniel Dunbar432cd5f2010-03-25 02:00:02 +0000410 assert(FragmentSize == DF.getContents().size() && "Invalid size!");
Daniel Dunbar3f4dcd92010-03-22 23:16:48 +0000411 OW->WriteBytes(DF.getContents().str());
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000412 break;
Daniel Dunbar3a30b822010-02-13 09:28:15 +0000413 }
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000414
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000415 case MCFragment::FT_Fill: {
416 MCFillFragment &FF = cast<MCFillFragment>(F);
Daniel Dunbare2fee5b2010-05-12 22:51:35 +0000417
418 assert(FF.getValueSize() && "Invalid virtual align in concrete fragment!");
419
Daniel Dunbar3153fec2010-05-12 22:51:32 +0000420 for (uint64_t i = 0, e = FF.getSize() / FF.getValueSize(); i != e; ++i) {
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000421 switch (FF.getValueSize()) {
422 default:
423 assert(0 && "Invalid size!");
Daniel Dunbarbdd92812010-03-19 09:28:55 +0000424 case 1: OW->Write8 (uint8_t (FF.getValue())); break;
425 case 2: OW->Write16(uint16_t(FF.getValue())); break;
426 case 4: OW->Write32(uint32_t(FF.getValue())); break;
427 case 8: OW->Write64(uint64_t(FF.getValue())); break;
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000428 }
429 }
430 break;
431 }
Daniel Dunbar7eb85192009-10-16 01:58:15 +0000432
Rafael Espindola179821a2010-12-06 19:08:48 +0000433 case MCFragment::FT_Inst: {
434 MCInstFragment &IF = cast<MCInstFragment>(F);
435 OW->WriteBytes(StringRef(IF.getCode().begin(), IF.getCode().size()));
Daniel Dunbar3f4dcd92010-03-22 23:16:48 +0000436 break;
Rafael Espindola179821a2010-12-06 19:08:48 +0000437 }
Daniel Dunbar3f4dcd92010-03-22 23:16:48 +0000438
Rafael Espindola3ff57092010-11-02 17:22:24 +0000439 case MCFragment::FT_LEB: {
440 MCLEBFragment &LF = cast<MCLEBFragment>(F);
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000441 OW->WriteBytes(LF.getContents().str());
Rafael Espindola3ff57092010-11-02 17:22:24 +0000442 break;
443 }
444
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000445 case MCFragment::FT_Org: {
446 MCOrgFragment &OF = cast<MCOrgFragment>(F);
447
Daniel Dunbar432cd5f2010-03-25 02:00:02 +0000448 for (uint64_t i = 0, e = FragmentSize; i != e; ++i)
Daniel Dunbarbdd92812010-03-19 09:28:55 +0000449 OW->Write8(uint8_t(OF.getValue()));
Daniel Dunbard6f761e2009-08-21 23:07:38 +0000450
451 break;
452 }
Kevin Enderbyc0957932010-09-30 16:52:03 +0000453
454 case MCFragment::FT_Dwarf: {
455 const MCDwarfLineAddrFragment &OF = cast<MCDwarfLineAddrFragment>(F);
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000456 OW->WriteBytes(OF.getContents().str());
Kevin Enderbyc0957932010-09-30 16:52:03 +0000457 break;
458 }
Rafael Espindola245a1e22010-12-28 05:39:27 +0000459 case MCFragment::FT_DwarfFrame: {
460 const MCDwarfCallFrameFragment &CF = cast<MCDwarfCallFrameFragment>(F);
461 OW->WriteBytes(CF.getContents().str());
462 break;
463 }
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000464 }
465
Daniel Dunbar432cd5f2010-03-25 02:00:02 +0000466 assert(OW->getStream().tell() - Start == FragmentSize);
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000467}
468
Daniel Dunbar53b23382010-03-19 09:28:59 +0000469void MCAssembler::WriteSectionData(const MCSectionData *SD,
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000470 const MCAsmLayout &Layout) const {
Daniel Dunbard5a8e982009-08-28 05:49:21 +0000471 // Ignore virtual sections.
Rafael Espindolaf2dc4aa2010-11-17 20:03:54 +0000472 if (SD->getSection().isVirtualSection()) {
Daniel Dunbar054be922010-05-13 03:50:50 +0000473 assert(Layout.getSectionFileSize(SD) == 0 && "Invalid size for section!");
Daniel Dunbare2fee5b2010-05-12 22:51:35 +0000474
475 // Check that contents are only things legal inside a virtual section.
476 for (MCSectionData::const_iterator it = SD->begin(),
477 ie = SD->end(); it != ie; ++it) {
478 switch (it->getKind()) {
479 default:
480 assert(0 && "Invalid fragment in virtual section!");
Daniel Dunbarc983b202010-08-18 18:22:37 +0000481 case MCFragment::FT_Data: {
482 // Check that we aren't trying to write a non-zero contents (or fixups)
483 // into a virtual section. This is to support clients which use standard
484 // directives to fill the contents of virtual sections.
485 MCDataFragment &DF = cast<MCDataFragment>(*it);
486 assert(DF.fixup_begin() == DF.fixup_end() &&
487 "Cannot have fixups in virtual section!");
488 for (unsigned i = 0, e = DF.getContents().size(); i != e; ++i)
489 assert(DF.getContents()[i] == 0 &&
490 "Invalid data value for virtual section!");
491 break;
492 }
Daniel Dunbare2fee5b2010-05-12 22:51:35 +0000493 case MCFragment::FT_Align:
Daniel Dunbarc983b202010-08-18 18:22:37 +0000494 // Check that we aren't trying to write a non-zero value into a virtual
495 // section.
496 assert((!cast<MCAlignFragment>(it)->getValueSize() ||
497 !cast<MCAlignFragment>(it)->getValue()) &&
Daniel Dunbare2fee5b2010-05-12 22:51:35 +0000498 "Invalid align in virtual section!");
499 break;
500 case MCFragment::FT_Fill:
501 assert(!cast<MCFillFragment>(it)->getValueSize() &&
502 "Invalid fill in virtual section!");
503 break;
Daniel Dunbare2fee5b2010-05-12 22:51:35 +0000504 }
505 }
506
Daniel Dunbard5a8e982009-08-28 05:49:21 +0000507 return;
508 }
509
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000510 uint64_t Start = getWriter().getStream().tell();
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000511 (void) Start;
Daniel Dunbar7eb85192009-10-16 01:58:15 +0000512
Daniel Dunbar53b23382010-03-19 09:28:59 +0000513 for (MCSectionData::const_iterator it = SD->begin(),
514 ie = SD->end(); it != ie; ++it)
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000515 WriteFragmentData(*this, Layout, *it);
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000516
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000517 assert(getWriter().getStream().tell() - Start ==
518 Layout.getSectionAddressSize(SD));
Daniel Dunbar0705fbf2009-08-21 18:29:01 +0000519}
520
Rafael Espindola179821a2010-12-06 19:08:48 +0000521
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000522uint64_t MCAssembler::HandleFixup(const MCAsmLayout &Layout,
523 MCFragment &F,
524 const MCFixup &Fixup) {
Rafael Espindola179821a2010-12-06 19:08:48 +0000525 // Evaluate the fixup.
526 MCValue Target;
527 uint64_t FixedValue;
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000528 if (!EvaluateFixup(Layout, Fixup, &F, Target, FixedValue)) {
Rafael Espindola179821a2010-12-06 19:08:48 +0000529 // The fixup was unresolved, we need a relocation. Inform the object
530 // writer of the relocation, and give it an opportunity to adjust the
531 // fixup value if need be.
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000532 getWriter().RecordRelocation(*this, Layout, &F, Fixup, Target, FixedValue);
Rafael Espindola179821a2010-12-06 19:08:48 +0000533 }
534 return FixedValue;
535 }
536
Daniel Dunbarfeb7ba32010-12-17 02:45:41 +0000537void MCAssembler::Finish() {
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000538 DEBUG_WITH_TYPE("mc-dump", {
539 llvm::errs() << "assembler backend - pre-layout\n--\n";
540 dump(); });
541
Daniel Dunbar61066db2010-05-13 02:34:14 +0000542 // Create the layout object.
Daniel Dunbar8d39eb42010-03-22 20:35:35 +0000543 MCAsmLayout Layout(*this);
Daniel Dunbar61066db2010-05-13 02:34:14 +0000544
Daniel Dunbar337718e2010-05-14 00:37:14 +0000545 // Create dummy fragments and assign section ordinals.
Daniel Dunbar49ed9212010-05-13 08:43:37 +0000546 unsigned SectionIndex = 0;
Daniel Dunbar49ed9212010-05-13 08:43:37 +0000547 for (MCAssembler::iterator it = begin(), ie = end(); it != ie; ++it) {
548 // Create dummy fragments to eliminate any empty sections, this simplifies
549 // layout.
Duncan Sands6f74f692010-06-29 13:30:08 +0000550 if (it->getFragmentList().empty())
Rafael Espindolad80781b2010-09-15 21:48:40 +0000551 new MCDataFragment(it);
Daniel Dunbar49ed9212010-05-13 08:43:37 +0000552
553 it->setOrdinal(SectionIndex++);
Daniel Dunbar337718e2010-05-14 00:37:14 +0000554 }
Daniel Dunbar49ed9212010-05-13 08:43:37 +0000555
Daniel Dunbar337718e2010-05-14 00:37:14 +0000556 // Assign layout order indices to sections and fragments.
Daniel Dunbar337718e2010-05-14 00:37:14 +0000557 for (unsigned i = 0, e = Layout.getSectionOrder().size(); i != e; ++i) {
558 MCSectionData *SD = Layout.getSectionOrder()[i];
559 SD->setLayoutOrder(i);
560
Rafael Espindola4f4363a2010-12-07 23:32:26 +0000561 unsigned FragmentIndex = 0;
Daniel Dunbar337718e2010-05-14 00:37:14 +0000562 for (MCSectionData::iterator it2 = SD->begin(),
563 ie2 = SD->end(); it2 != ie2; ++it2)
564 it2->setLayoutOrder(FragmentIndex++);
Daniel Dunbar49ed9212010-05-13 08:43:37 +0000565 }
566
Daniel Dunbar61066db2010-05-13 02:34:14 +0000567 // Layout until everything fits.
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000568 while (LayoutOnce(Layout))
Daniel Dunbarf08fde42010-03-12 22:07:14 +0000569 continue;
570
571 DEBUG_WITH_TYPE("mc-dump", {
Daniel Dunbar3f4dcd92010-03-22 23:16:48 +0000572 llvm::errs() << "assembler backend - post-relaxation\n--\n";
573 dump(); });
574
575 // Finalize the layout, including fragment lowering.
576 FinishLayout(Layout);
577
578 DEBUG_WITH_TYPE("mc-dump", {
579 llvm::errs() << "assembler backend - final-layout\n--\n";
Daniel Dunbarf08fde42010-03-12 22:07:14 +0000580 dump(); });
581
Daniel Dunbarff547842010-03-23 23:47:14 +0000582 uint64_t StartOffset = OS.tell();
Reid Klecknerc96a82a2010-07-22 05:58:53 +0000583
Daniel Dunbarbacba992010-03-19 07:09:33 +0000584 // Allow the object writer a chance to perform post-layout binding (for
585 // example, to set the index fields in the symbol data).
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000586 getWriter().ExecutePostLayoutBinding(*this, Layout);
Daniel Dunbarbacba992010-03-19 07:09:33 +0000587
Daniel Dunbarb1e98942010-03-19 07:09:47 +0000588 // Evaluate and apply the fixups, generating relocation entries as necessary.
Daniel Dunbarb1e98942010-03-19 07:09:47 +0000589 for (MCAssembler::iterator it = begin(), ie = end(); it != ie; ++it) {
590 for (MCSectionData::iterator it2 = it->begin(),
591 ie2 = it->end(); it2 != ie2; ++it2) {
592 MCDataFragment *DF = dyn_cast<MCDataFragment>(it2);
Rafael Espindola179821a2010-12-06 19:08:48 +0000593 if (DF) {
594 for (MCDataFragment::fixup_iterator it3 = DF->fixup_begin(),
595 ie3 = DF->fixup_end(); it3 != ie3; ++it3) {
596 MCFixup &Fixup = *it3;
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000597 uint64_t FixedValue = HandleFixup(Layout, *DF, Fixup);
Rafael Espindola179821a2010-12-06 19:08:48 +0000598 getBackend().ApplyFixup(Fixup, DF->getContents().data(),
599 DF->getContents().size(), FixedValue);
Daniel Dunbarb1e98942010-03-19 07:09:47 +0000600 }
Rafael Espindola179821a2010-12-06 19:08:48 +0000601 }
602 MCInstFragment *IF = dyn_cast<MCInstFragment>(it2);
603 if (IF) {
604 for (MCInstFragment::fixup_iterator it3 = IF->fixup_begin(),
605 ie3 = IF->fixup_end(); it3 != ie3; ++it3) {
606 MCFixup &Fixup = *it3;
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000607 uint64_t FixedValue = HandleFixup(Layout, *IF, Fixup);
Rafael Espindola179821a2010-12-06 19:08:48 +0000608 getBackend().ApplyFixup(Fixup, IF->getCode().data(),
609 IF->getCode().size(), FixedValue);
610 }
Daniel Dunbarb1e98942010-03-19 07:09:47 +0000611 }
612 }
613 }
614
Daniel Dunbarbacba992010-03-19 07:09:33 +0000615 // Write the object file.
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000616 getWriter().WriteObject(*this, Layout);
Daniel Dunbarff547842010-03-23 23:47:14 +0000617
618 stats::ObjectBytes += OS.tell() - StartOffset;
Daniel Dunbarf08fde42010-03-12 22:07:14 +0000619}
620
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000621bool MCAssembler::FixupNeedsRelaxation(const MCFixup &Fixup,
Daniel Dunbar9d39e612010-03-22 21:49:41 +0000622 const MCFragment *DF,
Daniel Dunbar8d39eb42010-03-22 20:35:35 +0000623 const MCAsmLayout &Layout) const {
Daniel Dunbarac2884a2010-03-25 22:49:09 +0000624 if (getRelaxAll())
625 return true;
626
Daniel Dunbarf08fde42010-03-12 22:07:14 +0000627 // If we cannot resolve the fixup value, it requires relaxation.
628 MCValue Target;
629 uint64_t Value;
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000630 if (!EvaluateFixup(Layout, Fixup, DF, Target, Value))
Daniel Dunbarf08fde42010-03-12 22:07:14 +0000631 return true;
632
633 // Otherwise, relax if the value is too big for a (signed) i8.
Daniel Dunbar31e8e1d2010-05-04 00:33:07 +0000634 //
635 // FIXME: This is target dependent!
Daniel Dunbarf08fde42010-03-12 22:07:14 +0000636 return int64_t(Value) != int64_t(int8_t(Value));
637}
638
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000639bool MCAssembler::FragmentNeedsRelaxation(const MCInstFragment *IF,
Daniel Dunbard8036fb2010-03-23 05:09:03 +0000640 const MCAsmLayout &Layout) const {
641 // If this inst doesn't ever need relaxation, ignore it. This occurs when we
642 // are intentionally pushing out inst fragments, or because we relaxed a
643 // previous instruction to one that doesn't need relaxation.
Daniel Dunbar84882522010-05-26 17:45:29 +0000644 if (!getBackend().MayNeedRelaxation(IF->getInst()))
Daniel Dunbard8036fb2010-03-23 05:09:03 +0000645 return false;
646
647 for (MCInstFragment::const_fixup_iterator it = IF->fixup_begin(),
648 ie = IF->fixup_end(); it != ie; ++it)
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000649 if (FixupNeedsRelaxation(*it, IF, Layout))
Daniel Dunbard8036fb2010-03-23 05:09:03 +0000650 return true;
651
652 return false;
653}
654
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000655bool MCAssembler::RelaxInstruction(MCAsmLayout &Layout,
Rafael Espindola3ff57092010-11-02 17:22:24 +0000656 MCInstFragment &IF) {
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000657 if (!FragmentNeedsRelaxation(&IF, Layout))
Rafael Espindola3ff57092010-11-02 17:22:24 +0000658 return false;
659
660 ++stats::RelaxedInstructions;
661
662 // FIXME-PERF: We could immediately lower out instructions if we can tell
663 // they are fully resolved, to avoid retesting on later passes.
664
665 // Relax the fragment.
666
667 MCInst Relaxed;
668 getBackend().RelaxInstruction(IF.getInst(), Relaxed);
669
670 // Encode the new instruction.
671 //
672 // FIXME-PERF: If it matters, we could let the target do this. It can
673 // probably do so more efficiently in many cases.
674 SmallVector<MCFixup, 4> Fixups;
675 SmallString<256> Code;
676 raw_svector_ostream VecOS(Code);
677 getEmitter().EncodeInstruction(Relaxed, VecOS, Fixups);
678 VecOS.flush();
679
680 // Update the instruction fragment.
Rafael Espindola3ff57092010-11-02 17:22:24 +0000681 IF.setInst(Relaxed);
682 IF.getCode() = Code;
683 IF.getFixups().clear();
684 // FIXME: Eliminate copy.
685 for (unsigned i = 0, e = Fixups.size(); i != e; ++i)
686 IF.getFixups().push_back(Fixups[i]);
687
Rafael Espindola3ff57092010-11-02 17:22:24 +0000688 return true;
689}
690
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000691bool MCAssembler::RelaxLEB(MCAsmLayout &Layout, MCLEBFragment &LF) {
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000692 int64_t Value = 0;
693 uint64_t OldSize = LF.getContents().size();
Rafael Espindolad88cac02011-04-26 02:17:58 +0000694 bool IsAbs = LF.getValue().EvaluateAsAbsolute(Value, Layout);
695 (void)IsAbs;
696 assert(IsAbs);
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000697 SmallString<8> &Data = LF.getContents();
698 Data.clear();
699 raw_svector_ostream OSE(Data);
Rafael Espindola3ff57092010-11-02 17:22:24 +0000700 if (LF.isSigned())
701 MCObjectWriter::EncodeSLEB128(Value, OSE);
702 else
703 MCObjectWriter::EncodeULEB128(Value, OSE);
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000704 OSE.flush();
705 return OldSize != LF.getContents().size();
Rafael Espindola3ff57092010-11-02 17:22:24 +0000706}
707
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000708bool MCAssembler::RelaxDwarfLineAddr(MCAsmLayout &Layout,
Rafael Espindola187d8332010-11-07 02:07:12 +0000709 MCDwarfLineAddrFragment &DF) {
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000710 int64_t AddrDelta = 0;
711 uint64_t OldSize = DF.getContents().size();
Rafael Espindola835439a2010-12-22 22:04:28 +0000712 bool IsAbs = DF.getAddrDelta().EvaluateAsAbsolute(AddrDelta, Layout);
713 (void)IsAbs;
714 assert(IsAbs);
Rafael Espindola187d8332010-11-07 02:07:12 +0000715 int64_t LineDelta;
716 LineDelta = DF.getLineDelta();
Rafael Espindoladb74aea2010-12-04 21:58:52 +0000717 SmallString<8> &Data = DF.getContents();
718 Data.clear();
719 raw_svector_ostream OSE(Data);
720 MCDwarfLineAddr::Encode(LineDelta, AddrDelta, OSE);
721 OSE.flush();
722 return OldSize != Data.size();
Rafael Espindola187d8332010-11-07 02:07:12 +0000723}
724
Rafael Espindola245a1e22010-12-28 05:39:27 +0000725bool MCAssembler::RelaxDwarfCallFrameFragment(MCAsmLayout &Layout,
726 MCDwarfCallFrameFragment &DF) {
727 int64_t AddrDelta = 0;
728 uint64_t OldSize = DF.getContents().size();
729 bool IsAbs = DF.getAddrDelta().EvaluateAsAbsolute(AddrDelta, Layout);
730 (void)IsAbs;
731 assert(IsAbs);
732 SmallString<8> &Data = DF.getContents();
733 Data.clear();
734 raw_svector_ostream OSE(Data);
735 MCDwarfFrameEmitter::EncodeAdvanceLoc(AddrDelta, OSE);
736 OSE.flush();
737 return OldSize != Data.size();
738}
739
Rafael Espindola62b83b62010-12-21 04:22:09 +0000740bool MCAssembler::LayoutSectionOnce(MCAsmLayout &Layout,
741 MCSectionData &SD) {
742 MCFragment *FirstInvalidFragment = NULL;
743 // Scan for fragments that need relaxation.
744 for (MCSectionData::iterator it2 = SD.begin(),
745 ie2 = SD.end(); it2 != ie2; ++it2) {
746 // Check if this is an fragment that needs relaxation.
747 bool relaxedFrag = false;
748 switch(it2->getKind()) {
749 default:
750 break;
Rafael Espindola62b83b62010-12-21 04:22:09 +0000751 case MCFragment::FT_Inst:
752 relaxedFrag = RelaxInstruction(Layout, *cast<MCInstFragment>(it2));
753 break;
Rafael Espindola62b83b62010-12-21 04:22:09 +0000754 case MCFragment::FT_Dwarf:
755 relaxedFrag = RelaxDwarfLineAddr(Layout,
756 *cast<MCDwarfLineAddrFragment>(it2));
757 break;
Rafael Espindola245a1e22010-12-28 05:39:27 +0000758 case MCFragment::FT_DwarfFrame:
759 relaxedFrag =
760 RelaxDwarfCallFrameFragment(Layout,
761 *cast<MCDwarfCallFrameFragment>(it2));
762 break;
763 case MCFragment::FT_LEB:
764 relaxedFrag = RelaxLEB(Layout, *cast<MCLEBFragment>(it2));
765 break;
Rafael Espindola62b83b62010-12-21 04:22:09 +0000766 }
767 // Update the layout, and remember that we relaxed.
768 if (relaxedFrag && !FirstInvalidFragment)
769 FirstInvalidFragment = it2;
770 }
771 if (FirstInvalidFragment) {
772 Layout.Invalidate(FirstInvalidFragment);
773 return true;
774 }
775 return false;
776}
777
Daniel Dunbar5d2477c2010-12-17 02:45:59 +0000778bool MCAssembler::LayoutOnce(MCAsmLayout &Layout) {
Daniel Dunbarff547842010-03-23 23:47:14 +0000779 ++stats::RelaxationSteps;
780
Daniel Dunbar0cc8bd42010-03-25 19:35:56 +0000781 bool WasRelaxed = false;
Daniel Dunbarf08fde42010-03-12 22:07:14 +0000782 for (iterator it = begin(), ie = end(); it != ie; ++it) {
783 MCSectionData &SD = *it;
Rafael Espindola62b83b62010-12-21 04:22:09 +0000784 while(LayoutSectionOnce(Layout, SD))
785 WasRelaxed = true;
Daniel Dunbarf08fde42010-03-12 22:07:14 +0000786 }
787
Daniel Dunbar0cc8bd42010-03-25 19:35:56 +0000788 return WasRelaxed;
Daniel Dunbarfb4a6b32009-08-21 09:11:24 +0000789}
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000790
Daniel Dunbar3f4dcd92010-03-22 23:16:48 +0000791void MCAssembler::FinishLayout(MCAsmLayout &Layout) {
Rafael Espindolab4172fa2010-12-04 22:47:22 +0000792 // The layout is done. Mark every fragment as valid.
Rafael Espindola4f4363a2010-12-07 23:32:26 +0000793 for (unsigned int i = 0, n = Layout.getSectionOrder().size(); i != n; ++i) {
794 Layout.getFragmentOffset(&*Layout.getSectionOrder()[i]->rbegin());
795 }
Daniel Dunbar3f4dcd92010-03-22 23:16:48 +0000796}
797
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000798// Debugging methods
799
800namespace llvm {
801
Daniel Dunbarc90e30a2010-05-26 15:18:56 +0000802raw_ostream &operator<<(raw_ostream &OS, const MCFixup &AF) {
803 OS << "<MCFixup" << " Offset:" << AF.getOffset()
Daniel Dunbar482ad802010-05-26 15:18:31 +0000804 << " Value:" << *AF.getValue()
805 << " Kind:" << AF.getKind() << ">";
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000806 return OS;
807}
808
809}
810
811void MCFragment::dump() {
812 raw_ostream &OS = llvm::errs();
813
Daniel Dunbare614e392010-05-26 06:50:57 +0000814 OS << "<";
815 switch (getKind()) {
816 case MCFragment::FT_Align: OS << "MCAlignFragment"; break;
817 case MCFragment::FT_Data: OS << "MCDataFragment"; break;
818 case MCFragment::FT_Fill: OS << "MCFillFragment"; break;
819 case MCFragment::FT_Inst: OS << "MCInstFragment"; break;
820 case MCFragment::FT_Org: OS << "MCOrgFragment"; break;
Kevin Enderbyc0957932010-09-30 16:52:03 +0000821 case MCFragment::FT_Dwarf: OS << "MCDwarfFragment"; break;
Rafael Espindola245a1e22010-12-28 05:39:27 +0000822 case MCFragment::FT_DwarfFrame: OS << "MCDwarfCallFrameFragment"; break;
Rafael Espindola3ff57092010-11-02 17:22:24 +0000823 case MCFragment::FT_LEB: OS << "MCLEBFragment"; break;
Daniel Dunbare614e392010-05-26 06:50:57 +0000824 }
825
Daniel Dunbar337718e2010-05-14 00:37:14 +0000826 OS << "<MCFragment " << (void*) this << " LayoutOrder:" << LayoutOrder
Rafael Espindola2bf6afc2010-12-15 08:45:53 +0000827 << " Offset:" << Offset << ">";
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000828
Daniel Dunbare614e392010-05-26 06:50:57 +0000829 switch (getKind()) {
830 case MCFragment::FT_Align: {
831 const MCAlignFragment *AF = cast<MCAlignFragment>(this);
832 if (AF->hasEmitNops())
833 OS << " (emit nops)";
Daniel Dunbare614e392010-05-26 06:50:57 +0000834 OS << "\n ";
835 OS << " Alignment:" << AF->getAlignment()
836 << " Value:" << AF->getValue() << " ValueSize:" << AF->getValueSize()
837 << " MaxBytesToEmit:" << AF->getMaxBytesToEmit() << ">";
838 break;
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000839 }
Daniel Dunbare614e392010-05-26 06:50:57 +0000840 case MCFragment::FT_Data: {
841 const MCDataFragment *DF = cast<MCDataFragment>(this);
842 OS << "\n ";
843 OS << " Contents:[";
844 const SmallVectorImpl<char> &Contents = DF->getContents();
845 for (unsigned i = 0, e = Contents.size(); i != e; ++i) {
846 if (i) OS << ",";
847 OS << hexdigit((Contents[i] >> 4) & 0xF) << hexdigit(Contents[i] & 0xF);
Daniel Dunbar0bcf0742010-02-13 09:28:43 +0000848 }
Daniel Dunbare614e392010-05-26 06:50:57 +0000849 OS << "] (" << Contents.size() << " bytes)";
850
851 if (!DF->getFixups().empty()) {
852 OS << ",\n ";
853 OS << " Fixups:[";
854 for (MCDataFragment::const_fixup_iterator it = DF->fixup_begin(),
855 ie = DF->fixup_end(); it != ie; ++it) {
856 if (it != DF->fixup_begin()) OS << ",\n ";
857 OS << *it;
858 }
859 OS << "]";
860 }
861 break;
Daniel Dunbar0bcf0742010-02-13 09:28:43 +0000862 }
Daniel Dunbare614e392010-05-26 06:50:57 +0000863 case MCFragment::FT_Fill: {
864 const MCFillFragment *FF = cast<MCFillFragment>(this);
865 OS << " Value:" << FF->getValue() << " ValueSize:" << FF->getValueSize()
866 << " Size:" << FF->getSize();
867 break;
868 }
869 case MCFragment::FT_Inst: {
870 const MCInstFragment *IF = cast<MCInstFragment>(this);
871 OS << "\n ";
872 OS << " Inst:";
873 IF->getInst().dump_pretty(OS);
874 break;
875 }
876 case MCFragment::FT_Org: {
877 const MCOrgFragment *OF = cast<MCOrgFragment>(this);
878 OS << "\n ";
879 OS << " Offset:" << OF->getOffset() << " Value:" << OF->getValue();
880 break;
881 }
Kevin Enderbyc0957932010-09-30 16:52:03 +0000882 case MCFragment::FT_Dwarf: {
883 const MCDwarfLineAddrFragment *OF = cast<MCDwarfLineAddrFragment>(this);
884 OS << "\n ";
885 OS << " AddrDelta:" << OF->getAddrDelta()
886 << " LineDelta:" << OF->getLineDelta();
887 break;
888 }
Rafael Espindola245a1e22010-12-28 05:39:27 +0000889 case MCFragment::FT_DwarfFrame: {
890 const MCDwarfCallFrameFragment *CF = cast<MCDwarfCallFrameFragment>(this);
891 OS << "\n ";
892 OS << " AddrDelta:" << CF->getAddrDelta();
893 break;
894 }
Rafael Espindola3ff57092010-11-02 17:22:24 +0000895 case MCFragment::FT_LEB: {
896 const MCLEBFragment *LF = cast<MCLEBFragment>(this);
897 OS << "\n ";
898 OS << " Value:" << LF->getValue() << " Signed:" << LF->isSigned();
899 break;
900 }
Daniel Dunbare614e392010-05-26 06:50:57 +0000901 }
Daniel Dunbar0bcf0742010-02-13 09:28:43 +0000902 OS << ">";
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000903}
904
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000905void MCSectionData::dump() {
906 raw_ostream &OS = llvm::errs();
907
908 OS << "<MCSectionData";
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000909 OS << " Alignment:" << getAlignment() << " Fragments:[\n ";
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000910 for (iterator it = begin(), ie = end(); it != ie; ++it) {
911 if (it != begin()) OS << ",\n ";
912 it->dump();
913 }
914 OS << "]>";
915}
916
917void MCSymbolData::dump() {
918 raw_ostream &OS = llvm::errs();
919
920 OS << "<MCSymbolData Symbol:" << getSymbol()
921 << " Fragment:" << getFragment() << " Offset:" << getOffset()
922 << " Flags:" << getFlags() << " Index:" << getIndex();
923 if (isCommon())
924 OS << " (common, size:" << getCommonSize()
925 << " align: " << getCommonAlignment() << ")";
926 if (isExternal())
927 OS << " (external)";
928 if (isPrivateExtern())
929 OS << " (private extern)";
930 OS << ">";
931}
932
933void MCAssembler::dump() {
934 raw_ostream &OS = llvm::errs();
935
936 OS << "<MCAssembler\n";
Daniel Dunbar45aefff2010-03-09 01:12:23 +0000937 OS << " Sections:[\n ";
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000938 for (iterator it = begin(), ie = end(); it != ie; ++it) {
939 if (it != begin()) OS << ",\n ";
940 it->dump();
941 }
942 OS << "],\n";
943 OS << " Symbols:[";
944
945 for (symbol_iterator it = symbol_begin(), ie = symbol_end(); it != ie; ++it) {
Daniel Dunbar45aefff2010-03-09 01:12:23 +0000946 if (it != symbol_begin()) OS << ",\n ";
Daniel Dunbarb7c3a4b2010-02-13 09:28:03 +0000947 it->dump();
948 }
949 OS << "]>\n";
950}