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 127