blob: 181bd413c5d1e14625230970a0159b754193d3d6 [file] [log] [blame]
Alp Toker1b935a82014-06-08 05:40:04 +00001//===---- 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"
Alp Toker1b935a82014-06-08 05:40:04 +000015#include "clang/AST/ASTContext.h"
16#include "clang/Basic/Diagnostic.h"
17#include "clang/Basic/TargetInfo.h"
Vassil Vassilev11ad3392017-03-23 15:11:07 +000018#include "clang/Parse/RAIIObjectsForParser.h"
Alp Toker1b935a82014-06-08 05:40:04 +000019#include "llvm/ADT/SmallString.h"
Marina Yatsina41c45fa2016-02-03 11:32:08 +000020#include "llvm/ADT/StringExtras.h"
Alp Toker1b935a82014-06-08 05:40:04 +000021#include "llvm/MC/MCAsmInfo.h"
22#include "llvm/MC/MCContext.h"
23#include "llvm/MC/MCInstPrinter.h"
24#include "llvm/MC/MCInstrInfo.h"
25#include "llvm/MC/MCObjectFileInfo.h"
26#include "llvm/MC/MCParser/MCAsmParser.h"
Benjamin Kramer5e456302016-01-27 10:01:30 +000027#include "llvm/MC/MCParser/MCTargetAsmParser.h"
Alp Toker1b935a82014-06-08 05:40:04 +000028#include "llvm/MC/MCRegisterInfo.h"
29#include "llvm/MC/MCStreamer.h"
30#include "llvm/MC/MCSubtargetInfo.h"
Alp Toker1b935a82014-06-08 05:40:04 +000031#include "llvm/MC/MCTargetOptions.h"
32#include "llvm/Support/SourceMgr.h"
33#include "llvm/Support/TargetRegistry.h"
34#include "llvm/Support/TargetSelect.h"
35using namespace clang;
36
37namespace {
38class ClangAsmParserCallback : public llvm::MCAsmParserSemaCallback {
39 Parser &TheParser;
40 SourceLocation AsmLoc;
41 StringRef AsmString;
42
43 /// The tokens we streamed into AsmString and handed off to MC.
44 ArrayRef<Token> AsmToks;
45
46 /// The offset of each token in AsmToks within AsmString.
47 ArrayRef<unsigned> AsmTokOffsets;
48
49public:
50 ClangAsmParserCallback(Parser &P, SourceLocation Loc, StringRef AsmString,
51 ArrayRef<Token> Toks, ArrayRef<unsigned> Offsets)
52 : TheParser(P), AsmLoc(Loc), AsmString(AsmString), AsmToks(Toks),
53 AsmTokOffsets(Offsets) {
54 assert(AsmToks.size() == AsmTokOffsets.size());
55 }
56
57 void *LookupInlineAsmIdentifier(StringRef &LineBuf,
58 llvm::InlineAsmIdentifierInfo &Info,
59 bool IsUnevaluatedContext) override {
60 // Collect the desired tokens.
61 SmallVector<Token, 16> LineToks;
62 const Token *FirstOrigToken = nullptr;
63 findTokensForString(LineBuf, LineToks, FirstOrigToken);
64
65 unsigned NumConsumedToks;
66 ExprResult Result = TheParser.ParseMSAsmIdentifier(
67 LineToks, NumConsumedToks, &Info, IsUnevaluatedContext);
68
69 // If we consumed the entire line, tell MC that.
70 // Also do this if we consumed nothing as a way of reporting failure.
71 if (NumConsumedToks == 0 || NumConsumedToks == LineToks.size()) {
72 // By not modifying LineBuf, we're implicitly consuming it all.
73
74 // Otherwise, consume up to the original tokens.
75 } else {
76 assert(FirstOrigToken && "not using original tokens?");
77
78 // Since we're using original tokens, apply that offset.
79 assert(FirstOrigToken[NumConsumedToks].getLocation() ==
80 LineToks[NumConsumedToks].getLocation());
81 unsigned FirstIndex = FirstOrigToken - AsmToks.begin();
82 unsigned LastIndex = FirstIndex + NumConsumedToks - 1;
83
84 // The total length we've consumed is the relative offset
85 // of the last token we consumed plus its length.
86 unsigned TotalOffset =
87 (AsmTokOffsets[LastIndex] + AsmToks[LastIndex].getLength() -
88 AsmTokOffsets[FirstIndex]);
89 LineBuf = LineBuf.substr(0, TotalOffset);
90 }
91
92 // Initialize the "decl" with the lookup result.
93 Info.OpDecl = static_cast<void *>(Result.get());
94 return Info.OpDecl;
95 }
96
Ehsan Akhgari31097582014-09-22 02:21:54 +000097 StringRef LookupInlineAsmLabel(StringRef Identifier, llvm::SourceMgr &LSM,
98 llvm::SMLoc Location,
99 bool Create) override {
100 SourceLocation Loc = translateLocation(LSM, Location);
101 LabelDecl *Label =
102 TheParser.getActions().GetOrCreateMSAsmLabel(Identifier, Loc, Create);
103 return Label->getMSAsmLabel();
104 }
105
Alp Toker1b935a82014-06-08 05:40:04 +0000106 bool LookupInlineAsmField(StringRef Base, StringRef Member,
107 unsigned &Offset) override {
108 return TheParser.getActions().LookupInlineAsmField(Base, Member, Offset,
109 AsmLoc);
110 }
111
112 static void DiagHandlerCallback(const llvm::SMDiagnostic &D, void *Context) {
113 ((ClangAsmParserCallback *)Context)->handleDiagnostic(D);
114 }
115
116private:
117 /// Collect the appropriate tokens for the given string.
118 void findTokensForString(StringRef Str, SmallVectorImpl<Token> &TempToks,
119 const Token *&FirstOrigToken) const {
120 // For now, assert that the string we're working with is a substring
121 // of what we gave to MC. This lets us use the original tokens.
122 assert(!std::less<const char *>()(Str.begin(), AsmString.begin()) &&
123 !std::less<const char *>()(AsmString.end(), Str.end()));
124
125 // Try to find a token whose offset matches the first token.
126 unsigned FirstCharOffset = Str.begin() - AsmString.begin();
127 const unsigned *FirstTokOffset = std::lower_bound(
128 AsmTokOffsets.begin(), AsmTokOffsets.end(), FirstCharOffset);
129
130 // For now, assert that the start of the string exactly
131 // corresponds to the start of a token.
132 assert(*FirstTokOffset == FirstCharOffset);
133
134 // Use all the original tokens for this line. (We assume the
135 // end of the line corresponds cleanly to a token break.)
136 unsigned FirstTokIndex = FirstTokOffset - AsmTokOffsets.begin();
137 FirstOrigToken = &AsmToks[FirstTokIndex];
138 unsigned LastCharOffset = Str.end() - AsmString.begin();
139 for (unsigned i = FirstTokIndex, e = AsmTokOffsets.size(); i != e; ++i) {
140 if (AsmTokOffsets[i] >= LastCharOffset)
141 break;
142 TempToks.push_back(AsmToks[i]);
143 }
144 }
145
Ehsan Akhgari31097582014-09-22 02:21:54 +0000146 SourceLocation translateLocation(const llvm::SourceMgr &LSM, llvm::SMLoc SMLoc) {
Alp Toker1b935a82014-06-08 05:40:04 +0000147 // Compute an offset into the inline asm buffer.
148 // FIXME: This isn't right if .macro is involved (but hopefully, no
149 // real-world code does that).
Alp Toker1b935a82014-06-08 05:40:04 +0000150 const llvm::MemoryBuffer *LBuf =
Ehsan Akhgari31097582014-09-22 02:21:54 +0000151 LSM.getMemoryBuffer(LSM.FindBufferContainingLoc(SMLoc));
152 unsigned Offset = SMLoc.getPointer() - LBuf->getBufferStart();
Alp Toker1b935a82014-06-08 05:40:04 +0000153
154 // Figure out which token that offset points into.
155 const unsigned *TokOffsetPtr =
156 std::lower_bound(AsmTokOffsets.begin(), AsmTokOffsets.end(), Offset);
157 unsigned TokIndex = TokOffsetPtr - AsmTokOffsets.begin();
158 unsigned TokOffset = *TokOffsetPtr;
159
160 // If we come up with an answer which seems sane, use it; otherwise,
161 // just point at the __asm keyword.
162 // FIXME: Assert the answer is sane once we handle .macro correctly.
163 SourceLocation Loc = AsmLoc;
164 if (TokIndex < AsmToks.size()) {
165 const Token &Tok = AsmToks[TokIndex];
166 Loc = Tok.getLocation();
167 Loc = Loc.getLocWithOffset(Offset - TokOffset);
168 }
Ehsan Akhgari31097582014-09-22 02:21:54 +0000169 return Loc;
170 }
171
172 void handleDiagnostic(const llvm::SMDiagnostic &D) {
173 const llvm::SourceMgr &LSM = *D.getSourceMgr();
174 SourceLocation Loc = translateLocation(LSM, D.getLoc());
Alp Toker1b935a82014-06-08 05:40:04 +0000175 TheParser.Diag(Loc, diag::err_inline_ms_asm_parsing) << D.getMessage();
176 }
177};
Alexander Kornienkoab9db512015-06-22 23:07:51 +0000178}
Alp Toker1b935a82014-06-08 05:40:04 +0000179
180/// Parse an identifier in an MS-style inline assembly block.
181///
182/// \param CastInfo - a void* so that we don't have to teach Parser.h
183/// about the actual type.
184ExprResult Parser::ParseMSAsmIdentifier(llvm::SmallVectorImpl<Token> &LineToks,
185 unsigned &NumLineToksConsumed,
186 void *CastInfo,
187 bool IsUnevaluatedContext) {
188 llvm::InlineAsmIdentifierInfo &Info =
189 *(llvm::InlineAsmIdentifierInfo *)CastInfo;
190
191 // Push a fake token on the end so that we don't overrun the token
192 // stream. We use ';' because it expression-parsing should never
193 // overrun it.
194 const tok::TokenKind EndOfStream = tok::semi;
195 Token EndOfStreamTok;
196 EndOfStreamTok.startToken();
197 EndOfStreamTok.setKind(EndOfStream);
198 LineToks.push_back(EndOfStreamTok);
199
200 // Also copy the current token over.
201 LineToks.push_back(Tok);
202
David Blaikie2eabcc92016-02-09 18:52:09 +0000203 PP.EnterTokenStream(LineToks, /*DisableMacroExpansions*/ true);
Alp Toker1b935a82014-06-08 05:40:04 +0000204
205 // Clear the current token and advance to the first token in LineToks.
206 ConsumeAnyToken();
207
208 // Parse an optional scope-specifier if we're in C++.
209 CXXScopeSpec SS;
210 if (getLangOpts().CPlusPlus) {
David Blaikieefdccaa2016-01-15 23:43:34 +0000211 ParseOptionalCXXScopeSpecifier(SS, nullptr, /*EnteringContext=*/false);
Alp Toker1b935a82014-06-08 05:40:04 +0000212 }
213
214 // Require an identifier here.
215 SourceLocation TemplateKWLoc;
216 UnqualifiedId Id;
Michael Zuckerman229158c2015-12-15 14:04:18 +0000217 bool Invalid = true;
218 ExprResult Result;
219 if (Tok.is(tok::kw_this)) {
220 Result = ParseCXXThis();
221 Invalid = false;
222 } else {
David Blaikieefdccaa2016-01-15 23:43:34 +0000223 Invalid = ParseUnqualifiedId(SS,
224 /*EnteringContext=*/false,
225 /*AllowDestructorName=*/false,
226 /*AllowConstructorName=*/false,
Richard Smith35845152017-02-07 01:37:30 +0000227 /*AllowDeductionGuide=*/false,
David Blaikieefdccaa2016-01-15 23:43:34 +0000228 /*ObjectType=*/nullptr, TemplateKWLoc, Id);
Michael Zuckerman229158c2015-12-15 14:04:18 +0000229 // Perform the lookup.
230 Result = Actions.LookupInlineAsmIdentifier(SS, TemplateKWLoc, Id, Info,
231 IsUnevaluatedContext);
232 }
Reid Kleckner14e96b42015-08-26 21:57:20 +0000233 // While the next two tokens are 'period' 'identifier', repeatedly parse it as
234 // a field access. We have to avoid consuming assembler directives that look
235 // like '.' 'else'.
236 while (Result.isUsable() && Tok.is(tok::period)) {
237 Token IdTok = PP.LookAhead(0);
238 if (IdTok.isNot(tok::identifier))
239 break;
240 ConsumeToken(); // Consume the period.
241 IdentifierInfo *Id = Tok.getIdentifierInfo();
242 ConsumeToken(); // Consume the identifier.
David Majnemer758e7982016-01-05 00:08:41 +0000243 Result = Actions.LookupInlineAsmVarDeclField(Result.get(), Id->getName(),
244 Info, Tok.getLocation());
Reid Kleckner14e96b42015-08-26 21:57:20 +0000245 }
246
Alp Toker1b935a82014-06-08 05:40:04 +0000247 // Figure out how many tokens we are into LineToks.
248 unsigned LineIndex = 0;
249 if (Tok.is(EndOfStream)) {
250 LineIndex = LineToks.size() - 2;
251 } else {
252 while (LineToks[LineIndex].getLocation() != Tok.getLocation()) {
253 LineIndex++;
254 assert(LineIndex < LineToks.size() - 2); // we added two extra tokens
255 }
256 }
257
258 // If we've run into the poison token we inserted before, or there
259 // was a parsing error, then claim the entire line.
260 if (Invalid || Tok.is(EndOfStream)) {
261 NumLineToksConsumed = LineToks.size() - 2;
262 } else {
263 // Otherwise, claim up to the start of the next token.
264 NumLineToksConsumed = LineIndex;
265 }
266
267 // Finally, restore the old parsing state by consuming all the tokens we
268 // staged before, implicitly killing off the token-lexer we pushed.
269 for (unsigned i = 0, e = LineToks.size() - LineIndex - 2; i != e; ++i) {
270 ConsumeAnyToken();
271 }
272 assert(Tok.is(EndOfStream));
273 ConsumeToken();
274
275 // Leave LineToks in its original state.
276 LineToks.pop_back();
277 LineToks.pop_back();
278
Reid Kleckner14e96b42015-08-26 21:57:20 +0000279 return Result;
Alp Toker1b935a82014-06-08 05:40:04 +0000280}
281
282/// Turn a sequence of our tokens back into a string that we can hand
283/// to the MC asm parser.
284static bool buildMSAsmString(Preprocessor &PP, SourceLocation AsmLoc,
285 ArrayRef<Token> AsmToks,
286 SmallVectorImpl<unsigned> &TokOffsets,
287 SmallString<512> &Asm) {
288 assert(!AsmToks.empty() && "Didn't expect an empty AsmToks!");
289
290 // Is this the start of a new assembly statement?
291 bool isNewStatement = true;
292
293 for (unsigned i = 0, e = AsmToks.size(); i < e; ++i) {
294 const Token &Tok = AsmToks[i];
295
296 // Start each new statement with a newline and a tab.
297 if (!isNewStatement && (Tok.is(tok::kw_asm) || Tok.isAtStartOfLine())) {
298 Asm += "\n\t";
299 isNewStatement = true;
300 }
301
302 // Preserve the existence of leading whitespace except at the
303 // start of a statement.
304 if (!isNewStatement && Tok.hasLeadingSpace())
305 Asm += ' ';
306
307 // Remember the offset of this token.
308 TokOffsets.push_back(Asm.size());
309
310 // Don't actually write '__asm' into the assembly stream.
311 if (Tok.is(tok::kw_asm)) {
312 // Complain about __asm at the end of the stream.
313 if (i + 1 == e) {
314 PP.Diag(AsmLoc, diag::err_asm_empty);
315 return true;
316 }
317
318 continue;
319 }
320
321 // Append the spelling of the token.
322 SmallString<32> SpellingBuffer;
323 bool SpellingInvalid = false;
324 Asm += PP.getSpelling(Tok, SpellingBuffer, &SpellingInvalid);
325 assert(!SpellingInvalid && "spelling was invalid after correct parse?");
326
327 // We are no longer at the start of a statement.
328 isNewStatement = false;
329 }
330
331 // Ensure that the buffer is null-terminated.
332 Asm.push_back('\0');
333 Asm.pop_back();
334
335 assert(TokOffsets.size() == AsmToks.size());
336 return false;
337}
338
Denis Zobnin628b0222016-04-21 10:59:18 +0000339/// isTypeQualifier - Return true if the current token could be the
340/// start of a type-qualifier-list.
341static bool isTypeQualifier(const Token &Tok) {
342 switch (Tok.getKind()) {
343 default: return false;
344 // type-qualifier
345 case tok::kw_const:
346 case tok::kw_volatile:
347 case tok::kw_restrict:
348 case tok::kw___private:
349 case tok::kw___local:
350 case tok::kw___global:
351 case tok::kw___constant:
352 case tok::kw___generic:
353 case tok::kw___read_only:
354 case tok::kw___read_write:
355 case tok::kw___write_only:
356 return true;
357 }
358}
359
360// Determine if this is a GCC-style asm statement.
361static bool isGCCAsmStatement(const Token &TokAfterAsm) {
362 return TokAfterAsm.is(tok::l_paren) || TokAfterAsm.is(tok::kw_goto) ||
363 isTypeQualifier(TokAfterAsm);
364}
365
Alp Toker1b935a82014-06-08 05:40:04 +0000366/// ParseMicrosoftAsmStatement. When -fms-extensions/-fasm-blocks is enabled,
367/// this routine is called to collect the tokens for an MS asm statement.
368///
369/// [MS] ms-asm-statement:
370/// ms-asm-block
371/// ms-asm-block ms-asm-statement
372///
373/// [MS] ms-asm-block:
374/// '__asm' ms-asm-line '\n'
375/// '__asm' '{' ms-asm-instruction-block[opt] '}' ';'[opt]
376///
377/// [MS] ms-asm-instruction-block
378/// ms-asm-line
379/// ms-asm-line '\n' ms-asm-instruction-block
380///
381StmtResult Parser::ParseMicrosoftAsmStatement(SourceLocation AsmLoc) {
382 SourceManager &SrcMgr = PP.getSourceManager();
383 SourceLocation EndLoc = AsmLoc;
384 SmallVector<Token, 4> AsmToks;
385
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000386 bool SingleLineMode = true;
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000387 unsigned BraceNesting = 0;
Ehsan Akhgari833ed942014-07-15 02:21:41 +0000388 unsigned short savedBraceCount = BraceCount;
Alp Toker1b935a82014-06-08 05:40:04 +0000389 bool InAsmComment = false;
390 FileID FID;
391 unsigned LineNo = 0;
392 unsigned NumTokensRead = 0;
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000393 SmallVector<SourceLocation, 4> LBraceLocs;
394 bool SkippedStartOfLine = false;
Alp Toker1b935a82014-06-08 05:40:04 +0000395
396 if (Tok.is(tok::l_brace)) {
397 // Braced inline asm: consume the opening brace.
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000398 SingleLineMode = false;
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000399 BraceNesting = 1;
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000400 EndLoc = ConsumeBrace();
401 LBraceLocs.push_back(EndLoc);
Alp Toker1b935a82014-06-08 05:40:04 +0000402 ++NumTokensRead;
403 } else {
404 // Single-line inline asm; compute which line it is on.
405 std::pair<FileID, unsigned> ExpAsmLoc =
406 SrcMgr.getDecomposedExpansionLoc(EndLoc);
407 FID = ExpAsmLoc.first;
408 LineNo = SrcMgr.getLineNumber(FID, ExpAsmLoc.second);
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000409 LBraceLocs.push_back(SourceLocation());
Alp Toker1b935a82014-06-08 05:40:04 +0000410 }
411
412 SourceLocation TokLoc = Tok.getLocation();
413 do {
414 // If we hit EOF, we're done, period.
415 if (isEofOrEom())
416 break;
417
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000418 if (!InAsmComment && Tok.is(tok::l_brace)) {
419 // Consume the opening brace.
420 SkippedStartOfLine = Tok.isAtStartOfLine();
Marina Yatsina5f776792016-03-07 18:10:25 +0000421 AsmToks.push_back(Tok);
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000422 EndLoc = ConsumeBrace();
423 BraceNesting++;
424 LBraceLocs.push_back(EndLoc);
425 TokLoc = Tok.getLocation();
426 ++NumTokensRead;
427 continue;
428 } else if (!InAsmComment && Tok.is(tok::semi)) {
Alp Toker1b935a82014-06-08 05:40:04 +0000429 // A semicolon in an asm is the start of a comment.
430 InAsmComment = true;
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000431 if (!SingleLineMode) {
Alp Toker1b935a82014-06-08 05:40:04 +0000432 // Compute which line the comment is on.
433 std::pair<FileID, unsigned> ExpSemiLoc =
434 SrcMgr.getDecomposedExpansionLoc(TokLoc);
435 FID = ExpSemiLoc.first;
436 LineNo = SrcMgr.getLineNumber(FID, ExpSemiLoc.second);
437 }
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000438 } else if (SingleLineMode || InAsmComment) {
Alp Toker1b935a82014-06-08 05:40:04 +0000439 // If end-of-line is significant, check whether this token is on a
440 // new line.
441 std::pair<FileID, unsigned> ExpLoc =
442 SrcMgr.getDecomposedExpansionLoc(TokLoc);
443 if (ExpLoc.first != FID ||
444 SrcMgr.getLineNumber(ExpLoc.first, ExpLoc.second) != LineNo) {
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000445 // If this is a single-line __asm, we're done, except if the next
Denis Zobnin628b0222016-04-21 10:59:18 +0000446 // line is MS-style asm too, in which case we finish a comment
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000447 // if needed and then keep processing the next line as a single
448 // line __asm.
449 bool isAsm = Tok.is(tok::kw_asm);
Denis Zobnin628b0222016-04-21 10:59:18 +0000450 if (SingleLineMode && (!isAsm || isGCCAsmStatement(NextToken())))
Alp Toker1b935a82014-06-08 05:40:04 +0000451 break;
452 // We're no longer in a comment.
453 InAsmComment = false;
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000454 if (isAsm) {
Simon Pilgrim2c518802017-03-30 14:13:19 +0000455 // If this is a new __asm {} block we want to process it separately
Marina Yatsina146d2ec2016-02-23 08:53:45 +0000456 // from the single-line __asm statements
457 if (PP.LookAhead(0).is(tok::l_brace))
458 break;
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000459 LineNo = SrcMgr.getLineNumber(ExpLoc.first, ExpLoc.second);
460 SkippedStartOfLine = Tok.isAtStartOfLine();
Coby Tayreec0fb36f2017-02-05 10:23:06 +0000461 } else if (Tok.is(tok::semi)) {
462 // A multi-line asm-statement, where next line is a comment
463 InAsmComment = true;
464 FID = ExpLoc.first;
465 LineNo = SrcMgr.getLineNumber(FID, ExpLoc.second);
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000466 }
Alp Toker1b935a82014-06-08 05:40:04 +0000467 } else if (!InAsmComment && Tok.is(tok::r_brace)) {
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000468 // In MSVC mode, braces only participate in brace matching and
469 // separating the asm statements. This is an intentional
470 // departure from the Apple gcc behavior.
471 if (!BraceNesting)
472 break;
Alp Toker1b935a82014-06-08 05:40:04 +0000473 }
474 }
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000475 if (!InAsmComment && BraceNesting && Tok.is(tok::r_brace) &&
476 BraceCount == (savedBraceCount + BraceNesting)) {
477 // Consume the closing brace.
478 SkippedStartOfLine = Tok.isAtStartOfLine();
Marina Yatsina5f776792016-03-07 18:10:25 +0000479 // Don't want to add the closing brace of the whole asm block
480 if (SingleLineMode || BraceNesting > 1) {
481 Tok.clearFlag(Token::LeadingSpace);
482 AsmToks.push_back(Tok);
483 }
Alp Toker1b935a82014-06-08 05:40:04 +0000484 EndLoc = ConsumeBrace();
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000485 BraceNesting--;
Nico Weber022e5072014-07-17 18:19:30 +0000486 // Finish if all of the opened braces in the inline asm section were
487 // consumed.
Ehsan Akhgari2f93b442014-07-25 02:27:14 +0000488 if (BraceNesting == 0 && !SingleLineMode)
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000489 break;
490 else {
491 LBraceLocs.pop_back();
492 TokLoc = Tok.getLocation();
493 ++NumTokensRead;
494 continue;
495 }
Alp Toker1b935a82014-06-08 05:40:04 +0000496 }
497
498 // Consume the next token; make sure we don't modify the brace count etc.
499 // if we are in a comment.
500 EndLoc = TokLoc;
501 if (InAsmComment)
502 PP.Lex(Tok);
503 else {
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000504 // Set the token as the start of line if we skipped the original start
505 // of line token in case it was a nested brace.
506 if (SkippedStartOfLine)
507 Tok.setFlag(Token::StartOfLine);
Alp Toker1b935a82014-06-08 05:40:04 +0000508 AsmToks.push_back(Tok);
509 ConsumeAnyToken();
510 }
511 TokLoc = Tok.getLocation();
512 ++NumTokensRead;
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000513 SkippedStartOfLine = false;
Alp Toker1b935a82014-06-08 05:40:04 +0000514 } while (1);
515
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000516 if (BraceNesting && BraceCount != savedBraceCount) {
Alp Toker1b935a82014-06-08 05:40:04 +0000517 // __asm without closing brace (this can happen at EOF).
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000518 for (unsigned i = 0; i < BraceNesting; ++i) {
519 Diag(Tok, diag::err_expected) << tok::r_brace;
520 Diag(LBraceLocs.back(), diag::note_matching) << tok::l_brace;
521 LBraceLocs.pop_back();
522 }
Alp Toker1b935a82014-06-08 05:40:04 +0000523 return StmtError();
524 } else if (NumTokensRead == 0) {
525 // Empty __asm.
526 Diag(Tok, diag::err_expected) << tok::l_brace;
527 return StmtError();
528 }
529
530 // Okay, prepare to use MC to parse the assembly.
531 SmallVector<StringRef, 4> ConstraintRefs;
532 SmallVector<Expr *, 4> Exprs;
533 SmallVector<StringRef, 4> ClobberRefs;
534
535 // We need an actual supported target.
536 const llvm::Triple &TheTriple = Actions.Context.getTargetInfo().getTriple();
537 llvm::Triple::ArchType ArchTy = TheTriple.getArch();
538 const std::string &TT = TheTriple.getTriple();
539 const llvm::Target *TheTarget = nullptr;
540 bool UnsupportedArch =
541 (ArchTy != llvm::Triple::x86 && ArchTy != llvm::Triple::x86_64);
542 if (UnsupportedArch) {
543 Diag(AsmLoc, diag::err_msasm_unsupported_arch) << TheTriple.getArchName();
544 } else {
545 std::string Error;
546 TheTarget = llvm::TargetRegistry::lookupTarget(TT, Error);
547 if (!TheTarget)
548 Diag(AsmLoc, diag::err_msasm_unable_to_create_target) << Error;
549 }
550
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000551 assert(!LBraceLocs.empty() && "Should have at least one location here");
552
Alp Toker1b935a82014-06-08 05:40:04 +0000553 // If we don't support assembly, or the assembly is empty, we don't
554 // need to instantiate the AsmParser, etc.
555 if (!TheTarget || AsmToks.empty()) {
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000556 return Actions.ActOnMSAsmStmt(AsmLoc, LBraceLocs[0], AsmToks, StringRef(),
Alp Toker1b935a82014-06-08 05:40:04 +0000557 /*NumOutputs*/ 0, /*NumInputs*/ 0,
558 ConstraintRefs, ClobberRefs, Exprs, EndLoc);
559 }
560
561 // Expand the tokens into a string buffer.
562 SmallString<512> AsmString;
563 SmallVector<unsigned, 8> TokOffsets;
564 if (buildMSAsmString(PP, AsmLoc, AsmToks, TokOffsets, AsmString))
565 return StmtError();
566
Marina Yatsina41c45fa2016-02-03 11:32:08 +0000567 TargetOptions TO = Actions.Context.getTargetInfo().getTargetOpts();
568 std::string FeaturesStr =
569 llvm::join(TO.Features.begin(), TO.Features.end(), ",");
570
Alp Toker1b935a82014-06-08 05:40:04 +0000571 std::unique_ptr<llvm::MCRegisterInfo> MRI(TheTarget->createMCRegInfo(TT));
572 std::unique_ptr<llvm::MCAsmInfo> MAI(TheTarget->createMCAsmInfo(*MRI, TT));
573 // Get the instruction descriptor.
574 std::unique_ptr<llvm::MCInstrInfo> MII(TheTarget->createMCInstrInfo());
575 std::unique_ptr<llvm::MCObjectFileInfo> MOFI(new llvm::MCObjectFileInfo());
576 std::unique_ptr<llvm::MCSubtargetInfo> STI(
Marina Yatsina41c45fa2016-02-03 11:32:08 +0000577 TheTarget->createMCSubtargetInfo(TT, TO.CPU, FeaturesStr));
Alp Toker1b935a82014-06-08 05:40:04 +0000578
579 llvm::SourceMgr TempSrcMgr;
580 llvm::MCContext Ctx(MAI.get(), MRI.get(), MOFI.get(), &TempSrcMgr);
Rafael Espindola2e8a7d32017-08-02 20:32:35 +0000581 MOFI->InitMCObjectFileInfo(TheTriple, /*PIC*/ false, Ctx);
Rafael Espindolad87f8d72014-08-27 20:03:29 +0000582 std::unique_ptr<llvm::MemoryBuffer> Buffer =
583 llvm::MemoryBuffer::getMemBuffer(AsmString, "<MS inline asm>");
Alp Toker1b935a82014-06-08 05:40:04 +0000584
585 // Tell SrcMgr about this buffer, which is what the parser will pick up.
David Blaikie9e095d92014-08-21 21:01:00 +0000586 TempSrcMgr.AddNewSourceBuffer(std::move(Buffer), llvm::SMLoc());
Alp Toker1b935a82014-06-08 05:40:04 +0000587
588 std::unique_ptr<llvm::MCStreamer> Str(createNullStreamer(Ctx));
589 std::unique_ptr<llvm::MCAsmParser> Parser(
590 createMCAsmParser(TempSrcMgr, Ctx, *Str.get(), *MAI));
591
592 // FIXME: init MCOptions from sanitizer flags here.
593 llvm::MCTargetOptions MCOptions;
594 std::unique_ptr<llvm::MCTargetAsmParser> TargetParser(
595 TheTarget->createMCAsmParser(*STI, *Parser, *MII, MCOptions));
596
Daniel Sanders50f17232015-09-15 16:17:27 +0000597 std::unique_ptr<llvm::MCInstPrinter> IP(
598 TheTarget->createMCInstPrinter(llvm::Triple(TT), 1, *MAI, *MII, *MRI));
Alp Toker1b935a82014-06-08 05:40:04 +0000599
600 // Change to the Intel dialect.
601 Parser->setAssemblerDialect(1);
602 Parser->setTargetParser(*TargetParser.get());
603 Parser->setParsingInlineAsm(true);
604 TargetParser->setParsingInlineAsm(true);
605
606 ClangAsmParserCallback Callback(*this, AsmLoc, AsmString, AsmToks,
607 TokOffsets);
608 TargetParser->setSemaCallback(&Callback);
609 TempSrcMgr.setDiagHandler(ClangAsmParserCallback::DiagHandlerCallback,
610 &Callback);
611
612 unsigned NumOutputs;
613 unsigned NumInputs;
614 std::string AsmStringIR;
615 SmallVector<std::pair<void *, bool>, 4> OpExprs;
616 SmallVector<std::string, 4> Constraints;
617 SmallVector<std::string, 4> Clobbers;
618 if (Parser->parseMSInlineAsm(AsmLoc.getPtrEncoding(), AsmStringIR, NumOutputs,
619 NumInputs, OpExprs, Constraints, Clobbers,
620 MII.get(), IP.get(), Callback))
621 return StmtError();
622
Reid Klecknerfb9f6472017-02-14 21:38:17 +0000623 // Filter out "fpsw" and "mxcsr". They aren't valid GCC asm clobber
624 // constraints. Clang always adds fpsr to the clobber list anyway.
625 llvm::erase_if(Clobbers, [](const std::string &C) {
626 return C == "fpsw" || C == "mxcsr";
627 });
Alp Toker1b935a82014-06-08 05:40:04 +0000628
629 // Build the vector of clobber StringRefs.
David Majnemer05c69862014-06-23 02:16:41 +0000630 ClobberRefs.insert(ClobberRefs.end(), Clobbers.begin(), Clobbers.end());
Alp Toker1b935a82014-06-08 05:40:04 +0000631
632 // Recast the void pointers and build the vector of constraint StringRefs.
633 unsigned NumExprs = NumOutputs + NumInputs;
634 ConstraintRefs.resize(NumExprs);
635 Exprs.resize(NumExprs);
636 for (unsigned i = 0, e = NumExprs; i != e; ++i) {
637 Expr *OpExpr = static_cast<Expr *>(OpExprs[i].first);
638 if (!OpExpr)
639 return StmtError();
640
641 // Need address of variable.
642 if (OpExprs[i].second)
643 OpExpr =
644 Actions.BuildUnaryOp(getCurScope(), AsmLoc, UO_AddrOf, OpExpr).get();
645
646 ConstraintRefs[i] = StringRef(Constraints[i]);
647 Exprs[i] = OpExpr;
648 }
649
650 // FIXME: We should be passing source locations for better diagnostics.
Ehsan Akhgari0f89fac2014-07-06 05:26:54 +0000651 return Actions.ActOnMSAsmStmt(AsmLoc, LBraceLocs[0], AsmToks, AsmStringIR,
Alp Toker1b935a82014-06-08 05:40:04 +0000652 NumOutputs, NumInputs, ConstraintRefs,
653 ClobberRefs, Exprs, EndLoc);
654}
655
656/// ParseAsmStatement - Parse a GNU extended asm statement.
657/// asm-statement:
658/// gnu-asm-statement
659/// ms-asm-statement
660///
661/// [GNU] gnu-asm-statement:
662/// 'asm' type-qualifier[opt] '(' asm-argument ')' ';'
663///
664/// [GNU] asm-argument:
665/// asm-string-literal
666/// asm-string-literal ':' asm-operands[opt]
667/// asm-string-literal ':' asm-operands[opt] ':' asm-operands[opt]
668/// asm-string-literal ':' asm-operands[opt] ':' asm-operands[opt]
669/// ':' asm-clobbers
670///
671/// [GNU] asm-clobbers:
672/// asm-string-literal
673/// asm-clobbers ',' asm-string-literal
674///
675StmtResult Parser::ParseAsmStatement(bool &msAsm) {
676 assert(Tok.is(tok::kw_asm) && "Not an asm stmt");
677 SourceLocation AsmLoc = ConsumeToken();
678
Denis Zobnin628b0222016-04-21 10:59:18 +0000679 if (getLangOpts().AsmBlocks && !isGCCAsmStatement(Tok)) {
Alp Toker1b935a82014-06-08 05:40:04 +0000680 msAsm = true;
681 return ParseMicrosoftAsmStatement(AsmLoc);
682 }
Steven Wucb0d13f2015-01-16 23:05:28 +0000683
Alp Toker1b935a82014-06-08 05:40:04 +0000684 DeclSpec DS(AttrFactory);
685 SourceLocation Loc = Tok.getLocation();
Aaron Ballman08b06592014-07-22 12:44:22 +0000686 ParseTypeQualifierListOpt(DS, AR_VendorAttributesParsed);
Alp Toker1b935a82014-06-08 05:40:04 +0000687
688 // GNU asms accept, but warn, about type-qualifiers other than volatile.
689 if (DS.getTypeQualifiers() & DeclSpec::TQ_const)
Richard Smith01d96982016-12-02 23:00:28 +0000690 Diag(Loc, diag::warn_asm_qualifier_ignored) << "const";
Alp Toker1b935a82014-06-08 05:40:04 +0000691 if (DS.getTypeQualifiers() & DeclSpec::TQ_restrict)
Richard Smith01d96982016-12-02 23:00:28 +0000692 Diag(Loc, diag::warn_asm_qualifier_ignored) << "restrict";
Alp Toker1b935a82014-06-08 05:40:04 +0000693 // FIXME: Once GCC supports _Atomic, check whether it permits it here.
694 if (DS.getTypeQualifiers() & DeclSpec::TQ_atomic)
Richard Smith01d96982016-12-02 23:00:28 +0000695 Diag(Loc, diag::warn_asm_qualifier_ignored) << "_Atomic";
Alp Toker1b935a82014-06-08 05:40:04 +0000696
697 // Remember if this was a volatile asm.
698 bool isVolatile = DS.getTypeQualifiers() & DeclSpec::TQ_volatile;
Denis Zobnin628b0222016-04-21 10:59:18 +0000699
700 // TODO: support "asm goto" constructs (PR#9295).
701 if (Tok.is(tok::kw_goto)) {
702 Diag(Tok, diag::err_asm_goto_not_supported_yet);
703 SkipUntil(tok::r_paren, StopAtSemi);
704 return StmtError();
705 }
706
Alp Toker1b935a82014-06-08 05:40:04 +0000707 if (Tok.isNot(tok::l_paren)) {
708 Diag(Tok, diag::err_expected_lparen_after) << "asm";
709 SkipUntil(tok::r_paren, StopAtSemi);
710 return StmtError();
711 }
712 BalancedDelimiterTracker T(*this, tok::l_paren);
713 T.consumeOpen();
714
715 ExprResult AsmString(ParseAsmStringLiteral());
Steven Wu18bbe192015-05-12 00:16:37 +0000716
717 // Check if GNU-style InlineAsm is disabled.
718 // Error on anything other than empty string.
719 if (!(getLangOpts().GNUAsm || AsmString.isInvalid())) {
720 const auto *SL = cast<StringLiteral>(AsmString.get());
721 if (!SL->getString().trim().empty())
722 Diag(Loc, diag::err_gnu_inline_asm_disabled);
723 }
724
Alp Toker1b935a82014-06-08 05:40:04 +0000725 if (AsmString.isInvalid()) {
726 // Consume up to and including the closing paren.
727 T.skipToEnd();
728 return StmtError();
729 }
730
731 SmallVector<IdentifierInfo *, 4> Names;
732 ExprVector Constraints;
733 ExprVector Exprs;
734 ExprVector Clobbers;
735
736 if (Tok.is(tok::r_paren)) {
737 // We have a simple asm expression like 'asm("foo")'.
738 T.consumeClose();
739 return Actions.ActOnGCCAsmStmt(AsmLoc, /*isSimple*/ true, isVolatile,
740 /*NumOutputs*/ 0, /*NumInputs*/ 0, nullptr,
741 Constraints, Exprs, AsmString.get(),
742 Clobbers, T.getCloseLocation());
743 }
744
745 // Parse Outputs, if present.
746 bool AteExtraColon = false;
747 if (Tok.is(tok::colon) || Tok.is(tok::coloncolon)) {
748 // In C++ mode, parse "::" like ": :".
749 AteExtraColon = Tok.is(tok::coloncolon);
750 ConsumeToken();
751
752 if (!AteExtraColon && ParseAsmOperandsOpt(Names, Constraints, Exprs))
753 return StmtError();
754 }
755
756 unsigned NumOutputs = Names.size();
757
758 // Parse Inputs, if present.
759 if (AteExtraColon || Tok.is(tok::colon) || Tok.is(tok::coloncolon)) {
760 // In C++ mode, parse "::" like ": :".
761 if (AteExtraColon)
762 AteExtraColon = false;
763 else {
764 AteExtraColon = Tok.is(tok::coloncolon);
765 ConsumeToken();
766 }
767
768 if (!AteExtraColon && ParseAsmOperandsOpt(Names, Constraints, Exprs))
769 return StmtError();
770 }
771
772 assert(Names.size() == Constraints.size() &&
773 Constraints.size() == Exprs.size() && "Input operand size mismatch!");
774
775 unsigned NumInputs = Names.size() - NumOutputs;
776
777 // Parse the clobbers, if present.
778 if (AteExtraColon || Tok.is(tok::colon)) {
779 if (!AteExtraColon)
780 ConsumeToken();
781
782 // Parse the asm-string list for clobbers if present.
783 if (Tok.isNot(tok::r_paren)) {
784 while (1) {
785 ExprResult Clobber(ParseAsmStringLiteral());
786
787 if (Clobber.isInvalid())
788 break;
789
790 Clobbers.push_back(Clobber.get());
791
792 if (!TryConsumeToken(tok::comma))
793 break;
794 }
795 }
796 }
797
798 T.consumeClose();
799 return Actions.ActOnGCCAsmStmt(
800 AsmLoc, false, isVolatile, NumOutputs, NumInputs, Names.data(),
801 Constraints, Exprs, AsmString.get(), Clobbers, T.getCloseLocation());
802}
803
804/// ParseAsmOperands - Parse the asm-operands production as used by
805/// asm-statement, assuming the leading ':' token was eaten.
806///
807/// [GNU] asm-operands:
808/// asm-operand
809/// asm-operands ',' asm-operand
810///
811/// [GNU] asm-operand:
812/// asm-string-literal '(' expression ')'
813/// '[' identifier ']' asm-string-literal '(' expression ')'
814///
815//
816// FIXME: Avoid unnecessary std::string trashing.
817bool Parser::ParseAsmOperandsOpt(SmallVectorImpl<IdentifierInfo *> &Names,
818 SmallVectorImpl<Expr *> &Constraints,
819 SmallVectorImpl<Expr *> &Exprs) {
820 // 'asm-operands' isn't present?
821 if (!isTokenStringLiteral() && Tok.isNot(tok::l_square))
822 return false;
823
824 while (1) {
825 // Read the [id] if present.
826 if (Tok.is(tok::l_square)) {
827 BalancedDelimiterTracker T(*this, tok::l_square);
828 T.consumeOpen();
829
830 if (Tok.isNot(tok::identifier)) {
831 Diag(Tok, diag::err_expected) << tok::identifier;
832 SkipUntil(tok::r_paren, StopAtSemi);
833 return true;
834 }
835
836 IdentifierInfo *II = Tok.getIdentifierInfo();
837 ConsumeToken();
838
839 Names.push_back(II);
840 T.consumeClose();
841 } else
842 Names.push_back(nullptr);
843
844 ExprResult Constraint(ParseAsmStringLiteral());
845 if (Constraint.isInvalid()) {
846 SkipUntil(tok::r_paren, StopAtSemi);
847 return true;
848 }
849 Constraints.push_back(Constraint.get());
850
851 if (Tok.isNot(tok::l_paren)) {
852 Diag(Tok, diag::err_expected_lparen_after) << "asm operand";
853 SkipUntil(tok::r_paren, StopAtSemi);
854 return true;
855 }
856
857 // Read the parenthesized expression.
858 BalancedDelimiterTracker T(*this, tok::l_paren);
859 T.consumeOpen();
Kaelyn Takata15867822014-11-21 18:48:04 +0000860 ExprResult Res = Actions.CorrectDelayedTyposInExpr(ParseExpression());
Alp Toker1b935a82014-06-08 05:40:04 +0000861 T.consumeClose();
862 if (Res.isInvalid()) {
863 SkipUntil(tok::r_paren, StopAtSemi);
864 return true;
865 }
866 Exprs.push_back(Res.get());
867 // Eat the comma and continue parsing if it exists.
868 if (!TryConsumeToken(tok::comma))
869 return false;
870 }
871}