contrib/llvm-project/llvm/lib/Target/AArch64/MCTargetDesc/AArch64WinCOFFObjectWriter.cpp

   1 //= AArch64WinCOFFObjectWriter.cpp - AArch64 Windows COFF Object Writer C++ =//
   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 #include "MCTargetDesc/AArch64FixupKinds.h"
  10 #include "MCTargetDesc/AArch64MCExpr.h"
  11 #include "llvm/ADT/Twine.h"
  12 #include "llvm/BinaryFormat/COFF.h"
  13 #include "llvm/MC/MCAsmBackend.h"
  14 #include "llvm/MC/MCExpr.h"
  15 #include "llvm/MC/MCFixup.h"
  16 #include "llvm/MC/MCFixupKindInfo.h"
  17 #include "llvm/MC/MCObjectWriter.h"
  18 #include "llvm/MC/MCValue.h"
  19 #include "llvm/MC/MCWinCOFFObjectWriter.h"
  20 #include "llvm/Support/ErrorHandling.h"
  21 #include "llvm/Support/raw_ostream.h"
  22 #include <cassert>
  23
  24 using namespace llvm;
  25
  26 namespace {
  27
  28 class AArch64WinCOFFObjectWriter : public MCWinCOFFObjectTargetWriter {
  29 public:
  30   AArch64WinCOFFObjectWriter()
  31       : MCWinCOFFObjectTargetWriter(COFF::IMAGE_FILE_MACHINE_ARM64) {}
  32
  33   ~AArch64WinCOFFObjectWriter() override = default;
  34
  35   unsigned getRelocType(MCContext &Ctx, const MCValue &Target,
  36                         const MCFixup &Fixup, bool IsCrossSection,
  37                         const MCAsmBackend &MAB) const override;
  38
  39   bool recordRelocation(const MCFixup &) const override;
  40 };
  41
  42 } // end anonymous namespace
  43
  44 unsigned AArch64WinCOFFObjectWriter::getRelocType(
  45     MCContext &Ctx, const MCValue &Target, const MCFixup &Fixup,
  46     bool IsCrossSection, const MCAsmBackend &MAB) const {
  47   auto Modifier = Target.isAbsolute() ? MCSymbolRefExpr::VK_None
  48                                       : Target.getSymA()->getKind();
  49   const MCExpr *Expr = Fixup.getValue();
  50
  51   switch (static_cast<unsigned>(Fixup.getKind())) {
  52   default: {
  53     const MCFixupKindInfo &Info = MAB.getFixupKindInfo(Fixup.getKind());
  54     report_fatal_error(Twine("unsupported relocation type: ") + Info.Name);
  55   }
  56
  57   case FK_Data_4:
  58     switch (Modifier) {
  59     default:
  60       return COFF::IMAGE_REL_ARM64_ADDR32;
  61     case MCSymbolRefExpr::VK_COFF_IMGREL32:
  62       return COFF::IMAGE_REL_ARM64_ADDR32NB;
  63     case MCSymbolRefExpr::VK_SECREL:
  64       return COFF::IMAGE_REL_ARM64_SECREL;
  65     }
  66
  67   case FK_Data_8:
  68     return COFF::IMAGE_REL_ARM64_ADDR64;
  69
  70   case FK_SecRel_2:
  71     return COFF::IMAGE_REL_ARM64_SECTION;
  72
  73   case FK_SecRel_4:
  74     return COFF::IMAGE_REL_ARM64_SECREL;
  75
  76   case AArch64::fixup_aarch64_add_imm12:
  77     if (const AArch64MCExpr *A64E = dyn_cast<AArch64MCExpr>(Expr)) {
  78       AArch64MCExpr::VariantKind RefKind = A64E->getKind();
  79       if (RefKind == AArch64MCExpr::VK_SECREL_LO12)
  80         return COFF::IMAGE_REL_ARM64_SECREL_LOW12A;
  81       if (RefKind == AArch64MCExpr::VK_SECREL_HI12)
  82         return COFF::IMAGE_REL_ARM64_SECREL_HIGH12A;
  83     }
  84     return COFF::IMAGE_REL_ARM64_PAGEOFFSET_12A;
  85
  86   case AArch64::fixup_aarch64_ldst_imm12_scale1:
  87   case AArch64::fixup_aarch64_ldst_imm12_scale2:
  88   case AArch64::fixup_aarch64_ldst_imm12_scale4:
  89   case AArch64::fixup_aarch64_ldst_imm12_scale8:
  90   case AArch64::fixup_aarch64_ldst_imm12_scale16:
  91     if (const AArch64MCExpr *A64E = dyn_cast<AArch64MCExpr>(Expr)) {
  92       AArch64MCExpr::VariantKind RefKind = A64E->getKind();
  93       if (RefKind == AArch64MCExpr::VK_SECREL_LO12)
  94         return COFF::IMAGE_REL_ARM64_SECREL_LOW12L;
  95     }
  96     return COFF::IMAGE_REL_ARM64_PAGEOFFSET_12L;
  97
  98   case AArch64::fixup_aarch64_pcrel_adr_imm21:
  99     return COFF::IMAGE_REL_ARM64_REL21;
 100
 101   case AArch64::fixup_aarch64_pcrel_adrp_imm21:
 102     return COFF::IMAGE_REL_ARM64_PAGEBASE_REL21;
 103
 104   case AArch64::fixup_aarch64_pcrel_branch14:
 105     return COFF::IMAGE_REL_ARM64_BRANCH14;
 106
 107   case AArch64::fixup_aarch64_pcrel_branch19:
 108     return COFF::IMAGE_REL_ARM64_BRANCH19;
 109
 110   case AArch64::fixup_aarch64_pcrel_branch26:
 111   case AArch64::fixup_aarch64_pcrel_call26:
 112     return COFF::IMAGE_REL_ARM64_BRANCH26;
 113   }
 114 }
 115
 116 bool AArch64WinCOFFObjectWriter::recordRelocation(const MCFixup &Fixup) const {
 117   return true;
 118 }
 119
 120 namespace llvm {
 121
 122 std::unique_ptr<MCObjectTargetWriter> createAArch64WinCOFFObjectWriter() {
 123   return llvm::make_unique<AArch64WinCOFFObjectWriter>();
 124 }
 125
 126 } // end namespace llvm