1 //===-- ARM64MCExpr.cpp - ARM64 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 // This file contains the implementation of the assembly expression modifiers
11 // accepted by the AArch64 architecture (e.g. ":lo12:", ":gottprel_g1:", ...).
13 //===----------------------------------------------------------------------===//
15 #define DEBUG_TYPE "aarch64symbolrefexpr"
16 #include "ARM64MCExpr.h"
17 #include "llvm/MC/MCAssembler.h"
18 #include "llvm/MC/MCContext.h"
19 #include "llvm/MC/MCELF.h"
20 #include "llvm/MC/MCSymbol.h"
21 #include "llvm/MC/MCValue.h"
22 #include "llvm/Object/ELF.h"
23 #include "llvm/Support/ErrorHandling.h"
27 const ARM64MCExpr *ARM64MCExpr::Create(const MCExpr *Expr, VariantKind Kind,
29 return new (Ctx) ARM64MCExpr(Expr, Kind);
32 StringRef ARM64MCExpr::getVariantKindName() const {
33 switch (static_cast<uint32_t>(getKind())) {
34 case VK_CALL: return "";
35 case VK_LO12: return ":lo12:";
36 case VK_ABS_G3: return ":abs_g3:";
37 case VK_ABS_G2: return ":abs_g2:";
38 case VK_ABS_G2_NC: return ":abs_g2_nc:";
39 case VK_ABS_G1: return ":abs_g1:";
40 case VK_ABS_G1_NC: return ":abs_g1_nc:";
41 case VK_ABS_G0: return ":abs_g0:";
42 case VK_ABS_G0_NC: return ":abs_g0_nc:";
43 case VK_DTPREL_G2: return ":dtprel_g2:";
44 case VK_DTPREL_G1: return ":dtprel_g1:";
45 case VK_DTPREL_G1_NC: return ":dtprel_g1_nc:";
46 case VK_DTPREL_G0: return ":dtprel_g0:";
47 case VK_DTPREL_G0_NC: return ":dtprel_g0_nc:";
48 case VK_DTPREL_LO12: return ":dtprel_lo12:";
49 case VK_DTPREL_LO12_NC: return ":dtprel_lo12_nc:";
50 case VK_TPREL_G2: return ":tprel_g2:";
51 case VK_TPREL_G1: return ":tprel_g1:";
52 case VK_TPREL_G1_NC: return ":tprel_g1_nc:";
53 case VK_TPREL_G0: return ":tprel_g0:";
54 case VK_TPREL_G0_NC: return ":tprel_g0_nc:";
55 case VK_TPREL_LO12: return ":tprel_lo12:";
56 case VK_TPREL_LO12_NC: return ":tprel_lo12_nc:";
57 case VK_TLSDESC_LO12: return ":tlsdesc_lo12:";
58 case VK_ABS_PAGE: return "";
59 case VK_GOT_PAGE: return ":got:";
60 case VK_GOT_LO12: return ":got_lo12:";
61 case VK_GOTTPREL_PAGE: return ":gottprel:";
62 case VK_GOTTPREL_LO12_NC: return ":gottprel_lo12:";
63 case VK_GOTTPREL_G1: return ":gottprel_g1:";
64 case VK_GOTTPREL_G0_NC: return ":gottprel_g0_nc:";
65 case VK_TLSDESC: return "";
66 case VK_TLSDESC_PAGE: return ":tlsdesc:";
68 llvm_unreachable("Invalid ELF symbol kind");
72 void ARM64MCExpr::PrintImpl(raw_ostream &OS) const {
73 if (getKind() != VK_NONE)
74 OS << getVariantKindName();
78 // FIXME: This basically copies MCObjectStreamer::AddValueSymbols. Perhaps
79 // that method should be made public?
80 // FIXME: really do above: now that two backends are using it.
81 static void AddValueSymbolsImpl(const MCExpr *Value, MCAssembler *Asm) {
82 switch (Value->getKind()) {
84 llvm_unreachable("Can't handle nested target expr!");
87 case MCExpr::Constant:
90 case MCExpr::Binary: {
91 const MCBinaryExpr *BE = cast<MCBinaryExpr>(Value);
92 AddValueSymbolsImpl(BE->getLHS(), Asm);
93 AddValueSymbolsImpl(BE->getRHS(), Asm);
97 case MCExpr::SymbolRef:
98 Asm->getOrCreateSymbolData(cast<MCSymbolRefExpr>(Value)->getSymbol());
102 AddValueSymbolsImpl(cast<MCUnaryExpr>(Value)->getSubExpr(), Asm);
107 void ARM64MCExpr::AddValueSymbols(MCAssembler *Asm) const {
108 AddValueSymbolsImpl(getSubExpr(), Asm);
111 const MCSection *ARM64MCExpr::FindAssociatedSection() const {
112 llvm_unreachable("FIXME: what goes here?");
115 bool ARM64MCExpr::EvaluateAsRelocatableImpl(MCValue &Res,
116 const MCAsmLayout *Layout) const {
117 if (!getSubExpr()->EvaluateAsRelocatable(Res, Layout))
121 MCValue::get(Res.getSymA(), Res.getSymB(), Res.getConstant(), getKind());
126 static void fixELFSymbolsInTLSFixupsImpl(const MCExpr *Expr, MCAssembler &Asm) {
127 switch (Expr->getKind()) {
129 llvm_unreachable("Can't handle nested target expression");
131 case MCExpr::Constant:
134 case MCExpr::Binary: {
135 const MCBinaryExpr *BE = cast<MCBinaryExpr>(Expr);
136 fixELFSymbolsInTLSFixupsImpl(BE->getLHS(), Asm);
137 fixELFSymbolsInTLSFixupsImpl(BE->getRHS(), Asm);
141 case MCExpr::SymbolRef: {
142 // We're known to be under a TLS fixup, so any symbol should be
143 // modified. There should be only one.
144 const MCSymbolRefExpr &SymRef = *cast<MCSymbolRefExpr>(Expr);
145 MCSymbolData &SD = Asm.getOrCreateSymbolData(SymRef.getSymbol());
146 MCELF::SetType(SD, ELF::STT_TLS);
151 fixELFSymbolsInTLSFixupsImpl(cast<MCUnaryExpr>(Expr)->getSubExpr(), Asm);
156 void ARM64MCExpr::fixELFSymbolsInTLSFixups(MCAssembler &Asm) const {
157 switch (getSymbolLoc(Kind)) {
167 fixELFSymbolsInTLSFixupsImpl(getSubExpr(), Asm);