lib/Target/AArch64/MCTargetDesc/AArch64MCExpr.cpp

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