1 //===- AMDGPUELFObjectWriter.cpp - AMDGPU ELF Writer ----------------------===// 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 "AMDGPUFixupKinds.h" 10 #include "AMDGPUMCTargetDesc.h" 11 #include "llvm/MC/MCContext.h" 12 #include "llvm/MC/MCELFObjectWriter.h" 13 #include "llvm/MC/MCValue.h" 14 15 using namespace llvm; 16 17 namespace { 18 19 class AMDGPUELFObjectWriter : public MCELFObjectTargetWriter { 20 public: 21 AMDGPUELFObjectWriter(bool Is64Bit, uint8_t OSABI, bool HasRelocationAddend, 22 uint8_t ABIVersion); 23 24 protected: 25 unsigned getRelocType(MCContext &Ctx, const MCValue &Target, 26 const MCFixup &Fixup, bool IsPCRel) const override; 27 }; 28 29 30 } // end anonymous namespace 31 32 AMDGPUELFObjectWriter::AMDGPUELFObjectWriter(bool Is64Bit, 33 uint8_t OSABI, 34 bool HasRelocationAddend, 35 uint8_t ABIVersion) 36 : MCELFObjectTargetWriter(Is64Bit, OSABI, ELF::EM_AMDGPU, 37 HasRelocationAddend, ABIVersion) {} 38 39 unsigned AMDGPUELFObjectWriter::getRelocType(MCContext &Ctx, 40 const MCValue &Target, 41 const MCFixup &Fixup, 42 bool IsPCRel) const { 43 if (const auto *SymA = Target.getSymA()) { 44 // SCRATCH_RSRC_DWORD[01] is a special global variable that represents 45 // the scratch buffer. 46 if (SymA->getSymbol().getName() == "SCRATCH_RSRC_DWORD0" || 47 SymA->getSymbol().getName() == "SCRATCH_RSRC_DWORD1") 48 return ELF::R_AMDGPU_ABS32_LO; 49 } 50 51 switch (Target.getAccessVariant()) { 52 default: 53 break; 54 case MCSymbolRefExpr::VK_GOTPCREL: 55 return ELF::R_AMDGPU_GOTPCREL; 56 case MCSymbolRefExpr::VK_AMDGPU_GOTPCREL32_LO: 57 return ELF::R_AMDGPU_GOTPCREL32_LO; 58 case MCSymbolRefExpr::VK_AMDGPU_GOTPCREL32_HI: 59 return ELF::R_AMDGPU_GOTPCREL32_HI; 60 case MCSymbolRefExpr::VK_AMDGPU_REL32_LO: 61 return ELF::R_AMDGPU_REL32_LO; 62 case MCSymbolRefExpr::VK_AMDGPU_REL32_HI: 63 return ELF::R_AMDGPU_REL32_HI; 64 case MCSymbolRefExpr::VK_AMDGPU_REL64: 65 return ELF::R_AMDGPU_REL64; 66 } 67 68 MCFixupKind Kind = Fixup.getKind(); 69 if (Kind >= FirstLiteralRelocationKind) 70 return Kind - FirstLiteralRelocationKind; 71 switch (Kind) { 72 default: break; 73 case FK_PCRel_4: 74 return ELF::R_AMDGPU_REL32; 75 case FK_Data_4: 76 case FK_SecRel_4: 77 return IsPCRel ? ELF::R_AMDGPU_REL32 : ELF::R_AMDGPU_ABS32; 78 case FK_Data_8: 79 return IsPCRel ? ELF::R_AMDGPU_REL64 : ELF::R_AMDGPU_ABS64; 80 } 81 82 if (Fixup.getTargetKind() == AMDGPU::fixup_si_sopp_br) { 83 const auto *SymA = Target.getSymA(); 84 assert(SymA); 85 86 if (SymA->getSymbol().isUndefined()) { 87 Ctx.reportError(Fixup.getLoc(), Twine("undefined label '") + 88 SymA->getSymbol().getName() + "'"); 89 return ELF::R_AMDGPU_NONE; 90 } 91 return ELF::R_AMDGPU_REL16; 92 } 93 94 llvm_unreachable("unhandled relocation type"); 95 } 96 97 std::unique_ptr<MCObjectTargetWriter> 98 llvm::createAMDGPUELFObjectWriter(bool Is64Bit, uint8_t OSABI, 99 bool HasRelocationAddend, 100 uint8_t ABIVersion) { 101 return std::make_unique<AMDGPUELFObjectWriter>(Is64Bit, OSABI, 102 HasRelocationAddend, 103 ABIVersion); 104 } 105