Line data    Source code

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