1 //===-- VEMCExpr.cpp - VE specific MC expression classes ------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file contains the implementation of the assembly expression modifiers
10 // accepted by the VE architecture (e.g. "%hi", "%lo", ...).
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "VEMCExpr.h"
15 #include "llvm/MC/MCAssembler.h"
16 #include "llvm/MC/MCContext.h"
17 #include "llvm/MC/MCObjectStreamer.h"
18 #include "llvm/MC/MCSymbolELF.h"
19 #include "llvm/BinaryFormat/ELF.h"
20 
21 using namespace llvm;
22 
23 #define DEBUG_TYPE "vemcexpr"
24 
25 const VEMCExpr *VEMCExpr::create(VariantKind Kind, const MCExpr *Expr,
26                                  MCContext &Ctx) {
27   return new (Ctx) VEMCExpr(Kind, Expr);
28 }
29 
30 void VEMCExpr::printImpl(raw_ostream &OS, const MCAsmInfo *MAI) const {
31 
32   bool closeParen = printVariantKind(OS, Kind);
33 
34   const MCExpr *Expr = getSubExpr();
35   Expr->print(OS, MAI);
36 
37   if (closeParen)
38     OS << ')';
39   printVariantKindSuffix(OS, Kind);
40 }
41 
42 bool VEMCExpr::printVariantKind(raw_ostream &OS, VariantKind Kind) {
43   switch (Kind) {
44   case VK_VE_None:
45     return false;
46 
47   case VK_VE_HI32:
48   case VK_VE_LO32:
49   case VK_VE_PC_HI32:
50   case VK_VE_PC_LO32:
51   case VK_VE_GOT_HI32:
52   case VK_VE_GOT_LO32:
53   case VK_VE_GOTOFF_HI32:
54   case VK_VE_GOTOFF_LO32:
55   case VK_VE_PLT_HI32:
56   case VK_VE_PLT_LO32:
57   case VK_VE_TLS_GD_HI32:
58   case VK_VE_TLS_GD_LO32:
59   case VK_VE_TPOFF_HI32:
60   case VK_VE_TPOFF_LO32:
61     return false; // OS << "@<text>(";  break;
62   }
63   return true;
64 }
65 
66 void VEMCExpr::printVariantKindSuffix(raw_ostream &OS, VariantKind Kind) {
67   switch (Kind) {
68   case VK_VE_None:
69     break;
70   case VK_VE_HI32:
71     OS << "@hi";
72     break;
73   case VK_VE_LO32:
74     OS << "@lo";
75     break;
76   case VK_VE_PC_HI32:
77     OS << "@pc_hi";
78     break;
79   case VK_VE_PC_LO32:
80     OS << "@pc_lo";
81     break;
82   case VK_VE_GOT_HI32:
83     OS << "@got_hi";
84     break;
85   case VK_VE_GOT_LO32:
86     OS << "@got_lo";
87     break;
88   case VK_VE_GOTOFF_HI32:
89     OS << "@gotoff_hi";
90     break;
91   case VK_VE_GOTOFF_LO32:
92     OS << "@gotoff_lo";
93     break;
94   case VK_VE_PLT_HI32:
95     OS << "@plt_hi";
96     break;
97   case VK_VE_TLS_GD_HI32:
98     OS << "@tls_gd_hi";
99     break;
100   case VK_VE_TLS_GD_LO32:
101     OS << "@tls_gd_lo";
102     break;
103   case VK_VE_PLT_LO32:
104     OS << "@plt_lo";
105     break;
106   case VK_VE_TPOFF_HI32:
107     OS << "@tpoff_hi";
108     break;
109   case VK_VE_TPOFF_LO32:
110     OS << "@tpoff_lo";
111     break;
112   }
113 }
114 
115 VEMCExpr::VariantKind VEMCExpr::parseVariantKind(StringRef name) {
116   return StringSwitch<VEMCExpr::VariantKind>(name)
117       .Case("hi", VK_VE_HI32)
118       .Case("lo", VK_VE_LO32)
119       .Case("pc_hi", VK_VE_PC_HI32)
120       .Case("pc_lo", VK_VE_PC_LO32)
121       .Case("got_hi", VK_VE_GOT_HI32)
122       .Case("got_lo", VK_VE_GOT_LO32)
123       .Case("gotoff_hi", VK_VE_GOTOFF_HI32)
124       .Case("gotoff_lo", VK_VE_GOTOFF_LO32)
125       .Case("plt_hi", VK_VE_PLT_HI32)
126       .Case("plt_lo", VK_VE_PLT_LO32)
127       .Case("tls_gd_hi", VK_VE_TLS_GD_HI32)
128       .Case("tls_gd_lo", VK_VE_TLS_GD_LO32)
129       .Case("tpoff_hi", VK_VE_TPOFF_HI32)
130       .Case("tpoff_lo", VK_VE_TPOFF_LO32)
131       .Default(VK_VE_None);
132 }
133 
134 VE::Fixups VEMCExpr::getFixupKind(VEMCExpr::VariantKind Kind) {
135   switch (Kind) {
136   default:
137     llvm_unreachable("Unhandled VEMCExpr::VariantKind");
138   case VK_VE_HI32:
139     return VE::fixup_ve_hi32;
140   case VK_VE_LO32:
141     return VE::fixup_ve_lo32;
142   case VK_VE_PC_HI32:
143     return VE::fixup_ve_pc_hi32;
144   case VK_VE_PC_LO32:
145     return VE::fixup_ve_pc_lo32;
146   case VK_VE_GOT_HI32:
147     return VE::fixup_ve_got_hi32;
148   case VK_VE_GOT_LO32:
149     return VE::fixup_ve_got_lo32;
150   case VK_VE_GOTOFF_HI32:
151     return VE::fixup_ve_gotoff_hi32;
152   case VK_VE_GOTOFF_LO32:
153     return VE::fixup_ve_gotoff_lo32;
154   case VK_VE_PLT_HI32:
155     return VE::fixup_ve_plt_hi32;
156   case VK_VE_PLT_LO32:
157     return VE::fixup_ve_plt_lo32;
158   case VK_VE_TLS_GD_HI32:
159     return VE::fixup_ve_tls_gd_hi32;
160   case VK_VE_TLS_GD_LO32:
161     return VE::fixup_ve_tls_gd_lo32;
162   }
163 }
164 
165 bool VEMCExpr::evaluateAsRelocatableImpl(MCValue &Res,
166                                          const MCAsmLayout *Layout,
167                                          const MCFixup *Fixup) const {
168   return getSubExpr()->evaluateAsRelocatable(Res, Layout, Fixup);
169 }
170 
171 static void fixELFSymbolsInTLSFixupsImpl(const MCExpr *Expr, MCAssembler &Asm) {
172   switch (Expr->getKind()) {
173   case MCExpr::Target:
174     llvm_unreachable("Can't handle nested target expr!");
175     break;
176 
177   case MCExpr::Constant:
178     break;
179 
180   case MCExpr::Binary: {
181     const MCBinaryExpr *BE = cast<MCBinaryExpr>(Expr);
182     fixELFSymbolsInTLSFixupsImpl(BE->getLHS(), Asm);
183     fixELFSymbolsInTLSFixupsImpl(BE->getRHS(), Asm);
184     break;
185   }
186 
187   case MCExpr::SymbolRef: {
188     const MCSymbolRefExpr &SymRef = *cast<MCSymbolRefExpr>(Expr);
189     cast<MCSymbolELF>(SymRef.getSymbol()).setType(ELF::STT_TLS);
190     break;
191   }
192 
193   case MCExpr::Unary:
194     fixELFSymbolsInTLSFixupsImpl(cast<MCUnaryExpr>(Expr)->getSubExpr(), Asm);
195     break;
196   }
197 }
198 
199 void VEMCExpr::visitUsedExpr(MCStreamer &Streamer) const {
200   Streamer.visitUsedExpr(*getSubExpr());
201 }
202 
203 void VEMCExpr::fixELFSymbolsInTLSFixups(MCAssembler &Asm) const {
204   fixELFSymbolsInTLSFixupsImpl(getSubExpr(), Asm);
205 }
206