xref: /llvm-project-15.0.7/lld/ELF/Arch/AMDGPU.cpp (revision 2e97d2aa)
1 //===- AMDGPU.cpp ---------------------------------------------------------===//
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 "InputFiles.h"
10 #include "Symbols.h"
11 #include "Target.h"
12 #include "lld/Common/ErrorHandler.h"
13 #include "llvm/Object/ELF.h"
14 #include "llvm/Support/Endian.h"
15 
16 using namespace llvm;
17 using namespace llvm::object;
18 using namespace llvm::support::endian;
19 using namespace llvm::ELF;
20 using namespace lld;
21 using namespace lld::elf;
22 
23 namespace {
24 class AMDGPU final : public TargetInfo {
25 public:
26   AMDGPU();
27   uint32_t calcEFlags() const override;
28   void relocateOne(uint8_t *Loc, RelType Type, uint64_t Val) const override;
29   RelExpr getRelExpr(RelType Type, const Symbol &S,
30                      const uint8_t *Loc) const override;
31   RelType getDynRel(RelType Type) const override;
32 };
33 } // namespace
34 
35 AMDGPU::AMDGPU() {
36   RelativeRel = R_AMDGPU_RELATIVE64;
37   GotRel = R_AMDGPU_ABS64;
38   NoneRel = R_AMDGPU_NONE;
39   SymbolicRel = R_AMDGPU_ABS64;
40 }
41 
42 static uint32_t getEFlags(InputFile *File) {
43   return cast<ObjFile<ELF64LE>>(File)->getObj().getHeader()->e_flags;
44 }
45 
46 uint32_t AMDGPU::calcEFlags() const {
47   assert(!ObjectFiles.empty());
48   uint32_t Ret = getEFlags(ObjectFiles[0]);
49 
50   // Verify that all input files have the same e_flags.
51   for (InputFile *F : makeArrayRef(ObjectFiles).slice(1)) {
52     if (Ret == getEFlags(F))
53       continue;
54     error("incompatible e_flags: " + toString(F));
55     return 0;
56   }
57   return Ret;
58 }
59 
60 void AMDGPU::relocateOne(uint8_t *Loc, RelType Type, uint64_t Val) const {
61   switch (Type) {
62   case R_AMDGPU_ABS32:
63   case R_AMDGPU_GOTPCREL:
64   case R_AMDGPU_GOTPCREL32_LO:
65   case R_AMDGPU_REL32:
66   case R_AMDGPU_REL32_LO:
67     write32le(Loc, Val);
68     break;
69   case R_AMDGPU_ABS64:
70   case R_AMDGPU_REL64:
71     write64le(Loc, Val);
72     break;
73   case R_AMDGPU_GOTPCREL32_HI:
74   case R_AMDGPU_REL32_HI:
75     write32le(Loc, Val >> 32);
76     break;
77   default:
78     llvm_unreachable("unknown relocation");
79   }
80 }
81 
82 RelExpr AMDGPU::getRelExpr(RelType Type, const Symbol &S,
83                            const uint8_t *Loc) const {
84   switch (Type) {
85   case R_AMDGPU_ABS32:
86   case R_AMDGPU_ABS64:
87     return R_ABS;
88   case R_AMDGPU_REL32:
89   case R_AMDGPU_REL32_LO:
90   case R_AMDGPU_REL32_HI:
91   case R_AMDGPU_REL64:
92     return R_PC;
93   case R_AMDGPU_GOTPCREL:
94   case R_AMDGPU_GOTPCREL32_LO:
95   case R_AMDGPU_GOTPCREL32_HI:
96     return R_GOT_PC;
97   default:
98     error(getErrorLocation(Loc) + "unknown relocation (" + Twine(Type) +
99           ") against symbol " + toString(S));
100     return R_NONE;
101   }
102 }
103 
104 RelType AMDGPU::getDynRel(RelType Type) const {
105   if (Type == R_AMDGPU_ABS64)
106     return Type;
107   return R_AMDGPU_NONE;
108 }
109 
110 TargetInfo *elf::getAMDGPUTargetInfo() {
111   static AMDGPU Target;
112   return &Target;
113 }
114