1 //===-- X86MachObjectWriter.cpp - X86 Mach-O Writer -----------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "MCTargetDesc/X86MCTargetDesc.h"
11 #include "MCTargetDesc/X86FixupKinds.h"
12 #include "llvm/ADT/Twine.h"
13 #include "llvm/MC/MCAsmLayout.h"
14 #include "llvm/MC/MCAssembler.h"
15 #include "llvm/MC/MCContext.h"
16 #include "llvm/MC/MCMachObjectWriter.h"
17 #include "llvm/MC/MCSectionMachO.h"
18 #include "llvm/MC/MCValue.h"
19 #include "llvm/Support/ErrorHandling.h"
20 #include "llvm/Support/Format.h"
21 #include "llvm/Support/MachO.h"
26 class X86MachObjectWriter : public MCMachObjectTargetWriter {
27 bool RecordScatteredRelocation(MachObjectWriter *Writer,
28 const MCAssembler &Asm,
29 const MCAsmLayout &Layout,
30 const MCFragment *Fragment,
34 uint64_t &FixedValue);
35 void RecordTLVPRelocation(MachObjectWriter *Writer,
36 const MCAssembler &Asm,
37 const MCAsmLayout &Layout,
38 const MCFragment *Fragment,
41 uint64_t &FixedValue);
43 void RecordX86Relocation(MachObjectWriter *Writer,
44 const MCAssembler &Asm,
45 const MCAsmLayout &Layout,
46 const MCFragment *Fragment,
49 uint64_t &FixedValue);
50 void RecordX86_64Relocation(MachObjectWriter *Writer,
51 const MCAssembler &Asm,
52 const MCAsmLayout &Layout,
53 const MCFragment *Fragment,
56 uint64_t &FixedValue);
58 X86MachObjectWriter(bool Is64Bit, uint32_t CPUType,
60 : MCMachObjectTargetWriter(Is64Bit, CPUType, CPUSubtype,
61 /*UseAggressiveSymbolFolding=*/Is64Bit) {}
63 void RecordRelocation(MachObjectWriter *Writer,
64 const MCAssembler &Asm, const MCAsmLayout &Layout,
65 const MCFragment *Fragment, const MCFixup &Fixup,
66 MCValue Target, uint64_t &FixedValue) {
67 if (Writer->is64Bit())
68 RecordX86_64Relocation(Writer, Asm, Layout, Fragment, Fixup, Target,
71 RecordX86Relocation(Writer, Asm, Layout, Fragment, Fixup, Target,
77 static bool isFixupKindRIPRel(unsigned Kind) {
78 return Kind == X86::reloc_riprel_4byte ||
79 Kind == X86::reloc_riprel_4byte_movq_load;
82 static unsigned getFixupKindLog2Size(unsigned Kind) {
85 llvm_unreachable("invalid fixup kind!");
87 case FK_Data_1: return 0;
89 case FK_Data_2: return 1;
91 // FIXME: Remove these!!!
92 case X86::reloc_riprel_4byte:
93 case X86::reloc_riprel_4byte_movq_load:
94 case X86::reloc_signed_4byte:
95 case FK_Data_4: return 2;
96 case FK_Data_8: return 3;
100 void X86MachObjectWriter::RecordX86_64Relocation(MachObjectWriter *Writer,
101 const MCAssembler &Asm,
102 const MCAsmLayout &Layout,
103 const MCFragment *Fragment,
104 const MCFixup &Fixup,
106 uint64_t &FixedValue) {
107 unsigned IsPCRel = Writer->isFixupKindPCRel(Asm, Fixup.getKind());
108 unsigned IsRIPRel = isFixupKindRIPRel(Fixup.getKind());
109 unsigned Log2Size = getFixupKindLog2Size(Fixup.getKind());
112 uint32_t FixupOffset =
113 Layout.getFragmentOffset(Fragment) + Fixup.getOffset();
114 uint32_t FixupAddress =
115 Writer->getFragmentAddress(Fragment, Layout) + Fixup.getOffset();
118 unsigned IsExtern = 0;
121 Value = Target.getConstant();
124 // Compensate for the relocation offset, Darwin x86_64 relocations only have
125 // the addend and appear to have attempted to define it to be the actual
126 // expression addend without the PCrel bias. However, instructions with data
127 // following the relocation are not accommodated for (see comment below
128 // regarding SIGNED{1,2,4}), so it isn't exactly that either.
129 Value += 1LL << Log2Size;
132 if (Target.isAbsolute()) { // constant
133 // SymbolNum of 0 indicates the absolute section.
134 Type = MachO::X86_64_RELOC_UNSIGNED;
137 // FIXME: I believe this is broken, I don't think the linker can understand
138 // it. I think it would require a local relocation, but I'm not sure if that
139 // would work either. The official way to get an absolute PCrel relocation
140 // is to use an absolute symbol (which we don't support yet).
143 Type = MachO::X86_64_RELOC_BRANCH;
145 } else if (Target.getSymB()) { // A - B + constant
146 const MCSymbol *A = &Target.getSymA()->getSymbol();
147 MCSymbolData &A_SD = Asm.getSymbolData(*A);
148 const MCSymbolData *A_Base = Asm.getAtom(&A_SD);
150 const MCSymbol *B = &Target.getSymB()->getSymbol();
151 MCSymbolData &B_SD = Asm.getSymbolData(*B);
152 const MCSymbolData *B_Base = Asm.getAtom(&B_SD);
154 // Neither symbol can be modified.
155 if (Target.getSymA()->getKind() != MCSymbolRefExpr::VK_None ||
156 Target.getSymB()->getKind() != MCSymbolRefExpr::VK_None)
157 report_fatal_error("unsupported relocation of modified symbol", false);
159 // We don't support PCrel relocations of differences. Darwin 'as' doesn't
160 // implement most of these correctly.
162 report_fatal_error("unsupported pc-relative relocation of difference",
165 // The support for the situation where one or both of the symbols would
166 // require a local relocation is handled just like if the symbols were
167 // external. This is certainly used in the case of debug sections where the
168 // section has only temporary symbols and thus the symbols don't have base
169 // symbols. This is encoded using the section ordinal and non-extern
170 // relocation entries.
172 // Darwin 'as' doesn't emit correct relocations for this (it ends up with a
173 // single SIGNED relocation); reject it for now. Except the case where both
174 // symbols don't have a base, equal but both NULL.
175 if (A_Base == B_Base && A_Base)
176 report_fatal_error("unsupported relocation with identical base", false);
178 // A subtraction expression where both symbols are undefined is a
179 // non-relocatable expression.
180 if (A->isUndefined() && B->isUndefined())
181 report_fatal_error("unsupported relocation with subtraction expression",
184 Value += Writer->getSymbolAddress(&A_SD, Layout) -
185 (A_Base == NULL ? 0 : Writer->getSymbolAddress(A_Base, Layout));
186 Value -= Writer->getSymbolAddress(&B_SD, Layout) -
187 (B_Base == NULL ? 0 : Writer->getSymbolAddress(B_Base, Layout));
190 Index = A_Base->getIndex();
194 Index = A_SD.getFragment()->getParent()->getOrdinal() + 1;
197 Type = MachO::X86_64_RELOC_UNSIGNED;
199 MachO::any_relocation_info MRE;
200 MRE.r_word0 = FixupOffset;
201 MRE.r_word1 = ((Index << 0) |
206 Writer->addRelocation(Fragment->getParent(), MRE);
209 Index = B_Base->getIndex();
213 Index = B_SD.getFragment()->getParent()->getOrdinal() + 1;
216 Type = MachO::X86_64_RELOC_SUBTRACTOR;
218 const MCSymbol *Symbol = &Target.getSymA()->getSymbol();
219 MCSymbolData &SD = Asm.getSymbolData(*Symbol);
220 const MCSymbolData *Base = Asm.getAtom(&SD);
222 // Relocations inside debug sections always use local relocations when
223 // possible. This seems to be done because the debugger doesn't fully
224 // understand x86_64 relocation entries, and expects to find values that
225 // have already been fixed up.
226 if (Symbol->isInSection()) {
227 const MCSectionMachO &Section = static_cast<const MCSectionMachO&>(
228 Fragment->getParent()->getSection());
229 if (Section.hasAttribute(MCSectionMachO::S_ATTR_DEBUG))
233 // x86_64 almost always uses external relocations, except when there is no
234 // symbol to use as a base address (a local symbol with no preceding
235 // non-local symbol).
237 Index = Base->getIndex();
240 // Add the local offset, if needed.
242 Value += Layout.getSymbolOffset(&SD) - Layout.getSymbolOffset(Base);
243 } else if (Symbol->isInSection() && !Symbol->isVariable()) {
244 // The index is the section ordinal (1-based).
245 Index = SD.getFragment()->getParent()->getOrdinal() + 1;
247 Value += Writer->getSymbolAddress(&SD, Layout);
250 Value -= FixupAddress + (1 << Log2Size);
251 } else if (Symbol->isVariable()) {
252 const MCExpr *Value = Symbol->getVariableValue();
254 bool isAbs = Value->EvaluateAsAbsolute(Res, Layout,
255 Writer->getSectionAddressMap());
260 report_fatal_error("unsupported relocation of variable '" +
261 Symbol->getName() + "'", false);
264 report_fatal_error("unsupported relocation of undefined symbol '" +
265 Symbol->getName() + "'", false);
268 MCSymbolRefExpr::VariantKind Modifier = Target.getSymA()->getKind();
271 if (Modifier == MCSymbolRefExpr::VK_GOTPCREL) {
272 // x86_64 distinguishes movq foo@GOTPCREL so that the linker can
273 // rewrite the movq to an leaq at link time if the symbol ends up in
274 // the same linkage unit.
275 if (unsigned(Fixup.getKind()) == X86::reloc_riprel_4byte_movq_load)
276 Type = MachO::X86_64_RELOC_GOT_LOAD;
278 Type = MachO::X86_64_RELOC_GOT;
279 } else if (Modifier == MCSymbolRefExpr::VK_TLVP) {
280 Type = MachO::X86_64_RELOC_TLV;
281 } else if (Modifier != MCSymbolRefExpr::VK_None) {
282 report_fatal_error("unsupported symbol modifier in relocation",
285 Type = MachO::X86_64_RELOC_SIGNED;
287 // The Darwin x86_64 relocation format has a problem where it cannot
288 // encode an address (L<foo> + <constant>) which is outside the atom
289 // containing L<foo>. Generally, this shouldn't occur but it does
290 // happen when we have a RIPrel instruction with data following the
291 // relocation entry (e.g., movb $012, L0(%rip)). Even with the PCrel
292 // adjustment Darwin x86_64 uses, the offset is still negative and the
293 // linker has no way to recognize this.
295 // To work around this, Darwin uses several special relocation types
296 // to indicate the offsets. However, the specification or
297 // implementation of these seems to also be incomplete; they should
298 // adjust the addend as well based on the actual encoded instruction
299 // (the additional bias), but instead appear to just look at the final
301 switch (-(Target.getConstant() + (1LL << Log2Size))) {
302 case 1: Type = MachO::X86_64_RELOC_SIGNED_1; break;
303 case 2: Type = MachO::X86_64_RELOC_SIGNED_2; break;
304 case 4: Type = MachO::X86_64_RELOC_SIGNED_4; break;
308 if (Modifier != MCSymbolRefExpr::VK_None)
309 report_fatal_error("unsupported symbol modifier in branch "
310 "relocation", false);
312 Type = MachO::X86_64_RELOC_BRANCH;
315 if (Modifier == MCSymbolRefExpr::VK_GOT) {
316 Type = MachO::X86_64_RELOC_GOT;
317 } else if (Modifier == MCSymbolRefExpr::VK_GOTPCREL) {
318 // GOTPCREL is allowed as a modifier on non-PCrel instructions, in which
319 // case all we do is set the PCrel bit in the relocation entry; this is
320 // used with exception handling, for example. The source is required to
321 // include any necessary offset directly.
322 Type = MachO::X86_64_RELOC_GOT;
324 } else if (Modifier == MCSymbolRefExpr::VK_TLVP) {
325 report_fatal_error("TLVP symbol modifier should have been rip-rel",
327 } else if (Modifier != MCSymbolRefExpr::VK_None)
328 report_fatal_error("unsupported symbol modifier in relocation", false);
330 Type = MachO::X86_64_RELOC_UNSIGNED;
331 unsigned Kind = Fixup.getKind();
332 if (Kind == X86::reloc_signed_4byte)
333 report_fatal_error("32-bit absolute addressing is not supported in "
334 "64-bit mode", false);
339 // x86_64 always writes custom values into the fixups.
342 // struct relocation_info (8 bytes)
343 MachO::any_relocation_info MRE;
344 MRE.r_word0 = FixupOffset;
345 MRE.r_word1 = ((Index << 0) |
350 Writer->addRelocation(Fragment->getParent(), MRE);
353 bool X86MachObjectWriter::RecordScatteredRelocation(MachObjectWriter *Writer,
354 const MCAssembler &Asm,
355 const MCAsmLayout &Layout,
356 const MCFragment *Fragment,
357 const MCFixup &Fixup,
360 uint64_t &FixedValue) {
361 uint32_t FixupOffset = Layout.getFragmentOffset(Fragment)+Fixup.getOffset();
362 unsigned IsPCRel = Writer->isFixupKindPCRel(Asm, Fixup.getKind());
363 unsigned Type = MachO::GENERIC_RELOC_VANILLA;
366 const MCSymbol *A = &Target.getSymA()->getSymbol();
367 MCSymbolData *A_SD = &Asm.getSymbolData(*A);
369 if (!A_SD->getFragment())
370 report_fatal_error("symbol '" + A->getName() +
371 "' can not be undefined in a subtraction expression",
374 uint32_t Value = Writer->getSymbolAddress(A_SD, Layout);
375 uint64_t SecAddr = Writer->getSectionAddress(A_SD->getFragment()->getParent());
376 FixedValue += SecAddr;
379 if (const MCSymbolRefExpr *B = Target.getSymB()) {
380 MCSymbolData *B_SD = &Asm.getSymbolData(B->getSymbol());
382 if (!B_SD->getFragment())
383 report_fatal_error("symbol '" + B->getSymbol().getName() +
384 "' can not be undefined in a subtraction expression",
387 // Select the appropriate difference relocation type.
389 // Note that there is no longer any semantic difference between these two
390 // relocation types from the linkers point of view, this is done solely for
391 // pedantic compatibility with 'as'.
392 Type = A_SD->isExternal() ? (unsigned)MachO::GENERIC_RELOC_SECTDIFF :
393 (unsigned)MachO::GENERIC_RELOC_LOCAL_SECTDIFF;
394 Value2 = Writer->getSymbolAddress(B_SD, Layout);
395 FixedValue -= Writer->getSectionAddress(B_SD->getFragment()->getParent());
398 // Relocations are written out in reverse order, so the PAIR comes first.
399 if (Type == MachO::GENERIC_RELOC_SECTDIFF ||
400 Type == MachO::GENERIC_RELOC_LOCAL_SECTDIFF) {
401 // If the offset is too large to fit in a scattered relocation,
402 // we're hosed. It's an unfortunate limitation of the MachO format.
403 if (FixupOffset > 0xffffff) {
405 format("0x%x", FixupOffset).print(Buffer, sizeof(Buffer));
406 Asm.getContext().FatalError(Fixup.getLoc(),
407 Twine("Section too large, can't encode "
408 "r_address (") + Buffer +
409 ") into 24 bits of scattered "
410 "relocation entry.");
411 llvm_unreachable("fatal error returned?!");
414 MachO::any_relocation_info MRE;
415 MRE.r_word0 = ((0 << 0) | // r_address
416 (MachO::GENERIC_RELOC_PAIR << 24) | // r_type
420 MRE.r_word1 = Value2;
421 Writer->addRelocation(Fragment->getParent(), MRE);
423 // If the offset is more than 24-bits, it won't fit in a scattered
424 // relocation offset field, so we fall back to using a non-scattered
425 // relocation. This is a bit risky, as if the offset reaches out of
426 // the block and the linker is doing scattered loading on this
427 // symbol, things can go badly.
429 // Required for 'as' compatibility.
430 if (FixupOffset > 0xffffff)
434 MachO::any_relocation_info MRE;
435 MRE.r_word0 = ((FixupOffset << 0) |
441 Writer->addRelocation(Fragment->getParent(), MRE);
445 void X86MachObjectWriter::RecordTLVPRelocation(MachObjectWriter *Writer,
446 const MCAssembler &Asm,
447 const MCAsmLayout &Layout,
448 const MCFragment *Fragment,
449 const MCFixup &Fixup,
451 uint64_t &FixedValue) {
452 assert(Target.getSymA()->getKind() == MCSymbolRefExpr::VK_TLVP &&
454 "Should only be called with a 32-bit TLVP relocation!");
456 unsigned Log2Size = getFixupKindLog2Size(Fixup.getKind());
457 uint32_t Value = Layout.getFragmentOffset(Fragment)+Fixup.getOffset();
458 unsigned IsPCRel = 0;
460 // Get the symbol data.
461 MCSymbolData *SD_A = &Asm.getSymbolData(Target.getSymA()->getSymbol());
462 unsigned Index = SD_A->getIndex();
464 // We're only going to have a second symbol in pic mode and it'll be a
465 // subtraction from the picbase. For 32-bit pic the addend is the difference
466 // between the picbase and the next address. For 32-bit static the addend is
468 if (Target.getSymB()) {
469 // If this is a subtraction then we're pcrel.
470 uint32_t FixupAddress =
471 Writer->getFragmentAddress(Fragment, Layout) + Fixup.getOffset();
472 MCSymbolData *SD_B = &Asm.getSymbolData(Target.getSymB()->getSymbol());
474 FixedValue = (FixupAddress - Writer->getSymbolAddress(SD_B, Layout) +
475 Target.getConstant());
476 FixedValue += 1ULL << Log2Size;
481 // struct relocation_info (8 bytes)
482 MachO::any_relocation_info MRE;
484 MRE.r_word1 = ((Index << 0) |
487 (1 << 27) | // r_extern
488 (MachO::GENERIC_RELOC_TLV << 28)); // r_type
489 Writer->addRelocation(Fragment->getParent(), MRE);
492 void X86MachObjectWriter::RecordX86Relocation(MachObjectWriter *Writer,
493 const MCAssembler &Asm,
494 const MCAsmLayout &Layout,
495 const MCFragment *Fragment,
496 const MCFixup &Fixup,
498 uint64_t &FixedValue) {
499 unsigned IsPCRel = Writer->isFixupKindPCRel(Asm, Fixup.getKind());
500 unsigned Log2Size = getFixupKindLog2Size(Fixup.getKind());
502 // If this is a 32-bit TLVP reloc it's handled a bit differently.
503 if (Target.getSymA() &&
504 Target.getSymA()->getKind() == MCSymbolRefExpr::VK_TLVP) {
505 RecordTLVPRelocation(Writer, Asm, Layout, Fragment, Fixup, Target,
510 // If this is a difference or a defined symbol plus an offset, then we need a
511 // scattered relocation entry. Differences always require scattered
513 if (Target.getSymB()) {
514 RecordScatteredRelocation(Writer, Asm, Layout, Fragment, Fixup,
515 Target, Log2Size, FixedValue);
519 // Get the symbol data, if any.
520 MCSymbolData *SD = 0;
521 if (Target.getSymA())
522 SD = &Asm.getSymbolData(Target.getSymA()->getSymbol());
524 // If this is an internal relocation with an offset, it also needs a scattered
526 uint32_t Offset = Target.getConstant();
528 Offset += 1 << Log2Size;
529 // Try to record the scattered relocation if needed. Fall back to non
530 // scattered if necessary (see comments in RecordScatteredRelocation()
532 if (Offset && SD && !Writer->doesSymbolRequireExternRelocation(SD) &&
533 RecordScatteredRelocation(Writer, Asm, Layout, Fragment, Fixup,
534 Target, Log2Size, FixedValue))
538 uint32_t FixupOffset = Layout.getFragmentOffset(Fragment)+Fixup.getOffset();
540 unsigned IsExtern = 0;
543 if (Target.isAbsolute()) { // constant
544 // SymbolNum of 0 indicates the absolute section.
546 // FIXME: Currently, these are never generated (see code below). I cannot
547 // find a case where they are actually emitted.
548 Type = MachO::GENERIC_RELOC_VANILLA;
550 // Resolve constant variables.
551 if (SD->getSymbol().isVariable()) {
553 if (SD->getSymbol().getVariableValue()->EvaluateAsAbsolute(
554 Res, Layout, Writer->getSectionAddressMap())) {
560 // Check whether we need an external or internal relocation.
561 if (Writer->doesSymbolRequireExternRelocation(SD)) {
563 Index = SD->getIndex();
564 // For external relocations, make sure to offset the fixup value to
565 // compensate for the addend of the symbol address, if it was
566 // undefined. This occurs with weak definitions, for example.
567 if (!SD->Symbol->isUndefined())
568 FixedValue -= Layout.getSymbolOffset(SD);
570 // The index is the section ordinal (1-based).
571 const MCSectionData &SymSD = Asm.getSectionData(
572 SD->getSymbol().getSection());
573 Index = SymSD.getOrdinal() + 1;
574 FixedValue += Writer->getSectionAddress(&SymSD);
577 FixedValue -= Writer->getSectionAddress(Fragment->getParent());
579 Type = MachO::GENERIC_RELOC_VANILLA;
582 // struct relocation_info (8 bytes)
583 MachO::any_relocation_info MRE;
584 MRE.r_word0 = FixupOffset;
585 MRE.r_word1 = ((Index << 0) |
590 Writer->addRelocation(Fragment->getParent(), MRE);
593 MCObjectWriter *llvm::createX86MachObjectWriter(raw_ostream &OS,
596 uint32_t CPUSubtype) {
597 return createMachObjectWriter(new X86MachObjectWriter(Is64Bit,
600 OS, /*IsLittleEndian=*/true);