14ba319b5SDimitry Andric //==- WebAssemblyAsmParser.cpp - Assembler for WebAssembly -*- C++ -*-==//
24ba319b5SDimitry Andric //
34ba319b5SDimitry Andric //                     The LLVM Compiler Infrastructure
44ba319b5SDimitry Andric //
54ba319b5SDimitry Andric // This file is distributed under the University of Illinois Open Source
64ba319b5SDimitry Andric // License. See LICENSE.TXT for details.
74ba319b5SDimitry Andric //
84ba319b5SDimitry Andric //===----------------------------------------------------------------------===//
94ba319b5SDimitry Andric ///
104ba319b5SDimitry Andric /// \file
114ba319b5SDimitry Andric /// This file is part of the WebAssembly Assembler.
124ba319b5SDimitry Andric ///
134ba319b5SDimitry Andric /// It contains code to translate a parsed .s file into MCInsts.
144ba319b5SDimitry Andric ///
154ba319b5SDimitry Andric //===----------------------------------------------------------------------===//
164ba319b5SDimitry Andric 
174ba319b5SDimitry Andric #include "MCTargetDesc/WebAssemblyMCTargetDesc.h"
184ba319b5SDimitry Andric #include "MCTargetDesc/WebAssemblyTargetStreamer.h"
194ba319b5SDimitry Andric #include "WebAssembly.h"
204ba319b5SDimitry Andric #include "llvm/MC/MCContext.h"
214ba319b5SDimitry Andric #include "llvm/MC/MCInst.h"
224ba319b5SDimitry Andric #include "llvm/MC/MCInstrInfo.h"
23*b5893f02SDimitry Andric #include "llvm/MC/MCParser/MCParsedAsmOperand.h"
24*b5893f02SDimitry Andric #include "llvm/MC/MCParser/MCTargetAsmParser.h"
25*b5893f02SDimitry Andric #include "llvm/MC/MCStreamer.h"
264ba319b5SDimitry Andric #include "llvm/MC/MCSubtargetInfo.h"
274ba319b5SDimitry Andric #include "llvm/MC/MCSymbol.h"
28*b5893f02SDimitry Andric #include "llvm/MC/MCSymbolWasm.h"
294ba319b5SDimitry Andric #include "llvm/Support/Endian.h"
304ba319b5SDimitry Andric #include "llvm/Support/TargetRegistry.h"
314ba319b5SDimitry Andric 
324ba319b5SDimitry Andric using namespace llvm;
334ba319b5SDimitry Andric 
344ba319b5SDimitry Andric #define DEBUG_TYPE "wasm-asm-parser"
354ba319b5SDimitry Andric 
364ba319b5SDimitry Andric namespace {
374ba319b5SDimitry Andric 
384ba319b5SDimitry Andric /// WebAssemblyOperand - Instances of this class represent the operands in a
394ba319b5SDimitry Andric /// parsed WASM machine instruction.
404ba319b5SDimitry Andric struct WebAssemblyOperand : public MCParsedAsmOperand {
41*b5893f02SDimitry Andric   enum KindTy { Token, Integer, Float, Symbol, BrList } Kind;
424ba319b5SDimitry Andric 
434ba319b5SDimitry Andric   SMLoc StartLoc, EndLoc;
444ba319b5SDimitry Andric 
454ba319b5SDimitry Andric   struct TokOp {
464ba319b5SDimitry Andric     StringRef Tok;
474ba319b5SDimitry Andric   };
484ba319b5SDimitry Andric 
494ba319b5SDimitry Andric   struct IntOp {
504ba319b5SDimitry Andric     int64_t Val;
514ba319b5SDimitry Andric   };
524ba319b5SDimitry Andric 
534ba319b5SDimitry Andric   struct FltOp {
544ba319b5SDimitry Andric     double Val;
554ba319b5SDimitry Andric   };
564ba319b5SDimitry Andric 
574ba319b5SDimitry Andric   struct SymOp {
584ba319b5SDimitry Andric     const MCExpr *Exp;
594ba319b5SDimitry Andric   };
604ba319b5SDimitry Andric 
61*b5893f02SDimitry Andric   struct BrLOp {
62*b5893f02SDimitry Andric     std::vector<unsigned> List;
63*b5893f02SDimitry Andric   };
64*b5893f02SDimitry Andric 
654ba319b5SDimitry Andric   union {
664ba319b5SDimitry Andric     struct TokOp Tok;
674ba319b5SDimitry Andric     struct IntOp Int;
684ba319b5SDimitry Andric     struct FltOp Flt;
694ba319b5SDimitry Andric     struct SymOp Sym;
70*b5893f02SDimitry Andric     struct BrLOp BrL;
714ba319b5SDimitry Andric   };
724ba319b5SDimitry Andric 
WebAssemblyOperand__anon1565d9e10111::WebAssemblyOperand734ba319b5SDimitry Andric   WebAssemblyOperand(KindTy K, SMLoc Start, SMLoc End, TokOp T)
744ba319b5SDimitry Andric       : Kind(K), StartLoc(Start), EndLoc(End), Tok(T) {}
WebAssemblyOperand__anon1565d9e10111::WebAssemblyOperand754ba319b5SDimitry Andric   WebAssemblyOperand(KindTy K, SMLoc Start, SMLoc End, IntOp I)
764ba319b5SDimitry Andric       : Kind(K), StartLoc(Start), EndLoc(End), Int(I) {}
WebAssemblyOperand__anon1565d9e10111::WebAssemblyOperand774ba319b5SDimitry Andric   WebAssemblyOperand(KindTy K, SMLoc Start, SMLoc End, FltOp F)
784ba319b5SDimitry Andric       : Kind(K), StartLoc(Start), EndLoc(End), Flt(F) {}
WebAssemblyOperand__anon1565d9e10111::WebAssemblyOperand794ba319b5SDimitry Andric   WebAssemblyOperand(KindTy K, SMLoc Start, SMLoc End, SymOp S)
804ba319b5SDimitry Andric       : Kind(K), StartLoc(Start), EndLoc(End), Sym(S) {}
WebAssemblyOperand__anon1565d9e10111::WebAssemblyOperand81*b5893f02SDimitry Andric   WebAssemblyOperand(KindTy K, SMLoc Start, SMLoc End)
82*b5893f02SDimitry Andric       : Kind(K), StartLoc(Start), EndLoc(End), BrL() {}
83*b5893f02SDimitry Andric 
~WebAssemblyOperand__anon1565d9e10111::WebAssemblyOperand84*b5893f02SDimitry Andric   ~WebAssemblyOperand() {
85*b5893f02SDimitry Andric     if (isBrList())
86*b5893f02SDimitry Andric       BrL.~BrLOp();
87*b5893f02SDimitry Andric   }
884ba319b5SDimitry Andric 
isToken__anon1565d9e10111::WebAssemblyOperand894ba319b5SDimitry Andric   bool isToken() const override { return Kind == Token; }
isImm__anon1565d9e10111::WebAssemblyOperand90*b5893f02SDimitry Andric   bool isImm() const override {
91*b5893f02SDimitry Andric     return Kind == Integer || Kind == Float || Kind == Symbol;
92*b5893f02SDimitry Andric   }
isMem__anon1565d9e10111::WebAssemblyOperand934ba319b5SDimitry Andric   bool isMem() const override { return false; }
isReg__anon1565d9e10111::WebAssemblyOperand94*b5893f02SDimitry Andric   bool isReg() const override { return false; }
isBrList__anon1565d9e10111::WebAssemblyOperand95*b5893f02SDimitry Andric   bool isBrList() const { return Kind == BrList; }
964ba319b5SDimitry Andric 
getReg__anon1565d9e10111::WebAssemblyOperand974ba319b5SDimitry Andric   unsigned getReg() const override {
98*b5893f02SDimitry Andric     llvm_unreachable("Assembly inspects a register operand");
99*b5893f02SDimitry Andric     return 0;
1004ba319b5SDimitry Andric   }
1014ba319b5SDimitry Andric 
getToken__anon1565d9e10111::WebAssemblyOperand1024ba319b5SDimitry Andric   StringRef getToken() const {
1034ba319b5SDimitry Andric     assert(isToken());
1044ba319b5SDimitry Andric     return Tok.Tok;
1054ba319b5SDimitry Andric   }
1064ba319b5SDimitry Andric 
getStartLoc__anon1565d9e10111::WebAssemblyOperand1074ba319b5SDimitry Andric   SMLoc getStartLoc() const override { return StartLoc; }
getEndLoc__anon1565d9e10111::WebAssemblyOperand1084ba319b5SDimitry Andric   SMLoc getEndLoc() const override { return EndLoc; }
1094ba319b5SDimitry Andric 
addRegOperands__anon1565d9e10111::WebAssemblyOperand110*b5893f02SDimitry Andric   void addRegOperands(MCInst &, unsigned) const {
111*b5893f02SDimitry Andric     // Required by the assembly matcher.
112*b5893f02SDimitry Andric     llvm_unreachable("Assembly matcher creates register operands");
1134ba319b5SDimitry Andric   }
1144ba319b5SDimitry Andric 
addImmOperands__anon1565d9e10111::WebAssemblyOperand1154ba319b5SDimitry Andric   void addImmOperands(MCInst &Inst, unsigned N) const {
1164ba319b5SDimitry Andric     assert(N == 1 && "Invalid number of operands!");
1174ba319b5SDimitry Andric     if (Kind == Integer)
1184ba319b5SDimitry Andric       Inst.addOperand(MCOperand::createImm(Int.Val));
1194ba319b5SDimitry Andric     else if (Kind == Float)
1204ba319b5SDimitry Andric       Inst.addOperand(MCOperand::createFPImm(Flt.Val));
1214ba319b5SDimitry Andric     else if (Kind == Symbol)
1224ba319b5SDimitry Andric       Inst.addOperand(MCOperand::createExpr(Sym.Exp));
1234ba319b5SDimitry Andric     else
1244ba319b5SDimitry Andric       llvm_unreachable("Should be immediate or symbol!");
1254ba319b5SDimitry Andric   }
1264ba319b5SDimitry Andric 
addBrListOperands__anon1565d9e10111::WebAssemblyOperand127*b5893f02SDimitry Andric   void addBrListOperands(MCInst &Inst, unsigned N) const {
128*b5893f02SDimitry Andric     assert(N == 1 && isBrList() && "Invalid BrList!");
129*b5893f02SDimitry Andric     for (auto Br : BrL.List)
130*b5893f02SDimitry Andric       Inst.addOperand(MCOperand::createImm(Br));
131*b5893f02SDimitry Andric   }
132*b5893f02SDimitry Andric 
print__anon1565d9e10111::WebAssemblyOperand1334ba319b5SDimitry Andric   void print(raw_ostream &OS) const override {
1344ba319b5SDimitry Andric     switch (Kind) {
1354ba319b5SDimitry Andric     case Token:
1364ba319b5SDimitry Andric       OS << "Tok:" << Tok.Tok;
1374ba319b5SDimitry Andric       break;
1384ba319b5SDimitry Andric     case Integer:
1394ba319b5SDimitry Andric       OS << "Int:" << Int.Val;
1404ba319b5SDimitry Andric       break;
1414ba319b5SDimitry Andric     case Float:
1424ba319b5SDimitry Andric       OS << "Flt:" << Flt.Val;
1434ba319b5SDimitry Andric       break;
1444ba319b5SDimitry Andric     case Symbol:
1454ba319b5SDimitry Andric       OS << "Sym:" << Sym.Exp;
1464ba319b5SDimitry Andric       break;
147*b5893f02SDimitry Andric     case BrList:
148*b5893f02SDimitry Andric       OS << "BrList:" << BrL.List.size();
149*b5893f02SDimitry Andric       break;
1504ba319b5SDimitry Andric     }
1514ba319b5SDimitry Andric   }
1524ba319b5SDimitry Andric };
1534ba319b5SDimitry Andric 
1544ba319b5SDimitry Andric class WebAssemblyAsmParser final : public MCTargetAsmParser {
1554ba319b5SDimitry Andric   MCAsmParser &Parser;
1564ba319b5SDimitry Andric   MCAsmLexer &Lexer;
157*b5893f02SDimitry Andric 
158*b5893f02SDimitry Andric   // Much like WebAssemblyAsmPrinter in the backend, we have to own these.
159*b5893f02SDimitry Andric   std::vector<std::unique_ptr<wasm::WasmSignature>> Signatures;
160*b5893f02SDimitry Andric 
161*b5893f02SDimitry Andric   // Order of labels, directives and instructions in a .s file have no
162*b5893f02SDimitry Andric   // syntactical enforcement. This class is a callback from the actual parser,
163*b5893f02SDimitry Andric   // and yet we have to be feeding data to the streamer in a very particular
164*b5893f02SDimitry Andric   // order to ensure a correct binary encoding that matches the regular backend
165*b5893f02SDimitry Andric   // (the streamer does not enforce this). This "state machine" enum helps
166*b5893f02SDimitry Andric   // guarantee that correct order.
167*b5893f02SDimitry Andric   enum ParserState {
168*b5893f02SDimitry Andric     FileStart,
169*b5893f02SDimitry Andric     Label,
170*b5893f02SDimitry Andric     FunctionStart,
171*b5893f02SDimitry Andric     FunctionLocals,
172*b5893f02SDimitry Andric     Instructions,
173*b5893f02SDimitry Andric   } CurrentState = FileStart;
174*b5893f02SDimitry Andric 
175*b5893f02SDimitry Andric   // For ensuring blocks are properly nested.
176*b5893f02SDimitry Andric   enum NestingType {
177*b5893f02SDimitry Andric     Function,
178*b5893f02SDimitry Andric     Block,
179*b5893f02SDimitry Andric     Loop,
180*b5893f02SDimitry Andric     Try,
181*b5893f02SDimitry Andric     If,
182*b5893f02SDimitry Andric     Else,
183*b5893f02SDimitry Andric     Undefined,
184*b5893f02SDimitry Andric   };
185*b5893f02SDimitry Andric   std::vector<NestingType> NestingStack;
186*b5893f02SDimitry Andric 
187*b5893f02SDimitry Andric   // We track this to see if a .functype following a label is the same,
188*b5893f02SDimitry Andric   // as this is how we recognize the start of a function.
189*b5893f02SDimitry Andric   MCSymbol *LastLabel = nullptr;
1904ba319b5SDimitry Andric 
1914ba319b5SDimitry Andric public:
WebAssemblyAsmParser(const MCSubtargetInfo & STI,MCAsmParser & Parser,const MCInstrInfo & MII,const MCTargetOptions & Options)192*b5893f02SDimitry Andric   WebAssemblyAsmParser(const MCSubtargetInfo &STI, MCAsmParser &Parser,
193*b5893f02SDimitry Andric                        const MCInstrInfo &MII, const MCTargetOptions &Options)
194*b5893f02SDimitry Andric       : MCTargetAsmParser(Options, STI, MII), Parser(Parser),
195*b5893f02SDimitry Andric         Lexer(Parser.getLexer()) {
196*b5893f02SDimitry Andric     setAvailableFeatures(ComputeAvailableFeatures(STI.getFeatureBits()));
1974ba319b5SDimitry Andric   }
1984ba319b5SDimitry Andric 
1994ba319b5SDimitry Andric #define GET_ASSEMBLER_HEADER
2004ba319b5SDimitry Andric #include "WebAssemblyGenAsmMatcher.inc"
2014ba319b5SDimitry Andric 
2024ba319b5SDimitry Andric   // TODO: This is required to be implemented, but appears unused.
ParseRegister(unsigned &,SMLoc &,SMLoc &)2034ba319b5SDimitry Andric   bool ParseRegister(unsigned & /*RegNo*/, SMLoc & /*StartLoc*/,
2044ba319b5SDimitry Andric                      SMLoc & /*EndLoc*/) override {
2054ba319b5SDimitry Andric     llvm_unreachable("ParseRegister is not implemented.");
2064ba319b5SDimitry Andric   }
2074ba319b5SDimitry Andric 
error(const Twine & Msg,const AsmToken & Tok)208*b5893f02SDimitry Andric   bool error(const Twine &Msg, const AsmToken &Tok) {
209*b5893f02SDimitry Andric     return Parser.Error(Tok.getLoc(), Msg + Tok.getString());
2104ba319b5SDimitry Andric   }
2114ba319b5SDimitry Andric 
error(const Twine & Msg)212*b5893f02SDimitry Andric   bool error(const Twine &Msg) {
213*b5893f02SDimitry Andric     return Parser.Error(Lexer.getTok().getLoc(), Msg);
2144ba319b5SDimitry Andric   }
2154ba319b5SDimitry Andric 
addSignature(std::unique_ptr<wasm::WasmSignature> && Sig)216*b5893f02SDimitry Andric   void addSignature(std::unique_ptr<wasm::WasmSignature> &&Sig) {
217*b5893f02SDimitry Andric     Signatures.push_back(std::move(Sig));
218*b5893f02SDimitry Andric   }
219*b5893f02SDimitry Andric 
nestingString(NestingType NT)220*b5893f02SDimitry Andric   std::pair<StringRef, StringRef> nestingString(NestingType NT) {
221*b5893f02SDimitry Andric     switch (NT) {
222*b5893f02SDimitry Andric     case Function:
223*b5893f02SDimitry Andric       return {"function", "end_function"};
224*b5893f02SDimitry Andric     case Block:
225*b5893f02SDimitry Andric       return {"block", "end_block"};
226*b5893f02SDimitry Andric     case Loop:
227*b5893f02SDimitry Andric       return {"loop", "end_loop"};
228*b5893f02SDimitry Andric     case Try:
229*b5893f02SDimitry Andric       return {"try", "end_try"};
230*b5893f02SDimitry Andric     case If:
231*b5893f02SDimitry Andric       return {"if", "end_if"};
232*b5893f02SDimitry Andric     case Else:
233*b5893f02SDimitry Andric       return {"else", "end_if"};
234*b5893f02SDimitry Andric     default:
235*b5893f02SDimitry Andric       llvm_unreachable("unknown NestingType");
236*b5893f02SDimitry Andric     }
237*b5893f02SDimitry Andric   }
238*b5893f02SDimitry Andric 
push(NestingType NT)239*b5893f02SDimitry Andric   void push(NestingType NT) { NestingStack.push_back(NT); }
240*b5893f02SDimitry Andric 
pop(StringRef Ins,NestingType NT1,NestingType NT2=Undefined)241*b5893f02SDimitry Andric   bool pop(StringRef Ins, NestingType NT1, NestingType NT2 = Undefined) {
242*b5893f02SDimitry Andric     if (NestingStack.empty())
243*b5893f02SDimitry Andric       return error(Twine("End of block construct with no start: ") + Ins);
244*b5893f02SDimitry Andric     auto Top = NestingStack.back();
245*b5893f02SDimitry Andric     if (Top != NT1 && Top != NT2)
246*b5893f02SDimitry Andric       return error(Twine("Block construct type mismatch, expected: ") +
247*b5893f02SDimitry Andric                    nestingString(Top).second + ", instead got: " + Ins);
248*b5893f02SDimitry Andric     NestingStack.pop_back();
249*b5893f02SDimitry Andric     return false;
250*b5893f02SDimitry Andric   }
251*b5893f02SDimitry Andric 
ensureEmptyNestingStack()252*b5893f02SDimitry Andric   bool ensureEmptyNestingStack() {
253*b5893f02SDimitry Andric     auto err = !NestingStack.empty();
254*b5893f02SDimitry Andric     while (!NestingStack.empty()) {
255*b5893f02SDimitry Andric       error(Twine("Unmatched block construct(s) at function end: ") +
256*b5893f02SDimitry Andric             nestingString(NestingStack.back()).first);
257*b5893f02SDimitry Andric       NestingStack.pop_back();
258*b5893f02SDimitry Andric     }
259*b5893f02SDimitry Andric     return err;
260*b5893f02SDimitry Andric   }
261*b5893f02SDimitry Andric 
isNext(AsmToken::TokenKind Kind)262*b5893f02SDimitry Andric   bool isNext(AsmToken::TokenKind Kind) {
263*b5893f02SDimitry Andric     auto Ok = Lexer.is(Kind);
264*b5893f02SDimitry Andric     if (Ok)
265*b5893f02SDimitry Andric       Parser.Lex();
266*b5893f02SDimitry Andric     return Ok;
267*b5893f02SDimitry Andric   }
268*b5893f02SDimitry Andric 
expect(AsmToken::TokenKind Kind,const char * KindName)269*b5893f02SDimitry Andric   bool expect(AsmToken::TokenKind Kind, const char *KindName) {
270*b5893f02SDimitry Andric     if (!isNext(Kind))
271*b5893f02SDimitry Andric       return error(std::string("Expected ") + KindName + ", instead got: ",
2724ba319b5SDimitry Andric                    Lexer.getTok());
2734ba319b5SDimitry Andric     return false;
2744ba319b5SDimitry Andric   }
2754ba319b5SDimitry Andric 
expectIdent()276*b5893f02SDimitry Andric   StringRef expectIdent() {
277*b5893f02SDimitry Andric     if (!Lexer.is(AsmToken::Identifier)) {
278*b5893f02SDimitry Andric       error("Expected identifier, got: ", Lexer.getTok());
279*b5893f02SDimitry Andric       return StringRef();
280*b5893f02SDimitry Andric     }
281*b5893f02SDimitry Andric     auto Name = Lexer.getTok().getString();
282*b5893f02SDimitry Andric     Parser.Lex();
283*b5893f02SDimitry Andric     return Name;
2844ba319b5SDimitry Andric   }
2854ba319b5SDimitry Andric 
parseType(const StringRef & Type)286*b5893f02SDimitry Andric   Optional<wasm::ValType> parseType(const StringRef &Type) {
287*b5893f02SDimitry Andric     // FIXME: can't use StringSwitch because wasm::ValType doesn't have a
288*b5893f02SDimitry Andric     // "invalid" value.
289*b5893f02SDimitry Andric     if (Type == "i32")
290*b5893f02SDimitry Andric       return wasm::ValType::I32;
291*b5893f02SDimitry Andric     if (Type == "i64")
292*b5893f02SDimitry Andric       return wasm::ValType::I64;
293*b5893f02SDimitry Andric     if (Type == "f32")
294*b5893f02SDimitry Andric       return wasm::ValType::F32;
295*b5893f02SDimitry Andric     if (Type == "f64")
296*b5893f02SDimitry Andric       return wasm::ValType::F64;
297*b5893f02SDimitry Andric     if (Type == "v128" || Type == "i8x16" || Type == "i16x8" ||
298*b5893f02SDimitry Andric         Type == "i32x4" || Type == "i64x2" || Type == "f32x4" ||
299*b5893f02SDimitry Andric         Type == "f64x2")
300*b5893f02SDimitry Andric       return wasm::ValType::V128;
301*b5893f02SDimitry Andric     return Optional<wasm::ValType>();
3024ba319b5SDimitry Andric   }
3034ba319b5SDimitry Andric 
parseBlockType(StringRef ID)304*b5893f02SDimitry Andric   WebAssembly::ExprType parseBlockType(StringRef ID) {
305*b5893f02SDimitry Andric     return StringSwitch<WebAssembly::ExprType>(ID)
306*b5893f02SDimitry Andric         .Case("i32", WebAssembly::ExprType::I32)
307*b5893f02SDimitry Andric         .Case("i64", WebAssembly::ExprType::I64)
308*b5893f02SDimitry Andric         .Case("f32", WebAssembly::ExprType::F32)
309*b5893f02SDimitry Andric         .Case("f64", WebAssembly::ExprType::F64)
310*b5893f02SDimitry Andric         .Case("v128", WebAssembly::ExprType::V128)
311*b5893f02SDimitry Andric         .Case("except_ref", WebAssembly::ExprType::ExceptRef)
312*b5893f02SDimitry Andric         .Case("void", WebAssembly::ExprType::Void)
313*b5893f02SDimitry Andric         .Default(WebAssembly::ExprType::Invalid);
3144ba319b5SDimitry Andric   }
315*b5893f02SDimitry Andric 
parseRegTypeList(SmallVectorImpl<wasm::ValType> & Types)316*b5893f02SDimitry Andric   bool parseRegTypeList(SmallVectorImpl<wasm::ValType> &Types) {
317*b5893f02SDimitry Andric     while (Lexer.is(AsmToken::Identifier)) {
318*b5893f02SDimitry Andric       auto Type = parseType(Lexer.getTok().getString());
319*b5893f02SDimitry Andric       if (!Type)
320*b5893f02SDimitry Andric         return true;
321*b5893f02SDimitry Andric       Types.push_back(Type.getValue());
3224ba319b5SDimitry Andric       Parser.Lex();
323*b5893f02SDimitry Andric       if (!isNext(AsmToken::Comma))
324*b5893f02SDimitry Andric         break;
3254ba319b5SDimitry Andric     }
3264ba319b5SDimitry Andric     return false;
3274ba319b5SDimitry Andric   }
3284ba319b5SDimitry Andric 
parseSingleInteger(bool IsNegative,OperandVector & Operands)329*b5893f02SDimitry Andric   void parseSingleInteger(bool IsNegative, OperandVector &Operands) {
3304ba319b5SDimitry Andric     auto &Int = Lexer.getTok();
3314ba319b5SDimitry Andric     int64_t Val = Int.getIntVal();
332*b5893f02SDimitry Andric     if (IsNegative)
333*b5893f02SDimitry Andric       Val = -Val;
3344ba319b5SDimitry Andric     Operands.push_back(make_unique<WebAssemblyOperand>(
335*b5893f02SDimitry Andric         WebAssemblyOperand::Integer, Int.getLoc(), Int.getEndLoc(),
336*b5893f02SDimitry Andric         WebAssemblyOperand::IntOp{Val}));
3374ba319b5SDimitry Andric     Parser.Lex();
3384ba319b5SDimitry Andric   }
3394ba319b5SDimitry Andric 
parseOperandStartingWithInteger(bool IsNegative,OperandVector & Operands,StringRef InstName)340*b5893f02SDimitry Andric   bool parseOperandStartingWithInteger(bool IsNegative, OperandVector &Operands,
341*b5893f02SDimitry Andric                                        StringRef InstName) {
342*b5893f02SDimitry Andric     parseSingleInteger(IsNegative, Operands);
343*b5893f02SDimitry Andric     // FIXME: there is probably a cleaner way to do this.
344*b5893f02SDimitry Andric     auto IsLoadStore = InstName.startswith("load") ||
345*b5893f02SDimitry Andric                        InstName.startswith("store") ||
346*b5893f02SDimitry Andric                        InstName.startswith("atomic_load") ||
347*b5893f02SDimitry Andric                        InstName.startswith("atomic_store");
348*b5893f02SDimitry Andric     if (IsLoadStore) {
349*b5893f02SDimitry Andric       // Parse load/store operands of the form: offset align
350*b5893f02SDimitry Andric       auto &Offset = Lexer.getTok();
351*b5893f02SDimitry Andric       if (Offset.is(AsmToken::Integer)) {
352*b5893f02SDimitry Andric         parseSingleInteger(false, Operands);
3534ba319b5SDimitry Andric       } else {
3544ba319b5SDimitry Andric         // Alignment not specified.
3554ba319b5SDimitry Andric         // FIXME: correctly derive a default from the instruction.
356*b5893f02SDimitry Andric         // We can't just call WebAssembly::GetDefaultP2Align since we don't have
357*b5893f02SDimitry Andric         // an opcode until after the assembly matcher.
3584ba319b5SDimitry Andric         Operands.push_back(make_unique<WebAssemblyOperand>(
359*b5893f02SDimitry Andric             WebAssemblyOperand::Integer, Offset.getLoc(), Offset.getEndLoc(),
360*b5893f02SDimitry Andric             WebAssemblyOperand::IntOp{0}));
3614ba319b5SDimitry Andric       }
3624ba319b5SDimitry Andric     }
3634ba319b5SDimitry Andric     return false;
3644ba319b5SDimitry Andric   }
3654ba319b5SDimitry Andric 
addBlockTypeOperand(OperandVector & Operands,SMLoc NameLoc,WebAssembly::ExprType BT)366*b5893f02SDimitry Andric   void addBlockTypeOperand(OperandVector &Operands, SMLoc NameLoc,
367*b5893f02SDimitry Andric                            WebAssembly::ExprType BT) {
368*b5893f02SDimitry Andric     Operands.push_back(make_unique<WebAssemblyOperand>(
369*b5893f02SDimitry Andric         WebAssemblyOperand::Integer, NameLoc, NameLoc,
370*b5893f02SDimitry Andric         WebAssemblyOperand::IntOp{static_cast<int64_t>(BT)}));
371*b5893f02SDimitry Andric   }
372*b5893f02SDimitry Andric 
ParseInstruction(ParseInstructionInfo &,StringRef Name,SMLoc NameLoc,OperandVector & Operands)3734ba319b5SDimitry Andric   bool ParseInstruction(ParseInstructionInfo & /*Info*/, StringRef Name,
3744ba319b5SDimitry Andric                         SMLoc NameLoc, OperandVector &Operands) override {
375*b5893f02SDimitry Andric     // Note: Name does NOT point into the sourcecode, but to a local, so
376*b5893f02SDimitry Andric     // use NameLoc instead.
377*b5893f02SDimitry Andric     Name = StringRef(NameLoc.getPointer(), Name.size());
378*b5893f02SDimitry Andric 
379*b5893f02SDimitry Andric     // WebAssembly has instructions with / in them, which AsmLexer parses
380*b5893f02SDimitry Andric     // as seperate tokens, so if we find such tokens immediately adjacent (no
381*b5893f02SDimitry Andric     // whitespace), expand the name to include them:
382*b5893f02SDimitry Andric     for (;;) {
383*b5893f02SDimitry Andric       auto &Sep = Lexer.getTok();
384*b5893f02SDimitry Andric       if (Sep.getLoc().getPointer() != Name.end() ||
385*b5893f02SDimitry Andric           Sep.getKind() != AsmToken::Slash)
386*b5893f02SDimitry Andric         break;
387*b5893f02SDimitry Andric       // Extend name with /
388*b5893f02SDimitry Andric       Name = StringRef(Name.begin(), Name.size() + Sep.getString().size());
389*b5893f02SDimitry Andric       Parser.Lex();
390*b5893f02SDimitry Andric       // We must now find another identifier, or error.
391*b5893f02SDimitry Andric       auto &Id = Lexer.getTok();
392*b5893f02SDimitry Andric       if (Id.getKind() != AsmToken::Identifier ||
393*b5893f02SDimitry Andric           Id.getLoc().getPointer() != Name.end())
394*b5893f02SDimitry Andric         return error("Incomplete instruction name: ", Id);
395*b5893f02SDimitry Andric       Name = StringRef(Name.begin(), Name.size() + Id.getString().size());
396*b5893f02SDimitry Andric       Parser.Lex();
397*b5893f02SDimitry Andric     }
398*b5893f02SDimitry Andric 
399*b5893f02SDimitry Andric     // Now construct the name as first operand.
400*b5893f02SDimitry Andric     Operands.push_back(make_unique<WebAssemblyOperand>(
401*b5893f02SDimitry Andric         WebAssemblyOperand::Token, NameLoc, SMLoc::getFromPointer(Name.end()),
402*b5893f02SDimitry Andric         WebAssemblyOperand::TokOp{Name}));
4034ba319b5SDimitry Andric     auto NamePair = Name.split('.');
4044ba319b5SDimitry Andric     // If no '.', there is no type prefix.
405*b5893f02SDimitry Andric     auto BaseName = NamePair.second.empty() ? NamePair.first : NamePair.second;
406*b5893f02SDimitry Andric 
407*b5893f02SDimitry Andric     // If this instruction is part of a control flow structure, ensure
408*b5893f02SDimitry Andric     // proper nesting.
409*b5893f02SDimitry Andric     bool ExpectBlockType = false;
410*b5893f02SDimitry Andric     if (BaseName == "block") {
411*b5893f02SDimitry Andric       push(Block);
412*b5893f02SDimitry Andric       ExpectBlockType = true;
413*b5893f02SDimitry Andric     } else if (BaseName == "loop") {
414*b5893f02SDimitry Andric       push(Loop);
415*b5893f02SDimitry Andric       ExpectBlockType = true;
416*b5893f02SDimitry Andric     } else if (BaseName == "try") {
417*b5893f02SDimitry Andric       push(Try);
418*b5893f02SDimitry Andric       ExpectBlockType = true;
419*b5893f02SDimitry Andric     } else if (BaseName == "if") {
420*b5893f02SDimitry Andric       push(If);
421*b5893f02SDimitry Andric       ExpectBlockType = true;
422*b5893f02SDimitry Andric     } else if (BaseName == "else") {
423*b5893f02SDimitry Andric       if (pop(BaseName, If))
424*b5893f02SDimitry Andric         return true;
425*b5893f02SDimitry Andric       push(Else);
426*b5893f02SDimitry Andric     } else if (BaseName == "catch") {
427*b5893f02SDimitry Andric       if (pop(BaseName, Try))
428*b5893f02SDimitry Andric         return true;
429*b5893f02SDimitry Andric       push(Try);
430*b5893f02SDimitry Andric     } else if (BaseName == "catch_all") {
431*b5893f02SDimitry Andric       if (pop(BaseName, Try))
432*b5893f02SDimitry Andric         return true;
433*b5893f02SDimitry Andric       push(Try);
434*b5893f02SDimitry Andric     } else if (BaseName == "end_if") {
435*b5893f02SDimitry Andric       if (pop(BaseName, If, Else))
436*b5893f02SDimitry Andric         return true;
437*b5893f02SDimitry Andric     } else if (BaseName == "end_try") {
438*b5893f02SDimitry Andric       if (pop(BaseName, Try))
439*b5893f02SDimitry Andric         return true;
440*b5893f02SDimitry Andric     } else if (BaseName == "end_loop") {
441*b5893f02SDimitry Andric       if (pop(BaseName, Loop))
442*b5893f02SDimitry Andric         return true;
443*b5893f02SDimitry Andric     } else if (BaseName == "end_block") {
444*b5893f02SDimitry Andric       if (pop(BaseName, Block))
445*b5893f02SDimitry Andric         return true;
446*b5893f02SDimitry Andric     } else if (BaseName == "end_function") {
447*b5893f02SDimitry Andric       if (pop(BaseName, Function) || ensureEmptyNestingStack())
448*b5893f02SDimitry Andric         return true;
449*b5893f02SDimitry Andric     }
450*b5893f02SDimitry Andric 
4514ba319b5SDimitry Andric     while (Lexer.isNot(AsmToken::EndOfStatement)) {
4524ba319b5SDimitry Andric       auto &Tok = Lexer.getTok();
4534ba319b5SDimitry Andric       switch (Tok.getKind()) {
4544ba319b5SDimitry Andric       case AsmToken::Identifier: {
4554ba319b5SDimitry Andric         auto &Id = Lexer.getTok();
456*b5893f02SDimitry Andric         if (ExpectBlockType) {
457*b5893f02SDimitry Andric           // Assume this identifier is a block_type.
458*b5893f02SDimitry Andric           auto BT = parseBlockType(Id.getString());
459*b5893f02SDimitry Andric           if (BT == WebAssembly::ExprType::Invalid)
460*b5893f02SDimitry Andric             return error("Unknown block type: ", Id);
461*b5893f02SDimitry Andric           addBlockTypeOperand(Operands, NameLoc, BT);
462*b5893f02SDimitry Andric           Parser.Lex();
463*b5893f02SDimitry Andric         } else {
464*b5893f02SDimitry Andric           // Assume this identifier is a label.
4654ba319b5SDimitry Andric           const MCExpr *Val;
4664ba319b5SDimitry Andric           SMLoc End;
4674ba319b5SDimitry Andric           if (Parser.parsePrimaryExpr(Val, End))
468*b5893f02SDimitry Andric             return error("Cannot parse symbol: ", Lexer.getTok());
4694ba319b5SDimitry Andric           Operands.push_back(make_unique<WebAssemblyOperand>(
470*b5893f02SDimitry Andric               WebAssemblyOperand::Symbol, Id.getLoc(), Id.getEndLoc(),
471*b5893f02SDimitry Andric               WebAssemblyOperand::SymOp{Val}));
472*b5893f02SDimitry Andric         }
4734ba319b5SDimitry Andric         break;
4744ba319b5SDimitry Andric       }
4754ba319b5SDimitry Andric       case AsmToken::Minus:
4764ba319b5SDimitry Andric         Parser.Lex();
4774ba319b5SDimitry Andric         if (Lexer.isNot(AsmToken::Integer))
478*b5893f02SDimitry Andric           return error("Expected integer instead got: ", Lexer.getTok());
479*b5893f02SDimitry Andric         if (parseOperandStartingWithInteger(true, Operands, BaseName))
4804ba319b5SDimitry Andric           return true;
4814ba319b5SDimitry Andric         break;
4824ba319b5SDimitry Andric       case AsmToken::Integer:
483*b5893f02SDimitry Andric         if (parseOperandStartingWithInteger(false, Operands, BaseName))
4844ba319b5SDimitry Andric           return true;
4854ba319b5SDimitry Andric         break;
4864ba319b5SDimitry Andric       case AsmToken::Real: {
4874ba319b5SDimitry Andric         double Val;
4884ba319b5SDimitry Andric         if (Tok.getString().getAsDouble(Val, false))
489*b5893f02SDimitry Andric           return error("Cannot parse real: ", Tok);
4904ba319b5SDimitry Andric         Operands.push_back(make_unique<WebAssemblyOperand>(
491*b5893f02SDimitry Andric             WebAssemblyOperand::Float, Tok.getLoc(), Tok.getEndLoc(),
492*b5893f02SDimitry Andric             WebAssemblyOperand::FltOp{Val}));
4934ba319b5SDimitry Andric         Parser.Lex();
4944ba319b5SDimitry Andric         break;
4954ba319b5SDimitry Andric       }
496*b5893f02SDimitry Andric       case AsmToken::LCurly: {
497*b5893f02SDimitry Andric         Parser.Lex();
498*b5893f02SDimitry Andric         auto Op = make_unique<WebAssemblyOperand>(
499*b5893f02SDimitry Andric             WebAssemblyOperand::BrList, Tok.getLoc(), Tok.getEndLoc());
500*b5893f02SDimitry Andric         if (!Lexer.is(AsmToken::RCurly))
501*b5893f02SDimitry Andric           for (;;) {
502*b5893f02SDimitry Andric             Op->BrL.List.push_back(Lexer.getTok().getIntVal());
503*b5893f02SDimitry Andric             expect(AsmToken::Integer, "integer");
504*b5893f02SDimitry Andric             if (!isNext(AsmToken::Comma))
505*b5893f02SDimitry Andric               break;
506*b5893f02SDimitry Andric           }
507*b5893f02SDimitry Andric         expect(AsmToken::RCurly, "}");
508*b5893f02SDimitry Andric         Operands.push_back(std::move(Op));
509*b5893f02SDimitry Andric         break;
510*b5893f02SDimitry Andric       }
5114ba319b5SDimitry Andric       default:
512*b5893f02SDimitry Andric         return error("Unexpected token in operand: ", Tok);
5134ba319b5SDimitry Andric       }
5144ba319b5SDimitry Andric       if (Lexer.isNot(AsmToken::EndOfStatement)) {
515*b5893f02SDimitry Andric         if (expect(AsmToken::Comma, ","))
516*b5893f02SDimitry Andric           return true;
5174ba319b5SDimitry Andric       }
5184ba319b5SDimitry Andric     }
519*b5893f02SDimitry Andric     if (ExpectBlockType && Operands.size() == 1) {
520*b5893f02SDimitry Andric       // Support blocks with no operands as default to void.
521*b5893f02SDimitry Andric       addBlockTypeOperand(Operands, NameLoc, WebAssembly::ExprType::Void);
522*b5893f02SDimitry Andric     }
5234ba319b5SDimitry Andric     Parser.Lex();
5244ba319b5SDimitry Andric     return false;
5254ba319b5SDimitry Andric   }
5264ba319b5SDimitry Andric 
onLabelParsed(MCSymbol * Symbol)5274ba319b5SDimitry Andric   void onLabelParsed(MCSymbol *Symbol) override {
5284ba319b5SDimitry Andric     LastLabel = Symbol;
529*b5893f02SDimitry Andric     CurrentState = Label;
5304ba319b5SDimitry Andric   }
5314ba319b5SDimitry Andric 
parseSignature(wasm::WasmSignature * Signature)532*b5893f02SDimitry Andric   bool parseSignature(wasm::WasmSignature *Signature) {
533*b5893f02SDimitry Andric     if (expect(AsmToken::LParen, "("))
534*b5893f02SDimitry Andric       return true;
535*b5893f02SDimitry Andric     if (parseRegTypeList(Signature->Params))
536*b5893f02SDimitry Andric       return true;
537*b5893f02SDimitry Andric     if (expect(AsmToken::RParen, ")"))
538*b5893f02SDimitry Andric       return true;
539*b5893f02SDimitry Andric     if (expect(AsmToken::MinusGreater, "->"))
540*b5893f02SDimitry Andric       return true;
541*b5893f02SDimitry Andric     if (expect(AsmToken::LParen, "("))
542*b5893f02SDimitry Andric       return true;
543*b5893f02SDimitry Andric     if (parseRegTypeList(Signature->Returns))
544*b5893f02SDimitry Andric       return true;
545*b5893f02SDimitry Andric     if (expect(AsmToken::RParen, ")"))
546*b5893f02SDimitry Andric       return true;
547*b5893f02SDimitry Andric     return false;
548*b5893f02SDimitry Andric   }
549*b5893f02SDimitry Andric 
550*b5893f02SDimitry Andric   // This function processes wasm-specific directives streamed to
551*b5893f02SDimitry Andric   // WebAssemblyTargetStreamer, all others go to the generic parser
552*b5893f02SDimitry Andric   // (see WasmAsmParser).
ParseDirective(AsmToken DirectiveID)5534ba319b5SDimitry Andric   bool ParseDirective(AsmToken DirectiveID) override {
554*b5893f02SDimitry Andric     // This function has a really weird return value behavior that is different
555*b5893f02SDimitry Andric     // from all the other parsing functions:
556*b5893f02SDimitry Andric     // - return true && no tokens consumed -> don't know this directive / let
557*b5893f02SDimitry Andric     //   the generic parser handle it.
558*b5893f02SDimitry Andric     // - return true && tokens consumed -> a parsing error occurred.
559*b5893f02SDimitry Andric     // - return false -> processed this directive successfully.
5604ba319b5SDimitry Andric     assert(DirectiveID.getKind() == AsmToken::Identifier);
5614ba319b5SDimitry Andric     auto &Out = getStreamer();
562*b5893f02SDimitry Andric     auto &TOut =
563*b5893f02SDimitry Andric         reinterpret_cast<WebAssemblyTargetStreamer &>(*Out.getTargetStreamer());
564*b5893f02SDimitry Andric 
565*b5893f02SDimitry Andric     // TODO: any time we return an error, at least one token must have been
566*b5893f02SDimitry Andric     // consumed, otherwise this will not signal an error to the caller.
567*b5893f02SDimitry Andric     if (DirectiveID.getString() == ".globaltype") {
568*b5893f02SDimitry Andric       auto SymName = expectIdent();
569*b5893f02SDimitry Andric       if (SymName.empty())
570*b5893f02SDimitry Andric         return true;
571*b5893f02SDimitry Andric       if (expect(AsmToken::Comma, ","))
572*b5893f02SDimitry Andric         return true;
573*b5893f02SDimitry Andric       auto TypeTok = Lexer.getTok();
574*b5893f02SDimitry Andric       auto TypeName = expectIdent();
575*b5893f02SDimitry Andric       if (TypeName.empty())
576*b5893f02SDimitry Andric         return true;
577*b5893f02SDimitry Andric       auto Type = parseType(TypeName);
578*b5893f02SDimitry Andric       if (!Type)
579*b5893f02SDimitry Andric         return error("Unknown type in .globaltype directive: ", TypeTok);
580*b5893f02SDimitry Andric       // Now set this symbol with the correct type.
581*b5893f02SDimitry Andric       auto WasmSym = cast<MCSymbolWasm>(
582*b5893f02SDimitry Andric           TOut.getStreamer().getContext().getOrCreateSymbol(SymName));
583*b5893f02SDimitry Andric       WasmSym->setType(wasm::WASM_SYMBOL_TYPE_GLOBAL);
584*b5893f02SDimitry Andric       WasmSym->setGlobalType(
585*b5893f02SDimitry Andric           wasm::WasmGlobalType{uint8_t(Type.getValue()), true});
586*b5893f02SDimitry Andric       // And emit the directive again.
587*b5893f02SDimitry Andric       TOut.emitGlobalType(WasmSym);
588*b5893f02SDimitry Andric       return expect(AsmToken::EndOfStatement, "EOL");
5894ba319b5SDimitry Andric     }
590*b5893f02SDimitry Andric 
591*b5893f02SDimitry Andric     if (DirectiveID.getString() == ".functype") {
592*b5893f02SDimitry Andric       // This code has to send things to the streamer similar to
593*b5893f02SDimitry Andric       // WebAssemblyAsmPrinter::EmitFunctionBodyStart.
594*b5893f02SDimitry Andric       // TODO: would be good to factor this into a common function, but the
595*b5893f02SDimitry Andric       // assembler and backend really don't share any common code, and this code
596*b5893f02SDimitry Andric       // parses the locals seperately.
597*b5893f02SDimitry Andric       auto SymName = expectIdent();
598*b5893f02SDimitry Andric       if (SymName.empty())
599*b5893f02SDimitry Andric         return true;
600*b5893f02SDimitry Andric       auto WasmSym = cast<MCSymbolWasm>(
601*b5893f02SDimitry Andric           TOut.getStreamer().getContext().getOrCreateSymbol(SymName));
602*b5893f02SDimitry Andric       if (CurrentState == Label && WasmSym == LastLabel) {
603*b5893f02SDimitry Andric         // This .functype indicates a start of a function.
604*b5893f02SDimitry Andric         if (ensureEmptyNestingStack())
605*b5893f02SDimitry Andric           return true;
606*b5893f02SDimitry Andric         CurrentState = FunctionStart;
607*b5893f02SDimitry Andric         push(Function);
6084ba319b5SDimitry Andric       }
609*b5893f02SDimitry Andric       auto Signature = make_unique<wasm::WasmSignature>();
610*b5893f02SDimitry Andric       if (parseSignature(Signature.get()))
611*b5893f02SDimitry Andric         return true;
612*b5893f02SDimitry Andric       WasmSym->setSignature(Signature.get());
613*b5893f02SDimitry Andric       addSignature(std::move(Signature));
614*b5893f02SDimitry Andric       WasmSym->setType(wasm::WASM_SYMBOL_TYPE_FUNCTION);
615*b5893f02SDimitry Andric       TOut.emitFunctionType(WasmSym);
616*b5893f02SDimitry Andric       // TODO: backend also calls TOut.emitIndIdx, but that is not implemented.
617*b5893f02SDimitry Andric       return expect(AsmToken::EndOfStatement, "EOL");
6184ba319b5SDimitry Andric     }
619*b5893f02SDimitry Andric 
620*b5893f02SDimitry Andric     if (DirectiveID.getString() == ".eventtype") {
621*b5893f02SDimitry Andric       auto SymName = expectIdent();
622*b5893f02SDimitry Andric       if (SymName.empty())
623*b5893f02SDimitry Andric         return true;
624*b5893f02SDimitry Andric       auto WasmSym = cast<MCSymbolWasm>(
625*b5893f02SDimitry Andric           TOut.getStreamer().getContext().getOrCreateSymbol(SymName));
626*b5893f02SDimitry Andric       auto Signature = make_unique<wasm::WasmSignature>();
627*b5893f02SDimitry Andric       if (parseRegTypeList(Signature->Params))
628*b5893f02SDimitry Andric         return true;
629*b5893f02SDimitry Andric       WasmSym->setSignature(Signature.get());
630*b5893f02SDimitry Andric       addSignature(std::move(Signature));
631*b5893f02SDimitry Andric       WasmSym->setType(wasm::WASM_SYMBOL_TYPE_EVENT);
632*b5893f02SDimitry Andric       TOut.emitEventType(WasmSym);
633*b5893f02SDimitry Andric       // TODO: backend also calls TOut.emitIndIdx, but that is not implemented.
634*b5893f02SDimitry Andric       return expect(AsmToken::EndOfStatement, "EOL");
635*b5893f02SDimitry Andric     }
636*b5893f02SDimitry Andric 
637*b5893f02SDimitry Andric     if (DirectiveID.getString() == ".local") {
638*b5893f02SDimitry Andric       if (CurrentState != FunctionStart)
639*b5893f02SDimitry Andric         return error(".local directive should follow the start of a function",
640*b5893f02SDimitry Andric                      Lexer.getTok());
641*b5893f02SDimitry Andric       SmallVector<wasm::ValType, 4> Locals;
642*b5893f02SDimitry Andric       if (parseRegTypeList(Locals))
643*b5893f02SDimitry Andric         return true;
6444ba319b5SDimitry Andric       TOut.emitLocal(Locals);
645*b5893f02SDimitry Andric       CurrentState = FunctionLocals;
646*b5893f02SDimitry Andric       return expect(AsmToken::EndOfStatement, "EOL");
6474ba319b5SDimitry Andric     }
648*b5893f02SDimitry Andric 
649*b5893f02SDimitry Andric     return true; // We didn't process this directive.
6504ba319b5SDimitry Andric   }
6514ba319b5SDimitry Andric 
MatchAndEmitInstruction(SMLoc IDLoc,unsigned &,OperandVector & Operands,MCStreamer & Out,uint64_t & ErrorInfo,bool MatchingInlineAsm)6524ba319b5SDimitry Andric   bool MatchAndEmitInstruction(SMLoc IDLoc, unsigned & /*Opcode*/,
653*b5893f02SDimitry Andric                                OperandVector &Operands, MCStreamer &Out,
654*b5893f02SDimitry Andric                                uint64_t &ErrorInfo,
6554ba319b5SDimitry Andric                                bool MatchingInlineAsm) override {
6564ba319b5SDimitry Andric     MCInst Inst;
6574ba319b5SDimitry Andric     unsigned MatchResult =
6584ba319b5SDimitry Andric         MatchInstructionImpl(Operands, Inst, ErrorInfo, MatchingInlineAsm);
6594ba319b5SDimitry Andric     switch (MatchResult) {
6604ba319b5SDimitry Andric     case Match_Success: {
661*b5893f02SDimitry Andric       if (CurrentState == FunctionStart) {
662*b5893f02SDimitry Andric         // This is the first instruction in a function, but we haven't seen
663*b5893f02SDimitry Andric         // a .local directive yet. The streamer requires locals to be encoded
664*b5893f02SDimitry Andric         // as a prelude to the instructions, so emit an empty list of locals
665*b5893f02SDimitry Andric         // here.
666*b5893f02SDimitry Andric         auto &TOut = reinterpret_cast<WebAssemblyTargetStreamer &>(
667*b5893f02SDimitry Andric             *Out.getTargetStreamer());
668*b5893f02SDimitry Andric         TOut.emitLocal(SmallVector<wasm::ValType, 0>());
669*b5893f02SDimitry Andric       }
670*b5893f02SDimitry Andric       CurrentState = Instructions;
6714ba319b5SDimitry Andric       Out.EmitInstruction(Inst, getSTI());
6724ba319b5SDimitry Andric       return false;
6734ba319b5SDimitry Andric     }
6744ba319b5SDimitry Andric     case Match_MissingFeature:
675*b5893f02SDimitry Andric       return Parser.Error(
676*b5893f02SDimitry Andric           IDLoc, "instruction requires a WASM feature not currently enabled");
6774ba319b5SDimitry Andric     case Match_MnemonicFail:
6784ba319b5SDimitry Andric       return Parser.Error(IDLoc, "invalid instruction");
6794ba319b5SDimitry Andric     case Match_NearMisses:
6804ba319b5SDimitry Andric       return Parser.Error(IDLoc, "ambiguous instruction");
6814ba319b5SDimitry Andric     case Match_InvalidTiedOperand:
6824ba319b5SDimitry Andric     case Match_InvalidOperand: {
6834ba319b5SDimitry Andric       SMLoc ErrorLoc = IDLoc;
6844ba319b5SDimitry Andric       if (ErrorInfo != ~0ULL) {
6854ba319b5SDimitry Andric         if (ErrorInfo >= Operands.size())
6864ba319b5SDimitry Andric           return Parser.Error(IDLoc, "too few operands for instruction");
6874ba319b5SDimitry Andric         ErrorLoc = Operands[ErrorInfo]->getStartLoc();
6884ba319b5SDimitry Andric         if (ErrorLoc == SMLoc())
6894ba319b5SDimitry Andric           ErrorLoc = IDLoc;
6904ba319b5SDimitry Andric       }
6914ba319b5SDimitry Andric       return Parser.Error(ErrorLoc, "invalid operand for instruction");
6924ba319b5SDimitry Andric     }
6934ba319b5SDimitry Andric     }
6944ba319b5SDimitry Andric     llvm_unreachable("Implement any new match types added!");
6954ba319b5SDimitry Andric   }
696*b5893f02SDimitry Andric 
onEndOfFile()697*b5893f02SDimitry Andric   void onEndOfFile() override { ensureEmptyNestingStack(); }
6984ba319b5SDimitry Andric };
6994ba319b5SDimitry Andric } // end anonymous namespace
7004ba319b5SDimitry Andric 
7014ba319b5SDimitry Andric // Force static initialization.
LLVMInitializeWebAssemblyAsmParser()7024ba319b5SDimitry Andric extern "C" void LLVMInitializeWebAssemblyAsmParser() {
7034ba319b5SDimitry Andric   RegisterMCAsmParser<WebAssemblyAsmParser> X(getTheWebAssemblyTarget32());
7044ba319b5SDimitry Andric   RegisterMCAsmParser<WebAssemblyAsmParser> Y(getTheWebAssemblyTarget64());
7054ba319b5SDimitry Andric }
7064ba319b5SDimitry Andric 
7074ba319b5SDimitry Andric #define GET_REGISTER_MATCHER
7084ba319b5SDimitry Andric #define GET_MATCHER_IMPLEMENTATION
7094ba319b5SDimitry Andric #include "WebAssemblyGenAsmMatcher.inc"
710