blob: c09f279a25f362cff8f096a63500f7c944486506 [file] [log] [blame]
Alex Lorenz8e0a1b42015-06-22 17:02:30 +00001//===- 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"
Alex Lorenz91370c52015-06-22 20:37:46 +000015#include "MILexer.h"
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000016#include "llvm/ADT/StringMap.h"
Alex Lorenzad156fb2015-07-31 20:49:21 +000017#include "llvm/AsmParser/Parser.h"
Alex Lorenz5d6108e2015-06-26 22:56:48 +000018#include "llvm/AsmParser/SlotMapping.h"
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000019#include "llvm/CodeGen/MachineBasicBlock.h"
20#include "llvm/CodeGen/MachineFunction.h"
Alex Lorenz7feaf7c2015-07-16 23:37:45 +000021#include "llvm/CodeGen/MachineFrameInfo.h"
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000022#include "llvm/CodeGen/MachineInstr.h"
Alex Lorenzcb268d42015-07-06 23:07:26 +000023#include "llvm/CodeGen/MachineInstrBuilder.h"
Alex Lorenz4af7e612015-08-03 23:08:19 +000024#include "llvm/CodeGen/MachineMemOperand.h"
Alex Lorenzf4baeb52015-07-21 22:28:27 +000025#include "llvm/CodeGen/MachineModuleInfo.h"
Alex Lorenz7feaf7c2015-07-16 23:37:45 +000026#include "llvm/IR/Instructions.h"
Alex Lorenzdeb53492015-07-28 17:28:03 +000027#include "llvm/IR/Constants.h"
Alex Lorenz5d6108e2015-06-26 22:56:48 +000028#include "llvm/IR/Module.h"
Alex Lorenz8a1915b2015-07-27 22:42:41 +000029#include "llvm/IR/ModuleSlotTracker.h"
Alex Lorenzdeb53492015-07-28 17:28:03 +000030#include "llvm/IR/ValueSymbolTable.h"
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000031#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
36using namespace llvm;
37
38namespace {
39
Alex Lorenz36962cd2015-07-07 02:08:46 +000040/// A wrapper struct around the 'MachineOperand' struct that includes a source
Alex Lorenzfeb6b432015-08-19 19:19:16 +000041/// range and other attributes.
42struct ParsedMachineOperand {
Alex Lorenz36962cd2015-07-07 02:08:46 +000043 MachineOperand Operand;
44 StringRef::iterator Begin;
45 StringRef::iterator End;
Alex Lorenz5ef93b02015-08-19 19:05:34 +000046 Optional<unsigned> TiedDefIdx;
Alex Lorenz36962cd2015-07-07 02:08:46 +000047
Alex Lorenzfeb6b432015-08-19 19:19:16 +000048 ParsedMachineOperand(const MachineOperand &Operand, StringRef::iterator Begin,
49 StringRef::iterator End, Optional<unsigned> &TiedDefIdx)
Alex Lorenz5ef93b02015-08-19 19:05:34 +000050 : Operand(Operand), Begin(Begin), End(End), TiedDefIdx(TiedDefIdx) {
51 if (TiedDefIdx)
52 assert(Operand.isReg() && Operand.isUse() &&
53 "Only used register operands can be tied");
54 }
Alex Lorenz36962cd2015-07-07 02:08:46 +000055};
56
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000057class MIParser {
58 SourceMgr &SM;
59 MachineFunction &MF;
60 SMDiagnostic &Error;
Alex Lorenz91370c52015-06-22 20:37:46 +000061 StringRef Source, CurrentSource;
62 MIToken Token;
Alex Lorenz7a503fa2015-07-07 17:46:43 +000063 const PerFunctionMIParsingState &PFS;
Alex Lorenz5d6108e2015-06-26 22:56:48 +000064 /// Maps from indices to unnamed global values and metadata nodes.
65 const SlotMapping &IRSlots;
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000066 /// Maps from instruction names to op codes.
67 StringMap<unsigned> Names2InstrOpCodes;
Alex Lorenzf3db51de2015-06-23 16:35:26 +000068 /// Maps from register names to registers.
69 StringMap<unsigned> Names2Regs;
Alex Lorenz8f6f4282015-06-29 16:57:06 +000070 /// Maps from register mask names to register masks.
71 StringMap<const uint32_t *> Names2RegMasks;
Alex Lorenz2eacca82015-07-13 23:24:34 +000072 /// Maps from subregister names to subregister indices.
73 StringMap<unsigned> Names2SubRegIndices;
Alex Lorenz8a1915b2015-07-27 22:42:41 +000074 /// Maps from slot numbers to function's unnamed basic blocks.
75 DenseMap<unsigned, const BasicBlock *> Slots2BasicBlocks;
Alex Lorenzdd13be02015-08-19 23:31:05 +000076 /// Maps from slot numbers to function's unnamed values.
77 DenseMap<unsigned, const Value *> Slots2Values;
Alex Lorenzef5c1962015-07-28 23:02:45 +000078 /// Maps from target index names to target indices.
79 StringMap<int> Names2TargetIndices;
Alex Lorenz49873a82015-08-06 00:44:07 +000080 /// Maps from direct target flag names to the direct target flag values.
81 StringMap<unsigned> Names2DirectTargetFlags;
Alex Lorenzf3630112015-08-18 22:52:15 +000082 /// Maps from direct target flag names to the bitmask target flag values.
83 StringMap<unsigned> Names2BitmaskTargetFlags;
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000084
85public:
86 MIParser(SourceMgr &SM, MachineFunction &MF, SMDiagnostic &Error,
Alex Lorenz7a503fa2015-07-07 17:46:43 +000087 StringRef Source, const PerFunctionMIParsingState &PFS,
Alex Lorenz5d6108e2015-06-26 22:56:48 +000088 const SlotMapping &IRSlots);
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000089
Alex Lorenz91370c52015-06-22 20:37:46 +000090 void lex();
91
Alex Lorenz8e0a1b42015-06-22 17:02:30 +000092 /// Report an error at the current location with the given message.
93 ///
94 /// This function always return true.
95 bool error(const Twine &Msg);
96
Alex Lorenz91370c52015-06-22 20:37:46 +000097 /// Report an error at the given location with the given message.
98 ///
99 /// This function always return true.
100 bool error(StringRef::iterator Loc, const Twine &Msg);
101
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000102 bool
103 parseBasicBlockDefinitions(DenseMap<unsigned, MachineBasicBlock *> &MBBSlots);
104 bool parseBasicBlocks();
Alex Lorenz3708a642015-06-30 17:47:50 +0000105 bool parse(MachineInstr *&MI);
Alex Lorenz1ea60892015-07-27 20:29:27 +0000106 bool parseStandaloneMBB(MachineBasicBlock *&MBB);
107 bool parseStandaloneNamedRegister(unsigned &Reg);
Alex Lorenz12045a42015-07-27 17:42:45 +0000108 bool parseStandaloneVirtualRegister(unsigned &Reg);
Alex Lorenza314d812015-08-18 22:26:26 +0000109 bool parseStandaloneStackObject(int &FI);
Alex Lorenzdf9e3c62015-08-19 00:13:25 +0000110 bool parseStandaloneMDNode(MDNode *&Node);
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000111
112 bool
113 parseBasicBlockDefinition(DenseMap<unsigned, MachineBasicBlock *> &MBBSlots);
114 bool parseBasicBlock(MachineBasicBlock &MBB);
115 bool parseBasicBlockLiveins(MachineBasicBlock &MBB);
116 bool parseBasicBlockSuccessors(MachineBasicBlock &MBB);
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000117
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000118 bool parseRegister(unsigned &Reg);
Alex Lorenzcb268d42015-07-06 23:07:26 +0000119 bool parseRegisterFlag(unsigned &Flags);
Alex Lorenz2eacca82015-07-13 23:24:34 +0000120 bool parseSubRegisterIndex(unsigned &SubReg);
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000121 bool parseRegisterTiedDefIndex(unsigned &TiedDefIdx);
122 bool parseRegisterOperand(MachineOperand &Dest,
123 Optional<unsigned> &TiedDefIdx, bool IsDef = false);
Alex Lorenz240fc1e2015-06-23 23:42:28 +0000124 bool parseImmediateOperand(MachineOperand &Dest);
Alex Lorenz5d8b0bd2015-08-21 21:48:22 +0000125 bool parseIRConstant(StringRef::iterator Loc, StringRef Source,
126 const Constant *&C);
Alex Lorenz7eaff4c2015-08-05 18:44:00 +0000127 bool parseIRConstant(StringRef::iterator Loc, const Constant *&C);
Alex Lorenz05e38822015-08-05 18:52:21 +0000128 bool parseTypedImmediateOperand(MachineOperand &Dest);
Alex Lorenzad156fb2015-07-31 20:49:21 +0000129 bool parseFPImmediateOperand(MachineOperand &Dest);
Alex Lorenzf09df002015-06-30 18:16:42 +0000130 bool parseMBBReference(MachineBasicBlock *&MBB);
Alex Lorenz33f0aef2015-06-26 16:46:11 +0000131 bool parseMBBOperand(MachineOperand &Dest);
Alex Lorenzdc9dadf2015-08-18 22:18:52 +0000132 bool parseStackFrameIndex(int &FI);
Alex Lorenz7feaf7c2015-07-16 23:37:45 +0000133 bool parseStackObjectOperand(MachineOperand &Dest);
Alex Lorenzea882122015-08-12 21:17:02 +0000134 bool parseFixedStackFrameIndex(int &FI);
Alex Lorenz7feaf7c2015-07-16 23:37:45 +0000135 bool parseFixedStackObjectOperand(MachineOperand &Dest);
Alex Lorenz41df7d32015-07-28 17:09:52 +0000136 bool parseGlobalValue(GlobalValue *&GV);
Alex Lorenz5d6108e2015-06-26 22:56:48 +0000137 bool parseGlobalAddressOperand(MachineOperand &Dest);
Alex Lorenzab980492015-07-20 20:51:18 +0000138 bool parseConstantPoolIndexOperand(MachineOperand &Dest);
Alex Lorenz31d70682015-07-15 23:38:35 +0000139 bool parseJumpTableIndexOperand(MachineOperand &Dest);
Alex Lorenz6ede3742015-07-21 16:59:53 +0000140 bool parseExternalSymbolOperand(MachineOperand &Dest);
Alex Lorenz44f29252015-07-22 21:07:04 +0000141 bool parseMDNode(MDNode *&Node);
Alex Lorenz35e44462015-07-22 17:58:46 +0000142 bool parseMetadataOperand(MachineOperand &Dest);
Alex Lorenzf4baeb52015-07-21 22:28:27 +0000143 bool parseCFIOffset(int &Offset);
Alex Lorenz8cfc6862015-07-23 23:09:07 +0000144 bool parseCFIRegister(unsigned &Reg);
Alex Lorenzf4baeb52015-07-21 22:28:27 +0000145 bool parseCFIOperand(MachineOperand &Dest);
Alex Lorenzdeb53492015-07-28 17:28:03 +0000146 bool parseIRBlock(BasicBlock *&BB, const Function &F);
147 bool parseBlockAddressOperand(MachineOperand &Dest);
Alex Lorenzef5c1962015-07-28 23:02:45 +0000148 bool parseTargetIndexOperand(MachineOperand &Dest);
Alex Lorenzb97c9ef2015-08-10 23:24:42 +0000149 bool parseLiveoutRegisterMaskOperand(MachineOperand &Dest);
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000150 bool parseMachineOperand(MachineOperand &Dest,
151 Optional<unsigned> &TiedDefIdx);
152 bool parseMachineOperandAndTargetFlags(MachineOperand &Dest,
153 Optional<unsigned> &TiedDefIdx);
Alex Lorenzdc24c172015-08-07 20:21:00 +0000154 bool parseOffset(int64_t &Offset);
Alex Lorenz620f8912015-08-13 20:33:33 +0000155 bool parseAlignment(unsigned &Alignment);
Alex Lorenz5672a892015-08-05 22:26:15 +0000156 bool parseOperandsOffset(MachineOperand &Op);
Alex Lorenz36593ac2015-08-19 23:27:07 +0000157 bool parseIRValue(const Value *&V);
Alex Lorenza518b792015-08-04 00:24:45 +0000158 bool parseMemoryOperandFlag(unsigned &Flags);
Alex Lorenz91097a32015-08-12 20:33:26 +0000159 bool parseMemoryPseudoSourceValue(const PseudoSourceValue *&PSV);
160 bool parseMachinePointerInfo(MachinePointerInfo &Dest);
Alex Lorenz4af7e612015-08-03 23:08:19 +0000161 bool parseMachineMemoryOperand(MachineMemOperand *&Dest);
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000162
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000163private:
Alex Lorenz33f0aef2015-06-26 16:46:11 +0000164 /// Convert the integer literal in the current token into an unsigned integer.
165 ///
166 /// Return true if an error occurred.
167 bool getUnsigned(unsigned &Result);
168
Alex Lorenz4af7e612015-08-03 23:08:19 +0000169 /// Convert the integer literal in the current token into an uint64.
170 ///
171 /// Return true if an error occurred.
172 bool getUint64(uint64_t &Result);
173
Alex Lorenz8cfc6862015-07-23 23:09:07 +0000174 /// If the current token is of the given kind, consume it and return false.
175 /// Otherwise report an error and return true.
176 bool expectAndConsume(MIToken::TokenKind TokenKind);
177
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000178 /// If the current token is of the given kind, consume it and return true.
179 /// Otherwise return false.
180 bool consumeIfPresent(MIToken::TokenKind TokenKind);
181
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000182 void initNames2InstrOpCodes();
183
184 /// Try to convert an instruction name to an opcode. Return true if the
185 /// instruction name is invalid.
186 bool parseInstrName(StringRef InstrName, unsigned &OpCode);
Alex Lorenz91370c52015-06-22 20:37:46 +0000187
Alex Lorenze5a44662015-07-17 00:24:15 +0000188 bool parseInstruction(unsigned &OpCode, unsigned &Flags);
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000189
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000190 bool assignRegisterTies(MachineInstr &MI,
Alex Lorenzfeb6b432015-08-19 19:19:16 +0000191 ArrayRef<ParsedMachineOperand> Operands);
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000192
Alex Lorenzfeb6b432015-08-19 19:19:16 +0000193 bool verifyImplicitOperands(ArrayRef<ParsedMachineOperand> Operands,
Alex Lorenz36962cd2015-07-07 02:08:46 +0000194 const MCInstrDesc &MCID);
195
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000196 void initNames2Regs();
197
198 /// Try to convert a register name to a register number. Return true if the
199 /// register name is invalid.
200 bool getRegisterByName(StringRef RegName, unsigned &Reg);
Alex Lorenz8f6f4282015-06-29 16:57:06 +0000201
202 void initNames2RegMasks();
203
204 /// Check if the given identifier is a name of a register mask.
205 ///
206 /// Return null if the identifier isn't a register mask.
207 const uint32_t *getRegMask(StringRef Identifier);
Alex Lorenz2eacca82015-07-13 23:24:34 +0000208
209 void initNames2SubRegIndices();
210
211 /// Check if the given identifier is a name of a subregister index.
212 ///
213 /// Return 0 if the name isn't a subregister index class.
214 unsigned getSubRegIndex(StringRef Name);
Alex Lorenz8a1915b2015-07-27 22:42:41 +0000215
Alex Lorenz8a1915b2015-07-27 22:42:41 +0000216 const BasicBlock *getIRBlock(unsigned Slot);
Alex Lorenzcba8c5f2015-08-06 23:57:04 +0000217 const BasicBlock *getIRBlock(unsigned Slot, const Function &F);
Alex Lorenzef5c1962015-07-28 23:02:45 +0000218
Alex Lorenzdd13be02015-08-19 23:31:05 +0000219 const Value *getIRValue(unsigned Slot);
220
Alex Lorenzef5c1962015-07-28 23:02:45 +0000221 void initNames2TargetIndices();
222
223 /// Try to convert a name of target index to the corresponding target index.
224 ///
225 /// Return true if the name isn't a name of a target index.
226 bool getTargetIndex(StringRef Name, int &Index);
Alex Lorenz49873a82015-08-06 00:44:07 +0000227
228 void initNames2DirectTargetFlags();
229
230 /// Try to convert a name of a direct target flag to the corresponding
231 /// target flag.
232 ///
233 /// Return true if the name isn't a name of a direct flag.
234 bool getDirectTargetFlag(StringRef Name, unsigned &Flag);
Alex Lorenzf3630112015-08-18 22:52:15 +0000235
236 void initNames2BitmaskTargetFlags();
237
238 /// Try to convert a name of a bitmask target flag to the corresponding
239 /// target flag.
240 ///
241 /// Return true if the name isn't a name of a bitmask target flag.
242 bool getBitmaskTargetFlag(StringRef Name, unsigned &Flag);
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000243};
244
245} // end anonymous namespace
246
247MIParser::MIParser(SourceMgr &SM, MachineFunction &MF, SMDiagnostic &Error,
Alex Lorenz7a503fa2015-07-07 17:46:43 +0000248 StringRef Source, const PerFunctionMIParsingState &PFS,
Alex Lorenz5d6108e2015-06-26 22:56:48 +0000249 const SlotMapping &IRSlots)
Alex Lorenz91370c52015-06-22 20:37:46 +0000250 : SM(SM), MF(MF), Error(Error), Source(Source), CurrentSource(Source),
Alex Lorenz3fb77682015-08-06 23:17:42 +0000251 PFS(PFS), IRSlots(IRSlots) {}
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000252
Alex Lorenz91370c52015-06-22 20:37:46 +0000253void MIParser::lex() {
254 CurrentSource = lexMIToken(
255 CurrentSource, Token,
256 [this](StringRef::iterator Loc, const Twine &Msg) { error(Loc, Msg); });
257}
258
259bool MIParser::error(const Twine &Msg) { return error(Token.location(), Msg); }
260
261bool MIParser::error(StringRef::iterator Loc, const Twine &Msg) {
Alex Lorenz91370c52015-06-22 20:37:46 +0000262 assert(Loc >= Source.data() && Loc <= (Source.data() + Source.size()));
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000263 const MemoryBuffer &Buffer = *SM.getMemoryBuffer(SM.getMainFileID());
264 if (Loc >= Buffer.getBufferStart() && Loc <= Buffer.getBufferEnd()) {
265 // Create an ordinary diagnostic when the source manager's buffer is the
266 // source string.
267 Error = SM.GetMessage(SMLoc::getFromPointer(Loc), SourceMgr::DK_Error, Msg);
268 return true;
269 }
270 // Create a diagnostic for a YAML string literal.
271 Error = SMDiagnostic(SM, SMLoc(), Buffer.getBufferIdentifier(), 1,
272 Loc - Source.data(), SourceMgr::DK_Error, Msg.str(),
273 Source, None, None);
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000274 return true;
275}
276
Alex Lorenz8cfc6862015-07-23 23:09:07 +0000277static const char *toString(MIToken::TokenKind TokenKind) {
278 switch (TokenKind) {
279 case MIToken::comma:
280 return "','";
Alex Lorenzfbe9c042015-07-29 18:51:21 +0000281 case MIToken::equal:
282 return "'='";
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000283 case MIToken::colon:
284 return "':'";
Alex Lorenzdeb53492015-07-28 17:28:03 +0000285 case MIToken::lparen:
286 return "'('";
287 case MIToken::rparen:
288 return "')'";
Alex Lorenz8cfc6862015-07-23 23:09:07 +0000289 default:
290 return "<unknown token>";
291 }
292}
293
294bool MIParser::expectAndConsume(MIToken::TokenKind TokenKind) {
295 if (Token.isNot(TokenKind))
296 return error(Twine("expected ") + toString(TokenKind));
297 lex();
298 return false;
299}
300
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000301bool MIParser::consumeIfPresent(MIToken::TokenKind TokenKind) {
302 if (Token.isNot(TokenKind))
303 return false;
Alex Lorenz91370c52015-06-22 20:37:46 +0000304 lex();
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000305 return true;
306}
Alex Lorenz91370c52015-06-22 20:37:46 +0000307
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000308bool MIParser::parseBasicBlockDefinition(
309 DenseMap<unsigned, MachineBasicBlock *> &MBBSlots) {
310 assert(Token.is(MIToken::MachineBasicBlockLabel));
311 unsigned ID = 0;
312 if (getUnsigned(ID))
313 return true;
314 auto Loc = Token.location();
315 auto Name = Token.stringValue();
316 lex();
317 bool HasAddressTaken = false;
318 bool IsLandingPad = false;
319 unsigned Alignment = 0;
320 BasicBlock *BB = nullptr;
321 if (consumeIfPresent(MIToken::lparen)) {
322 do {
323 // TODO: Report an error when multiple same attributes are specified.
324 switch (Token.kind()) {
325 case MIToken::kw_address_taken:
326 HasAddressTaken = true;
327 lex();
328 break;
329 case MIToken::kw_landing_pad:
330 IsLandingPad = true;
331 lex();
332 break;
333 case MIToken::kw_align:
334 if (parseAlignment(Alignment))
335 return true;
336 break;
337 case MIToken::IRBlock:
338 // TODO: Report an error when both name and ir block are specified.
339 if (parseIRBlock(BB, *MF.getFunction()))
340 return true;
341 lex();
342 break;
343 default:
344 break;
345 }
346 } while (consumeIfPresent(MIToken::comma));
347 if (expectAndConsume(MIToken::rparen))
348 return true;
349 }
350 if (expectAndConsume(MIToken::colon))
351 return true;
352
353 if (!Name.empty()) {
354 BB = dyn_cast_or_null<BasicBlock>(
355 MF.getFunction()->getValueSymbolTable().lookup(Name));
356 if (!BB)
357 return error(Loc, Twine("basic block '") + Name +
358 "' is not defined in the function '" +
359 MF.getName() + "'");
360 }
361 auto *MBB = MF.CreateMachineBasicBlock(BB);
362 MF.insert(MF.end(), MBB);
363 bool WasInserted = MBBSlots.insert(std::make_pair(ID, MBB)).second;
364 if (!WasInserted)
365 return error(Loc, Twine("redefinition of machine basic block with id #") +
366 Twine(ID));
367 if (Alignment)
368 MBB->setAlignment(Alignment);
369 if (HasAddressTaken)
370 MBB->setHasAddressTaken();
Reid Kleckner0e288232015-08-27 23:27:47 +0000371 MBB->setIsEHPad(IsLandingPad);
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000372 return false;
373}
374
375bool MIParser::parseBasicBlockDefinitions(
376 DenseMap<unsigned, MachineBasicBlock *> &MBBSlots) {
377 lex();
378 // Skip until the first machine basic block.
379 while (Token.is(MIToken::Newline))
380 lex();
381 if (Token.isErrorOrEOF())
382 return Token.isError();
383 if (Token.isNot(MIToken::MachineBasicBlockLabel))
384 return error("expected a basic block definition before instructions");
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000385 unsigned BraceDepth = 0;
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000386 do {
387 if (parseBasicBlockDefinition(MBBSlots))
388 return true;
389 bool IsAfterNewline = false;
390 // Skip until the next machine basic block.
391 while (true) {
392 if ((Token.is(MIToken::MachineBasicBlockLabel) && IsAfterNewline) ||
393 Token.isErrorOrEOF())
394 break;
395 else if (Token.is(MIToken::MachineBasicBlockLabel))
396 return error("basic block definition should be located at the start of "
397 "the line");
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000398 else if (consumeIfPresent(MIToken::Newline)) {
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000399 IsAfterNewline = true;
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000400 continue;
401 }
402 IsAfterNewline = false;
403 if (Token.is(MIToken::lbrace))
404 ++BraceDepth;
405 if (Token.is(MIToken::rbrace)) {
406 if (!BraceDepth)
407 return error("extraneous closing brace ('}')");
408 --BraceDepth;
409 }
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000410 lex();
411 }
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000412 // Verify that we closed all of the '{' at the end of a file or a block.
413 if (!Token.isError() && BraceDepth)
414 return error("expected '}'"); // FIXME: Report a note that shows '{'.
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000415 } while (!Token.isErrorOrEOF());
416 return Token.isError();
417}
418
419bool MIParser::parseBasicBlockLiveins(MachineBasicBlock &MBB) {
420 assert(Token.is(MIToken::kw_liveins));
421 lex();
422 if (expectAndConsume(MIToken::colon))
423 return true;
424 if (Token.isNewlineOrEOF()) // Allow an empty list of liveins.
425 return false;
426 do {
427 if (Token.isNot(MIToken::NamedRegister))
428 return error("expected a named register");
429 unsigned Reg = 0;
430 if (parseRegister(Reg))
431 return true;
432 MBB.addLiveIn(Reg);
433 lex();
434 } while (consumeIfPresent(MIToken::comma));
435 return false;
436}
437
438bool MIParser::parseBasicBlockSuccessors(MachineBasicBlock &MBB) {
439 assert(Token.is(MIToken::kw_successors));
440 lex();
441 if (expectAndConsume(MIToken::colon))
442 return true;
443 if (Token.isNewlineOrEOF()) // Allow an empty list of successors.
444 return false;
445 do {
446 if (Token.isNot(MIToken::MachineBasicBlock))
447 return error("expected a machine basic block reference");
448 MachineBasicBlock *SuccMBB = nullptr;
449 if (parseMBBReference(SuccMBB))
450 return true;
451 lex();
452 unsigned Weight = 0;
453 if (consumeIfPresent(MIToken::lparen)) {
454 if (Token.isNot(MIToken::IntegerLiteral))
455 return error("expected an integer literal after '('");
456 if (getUnsigned(Weight))
457 return true;
458 lex();
459 if (expectAndConsume(MIToken::rparen))
460 return true;
461 }
462 MBB.addSuccessor(SuccMBB, Weight);
463 } while (consumeIfPresent(MIToken::comma));
464 return false;
465}
466
467bool MIParser::parseBasicBlock(MachineBasicBlock &MBB) {
468 // Skip the definition.
469 assert(Token.is(MIToken::MachineBasicBlockLabel));
470 lex();
471 if (consumeIfPresent(MIToken::lparen)) {
472 while (Token.isNot(MIToken::rparen) && !Token.isErrorOrEOF())
473 lex();
474 consumeIfPresent(MIToken::rparen);
475 }
476 consumeIfPresent(MIToken::colon);
477
478 // Parse the liveins and successors.
479 // N.B: Multiple lists of successors and liveins are allowed and they're
480 // merged into one.
481 // Example:
482 // liveins: %edi
483 // liveins: %esi
484 //
485 // is equivalent to
486 // liveins: %edi, %esi
487 while (true) {
488 if (Token.is(MIToken::kw_successors)) {
489 if (parseBasicBlockSuccessors(MBB))
490 return true;
491 } else if (Token.is(MIToken::kw_liveins)) {
492 if (parseBasicBlockLiveins(MBB))
493 return true;
494 } else if (consumeIfPresent(MIToken::Newline)) {
495 continue;
496 } else
497 break;
498 if (!Token.isNewlineOrEOF())
499 return error("expected line break at the end of a list");
500 lex();
501 }
502
503 // Parse the instructions.
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000504 bool IsInBundle = false;
505 MachineInstr *PrevMI = nullptr;
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000506 while (true) {
507 if (Token.is(MIToken::MachineBasicBlockLabel) || Token.is(MIToken::Eof))
508 return false;
509 else if (consumeIfPresent(MIToken::Newline))
510 continue;
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000511 if (consumeIfPresent(MIToken::rbrace)) {
512 // The first parsing pass should verify that all closing '}' have an
513 // opening '{'.
514 assert(IsInBundle);
515 IsInBundle = false;
516 continue;
517 }
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000518 MachineInstr *MI = nullptr;
519 if (parse(MI))
520 return true;
521 MBB.insert(MBB.end(), MI);
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000522 if (IsInBundle) {
523 PrevMI->setFlag(MachineInstr::BundledSucc);
524 MI->setFlag(MachineInstr::BundledPred);
525 }
526 PrevMI = MI;
527 if (Token.is(MIToken::lbrace)) {
528 if (IsInBundle)
529 return error("nested instruction bundles are not allowed");
530 lex();
531 // This instruction is the start of the bundle.
532 MI->setFlag(MachineInstr::BundledSucc);
533 IsInBundle = true;
534 if (!Token.is(MIToken::Newline))
535 // The next instruction can be on the same line.
536 continue;
537 }
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000538 assert(Token.isNewlineOrEOF() && "MI is not fully parsed");
539 lex();
540 }
541 return false;
542}
543
544bool MIParser::parseBasicBlocks() {
545 lex();
546 // Skip until the first machine basic block.
547 while (Token.is(MIToken::Newline))
548 lex();
549 if (Token.isErrorOrEOF())
550 return Token.isError();
551 // The first parsing pass should have verified that this token is a MBB label
552 // in the 'parseBasicBlockDefinitions' method.
553 assert(Token.is(MIToken::MachineBasicBlockLabel));
554 do {
555 MachineBasicBlock *MBB = nullptr;
556 if (parseMBBReference(MBB))
557 return true;
558 if (parseBasicBlock(*MBB))
559 return true;
560 // The method 'parseBasicBlock' should parse the whole block until the next
561 // block or the end of file.
562 assert(Token.is(MIToken::MachineBasicBlockLabel) || Token.is(MIToken::Eof));
563 } while (Token.isNot(MIToken::Eof));
564 return false;
565}
566
567bool MIParser::parse(MachineInstr *&MI) {
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000568 // Parse any register operands before '='
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000569 MachineOperand MO = MachineOperand::CreateImm(0);
Alex Lorenzfeb6b432015-08-19 19:19:16 +0000570 SmallVector<ParsedMachineOperand, 8> Operands;
Alex Lorenzfbe9c042015-07-29 18:51:21 +0000571 while (Token.isRegister() || Token.isRegisterFlag()) {
Alex Lorenz36962cd2015-07-07 02:08:46 +0000572 auto Loc = Token.location();
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000573 Optional<unsigned> TiedDefIdx;
574 if (parseRegisterOperand(MO, TiedDefIdx, /*IsDef=*/true))
Alex Lorenz3708a642015-06-30 17:47:50 +0000575 return true;
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000576 Operands.push_back(
Alex Lorenzfeb6b432015-08-19 19:19:16 +0000577 ParsedMachineOperand(MO, Loc, Token.location(), TiedDefIdx));
Alex Lorenzfbe9c042015-07-29 18:51:21 +0000578 if (Token.isNot(MIToken::comma))
579 break;
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000580 lex();
581 }
Alex Lorenzfbe9c042015-07-29 18:51:21 +0000582 if (!Operands.empty() && expectAndConsume(MIToken::equal))
583 return true;
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000584
Alex Lorenze5a44662015-07-17 00:24:15 +0000585 unsigned OpCode, Flags = 0;
586 if (Token.isError() || parseInstruction(OpCode, Flags))
Alex Lorenz3708a642015-06-30 17:47:50 +0000587 return true;
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000588
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000589 // Parse the remaining machine operands.
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000590 while (!Token.isNewlineOrEOF() && Token.isNot(MIToken::kw_debug_location) &&
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000591 Token.isNot(MIToken::coloncolon) && Token.isNot(MIToken::lbrace)) {
Alex Lorenz36962cd2015-07-07 02:08:46 +0000592 auto Loc = Token.location();
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000593 Optional<unsigned> TiedDefIdx;
594 if (parseMachineOperandAndTargetFlags(MO, TiedDefIdx))
Alex Lorenz3708a642015-06-30 17:47:50 +0000595 return true;
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000596 Operands.push_back(
Alex Lorenzfeb6b432015-08-19 19:19:16 +0000597 ParsedMachineOperand(MO, Loc, Token.location(), TiedDefIdx));
Alex Lorenzf9a2b122015-08-14 18:57:24 +0000598 if (Token.isNewlineOrEOF() || Token.is(MIToken::coloncolon) ||
599 Token.is(MIToken::lbrace))
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000600 break;
Alex Lorenz3708a642015-06-30 17:47:50 +0000601 if (Token.isNot(MIToken::comma))
602 return error("expected ',' before the next machine operand");
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000603 lex();
604 }
605
Alex Lorenz46d760d2015-07-22 21:15:11 +0000606 DebugLoc DebugLocation;
607 if (Token.is(MIToken::kw_debug_location)) {
608 lex();
609 if (Token.isNot(MIToken::exclaim))
610 return error("expected a metadata node after 'debug-location'");
611 MDNode *Node = nullptr;
612 if (parseMDNode(Node))
613 return true;
614 DebugLocation = DebugLoc(Node);
615 }
616
Alex Lorenz4af7e612015-08-03 23:08:19 +0000617 // Parse the machine memory operands.
618 SmallVector<MachineMemOperand *, 2> MemOperands;
619 if (Token.is(MIToken::coloncolon)) {
620 lex();
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000621 while (!Token.isNewlineOrEOF()) {
Alex Lorenz4af7e612015-08-03 23:08:19 +0000622 MachineMemOperand *MemOp = nullptr;
623 if (parseMachineMemoryOperand(MemOp))
624 return true;
625 MemOperands.push_back(MemOp);
Alex Lorenz5022f6b2015-08-13 23:10:16 +0000626 if (Token.isNewlineOrEOF())
Alex Lorenz4af7e612015-08-03 23:08:19 +0000627 break;
628 if (Token.isNot(MIToken::comma))
629 return error("expected ',' before the next machine memory operand");
630 lex();
631 }
632 }
633
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000634 const auto &MCID = MF.getSubtarget().getInstrInfo()->get(OpCode);
Alex Lorenz36962cd2015-07-07 02:08:46 +0000635 if (!MCID.isVariadic()) {
636 // FIXME: Move the implicit operand verification to the machine verifier.
637 if (verifyImplicitOperands(Operands, MCID))
638 return true;
639 }
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000640
Alex Lorenzcb268d42015-07-06 23:07:26 +0000641 // TODO: Check for extraneous machine operands.
Alex Lorenz46d760d2015-07-22 21:15:11 +0000642 MI = MF.CreateMachineInstr(MCID, DebugLocation, /*NoImplicit=*/true);
Alex Lorenze5a44662015-07-17 00:24:15 +0000643 MI->setFlags(Flags);
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000644 for (const auto &Operand : Operands)
Alex Lorenz36962cd2015-07-07 02:08:46 +0000645 MI->addOperand(MF, Operand.Operand);
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000646 if (assignRegisterTies(*MI, Operands))
647 return true;
Alex Lorenz4af7e612015-08-03 23:08:19 +0000648 if (MemOperands.empty())
649 return false;
650 MachineInstr::mmo_iterator MemRefs =
651 MF.allocateMemRefsArray(MemOperands.size());
652 std::copy(MemOperands.begin(), MemOperands.end(), MemRefs);
653 MI->setMemRefs(MemRefs, MemRefs + MemOperands.size());
Alex Lorenz3708a642015-06-30 17:47:50 +0000654 return false;
Alex Lorenz8e0a1b42015-06-22 17:02:30 +0000655}
656
Alex Lorenz1ea60892015-07-27 20:29:27 +0000657bool MIParser::parseStandaloneMBB(MachineBasicBlock *&MBB) {
Alex Lorenzf09df002015-06-30 18:16:42 +0000658 lex();
659 if (Token.isNot(MIToken::MachineBasicBlock))
660 return error("expected a machine basic block reference");
661 if (parseMBBReference(MBB))
662 return true;
663 lex();
664 if (Token.isNot(MIToken::Eof))
665 return error(
666 "expected end of string after the machine basic block reference");
667 return false;
668}
669
Alex Lorenz1ea60892015-07-27 20:29:27 +0000670bool MIParser::parseStandaloneNamedRegister(unsigned &Reg) {
Alex Lorenz9fab3702015-07-14 21:24:41 +0000671 lex();
672 if (Token.isNot(MIToken::NamedRegister))
673 return error("expected a named register");
674 if (parseRegister(Reg))
Alex Lorenz607efb62015-08-18 22:57:36 +0000675 return true;
Alex Lorenz9fab3702015-07-14 21:24:41 +0000676 lex();
677 if (Token.isNot(MIToken::Eof))
678 return error("expected end of string after the register reference");
679 return false;
680}
681
Alex Lorenz12045a42015-07-27 17:42:45 +0000682bool MIParser::parseStandaloneVirtualRegister(unsigned &Reg) {
683 lex();
684 if (Token.isNot(MIToken::VirtualRegister))
685 return error("expected a virtual register");
686 if (parseRegister(Reg))
Alex Lorenz607efb62015-08-18 22:57:36 +0000687 return true;
Alex Lorenz12045a42015-07-27 17:42:45 +0000688 lex();
689 if (Token.isNot(MIToken::Eof))
690 return error("expected end of string after the register reference");
691 return false;
692}
693
Alex Lorenza314d812015-08-18 22:26:26 +0000694bool MIParser::parseStandaloneStackObject(int &FI) {
695 lex();
696 if (Token.isNot(MIToken::StackObject))
697 return error("expected a stack object");
698 if (parseStackFrameIndex(FI))
699 return true;
700 if (Token.isNot(MIToken::Eof))
701 return error("expected end of string after the stack object reference");
702 return false;
703}
704
Alex Lorenzdf9e3c62015-08-19 00:13:25 +0000705bool MIParser::parseStandaloneMDNode(MDNode *&Node) {
706 lex();
707 if (Token.isNot(MIToken::exclaim))
708 return error("expected a metadata node");
709 if (parseMDNode(Node))
710 return true;
711 if (Token.isNot(MIToken::Eof))
712 return error("expected end of string after the metadata node");
713 return false;
714}
715
Alex Lorenz36962cd2015-07-07 02:08:46 +0000716static const char *printImplicitRegisterFlag(const MachineOperand &MO) {
717 assert(MO.isImplicit());
718 return MO.isDef() ? "implicit-def" : "implicit";
719}
720
721static std::string getRegisterName(const TargetRegisterInfo *TRI,
722 unsigned Reg) {
723 assert(TargetRegisterInfo::isPhysicalRegister(Reg) && "expected phys reg");
724 return StringRef(TRI->getName(Reg)).lower();
725}
726
Alex Lorenzfeb6b432015-08-19 19:19:16 +0000727bool MIParser::verifyImplicitOperands(ArrayRef<ParsedMachineOperand> Operands,
728 const MCInstrDesc &MCID) {
Alex Lorenz36962cd2015-07-07 02:08:46 +0000729 if (MCID.isCall())
730 // We can't verify call instructions as they can contain arbitrary implicit
731 // register and register mask operands.
732 return false;
733
734 // Gather all the expected implicit operands.
735 SmallVector<MachineOperand, 4> ImplicitOperands;
736 if (MCID.ImplicitDefs)
737 for (const uint16_t *ImpDefs = MCID.getImplicitDefs(); *ImpDefs; ++ImpDefs)
738 ImplicitOperands.push_back(
739 MachineOperand::CreateReg(*ImpDefs, true, true));
740 if (MCID.ImplicitUses)
741 for (const uint16_t *ImpUses = MCID.getImplicitUses(); *ImpUses; ++ImpUses)
742 ImplicitOperands.push_back(
743 MachineOperand::CreateReg(*ImpUses, false, true));
744
745 const auto *TRI = MF.getSubtarget().getRegisterInfo();
746 assert(TRI && "Expected target register info");
747 size_t I = ImplicitOperands.size(), J = Operands.size();
748 while (I) {
749 --I;
750 if (J) {
751 --J;
752 const auto &ImplicitOperand = ImplicitOperands[I];
753 const auto &Operand = Operands[J].Operand;
754 if (ImplicitOperand.isIdenticalTo(Operand))
755 continue;
756 if (Operand.isReg() && Operand.isImplicit()) {
Alex Lorenzeb7c9be2015-08-18 17:17:13 +0000757 // Check if this implicit register is a subregister of an explicit
758 // register operand.
759 bool IsImplicitSubRegister = false;
760 for (size_t K = 0, E = Operands.size(); K < E; ++K) {
761 const auto &Op = Operands[K].Operand;
762 if (Op.isReg() && !Op.isImplicit() &&
763 TRI->isSubRegister(Op.getReg(), Operand.getReg())) {
764 IsImplicitSubRegister = true;
765 break;
766 }
767 }
768 if (IsImplicitSubRegister)
769 continue;
Alex Lorenz36962cd2015-07-07 02:08:46 +0000770 return error(Operands[J].Begin,
771 Twine("expected an implicit register operand '") +
772 printImplicitRegisterFlag(ImplicitOperand) + " %" +
773 getRegisterName(TRI, ImplicitOperand.getReg()) + "'");
774 }
775 }
776 // TODO: Fix source location when Operands[J].end is right before '=', i.e:
777 // insead of reporting an error at this location:
778 // %eax = MOV32r0
779 // ^
780 // report the error at the following location:
781 // %eax = MOV32r0
782 // ^
783 return error(J < Operands.size() ? Operands[J].End : Token.location(),
784 Twine("missing implicit register operand '") +
785 printImplicitRegisterFlag(ImplicitOperands[I]) + " %" +
786 getRegisterName(TRI, ImplicitOperands[I].getReg()) + "'");
787 }
788 return false;
789}
790
Alex Lorenze5a44662015-07-17 00:24:15 +0000791bool MIParser::parseInstruction(unsigned &OpCode, unsigned &Flags) {
792 if (Token.is(MIToken::kw_frame_setup)) {
793 Flags |= MachineInstr::FrameSetup;
794 lex();
795 }
Alex Lorenz91370c52015-06-22 20:37:46 +0000796 if (Token.isNot(MIToken::Identifier))
797 return error("expected a machine instruction");
798 StringRef InstrName = Token.stringValue();
799 if (parseInstrName(InstrName, OpCode))
800 return error(Twine("unknown machine instruction name '") + InstrName + "'");
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000801 lex();
802 return false;
803}
804
805bool MIParser::parseRegister(unsigned &Reg) {
806 switch (Token.kind()) {
Alex Lorenz12b554e2015-06-24 17:34:58 +0000807 case MIToken::underscore:
808 Reg = 0;
809 break;
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000810 case MIToken::NamedRegister: {
Alex Lorenz33f0aef2015-06-26 16:46:11 +0000811 StringRef Name = Token.stringValue();
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000812 if (getRegisterByName(Name, Reg))
813 return error(Twine("unknown register name '") + Name + "'");
814 break;
815 }
Alex Lorenz53464512015-07-10 22:51:20 +0000816 case MIToken::VirtualRegister: {
817 unsigned ID;
818 if (getUnsigned(ID))
819 return true;
820 const auto RegInfo = PFS.VirtualRegisterSlots.find(ID);
821 if (RegInfo == PFS.VirtualRegisterSlots.end())
822 return error(Twine("use of undefined virtual register '%") + Twine(ID) +
823 "'");
824 Reg = RegInfo->second;
825 break;
826 }
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000827 // TODO: Parse other register kinds.
828 default:
829 llvm_unreachable("The current token should be a register");
830 }
831 return false;
832}
833
Alex Lorenzcb268d42015-07-06 23:07:26 +0000834bool MIParser::parseRegisterFlag(unsigned &Flags) {
Alex Lorenz2b3cf192015-08-05 18:09:03 +0000835 const unsigned OldFlags = Flags;
Alex Lorenzcb268d42015-07-06 23:07:26 +0000836 switch (Token.kind()) {
837 case MIToken::kw_implicit:
838 Flags |= RegState::Implicit;
839 break;
840 case MIToken::kw_implicit_define:
841 Flags |= RegState::ImplicitDefine;
842 break;
Alex Lorenze66a7cc2015-08-19 18:55:47 +0000843 case MIToken::kw_def:
844 Flags |= RegState::Define;
845 break;
Alex Lorenzcbbfd0b2015-07-07 20:34:53 +0000846 case MIToken::kw_dead:
847 Flags |= RegState::Dead;
848 break;
Alex Lorenz495ad872015-07-08 21:23:34 +0000849 case MIToken::kw_killed:
850 Flags |= RegState::Kill;
851 break;
Alex Lorenz4d026b892015-07-08 23:58:31 +0000852 case MIToken::kw_undef:
853 Flags |= RegState::Undef;
854 break;
Alex Lorenz1039fd12015-08-14 19:07:07 +0000855 case MIToken::kw_internal:
856 Flags |= RegState::InternalRead;
857 break;
Alex Lorenz01c1a5e2015-08-05 17:49:03 +0000858 case MIToken::kw_early_clobber:
859 Flags |= RegState::EarlyClobber;
860 break;
Alex Lorenz90752582015-08-05 17:41:17 +0000861 case MIToken::kw_debug_use:
862 Flags |= RegState::Debug;
863 break;
Alex Lorenzcb268d42015-07-06 23:07:26 +0000864 default:
865 llvm_unreachable("The current token should be a register flag");
866 }
Alex Lorenz2b3cf192015-08-05 18:09:03 +0000867 if (OldFlags == Flags)
868 // We know that the same flag is specified more than once when the flags
869 // weren't modified.
870 return error("duplicate '" + Token.stringValue() + "' register flag");
Alex Lorenzcb268d42015-07-06 23:07:26 +0000871 lex();
872 return false;
873}
874
Alex Lorenz2eacca82015-07-13 23:24:34 +0000875bool MIParser::parseSubRegisterIndex(unsigned &SubReg) {
876 assert(Token.is(MIToken::colon));
877 lex();
878 if (Token.isNot(MIToken::Identifier))
879 return error("expected a subregister index after ':'");
880 auto Name = Token.stringValue();
881 SubReg = getSubRegIndex(Name);
882 if (!SubReg)
883 return error(Twine("use of unknown subregister index '") + Name + "'");
884 lex();
885 return false;
886}
887
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000888bool MIParser::parseRegisterTiedDefIndex(unsigned &TiedDefIdx) {
889 if (!consumeIfPresent(MIToken::kw_tied_def))
890 return error("expected 'tied-def' after '('");
891 if (Token.isNot(MIToken::IntegerLiteral))
892 return error("expected an integer literal after 'tied-def'");
893 if (getUnsigned(TiedDefIdx))
894 return true;
895 lex();
896 if (expectAndConsume(MIToken::rparen))
897 return true;
898 return false;
899}
900
Alex Lorenzfeb6b432015-08-19 19:19:16 +0000901bool MIParser::assignRegisterTies(MachineInstr &MI,
902 ArrayRef<ParsedMachineOperand> Operands) {
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000903 SmallVector<std::pair<unsigned, unsigned>, 4> TiedRegisterPairs;
904 for (unsigned I = 0, E = Operands.size(); I != E; ++I) {
905 if (!Operands[I].TiedDefIdx)
906 continue;
907 // The parser ensures that this operand is a register use, so we just have
908 // to check the tied-def operand.
909 unsigned DefIdx = Operands[I].TiedDefIdx.getValue();
910 if (DefIdx >= E)
911 return error(Operands[I].Begin,
912 Twine("use of invalid tied-def operand index '" +
913 Twine(DefIdx) + "'; instruction has only ") +
914 Twine(E) + " operands");
915 const auto &DefOperand = Operands[DefIdx].Operand;
916 if (!DefOperand.isReg() || !DefOperand.isDef())
917 // FIXME: add note with the def operand.
918 return error(Operands[I].Begin,
919 Twine("use of invalid tied-def operand index '") +
920 Twine(DefIdx) + "'; the operand #" + Twine(DefIdx) +
921 " isn't a defined register");
922 // Check that the tied-def operand wasn't tied elsewhere.
923 for (const auto &TiedPair : TiedRegisterPairs) {
924 if (TiedPair.first == DefIdx)
925 return error(Operands[I].Begin,
926 Twine("the tied-def operand #") + Twine(DefIdx) +
927 " is already tied with another register operand");
928 }
929 TiedRegisterPairs.push_back(std::make_pair(DefIdx, I));
930 }
931 // FIXME: Verify that for non INLINEASM instructions, the def and use tied
932 // indices must be less than tied max.
933 for (const auto &TiedPair : TiedRegisterPairs)
934 MI.tieOperands(TiedPair.first, TiedPair.second);
935 return false;
936}
937
938bool MIParser::parseRegisterOperand(MachineOperand &Dest,
939 Optional<unsigned> &TiedDefIdx,
940 bool IsDef) {
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000941 unsigned Reg;
Alex Lorenzcb268d42015-07-06 23:07:26 +0000942 unsigned Flags = IsDef ? RegState::Define : 0;
943 while (Token.isRegisterFlag()) {
944 if (parseRegisterFlag(Flags))
945 return true;
946 }
947 if (!Token.isRegister())
948 return error("expected a register after register flags");
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000949 if (parseRegister(Reg))
950 return true;
951 lex();
Alex Lorenz2eacca82015-07-13 23:24:34 +0000952 unsigned SubReg = 0;
953 if (Token.is(MIToken::colon)) {
954 if (parseSubRegisterIndex(SubReg))
955 return true;
956 }
Alex Lorenz5ef93b02015-08-19 19:05:34 +0000957 if ((Flags & RegState::Define) == 0 && consumeIfPresent(MIToken::lparen)) {
958 unsigned Idx;
959 if (parseRegisterTiedDefIndex(Idx))
960 return true;
961 TiedDefIdx = Idx;
962 }
Alex Lorenz495ad872015-07-08 21:23:34 +0000963 Dest = MachineOperand::CreateReg(
964 Reg, Flags & RegState::Define, Flags & RegState::Implicit,
Alex Lorenz2eacca82015-07-13 23:24:34 +0000965 Flags & RegState::Kill, Flags & RegState::Dead, Flags & RegState::Undef,
Alex Lorenz1039fd12015-08-14 19:07:07 +0000966 Flags & RegState::EarlyClobber, SubReg, Flags & RegState::Debug,
967 Flags & RegState::InternalRead);
Alex Lorenzf3db51de2015-06-23 16:35:26 +0000968 return false;
969}
970
Alex Lorenz240fc1e2015-06-23 23:42:28 +0000971bool MIParser::parseImmediateOperand(MachineOperand &Dest) {
972 assert(Token.is(MIToken::IntegerLiteral));
973 const APSInt &Int = Token.integerValue();
974 if (Int.getMinSignedBits() > 64)
Alex Lorenz3f2058d2015-08-05 19:03:42 +0000975 return error("integer literal is too large to be an immediate operand");
Alex Lorenz240fc1e2015-06-23 23:42:28 +0000976 Dest = MachineOperand::CreateImm(Int.getExtValue());
977 lex();
978 return false;
979}
980
Alex Lorenz5d8b0bd2015-08-21 21:48:22 +0000981bool MIParser::parseIRConstant(StringRef::iterator Loc, StringRef StringValue,
982 const Constant *&C) {
983 auto Source = StringValue.str(); // The source has to be null terminated.
Alex Lorenz7eaff4c2015-08-05 18:44:00 +0000984 SMDiagnostic Err;
Alex Lorenzc1136ef32015-08-21 21:54:12 +0000985 C = parseConstantValue(Source.c_str(), Err, *MF.getFunction()->getParent(),
986 &IRSlots);
Alex Lorenz7eaff4c2015-08-05 18:44:00 +0000987 if (!C)
988 return error(Loc + Err.getColumnNo(), Err.getMessage());
989 return false;
990}
991
Alex Lorenz5d8b0bd2015-08-21 21:48:22 +0000992bool MIParser::parseIRConstant(StringRef::iterator Loc, const Constant *&C) {
993 if (parseIRConstant(Loc, StringRef(Loc, Token.range().end() - Loc), C))
994 return true;
995 lex();
996 return false;
997}
998
Alex Lorenz05e38822015-08-05 18:52:21 +0000999bool MIParser::parseTypedImmediateOperand(MachineOperand &Dest) {
1000 assert(Token.is(MIToken::IntegerType));
1001 auto Loc = Token.location();
1002 lex();
1003 if (Token.isNot(MIToken::IntegerLiteral))
1004 return error("expected an integer literal");
1005 const Constant *C = nullptr;
1006 if (parseIRConstant(Loc, C))
1007 return true;
1008 Dest = MachineOperand::CreateCImm(cast<ConstantInt>(C));
1009 return false;
1010}
1011
Alex Lorenzad156fb2015-07-31 20:49:21 +00001012bool MIParser::parseFPImmediateOperand(MachineOperand &Dest) {
1013 auto Loc = Token.location();
1014 lex();
1015 if (Token.isNot(MIToken::FloatingPointLiteral))
1016 return error("expected a floating point literal");
Alex Lorenz7eaff4c2015-08-05 18:44:00 +00001017 const Constant *C = nullptr;
1018 if (parseIRConstant(Loc, C))
1019 return true;
Alex Lorenzad156fb2015-07-31 20:49:21 +00001020 Dest = MachineOperand::CreateFPImm(cast<ConstantFP>(C));
1021 return false;
1022}
1023
Alex Lorenz33f0aef2015-06-26 16:46:11 +00001024bool MIParser::getUnsigned(unsigned &Result) {
1025 assert(Token.hasIntegerValue() && "Expected a token with an integer value");
1026 const uint64_t Limit = uint64_t(std::numeric_limits<unsigned>::max()) + 1;
1027 uint64_t Val64 = Token.integerValue().getLimitedValue(Limit);
1028 if (Val64 == Limit)
1029 return error("expected 32-bit integer (too large)");
1030 Result = Val64;
1031 return false;
1032}
1033
Alex Lorenzf09df002015-06-30 18:16:42 +00001034bool MIParser::parseMBBReference(MachineBasicBlock *&MBB) {
Alex Lorenz5022f6b2015-08-13 23:10:16 +00001035 assert(Token.is(MIToken::MachineBasicBlock) ||
1036 Token.is(MIToken::MachineBasicBlockLabel));
Alex Lorenz33f0aef2015-06-26 16:46:11 +00001037 unsigned Number;
1038 if (getUnsigned(Number))
1039 return true;
Alex Lorenz7a503fa2015-07-07 17:46:43 +00001040 auto MBBInfo = PFS.MBBSlots.find(Number);
1041 if (MBBInfo == PFS.MBBSlots.end())
Alex Lorenz33f0aef2015-06-26 16:46:11 +00001042 return error(Twine("use of undefined machine basic block #") +
1043 Twine(Number));
Alex Lorenzf09df002015-06-30 18:16:42 +00001044 MBB = MBBInfo->second;
Alex Lorenz33f0aef2015-06-26 16:46:11 +00001045 if (!Token.stringValue().empty() && Token.stringValue() != MBB->getName())
1046 return error(Twine("the name of machine basic block #") + Twine(Number) +
1047 " isn't '" + Token.stringValue() + "'");
Alex Lorenzf09df002015-06-30 18:16:42 +00001048 return false;
1049}
1050
1051bool MIParser::parseMBBOperand(MachineOperand &Dest) {
1052 MachineBasicBlock *MBB;
1053 if (parseMBBReference(MBB))
1054 return true;
Alex Lorenz33f0aef2015-06-26 16:46:11 +00001055 Dest = MachineOperand::CreateMBB(MBB);
1056 lex();
1057 return false;
1058}
1059
Alex Lorenzdc9dadf2015-08-18 22:18:52 +00001060bool MIParser::parseStackFrameIndex(int &FI) {
Alex Lorenz7feaf7c2015-07-16 23:37:45 +00001061 assert(Token.is(MIToken::StackObject));
1062 unsigned ID;
1063 if (getUnsigned(ID))
1064 return true;
1065 auto ObjectInfo = PFS.StackObjectSlots.find(ID);
1066 if (ObjectInfo == PFS.StackObjectSlots.end())
1067 return error(Twine("use of undefined stack object '%stack.") + Twine(ID) +
1068 "'");
1069 StringRef Name;
1070 if (const auto *Alloca =
1071 MF.getFrameInfo()->getObjectAllocation(ObjectInfo->second))
1072 Name = Alloca->getName();
1073 if (!Token.stringValue().empty() && Token.stringValue() != Name)
1074 return error(Twine("the name of the stack object '%stack.") + Twine(ID) +
1075 "' isn't '" + Token.stringValue() + "'");
1076 lex();
Alex Lorenzdc9dadf2015-08-18 22:18:52 +00001077 FI = ObjectInfo->second;
1078 return false;
1079}
1080
1081bool MIParser::parseStackObjectOperand(MachineOperand &Dest) {
1082 int FI;
1083 if (parseStackFrameIndex(FI))
1084 return true;
1085 Dest = MachineOperand::CreateFI(FI);
Alex Lorenz7feaf7c2015-07-16 23:37:45 +00001086 return false;
1087}
1088
Alex Lorenzea882122015-08-12 21:17:02 +00001089bool MIParser::parseFixedStackFrameIndex(int &FI) {
Alex Lorenz7feaf7c2015-07-16 23:37:45 +00001090 assert(Token.is(MIToken::FixedStackObject));
1091 unsigned ID;
1092 if (getUnsigned(ID))
1093 return true;
1094 auto ObjectInfo = PFS.FixedStackObjectSlots.find(ID);
1095 if (ObjectInfo == PFS.FixedStackObjectSlots.end())
1096 return error(Twine("use of undefined fixed stack object '%fixed-stack.") +
1097 Twine(ID) + "'");
1098 lex();
Alex Lorenzea882122015-08-12 21:17:02 +00001099 FI = ObjectInfo->second;
1100 return false;
1101}
1102
1103bool MIParser::parseFixedStackObjectOperand(MachineOperand &Dest) {
1104 int FI;
1105 if (parseFixedStackFrameIndex(FI))
1106 return true;
1107 Dest = MachineOperand::CreateFI(FI);
Alex Lorenz7feaf7c2015-07-16 23:37:45 +00001108 return false;
1109}
1110
Alex Lorenz41df7d32015-07-28 17:09:52 +00001111bool MIParser::parseGlobalValue(GlobalValue *&GV) {
Alex Lorenz5d6108e2015-06-26 22:56:48 +00001112 switch (Token.kind()) {
Alex Lorenz970c12e2015-08-05 17:35:55 +00001113 case MIToken::NamedGlobalValue: {
Alex Lorenz5d6108e2015-06-26 22:56:48 +00001114 const Module *M = MF.getFunction()->getParent();
Alex Lorenz970c12e2015-08-05 17:35:55 +00001115 GV = M->getNamedValue(Token.stringValue());
Alex Lorenz41df7d32015-07-28 17:09:52 +00001116 if (!GV)
Alex Lorenz3fb77682015-08-06 23:17:42 +00001117 return error(Twine("use of undefined global value '") + Token.range() +
1118 "'");
Alex Lorenz41df7d32015-07-28 17:09:52 +00001119 break;
Alex Lorenz5d6108e2015-06-26 22:56:48 +00001120 }
1121 case MIToken::GlobalValue: {
1122 unsigned GVIdx;
1123 if (getUnsigned(GVIdx))
1124 return true;
1125 if (GVIdx >= IRSlots.GlobalValues.size())
1126 return error(Twine("use of undefined global value '@") + Twine(GVIdx) +
1127 "'");
Alex Lorenz41df7d32015-07-28 17:09:52 +00001128 GV = IRSlots.GlobalValues[GVIdx];
Alex Lorenz5d6108e2015-06-26 22:56:48 +00001129 break;
1130 }
1131 default:
1132 llvm_unreachable("The current token should be a global value");
1133 }
Alex Lorenz41df7d32015-07-28 17:09:52 +00001134 return false;
1135}
1136
1137bool MIParser::parseGlobalAddressOperand(MachineOperand &Dest) {
1138 GlobalValue *GV = nullptr;
1139 if (parseGlobalValue(GV))
1140 return true;
Alex Lorenz5d6108e2015-06-26 22:56:48 +00001141 lex();
Alex Lorenz5672a892015-08-05 22:26:15 +00001142 Dest = MachineOperand::CreateGA(GV, /*Offset=*/0);
Alex Lorenz5672a892015-08-05 22:26:15 +00001143 if (parseOperandsOffset(Dest))
1144 return true;
Alex Lorenz5d6108e2015-06-26 22:56:48 +00001145 return false;
1146}
1147
Alex Lorenzab980492015-07-20 20:51:18 +00001148bool MIParser::parseConstantPoolIndexOperand(MachineOperand &Dest) {
1149 assert(Token.is(MIToken::ConstantPoolItem));
1150 unsigned ID;
1151 if (getUnsigned(ID))
1152 return true;
1153 auto ConstantInfo = PFS.ConstantPoolSlots.find(ID);
1154 if (ConstantInfo == PFS.ConstantPoolSlots.end())
1155 return error("use of undefined constant '%const." + Twine(ID) + "'");
1156 lex();
Alex Lorenzab980492015-07-20 20:51:18 +00001157 Dest = MachineOperand::CreateCPI(ID, /*Offset=*/0);
Alex Lorenz5672a892015-08-05 22:26:15 +00001158 if (parseOperandsOffset(Dest))
1159 return true;
Alex Lorenzab980492015-07-20 20:51:18 +00001160 return false;
1161}
1162
Alex Lorenz31d70682015-07-15 23:38:35 +00001163bool MIParser::parseJumpTableIndexOperand(MachineOperand &Dest) {
1164 assert(Token.is(MIToken::JumpTableIndex));
1165 unsigned ID;
1166 if (getUnsigned(ID))
1167 return true;
1168 auto JumpTableEntryInfo = PFS.JumpTableSlots.find(ID);
1169 if (JumpTableEntryInfo == PFS.JumpTableSlots.end())
1170 return error("use of undefined jump table '%jump-table." + Twine(ID) + "'");
1171 lex();
Alex Lorenz31d70682015-07-15 23:38:35 +00001172 Dest = MachineOperand::CreateJTI(JumpTableEntryInfo->second);
1173 return false;
1174}
1175
Alex Lorenz6ede3742015-07-21 16:59:53 +00001176bool MIParser::parseExternalSymbolOperand(MachineOperand &Dest) {
Alex Lorenz970c12e2015-08-05 17:35:55 +00001177 assert(Token.is(MIToken::ExternalSymbol));
1178 const char *Symbol = MF.createExternalSymbolName(Token.stringValue());
Alex Lorenz6ede3742015-07-21 16:59:53 +00001179 lex();
Alex Lorenz6ede3742015-07-21 16:59:53 +00001180 Dest = MachineOperand::CreateES(Symbol);
Alex Lorenz5672a892015-08-05 22:26:15 +00001181 if (parseOperandsOffset(Dest))
1182 return true;
Alex Lorenz6ede3742015-07-21 16:59:53 +00001183 return false;
1184}
1185
Alex Lorenz44f29252015-07-22 21:07:04 +00001186bool MIParser::parseMDNode(MDNode *&Node) {
Alex Lorenz35e44462015-07-22 17:58:46 +00001187 assert(Token.is(MIToken::exclaim));
1188 auto Loc = Token.location();
1189 lex();
1190 if (Token.isNot(MIToken::IntegerLiteral) || Token.integerValue().isSigned())
1191 return error("expected metadata id after '!'");
1192 unsigned ID;
1193 if (getUnsigned(ID))
1194 return true;
1195 auto NodeInfo = IRSlots.MetadataNodes.find(ID);
1196 if (NodeInfo == IRSlots.MetadataNodes.end())
1197 return error(Loc, "use of undefined metadata '!" + Twine(ID) + "'");
1198 lex();
Alex Lorenz44f29252015-07-22 21:07:04 +00001199 Node = NodeInfo->second.get();
1200 return false;
1201}
1202
1203bool MIParser::parseMetadataOperand(MachineOperand &Dest) {
1204 MDNode *Node = nullptr;
1205 if (parseMDNode(Node))
1206 return true;
1207 Dest = MachineOperand::CreateMetadata(Node);
Alex Lorenz35e44462015-07-22 17:58:46 +00001208 return false;
1209}
1210
Alex Lorenzf4baeb52015-07-21 22:28:27 +00001211bool MIParser::parseCFIOffset(int &Offset) {
1212 if (Token.isNot(MIToken::IntegerLiteral))
1213 return error("expected a cfi offset");
1214 if (Token.integerValue().getMinSignedBits() > 32)
1215 return error("expected a 32 bit integer (the cfi offset is too large)");
1216 Offset = (int)Token.integerValue().getExtValue();
1217 lex();
1218 return false;
1219}
1220
Alex Lorenz8cfc6862015-07-23 23:09:07 +00001221bool MIParser::parseCFIRegister(unsigned &Reg) {
1222 if (Token.isNot(MIToken::NamedRegister))
1223 return error("expected a cfi register");
1224 unsigned LLVMReg;
1225 if (parseRegister(LLVMReg))
Alex Lorenzf4baeb52015-07-21 22:28:27 +00001226 return true;
Alex Lorenz8cfc6862015-07-23 23:09:07 +00001227 const auto *TRI = MF.getSubtarget().getRegisterInfo();
1228 assert(TRI && "Expected target register info");
1229 int DwarfReg = TRI->getDwarfRegNum(LLVMReg, true);
1230 if (DwarfReg < 0)
1231 return error("invalid DWARF register");
1232 Reg = (unsigned)DwarfReg;
1233 lex();
1234 return false;
1235}
1236
1237bool MIParser::parseCFIOperand(MachineOperand &Dest) {
1238 auto Kind = Token.kind();
1239 lex();
1240 auto &MMI = MF.getMMI();
1241 int Offset;
1242 unsigned Reg;
1243 unsigned CFIIndex;
1244 switch (Kind) {
Alex Lorenz577d2712015-08-14 21:55:58 +00001245 case MIToken::kw_cfi_same_value:
1246 if (parseCFIRegister(Reg))
1247 return true;
1248 CFIIndex =
1249 MMI.addFrameInst(MCCFIInstruction::createSameValue(nullptr, Reg));
1250 break;
Alex Lorenz8cfc6862015-07-23 23:09:07 +00001251 case MIToken::kw_cfi_offset:
1252 if (parseCFIRegister(Reg) || expectAndConsume(MIToken::comma) ||
1253 parseCFIOffset(Offset))
1254 return true;
1255 CFIIndex =
1256 MMI.addFrameInst(MCCFIInstruction::createOffset(nullptr, Reg, Offset));
1257 break;
Alex Lorenz5b0d5f62015-07-27 20:39:03 +00001258 case MIToken::kw_cfi_def_cfa_register:
1259 if (parseCFIRegister(Reg))
1260 return true;
1261 CFIIndex =
1262 MMI.addFrameInst(MCCFIInstruction::createDefCfaRegister(nullptr, Reg));
1263 break;
Alex Lorenz8cfc6862015-07-23 23:09:07 +00001264 case MIToken::kw_cfi_def_cfa_offset:
1265 if (parseCFIOffset(Offset))
1266 return true;
1267 // NB: MCCFIInstruction::createDefCfaOffset negates the offset.
1268 CFIIndex = MMI.addFrameInst(
1269 MCCFIInstruction::createDefCfaOffset(nullptr, -Offset));
1270 break;
Alex Lorenzb1393232015-07-29 18:57:23 +00001271 case MIToken::kw_cfi_def_cfa:
1272 if (parseCFIRegister(Reg) || expectAndConsume(MIToken::comma) ||
1273 parseCFIOffset(Offset))
1274 return true;
1275 // NB: MCCFIInstruction::createDefCfa negates the offset.
1276 CFIIndex =
1277 MMI.addFrameInst(MCCFIInstruction::createDefCfa(nullptr, Reg, -Offset));
1278 break;
Alex Lorenz8cfc6862015-07-23 23:09:07 +00001279 default:
1280 // TODO: Parse the other CFI operands.
1281 llvm_unreachable("The current token should be a cfi operand");
1282 }
1283 Dest = MachineOperand::CreateCFIIndex(CFIIndex);
Alex Lorenzf4baeb52015-07-21 22:28:27 +00001284 return false;
1285}
1286
Alex Lorenzdeb53492015-07-28 17:28:03 +00001287bool MIParser::parseIRBlock(BasicBlock *&BB, const Function &F) {
1288 switch (Token.kind()) {
Alex Lorenz970c12e2015-08-05 17:35:55 +00001289 case MIToken::NamedIRBlock: {
1290 BB = dyn_cast_or_null<BasicBlock>(
1291 F.getValueSymbolTable().lookup(Token.stringValue()));
Alex Lorenzdeb53492015-07-28 17:28:03 +00001292 if (!BB)
Alex Lorenz3fb77682015-08-06 23:17:42 +00001293 return error(Twine("use of undefined IR block '") + Token.range() + "'");
Alex Lorenzdeb53492015-07-28 17:28:03 +00001294 break;
1295 }
1296 case MIToken::IRBlock: {
1297 unsigned SlotNumber = 0;
1298 if (getUnsigned(SlotNumber))
1299 return true;
Alex Lorenzcba8c5f2015-08-06 23:57:04 +00001300 BB = const_cast<BasicBlock *>(getIRBlock(SlotNumber, F));
Alex Lorenzdeb53492015-07-28 17:28:03 +00001301 if (!BB)
1302 return error(Twine("use of undefined IR block '%ir-block.") +
1303 Twine(SlotNumber) + "'");
1304 break;
1305 }
1306 default:
1307 llvm_unreachable("The current token should be an IR block reference");
1308 }
1309 return false;
1310}
1311
1312bool MIParser::parseBlockAddressOperand(MachineOperand &Dest) {
1313 assert(Token.is(MIToken::kw_blockaddress));
1314 lex();
1315 if (expectAndConsume(MIToken::lparen))
1316 return true;
1317 if (Token.isNot(MIToken::GlobalValue) &&
Alex Lorenz970c12e2015-08-05 17:35:55 +00001318 Token.isNot(MIToken::NamedGlobalValue))
Alex Lorenzdeb53492015-07-28 17:28:03 +00001319 return error("expected a global value");
1320 GlobalValue *GV = nullptr;
1321 if (parseGlobalValue(GV))
1322 return true;
1323 auto *F = dyn_cast<Function>(GV);
1324 if (!F)
1325 return error("expected an IR function reference");
1326 lex();
1327 if (expectAndConsume(MIToken::comma))
1328 return true;
1329 BasicBlock *BB = nullptr;
Alex Lorenz970c12e2015-08-05 17:35:55 +00001330 if (Token.isNot(MIToken::IRBlock) && Token.isNot(MIToken::NamedIRBlock))
Alex Lorenzdeb53492015-07-28 17:28:03 +00001331 return error("expected an IR block reference");
1332 if (parseIRBlock(BB, *F))
1333 return true;
1334 lex();
1335 if (expectAndConsume(MIToken::rparen))
1336 return true;
Alex Lorenzdeb53492015-07-28 17:28:03 +00001337 Dest = MachineOperand::CreateBA(BlockAddress::get(F, BB), /*Offset=*/0);
Alex Lorenz5672a892015-08-05 22:26:15 +00001338 if (parseOperandsOffset(Dest))
1339 return true;
Alex Lorenzdeb53492015-07-28 17:28:03 +00001340 return false;
1341}
1342
Alex Lorenzef5c1962015-07-28 23:02:45 +00001343bool MIParser::parseTargetIndexOperand(MachineOperand &Dest) {
1344 assert(Token.is(MIToken::kw_target_index));
1345 lex();
1346 if (expectAndConsume(MIToken::lparen))
1347 return true;
1348 if (Token.isNot(MIToken::Identifier))
1349 return error("expected the name of the target index");
1350 int Index = 0;
1351 if (getTargetIndex(Token.stringValue(), Index))
1352 return error("use of undefined target index '" + Token.stringValue() + "'");
1353 lex();
1354 if (expectAndConsume(MIToken::rparen))
1355 return true;
Alex Lorenzef5c1962015-07-28 23:02:45 +00001356 Dest = MachineOperand::CreateTargetIndex(unsigned(Index), /*Offset=*/0);
Alex Lorenz5672a892015-08-05 22:26:15 +00001357 if (parseOperandsOffset(Dest))
1358 return true;
Alex Lorenzef5c1962015-07-28 23:02:45 +00001359 return false;
1360}
1361
Alex Lorenzb97c9ef2015-08-10 23:24:42 +00001362bool MIParser::parseLiveoutRegisterMaskOperand(MachineOperand &Dest) {
1363 assert(Token.is(MIToken::kw_liveout));
1364 const auto *TRI = MF.getSubtarget().getRegisterInfo();
1365 assert(TRI && "Expected target register info");
1366 uint32_t *Mask = MF.allocateRegisterMask(TRI->getNumRegs());
1367 lex();
1368 if (expectAndConsume(MIToken::lparen))
1369 return true;
1370 while (true) {
1371 if (Token.isNot(MIToken::NamedRegister))
1372 return error("expected a named register");
1373 unsigned Reg = 0;
1374 if (parseRegister(Reg))
1375 return true;
1376 lex();
1377 Mask[Reg / 32] |= 1U << (Reg % 32);
1378 // TODO: Report an error if the same register is used more than once.
1379 if (Token.isNot(MIToken::comma))
1380 break;
1381 lex();
1382 }
1383 if (expectAndConsume(MIToken::rparen))
1384 return true;
1385 Dest = MachineOperand::CreateRegLiveOut(Mask);
1386 return false;
1387}
1388
Alex Lorenz5ef93b02015-08-19 19:05:34 +00001389bool MIParser::parseMachineOperand(MachineOperand &Dest,
1390 Optional<unsigned> &TiedDefIdx) {
Alex Lorenzf3db51de2015-06-23 16:35:26 +00001391 switch (Token.kind()) {
Alex Lorenzcb268d42015-07-06 23:07:26 +00001392 case MIToken::kw_implicit:
1393 case MIToken::kw_implicit_define:
Alex Lorenze66a7cc2015-08-19 18:55:47 +00001394 case MIToken::kw_def:
Alex Lorenzcbbfd0b2015-07-07 20:34:53 +00001395 case MIToken::kw_dead:
Alex Lorenz495ad872015-07-08 21:23:34 +00001396 case MIToken::kw_killed:
Alex Lorenz4d026b892015-07-08 23:58:31 +00001397 case MIToken::kw_undef:
Alex Lorenz1039fd12015-08-14 19:07:07 +00001398 case MIToken::kw_internal:
Alex Lorenz01c1a5e2015-08-05 17:49:03 +00001399 case MIToken::kw_early_clobber:
Alex Lorenz90752582015-08-05 17:41:17 +00001400 case MIToken::kw_debug_use:
Alex Lorenz12b554e2015-06-24 17:34:58 +00001401 case MIToken::underscore:
Alex Lorenzf3db51de2015-06-23 16:35:26 +00001402 case MIToken::NamedRegister:
Alex Lorenz53464512015-07-10 22:51:20 +00001403 case MIToken::VirtualRegister:
Alex Lorenz5ef93b02015-08-19 19:05:34 +00001404 return parseRegisterOperand(Dest, TiedDefIdx);
Alex Lorenz240fc1e2015-06-23 23:42:28 +00001405 case MIToken::IntegerLiteral:
1406 return parseImmediateOperand(Dest);
Alex Lorenz05e38822015-08-05 18:52:21 +00001407 case MIToken::IntegerType:
1408 return parseTypedImmediateOperand(Dest);
Alex Lorenzad156fb2015-07-31 20:49:21 +00001409 case MIToken::kw_half:
1410 case MIToken::kw_float:
1411 case MIToken::kw_double:
1412 case MIToken::kw_x86_fp80:
1413 case MIToken::kw_fp128:
1414 case MIToken::kw_ppc_fp128:
1415 return parseFPImmediateOperand(Dest);
Alex Lorenz33f0aef2015-06-26 16:46:11 +00001416 case MIToken::MachineBasicBlock:
1417 return parseMBBOperand(Dest);
Alex Lorenz7feaf7c2015-07-16 23:37:45 +00001418 case MIToken::StackObject:
1419 return parseStackObjectOperand(Dest);
1420 case MIToken::FixedStackObject:
1421 return parseFixedStackObjectOperand(Dest);
Alex Lorenz5d6108e2015-06-26 22:56:48 +00001422 case MIToken::GlobalValue:
1423 case MIToken::NamedGlobalValue:
1424 return parseGlobalAddressOperand(Dest);
Alex Lorenzab980492015-07-20 20:51:18 +00001425 case MIToken::ConstantPoolItem:
1426 return parseConstantPoolIndexOperand(Dest);
Alex Lorenz31d70682015-07-15 23:38:35 +00001427 case MIToken::JumpTableIndex:
1428 return parseJumpTableIndexOperand(Dest);
Alex Lorenz6ede3742015-07-21 16:59:53 +00001429 case MIToken::ExternalSymbol:
Alex Lorenz6ede3742015-07-21 16:59:53 +00001430 return parseExternalSymbolOperand(Dest);
Alex Lorenz35e44462015-07-22 17:58:46 +00001431 case MIToken::exclaim:
1432 return parseMetadataOperand(Dest);
Alex Lorenz577d2712015-08-14 21:55:58 +00001433 case MIToken::kw_cfi_same_value:
Alex Lorenz8cfc6862015-07-23 23:09:07 +00001434 case MIToken::kw_cfi_offset:
Alex Lorenz5b0d5f62015-07-27 20:39:03 +00001435 case MIToken::kw_cfi_def_cfa_register:
Alex Lorenzf4baeb52015-07-21 22:28:27 +00001436 case MIToken::kw_cfi_def_cfa_offset:
Alex Lorenzb1393232015-07-29 18:57:23 +00001437 case MIToken::kw_cfi_def_cfa:
Alex Lorenzf4baeb52015-07-21 22:28:27 +00001438 return parseCFIOperand(Dest);
Alex Lorenzdeb53492015-07-28 17:28:03 +00001439 case MIToken::kw_blockaddress:
1440 return parseBlockAddressOperand(Dest);
Alex Lorenzef5c1962015-07-28 23:02:45 +00001441 case MIToken::kw_target_index:
1442 return parseTargetIndexOperand(Dest);
Alex Lorenzb97c9ef2015-08-10 23:24:42 +00001443 case MIToken::kw_liveout:
1444 return parseLiveoutRegisterMaskOperand(Dest);
Alex Lorenzf3db51de2015-06-23 16:35:26 +00001445 case MIToken::Error:
1446 return true;
Alex Lorenz8f6f4282015-06-29 16:57:06 +00001447 case MIToken::Identifier:
1448 if (const auto *RegMask = getRegMask(Token.stringValue())) {
1449 Dest = MachineOperand::CreateRegMask(RegMask);
1450 lex();
1451 break;
1452 }
1453 // fallthrough
Alex Lorenzf3db51de2015-06-23 16:35:26 +00001454 default:
Alex Lorenzf22ca8a2015-08-21 21:12:44 +00001455 // FIXME: Parse the MCSymbol machine operand.
Alex Lorenzf3db51de2015-06-23 16:35:26 +00001456 return error("expected a machine operand");
1457 }
Alex Lorenz91370c52015-06-22 20:37:46 +00001458 return false;
1459}
1460
Alex Lorenz5ef93b02015-08-19 19:05:34 +00001461bool MIParser::parseMachineOperandAndTargetFlags(
1462 MachineOperand &Dest, Optional<unsigned> &TiedDefIdx) {
Alex Lorenz49873a82015-08-06 00:44:07 +00001463 unsigned TF = 0;
1464 bool HasTargetFlags = false;
1465 if (Token.is(MIToken::kw_target_flags)) {
1466 HasTargetFlags = true;
1467 lex();
1468 if (expectAndConsume(MIToken::lparen))
1469 return true;
1470 if (Token.isNot(MIToken::Identifier))
1471 return error("expected the name of the target flag");
Alex Lorenzf3630112015-08-18 22:52:15 +00001472 if (getDirectTargetFlag(Token.stringValue(), TF)) {
1473 if (getBitmaskTargetFlag(Token.stringValue(), TF))
1474 return error("use of undefined target flag '" + Token.stringValue() +
1475 "'");
1476 }
Alex Lorenz49873a82015-08-06 00:44:07 +00001477 lex();
Alex Lorenzf3630112015-08-18 22:52:15 +00001478 while (Token.is(MIToken::comma)) {
1479 lex();
1480 if (Token.isNot(MIToken::Identifier))
1481 return error("expected the name of the target flag");
1482 unsigned BitFlag = 0;
1483 if (getBitmaskTargetFlag(Token.stringValue(), BitFlag))
1484 return error("use of undefined target flag '" + Token.stringValue() +
1485 "'");
1486 // TODO: Report an error when using a duplicate bit target flag.
1487 TF |= BitFlag;
1488 lex();
1489 }
Alex Lorenz49873a82015-08-06 00:44:07 +00001490 if (expectAndConsume(MIToken::rparen))
1491 return true;
1492 }
1493 auto Loc = Token.location();
Alex Lorenz5ef93b02015-08-19 19:05:34 +00001494 if (parseMachineOperand(Dest, TiedDefIdx))
Alex Lorenz49873a82015-08-06 00:44:07 +00001495 return true;
1496 if (!HasTargetFlags)
1497 return false;
1498 if (Dest.isReg())
1499 return error(Loc, "register operands can't have target flags");
1500 Dest.setTargetFlags(TF);
1501 return false;
1502}
1503
Alex Lorenzdc24c172015-08-07 20:21:00 +00001504bool MIParser::parseOffset(int64_t &Offset) {
Alex Lorenz5672a892015-08-05 22:26:15 +00001505 if (Token.isNot(MIToken::plus) && Token.isNot(MIToken::minus))
1506 return false;
Alex Lorenz3fb77682015-08-06 23:17:42 +00001507 StringRef Sign = Token.range();
Alex Lorenz5672a892015-08-05 22:26:15 +00001508 bool IsNegative = Token.is(MIToken::minus);
1509 lex();
1510 if (Token.isNot(MIToken::IntegerLiteral))
1511 return error("expected an integer literal after '" + Sign + "'");
1512 if (Token.integerValue().getMinSignedBits() > 64)
1513 return error("expected 64-bit integer (too large)");
Alex Lorenzdc24c172015-08-07 20:21:00 +00001514 Offset = Token.integerValue().getExtValue();
Alex Lorenz5672a892015-08-05 22:26:15 +00001515 if (IsNegative)
1516 Offset = -Offset;
1517 lex();
Alex Lorenzdc24c172015-08-07 20:21:00 +00001518 return false;
1519}
1520
Alex Lorenz620f8912015-08-13 20:33:33 +00001521bool MIParser::parseAlignment(unsigned &Alignment) {
1522 assert(Token.is(MIToken::kw_align));
1523 lex();
Alex Lorenz68661042015-08-13 20:55:01 +00001524 if (Token.isNot(MIToken::IntegerLiteral) || Token.integerValue().isSigned())
Alex Lorenz620f8912015-08-13 20:33:33 +00001525 return error("expected an integer literal after 'align'");
1526 if (getUnsigned(Alignment))
1527 return true;
1528 lex();
1529 return false;
1530}
1531
Alex Lorenzdc24c172015-08-07 20:21:00 +00001532bool MIParser::parseOperandsOffset(MachineOperand &Op) {
1533 int64_t Offset = 0;
1534 if (parseOffset(Offset))
1535 return true;
Alex Lorenz5672a892015-08-05 22:26:15 +00001536 Op.setOffset(Offset);
1537 return false;
1538}
1539
Alex Lorenz36593ac2015-08-19 23:27:07 +00001540bool MIParser::parseIRValue(const Value *&V) {
Alex Lorenz4af7e612015-08-03 23:08:19 +00001541 switch (Token.kind()) {
Alex Lorenz970c12e2015-08-05 17:35:55 +00001542 case MIToken::NamedIRValue: {
1543 V = MF.getFunction()->getValueSymbolTable().lookup(Token.stringValue());
Alex Lorenz4af7e612015-08-03 23:08:19 +00001544 break;
1545 }
Alex Lorenzdd13be02015-08-19 23:31:05 +00001546 case MIToken::IRValue: {
1547 unsigned SlotNumber = 0;
1548 if (getUnsigned(SlotNumber))
1549 return true;
1550 V = getIRValue(SlotNumber);
1551 break;
1552 }
Alex Lorenz36efd382015-08-20 00:20:03 +00001553 case MIToken::NamedGlobalValue:
1554 case MIToken::GlobalValue: {
1555 GlobalValue *GV = nullptr;
1556 if (parseGlobalValue(GV))
1557 return true;
1558 V = GV;
1559 break;
1560 }
Alex Lorenzc1136ef32015-08-21 21:54:12 +00001561 case MIToken::QuotedIRValue: {
1562 const Constant *C = nullptr;
1563 if (parseIRConstant(Token.location(), Token.stringValue(), C))
1564 return true;
1565 V = C;
1566 break;
1567 }
Alex Lorenz4af7e612015-08-03 23:08:19 +00001568 default:
1569 llvm_unreachable("The current token should be an IR block reference");
1570 }
Alex Lorenzdd13be02015-08-19 23:31:05 +00001571 if (!V)
1572 return error(Twine("use of undefined IR value '") + Token.range() + "'");
Alex Lorenz4af7e612015-08-03 23:08:19 +00001573 return false;
1574}
1575
1576bool MIParser::getUint64(uint64_t &Result) {
1577 assert(Token.hasIntegerValue());
1578 if (Token.integerValue().getActiveBits() > 64)
1579 return error("expected 64-bit integer (too large)");
1580 Result = Token.integerValue().getZExtValue();
1581 return false;
1582}
1583
Alex Lorenza518b792015-08-04 00:24:45 +00001584bool MIParser::parseMemoryOperandFlag(unsigned &Flags) {
Alex Lorenze86d5152015-08-06 18:26:36 +00001585 const unsigned OldFlags = Flags;
Alex Lorenza518b792015-08-04 00:24:45 +00001586 switch (Token.kind()) {
1587 case MIToken::kw_volatile:
1588 Flags |= MachineMemOperand::MOVolatile;
1589 break;
Alex Lorenz10fd0382015-08-06 16:49:30 +00001590 case MIToken::kw_non_temporal:
1591 Flags |= MachineMemOperand::MONonTemporal;
1592 break;
Alex Lorenzdc8de2a2015-08-06 16:55:53 +00001593 case MIToken::kw_invariant:
1594 Flags |= MachineMemOperand::MOInvariant;
1595 break;
Alex Lorenzdc8de2a2015-08-06 16:55:53 +00001596 // TODO: parse the target specific memory operand flags.
Alex Lorenza518b792015-08-04 00:24:45 +00001597 default:
1598 llvm_unreachable("The current token should be a memory operand flag");
1599 }
Alex Lorenze86d5152015-08-06 18:26:36 +00001600 if (OldFlags == Flags)
1601 // We know that the same flag is specified more than once when the flags
1602 // weren't modified.
1603 return error("duplicate '" + Token.stringValue() + "' memory operand flag");
Alex Lorenza518b792015-08-04 00:24:45 +00001604 lex();
1605 return false;
1606}
1607
Alex Lorenz91097a32015-08-12 20:33:26 +00001608bool MIParser::parseMemoryPseudoSourceValue(const PseudoSourceValue *&PSV) {
1609 switch (Token.kind()) {
Alex Lorenz46e95582015-08-12 20:44:16 +00001610 case MIToken::kw_stack:
1611 PSV = MF.getPSVManager().getStack();
1612 break;
Alex Lorenzd858f872015-08-12 21:00:22 +00001613 case MIToken::kw_got:
1614 PSV = MF.getPSVManager().getGOT();
1615 break;
Alex Lorenz4be56e92015-08-12 21:11:08 +00001616 case MIToken::kw_jump_table:
1617 PSV = MF.getPSVManager().getJumpTable();
1618 break;
Alex Lorenz91097a32015-08-12 20:33:26 +00001619 case MIToken::kw_constant_pool:
1620 PSV = MF.getPSVManager().getConstantPool();
1621 break;
Alex Lorenz0cc671b2015-08-12 21:23:17 +00001622 case MIToken::FixedStackObject: {
1623 int FI;
1624 if (parseFixedStackFrameIndex(FI))
1625 return true;
1626 PSV = MF.getPSVManager().getFixedStack(FI);
1627 // The token was already consumed, so use return here instead of break.
1628 return false;
1629 }
Alex Lorenz0d009642015-08-20 00:12:57 +00001630 case MIToken::kw_call_entry: {
1631 lex();
1632 switch (Token.kind()) {
1633 case MIToken::GlobalValue:
1634 case MIToken::NamedGlobalValue: {
1635 GlobalValue *GV = nullptr;
1636 if (parseGlobalValue(GV))
1637 return true;
1638 PSV = MF.getPSVManager().getGlobalValueCallEntry(GV);
1639 break;
1640 }
1641 case MIToken::ExternalSymbol:
1642 PSV = MF.getPSVManager().getExternalSymbolCallEntry(
1643 MF.createExternalSymbolName(Token.stringValue()));
1644 break;
1645 default:
1646 return error(
1647 "expected a global value or an external symbol after 'call-entry'");
1648 }
Alex Lorenz50b826f2015-08-14 21:08:30 +00001649 break;
1650 }
Alex Lorenz91097a32015-08-12 20:33:26 +00001651 default:
1652 llvm_unreachable("The current token should be pseudo source value");
1653 }
1654 lex();
1655 return false;
1656}
1657
1658bool MIParser::parseMachinePointerInfo(MachinePointerInfo &Dest) {
Alex Lorenzd858f872015-08-12 21:00:22 +00001659 if (Token.is(MIToken::kw_constant_pool) || Token.is(MIToken::kw_stack) ||
Alex Lorenz0cc671b2015-08-12 21:23:17 +00001660 Token.is(MIToken::kw_got) || Token.is(MIToken::kw_jump_table) ||
Alex Lorenz0d009642015-08-20 00:12:57 +00001661 Token.is(MIToken::FixedStackObject) || Token.is(MIToken::kw_call_entry)) {
Alex Lorenz91097a32015-08-12 20:33:26 +00001662 const PseudoSourceValue *PSV = nullptr;
1663 if (parseMemoryPseudoSourceValue(PSV))
1664 return true;
1665 int64_t Offset = 0;
1666 if (parseOffset(Offset))
1667 return true;
1668 Dest = MachinePointerInfo(PSV, Offset);
1669 return false;
1670 }
Alex Lorenz36efd382015-08-20 00:20:03 +00001671 if (Token.isNot(MIToken::NamedIRValue) && Token.isNot(MIToken::IRValue) &&
1672 Token.isNot(MIToken::GlobalValue) &&
Alex Lorenzc1136ef32015-08-21 21:54:12 +00001673 Token.isNot(MIToken::NamedGlobalValue) &&
1674 Token.isNot(MIToken::QuotedIRValue))
Alex Lorenz91097a32015-08-12 20:33:26 +00001675 return error("expected an IR value reference");
Alex Lorenz36593ac2015-08-19 23:27:07 +00001676 const Value *V = nullptr;
Alex Lorenz91097a32015-08-12 20:33:26 +00001677 if (parseIRValue(V))
1678 return true;
1679 if (!V->getType()->isPointerTy())
1680 return error("expected a pointer IR value");
1681 lex();
1682 int64_t Offset = 0;
1683 if (parseOffset(Offset))
1684 return true;
1685 Dest = MachinePointerInfo(V, Offset);
1686 return false;
1687}
1688
Alex Lorenz4af7e612015-08-03 23:08:19 +00001689bool MIParser::parseMachineMemoryOperand(MachineMemOperand *&Dest) {
1690 if (expectAndConsume(MIToken::lparen))
1691 return true;
Alex Lorenza518b792015-08-04 00:24:45 +00001692 unsigned Flags = 0;
1693 while (Token.isMemoryOperandFlag()) {
1694 if (parseMemoryOperandFlag(Flags))
1695 return true;
1696 }
Alex Lorenz4af7e612015-08-03 23:08:19 +00001697 if (Token.isNot(MIToken::Identifier) ||
1698 (Token.stringValue() != "load" && Token.stringValue() != "store"))
1699 return error("expected 'load' or 'store' memory operation");
Alex Lorenz4af7e612015-08-03 23:08:19 +00001700 if (Token.stringValue() == "load")
1701 Flags |= MachineMemOperand::MOLoad;
1702 else
1703 Flags |= MachineMemOperand::MOStore;
1704 lex();
1705
1706 if (Token.isNot(MIToken::IntegerLiteral))
1707 return error("expected the size integer literal after memory operation");
1708 uint64_t Size;
1709 if (getUint64(Size))
1710 return true;
1711 lex();
1712
1713 const char *Word = Flags & MachineMemOperand::MOLoad ? "from" : "into";
1714 if (Token.isNot(MIToken::Identifier) || Token.stringValue() != Word)
1715 return error(Twine("expected '") + Word + "'");
1716 lex();
1717
Alex Lorenz91097a32015-08-12 20:33:26 +00001718 MachinePointerInfo Ptr = MachinePointerInfo();
1719 if (parseMachinePointerInfo(Ptr))
Alex Lorenz83127732015-08-07 20:26:52 +00001720 return true;
Alex Lorenz61420f72015-08-07 20:48:30 +00001721 unsigned BaseAlignment = Size;
Alex Lorenza617c912015-08-17 22:05:15 +00001722 AAMDNodes AAInfo;
Alex Lorenzeb625682015-08-17 22:09:52 +00001723 MDNode *Range = nullptr;
Alex Lorenza617c912015-08-17 22:05:15 +00001724 while (consumeIfPresent(MIToken::comma)) {
1725 switch (Token.kind()) {
1726 case MIToken::kw_align:
1727 if (parseAlignment(BaseAlignment))
1728 return true;
1729 break;
1730 case MIToken::md_tbaa:
1731 lex();
1732 if (parseMDNode(AAInfo.TBAA))
1733 return true;
1734 break;
Alex Lorenza16f6242015-08-17 22:06:40 +00001735 case MIToken::md_alias_scope:
1736 lex();
1737 if (parseMDNode(AAInfo.Scope))
1738 return true;
1739 break;
Alex Lorenz03e940d2015-08-17 22:08:02 +00001740 case MIToken::md_noalias:
1741 lex();
1742 if (parseMDNode(AAInfo.NoAlias))
1743 return true;
1744 break;
Alex Lorenzeb625682015-08-17 22:09:52 +00001745 case MIToken::md_range:
1746 lex();
1747 if (parseMDNode(Range))
1748 return true;
1749 break;
Alex Lorenza617c912015-08-17 22:05:15 +00001750 // TODO: Report an error on duplicate metadata nodes.
1751 default:
Alex Lorenzeb625682015-08-17 22:09:52 +00001752 return error("expected 'align' or '!tbaa' or '!alias.scope' or "
1753 "'!noalias' or '!range'");
Alex Lorenza617c912015-08-17 22:05:15 +00001754 }
Alex Lorenz61420f72015-08-07 20:48:30 +00001755 }
Alex Lorenz4af7e612015-08-03 23:08:19 +00001756 if (expectAndConsume(MIToken::rparen))
1757 return true;
Alex Lorenzeb625682015-08-17 22:09:52 +00001758 Dest =
1759 MF.getMachineMemOperand(Ptr, Flags, Size, BaseAlignment, AAInfo, Range);
Alex Lorenz4af7e612015-08-03 23:08:19 +00001760 return false;
1761}
1762
Alex Lorenz8e0a1b42015-06-22 17:02:30 +00001763void MIParser::initNames2InstrOpCodes() {
1764 if (!Names2InstrOpCodes.empty())
1765 return;
1766 const auto *TII = MF.getSubtarget().getInstrInfo();
1767 assert(TII && "Expected target instruction info");
1768 for (unsigned I = 0, E = TII->getNumOpcodes(); I < E; ++I)
1769 Names2InstrOpCodes.insert(std::make_pair(StringRef(TII->getName(I)), I));
1770}
1771
1772bool MIParser::parseInstrName(StringRef InstrName, unsigned &OpCode) {
1773 initNames2InstrOpCodes();
1774 auto InstrInfo = Names2InstrOpCodes.find(InstrName);
1775 if (InstrInfo == Names2InstrOpCodes.end())
1776 return true;
1777 OpCode = InstrInfo->getValue();
1778 return false;
1779}
1780
Alex Lorenzf3db51de2015-06-23 16:35:26 +00001781void MIParser::initNames2Regs() {
1782 if (!Names2Regs.empty())
1783 return;
Alex Lorenz12b554e2015-06-24 17:34:58 +00001784 // The '%noreg' register is the register 0.
1785 Names2Regs.insert(std::make_pair("noreg", 0));
Alex Lorenzf3db51de2015-06-23 16:35:26 +00001786 const auto *TRI = MF.getSubtarget().getRegisterInfo();
1787 assert(TRI && "Expected target register info");
1788 for (unsigned I = 0, E = TRI->getNumRegs(); I < E; ++I) {
1789 bool WasInserted =
1790 Names2Regs.insert(std::make_pair(StringRef(TRI->getName(I)).lower(), I))
1791 .second;
1792 (void)WasInserted;
1793 assert(WasInserted && "Expected registers to be unique case-insensitively");
1794 }
1795}
1796
1797bool MIParser::getRegisterByName(StringRef RegName, unsigned &Reg) {
1798 initNames2Regs();
1799 auto RegInfo = Names2Regs.find(RegName);
1800 if (RegInfo == Names2Regs.end())
1801 return true;
1802 Reg = RegInfo->getValue();
1803 return false;
1804}
1805
Alex Lorenz8f6f4282015-06-29 16:57:06 +00001806void MIParser::initNames2RegMasks() {
1807 if (!Names2RegMasks.empty())
1808 return;
1809 const auto *TRI = MF.getSubtarget().getRegisterInfo();
1810 assert(TRI && "Expected target register info");
1811 ArrayRef<const uint32_t *> RegMasks = TRI->getRegMasks();
1812 ArrayRef<const char *> RegMaskNames = TRI->getRegMaskNames();
1813 assert(RegMasks.size() == RegMaskNames.size());
1814 for (size_t I = 0, E = RegMasks.size(); I < E; ++I)
1815 Names2RegMasks.insert(
1816 std::make_pair(StringRef(RegMaskNames[I]).lower(), RegMasks[I]));
1817}
1818
1819const uint32_t *MIParser::getRegMask(StringRef Identifier) {
1820 initNames2RegMasks();
1821 auto RegMaskInfo = Names2RegMasks.find(Identifier);
1822 if (RegMaskInfo == Names2RegMasks.end())
1823 return nullptr;
1824 return RegMaskInfo->getValue();
1825}
1826
Alex Lorenz2eacca82015-07-13 23:24:34 +00001827void MIParser::initNames2SubRegIndices() {
1828 if (!Names2SubRegIndices.empty())
1829 return;
1830 const TargetRegisterInfo *TRI = MF.getSubtarget().getRegisterInfo();
1831 for (unsigned I = 1, E = TRI->getNumSubRegIndices(); I < E; ++I)
1832 Names2SubRegIndices.insert(
1833 std::make_pair(StringRef(TRI->getSubRegIndexName(I)).lower(), I));
1834}
1835
1836unsigned MIParser::getSubRegIndex(StringRef Name) {
1837 initNames2SubRegIndices();
1838 auto SubRegInfo = Names2SubRegIndices.find(Name);
1839 if (SubRegInfo == Names2SubRegIndices.end())
1840 return 0;
1841 return SubRegInfo->getValue();
1842}
1843
Alex Lorenzcba8c5f2015-08-06 23:57:04 +00001844static void initSlots2BasicBlocks(
1845 const Function &F,
1846 DenseMap<unsigned, const BasicBlock *> &Slots2BasicBlocks) {
1847 ModuleSlotTracker MST(F.getParent(), /*ShouldInitializeAllMetadata=*/false);
Alex Lorenz8a1915b2015-07-27 22:42:41 +00001848 MST.incorporateFunction(F);
1849 for (auto &BB : F) {
1850 if (BB.hasName())
1851 continue;
1852 int Slot = MST.getLocalSlot(&BB);
1853 if (Slot == -1)
1854 continue;
1855 Slots2BasicBlocks.insert(std::make_pair(unsigned(Slot), &BB));
1856 }
1857}
1858
Alex Lorenzcba8c5f2015-08-06 23:57:04 +00001859static const BasicBlock *getIRBlockFromSlot(
1860 unsigned Slot,
1861 const DenseMap<unsigned, const BasicBlock *> &Slots2BasicBlocks) {
Alex Lorenz8a1915b2015-07-27 22:42:41 +00001862 auto BlockInfo = Slots2BasicBlocks.find(Slot);
1863 if (BlockInfo == Slots2BasicBlocks.end())
1864 return nullptr;
1865 return BlockInfo->second;
1866}
1867
Alex Lorenzcba8c5f2015-08-06 23:57:04 +00001868const BasicBlock *MIParser::getIRBlock(unsigned Slot) {
1869 if (Slots2BasicBlocks.empty())
1870 initSlots2BasicBlocks(*MF.getFunction(), Slots2BasicBlocks);
1871 return getIRBlockFromSlot(Slot, Slots2BasicBlocks);
1872}
1873
1874const BasicBlock *MIParser::getIRBlock(unsigned Slot, const Function &F) {
1875 if (&F == MF.getFunction())
1876 return getIRBlock(Slot);
1877 DenseMap<unsigned, const BasicBlock *> CustomSlots2BasicBlocks;
1878 initSlots2BasicBlocks(F, CustomSlots2BasicBlocks);
1879 return getIRBlockFromSlot(Slot, CustomSlots2BasicBlocks);
1880}
1881
Alex Lorenzdd13be02015-08-19 23:31:05 +00001882static void mapValueToSlot(const Value *V, ModuleSlotTracker &MST,
1883 DenseMap<unsigned, const Value *> &Slots2Values) {
1884 int Slot = MST.getLocalSlot(V);
1885 if (Slot == -1)
1886 return;
1887 Slots2Values.insert(std::make_pair(unsigned(Slot), V));
1888}
1889
1890/// Creates the mapping from slot numbers to function's unnamed IR values.
1891static void initSlots2Values(const Function &F,
1892 DenseMap<unsigned, const Value *> &Slots2Values) {
1893 ModuleSlotTracker MST(F.getParent(), /*ShouldInitializeAllMetadata=*/false);
1894 MST.incorporateFunction(F);
1895 for (const auto &Arg : F.args())
1896 mapValueToSlot(&Arg, MST, Slots2Values);
1897 for (const auto &BB : F) {
1898 mapValueToSlot(&BB, MST, Slots2Values);
1899 for (const auto &I : BB)
1900 mapValueToSlot(&I, MST, Slots2Values);
1901 }
1902}
1903
1904const Value *MIParser::getIRValue(unsigned Slot) {
1905 if (Slots2Values.empty())
1906 initSlots2Values(*MF.getFunction(), Slots2Values);
1907 auto ValueInfo = Slots2Values.find(Slot);
1908 if (ValueInfo == Slots2Values.end())
1909 return nullptr;
1910 return ValueInfo->second;
1911}
1912
Alex Lorenzef5c1962015-07-28 23:02:45 +00001913void MIParser::initNames2TargetIndices() {
1914 if (!Names2TargetIndices.empty())
1915 return;
1916 const auto *TII = MF.getSubtarget().getInstrInfo();
1917 assert(TII && "Expected target instruction info");
1918 auto Indices = TII->getSerializableTargetIndices();
1919 for (const auto &I : Indices)
1920 Names2TargetIndices.insert(std::make_pair(StringRef(I.second), I.first));
1921}
1922
1923bool MIParser::getTargetIndex(StringRef Name, int &Index) {
1924 initNames2TargetIndices();
1925 auto IndexInfo = Names2TargetIndices.find(Name);
1926 if (IndexInfo == Names2TargetIndices.end())
1927 return true;
1928 Index = IndexInfo->second;
1929 return false;
1930}
1931
Alex Lorenz49873a82015-08-06 00:44:07 +00001932void MIParser::initNames2DirectTargetFlags() {
1933 if (!Names2DirectTargetFlags.empty())
1934 return;
1935 const auto *TII = MF.getSubtarget().getInstrInfo();
1936 assert(TII && "Expected target instruction info");
1937 auto Flags = TII->getSerializableDirectMachineOperandTargetFlags();
1938 for (const auto &I : Flags)
1939 Names2DirectTargetFlags.insert(
1940 std::make_pair(StringRef(I.second), I.first));
1941}
1942
1943bool MIParser::getDirectTargetFlag(StringRef Name, unsigned &Flag) {
1944 initNames2DirectTargetFlags();
1945 auto FlagInfo = Names2DirectTargetFlags.find(Name);
1946 if (FlagInfo == Names2DirectTargetFlags.end())
1947 return true;
1948 Flag = FlagInfo->second;
1949 return false;
1950}
1951
Alex Lorenzf3630112015-08-18 22:52:15 +00001952void MIParser::initNames2BitmaskTargetFlags() {
1953 if (!Names2BitmaskTargetFlags.empty())
1954 return;
1955 const auto *TII = MF.getSubtarget().getInstrInfo();
1956 assert(TII && "Expected target instruction info");
1957 auto Flags = TII->getSerializableBitmaskMachineOperandTargetFlags();
1958 for (const auto &I : Flags)
1959 Names2BitmaskTargetFlags.insert(
1960 std::make_pair(StringRef(I.second), I.first));
1961}
1962
1963bool MIParser::getBitmaskTargetFlag(StringRef Name, unsigned &Flag) {
1964 initNames2BitmaskTargetFlags();
1965 auto FlagInfo = Names2BitmaskTargetFlags.find(Name);
1966 if (FlagInfo == Names2BitmaskTargetFlags.end())
1967 return true;
1968 Flag = FlagInfo->second;
1969 return false;
1970}
1971
Alex Lorenz5022f6b2015-08-13 23:10:16 +00001972bool llvm::parseMachineBasicBlockDefinitions(MachineFunction &MF, StringRef Src,
1973 PerFunctionMIParsingState &PFS,
1974 const SlotMapping &IRSlots,
1975 SMDiagnostic &Error) {
1976 SourceMgr SM;
1977 SM.AddNewSourceBuffer(
1978 MemoryBuffer::getMemBuffer(Src, "", /*RequiresNullTerminator=*/false),
1979 SMLoc());
1980 return MIParser(SM, MF, Error, Src, PFS, IRSlots)
1981 .parseBasicBlockDefinitions(PFS.MBBSlots);
1982}
1983
1984bool llvm::parseMachineInstructions(MachineFunction &MF, StringRef Src,
1985 const PerFunctionMIParsingState &PFS,
1986 const SlotMapping &IRSlots,
1987 SMDiagnostic &Error) {
1988 SourceMgr SM;
1989 SM.AddNewSourceBuffer(
1990 MemoryBuffer::getMemBuffer(Src, "", /*RequiresNullTerminator=*/false),
1991 SMLoc());
1992 return MIParser(SM, MF, Error, Src, PFS, IRSlots).parseBasicBlocks();
Alex Lorenz8e0a1b42015-06-22 17:02:30 +00001993}
Alex Lorenzf09df002015-06-30 18:16:42 +00001994
Alex Lorenz7a503fa2015-07-07 17:46:43 +00001995bool llvm::parseMBBReference(MachineBasicBlock *&MBB, SourceMgr &SM,
1996 MachineFunction &MF, StringRef Src,
1997 const PerFunctionMIParsingState &PFS,
1998 const SlotMapping &IRSlots, SMDiagnostic &Error) {
Alex Lorenz1ea60892015-07-27 20:29:27 +00001999 return MIParser(SM, MF, Error, Src, PFS, IRSlots).parseStandaloneMBB(MBB);
Alex Lorenzf09df002015-06-30 18:16:42 +00002000}
Alex Lorenz9fab3702015-07-14 21:24:41 +00002001
2002bool llvm::parseNamedRegisterReference(unsigned &Reg, SourceMgr &SM,
2003 MachineFunction &MF, StringRef Src,
2004 const PerFunctionMIParsingState &PFS,
2005 const SlotMapping &IRSlots,
2006 SMDiagnostic &Error) {
Alex Lorenz1ea60892015-07-27 20:29:27 +00002007 return MIParser(SM, MF, Error, Src, PFS, IRSlots)
2008 .parseStandaloneNamedRegister(Reg);
Alex Lorenz9fab3702015-07-14 21:24:41 +00002009}
Alex Lorenz12045a42015-07-27 17:42:45 +00002010
2011bool llvm::parseVirtualRegisterReference(unsigned &Reg, SourceMgr &SM,
2012 MachineFunction &MF, StringRef Src,
2013 const PerFunctionMIParsingState &PFS,
2014 const SlotMapping &IRSlots,
2015 SMDiagnostic &Error) {
2016 return MIParser(SM, MF, Error, Src, PFS, IRSlots)
2017 .parseStandaloneVirtualRegister(Reg);
2018}
Alex Lorenza314d812015-08-18 22:26:26 +00002019
2020bool llvm::parseStackObjectReference(int &FI, SourceMgr &SM,
2021 MachineFunction &MF, StringRef Src,
2022 const PerFunctionMIParsingState &PFS,
2023 const SlotMapping &IRSlots,
2024 SMDiagnostic &Error) {
2025 return MIParser(SM, MF, Error, Src, PFS, IRSlots)
2026 .parseStandaloneStackObject(FI);
2027}
Alex Lorenzdf9e3c62015-08-19 00:13:25 +00002028
2029bool llvm::parseMDNode(MDNode *&Node, SourceMgr &SM, MachineFunction &MF,
2030 StringRef Src, const PerFunctionMIParsingState &PFS,
2031 const SlotMapping &IRSlots, SMDiagnostic &Error) {
2032 return MIParser(SM, MF, Error, Src, PFS, IRSlots).parseStandaloneMDNode(Node);
2033}