109467b48Spatrick //===-- RISCVAsmBackend.cpp - RISCV Assembler Backend ---------------------===//
209467b48Spatrick //
309467b48Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
409467b48Spatrick // See https://llvm.org/LICENSE.txt for license information.
509467b48Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
609467b48Spatrick //
709467b48Spatrick //===----------------------------------------------------------------------===//
809467b48Spatrick
909467b48Spatrick #include "RISCVAsmBackend.h"
1009467b48Spatrick #include "RISCVMCExpr.h"
1109467b48Spatrick #include "llvm/ADT/APInt.h"
1273471bf0Spatrick #include "llvm/MC/MCAsmInfo.h"
1309467b48Spatrick #include "llvm/MC/MCAsmLayout.h"
1409467b48Spatrick #include "llvm/MC/MCAssembler.h"
1509467b48Spatrick #include "llvm/MC/MCContext.h"
1609467b48Spatrick #include "llvm/MC/MCDirectives.h"
1709467b48Spatrick #include "llvm/MC/MCELFObjectWriter.h"
1809467b48Spatrick #include "llvm/MC/MCExpr.h"
1909467b48Spatrick #include "llvm/MC/MCObjectWriter.h"
2009467b48Spatrick #include "llvm/MC/MCSymbol.h"
2109467b48Spatrick #include "llvm/MC/MCValue.h"
2273471bf0Spatrick #include "llvm/Support/Endian.h"
2373471bf0Spatrick #include "llvm/Support/EndianStream.h"
2409467b48Spatrick #include "llvm/Support/ErrorHandling.h"
2573471bf0Spatrick #include "llvm/Support/LEB128.h"
2609467b48Spatrick #include "llvm/Support/raw_ostream.h"
2709467b48Spatrick
2809467b48Spatrick using namespace llvm;
2909467b48Spatrick
getFixupKind(StringRef Name) const30*d415bd75Srobert std::optional<MCFixupKind> RISCVAsmBackend::getFixupKind(StringRef Name) const {
31097a140dSpatrick if (STI.getTargetTriple().isOSBinFormatELF()) {
32097a140dSpatrick unsigned Type;
33097a140dSpatrick Type = llvm::StringSwitch<unsigned>(Name)
34097a140dSpatrick #define ELF_RELOC(X, Y) .Case(#X, Y)
35097a140dSpatrick #include "llvm/BinaryFormat/ELFRelocs/RISCV.def"
36097a140dSpatrick #undef ELF_RELOC
3773471bf0Spatrick .Case("BFD_RELOC_NONE", ELF::R_RISCV_NONE)
3873471bf0Spatrick .Case("BFD_RELOC_32", ELF::R_RISCV_32)
3973471bf0Spatrick .Case("BFD_RELOC_64", ELF::R_RISCV_64)
40097a140dSpatrick .Default(-1u);
41097a140dSpatrick if (Type != -1u)
42097a140dSpatrick return static_cast<MCFixupKind>(FirstLiteralRelocationKind + Type);
43097a140dSpatrick }
44*d415bd75Srobert return std::nullopt;
45097a140dSpatrick }
46097a140dSpatrick
47097a140dSpatrick const MCFixupKindInfo &
getFixupKindInfo(MCFixupKind Kind) const48097a140dSpatrick RISCVAsmBackend::getFixupKindInfo(MCFixupKind Kind) const {
49097a140dSpatrick const static MCFixupKindInfo Infos[] = {
50097a140dSpatrick // This table *must* be in the order that the fixup_* kinds are defined in
51097a140dSpatrick // RISCVFixupKinds.h.
52097a140dSpatrick //
53097a140dSpatrick // name offset bits flags
54097a140dSpatrick {"fixup_riscv_hi20", 12, 20, 0},
55097a140dSpatrick {"fixup_riscv_lo12_i", 20, 12, 0},
56097a140dSpatrick {"fixup_riscv_lo12_s", 0, 32, 0},
57097a140dSpatrick {"fixup_riscv_pcrel_hi20", 12, 20,
58097a140dSpatrick MCFixupKindInfo::FKF_IsPCRel | MCFixupKindInfo::FKF_IsTarget},
59097a140dSpatrick {"fixup_riscv_pcrel_lo12_i", 20, 12,
60097a140dSpatrick MCFixupKindInfo::FKF_IsPCRel | MCFixupKindInfo::FKF_IsTarget},
61097a140dSpatrick {"fixup_riscv_pcrel_lo12_s", 0, 32,
62097a140dSpatrick MCFixupKindInfo::FKF_IsPCRel | MCFixupKindInfo::FKF_IsTarget},
63097a140dSpatrick {"fixup_riscv_got_hi20", 12, 20, MCFixupKindInfo::FKF_IsPCRel},
64097a140dSpatrick {"fixup_riscv_tprel_hi20", 12, 20, 0},
65097a140dSpatrick {"fixup_riscv_tprel_lo12_i", 20, 12, 0},
66097a140dSpatrick {"fixup_riscv_tprel_lo12_s", 0, 32, 0},
67097a140dSpatrick {"fixup_riscv_tprel_add", 0, 0, 0},
68097a140dSpatrick {"fixup_riscv_tls_got_hi20", 12, 20, MCFixupKindInfo::FKF_IsPCRel},
69097a140dSpatrick {"fixup_riscv_tls_gd_hi20", 12, 20, MCFixupKindInfo::FKF_IsPCRel},
70097a140dSpatrick {"fixup_riscv_jal", 12, 20, MCFixupKindInfo::FKF_IsPCRel},
71097a140dSpatrick {"fixup_riscv_branch", 0, 32, MCFixupKindInfo::FKF_IsPCRel},
72097a140dSpatrick {"fixup_riscv_rvc_jump", 2, 11, MCFixupKindInfo::FKF_IsPCRel},
73097a140dSpatrick {"fixup_riscv_rvc_branch", 0, 16, MCFixupKindInfo::FKF_IsPCRel},
74097a140dSpatrick {"fixup_riscv_call", 0, 64, MCFixupKindInfo::FKF_IsPCRel},
75097a140dSpatrick {"fixup_riscv_call_plt", 0, 64, MCFixupKindInfo::FKF_IsPCRel},
76097a140dSpatrick {"fixup_riscv_relax", 0, 0, 0},
7773471bf0Spatrick {"fixup_riscv_align", 0, 0, 0},
7873471bf0Spatrick
7973471bf0Spatrick {"fixup_riscv_set_8", 0, 8, 0},
8073471bf0Spatrick {"fixup_riscv_add_8", 0, 8, 0},
8173471bf0Spatrick {"fixup_riscv_sub_8", 0, 8, 0},
8273471bf0Spatrick
8373471bf0Spatrick {"fixup_riscv_set_16", 0, 16, 0},
8473471bf0Spatrick {"fixup_riscv_add_16", 0, 16, 0},
8573471bf0Spatrick {"fixup_riscv_sub_16", 0, 16, 0},
8673471bf0Spatrick
8773471bf0Spatrick {"fixup_riscv_set_32", 0, 32, 0},
8873471bf0Spatrick {"fixup_riscv_add_32", 0, 32, 0},
8973471bf0Spatrick {"fixup_riscv_sub_32", 0, 32, 0},
9073471bf0Spatrick
9173471bf0Spatrick {"fixup_riscv_add_64", 0, 64, 0},
9273471bf0Spatrick {"fixup_riscv_sub_64", 0, 64, 0},
9373471bf0Spatrick
9473471bf0Spatrick {"fixup_riscv_set_6b", 2, 6, 0},
9573471bf0Spatrick {"fixup_riscv_sub_6b", 2, 6, 0},
9673471bf0Spatrick };
97*d415bd75Srobert static_assert((std::size(Infos)) == RISCV::NumTargetFixupKinds,
98097a140dSpatrick "Not all fixup kinds added to Infos array");
99097a140dSpatrick
100097a140dSpatrick // Fixup kinds from .reloc directive are like R_RISCV_NONE. They
101097a140dSpatrick // do not require any extra processing.
102097a140dSpatrick if (Kind >= FirstLiteralRelocationKind)
103097a140dSpatrick return MCAsmBackend::getFixupKindInfo(FK_NONE);
104097a140dSpatrick
105097a140dSpatrick if (Kind < FirstTargetFixupKind)
106097a140dSpatrick return MCAsmBackend::getFixupKindInfo(Kind);
107097a140dSpatrick
108097a140dSpatrick assert(unsigned(Kind - FirstTargetFixupKind) < getNumFixupKinds() &&
109097a140dSpatrick "Invalid kind!");
110097a140dSpatrick return Infos[Kind - FirstTargetFixupKind];
111097a140dSpatrick }
112097a140dSpatrick
11309467b48Spatrick // If linker relaxation is enabled, or the relax option had previously been
11409467b48Spatrick // enabled, always emit relocations even if the fixup can be resolved. This is
11509467b48Spatrick // necessary for correctness as offsets may change during relaxation.
shouldForceRelocation(const MCAssembler & Asm,const MCFixup & Fixup,const MCValue & Target)11609467b48Spatrick bool RISCVAsmBackend::shouldForceRelocation(const MCAssembler &Asm,
11709467b48Spatrick const MCFixup &Fixup,
11809467b48Spatrick const MCValue &Target) {
119097a140dSpatrick if (Fixup.getKind() >= FirstLiteralRelocationKind)
120097a140dSpatrick return true;
12109467b48Spatrick switch (Fixup.getTargetKind()) {
12209467b48Spatrick default:
12309467b48Spatrick break;
12409467b48Spatrick case FK_Data_1:
12509467b48Spatrick case FK_Data_2:
12609467b48Spatrick case FK_Data_4:
12709467b48Spatrick case FK_Data_8:
12809467b48Spatrick if (Target.isAbsolute())
12909467b48Spatrick return false;
13009467b48Spatrick break;
13109467b48Spatrick case RISCV::fixup_riscv_got_hi20:
13209467b48Spatrick case RISCV::fixup_riscv_tls_got_hi20:
13309467b48Spatrick case RISCV::fixup_riscv_tls_gd_hi20:
13409467b48Spatrick return true;
13509467b48Spatrick }
13609467b48Spatrick
13709467b48Spatrick return STI.getFeatureBits()[RISCV::FeatureRelax] || ForceRelocs;
13809467b48Spatrick }
13909467b48Spatrick
fixupNeedsRelaxationAdvanced(const MCFixup & Fixup,bool Resolved,uint64_t Value,const MCRelaxableFragment * DF,const MCAsmLayout & Layout,const bool WasForced) const14009467b48Spatrick bool RISCVAsmBackend::fixupNeedsRelaxationAdvanced(const MCFixup &Fixup,
14109467b48Spatrick bool Resolved,
14209467b48Spatrick uint64_t Value,
14309467b48Spatrick const MCRelaxableFragment *DF,
14409467b48Spatrick const MCAsmLayout &Layout,
14509467b48Spatrick const bool WasForced) const {
14609467b48Spatrick // Return true if the symbol is actually unresolved.
14709467b48Spatrick // Resolved could be always false when shouldForceRelocation return true.
14809467b48Spatrick // We use !WasForced to indicate that the symbol is unresolved and not forced
14909467b48Spatrick // by shouldForceRelocation.
15009467b48Spatrick if (!Resolved && !WasForced)
15109467b48Spatrick return true;
15209467b48Spatrick
15309467b48Spatrick int64_t Offset = int64_t(Value);
15409467b48Spatrick switch (Fixup.getTargetKind()) {
15509467b48Spatrick default:
15609467b48Spatrick return false;
15709467b48Spatrick case RISCV::fixup_riscv_rvc_branch:
15809467b48Spatrick // For compressed branch instructions the immediate must be
15909467b48Spatrick // in the range [-256, 254].
16009467b48Spatrick return Offset > 254 || Offset < -256;
16109467b48Spatrick case RISCV::fixup_riscv_rvc_jump:
16209467b48Spatrick // For compressed jump instructions the immediate must be
16309467b48Spatrick // in the range [-2048, 2046].
16409467b48Spatrick return Offset > 2046 || Offset < -2048;
16509467b48Spatrick }
16609467b48Spatrick }
16709467b48Spatrick
relaxInstruction(MCInst & Inst,const MCSubtargetInfo & STI) const168097a140dSpatrick void RISCVAsmBackend::relaxInstruction(MCInst &Inst,
169097a140dSpatrick const MCSubtargetInfo &STI) const {
170097a140dSpatrick MCInst Res;
17109467b48Spatrick switch (Inst.getOpcode()) {
17209467b48Spatrick default:
17309467b48Spatrick llvm_unreachable("Opcode not expected!");
17409467b48Spatrick case RISCV::C_BEQZ:
17509467b48Spatrick case RISCV::C_BNEZ:
17609467b48Spatrick case RISCV::C_J:
17709467b48Spatrick case RISCV::C_JAL:
178*d415bd75Srobert bool Success = RISCVRVC::uncompress(Res, Inst, STI);
179*d415bd75Srobert assert(Success && "Can't uncompress instruction");
180*d415bd75Srobert (void)Success;
18109467b48Spatrick break;
18209467b48Spatrick }
183097a140dSpatrick Inst = std::move(Res);
18409467b48Spatrick }
18509467b48Spatrick
relaxDwarfLineAddr(MCDwarfLineAddrFragment & DF,MCAsmLayout & Layout,bool & WasRelaxed) const18673471bf0Spatrick bool RISCVAsmBackend::relaxDwarfLineAddr(MCDwarfLineAddrFragment &DF,
18773471bf0Spatrick MCAsmLayout &Layout,
18873471bf0Spatrick bool &WasRelaxed) const {
18973471bf0Spatrick MCContext &C = Layout.getAssembler().getContext();
19073471bf0Spatrick
19173471bf0Spatrick int64_t LineDelta = DF.getLineDelta();
19273471bf0Spatrick const MCExpr &AddrDelta = DF.getAddrDelta();
19373471bf0Spatrick SmallVectorImpl<char> &Data = DF.getContents();
19473471bf0Spatrick SmallVectorImpl<MCFixup> &Fixups = DF.getFixups();
19573471bf0Spatrick size_t OldSize = Data.size();
19673471bf0Spatrick
19773471bf0Spatrick int64_t Value;
19873471bf0Spatrick bool IsAbsolute = AddrDelta.evaluateKnownAbsolute(Value, Layout);
19973471bf0Spatrick assert(IsAbsolute && "CFA with invalid expression");
20073471bf0Spatrick (void)IsAbsolute;
20173471bf0Spatrick
20273471bf0Spatrick Data.clear();
20373471bf0Spatrick Fixups.clear();
20473471bf0Spatrick raw_svector_ostream OS(Data);
20573471bf0Spatrick
20673471bf0Spatrick // INT64_MAX is a signal that this is actually a DW_LNE_end_sequence.
20773471bf0Spatrick if (LineDelta != INT64_MAX) {
20873471bf0Spatrick OS << uint8_t(dwarf::DW_LNS_advance_line);
20973471bf0Spatrick encodeSLEB128(LineDelta, OS);
21073471bf0Spatrick }
21173471bf0Spatrick
21273471bf0Spatrick unsigned Offset;
21373471bf0Spatrick std::pair<unsigned, unsigned> Fixup;
21473471bf0Spatrick
21573471bf0Spatrick // According to the DWARF specification, the `DW_LNS_fixed_advance_pc` opcode
21673471bf0Spatrick // takes a single unsigned half (unencoded) operand. The maximum encodable
21773471bf0Spatrick // value is therefore 65535. Set a conservative upper bound for relaxation.
21873471bf0Spatrick if (Value > 60000) {
21973471bf0Spatrick unsigned PtrSize = C.getAsmInfo()->getCodePointerSize();
22073471bf0Spatrick
22173471bf0Spatrick OS << uint8_t(dwarf::DW_LNS_extended_op);
22273471bf0Spatrick encodeULEB128(PtrSize + 1, OS);
22373471bf0Spatrick
22473471bf0Spatrick OS << uint8_t(dwarf::DW_LNE_set_address);
22573471bf0Spatrick Offset = OS.tell();
22673471bf0Spatrick Fixup = PtrSize == 4 ? std::make_pair(RISCV::fixup_riscv_add_32,
22773471bf0Spatrick RISCV::fixup_riscv_sub_32)
22873471bf0Spatrick : std::make_pair(RISCV::fixup_riscv_add_64,
22973471bf0Spatrick RISCV::fixup_riscv_sub_64);
23073471bf0Spatrick OS.write_zeros(PtrSize);
23173471bf0Spatrick } else {
23273471bf0Spatrick OS << uint8_t(dwarf::DW_LNS_fixed_advance_pc);
23373471bf0Spatrick Offset = OS.tell();
23473471bf0Spatrick Fixup = {RISCV::fixup_riscv_add_16, RISCV::fixup_riscv_sub_16};
23573471bf0Spatrick support::endian::write<uint16_t>(OS, 0, support::little);
23673471bf0Spatrick }
23773471bf0Spatrick
23873471bf0Spatrick const MCBinaryExpr &MBE = cast<MCBinaryExpr>(AddrDelta);
23973471bf0Spatrick Fixups.push_back(MCFixup::create(
24073471bf0Spatrick Offset, MBE.getLHS(), static_cast<MCFixupKind>(std::get<0>(Fixup))));
24173471bf0Spatrick Fixups.push_back(MCFixup::create(
24273471bf0Spatrick Offset, MBE.getRHS(), static_cast<MCFixupKind>(std::get<1>(Fixup))));
24373471bf0Spatrick
24473471bf0Spatrick if (LineDelta == INT64_MAX) {
24573471bf0Spatrick OS << uint8_t(dwarf::DW_LNS_extended_op);
24673471bf0Spatrick OS << uint8_t(1);
24773471bf0Spatrick OS << uint8_t(dwarf::DW_LNE_end_sequence);
24873471bf0Spatrick } else {
24973471bf0Spatrick OS << uint8_t(dwarf::DW_LNS_copy);
25073471bf0Spatrick }
25173471bf0Spatrick
25273471bf0Spatrick WasRelaxed = OldSize != Data.size();
25373471bf0Spatrick return true;
25473471bf0Spatrick }
25573471bf0Spatrick
relaxDwarfCFA(MCDwarfCallFrameFragment & DF,MCAsmLayout & Layout,bool & WasRelaxed) const25673471bf0Spatrick bool RISCVAsmBackend::relaxDwarfCFA(MCDwarfCallFrameFragment &DF,
25773471bf0Spatrick MCAsmLayout &Layout,
25873471bf0Spatrick bool &WasRelaxed) const {
25973471bf0Spatrick
26073471bf0Spatrick const MCExpr &AddrDelta = DF.getAddrDelta();
26173471bf0Spatrick SmallVectorImpl<char> &Data = DF.getContents();
26273471bf0Spatrick SmallVectorImpl<MCFixup> &Fixups = DF.getFixups();
26373471bf0Spatrick size_t OldSize = Data.size();
26473471bf0Spatrick
26573471bf0Spatrick int64_t Value;
26673471bf0Spatrick bool IsAbsolute = AddrDelta.evaluateKnownAbsolute(Value, Layout);
26773471bf0Spatrick assert(IsAbsolute && "CFA with invalid expression");
26873471bf0Spatrick (void)IsAbsolute;
26973471bf0Spatrick
27073471bf0Spatrick Data.clear();
27173471bf0Spatrick Fixups.clear();
27273471bf0Spatrick raw_svector_ostream OS(Data);
27373471bf0Spatrick
27473471bf0Spatrick assert(
27573471bf0Spatrick Layout.getAssembler().getContext().getAsmInfo()->getMinInstAlignment() ==
27673471bf0Spatrick 1 &&
27773471bf0Spatrick "expected 1-byte alignment");
27873471bf0Spatrick if (Value == 0) {
27973471bf0Spatrick WasRelaxed = OldSize != Data.size();
28073471bf0Spatrick return true;
28173471bf0Spatrick }
28273471bf0Spatrick
28373471bf0Spatrick auto AddFixups = [&Fixups, &AddrDelta](unsigned Offset,
28473471bf0Spatrick std::pair<unsigned, unsigned> Fixup) {
28573471bf0Spatrick const MCBinaryExpr &MBE = cast<MCBinaryExpr>(AddrDelta);
28673471bf0Spatrick Fixups.push_back(MCFixup::create(
28773471bf0Spatrick Offset, MBE.getLHS(), static_cast<MCFixupKind>(std::get<0>(Fixup))));
28873471bf0Spatrick Fixups.push_back(MCFixup::create(
28973471bf0Spatrick Offset, MBE.getRHS(), static_cast<MCFixupKind>(std::get<1>(Fixup))));
29073471bf0Spatrick };
29173471bf0Spatrick
29273471bf0Spatrick if (isUIntN(6, Value)) {
29373471bf0Spatrick OS << uint8_t(dwarf::DW_CFA_advance_loc);
29473471bf0Spatrick AddFixups(0, {RISCV::fixup_riscv_set_6b, RISCV::fixup_riscv_sub_6b});
29573471bf0Spatrick } else if (isUInt<8>(Value)) {
29673471bf0Spatrick OS << uint8_t(dwarf::DW_CFA_advance_loc1);
29773471bf0Spatrick support::endian::write<uint8_t>(OS, 0, support::little);
29873471bf0Spatrick AddFixups(1, {RISCV::fixup_riscv_set_8, RISCV::fixup_riscv_sub_8});
29973471bf0Spatrick } else if (isUInt<16>(Value)) {
30073471bf0Spatrick OS << uint8_t(dwarf::DW_CFA_advance_loc2);
30173471bf0Spatrick support::endian::write<uint16_t>(OS, 0, support::little);
30273471bf0Spatrick AddFixups(1, {RISCV::fixup_riscv_set_16, RISCV::fixup_riscv_sub_16});
30373471bf0Spatrick } else if (isUInt<32>(Value)) {
30473471bf0Spatrick OS << uint8_t(dwarf::DW_CFA_advance_loc4);
30573471bf0Spatrick support::endian::write<uint32_t>(OS, 0, support::little);
30673471bf0Spatrick AddFixups(1, {RISCV::fixup_riscv_set_32, RISCV::fixup_riscv_sub_32});
30773471bf0Spatrick } else {
30873471bf0Spatrick llvm_unreachable("unsupported CFA encoding");
30973471bf0Spatrick }
31073471bf0Spatrick
31173471bf0Spatrick WasRelaxed = OldSize != Data.size();
31273471bf0Spatrick return true;
31373471bf0Spatrick }
31473471bf0Spatrick
31509467b48Spatrick // Given a compressed control flow instruction this function returns
31609467b48Spatrick // the expanded instruction.
getRelaxedOpcode(unsigned Op) const31709467b48Spatrick unsigned RISCVAsmBackend::getRelaxedOpcode(unsigned Op) const {
31809467b48Spatrick switch (Op) {
31909467b48Spatrick default:
32009467b48Spatrick return Op;
32109467b48Spatrick case RISCV::C_BEQZ:
32209467b48Spatrick return RISCV::BEQ;
32309467b48Spatrick case RISCV::C_BNEZ:
32409467b48Spatrick return RISCV::BNE;
32509467b48Spatrick case RISCV::C_J:
32609467b48Spatrick case RISCV::C_JAL: // fall through.
32709467b48Spatrick return RISCV::JAL;
32809467b48Spatrick }
32909467b48Spatrick }
33009467b48Spatrick
mayNeedRelaxation(const MCInst & Inst,const MCSubtargetInfo & STI) const33109467b48Spatrick bool RISCVAsmBackend::mayNeedRelaxation(const MCInst &Inst,
33209467b48Spatrick const MCSubtargetInfo &STI) const {
33309467b48Spatrick return getRelaxedOpcode(Inst.getOpcode()) != Inst.getOpcode();
33409467b48Spatrick }
33509467b48Spatrick
writeNopData(raw_ostream & OS,uint64_t Count,const MCSubtargetInfo * STI) const336*d415bd75Srobert bool RISCVAsmBackend::writeNopData(raw_ostream &OS, uint64_t Count,
337*d415bd75Srobert const MCSubtargetInfo *STI) const {
338*d415bd75Srobert // We mostly follow binutils' convention here: align to even boundary with a
339*d415bd75Srobert // 0-fill padding. We emit up to 1 2-byte nop, though we use c.nop if RVC is
340*d415bd75Srobert // enabled or 0-fill otherwise. The remainder is now padded with 4-byte nops.
34109467b48Spatrick
342*d415bd75Srobert // Instructions always are at even addresses. We must be in a data area or
343*d415bd75Srobert // be unaligned due to some other reason.
344*d415bd75Srobert if (Count % 2) {
345*d415bd75Srobert OS.write("\0", 1);
346*d415bd75Srobert Count -= 1;
347*d415bd75Srobert }
348*d415bd75Srobert
349*d415bd75Srobert bool HasStdExtC = STI->getFeatureBits()[RISCV::FeatureStdExtC];
350*d415bd75Srobert bool HasStdExtZca = STI->getFeatureBits()[RISCV::FeatureExtZca];
351*d415bd75Srobert // The canonical nop on RVC is c.nop.
352*d415bd75Srobert if (Count % 4 == 2) {
353*d415bd75Srobert OS.write((HasStdExtC || HasStdExtZca) ? "\x01\0" : "\0\0", 2);
354*d415bd75Srobert Count -= 2;
355*d415bd75Srobert }
35609467b48Spatrick
35709467b48Spatrick // The canonical nop on RISC-V is addi x0, x0, 0.
35809467b48Spatrick for (; Count >= 4; Count -= 4)
35909467b48Spatrick OS.write("\x13\0\0\0", 4);
36009467b48Spatrick
36109467b48Spatrick return true;
36209467b48Spatrick }
36309467b48Spatrick
adjustFixupValue(const MCFixup & Fixup,uint64_t Value,MCContext & Ctx)36409467b48Spatrick static uint64_t adjustFixupValue(const MCFixup &Fixup, uint64_t Value,
36509467b48Spatrick MCContext &Ctx) {
36609467b48Spatrick switch (Fixup.getTargetKind()) {
36709467b48Spatrick default:
36809467b48Spatrick llvm_unreachable("Unknown fixup kind!");
36909467b48Spatrick case RISCV::fixup_riscv_got_hi20:
37009467b48Spatrick case RISCV::fixup_riscv_tls_got_hi20:
37109467b48Spatrick case RISCV::fixup_riscv_tls_gd_hi20:
37209467b48Spatrick llvm_unreachable("Relocation should be unconditionally forced\n");
37373471bf0Spatrick case RISCV::fixup_riscv_set_8:
37473471bf0Spatrick case RISCV::fixup_riscv_add_8:
37573471bf0Spatrick case RISCV::fixup_riscv_sub_8:
37673471bf0Spatrick case RISCV::fixup_riscv_set_16:
37773471bf0Spatrick case RISCV::fixup_riscv_add_16:
37873471bf0Spatrick case RISCV::fixup_riscv_sub_16:
37973471bf0Spatrick case RISCV::fixup_riscv_set_32:
38073471bf0Spatrick case RISCV::fixup_riscv_add_32:
38173471bf0Spatrick case RISCV::fixup_riscv_sub_32:
38273471bf0Spatrick case RISCV::fixup_riscv_add_64:
38373471bf0Spatrick case RISCV::fixup_riscv_sub_64:
38409467b48Spatrick case FK_Data_1:
38509467b48Spatrick case FK_Data_2:
38609467b48Spatrick case FK_Data_4:
38709467b48Spatrick case FK_Data_8:
38809467b48Spatrick case FK_Data_6b:
38909467b48Spatrick return Value;
39073471bf0Spatrick case RISCV::fixup_riscv_set_6b:
39173471bf0Spatrick return Value & 0x03;
39209467b48Spatrick case RISCV::fixup_riscv_lo12_i:
39309467b48Spatrick case RISCV::fixup_riscv_pcrel_lo12_i:
39409467b48Spatrick case RISCV::fixup_riscv_tprel_lo12_i:
39509467b48Spatrick return Value & 0xfff;
39609467b48Spatrick case RISCV::fixup_riscv_lo12_s:
39709467b48Spatrick case RISCV::fixup_riscv_pcrel_lo12_s:
39809467b48Spatrick case RISCV::fixup_riscv_tprel_lo12_s:
39909467b48Spatrick return (((Value >> 5) & 0x7f) << 25) | ((Value & 0x1f) << 7);
40009467b48Spatrick case RISCV::fixup_riscv_hi20:
40109467b48Spatrick case RISCV::fixup_riscv_pcrel_hi20:
40209467b48Spatrick case RISCV::fixup_riscv_tprel_hi20:
40309467b48Spatrick // Add 1 if bit 11 is 1, to compensate for low 12 bits being negative.
40409467b48Spatrick return ((Value + 0x800) >> 12) & 0xfffff;
40509467b48Spatrick case RISCV::fixup_riscv_jal: {
40609467b48Spatrick if (!isInt<21>(Value))
40709467b48Spatrick Ctx.reportError(Fixup.getLoc(), "fixup value out of range");
40809467b48Spatrick if (Value & 0x1)
40909467b48Spatrick Ctx.reportError(Fixup.getLoc(), "fixup value must be 2-byte aligned");
41009467b48Spatrick // Need to produce imm[19|10:1|11|19:12] from the 21-bit Value.
41109467b48Spatrick unsigned Sbit = (Value >> 20) & 0x1;
41209467b48Spatrick unsigned Hi8 = (Value >> 12) & 0xff;
41309467b48Spatrick unsigned Mid1 = (Value >> 11) & 0x1;
41409467b48Spatrick unsigned Lo10 = (Value >> 1) & 0x3ff;
41509467b48Spatrick // Inst{31} = Sbit;
41609467b48Spatrick // Inst{30-21} = Lo10;
41709467b48Spatrick // Inst{20} = Mid1;
41809467b48Spatrick // Inst{19-12} = Hi8;
41909467b48Spatrick Value = (Sbit << 19) | (Lo10 << 9) | (Mid1 << 8) | Hi8;
42009467b48Spatrick return Value;
42109467b48Spatrick }
42209467b48Spatrick case RISCV::fixup_riscv_branch: {
42309467b48Spatrick if (!isInt<13>(Value))
42409467b48Spatrick Ctx.reportError(Fixup.getLoc(), "fixup value out of range");
42509467b48Spatrick if (Value & 0x1)
42609467b48Spatrick Ctx.reportError(Fixup.getLoc(), "fixup value must be 2-byte aligned");
42709467b48Spatrick // Need to extract imm[12], imm[10:5], imm[4:1], imm[11] from the 13-bit
42809467b48Spatrick // Value.
42909467b48Spatrick unsigned Sbit = (Value >> 12) & 0x1;
43009467b48Spatrick unsigned Hi1 = (Value >> 11) & 0x1;
43109467b48Spatrick unsigned Mid6 = (Value >> 5) & 0x3f;
43209467b48Spatrick unsigned Lo4 = (Value >> 1) & 0xf;
43309467b48Spatrick // Inst{31} = Sbit;
43409467b48Spatrick // Inst{30-25} = Mid6;
43509467b48Spatrick // Inst{11-8} = Lo4;
43609467b48Spatrick // Inst{7} = Hi1;
43709467b48Spatrick Value = (Sbit << 31) | (Mid6 << 25) | (Lo4 << 8) | (Hi1 << 7);
43809467b48Spatrick return Value;
43909467b48Spatrick }
44009467b48Spatrick case RISCV::fixup_riscv_call:
44109467b48Spatrick case RISCV::fixup_riscv_call_plt: {
44209467b48Spatrick // Jalr will add UpperImm with the sign-extended 12-bit LowerImm,
44309467b48Spatrick // we need to add 0x800ULL before extract upper bits to reflect the
44409467b48Spatrick // effect of the sign extension.
44509467b48Spatrick uint64_t UpperImm = (Value + 0x800ULL) & 0xfffff000ULL;
44609467b48Spatrick uint64_t LowerImm = Value & 0xfffULL;
44709467b48Spatrick return UpperImm | ((LowerImm << 20) << 32);
44809467b48Spatrick }
44909467b48Spatrick case RISCV::fixup_riscv_rvc_jump: {
45009467b48Spatrick // Need to produce offset[11|4|9:8|10|6|7|3:1|5] from the 11-bit Value.
45109467b48Spatrick unsigned Bit11 = (Value >> 11) & 0x1;
45209467b48Spatrick unsigned Bit4 = (Value >> 4) & 0x1;
45309467b48Spatrick unsigned Bit9_8 = (Value >> 8) & 0x3;
45409467b48Spatrick unsigned Bit10 = (Value >> 10) & 0x1;
45509467b48Spatrick unsigned Bit6 = (Value >> 6) & 0x1;
45609467b48Spatrick unsigned Bit7 = (Value >> 7) & 0x1;
45709467b48Spatrick unsigned Bit3_1 = (Value >> 1) & 0x7;
45809467b48Spatrick unsigned Bit5 = (Value >> 5) & 0x1;
45909467b48Spatrick Value = (Bit11 << 10) | (Bit4 << 9) | (Bit9_8 << 7) | (Bit10 << 6) |
46009467b48Spatrick (Bit6 << 5) | (Bit7 << 4) | (Bit3_1 << 1) | Bit5;
46109467b48Spatrick return Value;
46209467b48Spatrick }
46309467b48Spatrick case RISCV::fixup_riscv_rvc_branch: {
46409467b48Spatrick // Need to produce offset[8|4:3], [reg 3 bit], offset[7:6|2:1|5]
46509467b48Spatrick unsigned Bit8 = (Value >> 8) & 0x1;
46609467b48Spatrick unsigned Bit7_6 = (Value >> 6) & 0x3;
46709467b48Spatrick unsigned Bit5 = (Value >> 5) & 0x1;
46809467b48Spatrick unsigned Bit4_3 = (Value >> 3) & 0x3;
46909467b48Spatrick unsigned Bit2_1 = (Value >> 1) & 0x3;
47009467b48Spatrick Value = (Bit8 << 12) | (Bit4_3 << 10) | (Bit7_6 << 5) | (Bit2_1 << 3) |
47109467b48Spatrick (Bit5 << 2);
47209467b48Spatrick return Value;
47309467b48Spatrick }
47409467b48Spatrick
47509467b48Spatrick }
47609467b48Spatrick }
47709467b48Spatrick
evaluateTargetFixup(const MCAssembler & Asm,const MCAsmLayout & Layout,const MCFixup & Fixup,const MCFragment * DF,const MCValue & Target,uint64_t & Value,bool & WasForced)47809467b48Spatrick bool RISCVAsmBackend::evaluateTargetFixup(
47909467b48Spatrick const MCAssembler &Asm, const MCAsmLayout &Layout, const MCFixup &Fixup,
48009467b48Spatrick const MCFragment *DF, const MCValue &Target, uint64_t &Value,
48109467b48Spatrick bool &WasForced) {
48209467b48Spatrick const MCFixup *AUIPCFixup;
48309467b48Spatrick const MCFragment *AUIPCDF;
48409467b48Spatrick MCValue AUIPCTarget;
48509467b48Spatrick switch (Fixup.getTargetKind()) {
48609467b48Spatrick default:
48709467b48Spatrick llvm_unreachable("Unexpected fixup kind!");
48809467b48Spatrick case RISCV::fixup_riscv_pcrel_hi20:
48909467b48Spatrick AUIPCFixup = &Fixup;
49009467b48Spatrick AUIPCDF = DF;
49109467b48Spatrick AUIPCTarget = Target;
49209467b48Spatrick break;
49309467b48Spatrick case RISCV::fixup_riscv_pcrel_lo12_i:
49409467b48Spatrick case RISCV::fixup_riscv_pcrel_lo12_s: {
49509467b48Spatrick AUIPCFixup = cast<RISCVMCExpr>(Fixup.getValue())->getPCRelHiFixup(&AUIPCDF);
49609467b48Spatrick if (!AUIPCFixup) {
49709467b48Spatrick Asm.getContext().reportError(Fixup.getLoc(),
49809467b48Spatrick "could not find corresponding %pcrel_hi");
49909467b48Spatrick return true;
50009467b48Spatrick }
50109467b48Spatrick
50209467b48Spatrick // MCAssembler::evaluateFixup will emit an error for this case when it sees
50309467b48Spatrick // the %pcrel_hi, so don't duplicate it when also seeing the %pcrel_lo.
50409467b48Spatrick const MCExpr *AUIPCExpr = AUIPCFixup->getValue();
50509467b48Spatrick if (!AUIPCExpr->evaluateAsRelocatable(AUIPCTarget, &Layout, AUIPCFixup))
50609467b48Spatrick return true;
50709467b48Spatrick break;
50809467b48Spatrick }
50909467b48Spatrick }
51009467b48Spatrick
51109467b48Spatrick if (!AUIPCTarget.getSymA() || AUIPCTarget.getSymB())
51209467b48Spatrick return false;
51309467b48Spatrick
51409467b48Spatrick const MCSymbolRefExpr *A = AUIPCTarget.getSymA();
51509467b48Spatrick const MCSymbol &SA = A->getSymbol();
51609467b48Spatrick if (A->getKind() != MCSymbolRefExpr::VK_None || SA.isUndefined())
51709467b48Spatrick return false;
51809467b48Spatrick
51909467b48Spatrick auto *Writer = Asm.getWriterPtr();
52009467b48Spatrick if (!Writer)
52109467b48Spatrick return false;
52209467b48Spatrick
52309467b48Spatrick bool IsResolved = Writer->isSymbolRefDifferenceFullyResolvedImpl(
52409467b48Spatrick Asm, SA, *AUIPCDF, false, true);
52509467b48Spatrick if (!IsResolved)
52609467b48Spatrick return false;
52709467b48Spatrick
52809467b48Spatrick Value = Layout.getSymbolOffset(SA) + AUIPCTarget.getConstant();
52909467b48Spatrick Value -= Layout.getFragmentOffset(AUIPCDF) + AUIPCFixup->getOffset();
53009467b48Spatrick
53109467b48Spatrick if (shouldForceRelocation(Asm, *AUIPCFixup, AUIPCTarget)) {
53209467b48Spatrick WasForced = true;
53309467b48Spatrick return false;
53409467b48Spatrick }
53509467b48Spatrick
53609467b48Spatrick return true;
53709467b48Spatrick }
53809467b48Spatrick
applyFixup(const MCAssembler & Asm,const MCFixup & Fixup,const MCValue & Target,MutableArrayRef<char> Data,uint64_t Value,bool IsResolved,const MCSubtargetInfo * STI) const53909467b48Spatrick void RISCVAsmBackend::applyFixup(const MCAssembler &Asm, const MCFixup &Fixup,
54009467b48Spatrick const MCValue &Target,
54109467b48Spatrick MutableArrayRef<char> Data, uint64_t Value,
54209467b48Spatrick bool IsResolved,
54309467b48Spatrick const MCSubtargetInfo *STI) const {
544097a140dSpatrick MCFixupKind Kind = Fixup.getKind();
545097a140dSpatrick if (Kind >= FirstLiteralRelocationKind)
546097a140dSpatrick return;
54709467b48Spatrick MCContext &Ctx = Asm.getContext();
548097a140dSpatrick MCFixupKindInfo Info = getFixupKindInfo(Kind);
54909467b48Spatrick if (!Value)
55009467b48Spatrick return; // Doesn't change encoding.
55109467b48Spatrick // Apply any target-specific value adjustments.
55209467b48Spatrick Value = adjustFixupValue(Fixup, Value, Ctx);
55309467b48Spatrick
55409467b48Spatrick // Shift the value into position.
55509467b48Spatrick Value <<= Info.TargetOffset;
55609467b48Spatrick
55709467b48Spatrick unsigned Offset = Fixup.getOffset();
55809467b48Spatrick unsigned NumBytes = alignTo(Info.TargetSize + Info.TargetOffset, 8) / 8;
55909467b48Spatrick
56009467b48Spatrick assert(Offset + NumBytes <= Data.size() && "Invalid fixup offset!");
56109467b48Spatrick
56209467b48Spatrick // For each byte of the fragment that the fixup touches, mask in the
56309467b48Spatrick // bits from the fixup value.
56409467b48Spatrick for (unsigned i = 0; i != NumBytes; ++i) {
56509467b48Spatrick Data[Offset + i] |= uint8_t((Value >> (i * 8)) & 0xff);
56609467b48Spatrick }
56709467b48Spatrick }
56809467b48Spatrick
56909467b48Spatrick // Linker relaxation may change code size. We have to insert Nops
57009467b48Spatrick // for .align directive when linker relaxation enabled. So then Linker
57109467b48Spatrick // could satisfy alignment by removing Nops.
57209467b48Spatrick // The function return the total Nops Size we need to insert.
shouldInsertExtraNopBytesForCodeAlign(const MCAlignFragment & AF,unsigned & Size)57309467b48Spatrick bool RISCVAsmBackend::shouldInsertExtraNopBytesForCodeAlign(
57409467b48Spatrick const MCAlignFragment &AF, unsigned &Size) {
57509467b48Spatrick // Calculate Nops Size only when linker relaxation enabled.
576*d415bd75Srobert const MCSubtargetInfo *STI = AF.getSubtargetInfo();
577*d415bd75Srobert if (!STI->getFeatureBits()[RISCV::FeatureRelax])
57809467b48Spatrick return false;
57909467b48Spatrick
580*d415bd75Srobert bool UseCompressedNop = STI->getFeatureBits()[RISCV::FeatureStdExtC] ||
581*d415bd75Srobert STI->getFeatureBits()[RISCV::FeatureExtZca];
582*d415bd75Srobert unsigned MinNopLen = UseCompressedNop ? 2 : 4;
58309467b48Spatrick
58409467b48Spatrick if (AF.getAlignment() <= MinNopLen) {
58509467b48Spatrick return false;
58609467b48Spatrick } else {
587*d415bd75Srobert Size = AF.getAlignment().value() - MinNopLen;
58809467b48Spatrick return true;
58909467b48Spatrick }
59009467b48Spatrick }
59109467b48Spatrick
59209467b48Spatrick // We need to insert R_RISCV_ALIGN relocation type to indicate the
59309467b48Spatrick // position of Nops and the total bytes of the Nops have been inserted
59409467b48Spatrick // when linker relaxation enabled.
59509467b48Spatrick // The function insert fixup_riscv_align fixup which eventually will
59609467b48Spatrick // transfer to R_RISCV_ALIGN relocation type.
shouldInsertFixupForCodeAlign(MCAssembler & Asm,const MCAsmLayout & Layout,MCAlignFragment & AF)59709467b48Spatrick bool RISCVAsmBackend::shouldInsertFixupForCodeAlign(MCAssembler &Asm,
59809467b48Spatrick const MCAsmLayout &Layout,
59909467b48Spatrick MCAlignFragment &AF) {
60009467b48Spatrick // Insert the fixup only when linker relaxation enabled.
601*d415bd75Srobert const MCSubtargetInfo *STI = AF.getSubtargetInfo();
602*d415bd75Srobert if (!STI->getFeatureBits()[RISCV::FeatureRelax])
60309467b48Spatrick return false;
60409467b48Spatrick
60509467b48Spatrick // Calculate total Nops we need to insert. If there are none to insert
60609467b48Spatrick // then simply return.
60709467b48Spatrick unsigned Count;
60809467b48Spatrick if (!shouldInsertExtraNopBytesForCodeAlign(AF, Count) || (Count == 0))
60909467b48Spatrick return false;
61009467b48Spatrick
61109467b48Spatrick MCContext &Ctx = Asm.getContext();
61209467b48Spatrick const MCExpr *Dummy = MCConstantExpr::create(0, Ctx);
61309467b48Spatrick // Create fixup_riscv_align fixup.
61409467b48Spatrick MCFixup Fixup =
61509467b48Spatrick MCFixup::create(0, Dummy, MCFixupKind(RISCV::fixup_riscv_align), SMLoc());
61609467b48Spatrick
61709467b48Spatrick uint64_t FixedValue = 0;
61809467b48Spatrick MCValue NopBytes = MCValue::get(Count);
61909467b48Spatrick
62009467b48Spatrick Asm.getWriter().recordRelocation(Asm, Layout, &AF, Fixup, NopBytes,
62109467b48Spatrick FixedValue);
62209467b48Spatrick
62309467b48Spatrick return true;
62409467b48Spatrick }
62509467b48Spatrick
62609467b48Spatrick std::unique_ptr<MCObjectTargetWriter>
createObjectTargetWriter() const62709467b48Spatrick RISCVAsmBackend::createObjectTargetWriter() const {
62809467b48Spatrick return createRISCVELFObjectWriter(OSABI, Is64Bit);
62909467b48Spatrick }
63009467b48Spatrick
createRISCVAsmBackend(const Target & T,const MCSubtargetInfo & STI,const MCRegisterInfo & MRI,const MCTargetOptions & Options)63109467b48Spatrick MCAsmBackend *llvm::createRISCVAsmBackend(const Target &T,
63209467b48Spatrick const MCSubtargetInfo &STI,
63309467b48Spatrick const MCRegisterInfo &MRI,
63409467b48Spatrick const MCTargetOptions &Options) {
63509467b48Spatrick const Triple &TT = STI.getTargetTriple();
63609467b48Spatrick uint8_t OSABI = MCELFObjectTargetWriter::getOSABI(TT.getOS());
63709467b48Spatrick return new RISCVAsmBackend(STI, OSABI, TT.isArch64Bit(), Options);
63809467b48Spatrick }
639