1 //===---- ParseStmtAsm.cpp - Assembly Statement Parser --------------------===//
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 parsing for GCC and Microsoft inline assembly.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/Parse/Parser.h"
15 #include "RAIIObjectsForParser.h"
16 #include "clang/AST/ASTContext.h"
17 #include "clang/Basic/Diagnostic.h"
18 #include "clang/Basic/TargetInfo.h"
19 #include "llvm/ADT/SmallString.h"
20 #include "llvm/MC/MCAsmInfo.h"
21 #include "llvm/MC/MCContext.h"
22 #include "llvm/MC/MCInstPrinter.h"
23 #include "llvm/MC/MCInstrInfo.h"
24 #include "llvm/MC/MCObjectFileInfo.h"
25 #include "llvm/MC/MCParser/MCAsmParser.h"
26 #include "llvm/MC/MCRegisterInfo.h"
27 #include "llvm/MC/MCStreamer.h"
28 #include "llvm/MC/MCSubtargetInfo.h"
29 #include "llvm/MC/MCTargetAsmParser.h"
30 #include "llvm/MC/MCTargetOptions.h"
31 #include "llvm/Support/SourceMgr.h"
32 #include "llvm/Support/TargetRegistry.h"
33 #include "llvm/Support/TargetSelect.h"
34 using namespace clang;
35 
36 namespace {
37 class ClangAsmParserCallback : public llvm::MCAsmParserSemaCallback {
38   Parser &TheParser;
39   SourceLocation AsmLoc;
40   StringRef AsmString;
41 
42   /// The tokens we streamed into AsmString and handed off to MC.
43   ArrayRef<Token> AsmToks;
44 
45   /// The offset of each token in AsmToks within AsmString.
46   ArrayRef<unsigned> AsmTokOffsets;
47 
48 public:
ClangAsmParserCallback(Parser & P,SourceLocation Loc,StringRef AsmString,ArrayRef<Token> Toks,ArrayRef<unsigned> Offsets)49   ClangAsmParserCallback(Parser &P, SourceLocation Loc, StringRef AsmString,
50                          ArrayRef<Token> Toks, ArrayRef<unsigned> Offsets)
51       : TheParser(P), AsmLoc(Loc), AsmString(AsmString), AsmToks(Toks),
52         AsmTokOffsets(Offsets) {
53     assert(AsmToks.size() == AsmTokOffsets.size());
54   }
55 
LookupInlineAsmIdentifier(StringRef & LineBuf,llvm::InlineAsmIdentifierInfo & Info,bool IsUnevaluatedContext)56   void *LookupInlineAsmIdentifier(StringRef &LineBuf,
57                                   llvm::InlineAsmIdentifierInfo &Info,
58                                   bool IsUnevaluatedContext) override {
59     // Collect the desired tokens.
60     SmallVector<Token, 16> LineToks;
61     const Token *FirstOrigToken = nullptr;
62     findTokensForString(LineBuf, LineToks, FirstOrigToken);
63 
64     unsigned NumConsumedToks;
65     ExprResult Result = TheParser.ParseMSAsmIdentifier(
66         LineToks, NumConsumedToks, &Info, IsUnevaluatedContext);
67 
68     // If we consumed the entire line, tell MC that.
69     // Also do this if we consumed nothing as a way of reporting failure.
70     if (NumConsumedToks == 0 || NumConsumedToks == LineToks.size()) {
71       // By not modifying LineBuf, we're implicitly consuming it all.
72 
73       // Otherwise, consume up to the original tokens.
74     } else {
75       assert(FirstOrigToken && "not using original tokens?");
76 
77       // Since we're using original tokens, apply that offset.
78       assert(FirstOrigToken[NumConsumedToks].getLocation() ==
79              LineToks[NumConsumedToks].getLocation());
80       unsigned FirstIndex = FirstOrigToken - AsmToks.begin();
81       unsigned LastIndex = FirstIndex + NumConsumedToks - 1;
82 
83       // The total length we've consumed is the relative offset
84       // of the last token we consumed plus its length.
85       unsigned TotalOffset =
86           (AsmTokOffsets[LastIndex] + AsmToks[LastIndex].getLength() -
87            AsmTokOffsets[FirstIndex]);
88       LineBuf = LineBuf.substr(0, TotalOffset);
89     }
90 
91     // Initialize the "decl" with the lookup result.
92     Info.OpDecl = static_cast<void *>(Result.get());
93     return Info.OpDecl;
94   }
95 
LookupInlineAsmLabel(StringRef Identifier,llvm::SourceMgr & LSM,llvm::SMLoc Location,bool Create)96   StringRef LookupInlineAsmLabel(StringRef Identifier, llvm::SourceMgr &LSM,
97                                  llvm::SMLoc Location,
98                                  bool Create) override {
99     SourceLocation Loc = translateLocation(LSM, Location);
100     LabelDecl *Label =
101       TheParser.getActions().GetOrCreateMSAsmLabel(Identifier, Loc, Create);
102     return Label->getMSAsmLabel();
103   }
104 
LookupInlineAsmField(StringRef Base,StringRef Member,unsigned & Offset)105   bool LookupInlineAsmField(StringRef Base, StringRef Member,
106                             unsigned &Offset) override {
107     return TheParser.getActions().LookupInlineAsmField(Base, Member, Offset,
108                                                        AsmLoc);
109   }
110 
DiagHandlerCallback(const llvm::SMDiagnostic & D,void * Context)111   static void DiagHandlerCallback(const llvm::SMDiagnostic &D, void *Context) {
112     ((ClangAsmParserCallback *)Context)->handleDiagnostic(D);
113   }
114 
115 private:
116   /// Collect the appropriate tokens for the given string.
findTokensForString(StringRef Str,SmallVectorImpl<Token> & TempToks,const Token * & FirstOrigToken) const117   void findTokensForString(StringRef Str, SmallVectorImpl<Token> &TempToks,
118                            const Token *&FirstOrigToken) const {
119     // For now, assert that the string we're working with is a substring
120     // of what we gave to MC.  This lets us use the original tokens.
121     assert(!std::less<const char *>()(Str.begin(), AsmString.begin()) &&
122            !std::less<const char *>()(AsmString.end(), Str.end()));
123 
124     // Try to find a token whose offset matches the first token.
125     unsigned FirstCharOffset = Str.begin() - AsmString.begin();
126     const unsigned *FirstTokOffset = std::lower_bound(
127         AsmTokOffsets.begin(), AsmTokOffsets.end(), FirstCharOffset);
128 
129     // For now, assert that the start of the string exactly
130     // corresponds to the start of a token.
131     assert(*FirstTokOffset == FirstCharOffset);
132 
133     // Use all the original tokens for this line.  (We assume the
134     // end of the line corresponds cleanly to a token break.)
135     unsigned FirstTokIndex = FirstTokOffset - AsmTokOffsets.begin();
136     FirstOrigToken = &AsmToks[FirstTokIndex];
137     unsigned LastCharOffset = Str.end() - AsmString.begin();
138     for (unsigned i = FirstTokIndex, e = AsmTokOffsets.size(); i != e; ++i) {
139       if (AsmTokOffsets[i] >= LastCharOffset)
140         break;
141       TempToks.push_back(AsmToks[i]);
142     }
143   }
144 
translateLocation(const llvm::SourceMgr & LSM,llvm::SMLoc SMLoc)145   SourceLocation translateLocation(const llvm::SourceMgr &LSM, llvm::SMLoc SMLoc) {
146     // Compute an offset into the inline asm buffer.
147     // FIXME: This isn't right if .macro is involved (but hopefully, no
148     // real-world code does that).
149     const llvm::MemoryBuffer *LBuf =
150         LSM.getMemoryBuffer(LSM.FindBufferContainingLoc(SMLoc));
151     unsigned Offset = SMLoc.getPointer() - LBuf->getBufferStart();
152 
153     // Figure out which token that offset points into.
154     const unsigned *TokOffsetPtr =
155         std::lower_bound(AsmTokOffsets.begin(), AsmTokOffsets.end(), Offset);
156     unsigned TokIndex = TokOffsetPtr - AsmTokOffsets.begin();
157     unsigned TokOffset = *TokOffsetPtr;
158 
159     // If we come up with an answer which seems sane, use it; otherwise,
160     // just point at the __asm keyword.
161     // FIXME: Assert the answer is sane once we handle .macro correctly.
162     SourceLocation Loc = AsmLoc;
163     if (TokIndex < AsmToks.size()) {
164       const Token &Tok = AsmToks[TokIndex];
165       Loc = Tok.getLocation();
166       Loc = Loc.getLocWithOffset(Offset - TokOffset);
167     }
168     return Loc;
169   }
170 
handleDiagnostic(const llvm::SMDiagnostic & D)171   void handleDiagnostic(const llvm::SMDiagnostic &D) {
172     const llvm::SourceMgr &LSM = *D.getSourceMgr();
173     SourceLocation Loc = translateLocation(LSM, D.getLoc());
174     TheParser.Diag(Loc, diag::err_inline_ms_asm_parsing) << D.getMessage();
175   }
176 };
177 }
178 
179 /// Parse an identifier in an MS-style inline assembly block.
180 ///
181 /// \param CastInfo - a void* so that we don't have to teach Parser.h
182 ///   about the actual type.
ParseMSAsmIdentifier(llvm::SmallVectorImpl<Token> & LineToks,unsigned & NumLineToksConsumed,void * CastInfo,bool IsUnevaluatedContext)183 ExprResult Parser::ParseMSAsmIdentifier(llvm::SmallVectorImpl<Token> &LineToks,
184                                         unsigned &NumLineToksConsumed,
185                                         void *CastInfo,
186                                         bool IsUnevaluatedContext) {
187   llvm::InlineAsmIdentifierInfo &Info =
188       *(llvm::InlineAsmIdentifierInfo *)CastInfo;
189 
190   // Push a fake token on the end so that we don't overrun the token
191   // stream.  We use ';' because it expression-parsing should never
192   // overrun it.
193   const tok::TokenKind EndOfStream = tok::semi;
194   Token EndOfStreamTok;
195   EndOfStreamTok.startToken();
196   EndOfStreamTok.setKind(EndOfStream);
197   LineToks.push_back(EndOfStreamTok);
198 
199   // Also copy the current token over.
200   LineToks.push_back(Tok);
201 
202   PP.EnterTokenStream(LineToks.begin(), LineToks.size(),
203                       /*disable macros*/ true,
204                       /*owns tokens*/ false);
205 
206   // Clear the current token and advance to the first token in LineToks.
207   ConsumeAnyToken();
208 
209   // Parse an optional scope-specifier if we're in C++.
210   CXXScopeSpec SS;
211   if (getLangOpts().CPlusPlus) {
212     ParseOptionalCXXScopeSpecifier(SS, ParsedType(), /*EnteringContext=*/false);
213   }
214 
215   // Require an identifier here.
216   SourceLocation TemplateKWLoc;
217   UnqualifiedId Id;
218   bool Invalid =
219       ParseUnqualifiedId(SS,
220                          /*EnteringContext=*/false,
221                          /*AllowDestructorName=*/false,
222                          /*AllowConstructorName=*/false,
223                          /*ObjectType=*/ParsedType(), TemplateKWLoc, Id);
224 
225   // Figure out how many tokens we are into LineToks.
226   unsigned LineIndex = 0;
227   if (Tok.is(EndOfStream)) {
228     LineIndex = LineToks.size() - 2;
229   } else {
230     while (LineToks[LineIndex].getLocation() != Tok.getLocation()) {
231       LineIndex++;
232       assert(LineIndex < LineToks.size() - 2); // we added two extra tokens
233     }
234   }
235 
236   // If we've run into the poison token we inserted before, or there
237   // was a parsing error, then claim the entire line.
238   if (Invalid || Tok.is(EndOfStream)) {
239     NumLineToksConsumed = LineToks.size() - 2;
240   } else {
241     // Otherwise, claim up to the start of the next token.
242     NumLineToksConsumed = LineIndex;
243   }
244 
245   // Finally, restore the old parsing state by consuming all the tokens we
246   // staged before, implicitly killing off the token-lexer we pushed.
247   for (unsigned i = 0, e = LineToks.size() - LineIndex - 2; i != e; ++i) {
248     ConsumeAnyToken();
249   }
250   assert(Tok.is(EndOfStream));
251   ConsumeToken();
252 
253   // Leave LineToks in its original state.
254   LineToks.pop_back();
255   LineToks.pop_back();
256 
257   // Perform the lookup.
258   return Actions.LookupInlineAsmIdentifier(SS, TemplateKWLoc, Id, Info,
259                                            IsUnevaluatedContext);
260 }
261 
262 /// Turn a sequence of our tokens back into a string that we can hand
263 /// to the MC asm parser.
buildMSAsmString(Preprocessor & PP,SourceLocation AsmLoc,ArrayRef<Token> AsmToks,SmallVectorImpl<unsigned> & TokOffsets,SmallString<512> & Asm)264 static bool buildMSAsmString(Preprocessor &PP, SourceLocation AsmLoc,
265                              ArrayRef<Token> AsmToks,
266                              SmallVectorImpl<unsigned> &TokOffsets,
267                              SmallString<512> &Asm) {
268   assert(!AsmToks.empty() && "Didn't expect an empty AsmToks!");
269 
270   // Is this the start of a new assembly statement?
271   bool isNewStatement = true;
272 
273   for (unsigned i = 0, e = AsmToks.size(); i < e; ++i) {
274     const Token &Tok = AsmToks[i];
275 
276     // Start each new statement with a newline and a tab.
277     if (!isNewStatement && (Tok.is(tok::kw_asm) || Tok.isAtStartOfLine())) {
278       Asm += "\n\t";
279       isNewStatement = true;
280     }
281 
282     // Preserve the existence of leading whitespace except at the
283     // start of a statement.
284     if (!isNewStatement && Tok.hasLeadingSpace())
285       Asm += ' ';
286 
287     // Remember the offset of this token.
288     TokOffsets.push_back(Asm.size());
289 
290     // Don't actually write '__asm' into the assembly stream.
291     if (Tok.is(tok::kw_asm)) {
292       // Complain about __asm at the end of the stream.
293       if (i + 1 == e) {
294         PP.Diag(AsmLoc, diag::err_asm_empty);
295         return true;
296       }
297 
298       continue;
299     }
300 
301     // Append the spelling of the token.
302     SmallString<32> SpellingBuffer;
303     bool SpellingInvalid = false;
304     Asm += PP.getSpelling(Tok, SpellingBuffer, &SpellingInvalid);
305     assert(!SpellingInvalid && "spelling was invalid after correct parse?");
306 
307     // We are no longer at the start of a statement.
308     isNewStatement = false;
309   }
310 
311   // Ensure that the buffer is null-terminated.
312   Asm.push_back('\0');
313   Asm.pop_back();
314 
315   assert(TokOffsets.size() == AsmToks.size());
316   return false;
317 }
318 
319 /// ParseMicrosoftAsmStatement. When -fms-extensions/-fasm-blocks is enabled,
320 /// this routine is called to collect the tokens for an MS asm statement.
321 ///
322 /// [MS]  ms-asm-statement:
323 ///         ms-asm-block
324 ///         ms-asm-block ms-asm-statement
325 ///
326 /// [MS]  ms-asm-block:
327 ///         '__asm' ms-asm-line '\n'
328 ///         '__asm' '{' ms-asm-instruction-block[opt] '}' ';'[opt]
329 ///
330 /// [MS]  ms-asm-instruction-block
331 ///         ms-asm-line
332 ///         ms-asm-line '\n' ms-asm-instruction-block
333 ///
ParseMicrosoftAsmStatement(SourceLocation AsmLoc)334 StmtResult Parser::ParseMicrosoftAsmStatement(SourceLocation AsmLoc) {
335   SourceManager &SrcMgr = PP.getSourceManager();
336   SourceLocation EndLoc = AsmLoc;
337   SmallVector<Token, 4> AsmToks;
338 
339   bool SingleLineMode = true;
340   unsigned BraceNesting = 0;
341   unsigned short savedBraceCount = BraceCount;
342   bool InAsmComment = false;
343   FileID FID;
344   unsigned LineNo = 0;
345   unsigned NumTokensRead = 0;
346   SmallVector<SourceLocation, 4> LBraceLocs;
347   bool SkippedStartOfLine = false;
348 
349   if (Tok.is(tok::l_brace)) {
350     // Braced inline asm: consume the opening brace.
351     SingleLineMode = false;
352     BraceNesting = 1;
353     EndLoc = ConsumeBrace();
354     LBraceLocs.push_back(EndLoc);
355     ++NumTokensRead;
356   } else {
357     // Single-line inline asm; compute which line it is on.
358     std::pair<FileID, unsigned> ExpAsmLoc =
359         SrcMgr.getDecomposedExpansionLoc(EndLoc);
360     FID = ExpAsmLoc.first;
361     LineNo = SrcMgr.getLineNumber(FID, ExpAsmLoc.second);
362     LBraceLocs.push_back(SourceLocation());
363   }
364 
365   SourceLocation TokLoc = Tok.getLocation();
366   do {
367     // If we hit EOF, we're done, period.
368     if (isEofOrEom())
369       break;
370 
371     if (!InAsmComment && Tok.is(tok::l_brace)) {
372       // Consume the opening brace.
373       SkippedStartOfLine = Tok.isAtStartOfLine();
374       EndLoc = ConsumeBrace();
375       BraceNesting++;
376       LBraceLocs.push_back(EndLoc);
377       TokLoc = Tok.getLocation();
378       ++NumTokensRead;
379       continue;
380     } else if (!InAsmComment && Tok.is(tok::semi)) {
381       // A semicolon in an asm is the start of a comment.
382       InAsmComment = true;
383       if (!SingleLineMode) {
384         // Compute which line the comment is on.
385         std::pair<FileID, unsigned> ExpSemiLoc =
386             SrcMgr.getDecomposedExpansionLoc(TokLoc);
387         FID = ExpSemiLoc.first;
388         LineNo = SrcMgr.getLineNumber(FID, ExpSemiLoc.second);
389       }
390     } else if (SingleLineMode || InAsmComment) {
391       // If end-of-line is significant, check whether this token is on a
392       // new line.
393       std::pair<FileID, unsigned> ExpLoc =
394           SrcMgr.getDecomposedExpansionLoc(TokLoc);
395       if (ExpLoc.first != FID ||
396           SrcMgr.getLineNumber(ExpLoc.first, ExpLoc.second) != LineNo) {
397         // If this is a single-line __asm, we're done, except if the next
398         // line begins with an __asm too, in which case we finish a comment
399         // if needed and then keep processing the next line as a single
400         // line __asm.
401         bool isAsm = Tok.is(tok::kw_asm);
402         if (SingleLineMode && !isAsm)
403           break;
404         // We're no longer in a comment.
405         InAsmComment = false;
406         if (isAsm) {
407           LineNo = SrcMgr.getLineNumber(ExpLoc.first, ExpLoc.second);
408           SkippedStartOfLine = Tok.isAtStartOfLine();
409         }
410       } else if (!InAsmComment && Tok.is(tok::r_brace)) {
411         // In MSVC mode, braces only participate in brace matching and
412         // separating the asm statements.  This is an intentional
413         // departure from the Apple gcc behavior.
414         if (!BraceNesting)
415           break;
416       }
417     }
418     if (!InAsmComment && BraceNesting && Tok.is(tok::r_brace) &&
419         BraceCount == (savedBraceCount + BraceNesting)) {
420       // Consume the closing brace.
421       SkippedStartOfLine = Tok.isAtStartOfLine();
422       EndLoc = ConsumeBrace();
423       BraceNesting--;
424       // Finish if all of the opened braces in the inline asm section were
425       // consumed.
426       if (BraceNesting == 0 && !SingleLineMode)
427         break;
428       else {
429         LBraceLocs.pop_back();
430         TokLoc = Tok.getLocation();
431         ++NumTokensRead;
432         continue;
433       }
434     }
435 
436     // Consume the next token; make sure we don't modify the brace count etc.
437     // if we are in a comment.
438     EndLoc = TokLoc;
439     if (InAsmComment)
440       PP.Lex(Tok);
441     else {
442       // Set the token as the start of line if we skipped the original start
443       // of line token in case it was a nested brace.
444       if (SkippedStartOfLine)
445         Tok.setFlag(Token::StartOfLine);
446       AsmToks.push_back(Tok);
447       ConsumeAnyToken();
448     }
449     TokLoc = Tok.getLocation();
450     ++NumTokensRead;
451     SkippedStartOfLine = false;
452   } while (1);
453 
454   if (BraceNesting && BraceCount != savedBraceCount) {
455     // __asm without closing brace (this can happen at EOF).
456     for (unsigned i = 0; i < BraceNesting; ++i) {
457       Diag(Tok, diag::err_expected) << tok::r_brace;
458       Diag(LBraceLocs.back(), diag::note_matching) << tok::l_brace;
459       LBraceLocs.pop_back();
460     }
461     return StmtError();
462   } else if (NumTokensRead == 0) {
463     // Empty __asm.
464     Diag(Tok, diag::err_expected) << tok::l_brace;
465     return StmtError();
466   }
467 
468   // Okay, prepare to use MC to parse the assembly.
469   SmallVector<StringRef, 4> ConstraintRefs;
470   SmallVector<Expr *, 4> Exprs;
471   SmallVector<StringRef, 4> ClobberRefs;
472 
473   // We need an actual supported target.
474   const llvm::Triple &TheTriple = Actions.Context.getTargetInfo().getTriple();
475   llvm::Triple::ArchType ArchTy = TheTriple.getArch();
476   const std::string &TT = TheTriple.getTriple();
477   const llvm::Target *TheTarget = nullptr;
478   bool UnsupportedArch =
479       (ArchTy != llvm::Triple::x86 && ArchTy != llvm::Triple::x86_64);
480   if (UnsupportedArch) {
481     Diag(AsmLoc, diag::err_msasm_unsupported_arch) << TheTriple.getArchName();
482   } else {
483     std::string Error;
484     TheTarget = llvm::TargetRegistry::lookupTarget(TT, Error);
485     if (!TheTarget)
486       Diag(AsmLoc, diag::err_msasm_unable_to_create_target) << Error;
487   }
488 
489   assert(!LBraceLocs.empty() && "Should have at least one location here");
490 
491   // If we don't support assembly, or the assembly is empty, we don't
492   // need to instantiate the AsmParser, etc.
493   if (!TheTarget || AsmToks.empty()) {
494     return Actions.ActOnMSAsmStmt(AsmLoc, LBraceLocs[0], AsmToks, StringRef(),
495                                   /*NumOutputs*/ 0, /*NumInputs*/ 0,
496                                   ConstraintRefs, ClobberRefs, Exprs, EndLoc);
497   }
498 
499   // Expand the tokens into a string buffer.
500   SmallString<512> AsmString;
501   SmallVector<unsigned, 8> TokOffsets;
502   if (buildMSAsmString(PP, AsmLoc, AsmToks, TokOffsets, AsmString))
503     return StmtError();
504 
505   std::unique_ptr<llvm::MCRegisterInfo> MRI(TheTarget->createMCRegInfo(TT));
506   std::unique_ptr<llvm::MCAsmInfo> MAI(TheTarget->createMCAsmInfo(*MRI, TT));
507   // Get the instruction descriptor.
508   std::unique_ptr<llvm::MCInstrInfo> MII(TheTarget->createMCInstrInfo());
509   std::unique_ptr<llvm::MCObjectFileInfo> MOFI(new llvm::MCObjectFileInfo());
510   std::unique_ptr<llvm::MCSubtargetInfo> STI(
511       TheTarget->createMCSubtargetInfo(TT, "", ""));
512 
513   llvm::SourceMgr TempSrcMgr;
514   llvm::MCContext Ctx(MAI.get(), MRI.get(), MOFI.get(), &TempSrcMgr);
515   MOFI->InitMCObjectFileInfo(TT, llvm::Reloc::Default, llvm::CodeModel::Default,
516                              Ctx);
517   std::unique_ptr<llvm::MemoryBuffer> Buffer =
518       llvm::MemoryBuffer::getMemBuffer(AsmString, "<MS inline asm>");
519 
520   // Tell SrcMgr about this buffer, which is what the parser will pick up.
521   TempSrcMgr.AddNewSourceBuffer(std::move(Buffer), llvm::SMLoc());
522 
523   std::unique_ptr<llvm::MCStreamer> Str(createNullStreamer(Ctx));
524   std::unique_ptr<llvm::MCAsmParser> Parser(
525       createMCAsmParser(TempSrcMgr, Ctx, *Str.get(), *MAI));
526 
527   // FIXME: init MCOptions from sanitizer flags here.
528   llvm::MCTargetOptions MCOptions;
529   std::unique_ptr<llvm::MCTargetAsmParser> TargetParser(
530       TheTarget->createMCAsmParser(*STI, *Parser, *MII, MCOptions));
531 
532   std::unique_ptr<llvm::MCInstPrinter> IP(
533       TheTarget->createMCInstPrinter(llvm::Triple(TT), 1, *MAI, *MII, *MRI));
534 
535   // Change to the Intel dialect.
536   Parser->setAssemblerDialect(1);
537   Parser->setTargetParser(*TargetParser.get());
538   Parser->setParsingInlineAsm(true);
539   TargetParser->setParsingInlineAsm(true);
540 
541   ClangAsmParserCallback Callback(*this, AsmLoc, AsmString, AsmToks,
542                                   TokOffsets);
543   TargetParser->setSemaCallback(&Callback);
544   TempSrcMgr.setDiagHandler(ClangAsmParserCallback::DiagHandlerCallback,
545                             &Callback);
546 
547   unsigned NumOutputs;
548   unsigned NumInputs;
549   std::string AsmStringIR;
550   SmallVector<std::pair<void *, bool>, 4> OpExprs;
551   SmallVector<std::string, 4> Constraints;
552   SmallVector<std::string, 4> Clobbers;
553   if (Parser->parseMSInlineAsm(AsmLoc.getPtrEncoding(), AsmStringIR, NumOutputs,
554                                NumInputs, OpExprs, Constraints, Clobbers,
555                                MII.get(), IP.get(), Callback))
556     return StmtError();
557 
558   // Filter out "fpsw".  Clang doesn't accept it, and it always lists flags and
559   // fpsr as clobbers.
560   auto End = std::remove(Clobbers.begin(), Clobbers.end(), "fpsw");
561   Clobbers.erase(End, Clobbers.end());
562 
563   // Build the vector of clobber StringRefs.
564   ClobberRefs.insert(ClobberRefs.end(), Clobbers.begin(), Clobbers.end());
565 
566   // Recast the void pointers and build the vector of constraint StringRefs.
567   unsigned NumExprs = NumOutputs + NumInputs;
568   ConstraintRefs.resize(NumExprs);
569   Exprs.resize(NumExprs);
570   for (unsigned i = 0, e = NumExprs; i != e; ++i) {
571     Expr *OpExpr = static_cast<Expr *>(OpExprs[i].first);
572     if (!OpExpr)
573       return StmtError();
574 
575     // Need address of variable.
576     if (OpExprs[i].second)
577       OpExpr =
578           Actions.BuildUnaryOp(getCurScope(), AsmLoc, UO_AddrOf, OpExpr).get();
579 
580     ConstraintRefs[i] = StringRef(Constraints[i]);
581     Exprs[i] = OpExpr;
582   }
583 
584   // FIXME: We should be passing source locations for better diagnostics.
585   return Actions.ActOnMSAsmStmt(AsmLoc, LBraceLocs[0], AsmToks, AsmStringIR,
586                                 NumOutputs, NumInputs, ConstraintRefs,
587                                 ClobberRefs, Exprs, EndLoc);
588 }
589 
590 /// ParseAsmStatement - Parse a GNU extended asm statement.
591 ///       asm-statement:
592 ///         gnu-asm-statement
593 ///         ms-asm-statement
594 ///
595 /// [GNU] gnu-asm-statement:
596 ///         'asm' type-qualifier[opt] '(' asm-argument ')' ';'
597 ///
598 /// [GNU] asm-argument:
599 ///         asm-string-literal
600 ///         asm-string-literal ':' asm-operands[opt]
601 ///         asm-string-literal ':' asm-operands[opt] ':' asm-operands[opt]
602 ///         asm-string-literal ':' asm-operands[opt] ':' asm-operands[opt]
603 ///                 ':' asm-clobbers
604 ///
605 /// [GNU] asm-clobbers:
606 ///         asm-string-literal
607 ///         asm-clobbers ',' asm-string-literal
608 ///
ParseAsmStatement(bool & msAsm)609 StmtResult Parser::ParseAsmStatement(bool &msAsm) {
610   assert(Tok.is(tok::kw_asm) && "Not an asm stmt");
611   SourceLocation AsmLoc = ConsumeToken();
612 
613   if (getLangOpts().AsmBlocks && Tok.isNot(tok::l_paren) &&
614       !isTypeQualifier()) {
615     msAsm = true;
616     return ParseMicrosoftAsmStatement(AsmLoc);
617   }
618 
619   // Check if GNU-style inline Asm is disabled.
620   if (!getLangOpts().GNUAsm)
621     Diag(AsmLoc, diag::err_gnu_inline_asm_disabled);
622 
623   DeclSpec DS(AttrFactory);
624   SourceLocation Loc = Tok.getLocation();
625   ParseTypeQualifierListOpt(DS, AR_VendorAttributesParsed);
626 
627   // GNU asms accept, but warn, about type-qualifiers other than volatile.
628   if (DS.getTypeQualifiers() & DeclSpec::TQ_const)
629     Diag(Loc, diag::w_asm_qualifier_ignored) << "const";
630   if (DS.getTypeQualifiers() & DeclSpec::TQ_restrict)
631     Diag(Loc, diag::w_asm_qualifier_ignored) << "restrict";
632   // FIXME: Once GCC supports _Atomic, check whether it permits it here.
633   if (DS.getTypeQualifiers() & DeclSpec::TQ_atomic)
634     Diag(Loc, diag::w_asm_qualifier_ignored) << "_Atomic";
635 
636   // Remember if this was a volatile asm.
637   bool isVolatile = DS.getTypeQualifiers() & DeclSpec::TQ_volatile;
638   if (Tok.isNot(tok::l_paren)) {
639     Diag(Tok, diag::err_expected_lparen_after) << "asm";
640     SkipUntil(tok::r_paren, StopAtSemi);
641     return StmtError();
642   }
643   BalancedDelimiterTracker T(*this, tok::l_paren);
644   T.consumeOpen();
645 
646   ExprResult AsmString(ParseAsmStringLiteral());
647   if (AsmString.isInvalid()) {
648     // Consume up to and including the closing paren.
649     T.skipToEnd();
650     return StmtError();
651   }
652 
653   SmallVector<IdentifierInfo *, 4> Names;
654   ExprVector Constraints;
655   ExprVector Exprs;
656   ExprVector Clobbers;
657 
658   if (Tok.is(tok::r_paren)) {
659     // We have a simple asm expression like 'asm("foo")'.
660     T.consumeClose();
661     return Actions.ActOnGCCAsmStmt(AsmLoc, /*isSimple*/ true, isVolatile,
662                                    /*NumOutputs*/ 0, /*NumInputs*/ 0, nullptr,
663                                    Constraints, Exprs, AsmString.get(),
664                                    Clobbers, T.getCloseLocation());
665   }
666 
667   // Parse Outputs, if present.
668   bool AteExtraColon = false;
669   if (Tok.is(tok::colon) || Tok.is(tok::coloncolon)) {
670     // In C++ mode, parse "::" like ": :".
671     AteExtraColon = Tok.is(tok::coloncolon);
672     ConsumeToken();
673 
674     if (!AteExtraColon && ParseAsmOperandsOpt(Names, Constraints, Exprs))
675       return StmtError();
676   }
677 
678   unsigned NumOutputs = Names.size();
679 
680   // Parse Inputs, if present.
681   if (AteExtraColon || Tok.is(tok::colon) || Tok.is(tok::coloncolon)) {
682     // In C++ mode, parse "::" like ": :".
683     if (AteExtraColon)
684       AteExtraColon = false;
685     else {
686       AteExtraColon = Tok.is(tok::coloncolon);
687       ConsumeToken();
688     }
689 
690     if (!AteExtraColon && ParseAsmOperandsOpt(Names, Constraints, Exprs))
691       return StmtError();
692   }
693 
694   assert(Names.size() == Constraints.size() &&
695          Constraints.size() == Exprs.size() && "Input operand size mismatch!");
696 
697   unsigned NumInputs = Names.size() - NumOutputs;
698 
699   // Parse the clobbers, if present.
700   if (AteExtraColon || Tok.is(tok::colon)) {
701     if (!AteExtraColon)
702       ConsumeToken();
703 
704     // Parse the asm-string list for clobbers if present.
705     if (Tok.isNot(tok::r_paren)) {
706       while (1) {
707         ExprResult Clobber(ParseAsmStringLiteral());
708 
709         if (Clobber.isInvalid())
710           break;
711 
712         Clobbers.push_back(Clobber.get());
713 
714         if (!TryConsumeToken(tok::comma))
715           break;
716       }
717     }
718   }
719 
720   T.consumeClose();
721   return Actions.ActOnGCCAsmStmt(
722       AsmLoc, false, isVolatile, NumOutputs, NumInputs, Names.data(),
723       Constraints, Exprs, AsmString.get(), Clobbers, T.getCloseLocation());
724 }
725 
726 /// ParseAsmOperands - Parse the asm-operands production as used by
727 /// asm-statement, assuming the leading ':' token was eaten.
728 ///
729 /// [GNU] asm-operands:
730 ///         asm-operand
731 ///         asm-operands ',' asm-operand
732 ///
733 /// [GNU] asm-operand:
734 ///         asm-string-literal '(' expression ')'
735 ///         '[' identifier ']' asm-string-literal '(' expression ')'
736 ///
737 //
738 // FIXME: Avoid unnecessary std::string trashing.
ParseAsmOperandsOpt(SmallVectorImpl<IdentifierInfo * > & Names,SmallVectorImpl<Expr * > & Constraints,SmallVectorImpl<Expr * > & Exprs)739 bool Parser::ParseAsmOperandsOpt(SmallVectorImpl<IdentifierInfo *> &Names,
740                                  SmallVectorImpl<Expr *> &Constraints,
741                                  SmallVectorImpl<Expr *> &Exprs) {
742   // 'asm-operands' isn't present?
743   if (!isTokenStringLiteral() && Tok.isNot(tok::l_square))
744     return false;
745 
746   while (1) {
747     // Read the [id] if present.
748     if (Tok.is(tok::l_square)) {
749       BalancedDelimiterTracker T(*this, tok::l_square);
750       T.consumeOpen();
751 
752       if (Tok.isNot(tok::identifier)) {
753         Diag(Tok, diag::err_expected) << tok::identifier;
754         SkipUntil(tok::r_paren, StopAtSemi);
755         return true;
756       }
757 
758       IdentifierInfo *II = Tok.getIdentifierInfo();
759       ConsumeToken();
760 
761       Names.push_back(II);
762       T.consumeClose();
763     } else
764       Names.push_back(nullptr);
765 
766     ExprResult Constraint(ParseAsmStringLiteral());
767     if (Constraint.isInvalid()) {
768       SkipUntil(tok::r_paren, StopAtSemi);
769       return true;
770     }
771     Constraints.push_back(Constraint.get());
772 
773     if (Tok.isNot(tok::l_paren)) {
774       Diag(Tok, diag::err_expected_lparen_after) << "asm operand";
775       SkipUntil(tok::r_paren, StopAtSemi);
776       return true;
777     }
778 
779     // Read the parenthesized expression.
780     BalancedDelimiterTracker T(*this, tok::l_paren);
781     T.consumeOpen();
782     ExprResult Res = Actions.CorrectDelayedTyposInExpr(ParseExpression());
783     T.consumeClose();
784     if (Res.isInvalid()) {
785       SkipUntil(tok::r_paren, StopAtSemi);
786       return true;
787     }
788     Exprs.push_back(Res.get());
789     // Eat the comma and continue parsing if it exists.
790     if (!TryConsumeToken(tok::comma))
791       return false;
792   }
793 }
794