lib/CodeGen/MIRParser/MIParser.cpp

   1 //===- MIParser.cpp - Machine instructions parser implementation ----------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file implements the parsing of machine instructions.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "MIParser.h"
  15 #include "MILexer.h"
  16 #include "llvm/ADT/StringMap.h"
  17 #include "llvm/AsmParser/Parser.h"
  18 #include "llvm/AsmParser/SlotMapping.h"
  19 #include "llvm/CodeGen/MachineBasicBlock.h"
  20 #include "llvm/CodeGen/MachineFunction.h"
  21 #include "llvm/CodeGen/MachineFrameInfo.h"
  22 #include "llvm/CodeGen/MachineInstr.h"
  23 #include "llvm/CodeGen/MachineInstrBuilder.h"
  24 #include "llvm/CodeGen/MachineMemOperand.h"
  25 #include "llvm/CodeGen/MachineModuleInfo.h"
  26 #include "llvm/IR/Instructions.h"
  27 #include "llvm/IR/Constants.h"
  28 #include "llvm/IR/Module.h"
  29 #include "llvm/IR/ModuleSlotTracker.h"
  30 #include "llvm/IR/ValueSymbolTable.h"
  31 #include "llvm/Support/raw_ostream.h"
  32 #include "llvm/Support/SourceMgr.h"
  33 #include "llvm/Target/TargetSubtargetInfo.h"
  34 #include "llvm/Target/TargetInstrInfo.h"
  35
  36 using namespace llvm;
  37
  38 namespace {
  39
  40 /// A wrapper struct around the 'MachineOperand' struct that includes a source
  41 /// range.
  42 struct MachineOperandWithLocation {
  43   MachineOperand Operand;
  44   StringRef::iterator Begin;
  45   StringRef::iterator End;
  46
  47   MachineOperandWithLocation(const MachineOperand &Operand,
  48                              StringRef::iterator Begin, StringRef::iterator End)
  49       : Operand(Operand), Begin(Begin), End(End) {}
  50 };
  51
  52 class MIParser {
  53   SourceMgr &SM;
  54   MachineFunction &MF;
  55   SMDiagnostic &Error;
  56   StringRef Source, CurrentSource;
  57   MIToken Token;
  58   const PerFunctionMIParsingState &PFS;
  59   /// Maps from indices to unnamed global values and metadata nodes.
  60   const SlotMapping &IRSlots;
  61   /// Maps from instruction names to op codes.
  62   StringMap<unsigned> Names2InstrOpCodes;
  63   /// Maps from register names to registers.
  64   StringMap<unsigned> Names2Regs;
  65   /// Maps from register mask names to register masks.
  66   StringMap<const uint32_t *> Names2RegMasks;
  67   /// Maps from subregister names to subregister indices.
  68   StringMap<unsigned> Names2SubRegIndices;
  69   /// Maps from slot numbers to function's unnamed basic blocks.
  70   DenseMap<unsigned, const BasicBlock *> Slots2BasicBlocks;
  71   /// Maps from target index names to target indices.
  72   StringMap<int> Names2TargetIndices;
  73
  74 public:
  75   MIParser(SourceMgr &SM, MachineFunction &MF, SMDiagnostic &Error,
  76            StringRef Source, const PerFunctionMIParsingState &PFS,
  77            const SlotMapping &IRSlots);
  78
  79   void lex();
  80
  81   /// Report an error at the current location with the given message.
  82   ///
  83   /// This function always return true.
  84   bool error(const Twine &Msg);
  85
  86   /// Report an error at the given location with the given message.
  87   ///
  88   /// This function always return true.
  89   bool error(StringRef::iterator Loc, const Twine &Msg);
  90
  91   bool parse(MachineInstr *&MI);
  92   bool parseStandaloneMBB(MachineBasicBlock *&MBB);
  93   bool parseStandaloneNamedRegister(unsigned &Reg);
  94   bool parseStandaloneVirtualRegister(unsigned &Reg);
  95   bool parseStandaloneIRBlockReference(const BasicBlock *&BB);
  96
  97   bool parseRegister(unsigned &Reg);
  98   bool parseRegisterFlag(unsigned &Flags);
  99   bool parseSubRegisterIndex(unsigned &SubReg);
 100   bool parseRegisterOperand(MachineOperand &Dest, bool IsDef = false);
 101   bool parseImmediateOperand(MachineOperand &Dest);
 102   bool parseIRConstant(StringRef::iterator Loc, const Constant *&C);
 103   bool parseTypedImmediateOperand(MachineOperand &Dest);
 104   bool parseFPImmediateOperand(MachineOperand &Dest);
 105   bool parseMBBReference(MachineBasicBlock *&MBB);
 106   bool parseMBBOperand(MachineOperand &Dest);
 107   bool parseStackObjectOperand(MachineOperand &Dest);
 108   bool parseFixedStackObjectOperand(MachineOperand &Dest);
 109   bool parseGlobalValue(GlobalValue *&GV);
 110   bool parseGlobalAddressOperand(MachineOperand &Dest);
 111   bool parseConstantPoolIndexOperand(MachineOperand &Dest);
 112   bool parseJumpTableIndexOperand(MachineOperand &Dest);
 113   bool parseExternalSymbolOperand(MachineOperand &Dest);
 114   bool parseMDNode(MDNode *&Node);
 115   bool parseMetadataOperand(MachineOperand &Dest);
 116   bool parseCFIOffset(int &Offset);
 117   bool parseCFIRegister(unsigned &Reg);
 118   bool parseCFIOperand(MachineOperand &Dest);
 119   bool parseIRBlock(BasicBlock *&BB, const Function &F);
 120   bool parseBlockAddressOperand(MachineOperand &Dest);
 121   bool parseTargetIndexOperand(MachineOperand &Dest);
 122   bool parseMachineOperand(MachineOperand &Dest);
 123   bool parseOperandsOffset(MachineOperand &Op);
 124   bool parseIRValue(Value *&V);
 125   bool parseMemoryOperandFlag(unsigned &Flags);
 126   bool parseMachineMemoryOperand(MachineMemOperand *&Dest);
 127
 128 private:
 129   /// Convert the integer literal in the current token into an unsigned integer.
 130   ///
 131   /// Return true if an error occurred.
 132   bool getUnsigned(unsigned &Result);
 133
 134   /// Convert the integer literal in the current token into an uint64.
 135   ///
 136   /// Return true if an error occurred.
 137   bool getUint64(uint64_t &Result);
 138
 139   /// If the current token is of the given kind, consume it and return false.
 140   /// Otherwise report an error and return true.
 141   bool expectAndConsume(MIToken::TokenKind TokenKind);
 142
 143   void initNames2InstrOpCodes();
 144
 145   /// Try to convert an instruction name to an opcode. Return true if the
 146   /// instruction name is invalid.
 147   bool parseInstrName(StringRef InstrName, unsigned &OpCode);
 148
 149   bool parseInstruction(unsigned &OpCode, unsigned &Flags);
 150
 151   bool verifyImplicitOperands(ArrayRef<MachineOperandWithLocation> Operands,
 152                               const MCInstrDesc &MCID);
 153
 154   void initNames2Regs();
 155
 156   /// Try to convert a register name to a register number. Return true if the
 157   /// register name is invalid.
 158   bool getRegisterByName(StringRef RegName, unsigned &Reg);
 159
 160   void initNames2RegMasks();
 161
 162   /// Check if the given identifier is a name of a register mask.
 163   ///
 164   /// Return null if the identifier isn't a register mask.
 165   const uint32_t *getRegMask(StringRef Identifier);
 166
 167   void initNames2SubRegIndices();
 168
 169   /// Check if the given identifier is a name of a subregister index.
 170   ///
 171   /// Return 0 if the name isn't a subregister index class.
 172   unsigned getSubRegIndex(StringRef Name);
 173
 174   void initSlots2BasicBlocks();
 175
 176   const BasicBlock *getIRBlock(unsigned Slot);
 177
 178   void initNames2TargetIndices();
 179
 180   /// Try to convert a name of target index to the corresponding target index.
 181   ///
 182   /// Return true if the name isn't a name of a target index.
 183   bool getTargetIndex(StringRef Name, int &Index);
 184 };
 185
 186 } // end anonymous namespace
 187
 188 MIParser::MIParser(SourceMgr &SM, MachineFunction &MF, SMDiagnostic &Error,
 189                    StringRef Source, const PerFunctionMIParsingState &PFS,
 190                    const SlotMapping &IRSlots)
 191     : SM(SM), MF(MF), Error(Error), Source(Source), CurrentSource(Source),
 192       Token(MIToken::Error, StringRef()), PFS(PFS), IRSlots(IRSlots) {}
 193
 194 void MIParser::lex() {
 195   CurrentSource = lexMIToken(
 196       CurrentSource, Token,
 197       [this](StringRef::iterator Loc, const Twine &Msg) { error(Loc, Msg); });
 198 }
 199
 200 bool MIParser::error(const Twine &Msg) { return error(Token.location(), Msg); }
 201
 202 bool MIParser::error(StringRef::iterator Loc, const Twine &Msg) {
 203   assert(Loc >= Source.data() && Loc <= (Source.data() + Source.size()));
 204   Error = SMDiagnostic(
 205       SM, SMLoc(),
 206       SM.getMemoryBuffer(SM.getMainFileID())->getBufferIdentifier(), 1,
 207       Loc - Source.data(), SourceMgr::DK_Error, Msg.str(), Source, None, None);
 208   return true;
 209 }
 210
 211 static const char *toString(MIToken::TokenKind TokenKind) {
 212   switch (TokenKind) {
 213   case MIToken::comma:
 214     return "','";
 215   case MIToken::equal:
 216     return "'='";
 217   case MIToken::lparen:
 218     return "'('";
 219   case MIToken::rparen:
 220     return "')'";
 221   default:
 222     return "<unknown token>";
 223   }
 224 }
 225
 226 bool MIParser::expectAndConsume(MIToken::TokenKind TokenKind) {
 227   if (Token.isNot(TokenKind))
 228     return error(Twine("expected ") + toString(TokenKind));
 229   lex();
 230   return false;
 231 }
 232
 233 bool MIParser::parse(MachineInstr *&MI) {
 234   lex();
 235
 236   // Parse any register operands before '='
 237   MachineOperand MO = MachineOperand::CreateImm(0);
 238   SmallVector<MachineOperandWithLocation, 8> Operands;
 239   while (Token.isRegister() || Token.isRegisterFlag()) {
 240     auto Loc = Token.location();
 241     if (parseRegisterOperand(MO, /*IsDef=*/true))
 242       return true;
 243     Operands.push_back(MachineOperandWithLocation(MO, Loc, Token.location()));
 244     if (Token.isNot(MIToken::comma))
 245       break;
 246     lex();
 247   }
 248   if (!Operands.empty() && expectAndConsume(MIToken::equal))
 249     return true;
 250
 251   unsigned OpCode, Flags = 0;
 252   if (Token.isError() || parseInstruction(OpCode, Flags))
 253     return true;
 254
 255   // TODO: Parse the bundle instruction flags.
 256
 257   // Parse the remaining machine operands.
 258   while (Token.isNot(MIToken::Eof) && Token.isNot(MIToken::kw_debug_location) &&
 259          Token.isNot(MIToken::coloncolon)) {
 260     auto Loc = Token.location();
 261     if (parseMachineOperand(MO))
 262       return true;
 263     Operands.push_back(MachineOperandWithLocation(MO, Loc, Token.location()));
 264     if (Token.is(MIToken::Eof) || Token.is(MIToken::coloncolon))
 265       break;
 266     if (Token.isNot(MIToken::comma))
 267       return error("expected ',' before the next machine operand");
 268     lex();
 269   }
 270
 271   DebugLoc DebugLocation;
 272   if (Token.is(MIToken::kw_debug_location)) {
 273     lex();
 274     if (Token.isNot(MIToken::exclaim))
 275       return error("expected a metadata node after 'debug-location'");
 276     MDNode *Node = nullptr;
 277     if (parseMDNode(Node))
 278       return true;
 279     DebugLocation = DebugLoc(Node);
 280   }
 281
 282   // Parse the machine memory operands.
 283   SmallVector<MachineMemOperand *, 2> MemOperands;
 284   if (Token.is(MIToken::coloncolon)) {
 285     lex();
 286     while (Token.isNot(MIToken::Eof)) {
 287       MachineMemOperand *MemOp = nullptr;
 288       if (parseMachineMemoryOperand(MemOp))
 289         return true;
 290       MemOperands.push_back(MemOp);
 291       if (Token.is(MIToken::Eof))
 292         break;
 293       if (Token.isNot(MIToken::comma))
 294         return error("expected ',' before the next machine memory operand");
 295       lex();
 296     }
 297   }
 298
 299   const auto &MCID = MF.getSubtarget().getInstrInfo()->get(OpCode);
 300   if (!MCID.isVariadic()) {
 301     // FIXME: Move the implicit operand verification to the machine verifier.
 302     if (verifyImplicitOperands(Operands, MCID))
 303       return true;
 304   }
 305
 306   // TODO: Check for extraneous machine operands.
 307   MI = MF.CreateMachineInstr(MCID, DebugLocation, /*NoImplicit=*/true);
 308   MI->setFlags(Flags);
 309   for (const auto &Operand : Operands)
 310     MI->addOperand(MF, Operand.Operand);
 311   if (MemOperands.empty())
 312     return false;
 313   MachineInstr::mmo_iterator MemRefs =
 314       MF.allocateMemRefsArray(MemOperands.size());
 315   std::copy(MemOperands.begin(), MemOperands.end(), MemRefs);
 316   MI->setMemRefs(MemRefs, MemRefs + MemOperands.size());
 317   return false;
 318 }
 319
 320 bool MIParser::parseStandaloneMBB(MachineBasicBlock *&MBB) {
 321   lex();
 322   if (Token.isNot(MIToken::MachineBasicBlock))
 323     return error("expected a machine basic block reference");
 324   if (parseMBBReference(MBB))
 325     return true;
 326   lex();
 327   if (Token.isNot(MIToken::Eof))
 328     return error(
 329         "expected end of string after the machine basic block reference");
 330   return false;
 331 }
 332
 333 bool MIParser::parseStandaloneNamedRegister(unsigned &Reg) {
 334   lex();
 335   if (Token.isNot(MIToken::NamedRegister))
 336     return error("expected a named register");
 337   if (parseRegister(Reg))
 338     return 0;
 339   lex();
 340   if (Token.isNot(MIToken::Eof))
 341     return error("expected end of string after the register reference");
 342   return false;
 343 }
 344
 345 bool MIParser::parseStandaloneVirtualRegister(unsigned &Reg) {
 346   lex();
 347   if (Token.isNot(MIToken::VirtualRegister))
 348     return error("expected a virtual register");
 349   if (parseRegister(Reg))
 350     return 0;
 351   lex();
 352   if (Token.isNot(MIToken::Eof))
 353     return error("expected end of string after the register reference");
 354   return false;
 355 }
 356
 357 bool MIParser::parseStandaloneIRBlockReference(const BasicBlock *&BB) {
 358   lex();
 359   if (Token.isNot(MIToken::IRBlock))
 360     return error("expected an IR block reference");
 361   unsigned SlotNumber = 0;
 362   if (getUnsigned(SlotNumber))
 363     return true;
 364   BB = getIRBlock(SlotNumber);
 365   if (!BB)
 366     return error(Twine("use of undefined IR block '%ir-block.") +
 367                  Twine(SlotNumber) + "'");
 368   lex();
 369   if (Token.isNot(MIToken::Eof))
 370     return error("expected end of string after the IR block reference");
 371   return false;
 372 }
 373
 374 static const char *printImplicitRegisterFlag(const MachineOperand &MO) {
 375   assert(MO.isImplicit());
 376   return MO.isDef() ? "implicit-def" : "implicit";
 377 }
 378
 379 static std::string getRegisterName(const TargetRegisterInfo *TRI,
 380                                    unsigned Reg) {
 381   assert(TargetRegisterInfo::isPhysicalRegister(Reg) && "expected phys reg");
 382   return StringRef(TRI->getName(Reg)).lower();
 383 }
 384
 385 bool MIParser::verifyImplicitOperands(
 386     ArrayRef<MachineOperandWithLocation> Operands, const MCInstrDesc &MCID) {
 387   if (MCID.isCall())
 388     // We can't verify call instructions as they can contain arbitrary implicit
 389     // register and register mask operands.
 390     return false;
 391
 392   // Gather all the expected implicit operands.
 393   SmallVector<MachineOperand, 4> ImplicitOperands;
 394   if (MCID.ImplicitDefs)
 395     for (const uint16_t *ImpDefs = MCID.getImplicitDefs(); *ImpDefs; ++ImpDefs)
 396       ImplicitOperands.push_back(
 397           MachineOperand::CreateReg(*ImpDefs, true, true));
 398   if (MCID.ImplicitUses)
 399     for (const uint16_t *ImpUses = MCID.getImplicitUses(); *ImpUses; ++ImpUses)
 400       ImplicitOperands.push_back(
 401           MachineOperand::CreateReg(*ImpUses, false, true));
 402
 403   const auto *TRI = MF.getSubtarget().getRegisterInfo();
 404   assert(TRI && "Expected target register info");
 405   size_t I = ImplicitOperands.size(), J = Operands.size();
 406   while (I) {
 407     --I;
 408     if (J) {
 409       --J;
 410       const auto &ImplicitOperand = ImplicitOperands[I];
 411       const auto &Operand = Operands[J].Operand;
 412       if (ImplicitOperand.isIdenticalTo(Operand))
 413         continue;
 414       if (Operand.isReg() && Operand.isImplicit()) {
 415         return error(Operands[J].Begin,
 416                      Twine("expected an implicit register operand '") +
 417                          printImplicitRegisterFlag(ImplicitOperand) + " %" +
 418                          getRegisterName(TRI, ImplicitOperand.getReg()) + "'");
 419       }
 420     }
 421     // TODO: Fix source location when Operands[J].end is right before '=', i.e:
 422     // insead of reporting an error at this location:
 423     //            %eax = MOV32r0
 424     //                 ^
 425     // report the error at the following location:
 426     //            %eax = MOV32r0
 427     //                          ^
 428     return error(J < Operands.size() ? Operands[J].End : Token.location(),
 429                  Twine("missing implicit register operand '") +
 430                      printImplicitRegisterFlag(ImplicitOperands[I]) + " %" +
 431                      getRegisterName(TRI, ImplicitOperands[I].getReg()) + "'");
 432   }
 433   return false;
 434 }
 435
 436 bool MIParser::parseInstruction(unsigned &OpCode, unsigned &Flags) {
 437   if (Token.is(MIToken::kw_frame_setup)) {
 438     Flags |= MachineInstr::FrameSetup;
 439     lex();
 440   }
 441   if (Token.isNot(MIToken::Identifier))
 442     return error("expected a machine instruction");
 443   StringRef InstrName = Token.stringValue();
 444   if (parseInstrName(InstrName, OpCode))
 445     return error(Twine("unknown machine instruction name '") + InstrName + "'");
 446   lex();
 447   return false;
 448 }
 449
 450 bool MIParser::parseRegister(unsigned &Reg) {
 451   switch (Token.kind()) {
 452   case MIToken::underscore:
 453     Reg = 0;
 454     break;
 455   case MIToken::NamedRegister: {
 456     StringRef Name = Token.stringValue();
 457     if (getRegisterByName(Name, Reg))
 458       return error(Twine("unknown register name '") + Name + "'");
 459     break;
 460   }
 461   case MIToken::VirtualRegister: {
 462     unsigned ID;
 463     if (getUnsigned(ID))
 464       return true;
 465     const auto RegInfo = PFS.VirtualRegisterSlots.find(ID);
 466     if (RegInfo == PFS.VirtualRegisterSlots.end())
 467       return error(Twine("use of undefined virtual register '%") + Twine(ID) +
 468                    "'");
 469     Reg = RegInfo->second;
 470     break;
 471   }
 472   // TODO: Parse other register kinds.
 473   default:
 474     llvm_unreachable("The current token should be a register");
 475   }
 476   return false;
 477 }
 478
 479 bool MIParser::parseRegisterFlag(unsigned &Flags) {
 480   const unsigned OldFlags = Flags;
 481   switch (Token.kind()) {
 482   case MIToken::kw_implicit:
 483     Flags |= RegState::Implicit;
 484     break;
 485   case MIToken::kw_implicit_define:
 486     Flags |= RegState::ImplicitDefine;
 487     break;
 488   case MIToken::kw_dead:
 489     Flags |= RegState::Dead;
 490     break;
 491   case MIToken::kw_killed:
 492     Flags |= RegState::Kill;
 493     break;
 494   case MIToken::kw_undef:
 495     Flags |= RegState::Undef;
 496     break;
 497   case MIToken::kw_early_clobber:
 498     Flags |= RegState::EarlyClobber;
 499     break;
 500   case MIToken::kw_debug_use:
 501     Flags |= RegState::Debug;
 502     break;
 503   // TODO: parse the other register flags.
 504   default:
 505     llvm_unreachable("The current token should be a register flag");
 506   }
 507   if (OldFlags == Flags)
 508     // We know that the same flag is specified more than once when the flags
 509     // weren't modified.
 510     return error("duplicate '" + Token.stringValue() + "' register flag");
 511   lex();
 512   return false;
 513 }
 514
 515 bool MIParser::parseSubRegisterIndex(unsigned &SubReg) {
 516   assert(Token.is(MIToken::colon));
 517   lex();
 518   if (Token.isNot(MIToken::Identifier))
 519     return error("expected a subregister index after ':'");
 520   auto Name = Token.stringValue();
 521   SubReg = getSubRegIndex(Name);
 522   if (!SubReg)
 523     return error(Twine("use of unknown subregister index '") + Name + "'");
 524   lex();
 525   return false;
 526 }
 527
 528 bool MIParser::parseRegisterOperand(MachineOperand &Dest, bool IsDef) {
 529   unsigned Reg;
 530   unsigned Flags = IsDef ? RegState::Define : 0;
 531   while (Token.isRegisterFlag()) {
 532     if (parseRegisterFlag(Flags))
 533       return true;
 534   }
 535   if (!Token.isRegister())
 536     return error("expected a register after register flags");
 537   if (parseRegister(Reg))
 538     return true;
 539   lex();
 540   unsigned SubReg = 0;
 541   if (Token.is(MIToken::colon)) {
 542     if (parseSubRegisterIndex(SubReg))
 543       return true;
 544   }
 545   Dest = MachineOperand::CreateReg(
 546       Reg, Flags & RegState::Define, Flags & RegState::Implicit,
 547       Flags & RegState::Kill, Flags & RegState::Dead, Flags & RegState::Undef,
 548       Flags & RegState::EarlyClobber, SubReg, Flags & RegState::Debug);
 549   return false;
 550 }
 551
 552 bool MIParser::parseImmediateOperand(MachineOperand &Dest) {
 553   assert(Token.is(MIToken::IntegerLiteral));
 554   const APSInt &Int = Token.integerValue();
 555   if (Int.getMinSignedBits() > 64)
 556     return error("integer literal is too large to be an immediate operand");
 557   Dest = MachineOperand::CreateImm(Int.getExtValue());
 558   lex();
 559   return false;
 560 }
 561
 562 bool MIParser::parseIRConstant(StringRef::iterator Loc, const Constant *&C) {
 563   auto Source = StringRef(Loc, Token.stringValue().end() - Loc).str();
 564   lex();
 565   SMDiagnostic Err;
 566   C = parseConstantValue(Source.c_str(), Err, *MF.getFunction()->getParent());
 567   if (!C)
 568     return error(Loc + Err.getColumnNo(), Err.getMessage());
 569   return false;
 570 }
 571
 572 bool MIParser::parseTypedImmediateOperand(MachineOperand &Dest) {
 573   assert(Token.is(MIToken::IntegerType));
 574   auto Loc = Token.location();
 575   lex();
 576   if (Token.isNot(MIToken::IntegerLiteral))
 577     return error("expected an integer literal");
 578   const Constant *C = nullptr;
 579   if (parseIRConstant(Loc, C))
 580     return true;
 581   Dest = MachineOperand::CreateCImm(cast<ConstantInt>(C));
 582   return false;
 583 }
 584
 585 bool MIParser::parseFPImmediateOperand(MachineOperand &Dest) {
 586   auto Loc = Token.location();
 587   lex();
 588   if (Token.isNot(MIToken::FloatingPointLiteral))
 589     return error("expected a floating point literal");
 590   const Constant *C = nullptr;
 591   if (parseIRConstant(Loc, C))
 592     return true;
 593   Dest = MachineOperand::CreateFPImm(cast<ConstantFP>(C));
 594   return false;
 595 }
 596
 597 bool MIParser::getUnsigned(unsigned &Result) {
 598   assert(Token.hasIntegerValue() && "Expected a token with an integer value");
 599   const uint64_t Limit = uint64_t(std::numeric_limits<unsigned>::max()) + 1;
 600   uint64_t Val64 = Token.integerValue().getLimitedValue(Limit);
 601   if (Val64 == Limit)
 602     return error("expected 32-bit integer (too large)");
 603   Result = Val64;
 604   return false;
 605 }
 606
 607 bool MIParser::parseMBBReference(MachineBasicBlock *&MBB) {
 608   assert(Token.is(MIToken::MachineBasicBlock));
 609   unsigned Number;
 610   if (getUnsigned(Number))
 611     return true;
 612   auto MBBInfo = PFS.MBBSlots.find(Number);
 613   if (MBBInfo == PFS.MBBSlots.end())
 614     return error(Twine("use of undefined machine basic block #") +
 615                  Twine(Number));
 616   MBB = MBBInfo->second;
 617   if (!Token.stringValue().empty() && Token.stringValue() != MBB->getName())
 618     return error(Twine("the name of machine basic block #") + Twine(Number) +
 619                  " isn't '" + Token.stringValue() + "'");
 620   return false;
 621 }
 622
 623 bool MIParser::parseMBBOperand(MachineOperand &Dest) {
 624   MachineBasicBlock *MBB;
 625   if (parseMBBReference(MBB))
 626     return true;
 627   Dest = MachineOperand::CreateMBB(MBB);
 628   lex();
 629   return false;
 630 }
 631
 632 bool MIParser::parseStackObjectOperand(MachineOperand &Dest) {
 633   assert(Token.is(MIToken::StackObject));
 634   unsigned ID;
 635   if (getUnsigned(ID))
 636     return true;
 637   auto ObjectInfo = PFS.StackObjectSlots.find(ID);
 638   if (ObjectInfo == PFS.StackObjectSlots.end())
 639     return error(Twine("use of undefined stack object '%stack.") + Twine(ID) +
 640                  "'");
 641   StringRef Name;
 642   if (const auto *Alloca =
 643           MF.getFrameInfo()->getObjectAllocation(ObjectInfo->second))
 644     Name = Alloca->getName();
 645   if (!Token.stringValue().empty() && Token.stringValue() != Name)
 646     return error(Twine("the name of the stack object '%stack.") + Twine(ID) +
 647                  "' isn't '" + Token.stringValue() + "'");
 648   lex();
 649   Dest = MachineOperand::CreateFI(ObjectInfo->second);
 650   return false;
 651 }
 652
 653 bool MIParser::parseFixedStackObjectOperand(MachineOperand &Dest) {
 654   assert(Token.is(MIToken::FixedStackObject));
 655   unsigned ID;
 656   if (getUnsigned(ID))
 657     return true;
 658   auto ObjectInfo = PFS.FixedStackObjectSlots.find(ID);
 659   if (ObjectInfo == PFS.FixedStackObjectSlots.end())
 660     return error(Twine("use of undefined fixed stack object '%fixed-stack.") +
 661                  Twine(ID) + "'");
 662   lex();
 663   Dest = MachineOperand::CreateFI(ObjectInfo->second);
 664   return false;
 665 }
 666
 667 bool MIParser::parseGlobalValue(GlobalValue *&GV) {
 668   switch (Token.kind()) {
 669   case MIToken::NamedGlobalValue: {
 670     const Module *M = MF.getFunction()->getParent();
 671     GV = M->getNamedValue(Token.stringValue());
 672     if (!GV)
 673       return error(Twine("use of undefined global value '@") +
 674                    Token.rawStringValue() + "'");
 675     break;
 676   }
 677   case MIToken::GlobalValue: {
 678     unsigned GVIdx;
 679     if (getUnsigned(GVIdx))
 680       return true;
 681     if (GVIdx >= IRSlots.GlobalValues.size())
 682       return error(Twine("use of undefined global value '@") + Twine(GVIdx) +
 683                    "'");
 684     GV = IRSlots.GlobalValues[GVIdx];
 685     break;
 686   }
 687   default:
 688     llvm_unreachable("The current token should be a global value");
 689   }
 690   return false;
 691 }
 692
 693 bool MIParser::parseGlobalAddressOperand(MachineOperand &Dest) {
 694   GlobalValue *GV = nullptr;
 695   if (parseGlobalValue(GV))
 696     return true;
 697   lex();
 698   Dest = MachineOperand::CreateGA(GV, /*Offset=*/0);
 699   // TODO: Parse the target flags.
 700   if (parseOperandsOffset(Dest))
 701     return true;
 702   return false;
 703 }
 704
 705 bool MIParser::parseConstantPoolIndexOperand(MachineOperand &Dest) {
 706   assert(Token.is(MIToken::ConstantPoolItem));
 707   unsigned ID;
 708   if (getUnsigned(ID))
 709     return true;
 710   auto ConstantInfo = PFS.ConstantPoolSlots.find(ID);
 711   if (ConstantInfo == PFS.ConstantPoolSlots.end())
 712     return error("use of undefined constant '%const." + Twine(ID) + "'");
 713   lex();
 714   // TODO: Parse the target flags.
 715   Dest = MachineOperand::CreateCPI(ID, /*Offset=*/0);
 716   if (parseOperandsOffset(Dest))
 717     return true;
 718   return false;
 719 }
 720
 721 bool MIParser::parseJumpTableIndexOperand(MachineOperand &Dest) {
 722   assert(Token.is(MIToken::JumpTableIndex));
 723   unsigned ID;
 724   if (getUnsigned(ID))
 725     return true;
 726   auto JumpTableEntryInfo = PFS.JumpTableSlots.find(ID);
 727   if (JumpTableEntryInfo == PFS.JumpTableSlots.end())
 728     return error("use of undefined jump table '%jump-table." + Twine(ID) + "'");
 729   lex();
 730   // TODO: Parse target flags.
 731   Dest = MachineOperand::CreateJTI(JumpTableEntryInfo->second);
 732   return false;
 733 }
 734
 735 bool MIParser::parseExternalSymbolOperand(MachineOperand &Dest) {
 736   assert(Token.is(MIToken::ExternalSymbol));
 737   const char *Symbol = MF.createExternalSymbolName(Token.stringValue());
 738   lex();
 739   // TODO: Parse the target flags.
 740   Dest = MachineOperand::CreateES(Symbol);
 741   if (parseOperandsOffset(Dest))
 742     return true;
 743   return false;
 744 }
 745
 746 bool MIParser::parseMDNode(MDNode *&Node) {
 747   assert(Token.is(MIToken::exclaim));
 748   auto Loc = Token.location();
 749   lex();
 750   if (Token.isNot(MIToken::IntegerLiteral) || Token.integerValue().isSigned())
 751     return error("expected metadata id after '!'");
 752   unsigned ID;
 753   if (getUnsigned(ID))
 754     return true;
 755   auto NodeInfo = IRSlots.MetadataNodes.find(ID);
 756   if (NodeInfo == IRSlots.MetadataNodes.end())
 757     return error(Loc, "use of undefined metadata '!" + Twine(ID) + "'");
 758   lex();
 759   Node = NodeInfo->second.get();
 760   return false;
 761 }
 762
 763 bool MIParser::parseMetadataOperand(MachineOperand &Dest) {
 764   MDNode *Node = nullptr;
 765   if (parseMDNode(Node))
 766     return true;
 767   Dest = MachineOperand::CreateMetadata(Node);
 768   return false;
 769 }
 770
 771 bool MIParser::parseCFIOffset(int &Offset) {
 772   if (Token.isNot(MIToken::IntegerLiteral))
 773     return error("expected a cfi offset");
 774   if (Token.integerValue().getMinSignedBits() > 32)
 775     return error("expected a 32 bit integer (the cfi offset is too large)");
 776   Offset = (int)Token.integerValue().getExtValue();
 777   lex();
 778   return false;
 779 }
 780
 781 bool MIParser::parseCFIRegister(unsigned &Reg) {
 782   if (Token.isNot(MIToken::NamedRegister))
 783     return error("expected a cfi register");
 784   unsigned LLVMReg;
 785   if (parseRegister(LLVMReg))
 786     return true;
 787   const auto *TRI = MF.getSubtarget().getRegisterInfo();
 788   assert(TRI && "Expected target register info");
 789   int DwarfReg = TRI->getDwarfRegNum(LLVMReg, true);
 790   if (DwarfReg < 0)
 791     return error("invalid DWARF register");
 792   Reg = (unsigned)DwarfReg;
 793   lex();
 794   return false;
 795 }
 796
 797 bool MIParser::parseCFIOperand(MachineOperand &Dest) {
 798   auto Kind = Token.kind();
 799   lex();
 800   auto &MMI = MF.getMMI();
 801   int Offset;
 802   unsigned Reg;
 803   unsigned CFIIndex;
 804   switch (Kind) {
 805   case MIToken::kw_cfi_offset:
 806     if (parseCFIRegister(Reg) || expectAndConsume(MIToken::comma) ||
 807         parseCFIOffset(Offset))
 808       return true;
 809     CFIIndex =
 810         MMI.addFrameInst(MCCFIInstruction::createOffset(nullptr, Reg, Offset));
 811     break;
 812   case MIToken::kw_cfi_def_cfa_register:
 813     if (parseCFIRegister(Reg))
 814       return true;
 815     CFIIndex =
 816         MMI.addFrameInst(MCCFIInstruction::createDefCfaRegister(nullptr, Reg));
 817     break;
 818   case MIToken::kw_cfi_def_cfa_offset:
 819     if (parseCFIOffset(Offset))
 820       return true;
 821     // NB: MCCFIInstruction::createDefCfaOffset negates the offset.
 822     CFIIndex = MMI.addFrameInst(
 823         MCCFIInstruction::createDefCfaOffset(nullptr, -Offset));
 824     break;
 825   case MIToken::kw_cfi_def_cfa:
 826     if (parseCFIRegister(Reg) || expectAndConsume(MIToken::comma) ||
 827         parseCFIOffset(Offset))
 828       return true;
 829     // NB: MCCFIInstruction::createDefCfa negates the offset.
 830     CFIIndex =
 831         MMI.addFrameInst(MCCFIInstruction::createDefCfa(nullptr, Reg, -Offset));
 832     break;
 833   default:
 834     // TODO: Parse the other CFI operands.
 835     llvm_unreachable("The current token should be a cfi operand");
 836   }
 837   Dest = MachineOperand::CreateCFIIndex(CFIIndex);
 838   return false;
 839 }
 840
 841 bool MIParser::parseIRBlock(BasicBlock *&BB, const Function &F) {
 842   switch (Token.kind()) {
 843   case MIToken::NamedIRBlock: {
 844     BB = dyn_cast_or_null<BasicBlock>(
 845         F.getValueSymbolTable().lookup(Token.stringValue()));
 846     if (!BB)
 847       return error(Twine("use of undefined IR block '%ir-block.") +
 848                    Token.rawStringValue() + "'");
 849     break;
 850   }
 851   case MIToken::IRBlock: {
 852     unsigned SlotNumber = 0;
 853     if (getUnsigned(SlotNumber))
 854       return true;
 855     BB = const_cast<BasicBlock *>(getIRBlock(SlotNumber));
 856     if (!BB)
 857       return error(Twine("use of undefined IR block '%ir-block.") +
 858                    Twine(SlotNumber) + "'");
 859     break;
 860   }
 861   default:
 862     llvm_unreachable("The current token should be an IR block reference");
 863   }
 864   return false;
 865 }
 866
 867 bool MIParser::parseBlockAddressOperand(MachineOperand &Dest) {
 868   assert(Token.is(MIToken::kw_blockaddress));
 869   lex();
 870   if (expectAndConsume(MIToken::lparen))
 871     return true;
 872   if (Token.isNot(MIToken::GlobalValue) &&
 873       Token.isNot(MIToken::NamedGlobalValue))
 874     return error("expected a global value");
 875   GlobalValue *GV = nullptr;
 876   if (parseGlobalValue(GV))
 877     return true;
 878   auto *F = dyn_cast<Function>(GV);
 879   if (!F)
 880     return error("expected an IR function reference");
 881   lex();
 882   if (expectAndConsume(MIToken::comma))
 883     return true;
 884   BasicBlock *BB = nullptr;
 885   if (Token.isNot(MIToken::IRBlock) && Token.isNot(MIToken::NamedIRBlock))
 886     return error("expected an IR block reference");
 887   if (parseIRBlock(BB, *F))
 888     return true;
 889   lex();
 890   if (expectAndConsume(MIToken::rparen))
 891     return true;
 892   // TODO: parse the target flags.
 893   Dest = MachineOperand::CreateBA(BlockAddress::get(F, BB), /*Offset=*/0);
 894   if (parseOperandsOffset(Dest))
 895     return true;
 896   return false;
 897 }
 898
 899 bool MIParser::parseTargetIndexOperand(MachineOperand &Dest) {
 900   assert(Token.is(MIToken::kw_target_index));
 901   lex();
 902   if (expectAndConsume(MIToken::lparen))
 903     return true;
 904   if (Token.isNot(MIToken::Identifier))
 905     return error("expected the name of the target index");
 906   int Index = 0;
 907   if (getTargetIndex(Token.stringValue(), Index))
 908     return error("use of undefined target index '" + Token.stringValue() + "'");
 909   lex();
 910   if (expectAndConsume(MIToken::rparen))
 911     return true;
 912   // TODO: Parse the target flags.
 913   Dest = MachineOperand::CreateTargetIndex(unsigned(Index), /*Offset=*/0);
 914   if (parseOperandsOffset(Dest))
 915     return true;
 916   return false;
 917 }
 918
 919 bool MIParser::parseMachineOperand(MachineOperand &Dest) {
 920   switch (Token.kind()) {
 921   case MIToken::kw_implicit:
 922   case MIToken::kw_implicit_define:
 923   case MIToken::kw_dead:
 924   case MIToken::kw_killed:
 925   case MIToken::kw_undef:
 926   case MIToken::kw_early_clobber:
 927   case MIToken::kw_debug_use:
 928   case MIToken::underscore:
 929   case MIToken::NamedRegister:
 930   case MIToken::VirtualRegister:
 931     return parseRegisterOperand(Dest);
 932   case MIToken::IntegerLiteral:
 933     return parseImmediateOperand(Dest);
 934   case MIToken::IntegerType:
 935     return parseTypedImmediateOperand(Dest);
 936   case MIToken::kw_half:
 937   case MIToken::kw_float:
 938   case MIToken::kw_double:
 939   case MIToken::kw_x86_fp80:
 940   case MIToken::kw_fp128:
 941   case MIToken::kw_ppc_fp128:
 942     return parseFPImmediateOperand(Dest);
 943   case MIToken::MachineBasicBlock:
 944     return parseMBBOperand(Dest);
 945   case MIToken::StackObject:
 946     return parseStackObjectOperand(Dest);
 947   case MIToken::FixedStackObject:
 948     return parseFixedStackObjectOperand(Dest);
 949   case MIToken::GlobalValue:
 950   case MIToken::NamedGlobalValue:
 951     return parseGlobalAddressOperand(Dest);
 952   case MIToken::ConstantPoolItem:
 953     return parseConstantPoolIndexOperand(Dest);
 954   case MIToken::JumpTableIndex:
 955     return parseJumpTableIndexOperand(Dest);
 956   case MIToken::ExternalSymbol:
 957     return parseExternalSymbolOperand(Dest);
 958   case MIToken::exclaim:
 959     return parseMetadataOperand(Dest);
 960   case MIToken::kw_cfi_offset:
 961   case MIToken::kw_cfi_def_cfa_register:
 962   case MIToken::kw_cfi_def_cfa_offset:
 963   case MIToken::kw_cfi_def_cfa:
 964     return parseCFIOperand(Dest);
 965   case MIToken::kw_blockaddress:
 966     return parseBlockAddressOperand(Dest);
 967   case MIToken::kw_target_index:
 968     return parseTargetIndexOperand(Dest);
 969   case MIToken::Error:
 970     return true;
 971   case MIToken::Identifier:
 972     if (const auto *RegMask = getRegMask(Token.stringValue())) {
 973       Dest = MachineOperand::CreateRegMask(RegMask);
 974       lex();
 975       break;
 976     }
 977   // fallthrough
 978   default:
 979     // TODO: parse the other machine operands.
 980     return error("expected a machine operand");
 981   }
 982   return false;
 983 }
 984
 985 bool MIParser::parseOperandsOffset(MachineOperand &Op) {
 986   if (Token.isNot(MIToken::plus) && Token.isNot(MIToken::minus))
 987     return false;
 988   StringRef Sign = Token.stringValue();
 989   bool IsNegative = Token.is(MIToken::minus);
 990   lex();
 991   if (Token.isNot(MIToken::IntegerLiteral))
 992     return error("expected an integer literal after '" + Sign + "'");
 993   if (Token.integerValue().getMinSignedBits() > 64)
 994     return error("expected 64-bit integer (too large)");
 995   int64_t Offset = Token.integerValue().getExtValue();
 996   if (IsNegative)
 997     Offset = -Offset;
 998   lex();
 999   Op.setOffset(Offset);
1000   return false;
1001 }
1002
1003 bool MIParser::parseIRValue(Value *&V) {
1004   switch (Token.kind()) {
1005   case MIToken::NamedIRValue: {
1006     V = MF.getFunction()->getValueSymbolTable().lookup(Token.stringValue());
1007     if (!V)
1008       return error(Twine("use of undefined IR value '%ir.") +
1009                    Token.rawStringValue() + "'");
1010     break;
1011   }
1012   // TODO: Parse unnamed IR value references.
1013   default:
1014     llvm_unreachable("The current token should be an IR block reference");
1015   }
1016   return false;
1017 }
1018
1019 bool MIParser::getUint64(uint64_t &Result) {
1020   assert(Token.hasIntegerValue());
1021   if (Token.integerValue().getActiveBits() > 64)
1022     return error("expected 64-bit integer (too large)");
1023   Result = Token.integerValue().getZExtValue();
1024   return false;
1025 }
1026
1027 bool MIParser::parseMemoryOperandFlag(unsigned &Flags) {
1028   switch (Token.kind()) {
1029   case MIToken::kw_volatile:
1030     Flags |= MachineMemOperand::MOVolatile;
1031     break;
1032   // TODO: report an error when we specify the same flag more than once.
1033   // TODO: parse the other memory operand flags.
1034   default:
1035     llvm_unreachable("The current token should be a memory operand flag");
1036   }
1037   lex();
1038   return false;
1039 }
1040
1041 bool MIParser::parseMachineMemoryOperand(MachineMemOperand *&Dest) {
1042   if (expectAndConsume(MIToken::lparen))
1043     return true;
1044   unsigned Flags = 0;
1045   while (Token.isMemoryOperandFlag()) {
1046     if (parseMemoryOperandFlag(Flags))
1047       return true;
1048   }
1049   if (Token.isNot(MIToken::Identifier) ||
1050       (Token.stringValue() != "load" && Token.stringValue() != "store"))
1051     return error("expected 'load' or 'store' memory operation");
1052   if (Token.stringValue() == "load")
1053     Flags |= MachineMemOperand::MOLoad;
1054   else
1055     Flags |= MachineMemOperand::MOStore;
1056   lex();
1057
1058   if (Token.isNot(MIToken::IntegerLiteral))
1059     return error("expected the size integer literal after memory operation");
1060   uint64_t Size;
1061   if (getUint64(Size))
1062     return true;
1063   lex();
1064
1065   const char *Word = Flags & MachineMemOperand::MOLoad ? "from" : "into";
1066   if (Token.isNot(MIToken::Identifier) || Token.stringValue() != Word)
1067     return error(Twine("expected '") + Word + "'");
1068   lex();
1069
1070   // TODO: Parse pseudo source values.
1071   if (Token.isNot(MIToken::NamedIRValue))
1072     return error("expected an IR value reference");
1073   Value *V = nullptr;
1074   if (parseIRValue(V))
1075     return true;
1076   if (!V->getType()->isPointerTy())
1077     return error("expected a pointer IR value");
1078   lex();
1079   // TODO: Parse the base alignment.
1080   // TODO: Parse the attached metadata nodes.
1081   if (expectAndConsume(MIToken::rparen))
1082     return true;
1083
1084   Dest = MF.getMachineMemOperand(MachinePointerInfo(V), Flags, Size, Size);
1085   return false;
1086 }
1087
1088 void MIParser::initNames2InstrOpCodes() {
1089   if (!Names2InstrOpCodes.empty())
1090     return;
1091   const auto *TII = MF.getSubtarget().getInstrInfo();
1092   assert(TII && "Expected target instruction info");
1093   for (unsigned I = 0, E = TII->getNumOpcodes(); I < E; ++I)
1094     Names2InstrOpCodes.insert(std::make_pair(StringRef(TII->getName(I)), I));
1095 }
1096
1097 bool MIParser::parseInstrName(StringRef InstrName, unsigned &OpCode) {
1098   initNames2InstrOpCodes();
1099   auto InstrInfo = Names2InstrOpCodes.find(InstrName);
1100   if (InstrInfo == Names2InstrOpCodes.end())
1101     return true;
1102   OpCode = InstrInfo->getValue();
1103   return false;
1104 }
1105
1106 void MIParser::initNames2Regs() {
1107   if (!Names2Regs.empty())
1108     return;
1109   // The '%noreg' register is the register 0.
1110   Names2Regs.insert(std::make_pair("noreg", 0));
1111   const auto *TRI = MF.getSubtarget().getRegisterInfo();
1112   assert(TRI && "Expected target register info");
1113   for (unsigned I = 0, E = TRI->getNumRegs(); I < E; ++I) {
1114     bool WasInserted =
1115         Names2Regs.insert(std::make_pair(StringRef(TRI->getName(I)).lower(), I))
1116             .second;
1117     (void)WasInserted;
1118     assert(WasInserted && "Expected registers to be unique case-insensitively");
1119   }
1120 }
1121
1122 bool MIParser::getRegisterByName(StringRef RegName, unsigned &Reg) {
1123   initNames2Regs();
1124   auto RegInfo = Names2Regs.find(RegName);
1125   if (RegInfo == Names2Regs.end())
1126     return true;
1127   Reg = RegInfo->getValue();
1128   return false;
1129 }
1130
1131 void MIParser::initNames2RegMasks() {
1132   if (!Names2RegMasks.empty())
1133     return;
1134   const auto *TRI = MF.getSubtarget().getRegisterInfo();
1135   assert(TRI && "Expected target register info");
1136   ArrayRef<const uint32_t *> RegMasks = TRI->getRegMasks();
1137   ArrayRef<const char *> RegMaskNames = TRI->getRegMaskNames();
1138   assert(RegMasks.size() == RegMaskNames.size());
1139   for (size_t I = 0, E = RegMasks.size(); I < E; ++I)
1140     Names2RegMasks.insert(
1141         std::make_pair(StringRef(RegMaskNames[I]).lower(), RegMasks[I]));
1142 }
1143
1144 const uint32_t *MIParser::getRegMask(StringRef Identifier) {
1145   initNames2RegMasks();
1146   auto RegMaskInfo = Names2RegMasks.find(Identifier);
1147   if (RegMaskInfo == Names2RegMasks.end())
1148     return nullptr;
1149   return RegMaskInfo->getValue();
1150 }
1151
1152 void MIParser::initNames2SubRegIndices() {
1153   if (!Names2SubRegIndices.empty())
1154     return;
1155   const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
1156   for (unsigned I = 1, E = TRI->getNumSubRegIndices(); I < E; ++I)
1157     Names2SubRegIndices.insert(
1158         std::make_pair(StringRef(TRI->getSubRegIndexName(I)).lower(), I));
1159 }
1160
1161 unsigned MIParser::getSubRegIndex(StringRef Name) {
1162   initNames2SubRegIndices();
1163   auto SubRegInfo = Names2SubRegIndices.find(Name);
1164   if (SubRegInfo == Names2SubRegIndices.end())
1165     return 0;
1166   return SubRegInfo->getValue();
1167 }
1168
1169 void MIParser::initSlots2BasicBlocks() {
1170   if (!Slots2BasicBlocks.empty())
1171     return;
1172   const auto &F = *MF.getFunction();
1173   ModuleSlotTracker MST(F.getParent());
1174   MST.incorporateFunction(F);
1175   for (auto &BB : F) {
1176     if (BB.hasName())
1177       continue;
1178     int Slot = MST.getLocalSlot(&BB);
1179     if (Slot == -1)
1180       continue;
1181     Slots2BasicBlocks.insert(std::make_pair(unsigned(Slot), &BB));
1182   }
1183 }
1184
1185 const BasicBlock *MIParser::getIRBlock(unsigned Slot) {
1186   initSlots2BasicBlocks();
1187   auto BlockInfo = Slots2BasicBlocks.find(Slot);
1188   if (BlockInfo == Slots2BasicBlocks.end())
1189     return nullptr;
1190   return BlockInfo->second;
1191 }
1192
1193 void MIParser::initNames2TargetIndices() {
1194   if (!Names2TargetIndices.empty())
1195     return;
1196   const auto *TII = MF.getSubtarget().getInstrInfo();
1197   assert(TII && "Expected target instruction info");
1198   auto Indices = TII->getSerializableTargetIndices();
1199   for (const auto &I : Indices)
1200     Names2TargetIndices.insert(std::make_pair(StringRef(I.second), I.first));
1201 }
1202
1203 bool MIParser::getTargetIndex(StringRef Name, int &Index) {
1204   initNames2TargetIndices();
1205   auto IndexInfo = Names2TargetIndices.find(Name);
1206   if (IndexInfo == Names2TargetIndices.end())
1207     return true;
1208   Index = IndexInfo->second;
1209   return false;
1210 }
1211
1212 bool llvm::parseMachineInstr(MachineInstr *&MI, SourceMgr &SM,
1213                              MachineFunction &MF, StringRef Src,
1214                              const PerFunctionMIParsingState &PFS,
1215                              const SlotMapping &IRSlots, SMDiagnostic &Error) {
1216   return MIParser(SM, MF, Error, Src, PFS, IRSlots).parse(MI);
1217 }
1218
1219 bool llvm::parseMBBReference(MachineBasicBlock *&MBB, SourceMgr &SM,
1220                              MachineFunction &MF, StringRef Src,
1221                              const PerFunctionMIParsingState &PFS,
1222                              const SlotMapping &IRSlots, SMDiagnostic &Error) {
1223   return MIParser(SM, MF, Error, Src, PFS, IRSlots).parseStandaloneMBB(MBB);
1224 }
1225
1226 bool llvm::parseNamedRegisterReference(unsigned &Reg, SourceMgr &SM,
1227                                        MachineFunction &MF, StringRef Src,
1228                                        const PerFunctionMIParsingState &PFS,
1229                                        const SlotMapping &IRSlots,
1230                                        SMDiagnostic &Error) {
1231   return MIParser(SM, MF, Error, Src, PFS, IRSlots)
1232       .parseStandaloneNamedRegister(Reg);
1233 }
1234
1235 bool llvm::parseVirtualRegisterReference(unsigned &Reg, SourceMgr &SM,
1236                                          MachineFunction &MF, StringRef Src,
1237                                          const PerFunctionMIParsingState &PFS,
1238                                          const SlotMapping &IRSlots,
1239                                          SMDiagnostic &Error) {
1240   return MIParser(SM, MF, Error, Src, PFS, IRSlots)
1241       .parseStandaloneVirtualRegister(Reg);
1242 }
1243
1244 bool llvm::parseIRBlockReference(const BasicBlock *&BB, SourceMgr &SM,
1245                                  MachineFunction &MF, StringRef Src,
1246                                  const PerFunctionMIParsingState &PFS,
1247                                  const SlotMapping &IRSlots,
1248                                  SMDiagnostic &Error) {
1249   return MIParser(SM, MF, Error, Src, PFS, IRSlots)
1250       .parseStandaloneIRBlockReference(BB);
1251 }