xref: /freebsd/contrib/llvm-project/llvm/lib/Target/SystemZ/MCTargetDesc/SystemZMCCodeEmitter.cpp (revision a0ca4af9455b844c5e094fc1b09b1390ffa979fc)
1 //===-- SystemZMCCodeEmitter.cpp - Convert SystemZ code to machine code ---===//
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 // This file implements the SystemZMCCodeEmitter class.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "MCTargetDesc/SystemZMCFixups.h"
14 #include "MCTargetDesc/SystemZMCTargetDesc.h"
15 #include "llvm/ADT/SmallVector.h"
16 #include "llvm/MC/MCCodeEmitter.h"
17 #include "llvm/MC/MCContext.h"
18 #include "llvm/MC/MCExpr.h"
19 #include "llvm/MC/MCFixup.h"
20 #include "llvm/MC/MCInst.h"
21 #include "llvm/MC/MCInstrInfo.h"
22 #include "llvm/MC/MCRegisterInfo.h"
23 #include "llvm/MC/MCSubtargetInfo.h"
24 #include "llvm/Support/ErrorHandling.h"
25 #include <cassert>
26 #include <cstdint>
27 
28 using namespace llvm;
29 
30 #define DEBUG_TYPE "mccodeemitter"
31 
32 namespace {
33 
34 class SystemZMCCodeEmitter : public MCCodeEmitter {
35   const MCInstrInfo &MCII;
36   MCContext &Ctx;
37 
38 public:
39   SystemZMCCodeEmitter(const MCInstrInfo &MCII, MCContext &Ctx)
40       : MCII(MCII), Ctx(Ctx) {}
41 
42   ~SystemZMCCodeEmitter() override = default;
43 
44   // OVerride MCCodeEmitter.
45   void encodeInstruction(const MCInst &MI, SmallVectorImpl<char> &CB,
46                          SmallVectorImpl<MCFixup> &Fixups,
47                          const MCSubtargetInfo &STI) const override;
48 
49 private:
50   // Automatically generated by TableGen.
51   uint64_t getBinaryCodeForInstr(const MCInst &MI,
52                                  SmallVectorImpl<MCFixup> &Fixups,
53                                  const MCSubtargetInfo &STI) const;
54   uint32_t getOperandBitOffset(const MCInst &MI, unsigned OpNum,
55                                const MCSubtargetInfo &STI) const;
56 
57   // Called by the TableGen code to get the binary encoding of operand
58   // MO in MI.  Fixups is the list of fixups against MI.
59   uint64_t getMachineOpValue(const MCInst &MI, const MCOperand &MO,
60                              SmallVectorImpl<MCFixup> &Fixups,
61                              const MCSubtargetInfo &STI) const;
62 
63   // Return the encoded immediate value for the OpNum operand. If it is a
64   // symbol, add a fixup for it and return 0.
65   template <SystemZ::FixupKind Kind>
66   uint64_t getImmOpValue(const MCInst &MI, unsigned OpNum,
67                          SmallVectorImpl<MCFixup> &Fixups,
68                          const MCSubtargetInfo &STI) const;
69 
70   // Called by the TableGen code to get the binary encoding of a length value.
71   // Length values are encoded by subtracting 1 from the actual value.
72   template <SystemZ::FixupKind Kind>
73   uint64_t getLenEncoding(const MCInst &MI, unsigned OpNum,
74                           SmallVectorImpl<MCFixup> &Fixups,
75                           const MCSubtargetInfo &STI) const;
76 
77   // Operand OpNum of MI needs a PC-relative fixup of kind Kind at
78   // Offset bytes from the start of MI.  Add the fixup to Fixups
79   // and return the in-place addend, which since we're a RELA target
80   // is always 0.  If AllowTLS is true and optional operand OpNum + 1
81   // is present, also emit a TLS call fixup for it.
82   uint64_t getPCRelEncoding(const MCInst &MI, unsigned OpNum,
83                             SmallVectorImpl<MCFixup> &Fixups,
84                             unsigned Kind, int64_t Offset,
85                             bool AllowTLS) const;
86 
87   uint64_t getPC16DBLEncoding(const MCInst &MI, unsigned OpNum,
88                               SmallVectorImpl<MCFixup> &Fixups,
89                               const MCSubtargetInfo &STI) const {
90     return getPCRelEncoding(MI, OpNum, Fixups,
91                             SystemZ::FK_390_PC16DBL, 2, false);
92   }
93   uint64_t getPC32DBLEncoding(const MCInst &MI, unsigned OpNum,
94                               SmallVectorImpl<MCFixup> &Fixups,
95                               const MCSubtargetInfo &STI) const {
96     return getPCRelEncoding(MI, OpNum, Fixups,
97                             SystemZ::FK_390_PC32DBL, 2, false);
98   }
99   uint64_t getPC16DBLTLSEncoding(const MCInst &MI, unsigned OpNum,
100                                  SmallVectorImpl<MCFixup> &Fixups,
101                                  const MCSubtargetInfo &STI) const {
102     return getPCRelEncoding(MI, OpNum, Fixups,
103                             SystemZ::FK_390_PC16DBL, 2, true);
104   }
105   uint64_t getPC32DBLTLSEncoding(const MCInst &MI, unsigned OpNum,
106                                  SmallVectorImpl<MCFixup> &Fixups,
107                                  const MCSubtargetInfo &STI) const {
108     return getPCRelEncoding(MI, OpNum, Fixups,
109                             SystemZ::FK_390_PC32DBL, 2, true);
110   }
111   uint64_t getPC12DBLBPPEncoding(const MCInst &MI, unsigned OpNum,
112                                  SmallVectorImpl<MCFixup> &Fixups,
113                                  const MCSubtargetInfo &STI) const {
114     return getPCRelEncoding(MI, OpNum, Fixups,
115                             SystemZ::FK_390_PC12DBL, 1, false);
116   }
117   uint64_t getPC16DBLBPPEncoding(const MCInst &MI, unsigned OpNum,
118                                  SmallVectorImpl<MCFixup> &Fixups,
119                                  const MCSubtargetInfo &STI) const {
120     return getPCRelEncoding(MI, OpNum, Fixups,
121                             SystemZ::FK_390_PC16DBL, 4, false);
122   }
123   uint64_t getPC24DBLBPPEncoding(const MCInst &MI, unsigned OpNum,
124                                  SmallVectorImpl<MCFixup> &Fixups,
125                                  const MCSubtargetInfo &STI) const {
126     return getPCRelEncoding(MI, OpNum, Fixups,
127                             SystemZ::FK_390_PC24DBL, 3, false);
128   }
129 };
130 
131 } // end anonymous namespace
132 
133 void SystemZMCCodeEmitter::encodeInstruction(const MCInst &MI,
134                                              SmallVectorImpl<char> &CB,
135                                              SmallVectorImpl<MCFixup> &Fixups,
136                                              const MCSubtargetInfo &STI) const {
137   uint64_t Bits = getBinaryCodeForInstr(MI, Fixups, STI);
138   unsigned Size = MCII.get(MI.getOpcode()).getSize();
139   // Big-endian insertion of Size bytes.
140   unsigned ShiftValue = (Size * 8) - 8;
141   for (unsigned I = 0; I != Size; ++I) {
142     CB.push_back(uint8_t(Bits >> ShiftValue));
143     ShiftValue -= 8;
144   }
145 }
146 
147 uint64_t SystemZMCCodeEmitter::
148 getMachineOpValue(const MCInst &MI, const MCOperand &MO,
149                   SmallVectorImpl<MCFixup> &Fixups,
150                   const MCSubtargetInfo &STI) const {
151   if (MO.isReg())
152     return Ctx.getRegisterInfo()->getEncodingValue(MO.getReg());
153   // SystemZAsmParser::parseAnyRegister() produces KindImm when registers are
154   // specified as integers.
155   if (MO.isImm())
156     return static_cast<uint64_t>(MO.getImm());
157   llvm_unreachable("Unexpected operand type!");
158 }
159 
160 template <SystemZ::FixupKind Kind>
161 uint64_t SystemZMCCodeEmitter::getImmOpValue(const MCInst &MI, unsigned OpNum,
162                                              SmallVectorImpl<MCFixup> &Fixups,
163                                              const MCSubtargetInfo &STI) const {
164   const MCOperand &MO = MI.getOperand(OpNum);
165   if (MO.isImm())
166     return static_cast<uint64_t>(MO.getImm());
167   if (MO.isExpr()) {
168     unsigned MIBitSize = MCII.get(MI.getOpcode()).getSize() * 8;
169     uint32_t RawBitOffset = getOperandBitOffset(MI, OpNum, STI);
170     unsigned OpBitSize =
171         SystemZ::MCFixupKindInfos[Kind - FirstTargetFixupKind].TargetSize;
172     uint32_t BitOffset = MIBitSize - RawBitOffset - OpBitSize;
173     Fixups.push_back(MCFixup::create(BitOffset >> 3, MO.getExpr(),
174                                      (MCFixupKind)Kind, MI.getLoc()));
175     assert(Fixups.size() <= 2 && "More than two memory operands in MI?");
176     return 0;
177   }
178   llvm_unreachable("Unexpected operand type!");
179 }
180 
181 template <SystemZ::FixupKind Kind>
182 uint64_t
183 SystemZMCCodeEmitter::getLenEncoding(const MCInst &MI, unsigned OpNum,
184                                      SmallVectorImpl<MCFixup> &Fixups,
185                                      const MCSubtargetInfo &STI) const {
186   return getImmOpValue<Kind>(MI, OpNum, Fixups, STI) - 1;
187 }
188 
189 uint64_t
190 SystemZMCCodeEmitter::getPCRelEncoding(const MCInst &MI, unsigned OpNum,
191                                        SmallVectorImpl<MCFixup> &Fixups,
192                                        unsigned Kind, int64_t Offset,
193                                        bool AllowTLS) const {
194   SMLoc Loc = MI.getLoc();
195   const MCOperand &MO = MI.getOperand(OpNum);
196   const MCExpr *Expr;
197   if (MO.isImm())
198     Expr = MCConstantExpr::create(MO.getImm() + Offset, Ctx);
199   else {
200     Expr = MO.getExpr();
201     if (Offset) {
202       // The operand value is relative to the start of MI, but the fixup
203       // is relative to the operand field itself, which is Offset bytes
204       // into MI.  Add Offset to the relocation value to cancel out
205       // this difference.
206       const MCExpr *OffsetExpr = MCConstantExpr::create(Offset, Ctx);
207       Expr = MCBinaryExpr::createAdd(Expr, OffsetExpr, Ctx);
208     }
209   }
210   Fixups.push_back(MCFixup::create(Offset, Expr, (MCFixupKind)Kind, Loc));
211 
212   // Output the fixup for the TLS marker if present.
213   if (AllowTLS && OpNum + 1 < MI.getNumOperands()) {
214     const MCOperand &MOTLS = MI.getOperand(OpNum + 1);
215     Fixups.push_back(MCFixup::create(
216         0, MOTLS.getExpr(), (MCFixupKind)SystemZ::FK_390_TLS_CALL, Loc));
217   }
218   return 0;
219 }
220 
221 #define GET_OPERAND_BIT_OFFSET
222 #include "SystemZGenMCCodeEmitter.inc"
223 
224 MCCodeEmitter *llvm::createSystemZMCCodeEmitter(const MCInstrInfo &MCII,
225                                                 MCContext &Ctx) {
226   return new SystemZMCCodeEmitter(MCII, Ctx);
227 }
228