1 //===-- LanaiMCCodeEmitter.cpp - Convert Lanai 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 LanaiMCCodeEmitter class.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "LanaiAluCode.h"
14 #include "MCTargetDesc/LanaiBaseInfo.h"
15 #include "MCTargetDesc/LanaiFixupKinds.h"
16 #include "MCTargetDesc/LanaiMCExpr.h"
17 #include "llvm/ADT/SmallVector.h"
18 #include "llvm/ADT/Statistic.h"
19 #include "llvm/MC/MCCodeEmitter.h"
20 #include "llvm/MC/MCExpr.h"
21 #include "llvm/MC/MCFixup.h"
22 #include "llvm/MC/MCInst.h"
23 #include "llvm/MC/MCInstrInfo.h"
24 #include "llvm/MC/MCRegisterInfo.h"
25 #include "llvm/MC/MCSubtargetInfo.h"
26 #include "llvm/Support/Casting.h"
27 #include "llvm/Support/EndianStream.h"
28 #include "llvm/Support/raw_ostream.h"
29 #include <cassert>
30 #include <cstdint>
31 
32 #define DEBUG_TYPE "mccodeemitter"
33 
34 STATISTIC(MCNumEmitted, "Number of MC instructions emitted");
35 
36 namespace llvm {
37 
38 namespace {
39 
40 class LanaiMCCodeEmitter : public MCCodeEmitter {
41 public:
LanaiMCCodeEmitter(const MCInstrInfo & MCII,MCContext & C)42   LanaiMCCodeEmitter(const MCInstrInfo &MCII, MCContext &C) {}
43   LanaiMCCodeEmitter(const LanaiMCCodeEmitter &) = delete;
44   void operator=(const LanaiMCCodeEmitter &) = delete;
45   ~LanaiMCCodeEmitter() override = default;
46 
47   // The functions below are called by TableGen generated functions for getting
48   // the binary encoding of instructions/opereands.
49 
50   // getBinaryCodeForInstr - TableGen'erated function for getting the
51   // binary encoding for an instruction.
52   uint64_t getBinaryCodeForInstr(const MCInst &Inst,
53                                  SmallVectorImpl<MCFixup> &Fixups,
54                                  const MCSubtargetInfo &SubtargetInfo) const;
55 
56   // getMachineOpValue - Return binary encoding of operand. If the machine
57   // operand requires relocation, record the relocation and return zero.
58   unsigned getMachineOpValue(const MCInst &Inst, const MCOperand &MCOp,
59                              SmallVectorImpl<MCFixup> &Fixups,
60                              const MCSubtargetInfo &SubtargetInfo) const;
61 
62   unsigned getRiMemoryOpValue(const MCInst &Inst, unsigned OpNo,
63                               SmallVectorImpl<MCFixup> &Fixups,
64                               const MCSubtargetInfo &SubtargetInfo) const;
65 
66   unsigned getRrMemoryOpValue(const MCInst &Inst, unsigned OpNo,
67                               SmallVectorImpl<MCFixup> &Fixups,
68                               const MCSubtargetInfo &SubtargetInfo) const;
69 
70   unsigned getSplsOpValue(const MCInst &Inst, unsigned OpNo,
71                           SmallVectorImpl<MCFixup> &Fixups,
72                           const MCSubtargetInfo &SubtargetInfo) const;
73 
74   unsigned getBranchTargetOpValue(const MCInst &Inst, unsigned OpNo,
75                                   SmallVectorImpl<MCFixup> &Fixups,
76                                   const MCSubtargetInfo &SubtargetInfo) const;
77 
78   void encodeInstruction(const MCInst &Inst, SmallVectorImpl<char> &CB,
79                          SmallVectorImpl<MCFixup> &Fixups,
80                          const MCSubtargetInfo &SubtargetInfo) const override;
81 
82   unsigned adjustPqBitsRmAndRrm(const MCInst &Inst, unsigned Value,
83                                 const MCSubtargetInfo &STI) const;
84 
85   unsigned adjustPqBitsSpls(const MCInst &Inst, unsigned Value,
86                             const MCSubtargetInfo &STI) const;
87 };
88 
89 } // end anonymous namespace
90 
FixupKind(const MCExpr * Expr)91 static Lanai::Fixups FixupKind(const MCExpr *Expr) {
92   if (isa<MCSymbolRefExpr>(Expr))
93     return Lanai::FIXUP_LANAI_21;
94   if (const LanaiMCExpr *McExpr = dyn_cast<LanaiMCExpr>(Expr)) {
95     LanaiMCExpr::VariantKind ExprKind = McExpr->getKind();
96     switch (ExprKind) {
97     case LanaiMCExpr::VK_Lanai_None:
98       return Lanai::FIXUP_LANAI_21;
99     case LanaiMCExpr::VK_Lanai_ABS_HI:
100       return Lanai::FIXUP_LANAI_HI16;
101     case LanaiMCExpr::VK_Lanai_ABS_LO:
102       return Lanai::FIXUP_LANAI_LO16;
103     }
104   }
105   return Lanai::Fixups(0);
106 }
107 
108 // getMachineOpValue - Return binary encoding of operand. If the machine
109 // operand requires relocation, record the relocation and return zero.
getMachineOpValue(const MCInst & Inst,const MCOperand & MCOp,SmallVectorImpl<MCFixup> & Fixups,const MCSubtargetInfo & SubtargetInfo) const110 unsigned LanaiMCCodeEmitter::getMachineOpValue(
111     const MCInst &Inst, const MCOperand &MCOp, SmallVectorImpl<MCFixup> &Fixups,
112     const MCSubtargetInfo &SubtargetInfo) const {
113   if (MCOp.isReg())
114     return getLanaiRegisterNumbering(MCOp.getReg());
115   if (MCOp.isImm())
116     return static_cast<unsigned>(MCOp.getImm());
117 
118   // MCOp must be an expression
119   assert(MCOp.isExpr());
120   const MCExpr *Expr = MCOp.getExpr();
121 
122   // Extract the symbolic reference side of a binary expression.
123   if (Expr->getKind() == MCExpr::Binary) {
124     const MCBinaryExpr *BinaryExpr = static_cast<const MCBinaryExpr *>(Expr);
125     Expr = BinaryExpr->getLHS();
126   }
127 
128   assert(isa<LanaiMCExpr>(Expr) || Expr->getKind() == MCExpr::SymbolRef);
129   // Push fixup (all info is contained within)
130   Fixups.push_back(
131       MCFixup::create(0, MCOp.getExpr(), MCFixupKind(FixupKind(Expr))));
132   return 0;
133 }
134 
135 // Helper function to adjust P and Q bits on load and store instructions.
adjustPqBits(const MCInst & Inst,unsigned Value,unsigned PBitShift,unsigned QBitShift)136 static unsigned adjustPqBits(const MCInst &Inst, unsigned Value,
137                              unsigned PBitShift, unsigned QBitShift) {
138   const MCOperand AluOp = Inst.getOperand(3);
139   unsigned AluCode = AluOp.getImm();
140 
141   // Set the P bit to one iff the immediate is nonzero and not a post-op
142   // instruction.
143   const MCOperand Op2 = Inst.getOperand(2);
144   Value &= ~(1 << PBitShift);
145   if (!LPAC::isPostOp(AluCode) &&
146       ((Op2.isImm() && Op2.getImm() != 0) ||
147        (Op2.isReg() && Op2.getReg() != Lanai::R0) || (Op2.isExpr())))
148     Value |= (1 << PBitShift);
149 
150   // Set the Q bit to one iff it is a post- or pre-op instruction.
151   assert(Inst.getOperand(0).isReg() && Inst.getOperand(1).isReg() &&
152          "Expected register operand.");
153   Value &= ~(1 << QBitShift);
154   if (LPAC::modifiesOp(AluCode) && ((Op2.isImm() && Op2.getImm() != 0) ||
155                                     (Op2.isReg() && Op2.getReg() != Lanai::R0)))
156     Value |= (1 << QBitShift);
157 
158   return Value;
159 }
160 
161 unsigned
adjustPqBitsRmAndRrm(const MCInst & Inst,unsigned Value,const MCSubtargetInfo & STI) const162 LanaiMCCodeEmitter::adjustPqBitsRmAndRrm(const MCInst &Inst, unsigned Value,
163                                          const MCSubtargetInfo &STI) const {
164   return adjustPqBits(Inst, Value, 17, 16);
165 }
166 
167 unsigned
adjustPqBitsSpls(const MCInst & Inst,unsigned Value,const MCSubtargetInfo & STI) const168 LanaiMCCodeEmitter::adjustPqBitsSpls(const MCInst &Inst, unsigned Value,
169                                      const MCSubtargetInfo &STI) const {
170   return adjustPqBits(Inst, Value, 11, 10);
171 }
172 
encodeInstruction(const MCInst & Inst,SmallVectorImpl<char> & CB,SmallVectorImpl<MCFixup> & Fixups,const MCSubtargetInfo & SubtargetInfo) const173 void LanaiMCCodeEmitter::encodeInstruction(
174     const MCInst &Inst, SmallVectorImpl<char> &CB,
175     SmallVectorImpl<MCFixup> &Fixups,
176     const MCSubtargetInfo &SubtargetInfo) const {
177   // Get instruction encoding and emit it
178   unsigned Value = getBinaryCodeForInstr(Inst, Fixups, SubtargetInfo);
179   ++MCNumEmitted; // Keep track of the number of emitted insns.
180 
181   support::endian::write<uint32_t>(CB, Value, llvm::endianness::big);
182 }
183 
184 // Encode Lanai Memory Operand
getRiMemoryOpValue(const MCInst & Inst,unsigned OpNo,SmallVectorImpl<MCFixup> & Fixups,const MCSubtargetInfo & SubtargetInfo) const185 unsigned LanaiMCCodeEmitter::getRiMemoryOpValue(
186     const MCInst &Inst, unsigned OpNo, SmallVectorImpl<MCFixup> &Fixups,
187     const MCSubtargetInfo &SubtargetInfo) const {
188   unsigned Encoding;
189   const MCOperand Op1 = Inst.getOperand(OpNo + 0);
190   const MCOperand Op2 = Inst.getOperand(OpNo + 1);
191   const MCOperand AluOp = Inst.getOperand(OpNo + 2);
192 
193   assert(Op1.isReg() && "First operand is not register.");
194   assert((Op2.isImm() || Op2.isExpr()) &&
195          "Second operand is neither an immediate nor an expression.");
196   assert((LPAC::getAluOp(AluOp.getImm()) == LPAC::ADD) &&
197          "Register immediate only supports addition operator");
198 
199   Encoding = (getLanaiRegisterNumbering(Op1.getReg()) << 18);
200   if (Op2.isImm()) {
201     assert(isInt<16>(Op2.getImm()) &&
202            "Constant value truncated (limited to 16-bit)");
203 
204     Encoding |= (Op2.getImm() & 0xffff);
205     if (Op2.getImm() != 0) {
206       if (LPAC::isPreOp(AluOp.getImm()))
207         Encoding |= (0x3 << 16);
208       if (LPAC::isPostOp(AluOp.getImm()))
209         Encoding |= (0x1 << 16);
210     }
211   } else
212     getMachineOpValue(Inst, Op2, Fixups, SubtargetInfo);
213 
214   return Encoding;
215 }
216 
getRrMemoryOpValue(const MCInst & Inst,unsigned OpNo,SmallVectorImpl<MCFixup> & Fixups,const MCSubtargetInfo & SubtargetInfo) const217 unsigned LanaiMCCodeEmitter::getRrMemoryOpValue(
218     const MCInst &Inst, unsigned OpNo, SmallVectorImpl<MCFixup> &Fixups,
219     const MCSubtargetInfo &SubtargetInfo) const {
220   unsigned Encoding;
221   const MCOperand Op1 = Inst.getOperand(OpNo + 0);
222   const MCOperand Op2 = Inst.getOperand(OpNo + 1);
223   const MCOperand AluMCOp = Inst.getOperand(OpNo + 2);
224 
225   assert(Op1.isReg() && "First operand is not register.");
226   Encoding = (getLanaiRegisterNumbering(Op1.getReg()) << 15);
227   assert(Op2.isReg() && "Second operand is not register.");
228   Encoding |= (getLanaiRegisterNumbering(Op2.getReg()) << 10);
229 
230   assert(AluMCOp.isImm() && "Third operator is not immediate.");
231   // Set BBB
232   unsigned AluOp = AluMCOp.getImm();
233   Encoding |= LPAC::encodeLanaiAluCode(AluOp) << 5;
234   // Set P and Q
235   if (LPAC::isPreOp(AluOp))
236     Encoding |= (0x3 << 8);
237   if (LPAC::isPostOp(AluOp))
238     Encoding |= (0x1 << 8);
239   // Set JJJJ
240   switch (LPAC::getAluOp(AluOp)) {
241   case LPAC::SHL:
242   case LPAC::SRL:
243     Encoding |= 0x10;
244     break;
245   case LPAC::SRA:
246     Encoding |= 0x18;
247     break;
248   default:
249     break;
250   }
251 
252   return Encoding;
253 }
254 
255 unsigned
getSplsOpValue(const MCInst & Inst,unsigned OpNo,SmallVectorImpl<MCFixup> & Fixups,const MCSubtargetInfo & SubtargetInfo) const256 LanaiMCCodeEmitter::getSplsOpValue(const MCInst &Inst, unsigned OpNo,
257                                    SmallVectorImpl<MCFixup> &Fixups,
258                                    const MCSubtargetInfo &SubtargetInfo) const {
259   unsigned Encoding;
260   const MCOperand Op1 = Inst.getOperand(OpNo + 0);
261   const MCOperand Op2 = Inst.getOperand(OpNo + 1);
262   const MCOperand AluOp = Inst.getOperand(OpNo + 2);
263 
264   assert(Op1.isReg() && "First operand is not register.");
265   assert((Op2.isImm() || Op2.isExpr()) &&
266          "Second operand is neither an immediate nor an expression.");
267   assert((LPAC::getAluOp(AluOp.getImm()) == LPAC::ADD) &&
268          "Register immediate only supports addition operator");
269 
270   Encoding = (getLanaiRegisterNumbering(Op1.getReg()) << 12);
271   if (Op2.isImm()) {
272     assert(isInt<10>(Op2.getImm()) &&
273            "Constant value truncated (limited to 10-bit)");
274 
275     Encoding |= (Op2.getImm() & 0x3ff);
276     if (Op2.getImm() != 0) {
277       if (LPAC::isPreOp(AluOp.getImm()))
278         Encoding |= (0x3 << 10);
279       if (LPAC::isPostOp(AluOp.getImm()))
280         Encoding |= (0x1 << 10);
281     }
282   } else
283     getMachineOpValue(Inst, Op2, Fixups, SubtargetInfo);
284 
285   return Encoding;
286 }
287 
getBranchTargetOpValue(const MCInst & Inst,unsigned OpNo,SmallVectorImpl<MCFixup> & Fixups,const MCSubtargetInfo & SubtargetInfo) const288 unsigned LanaiMCCodeEmitter::getBranchTargetOpValue(
289     const MCInst &Inst, unsigned OpNo, SmallVectorImpl<MCFixup> &Fixups,
290     const MCSubtargetInfo &SubtargetInfo) const {
291   const MCOperand &MCOp = Inst.getOperand(OpNo);
292   if (MCOp.isReg() || MCOp.isImm())
293     return getMachineOpValue(Inst, MCOp, Fixups, SubtargetInfo);
294 
295   Fixups.push_back(MCFixup::create(
296       0, MCOp.getExpr(), static_cast<MCFixupKind>(Lanai::FIXUP_LANAI_25)));
297 
298   return 0;
299 }
300 
301 #include "LanaiGenMCCodeEmitter.inc"
302 
303 } // end namespace llvm
304 
305 llvm::MCCodeEmitter *
createLanaiMCCodeEmitter(const MCInstrInfo & InstrInfo,MCContext & context)306 llvm::createLanaiMCCodeEmitter(const MCInstrInfo &InstrInfo,
307                                MCContext &context) {
308   return new LanaiMCCodeEmitter(InstrInfo, context);
309 }
310