1 //===--- Parser.cpp - C Language Family 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 the Parser interfaces.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/Parse/Parser.h"
15 #include "clang/AST/ASTConsumer.h"
16 #include "clang/AST/ASTContext.h"
17 #include "clang/AST/DeclTemplate.h"
18 #include "clang/Parse/ParseDiagnostic.h"
19 #include "clang/Parse/RAIIObjectsForParser.h"
20 #include "clang/Sema/DeclSpec.h"
21 #include "clang/Sema/ParsedTemplate.h"
22 #include "clang/Sema/Scope.h"
23 #include "llvm/Support/Path.h"
24 using namespace clang;
25 
26 
27 namespace {
28 /// A comment handler that passes comments found by the preprocessor
29 /// to the parser action.
30 class ActionCommentHandler : public CommentHandler {
31   Sema &S;
32 
33 public:
ActionCommentHandler(Sema & S)34   explicit ActionCommentHandler(Sema &S) : S(S) { }
35 
HandleComment(Preprocessor & PP,SourceRange Comment)36   bool HandleComment(Preprocessor &PP, SourceRange Comment) override {
37     S.ActOnComment(Comment);
38     return false;
39   }
40 };
41 } // end anonymous namespace
42 
getSEHExceptKeyword()43 IdentifierInfo *Parser::getSEHExceptKeyword() {
44   // __except is accepted as a (contextual) keyword
45   if (!Ident__except && (getLangOpts().MicrosoftExt || getLangOpts().Borland))
46     Ident__except = PP.getIdentifierInfo("__except");
47 
48   return Ident__except;
49 }
50 
Parser(Preprocessor & pp,Sema & actions,bool skipFunctionBodies)51 Parser::Parser(Preprocessor &pp, Sema &actions, bool skipFunctionBodies)
52   : PP(pp), Actions(actions), Diags(PP.getDiagnostics()),
53     GreaterThanIsOperator(true), ColonIsSacred(false),
54     InMessageExpression(false), TemplateParameterDepth(0),
55     ParsingInObjCContainer(false) {
56   SkipFunctionBodies = pp.isCodeCompletionEnabled() || skipFunctionBodies;
57   Tok.startToken();
58   Tok.setKind(tok::eof);
59   Actions.CurScope = nullptr;
60   NumCachedScopes = 0;
61   CurParsedObjCImpl = nullptr;
62 
63   // Add #pragma handlers. These are removed and destroyed in the
64   // destructor.
65   initializePragmaHandlers();
66 
67   CommentSemaHandler.reset(new ActionCommentHandler(actions));
68   PP.addCommentHandler(CommentSemaHandler.get());
69 
70   PP.setCodeCompletionHandler(*this);
71 }
72 
Diag(SourceLocation Loc,unsigned DiagID)73 DiagnosticBuilder Parser::Diag(SourceLocation Loc, unsigned DiagID) {
74   return Diags.Report(Loc, DiagID);
75 }
76 
Diag(const Token & Tok,unsigned DiagID)77 DiagnosticBuilder Parser::Diag(const Token &Tok, unsigned DiagID) {
78   return Diag(Tok.getLocation(), DiagID);
79 }
80 
81 /// Emits a diagnostic suggesting parentheses surrounding a
82 /// given range.
83 ///
84 /// \param Loc The location where we'll emit the diagnostic.
85 /// \param DK The kind of diagnostic to emit.
86 /// \param ParenRange Source range enclosing code that should be parenthesized.
SuggestParentheses(SourceLocation Loc,unsigned DK,SourceRange ParenRange)87 void Parser::SuggestParentheses(SourceLocation Loc, unsigned DK,
88                                 SourceRange ParenRange) {
89   SourceLocation EndLoc = PP.getLocForEndOfToken(ParenRange.getEnd());
90   if (!ParenRange.getEnd().isFileID() || EndLoc.isInvalid()) {
91     // We can't display the parentheses, so just dig the
92     // warning/error and return.
93     Diag(Loc, DK);
94     return;
95   }
96 
97   Diag(Loc, DK)
98     << FixItHint::CreateInsertion(ParenRange.getBegin(), "(")
99     << FixItHint::CreateInsertion(EndLoc, ")");
100 }
101 
IsCommonTypo(tok::TokenKind ExpectedTok,const Token & Tok)102 static bool IsCommonTypo(tok::TokenKind ExpectedTok, const Token &Tok) {
103   switch (ExpectedTok) {
104   case tok::semi:
105     return Tok.is(tok::colon) || Tok.is(tok::comma); // : or , for ;
106   default: return false;
107   }
108 }
109 
ExpectAndConsume(tok::TokenKind ExpectedTok,unsigned DiagID,StringRef Msg)110 bool Parser::ExpectAndConsume(tok::TokenKind ExpectedTok, unsigned DiagID,
111                               StringRef Msg) {
112   if (Tok.is(ExpectedTok) || Tok.is(tok::code_completion)) {
113     ConsumeAnyToken();
114     return false;
115   }
116 
117   // Detect common single-character typos and resume.
118   if (IsCommonTypo(ExpectedTok, Tok)) {
119     SourceLocation Loc = Tok.getLocation();
120     {
121       DiagnosticBuilder DB = Diag(Loc, DiagID);
122       DB << FixItHint::CreateReplacement(
123                 SourceRange(Loc), tok::getPunctuatorSpelling(ExpectedTok));
124       if (DiagID == diag::err_expected)
125         DB << ExpectedTok;
126       else if (DiagID == diag::err_expected_after)
127         DB << Msg << ExpectedTok;
128       else
129         DB << Msg;
130     }
131 
132     // Pretend there wasn't a problem.
133     ConsumeAnyToken();
134     return false;
135   }
136 
137   SourceLocation EndLoc = PP.getLocForEndOfToken(PrevTokLocation);
138   const char *Spelling = nullptr;
139   if (EndLoc.isValid())
140     Spelling = tok::getPunctuatorSpelling(ExpectedTok);
141 
142   DiagnosticBuilder DB =
143       Spelling
144           ? Diag(EndLoc, DiagID) << FixItHint::CreateInsertion(EndLoc, Spelling)
145           : Diag(Tok, DiagID);
146   if (DiagID == diag::err_expected)
147     DB << ExpectedTok;
148   else if (DiagID == diag::err_expected_after)
149     DB << Msg << ExpectedTok;
150   else
151     DB << Msg;
152 
153   return true;
154 }
155 
ExpectAndConsumeSemi(unsigned DiagID)156 bool Parser::ExpectAndConsumeSemi(unsigned DiagID) {
157   if (TryConsumeToken(tok::semi))
158     return false;
159 
160   if (Tok.is(tok::code_completion)) {
161     handleUnexpectedCodeCompletionToken();
162     return false;
163   }
164 
165   if ((Tok.is(tok::r_paren) || Tok.is(tok::r_square)) &&
166       NextToken().is(tok::semi)) {
167     Diag(Tok, diag::err_extraneous_token_before_semi)
168       << PP.getSpelling(Tok)
169       << FixItHint::CreateRemoval(Tok.getLocation());
170     ConsumeAnyToken(); // The ')' or ']'.
171     ConsumeToken(); // The ';'.
172     return false;
173   }
174 
175   return ExpectAndConsume(tok::semi, DiagID);
176 }
177 
ConsumeExtraSemi(ExtraSemiKind Kind,unsigned TST)178 void Parser::ConsumeExtraSemi(ExtraSemiKind Kind, unsigned TST) {
179   if (!Tok.is(tok::semi)) return;
180 
181   bool HadMultipleSemis = false;
182   SourceLocation StartLoc = Tok.getLocation();
183   SourceLocation EndLoc = Tok.getLocation();
184   ConsumeToken();
185 
186   while ((Tok.is(tok::semi) && !Tok.isAtStartOfLine())) {
187     HadMultipleSemis = true;
188     EndLoc = Tok.getLocation();
189     ConsumeToken();
190   }
191 
192   // C++11 allows extra semicolons at namespace scope, but not in any of the
193   // other contexts.
194   if (Kind == OutsideFunction && getLangOpts().CPlusPlus) {
195     if (getLangOpts().CPlusPlus11)
196       Diag(StartLoc, diag::warn_cxx98_compat_top_level_semi)
197           << FixItHint::CreateRemoval(SourceRange(StartLoc, EndLoc));
198     else
199       Diag(StartLoc, diag::ext_extra_semi_cxx11)
200           << FixItHint::CreateRemoval(SourceRange(StartLoc, EndLoc));
201     return;
202   }
203 
204   if (Kind != AfterMemberFunctionDefinition || HadMultipleSemis)
205     Diag(StartLoc, diag::ext_extra_semi)
206         << Kind << DeclSpec::getSpecifierName((DeclSpec::TST)TST,
207                                     Actions.getASTContext().getPrintingPolicy())
208         << FixItHint::CreateRemoval(SourceRange(StartLoc, EndLoc));
209   else
210     // A single semicolon is valid after a member function definition.
211     Diag(StartLoc, diag::warn_extra_semi_after_mem_fn_def)
212       << FixItHint::CreateRemoval(SourceRange(StartLoc, EndLoc));
213 }
214 
expectIdentifier()215 bool Parser::expectIdentifier() {
216   if (Tok.is(tok::identifier))
217     return false;
218   if (const auto *II = Tok.getIdentifierInfo()) {
219     if (II->isCPlusPlusKeyword(getLangOpts())) {
220       Diag(Tok, diag::err_expected_token_instead_of_objcxx_keyword)
221           << tok::identifier << Tok.getIdentifierInfo();
222       // Objective-C++: Recover by treating this keyword as a valid identifier.
223       return false;
224     }
225   }
226   Diag(Tok, diag::err_expected) << tok::identifier;
227   return true;
228 }
229 
230 //===----------------------------------------------------------------------===//
231 // Error recovery.
232 //===----------------------------------------------------------------------===//
233 
HasFlagsSet(Parser::SkipUntilFlags L,Parser::SkipUntilFlags R)234 static bool HasFlagsSet(Parser::SkipUntilFlags L, Parser::SkipUntilFlags R) {
235   return (static_cast<unsigned>(L) & static_cast<unsigned>(R)) != 0;
236 }
237 
238 /// SkipUntil - Read tokens until we get to the specified token, then consume
239 /// it (unless no flag StopBeforeMatch).  Because we cannot guarantee that the
240 /// token will ever occur, this skips to the next token, or to some likely
241 /// good stopping point.  If StopAtSemi is true, skipping will stop at a ';'
242 /// character.
243 ///
244 /// If SkipUntil finds the specified token, it returns true, otherwise it
245 /// returns false.
SkipUntil(ArrayRef<tok::TokenKind> Toks,SkipUntilFlags Flags)246 bool Parser::SkipUntil(ArrayRef<tok::TokenKind> Toks, SkipUntilFlags Flags) {
247   // We always want this function to skip at least one token if the first token
248   // isn't T and if not at EOF.
249   bool isFirstTokenSkipped = true;
250   while (1) {
251     // If we found one of the tokens, stop and return true.
252     for (unsigned i = 0, NumToks = Toks.size(); i != NumToks; ++i) {
253       if (Tok.is(Toks[i])) {
254         if (HasFlagsSet(Flags, StopBeforeMatch)) {
255           // Noop, don't consume the token.
256         } else {
257           ConsumeAnyToken();
258         }
259         return true;
260       }
261     }
262 
263     // Important special case: The caller has given up and just wants us to
264     // skip the rest of the file. Do this without recursing, since we can
265     // get here precisely because the caller detected too much recursion.
266     if (Toks.size() == 1 && Toks[0] == tok::eof &&
267         !HasFlagsSet(Flags, StopAtSemi) &&
268         !HasFlagsSet(Flags, StopAtCodeCompletion)) {
269       while (Tok.isNot(tok::eof))
270         ConsumeAnyToken();
271       return true;
272     }
273 
274     switch (Tok.getKind()) {
275     case tok::eof:
276       // Ran out of tokens.
277       return false;
278 
279     case tok::annot_pragma_openmp:
280     case tok::annot_pragma_openmp_end:
281       // Stop before an OpenMP pragma boundary.
282     case tok::annot_module_begin:
283     case tok::annot_module_end:
284     case tok::annot_module_include:
285       // Stop before we change submodules. They generally indicate a "good"
286       // place to pick up parsing again (except in the special case where
287       // we're trying to skip to EOF).
288       return false;
289 
290     case tok::code_completion:
291       if (!HasFlagsSet(Flags, StopAtCodeCompletion))
292         handleUnexpectedCodeCompletionToken();
293       return false;
294 
295     case tok::l_paren:
296       // Recursively skip properly-nested parens.
297       ConsumeParen();
298       if (HasFlagsSet(Flags, StopAtCodeCompletion))
299         SkipUntil(tok::r_paren, StopAtCodeCompletion);
300       else
301         SkipUntil(tok::r_paren);
302       break;
303     case tok::l_square:
304       // Recursively skip properly-nested square brackets.
305       ConsumeBracket();
306       if (HasFlagsSet(Flags, StopAtCodeCompletion))
307         SkipUntil(tok::r_square, StopAtCodeCompletion);
308       else
309         SkipUntil(tok::r_square);
310       break;
311     case tok::l_brace:
312       // Recursively skip properly-nested braces.
313       ConsumeBrace();
314       if (HasFlagsSet(Flags, StopAtCodeCompletion))
315         SkipUntil(tok::r_brace, StopAtCodeCompletion);
316       else
317         SkipUntil(tok::r_brace);
318       break;
319 
320     // Okay, we found a ']' or '}' or ')', which we think should be balanced.
321     // Since the user wasn't looking for this token (if they were, it would
322     // already be handled), this isn't balanced.  If there is a LHS token at a
323     // higher level, we will assume that this matches the unbalanced token
324     // and return it.  Otherwise, this is a spurious RHS token, which we skip.
325     case tok::r_paren:
326       if (ParenCount && !isFirstTokenSkipped)
327         return false;  // Matches something.
328       ConsumeParen();
329       break;
330     case tok::r_square:
331       if (BracketCount && !isFirstTokenSkipped)
332         return false;  // Matches something.
333       ConsumeBracket();
334       break;
335     case tok::r_brace:
336       if (BraceCount && !isFirstTokenSkipped)
337         return false;  // Matches something.
338       ConsumeBrace();
339       break;
340 
341     case tok::semi:
342       if (HasFlagsSet(Flags, StopAtSemi))
343         return false;
344       LLVM_FALLTHROUGH;
345     default:
346       // Skip this token.
347       ConsumeAnyToken();
348       break;
349     }
350     isFirstTokenSkipped = false;
351   }
352 }
353 
354 //===----------------------------------------------------------------------===//
355 // Scope manipulation
356 //===----------------------------------------------------------------------===//
357 
358 /// EnterScope - Start a new scope.
EnterScope(unsigned ScopeFlags)359 void Parser::EnterScope(unsigned ScopeFlags) {
360   if (NumCachedScopes) {
361     Scope *N = ScopeCache[--NumCachedScopes];
362     N->Init(getCurScope(), ScopeFlags);
363     Actions.CurScope = N;
364   } else {
365     Actions.CurScope = new Scope(getCurScope(), ScopeFlags, Diags);
366   }
367 }
368 
369 /// ExitScope - Pop a scope off the scope stack.
ExitScope()370 void Parser::ExitScope() {
371   assert(getCurScope() && "Scope imbalance!");
372 
373   // Inform the actions module that this scope is going away if there are any
374   // decls in it.
375   Actions.ActOnPopScope(Tok.getLocation(), getCurScope());
376 
377   Scope *OldScope = getCurScope();
378   Actions.CurScope = OldScope->getParent();
379 
380   if (NumCachedScopes == ScopeCacheSize)
381     delete OldScope;
382   else
383     ScopeCache[NumCachedScopes++] = OldScope;
384 }
385 
386 /// Set the flags for the current scope to ScopeFlags. If ManageFlags is false,
387 /// this object does nothing.
ParseScopeFlags(Parser * Self,unsigned ScopeFlags,bool ManageFlags)388 Parser::ParseScopeFlags::ParseScopeFlags(Parser *Self, unsigned ScopeFlags,
389                                  bool ManageFlags)
390   : CurScope(ManageFlags ? Self->getCurScope() : nullptr) {
391   if (CurScope) {
392     OldFlags = CurScope->getFlags();
393     CurScope->setFlags(ScopeFlags);
394   }
395 }
396 
397 /// Restore the flags for the current scope to what they were before this
398 /// object overrode them.
~ParseScopeFlags()399 Parser::ParseScopeFlags::~ParseScopeFlags() {
400   if (CurScope)
401     CurScope->setFlags(OldFlags);
402 }
403 
404 
405 //===----------------------------------------------------------------------===//
406 // C99 6.9: External Definitions.
407 //===----------------------------------------------------------------------===//
408 
~Parser()409 Parser::~Parser() {
410   // If we still have scopes active, delete the scope tree.
411   delete getCurScope();
412   Actions.CurScope = nullptr;
413 
414   // Free the scope cache.
415   for (unsigned i = 0, e = NumCachedScopes; i != e; ++i)
416     delete ScopeCache[i];
417 
418   resetPragmaHandlers();
419 
420   PP.removeCommentHandler(CommentSemaHandler.get());
421 
422   PP.clearCodeCompletionHandler();
423 
424   if (getLangOpts().DelayedTemplateParsing &&
425       !PP.isIncrementalProcessingEnabled() && !TemplateIds.empty()) {
426     // If an ASTConsumer parsed delay-parsed templates in their
427     // HandleTranslationUnit() method, TemplateIds created there were not
428     // guarded by a DestroyTemplateIdAnnotationsRAIIObj object in
429     // ParseTopLevelDecl(). Destroy them here.
430     DestroyTemplateIdAnnotationsRAIIObj CleanupRAII(TemplateIds);
431   }
432 
433   assert(TemplateIds.empty() && "Still alive TemplateIdAnnotations around?");
434 }
435 
436 /// Initialize - Warm up the parser.
437 ///
Initialize()438 void Parser::Initialize() {
439   // Create the translation unit scope.  Install it as the current scope.
440   assert(getCurScope() == nullptr && "A scope is already active?");
441   EnterScope(Scope::DeclScope);
442   Actions.ActOnTranslationUnitScope(getCurScope());
443 
444   // Initialization for Objective-C context sensitive keywords recognition.
445   // Referenced in Parser::ParseObjCTypeQualifierList.
446   if (getLangOpts().ObjC) {
447     ObjCTypeQuals[objc_in] = &PP.getIdentifierTable().get("in");
448     ObjCTypeQuals[objc_out] = &PP.getIdentifierTable().get("out");
449     ObjCTypeQuals[objc_inout] = &PP.getIdentifierTable().get("inout");
450     ObjCTypeQuals[objc_oneway] = &PP.getIdentifierTable().get("oneway");
451     ObjCTypeQuals[objc_bycopy] = &PP.getIdentifierTable().get("bycopy");
452     ObjCTypeQuals[objc_byref] = &PP.getIdentifierTable().get("byref");
453     ObjCTypeQuals[objc_nonnull] = &PP.getIdentifierTable().get("nonnull");
454     ObjCTypeQuals[objc_nullable] = &PP.getIdentifierTable().get("nullable");
455     ObjCTypeQuals[objc_null_unspecified]
456       = &PP.getIdentifierTable().get("null_unspecified");
457   }
458 
459   Ident_instancetype = nullptr;
460   Ident_final = nullptr;
461   Ident_sealed = nullptr;
462   Ident_override = nullptr;
463   Ident_GNU_final = nullptr;
464 
465   Ident_super = &PP.getIdentifierTable().get("super");
466 
467   Ident_vector = nullptr;
468   Ident_bool = nullptr;
469   Ident_pixel = nullptr;
470   if (getLangOpts().AltiVec || getLangOpts().ZVector) {
471     Ident_vector = &PP.getIdentifierTable().get("vector");
472     Ident_bool = &PP.getIdentifierTable().get("bool");
473   }
474   if (getLangOpts().AltiVec)
475     Ident_pixel = &PP.getIdentifierTable().get("pixel");
476 
477   Ident_introduced = nullptr;
478   Ident_deprecated = nullptr;
479   Ident_obsoleted = nullptr;
480   Ident_unavailable = nullptr;
481   Ident_strict = nullptr;
482   Ident_replacement = nullptr;
483 
484   Ident_language = Ident_defined_in = Ident_generated_declaration = nullptr;
485 
486   Ident__except = nullptr;
487 
488   Ident__exception_code = Ident__exception_info = nullptr;
489   Ident__abnormal_termination = Ident___exception_code = nullptr;
490   Ident___exception_info = Ident___abnormal_termination = nullptr;
491   Ident_GetExceptionCode = Ident_GetExceptionInfo = nullptr;
492   Ident_AbnormalTermination = nullptr;
493 
494   if(getLangOpts().Borland) {
495     Ident__exception_info        = PP.getIdentifierInfo("_exception_info");
496     Ident___exception_info       = PP.getIdentifierInfo("__exception_info");
497     Ident_GetExceptionInfo       = PP.getIdentifierInfo("GetExceptionInformation");
498     Ident__exception_code        = PP.getIdentifierInfo("_exception_code");
499     Ident___exception_code       = PP.getIdentifierInfo("__exception_code");
500     Ident_GetExceptionCode       = PP.getIdentifierInfo("GetExceptionCode");
501     Ident__abnormal_termination  = PP.getIdentifierInfo("_abnormal_termination");
502     Ident___abnormal_termination = PP.getIdentifierInfo("__abnormal_termination");
503     Ident_AbnormalTermination    = PP.getIdentifierInfo("AbnormalTermination");
504 
505     PP.SetPoisonReason(Ident__exception_code,diag::err_seh___except_block);
506     PP.SetPoisonReason(Ident___exception_code,diag::err_seh___except_block);
507     PP.SetPoisonReason(Ident_GetExceptionCode,diag::err_seh___except_block);
508     PP.SetPoisonReason(Ident__exception_info,diag::err_seh___except_filter);
509     PP.SetPoisonReason(Ident___exception_info,diag::err_seh___except_filter);
510     PP.SetPoisonReason(Ident_GetExceptionInfo,diag::err_seh___except_filter);
511     PP.SetPoisonReason(Ident__abnormal_termination,diag::err_seh___finally_block);
512     PP.SetPoisonReason(Ident___abnormal_termination,diag::err_seh___finally_block);
513     PP.SetPoisonReason(Ident_AbnormalTermination,diag::err_seh___finally_block);
514   }
515 
516   Actions.Initialize();
517 
518   // Prime the lexer look-ahead.
519   ConsumeToken();
520 }
521 
LateTemplateParserCleanupCallback(void * P)522 void Parser::LateTemplateParserCleanupCallback(void *P) {
523   // While this RAII helper doesn't bracket any actual work, the destructor will
524   // clean up annotations that were created during ActOnEndOfTranslationUnit
525   // when incremental processing is enabled.
526   DestroyTemplateIdAnnotationsRAIIObj CleanupRAII(((Parser *)P)->TemplateIds);
527 }
528 
ParseFirstTopLevelDecl(DeclGroupPtrTy & Result)529 bool Parser::ParseFirstTopLevelDecl(DeclGroupPtrTy &Result) {
530   Actions.ActOnStartOfTranslationUnit();
531 
532   // C11 6.9p1 says translation units must have at least one top-level
533   // declaration. C++ doesn't have this restriction. We also don't want to
534   // complain if we have a precompiled header, although technically if the PCH
535   // is empty we should still emit the (pedantic) diagnostic.
536   bool NoTopLevelDecls = ParseTopLevelDecl(Result);
537   if (NoTopLevelDecls && !Actions.getASTContext().getExternalSource() &&
538       !getLangOpts().CPlusPlus)
539     Diag(diag::ext_empty_translation_unit);
540 
541   return NoTopLevelDecls;
542 }
543 
544 /// ParseTopLevelDecl - Parse one top-level declaration, return whatever the
545 /// action tells us to.  This returns true if the EOF was encountered.
ParseTopLevelDecl(DeclGroupPtrTy & Result)546 bool Parser::ParseTopLevelDecl(DeclGroupPtrTy &Result) {
547   DestroyTemplateIdAnnotationsRAIIObj CleanupRAII(TemplateIds);
548 
549   // Skip over the EOF token, flagging end of previous input for incremental
550   // processing
551   if (PP.isIncrementalProcessingEnabled() && Tok.is(tok::eof))
552     ConsumeToken();
553 
554   Result = nullptr;
555   switch (Tok.getKind()) {
556   case tok::annot_pragma_unused:
557     HandlePragmaUnused();
558     return false;
559 
560   case tok::kw_export:
561     if (NextToken().isNot(tok::kw_module))
562       break;
563     LLVM_FALLTHROUGH;
564   case tok::kw_module:
565     Result = ParseModuleDecl();
566     return false;
567 
568   case tok::annot_module_include:
569     Actions.ActOnModuleInclude(Tok.getLocation(),
570                                reinterpret_cast<Module *>(
571                                    Tok.getAnnotationValue()));
572     ConsumeAnnotationToken();
573     return false;
574 
575   case tok::annot_module_begin:
576     Actions.ActOnModuleBegin(Tok.getLocation(), reinterpret_cast<Module *>(
577                                                     Tok.getAnnotationValue()));
578     ConsumeAnnotationToken();
579     return false;
580 
581   case tok::annot_module_end:
582     Actions.ActOnModuleEnd(Tok.getLocation(), reinterpret_cast<Module *>(
583                                                   Tok.getAnnotationValue()));
584     ConsumeAnnotationToken();
585     return false;
586 
587   case tok::annot_pragma_attribute:
588     HandlePragmaAttribute();
589     return false;
590 
591   case tok::eof:
592     // Late template parsing can begin.
593     if (getLangOpts().DelayedTemplateParsing)
594       Actions.SetLateTemplateParser(LateTemplateParserCallback,
595                                     PP.isIncrementalProcessingEnabled() ?
596                                     LateTemplateParserCleanupCallback : nullptr,
597                                     this);
598     if (!PP.isIncrementalProcessingEnabled())
599       Actions.ActOnEndOfTranslationUnit();
600     //else don't tell Sema that we ended parsing: more input might come.
601     return true;
602 
603   default:
604     break;
605   }
606 
607   ParsedAttributesWithRange attrs(AttrFactory);
608   MaybeParseCXX11Attributes(attrs);
609 
610   Result = ParseExternalDeclaration(attrs);
611   return false;
612 }
613 
614 /// ParseExternalDeclaration:
615 ///
616 ///       external-declaration: [C99 6.9], declaration: [C++ dcl.dcl]
617 ///         function-definition
618 ///         declaration
619 /// [GNU]   asm-definition
620 /// [GNU]   __extension__ external-declaration
621 /// [OBJC]  objc-class-definition
622 /// [OBJC]  objc-class-declaration
623 /// [OBJC]  objc-alias-declaration
624 /// [OBJC]  objc-protocol-definition
625 /// [OBJC]  objc-method-definition
626 /// [OBJC]  @end
627 /// [C++]   linkage-specification
628 /// [GNU] asm-definition:
629 ///         simple-asm-expr ';'
630 /// [C++11] empty-declaration
631 /// [C++11] attribute-declaration
632 ///
633 /// [C++11] empty-declaration:
634 ///           ';'
635 ///
636 /// [C++0x/GNU] 'extern' 'template' declaration
637 ///
638 /// [Modules-TS] module-import-declaration
639 ///
640 Parser::DeclGroupPtrTy
ParseExternalDeclaration(ParsedAttributesWithRange & attrs,ParsingDeclSpec * DS)641 Parser::ParseExternalDeclaration(ParsedAttributesWithRange &attrs,
642                                  ParsingDeclSpec *DS) {
643   DestroyTemplateIdAnnotationsRAIIObj CleanupRAII(TemplateIds);
644   ParenBraceBracketBalancer BalancerRAIIObj(*this);
645 
646   if (PP.isCodeCompletionReached()) {
647     cutOffParsing();
648     return nullptr;
649   }
650 
651   Decl *SingleDecl = nullptr;
652   switch (Tok.getKind()) {
653   case tok::annot_pragma_vis:
654     HandlePragmaVisibility();
655     return nullptr;
656   case tok::annot_pragma_pack:
657     HandlePragmaPack();
658     return nullptr;
659   case tok::annot_pragma_msstruct:
660     HandlePragmaMSStruct();
661     return nullptr;
662   case tok::annot_pragma_align:
663     HandlePragmaAlign();
664     return nullptr;
665   case tok::annot_pragma_weak:
666     HandlePragmaWeak();
667     return nullptr;
668   case tok::annot_pragma_weakalias:
669     HandlePragmaWeakAlias();
670     return nullptr;
671   case tok::annot_pragma_redefine_extname:
672     HandlePragmaRedefineExtname();
673     return nullptr;
674   case tok::annot_pragma_fp_contract:
675     HandlePragmaFPContract();
676     return nullptr;
677   case tok::annot_pragma_fenv_access:
678     HandlePragmaFEnvAccess();
679     return nullptr;
680   case tok::annot_pragma_fp:
681     HandlePragmaFP();
682     break;
683   case tok::annot_pragma_opencl_extension:
684     HandlePragmaOpenCLExtension();
685     return nullptr;
686   case tok::annot_pragma_openmp: {
687     AccessSpecifier AS = AS_none;
688     return ParseOpenMPDeclarativeDirectiveWithExtDecl(AS, attrs);
689   }
690   case tok::annot_pragma_ms_pointers_to_members:
691     HandlePragmaMSPointersToMembers();
692     return nullptr;
693   case tok::annot_pragma_ms_vtordisp:
694     HandlePragmaMSVtorDisp();
695     return nullptr;
696   case tok::annot_pragma_ms_pragma:
697     HandlePragmaMSPragma();
698     return nullptr;
699   case tok::annot_pragma_dump:
700     HandlePragmaDump();
701     return nullptr;
702   case tok::semi:
703     // Either a C++11 empty-declaration or attribute-declaration.
704     SingleDecl =
705         Actions.ActOnEmptyDeclaration(getCurScope(), attrs, Tok.getLocation());
706     ConsumeExtraSemi(OutsideFunction);
707     break;
708   case tok::r_brace:
709     Diag(Tok, diag::err_extraneous_closing_brace);
710     ConsumeBrace();
711     return nullptr;
712   case tok::eof:
713     Diag(Tok, diag::err_expected_external_declaration);
714     return nullptr;
715   case tok::kw___extension__: {
716     // __extension__ silences extension warnings in the subexpression.
717     ExtensionRAIIObject O(Diags);  // Use RAII to do this.
718     ConsumeToken();
719     return ParseExternalDeclaration(attrs);
720   }
721   case tok::kw_asm: {
722     ProhibitAttributes(attrs);
723 
724     SourceLocation StartLoc = Tok.getLocation();
725     SourceLocation EndLoc;
726 
727     ExprResult Result(ParseSimpleAsm(&EndLoc));
728 
729     // Check if GNU-style InlineAsm is disabled.
730     // Empty asm string is allowed because it will not introduce
731     // any assembly code.
732     if (!(getLangOpts().GNUAsm || Result.isInvalid())) {
733       const auto *SL = cast<StringLiteral>(Result.get());
734       if (!SL->getString().trim().empty())
735         Diag(StartLoc, diag::err_gnu_inline_asm_disabled);
736     }
737 
738     ExpectAndConsume(tok::semi, diag::err_expected_after,
739                      "top-level asm block");
740 
741     if (Result.isInvalid())
742       return nullptr;
743     SingleDecl = Actions.ActOnFileScopeAsmDecl(Result.get(), StartLoc, EndLoc);
744     break;
745   }
746   case tok::at:
747     return ParseObjCAtDirectives(attrs);
748   case tok::minus:
749   case tok::plus:
750     if (!getLangOpts().ObjC) {
751       Diag(Tok, diag::err_expected_external_declaration);
752       ConsumeToken();
753       return nullptr;
754     }
755     SingleDecl = ParseObjCMethodDefinition();
756     break;
757   case tok::code_completion:
758     if (CurParsedObjCImpl) {
759       // Code-complete Objective-C methods even without leading '-'/'+' prefix.
760       Actions.CodeCompleteObjCMethodDecl(getCurScope(),
761                                          /*IsInstanceMethod=*/None,
762                                          /*ReturnType=*/nullptr);
763     }
764     Actions.CodeCompleteOrdinaryName(
765         getCurScope(),
766         CurParsedObjCImpl ? Sema::PCC_ObjCImplementation : Sema::PCC_Namespace);
767     cutOffParsing();
768     return nullptr;
769   case tok::kw_import:
770     SingleDecl = ParseModuleImport(SourceLocation());
771     break;
772   case tok::kw_export:
773     if (getLangOpts().ModulesTS) {
774       SingleDecl = ParseExportDeclaration();
775       break;
776     }
777     // This must be 'export template'. Parse it so we can diagnose our lack
778     // of support.
779     LLVM_FALLTHROUGH;
780   case tok::kw_using:
781   case tok::kw_namespace:
782   case tok::kw_typedef:
783   case tok::kw_template:
784   case tok::kw_static_assert:
785   case tok::kw__Static_assert:
786     // A function definition cannot start with any of these keywords.
787     {
788       SourceLocation DeclEnd;
789       return ParseDeclaration(DeclaratorContext::FileContext, DeclEnd, attrs);
790     }
791 
792   case tok::kw_static:
793     // Parse (then ignore) 'static' prior to a template instantiation. This is
794     // a GCC extension that we intentionally do not support.
795     if (getLangOpts().CPlusPlus && NextToken().is(tok::kw_template)) {
796       Diag(ConsumeToken(), diag::warn_static_inline_explicit_inst_ignored)
797         << 0;
798       SourceLocation DeclEnd;
799       return ParseDeclaration(DeclaratorContext::FileContext, DeclEnd, attrs);
800     }
801     goto dont_know;
802 
803   case tok::kw_inline:
804     if (getLangOpts().CPlusPlus) {
805       tok::TokenKind NextKind = NextToken().getKind();
806 
807       // Inline namespaces. Allowed as an extension even in C++03.
808       if (NextKind == tok::kw_namespace) {
809         SourceLocation DeclEnd;
810         return ParseDeclaration(DeclaratorContext::FileContext, DeclEnd, attrs);
811       }
812 
813       // Parse (then ignore) 'inline' prior to a template instantiation. This is
814       // a GCC extension that we intentionally do not support.
815       if (NextKind == tok::kw_template) {
816         Diag(ConsumeToken(), diag::warn_static_inline_explicit_inst_ignored)
817           << 1;
818         SourceLocation DeclEnd;
819         return ParseDeclaration(DeclaratorContext::FileContext, DeclEnd, attrs);
820       }
821     }
822     goto dont_know;
823 
824   case tok::kw_extern:
825     if (getLangOpts().CPlusPlus && NextToken().is(tok::kw_template)) {
826       // Extern templates
827       SourceLocation ExternLoc = ConsumeToken();
828       SourceLocation TemplateLoc = ConsumeToken();
829       Diag(ExternLoc, getLangOpts().CPlusPlus11 ?
830              diag::warn_cxx98_compat_extern_template :
831              diag::ext_extern_template) << SourceRange(ExternLoc, TemplateLoc);
832       SourceLocation DeclEnd;
833       return Actions.ConvertDeclToDeclGroup(
834           ParseExplicitInstantiation(DeclaratorContext::FileContext, ExternLoc,
835                                      TemplateLoc, DeclEnd, attrs));
836     }
837     goto dont_know;
838 
839   case tok::kw___if_exists:
840   case tok::kw___if_not_exists:
841     ParseMicrosoftIfExistsExternalDeclaration();
842     return nullptr;
843 
844   case tok::kw_module:
845     Diag(Tok, diag::err_unexpected_module_decl);
846     SkipUntil(tok::semi);
847     return nullptr;
848 
849   default:
850   dont_know:
851     if (Tok.isEditorPlaceholder()) {
852       ConsumeToken();
853       return nullptr;
854     }
855     // We can't tell whether this is a function-definition or declaration yet.
856     return ParseDeclarationOrFunctionDefinition(attrs, DS);
857   }
858 
859   // This routine returns a DeclGroup, if the thing we parsed only contains a
860   // single decl, convert it now.
861   return Actions.ConvertDeclToDeclGroup(SingleDecl);
862 }
863 
864 /// Determine whether the current token, if it occurs after a
865 /// declarator, continues a declaration or declaration list.
isDeclarationAfterDeclarator()866 bool Parser::isDeclarationAfterDeclarator() {
867   // Check for '= delete' or '= default'
868   if (getLangOpts().CPlusPlus && Tok.is(tok::equal)) {
869     const Token &KW = NextToken();
870     if (KW.is(tok::kw_default) || KW.is(tok::kw_delete))
871       return false;
872   }
873 
874   return Tok.is(tok::equal) ||      // int X()=  -> not a function def
875     Tok.is(tok::comma) ||           // int X(),  -> not a function def
876     Tok.is(tok::semi)  ||           // int X();  -> not a function def
877     Tok.is(tok::kw_asm) ||          // int X() __asm__ -> not a function def
878     Tok.is(tok::kw___attribute) ||  // int X() __attr__ -> not a function def
879     (getLangOpts().CPlusPlus &&
880      Tok.is(tok::l_paren));         // int X(0) -> not a function def [C++]
881 }
882 
883 /// Determine whether the current token, if it occurs after a
884 /// declarator, indicates the start of a function definition.
isStartOfFunctionDefinition(const ParsingDeclarator & Declarator)885 bool Parser::isStartOfFunctionDefinition(const ParsingDeclarator &Declarator) {
886   assert(Declarator.isFunctionDeclarator() && "Isn't a function declarator");
887   if (Tok.is(tok::l_brace))   // int X() {}
888     return true;
889 
890   // Handle K&R C argument lists: int X(f) int f; {}
891   if (!getLangOpts().CPlusPlus &&
892       Declarator.getFunctionTypeInfo().isKNRPrototype())
893     return isDeclarationSpecifier();
894 
895   if (getLangOpts().CPlusPlus && Tok.is(tok::equal)) {
896     const Token &KW = NextToken();
897     return KW.is(tok::kw_default) || KW.is(tok::kw_delete);
898   }
899 
900   return Tok.is(tok::colon) ||         // X() : Base() {} (used for ctors)
901          Tok.is(tok::kw_try);          // X() try { ... }
902 }
903 
904 /// Parse either a function-definition or a declaration.  We can't tell which
905 /// we have until we read up to the compound-statement in function-definition.
906 /// TemplateParams, if non-NULL, provides the template parameters when we're
907 /// parsing a C++ template-declaration.
908 ///
909 ///       function-definition: [C99 6.9.1]
910 ///         decl-specs      declarator declaration-list[opt] compound-statement
911 /// [C90] function-definition: [C99 6.7.1] - implicit int result
912 /// [C90]   decl-specs[opt] declarator declaration-list[opt] compound-statement
913 ///
914 ///       declaration: [C99 6.7]
915 ///         declaration-specifiers init-declarator-list[opt] ';'
916 /// [!C99]  init-declarator-list ';'                   [TODO: warn in c99 mode]
917 /// [OMP]   threadprivate-directive                              [TODO]
918 ///
919 Parser::DeclGroupPtrTy
ParseDeclOrFunctionDefInternal(ParsedAttributesWithRange & attrs,ParsingDeclSpec & DS,AccessSpecifier AS)920 Parser::ParseDeclOrFunctionDefInternal(ParsedAttributesWithRange &attrs,
921                                        ParsingDeclSpec &DS,
922                                        AccessSpecifier AS) {
923   MaybeParseMicrosoftAttributes(DS.getAttributes());
924   // Parse the common declaration-specifiers piece.
925   ParseDeclarationSpecifiers(DS, ParsedTemplateInfo(), AS,
926                              DeclSpecContext::DSC_top_level);
927 
928   // If we had a free-standing type definition with a missing semicolon, we
929   // may get this far before the problem becomes obvious.
930   if (DS.hasTagDefinition() && DiagnoseMissingSemiAfterTagDefinition(
931                                    DS, AS, DeclSpecContext::DSC_top_level))
932     return nullptr;
933 
934   // C99 6.7.2.3p6: Handle "struct-or-union identifier;", "enum { X };"
935   // declaration-specifiers init-declarator-list[opt] ';'
936   if (Tok.is(tok::semi)) {
937     auto LengthOfTSTToken = [](DeclSpec::TST TKind) {
938       assert(DeclSpec::isDeclRep(TKind));
939       switch(TKind) {
940       case DeclSpec::TST_class:
941         return 5;
942       case DeclSpec::TST_struct:
943         return 6;
944       case DeclSpec::TST_union:
945         return 5;
946       case DeclSpec::TST_enum:
947         return 4;
948       case DeclSpec::TST_interface:
949         return 9;
950       default:
951         llvm_unreachable("we only expect to get the length of the class/struct/union/enum");
952       }
953 
954     };
955     // Suggest correct location to fix '[[attrib]] struct' to 'struct [[attrib]]'
956     SourceLocation CorrectLocationForAttributes =
957         DeclSpec::isDeclRep(DS.getTypeSpecType())
958             ? DS.getTypeSpecTypeLoc().getLocWithOffset(
959                   LengthOfTSTToken(DS.getTypeSpecType()))
960             : SourceLocation();
961     ProhibitAttributes(attrs, CorrectLocationForAttributes);
962     ConsumeToken();
963     RecordDecl *AnonRecord = nullptr;
964     Decl *TheDecl = Actions.ParsedFreeStandingDeclSpec(getCurScope(), AS_none,
965                                                        DS, AnonRecord);
966     DS.complete(TheDecl);
967     if (getLangOpts().OpenCL)
968       Actions.setCurrentOpenCLExtensionForDecl(TheDecl);
969     if (AnonRecord) {
970       Decl* decls[] = {AnonRecord, TheDecl};
971       return Actions.BuildDeclaratorGroup(decls);
972     }
973     return Actions.ConvertDeclToDeclGroup(TheDecl);
974   }
975 
976   DS.takeAttributesFrom(attrs);
977 
978   // ObjC2 allows prefix attributes on class interfaces and protocols.
979   // FIXME: This still needs better diagnostics. We should only accept
980   // attributes here, no types, etc.
981   if (getLangOpts().ObjC && Tok.is(tok::at)) {
982     SourceLocation AtLoc = ConsumeToken(); // the "@"
983     if (!Tok.isObjCAtKeyword(tok::objc_interface) &&
984         !Tok.isObjCAtKeyword(tok::objc_protocol)) {
985       Diag(Tok, diag::err_objc_unexpected_attr);
986       SkipUntil(tok::semi); // FIXME: better skip?
987       return nullptr;
988     }
989 
990     DS.abort();
991 
992     const char *PrevSpec = nullptr;
993     unsigned DiagID;
994     if (DS.SetTypeSpecType(DeclSpec::TST_unspecified, AtLoc, PrevSpec, DiagID,
995                            Actions.getASTContext().getPrintingPolicy()))
996       Diag(AtLoc, DiagID) << PrevSpec;
997 
998     if (Tok.isObjCAtKeyword(tok::objc_protocol))
999       return ParseObjCAtProtocolDeclaration(AtLoc, DS.getAttributes());
1000 
1001     return Actions.ConvertDeclToDeclGroup(
1002             ParseObjCAtInterfaceDeclaration(AtLoc, DS.getAttributes()));
1003   }
1004 
1005   // If the declspec consisted only of 'extern' and we have a string
1006   // literal following it, this must be a C++ linkage specifier like
1007   // 'extern "C"'.
1008   if (getLangOpts().CPlusPlus && isTokenStringLiteral() &&
1009       DS.getStorageClassSpec() == DeclSpec::SCS_extern &&
1010       DS.getParsedSpecifiers() == DeclSpec::PQ_StorageClassSpecifier) {
1011     Decl *TheDecl = ParseLinkage(DS, DeclaratorContext::FileContext);
1012     return Actions.ConvertDeclToDeclGroup(TheDecl);
1013   }
1014 
1015   return ParseDeclGroup(DS, DeclaratorContext::FileContext);
1016 }
1017 
1018 Parser::DeclGroupPtrTy
ParseDeclarationOrFunctionDefinition(ParsedAttributesWithRange & attrs,ParsingDeclSpec * DS,AccessSpecifier AS)1019 Parser::ParseDeclarationOrFunctionDefinition(ParsedAttributesWithRange &attrs,
1020                                              ParsingDeclSpec *DS,
1021                                              AccessSpecifier AS) {
1022   if (DS) {
1023     return ParseDeclOrFunctionDefInternal(attrs, *DS, AS);
1024   } else {
1025     ParsingDeclSpec PDS(*this);
1026     // Must temporarily exit the objective-c container scope for
1027     // parsing c constructs and re-enter objc container scope
1028     // afterwards.
1029     ObjCDeclContextSwitch ObjCDC(*this);
1030 
1031     return ParseDeclOrFunctionDefInternal(attrs, PDS, AS);
1032   }
1033 }
1034 
1035 /// ParseFunctionDefinition - We parsed and verified that the specified
1036 /// Declarator is well formed.  If this is a K&R-style function, read the
1037 /// parameters declaration-list, then start the compound-statement.
1038 ///
1039 ///       function-definition: [C99 6.9.1]
1040 ///         decl-specs      declarator declaration-list[opt] compound-statement
1041 /// [C90] function-definition: [C99 6.7.1] - implicit int result
1042 /// [C90]   decl-specs[opt] declarator declaration-list[opt] compound-statement
1043 /// [C++] function-definition: [C++ 8.4]
1044 ///         decl-specifier-seq[opt] declarator ctor-initializer[opt]
1045 ///         function-body
1046 /// [C++] function-definition: [C++ 8.4]
1047 ///         decl-specifier-seq[opt] declarator function-try-block
1048 ///
ParseFunctionDefinition(ParsingDeclarator & D,const ParsedTemplateInfo & TemplateInfo,LateParsedAttrList * LateParsedAttrs)1049 Decl *Parser::ParseFunctionDefinition(ParsingDeclarator &D,
1050                                       const ParsedTemplateInfo &TemplateInfo,
1051                                       LateParsedAttrList *LateParsedAttrs) {
1052   // Poison SEH identifiers so they are flagged as illegal in function bodies.
1053   PoisonSEHIdentifiersRAIIObject PoisonSEHIdentifiers(*this, true);
1054   const DeclaratorChunk::FunctionTypeInfo &FTI = D.getFunctionTypeInfo();
1055 
1056   // If this is C90 and the declspecs were completely missing, fudge in an
1057   // implicit int.  We do this here because this is the only place where
1058   // declaration-specifiers are completely optional in the grammar.
1059   if (getLangOpts().ImplicitInt && D.getDeclSpec().isEmpty()) {
1060     const char *PrevSpec;
1061     unsigned DiagID;
1062     const PrintingPolicy &Policy = Actions.getASTContext().getPrintingPolicy();
1063     D.getMutableDeclSpec().SetTypeSpecType(DeclSpec::TST_int,
1064                                            D.getIdentifierLoc(),
1065                                            PrevSpec, DiagID,
1066                                            Policy);
1067     D.SetRangeBegin(D.getDeclSpec().getSourceRange().getBegin());
1068   }
1069 
1070   // If this declaration was formed with a K&R-style identifier list for the
1071   // arguments, parse declarations for all of the args next.
1072   // int foo(a,b) int a; float b; {}
1073   if (FTI.isKNRPrototype())
1074     ParseKNRParamDeclarations(D);
1075 
1076   // We should have either an opening brace or, in a C++ constructor,
1077   // we may have a colon.
1078   if (Tok.isNot(tok::l_brace) &&
1079       (!getLangOpts().CPlusPlus ||
1080        (Tok.isNot(tok::colon) && Tok.isNot(tok::kw_try) &&
1081         Tok.isNot(tok::equal)))) {
1082     Diag(Tok, diag::err_expected_fn_body);
1083 
1084     // Skip over garbage, until we get to '{'.  Don't eat the '{'.
1085     SkipUntil(tok::l_brace, StopAtSemi | StopBeforeMatch);
1086 
1087     // If we didn't find the '{', bail out.
1088     if (Tok.isNot(tok::l_brace))
1089       return nullptr;
1090   }
1091 
1092   // Check to make sure that any normal attributes are allowed to be on
1093   // a definition.  Late parsed attributes are checked at the end.
1094   if (Tok.isNot(tok::equal)) {
1095     for (const ParsedAttr &AL : D.getAttributes())
1096       if (AL.isKnownToGCC() && !AL.isCXX11Attribute())
1097         Diag(AL.getLoc(), diag::warn_attribute_on_function_definition)
1098             << AL.getName();
1099   }
1100 
1101   // In delayed template parsing mode, for function template we consume the
1102   // tokens and store them for late parsing at the end of the translation unit.
1103   if (getLangOpts().DelayedTemplateParsing && Tok.isNot(tok::equal) &&
1104       TemplateInfo.Kind == ParsedTemplateInfo::Template &&
1105       Actions.canDelayFunctionBody(D)) {
1106     MultiTemplateParamsArg TemplateParameterLists(*TemplateInfo.TemplateParams);
1107 
1108     ParseScope BodyScope(this, Scope::FnScope | Scope::DeclScope |
1109                                    Scope::CompoundStmtScope);
1110     Scope *ParentScope = getCurScope()->getParent();
1111 
1112     D.setFunctionDefinitionKind(FDK_Definition);
1113     Decl *DP = Actions.HandleDeclarator(ParentScope, D,
1114                                         TemplateParameterLists);
1115     D.complete(DP);
1116     D.getMutableDeclSpec().abort();
1117 
1118     if (SkipFunctionBodies && (!DP || Actions.canSkipFunctionBody(DP)) &&
1119         trySkippingFunctionBody()) {
1120       BodyScope.Exit();
1121       return Actions.ActOnSkippedFunctionBody(DP);
1122     }
1123 
1124     CachedTokens Toks;
1125     LexTemplateFunctionForLateParsing(Toks);
1126 
1127     if (DP) {
1128       FunctionDecl *FnD = DP->getAsFunction();
1129       Actions.CheckForFunctionRedefinition(FnD);
1130       Actions.MarkAsLateParsedTemplate(FnD, DP, Toks);
1131     }
1132     return DP;
1133   }
1134   else if (CurParsedObjCImpl &&
1135            !TemplateInfo.TemplateParams &&
1136            (Tok.is(tok::l_brace) || Tok.is(tok::kw_try) ||
1137             Tok.is(tok::colon)) &&
1138       Actions.CurContext->isTranslationUnit()) {
1139     ParseScope BodyScope(this, Scope::FnScope | Scope::DeclScope |
1140                                    Scope::CompoundStmtScope);
1141     Scope *ParentScope = getCurScope()->getParent();
1142 
1143     D.setFunctionDefinitionKind(FDK_Definition);
1144     Decl *FuncDecl = Actions.HandleDeclarator(ParentScope, D,
1145                                               MultiTemplateParamsArg());
1146     D.complete(FuncDecl);
1147     D.getMutableDeclSpec().abort();
1148     if (FuncDecl) {
1149       // Consume the tokens and store them for later parsing.
1150       StashAwayMethodOrFunctionBodyTokens(FuncDecl);
1151       CurParsedObjCImpl->HasCFunction = true;
1152       return FuncDecl;
1153     }
1154     // FIXME: Should we really fall through here?
1155   }
1156 
1157   // Enter a scope for the function body.
1158   ParseScope BodyScope(this, Scope::FnScope | Scope::DeclScope |
1159                                  Scope::CompoundStmtScope);
1160 
1161   // Tell the actions module that we have entered a function definition with the
1162   // specified Declarator for the function.
1163   Sema::SkipBodyInfo SkipBody;
1164   Decl *Res = Actions.ActOnStartOfFunctionDef(getCurScope(), D,
1165                                               TemplateInfo.TemplateParams
1166                                                   ? *TemplateInfo.TemplateParams
1167                                                   : MultiTemplateParamsArg(),
1168                                               &SkipBody);
1169 
1170   if (SkipBody.ShouldSkip) {
1171     SkipFunctionBody();
1172     return Res;
1173   }
1174 
1175   // Break out of the ParsingDeclarator context before we parse the body.
1176   D.complete(Res);
1177 
1178   // Break out of the ParsingDeclSpec context, too.  This const_cast is
1179   // safe because we're always the sole owner.
1180   D.getMutableDeclSpec().abort();
1181 
1182   if (TryConsumeToken(tok::equal)) {
1183     assert(getLangOpts().CPlusPlus && "Only C++ function definitions have '='");
1184 
1185     bool Delete = false;
1186     SourceLocation KWLoc;
1187     if (TryConsumeToken(tok::kw_delete, KWLoc)) {
1188       Diag(KWLoc, getLangOpts().CPlusPlus11
1189                       ? diag::warn_cxx98_compat_defaulted_deleted_function
1190                       : diag::ext_defaulted_deleted_function)
1191         << 1 /* deleted */;
1192       Actions.SetDeclDeleted(Res, KWLoc);
1193       Delete = true;
1194     } else if (TryConsumeToken(tok::kw_default, KWLoc)) {
1195       Diag(KWLoc, getLangOpts().CPlusPlus11
1196                       ? diag::warn_cxx98_compat_defaulted_deleted_function
1197                       : diag::ext_defaulted_deleted_function)
1198         << 0 /* defaulted */;
1199       Actions.SetDeclDefaulted(Res, KWLoc);
1200     } else {
1201       llvm_unreachable("function definition after = not 'delete' or 'default'");
1202     }
1203 
1204     if (Tok.is(tok::comma)) {
1205       Diag(KWLoc, diag::err_default_delete_in_multiple_declaration)
1206         << Delete;
1207       SkipUntil(tok::semi);
1208     } else if (ExpectAndConsume(tok::semi, diag::err_expected_after,
1209                                 Delete ? "delete" : "default")) {
1210       SkipUntil(tok::semi);
1211     }
1212 
1213     Stmt *GeneratedBody = Res ? Res->getBody() : nullptr;
1214     Actions.ActOnFinishFunctionBody(Res, GeneratedBody, false);
1215     return Res;
1216   }
1217 
1218   if (SkipFunctionBodies && (!Res || Actions.canSkipFunctionBody(Res)) &&
1219       trySkippingFunctionBody()) {
1220     BodyScope.Exit();
1221     Actions.ActOnSkippedFunctionBody(Res);
1222     return Actions.ActOnFinishFunctionBody(Res, nullptr, false);
1223   }
1224 
1225   if (Tok.is(tok::kw_try))
1226     return ParseFunctionTryBlock(Res, BodyScope);
1227 
1228   // If we have a colon, then we're probably parsing a C++
1229   // ctor-initializer.
1230   if (Tok.is(tok::colon)) {
1231     ParseConstructorInitializer(Res);
1232 
1233     // Recover from error.
1234     if (!Tok.is(tok::l_brace)) {
1235       BodyScope.Exit();
1236       Actions.ActOnFinishFunctionBody(Res, nullptr);
1237       return Res;
1238     }
1239   } else
1240     Actions.ActOnDefaultCtorInitializers(Res);
1241 
1242   // Late attributes are parsed in the same scope as the function body.
1243   if (LateParsedAttrs)
1244     ParseLexedAttributeList(*LateParsedAttrs, Res, false, true);
1245 
1246   return ParseFunctionStatementBody(Res, BodyScope);
1247 }
1248 
SkipFunctionBody()1249 void Parser::SkipFunctionBody() {
1250   if (Tok.is(tok::equal)) {
1251     SkipUntil(tok::semi);
1252     return;
1253   }
1254 
1255   bool IsFunctionTryBlock = Tok.is(tok::kw_try);
1256   if (IsFunctionTryBlock)
1257     ConsumeToken();
1258 
1259   CachedTokens Skipped;
1260   if (ConsumeAndStoreFunctionPrologue(Skipped))
1261     SkipMalformedDecl();
1262   else {
1263     SkipUntil(tok::r_brace);
1264     while (IsFunctionTryBlock && Tok.is(tok::kw_catch)) {
1265       SkipUntil(tok::l_brace);
1266       SkipUntil(tok::r_brace);
1267     }
1268   }
1269 }
1270 
1271 /// ParseKNRParamDeclarations - Parse 'declaration-list[opt]' which provides
1272 /// types for a function with a K&R-style identifier list for arguments.
ParseKNRParamDeclarations(Declarator & D)1273 void Parser::ParseKNRParamDeclarations(Declarator &D) {
1274   // We know that the top-level of this declarator is a function.
1275   DeclaratorChunk::FunctionTypeInfo &FTI = D.getFunctionTypeInfo();
1276 
1277   // Enter function-declaration scope, limiting any declarators to the
1278   // function prototype scope, including parameter declarators.
1279   ParseScope PrototypeScope(this, Scope::FunctionPrototypeScope |
1280                             Scope::FunctionDeclarationScope | Scope::DeclScope);
1281 
1282   // Read all the argument declarations.
1283   while (isDeclarationSpecifier()) {
1284     SourceLocation DSStart = Tok.getLocation();
1285 
1286     // Parse the common declaration-specifiers piece.
1287     DeclSpec DS(AttrFactory);
1288     ParseDeclarationSpecifiers(DS);
1289 
1290     // C99 6.9.1p6: 'each declaration in the declaration list shall have at
1291     // least one declarator'.
1292     // NOTE: GCC just makes this an ext-warn.  It's not clear what it does with
1293     // the declarations though.  It's trivial to ignore them, really hard to do
1294     // anything else with them.
1295     if (TryConsumeToken(tok::semi)) {
1296       Diag(DSStart, diag::err_declaration_does_not_declare_param);
1297       continue;
1298     }
1299 
1300     // C99 6.9.1p6: Declarations shall contain no storage-class specifiers other
1301     // than register.
1302     if (DS.getStorageClassSpec() != DeclSpec::SCS_unspecified &&
1303         DS.getStorageClassSpec() != DeclSpec::SCS_register) {
1304       Diag(DS.getStorageClassSpecLoc(),
1305            diag::err_invalid_storage_class_in_func_decl);
1306       DS.ClearStorageClassSpecs();
1307     }
1308     if (DS.getThreadStorageClassSpec() != DeclSpec::TSCS_unspecified) {
1309       Diag(DS.getThreadStorageClassSpecLoc(),
1310            diag::err_invalid_storage_class_in_func_decl);
1311       DS.ClearStorageClassSpecs();
1312     }
1313 
1314     // Parse the first declarator attached to this declspec.
1315     Declarator ParmDeclarator(DS, DeclaratorContext::KNRTypeListContext);
1316     ParseDeclarator(ParmDeclarator);
1317 
1318     // Handle the full declarator list.
1319     while (1) {
1320       // If attributes are present, parse them.
1321       MaybeParseGNUAttributes(ParmDeclarator);
1322 
1323       // Ask the actions module to compute the type for this declarator.
1324       Decl *Param =
1325         Actions.ActOnParamDeclarator(getCurScope(), ParmDeclarator);
1326 
1327       if (Param &&
1328           // A missing identifier has already been diagnosed.
1329           ParmDeclarator.getIdentifier()) {
1330 
1331         // Scan the argument list looking for the correct param to apply this
1332         // type.
1333         for (unsigned i = 0; ; ++i) {
1334           // C99 6.9.1p6: those declarators shall declare only identifiers from
1335           // the identifier list.
1336           if (i == FTI.NumParams) {
1337             Diag(ParmDeclarator.getIdentifierLoc(), diag::err_no_matching_param)
1338               << ParmDeclarator.getIdentifier();
1339             break;
1340           }
1341 
1342           if (FTI.Params[i].Ident == ParmDeclarator.getIdentifier()) {
1343             // Reject redefinitions of parameters.
1344             if (FTI.Params[i].Param) {
1345               Diag(ParmDeclarator.getIdentifierLoc(),
1346                    diag::err_param_redefinition)
1347                  << ParmDeclarator.getIdentifier();
1348             } else {
1349               FTI.Params[i].Param = Param;
1350             }
1351             break;
1352           }
1353         }
1354       }
1355 
1356       // If we don't have a comma, it is either the end of the list (a ';') or
1357       // an error, bail out.
1358       if (Tok.isNot(tok::comma))
1359         break;
1360 
1361       ParmDeclarator.clear();
1362 
1363       // Consume the comma.
1364       ParmDeclarator.setCommaLoc(ConsumeToken());
1365 
1366       // Parse the next declarator.
1367       ParseDeclarator(ParmDeclarator);
1368     }
1369 
1370     // Consume ';' and continue parsing.
1371     if (!ExpectAndConsumeSemi(diag::err_expected_semi_declaration))
1372       continue;
1373 
1374     // Otherwise recover by skipping to next semi or mandatory function body.
1375     if (SkipUntil(tok::l_brace, StopAtSemi | StopBeforeMatch))
1376       break;
1377     TryConsumeToken(tok::semi);
1378   }
1379 
1380   // The actions module must verify that all arguments were declared.
1381   Actions.ActOnFinishKNRParamDeclarations(getCurScope(), D, Tok.getLocation());
1382 }
1383 
1384 
1385 /// ParseAsmStringLiteral - This is just a normal string-literal, but is not
1386 /// allowed to be a wide string, and is not subject to character translation.
1387 ///
1388 /// [GNU] asm-string-literal:
1389 ///         string-literal
1390 ///
ParseAsmStringLiteral()1391 ExprResult Parser::ParseAsmStringLiteral() {
1392   if (!isTokenStringLiteral()) {
1393     Diag(Tok, diag::err_expected_string_literal)
1394       << /*Source='in...'*/0 << "'asm'";
1395     return ExprError();
1396   }
1397 
1398   ExprResult AsmString(ParseStringLiteralExpression());
1399   if (!AsmString.isInvalid()) {
1400     const auto *SL = cast<StringLiteral>(AsmString.get());
1401     if (!SL->isAscii()) {
1402       Diag(Tok, diag::err_asm_operand_wide_string_literal)
1403         << SL->isWide()
1404         << SL->getSourceRange();
1405       return ExprError();
1406     }
1407   }
1408   return AsmString;
1409 }
1410 
1411 /// ParseSimpleAsm
1412 ///
1413 /// [GNU] simple-asm-expr:
1414 ///         'asm' '(' asm-string-literal ')'
1415 ///
ParseSimpleAsm(SourceLocation * EndLoc)1416 ExprResult Parser::ParseSimpleAsm(SourceLocation *EndLoc) {
1417   assert(Tok.is(tok::kw_asm) && "Not an asm!");
1418   SourceLocation Loc = ConsumeToken();
1419 
1420   if (Tok.is(tok::kw_volatile)) {
1421     // Remove from the end of 'asm' to the end of 'volatile'.
1422     SourceRange RemovalRange(PP.getLocForEndOfToken(Loc),
1423                              PP.getLocForEndOfToken(Tok.getLocation()));
1424 
1425     Diag(Tok, diag::warn_file_asm_volatile)
1426       << FixItHint::CreateRemoval(RemovalRange);
1427     ConsumeToken();
1428   }
1429 
1430   BalancedDelimiterTracker T(*this, tok::l_paren);
1431   if (T.consumeOpen()) {
1432     Diag(Tok, diag::err_expected_lparen_after) << "asm";
1433     return ExprError();
1434   }
1435 
1436   ExprResult Result(ParseAsmStringLiteral());
1437 
1438   if (!Result.isInvalid()) {
1439     // Close the paren and get the location of the end bracket
1440     T.consumeClose();
1441     if (EndLoc)
1442       *EndLoc = T.getCloseLocation();
1443   } else if (SkipUntil(tok::r_paren, StopAtSemi | StopBeforeMatch)) {
1444     if (EndLoc)
1445       *EndLoc = Tok.getLocation();
1446     ConsumeParen();
1447   }
1448 
1449   return Result;
1450 }
1451 
1452 /// Get the TemplateIdAnnotation from the token and put it in the
1453 /// cleanup pool so that it gets destroyed when parsing the current top level
1454 /// declaration is finished.
takeTemplateIdAnnotation(const Token & tok)1455 TemplateIdAnnotation *Parser::takeTemplateIdAnnotation(const Token &tok) {
1456   assert(tok.is(tok::annot_template_id) && "Expected template-id token");
1457   TemplateIdAnnotation *
1458       Id = static_cast<TemplateIdAnnotation *>(tok.getAnnotationValue());
1459   return Id;
1460 }
1461 
AnnotateScopeToken(CXXScopeSpec & SS,bool IsNewAnnotation)1462 void Parser::AnnotateScopeToken(CXXScopeSpec &SS, bool IsNewAnnotation) {
1463   // Push the current token back into the token stream (or revert it if it is
1464   // cached) and use an annotation scope token for current token.
1465   if (PP.isBacktrackEnabled())
1466     PP.RevertCachedTokens(1);
1467   else
1468     PP.EnterToken(Tok);
1469   Tok.setKind(tok::annot_cxxscope);
1470   Tok.setAnnotationValue(Actions.SaveNestedNameSpecifierAnnotation(SS));
1471   Tok.setAnnotationRange(SS.getRange());
1472 
1473   // In case the tokens were cached, have Preprocessor replace them
1474   // with the annotation token.  We don't need to do this if we've
1475   // just reverted back to a prior state.
1476   if (IsNewAnnotation)
1477     PP.AnnotateCachedTokens(Tok);
1478 }
1479 
1480 /// Attempt to classify the name at the current token position. This may
1481 /// form a type, scope or primary expression annotation, or replace the token
1482 /// with a typo-corrected keyword. This is only appropriate when the current
1483 /// name must refer to an entity which has already been declared.
1484 ///
1485 /// \param IsAddressOfOperand Must be \c true if the name is preceded by an '&'
1486 ///        and might possibly have a dependent nested name specifier.
1487 /// \param CCC Indicates how to perform typo-correction for this name. If NULL,
1488 ///        no typo correction will be performed.
1489 Parser::AnnotatedNameKind
TryAnnotateName(bool IsAddressOfOperand,std::unique_ptr<CorrectionCandidateCallback> CCC)1490 Parser::TryAnnotateName(bool IsAddressOfOperand,
1491                         std::unique_ptr<CorrectionCandidateCallback> CCC) {
1492   assert(Tok.is(tok::identifier) || Tok.is(tok::annot_cxxscope));
1493 
1494   const bool EnteringContext = false;
1495   const bool WasScopeAnnotation = Tok.is(tok::annot_cxxscope);
1496 
1497   CXXScopeSpec SS;
1498   if (getLangOpts().CPlusPlus &&
1499       ParseOptionalCXXScopeSpecifier(SS, nullptr, EnteringContext))
1500     return ANK_Error;
1501 
1502   if (Tok.isNot(tok::identifier) || SS.isInvalid()) {
1503     if (TryAnnotateTypeOrScopeTokenAfterScopeSpec(SS, !WasScopeAnnotation))
1504       return ANK_Error;
1505     return ANK_Unresolved;
1506   }
1507 
1508   IdentifierInfo *Name = Tok.getIdentifierInfo();
1509   SourceLocation NameLoc = Tok.getLocation();
1510 
1511   // FIXME: Move the tentative declaration logic into ClassifyName so we can
1512   // typo-correct to tentatively-declared identifiers.
1513   if (isTentativelyDeclared(Name)) {
1514     // Identifier has been tentatively declared, and thus cannot be resolved as
1515     // an expression. Fall back to annotating it as a type.
1516     if (TryAnnotateTypeOrScopeTokenAfterScopeSpec(SS, !WasScopeAnnotation))
1517       return ANK_Error;
1518     return Tok.is(tok::annot_typename) ? ANK_Success : ANK_TentativeDecl;
1519   }
1520 
1521   Token Next = NextToken();
1522 
1523   // Look up and classify the identifier. We don't perform any typo-correction
1524   // after a scope specifier, because in general we can't recover from typos
1525   // there (eg, after correcting 'A::template B<X>::C' [sic], we would need to
1526   // jump back into scope specifier parsing).
1527   Sema::NameClassification Classification = Actions.ClassifyName(
1528       getCurScope(), SS, Name, NameLoc, Next, IsAddressOfOperand,
1529       SS.isEmpty() ? std::move(CCC) : nullptr);
1530 
1531   switch (Classification.getKind()) {
1532   case Sema::NC_Error:
1533     return ANK_Error;
1534 
1535   case Sema::NC_Keyword:
1536     // The identifier was typo-corrected to a keyword.
1537     Tok.setIdentifierInfo(Name);
1538     Tok.setKind(Name->getTokenID());
1539     PP.TypoCorrectToken(Tok);
1540     if (SS.isNotEmpty())
1541       AnnotateScopeToken(SS, !WasScopeAnnotation);
1542     // We've "annotated" this as a keyword.
1543     return ANK_Success;
1544 
1545   case Sema::NC_Unknown:
1546     // It's not something we know about. Leave it unannotated.
1547     break;
1548 
1549   case Sema::NC_Type: {
1550     SourceLocation BeginLoc = NameLoc;
1551     if (SS.isNotEmpty())
1552       BeginLoc = SS.getBeginLoc();
1553 
1554     /// An Objective-C object type followed by '<' is a specialization of
1555     /// a parameterized class type or a protocol-qualified type.
1556     ParsedType Ty = Classification.getType();
1557     if (getLangOpts().ObjC && NextToken().is(tok::less) &&
1558         (Ty.get()->isObjCObjectType() ||
1559          Ty.get()->isObjCObjectPointerType())) {
1560       // Consume the name.
1561       SourceLocation IdentifierLoc = ConsumeToken();
1562       SourceLocation NewEndLoc;
1563       TypeResult NewType
1564           = parseObjCTypeArgsAndProtocolQualifiers(IdentifierLoc, Ty,
1565                                                    /*consumeLastToken=*/false,
1566                                                    NewEndLoc);
1567       if (NewType.isUsable())
1568         Ty = NewType.get();
1569       else if (Tok.is(tok::eof)) // Nothing to do here, bail out...
1570         return ANK_Error;
1571     }
1572 
1573     Tok.setKind(tok::annot_typename);
1574     setTypeAnnotation(Tok, Ty);
1575     Tok.setAnnotationEndLoc(Tok.getLocation());
1576     Tok.setLocation(BeginLoc);
1577     PP.AnnotateCachedTokens(Tok);
1578     return ANK_Success;
1579   }
1580 
1581   case Sema::NC_Expression:
1582     Tok.setKind(tok::annot_primary_expr);
1583     setExprAnnotation(Tok, Classification.getExpression());
1584     Tok.setAnnotationEndLoc(NameLoc);
1585     if (SS.isNotEmpty())
1586       Tok.setLocation(SS.getBeginLoc());
1587     PP.AnnotateCachedTokens(Tok);
1588     return ANK_Success;
1589 
1590   case Sema::NC_TypeTemplate:
1591     if (Next.isNot(tok::less)) {
1592       // This may be a type template being used as a template template argument.
1593       if (SS.isNotEmpty())
1594         AnnotateScopeToken(SS, !WasScopeAnnotation);
1595       return ANK_TemplateName;
1596     }
1597     LLVM_FALLTHROUGH;
1598   case Sema::NC_VarTemplate:
1599   case Sema::NC_FunctionTemplate: {
1600     // We have a type, variable or function template followed by '<'.
1601     ConsumeToken();
1602     UnqualifiedId Id;
1603     Id.setIdentifier(Name, NameLoc);
1604     if (AnnotateTemplateIdToken(
1605             TemplateTy::make(Classification.getTemplateName()),
1606             Classification.getTemplateNameKind(), SS, SourceLocation(), Id))
1607       return ANK_Error;
1608     return ANK_Success;
1609   }
1610 
1611   case Sema::NC_NestedNameSpecifier:
1612     llvm_unreachable("already parsed nested name specifier");
1613   }
1614 
1615   // Unable to classify the name, but maybe we can annotate a scope specifier.
1616   if (SS.isNotEmpty())
1617     AnnotateScopeToken(SS, !WasScopeAnnotation);
1618   return ANK_Unresolved;
1619 }
1620 
TryKeywordIdentFallback(bool DisableKeyword)1621 bool Parser::TryKeywordIdentFallback(bool DisableKeyword) {
1622   assert(Tok.isNot(tok::identifier));
1623   Diag(Tok, diag::ext_keyword_as_ident)
1624     << PP.getSpelling(Tok)
1625     << DisableKeyword;
1626   if (DisableKeyword)
1627     Tok.getIdentifierInfo()->revertTokenIDToIdentifier();
1628   Tok.setKind(tok::identifier);
1629   return true;
1630 }
1631 
1632 /// TryAnnotateTypeOrScopeToken - If the current token position is on a
1633 /// typename (possibly qualified in C++) or a C++ scope specifier not followed
1634 /// by a typename, TryAnnotateTypeOrScopeToken will replace one or more tokens
1635 /// with a single annotation token representing the typename or C++ scope
1636 /// respectively.
1637 /// This simplifies handling of C++ scope specifiers and allows efficient
1638 /// backtracking without the need to re-parse and resolve nested-names and
1639 /// typenames.
1640 /// It will mainly be called when we expect to treat identifiers as typenames
1641 /// (if they are typenames). For example, in C we do not expect identifiers
1642 /// inside expressions to be treated as typenames so it will not be called
1643 /// for expressions in C.
1644 /// The benefit for C/ObjC is that a typename will be annotated and
1645 /// Actions.getTypeName will not be needed to be called again (e.g. getTypeName
1646 /// will not be called twice, once to check whether we have a declaration
1647 /// specifier, and another one to get the actual type inside
1648 /// ParseDeclarationSpecifiers).
1649 ///
1650 /// This returns true if an error occurred.
1651 ///
1652 /// Note that this routine emits an error if you call it with ::new or ::delete
1653 /// as the current tokens, so only call it in contexts where these are invalid.
TryAnnotateTypeOrScopeToken()1654 bool Parser::TryAnnotateTypeOrScopeToken() {
1655   assert((Tok.is(tok::identifier) || Tok.is(tok::coloncolon) ||
1656           Tok.is(tok::kw_typename) || Tok.is(tok::annot_cxxscope) ||
1657           Tok.is(tok::kw_decltype) || Tok.is(tok::annot_template_id) ||
1658           Tok.is(tok::kw___super)) &&
1659          "Cannot be a type or scope token!");
1660 
1661   if (Tok.is(tok::kw_typename)) {
1662     // MSVC lets you do stuff like:
1663     //   typename typedef T_::D D;
1664     //
1665     // We will consume the typedef token here and put it back after we have
1666     // parsed the first identifier, transforming it into something more like:
1667     //   typename T_::D typedef D;
1668     if (getLangOpts().MSVCCompat && NextToken().is(tok::kw_typedef)) {
1669       Token TypedefToken;
1670       PP.Lex(TypedefToken);
1671       bool Result = TryAnnotateTypeOrScopeToken();
1672       PP.EnterToken(Tok);
1673       Tok = TypedefToken;
1674       if (!Result)
1675         Diag(Tok.getLocation(), diag::warn_expected_qualified_after_typename);
1676       return Result;
1677     }
1678 
1679     // Parse a C++ typename-specifier, e.g., "typename T::type".
1680     //
1681     //   typename-specifier:
1682     //     'typename' '::' [opt] nested-name-specifier identifier
1683     //     'typename' '::' [opt] nested-name-specifier template [opt]
1684     //            simple-template-id
1685     SourceLocation TypenameLoc = ConsumeToken();
1686     CXXScopeSpec SS;
1687     if (ParseOptionalCXXScopeSpecifier(SS, /*ObjectType=*/nullptr,
1688                                        /*EnteringContext=*/false, nullptr,
1689                                        /*IsTypename*/ true))
1690       return true;
1691     if (!SS.isSet()) {
1692       if (Tok.is(tok::identifier) || Tok.is(tok::annot_template_id) ||
1693           Tok.is(tok::annot_decltype)) {
1694         // Attempt to recover by skipping the invalid 'typename'
1695         if (Tok.is(tok::annot_decltype) ||
1696             (!TryAnnotateTypeOrScopeToken() && Tok.isAnnotation())) {
1697           unsigned DiagID = diag::err_expected_qualified_after_typename;
1698           // MS compatibility: MSVC permits using known types with typename.
1699           // e.g. "typedef typename T* pointer_type"
1700           if (getLangOpts().MicrosoftExt)
1701             DiagID = diag::warn_expected_qualified_after_typename;
1702           Diag(Tok.getLocation(), DiagID);
1703           return false;
1704         }
1705       }
1706       if (Tok.isEditorPlaceholder())
1707         return true;
1708 
1709       Diag(Tok.getLocation(), diag::err_expected_qualified_after_typename);
1710       return true;
1711     }
1712 
1713     TypeResult Ty;
1714     if (Tok.is(tok::identifier)) {
1715       // FIXME: check whether the next token is '<', first!
1716       Ty = Actions.ActOnTypenameType(getCurScope(), TypenameLoc, SS,
1717                                      *Tok.getIdentifierInfo(),
1718                                      Tok.getLocation());
1719     } else if (Tok.is(tok::annot_template_id)) {
1720       TemplateIdAnnotation *TemplateId = takeTemplateIdAnnotation(Tok);
1721       if (TemplateId->Kind != TNK_Type_template &&
1722           TemplateId->Kind != TNK_Dependent_template_name) {
1723         Diag(Tok, diag::err_typename_refers_to_non_type_template)
1724           << Tok.getAnnotationRange();
1725         return true;
1726       }
1727 
1728       ASTTemplateArgsPtr TemplateArgsPtr(TemplateId->getTemplateArgs(),
1729                                          TemplateId->NumArgs);
1730 
1731       Ty = Actions.ActOnTypenameType(getCurScope(), TypenameLoc, SS,
1732                                      TemplateId->TemplateKWLoc,
1733                                      TemplateId->Template,
1734                                      TemplateId->Name,
1735                                      TemplateId->TemplateNameLoc,
1736                                      TemplateId->LAngleLoc,
1737                                      TemplateArgsPtr,
1738                                      TemplateId->RAngleLoc);
1739     } else {
1740       Diag(Tok, diag::err_expected_type_name_after_typename)
1741         << SS.getRange();
1742       return true;
1743     }
1744 
1745     SourceLocation EndLoc = Tok.getLastLoc();
1746     Tok.setKind(tok::annot_typename);
1747     setTypeAnnotation(Tok, Ty.isInvalid() ? nullptr : Ty.get());
1748     Tok.setAnnotationEndLoc(EndLoc);
1749     Tok.setLocation(TypenameLoc);
1750     PP.AnnotateCachedTokens(Tok);
1751     return false;
1752   }
1753 
1754   // Remembers whether the token was originally a scope annotation.
1755   bool WasScopeAnnotation = Tok.is(tok::annot_cxxscope);
1756 
1757   CXXScopeSpec SS;
1758   if (getLangOpts().CPlusPlus)
1759     if (ParseOptionalCXXScopeSpecifier(SS, nullptr, /*EnteringContext*/false))
1760       return true;
1761 
1762   return TryAnnotateTypeOrScopeTokenAfterScopeSpec(SS, !WasScopeAnnotation);
1763 }
1764 
1765 /// Try to annotate a type or scope token, having already parsed an
1766 /// optional scope specifier. \p IsNewScope should be \c true unless the scope
1767 /// specifier was extracted from an existing tok::annot_cxxscope annotation.
TryAnnotateTypeOrScopeTokenAfterScopeSpec(CXXScopeSpec & SS,bool IsNewScope)1768 bool Parser::TryAnnotateTypeOrScopeTokenAfterScopeSpec(CXXScopeSpec &SS,
1769                                                        bool IsNewScope) {
1770   if (Tok.is(tok::identifier)) {
1771     // Determine whether the identifier is a type name.
1772     if (ParsedType Ty = Actions.getTypeName(
1773             *Tok.getIdentifierInfo(), Tok.getLocation(), getCurScope(), &SS,
1774             false, NextToken().is(tok::period), nullptr,
1775             /*IsCtorOrDtorName=*/false,
1776             /*NonTrivialTypeSourceInfo*/true,
1777             /*IsClassTemplateDeductionContext*/true)) {
1778       SourceLocation BeginLoc = Tok.getLocation();
1779       if (SS.isNotEmpty()) // it was a C++ qualified type name.
1780         BeginLoc = SS.getBeginLoc();
1781 
1782       /// An Objective-C object type followed by '<' is a specialization of
1783       /// a parameterized class type or a protocol-qualified type.
1784       if (getLangOpts().ObjC && NextToken().is(tok::less) &&
1785           (Ty.get()->isObjCObjectType() ||
1786            Ty.get()->isObjCObjectPointerType())) {
1787         // Consume the name.
1788         SourceLocation IdentifierLoc = ConsumeToken();
1789         SourceLocation NewEndLoc;
1790         TypeResult NewType
1791           = parseObjCTypeArgsAndProtocolQualifiers(IdentifierLoc, Ty,
1792                                                    /*consumeLastToken=*/false,
1793                                                    NewEndLoc);
1794         if (NewType.isUsable())
1795           Ty = NewType.get();
1796         else if (Tok.is(tok::eof)) // Nothing to do here, bail out...
1797           return false;
1798       }
1799 
1800       // This is a typename. Replace the current token in-place with an
1801       // annotation type token.
1802       Tok.setKind(tok::annot_typename);
1803       setTypeAnnotation(Tok, Ty);
1804       Tok.setAnnotationEndLoc(Tok.getLocation());
1805       Tok.setLocation(BeginLoc);
1806 
1807       // In case the tokens were cached, have Preprocessor replace
1808       // them with the annotation token.
1809       PP.AnnotateCachedTokens(Tok);
1810       return false;
1811     }
1812 
1813     if (!getLangOpts().CPlusPlus) {
1814       // If we're in C, we can't have :: tokens at all (the lexer won't return
1815       // them).  If the identifier is not a type, then it can't be scope either,
1816       // just early exit.
1817       return false;
1818     }
1819 
1820     // If this is a template-id, annotate with a template-id or type token.
1821     if (NextToken().is(tok::less)) {
1822       TemplateTy Template;
1823       UnqualifiedId TemplateName;
1824       TemplateName.setIdentifier(Tok.getIdentifierInfo(), Tok.getLocation());
1825       bool MemberOfUnknownSpecialization;
1826       if (TemplateNameKind TNK = Actions.isTemplateName(
1827               getCurScope(), SS,
1828               /*hasTemplateKeyword=*/false, TemplateName,
1829               /*ObjectType=*/nullptr, /*EnteringContext*/false, Template,
1830               MemberOfUnknownSpecialization)) {
1831         // Consume the identifier.
1832         ConsumeToken();
1833         if (AnnotateTemplateIdToken(Template, TNK, SS, SourceLocation(),
1834                                     TemplateName)) {
1835           // If an unrecoverable error occurred, we need to return true here,
1836           // because the token stream is in a damaged state.  We may not return
1837           // a valid identifier.
1838           return true;
1839         }
1840       }
1841     }
1842 
1843     // The current token, which is either an identifier or a
1844     // template-id, is not part of the annotation. Fall through to
1845     // push that token back into the stream and complete the C++ scope
1846     // specifier annotation.
1847   }
1848 
1849   if (Tok.is(tok::annot_template_id)) {
1850     TemplateIdAnnotation *TemplateId = takeTemplateIdAnnotation(Tok);
1851     if (TemplateId->Kind == TNK_Type_template) {
1852       // A template-id that refers to a type was parsed into a
1853       // template-id annotation in a context where we weren't allowed
1854       // to produce a type annotation token. Update the template-id
1855       // annotation token to a type annotation token now.
1856       AnnotateTemplateIdTokenAsType();
1857       return false;
1858     }
1859   }
1860 
1861   if (SS.isEmpty())
1862     return false;
1863 
1864   // A C++ scope specifier that isn't followed by a typename.
1865   AnnotateScopeToken(SS, IsNewScope);
1866   return false;
1867 }
1868 
1869 /// TryAnnotateScopeToken - Like TryAnnotateTypeOrScopeToken but only
1870 /// annotates C++ scope specifiers and template-ids.  This returns
1871 /// true if there was an error that could not be recovered from.
1872 ///
1873 /// Note that this routine emits an error if you call it with ::new or ::delete
1874 /// as the current tokens, so only call it in contexts where these are invalid.
TryAnnotateCXXScopeToken(bool EnteringContext)1875 bool Parser::TryAnnotateCXXScopeToken(bool EnteringContext) {
1876   assert(getLangOpts().CPlusPlus &&
1877          "Call sites of this function should be guarded by checking for C++");
1878   assert((Tok.is(tok::identifier) || Tok.is(tok::coloncolon) ||
1879           (Tok.is(tok::annot_template_id) && NextToken().is(tok::coloncolon)) ||
1880           Tok.is(tok::kw_decltype) || Tok.is(tok::kw___super)) &&
1881          "Cannot be a type or scope token!");
1882 
1883   CXXScopeSpec SS;
1884   if (ParseOptionalCXXScopeSpecifier(SS, nullptr, EnteringContext))
1885     return true;
1886   if (SS.isEmpty())
1887     return false;
1888 
1889   AnnotateScopeToken(SS, true);
1890   return false;
1891 }
1892 
isTokenEqualOrEqualTypo()1893 bool Parser::isTokenEqualOrEqualTypo() {
1894   tok::TokenKind Kind = Tok.getKind();
1895   switch (Kind) {
1896   default:
1897     return false;
1898   case tok::ampequal:            // &=
1899   case tok::starequal:           // *=
1900   case tok::plusequal:           // +=
1901   case tok::minusequal:          // -=
1902   case tok::exclaimequal:        // !=
1903   case tok::slashequal:          // /=
1904   case tok::percentequal:        // %=
1905   case tok::lessequal:           // <=
1906   case tok::lesslessequal:       // <<=
1907   case tok::greaterequal:        // >=
1908   case tok::greatergreaterequal: // >>=
1909   case tok::caretequal:          // ^=
1910   case tok::pipeequal:           // |=
1911   case tok::equalequal:          // ==
1912     Diag(Tok, diag::err_invalid_token_after_declarator_suggest_equal)
1913         << Kind
1914         << FixItHint::CreateReplacement(SourceRange(Tok.getLocation()), "=");
1915     LLVM_FALLTHROUGH;
1916   case tok::equal:
1917     return true;
1918   }
1919 }
1920 
handleUnexpectedCodeCompletionToken()1921 SourceLocation Parser::handleUnexpectedCodeCompletionToken() {
1922   assert(Tok.is(tok::code_completion));
1923   PrevTokLocation = Tok.getLocation();
1924 
1925   for (Scope *S = getCurScope(); S; S = S->getParent()) {
1926     if (S->getFlags() & Scope::FnScope) {
1927       Actions.CodeCompleteOrdinaryName(getCurScope(),
1928                                        Sema::PCC_RecoveryInFunction);
1929       cutOffParsing();
1930       return PrevTokLocation;
1931     }
1932 
1933     if (S->getFlags() & Scope::ClassScope) {
1934       Actions.CodeCompleteOrdinaryName(getCurScope(), Sema::PCC_Class);
1935       cutOffParsing();
1936       return PrevTokLocation;
1937     }
1938   }
1939 
1940   Actions.CodeCompleteOrdinaryName(getCurScope(), Sema::PCC_Namespace);
1941   cutOffParsing();
1942   return PrevTokLocation;
1943 }
1944 
1945 // Code-completion pass-through functions
1946 
CodeCompleteDirective(bool InConditional)1947 void Parser::CodeCompleteDirective(bool InConditional) {
1948   Actions.CodeCompletePreprocessorDirective(InConditional);
1949 }
1950 
CodeCompleteInConditionalExclusion()1951 void Parser::CodeCompleteInConditionalExclusion() {
1952   Actions.CodeCompleteInPreprocessorConditionalExclusion(getCurScope());
1953 }
1954 
CodeCompleteMacroName(bool IsDefinition)1955 void Parser::CodeCompleteMacroName(bool IsDefinition) {
1956   Actions.CodeCompletePreprocessorMacroName(IsDefinition);
1957 }
1958 
CodeCompletePreprocessorExpression()1959 void Parser::CodeCompletePreprocessorExpression() {
1960   Actions.CodeCompletePreprocessorExpression();
1961 }
1962 
CodeCompleteMacroArgument(IdentifierInfo * Macro,MacroInfo * MacroInfo,unsigned ArgumentIndex)1963 void Parser::CodeCompleteMacroArgument(IdentifierInfo *Macro,
1964                                        MacroInfo *MacroInfo,
1965                                        unsigned ArgumentIndex) {
1966   Actions.CodeCompletePreprocessorMacroArgument(getCurScope(), Macro, MacroInfo,
1967                                                 ArgumentIndex);
1968 }
1969 
CodeCompleteIncludedFile(llvm::StringRef Dir,bool IsAngled)1970 void Parser::CodeCompleteIncludedFile(llvm::StringRef Dir, bool IsAngled) {
1971   Actions.CodeCompleteIncludedFile(Dir, IsAngled);
1972 }
1973 
CodeCompleteNaturalLanguage()1974 void Parser::CodeCompleteNaturalLanguage() {
1975   Actions.CodeCompleteNaturalLanguage();
1976 }
1977 
ParseMicrosoftIfExistsCondition(IfExistsCondition & Result)1978 bool Parser::ParseMicrosoftIfExistsCondition(IfExistsCondition& Result) {
1979   assert((Tok.is(tok::kw___if_exists) || Tok.is(tok::kw___if_not_exists)) &&
1980          "Expected '__if_exists' or '__if_not_exists'");
1981   Result.IsIfExists = Tok.is(tok::kw___if_exists);
1982   Result.KeywordLoc = ConsumeToken();
1983 
1984   BalancedDelimiterTracker T(*this, tok::l_paren);
1985   if (T.consumeOpen()) {
1986     Diag(Tok, diag::err_expected_lparen_after)
1987       << (Result.IsIfExists? "__if_exists" : "__if_not_exists");
1988     return true;
1989   }
1990 
1991   // Parse nested-name-specifier.
1992   if (getLangOpts().CPlusPlus)
1993     ParseOptionalCXXScopeSpecifier(Result.SS, nullptr,
1994                                    /*EnteringContext=*/false);
1995 
1996   // Check nested-name specifier.
1997   if (Result.SS.isInvalid()) {
1998     T.skipToEnd();
1999     return true;
2000   }
2001 
2002   // Parse the unqualified-id.
2003   SourceLocation TemplateKWLoc; // FIXME: parsed, but unused.
2004   if (ParseUnqualifiedId(
2005           Result.SS, /*EnteringContext*/false, /*AllowDestructorName*/true,
2006           /*AllowConstructorName*/true, /*AllowDeductionGuide*/false, nullptr,
2007           &TemplateKWLoc, Result.Name)) {
2008     T.skipToEnd();
2009     return true;
2010   }
2011 
2012   if (T.consumeClose())
2013     return true;
2014 
2015   // Check if the symbol exists.
2016   switch (Actions.CheckMicrosoftIfExistsSymbol(getCurScope(), Result.KeywordLoc,
2017                                                Result.IsIfExists, Result.SS,
2018                                                Result.Name)) {
2019   case Sema::IER_Exists:
2020     Result.Behavior = Result.IsIfExists ? IEB_Parse : IEB_Skip;
2021     break;
2022 
2023   case Sema::IER_DoesNotExist:
2024     Result.Behavior = !Result.IsIfExists ? IEB_Parse : IEB_Skip;
2025     break;
2026 
2027   case Sema::IER_Dependent:
2028     Result.Behavior = IEB_Dependent;
2029     break;
2030 
2031   case Sema::IER_Error:
2032     return true;
2033   }
2034 
2035   return false;
2036 }
2037 
ParseMicrosoftIfExistsExternalDeclaration()2038 void Parser::ParseMicrosoftIfExistsExternalDeclaration() {
2039   IfExistsCondition Result;
2040   if (ParseMicrosoftIfExistsCondition(Result))
2041     return;
2042 
2043   BalancedDelimiterTracker Braces(*this, tok::l_brace);
2044   if (Braces.consumeOpen()) {
2045     Diag(Tok, diag::err_expected) << tok::l_brace;
2046     return;
2047   }
2048 
2049   switch (Result.Behavior) {
2050   case IEB_Parse:
2051     // Parse declarations below.
2052     break;
2053 
2054   case IEB_Dependent:
2055     llvm_unreachable("Cannot have a dependent external declaration");
2056 
2057   case IEB_Skip:
2058     Braces.skipToEnd();
2059     return;
2060   }
2061 
2062   // Parse the declarations.
2063   // FIXME: Support module import within __if_exists?
2064   while (Tok.isNot(tok::r_brace) && !isEofOrEom()) {
2065     ParsedAttributesWithRange attrs(AttrFactory);
2066     MaybeParseCXX11Attributes(attrs);
2067     DeclGroupPtrTy Result = ParseExternalDeclaration(attrs);
2068     if (Result && !getCurScope()->getParent())
2069       Actions.getASTConsumer().HandleTopLevelDecl(Result.get());
2070   }
2071   Braces.consumeClose();
2072 }
2073 
2074 /// Parse a C++ Modules TS module declaration, which appears at the beginning
2075 /// of a module interface, module partition, or module implementation file.
2076 ///
2077 ///   module-declaration:   [Modules TS + P0273R0 + P0629R0]
2078 ///     'export'[opt] 'module' 'partition'[opt]
2079 ///            module-name attribute-specifier-seq[opt] ';'
2080 ///
2081 /// Note that 'partition' is a context-sensitive keyword.
ParseModuleDecl()2082 Parser::DeclGroupPtrTy Parser::ParseModuleDecl() {
2083   SourceLocation StartLoc = Tok.getLocation();
2084 
2085   Sema::ModuleDeclKind MDK = TryConsumeToken(tok::kw_export)
2086                                  ? Sema::ModuleDeclKind::Interface
2087                                  : Sema::ModuleDeclKind::Implementation;
2088 
2089   assert(Tok.is(tok::kw_module) && "not a module declaration");
2090   SourceLocation ModuleLoc = ConsumeToken();
2091 
2092   if (Tok.is(tok::identifier) && NextToken().is(tok::identifier) &&
2093       Tok.getIdentifierInfo()->isStr("partition")) {
2094     // If 'partition' is present, this must be a module interface unit.
2095     if (MDK != Sema::ModuleDeclKind::Interface)
2096       Diag(Tok.getLocation(), diag::err_module_implementation_partition)
2097         << FixItHint::CreateInsertion(ModuleLoc, "export ");
2098     MDK = Sema::ModuleDeclKind::Partition;
2099     ConsumeToken();
2100   }
2101 
2102   SmallVector<std::pair<IdentifierInfo *, SourceLocation>, 2> Path;
2103   if (ParseModuleName(ModuleLoc, Path, /*IsImport*/false))
2104     return nullptr;
2105 
2106   // We don't support any module attributes yet; just parse them and diagnose.
2107   ParsedAttributesWithRange Attrs(AttrFactory);
2108   MaybeParseCXX11Attributes(Attrs);
2109   ProhibitCXX11Attributes(Attrs, diag::err_attribute_not_module_attr);
2110 
2111   ExpectAndConsumeSemi(diag::err_module_expected_semi);
2112 
2113   return Actions.ActOnModuleDecl(StartLoc, ModuleLoc, MDK, Path);
2114 }
2115 
2116 /// Parse a module import declaration. This is essentially the same for
2117 /// Objective-C and the C++ Modules TS, except for the leading '@' (in ObjC)
2118 /// and the trailing optional attributes (in C++).
2119 ///
2120 /// [ObjC]  @import declaration:
2121 ///           '@' 'import' module-name ';'
2122 /// [ModTS] module-import-declaration:
2123 ///           'import' module-name attribute-specifier-seq[opt] ';'
ParseModuleImport(SourceLocation AtLoc)2124 Decl *Parser::ParseModuleImport(SourceLocation AtLoc) {
2125   assert((AtLoc.isInvalid() ? Tok.is(tok::kw_import)
2126                             : Tok.isObjCAtKeyword(tok::objc_import)) &&
2127          "Improper start to module import");
2128   bool IsObjCAtImport = Tok.isObjCAtKeyword(tok::objc_import);
2129   SourceLocation ImportLoc = ConsumeToken();
2130   SourceLocation StartLoc = AtLoc.isInvalid() ? ImportLoc : AtLoc;
2131 
2132   SmallVector<std::pair<IdentifierInfo *, SourceLocation>, 2> Path;
2133   if (ParseModuleName(ImportLoc, Path, /*IsImport*/true))
2134     return nullptr;
2135 
2136   ParsedAttributesWithRange Attrs(AttrFactory);
2137   MaybeParseCXX11Attributes(Attrs);
2138   // We don't support any module import attributes yet.
2139   ProhibitCXX11Attributes(Attrs, diag::err_attribute_not_import_attr);
2140 
2141   if (PP.hadModuleLoaderFatalFailure()) {
2142     // With a fatal failure in the module loader, we abort parsing.
2143     cutOffParsing();
2144     return nullptr;
2145   }
2146 
2147   DeclResult Import = Actions.ActOnModuleImport(StartLoc, ImportLoc, Path);
2148   ExpectAndConsumeSemi(diag::err_module_expected_semi);
2149   if (Import.isInvalid())
2150     return nullptr;
2151 
2152   // Using '@import' in framework headers requires modules to be enabled so that
2153   // the header is parseable. Emit a warning to make the user aware.
2154   if (IsObjCAtImport && AtLoc.isValid()) {
2155     auto &SrcMgr = PP.getSourceManager();
2156     auto *FE = SrcMgr.getFileEntryForID(SrcMgr.getFileID(AtLoc));
2157     if (FE && llvm::sys::path::parent_path(FE->getDir()->getName())
2158                   .endswith(".framework"))
2159       Diags.Report(AtLoc, diag::warn_atimport_in_framework_header);
2160   }
2161 
2162   return Import.get();
2163 }
2164 
2165 /// Parse a C++ Modules TS / Objective-C module name (both forms use the same
2166 /// grammar).
2167 ///
2168 ///         module-name:
2169 ///           module-name-qualifier[opt] identifier
2170 ///         module-name-qualifier:
2171 ///           module-name-qualifier[opt] identifier '.'
ParseModuleName(SourceLocation UseLoc,SmallVectorImpl<std::pair<IdentifierInfo *,SourceLocation>> & Path,bool IsImport)2172 bool Parser::ParseModuleName(
2173     SourceLocation UseLoc,
2174     SmallVectorImpl<std::pair<IdentifierInfo *, SourceLocation>> &Path,
2175     bool IsImport) {
2176   // Parse the module path.
2177   while (true) {
2178     if (!Tok.is(tok::identifier)) {
2179       if (Tok.is(tok::code_completion)) {
2180         Actions.CodeCompleteModuleImport(UseLoc, Path);
2181         cutOffParsing();
2182         return true;
2183       }
2184 
2185       Diag(Tok, diag::err_module_expected_ident) << IsImport;
2186       SkipUntil(tok::semi);
2187       return true;
2188     }
2189 
2190     // Record this part of the module path.
2191     Path.push_back(std::make_pair(Tok.getIdentifierInfo(), Tok.getLocation()));
2192     ConsumeToken();
2193 
2194     if (Tok.isNot(tok::period))
2195       return false;
2196 
2197     ConsumeToken();
2198   }
2199 }
2200 
2201 /// Try recover parser when module annotation appears where it must not
2202 /// be found.
2203 /// \returns false if the recover was successful and parsing may be continued, or
2204 /// true if parser must bail out to top level and handle the token there.
parseMisplacedModuleImport()2205 bool Parser::parseMisplacedModuleImport() {
2206   while (true) {
2207     switch (Tok.getKind()) {
2208     case tok::annot_module_end:
2209       // If we recovered from a misplaced module begin, we expect to hit a
2210       // misplaced module end too. Stay in the current context when this
2211       // happens.
2212       if (MisplacedModuleBeginCount) {
2213         --MisplacedModuleBeginCount;
2214         Actions.ActOnModuleEnd(Tok.getLocation(),
2215                                reinterpret_cast<Module *>(
2216                                    Tok.getAnnotationValue()));
2217         ConsumeAnnotationToken();
2218         continue;
2219       }
2220       // Inform caller that recovery failed, the error must be handled at upper
2221       // level. This will generate the desired "missing '}' at end of module"
2222       // diagnostics on the way out.
2223       return true;
2224     case tok::annot_module_begin:
2225       // Recover by entering the module (Sema will diagnose).
2226       Actions.ActOnModuleBegin(Tok.getLocation(),
2227                                reinterpret_cast<Module *>(
2228                                    Tok.getAnnotationValue()));
2229       ConsumeAnnotationToken();
2230       ++MisplacedModuleBeginCount;
2231       continue;
2232     case tok::annot_module_include:
2233       // Module import found where it should not be, for instance, inside a
2234       // namespace. Recover by importing the module.
2235       Actions.ActOnModuleInclude(Tok.getLocation(),
2236                                  reinterpret_cast<Module *>(
2237                                      Tok.getAnnotationValue()));
2238       ConsumeAnnotationToken();
2239       // If there is another module import, process it.
2240       continue;
2241     default:
2242       return false;
2243     }
2244   }
2245   return false;
2246 }
2247 
diagnoseOverflow()2248 bool BalancedDelimiterTracker::diagnoseOverflow() {
2249   P.Diag(P.Tok, diag::err_bracket_depth_exceeded)
2250     << P.getLangOpts().BracketDepth;
2251   P.Diag(P.Tok, diag::note_bracket_depth);
2252   P.cutOffParsing();
2253   return true;
2254 }
2255 
expectAndConsume(unsigned DiagID,const char * Msg,tok::TokenKind SkipToTok)2256 bool BalancedDelimiterTracker::expectAndConsume(unsigned DiagID,
2257                                                 const char *Msg,
2258                                                 tok::TokenKind SkipToTok) {
2259   LOpen = P.Tok.getLocation();
2260   if (P.ExpectAndConsume(Kind, DiagID, Msg)) {
2261     if (SkipToTok != tok::unknown)
2262       P.SkipUntil(SkipToTok, Parser::StopAtSemi);
2263     return true;
2264   }
2265 
2266   if (getDepth() < P.getLangOpts().BracketDepth)
2267     return false;
2268 
2269   return diagnoseOverflow();
2270 }
2271 
diagnoseMissingClose()2272 bool BalancedDelimiterTracker::diagnoseMissingClose() {
2273   assert(!P.Tok.is(Close) && "Should have consumed closing delimiter");
2274 
2275   if (P.Tok.is(tok::annot_module_end))
2276     P.Diag(P.Tok, diag::err_missing_before_module_end) << Close;
2277   else
2278     P.Diag(P.Tok, diag::err_expected) << Close;
2279   P.Diag(LOpen, diag::note_matching) << Kind;
2280 
2281   // If we're not already at some kind of closing bracket, skip to our closing
2282   // token.
2283   if (P.Tok.isNot(tok::r_paren) && P.Tok.isNot(tok::r_brace) &&
2284       P.Tok.isNot(tok::r_square) &&
2285       P.SkipUntil(Close, FinalToken,
2286                   Parser::StopAtSemi | Parser::StopBeforeMatch) &&
2287       P.Tok.is(Close))
2288     LClose = P.ConsumeAnyToken();
2289   return true;
2290 }
2291 
skipToEnd()2292 void BalancedDelimiterTracker::skipToEnd() {
2293   P.SkipUntil(Close, Parser::StopBeforeMatch);
2294   consumeClose();
2295 }
2296