1 //===-- RISCVAsmParser.cpp - Parse RISCV assembly to MCInst instructions --===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "MCTargetDesc/RISCVAsmBackend.h" 10 #include "MCTargetDesc/RISCVMCExpr.h" 11 #include "MCTargetDesc/RISCVMCTargetDesc.h" 12 #include "MCTargetDesc/RISCVTargetStreamer.h" 13 #include "RISCVInstrInfo.h" 14 #include "TargetInfo/RISCVTargetInfo.h" 15 #include "Utils/RISCVBaseInfo.h" 16 #include "Utils/RISCVMatInt.h" 17 #include "llvm/ADT/STLExtras.h" 18 #include "llvm/ADT/SmallBitVector.h" 19 #include "llvm/ADT/SmallString.h" 20 #include "llvm/ADT/SmallVector.h" 21 #include "llvm/ADT/Statistic.h" 22 #include "llvm/ADT/StringSwitch.h" 23 #include "llvm/CodeGen/Register.h" 24 #include "llvm/MC/MCAssembler.h" 25 #include "llvm/MC/MCContext.h" 26 #include "llvm/MC/MCExpr.h" 27 #include "llvm/MC/MCInst.h" 28 #include "llvm/MC/MCInstBuilder.h" 29 #include "llvm/MC/MCObjectFileInfo.h" 30 #include "llvm/MC/MCParser/MCAsmLexer.h" 31 #include "llvm/MC/MCParser/MCParsedAsmOperand.h" 32 #include "llvm/MC/MCParser/MCTargetAsmParser.h" 33 #include "llvm/MC/MCRegisterInfo.h" 34 #include "llvm/MC/MCStreamer.h" 35 #include "llvm/MC/MCSubtargetInfo.h" 36 #include "llvm/Support/Casting.h" 37 #include "llvm/Support/MathExtras.h" 38 #include "llvm/Support/RISCVAttributes.h" 39 #include "llvm/Support/TargetRegistry.h" 40 41 #include <limits> 42 43 using namespace llvm; 44 45 #define DEBUG_TYPE "riscv-asm-parser" 46 47 // Include the auto-generated portion of the compress emitter. 48 #define GEN_COMPRESS_INSTR 49 #include "RISCVGenCompressInstEmitter.inc" 50 51 STATISTIC(RISCVNumInstrsCompressed, 52 "Number of RISC-V Compressed instructions emitted"); 53 54 namespace { 55 struct RISCVOperand; 56 57 struct ParserOptionsSet { 58 bool IsPicEnabled; 59 }; 60 61 class RISCVAsmParser : public MCTargetAsmParser { 62 SmallVector<FeatureBitset, 4> FeatureBitStack; 63 64 SmallVector<ParserOptionsSet, 4> ParserOptionsStack; 65 ParserOptionsSet ParserOptions; 66 67 SMLoc getLoc() const { return getParser().getTok().getLoc(); } 68 bool isRV64() const { return getSTI().hasFeature(RISCV::Feature64Bit); } 69 bool isRV32E() const { return getSTI().hasFeature(RISCV::FeatureRV32E); } 70 71 RISCVTargetStreamer &getTargetStreamer() { 72 MCTargetStreamer &TS = *getParser().getStreamer().getTargetStreamer(); 73 return static_cast<RISCVTargetStreamer &>(TS); 74 } 75 76 unsigned validateTargetOperandClass(MCParsedAsmOperand &Op, 77 unsigned Kind) override; 78 79 bool generateImmOutOfRangeError(OperandVector &Operands, uint64_t ErrorInfo, 80 int64_t Lower, int64_t Upper, Twine Msg); 81 82 bool MatchAndEmitInstruction(SMLoc IDLoc, unsigned &Opcode, 83 OperandVector &Operands, MCStreamer &Out, 84 uint64_t &ErrorInfo, 85 bool MatchingInlineAsm) override; 86 87 bool ParseRegister(unsigned &RegNo, SMLoc &StartLoc, SMLoc &EndLoc) override; 88 OperandMatchResultTy tryParseRegister(unsigned &RegNo, SMLoc &StartLoc, 89 SMLoc &EndLoc) override; 90 91 bool ParseInstruction(ParseInstructionInfo &Info, StringRef Name, 92 SMLoc NameLoc, OperandVector &Operands) override; 93 94 bool ParseDirective(AsmToken DirectiveID) override; 95 96 // Helper to actually emit an instruction to the MCStreamer. Also, when 97 // possible, compression of the instruction is performed. 98 void emitToStreamer(MCStreamer &S, const MCInst &Inst); 99 100 // Helper to emit a combination of LUI, ADDI(W), and SLLI instructions that 101 // synthesize the desired immedate value into the destination register. 102 void emitLoadImm(Register DestReg, int64_t Value, MCStreamer &Out); 103 104 // Helper to emit a combination of AUIPC and SecondOpcode. Used to implement 105 // helpers such as emitLoadLocalAddress and emitLoadAddress. 106 void emitAuipcInstPair(MCOperand DestReg, MCOperand TmpReg, 107 const MCExpr *Symbol, RISCVMCExpr::VariantKind VKHi, 108 unsigned SecondOpcode, SMLoc IDLoc, MCStreamer &Out); 109 110 // Helper to emit pseudo instruction "lla" used in PC-rel addressing. 111 void emitLoadLocalAddress(MCInst &Inst, SMLoc IDLoc, MCStreamer &Out); 112 113 // Helper to emit pseudo instruction "la" used in GOT/PC-rel addressing. 114 void emitLoadAddress(MCInst &Inst, SMLoc IDLoc, MCStreamer &Out); 115 116 // Helper to emit pseudo instruction "la.tls.ie" used in initial-exec TLS 117 // addressing. 118 void emitLoadTLSIEAddress(MCInst &Inst, SMLoc IDLoc, MCStreamer &Out); 119 120 // Helper to emit pseudo instruction "la.tls.gd" used in global-dynamic TLS 121 // addressing. 122 void emitLoadTLSGDAddress(MCInst &Inst, SMLoc IDLoc, MCStreamer &Out); 123 124 // Helper to emit pseudo load/store instruction with a symbol. 125 void emitLoadStoreSymbol(MCInst &Inst, unsigned Opcode, SMLoc IDLoc, 126 MCStreamer &Out, bool HasTmpReg); 127 128 // Checks that a PseudoAddTPRel is using x4/tp in its second input operand. 129 // Enforcing this using a restricted register class for the second input 130 // operand of PseudoAddTPRel results in a poor diagnostic due to the fact 131 // 'add' is an overloaded mnemonic. 132 bool checkPseudoAddTPRel(MCInst &Inst, OperandVector &Operands); 133 134 // Check instruction constraints. 135 bool validateInstruction(MCInst &Inst, OperandVector &Operands); 136 137 /// Helper for processing MC instructions that have been successfully matched 138 /// by MatchAndEmitInstruction. Modifications to the emitted instructions, 139 /// like the expansion of pseudo instructions (e.g., "li"), can be performed 140 /// in this method. 141 bool processInstruction(MCInst &Inst, SMLoc IDLoc, OperandVector &Operands, 142 MCStreamer &Out); 143 144 // Auto-generated instruction matching functions 145 #define GET_ASSEMBLER_HEADER 146 #include "RISCVGenAsmMatcher.inc" 147 148 OperandMatchResultTy parseCSRSystemRegister(OperandVector &Operands); 149 OperandMatchResultTy parseImmediate(OperandVector &Operands); 150 OperandMatchResultTy parseRegister(OperandVector &Operands, 151 bool AllowParens = false); 152 OperandMatchResultTy parseMemOpBaseReg(OperandVector &Operands); 153 OperandMatchResultTy parseAtomicMemOp(OperandVector &Operands); 154 OperandMatchResultTy parseOperandWithModifier(OperandVector &Operands); 155 OperandMatchResultTy parseBareSymbol(OperandVector &Operands); 156 OperandMatchResultTy parseCallSymbol(OperandVector &Operands); 157 OperandMatchResultTy parsePseudoJumpSymbol(OperandVector &Operands); 158 OperandMatchResultTy parseJALOffset(OperandVector &Operands); 159 OperandMatchResultTy parseVTypeI(OperandVector &Operands); 160 OperandMatchResultTy parseMaskReg(OperandVector &Operands); 161 162 bool parseOperand(OperandVector &Operands, StringRef Mnemonic); 163 164 bool parseDirectiveOption(); 165 bool parseDirectiveAttribute(); 166 167 void setFeatureBits(uint64_t Feature, StringRef FeatureString) { 168 if (!(getSTI().getFeatureBits()[Feature])) { 169 MCSubtargetInfo &STI = copySTI(); 170 setAvailableFeatures( 171 ComputeAvailableFeatures(STI.ToggleFeature(FeatureString))); 172 } 173 } 174 175 bool getFeatureBits(uint64_t Feature) { 176 return getSTI().getFeatureBits()[Feature]; 177 } 178 179 void clearFeatureBits(uint64_t Feature, StringRef FeatureString) { 180 if (getSTI().getFeatureBits()[Feature]) { 181 MCSubtargetInfo &STI = copySTI(); 182 setAvailableFeatures( 183 ComputeAvailableFeatures(STI.ToggleFeature(FeatureString))); 184 } 185 } 186 187 void pushFeatureBits() { 188 assert(FeatureBitStack.size() == ParserOptionsStack.size() && 189 "These two stacks must be kept synchronized"); 190 FeatureBitStack.push_back(getSTI().getFeatureBits()); 191 ParserOptionsStack.push_back(ParserOptions); 192 } 193 194 bool popFeatureBits() { 195 assert(FeatureBitStack.size() == ParserOptionsStack.size() && 196 "These two stacks must be kept synchronized"); 197 if (FeatureBitStack.empty()) 198 return true; 199 200 FeatureBitset FeatureBits = FeatureBitStack.pop_back_val(); 201 copySTI().setFeatureBits(FeatureBits); 202 setAvailableFeatures(ComputeAvailableFeatures(FeatureBits)); 203 204 ParserOptions = ParserOptionsStack.pop_back_val(); 205 206 return false; 207 } 208 209 std::unique_ptr<RISCVOperand> defaultMaskRegOp() const; 210 211 public: 212 enum RISCVMatchResultTy { 213 Match_Dummy = FIRST_TARGET_MATCH_RESULT_TY, 214 #define GET_OPERAND_DIAGNOSTIC_TYPES 215 #include "RISCVGenAsmMatcher.inc" 216 #undef GET_OPERAND_DIAGNOSTIC_TYPES 217 }; 218 219 static bool classifySymbolRef(const MCExpr *Expr, 220 RISCVMCExpr::VariantKind &Kind, 221 int64_t &Addend); 222 223 RISCVAsmParser(const MCSubtargetInfo &STI, MCAsmParser &Parser, 224 const MCInstrInfo &MII, const MCTargetOptions &Options) 225 : MCTargetAsmParser(Options, STI, MII) { 226 Parser.addAliasForDirective(".half", ".2byte"); 227 Parser.addAliasForDirective(".hword", ".2byte"); 228 Parser.addAliasForDirective(".word", ".4byte"); 229 Parser.addAliasForDirective(".dword", ".8byte"); 230 setAvailableFeatures(ComputeAvailableFeatures(STI.getFeatureBits())); 231 232 auto ABIName = StringRef(Options.ABIName); 233 if (ABIName.endswith("f") && 234 !getSTI().getFeatureBits()[RISCV::FeatureStdExtF]) { 235 errs() << "Hard-float 'f' ABI can't be used for a target that " 236 "doesn't support the F instruction set extension (ignoring " 237 "target-abi)\n"; 238 } else if (ABIName.endswith("d") && 239 !getSTI().getFeatureBits()[RISCV::FeatureStdExtD]) { 240 errs() << "Hard-float 'd' ABI can't be used for a target that " 241 "doesn't support the D instruction set extension (ignoring " 242 "target-abi)\n"; 243 } 244 245 const MCObjectFileInfo *MOFI = Parser.getContext().getObjectFileInfo(); 246 ParserOptions.IsPicEnabled = MOFI->isPositionIndependent(); 247 } 248 }; 249 250 /// RISCVOperand - Instances of this class represent a parsed machine 251 /// instruction 252 struct RISCVOperand : public MCParsedAsmOperand { 253 254 enum class KindTy { 255 Token, 256 Register, 257 Immediate, 258 SystemRegister, 259 VType, 260 } Kind; 261 262 bool IsRV64; 263 264 struct RegOp { 265 Register RegNum; 266 }; 267 268 struct ImmOp { 269 const MCExpr *Val; 270 }; 271 272 struct SysRegOp { 273 const char *Data; 274 unsigned Length; 275 unsigned Encoding; 276 // FIXME: Add the Encoding parsed fields as needed for checks, 277 // e.g.: read/write or user/supervisor/machine privileges. 278 }; 279 280 enum class VSEW { 281 SEW_8 = 0, 282 SEW_16, 283 SEW_32, 284 SEW_64, 285 SEW_128, 286 SEW_256, 287 SEW_512, 288 SEW_1024, 289 }; 290 291 enum class VLMUL { 292 LMUL_1 = 0, 293 LMUL_2, 294 LMUL_4, 295 LMUL_8, 296 LMUL_F8 = 5, 297 LMUL_F4, 298 LMUL_F2 299 }; 300 301 struct VTypeOp { 302 VSEW Sew; 303 VLMUL Lmul; 304 bool TailAgnostic; 305 bool MaskedoffAgnostic; 306 unsigned Encoding; 307 }; 308 309 SMLoc StartLoc, EndLoc; 310 union { 311 StringRef Tok; 312 RegOp Reg; 313 ImmOp Imm; 314 struct SysRegOp SysReg; 315 struct VTypeOp VType; 316 }; 317 318 RISCVOperand(KindTy K) : MCParsedAsmOperand(), Kind(K) {} 319 320 public: 321 RISCVOperand(const RISCVOperand &o) : MCParsedAsmOperand() { 322 Kind = o.Kind; 323 IsRV64 = o.IsRV64; 324 StartLoc = o.StartLoc; 325 EndLoc = o.EndLoc; 326 switch (Kind) { 327 case KindTy::Register: 328 Reg = o.Reg; 329 break; 330 case KindTy::Immediate: 331 Imm = o.Imm; 332 break; 333 case KindTy::Token: 334 Tok = o.Tok; 335 break; 336 case KindTy::SystemRegister: 337 SysReg = o.SysReg; 338 break; 339 case KindTy::VType: 340 VType = o.VType; 341 break; 342 } 343 } 344 345 bool isToken() const override { return Kind == KindTy::Token; } 346 bool isReg() const override { return Kind == KindTy::Register; } 347 bool isV0Reg() const { 348 return Kind == KindTy::Register && Reg.RegNum == RISCV::V0; 349 } 350 bool isImm() const override { return Kind == KindTy::Immediate; } 351 bool isMem() const override { return false; } 352 bool isSystemRegister() const { return Kind == KindTy::SystemRegister; } 353 bool isVType() const { return Kind == KindTy::VType; } 354 355 bool isGPR() const { 356 return Kind == KindTy::Register && 357 RISCVMCRegisterClasses[RISCV::GPRRegClassID].contains(Reg.RegNum); 358 } 359 360 static bool evaluateConstantImm(const MCExpr *Expr, int64_t &Imm, 361 RISCVMCExpr::VariantKind &VK) { 362 if (auto *RE = dyn_cast<RISCVMCExpr>(Expr)) { 363 VK = RE->getKind(); 364 return RE->evaluateAsConstant(Imm); 365 } 366 367 if (auto CE = dyn_cast<MCConstantExpr>(Expr)) { 368 VK = RISCVMCExpr::VK_RISCV_None; 369 Imm = CE->getValue(); 370 return true; 371 } 372 373 return false; 374 } 375 376 // True if operand is a symbol with no modifiers, or a constant with no 377 // modifiers and isShiftedInt<N-1, 1>(Op). 378 template <int N> bool isBareSimmNLsb0() const { 379 int64_t Imm; 380 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 381 if (!isImm()) 382 return false; 383 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 384 bool IsValid; 385 if (!IsConstantImm) 386 IsValid = RISCVAsmParser::classifySymbolRef(getImm(), VK, Imm); 387 else 388 IsValid = isShiftedInt<N - 1, 1>(Imm); 389 return IsValid && VK == RISCVMCExpr::VK_RISCV_None; 390 } 391 392 // Predicate methods for AsmOperands defined in RISCVInstrInfo.td 393 394 bool isBareSymbol() const { 395 int64_t Imm; 396 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 397 // Must be of 'immediate' type but not a constant. 398 if (!isImm() || evaluateConstantImm(getImm(), Imm, VK)) 399 return false; 400 return RISCVAsmParser::classifySymbolRef(getImm(), VK, Imm) && 401 VK == RISCVMCExpr::VK_RISCV_None; 402 } 403 404 bool isCallSymbol() const { 405 int64_t Imm; 406 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 407 // Must be of 'immediate' type but not a constant. 408 if (!isImm() || evaluateConstantImm(getImm(), Imm, VK)) 409 return false; 410 return RISCVAsmParser::classifySymbolRef(getImm(), VK, Imm) && 411 (VK == RISCVMCExpr::VK_RISCV_CALL || 412 VK == RISCVMCExpr::VK_RISCV_CALL_PLT); 413 } 414 415 bool isPseudoJumpSymbol() const { 416 int64_t Imm; 417 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 418 // Must be of 'immediate' type but not a constant. 419 if (!isImm() || evaluateConstantImm(getImm(), Imm, VK)) 420 return false; 421 return RISCVAsmParser::classifySymbolRef(getImm(), VK, Imm) && 422 VK == RISCVMCExpr::VK_RISCV_CALL; 423 } 424 425 bool isTPRelAddSymbol() const { 426 int64_t Imm; 427 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 428 // Must be of 'immediate' type but not a constant. 429 if (!isImm() || evaluateConstantImm(getImm(), Imm, VK)) 430 return false; 431 return RISCVAsmParser::classifySymbolRef(getImm(), VK, Imm) && 432 VK == RISCVMCExpr::VK_RISCV_TPREL_ADD; 433 } 434 435 bool isCSRSystemRegister() const { return isSystemRegister(); } 436 437 bool isVTypeI() const { return isVType(); } 438 439 /// Return true if the operand is a valid for the fence instruction e.g. 440 /// ('iorw'). 441 bool isFenceArg() const { 442 if (!isImm()) 443 return false; 444 const MCExpr *Val = getImm(); 445 auto *SVal = dyn_cast<MCSymbolRefExpr>(Val); 446 if (!SVal || SVal->getKind() != MCSymbolRefExpr::VK_None) 447 return false; 448 449 StringRef Str = SVal->getSymbol().getName(); 450 // Letters must be unique, taken from 'iorw', and in ascending order. This 451 // holds as long as each individual character is one of 'iorw' and is 452 // greater than the previous character. 453 char Prev = '\0'; 454 for (char c : Str) { 455 if (c != 'i' && c != 'o' && c != 'r' && c != 'w') 456 return false; 457 if (c <= Prev) 458 return false; 459 Prev = c; 460 } 461 return true; 462 } 463 464 /// Return true if the operand is a valid floating point rounding mode. 465 bool isFRMArg() const { 466 if (!isImm()) 467 return false; 468 const MCExpr *Val = getImm(); 469 auto *SVal = dyn_cast<MCSymbolRefExpr>(Val); 470 if (!SVal || SVal->getKind() != MCSymbolRefExpr::VK_None) 471 return false; 472 473 StringRef Str = SVal->getSymbol().getName(); 474 475 return RISCVFPRndMode::stringToRoundingMode(Str) != RISCVFPRndMode::Invalid; 476 } 477 478 bool isImmXLenLI() const { 479 int64_t Imm; 480 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 481 if (!isImm()) 482 return false; 483 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 484 if (VK == RISCVMCExpr::VK_RISCV_LO || VK == RISCVMCExpr::VK_RISCV_PCREL_LO) 485 return true; 486 // Given only Imm, ensuring that the actually specified constant is either 487 // a signed or unsigned 64-bit number is unfortunately impossible. 488 return IsConstantImm && VK == RISCVMCExpr::VK_RISCV_None && 489 (isRV64() || (isInt<32>(Imm) || isUInt<32>(Imm))); 490 } 491 492 bool isUImmLog2XLen() const { 493 int64_t Imm; 494 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 495 if (!isImm()) 496 return false; 497 if (!evaluateConstantImm(getImm(), Imm, VK) || 498 VK != RISCVMCExpr::VK_RISCV_None) 499 return false; 500 return (isRV64() && isUInt<6>(Imm)) || isUInt<5>(Imm); 501 } 502 503 bool isUImmLog2XLenNonZero() const { 504 int64_t Imm; 505 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 506 if (!isImm()) 507 return false; 508 if (!evaluateConstantImm(getImm(), Imm, VK) || 509 VK != RISCVMCExpr::VK_RISCV_None) 510 return false; 511 if (Imm == 0) 512 return false; 513 return (isRV64() && isUInt<6>(Imm)) || isUInt<5>(Imm); 514 } 515 516 bool isUImmLog2XLenHalf() const { 517 int64_t Imm; 518 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 519 if (!isImm()) 520 return false; 521 if (!evaluateConstantImm(getImm(), Imm, VK) || 522 VK != RISCVMCExpr::VK_RISCV_None) 523 return false; 524 return (isRV64() && isUInt<5>(Imm)) || isUInt<4>(Imm); 525 } 526 527 bool isUImm5() const { 528 int64_t Imm; 529 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 530 if (!isImm()) 531 return false; 532 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 533 return IsConstantImm && isUInt<5>(Imm) && VK == RISCVMCExpr::VK_RISCV_None; 534 } 535 536 bool isUImm5NonZero() const { 537 int64_t Imm; 538 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 539 if (!isImm()) 540 return false; 541 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 542 return IsConstantImm && isUInt<5>(Imm) && (Imm != 0) && 543 VK == RISCVMCExpr::VK_RISCV_None; 544 } 545 546 bool isSImm5() const { 547 if (!isImm()) 548 return false; 549 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 550 int64_t Imm; 551 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 552 return IsConstantImm && isInt<5>(Imm) && VK == RISCVMCExpr::VK_RISCV_None; 553 } 554 555 bool isSImm6() const { 556 if (!isImm()) 557 return false; 558 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 559 int64_t Imm; 560 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 561 return IsConstantImm && isInt<6>(Imm) && 562 VK == RISCVMCExpr::VK_RISCV_None; 563 } 564 565 bool isSImm6NonZero() const { 566 if (!isImm()) 567 return false; 568 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 569 int64_t Imm; 570 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 571 return IsConstantImm && isInt<6>(Imm) && (Imm != 0) && 572 VK == RISCVMCExpr::VK_RISCV_None; 573 } 574 575 bool isCLUIImm() const { 576 if (!isImm()) 577 return false; 578 int64_t Imm; 579 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 580 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 581 return IsConstantImm && (Imm != 0) && 582 (isUInt<5>(Imm) || (Imm >= 0xfffe0 && Imm <= 0xfffff)) && 583 VK == RISCVMCExpr::VK_RISCV_None; 584 } 585 586 bool isUImm7Lsb00() const { 587 if (!isImm()) 588 return false; 589 int64_t Imm; 590 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 591 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 592 return IsConstantImm && isShiftedUInt<5, 2>(Imm) && 593 VK == RISCVMCExpr::VK_RISCV_None; 594 } 595 596 bool isUImm8Lsb00() const { 597 if (!isImm()) 598 return false; 599 int64_t Imm; 600 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 601 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 602 return IsConstantImm && isShiftedUInt<6, 2>(Imm) && 603 VK == RISCVMCExpr::VK_RISCV_None; 604 } 605 606 bool isUImm8Lsb000() const { 607 if (!isImm()) 608 return false; 609 int64_t Imm; 610 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 611 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 612 return IsConstantImm && isShiftedUInt<5, 3>(Imm) && 613 VK == RISCVMCExpr::VK_RISCV_None; 614 } 615 616 bool isSImm9Lsb0() const { return isBareSimmNLsb0<9>(); } 617 618 bool isUImm9Lsb000() const { 619 if (!isImm()) 620 return false; 621 int64_t Imm; 622 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 623 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 624 return IsConstantImm && isShiftedUInt<6, 3>(Imm) && 625 VK == RISCVMCExpr::VK_RISCV_None; 626 } 627 628 bool isUImm10Lsb00NonZero() const { 629 if (!isImm()) 630 return false; 631 int64_t Imm; 632 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 633 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 634 return IsConstantImm && isShiftedUInt<8, 2>(Imm) && (Imm != 0) && 635 VK == RISCVMCExpr::VK_RISCV_None; 636 } 637 638 bool isSImm12() const { 639 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 640 int64_t Imm; 641 bool IsValid; 642 if (!isImm()) 643 return false; 644 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 645 if (!IsConstantImm) 646 IsValid = RISCVAsmParser::classifySymbolRef(getImm(), VK, Imm); 647 else 648 IsValid = isInt<12>(Imm); 649 return IsValid && ((IsConstantImm && VK == RISCVMCExpr::VK_RISCV_None) || 650 VK == RISCVMCExpr::VK_RISCV_LO || 651 VK == RISCVMCExpr::VK_RISCV_PCREL_LO || 652 VK == RISCVMCExpr::VK_RISCV_TPREL_LO); 653 } 654 655 bool isSImm12Lsb0() const { return isBareSimmNLsb0<12>(); } 656 657 bool isSImm13Lsb0() const { return isBareSimmNLsb0<13>(); } 658 659 bool isSImm10Lsb0000NonZero() const { 660 if (!isImm()) 661 return false; 662 int64_t Imm; 663 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 664 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 665 return IsConstantImm && (Imm != 0) && isShiftedInt<6, 4>(Imm) && 666 VK == RISCVMCExpr::VK_RISCV_None; 667 } 668 669 bool isUImm20LUI() const { 670 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 671 int64_t Imm; 672 bool IsValid; 673 if (!isImm()) 674 return false; 675 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 676 if (!IsConstantImm) { 677 IsValid = RISCVAsmParser::classifySymbolRef(getImm(), VK, Imm); 678 return IsValid && (VK == RISCVMCExpr::VK_RISCV_HI || 679 VK == RISCVMCExpr::VK_RISCV_TPREL_HI); 680 } else { 681 return isUInt<20>(Imm) && (VK == RISCVMCExpr::VK_RISCV_None || 682 VK == RISCVMCExpr::VK_RISCV_HI || 683 VK == RISCVMCExpr::VK_RISCV_TPREL_HI); 684 } 685 } 686 687 bool isUImm20AUIPC() const { 688 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 689 int64_t Imm; 690 bool IsValid; 691 if (!isImm()) 692 return false; 693 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 694 if (!IsConstantImm) { 695 IsValid = RISCVAsmParser::classifySymbolRef(getImm(), VK, Imm); 696 return IsValid && (VK == RISCVMCExpr::VK_RISCV_PCREL_HI || 697 VK == RISCVMCExpr::VK_RISCV_GOT_HI || 698 VK == RISCVMCExpr::VK_RISCV_TLS_GOT_HI || 699 VK == RISCVMCExpr::VK_RISCV_TLS_GD_HI); 700 } else { 701 return isUInt<20>(Imm) && (VK == RISCVMCExpr::VK_RISCV_None || 702 VK == RISCVMCExpr::VK_RISCV_PCREL_HI || 703 VK == RISCVMCExpr::VK_RISCV_GOT_HI || 704 VK == RISCVMCExpr::VK_RISCV_TLS_GOT_HI || 705 VK == RISCVMCExpr::VK_RISCV_TLS_GD_HI); 706 } 707 } 708 709 bool isSImm21Lsb0JAL() const { return isBareSimmNLsb0<21>(); } 710 711 bool isImmZero() const { 712 if (!isImm()) 713 return false; 714 int64_t Imm; 715 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 716 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 717 return IsConstantImm && (Imm == 0) && VK == RISCVMCExpr::VK_RISCV_None; 718 } 719 720 bool isSImm5Plus1() const { 721 if (!isImm()) 722 return false; 723 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 724 int64_t Imm; 725 bool IsConstantImm = evaluateConstantImm(getImm(), Imm, VK); 726 return IsConstantImm && isInt<5>(Imm - 1) && 727 VK == RISCVMCExpr::VK_RISCV_None; 728 } 729 730 /// getStartLoc - Gets location of the first token of this operand 731 SMLoc getStartLoc() const override { return StartLoc; } 732 /// getEndLoc - Gets location of the last token of this operand 733 SMLoc getEndLoc() const override { return EndLoc; } 734 /// True if this operand is for an RV64 instruction 735 bool isRV64() const { return IsRV64; } 736 737 unsigned getReg() const override { 738 assert(Kind == KindTy::Register && "Invalid type access!"); 739 return Reg.RegNum.id(); 740 } 741 742 StringRef getSysReg() const { 743 assert(Kind == KindTy::SystemRegister && "Invalid access!"); 744 return StringRef(SysReg.Data, SysReg.Length); 745 } 746 747 const MCExpr *getImm() const { 748 assert(Kind == KindTy::Immediate && "Invalid type access!"); 749 return Imm.Val; 750 } 751 752 StringRef getToken() const { 753 assert(Kind == KindTy::Token && "Invalid type access!"); 754 return Tok; 755 } 756 757 static StringRef getSEWStr(VSEW Sew) { 758 switch (Sew) { 759 case VSEW::SEW_8: 760 return "e8"; 761 case VSEW::SEW_16: 762 return "e16"; 763 case VSEW::SEW_32: 764 return "e32"; 765 case VSEW::SEW_64: 766 return "e64"; 767 case VSEW::SEW_128: 768 return "e128"; 769 case VSEW::SEW_256: 770 return "e256"; 771 case VSEW::SEW_512: 772 return "e512"; 773 case VSEW::SEW_1024: 774 return "e1024"; 775 } 776 llvm_unreachable("Unknown SEW."); 777 } 778 779 static StringRef getLMULStr(VLMUL Lmul) { 780 switch (Lmul) { 781 case VLMUL::LMUL_1: 782 return "m1"; 783 case VLMUL::LMUL_2: 784 return "m2"; 785 case VLMUL::LMUL_4: 786 return "m4"; 787 case VLMUL::LMUL_8: 788 return "m8"; 789 case VLMUL::LMUL_F2: 790 return "mf2"; 791 case VLMUL::LMUL_F4: 792 return "mf4"; 793 case VLMUL::LMUL_F8: 794 return "mf8"; 795 } 796 llvm_unreachable("Unknown LMUL."); 797 } 798 799 StringRef getVType(SmallString<32> &Buf) const { 800 assert(Kind == KindTy::VType && "Invalid access!"); 801 Buf.append(getSEWStr(VType.Sew)); 802 Buf.append(","); 803 Buf.append(getLMULStr(VType.Lmul)); 804 805 return Buf.str(); 806 } 807 808 void print(raw_ostream &OS) const override { 809 switch (Kind) { 810 case KindTy::Immediate: 811 OS << *getImm(); 812 break; 813 case KindTy::Register: 814 OS << "<register x"; 815 OS << getReg() << ">"; 816 break; 817 case KindTy::Token: 818 OS << "'" << getToken() << "'"; 819 break; 820 case KindTy::SystemRegister: 821 OS << "<sysreg: " << getSysReg() << '>'; 822 break; 823 case KindTy::VType: 824 SmallString<32> VTypeBuf; 825 OS << "<vtype: " << getVType(VTypeBuf) << '>'; 826 break; 827 } 828 } 829 830 static std::unique_ptr<RISCVOperand> createToken(StringRef Str, SMLoc S, 831 bool IsRV64) { 832 auto Op = std::make_unique<RISCVOperand>(KindTy::Token); 833 Op->Tok = Str; 834 Op->StartLoc = S; 835 Op->EndLoc = S; 836 Op->IsRV64 = IsRV64; 837 return Op; 838 } 839 840 static std::unique_ptr<RISCVOperand> createReg(unsigned RegNo, SMLoc S, 841 SMLoc E, bool IsRV64) { 842 auto Op = std::make_unique<RISCVOperand>(KindTy::Register); 843 Op->Reg.RegNum = RegNo; 844 Op->StartLoc = S; 845 Op->EndLoc = E; 846 Op->IsRV64 = IsRV64; 847 return Op; 848 } 849 850 static std::unique_ptr<RISCVOperand> createImm(const MCExpr *Val, SMLoc S, 851 SMLoc E, bool IsRV64) { 852 auto Op = std::make_unique<RISCVOperand>(KindTy::Immediate); 853 Op->Imm.Val = Val; 854 Op->StartLoc = S; 855 Op->EndLoc = E; 856 Op->IsRV64 = IsRV64; 857 return Op; 858 } 859 860 static std::unique_ptr<RISCVOperand> 861 createSysReg(StringRef Str, SMLoc S, unsigned Encoding, bool IsRV64) { 862 auto Op = std::make_unique<RISCVOperand>(KindTy::SystemRegister); 863 Op->SysReg.Data = Str.data(); 864 Op->SysReg.Length = Str.size(); 865 Op->SysReg.Encoding = Encoding; 866 Op->StartLoc = S; 867 Op->IsRV64 = IsRV64; 868 return Op; 869 } 870 871 static std::unique_ptr<RISCVOperand> 872 createVType(APInt Sew, APInt Lmul, bool Fractional, bool TailAgnostic, 873 bool MaskedoffAgnostic, SMLoc S, bool IsRV64) { 874 auto Op = std::make_unique<RISCVOperand>(KindTy::VType); 875 Sew.ashrInPlace(3); 876 unsigned SewLog2 = Sew.logBase2(); 877 unsigned LmulLog2 = Lmul.logBase2(); 878 Op->VType.Sew = static_cast<VSEW>(SewLog2); 879 if (Fractional) { 880 unsigned Flmul = 8 - LmulLog2; 881 Op->VType.Lmul = static_cast<VLMUL>(Flmul); 882 Op->VType.Encoding = 883 ((Flmul & 0x4) << 3) | ((SewLog2 & 0x7) << 2) | (Flmul & 0x3); 884 } else { 885 Op->VType.Lmul = static_cast<VLMUL>(LmulLog2); 886 Op->VType.Encoding = (SewLog2 << 2) | LmulLog2; 887 } 888 if (TailAgnostic) { 889 Op->VType.Encoding |= 0x40; 890 } 891 if (MaskedoffAgnostic) { 892 Op->VType.Encoding |= 0x80; 893 } 894 Op->VType.TailAgnostic = TailAgnostic; 895 Op->VType.MaskedoffAgnostic = MaskedoffAgnostic; 896 Op->StartLoc = S; 897 Op->IsRV64 = IsRV64; 898 return Op; 899 } 900 901 void addExpr(MCInst &Inst, const MCExpr *Expr) const { 902 assert(Expr && "Expr shouldn't be null!"); 903 int64_t Imm = 0; 904 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 905 bool IsConstant = evaluateConstantImm(Expr, Imm, VK); 906 907 if (IsConstant) 908 Inst.addOperand(MCOperand::createImm(Imm)); 909 else 910 Inst.addOperand(MCOperand::createExpr(Expr)); 911 } 912 913 // Used by the TableGen Code 914 void addRegOperands(MCInst &Inst, unsigned N) const { 915 assert(N == 1 && "Invalid number of operands!"); 916 Inst.addOperand(MCOperand::createReg(getReg())); 917 } 918 919 void addImmOperands(MCInst &Inst, unsigned N) const { 920 assert(N == 1 && "Invalid number of operands!"); 921 addExpr(Inst, getImm()); 922 } 923 924 void addSImm5Plus1Operands(MCInst &Inst, unsigned N) const { 925 assert(N == 1 && "Invalid number of operands!"); 926 int64_t Imm = 0; 927 RISCVMCExpr::VariantKind VK = RISCVMCExpr::VK_RISCV_None; 928 bool IsConstant = evaluateConstantImm(getImm(), Imm, VK); 929 assert(IsConstant && "Expect constant value!"); 930 (void)IsConstant; 931 Inst.addOperand(MCOperand::createImm(Imm - 1)); 932 } 933 934 void addFenceArgOperands(MCInst &Inst, unsigned N) const { 935 assert(N == 1 && "Invalid number of operands!"); 936 // isFenceArg has validated the operand, meaning this cast is safe 937 auto SE = cast<MCSymbolRefExpr>(getImm()); 938 939 unsigned Imm = 0; 940 for (char c : SE->getSymbol().getName()) { 941 switch (c) { 942 default: 943 llvm_unreachable("FenceArg must contain only [iorw]"); 944 case 'i': Imm |= RISCVFenceField::I; break; 945 case 'o': Imm |= RISCVFenceField::O; break; 946 case 'r': Imm |= RISCVFenceField::R; break; 947 case 'w': Imm |= RISCVFenceField::W; break; 948 } 949 } 950 Inst.addOperand(MCOperand::createImm(Imm)); 951 } 952 953 void addCSRSystemRegisterOperands(MCInst &Inst, unsigned N) const { 954 assert(N == 1 && "Invalid number of operands!"); 955 Inst.addOperand(MCOperand::createImm(SysReg.Encoding)); 956 } 957 958 void addVTypeIOperands(MCInst &Inst, unsigned N) const { 959 assert(N == 1 && "Invalid number of operands!"); 960 Inst.addOperand(MCOperand::createImm(VType.Encoding)); 961 } 962 963 // Returns the rounding mode represented by this RISCVOperand. Should only 964 // be called after checking isFRMArg. 965 RISCVFPRndMode::RoundingMode getRoundingMode() const { 966 // isFRMArg has validated the operand, meaning this cast is safe. 967 auto SE = cast<MCSymbolRefExpr>(getImm()); 968 RISCVFPRndMode::RoundingMode FRM = 969 RISCVFPRndMode::stringToRoundingMode(SE->getSymbol().getName()); 970 assert(FRM != RISCVFPRndMode::Invalid && "Invalid rounding mode"); 971 return FRM; 972 } 973 974 void addFRMArgOperands(MCInst &Inst, unsigned N) const { 975 assert(N == 1 && "Invalid number of operands!"); 976 Inst.addOperand(MCOperand::createImm(getRoundingMode())); 977 } 978 }; 979 } // end anonymous namespace. 980 981 #define GET_REGISTER_MATCHER 982 #define GET_SUBTARGET_FEATURE_NAME 983 #define GET_MATCHER_IMPLEMENTATION 984 #define GET_MNEMONIC_SPELL_CHECKER 985 #include "RISCVGenAsmMatcher.inc" 986 987 static Register convertFPR64ToFPR32(Register Reg) { 988 assert(Reg >= RISCV::F0_D && Reg <= RISCV::F31_D && "Invalid register"); 989 return Reg - RISCV::F0_D + RISCV::F0_F; 990 } 991 992 unsigned RISCVAsmParser::validateTargetOperandClass(MCParsedAsmOperand &AsmOp, 993 unsigned Kind) { 994 RISCVOperand &Op = static_cast<RISCVOperand &>(AsmOp); 995 if (!Op.isReg()) 996 return Match_InvalidOperand; 997 998 Register Reg = Op.getReg(); 999 bool IsRegFPR64 = 1000 RISCVMCRegisterClasses[RISCV::FPR64RegClassID].contains(Reg); 1001 bool IsRegFPR64C = 1002 RISCVMCRegisterClasses[RISCV::FPR64CRegClassID].contains(Reg); 1003 1004 // As the parser couldn't differentiate an FPR32 from an FPR64, coerce the 1005 // register from FPR64 to FPR32 or FPR64C to FPR32C if necessary. 1006 if ((IsRegFPR64 && Kind == MCK_FPR32) || 1007 (IsRegFPR64C && Kind == MCK_FPR32C)) { 1008 Op.Reg.RegNum = convertFPR64ToFPR32(Reg); 1009 return Match_Success; 1010 } 1011 return Match_InvalidOperand; 1012 } 1013 1014 bool RISCVAsmParser::generateImmOutOfRangeError( 1015 OperandVector &Operands, uint64_t ErrorInfo, int64_t Lower, int64_t Upper, 1016 Twine Msg = "immediate must be an integer in the range") { 1017 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1018 return Error(ErrorLoc, Msg + " [" + Twine(Lower) + ", " + Twine(Upper) + "]"); 1019 } 1020 1021 static std::string RISCVMnemonicSpellCheck(StringRef S, 1022 const FeatureBitset &FBS, 1023 unsigned VariantID = 0); 1024 1025 bool RISCVAsmParser::MatchAndEmitInstruction(SMLoc IDLoc, unsigned &Opcode, 1026 OperandVector &Operands, 1027 MCStreamer &Out, 1028 uint64_t &ErrorInfo, 1029 bool MatchingInlineAsm) { 1030 MCInst Inst; 1031 FeatureBitset MissingFeatures; 1032 1033 auto Result = 1034 MatchInstructionImpl(Operands, Inst, ErrorInfo, MissingFeatures, 1035 MatchingInlineAsm); 1036 switch (Result) { 1037 default: 1038 break; 1039 case Match_Success: 1040 if (validateInstruction(Inst, Operands)) 1041 return true; 1042 return processInstruction(Inst, IDLoc, Operands, Out); 1043 case Match_MissingFeature: { 1044 assert(MissingFeatures.any() && "Unknown missing features!"); 1045 bool FirstFeature = true; 1046 std::string Msg = "instruction requires the following:"; 1047 for (unsigned i = 0, e = MissingFeatures.size(); i != e; ++i) { 1048 if (MissingFeatures[i]) { 1049 Msg += FirstFeature ? " " : ", "; 1050 Msg += getSubtargetFeatureName(i); 1051 FirstFeature = false; 1052 } 1053 } 1054 return Error(IDLoc, Msg); 1055 } 1056 case Match_MnemonicFail: { 1057 FeatureBitset FBS = ComputeAvailableFeatures(getSTI().getFeatureBits()); 1058 std::string Suggestion = RISCVMnemonicSpellCheck( 1059 ((RISCVOperand &)*Operands[0]).getToken(), FBS); 1060 return Error(IDLoc, "unrecognized instruction mnemonic" + Suggestion); 1061 } 1062 case Match_InvalidOperand: { 1063 SMLoc ErrorLoc = IDLoc; 1064 if (ErrorInfo != ~0U) { 1065 if (ErrorInfo >= Operands.size()) 1066 return Error(ErrorLoc, "too few operands for instruction"); 1067 1068 ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1069 if (ErrorLoc == SMLoc()) 1070 ErrorLoc = IDLoc; 1071 } 1072 return Error(ErrorLoc, "invalid operand for instruction"); 1073 } 1074 } 1075 1076 // Handle the case when the error message is of specific type 1077 // other than the generic Match_InvalidOperand, and the 1078 // corresponding operand is missing. 1079 if (Result > FIRST_TARGET_MATCH_RESULT_TY) { 1080 SMLoc ErrorLoc = IDLoc; 1081 if (ErrorInfo != ~0U && ErrorInfo >= Operands.size()) 1082 return Error(ErrorLoc, "too few operands for instruction"); 1083 } 1084 1085 switch(Result) { 1086 default: 1087 break; 1088 case Match_InvalidImmXLenLI: 1089 if (isRV64()) { 1090 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1091 return Error(ErrorLoc, "operand must be a constant 64-bit integer"); 1092 } 1093 return generateImmOutOfRangeError(Operands, ErrorInfo, 1094 std::numeric_limits<int32_t>::min(), 1095 std::numeric_limits<uint32_t>::max()); 1096 case Match_InvalidImmZero: { 1097 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1098 return Error(ErrorLoc, "immediate must be zero"); 1099 } 1100 case Match_InvalidUImmLog2XLen: 1101 if (isRV64()) 1102 return generateImmOutOfRangeError(Operands, ErrorInfo, 0, (1 << 6) - 1); 1103 return generateImmOutOfRangeError(Operands, ErrorInfo, 0, (1 << 5) - 1); 1104 case Match_InvalidUImmLog2XLenNonZero: 1105 if (isRV64()) 1106 return generateImmOutOfRangeError(Operands, ErrorInfo, 1, (1 << 6) - 1); 1107 return generateImmOutOfRangeError(Operands, ErrorInfo, 1, (1 << 5) - 1); 1108 case Match_InvalidUImmLog2XLenHalf: 1109 if (isRV64()) 1110 return generateImmOutOfRangeError(Operands, ErrorInfo, 0, (1 << 5) - 1); 1111 return generateImmOutOfRangeError(Operands, ErrorInfo, 0, (1 << 4) - 1); 1112 case Match_InvalidUImm5: 1113 return generateImmOutOfRangeError(Operands, ErrorInfo, 0, (1 << 5) - 1); 1114 case Match_InvalidSImm6: 1115 return generateImmOutOfRangeError(Operands, ErrorInfo, -(1 << 5), 1116 (1 << 5) - 1); 1117 case Match_InvalidSImm6NonZero: 1118 return generateImmOutOfRangeError( 1119 Operands, ErrorInfo, -(1 << 5), (1 << 5) - 1, 1120 "immediate must be non-zero in the range"); 1121 case Match_InvalidCLUIImm: 1122 return generateImmOutOfRangeError( 1123 Operands, ErrorInfo, 1, (1 << 5) - 1, 1124 "immediate must be in [0xfffe0, 0xfffff] or"); 1125 case Match_InvalidUImm7Lsb00: 1126 return generateImmOutOfRangeError( 1127 Operands, ErrorInfo, 0, (1 << 7) - 4, 1128 "immediate must be a multiple of 4 bytes in the range"); 1129 case Match_InvalidUImm8Lsb00: 1130 return generateImmOutOfRangeError( 1131 Operands, ErrorInfo, 0, (1 << 8) - 4, 1132 "immediate must be a multiple of 4 bytes in the range"); 1133 case Match_InvalidUImm8Lsb000: 1134 return generateImmOutOfRangeError( 1135 Operands, ErrorInfo, 0, (1 << 8) - 8, 1136 "immediate must be a multiple of 8 bytes in the range"); 1137 case Match_InvalidSImm9Lsb0: 1138 return generateImmOutOfRangeError( 1139 Operands, ErrorInfo, -(1 << 8), (1 << 8) - 2, 1140 "immediate must be a multiple of 2 bytes in the range"); 1141 case Match_InvalidUImm9Lsb000: 1142 return generateImmOutOfRangeError( 1143 Operands, ErrorInfo, 0, (1 << 9) - 8, 1144 "immediate must be a multiple of 8 bytes in the range"); 1145 case Match_InvalidUImm10Lsb00NonZero: 1146 return generateImmOutOfRangeError( 1147 Operands, ErrorInfo, 4, (1 << 10) - 4, 1148 "immediate must be a multiple of 4 bytes in the range"); 1149 case Match_InvalidSImm10Lsb0000NonZero: 1150 return generateImmOutOfRangeError( 1151 Operands, ErrorInfo, -(1 << 9), (1 << 9) - 16, 1152 "immediate must be a multiple of 16 bytes and non-zero in the range"); 1153 case Match_InvalidSImm12: 1154 return generateImmOutOfRangeError( 1155 Operands, ErrorInfo, -(1 << 11), (1 << 11) - 1, 1156 "operand must be a symbol with %lo/%pcrel_lo/%tprel_lo modifier or an " 1157 "integer in the range"); 1158 case Match_InvalidSImm12Lsb0: 1159 return generateImmOutOfRangeError( 1160 Operands, ErrorInfo, -(1 << 11), (1 << 11) - 2, 1161 "immediate must be a multiple of 2 bytes in the range"); 1162 case Match_InvalidSImm13Lsb0: 1163 return generateImmOutOfRangeError( 1164 Operands, ErrorInfo, -(1 << 12), (1 << 12) - 2, 1165 "immediate must be a multiple of 2 bytes in the range"); 1166 case Match_InvalidUImm20LUI: 1167 return generateImmOutOfRangeError(Operands, ErrorInfo, 0, (1 << 20) - 1, 1168 "operand must be a symbol with " 1169 "%hi/%tprel_hi modifier or an integer in " 1170 "the range"); 1171 case Match_InvalidUImm20AUIPC: 1172 return generateImmOutOfRangeError( 1173 Operands, ErrorInfo, 0, (1 << 20) - 1, 1174 "operand must be a symbol with a " 1175 "%pcrel_hi/%got_pcrel_hi/%tls_ie_pcrel_hi/%tls_gd_pcrel_hi modifier or " 1176 "an integer in the range"); 1177 case Match_InvalidSImm21Lsb0JAL: 1178 return generateImmOutOfRangeError( 1179 Operands, ErrorInfo, -(1 << 20), (1 << 20) - 2, 1180 "immediate must be a multiple of 2 bytes in the range"); 1181 case Match_InvalidCSRSystemRegister: { 1182 return generateImmOutOfRangeError(Operands, ErrorInfo, 0, (1 << 12) - 1, 1183 "operand must be a valid system register " 1184 "name or an integer in the range"); 1185 } 1186 case Match_InvalidFenceArg: { 1187 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1188 return Error( 1189 ErrorLoc, 1190 "operand must be formed of letters selected in-order from 'iorw'"); 1191 } 1192 case Match_InvalidFRMArg: { 1193 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1194 return Error( 1195 ErrorLoc, 1196 "operand must be a valid floating point rounding mode mnemonic"); 1197 } 1198 case Match_InvalidBareSymbol: { 1199 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1200 return Error(ErrorLoc, "operand must be a bare symbol name"); 1201 } 1202 case Match_InvalidPseudoJumpSymbol: { 1203 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1204 return Error(ErrorLoc, "operand must be a valid jump target"); 1205 } 1206 case Match_InvalidCallSymbol: { 1207 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1208 return Error(ErrorLoc, "operand must be a bare symbol name"); 1209 } 1210 case Match_InvalidTPRelAddSymbol: { 1211 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1212 return Error(ErrorLoc, "operand must be a symbol with %tprel_add modifier"); 1213 } 1214 case Match_InvalidVTypeI: { 1215 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1216 return Error( 1217 ErrorLoc, 1218 "operand must be " 1219 "e[8|16|32|64|128|256|512|1024],m[1|2|4|8|f2|f4|f8],[ta|tu],[ma|mu]"); 1220 } 1221 case Match_InvalidVMaskRegister: { 1222 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[ErrorInfo]).getStartLoc(); 1223 return Error(ErrorLoc, "operand must be v0.t"); 1224 } 1225 case Match_InvalidSImm5Plus1: { 1226 return generateImmOutOfRangeError(Operands, ErrorInfo, -(1 << 4) + 1, 1227 (1 << 4), 1228 "immediate must be in the range"); 1229 } 1230 } 1231 1232 llvm_unreachable("Unknown match type detected!"); 1233 } 1234 1235 // Attempts to match Name as a register (either using the default name or 1236 // alternative ABI names), setting RegNo to the matching register. Upon 1237 // failure, returns true and sets RegNo to 0. If IsRV32E then registers 1238 // x16-x31 will be rejected. 1239 static bool matchRegisterNameHelper(bool IsRV32E, Register &RegNo, 1240 StringRef Name) { 1241 RegNo = MatchRegisterName(Name); 1242 // The 32- and 64-bit FPRs have the same asm name. Check that the initial 1243 // match always matches the 64-bit variant, and not the 32-bit one. 1244 assert(!(RegNo >= RISCV::F0_F && RegNo <= RISCV::F31_F)); 1245 // The default FPR register class is based on the tablegen enum ordering. 1246 static_assert(RISCV::F0_D < RISCV::F0_F, "FPR matching must be updated"); 1247 if (RegNo == RISCV::NoRegister) 1248 RegNo = MatchRegisterAltName(Name); 1249 if (IsRV32E && RegNo >= RISCV::X16 && RegNo <= RISCV::X31) 1250 RegNo = RISCV::NoRegister; 1251 return RegNo == RISCV::NoRegister; 1252 } 1253 1254 bool RISCVAsmParser::ParseRegister(unsigned &RegNo, SMLoc &StartLoc, 1255 SMLoc &EndLoc) { 1256 if (tryParseRegister(RegNo, StartLoc, EndLoc) != MatchOperand_Success) 1257 return Error(StartLoc, "invalid register name"); 1258 return false; 1259 } 1260 1261 OperandMatchResultTy RISCVAsmParser::tryParseRegister(unsigned &RegNo, 1262 SMLoc &StartLoc, 1263 SMLoc &EndLoc) { 1264 const AsmToken &Tok = getParser().getTok(); 1265 StartLoc = Tok.getLoc(); 1266 EndLoc = Tok.getEndLoc(); 1267 RegNo = 0; 1268 StringRef Name = getLexer().getTok().getIdentifier(); 1269 1270 if (matchRegisterNameHelper(isRV32E(), (Register &)RegNo, Name)) 1271 return MatchOperand_NoMatch; 1272 1273 getParser().Lex(); // Eat identifier token. 1274 return MatchOperand_Success; 1275 } 1276 1277 OperandMatchResultTy RISCVAsmParser::parseRegister(OperandVector &Operands, 1278 bool AllowParens) { 1279 SMLoc FirstS = getLoc(); 1280 bool HadParens = false; 1281 AsmToken LParen; 1282 1283 // If this is an LParen and a parenthesised register name is allowed, parse it 1284 // atomically. 1285 if (AllowParens && getLexer().is(AsmToken::LParen)) { 1286 AsmToken Buf[2]; 1287 size_t ReadCount = getLexer().peekTokens(Buf); 1288 if (ReadCount == 2 && Buf[1].getKind() == AsmToken::RParen) { 1289 HadParens = true; 1290 LParen = getParser().getTok(); 1291 getParser().Lex(); // Eat '(' 1292 } 1293 } 1294 1295 switch (getLexer().getKind()) { 1296 default: 1297 if (HadParens) 1298 getLexer().UnLex(LParen); 1299 return MatchOperand_NoMatch; 1300 case AsmToken::Identifier: 1301 StringRef Name = getLexer().getTok().getIdentifier(); 1302 Register RegNo; 1303 matchRegisterNameHelper(isRV32E(), RegNo, Name); 1304 1305 if (RegNo == RISCV::NoRegister) { 1306 if (HadParens) 1307 getLexer().UnLex(LParen); 1308 return MatchOperand_NoMatch; 1309 } 1310 if (HadParens) 1311 Operands.push_back(RISCVOperand::createToken("(", FirstS, isRV64())); 1312 SMLoc S = getLoc(); 1313 SMLoc E = SMLoc::getFromPointer(S.getPointer() - 1); 1314 getLexer().Lex(); 1315 Operands.push_back(RISCVOperand::createReg(RegNo, S, E, isRV64())); 1316 } 1317 1318 if (HadParens) { 1319 getParser().Lex(); // Eat ')' 1320 Operands.push_back(RISCVOperand::createToken(")", getLoc(), isRV64())); 1321 } 1322 1323 return MatchOperand_Success; 1324 } 1325 1326 OperandMatchResultTy 1327 RISCVAsmParser::parseCSRSystemRegister(OperandVector &Operands) { 1328 SMLoc S = getLoc(); 1329 const MCExpr *Res; 1330 1331 switch (getLexer().getKind()) { 1332 default: 1333 return MatchOperand_NoMatch; 1334 case AsmToken::LParen: 1335 case AsmToken::Minus: 1336 case AsmToken::Plus: 1337 case AsmToken::Exclaim: 1338 case AsmToken::Tilde: 1339 case AsmToken::Integer: 1340 case AsmToken::String: { 1341 if (getParser().parseExpression(Res)) 1342 return MatchOperand_ParseFail; 1343 1344 auto *CE = dyn_cast<MCConstantExpr>(Res); 1345 if (CE) { 1346 int64_t Imm = CE->getValue(); 1347 if (isUInt<12>(Imm)) { 1348 auto SysReg = RISCVSysReg::lookupSysRegByEncoding(Imm); 1349 // Accept an immediate representing a named or un-named Sys Reg 1350 // if the range is valid, regardless of the required features. 1351 Operands.push_back(RISCVOperand::createSysReg( 1352 SysReg ? SysReg->Name : "", S, Imm, isRV64())); 1353 return MatchOperand_Success; 1354 } 1355 } 1356 1357 Twine Msg = "immediate must be an integer in the range"; 1358 Error(S, Msg + " [" + Twine(0) + ", " + Twine((1 << 12) - 1) + "]"); 1359 return MatchOperand_ParseFail; 1360 } 1361 case AsmToken::Identifier: { 1362 StringRef Identifier; 1363 if (getParser().parseIdentifier(Identifier)) 1364 return MatchOperand_ParseFail; 1365 1366 auto SysReg = RISCVSysReg::lookupSysRegByName(Identifier); 1367 if (!SysReg) 1368 SysReg = RISCVSysReg::lookupSysRegByAltName(Identifier); 1369 // Accept a named Sys Reg if the required features are present. 1370 if (SysReg) { 1371 if (!SysReg->haveRequiredFeatures(getSTI().getFeatureBits())) { 1372 Error(S, "system register use requires an option to be enabled"); 1373 return MatchOperand_ParseFail; 1374 } 1375 Operands.push_back(RISCVOperand::createSysReg( 1376 Identifier, S, SysReg->Encoding, isRV64())); 1377 return MatchOperand_Success; 1378 } 1379 1380 Twine Msg = "operand must be a valid system register name " 1381 "or an integer in the range"; 1382 Error(S, Msg + " [" + Twine(0) + ", " + Twine((1 << 12) - 1) + "]"); 1383 return MatchOperand_ParseFail; 1384 } 1385 case AsmToken::Percent: { 1386 // Discard operand with modifier. 1387 Twine Msg = "immediate must be an integer in the range"; 1388 Error(S, Msg + " [" + Twine(0) + ", " + Twine((1 << 12) - 1) + "]"); 1389 return MatchOperand_ParseFail; 1390 } 1391 } 1392 1393 return MatchOperand_NoMatch; 1394 } 1395 1396 OperandMatchResultTy RISCVAsmParser::parseImmediate(OperandVector &Operands) { 1397 SMLoc S = getLoc(); 1398 SMLoc E = SMLoc::getFromPointer(S.getPointer() - 1); 1399 const MCExpr *Res; 1400 1401 switch (getLexer().getKind()) { 1402 default: 1403 return MatchOperand_NoMatch; 1404 case AsmToken::LParen: 1405 case AsmToken::Dot: 1406 case AsmToken::Minus: 1407 case AsmToken::Plus: 1408 case AsmToken::Exclaim: 1409 case AsmToken::Tilde: 1410 case AsmToken::Integer: 1411 case AsmToken::String: 1412 case AsmToken::Identifier: 1413 if (getParser().parseExpression(Res)) 1414 return MatchOperand_ParseFail; 1415 break; 1416 case AsmToken::Percent: 1417 return parseOperandWithModifier(Operands); 1418 } 1419 1420 Operands.push_back(RISCVOperand::createImm(Res, S, E, isRV64())); 1421 return MatchOperand_Success; 1422 } 1423 1424 OperandMatchResultTy 1425 RISCVAsmParser::parseOperandWithModifier(OperandVector &Operands) { 1426 SMLoc S = getLoc(); 1427 SMLoc E = SMLoc::getFromPointer(S.getPointer() - 1); 1428 1429 if (getLexer().getKind() != AsmToken::Percent) { 1430 Error(getLoc(), "expected '%' for operand modifier"); 1431 return MatchOperand_ParseFail; 1432 } 1433 1434 getParser().Lex(); // Eat '%' 1435 1436 if (getLexer().getKind() != AsmToken::Identifier) { 1437 Error(getLoc(), "expected valid identifier for operand modifier"); 1438 return MatchOperand_ParseFail; 1439 } 1440 StringRef Identifier = getParser().getTok().getIdentifier(); 1441 RISCVMCExpr::VariantKind VK = RISCVMCExpr::getVariantKindForName(Identifier); 1442 if (VK == RISCVMCExpr::VK_RISCV_Invalid) { 1443 Error(getLoc(), "unrecognized operand modifier"); 1444 return MatchOperand_ParseFail; 1445 } 1446 1447 getParser().Lex(); // Eat the identifier 1448 if (getLexer().getKind() != AsmToken::LParen) { 1449 Error(getLoc(), "expected '('"); 1450 return MatchOperand_ParseFail; 1451 } 1452 getParser().Lex(); // Eat '(' 1453 1454 const MCExpr *SubExpr; 1455 if (getParser().parseParenExpression(SubExpr, E)) { 1456 return MatchOperand_ParseFail; 1457 } 1458 1459 const MCExpr *ModExpr = RISCVMCExpr::create(SubExpr, VK, getContext()); 1460 Operands.push_back(RISCVOperand::createImm(ModExpr, S, E, isRV64())); 1461 return MatchOperand_Success; 1462 } 1463 1464 OperandMatchResultTy RISCVAsmParser::parseBareSymbol(OperandVector &Operands) { 1465 SMLoc S = getLoc(); 1466 SMLoc E = SMLoc::getFromPointer(S.getPointer() - 1); 1467 const MCExpr *Res; 1468 1469 if (getLexer().getKind() != AsmToken::Identifier) 1470 return MatchOperand_NoMatch; 1471 1472 StringRef Identifier; 1473 AsmToken Tok = getLexer().getTok(); 1474 1475 if (getParser().parseIdentifier(Identifier)) 1476 return MatchOperand_ParseFail; 1477 1478 if (Identifier.consume_back("@plt")) { 1479 Error(getLoc(), "'@plt' operand not valid for instruction"); 1480 return MatchOperand_ParseFail; 1481 } 1482 1483 MCSymbol *Sym = getContext().getOrCreateSymbol(Identifier); 1484 1485 if (Sym->isVariable()) { 1486 const MCExpr *V = Sym->getVariableValue(/*SetUsed=*/false); 1487 if (!isa<MCSymbolRefExpr>(V)) { 1488 getLexer().UnLex(Tok); // Put back if it's not a bare symbol. 1489 return MatchOperand_NoMatch; 1490 } 1491 Res = V; 1492 } else 1493 Res = MCSymbolRefExpr::create(Sym, MCSymbolRefExpr::VK_None, getContext()); 1494 1495 MCBinaryExpr::Opcode Opcode; 1496 switch (getLexer().getKind()) { 1497 default: 1498 Operands.push_back(RISCVOperand::createImm(Res, S, E, isRV64())); 1499 return MatchOperand_Success; 1500 case AsmToken::Plus: 1501 Opcode = MCBinaryExpr::Add; 1502 break; 1503 case AsmToken::Minus: 1504 Opcode = MCBinaryExpr::Sub; 1505 break; 1506 } 1507 1508 const MCExpr *Expr; 1509 if (getParser().parseExpression(Expr)) 1510 return MatchOperand_ParseFail; 1511 Res = MCBinaryExpr::create(Opcode, Res, Expr, getContext()); 1512 Operands.push_back(RISCVOperand::createImm(Res, S, E, isRV64())); 1513 return MatchOperand_Success; 1514 } 1515 1516 OperandMatchResultTy RISCVAsmParser::parseCallSymbol(OperandVector &Operands) { 1517 SMLoc S = getLoc(); 1518 SMLoc E = SMLoc::getFromPointer(S.getPointer() - 1); 1519 const MCExpr *Res; 1520 1521 if (getLexer().getKind() != AsmToken::Identifier) 1522 return MatchOperand_NoMatch; 1523 1524 // Avoid parsing the register in `call rd, foo` as a call symbol. 1525 if (getLexer().peekTok().getKind() != AsmToken::EndOfStatement) 1526 return MatchOperand_NoMatch; 1527 1528 StringRef Identifier; 1529 if (getParser().parseIdentifier(Identifier)) 1530 return MatchOperand_ParseFail; 1531 1532 RISCVMCExpr::VariantKind Kind = RISCVMCExpr::VK_RISCV_CALL; 1533 if (Identifier.consume_back("@plt")) 1534 Kind = RISCVMCExpr::VK_RISCV_CALL_PLT; 1535 1536 MCSymbol *Sym = getContext().getOrCreateSymbol(Identifier); 1537 Res = MCSymbolRefExpr::create(Sym, MCSymbolRefExpr::VK_None, getContext()); 1538 Res = RISCVMCExpr::create(Res, Kind, getContext()); 1539 Operands.push_back(RISCVOperand::createImm(Res, S, E, isRV64())); 1540 return MatchOperand_Success; 1541 } 1542 1543 OperandMatchResultTy 1544 RISCVAsmParser::parsePseudoJumpSymbol(OperandVector &Operands) { 1545 SMLoc S = getLoc(); 1546 SMLoc E = SMLoc::getFromPointer(S.getPointer() - 1); 1547 const MCExpr *Res; 1548 1549 if (getParser().parseExpression(Res)) 1550 return MatchOperand_ParseFail; 1551 1552 if (Res->getKind() != MCExpr::ExprKind::SymbolRef || 1553 cast<MCSymbolRefExpr>(Res)->getKind() == 1554 MCSymbolRefExpr::VariantKind::VK_PLT) { 1555 Error(S, "operand must be a valid jump target"); 1556 return MatchOperand_ParseFail; 1557 } 1558 1559 Res = RISCVMCExpr::create(Res, RISCVMCExpr::VK_RISCV_CALL, getContext()); 1560 Operands.push_back(RISCVOperand::createImm(Res, S, E, isRV64())); 1561 return MatchOperand_Success; 1562 } 1563 1564 OperandMatchResultTy RISCVAsmParser::parseJALOffset(OperandVector &Operands) { 1565 // Parsing jal operands is fiddly due to the `jal foo` and `jal ra, foo` 1566 // both being acceptable forms. When parsing `jal ra, foo` this function 1567 // will be called for the `ra` register operand in an attempt to match the 1568 // single-operand alias. parseJALOffset must fail for this case. It would 1569 // seem logical to try parse the operand using parseImmediate and return 1570 // NoMatch if the next token is a comma (meaning we must be parsing a jal in 1571 // the second form rather than the first). We can't do this as there's no 1572 // way of rewinding the lexer state. Instead, return NoMatch if this operand 1573 // is an identifier and is followed by a comma. 1574 if (getLexer().is(AsmToken::Identifier) && 1575 getLexer().peekTok().is(AsmToken::Comma)) 1576 return MatchOperand_NoMatch; 1577 1578 return parseImmediate(Operands); 1579 } 1580 1581 OperandMatchResultTy RISCVAsmParser::parseVTypeI(OperandVector &Operands) { 1582 SMLoc S = getLoc(); 1583 if (getLexer().getKind() != AsmToken::Identifier) 1584 return MatchOperand_NoMatch; 1585 1586 // Parse "e8,m1,t[a|u],m[a|u]" 1587 StringRef Name = getLexer().getTok().getIdentifier(); 1588 if (!Name.consume_front("e")) 1589 return MatchOperand_NoMatch; 1590 APInt Sew(16, Name, 10); 1591 if (Sew != 8 && Sew != 16 && Sew != 32 && Sew != 64 && Sew != 128 && 1592 Sew != 256 && Sew != 512 && Sew != 1024) 1593 return MatchOperand_NoMatch; 1594 getLexer().Lex(); 1595 1596 if (!getLexer().is(AsmToken::Comma)) 1597 return MatchOperand_NoMatch; 1598 getLexer().Lex(); 1599 1600 Name = getLexer().getTok().getIdentifier(); 1601 if (!Name.consume_front("m")) 1602 return MatchOperand_NoMatch; 1603 // "m" or "mf" 1604 bool Fractional = false; 1605 if (Name.consume_front("f")) { 1606 Fractional = true; 1607 } 1608 APInt Lmul(16, Name, 10); 1609 if (Lmul != 1 && Lmul != 2 && Lmul != 4 && Lmul != 8) 1610 return MatchOperand_NoMatch; 1611 getLexer().Lex(); 1612 1613 if (!getLexer().is(AsmToken::Comma)) 1614 return MatchOperand_NoMatch; 1615 getLexer().Lex(); 1616 1617 Name = getLexer().getTok().getIdentifier(); 1618 // ta or tu 1619 bool TailAgnostic; 1620 if (Name.consume_front("ta")) 1621 TailAgnostic = true; 1622 else if (Name.consume_front("tu")) 1623 TailAgnostic = false; 1624 else 1625 return MatchOperand_NoMatch; 1626 getLexer().Lex(); 1627 1628 if (!getLexer().is(AsmToken::Comma)) 1629 return MatchOperand_NoMatch; 1630 getLexer().Lex(); 1631 1632 Name = getLexer().getTok().getIdentifier(); 1633 // ma or mu 1634 bool MaskedoffAgnostic; 1635 if (Name.consume_front("ma")) 1636 MaskedoffAgnostic = true; 1637 else if (Name.consume_front("mu")) 1638 MaskedoffAgnostic = false; 1639 else 1640 return MatchOperand_NoMatch; 1641 getLexer().Lex(); 1642 1643 if (getLexer().getKind() != AsmToken::EndOfStatement) 1644 return MatchOperand_NoMatch; 1645 1646 Operands.push_back(RISCVOperand::createVType( 1647 Sew, Lmul, Fractional, TailAgnostic, MaskedoffAgnostic, S, isRV64())); 1648 1649 return MatchOperand_Success; 1650 } 1651 1652 OperandMatchResultTy RISCVAsmParser::parseMaskReg(OperandVector &Operands) { 1653 switch (getLexer().getKind()) { 1654 default: 1655 return MatchOperand_NoMatch; 1656 case AsmToken::Identifier: 1657 StringRef Name = getLexer().getTok().getIdentifier(); 1658 if (!Name.consume_back(".t")) { 1659 Error(getLoc(), "expected '.t' suffix"); 1660 return MatchOperand_ParseFail; 1661 } 1662 Register RegNo; 1663 matchRegisterNameHelper(isRV32E(), RegNo, Name); 1664 1665 if (RegNo == RISCV::NoRegister) 1666 return MatchOperand_NoMatch; 1667 if (RegNo != RISCV::V0) 1668 return MatchOperand_NoMatch; 1669 SMLoc S = getLoc(); 1670 SMLoc E = SMLoc::getFromPointer(S.getPointer() - 1); 1671 getLexer().Lex(); 1672 Operands.push_back(RISCVOperand::createReg(RegNo, S, E, isRV64())); 1673 } 1674 1675 return MatchOperand_Success; 1676 } 1677 1678 OperandMatchResultTy 1679 RISCVAsmParser::parseMemOpBaseReg(OperandVector &Operands) { 1680 if (getLexer().isNot(AsmToken::LParen)) { 1681 Error(getLoc(), "expected '('"); 1682 return MatchOperand_ParseFail; 1683 } 1684 1685 getParser().Lex(); // Eat '(' 1686 Operands.push_back(RISCVOperand::createToken("(", getLoc(), isRV64())); 1687 1688 if (parseRegister(Operands) != MatchOperand_Success) { 1689 Error(getLoc(), "expected register"); 1690 return MatchOperand_ParseFail; 1691 } 1692 1693 if (getLexer().isNot(AsmToken::RParen)) { 1694 Error(getLoc(), "expected ')'"); 1695 return MatchOperand_ParseFail; 1696 } 1697 1698 getParser().Lex(); // Eat ')' 1699 Operands.push_back(RISCVOperand::createToken(")", getLoc(), isRV64())); 1700 1701 return MatchOperand_Success; 1702 } 1703 1704 OperandMatchResultTy RISCVAsmParser::parseAtomicMemOp(OperandVector &Operands) { 1705 // Atomic operations such as lr.w, sc.w, and amo*.w accept a "memory operand" 1706 // as one of their register operands, such as `(a0)`. This just denotes that 1707 // the register (in this case `a0`) contains a memory address. 1708 // 1709 // Normally, we would be able to parse these by putting the parens into the 1710 // instruction string. However, GNU as also accepts a zero-offset memory 1711 // operand (such as `0(a0)`), and ignores the 0. Normally this would be parsed 1712 // with parseImmediate followed by parseMemOpBaseReg, but these instructions 1713 // do not accept an immediate operand, and we do not want to add a "dummy" 1714 // operand that is silently dropped. 1715 // 1716 // Instead, we use this custom parser. This will: allow (and discard) an 1717 // offset if it is zero; require (and discard) parentheses; and add only the 1718 // parsed register operand to `Operands`. 1719 // 1720 // These operands are printed with RISCVInstPrinter::printAtomicMemOp, which 1721 // will only print the register surrounded by parentheses (which GNU as also 1722 // uses as its canonical representation for these operands). 1723 std::unique_ptr<RISCVOperand> OptionalImmOp; 1724 1725 if (getLexer().isNot(AsmToken::LParen)) { 1726 // Parse an Integer token. We do not accept arbritrary constant expressions 1727 // in the offset field (because they may include parens, which complicates 1728 // parsing a lot). 1729 int64_t ImmVal; 1730 SMLoc ImmStart = getLoc(); 1731 if (getParser().parseIntToken(ImmVal, 1732 "expected '(' or optional integer offset")) 1733 return MatchOperand_ParseFail; 1734 1735 // Create a RISCVOperand for checking later (so the error messages are 1736 // nicer), but we don't add it to Operands. 1737 SMLoc ImmEnd = getLoc(); 1738 OptionalImmOp = 1739 RISCVOperand::createImm(MCConstantExpr::create(ImmVal, getContext()), 1740 ImmStart, ImmEnd, isRV64()); 1741 } 1742 1743 if (getLexer().isNot(AsmToken::LParen)) { 1744 Error(getLoc(), OptionalImmOp ? "expected '(' after optional integer offset" 1745 : "expected '(' or optional integer offset"); 1746 return MatchOperand_ParseFail; 1747 } 1748 getParser().Lex(); // Eat '(' 1749 1750 if (parseRegister(Operands) != MatchOperand_Success) { 1751 Error(getLoc(), "expected register"); 1752 return MatchOperand_ParseFail; 1753 } 1754 1755 if (getLexer().isNot(AsmToken::RParen)) { 1756 Error(getLoc(), "expected ')'"); 1757 return MatchOperand_ParseFail; 1758 } 1759 getParser().Lex(); // Eat ')' 1760 1761 // Deferred Handling of non-zero offsets. This makes the error messages nicer. 1762 if (OptionalImmOp && !OptionalImmOp->isImmZero()) { 1763 Error(OptionalImmOp->getStartLoc(), "optional integer offset must be 0", 1764 SMRange(OptionalImmOp->getStartLoc(), OptionalImmOp->getEndLoc())); 1765 return MatchOperand_ParseFail; 1766 } 1767 1768 return MatchOperand_Success; 1769 } 1770 1771 /// Looks at a token type and creates the relevant operand from this 1772 /// information, adding to Operands. If operand was parsed, returns false, else 1773 /// true. 1774 bool RISCVAsmParser::parseOperand(OperandVector &Operands, StringRef Mnemonic) { 1775 // Check if the current operand has a custom associated parser, if so, try to 1776 // custom parse the operand, or fallback to the general approach. 1777 OperandMatchResultTy Result = 1778 MatchOperandParserImpl(Operands, Mnemonic, /*ParseForAllFeatures=*/true); 1779 if (Result == MatchOperand_Success) 1780 return false; 1781 if (Result == MatchOperand_ParseFail) 1782 return true; 1783 1784 // Attempt to parse token as a register. 1785 if (parseRegister(Operands, true) == MatchOperand_Success) 1786 return false; 1787 1788 // Attempt to parse token as an immediate 1789 if (parseImmediate(Operands) == MatchOperand_Success) { 1790 // Parse memory base register if present 1791 if (getLexer().is(AsmToken::LParen)) 1792 return parseMemOpBaseReg(Operands) != MatchOperand_Success; 1793 return false; 1794 } 1795 1796 // Finally we have exhausted all options and must declare defeat. 1797 Error(getLoc(), "unknown operand"); 1798 return true; 1799 } 1800 1801 bool RISCVAsmParser::ParseInstruction(ParseInstructionInfo &Info, 1802 StringRef Name, SMLoc NameLoc, 1803 OperandVector &Operands) { 1804 // Ensure that if the instruction occurs when relaxation is enabled, 1805 // relocations are forced for the file. Ideally this would be done when there 1806 // is enough information to reliably determine if the instruction itself may 1807 // cause relaxations. Unfortunately instruction processing stage occurs in the 1808 // same pass as relocation emission, so it's too late to set a 'sticky bit' 1809 // for the entire file. 1810 if (getSTI().getFeatureBits()[RISCV::FeatureRelax]) { 1811 auto *Assembler = getTargetStreamer().getStreamer().getAssemblerPtr(); 1812 if (Assembler != nullptr) { 1813 RISCVAsmBackend &MAB = 1814 static_cast<RISCVAsmBackend &>(Assembler->getBackend()); 1815 MAB.setForceRelocs(); 1816 } 1817 } 1818 1819 // First operand is token for instruction 1820 Operands.push_back(RISCVOperand::createToken(Name, NameLoc, isRV64())); 1821 1822 // If there are no more operands, then finish 1823 if (getLexer().is(AsmToken::EndOfStatement)) 1824 return false; 1825 1826 // Parse first operand 1827 if (parseOperand(Operands, Name)) 1828 return true; 1829 1830 // Parse until end of statement, consuming commas between operands 1831 unsigned OperandIdx = 1; 1832 while (getLexer().is(AsmToken::Comma)) { 1833 // Consume comma token 1834 getLexer().Lex(); 1835 1836 // Parse next operand 1837 if (parseOperand(Operands, Name)) 1838 return true; 1839 1840 ++OperandIdx; 1841 } 1842 1843 if (getLexer().isNot(AsmToken::EndOfStatement)) { 1844 SMLoc Loc = getLexer().getLoc(); 1845 getParser().eatToEndOfStatement(); 1846 return Error(Loc, "unexpected token"); 1847 } 1848 1849 getParser().Lex(); // Consume the EndOfStatement. 1850 return false; 1851 } 1852 1853 bool RISCVAsmParser::classifySymbolRef(const MCExpr *Expr, 1854 RISCVMCExpr::VariantKind &Kind, 1855 int64_t &Addend) { 1856 Kind = RISCVMCExpr::VK_RISCV_None; 1857 Addend = 0; 1858 1859 if (const RISCVMCExpr *RE = dyn_cast<RISCVMCExpr>(Expr)) { 1860 Kind = RE->getKind(); 1861 Expr = RE->getSubExpr(); 1862 } 1863 1864 // It's a simple symbol reference or constant with no addend. 1865 if (isa<MCConstantExpr>(Expr) || isa<MCSymbolRefExpr>(Expr)) 1866 return true; 1867 1868 const MCBinaryExpr *BE = dyn_cast<MCBinaryExpr>(Expr); 1869 if (!BE) 1870 return false; 1871 1872 if (!isa<MCSymbolRefExpr>(BE->getLHS())) 1873 return false; 1874 1875 if (BE->getOpcode() != MCBinaryExpr::Add && 1876 BE->getOpcode() != MCBinaryExpr::Sub) 1877 return false; 1878 1879 // We are able to support the subtraction of two symbol references 1880 if (BE->getOpcode() == MCBinaryExpr::Sub && 1881 isa<MCSymbolRefExpr>(BE->getRHS())) 1882 return true; 1883 1884 // See if the addend is a constant, otherwise there's more going 1885 // on here than we can deal with. 1886 auto AddendExpr = dyn_cast<MCConstantExpr>(BE->getRHS()); 1887 if (!AddendExpr) 1888 return false; 1889 1890 Addend = AddendExpr->getValue(); 1891 if (BE->getOpcode() == MCBinaryExpr::Sub) 1892 Addend = -Addend; 1893 1894 // It's some symbol reference + a constant addend 1895 return Kind != RISCVMCExpr::VK_RISCV_Invalid; 1896 } 1897 1898 bool RISCVAsmParser::ParseDirective(AsmToken DirectiveID) { 1899 // This returns false if this function recognizes the directive 1900 // regardless of whether it is successfully handles or reports an 1901 // error. Otherwise it returns true to give the generic parser a 1902 // chance at recognizing it. 1903 StringRef IDVal = DirectiveID.getString(); 1904 1905 if (IDVal == ".option") 1906 return parseDirectiveOption(); 1907 else if (IDVal == ".attribute") 1908 return parseDirectiveAttribute(); 1909 1910 return true; 1911 } 1912 1913 bool RISCVAsmParser::parseDirectiveOption() { 1914 MCAsmParser &Parser = getParser(); 1915 // Get the option token. 1916 AsmToken Tok = Parser.getTok(); 1917 // At the moment only identifiers are supported. 1918 if (Tok.isNot(AsmToken::Identifier)) 1919 return Error(Parser.getTok().getLoc(), 1920 "unexpected token, expected identifier"); 1921 1922 StringRef Option = Tok.getIdentifier(); 1923 1924 if (Option == "push") { 1925 getTargetStreamer().emitDirectiveOptionPush(); 1926 1927 Parser.Lex(); 1928 if (Parser.getTok().isNot(AsmToken::EndOfStatement)) 1929 return Error(Parser.getTok().getLoc(), 1930 "unexpected token, expected end of statement"); 1931 1932 pushFeatureBits(); 1933 return false; 1934 } 1935 1936 if (Option == "pop") { 1937 SMLoc StartLoc = Parser.getTok().getLoc(); 1938 getTargetStreamer().emitDirectiveOptionPop(); 1939 1940 Parser.Lex(); 1941 if (Parser.getTok().isNot(AsmToken::EndOfStatement)) 1942 return Error(Parser.getTok().getLoc(), 1943 "unexpected token, expected end of statement"); 1944 1945 if (popFeatureBits()) 1946 return Error(StartLoc, ".option pop with no .option push"); 1947 1948 return false; 1949 } 1950 1951 if (Option == "rvc") { 1952 getTargetStreamer().emitDirectiveOptionRVC(); 1953 1954 Parser.Lex(); 1955 if (Parser.getTok().isNot(AsmToken::EndOfStatement)) 1956 return Error(Parser.getTok().getLoc(), 1957 "unexpected token, expected end of statement"); 1958 1959 setFeatureBits(RISCV::FeatureStdExtC, "c"); 1960 return false; 1961 } 1962 1963 if (Option == "norvc") { 1964 getTargetStreamer().emitDirectiveOptionNoRVC(); 1965 1966 Parser.Lex(); 1967 if (Parser.getTok().isNot(AsmToken::EndOfStatement)) 1968 return Error(Parser.getTok().getLoc(), 1969 "unexpected token, expected end of statement"); 1970 1971 clearFeatureBits(RISCV::FeatureStdExtC, "c"); 1972 return false; 1973 } 1974 1975 if (Option == "pic") { 1976 getTargetStreamer().emitDirectiveOptionPIC(); 1977 1978 Parser.Lex(); 1979 if (Parser.getTok().isNot(AsmToken::EndOfStatement)) 1980 return Error(Parser.getTok().getLoc(), 1981 "unexpected token, expected end of statement"); 1982 1983 ParserOptions.IsPicEnabled = true; 1984 return false; 1985 } 1986 1987 if (Option == "nopic") { 1988 getTargetStreamer().emitDirectiveOptionNoPIC(); 1989 1990 Parser.Lex(); 1991 if (Parser.getTok().isNot(AsmToken::EndOfStatement)) 1992 return Error(Parser.getTok().getLoc(), 1993 "unexpected token, expected end of statement"); 1994 1995 ParserOptions.IsPicEnabled = false; 1996 return false; 1997 } 1998 1999 if (Option == "relax") { 2000 getTargetStreamer().emitDirectiveOptionRelax(); 2001 2002 Parser.Lex(); 2003 if (Parser.getTok().isNot(AsmToken::EndOfStatement)) 2004 return Error(Parser.getTok().getLoc(), 2005 "unexpected token, expected end of statement"); 2006 2007 setFeatureBits(RISCV::FeatureRelax, "relax"); 2008 return false; 2009 } 2010 2011 if (Option == "norelax") { 2012 getTargetStreamer().emitDirectiveOptionNoRelax(); 2013 2014 Parser.Lex(); 2015 if (Parser.getTok().isNot(AsmToken::EndOfStatement)) 2016 return Error(Parser.getTok().getLoc(), 2017 "unexpected token, expected end of statement"); 2018 2019 clearFeatureBits(RISCV::FeatureRelax, "relax"); 2020 return false; 2021 } 2022 2023 // Unknown option. 2024 Warning(Parser.getTok().getLoc(), 2025 "unknown option, expected 'push', 'pop', 'rvc', 'norvc', 'relax' or " 2026 "'norelax'"); 2027 Parser.eatToEndOfStatement(); 2028 return false; 2029 } 2030 2031 /// parseDirectiveAttribute 2032 /// ::= .attribute expression ',' ( expression | "string" ) 2033 /// ::= .attribute identifier ',' ( expression | "string" ) 2034 bool RISCVAsmParser::parseDirectiveAttribute() { 2035 MCAsmParser &Parser = getParser(); 2036 int64_t Tag; 2037 SMLoc TagLoc; 2038 TagLoc = Parser.getTok().getLoc(); 2039 if (Parser.getTok().is(AsmToken::Identifier)) { 2040 StringRef Name = Parser.getTok().getIdentifier(); 2041 Optional<unsigned> Ret = 2042 ELFAttrs::attrTypeFromString(Name, RISCVAttrs::RISCVAttributeTags); 2043 if (!Ret.hasValue()) { 2044 Error(TagLoc, "attribute name not recognised: " + Name); 2045 return false; 2046 } 2047 Tag = Ret.getValue(); 2048 Parser.Lex(); 2049 } else { 2050 const MCExpr *AttrExpr; 2051 2052 TagLoc = Parser.getTok().getLoc(); 2053 if (Parser.parseExpression(AttrExpr)) 2054 return true; 2055 2056 const MCConstantExpr *CE = dyn_cast<MCConstantExpr>(AttrExpr); 2057 if (check(!CE, TagLoc, "expected numeric constant")) 2058 return true; 2059 2060 Tag = CE->getValue(); 2061 } 2062 2063 if (Parser.parseToken(AsmToken::Comma, "comma expected")) 2064 return true; 2065 2066 StringRef StringValue; 2067 int64_t IntegerValue = 0; 2068 bool IsIntegerValue = true; 2069 2070 // RISC-V attributes have a string value if the tag number is odd 2071 // and an integer value if the tag number is even. 2072 if (Tag % 2) 2073 IsIntegerValue = false; 2074 2075 SMLoc ValueExprLoc = Parser.getTok().getLoc(); 2076 if (IsIntegerValue) { 2077 const MCExpr *ValueExpr; 2078 if (Parser.parseExpression(ValueExpr)) 2079 return true; 2080 2081 const MCConstantExpr *CE = dyn_cast<MCConstantExpr>(ValueExpr); 2082 if (!CE) 2083 return Error(ValueExprLoc, "expected numeric constant"); 2084 IntegerValue = CE->getValue(); 2085 } else { 2086 if (Parser.getTok().isNot(AsmToken::String)) 2087 return Error(Parser.getTok().getLoc(), "expected string constant"); 2088 2089 StringValue = Parser.getTok().getStringContents(); 2090 Parser.Lex(); 2091 } 2092 2093 if (Parser.parseToken(AsmToken::EndOfStatement, 2094 "unexpected token in '.attribute' directive")) 2095 return true; 2096 2097 if (Tag == RISCVAttrs::ARCH) { 2098 StringRef Arch = StringValue; 2099 if (Arch.consume_front("rv32")) 2100 clearFeatureBits(RISCV::Feature64Bit, "64bit"); 2101 else if (Arch.consume_front("rv64")) 2102 setFeatureBits(RISCV::Feature64Bit, "64bit"); 2103 else 2104 return Error(ValueExprLoc, "bad arch string " + Arch); 2105 2106 while (!Arch.empty()) { 2107 if (Arch[0] == 'i') 2108 clearFeatureBits(RISCV::FeatureRV32E, "e"); 2109 else if (Arch[0] == 'e') 2110 setFeatureBits(RISCV::FeatureRV32E, "e"); 2111 else if (Arch[0] == 'g') { 2112 clearFeatureBits(RISCV::FeatureRV32E, "e"); 2113 setFeatureBits(RISCV::FeatureStdExtM, "m"); 2114 setFeatureBits(RISCV::FeatureStdExtA, "a"); 2115 setFeatureBits(RISCV::FeatureStdExtF, "f"); 2116 setFeatureBits(RISCV::FeatureStdExtD, "d"); 2117 } else if (Arch[0] == 'm') 2118 setFeatureBits(RISCV::FeatureStdExtM, "m"); 2119 else if (Arch[0] == 'a') 2120 setFeatureBits(RISCV::FeatureStdExtA, "a"); 2121 else if (Arch[0] == 'f') 2122 setFeatureBits(RISCV::FeatureStdExtF, "f"); 2123 else if (Arch[0] == 'd') { 2124 setFeatureBits(RISCV::FeatureStdExtF, "f"); 2125 setFeatureBits(RISCV::FeatureStdExtD, "d"); 2126 } else if (Arch[0] == 'c') { 2127 setFeatureBits(RISCV::FeatureStdExtC, "c"); 2128 } else 2129 return Error(ValueExprLoc, "bad arch string " + Arch); 2130 2131 Arch = Arch.drop_front(1); 2132 int major = 0; 2133 int minor = 0; 2134 Arch.consumeInteger(10, major); 2135 Arch.consume_front("p"); 2136 Arch.consumeInteger(10, minor); 2137 if (major != 0 || minor != 0) { 2138 Arch = Arch.drop_until([](char c) { return c == '_' || c == '"'; }); 2139 Arch = Arch.drop_while([](char c) { return c == '_'; }); 2140 } 2141 } 2142 } 2143 2144 if (IsIntegerValue) 2145 getTargetStreamer().emitAttribute(Tag, IntegerValue); 2146 else { 2147 if (Tag != RISCVAttrs::ARCH) { 2148 getTargetStreamer().emitTextAttribute(Tag, StringValue); 2149 } else { 2150 std::string formalArchStr = "rv32"; 2151 if (getFeatureBits(RISCV::Feature64Bit)) 2152 formalArchStr = "rv64"; 2153 if (getFeatureBits(RISCV::FeatureRV32E)) 2154 formalArchStr = (Twine(formalArchStr) + "e1p9").str(); 2155 else 2156 formalArchStr = (Twine(formalArchStr) + "i2p0").str(); 2157 2158 if (getFeatureBits(RISCV::FeatureStdExtM)) 2159 formalArchStr = (Twine(formalArchStr) + "_m2p0").str(); 2160 if (getFeatureBits(RISCV::FeatureStdExtA)) 2161 formalArchStr = (Twine(formalArchStr) + "_a2p0").str(); 2162 if (getFeatureBits(RISCV::FeatureStdExtF)) 2163 formalArchStr = (Twine(formalArchStr) + "_f2p0").str(); 2164 if (getFeatureBits(RISCV::FeatureStdExtD)) 2165 formalArchStr = (Twine(formalArchStr) + "_d2p0").str(); 2166 if (getFeatureBits(RISCV::FeatureStdExtC)) 2167 formalArchStr = (Twine(formalArchStr) + "_c2p0").str(); 2168 2169 getTargetStreamer().emitTextAttribute(Tag, formalArchStr); 2170 } 2171 } 2172 2173 return false; 2174 } 2175 2176 void RISCVAsmParser::emitToStreamer(MCStreamer &S, const MCInst &Inst) { 2177 MCInst CInst; 2178 bool Res = compressInst(CInst, Inst, getSTI(), S.getContext()); 2179 if (Res) 2180 ++RISCVNumInstrsCompressed; 2181 S.emitInstruction((Res ? CInst : Inst), getSTI()); 2182 } 2183 2184 void RISCVAsmParser::emitLoadImm(Register DestReg, int64_t Value, 2185 MCStreamer &Out) { 2186 RISCVMatInt::InstSeq Seq; 2187 RISCVMatInt::generateInstSeq(Value, isRV64(), Seq); 2188 2189 Register SrcReg = RISCV::X0; 2190 for (RISCVMatInt::Inst &Inst : Seq) { 2191 if (Inst.Opc == RISCV::LUI) { 2192 emitToStreamer( 2193 Out, MCInstBuilder(RISCV::LUI).addReg(DestReg).addImm(Inst.Imm)); 2194 } else { 2195 emitToStreamer( 2196 Out, MCInstBuilder(Inst.Opc).addReg(DestReg).addReg(SrcReg).addImm( 2197 Inst.Imm)); 2198 } 2199 2200 // Only the first instruction has X0 as its source. 2201 SrcReg = DestReg; 2202 } 2203 } 2204 2205 void RISCVAsmParser::emitAuipcInstPair(MCOperand DestReg, MCOperand TmpReg, 2206 const MCExpr *Symbol, 2207 RISCVMCExpr::VariantKind VKHi, 2208 unsigned SecondOpcode, SMLoc IDLoc, 2209 MCStreamer &Out) { 2210 // A pair of instructions for PC-relative addressing; expands to 2211 // TmpLabel: AUIPC TmpReg, VKHi(symbol) 2212 // OP DestReg, TmpReg, %pcrel_lo(TmpLabel) 2213 MCContext &Ctx = getContext(); 2214 2215 MCSymbol *TmpLabel = Ctx.createTempSymbol( 2216 "pcrel_hi", /* AlwaysAddSuffix */ true, /* CanBeUnnamed */ false); 2217 Out.emitLabel(TmpLabel); 2218 2219 const RISCVMCExpr *SymbolHi = RISCVMCExpr::create(Symbol, VKHi, Ctx); 2220 emitToStreamer( 2221 Out, MCInstBuilder(RISCV::AUIPC).addOperand(TmpReg).addExpr(SymbolHi)); 2222 2223 const MCExpr *RefToLinkTmpLabel = 2224 RISCVMCExpr::create(MCSymbolRefExpr::create(TmpLabel, Ctx), 2225 RISCVMCExpr::VK_RISCV_PCREL_LO, Ctx); 2226 2227 emitToStreamer(Out, MCInstBuilder(SecondOpcode) 2228 .addOperand(DestReg) 2229 .addOperand(TmpReg) 2230 .addExpr(RefToLinkTmpLabel)); 2231 } 2232 2233 void RISCVAsmParser::emitLoadLocalAddress(MCInst &Inst, SMLoc IDLoc, 2234 MCStreamer &Out) { 2235 // The load local address pseudo-instruction "lla" is used in PC-relative 2236 // addressing of local symbols: 2237 // lla rdest, symbol 2238 // expands to 2239 // TmpLabel: AUIPC rdest, %pcrel_hi(symbol) 2240 // ADDI rdest, rdest, %pcrel_lo(TmpLabel) 2241 MCOperand DestReg = Inst.getOperand(0); 2242 const MCExpr *Symbol = Inst.getOperand(1).getExpr(); 2243 emitAuipcInstPair(DestReg, DestReg, Symbol, RISCVMCExpr::VK_RISCV_PCREL_HI, 2244 RISCV::ADDI, IDLoc, Out); 2245 } 2246 2247 void RISCVAsmParser::emitLoadAddress(MCInst &Inst, SMLoc IDLoc, 2248 MCStreamer &Out) { 2249 // The load address pseudo-instruction "la" is used in PC-relative and 2250 // GOT-indirect addressing of global symbols: 2251 // la rdest, symbol 2252 // expands to either (for non-PIC) 2253 // TmpLabel: AUIPC rdest, %pcrel_hi(symbol) 2254 // ADDI rdest, rdest, %pcrel_lo(TmpLabel) 2255 // or (for PIC) 2256 // TmpLabel: AUIPC rdest, %got_pcrel_hi(symbol) 2257 // Lx rdest, %pcrel_lo(TmpLabel)(rdest) 2258 MCOperand DestReg = Inst.getOperand(0); 2259 const MCExpr *Symbol = Inst.getOperand(1).getExpr(); 2260 unsigned SecondOpcode; 2261 RISCVMCExpr::VariantKind VKHi; 2262 if (ParserOptions.IsPicEnabled) { 2263 SecondOpcode = isRV64() ? RISCV::LD : RISCV::LW; 2264 VKHi = RISCVMCExpr::VK_RISCV_GOT_HI; 2265 } else { 2266 SecondOpcode = RISCV::ADDI; 2267 VKHi = RISCVMCExpr::VK_RISCV_PCREL_HI; 2268 } 2269 emitAuipcInstPair(DestReg, DestReg, Symbol, VKHi, SecondOpcode, IDLoc, Out); 2270 } 2271 2272 void RISCVAsmParser::emitLoadTLSIEAddress(MCInst &Inst, SMLoc IDLoc, 2273 MCStreamer &Out) { 2274 // The load TLS IE address pseudo-instruction "la.tls.ie" is used in 2275 // initial-exec TLS model addressing of global symbols: 2276 // la.tls.ie rdest, symbol 2277 // expands to 2278 // TmpLabel: AUIPC rdest, %tls_ie_pcrel_hi(symbol) 2279 // Lx rdest, %pcrel_lo(TmpLabel)(rdest) 2280 MCOperand DestReg = Inst.getOperand(0); 2281 const MCExpr *Symbol = Inst.getOperand(1).getExpr(); 2282 unsigned SecondOpcode = isRV64() ? RISCV::LD : RISCV::LW; 2283 emitAuipcInstPair(DestReg, DestReg, Symbol, RISCVMCExpr::VK_RISCV_TLS_GOT_HI, 2284 SecondOpcode, IDLoc, Out); 2285 } 2286 2287 void RISCVAsmParser::emitLoadTLSGDAddress(MCInst &Inst, SMLoc IDLoc, 2288 MCStreamer &Out) { 2289 // The load TLS GD address pseudo-instruction "la.tls.gd" is used in 2290 // global-dynamic TLS model addressing of global symbols: 2291 // la.tls.gd rdest, symbol 2292 // expands to 2293 // TmpLabel: AUIPC rdest, %tls_gd_pcrel_hi(symbol) 2294 // ADDI rdest, rdest, %pcrel_lo(TmpLabel) 2295 MCOperand DestReg = Inst.getOperand(0); 2296 const MCExpr *Symbol = Inst.getOperand(1).getExpr(); 2297 emitAuipcInstPair(DestReg, DestReg, Symbol, RISCVMCExpr::VK_RISCV_TLS_GD_HI, 2298 RISCV::ADDI, IDLoc, Out); 2299 } 2300 2301 void RISCVAsmParser::emitLoadStoreSymbol(MCInst &Inst, unsigned Opcode, 2302 SMLoc IDLoc, MCStreamer &Out, 2303 bool HasTmpReg) { 2304 // The load/store pseudo-instruction does a pc-relative load with 2305 // a symbol. 2306 // 2307 // The expansion looks like this 2308 // 2309 // TmpLabel: AUIPC tmp, %pcrel_hi(symbol) 2310 // [S|L]X rd, %pcrel_lo(TmpLabel)(tmp) 2311 MCOperand DestReg = Inst.getOperand(0); 2312 unsigned SymbolOpIdx = HasTmpReg ? 2 : 1; 2313 unsigned TmpRegOpIdx = HasTmpReg ? 1 : 0; 2314 MCOperand TmpReg = Inst.getOperand(TmpRegOpIdx); 2315 const MCExpr *Symbol = Inst.getOperand(SymbolOpIdx).getExpr(); 2316 emitAuipcInstPair(DestReg, TmpReg, Symbol, RISCVMCExpr::VK_RISCV_PCREL_HI, 2317 Opcode, IDLoc, Out); 2318 } 2319 2320 bool RISCVAsmParser::checkPseudoAddTPRel(MCInst &Inst, 2321 OperandVector &Operands) { 2322 assert(Inst.getOpcode() == RISCV::PseudoAddTPRel && "Invalid instruction"); 2323 assert(Inst.getOperand(2).isReg() && "Unexpected second operand kind"); 2324 if (Inst.getOperand(2).getReg() != RISCV::X4) { 2325 SMLoc ErrorLoc = ((RISCVOperand &)*Operands[3]).getStartLoc(); 2326 return Error(ErrorLoc, "the second input operand must be tp/x4 when using " 2327 "%tprel_add modifier"); 2328 } 2329 2330 return false; 2331 } 2332 2333 std::unique_ptr<RISCVOperand> RISCVAsmParser::defaultMaskRegOp() const { 2334 return RISCVOperand::createReg(RISCV::NoRegister, llvm::SMLoc(), 2335 llvm::SMLoc(), isRV64()); 2336 } 2337 2338 bool RISCVAsmParser::validateInstruction(MCInst &Inst, 2339 OperandVector &Operands) { 2340 const MCInstrDesc &MCID = MII.get(Inst.getOpcode()); 2341 unsigned TargetFlags = 2342 (MCID.TSFlags >> RISCV::ConstraintOffset) & RISCV::ConstraintMask; 2343 if (TargetFlags == RISCV::NoConstraint) 2344 return false; 2345 2346 unsigned DestReg = Inst.getOperand(0).getReg(); 2347 unsigned CheckReg; 2348 // Operands[1] will be the first operand, DestReg. 2349 SMLoc Loc = Operands[1]->getStartLoc(); 2350 if (TargetFlags & RISCV::VS2Constraint) { 2351 CheckReg = Inst.getOperand(1).getReg(); 2352 if (DestReg == CheckReg) 2353 return Error(Loc, "The destination vector register group cannot overlap" 2354 " the source vector register group."); 2355 } 2356 if ((TargetFlags & RISCV::VS1Constraint) && (Inst.getOperand(2).isReg())) { 2357 CheckReg = Inst.getOperand(2).getReg(); 2358 if (DestReg == CheckReg) 2359 return Error(Loc, "The destination vector register group cannot overlap" 2360 " the source vector register group."); 2361 } 2362 if ((TargetFlags & RISCV::VMConstraint) && (DestReg == RISCV::V0)) { 2363 // vadc, vsbc are special cases. These instructions have no mask register. 2364 // The destination register could not be V0. 2365 unsigned Opcode = Inst.getOpcode(); 2366 if (Opcode == RISCV::VADC_VVM || Opcode == RISCV::VADC_VXM || 2367 Opcode == RISCV::VADC_VIM || Opcode == RISCV::VSBC_VVM || 2368 Opcode == RISCV::VSBC_VXM) 2369 return Error(Loc, "The destination vector register group cannot be V0."); 2370 2371 // Regardless masked or unmasked version, the number of operands is the 2372 // same. For example, "viota.m v0, v2" is "viota.m v0, v2, NoRegister" 2373 // actually. We need to check the last operand to ensure whether it is 2374 // masked or not. 2375 if ((TargetFlags & RISCV::OneInput) && (Inst.getNumOperands() == 3)) 2376 CheckReg = Inst.getOperand(2).getReg(); 2377 else if (Inst.getNumOperands() == 4) 2378 CheckReg = Inst.getOperand(3).getReg(); 2379 if (DestReg == CheckReg) 2380 return Error(Loc, "The destination vector register group cannot overlap" 2381 " the mask register."); 2382 } 2383 return false; 2384 } 2385 2386 bool RISCVAsmParser::processInstruction(MCInst &Inst, SMLoc IDLoc, 2387 OperandVector &Operands, 2388 MCStreamer &Out) { 2389 Inst.setLoc(IDLoc); 2390 2391 switch (Inst.getOpcode()) { 2392 default: 2393 break; 2394 case RISCV::PseudoLI: { 2395 Register Reg = Inst.getOperand(0).getReg(); 2396 const MCOperand &Op1 = Inst.getOperand(1); 2397 if (Op1.isExpr()) { 2398 // We must have li reg, %lo(sym) or li reg, %pcrel_lo(sym) or similar. 2399 // Just convert to an addi. This allows compatibility with gas. 2400 emitToStreamer(Out, MCInstBuilder(RISCV::ADDI) 2401 .addReg(Reg) 2402 .addReg(RISCV::X0) 2403 .addExpr(Op1.getExpr())); 2404 return false; 2405 } 2406 int64_t Imm = Inst.getOperand(1).getImm(); 2407 // On RV32 the immediate here can either be a signed or an unsigned 2408 // 32-bit number. Sign extension has to be performed to ensure that Imm 2409 // represents the expected signed 64-bit number. 2410 if (!isRV64()) 2411 Imm = SignExtend64<32>(Imm); 2412 emitLoadImm(Reg, Imm, Out); 2413 return false; 2414 } 2415 case RISCV::PseudoLLA: 2416 emitLoadLocalAddress(Inst, IDLoc, Out); 2417 return false; 2418 case RISCV::PseudoLA: 2419 emitLoadAddress(Inst, IDLoc, Out); 2420 return false; 2421 case RISCV::PseudoLA_TLS_IE: 2422 emitLoadTLSIEAddress(Inst, IDLoc, Out); 2423 return false; 2424 case RISCV::PseudoLA_TLS_GD: 2425 emitLoadTLSGDAddress(Inst, IDLoc, Out); 2426 return false; 2427 case RISCV::PseudoLB: 2428 emitLoadStoreSymbol(Inst, RISCV::LB, IDLoc, Out, /*HasTmpReg=*/false); 2429 return false; 2430 case RISCV::PseudoLBU: 2431 emitLoadStoreSymbol(Inst, RISCV::LBU, IDLoc, Out, /*HasTmpReg=*/false); 2432 return false; 2433 case RISCV::PseudoLH: 2434 emitLoadStoreSymbol(Inst, RISCV::LH, IDLoc, Out, /*HasTmpReg=*/false); 2435 return false; 2436 case RISCV::PseudoLHU: 2437 emitLoadStoreSymbol(Inst, RISCV::LHU, IDLoc, Out, /*HasTmpReg=*/false); 2438 return false; 2439 case RISCV::PseudoLW: 2440 emitLoadStoreSymbol(Inst, RISCV::LW, IDLoc, Out, /*HasTmpReg=*/false); 2441 return false; 2442 case RISCV::PseudoLWU: 2443 emitLoadStoreSymbol(Inst, RISCV::LWU, IDLoc, Out, /*HasTmpReg=*/false); 2444 return false; 2445 case RISCV::PseudoLD: 2446 emitLoadStoreSymbol(Inst, RISCV::LD, IDLoc, Out, /*HasTmpReg=*/false); 2447 return false; 2448 case RISCV::PseudoFLW: 2449 emitLoadStoreSymbol(Inst, RISCV::FLW, IDLoc, Out, /*HasTmpReg=*/true); 2450 return false; 2451 case RISCV::PseudoFLD: 2452 emitLoadStoreSymbol(Inst, RISCV::FLD, IDLoc, Out, /*HasTmpReg=*/true); 2453 return false; 2454 case RISCV::PseudoSB: 2455 emitLoadStoreSymbol(Inst, RISCV::SB, IDLoc, Out, /*HasTmpReg=*/true); 2456 return false; 2457 case RISCV::PseudoSH: 2458 emitLoadStoreSymbol(Inst, RISCV::SH, IDLoc, Out, /*HasTmpReg=*/true); 2459 return false; 2460 case RISCV::PseudoSW: 2461 emitLoadStoreSymbol(Inst, RISCV::SW, IDLoc, Out, /*HasTmpReg=*/true); 2462 return false; 2463 case RISCV::PseudoSD: 2464 emitLoadStoreSymbol(Inst, RISCV::SD, IDLoc, Out, /*HasTmpReg=*/true); 2465 return false; 2466 case RISCV::PseudoFSW: 2467 emitLoadStoreSymbol(Inst, RISCV::FSW, IDLoc, Out, /*HasTmpReg=*/true); 2468 return false; 2469 case RISCV::PseudoFSD: 2470 emitLoadStoreSymbol(Inst, RISCV::FSD, IDLoc, Out, /*HasTmpReg=*/true); 2471 return false; 2472 case RISCV::PseudoAddTPRel: 2473 if (checkPseudoAddTPRel(Inst, Operands)) 2474 return true; 2475 break; 2476 } 2477 2478 emitToStreamer(Out, Inst); 2479 return false; 2480 } 2481 2482 extern "C" LLVM_EXTERNAL_VISIBILITY void LLVMInitializeRISCVAsmParser() { 2483 RegisterMCAsmParser<RISCVAsmParser> X(getTheRISCV32Target()); 2484 RegisterMCAsmParser<RISCVAsmParser> Y(getTheRISCV64Target()); 2485 } 2486