ParseCXXInlineMethods.cpp revision d01c915dda27bb0045687f0a08bbcab1dd40e652
14cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis//===--- ParseCXXInlineMethods.cpp - C++ class inline methods parsing------===// 24cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// 34cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// The LLVM Compiler Infrastructure 44cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// 54cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// This file is distributed under the University of Illinois Open Source 64cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// License. See LICENSE.TXT for details. 74cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// 84cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis//===----------------------------------------------------------------------===// 94cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// 104cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// This file implements parsing for C++ class inline methods. 114cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis// 124cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis//===----------------------------------------------------------------------===// 134cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 14500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/Parse/ParseDiagnostic.h" 154cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis#include "clang/Parse/Parser.h" 164cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis#include "clang/Parse/DeclSpec.h" 174cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis#include "clang/Parse/Scope.h" 184cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidisusing namespace clang; 194cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 20d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl/// ParseCXXInlineMethodDef - We parsed and verified that the specified 214cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis/// Declarator is a well formed C++ inline method definition. Now lex its body 224cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis/// and store its tokens for parsing after the C++ class is complete. 23b28317a8e5e0e2953d1e5406d753d6c3c7f1e7d2Chris LattnerParser::DeclPtrTy 244cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios KyrtzidisParser::ParseCXXInlineMethodDef(AccessSpecifier AS, Declarator &D) { 254cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis assert(D.getTypeObject(0).Kind == DeclaratorChunk::Function && 264cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis "This isn't a function declarator!"); 27d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl assert((Tok.is(tok::l_brace) || Tok.is(tok::colon) || Tok.is(tok::kw_try)) && 28d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl "Current token not a '{', ':' or 'try'!"); 294cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 30682bf92db408a6cbc3d37b5496a99b6ef85041ecChris Lattner DeclPtrTy FnD = Actions.ActOnCXXMemberDeclarator(CurScope, AS, D, 0, 0); 314cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 324cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Consume the tokens and store them for later parsing. 334cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 346569d68745c8213709740337d2be52b031384f58Douglas Gregor getCurrentClass().MethodDefs.push_back(LexedMethod(FnD)); 356569d68745c8213709740337d2be52b031384f58Douglas Gregor CachedTokens &Toks = getCurrentClass().MethodDefs.back().Toks; 364cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 37d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl tok::TokenKind kind = Tok.getKind(); 38d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl // We may have a constructor initializer or function-try-block here. 39d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl if (kind == tok::colon || kind == tok::kw_try) { 407ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor // Consume everything up to (and including) the left brace. 4172b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor if (!ConsumeAndStoreUntil(tok::l_brace, tok::unknown, Toks, tok::semi)) { 423f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor // We didn't find the left-brace we expected after the 433f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor // constructor initializer. 443f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor if (Tok.is(tok::semi)) { 453f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor // We found a semicolon; complain, consume the semicolon, and 463f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor // don't try to parse this method later. 473f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor Diag(Tok.getLocation(), diag::err_expected_lbrace); 483f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor ConsumeAnyToken(); 496569d68745c8213709740337d2be52b031384f58Douglas Gregor getCurrentClass().MethodDefs.pop_back(); 503f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor return FnD; 513f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor } 523f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor } 533f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor 547ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor } else { 557ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor // Begin by storing the '{' token. 567ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor Toks.push_back(Tok); 577ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor ConsumeBrace(); 587ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor } 597ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor // Consume everything up to (and including) the matching right brace. 6072b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor ConsumeAndStoreUntil(tok::r_brace, tok::unknown, Toks); 614cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 62d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl // If we're in a function-try-block, we need to store all the catch blocks. 63d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl if (kind == tok::kw_try) { 64d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl while (Tok.is(tok::kw_catch)) { 65d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl ConsumeAndStoreUntil(tok::l_brace, tok::unknown, Toks); 66d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl ConsumeAndStoreUntil(tok::r_brace, tok::unknown, Toks); 67d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl } 68d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl } 69d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl 704cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis return FnD; 714cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis} 724cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 7372b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor/// ParseLexedMethodDeclarations - We finished parsing the member 7472b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor/// specification of a top (non-nested) C++ class. Now go over the 7572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor/// stack of method declarations with some parts for which parsing was 7672b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor/// delayed (such as default arguments) and parse them. 776569d68745c8213709740337d2be52b031384f58Douglas Gregorvoid Parser::ParseLexedMethodDeclarations(ParsingClass &Class) { 786569d68745c8213709740337d2be52b031384f58Douglas Gregor bool HasTemplateScope = !Class.TopLevelClass && Class.TemplateScope; 796569d68745c8213709740337d2be52b031384f58Douglas Gregor ParseScope TemplateScope(this, Scope::TemplateParamScope, HasTemplateScope); 806569d68745c8213709740337d2be52b031384f58Douglas Gregor if (HasTemplateScope) 816569d68745c8213709740337d2be52b031384f58Douglas Gregor Actions.ActOnReenterTemplateScope(CurScope, Class.TagOrTemplate); 826569d68745c8213709740337d2be52b031384f58Douglas Gregor 836569d68745c8213709740337d2be52b031384f58Douglas Gregor bool HasClassScope = !Class.TopLevelClass; 846569d68745c8213709740337d2be52b031384f58Douglas Gregor ParseScope ClassScope(this, Scope::ClassScope|Scope::DeclScope, 856569d68745c8213709740337d2be52b031384f58Douglas Gregor HasClassScope); 866569d68745c8213709740337d2be52b031384f58Douglas Gregor 876569d68745c8213709740337d2be52b031384f58Douglas Gregor for (; !Class.MethodDecls.empty(); Class.MethodDecls.pop_front()) { 886569d68745c8213709740337d2be52b031384f58Douglas Gregor LateParsedMethodDeclaration &LM = Class.MethodDecls.front(); 8972b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor 906569d68745c8213709740337d2be52b031384f58Douglas Gregor // FIXME: For member function templates, we'll need to introduce a 916569d68745c8213709740337d2be52b031384f58Douglas Gregor // scope for the template parameters. 926569d68745c8213709740337d2be52b031384f58Douglas Gregor 9372b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor // Start the delayed C++ method declaration 9472b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor Actions.ActOnStartDelayedCXXMethodDeclaration(CurScope, LM.Method); 9572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor 9672b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor // Introduce the parameters into scope and parse their default 9772b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor // arguments. 983218c4bb3b5d7250f12420de6db7ef3e3f805a75Douglas Gregor ParseScope PrototypeScope(this, 993218c4bb3b5d7250f12420de6db7ef3e3f805a75Douglas Gregor Scope::FunctionPrototypeScope|Scope::DeclScope); 10072b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor for (unsigned I = 0, N = LM.DefaultArgs.size(); I != N; ++I) { 10172b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor // Introduce the parameter into scope. 10272b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor Actions.ActOnDelayedCXXMethodParameter(CurScope, LM.DefaultArgs[I].Param); 10372b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor 10472b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor if (CachedTokens *Toks = LM.DefaultArgs[I].Toks) { 10572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor // Parse the default argument from its saved token stream. 10672b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor Toks->push_back(Tok); // So that the current token doesn't get lost 10772b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor PP.EnterTokenStream(&Toks->front(), Toks->size(), true, false); 10872b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor 10972b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor // Consume the previously-pushed token. 11072b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor ConsumeAnyToken(); 11172b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor 11272b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor // Consume the '='. 11372b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor assert(Tok.is(tok::equal) && "Default argument not starting with '='"); 11472b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor SourceLocation EqualLoc = ConsumeToken(); 11572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor 11672b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor OwningExprResult DefArgResult(ParseAssignmentExpression()); 11772b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor if (DefArgResult.isInvalid()) 11872b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor Actions.ActOnParamDefaultArgumentError(LM.DefaultArgs[I].Param); 11972b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor else 12072b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor Actions.ActOnParamDefaultArgument(LM.DefaultArgs[I].Param, EqualLoc, 121f53597fb16142bdb4a66901f8c0b768db4f2a548Sebastian Redl move(DefArgResult)); 12272b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor delete Toks; 12372b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor LM.DefaultArgs[I].Toks = 0; 12472b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor } 12572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor } 12672b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor PrototypeScope.Exit(); 12772b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor 12872b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor // Finish the delayed C++ method declaration. 12972b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor Actions.ActOnFinishDelayedCXXMethodDeclaration(CurScope, LM.Method); 13072b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor } 1316569d68745c8213709740337d2be52b031384f58Douglas Gregor 1326569d68745c8213709740337d2be52b031384f58Douglas Gregor for (unsigned I = 0, N = Class.NestedClasses.size(); I != N; ++I) 1336569d68745c8213709740337d2be52b031384f58Douglas Gregor ParseLexedMethodDeclarations(*Class.NestedClasses[I]); 13472b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor} 13572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor 1364cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis/// ParseLexedMethodDefs - We finished parsing the member specification of a top 1374cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis/// (non-nested) C++ class. Now go over the stack of lexed methods that were 1384cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis/// collected during its parsing and parse them all. 1396569d68745c8213709740337d2be52b031384f58Douglas Gregorvoid Parser::ParseLexedMethodDefs(ParsingClass &Class) { 1406569d68745c8213709740337d2be52b031384f58Douglas Gregor bool HasTemplateScope = !Class.TopLevelClass && Class.TemplateScope; 1416569d68745c8213709740337d2be52b031384f58Douglas Gregor ParseScope TemplateScope(this, Scope::TemplateParamScope, HasTemplateScope); 1426569d68745c8213709740337d2be52b031384f58Douglas Gregor if (HasTemplateScope) 1436569d68745c8213709740337d2be52b031384f58Douglas Gregor Actions.ActOnReenterTemplateScope(CurScope, Class.TagOrTemplate); 1446569d68745c8213709740337d2be52b031384f58Douglas Gregor 1456569d68745c8213709740337d2be52b031384f58Douglas Gregor bool HasClassScope = !Class.TopLevelClass; 1466569d68745c8213709740337d2be52b031384f58Douglas Gregor ParseScope ClassScope(this, Scope::ClassScope|Scope::DeclScope, 1476569d68745c8213709740337d2be52b031384f58Douglas Gregor HasClassScope); 1486569d68745c8213709740337d2be52b031384f58Douglas Gregor 1496569d68745c8213709740337d2be52b031384f58Douglas Gregor for (; !Class.MethodDefs.empty(); Class.MethodDefs.pop_front()) { 1506569d68745c8213709740337d2be52b031384f58Douglas Gregor LexedMethod &LM = Class.MethodDefs.front(); 1514cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 1524cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis assert(!LM.Toks.empty() && "Empty body!"); 1534cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Append the current token at the end of the new token stream so that it 1544cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // doesn't get lost. 1554cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis LM.Toks.push_back(Tok); 1564cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis PP.EnterTokenStream(&LM.Toks.front(), LM.Toks.size(), true, false); 1574cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 1584cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Consume the previously pushed token. 1594cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeAnyToken(); 160d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl assert((Tok.is(tok::l_brace) || Tok.is(tok::colon) || Tok.is(tok::kw_try)) 161d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl && "Inline method not starting with '{', ':' or 'try'"); 1624cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 1634cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Parse the method body. Function body parsing code is similar enough 1644cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // to be re-used for method bodies as well. 1658935b8b49053122ddd3ab4cd59af0fe5eb9c23cfDouglas Gregor ParseScope FnScope(this, Scope::FnScope|Scope::DeclScope); 1664cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Actions.ActOnStartOfFunctionDef(CurScope, LM.D); 1674cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 168d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl if (Tok.is(tok::kw_try)) { 169d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl ParseFunctionTryBlock(LM.D); 170de1b60a9868f80f0872ed05d78df3b40a10ba5caSebastian Redl continue; 171d3a413d3b8eb39bcee5944bc545d9997c1abe492Sebastian Redl } 1727ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor if (Tok.is(tok::colon)) 1737ad8390f7992ab7f19b1460c5f0b9d96f165c4e9Douglas Gregor ParseConstructorInitializer(LM.D); 174d01c915dda27bb0045687f0a08bbcab1dd40e652Fariborz Jahanian else { 175d01c915dda27bb0045687f0a08bbcab1dd40e652Fariborz Jahanian Actions.ActOnDefaultInitializers(LM.D); 176d01c915dda27bb0045687f0a08bbcab1dd40e652Fariborz Jahanian } 177d01c915dda27bb0045687f0a08bbcab1dd40e652Fariborz Jahanian 17840e9bc84a2ab49fc33c2b1a95c6674ab2b820e9eChris Lattner // FIXME: What if ParseConstructorInitializer doesn't leave us with a '{'?? 17940e9bc84a2ab49fc33c2b1a95c6674ab2b820e9eChris Lattner ParseFunctionStatementBody(LM.D); 1804cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis } 1816569d68745c8213709740337d2be52b031384f58Douglas Gregor 1826569d68745c8213709740337d2be52b031384f58Douglas Gregor for (unsigned I = 0, N = Class.NestedClasses.size(); I != N; ++I) 1836569d68745c8213709740337d2be52b031384f58Douglas Gregor ParseLexedMethodDefs(*Class.NestedClasses[I]); 1844cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis} 1854cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 1864cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis/// ConsumeAndStoreUntil - Consume and store the token at the passed token 18772b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor/// container until the token 'T' is reached (which gets 18872b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor/// consumed/stored too, if ConsumeFinalToken). 1893f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor/// If EarlyAbortIf is specified, then we will stop early if we find that 1903f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor/// token at the top level. 19172b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor/// Returns true if token 'T1' or 'T2' was found. 1924cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis/// NOTE: This is a specialized version of Parser::SkipUntil. 19372b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregorbool Parser::ConsumeAndStoreUntil(tok::TokenKind T1, tok::TokenKind T2, 19472b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor CachedTokens &Toks, 19572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor tok::TokenKind EarlyAbortIf, 19672b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor bool ConsumeFinalToken) { 1974cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // We always want this function to consume at least one token if the first 1984cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // token isn't T and if not at EOF. 1994cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis bool isFirstTokenConsumed = true; 2004cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis while (1) { 2014cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // If we found one of the tokens, stop and return true. 20272b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor if (Tok.is(T1) || Tok.is(T2)) { 20372b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor if (ConsumeFinalToken) { 20472b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor Toks.push_back(Tok); 20572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor ConsumeAnyToken(); 20672b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor } 2074cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis return true; 2084cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis } 2094cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 2103f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor // If we found the early-abort token, return. 2113f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor if (Tok.is(EarlyAbortIf)) 2123f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor return false; 2133f08d181f620e6bf4971c436fc9878f98a02bbe3Douglas Gregor 2144cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis switch (Tok.getKind()) { 2154cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::eof: 2164cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Ran out of tokens. 2174cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis return false; 2184cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 2194cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::l_paren: 2204cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Recursively consume properly-nested parens. 2214cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Toks.push_back(Tok); 2224cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeParen(); 22372b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor ConsumeAndStoreUntil(tok::r_paren, tok::unknown, Toks); 2244cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis break; 2254cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::l_square: 2264cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Recursively consume properly-nested square brackets. 2274cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Toks.push_back(Tok); 2284cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeBracket(); 22972b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor ConsumeAndStoreUntil(tok::r_square, tok::unknown, Toks); 2304cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis break; 2314cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::l_brace: 2324cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Recursively consume properly-nested braces. 2334cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Toks.push_back(Tok); 2344cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeBrace(); 23572b505b7904b3c9320a1312998800ba76e4f5841Douglas Gregor ConsumeAndStoreUntil(tok::r_brace, tok::unknown, Toks); 2364cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis break; 2374cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 2384cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Okay, we found a ']' or '}' or ')', which we think should be balanced. 2394cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // Since the user wasn't looking for this token (if they were, it would 2404cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // already be handled), this isn't balanced. If there is a LHS token at a 2414cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // higher level, we will assume that this matches the unbalanced token 2424cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // and return it. Otherwise, this is a spurious RHS token, which we skip. 2434cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::r_paren: 2444cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis if (ParenCount && !isFirstTokenConsumed) 2454cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis return false; // Matches something. 2464cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Toks.push_back(Tok); 2474cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeParen(); 2484cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis break; 2494cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::r_square: 2504cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis if (BracketCount && !isFirstTokenConsumed) 2514cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis return false; // Matches something. 2524cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Toks.push_back(Tok); 2534cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeBracket(); 2544cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis break; 2554cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::r_brace: 2564cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis if (BraceCount && !isFirstTokenConsumed) 2574cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis return false; // Matches something. 2584cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Toks.push_back(Tok); 2594cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeBrace(); 2604cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis break; 2614cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis 2624cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::string_literal: 2634cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis case tok::wide_string_literal: 2644cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Toks.push_back(Tok); 2654cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeStringToken(); 2664cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis break; 2674cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis default: 2684cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis // consume this token. 2694cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis Toks.push_back(Tok); 2704cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis ConsumeToken(); 2714cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis break; 2724cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis } 2734cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis isFirstTokenConsumed = false; 2744cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis } 2754cc18a4d5222e04bd568b1e3e4d86127dbbcdf3fArgyrios Kyrtzidis} 276