lib/Target/ARM64/MCTargetDesc/ARM64MCExpr.cpp

   1 //===-- ARM64MCExpr.cpp - ARM64 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 "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"
  24
  25 using namespace llvm;
  26
  27 const ARM64MCExpr *ARM64MCExpr::Create(const MCExpr *Expr, VariantKind Kind,
  28                                        MCContext &Ctx) {
  29   return new (Ctx) ARM64MCExpr(Expr, Kind);
  30 }
  31
  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:";
  67   default:
  68     llvm_unreachable("Invalid ELF symbol kind");
  69   }
  70 }
  71
  72 void ARM64MCExpr::PrintImpl(raw_ostream &OS) const {
  73   if (getKind() != VK_NONE)
  74     OS << getVariantKindName();
  75   OS << *Expr;
  76 }
  77
  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()) {
  83   case MCExpr::Target:
  84     llvm_unreachable("Can't handle nested target expr!");
  85     break;
  86
  87   case MCExpr::Constant:
  88     break;
  89
  90   case MCExpr::Binary: {
  91     const MCBinaryExpr *BE = cast<MCBinaryExpr>(Value);
  92     AddValueSymbolsImpl(BE->getLHS(), Asm);
  93     AddValueSymbolsImpl(BE->getRHS(), Asm);
  94     break;
  95   }
  96
  97   case MCExpr::SymbolRef:
  98     Asm->getOrCreateSymbolData(cast<MCSymbolRefExpr>(Value)->getSymbol());
  99     break;
 100
 101   case MCExpr::Unary:
 102     AddValueSymbolsImpl(cast<MCUnaryExpr>(Value)->getSubExpr(), Asm);
 103     break;
 104   }
 105 }
 106
 107 void ARM64MCExpr::AddValueSymbols(MCAssembler *Asm) const {
 108   AddValueSymbolsImpl(getSubExpr(), Asm);
 109 }
 110
 111 const MCSection *ARM64MCExpr::FindAssociatedSection() const {
 112   llvm_unreachable("FIXME: what goes here?");
 113 }
 114
 115 bool ARM64MCExpr::EvaluateAsRelocatableImpl(MCValue &Res,
 116                                             const MCAsmLayout *Layout) const {
 117   if (!getSubExpr()->EvaluateAsRelocatable(Res, Layout))
 118     return false;
 119
 120   Res =
 121       MCValue::get(Res.getSymA(), Res.getSymB(), Res.getConstant(), getKind());
 122
 123   return true;
 124 }
 125
 126 static void fixELFSymbolsInTLSFixupsImpl(const MCExpr *Expr, MCAssembler &Asm) {
 127   switch (Expr->getKind()) {
 128   case MCExpr::Target:
 129     llvm_unreachable("Can't handle nested target expression");
 130     break;
 131   case MCExpr::Constant:
 132     break;
 133
 134   case MCExpr::Binary: {
 135     const MCBinaryExpr *BE = cast<MCBinaryExpr>(Expr);
 136     fixELFSymbolsInTLSFixupsImpl(BE->getLHS(), Asm);
 137     fixELFSymbolsInTLSFixupsImpl(BE->getRHS(), Asm);
 138     break;
 139   }
 140
 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);
 147     break;
 148   }
 149
 150   case MCExpr::Unary:
 151     fixELFSymbolsInTLSFixupsImpl(cast<MCUnaryExpr>(Expr)->getSubExpr(), Asm);
 152     break;
 153   }
 154 }
 155
 156 void ARM64MCExpr::fixELFSymbolsInTLSFixups(MCAssembler &Asm) const {
 157   switch (getSymbolLoc(Kind)) {
 158   default:
 159     return;
 160   case VK_DTPREL:
 161   case VK_GOTTPREL:
 162   case VK_TPREL:
 163   case VK_TLSDESC:
 164     break;
 165   }
 166
 167   fixELFSymbolsInTLSFixupsImpl(getSubExpr(), Asm);
 168 }