xref: /freebsd-src/contrib/llvm-project/llvm/lib/Target/RISCV/MCTargetDesc/RISCVMCCodeEmitter.cpp (revision 415efcecd8b80f68e76376ef2b854cb6f5c84b5a)
106c3fb27SDimitry Andric //===-- RISCVMCCodeEmitter.cpp - Convert RISC-V code to machine code ------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric // This file implements the RISCVMCCodeEmitter class.
100b57cec5SDimitry Andric //
110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
120b57cec5SDimitry Andric 
13e8d8bef9SDimitry Andric #include "MCTargetDesc/RISCVBaseInfo.h"
140b57cec5SDimitry Andric #include "MCTargetDesc/RISCVFixupKinds.h"
150b57cec5SDimitry Andric #include "MCTargetDesc/RISCVMCExpr.h"
160b57cec5SDimitry Andric #include "MCTargetDesc/RISCVMCTargetDesc.h"
170b57cec5SDimitry Andric #include "llvm/ADT/Statistic.h"
180b57cec5SDimitry Andric #include "llvm/MC/MCAsmInfo.h"
190b57cec5SDimitry Andric #include "llvm/MC/MCCodeEmitter.h"
200b57cec5SDimitry Andric #include "llvm/MC/MCContext.h"
210b57cec5SDimitry Andric #include "llvm/MC/MCExpr.h"
220b57cec5SDimitry Andric #include "llvm/MC/MCInst.h"
230b57cec5SDimitry Andric #include "llvm/MC/MCInstBuilder.h"
240b57cec5SDimitry Andric #include "llvm/MC/MCInstrInfo.h"
250b57cec5SDimitry Andric #include "llvm/MC/MCRegisterInfo.h"
2681ad6265SDimitry Andric #include "llvm/MC/MCSubtargetInfo.h"
270b57cec5SDimitry Andric #include "llvm/MC/MCSymbol.h"
280b57cec5SDimitry Andric #include "llvm/Support/Casting.h"
290b57cec5SDimitry Andric #include "llvm/Support/EndianStream.h"
300b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h"
310b57cec5SDimitry Andric 
320b57cec5SDimitry Andric using namespace llvm;
330b57cec5SDimitry Andric 
340b57cec5SDimitry Andric #define DEBUG_TYPE "mccodeemitter"
350b57cec5SDimitry Andric 
360b57cec5SDimitry Andric STATISTIC(MCNumEmitted, "Number of MC instructions emitted");
370b57cec5SDimitry Andric STATISTIC(MCNumFixups, "Number of MC fixups created");
380b57cec5SDimitry Andric 
390b57cec5SDimitry Andric namespace {
400b57cec5SDimitry Andric class RISCVMCCodeEmitter : public MCCodeEmitter {
410b57cec5SDimitry Andric   RISCVMCCodeEmitter(const RISCVMCCodeEmitter &) = delete;
420b57cec5SDimitry Andric   void operator=(const RISCVMCCodeEmitter &) = delete;
430b57cec5SDimitry Andric   MCContext &Ctx;
440b57cec5SDimitry Andric   MCInstrInfo const &MCII;
450b57cec5SDimitry Andric 
460b57cec5SDimitry Andric public:
470b57cec5SDimitry Andric   RISCVMCCodeEmitter(MCContext &ctx, MCInstrInfo const &MCII)
480b57cec5SDimitry Andric       : Ctx(ctx), MCII(MCII) {}
490b57cec5SDimitry Andric 
5081ad6265SDimitry Andric   ~RISCVMCCodeEmitter() override = default;
510b57cec5SDimitry Andric 
5206c3fb27SDimitry Andric   void encodeInstruction(const MCInst &MI, SmallVectorImpl<char> &CB,
530b57cec5SDimitry Andric                          SmallVectorImpl<MCFixup> &Fixups,
540b57cec5SDimitry Andric                          const MCSubtargetInfo &STI) const override;
550b57cec5SDimitry Andric 
5606c3fb27SDimitry Andric   void expandFunctionCall(const MCInst &MI, SmallVectorImpl<char> &CB,
570b57cec5SDimitry Andric                           SmallVectorImpl<MCFixup> &Fixups,
580b57cec5SDimitry Andric                           const MCSubtargetInfo &STI) const;
590b57cec5SDimitry Andric 
607a6dacacSDimitry Andric   void expandTLSDESCCall(const MCInst &MI, SmallVectorImpl<char> &CB,
617a6dacacSDimitry Andric                          SmallVectorImpl<MCFixup> &Fixups,
627a6dacacSDimitry Andric                          const MCSubtargetInfo &STI) const;
637a6dacacSDimitry Andric 
6406c3fb27SDimitry Andric   void expandAddTPRel(const MCInst &MI, SmallVectorImpl<char> &CB,
6506c3fb27SDimitry Andric                       SmallVectorImpl<MCFixup> &Fixups,
6606c3fb27SDimitry Andric                       const MCSubtargetInfo &STI) const;
6706c3fb27SDimitry Andric 
6806c3fb27SDimitry Andric   void expandLongCondBr(const MCInst &MI, SmallVectorImpl<char> &CB,
690b57cec5SDimitry Andric                         SmallVectorImpl<MCFixup> &Fixups,
700b57cec5SDimitry Andric                         const MCSubtargetInfo &STI) const;
710b57cec5SDimitry Andric 
720b57cec5SDimitry Andric   /// TableGen'erated function for getting the binary encoding for an
730b57cec5SDimitry Andric   /// instruction.
740b57cec5SDimitry Andric   uint64_t getBinaryCodeForInstr(const MCInst &MI,
750b57cec5SDimitry Andric                                  SmallVectorImpl<MCFixup> &Fixups,
760b57cec5SDimitry Andric                                  const MCSubtargetInfo &STI) const;
770b57cec5SDimitry Andric 
780b57cec5SDimitry Andric   /// Return binary encoding of operand. If the machine operand requires
790b57cec5SDimitry Andric   /// relocation, record the relocation and return zero.
800b57cec5SDimitry Andric   unsigned getMachineOpValue(const MCInst &MI, const MCOperand &MO,
810b57cec5SDimitry Andric                              SmallVectorImpl<MCFixup> &Fixups,
820b57cec5SDimitry Andric                              const MCSubtargetInfo &STI) const;
830b57cec5SDimitry Andric 
840b57cec5SDimitry Andric   unsigned getImmOpValueAsr1(const MCInst &MI, unsigned OpNo,
850b57cec5SDimitry Andric                              SmallVectorImpl<MCFixup> &Fixups,
860b57cec5SDimitry Andric                              const MCSubtargetInfo &STI) const;
870b57cec5SDimitry Andric 
880b57cec5SDimitry Andric   unsigned getImmOpValue(const MCInst &MI, unsigned OpNo,
890b57cec5SDimitry Andric                          SmallVectorImpl<MCFixup> &Fixups,
900b57cec5SDimitry Andric                          const MCSubtargetInfo &STI) const;
915ffd83dbSDimitry Andric 
925ffd83dbSDimitry Andric   unsigned getVMaskReg(const MCInst &MI, unsigned OpNo,
935ffd83dbSDimitry Andric                        SmallVectorImpl<MCFixup> &Fixups,
945ffd83dbSDimitry Andric                        const MCSubtargetInfo &STI) const;
9506c3fb27SDimitry Andric 
9606c3fb27SDimitry Andric   unsigned getRlistOpValue(const MCInst &MI, unsigned OpNo,
9706c3fb27SDimitry Andric                            SmallVectorImpl<MCFixup> &Fixups,
9806c3fb27SDimitry Andric                            const MCSubtargetInfo &STI) const;
995f757f3fSDimitry Andric 
1005f757f3fSDimitry Andric   unsigned getRegReg(const MCInst &MI, unsigned OpNo,
1015f757f3fSDimitry Andric                      SmallVectorImpl<MCFixup> &Fixups,
1025f757f3fSDimitry Andric                      const MCSubtargetInfo &STI) const;
1030b57cec5SDimitry Andric };
1040b57cec5SDimitry Andric } // end anonymous namespace
1050b57cec5SDimitry Andric 
1060b57cec5SDimitry Andric MCCodeEmitter *llvm::createRISCVMCCodeEmitter(const MCInstrInfo &MCII,
1070b57cec5SDimitry Andric                                               MCContext &Ctx) {
1080b57cec5SDimitry Andric   return new RISCVMCCodeEmitter(Ctx, MCII);
1090b57cec5SDimitry Andric }
1100b57cec5SDimitry Andric 
1115ffd83dbSDimitry Andric // Expand PseudoCALL(Reg), PseudoTAIL and PseudoJump to AUIPC and JALR with
1125ffd83dbSDimitry Andric // relocation types. We expand those pseudo-instructions while encoding them,
11306c3fb27SDimitry Andric // meaning AUIPC and JALR won't go through RISC-V MC to MC compressed
1145ffd83dbSDimitry Andric // instruction transformation. This is acceptable because AUIPC has no 16-bit
1155ffd83dbSDimitry Andric // form and C_JALR has no immediate operand field.  We let linker relaxation
1165ffd83dbSDimitry Andric // deal with it. When linker relaxation is enabled, AUIPC and JALR have a
1175ffd83dbSDimitry Andric // chance to relax to JAL.
1185ffd83dbSDimitry Andric // If the C extension is enabled, JAL has a chance relax to C_JAL.
11906c3fb27SDimitry Andric void RISCVMCCodeEmitter::expandFunctionCall(const MCInst &MI,
12006c3fb27SDimitry Andric                                             SmallVectorImpl<char> &CB,
1210b57cec5SDimitry Andric                                             SmallVectorImpl<MCFixup> &Fixups,
1220b57cec5SDimitry Andric                                             const MCSubtargetInfo &STI) const {
1230b57cec5SDimitry Andric   MCInst TmpInst;
1240b57cec5SDimitry Andric   MCOperand Func;
125e8d8bef9SDimitry Andric   MCRegister Ra;
1260b57cec5SDimitry Andric   if (MI.getOpcode() == RISCV::PseudoTAIL) {
1270b57cec5SDimitry Andric     Func = MI.getOperand(0);
1280b57cec5SDimitry Andric     Ra = RISCV::X6;
1290fca6ea1SDimitry Andric     // For Zicfilp, PseudoTAIL should be expanded to a software guarded branch.
1300fca6ea1SDimitry Andric     // It means to use t2(x7) as rs1 of JALR to expand PseudoTAIL.
1310fca6ea1SDimitry Andric     if (STI.hasFeature(RISCV::FeatureStdExtZicfilp))
1320fca6ea1SDimitry Andric       Ra = RISCV::X7;
1330b57cec5SDimitry Andric   } else if (MI.getOpcode() == RISCV::PseudoCALLReg) {
1340b57cec5SDimitry Andric     Func = MI.getOperand(1);
1350b57cec5SDimitry Andric     Ra = MI.getOperand(0).getReg();
1365ffd83dbSDimitry Andric   } else if (MI.getOpcode() == RISCV::PseudoCALL) {
1370b57cec5SDimitry Andric     Func = MI.getOperand(0);
1380b57cec5SDimitry Andric     Ra = RISCV::X1;
1395ffd83dbSDimitry Andric   } else if (MI.getOpcode() == RISCV::PseudoJump) {
1405ffd83dbSDimitry Andric     Func = MI.getOperand(1);
1415ffd83dbSDimitry Andric     Ra = MI.getOperand(0).getReg();
1420b57cec5SDimitry Andric   }
1430b57cec5SDimitry Andric   uint32_t Binary;
1440b57cec5SDimitry Andric 
1450b57cec5SDimitry Andric   assert(Func.isExpr() && "Expected expression");
1460b57cec5SDimitry Andric 
1470b57cec5SDimitry Andric   const MCExpr *CallExpr = Func.getExpr();
1480b57cec5SDimitry Andric 
1490b57cec5SDimitry Andric   // Emit AUIPC Ra, Func with R_RISCV_CALL relocation type.
15081ad6265SDimitry Andric   TmpInst = MCInstBuilder(RISCV::AUIPC).addReg(Ra).addExpr(CallExpr);
1510b57cec5SDimitry Andric   Binary = getBinaryCodeForInstr(TmpInst, Fixups, STI);
1525f757f3fSDimitry Andric   support::endian::write(CB, Binary, llvm::endianness::little);
1530b57cec5SDimitry Andric 
1545ffd83dbSDimitry Andric   if (MI.getOpcode() == RISCV::PseudoTAIL ||
1555ffd83dbSDimitry Andric       MI.getOpcode() == RISCV::PseudoJump)
1565ffd83dbSDimitry Andric     // Emit JALR X0, Ra, 0
1570b57cec5SDimitry Andric     TmpInst = MCInstBuilder(RISCV::JALR).addReg(RISCV::X0).addReg(Ra).addImm(0);
1580b57cec5SDimitry Andric   else
1590b57cec5SDimitry Andric     // Emit JALR Ra, Ra, 0
1600b57cec5SDimitry Andric     TmpInst = MCInstBuilder(RISCV::JALR).addReg(Ra).addReg(Ra).addImm(0);
1610b57cec5SDimitry Andric   Binary = getBinaryCodeForInstr(TmpInst, Fixups, STI);
1625f757f3fSDimitry Andric   support::endian::write(CB, Binary, llvm::endianness::little);
1630b57cec5SDimitry Andric }
1640b57cec5SDimitry Andric 
1657a6dacacSDimitry Andric void RISCVMCCodeEmitter::expandTLSDESCCall(const MCInst &MI,
1667a6dacacSDimitry Andric                                            SmallVectorImpl<char> &CB,
1677a6dacacSDimitry Andric                                            SmallVectorImpl<MCFixup> &Fixups,
1687a6dacacSDimitry Andric                                            const MCSubtargetInfo &STI) const {
1697a6dacacSDimitry Andric   MCOperand SrcSymbol = MI.getOperand(3);
1707a6dacacSDimitry Andric   assert(SrcSymbol.isExpr() &&
1717a6dacacSDimitry Andric          "Expected expression as first input to TLSDESCCALL");
1727a6dacacSDimitry Andric   const RISCVMCExpr *Expr = dyn_cast<RISCVMCExpr>(SrcSymbol.getExpr());
1737a6dacacSDimitry Andric   MCRegister Link = MI.getOperand(0).getReg();
1747a6dacacSDimitry Andric   MCRegister Dest = MI.getOperand(1).getReg();
1757a6dacacSDimitry Andric   MCRegister Imm = MI.getOperand(2).getImm();
1767a6dacacSDimitry Andric   Fixups.push_back(MCFixup::create(
1777a6dacacSDimitry Andric       0, Expr, MCFixupKind(RISCV::fixup_riscv_tlsdesc_call), MI.getLoc()));
1787a6dacacSDimitry Andric   MCInst Call =
1797a6dacacSDimitry Andric       MCInstBuilder(RISCV::JALR).addReg(Link).addReg(Dest).addImm(Imm);
1807a6dacacSDimitry Andric 
1817a6dacacSDimitry Andric   uint32_t Binary = getBinaryCodeForInstr(Call, Fixups, STI);
1827a6dacacSDimitry Andric   support::endian::write(CB, Binary, llvm::endianness::little);
1837a6dacacSDimitry Andric }
1847a6dacacSDimitry Andric 
1850b57cec5SDimitry Andric // Expand PseudoAddTPRel to a simple ADD with the correct relocation.
18606c3fb27SDimitry Andric void RISCVMCCodeEmitter::expandAddTPRel(const MCInst &MI,
18706c3fb27SDimitry Andric                                         SmallVectorImpl<char> &CB,
1880b57cec5SDimitry Andric                                         SmallVectorImpl<MCFixup> &Fixups,
1890b57cec5SDimitry Andric                                         const MCSubtargetInfo &STI) const {
1900b57cec5SDimitry Andric   MCOperand DestReg = MI.getOperand(0);
1910b57cec5SDimitry Andric   MCOperand SrcReg = MI.getOperand(1);
1920b57cec5SDimitry Andric   MCOperand TPReg = MI.getOperand(2);
1930b57cec5SDimitry Andric   assert(TPReg.isReg() && TPReg.getReg() == RISCV::X4 &&
1940b57cec5SDimitry Andric          "Expected thread pointer as second input to TP-relative add");
1950b57cec5SDimitry Andric 
1960b57cec5SDimitry Andric   MCOperand SrcSymbol = MI.getOperand(3);
1970b57cec5SDimitry Andric   assert(SrcSymbol.isExpr() &&
1980b57cec5SDimitry Andric          "Expected expression as third input to TP-relative add");
1990b57cec5SDimitry Andric 
2000b57cec5SDimitry Andric   const RISCVMCExpr *Expr = dyn_cast<RISCVMCExpr>(SrcSymbol.getExpr());
2010b57cec5SDimitry Andric   assert(Expr && Expr->getKind() == RISCVMCExpr::VK_RISCV_TPREL_ADD &&
2020b57cec5SDimitry Andric          "Expected tprel_add relocation on TP-relative symbol");
2030b57cec5SDimitry Andric 
2040b57cec5SDimitry Andric   // Emit the correct tprel_add relocation for the symbol.
2050b57cec5SDimitry Andric   Fixups.push_back(MCFixup::create(
2060b57cec5SDimitry Andric       0, Expr, MCFixupKind(RISCV::fixup_riscv_tprel_add), MI.getLoc()));
2070b57cec5SDimitry Andric 
2080b57cec5SDimitry Andric   // Emit fixup_riscv_relax for tprel_add where the relax feature is enabled.
20906c3fb27SDimitry Andric   if (STI.hasFeature(RISCV::FeatureRelax)) {
2100b57cec5SDimitry Andric     const MCConstantExpr *Dummy = MCConstantExpr::create(0, Ctx);
2110b57cec5SDimitry Andric     Fixups.push_back(MCFixup::create(
2120b57cec5SDimitry Andric         0, Dummy, MCFixupKind(RISCV::fixup_riscv_relax), MI.getLoc()));
2130b57cec5SDimitry Andric   }
2140b57cec5SDimitry Andric 
2150b57cec5SDimitry Andric   // Emit a normal ADD instruction with the given operands.
2160b57cec5SDimitry Andric   MCInst TmpInst = MCInstBuilder(RISCV::ADD)
2170b57cec5SDimitry Andric                        .addOperand(DestReg)
2180b57cec5SDimitry Andric                        .addOperand(SrcReg)
2190b57cec5SDimitry Andric                        .addOperand(TPReg);
2200b57cec5SDimitry Andric   uint32_t Binary = getBinaryCodeForInstr(TmpInst, Fixups, STI);
2215f757f3fSDimitry Andric   support::endian::write(CB, Binary, llvm::endianness::little);
2220b57cec5SDimitry Andric }
2230b57cec5SDimitry Andric 
22406c3fb27SDimitry Andric static unsigned getInvertedBranchOp(unsigned BrOp) {
22506c3fb27SDimitry Andric   switch (BrOp) {
22606c3fb27SDimitry Andric   default:
22706c3fb27SDimitry Andric     llvm_unreachable("Unexpected branch opcode!");
22806c3fb27SDimitry Andric   case RISCV::PseudoLongBEQ:
22906c3fb27SDimitry Andric     return RISCV::BNE;
23006c3fb27SDimitry Andric   case RISCV::PseudoLongBNE:
23106c3fb27SDimitry Andric     return RISCV::BEQ;
23206c3fb27SDimitry Andric   case RISCV::PseudoLongBLT:
23306c3fb27SDimitry Andric     return RISCV::BGE;
23406c3fb27SDimitry Andric   case RISCV::PseudoLongBGE:
23506c3fb27SDimitry Andric     return RISCV::BLT;
23606c3fb27SDimitry Andric   case RISCV::PseudoLongBLTU:
23706c3fb27SDimitry Andric     return RISCV::BGEU;
23806c3fb27SDimitry Andric   case RISCV::PseudoLongBGEU:
23906c3fb27SDimitry Andric     return RISCV::BLTU;
24006c3fb27SDimitry Andric   }
24106c3fb27SDimitry Andric }
24206c3fb27SDimitry Andric 
24306c3fb27SDimitry Andric // Expand PseudoLongBxx to an inverted conditional branch and an unconditional
24406c3fb27SDimitry Andric // jump.
24506c3fb27SDimitry Andric void RISCVMCCodeEmitter::expandLongCondBr(const MCInst &MI,
24606c3fb27SDimitry Andric                                           SmallVectorImpl<char> &CB,
24706c3fb27SDimitry Andric                                           SmallVectorImpl<MCFixup> &Fixups,
24806c3fb27SDimitry Andric                                           const MCSubtargetInfo &STI) const {
24906c3fb27SDimitry Andric   MCRegister SrcReg1 = MI.getOperand(0).getReg();
25006c3fb27SDimitry Andric   MCRegister SrcReg2 = MI.getOperand(1).getReg();
25106c3fb27SDimitry Andric   MCOperand SrcSymbol = MI.getOperand(2);
25206c3fb27SDimitry Andric   unsigned Opcode = MI.getOpcode();
25306c3fb27SDimitry Andric   bool IsEqTest =
25406c3fb27SDimitry Andric       Opcode == RISCV::PseudoLongBNE || Opcode == RISCV::PseudoLongBEQ;
25506c3fb27SDimitry Andric 
25606c3fb27SDimitry Andric   bool UseCompressedBr = false;
25706c3fb27SDimitry Andric   if (IsEqTest && (STI.hasFeature(RISCV::FeatureStdExtC) ||
25806c3fb27SDimitry Andric                    STI.hasFeature(RISCV::FeatureStdExtZca))) {
25906c3fb27SDimitry Andric     if (RISCV::X8 <= SrcReg1.id() && SrcReg1.id() <= RISCV::X15 &&
26006c3fb27SDimitry Andric         SrcReg2.id() == RISCV::X0) {
26106c3fb27SDimitry Andric       UseCompressedBr = true;
26206c3fb27SDimitry Andric     } else if (RISCV::X8 <= SrcReg2.id() && SrcReg2.id() <= RISCV::X15 &&
26306c3fb27SDimitry Andric                SrcReg1.id() == RISCV::X0) {
26406c3fb27SDimitry Andric       std::swap(SrcReg1, SrcReg2);
26506c3fb27SDimitry Andric       UseCompressedBr = true;
26606c3fb27SDimitry Andric     }
26706c3fb27SDimitry Andric   }
26806c3fb27SDimitry Andric 
26906c3fb27SDimitry Andric   uint32_t Offset;
27006c3fb27SDimitry Andric   if (UseCompressedBr) {
27106c3fb27SDimitry Andric     unsigned InvOpc =
27206c3fb27SDimitry Andric         Opcode == RISCV::PseudoLongBNE ? RISCV::C_BEQZ : RISCV::C_BNEZ;
27306c3fb27SDimitry Andric     MCInst TmpInst = MCInstBuilder(InvOpc).addReg(SrcReg1).addImm(6);
27406c3fb27SDimitry Andric     uint16_t Binary = getBinaryCodeForInstr(TmpInst, Fixups, STI);
2755f757f3fSDimitry Andric     support::endian::write<uint16_t>(CB, Binary, llvm::endianness::little);
27606c3fb27SDimitry Andric     Offset = 2;
27706c3fb27SDimitry Andric   } else {
27806c3fb27SDimitry Andric     unsigned InvOpc = getInvertedBranchOp(Opcode);
27906c3fb27SDimitry Andric     MCInst TmpInst =
28006c3fb27SDimitry Andric         MCInstBuilder(InvOpc).addReg(SrcReg1).addReg(SrcReg2).addImm(8);
28106c3fb27SDimitry Andric     uint32_t Binary = getBinaryCodeForInstr(TmpInst, Fixups, STI);
2825f757f3fSDimitry Andric     support::endian::write(CB, Binary, llvm::endianness::little);
28306c3fb27SDimitry Andric     Offset = 4;
28406c3fb27SDimitry Andric   }
28506c3fb27SDimitry Andric 
286*415efcecSDimitry Andric   // Save the number fixups.
287*415efcecSDimitry Andric   size_t FixupStartIndex = Fixups.size();
288*415efcecSDimitry Andric 
28906c3fb27SDimitry Andric   // Emit an unconditional jump to the destination.
29006c3fb27SDimitry Andric   MCInst TmpInst =
29106c3fb27SDimitry Andric       MCInstBuilder(RISCV::JAL).addReg(RISCV::X0).addOperand(SrcSymbol);
29206c3fb27SDimitry Andric   uint32_t Binary = getBinaryCodeForInstr(TmpInst, Fixups, STI);
2935f757f3fSDimitry Andric   support::endian::write(CB, Binary, llvm::endianness::little);
29406c3fb27SDimitry Andric 
295*415efcecSDimitry Andric   // Drop any fixup added so we can add the correct one.
296*415efcecSDimitry Andric   Fixups.resize(FixupStartIndex);
297*415efcecSDimitry Andric 
29806c3fb27SDimitry Andric   if (SrcSymbol.isExpr()) {
29906c3fb27SDimitry Andric     Fixups.push_back(MCFixup::create(Offset, SrcSymbol.getExpr(),
30006c3fb27SDimitry Andric                                      MCFixupKind(RISCV::fixup_riscv_jal),
30106c3fb27SDimitry Andric                                      MI.getLoc()));
30206c3fb27SDimitry Andric   }
30306c3fb27SDimitry Andric }
30406c3fb27SDimitry Andric 
30506c3fb27SDimitry Andric void RISCVMCCodeEmitter::encodeInstruction(const MCInst &MI,
30606c3fb27SDimitry Andric                                            SmallVectorImpl<char> &CB,
3070b57cec5SDimitry Andric                                            SmallVectorImpl<MCFixup> &Fixups,
3080b57cec5SDimitry Andric                                            const MCSubtargetInfo &STI) const {
3090b57cec5SDimitry Andric   const MCInstrDesc &Desc = MCII.get(MI.getOpcode());
3100b57cec5SDimitry Andric   // Get byte count of instruction.
3110b57cec5SDimitry Andric   unsigned Size = Desc.getSize();
3120b57cec5SDimitry Andric 
3131fd87a68SDimitry Andric   // RISCVInstrInfo::getInstSizeInBytes expects that the total size of the
3141fd87a68SDimitry Andric   // expanded instructions for each pseudo is correct in the Size field of the
3151fd87a68SDimitry Andric   // tablegen definition for the pseudo.
31606c3fb27SDimitry Andric   switch (MI.getOpcode()) {
31706c3fb27SDimitry Andric   default:
31806c3fb27SDimitry Andric     break;
31906c3fb27SDimitry Andric   case RISCV::PseudoCALLReg:
32006c3fb27SDimitry Andric   case RISCV::PseudoCALL:
32106c3fb27SDimitry Andric   case RISCV::PseudoTAIL:
32206c3fb27SDimitry Andric   case RISCV::PseudoJump:
32306c3fb27SDimitry Andric     expandFunctionCall(MI, CB, Fixups, STI);
3240b57cec5SDimitry Andric     MCNumEmitted += 2;
3250b57cec5SDimitry Andric     return;
32606c3fb27SDimitry Andric   case RISCV::PseudoAddTPRel:
32706c3fb27SDimitry Andric     expandAddTPRel(MI, CB, Fixups, STI);
3280b57cec5SDimitry Andric     MCNumEmitted += 1;
3290b57cec5SDimitry Andric     return;
33006c3fb27SDimitry Andric   case RISCV::PseudoLongBEQ:
33106c3fb27SDimitry Andric   case RISCV::PseudoLongBNE:
33206c3fb27SDimitry Andric   case RISCV::PseudoLongBLT:
33306c3fb27SDimitry Andric   case RISCV::PseudoLongBGE:
33406c3fb27SDimitry Andric   case RISCV::PseudoLongBLTU:
33506c3fb27SDimitry Andric   case RISCV::PseudoLongBGEU:
33606c3fb27SDimitry Andric     expandLongCondBr(MI, CB, Fixups, STI);
33706c3fb27SDimitry Andric     MCNumEmitted += 2;
33806c3fb27SDimitry Andric     return;
3397a6dacacSDimitry Andric   case RISCV::PseudoTLSDESCCall:
3407a6dacacSDimitry Andric     expandTLSDESCCall(MI, CB, Fixups, STI);
3417a6dacacSDimitry Andric     MCNumEmitted += 1;
3427a6dacacSDimitry Andric     return;
3430b57cec5SDimitry Andric   }
3440b57cec5SDimitry Andric 
3450b57cec5SDimitry Andric   switch (Size) {
3460b57cec5SDimitry Andric   default:
3470b57cec5SDimitry Andric     llvm_unreachable("Unhandled encodeInstruction length!");
3480b57cec5SDimitry Andric   case 2: {
3490b57cec5SDimitry Andric     uint16_t Bits = getBinaryCodeForInstr(MI, Fixups, STI);
3505f757f3fSDimitry Andric     support::endian::write<uint16_t>(CB, Bits, llvm::endianness::little);
3510b57cec5SDimitry Andric     break;
3520b57cec5SDimitry Andric   }
3530b57cec5SDimitry Andric   case 4: {
3540b57cec5SDimitry Andric     uint32_t Bits = getBinaryCodeForInstr(MI, Fixups, STI);
3555f757f3fSDimitry Andric     support::endian::write(CB, Bits, llvm::endianness::little);
3560b57cec5SDimitry Andric     break;
3570b57cec5SDimitry Andric   }
3580b57cec5SDimitry Andric   }
3590b57cec5SDimitry Andric 
3600b57cec5SDimitry Andric   ++MCNumEmitted; // Keep track of the # of mi's emitted.
3610b57cec5SDimitry Andric }
3620b57cec5SDimitry Andric 
3630b57cec5SDimitry Andric unsigned
3640b57cec5SDimitry Andric RISCVMCCodeEmitter::getMachineOpValue(const MCInst &MI, const MCOperand &MO,
3650b57cec5SDimitry Andric                                       SmallVectorImpl<MCFixup> &Fixups,
3660b57cec5SDimitry Andric                                       const MCSubtargetInfo &STI) const {
3670b57cec5SDimitry Andric 
3680b57cec5SDimitry Andric   if (MO.isReg())
3690b57cec5SDimitry Andric     return Ctx.getRegisterInfo()->getEncodingValue(MO.getReg());
3700b57cec5SDimitry Andric 
3710b57cec5SDimitry Andric   if (MO.isImm())
3720b57cec5SDimitry Andric     return static_cast<unsigned>(MO.getImm());
3730b57cec5SDimitry Andric 
3740b57cec5SDimitry Andric   llvm_unreachable("Unhandled expression!");
3750b57cec5SDimitry Andric   return 0;
3760b57cec5SDimitry Andric }
3770b57cec5SDimitry Andric 
3780b57cec5SDimitry Andric unsigned
3790b57cec5SDimitry Andric RISCVMCCodeEmitter::getImmOpValueAsr1(const MCInst &MI, unsigned OpNo,
3800b57cec5SDimitry Andric                                       SmallVectorImpl<MCFixup> &Fixups,
3810b57cec5SDimitry Andric                                       const MCSubtargetInfo &STI) const {
3820b57cec5SDimitry Andric   const MCOperand &MO = MI.getOperand(OpNo);
3830b57cec5SDimitry Andric 
3840b57cec5SDimitry Andric   if (MO.isImm()) {
3850b57cec5SDimitry Andric     unsigned Res = MO.getImm();
3860b57cec5SDimitry Andric     assert((Res & 1) == 0 && "LSB is non-zero");
3870b57cec5SDimitry Andric     return Res >> 1;
3880b57cec5SDimitry Andric   }
3890b57cec5SDimitry Andric 
3900b57cec5SDimitry Andric   return getImmOpValue(MI, OpNo, Fixups, STI);
3910b57cec5SDimitry Andric }
3920b57cec5SDimitry Andric 
3930b57cec5SDimitry Andric unsigned RISCVMCCodeEmitter::getImmOpValue(const MCInst &MI, unsigned OpNo,
3940b57cec5SDimitry Andric                                            SmallVectorImpl<MCFixup> &Fixups,
3950b57cec5SDimitry Andric                                            const MCSubtargetInfo &STI) const {
39606c3fb27SDimitry Andric   bool EnableRelax = STI.hasFeature(RISCV::FeatureRelax);
3970b57cec5SDimitry Andric   const MCOperand &MO = MI.getOperand(OpNo);
3980b57cec5SDimitry Andric 
3990b57cec5SDimitry Andric   MCInstrDesc const &Desc = MCII.get(MI.getOpcode());
400fe6060f1SDimitry Andric   unsigned MIFrm = RISCVII::getFormat(Desc.TSFlags);
4010b57cec5SDimitry Andric 
4020b57cec5SDimitry Andric   // If the destination is an immediate, there is nothing to do.
4030b57cec5SDimitry Andric   if (MO.isImm())
4040b57cec5SDimitry Andric     return MO.getImm();
4050b57cec5SDimitry Andric 
4060b57cec5SDimitry Andric   assert(MO.isExpr() &&
4070b57cec5SDimitry Andric          "getImmOpValue expects only expressions or immediates");
4080b57cec5SDimitry Andric   const MCExpr *Expr = MO.getExpr();
4090b57cec5SDimitry Andric   MCExpr::ExprKind Kind = Expr->getKind();
4100b57cec5SDimitry Andric   RISCV::Fixups FixupKind = RISCV::fixup_riscv_invalid;
4110b57cec5SDimitry Andric   bool RelaxCandidate = false;
4120b57cec5SDimitry Andric   if (Kind == MCExpr::Target) {
4130b57cec5SDimitry Andric     const RISCVMCExpr *RVExpr = cast<RISCVMCExpr>(Expr);
4140b57cec5SDimitry Andric 
4150b57cec5SDimitry Andric     switch (RVExpr->getKind()) {
4160b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_None:
4170b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_Invalid:
4180b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_32_PCREL:
4190b57cec5SDimitry Andric       llvm_unreachable("Unhandled fixup kind!");
4200b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_TPREL_ADD:
4210b57cec5SDimitry Andric       // tprel_add is only used to indicate that a relocation should be emitted
4220b57cec5SDimitry Andric       // for an add instruction used in TP-relative addressing. It should not be
4230b57cec5SDimitry Andric       // expanded as if representing an actual instruction operand and so to
4240b57cec5SDimitry Andric       // encounter it here is an error.
4250b57cec5SDimitry Andric       llvm_unreachable(
4260b57cec5SDimitry Andric           "VK_RISCV_TPREL_ADD should not represent an instruction operand");
4270b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_LO:
4280b57cec5SDimitry Andric       if (MIFrm == RISCVII::InstFormatI)
4290b57cec5SDimitry Andric         FixupKind = RISCV::fixup_riscv_lo12_i;
4300b57cec5SDimitry Andric       else if (MIFrm == RISCVII::InstFormatS)
4310b57cec5SDimitry Andric         FixupKind = RISCV::fixup_riscv_lo12_s;
4320b57cec5SDimitry Andric       else
4330b57cec5SDimitry Andric         llvm_unreachable("VK_RISCV_LO used with unexpected instruction format");
4340b57cec5SDimitry Andric       RelaxCandidate = true;
4350b57cec5SDimitry Andric       break;
4360b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_HI:
4370b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_hi20;
4380b57cec5SDimitry Andric       RelaxCandidate = true;
4390b57cec5SDimitry Andric       break;
4400b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_PCREL_LO:
4410b57cec5SDimitry Andric       if (MIFrm == RISCVII::InstFormatI)
4420b57cec5SDimitry Andric         FixupKind = RISCV::fixup_riscv_pcrel_lo12_i;
4430b57cec5SDimitry Andric       else if (MIFrm == RISCVII::InstFormatS)
4440b57cec5SDimitry Andric         FixupKind = RISCV::fixup_riscv_pcrel_lo12_s;
4450b57cec5SDimitry Andric       else
4460b57cec5SDimitry Andric         llvm_unreachable(
4470b57cec5SDimitry Andric             "VK_RISCV_PCREL_LO used with unexpected instruction format");
4480b57cec5SDimitry Andric       RelaxCandidate = true;
4490b57cec5SDimitry Andric       break;
4500b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_PCREL_HI:
4510b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_pcrel_hi20;
4520b57cec5SDimitry Andric       RelaxCandidate = true;
4530b57cec5SDimitry Andric       break;
4540b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_GOT_HI:
4550b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_got_hi20;
4560b57cec5SDimitry Andric       break;
4570b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_TPREL_LO:
4580b57cec5SDimitry Andric       if (MIFrm == RISCVII::InstFormatI)
4590b57cec5SDimitry Andric         FixupKind = RISCV::fixup_riscv_tprel_lo12_i;
4600b57cec5SDimitry Andric       else if (MIFrm == RISCVII::InstFormatS)
4610b57cec5SDimitry Andric         FixupKind = RISCV::fixup_riscv_tprel_lo12_s;
4620b57cec5SDimitry Andric       else
4630b57cec5SDimitry Andric         llvm_unreachable(
4640b57cec5SDimitry Andric             "VK_RISCV_TPREL_LO used with unexpected instruction format");
4650b57cec5SDimitry Andric       RelaxCandidate = true;
4660b57cec5SDimitry Andric       break;
4670b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_TPREL_HI:
4680b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_tprel_hi20;
4690b57cec5SDimitry Andric       RelaxCandidate = true;
4700b57cec5SDimitry Andric       break;
4710b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_TLS_GOT_HI:
4720b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_tls_got_hi20;
4730b57cec5SDimitry Andric       break;
4740b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_TLS_GD_HI:
4750b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_tls_gd_hi20;
4760b57cec5SDimitry Andric       break;
4770b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_CALL:
4780b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_call;
4790b57cec5SDimitry Andric       RelaxCandidate = true;
4800b57cec5SDimitry Andric       break;
4810b57cec5SDimitry Andric     case RISCVMCExpr::VK_RISCV_CALL_PLT:
4820b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_call_plt;
4830b57cec5SDimitry Andric       RelaxCandidate = true;
4840b57cec5SDimitry Andric       break;
4857a6dacacSDimitry Andric     case RISCVMCExpr::VK_RISCV_TLSDESC_HI:
4867a6dacacSDimitry Andric       FixupKind = RISCV::fixup_riscv_tlsdesc_hi20;
4877a6dacacSDimitry Andric       break;
4887a6dacacSDimitry Andric     case RISCVMCExpr::VK_RISCV_TLSDESC_LOAD_LO:
4897a6dacacSDimitry Andric       FixupKind = RISCV::fixup_riscv_tlsdesc_load_lo12;
4907a6dacacSDimitry Andric       break;
4917a6dacacSDimitry Andric     case RISCVMCExpr::VK_RISCV_TLSDESC_ADD_LO:
4927a6dacacSDimitry Andric       FixupKind = RISCV::fixup_riscv_tlsdesc_add_lo12;
4937a6dacacSDimitry Andric       break;
4947a6dacacSDimitry Andric     case RISCVMCExpr::VK_RISCV_TLSDESC_CALL:
4957a6dacacSDimitry Andric       FixupKind = RISCV::fixup_riscv_tlsdesc_call;
4967a6dacacSDimitry Andric       break;
4970b57cec5SDimitry Andric     }
4985f757f3fSDimitry Andric   } else if ((Kind == MCExpr::SymbolRef &&
4995f757f3fSDimitry Andric                  cast<MCSymbolRefExpr>(Expr)->getKind() ==
5005f757f3fSDimitry Andric                      MCSymbolRefExpr::VK_None) ||
5015f757f3fSDimitry Andric              Kind == MCExpr::Binary) {
5025f757f3fSDimitry Andric     // FIXME: Sub kind binary exprs have chance of underflow.
503349cc55cSDimitry Andric     if (MIFrm == RISCVII::InstFormatJ) {
5040b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_jal;
5050b57cec5SDimitry Andric     } else if (MIFrm == RISCVII::InstFormatB) {
5060b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_branch;
5070b57cec5SDimitry Andric     } else if (MIFrm == RISCVII::InstFormatCJ) {
5080b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_rvc_jump;
5090b57cec5SDimitry Andric     } else if (MIFrm == RISCVII::InstFormatCB) {
5100b57cec5SDimitry Andric       FixupKind = RISCV::fixup_riscv_rvc_branch;
51106c3fb27SDimitry Andric     } else if (MIFrm == RISCVII::InstFormatI) {
51206c3fb27SDimitry Andric       FixupKind = RISCV::fixup_riscv_12_i;
5130b57cec5SDimitry Andric     }
5140b57cec5SDimitry Andric   }
5150b57cec5SDimitry Andric 
5160b57cec5SDimitry Andric   assert(FixupKind != RISCV::fixup_riscv_invalid && "Unhandled expression!");
5170b57cec5SDimitry Andric 
5180b57cec5SDimitry Andric   Fixups.push_back(
5190b57cec5SDimitry Andric       MCFixup::create(0, Expr, MCFixupKind(FixupKind), MI.getLoc()));
5200b57cec5SDimitry Andric   ++MCNumFixups;
5210b57cec5SDimitry Andric 
5220b57cec5SDimitry Andric   // Ensure an R_RISCV_RELAX relocation will be emitted if linker relaxation is
5230b57cec5SDimitry Andric   // enabled and the current fixup will result in a relocation that may be
5240b57cec5SDimitry Andric   // relaxed.
5250b57cec5SDimitry Andric   if (EnableRelax && RelaxCandidate) {
5260b57cec5SDimitry Andric     const MCConstantExpr *Dummy = MCConstantExpr::create(0, Ctx);
5270b57cec5SDimitry Andric     Fixups.push_back(
5280b57cec5SDimitry Andric     MCFixup::create(0, Dummy, MCFixupKind(RISCV::fixup_riscv_relax),
5290b57cec5SDimitry Andric                     MI.getLoc()));
5300b57cec5SDimitry Andric     ++MCNumFixups;
5310b57cec5SDimitry Andric   }
5320b57cec5SDimitry Andric 
5330b57cec5SDimitry Andric   return 0;
5340b57cec5SDimitry Andric }
5350b57cec5SDimitry Andric 
5365ffd83dbSDimitry Andric unsigned RISCVMCCodeEmitter::getVMaskReg(const MCInst &MI, unsigned OpNo,
5375ffd83dbSDimitry Andric                                          SmallVectorImpl<MCFixup> &Fixups,
5385ffd83dbSDimitry Andric                                          const MCSubtargetInfo &STI) const {
5395ffd83dbSDimitry Andric   MCOperand MO = MI.getOperand(OpNo);
5405ffd83dbSDimitry Andric   assert(MO.isReg() && "Expected a register.");
5415ffd83dbSDimitry Andric 
5425ffd83dbSDimitry Andric   switch (MO.getReg()) {
5435ffd83dbSDimitry Andric   default:
5445ffd83dbSDimitry Andric     llvm_unreachable("Invalid mask register.");
5455ffd83dbSDimitry Andric   case RISCV::V0:
5465ffd83dbSDimitry Andric     return 0;
5475ffd83dbSDimitry Andric   case RISCV::NoRegister:
5485ffd83dbSDimitry Andric     return 1;
5495ffd83dbSDimitry Andric   }
5505ffd83dbSDimitry Andric }
5515ffd83dbSDimitry Andric 
55206c3fb27SDimitry Andric unsigned RISCVMCCodeEmitter::getRlistOpValue(const MCInst &MI, unsigned OpNo,
55306c3fb27SDimitry Andric                                              SmallVectorImpl<MCFixup> &Fixups,
55406c3fb27SDimitry Andric                                              const MCSubtargetInfo &STI) const {
55506c3fb27SDimitry Andric   const MCOperand &MO = MI.getOperand(OpNo);
55606c3fb27SDimitry Andric   assert(MO.isImm() && "Rlist operand must be immediate");
55706c3fb27SDimitry Andric   auto Imm = MO.getImm();
55806c3fb27SDimitry Andric   assert(Imm >= 4 && "EABI is currently not implemented");
55906c3fb27SDimitry Andric   return Imm;
56006c3fb27SDimitry Andric }
56106c3fb27SDimitry Andric 
5625f757f3fSDimitry Andric unsigned RISCVMCCodeEmitter::getRegReg(const MCInst &MI, unsigned OpNo,
5635f757f3fSDimitry Andric                                        SmallVectorImpl<MCFixup> &Fixups,
5645f757f3fSDimitry Andric                                        const MCSubtargetInfo &STI) const {
5655f757f3fSDimitry Andric   const MCOperand &MO = MI.getOperand(OpNo);
5665f757f3fSDimitry Andric   const MCOperand &MO1 = MI.getOperand(OpNo + 1);
5675f757f3fSDimitry Andric   assert(MO.isReg() && MO1.isReg() && "Expected registers.");
5685f757f3fSDimitry Andric 
5695f757f3fSDimitry Andric   unsigned Op = Ctx.getRegisterInfo()->getEncodingValue(MO.getReg());
5705f757f3fSDimitry Andric   unsigned Op1 = Ctx.getRegisterInfo()->getEncodingValue(MO1.getReg());
5715f757f3fSDimitry Andric 
5725f757f3fSDimitry Andric   return Op | Op1 << 5;
5735f757f3fSDimitry Andric }
5745f757f3fSDimitry Andric 
5750b57cec5SDimitry Andric #include "RISCVGenMCCodeEmitter.inc"
576