1 //===--- MacroExpansion.cpp - Top level Macro Expansion -------------------===// 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 top level handling of macro expasion for the 11 // preprocessor. 12 // 13 //===----------------------------------------------------------------------===// 14 15 #include "clang/Lex/Preprocessor.h" 16 #include "clang/Lex/MacroArgs.h" 17 #include "clang/Basic/FileManager.h" 18 #include "clang/Basic/SourceManager.h" 19 #include "clang/Basic/TargetInfo.h" 20 #include "clang/Lex/CodeCompletionHandler.h" 21 #include "clang/Lex/ExternalPreprocessorSource.h" 22 #include "clang/Lex/LexDiagnostic.h" 23 #include "clang/Lex/MacroInfo.h" 24 #include "llvm/ADT/STLExtras.h" 25 #include "llvm/ADT/SmallString.h" 26 #include "llvm/ADT/StringSwitch.h" 27 #include "llvm/Config/llvm-config.h" 28 #include "llvm/Support/ErrorHandling.h" 29 #include "llvm/Support/Format.h" 30 #include "llvm/Support/raw_ostream.h" 31 #include <cstdio> 32 #include <ctime> 33 using namespace clang; 34 35 MacroDirective * 36 Preprocessor::getMacroDirectiveHistory(const IdentifierInfo *II) const { 37 assert(II->hadMacroDefinition() && "Identifier has not been not a macro!"); 38 39 macro_iterator Pos = Macros.find(II); 40 assert(Pos != Macros.end() && "Identifier macro info is missing!"); 41 return Pos->second; 42 } 43 44 void Preprocessor::appendMacroDirective(IdentifierInfo *II, MacroDirective *MD){ 45 assert(MD && "MacroDirective should be non-zero!"); 46 assert(!MD->getPrevious() && "Already attached to a MacroDirective history."); 47 48 MacroDirective *&StoredMD = Macros[II]; 49 MD->setPrevious(StoredMD); 50 StoredMD = MD; 51 II->setHasMacroDefinition(MD->isDefined()); 52 bool isImportedMacro = isa<DefMacroDirective>(MD) && 53 cast<DefMacroDirective>(MD)->isImported(); 54 if (II->isFromAST() && !isImportedMacro) 55 II->setChangedSinceDeserialization(); 56 } 57 58 void Preprocessor::setLoadedMacroDirective(IdentifierInfo *II, 59 MacroDirective *MD) { 60 assert(II && MD); 61 MacroDirective *&StoredMD = Macros[II]; 62 assert(!StoredMD && 63 "the macro history was modified before initializing it from a pch"); 64 StoredMD = MD; 65 // Setup the identifier as having associated macro history. 66 II->setHasMacroDefinition(true); 67 if (!MD->isDefined()) 68 II->setHasMacroDefinition(false); 69 } 70 71 /// RegisterBuiltinMacro - Register the specified identifier in the identifier 72 /// table and mark it as a builtin macro to be expanded. 73 static IdentifierInfo *RegisterBuiltinMacro(Preprocessor &PP, const char *Name){ 74 // Get the identifier. 75 IdentifierInfo *Id = PP.getIdentifierInfo(Name); 76 77 // Mark it as being a macro that is builtin. 78 MacroInfo *MI = PP.AllocateMacroInfo(SourceLocation()); 79 MI->setIsBuiltinMacro(); 80 PP.appendDefMacroDirective(Id, MI); 81 return Id; 82 } 83 84 85 /// RegisterBuiltinMacros - Register builtin macros, such as __LINE__ with the 86 /// identifier table. 87 void Preprocessor::RegisterBuiltinMacros() { 88 Ident__LINE__ = RegisterBuiltinMacro(*this, "__LINE__"); 89 Ident__FILE__ = RegisterBuiltinMacro(*this, "__FILE__"); 90 Ident__DATE__ = RegisterBuiltinMacro(*this, "__DATE__"); 91 Ident__TIME__ = RegisterBuiltinMacro(*this, "__TIME__"); 92 Ident__COUNTER__ = RegisterBuiltinMacro(*this, "__COUNTER__"); 93 Ident_Pragma = RegisterBuiltinMacro(*this, "_Pragma"); 94 95 // GCC Extensions. 96 Ident__BASE_FILE__ = RegisterBuiltinMacro(*this, "__BASE_FILE__"); 97 Ident__INCLUDE_LEVEL__ = RegisterBuiltinMacro(*this, "__INCLUDE_LEVEL__"); 98 Ident__TIMESTAMP__ = RegisterBuiltinMacro(*this, "__TIMESTAMP__"); 99 100 // Clang Extensions. 101 Ident__has_feature = RegisterBuiltinMacro(*this, "__has_feature"); 102 Ident__has_extension = RegisterBuiltinMacro(*this, "__has_extension"); 103 Ident__has_builtin = RegisterBuiltinMacro(*this, "__has_builtin"); 104 Ident__has_attribute = RegisterBuiltinMacro(*this, "__has_attribute"); 105 Ident__has_include = RegisterBuiltinMacro(*this, "__has_include"); 106 Ident__has_include_next = RegisterBuiltinMacro(*this, "__has_include_next"); 107 Ident__has_warning = RegisterBuiltinMacro(*this, "__has_warning"); 108 109 // Modules. 110 if (LangOpts.Modules) { 111 Ident__building_module = RegisterBuiltinMacro(*this, "__building_module"); 112 113 // __MODULE__ 114 if (!LangOpts.CurrentModule.empty()) 115 Ident__MODULE__ = RegisterBuiltinMacro(*this, "__MODULE__"); 116 else 117 Ident__MODULE__ = 0; 118 } else { 119 Ident__building_module = 0; 120 Ident__MODULE__ = 0; 121 } 122 123 // Microsoft Extensions. 124 if (LangOpts.MicrosoftExt) 125 Ident__pragma = RegisterBuiltinMacro(*this, "__pragma"); 126 else 127 Ident__pragma = 0; 128 } 129 130 /// isTrivialSingleTokenExpansion - Return true if MI, which has a single token 131 /// in its expansion, currently expands to that token literally. 132 static bool isTrivialSingleTokenExpansion(const MacroInfo *MI, 133 const IdentifierInfo *MacroIdent, 134 Preprocessor &PP) { 135 IdentifierInfo *II = MI->getReplacementToken(0).getIdentifierInfo(); 136 137 // If the token isn't an identifier, it's always literally expanded. 138 if (II == 0) return true; 139 140 // If the information about this identifier is out of date, update it from 141 // the external source. 142 if (II->isOutOfDate()) 143 PP.getExternalSource()->updateOutOfDateIdentifier(*II); 144 145 // If the identifier is a macro, and if that macro is enabled, it may be 146 // expanded so it's not a trivial expansion. 147 if (II->hasMacroDefinition() && PP.getMacroInfo(II)->isEnabled() && 148 // Fast expanding "#define X X" is ok, because X would be disabled. 149 II != MacroIdent) 150 return false; 151 152 // If this is an object-like macro invocation, it is safe to trivially expand 153 // it. 154 if (MI->isObjectLike()) return true; 155 156 // If this is a function-like macro invocation, it's safe to trivially expand 157 // as long as the identifier is not a macro argument. 158 for (MacroInfo::arg_iterator I = MI->arg_begin(), E = MI->arg_end(); 159 I != E; ++I) 160 if (*I == II) 161 return false; // Identifier is a macro argument. 162 163 return true; 164 } 165 166 167 /// isNextPPTokenLParen - Determine whether the next preprocessor token to be 168 /// lexed is a '('. If so, consume the token and return true, if not, this 169 /// method should have no observable side-effect on the lexed tokens. 170 bool Preprocessor::isNextPPTokenLParen() { 171 // Do some quick tests for rejection cases. 172 unsigned Val; 173 if (CurLexer) 174 Val = CurLexer->isNextPPTokenLParen(); 175 else if (CurPTHLexer) 176 Val = CurPTHLexer->isNextPPTokenLParen(); 177 else 178 Val = CurTokenLexer->isNextTokenLParen(); 179 180 if (Val == 2) { 181 // We have run off the end. If it's a source file we don't 182 // examine enclosing ones (C99 5.1.1.2p4). Otherwise walk up the 183 // macro stack. 184 if (CurPPLexer) 185 return false; 186 for (unsigned i = IncludeMacroStack.size(); i != 0; --i) { 187 IncludeStackInfo &Entry = IncludeMacroStack[i-1]; 188 if (Entry.TheLexer) 189 Val = Entry.TheLexer->isNextPPTokenLParen(); 190 else if (Entry.ThePTHLexer) 191 Val = Entry.ThePTHLexer->isNextPPTokenLParen(); 192 else 193 Val = Entry.TheTokenLexer->isNextTokenLParen(); 194 195 if (Val != 2) 196 break; 197 198 // Ran off the end of a source file? 199 if (Entry.ThePPLexer) 200 return false; 201 } 202 } 203 204 // Okay, if we know that the token is a '(', lex it and return. Otherwise we 205 // have found something that isn't a '(' or we found the end of the 206 // translation unit. In either case, return false. 207 return Val == 1; 208 } 209 210 /// HandleMacroExpandedIdentifier - If an identifier token is read that is to be 211 /// expanded as a macro, handle it and return the next token as 'Identifier'. 212 bool Preprocessor::HandleMacroExpandedIdentifier(Token &Identifier, 213 MacroDirective *MD) { 214 MacroDirective::DefInfo Def = MD->getDefinition(); 215 assert(Def.isValid()); 216 MacroInfo *MI = Def.getMacroInfo(); 217 218 // If this is a macro expansion in the "#if !defined(x)" line for the file, 219 // then the macro could expand to different things in other contexts, we need 220 // to disable the optimization in this case. 221 if (CurPPLexer) CurPPLexer->MIOpt.ExpandedMacro(); 222 223 // If this is a builtin macro, like __LINE__ or _Pragma, handle it specially. 224 if (MI->isBuiltinMacro()) { 225 if (Callbacks) Callbacks->MacroExpands(Identifier, MD, 226 Identifier.getLocation(),/*Args=*/0); 227 ExpandBuiltinMacro(Identifier); 228 return false; 229 } 230 231 /// Args - If this is a function-like macro expansion, this contains, 232 /// for each macro argument, the list of tokens that were provided to the 233 /// invocation. 234 MacroArgs *Args = 0; 235 236 // Remember where the end of the expansion occurred. For an object-like 237 // macro, this is the identifier. For a function-like macro, this is the ')'. 238 SourceLocation ExpansionEnd = Identifier.getLocation(); 239 240 // If this is a function-like macro, read the arguments. 241 if (MI->isFunctionLike()) { 242 // C99 6.10.3p10: If the preprocessing token immediately after the macro 243 // name isn't a '(', this macro should not be expanded. 244 if (!isNextPPTokenLParen()) 245 return true; 246 247 // Remember that we are now parsing the arguments to a macro invocation. 248 // Preprocessor directives used inside macro arguments are not portable, and 249 // this enables the warning. 250 InMacroArgs = true; 251 Args = ReadFunctionLikeMacroArgs(Identifier, MI, ExpansionEnd); 252 253 // Finished parsing args. 254 InMacroArgs = false; 255 256 // If there was an error parsing the arguments, bail out. 257 if (Args == 0) return false; 258 259 ++NumFnMacroExpanded; 260 } else { 261 ++NumMacroExpanded; 262 } 263 264 // Notice that this macro has been used. 265 markMacroAsUsed(MI); 266 267 // Remember where the token is expanded. 268 SourceLocation ExpandLoc = Identifier.getLocation(); 269 SourceRange ExpansionRange(ExpandLoc, ExpansionEnd); 270 271 if (Callbacks) { 272 if (InMacroArgs) { 273 // We can have macro expansion inside a conditional directive while 274 // reading the function macro arguments. To ensure, in that case, that 275 // MacroExpands callbacks still happen in source order, queue this 276 // callback to have it happen after the function macro callback. 277 DelayedMacroExpandsCallbacks.push_back( 278 MacroExpandsInfo(Identifier, MD, ExpansionRange)); 279 } else { 280 Callbacks->MacroExpands(Identifier, MD, ExpansionRange, Args); 281 if (!DelayedMacroExpandsCallbacks.empty()) { 282 for (unsigned i=0, e = DelayedMacroExpandsCallbacks.size(); i!=e; ++i) { 283 MacroExpandsInfo &Info = DelayedMacroExpandsCallbacks[i]; 284 // FIXME: We lose macro args info with delayed callback. 285 Callbacks->MacroExpands(Info.Tok, Info.MD, Info.Range, /*Args=*/0); 286 } 287 DelayedMacroExpandsCallbacks.clear(); 288 } 289 } 290 } 291 292 // If the macro definition is ambiguous, complain. 293 if (Def.getDirective()->isAmbiguous()) { 294 Diag(Identifier, diag::warn_pp_ambiguous_macro) 295 << Identifier.getIdentifierInfo(); 296 Diag(MI->getDefinitionLoc(), diag::note_pp_ambiguous_macro_chosen) 297 << Identifier.getIdentifierInfo(); 298 for (MacroDirective::DefInfo PrevDef = Def.getPreviousDefinition(); 299 PrevDef && !PrevDef.isUndefined(); 300 PrevDef = PrevDef.getPreviousDefinition()) { 301 if (PrevDef.getDirective()->isAmbiguous()) { 302 Diag(PrevDef.getMacroInfo()->getDefinitionLoc(), 303 diag::note_pp_ambiguous_macro_other) 304 << Identifier.getIdentifierInfo(); 305 } 306 } 307 } 308 309 // If we started lexing a macro, enter the macro expansion body. 310 311 // If this macro expands to no tokens, don't bother to push it onto the 312 // expansion stack, only to take it right back off. 313 if (MI->getNumTokens() == 0) { 314 // No need for arg info. 315 if (Args) Args->destroy(*this); 316 317 // Ignore this macro use, just return the next token in the current 318 // buffer. 319 bool HadLeadingSpace = Identifier.hasLeadingSpace(); 320 bool IsAtStartOfLine = Identifier.isAtStartOfLine(); 321 322 Lex(Identifier); 323 324 // If the identifier isn't on some OTHER line, inherit the leading 325 // whitespace/first-on-a-line property of this token. This handles 326 // stuff like "! XX," -> "! ," and " XX," -> " ,", when XX is 327 // empty. 328 if (!Identifier.isAtStartOfLine()) { 329 if (IsAtStartOfLine) Identifier.setFlag(Token::StartOfLine); 330 if (HadLeadingSpace) Identifier.setFlag(Token::LeadingSpace); 331 } 332 Identifier.setFlag(Token::LeadingEmptyMacro); 333 ++NumFastMacroExpanded; 334 return false; 335 336 } else if (MI->getNumTokens() == 1 && 337 isTrivialSingleTokenExpansion(MI, Identifier.getIdentifierInfo(), 338 *this)) { 339 // Otherwise, if this macro expands into a single trivially-expanded 340 // token: expand it now. This handles common cases like 341 // "#define VAL 42". 342 343 // No need for arg info. 344 if (Args) Args->destroy(*this); 345 346 // Propagate the isAtStartOfLine/hasLeadingSpace markers of the macro 347 // identifier to the expanded token. 348 bool isAtStartOfLine = Identifier.isAtStartOfLine(); 349 bool hasLeadingSpace = Identifier.hasLeadingSpace(); 350 351 // Replace the result token. 352 Identifier = MI->getReplacementToken(0); 353 354 // Restore the StartOfLine/LeadingSpace markers. 355 Identifier.setFlagValue(Token::StartOfLine , isAtStartOfLine); 356 Identifier.setFlagValue(Token::LeadingSpace, hasLeadingSpace); 357 358 // Update the tokens location to include both its expansion and physical 359 // locations. 360 SourceLocation Loc = 361 SourceMgr.createExpansionLoc(Identifier.getLocation(), ExpandLoc, 362 ExpansionEnd,Identifier.getLength()); 363 Identifier.setLocation(Loc); 364 365 // If this is a disabled macro or #define X X, we must mark the result as 366 // unexpandable. 367 if (IdentifierInfo *NewII = Identifier.getIdentifierInfo()) { 368 if (MacroInfo *NewMI = getMacroInfo(NewII)) 369 if (!NewMI->isEnabled() || NewMI == MI) { 370 Identifier.setFlag(Token::DisableExpand); 371 // Don't warn for "#define X X" like "#define bool bool" from 372 // stdbool.h. 373 if (NewMI != MI || MI->isFunctionLike()) 374 Diag(Identifier, diag::pp_disabled_macro_expansion); 375 } 376 } 377 378 // Since this is not an identifier token, it can't be macro expanded, so 379 // we're done. 380 ++NumFastMacroExpanded; 381 return false; 382 } 383 384 // Start expanding the macro. 385 EnterMacro(Identifier, ExpansionEnd, MI, Args); 386 387 // Now that the macro is at the top of the include stack, ask the 388 // preprocessor to read the next token from it. 389 Lex(Identifier); 390 return false; 391 } 392 393 /// ReadFunctionLikeMacroArgs - After reading "MACRO" and knowing that the next 394 /// token is the '(' of the macro, this method is invoked to read all of the 395 /// actual arguments specified for the macro invocation. This returns null on 396 /// error. 397 MacroArgs *Preprocessor::ReadFunctionLikeMacroArgs(Token &MacroName, 398 MacroInfo *MI, 399 SourceLocation &MacroEnd) { 400 // The number of fixed arguments to parse. 401 unsigned NumFixedArgsLeft = MI->getNumArgs(); 402 bool isVariadic = MI->isVariadic(); 403 404 // Outer loop, while there are more arguments, keep reading them. 405 Token Tok; 406 407 // Read arguments as unexpanded tokens. This avoids issues, e.g., where 408 // an argument value in a macro could expand to ',' or '(' or ')'. 409 LexUnexpandedToken(Tok); 410 assert(Tok.is(tok::l_paren) && "Error computing l-paren-ness?"); 411 412 // ArgTokens - Build up a list of tokens that make up each argument. Each 413 // argument is separated by an EOF token. Use a SmallVector so we can avoid 414 // heap allocations in the common case. 415 SmallVector<Token, 64> ArgTokens; 416 bool ContainsCodeCompletionTok = false; 417 418 unsigned NumActuals = 0; 419 while (Tok.isNot(tok::r_paren)) { 420 if (ContainsCodeCompletionTok && (Tok.is(tok::eof) || Tok.is(tok::eod))) 421 break; 422 423 assert((Tok.is(tok::l_paren) || Tok.is(tok::comma)) && 424 "only expect argument separators here"); 425 426 unsigned ArgTokenStart = ArgTokens.size(); 427 SourceLocation ArgStartLoc = Tok.getLocation(); 428 429 // C99 6.10.3p11: Keep track of the number of l_parens we have seen. Note 430 // that we already consumed the first one. 431 unsigned NumParens = 0; 432 433 while (1) { 434 // Read arguments as unexpanded tokens. This avoids issues, e.g., where 435 // an argument value in a macro could expand to ',' or '(' or ')'. 436 LexUnexpandedToken(Tok); 437 438 if (Tok.is(tok::eof) || Tok.is(tok::eod)) { // "#if f(<eof>" & "#if f(\n" 439 if (!ContainsCodeCompletionTok) { 440 Diag(MacroName, diag::err_unterm_macro_invoc); 441 Diag(MI->getDefinitionLoc(), diag::note_macro_here) 442 << MacroName.getIdentifierInfo(); 443 // Do not lose the EOF/EOD. Return it to the client. 444 MacroName = Tok; 445 return 0; 446 } else { 447 // Do not lose the EOF/EOD. 448 Token *Toks = new Token[1]; 449 Toks[0] = Tok; 450 EnterTokenStream(Toks, 1, true, true); 451 break; 452 } 453 } else if (Tok.is(tok::r_paren)) { 454 // If we found the ) token, the macro arg list is done. 455 if (NumParens-- == 0) { 456 MacroEnd = Tok.getLocation(); 457 break; 458 } 459 } else if (Tok.is(tok::l_paren)) { 460 ++NumParens; 461 } else if (Tok.is(tok::comma) && NumParens == 0) { 462 // Comma ends this argument if there are more fixed arguments expected. 463 // However, if this is a variadic macro, and this is part of the 464 // variadic part, then the comma is just an argument token. 465 if (!isVariadic) break; 466 if (NumFixedArgsLeft > 1) 467 break; 468 } else if (Tok.is(tok::comment) && !KeepMacroComments) { 469 // If this is a comment token in the argument list and we're just in 470 // -C mode (not -CC mode), discard the comment. 471 continue; 472 } else if (Tok.getIdentifierInfo() != 0) { 473 // Reading macro arguments can cause macros that we are currently 474 // expanding from to be popped off the expansion stack. Doing so causes 475 // them to be reenabled for expansion. Here we record whether any 476 // identifiers we lex as macro arguments correspond to disabled macros. 477 // If so, we mark the token as noexpand. This is a subtle aspect of 478 // C99 6.10.3.4p2. 479 if (MacroInfo *MI = getMacroInfo(Tok.getIdentifierInfo())) 480 if (!MI->isEnabled()) 481 Tok.setFlag(Token::DisableExpand); 482 } else if (Tok.is(tok::code_completion)) { 483 ContainsCodeCompletionTok = true; 484 if (CodeComplete) 485 CodeComplete->CodeCompleteMacroArgument(MacroName.getIdentifierInfo(), 486 MI, NumActuals); 487 // Don't mark that we reached the code-completion point because the 488 // parser is going to handle the token and there will be another 489 // code-completion callback. 490 } 491 492 ArgTokens.push_back(Tok); 493 } 494 495 // If this was an empty argument list foo(), don't add this as an empty 496 // argument. 497 if (ArgTokens.empty() && Tok.getKind() == tok::r_paren) 498 break; 499 500 // If this is not a variadic macro, and too many args were specified, emit 501 // an error. 502 if (!isVariadic && NumFixedArgsLeft == 0) { 503 if (ArgTokens.size() != ArgTokenStart) 504 ArgStartLoc = ArgTokens[ArgTokenStart].getLocation(); 505 506 if (!ContainsCodeCompletionTok) { 507 // Emit the diagnostic at the macro name in case there is a missing ). 508 // Emitting it at the , could be far away from the macro name. 509 Diag(ArgStartLoc, diag::err_too_many_args_in_macro_invoc); 510 Diag(MI->getDefinitionLoc(), diag::note_macro_here) 511 << MacroName.getIdentifierInfo(); 512 return 0; 513 } 514 } 515 516 // Empty arguments are standard in C99 and C++0x, and are supported as an extension in 517 // other modes. 518 if (ArgTokens.size() == ArgTokenStart && !LangOpts.C99) 519 Diag(Tok, LangOpts.CPlusPlus11 ? 520 diag::warn_cxx98_compat_empty_fnmacro_arg : 521 diag::ext_empty_fnmacro_arg); 522 523 // Add a marker EOF token to the end of the token list for this argument. 524 Token EOFTok; 525 EOFTok.startToken(); 526 EOFTok.setKind(tok::eof); 527 EOFTok.setLocation(Tok.getLocation()); 528 EOFTok.setLength(0); 529 ArgTokens.push_back(EOFTok); 530 ++NumActuals; 531 if (!ContainsCodeCompletionTok || NumFixedArgsLeft != 0) { 532 assert(NumFixedArgsLeft != 0 && "Too many arguments parsed"); 533 --NumFixedArgsLeft; 534 } 535 } 536 537 // Okay, we either found the r_paren. Check to see if we parsed too few 538 // arguments. 539 unsigned MinArgsExpected = MI->getNumArgs(); 540 541 // See MacroArgs instance var for description of this. 542 bool isVarargsElided = false; 543 544 if (ContainsCodeCompletionTok) { 545 // Recover from not-fully-formed macro invocation during code-completion. 546 Token EOFTok; 547 EOFTok.startToken(); 548 EOFTok.setKind(tok::eof); 549 EOFTok.setLocation(Tok.getLocation()); 550 EOFTok.setLength(0); 551 for (; NumActuals < MinArgsExpected; ++NumActuals) 552 ArgTokens.push_back(EOFTok); 553 } 554 555 if (NumActuals < MinArgsExpected) { 556 // There are several cases where too few arguments is ok, handle them now. 557 if (NumActuals == 0 && MinArgsExpected == 1) { 558 // #define A(X) or #define A(...) ---> A() 559 560 // If there is exactly one argument, and that argument is missing, 561 // then we have an empty "()" argument empty list. This is fine, even if 562 // the macro expects one argument (the argument is just empty). 563 isVarargsElided = MI->isVariadic(); 564 } else if (MI->isVariadic() && 565 (NumActuals+1 == MinArgsExpected || // A(x, ...) -> A(X) 566 (NumActuals == 0 && MinArgsExpected == 2))) {// A(x,...) -> A() 567 // Varargs where the named vararg parameter is missing: OK as extension. 568 // #define A(x, ...) 569 // A("blah") 570 // 571 // If the macro contains the comma pasting extension, the diagnostic 572 // is suppressed; we know we'll get another diagnostic later. 573 if (!MI->hasCommaPasting()) { 574 Diag(Tok, diag::ext_missing_varargs_arg); 575 Diag(MI->getDefinitionLoc(), diag::note_macro_here) 576 << MacroName.getIdentifierInfo(); 577 } 578 579 // Remember this occurred, allowing us to elide the comma when used for 580 // cases like: 581 // #define A(x, foo...) blah(a, ## foo) 582 // #define B(x, ...) blah(a, ## __VA_ARGS__) 583 // #define C(...) blah(a, ## __VA_ARGS__) 584 // A(x) B(x) C() 585 isVarargsElided = true; 586 } else if (!ContainsCodeCompletionTok) { 587 // Otherwise, emit the error. 588 Diag(Tok, diag::err_too_few_args_in_macro_invoc); 589 Diag(MI->getDefinitionLoc(), diag::note_macro_here) 590 << MacroName.getIdentifierInfo(); 591 return 0; 592 } 593 594 // Add a marker EOF token to the end of the token list for this argument. 595 SourceLocation EndLoc = Tok.getLocation(); 596 Tok.startToken(); 597 Tok.setKind(tok::eof); 598 Tok.setLocation(EndLoc); 599 Tok.setLength(0); 600 ArgTokens.push_back(Tok); 601 602 // If we expect two arguments, add both as empty. 603 if (NumActuals == 0 && MinArgsExpected == 2) 604 ArgTokens.push_back(Tok); 605 606 } else if (NumActuals > MinArgsExpected && !MI->isVariadic() && 607 !ContainsCodeCompletionTok) { 608 // Emit the diagnostic at the macro name in case there is a missing ). 609 // Emitting it at the , could be far away from the macro name. 610 Diag(MacroName, diag::err_too_many_args_in_macro_invoc); 611 Diag(MI->getDefinitionLoc(), diag::note_macro_here) 612 << MacroName.getIdentifierInfo(); 613 return 0; 614 } 615 616 return MacroArgs::create(MI, ArgTokens, isVarargsElided, *this); 617 } 618 619 /// \brief Keeps macro expanded tokens for TokenLexers. 620 // 621 /// Works like a stack; a TokenLexer adds the macro expanded tokens that is 622 /// going to lex in the cache and when it finishes the tokens are removed 623 /// from the end of the cache. 624 Token *Preprocessor::cacheMacroExpandedTokens(TokenLexer *tokLexer, 625 ArrayRef<Token> tokens) { 626 assert(tokLexer); 627 if (tokens.empty()) 628 return 0; 629 630 size_t newIndex = MacroExpandedTokens.size(); 631 bool cacheNeedsToGrow = tokens.size() > 632 MacroExpandedTokens.capacity()-MacroExpandedTokens.size(); 633 MacroExpandedTokens.append(tokens.begin(), tokens.end()); 634 635 if (cacheNeedsToGrow) { 636 // Go through all the TokenLexers whose 'Tokens' pointer points in the 637 // buffer and update the pointers to the (potential) new buffer array. 638 for (unsigned i = 0, e = MacroExpandingLexersStack.size(); i != e; ++i) { 639 TokenLexer *prevLexer; 640 size_t tokIndex; 641 llvm::tie(prevLexer, tokIndex) = MacroExpandingLexersStack[i]; 642 prevLexer->Tokens = MacroExpandedTokens.data() + tokIndex; 643 } 644 } 645 646 MacroExpandingLexersStack.push_back(std::make_pair(tokLexer, newIndex)); 647 return MacroExpandedTokens.data() + newIndex; 648 } 649 650 void Preprocessor::removeCachedMacroExpandedTokensOfLastLexer() { 651 assert(!MacroExpandingLexersStack.empty()); 652 size_t tokIndex = MacroExpandingLexersStack.back().second; 653 assert(tokIndex < MacroExpandedTokens.size()); 654 // Pop the cached macro expanded tokens from the end. 655 MacroExpandedTokens.resize(tokIndex); 656 MacroExpandingLexersStack.pop_back(); 657 } 658 659 /// ComputeDATE_TIME - Compute the current time, enter it into the specified 660 /// scratch buffer, then return DATELoc/TIMELoc locations with the position of 661 /// the identifier tokens inserted. 662 static void ComputeDATE_TIME(SourceLocation &DATELoc, SourceLocation &TIMELoc, 663 Preprocessor &PP) { 664 time_t TT = time(0); 665 struct tm *TM = localtime(&TT); 666 667 static const char * const Months[] = { 668 "Jan","Feb","Mar","Apr","May","Jun","Jul","Aug","Sep","Oct","Nov","Dec" 669 }; 670 671 { 672 SmallString<32> TmpBuffer; 673 llvm::raw_svector_ostream TmpStream(TmpBuffer); 674 TmpStream << llvm::format("\"%s %2d %4d\"", Months[TM->tm_mon], 675 TM->tm_mday, TM->tm_year + 1900); 676 Token TmpTok; 677 TmpTok.startToken(); 678 PP.CreateString(TmpStream.str(), TmpTok); 679 DATELoc = TmpTok.getLocation(); 680 } 681 682 { 683 SmallString<32> TmpBuffer; 684 llvm::raw_svector_ostream TmpStream(TmpBuffer); 685 TmpStream << llvm::format("\"%02d:%02d:%02d\"", 686 TM->tm_hour, TM->tm_min, TM->tm_sec); 687 Token TmpTok; 688 TmpTok.startToken(); 689 PP.CreateString(TmpStream.str(), TmpTok); 690 TIMELoc = TmpTok.getLocation(); 691 } 692 } 693 694 695 /// HasFeature - Return true if we recognize and implement the feature 696 /// specified by the identifier as a standard language feature. 697 static bool HasFeature(const Preprocessor &PP, const IdentifierInfo *II) { 698 const LangOptions &LangOpts = PP.getLangOpts(); 699 StringRef Feature = II->getName(); 700 701 // Normalize the feature name, __foo__ becomes foo. 702 if (Feature.startswith("__") && Feature.endswith("__") && Feature.size() >= 4) 703 Feature = Feature.substr(2, Feature.size() - 4); 704 705 return llvm::StringSwitch<bool>(Feature) 706 .Case("address_sanitizer", LangOpts.Sanitize.Address) 707 .Case("attribute_analyzer_noreturn", true) 708 .Case("attribute_availability", true) 709 .Case("attribute_availability_with_message", true) 710 .Case("attribute_cf_returns_not_retained", true) 711 .Case("attribute_cf_returns_retained", true) 712 .Case("attribute_deprecated_with_message", true) 713 .Case("attribute_ext_vector_type", true) 714 .Case("attribute_ns_returns_not_retained", true) 715 .Case("attribute_ns_returns_retained", true) 716 .Case("attribute_ns_consumes_self", true) 717 .Case("attribute_ns_consumed", true) 718 .Case("attribute_cf_consumed", true) 719 .Case("attribute_objc_ivar_unused", true) 720 .Case("attribute_objc_method_family", true) 721 .Case("attribute_overloadable", true) 722 .Case("attribute_unavailable_with_message", true) 723 .Case("attribute_unused_on_fields", true) 724 .Case("blocks", LangOpts.Blocks) 725 .Case("cxx_exceptions", LangOpts.Exceptions) 726 .Case("cxx_rtti", LangOpts.RTTI) 727 .Case("enumerator_attributes", true) 728 .Case("memory_sanitizer", LangOpts.Sanitize.Memory) 729 .Case("thread_sanitizer", LangOpts.Sanitize.Thread) 730 // Objective-C features 731 .Case("objc_arr", LangOpts.ObjCAutoRefCount) // FIXME: REMOVE? 732 .Case("objc_arc", LangOpts.ObjCAutoRefCount) 733 .Case("objc_arc_weak", LangOpts.ObjCARCWeak) 734 .Case("objc_default_synthesize_properties", LangOpts.ObjC2) 735 .Case("objc_fixed_enum", LangOpts.ObjC2) 736 .Case("objc_instancetype", LangOpts.ObjC2) 737 .Case("objc_modules", LangOpts.ObjC2 && LangOpts.Modules) 738 .Case("objc_nonfragile_abi", LangOpts.ObjCRuntime.isNonFragile()) 739 .Case("objc_property_explicit_atomic", true) // Does clang support explicit "atomic" keyword? 740 .Case("objc_weak_class", LangOpts.ObjCRuntime.hasWeakClassImport()) 741 .Case("ownership_holds", true) 742 .Case("ownership_returns", true) 743 .Case("ownership_takes", true) 744 .Case("objc_bool", true) 745 .Case("objc_subscripting", LangOpts.ObjCRuntime.isNonFragile()) 746 .Case("objc_array_literals", LangOpts.ObjC2) 747 .Case("objc_dictionary_literals", LangOpts.ObjC2) 748 .Case("objc_boxed_expressions", LangOpts.ObjC2) 749 .Case("arc_cf_code_audited", true) 750 // C11 features 751 .Case("c_alignas", LangOpts.C11) 752 .Case("c_atomic", LangOpts.C11) 753 .Case("c_generic_selections", LangOpts.C11) 754 .Case("c_static_assert", LangOpts.C11) 755 .Case("c_thread_local", 756 LangOpts.C11 && PP.getTargetInfo().isTLSSupported()) 757 // C++11 features 758 .Case("cxx_access_control_sfinae", LangOpts.CPlusPlus11) 759 .Case("cxx_alias_templates", LangOpts.CPlusPlus11) 760 .Case("cxx_alignas", LangOpts.CPlusPlus11) 761 .Case("cxx_atomic", LangOpts.CPlusPlus11) 762 .Case("cxx_attributes", LangOpts.CPlusPlus11) 763 .Case("cxx_auto_type", LangOpts.CPlusPlus11) 764 .Case("cxx_constexpr", LangOpts.CPlusPlus11) 765 .Case("cxx_decltype", LangOpts.CPlusPlus11) 766 .Case("cxx_decltype_incomplete_return_types", LangOpts.CPlusPlus11) 767 .Case("cxx_default_function_template_args", LangOpts.CPlusPlus11) 768 .Case("cxx_defaulted_functions", LangOpts.CPlusPlus11) 769 .Case("cxx_delegating_constructors", LangOpts.CPlusPlus11) 770 .Case("cxx_deleted_functions", LangOpts.CPlusPlus11) 771 .Case("cxx_explicit_conversions", LangOpts.CPlusPlus11) 772 .Case("cxx_generalized_initializers", LangOpts.CPlusPlus11) 773 .Case("cxx_implicit_moves", LangOpts.CPlusPlus11) 774 .Case("cxx_inheriting_constructors", LangOpts.CPlusPlus11) 775 .Case("cxx_inline_namespaces", LangOpts.CPlusPlus11) 776 .Case("cxx_lambdas", LangOpts.CPlusPlus11) 777 .Case("cxx_local_type_template_args", LangOpts.CPlusPlus11) 778 .Case("cxx_nonstatic_member_init", LangOpts.CPlusPlus11) 779 .Case("cxx_noexcept", LangOpts.CPlusPlus11) 780 .Case("cxx_nullptr", LangOpts.CPlusPlus11) 781 .Case("cxx_override_control", LangOpts.CPlusPlus11) 782 .Case("cxx_range_for", LangOpts.CPlusPlus11) 783 .Case("cxx_raw_string_literals", LangOpts.CPlusPlus11) 784 .Case("cxx_reference_qualified_functions", LangOpts.CPlusPlus11) 785 .Case("cxx_rvalue_references", LangOpts.CPlusPlus11) 786 .Case("cxx_strong_enums", LangOpts.CPlusPlus11) 787 .Case("cxx_static_assert", LangOpts.CPlusPlus11) 788 .Case("cxx_thread_local", 789 LangOpts.CPlusPlus11 && PP.getTargetInfo().isTLSSupported()) 790 .Case("cxx_trailing_return", LangOpts.CPlusPlus11) 791 .Case("cxx_unicode_literals", LangOpts.CPlusPlus11) 792 .Case("cxx_unrestricted_unions", LangOpts.CPlusPlus11) 793 .Case("cxx_user_literals", LangOpts.CPlusPlus11) 794 .Case("cxx_variadic_templates", LangOpts.CPlusPlus11) 795 // C++1y features 796 .Case("cxx_binary_literals", LangOpts.CPlusPlus1y) 797 //.Case("cxx_contextual_conversions", LangOpts.CPlusPlus1y) 798 //.Case("cxx_generalized_capture", LangOpts.CPlusPlus1y) 799 //.Case("cxx_generic_lambda", LangOpts.CPlusPlus1y) 800 //.Case("cxx_relaxed_constexpr", LangOpts.CPlusPlus1y) 801 .Case("cxx_return_type_deduction", LangOpts.CPlusPlus1y) 802 //.Case("cxx_runtime_array", LangOpts.CPlusPlus1y) 803 .Case("cxx_aggregate_nsdmi", LangOpts.CPlusPlus1y) 804 //.Case("cxx_variable_templates", LangOpts.CPlusPlus1y) 805 // Type traits 806 .Case("has_nothrow_assign", LangOpts.CPlusPlus) 807 .Case("has_nothrow_copy", LangOpts.CPlusPlus) 808 .Case("has_nothrow_constructor", LangOpts.CPlusPlus) 809 .Case("has_trivial_assign", LangOpts.CPlusPlus) 810 .Case("has_trivial_copy", LangOpts.CPlusPlus) 811 .Case("has_trivial_constructor", LangOpts.CPlusPlus) 812 .Case("has_trivial_destructor", LangOpts.CPlusPlus) 813 .Case("has_virtual_destructor", LangOpts.CPlusPlus) 814 .Case("is_abstract", LangOpts.CPlusPlus) 815 .Case("is_base_of", LangOpts.CPlusPlus) 816 .Case("is_class", LangOpts.CPlusPlus) 817 .Case("is_convertible_to", LangOpts.CPlusPlus) 818 .Case("is_empty", LangOpts.CPlusPlus) 819 .Case("is_enum", LangOpts.CPlusPlus) 820 .Case("is_final", LangOpts.CPlusPlus) 821 .Case("is_literal", LangOpts.CPlusPlus) 822 .Case("is_standard_layout", LangOpts.CPlusPlus) 823 .Case("is_pod", LangOpts.CPlusPlus) 824 .Case("is_polymorphic", LangOpts.CPlusPlus) 825 .Case("is_trivial", LangOpts.CPlusPlus) 826 .Case("is_trivially_assignable", LangOpts.CPlusPlus) 827 .Case("is_trivially_constructible", LangOpts.CPlusPlus) 828 .Case("is_trivially_copyable", LangOpts.CPlusPlus) 829 .Case("is_union", LangOpts.CPlusPlus) 830 .Case("modules", LangOpts.Modules) 831 .Case("tls", PP.getTargetInfo().isTLSSupported()) 832 .Case("underlying_type", LangOpts.CPlusPlus) 833 .Default(false); 834 } 835 836 /// HasExtension - Return true if we recognize and implement the feature 837 /// specified by the identifier, either as an extension or a standard language 838 /// feature. 839 static bool HasExtension(const Preprocessor &PP, const IdentifierInfo *II) { 840 if (HasFeature(PP, II)) 841 return true; 842 843 // If the use of an extension results in an error diagnostic, extensions are 844 // effectively unavailable, so just return false here. 845 if (PP.getDiagnostics().getExtensionHandlingBehavior() == 846 DiagnosticsEngine::Ext_Error) 847 return false; 848 849 const LangOptions &LangOpts = PP.getLangOpts(); 850 StringRef Extension = II->getName(); 851 852 // Normalize the extension name, __foo__ becomes foo. 853 if (Extension.startswith("__") && Extension.endswith("__") && 854 Extension.size() >= 4) 855 Extension = Extension.substr(2, Extension.size() - 4); 856 857 // Because we inherit the feature list from HasFeature, this string switch 858 // must be less restrictive than HasFeature's. 859 return llvm::StringSwitch<bool>(Extension) 860 // C11 features supported by other languages as extensions. 861 .Case("c_alignas", true) 862 .Case("c_atomic", true) 863 .Case("c_generic_selections", true) 864 .Case("c_static_assert", true) 865 // C++11 features supported by other languages as extensions. 866 .Case("cxx_atomic", LangOpts.CPlusPlus) 867 .Case("cxx_deleted_functions", LangOpts.CPlusPlus) 868 .Case("cxx_explicit_conversions", LangOpts.CPlusPlus) 869 .Case("cxx_inline_namespaces", LangOpts.CPlusPlus) 870 .Case("cxx_local_type_template_args", LangOpts.CPlusPlus) 871 .Case("cxx_nonstatic_member_init", LangOpts.CPlusPlus) 872 .Case("cxx_override_control", LangOpts.CPlusPlus) 873 .Case("cxx_range_for", LangOpts.CPlusPlus) 874 .Case("cxx_reference_qualified_functions", LangOpts.CPlusPlus) 875 .Case("cxx_rvalue_references", LangOpts.CPlusPlus) 876 // C++1y features supported by other languages as extensions. 877 .Case("cxx_binary_literals", true) 878 .Default(false); 879 } 880 881 /// HasAttribute - Return true if we recognize and implement the attribute 882 /// specified by the given identifier. 883 static bool HasAttribute(const IdentifierInfo *II) { 884 StringRef Name = II->getName(); 885 // Normalize the attribute name, __foo__ becomes foo. 886 if (Name.startswith("__") && Name.endswith("__") && Name.size() >= 4) 887 Name = Name.substr(2, Name.size() - 4); 888 889 // FIXME: Do we need to handle namespaces here? 890 return llvm::StringSwitch<bool>(Name) 891 #include "clang/Lex/AttrSpellings.inc" 892 .Default(false); 893 } 894 895 /// EvaluateHasIncludeCommon - Process a '__has_include("path")' 896 /// or '__has_include_next("path")' expression. 897 /// Returns true if successful. 898 static bool EvaluateHasIncludeCommon(Token &Tok, 899 IdentifierInfo *II, Preprocessor &PP, 900 const DirectoryLookup *LookupFrom) { 901 // Save the location of the current token. If a '(' is later found, use 902 // that location. If not, use the end of this location instead. 903 SourceLocation LParenLoc = Tok.getLocation(); 904 905 // These expressions are only allowed within a preprocessor directive. 906 if (!PP.isParsingIfOrElifDirective()) { 907 PP.Diag(LParenLoc, diag::err_pp_directive_required) << II->getName(); 908 return false; 909 } 910 911 // Get '('. 912 PP.LexNonComment(Tok); 913 914 // Ensure we have a '('. 915 if (Tok.isNot(tok::l_paren)) { 916 // No '(', use end of last token. 917 LParenLoc = PP.getLocForEndOfToken(LParenLoc); 918 PP.Diag(LParenLoc, diag::err_pp_missing_lparen) << II->getName(); 919 // If the next token looks like a filename or the start of one, 920 // assume it is and process it as such. 921 if (!Tok.is(tok::angle_string_literal) && !Tok.is(tok::string_literal) && 922 !Tok.is(tok::less)) 923 return false; 924 } else { 925 // Save '(' location for possible missing ')' message. 926 LParenLoc = Tok.getLocation(); 927 928 if (PP.getCurrentLexer()) { 929 // Get the file name. 930 PP.getCurrentLexer()->LexIncludeFilename(Tok); 931 } else { 932 // We're in a macro, so we can't use LexIncludeFilename; just 933 // grab the next token. 934 PP.Lex(Tok); 935 } 936 } 937 938 // Reserve a buffer to get the spelling. 939 SmallString<128> FilenameBuffer; 940 StringRef Filename; 941 SourceLocation EndLoc; 942 943 switch (Tok.getKind()) { 944 case tok::eod: 945 // If the token kind is EOD, the error has already been diagnosed. 946 return false; 947 948 case tok::angle_string_literal: 949 case tok::string_literal: { 950 bool Invalid = false; 951 Filename = PP.getSpelling(Tok, FilenameBuffer, &Invalid); 952 if (Invalid) 953 return false; 954 break; 955 } 956 957 case tok::less: 958 // This could be a <foo/bar.h> file coming from a macro expansion. In this 959 // case, glue the tokens together into FilenameBuffer and interpret those. 960 FilenameBuffer.push_back('<'); 961 if (PP.ConcatenateIncludeName(FilenameBuffer, EndLoc)) { 962 // Let the caller know a <eod> was found by changing the Token kind. 963 Tok.setKind(tok::eod); 964 return false; // Found <eod> but no ">"? Diagnostic already emitted. 965 } 966 Filename = FilenameBuffer.str(); 967 break; 968 default: 969 PP.Diag(Tok.getLocation(), diag::err_pp_expects_filename); 970 return false; 971 } 972 973 SourceLocation FilenameLoc = Tok.getLocation(); 974 975 // Get ')'. 976 PP.LexNonComment(Tok); 977 978 // Ensure we have a trailing ). 979 if (Tok.isNot(tok::r_paren)) { 980 PP.Diag(PP.getLocForEndOfToken(FilenameLoc), diag::err_pp_missing_rparen) 981 << II->getName(); 982 PP.Diag(LParenLoc, diag::note_matching) << "("; 983 return false; 984 } 985 986 bool isAngled = PP.GetIncludeFilenameSpelling(Tok.getLocation(), Filename); 987 // If GetIncludeFilenameSpelling set the start ptr to null, there was an 988 // error. 989 if (Filename.empty()) 990 return false; 991 992 // Search include directories. 993 const DirectoryLookup *CurDir; 994 const FileEntry *File = 995 PP.LookupFile(FilenameLoc, Filename, isAngled, LookupFrom, CurDir, NULL, 996 NULL, NULL); 997 998 // Get the result value. A result of true means the file exists. 999 return File != 0; 1000 } 1001 1002 /// EvaluateHasInclude - Process a '__has_include("path")' expression. 1003 /// Returns true if successful. 1004 static bool EvaluateHasInclude(Token &Tok, IdentifierInfo *II, 1005 Preprocessor &PP) { 1006 return EvaluateHasIncludeCommon(Tok, II, PP, NULL); 1007 } 1008 1009 /// EvaluateHasIncludeNext - Process '__has_include_next("path")' expression. 1010 /// Returns true if successful. 1011 static bool EvaluateHasIncludeNext(Token &Tok, 1012 IdentifierInfo *II, Preprocessor &PP) { 1013 // __has_include_next is like __has_include, except that we start 1014 // searching after the current found directory. If we can't do this, 1015 // issue a diagnostic. 1016 const DirectoryLookup *Lookup = PP.GetCurDirLookup(); 1017 if (PP.isInPrimaryFile()) { 1018 Lookup = 0; 1019 PP.Diag(Tok, diag::pp_include_next_in_primary); 1020 } else if (Lookup == 0) { 1021 PP.Diag(Tok, diag::pp_include_next_absolute_path); 1022 } else { 1023 // Start looking up in the next directory. 1024 ++Lookup; 1025 } 1026 1027 return EvaluateHasIncludeCommon(Tok, II, PP, Lookup); 1028 } 1029 1030 /// \brief Process __building_module(identifier) expression. 1031 /// \returns true if we are building the named module, false otherwise. 1032 static bool EvaluateBuildingModule(Token &Tok, 1033 IdentifierInfo *II, Preprocessor &PP) { 1034 // Get '('. 1035 PP.LexNonComment(Tok); 1036 1037 // Ensure we have a '('. 1038 if (Tok.isNot(tok::l_paren)) { 1039 PP.Diag(Tok.getLocation(), diag::err_pp_missing_lparen) << II->getName(); 1040 return false; 1041 } 1042 1043 // Save '(' location for possible missing ')' message. 1044 SourceLocation LParenLoc = Tok.getLocation(); 1045 1046 // Get the module name. 1047 PP.LexNonComment(Tok); 1048 1049 // Ensure that we have an identifier. 1050 if (Tok.isNot(tok::identifier)) { 1051 PP.Diag(Tok.getLocation(), diag::err_expected_id_building_module); 1052 return false; 1053 } 1054 1055 bool Result 1056 = Tok.getIdentifierInfo()->getName() == PP.getLangOpts().CurrentModule; 1057 1058 // Get ')'. 1059 PP.LexNonComment(Tok); 1060 1061 // Ensure we have a trailing ). 1062 if (Tok.isNot(tok::r_paren)) { 1063 PP.Diag(Tok.getLocation(), diag::err_pp_missing_rparen) << II->getName(); 1064 PP.Diag(LParenLoc, diag::note_matching) << "("; 1065 return false; 1066 } 1067 1068 return Result; 1069 } 1070 1071 /// ExpandBuiltinMacro - If an identifier token is read that is to be expanded 1072 /// as a builtin macro, handle it and return the next token as 'Tok'. 1073 void Preprocessor::ExpandBuiltinMacro(Token &Tok) { 1074 // Figure out which token this is. 1075 IdentifierInfo *II = Tok.getIdentifierInfo(); 1076 assert(II && "Can't be a macro without id info!"); 1077 1078 // If this is an _Pragma or Microsoft __pragma directive, expand it, 1079 // invoke the pragma handler, then lex the token after it. 1080 if (II == Ident_Pragma) 1081 return Handle_Pragma(Tok); 1082 else if (II == Ident__pragma) // in non-MS mode this is null 1083 return HandleMicrosoft__pragma(Tok); 1084 1085 ++NumBuiltinMacroExpanded; 1086 1087 SmallString<128> TmpBuffer; 1088 llvm::raw_svector_ostream OS(TmpBuffer); 1089 1090 // Set up the return result. 1091 Tok.setIdentifierInfo(0); 1092 Tok.clearFlag(Token::NeedsCleaning); 1093 1094 if (II == Ident__LINE__) { 1095 // C99 6.10.8: "__LINE__: The presumed line number (within the current 1096 // source file) of the current source line (an integer constant)". This can 1097 // be affected by #line. 1098 SourceLocation Loc = Tok.getLocation(); 1099 1100 // Advance to the location of the first _, this might not be the first byte 1101 // of the token if it starts with an escaped newline. 1102 Loc = AdvanceToTokenCharacter(Loc, 0); 1103 1104 // One wrinkle here is that GCC expands __LINE__ to location of the *end* of 1105 // a macro expansion. This doesn't matter for object-like macros, but 1106 // can matter for a function-like macro that expands to contain __LINE__. 1107 // Skip down through expansion points until we find a file loc for the 1108 // end of the expansion history. 1109 Loc = SourceMgr.getExpansionRange(Loc).second; 1110 PresumedLoc PLoc = SourceMgr.getPresumedLoc(Loc); 1111 1112 // __LINE__ expands to a simple numeric value. 1113 OS << (PLoc.isValid()? PLoc.getLine() : 1); 1114 Tok.setKind(tok::numeric_constant); 1115 } else if (II == Ident__FILE__ || II == Ident__BASE_FILE__) { 1116 // C99 6.10.8: "__FILE__: The presumed name of the current source file (a 1117 // character string literal)". This can be affected by #line. 1118 PresumedLoc PLoc = SourceMgr.getPresumedLoc(Tok.getLocation()); 1119 1120 // __BASE_FILE__ is a GNU extension that returns the top of the presumed 1121 // #include stack instead of the current file. 1122 if (II == Ident__BASE_FILE__ && PLoc.isValid()) { 1123 SourceLocation NextLoc = PLoc.getIncludeLoc(); 1124 while (NextLoc.isValid()) { 1125 PLoc = SourceMgr.getPresumedLoc(NextLoc); 1126 if (PLoc.isInvalid()) 1127 break; 1128 1129 NextLoc = PLoc.getIncludeLoc(); 1130 } 1131 } 1132 1133 // Escape this filename. Turn '\' -> '\\' '"' -> '\"' 1134 SmallString<128> FN; 1135 if (PLoc.isValid()) { 1136 FN += PLoc.getFilename(); 1137 Lexer::Stringify(FN); 1138 OS << '"' << FN.str() << '"'; 1139 } 1140 Tok.setKind(tok::string_literal); 1141 } else if (II == Ident__DATE__) { 1142 if (!DATELoc.isValid()) 1143 ComputeDATE_TIME(DATELoc, TIMELoc, *this); 1144 Tok.setKind(tok::string_literal); 1145 Tok.setLength(strlen("\"Mmm dd yyyy\"")); 1146 Tok.setLocation(SourceMgr.createExpansionLoc(DATELoc, Tok.getLocation(), 1147 Tok.getLocation(), 1148 Tok.getLength())); 1149 return; 1150 } else if (II == Ident__TIME__) { 1151 if (!TIMELoc.isValid()) 1152 ComputeDATE_TIME(DATELoc, TIMELoc, *this); 1153 Tok.setKind(tok::string_literal); 1154 Tok.setLength(strlen("\"hh:mm:ss\"")); 1155 Tok.setLocation(SourceMgr.createExpansionLoc(TIMELoc, Tok.getLocation(), 1156 Tok.getLocation(), 1157 Tok.getLength())); 1158 return; 1159 } else if (II == Ident__INCLUDE_LEVEL__) { 1160 // Compute the presumed include depth of this token. This can be affected 1161 // by GNU line markers. 1162 unsigned Depth = 0; 1163 1164 PresumedLoc PLoc = SourceMgr.getPresumedLoc(Tok.getLocation()); 1165 if (PLoc.isValid()) { 1166 PLoc = SourceMgr.getPresumedLoc(PLoc.getIncludeLoc()); 1167 for (; PLoc.isValid(); ++Depth) 1168 PLoc = SourceMgr.getPresumedLoc(PLoc.getIncludeLoc()); 1169 } 1170 1171 // __INCLUDE_LEVEL__ expands to a simple numeric value. 1172 OS << Depth; 1173 Tok.setKind(tok::numeric_constant); 1174 } else if (II == Ident__TIMESTAMP__) { 1175 // MSVC, ICC, GCC, VisualAge C++ extension. The generated string should be 1176 // of the form "Ddd Mmm dd hh::mm::ss yyyy", which is returned by asctime. 1177 1178 // Get the file that we are lexing out of. If we're currently lexing from 1179 // a macro, dig into the include stack. 1180 const FileEntry *CurFile = 0; 1181 PreprocessorLexer *TheLexer = getCurrentFileLexer(); 1182 1183 if (TheLexer) 1184 CurFile = SourceMgr.getFileEntryForID(TheLexer->getFileID()); 1185 1186 const char *Result; 1187 if (CurFile) { 1188 time_t TT = CurFile->getModificationTime(); 1189 struct tm *TM = localtime(&TT); 1190 Result = asctime(TM); 1191 } else { 1192 Result = "??? ??? ?? ??:??:?? ????\n"; 1193 } 1194 // Surround the string with " and strip the trailing newline. 1195 OS << '"' << StringRef(Result, strlen(Result)-1) << '"'; 1196 Tok.setKind(tok::string_literal); 1197 } else if (II == Ident__COUNTER__) { 1198 // __COUNTER__ expands to a simple numeric value. 1199 OS << CounterValue++; 1200 Tok.setKind(tok::numeric_constant); 1201 } else if (II == Ident__has_feature || 1202 II == Ident__has_extension || 1203 II == Ident__has_builtin || 1204 II == Ident__has_attribute) { 1205 // The argument to these builtins should be a parenthesized identifier. 1206 SourceLocation StartLoc = Tok.getLocation(); 1207 1208 bool IsValid = false; 1209 IdentifierInfo *FeatureII = 0; 1210 1211 // Read the '('. 1212 LexUnexpandedToken(Tok); 1213 if (Tok.is(tok::l_paren)) { 1214 // Read the identifier 1215 LexUnexpandedToken(Tok); 1216 if (Tok.is(tok::identifier) || Tok.is(tok::kw_const)) { 1217 FeatureII = Tok.getIdentifierInfo(); 1218 1219 // Read the ')'. 1220 LexUnexpandedToken(Tok); 1221 if (Tok.is(tok::r_paren)) 1222 IsValid = true; 1223 } 1224 } 1225 1226 bool Value = false; 1227 if (!IsValid) 1228 Diag(StartLoc, diag::err_feature_check_malformed); 1229 else if (II == Ident__has_builtin) { 1230 // Check for a builtin is trivial. 1231 Value = FeatureII->getBuiltinID() != 0; 1232 } else if (II == Ident__has_attribute) 1233 Value = HasAttribute(FeatureII); 1234 else if (II == Ident__has_extension) 1235 Value = HasExtension(*this, FeatureII); 1236 else { 1237 assert(II == Ident__has_feature && "Must be feature check"); 1238 Value = HasFeature(*this, FeatureII); 1239 } 1240 1241 OS << (int)Value; 1242 if (IsValid) 1243 Tok.setKind(tok::numeric_constant); 1244 } else if (II == Ident__has_include || 1245 II == Ident__has_include_next) { 1246 // The argument to these two builtins should be a parenthesized 1247 // file name string literal using angle brackets (<>) or 1248 // double-quotes (""). 1249 bool Value; 1250 if (II == Ident__has_include) 1251 Value = EvaluateHasInclude(Tok, II, *this); 1252 else 1253 Value = EvaluateHasIncludeNext(Tok, II, *this); 1254 OS << (int)Value; 1255 if (Tok.is(tok::r_paren)) 1256 Tok.setKind(tok::numeric_constant); 1257 } else if (II == Ident__has_warning) { 1258 // The argument should be a parenthesized string literal. 1259 // The argument to these builtins should be a parenthesized identifier. 1260 SourceLocation StartLoc = Tok.getLocation(); 1261 bool IsValid = false; 1262 bool Value = false; 1263 // Read the '('. 1264 LexUnexpandedToken(Tok); 1265 do { 1266 if (Tok.isNot(tok::l_paren)) { 1267 Diag(StartLoc, diag::err_warning_check_malformed); 1268 break; 1269 } 1270 1271 LexUnexpandedToken(Tok); 1272 std::string WarningName; 1273 SourceLocation StrStartLoc = Tok.getLocation(); 1274 if (!FinishLexStringLiteral(Tok, WarningName, "'__has_warning'", 1275 /*MacroExpansion=*/false)) { 1276 // Eat tokens until ')'. 1277 while (Tok.isNot(tok::r_paren) && Tok.isNot(tok::eod) && 1278 Tok.isNot(tok::eof)) 1279 LexUnexpandedToken(Tok); 1280 break; 1281 } 1282 1283 // Is the end a ')'? 1284 if (!(IsValid = Tok.is(tok::r_paren))) { 1285 Diag(StartLoc, diag::err_warning_check_malformed); 1286 break; 1287 } 1288 1289 if (WarningName.size() < 3 || WarningName[0] != '-' || 1290 WarningName[1] != 'W') { 1291 Diag(StrStartLoc, diag::warn_has_warning_invalid_option); 1292 break; 1293 } 1294 1295 // Finally, check if the warning flags maps to a diagnostic group. 1296 // We construct a SmallVector here to talk to getDiagnosticIDs(). 1297 // Although we don't use the result, this isn't a hot path, and not 1298 // worth special casing. 1299 SmallVector<diag::kind, 10> Diags; 1300 Value = !getDiagnostics().getDiagnosticIDs()-> 1301 getDiagnosticsInGroup(WarningName.substr(2), Diags); 1302 } while (false); 1303 1304 OS << (int)Value; 1305 if (IsValid) 1306 Tok.setKind(tok::numeric_constant); 1307 } else if (II == Ident__building_module) { 1308 // The argument to this builtin should be an identifier. The 1309 // builtin evaluates to 1 when that identifier names the module we are 1310 // currently building. 1311 OS << (int)EvaluateBuildingModule(Tok, II, *this); 1312 Tok.setKind(tok::numeric_constant); 1313 } else if (II == Ident__MODULE__) { 1314 // The current module as an identifier. 1315 OS << getLangOpts().CurrentModule; 1316 IdentifierInfo *ModuleII = getIdentifierInfo(getLangOpts().CurrentModule); 1317 Tok.setIdentifierInfo(ModuleII); 1318 Tok.setKind(ModuleII->getTokenID()); 1319 } else { 1320 llvm_unreachable("Unknown identifier!"); 1321 } 1322 CreateString(OS.str(), Tok, Tok.getLocation(), Tok.getLocation()); 1323 } 1324 1325 void Preprocessor::markMacroAsUsed(MacroInfo *MI) { 1326 // If the 'used' status changed, and the macro requires 'unused' warning, 1327 // remove its SourceLocation from the warn-for-unused-macro locations. 1328 if (MI->isWarnIfUnused() && !MI->isUsed()) 1329 WarnUnusedMacroLocs.erase(MI->getDefinitionLoc()); 1330 MI->setIsUsed(true); 1331 } 1332