1 //===-- SystemZMCAsmBackend.cpp - SystemZ assembler backend ---------------===//
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/SystemZMCFixups.h"
10 #include "MCTargetDesc/SystemZMCTargetDesc.h"
11 #include "llvm/ADT/StringSwitch.h"
12 #include "llvm/MC/MCAsmBackend.h"
13 #include "llvm/MC/MCAssembler.h"
14 #include "llvm/MC/MCContext.h"
15 #include "llvm/MC/MCELFObjectWriter.h"
16 #include "llvm/MC/MCFixupKindInfo.h"
17 #include "llvm/MC/MCInst.h"
18 #include "llvm/MC/MCObjectWriter.h"
19 #include "llvm/MC/MCSubtargetInfo.h"
20 
21 using namespace llvm;
22 
23 // Value is a fully-resolved relocation value: Symbol + Addend [- Pivot].
24 // Return the bits that should be installed in a relocation field for
25 // fixup kind Kind.
extractBitsForFixup(MCFixupKind Kind,uint64_t Value,const MCFixup & Fixup,MCContext & Ctx)26 static uint64_t extractBitsForFixup(MCFixupKind Kind, uint64_t Value,
27                                     const MCFixup &Fixup, MCContext &Ctx) {
28   if (Kind < FirstTargetFixupKind)
29     return Value;
30 
31   auto checkFixupInRange = [&](int64_t Min, int64_t Max) -> bool {
32     int64_t SVal = int64_t(Value);
33     if (SVal < Min || SVal > Max) {
34       Ctx.reportError(Fixup.getLoc(), "operand out of range (" + Twine(SVal) +
35                                           " not between " + Twine(Min) +
36                                           " and " + Twine(Max) + ")");
37       return false;
38     }
39     return true;
40   };
41 
42   auto handlePCRelFixupValue = [&](unsigned W) -> uint64_t {
43     if (Value % 2 != 0)
44       Ctx.reportError(Fixup.getLoc(), "Non-even PC relative offset.");
45     if (!checkFixupInRange(minIntN(W) * 2, maxIntN(W) * 2))
46       return 0;
47     return (int64_t)Value / 2;
48   };
49 
50   auto handleImmValue = [&](bool IsSigned, unsigned W) -> uint64_t {
51     if (!(IsSigned ? checkFixupInRange(minIntN(W), maxIntN(W))
52                    : checkFixupInRange(0, maxUIntN(W))))
53       return 0;
54     return Value;
55   };
56 
57   switch (unsigned(Kind)) {
58   case SystemZ::FK_390_PC12DBL:
59     return handlePCRelFixupValue(12);
60   case SystemZ::FK_390_PC16DBL:
61     return handlePCRelFixupValue(16);
62   case SystemZ::FK_390_PC24DBL:
63     return handlePCRelFixupValue(24);
64   case SystemZ::FK_390_PC32DBL:
65     return handlePCRelFixupValue(32);
66 
67   case SystemZ::FK_390_TLS_CALL:
68     return 0;
69 
70   case SystemZ::FK_390_S8Imm:
71     return handleImmValue(true, 8);
72   case SystemZ::FK_390_S16Imm:
73     return handleImmValue(true, 16);
74   case SystemZ::FK_390_S20Imm: {
75     Value = handleImmValue(true, 20);
76     // S20Imm is used only for signed 20-bit displacements.
77     // The high byte of a 20 bit displacement value comes first.
78     uint64_t DLo = Value & 0xfff;
79     uint64_t DHi = (Value >> 12) & 0xff;
80     return (DLo << 8) | DHi;
81   }
82   case SystemZ::FK_390_S32Imm:
83     return handleImmValue(true, 32);
84   case SystemZ::FK_390_U1Imm:
85     return handleImmValue(false, 1);
86   case SystemZ::FK_390_U2Imm:
87     return handleImmValue(false, 2);
88   case SystemZ::FK_390_U3Imm:
89     return handleImmValue(false, 3);
90   case SystemZ::FK_390_U4Imm:
91     return handleImmValue(false, 4);
92   case SystemZ::FK_390_U8Imm:
93     return handleImmValue(false, 8);
94   case SystemZ::FK_390_U12Imm:
95     return handleImmValue(false, 12);
96   case SystemZ::FK_390_U16Imm:
97     return handleImmValue(false, 16);
98   case SystemZ::FK_390_U32Imm:
99     return handleImmValue(false, 32);
100   case SystemZ::FK_390_U48Imm:
101     return handleImmValue(false, 48);
102   }
103 
104   llvm_unreachable("Unknown fixup kind!");
105 }
106 
107 namespace {
108 class SystemZMCAsmBackend : public MCAsmBackend {
109 public:
SystemZMCAsmBackend()110   SystemZMCAsmBackend() : MCAsmBackend(llvm::endianness::big) {}
111 
112   // Override MCAsmBackend
getNumFixupKinds() const113   unsigned getNumFixupKinds() const override {
114     return SystemZ::NumTargetFixupKinds;
115   }
116   std::optional<MCFixupKind> getFixupKind(StringRef Name) const override;
117   const MCFixupKindInfo &getFixupKindInfo(MCFixupKind Kind) const override;
118   bool shouldForceRelocation(const MCAssembler &Asm, const MCFixup &Fixup,
119                              const MCValue &Target,
120                              const MCSubtargetInfo *STI) override;
121   void applyFixup(const MCAssembler &Asm, const MCFixup &Fixup,
122                   const MCValue &Target, MutableArrayRef<char> Data,
123                   uint64_t Value, bool IsResolved,
124                   const MCSubtargetInfo *STI) const override;
fixupNeedsRelaxation(const MCFixup & Fixup,uint64_t Value,const MCRelaxableFragment * Fragment,const MCAsmLayout & Layout) const125   bool fixupNeedsRelaxation(const MCFixup &Fixup, uint64_t Value,
126                             const MCRelaxableFragment *Fragment,
127                             const MCAsmLayout &Layout) const override {
128     return false;
129   }
130   bool writeNopData(raw_ostream &OS, uint64_t Count,
131                     const MCSubtargetInfo *STI) const override;
132 };
133 } // end anonymous namespace
134 
135 std::optional<MCFixupKind>
getFixupKind(StringRef Name) const136 SystemZMCAsmBackend::getFixupKind(StringRef Name) const {
137   unsigned Type = llvm::StringSwitch<unsigned>(Name)
138 #define ELF_RELOC(X, Y) .Case(#X, Y)
139 #include "llvm/BinaryFormat/ELFRelocs/SystemZ.def"
140 #undef ELF_RELOC
141 			.Case("BFD_RELOC_NONE", ELF::R_390_NONE)
142 			.Case("BFD_RELOC_8", ELF::R_390_8)
143 			.Case("BFD_RELOC_16", ELF::R_390_16)
144 			.Case("BFD_RELOC_32", ELF::R_390_32)
145 			.Case("BFD_RELOC_64", ELF::R_390_64)
146 			.Default(-1u);
147   if (Type != -1u)
148     return static_cast<MCFixupKind>(FirstLiteralRelocationKind + Type);
149   return std::nullopt;
150 }
151 
152 const MCFixupKindInfo &
getFixupKindInfo(MCFixupKind Kind) const153 SystemZMCAsmBackend::getFixupKindInfo(MCFixupKind Kind) const {
154   // Fixup kinds from .reloc directive are like R_390_NONE. They
155   // do not require any extra processing.
156   if (Kind >= FirstLiteralRelocationKind)
157     return MCAsmBackend::getFixupKindInfo(FK_NONE);
158 
159   if (Kind < FirstTargetFixupKind)
160     return MCAsmBackend::getFixupKindInfo(Kind);
161 
162   assert(unsigned(Kind - FirstTargetFixupKind) < getNumFixupKinds() &&
163          "Invalid kind!");
164   return SystemZ::MCFixupKindInfos[Kind - FirstTargetFixupKind];
165 }
166 
shouldForceRelocation(const MCAssembler &,const MCFixup & Fixup,const MCValue &,const MCSubtargetInfo * STI)167 bool SystemZMCAsmBackend::shouldForceRelocation(const MCAssembler &,
168                                                 const MCFixup &Fixup,
169                                                 const MCValue &,
170                                                 const MCSubtargetInfo *STI) {
171   return Fixup.getKind() >= FirstLiteralRelocationKind;
172 }
173 
applyFixup(const MCAssembler & Asm,const MCFixup & Fixup,const MCValue & Target,MutableArrayRef<char> Data,uint64_t Value,bool IsResolved,const MCSubtargetInfo * STI) const174 void SystemZMCAsmBackend::applyFixup(const MCAssembler &Asm,
175                                      const MCFixup &Fixup,
176                                      const MCValue &Target,
177                                      MutableArrayRef<char> Data, uint64_t Value,
178                                      bool IsResolved,
179                                      const MCSubtargetInfo *STI) const {
180   MCFixupKind Kind = Fixup.getKind();
181   if (Kind >= FirstLiteralRelocationKind)
182     return;
183   unsigned Offset = Fixup.getOffset();
184   unsigned BitSize = getFixupKindInfo(Kind).TargetSize;
185   unsigned Size = (BitSize + 7) / 8;
186 
187   assert(Offset + Size <= Data.size() && "Invalid fixup offset!");
188 
189   // Big-endian insertion of Size bytes.
190   Value = extractBitsForFixup(Kind, Value, Fixup, Asm.getContext());
191   if (BitSize < 64)
192     Value &= ((uint64_t)1 << BitSize) - 1;
193   unsigned ShiftValue = (Size * 8) - 8;
194   for (unsigned I = 0; I != Size; ++I) {
195     Data[Offset + I] |= uint8_t(Value >> ShiftValue);
196     ShiftValue -= 8;
197   }
198 }
199 
writeNopData(raw_ostream & OS,uint64_t Count,const MCSubtargetInfo * STI) const200 bool SystemZMCAsmBackend::writeNopData(raw_ostream &OS, uint64_t Count,
201                                        const MCSubtargetInfo *STI) const {
202   for (uint64_t I = 0; I != Count; ++I)
203     OS << '\x7';
204   return true;
205 }
206 
207 namespace {
208 class ELFSystemZAsmBackend : public SystemZMCAsmBackend {
209   uint8_t OSABI;
210 
211 public:
ELFSystemZAsmBackend(uint8_t OsABI)212   ELFSystemZAsmBackend(uint8_t OsABI) : SystemZMCAsmBackend(), OSABI(OsABI){};
213 
214   std::unique_ptr<MCObjectTargetWriter>
createObjectTargetWriter() const215   createObjectTargetWriter() const override {
216     return createSystemZELFObjectWriter(OSABI);
217   }
218 };
219 
220 class GOFFSystemZAsmBackend : public SystemZMCAsmBackend {
221 public:
GOFFSystemZAsmBackend()222   GOFFSystemZAsmBackend() : SystemZMCAsmBackend(){};
223 
224   std::unique_ptr<MCObjectTargetWriter>
createObjectTargetWriter() const225   createObjectTargetWriter() const override {
226     return createSystemZGOFFObjectWriter();
227   }
228 };
229 } // namespace
230 
createSystemZMCAsmBackend(const Target & T,const MCSubtargetInfo & STI,const MCRegisterInfo & MRI,const MCTargetOptions & Options)231 MCAsmBackend *llvm::createSystemZMCAsmBackend(const Target &T,
232                                               const MCSubtargetInfo &STI,
233                                               const MCRegisterInfo &MRI,
234                                               const MCTargetOptions &Options) {
235   if (STI.getTargetTriple().isOSzOS()) {
236     return new GOFFSystemZAsmBackend();
237   }
238 
239   uint8_t OSABI =
240       MCELFObjectTargetWriter::getOSABI(STI.getTargetTriple().getOS());
241   return new ELFSystemZAsmBackend(OSABI);
242 }
243