1 //===-- AArch64MCExpr.cpp - AArch64 specific MC expression classes --------===//
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 #define DEBUG_TYPE "aarch64mcexpr"
11 #include "AArch64MCExpr.h"
12 #include "llvm/MC/MCContext.h"
13 #include "llvm/MC/MCAssembler.h"
14 #include "llvm/MC/MCELF.h"
15 #include "llvm/Object/ELF.h"
20 AArch64MCExpr::Create(VariantKind Kind, const MCExpr *Expr,
22 return new (Ctx) AArch64MCExpr(Kind, Expr);
25 void AArch64MCExpr::PrintImpl(raw_ostream &OS) const {
27 default: llvm_unreachable("Invalid kind!");
28 case VK_AARCH64_GOT: OS << ":got:"; break;
29 case VK_AARCH64_GOT_LO12: OS << ":got_lo12:"; break;
30 case VK_AARCH64_LO12: OS << ":lo12:"; break;
31 case VK_AARCH64_ABS_G0: OS << ":abs_g0:"; break;
32 case VK_AARCH64_ABS_G0_NC: OS << ":abs_g0_nc:"; break;
33 case VK_AARCH64_ABS_G1: OS << ":abs_g1:"; break;
34 case VK_AARCH64_ABS_G1_NC: OS << ":abs_g1_nc:"; break;
35 case VK_AARCH64_ABS_G2: OS << ":abs_g2:"; break;
36 case VK_AARCH64_ABS_G2_NC: OS << ":abs_g2_nc:"; break;
37 case VK_AARCH64_ABS_G3: OS << ":abs_g3:"; break;
38 case VK_AARCH64_SABS_G0: OS << ":abs_g0_s:"; break;
39 case VK_AARCH64_SABS_G1: OS << ":abs_g1_s:"; break;
40 case VK_AARCH64_SABS_G2: OS << ":abs_g2_s:"; break;
41 case VK_AARCH64_DTPREL_G2: OS << ":dtprel_g2:"; break;
42 case VK_AARCH64_DTPREL_G1: OS << ":dtprel_g1:"; break;
43 case VK_AARCH64_DTPREL_G1_NC: OS << ":dtprel_g1_nc:"; break;
44 case VK_AARCH64_DTPREL_G0: OS << ":dtprel_g0:"; break;
45 case VK_AARCH64_DTPREL_G0_NC: OS << ":dtprel_g0_nc:"; break;
46 case VK_AARCH64_DTPREL_HI12: OS << ":dtprel_hi12:"; break;
47 case VK_AARCH64_DTPREL_LO12: OS << ":dtprel_lo12:"; break;
48 case VK_AARCH64_DTPREL_LO12_NC: OS << ":dtprel_lo12_nc:"; break;
49 case VK_AARCH64_GOTTPREL_G1: OS << ":gottprel_g1:"; break;
50 case VK_AARCH64_GOTTPREL_G0_NC: OS << ":gottprel_g0_nc:"; break;
51 case VK_AARCH64_GOTTPREL: OS << ":gottprel:"; break;
52 case VK_AARCH64_GOTTPREL_LO12: OS << ":gottprel_lo12:"; break;
53 case VK_AARCH64_TPREL_G2: OS << ":tprel_g2:"; break;
54 case VK_AARCH64_TPREL_G1: OS << ":tprel_g1:"; break;
55 case VK_AARCH64_TPREL_G1_NC: OS << ":tprel_g1_nc:"; break;
56 case VK_AARCH64_TPREL_G0: OS << ":tprel_g0:"; break;
57 case VK_AARCH64_TPREL_G0_NC: OS << ":tprel_g0_nc:"; break;
58 case VK_AARCH64_TPREL_HI12: OS << ":tprel_hi12:"; break;
59 case VK_AARCH64_TPREL_LO12: OS << ":tprel_lo12:"; break;
60 case VK_AARCH64_TPREL_LO12_NC: OS << ":tprel_lo12_nc:"; break;
61 case VK_AARCH64_TLSDESC: OS << ":tlsdesc:"; break;
62 case VK_AARCH64_TLSDESC_LO12: OS << ":tlsdesc_lo12:"; break;
66 const MCExpr *Expr = getSubExpr();
67 if (Expr->getKind() != MCExpr::SymbolRef)
70 if (Expr->getKind() != MCExpr::SymbolRef)
75 AArch64MCExpr::EvaluateAsRelocatableImpl(MCValue &Res,
76 const MCAsmLayout *Layout) const {
77 return getSubExpr()->EvaluateAsRelocatable(Res, *Layout);
80 static void fixELFSymbolsInTLSFixupsImpl(const MCExpr *Expr, MCAssembler &Asm) {
81 switch (Expr->getKind()) {
83 llvm_unreachable("Can't handle nested target expression");
85 case MCExpr::Constant:
88 case MCExpr::Binary: {
89 const MCBinaryExpr *BE = cast<MCBinaryExpr>(Expr);
90 fixELFSymbolsInTLSFixupsImpl(BE->getLHS(), Asm);
91 fixELFSymbolsInTLSFixupsImpl(BE->getRHS(), Asm);
95 case MCExpr::SymbolRef: {
96 // We're known to be under a TLS fixup, so any symbol should be
97 // modified. There should be only one.
98 const MCSymbolRefExpr &SymRef = *cast<MCSymbolRefExpr>(Expr);
99 MCSymbolData &SD = Asm.getOrCreateSymbolData(SymRef.getSymbol());
100 MCELF::SetType(SD, ELF::STT_TLS);
105 fixELFSymbolsInTLSFixupsImpl(cast<MCUnaryExpr>(Expr)->getSubExpr(), Asm);
110 void AArch64MCExpr::fixELFSymbolsInTLSFixups(MCAssembler &Asm) const {
114 case VK_AARCH64_DTPREL_G2:
115 case VK_AARCH64_DTPREL_G1:
116 case VK_AARCH64_DTPREL_G1_NC:
117 case VK_AARCH64_DTPREL_G0:
118 case VK_AARCH64_DTPREL_G0_NC:
119 case VK_AARCH64_DTPREL_HI12:
120 case VK_AARCH64_DTPREL_LO12:
121 case VK_AARCH64_DTPREL_LO12_NC:
122 case VK_AARCH64_GOTTPREL_G1:
123 case VK_AARCH64_GOTTPREL_G0_NC:
124 case VK_AARCH64_GOTTPREL:
125 case VK_AARCH64_GOTTPREL_LO12:
126 case VK_AARCH64_TPREL_G2:
127 case VK_AARCH64_TPREL_G1:
128 case VK_AARCH64_TPREL_G1_NC:
129 case VK_AARCH64_TPREL_G0:
130 case VK_AARCH64_TPREL_G0_NC:
131 case VK_AARCH64_TPREL_HI12:
132 case VK_AARCH64_TPREL_LO12:
133 case VK_AARCH64_TPREL_LO12_NC:
134 case VK_AARCH64_TLSDESC:
135 case VK_AARCH64_TLSDESC_LO12:
139 fixELFSymbolsInTLSFixupsImpl(getSubExpr(), Asm);
142 // FIXME: This basically copies MCObjectStreamer::AddValueSymbols. Perhaps
143 // that method should be made public?
144 // FIXME: really do above: now that two backends are using it.
145 static void AddValueSymbolsImpl(const MCExpr *Value, MCAssembler *Asm) {
146 switch (Value->getKind()) {
148 llvm_unreachable("Can't handle nested target expr!");
151 case MCExpr::Constant:
154 case MCExpr::Binary: {
155 const MCBinaryExpr *BE = cast<MCBinaryExpr>(Value);
156 AddValueSymbolsImpl(BE->getLHS(), Asm);
157 AddValueSymbolsImpl(BE->getRHS(), Asm);
161 case MCExpr::SymbolRef:
162 Asm->getOrCreateSymbolData(cast<MCSymbolRefExpr>(Value)->getSymbol());
166 AddValueSymbolsImpl(cast<MCUnaryExpr>(Value)->getSubExpr(), Asm);
171 void AArch64MCExpr::AddValueSymbols(MCAssembler *Asm) const {
172 AddValueSymbolsImpl(getSubExpr(), Asm);