blob: f59222263dab6a7b104e4b5e4bab4c0e1115d6e9 [file] [log] [blame]
Daniel Dunbar28c251b2009-08-31 08:06:59 +00001//===- MCExpr.cpp - Assembly Level Expression Implementation --------------===//
Daniel Dunbarfc6877a2009-06-29 20:40:36 +00002//
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
Daniel Dunbarff547842010-03-23 23:47:14 +000010#define DEBUG_TYPE "mcexpr"
Daniel Dunbar28c251b2009-08-31 08:06:59 +000011#include "llvm/MC/MCExpr.h"
Daniel Dunbarff547842010-03-23 23:47:14 +000012#include "llvm/ADT/Statistic.h"
Daniel Dunbar4e815f82010-03-15 23:51:06 +000013#include "llvm/ADT/StringSwitch.h"
Daniel Dunbar979ba5b2010-03-11 05:53:37 +000014#include "llvm/MC/MCAsmLayout.h"
15#include "llvm/MC/MCAssembler.h"
Daniel Dunbarfc6877a2009-06-29 20:40:36 +000016#include "llvm/MC/MCContext.h"
Craig Topperf1d0f772012-03-26 06:58:25 +000017#include "llvm/MC/MCObjectWriter.h"
Daniel Dunbar15d17072009-06-30 01:49:52 +000018#include "llvm/MC/MCSymbol.h"
Daniel Dunbarfc6877a2009-06-29 20:40:36 +000019#include "llvm/MC/MCValue.h"
David Greene5d2bcb12010-01-05 01:28:07 +000020#include "llvm/Support/Debug.h"
Chandler Carruth732f05c2012-01-10 18:08:01 +000021#include "llvm/Support/ErrorHandling.h"
Daniel Dunbar87392fd2009-08-31 08:07:33 +000022#include "llvm/Support/raw_ostream.h"
Daniel Dunbarfc6877a2009-06-29 20:40:36 +000023using namespace llvm;
24
Daniel Dunbarff547842010-03-23 23:47:14 +000025namespace {
26namespace stats {
27STATISTIC(MCExprEvaluate, "Number of MCExpr evaluations");
28}
29}
30
Chris Lattner8cb9a3b2010-01-18 00:37:40 +000031void MCExpr::print(raw_ostream &OS) const {
Daniel Dunbar87392fd2009-08-31 08:07:33 +000032 switch (getKind()) {
Chris Lattner5d917a82010-02-08 19:41:07 +000033 case MCExpr::Target:
34 return cast<MCTargetExpr>(this)->PrintImpl(OS);
Daniel Dunbar87392fd2009-08-31 08:07:33 +000035 case MCExpr::Constant:
36 OS << cast<MCConstantExpr>(*this).getValue();
37 return;
38
Chris Lattnerd50c2b92009-09-08 23:20:50 +000039 case MCExpr::SymbolRef: {
Daniel Dunbar4e815f82010-03-15 23:51:06 +000040 const MCSymbolRefExpr &SRE = cast<MCSymbolRefExpr>(*this);
41 const MCSymbol &Sym = SRE.getSymbol();
Chris Lattner1e61e692010-11-15 02:46:57 +000042 // Parenthesize names that start with $ so that they don't look like
43 // absolute names.
44 bool UseParens = Sym.getName()[0] == '$';
Chris Lattner1e61e692010-11-15 02:46:57 +000045 if (UseParens)
Chris Lattner10b318b2010-01-17 21:43:43 +000046 OS << '(' << Sym << ')';
47 else
48 OS << Sym;
Daniel Dunbar4e815f82010-03-15 23:51:06 +000049
Logan Chien37c74612012-12-12 07:14:46 +000050 if (SRE.getKind() == MCSymbolRefExpr::VK_ARM_NONE ||
51 SRE.getKind() == MCSymbolRefExpr::VK_ARM_PLT ||
Jim Grosbach2c4d5122010-11-10 03:26:07 +000052 SRE.getKind() == MCSymbolRefExpr::VK_ARM_TLSGD ||
53 SRE.getKind() == MCSymbolRefExpr::VK_ARM_GOT ||
54 SRE.getKind() == MCSymbolRefExpr::VK_ARM_GOTOFF ||
55 SRE.getKind() == MCSymbolRefExpr::VK_ARM_TPOFF ||
James Molloy34982572012-01-26 09:25:43 +000056 SRE.getKind() == MCSymbolRefExpr::VK_ARM_GOTTPOFF ||
Anton Korobeynikov12cfa112012-11-09 20:20:12 +000057 SRE.getKind() == MCSymbolRefExpr::VK_ARM_TARGET1 ||
Logan Chien37c74612012-12-12 07:14:46 +000058 SRE.getKind() == MCSymbolRefExpr::VK_ARM_TARGET2 ||
59 SRE.getKind() == MCSymbolRefExpr::VK_ARM_PREL31)
Jim Grosbach637d89f2010-09-22 23:27:36 +000060 OS << MCSymbolRefExpr::getVariantKindName(SRE.getKind());
Ulrich Weigandedaa58e2013-05-23 22:26:41 +000061 else if (SRE.getKind() != MCSymbolRefExpr::VK_None)
Daniel Dunbar4e815f82010-03-15 23:51:06 +000062 OS << '@' << MCSymbolRefExpr::getVariantKindName(SRE.getKind());
63
Daniel Dunbar87392fd2009-08-31 08:07:33 +000064 return;
Chris Lattnerd50c2b92009-09-08 23:20:50 +000065 }
Daniel Dunbar87392fd2009-08-31 08:07:33 +000066
67 case MCExpr::Unary: {
68 const MCUnaryExpr &UE = cast<MCUnaryExpr>(*this);
69 switch (UE.getOpcode()) {
Daniel Dunbar87392fd2009-08-31 08:07:33 +000070 case MCUnaryExpr::LNot: OS << '!'; break;
71 case MCUnaryExpr::Minus: OS << '-'; break;
72 case MCUnaryExpr::Not: OS << '~'; break;
73 case MCUnaryExpr::Plus: OS << '+'; break;
74 }
Chris Lattner8cb9a3b2010-01-18 00:37:40 +000075 OS << *UE.getSubExpr();
Daniel Dunbar87392fd2009-08-31 08:07:33 +000076 return;
77 }
78
79 case MCExpr::Binary: {
80 const MCBinaryExpr &BE = cast<MCBinaryExpr>(*this);
Daniel Dunbar9a1d2002010-03-18 00:59:10 +000081
Chris Lattnerbe73e8c2009-09-08 06:34:07 +000082 // Only print parens around the LHS if it is non-trivial.
83 if (isa<MCConstantExpr>(BE.getLHS()) || isa<MCSymbolRefExpr>(BE.getLHS())) {
Chris Lattner8cb9a3b2010-01-18 00:37:40 +000084 OS << *BE.getLHS();
Chris Lattnerbe73e8c2009-09-08 06:34:07 +000085 } else {
Chris Lattner8cb9a3b2010-01-18 00:37:40 +000086 OS << '(' << *BE.getLHS() << ')';
Chris Lattnerbe73e8c2009-09-08 06:34:07 +000087 }
Daniel Dunbar9a1d2002010-03-18 00:59:10 +000088
Daniel Dunbar87392fd2009-08-31 08:07:33 +000089 switch (BE.getOpcode()) {
Chris Lattnerd19ceb92009-09-08 06:37:35 +000090 case MCBinaryExpr::Add:
91 // Print "X-42" instead of "X+-42".
92 if (const MCConstantExpr *RHSC = dyn_cast<MCConstantExpr>(BE.getRHS())) {
93 if (RHSC->getValue() < 0) {
94 OS << RHSC->getValue();
95 return;
96 }
97 }
Daniel Dunbar9a1d2002010-03-18 00:59:10 +000098
Chris Lattnerd19ceb92009-09-08 06:37:35 +000099 OS << '+';
100 break;
Daniel Dunbar87392fd2009-08-31 08:07:33 +0000101 case MCBinaryExpr::And: OS << '&'; break;
102 case MCBinaryExpr::Div: OS << '/'; break;
103 case MCBinaryExpr::EQ: OS << "=="; break;
104 case MCBinaryExpr::GT: OS << '>'; break;
105 case MCBinaryExpr::GTE: OS << ">="; break;
106 case MCBinaryExpr::LAnd: OS << "&&"; break;
107 case MCBinaryExpr::LOr: OS << "||"; break;
108 case MCBinaryExpr::LT: OS << '<'; break;
109 case MCBinaryExpr::LTE: OS << "<="; break;
110 case MCBinaryExpr::Mod: OS << '%'; break;
111 case MCBinaryExpr::Mul: OS << '*'; break;
112 case MCBinaryExpr::NE: OS << "!="; break;
113 case MCBinaryExpr::Or: OS << '|'; break;
114 case MCBinaryExpr::Shl: OS << "<<"; break;
115 case MCBinaryExpr::Shr: OS << ">>"; break;
116 case MCBinaryExpr::Sub: OS << '-'; break;
117 case MCBinaryExpr::Xor: OS << '^'; break;
118 }
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000119
Chris Lattnerbe73e8c2009-09-08 06:34:07 +0000120 // Only print parens around the LHS if it is non-trivial.
121 if (isa<MCConstantExpr>(BE.getRHS()) || isa<MCSymbolRefExpr>(BE.getRHS())) {
Chris Lattner8cb9a3b2010-01-18 00:37:40 +0000122 OS << *BE.getRHS();
Chris Lattnerbe73e8c2009-09-08 06:34:07 +0000123 } else {
Chris Lattner8cb9a3b2010-01-18 00:37:40 +0000124 OS << '(' << *BE.getRHS() << ')';
Chris Lattnerbe73e8c2009-09-08 06:34:07 +0000125 }
Daniel Dunbar87392fd2009-08-31 08:07:33 +0000126 return;
127 }
128 }
129
Craig Topper85814382012-02-07 05:05:23 +0000130 llvm_unreachable("Invalid expression kind!");
Daniel Dunbar87392fd2009-08-31 08:07:33 +0000131}
132
Manman Ren286c4dc2012-09-12 05:06:18 +0000133#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
Daniel Dunbar87392fd2009-08-31 08:07:33 +0000134void MCExpr::dump() const {
Chris Lattner8cb9a3b2010-01-18 00:37:40 +0000135 print(dbgs());
David Greene5d2bcb12010-01-05 01:28:07 +0000136 dbgs() << '\n';
Daniel Dunbar87392fd2009-08-31 08:07:33 +0000137}
Manman Rencc77eec2012-09-06 19:55:56 +0000138#endif
Daniel Dunbar87392fd2009-08-31 08:07:33 +0000139
140/* *** */
141
Chris Lattner8b4ada22009-09-08 06:27:48 +0000142const MCBinaryExpr *MCBinaryExpr::Create(Opcode Opc, const MCExpr *LHS,
143 const MCExpr *RHS, MCContext &Ctx) {
Daniel Dunbar9643ac52009-08-31 08:07:22 +0000144 return new (Ctx) MCBinaryExpr(Opc, LHS, RHS);
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000145}
146
Chris Lattner8b4ada22009-09-08 06:27:48 +0000147const MCUnaryExpr *MCUnaryExpr::Create(Opcode Opc, const MCExpr *Expr,
148 MCContext &Ctx) {
Daniel Dunbar9643ac52009-08-31 08:07:22 +0000149 return new (Ctx) MCUnaryExpr(Opc, Expr);
150}
151
152const MCConstantExpr *MCConstantExpr::Create(int64_t Value, MCContext &Ctx) {
153 return new (Ctx) MCConstantExpr(Value);
154}
155
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000156/* *** */
157
Daniel Dunbar9643ac52009-08-31 08:07:22 +0000158const MCSymbolRefExpr *MCSymbolRefExpr::Create(const MCSymbol *Sym,
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000159 VariantKind Kind,
Daniel Dunbar9643ac52009-08-31 08:07:22 +0000160 MCContext &Ctx) {
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000161 return new (Ctx) MCSymbolRefExpr(Sym, Kind);
Daniel Dunbar9643ac52009-08-31 08:07:22 +0000162}
163
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000164const MCSymbolRefExpr *MCSymbolRefExpr::Create(StringRef Name, VariantKind Kind,
165 MCContext &Ctx) {
166 return Create(Ctx.GetOrCreateSymbol(Name), Kind, Ctx);
Chris Lattner4f3e7aa2009-09-16 01:26:31 +0000167}
168
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000169StringRef MCSymbolRefExpr::getVariantKindName(VariantKind Kind) {
170 switch (Kind) {
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000171 case VK_Invalid: return "<<invalid>>";
172 case VK_None: return "<<none>>";
173
174 case VK_GOT: return "GOT";
175 case VK_GOTOFF: return "GOTOFF";
176 case VK_GOTPCREL: return "GOTPCREL";
177 case VK_GOTTPOFF: return "GOTTPOFF";
178 case VK_INDNTPOFF: return "INDNTPOFF";
179 case VK_NTPOFF: return "NTPOFF";
Rafael Espindolaa0a2f872010-10-28 14:22:44 +0000180 case VK_GOTNTPOFF: return "GOTNTPOFF";
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000181 case VK_PLT: return "PLT";
182 case VK_TLSGD: return "TLSGD";
Rafael Espindolab4d17212010-10-28 15:02:40 +0000183 case VK_TLSLD: return "TLSLD";
Rafael Espindolaa264f722010-10-28 14:37:09 +0000184 case VK_TLSLDM: return "TLSLDM";
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000185 case VK_TPOFF: return "TPOFF";
Rafael Espindola0cf15d62010-10-28 14:48:59 +0000186 case VK_DTPOFF: return "DTPOFF";
Chris Lattner6135a962010-11-14 22:22:59 +0000187 case VK_TLVP: return "TLVP";
Benjamin Kramer42734cf2013-03-30 16:21:50 +0000188 case VK_SECREL: return "SECREL32";
Logan Chien37c74612012-12-12 07:14:46 +0000189 case VK_ARM_NONE: return "(NONE)";
Jim Grosbach637d89f2010-09-22 23:27:36 +0000190 case VK_ARM_PLT: return "(PLT)";
Jim Grosbach2c4d5122010-11-10 03:26:07 +0000191 case VK_ARM_GOT: return "(GOT)";
192 case VK_ARM_GOTOFF: return "(GOTOFF)";
193 case VK_ARM_TPOFF: return "(tpoff)";
194 case VK_ARM_GOTTPOFF: return "(gottpoff)";
Jim Grosbach4d088632010-11-17 19:30:11 +0000195 case VK_ARM_TLSGD: return "(tlsgd)";
James Molloy34982572012-01-26 09:25:43 +0000196 case VK_ARM_TARGET1: return "(target1)";
Anton Korobeynikov12cfa112012-11-09 20:20:12 +0000197 case VK_ARM_TARGET2: return "(target2)";
Logan Chien37c74612012-12-12 07:14:46 +0000198 case VK_ARM_PREL31: return "(prel31)";
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000199 case VK_PPC_LO: return "l";
Ulrich Weigandd2849572013-06-21 14:42:49 +0000200 case VK_PPC_HI: return "h";
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000201 case VK_PPC_HA: return "ha";
Ulrich Weigand84656592013-06-20 22:39:42 +0000202 case VK_PPC_TOCBASE: return "tocbase";
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000203 case VK_PPC_TOC: return "toc";
204 case VK_PPC_TOC_LO: return "toc@l";
205 case VK_PPC_TOC_HA: return "toc@ha";
206 case VK_PPC_TPREL_LO: return "tprel@l";
207 case VK_PPC_TPREL_HA: return "tprel@ha";
208 case VK_PPC_DTPREL_LO: return "dtprel@l";
209 case VK_PPC_DTPREL_HA: return "dtprel@ha";
210 case VK_PPC_GOT_TPREL_LO: return "got@tprel@l";
211 case VK_PPC_GOT_TPREL_HA: return "got@tprel@ha";
Bill Schmidtd7802bf2012-12-04 16:18:08 +0000212 case VK_PPC_TLS: return "tls";
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000213 case VK_PPC_GOT_TLSGD_LO: return "got@tlsgd@l";
214 case VK_PPC_GOT_TLSGD_HA: return "got@tlsgd@ha";
Bill Schmidt57ac1f42012-12-11 20:30:11 +0000215 case VK_PPC_TLSGD: return "tlsgd";
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000216 case VK_PPC_GOT_TLSLD_LO: return "got@tlsld@l";
217 case VK_PPC_GOT_TLSLD_HA: return "got@tlsld@ha";
Bill Schmidt349c2782012-12-12 19:29:35 +0000218 case VK_PPC_TLSLD: return "tlsld";
Akira Hatanakabc249852011-12-22 01:05:17 +0000219 case VK_Mips_GPREL: return "GPREL";
220 case VK_Mips_GOT_CALL: return "GOT_CALL";
221 case VK_Mips_GOT16: return "GOT16";
222 case VK_Mips_GOT: return "GOT";
223 case VK_Mips_ABS_HI: return "ABS_HI";
224 case VK_Mips_ABS_LO: return "ABS_LO";
225 case VK_Mips_TLSGD: return "TLSGD";
226 case VK_Mips_TLSLDM: return "TLSLDM";
227 case VK_Mips_DTPREL_HI: return "DTPREL_HI";
228 case VK_Mips_DTPREL_LO: return "DTPREL_LO";
229 case VK_Mips_GOTTPREL: return "GOTTPREL";
230 case VK_Mips_TPREL_HI: return "TPREL_HI";
231 case VK_Mips_TPREL_LO: return "TPREL_LO";
232 case VK_Mips_GPOFF_HI: return "GPOFF_HI";
233 case VK_Mips_GPOFF_LO: return "GPOFF_LO";
234 case VK_Mips_GOT_DISP: return "GOT_DISP";
235 case VK_Mips_GOT_PAGE: return "GOT_PAGE";
236 case VK_Mips_GOT_OFST: return "GOT_OFST";
Akira Hatanakab7dd9fc2012-07-21 02:15:19 +0000237 case VK_Mips_HIGHER: return "HIGHER";
238 case VK_Mips_HIGHEST: return "HIGHEST";
Akira Hatanakae3904342012-11-21 19:50:22 +0000239 case VK_Mips_GOT_HI16: return "GOT_HI16";
240 case VK_Mips_GOT_LO16: return "GOT_LO16";
241 case VK_Mips_CALL_HI16: return "CALL_HI16";
242 case VK_Mips_CALL_LO16: return "CALL_LO16";
Nico Rieck18d49ac2013-04-10 23:28:17 +0000243 case VK_COFF_IMGREL32: return "IMGREL32";
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000244 }
Chandler Carruth732f05c2012-01-10 18:08:01 +0000245 llvm_unreachable("Invalid variant kind");
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000246}
247
248MCSymbolRefExpr::VariantKind
249MCSymbolRefExpr::getVariantKindForName(StringRef Name) {
250 return StringSwitch<VariantKind>(Name)
251 .Case("GOT", VK_GOT)
Rafael Espindola54104db2011-01-23 16:11:25 +0000252 .Case("got", VK_GOT)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000253 .Case("GOTOFF", VK_GOTOFF)
Rafael Espindola54104db2011-01-23 16:11:25 +0000254 .Case("gotoff", VK_GOTOFF)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000255 .Case("GOTPCREL", VK_GOTPCREL)
Rafael Espindola54104db2011-01-23 16:11:25 +0000256 .Case("gotpcrel", VK_GOTPCREL)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000257 .Case("GOTTPOFF", VK_GOTTPOFF)
Rafael Espindola54104db2011-01-23 16:11:25 +0000258 .Case("gottpoff", VK_GOTTPOFF)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000259 .Case("INDNTPOFF", VK_INDNTPOFF)
Rafael Espindola54104db2011-01-23 16:11:25 +0000260 .Case("indntpoff", VK_INDNTPOFF)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000261 .Case("NTPOFF", VK_NTPOFF)
Rafael Espindola54104db2011-01-23 16:11:25 +0000262 .Case("ntpoff", VK_NTPOFF)
Rafael Espindolaa0a2f872010-10-28 14:22:44 +0000263 .Case("GOTNTPOFF", VK_GOTNTPOFF)
Rafael Espindola54104db2011-01-23 16:11:25 +0000264 .Case("gotntpoff", VK_GOTNTPOFF)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000265 .Case("PLT", VK_PLT)
Rafael Espindola54104db2011-01-23 16:11:25 +0000266 .Case("plt", VK_PLT)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000267 .Case("TLSGD", VK_TLSGD)
Rafael Espindola54104db2011-01-23 16:11:25 +0000268 .Case("tlsgd", VK_TLSGD)
Rafael Espindolab4d17212010-10-28 15:02:40 +0000269 .Case("TLSLD", VK_TLSLD)
Rafael Espindola54104db2011-01-23 16:11:25 +0000270 .Case("tlsld", VK_TLSLD)
Rafael Espindolaa264f722010-10-28 14:37:09 +0000271 .Case("TLSLDM", VK_TLSLDM)
Rafael Espindola54104db2011-01-23 16:11:25 +0000272 .Case("tlsldm", VK_TLSLDM)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000273 .Case("TPOFF", VK_TPOFF)
Rafael Espindola54104db2011-01-23 16:11:25 +0000274 .Case("tpoff", VK_TPOFF)
Rafael Espindola0cf15d62010-10-28 14:48:59 +0000275 .Case("DTPOFF", VK_DTPOFF)
Rafael Espindola54104db2011-01-23 16:11:25 +0000276 .Case("dtpoff", VK_DTPOFF)
Eric Christopher96ac5152010-05-26 00:02:12 +0000277 .Case("TLVP", VK_TLVP)
Rafael Espindola54104db2011-01-23 16:11:25 +0000278 .Case("tlvp", VK_TLVP)
Nico Rieck18d49ac2013-04-10 23:28:17 +0000279 .Case("IMGREL", VK_COFF_IMGREL32)
280 .Case("imgrel", VK_COFF_IMGREL32)
Rafael Espindola93d0b062013-04-25 19:27:05 +0000281 .Case("SECREL32", VK_SECREL)
282 .Case("secrel32", VK_SECREL)
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000283 .Case("L", VK_PPC_LO)
284 .Case("l", VK_PPC_LO)
Ulrich Weigandd2849572013-06-21 14:42:49 +0000285 .Case("H", VK_PPC_HI)
286 .Case("h", VK_PPC_HI)
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000287 .Case("HA", VK_PPC_HA)
288 .Case("ha", VK_PPC_HA)
Ulrich Weigand84656592013-06-20 22:39:42 +0000289 .Case("TOCBASE", VK_PPC_TOCBASE)
290 .Case("tocbase", VK_PPC_TOCBASE)
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000291 .Case("TOC", VK_PPC_TOC)
292 .Case("toc", VK_PPC_TOC)
293 .Case("TOC@L", VK_PPC_TOC_LO)
294 .Case("toc@l", VK_PPC_TOC_LO)
295 .Case("TOC@HA", VK_PPC_TOC_HA)
296 .Case("toc@ha", VK_PPC_TOC_HA)
Ulrich Weiganda7e5e6b2013-05-03 19:52:35 +0000297 .Case("TLS", VK_PPC_TLS)
298 .Case("tls", VK_PPC_TLS)
Ulrich Weigand92cfa612013-06-21 14:42:20 +0000299 .Case("TPREL@L", VK_PPC_TPREL_LO)
300 .Case("tprel@l", VK_PPC_TPREL_LO)
301 .Case("TPREL@HA", VK_PPC_TPREL_HA)
302 .Case("tprel@ha", VK_PPC_TPREL_HA)
303 .Case("DTPREL@L", VK_PPC_DTPREL_LO)
304 .Case("dtprel@l", VK_PPC_DTPREL_LO)
305 .Case("DTPREL@HA", VK_PPC_DTPREL_HA)
306 .Case("dtprel@ha", VK_PPC_DTPREL_HA)
307 .Case("GOT@TPREL@L", VK_PPC_GOT_TPREL_LO)
308 .Case("got@tprel@l", VK_PPC_GOT_TPREL_LO)
309 .Case("GOT@TPREL@HA", VK_PPC_GOT_TPREL_HA)
310 .Case("got@tprel@ha", VK_PPC_GOT_TPREL_HA)
311 .Case("GOT@TLSGD@L", VK_PPC_GOT_TLSGD_LO)
312 .Case("got@tlsgd@l", VK_PPC_GOT_TLSGD_LO)
313 .Case("GOT@TLSGD@HA", VK_PPC_GOT_TLSGD_HA)
314 .Case("got@tlsgd@ha", VK_PPC_GOT_TLSGD_HA)
315 .Case("GOT@TLSLD@L", VK_PPC_GOT_TLSLD_LO)
316 .Case("got@tlsld@l", VK_PPC_GOT_TLSLD_LO)
317 .Case("GOT@TLSLD@HA", VK_PPC_GOT_TLSLD_HA)
318 .Case("got@tlsld@ha", VK_PPC_GOT_TLSLD_HA)
Daniel Dunbar4e815f82010-03-15 23:51:06 +0000319 .Default(VK_Invalid);
320}
321
322/* *** */
323
Craig Topper4bb51cc2012-09-26 06:36:36 +0000324void MCTargetExpr::anchor() {}
Chris Lattner4f3e7aa2009-09-16 01:26:31 +0000325
Daniel Dunbar9643ac52009-08-31 08:07:22 +0000326/* *** */
327
Rafael Espindolad0764822010-12-18 03:57:21 +0000328bool MCExpr::EvaluateAsAbsolute(int64_t &Res) const {
329 return EvaluateAsAbsolute(Res, 0, 0, 0);
330}
331
332bool MCExpr::EvaluateAsAbsolute(int64_t &Res,
333 const MCAsmLayout &Layout) const {
334 return EvaluateAsAbsolute(Res, &Layout.getAssembler(), &Layout, 0);
335}
336
337bool MCExpr::EvaluateAsAbsolute(int64_t &Res,
338 const MCAsmLayout &Layout,
339 const SectionAddrMap &Addrs) const {
340 return EvaluateAsAbsolute(Res, &Layout.getAssembler(), &Layout, &Addrs);
341}
342
343bool MCExpr::EvaluateAsAbsolute(int64_t &Res, const MCAssembler &Asm) const {
344 return EvaluateAsAbsolute(Res, &Asm, 0, 0);
345}
346
347bool MCExpr::EvaluateAsAbsolute(int64_t &Res, const MCAssembler *Asm,
348 const MCAsmLayout *Layout,
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000349 const SectionAddrMap *Addrs) const {
Daniel Dunbar15d17072009-06-30 01:49:52 +0000350 MCValue Value;
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000351
Daniel Dunbaref6e96f2010-03-23 23:47:07 +0000352 // Fast path constants.
353 if (const MCConstantExpr *CE = dyn_cast<MCConstantExpr>(this)) {
354 Res = CE->getValue();
355 return true;
356 }
357
Rafael Espindola3f037ef2010-12-18 04:19:20 +0000358 // FIXME: The use if InSet = Addrs is a hack. Setting InSet causes us
359 // absolutize differences across sections and that is what the MachO writer
360 // uses Addrs for.
Rafael Espindola10b6d332010-12-22 22:16:24 +0000361 bool IsRelocatable =
362 EvaluateAsRelocatableImpl(Value, Asm, Layout, Addrs, /*InSet*/ Addrs);
Daniel Dunbar15d17072009-06-30 01:49:52 +0000363
Rafael Espindola10b6d332010-12-22 22:16:24 +0000364 // Record the current value.
Daniel Dunbar15d17072009-06-30 01:49:52 +0000365 Res = Value.getConstant();
Rafael Espindola10b6d332010-12-22 22:16:24 +0000366
367 return IsRelocatable && Value.isAbsolute();
Daniel Dunbar15d17072009-06-30 01:49:52 +0000368}
369
Daniel Dunbar17304b32010-12-17 05:50:33 +0000370/// \brief Helper method for \see EvaluateSymbolAdd().
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000371static void AttemptToFoldSymbolOffsetDifference(const MCAssembler *Asm,
372 const MCAsmLayout *Layout,
373 const SectionAddrMap *Addrs,
374 bool InSet,
Daniel Dunbar17304b32010-12-17 05:50:33 +0000375 const MCSymbolRefExpr *&A,
376 const MCSymbolRefExpr *&B,
377 int64_t &Addend) {
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000378 if (!A || !B)
379 return;
Daniel Dunbar17304b32010-12-17 05:50:33 +0000380
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000381 const MCSymbol &SA = A->getSymbol();
382 const MCSymbol &SB = B->getSymbol();
383
384 if (SA.isUndefined() || SB.isUndefined())
385 return;
386
387 if (!Asm->getWriter().IsSymbolRefDifferenceFullyResolved(*Asm, A, B, InSet))
388 return;
389
390 MCSymbolData &AD = Asm->getSymbolData(SA);
391 MCSymbolData &BD = Asm->getSymbolData(SB);
392
393 if (AD.getFragment() == BD.getFragment()) {
394 Addend += (AD.getOffset() - BD.getOffset());
Daniel Dunbar17304b32010-12-17 05:50:33 +0000395
Owen Andersonce3a1ba2011-03-21 23:13:43 +0000396 // Pointers to Thumb symbols need to have their low-bit set to allow
397 // for interworking.
398 if (Asm->isThumbFunc(&SA))
399 Addend |= 1;
400
Daniel Dunbar17304b32010-12-17 05:50:33 +0000401 // Clear the symbol expr pointers to indicate we have folded these
402 // operands.
403 A = B = 0;
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000404 return;
Daniel Dunbar17304b32010-12-17 05:50:33 +0000405 }
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000406
407 if (!Layout)
408 return;
409
410 const MCSectionData &SecA = *AD.getFragment()->getParent();
411 const MCSectionData &SecB = *BD.getFragment()->getParent();
412
413 if ((&SecA != &SecB) && !Addrs)
414 return;
415
416 // Eagerly evaluate.
417 Addend += (Layout->getSymbolOffset(&Asm->getSymbolData(A->getSymbol())) -
418 Layout->getSymbolOffset(&Asm->getSymbolData(B->getSymbol())));
419 if (Addrs && (&SecA != &SecB))
420 Addend += (Addrs->lookup(&SecA) - Addrs->lookup(&SecB));
421
Jim Grosbach0db12412012-02-24 05:12:18 +0000422 // Pointers to Thumb symbols need to have their low-bit set to allow
423 // for interworking.
424 if (Asm->isThumbFunc(&SA))
425 Addend |= 1;
426
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000427 // Clear the symbol expr pointers to indicate we have folded these
428 // operands.
429 A = B = 0;
Daniel Dunbar17304b32010-12-17 05:50:33 +0000430}
431
Daniel Dunbardd18e282010-12-16 18:36:25 +0000432/// \brief Evaluate the result of an add between (conceptually) two MCValues.
433///
434/// This routine conceptually attempts to construct an MCValue:
435/// Result = (Result_A - Result_B + Result_Cst)
436/// from two MCValue's LHS and RHS where
437/// Result = LHS + RHS
438/// and
439/// Result = (LHS_A - LHS_B + LHS_Cst) + (RHS_A - RHS_B + RHS_Cst).
440///
441/// This routine attempts to aggresively fold the operands such that the result
442/// is representable in an MCValue, but may not always succeed.
443///
444/// \returns True on success, false if the result is not representable in an
445/// MCValue.
Rafael Espindolad0764822010-12-18 03:57:21 +0000446
447/// NOTE: It is really important to have both the Asm and Layout arguments.
448/// They might look redundant, but this function can be used before layout
449/// is done (see the object streamer for example) and having the Asm argument
Rafael Espindolaf705a7e2010-12-18 04:01:45 +0000450/// lets us avoid relaxations early.
Rafael Espindolad0764822010-12-18 03:57:21 +0000451static bool EvaluateSymbolicAdd(const MCAssembler *Asm,
452 const MCAsmLayout *Layout,
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000453 const SectionAddrMap *Addrs,
Rafael Espindola32a006e2010-12-03 00:55:40 +0000454 bool InSet,
Rafael Espindolaf230df92010-10-16 18:23:53 +0000455 const MCValue &LHS,const MCSymbolRefExpr *RHS_A,
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000456 const MCSymbolRefExpr *RHS_B, int64_t RHS_Cst,
Daniel Dunbar35976042009-06-30 02:08:27 +0000457 MCValue &Res) {
Daniel Dunbarf2ed62d2010-12-17 01:07:20 +0000458 // FIXME: This routine (and other evaluation parts) are *incredibly* sloppy
459 // about dealing with modifiers. This will ultimately bite us, one day.
460 const MCSymbolRefExpr *LHS_A = LHS.getSymA();
461 const MCSymbolRefExpr *LHS_B = LHS.getSymB();
462 int64_t LHS_Cst = LHS.getConstant();
463
464 // Fold the result constant immediately.
465 int64_t Result_Cst = LHS_Cst + RHS_Cst;
466
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000467 assert((!Layout || Asm) &&
468 "Must have an assembler object if layout is given!");
469
Daniel Dunbar17304b32010-12-17 05:50:33 +0000470 // If we have a layout, we can fold resolved differences.
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000471 if (Asm) {
Daniel Dunbar17304b32010-12-17 05:50:33 +0000472 // First, fold out any differences which are fully resolved. By
473 // reassociating terms in
474 // Result = (LHS_A - LHS_B + LHS_Cst) + (RHS_A - RHS_B + RHS_Cst).
475 // we have the four possible differences:
476 // (LHS_A - LHS_B),
477 // (LHS_A - RHS_B),
478 // (RHS_A - LHS_B),
479 // (RHS_A - RHS_B).
Chris Lattner7a2bdde2011-04-15 05:18:47 +0000480 // Since we are attempting to be as aggressive as possible about folding, we
Daniel Dunbar17304b32010-12-17 05:50:33 +0000481 // attempt to evaluate each possible alternative.
Rafael Espindolac5b0e442010-12-19 04:18:56 +0000482 AttemptToFoldSymbolOffsetDifference(Asm, Layout, Addrs, InSet, LHS_A, LHS_B,
483 Result_Cst);
484 AttemptToFoldSymbolOffsetDifference(Asm, Layout, Addrs, InSet, LHS_A, RHS_B,
485 Result_Cst);
486 AttemptToFoldSymbolOffsetDifference(Asm, Layout, Addrs, InSet, RHS_A, LHS_B,
487 Result_Cst);
488 AttemptToFoldSymbolOffsetDifference(Asm, Layout, Addrs, InSet, RHS_A, RHS_B,
489 Result_Cst);
Daniel Dunbar17304b32010-12-17 05:50:33 +0000490 }
491
Daniel Dunbarf2ed62d2010-12-17 01:07:20 +0000492 // We can't represent the addition or subtraction of two symbols.
493 if ((LHS_A && RHS_A) || (LHS_B && RHS_B))
Daniel Dunbar35976042009-06-30 02:08:27 +0000494 return false;
495
Daniel Dunbar17304b32010-12-17 05:50:33 +0000496 // At this point, we have at most one additive symbol and one subtractive
497 // symbol -- find them.
Daniel Dunbarf2ed62d2010-12-17 01:07:20 +0000498 const MCSymbolRefExpr *A = LHS_A ? LHS_A : RHS_A;
499 const MCSymbolRefExpr *B = LHS_B ? LHS_B : RHS_B;
Daniel Dunbar17304b32010-12-17 05:50:33 +0000500
501 // If we have a negated symbol, then we must have also have a non-negated
502 // symbol in order to encode the expression.
503 if (B && !A)
504 return false;
Rafael Espindolaf230df92010-10-16 18:23:53 +0000505
Daniel Dunbarf2ed62d2010-12-17 01:07:20 +0000506 Res = MCValue::get(A, B, Result_Cst);
Daniel Dunbar35976042009-06-30 02:08:27 +0000507 return true;
508}
509
Rafael Espindolad0764822010-12-18 03:57:21 +0000510bool MCExpr::EvaluateAsRelocatable(MCValue &Res,
Rafael Espindola33a38a12010-12-22 16:11:57 +0000511 const MCAsmLayout &Layout) const {
512 return EvaluateAsRelocatableImpl(Res, &Layout.getAssembler(), &Layout,
513 0, false);
Rafael Espindolad0764822010-12-18 03:57:21 +0000514}
515
Rafael Espindolaf230df92010-10-16 18:23:53 +0000516bool MCExpr::EvaluateAsRelocatableImpl(MCValue &Res,
Rafael Espindolad0764822010-12-18 03:57:21 +0000517 const MCAssembler *Asm,
Rafael Espindolaf230df92010-10-16 18:23:53 +0000518 const MCAsmLayout *Layout,
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000519 const SectionAddrMap *Addrs,
Rafael Espindolaf230df92010-10-16 18:23:53 +0000520 bool InSet) const {
Daniel Dunbarff547842010-03-23 23:47:14 +0000521 ++stats::MCExprEvaluate;
522
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000523 switch (getKind()) {
Chris Lattner5d917a82010-02-08 19:41:07 +0000524 case Target:
Daniel Dunbarf82f4492010-03-11 02:28:59 +0000525 return cast<MCTargetExpr>(this)->EvaluateAsRelocatableImpl(Res, Layout);
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000526
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000527 case Constant:
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000528 Res = MCValue::get(cast<MCConstantExpr>(this)->getValue());
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000529 return true;
530
531 case SymbolRef: {
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000532 const MCSymbolRefExpr *SRE = cast<MCSymbolRefExpr>(this);
533 const MCSymbol &Sym = SRE->getSymbol();
Daniel Dunbar7c3600d2009-10-16 01:33:57 +0000534
535 // Evaluate recursively if this is a variable.
Rafael Espindola94ed5fc2010-11-15 16:33:49 +0000536 if (Sym.isVariable() && SRE->getKind() == MCSymbolRefExpr::VK_None) {
Rafael Espindolad0764822010-12-18 03:57:21 +0000537 bool Ret = Sym.getVariableValue()->EvaluateAsRelocatableImpl(Res, Asm,
538 Layout,
539 Addrs,
540 true);
Rafael Espindola94ed5fc2010-11-15 16:33:49 +0000541 // If we failed to simplify this to a constant, let the target
542 // handle it.
543 if (Ret && !Res.getSymA() && !Res.getSymB())
544 return true;
545 }
Daniel Dunbar7c3600d2009-10-16 01:33:57 +0000546
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000547 Res = MCValue::get(SRE, 0, 0);
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000548 return true;
549 }
550
551 case Unary: {
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000552 const MCUnaryExpr *AUE = cast<MCUnaryExpr>(this);
Daniel Dunbar15d17072009-06-30 01:49:52 +0000553 MCValue Value;
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000554
Rafael Espindolad0764822010-12-18 03:57:21 +0000555 if (!AUE->getSubExpr()->EvaluateAsRelocatableImpl(Value, Asm, Layout,
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000556 Addrs, InSet))
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000557 return false;
558
559 switch (AUE->getOpcode()) {
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000560 case MCUnaryExpr::LNot:
Daniel Dunbar80f62d02009-07-01 06:48:00 +0000561 if (!Value.isAbsolute())
Daniel Dunbar15d17072009-06-30 01:49:52 +0000562 return false;
563 Res = MCValue::get(!Value.getConstant());
564 break;
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000565 case MCUnaryExpr::Minus:
Daniel Dunbar15d17072009-06-30 01:49:52 +0000566 /// -(a - b + const) ==> (b - a - const)
Daniel Dunbarb27a41b2009-08-11 17:47:52 +0000567 if (Value.getSymA() && !Value.getSymB())
Daniel Dunbar15d17072009-06-30 01:49:52 +0000568 return false;
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000569 Res = MCValue::get(Value.getSymB(), Value.getSymA(),
570 -Value.getConstant());
Daniel Dunbar15d17072009-06-30 01:49:52 +0000571 break;
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000572 case MCUnaryExpr::Not:
Daniel Dunbar80f62d02009-07-01 06:48:00 +0000573 if (!Value.isAbsolute())
Daniel Dunbar15d17072009-06-30 01:49:52 +0000574 return false;
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000575 Res = MCValue::get(~Value.getConstant());
Daniel Dunbar15d17072009-06-30 01:49:52 +0000576 break;
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000577 case MCUnaryExpr::Plus:
Daniel Dunbar15d17072009-06-30 01:49:52 +0000578 Res = Value;
579 break;
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000580 }
581
582 return true;
583 }
584
585 case Binary: {
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000586 const MCBinaryExpr *ABE = cast<MCBinaryExpr>(this);
Daniel Dunbar15d17072009-06-30 01:49:52 +0000587 MCValue LHSValue, RHSValue;
Daniel Dunbar9a1d2002010-03-18 00:59:10 +0000588
Rafael Espindolad0764822010-12-18 03:57:21 +0000589 if (!ABE->getLHS()->EvaluateAsRelocatableImpl(LHSValue, Asm, Layout,
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000590 Addrs, InSet) ||
Rafael Espindolad0764822010-12-18 03:57:21 +0000591 !ABE->getRHS()->EvaluateAsRelocatableImpl(RHSValue, Asm, Layout,
Rafael Espindola85f2ecc2010-12-07 00:27:36 +0000592 Addrs, InSet))
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000593 return false;
594
Daniel Dunbar15d17072009-06-30 01:49:52 +0000595 // We only support a few operations on non-constant expressions, handle
596 // those first.
Daniel Dunbar80f62d02009-07-01 06:48:00 +0000597 if (!LHSValue.isAbsolute() || !RHSValue.isAbsolute()) {
Daniel Dunbar15d17072009-06-30 01:49:52 +0000598 switch (ABE->getOpcode()) {
599 default:
600 return false;
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000601 case MCBinaryExpr::Sub:
Daniel Dunbar35976042009-06-30 02:08:27 +0000602 // Negate RHS and add.
Rafael Espindolad0764822010-12-18 03:57:21 +0000603 return EvaluateSymbolicAdd(Asm, Layout, Addrs, InSet, LHSValue,
Daniel Dunbar35976042009-06-30 02:08:27 +0000604 RHSValue.getSymB(), RHSValue.getSymA(),
605 -RHSValue.getConstant(),
606 Res);
607
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000608 case MCBinaryExpr::Add:
Rafael Espindolad0764822010-12-18 03:57:21 +0000609 return EvaluateSymbolicAdd(Asm, Layout, Addrs, InSet, LHSValue,
Daniel Dunbar35976042009-06-30 02:08:27 +0000610 RHSValue.getSymA(), RHSValue.getSymB(),
611 RHSValue.getConstant(),
612 Res);
Daniel Dunbar15d17072009-06-30 01:49:52 +0000613 }
614 }
615
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000616 // FIXME: We need target hooks for the evaluation. It may be limited in
Dan Gohman036c1302010-02-08 23:58:47 +0000617 // width, and gas defines the result of comparisons and right shifts
618 // differently from Apple as.
Daniel Dunbarb79742c2009-06-30 16:02:47 +0000619 int64_t LHS = LHSValue.getConstant(), RHS = RHSValue.getConstant();
620 int64_t Result = 0;
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000621 switch (ABE->getOpcode()) {
Daniel Dunbar28c251b2009-08-31 08:06:59 +0000622 case MCBinaryExpr::Add: Result = LHS + RHS; break;
623 case MCBinaryExpr::And: Result = LHS & RHS; break;
624 case MCBinaryExpr::Div: Result = LHS / RHS; break;
625 case MCBinaryExpr::EQ: Result = LHS == RHS; break;
626 case MCBinaryExpr::GT: Result = LHS > RHS; break;
627 case MCBinaryExpr::GTE: Result = LHS >= RHS; break;
628 case MCBinaryExpr::LAnd: Result = LHS && RHS; break;
629 case MCBinaryExpr::LOr: Result = LHS || RHS; break;
630 case MCBinaryExpr::LT: Result = LHS < RHS; break;
631 case MCBinaryExpr::LTE: Result = LHS <= RHS; break;
632 case MCBinaryExpr::Mod: Result = LHS % RHS; break;
633 case MCBinaryExpr::Mul: Result = LHS * RHS; break;
634 case MCBinaryExpr::NE: Result = LHS != RHS; break;
635 case MCBinaryExpr::Or: Result = LHS | RHS; break;
636 case MCBinaryExpr::Shl: Result = LHS << RHS; break;
637 case MCBinaryExpr::Shr: Result = LHS >> RHS; break;
638 case MCBinaryExpr::Sub: Result = LHS - RHS; break;
639 case MCBinaryExpr::Xor: Result = LHS ^ RHS; break;
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000640 }
641
Daniel Dunbar15d17072009-06-30 01:49:52 +0000642 Res = MCValue::get(Result);
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000643 return true;
644 }
645 }
Daniel Dunbarfc6877a2009-06-29 20:40:36 +0000646
Craig Topper85814382012-02-07 05:05:23 +0000647 llvm_unreachable("Invalid assembly expression kind!");
Daniel Dunbar9643ac52009-08-31 08:07:22 +0000648}
Daniel Dunbar0eab5c42011-04-29 18:00:03 +0000649
650const MCSection *MCExpr::FindAssociatedSection() const {
651 switch (getKind()) {
652 case Target:
653 // We never look through target specific expressions.
654 return cast<MCTargetExpr>(this)->FindAssociatedSection();
655
656 case Constant:
657 return MCSymbol::AbsolutePseudoSection;
658
659 case SymbolRef: {
660 const MCSymbolRefExpr *SRE = cast<MCSymbolRefExpr>(this);
661 const MCSymbol &Sym = SRE->getSymbol();
662
663 if (Sym.isDefined())
664 return &Sym.getSection();
665
666 return 0;
667 }
668
669 case Unary:
670 return cast<MCUnaryExpr>(this)->getSubExpr()->FindAssociatedSection();
671
672 case Binary: {
673 const MCBinaryExpr *BE = cast<MCBinaryExpr>(this);
674 const MCSection *LHS_S = BE->getLHS()->FindAssociatedSection();
675 const MCSection *RHS_S = BE->getRHS()->FindAssociatedSection();
676
677 // If either section is absolute, return the other.
678 if (LHS_S == MCSymbol::AbsolutePseudoSection)
679 return RHS_S;
680 if (RHS_S == MCSymbol::AbsolutePseudoSection)
681 return LHS_S;
682
683 // Otherwise, return the first non-null section.
684 return LHS_S ? LHS_S : RHS_S;
685 }
686 }
687
Craig Topper85814382012-02-07 05:05:23 +0000688 llvm_unreachable("Invalid assembly expression kind!");
Daniel Dunbar0eab5c42011-04-29 18:00:03 +0000689}