1 //===-- AArch64MCExpr.cpp - AArch64 specific MC expression classes --------===//
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 //
10 // This file contains the implementation of the assembly expression modifiers
11 // accepted by the AArch64 architecture (e.g. ":lo12:", ":gottprel_g1:", ...).
12 //
13 //===----------------------------------------------------------------------===//
14
15 #define DEBUG_TYPE "aarch64mcexpr"
16 #include "AArch64MCExpr.h"
17 #include "llvm/MC/MCContext.h"
18 #include "llvm/MC/MCAssembler.h"
19 #include "llvm/MC/MCELF.h"
20 #include "llvm/Object/ELF.h"
21
22 using namespace llvm;
23
24 const AArch64MCExpr*
Create(VariantKind Kind,const MCExpr * Expr,MCContext & Ctx)25 AArch64MCExpr::Create(VariantKind Kind, const MCExpr *Expr,
26 MCContext &Ctx) {
27 return new (Ctx) AArch64MCExpr(Kind, Expr);
28 }
29
PrintImpl(raw_ostream & OS) const30 void AArch64MCExpr::PrintImpl(raw_ostream &OS) const {
31 switch (Kind) {
32 default: llvm_unreachable("Invalid kind!");
33 case VK_AARCH64_GOT: OS << ":got:"; break;
34 case VK_AARCH64_GOT_LO12: OS << ":got_lo12:"; break;
35 case VK_AARCH64_LO12: OS << ":lo12:"; break;
36 case VK_AARCH64_ABS_G0: OS << ":abs_g0:"; break;
37 case VK_AARCH64_ABS_G0_NC: OS << ":abs_g0_nc:"; break;
38 case VK_AARCH64_ABS_G1: OS << ":abs_g1:"; break;
39 case VK_AARCH64_ABS_G1_NC: OS << ":abs_g1_nc:"; break;
40 case VK_AARCH64_ABS_G2: OS << ":abs_g2:"; break;
41 case VK_AARCH64_ABS_G2_NC: OS << ":abs_g2_nc:"; break;
42 case VK_AARCH64_ABS_G3: OS << ":abs_g3:"; break;
43 case VK_AARCH64_SABS_G0: OS << ":abs_g0_s:"; break;
44 case VK_AARCH64_SABS_G1: OS << ":abs_g1_s:"; break;
45 case VK_AARCH64_SABS_G2: OS << ":abs_g2_s:"; break;
46 case VK_AARCH64_DTPREL_G2: OS << ":dtprel_g2:"; break;
47 case VK_AARCH64_DTPREL_G1: OS << ":dtprel_g1:"; break;
48 case VK_AARCH64_DTPREL_G1_NC: OS << ":dtprel_g1_nc:"; break;
49 case VK_AARCH64_DTPREL_G0: OS << ":dtprel_g0:"; break;
50 case VK_AARCH64_DTPREL_G0_NC: OS << ":dtprel_g0_nc:"; break;
51 case VK_AARCH64_DTPREL_HI12: OS << ":dtprel_hi12:"; break;
52 case VK_AARCH64_DTPREL_LO12: OS << ":dtprel_lo12:"; break;
53 case VK_AARCH64_DTPREL_LO12_NC: OS << ":dtprel_lo12_nc:"; break;
54 case VK_AARCH64_GOTTPREL_G1: OS << ":gottprel_g1:"; break;
55 case VK_AARCH64_GOTTPREL_G0_NC: OS << ":gottprel_g0_nc:"; break;
56 case VK_AARCH64_GOTTPREL: OS << ":gottprel:"; break;
57 case VK_AARCH64_GOTTPREL_LO12: OS << ":gottprel_lo12:"; break;
58 case VK_AARCH64_TPREL_G2: OS << ":tprel_g2:"; break;
59 case VK_AARCH64_TPREL_G1: OS << ":tprel_g1:"; break;
60 case VK_AARCH64_TPREL_G1_NC: OS << ":tprel_g1_nc:"; break;
61 case VK_AARCH64_TPREL_G0: OS << ":tprel_g0:"; break;
62 case VK_AARCH64_TPREL_G0_NC: OS << ":tprel_g0_nc:"; break;
63 case VK_AARCH64_TPREL_HI12: OS << ":tprel_hi12:"; break;
64 case VK_AARCH64_TPREL_LO12: OS << ":tprel_lo12:"; break;
65 case VK_AARCH64_TPREL_LO12_NC: OS << ":tprel_lo12_nc:"; break;
66 case VK_AARCH64_TLSDESC: OS << ":tlsdesc:"; break;
67 case VK_AARCH64_TLSDESC_LO12: OS << ":tlsdesc_lo12:"; break;
68
69 }
70
71 const MCExpr *Expr = getSubExpr();
72 if (Expr->getKind() != MCExpr::SymbolRef)
73 OS << '(';
74 Expr->print(OS);
75 if (Expr->getKind() != MCExpr::SymbolRef)
76 OS << ')';
77 }
78
79 bool
EvaluateAsRelocatableImpl(MCValue & Res,const MCAsmLayout * Layout) const80 AArch64MCExpr::EvaluateAsRelocatableImpl(MCValue &Res,
81 const MCAsmLayout *Layout) const {
82 return getSubExpr()->EvaluateAsRelocatable(Res, *Layout);
83 }
84
fixELFSymbolsInTLSFixupsImpl(const MCExpr * Expr,MCAssembler & Asm)85 static void fixELFSymbolsInTLSFixupsImpl(const MCExpr *Expr, MCAssembler &Asm) {
86 switch (Expr->getKind()) {
87 case MCExpr::Target:
88 llvm_unreachable("Can't handle nested target expression");
89 break;
90 case MCExpr::Constant:
91 break;
92
93 case MCExpr::Binary: {
94 const MCBinaryExpr *BE = cast<MCBinaryExpr>(Expr);
95 fixELFSymbolsInTLSFixupsImpl(BE->getLHS(), Asm);
96 fixELFSymbolsInTLSFixupsImpl(BE->getRHS(), Asm);
97 break;
98 }
99
100 case MCExpr::SymbolRef: {
101 // We're known to be under a TLS fixup, so any symbol should be
102 // modified. There should be only one.
103 const MCSymbolRefExpr &SymRef = *cast<MCSymbolRefExpr>(Expr);
104 MCSymbolData &SD = Asm.getOrCreateSymbolData(SymRef.getSymbol());
105 MCELF::SetType(SD, ELF::STT_TLS);
106 break;
107 }
108
109 case MCExpr::Unary:
110 fixELFSymbolsInTLSFixupsImpl(cast<MCUnaryExpr>(Expr)->getSubExpr(), Asm);
111 break;
112 }
113 }
114
fixELFSymbolsInTLSFixups(MCAssembler & Asm) const115 void AArch64MCExpr::fixELFSymbolsInTLSFixups(MCAssembler &Asm) const {
116 switch (getKind()) {
117 default:
118 return;
119 case VK_AARCH64_DTPREL_G2:
120 case VK_AARCH64_DTPREL_G1:
121 case VK_AARCH64_DTPREL_G1_NC:
122 case VK_AARCH64_DTPREL_G0:
123 case VK_AARCH64_DTPREL_G0_NC:
124 case VK_AARCH64_DTPREL_HI12:
125 case VK_AARCH64_DTPREL_LO12:
126 case VK_AARCH64_DTPREL_LO12_NC:
127 case VK_AARCH64_GOTTPREL_G1:
128 case VK_AARCH64_GOTTPREL_G0_NC:
129 case VK_AARCH64_GOTTPREL:
130 case VK_AARCH64_GOTTPREL_LO12:
131 case VK_AARCH64_TPREL_G2:
132 case VK_AARCH64_TPREL_G1:
133 case VK_AARCH64_TPREL_G1_NC:
134 case VK_AARCH64_TPREL_G0:
135 case VK_AARCH64_TPREL_G0_NC:
136 case VK_AARCH64_TPREL_HI12:
137 case VK_AARCH64_TPREL_LO12:
138 case VK_AARCH64_TPREL_LO12_NC:
139 case VK_AARCH64_TLSDESC:
140 case VK_AARCH64_TLSDESC_LO12:
141 break;
142 }
143
144 fixELFSymbolsInTLSFixupsImpl(getSubExpr(), Asm);
145 }
146
147 // FIXME: This basically copies MCObjectStreamer::AddValueSymbols. Perhaps
148 // that method should be made public?
149 // FIXME: really do above: now that two backends are using it.
AddValueSymbolsImpl(const MCExpr * Value,MCAssembler * Asm)150 static void AddValueSymbolsImpl(const MCExpr *Value, MCAssembler *Asm) {
151 switch (Value->getKind()) {
152 case MCExpr::Target:
153 llvm_unreachable("Can't handle nested target expr!");
154 break;
155
156 case MCExpr::Constant:
157 break;
158
159 case MCExpr::Binary: {
160 const MCBinaryExpr *BE = cast<MCBinaryExpr>(Value);
161 AddValueSymbolsImpl(BE->getLHS(), Asm);
162 AddValueSymbolsImpl(BE->getRHS(), Asm);
163 break;
164 }
165
166 case MCExpr::SymbolRef:
167 Asm->getOrCreateSymbolData(cast<MCSymbolRefExpr>(Value)->getSymbol());
168 break;
169
170 case MCExpr::Unary:
171 AddValueSymbolsImpl(cast<MCUnaryExpr>(Value)->getSubExpr(), Asm);
172 break;
173 }
174 }
175
AddValueSymbols(MCAssembler * Asm) const176 void AArch64MCExpr::AddValueSymbols(MCAssembler *Asm) const {
177 AddValueSymbolsImpl(getSubExpr(), Asm);
178 }
179