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 "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 
50   switch (static_cast<unsigned>(Fixup.getKind())) {
51   default: {
52     const MCFixupKindInfo &Info = MAB.getFixupKindInfo(Fixup.getKind());
53     report_fatal_error(Twine("unsupported relocation type: ") + Info.Name);
54   }
55 
56   case FK_Data_4:
57     switch (Modifier) {
58     default:
59       return COFF::IMAGE_REL_ARM64_ADDR32;
60     case MCSymbolRefExpr::VK_COFF_IMGREL32:
61       return COFF::IMAGE_REL_ARM64_ADDR32NB;
62     case MCSymbolRefExpr::VK_SECREL:
63       return COFF::IMAGE_REL_ARM64_SECREL;
64     }
65 
66   case FK_Data_8:
67     return COFF::IMAGE_REL_ARM64_ADDR64;
68 
69   case FK_SecRel_2:
70     return COFF::IMAGE_REL_ARM64_SECTION;
71 
72   case FK_SecRel_4:
73     return COFF::IMAGE_REL_ARM64_SECREL;
74 
75   case AArch64::fixup_aarch64_add_imm12:
76     return COFF::IMAGE_REL_ARM64_PAGEOFFSET_12A;
77 
78   case AArch64::fixup_aarch64_ldst_imm12_scale1:
79   case AArch64::fixup_aarch64_ldst_imm12_scale2:
80   case AArch64::fixup_aarch64_ldst_imm12_scale4:
81   case AArch64::fixup_aarch64_ldst_imm12_scale8:
82   case AArch64::fixup_aarch64_ldst_imm12_scale16:
83     return COFF::IMAGE_REL_ARM64_PAGEOFFSET_12L;
84 
85   case AArch64::fixup_aarch64_pcrel_adrp_imm21:
86     return COFF::IMAGE_REL_ARM64_PAGEBASE_REL21;
87 
88   case AArch64::fixup_aarch64_pcrel_branch26:
89   case AArch64::fixup_aarch64_pcrel_call26:
90     return COFF::IMAGE_REL_ARM64_BRANCH26;
91   }
92 }
93 
94 bool AArch64WinCOFFObjectWriter::recordRelocation(const MCFixup &Fixup) const {
95   return true;
96 }
97 
98 namespace llvm {
99 
100 std::unique_ptr<MCObjectWriter>
101 createAArch64WinCOFFObjectWriter(raw_pwrite_stream &OS) {
102   auto MOTW = llvm::make_unique<AArch64WinCOFFObjectWriter>();
103   return createWinCOFFObjectWriter(std::move(MOTW), OS);
104 }
105 
106 } // end namespace llvm
107