1 //===--- Preprocess.cpp - C Language Family Preprocessor Implementation ---===// 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 Preprocessor interface. 11 // 12 //===----------------------------------------------------------------------===// 13 // 14 // Options to support: 15 // -H - Print the name of each header file used. 16 // -d[DNI] - Dump various things. 17 // -fworking-directory - #line's with preprocessor's working dir. 18 // -fpreprocessed 19 // -dependency-file,-M,-MM,-MF,-MG,-MP,-MT,-MQ,-MD,-MMD 20 // -W* 21 // -w 22 // 23 // Messages to emit: 24 // "Multiple include guards may be useful for:\n" 25 // 26 //===----------------------------------------------------------------------===// 27 28 #include "clang/Lex/Preprocessor.h" 29 #include "clang/Basic/FileManager.h" 30 #include "clang/Basic/FileSystemStatCache.h" 31 #include "clang/Basic/SourceManager.h" 32 #include "clang/Basic/TargetInfo.h" 33 #include "clang/Lex/CodeCompletionHandler.h" 34 #include "clang/Lex/ExternalPreprocessorSource.h" 35 #include "clang/Lex/HeaderSearch.h" 36 #include "clang/Lex/LexDiagnostic.h" 37 #include "clang/Lex/LiteralSupport.h" 38 #include "clang/Lex/MacroArgs.h" 39 #include "clang/Lex/MacroInfo.h" 40 #include "clang/Lex/ModuleLoader.h" 41 #include "clang/Lex/Pragma.h" 42 #include "clang/Lex/PreprocessingRecord.h" 43 #include "clang/Lex/PreprocessorOptions.h" 44 #include "clang/Lex/ScratchBuffer.h" 45 #include "llvm/ADT/APFloat.h" 46 #include "llvm/ADT/STLExtras.h" 47 #include "llvm/ADT/SmallString.h" 48 #include "llvm/ADT/StringExtras.h" 49 #include "llvm/Support/Capacity.h" 50 #include "llvm/Support/ConvertUTF.h" 51 #include "llvm/Support/MemoryBuffer.h" 52 #include "llvm/Support/raw_ostream.h" 53 using namespace clang; 54 55 //===----------------------------------------------------------------------===// 56 ExternalPreprocessorSource::~ExternalPreprocessorSource() { } 57 58 Preprocessor::Preprocessor(IntrusiveRefCntPtr<PreprocessorOptions> PPOpts, 59 DiagnosticsEngine &diags, LangOptions &opts, 60 SourceManager &SM, HeaderSearch &Headers, 61 ModuleLoader &TheModuleLoader, 62 IdentifierInfoLookup *IILookup, bool OwnsHeaders, 63 TranslationUnitKind TUKind) 64 : PPOpts(PPOpts), Diags(&diags), LangOpts(opts), Target(nullptr), 65 FileMgr(Headers.getFileMgr()), SourceMgr(SM), 66 ScratchBuf(new ScratchBuffer(SourceMgr)),HeaderInfo(Headers), 67 TheModuleLoader(TheModuleLoader), ExternalSource(nullptr), 68 Identifiers(opts, IILookup), 69 PragmaHandlers(new PragmaNamespace(StringRef())), 70 IncrementalProcessing(false), TUKind(TUKind), 71 CodeComplete(nullptr), CodeCompletionFile(nullptr), 72 CodeCompletionOffset(0), LastTokenWasAt(false), 73 ModuleImportExpectsIdentifier(false), CodeCompletionReached(0), 74 SkipMainFilePreamble(0, true), CurPPLexer(nullptr), 75 CurDirLookup(nullptr), CurLexerKind(CLK_Lexer), CurSubmodule(nullptr), 76 Callbacks(nullptr), MacroArgCache(nullptr), Record(nullptr), 77 MIChainHead(nullptr), DeserialMIChainHead(nullptr) { 78 OwnsHeaderSearch = OwnsHeaders; 79 80 CounterValue = 0; // __COUNTER__ starts at 0. 81 82 // Clear stats. 83 NumDirectives = NumDefined = NumUndefined = NumPragma = 0; 84 NumIf = NumElse = NumEndif = 0; 85 NumEnteredSourceFiles = 0; 86 NumMacroExpanded = NumFnMacroExpanded = NumBuiltinMacroExpanded = 0; 87 NumFastMacroExpanded = NumTokenPaste = NumFastTokenPaste = 0; 88 MaxIncludeStackDepth = 0; 89 NumSkipped = 0; 90 91 // Default to discarding comments. 92 KeepComments = false; 93 KeepMacroComments = false; 94 SuppressIncludeNotFoundError = false; 95 96 // Macro expansion is enabled. 97 DisableMacroExpansion = false; 98 MacroExpansionInDirectivesOverride = false; 99 InMacroArgs = false; 100 InMacroArgPreExpansion = false; 101 NumCachedTokenLexers = 0; 102 PragmasEnabled = true; 103 ParsingIfOrElifDirective = false; 104 PreprocessedOutput = false; 105 106 CachedLexPos = 0; 107 108 // We haven't read anything from the external source. 109 ReadMacrosFromExternalSource = false; 110 111 // "Poison" __VA_ARGS__, which can only appear in the expansion of a macro. 112 // This gets unpoisoned where it is allowed. 113 (Ident__VA_ARGS__ = getIdentifierInfo("__VA_ARGS__"))->setIsPoisoned(); 114 SetPoisonReason(Ident__VA_ARGS__,diag::ext_pp_bad_vaargs_use); 115 116 // Initialize the pragma handlers. 117 RegisterBuiltinPragmas(); 118 119 // Initialize builtin macros like __LINE__ and friends. 120 RegisterBuiltinMacros(); 121 122 if(LangOpts.Borland) { 123 Ident__exception_info = getIdentifierInfo("_exception_info"); 124 Ident___exception_info = getIdentifierInfo("__exception_info"); 125 Ident_GetExceptionInfo = getIdentifierInfo("GetExceptionInformation"); 126 Ident__exception_code = getIdentifierInfo("_exception_code"); 127 Ident___exception_code = getIdentifierInfo("__exception_code"); 128 Ident_GetExceptionCode = getIdentifierInfo("GetExceptionCode"); 129 Ident__abnormal_termination = getIdentifierInfo("_abnormal_termination"); 130 Ident___abnormal_termination = getIdentifierInfo("__abnormal_termination"); 131 Ident_AbnormalTermination = getIdentifierInfo("AbnormalTermination"); 132 } else { 133 Ident__exception_info = Ident__exception_code = nullptr; 134 Ident__abnormal_termination = Ident___exception_info = nullptr; 135 Ident___exception_code = Ident___abnormal_termination = nullptr; 136 Ident_GetExceptionInfo = Ident_GetExceptionCode = nullptr; 137 Ident_AbnormalTermination = nullptr; 138 } 139 } 140 141 Preprocessor::~Preprocessor() { 142 assert(BacktrackPositions.empty() && "EnableBacktrack/Backtrack imbalance!"); 143 144 IncludeMacroStack.clear(); 145 146 // Destroy any macro definitions. 147 while (MacroInfoChain *I = MIChainHead) { 148 MIChainHead = I->Next; 149 I->~MacroInfoChain(); 150 } 151 152 // Free any cached macro expanders. 153 // This populates MacroArgCache, so all TokenLexers need to be destroyed 154 // before the code below that frees up the MacroArgCache list. 155 std::fill(TokenLexerCache, TokenLexerCache + NumCachedTokenLexers, nullptr); 156 CurTokenLexer.reset(); 157 158 while (DeserializedMacroInfoChain *I = DeserialMIChainHead) { 159 DeserialMIChainHead = I->Next; 160 I->~DeserializedMacroInfoChain(); 161 } 162 163 // Free any cached MacroArgs. 164 for (MacroArgs *ArgList = MacroArgCache; ArgList;) 165 ArgList = ArgList->deallocate(); 166 167 // Delete the header search info, if we own it. 168 if (OwnsHeaderSearch) 169 delete &HeaderInfo; 170 } 171 172 void Preprocessor::Initialize(const TargetInfo &Target) { 173 assert((!this->Target || this->Target == &Target) && 174 "Invalid override of target information"); 175 this->Target = &Target; 176 177 // Initialize information about built-ins. 178 BuiltinInfo.InitializeTarget(Target); 179 HeaderInfo.setTarget(Target); 180 } 181 182 void Preprocessor::InitializeForModelFile() { 183 NumEnteredSourceFiles = 0; 184 185 // Reset pragmas 186 PragmaHandlersBackup = std::move(PragmaHandlers); 187 PragmaHandlers = llvm::make_unique<PragmaNamespace>(StringRef()); 188 RegisterBuiltinPragmas(); 189 190 // Reset PredefinesFileID 191 PredefinesFileID = FileID(); 192 } 193 194 void Preprocessor::FinalizeForModelFile() { 195 NumEnteredSourceFiles = 1; 196 197 PragmaHandlers = std::move(PragmaHandlersBackup); 198 } 199 200 void Preprocessor::setPTHManager(PTHManager* pm) { 201 PTH.reset(pm); 202 FileMgr.addStatCache(PTH->createStatCache()); 203 } 204 205 void Preprocessor::DumpToken(const Token &Tok, bool DumpFlags) const { 206 llvm::errs() << tok::getTokenName(Tok.getKind()) << " '" 207 << getSpelling(Tok) << "'"; 208 209 if (!DumpFlags) return; 210 211 llvm::errs() << "\t"; 212 if (Tok.isAtStartOfLine()) 213 llvm::errs() << " [StartOfLine]"; 214 if (Tok.hasLeadingSpace()) 215 llvm::errs() << " [LeadingSpace]"; 216 if (Tok.isExpandDisabled()) 217 llvm::errs() << " [ExpandDisabled]"; 218 if (Tok.needsCleaning()) { 219 const char *Start = SourceMgr.getCharacterData(Tok.getLocation()); 220 llvm::errs() << " [UnClean='" << StringRef(Start, Tok.getLength()) 221 << "']"; 222 } 223 224 llvm::errs() << "\tLoc=<"; 225 DumpLocation(Tok.getLocation()); 226 llvm::errs() << ">"; 227 } 228 229 void Preprocessor::DumpLocation(SourceLocation Loc) const { 230 Loc.dump(SourceMgr); 231 } 232 233 void Preprocessor::DumpMacro(const MacroInfo &MI) const { 234 llvm::errs() << "MACRO: "; 235 for (unsigned i = 0, e = MI.getNumTokens(); i != e; ++i) { 236 DumpToken(MI.getReplacementToken(i)); 237 llvm::errs() << " "; 238 } 239 llvm::errs() << "\n"; 240 } 241 242 void Preprocessor::PrintStats() { 243 llvm::errs() << "\n*** Preprocessor Stats:\n"; 244 llvm::errs() << NumDirectives << " directives found:\n"; 245 llvm::errs() << " " << NumDefined << " #define.\n"; 246 llvm::errs() << " " << NumUndefined << " #undef.\n"; 247 llvm::errs() << " #include/#include_next/#import:\n"; 248 llvm::errs() << " " << NumEnteredSourceFiles << " source files entered.\n"; 249 llvm::errs() << " " << MaxIncludeStackDepth << " max include stack depth\n"; 250 llvm::errs() << " " << NumIf << " #if/#ifndef/#ifdef.\n"; 251 llvm::errs() << " " << NumElse << " #else/#elif.\n"; 252 llvm::errs() << " " << NumEndif << " #endif.\n"; 253 llvm::errs() << " " << NumPragma << " #pragma.\n"; 254 llvm::errs() << NumSkipped << " #if/#ifndef#ifdef regions skipped\n"; 255 256 llvm::errs() << NumMacroExpanded << "/" << NumFnMacroExpanded << "/" 257 << NumBuiltinMacroExpanded << " obj/fn/builtin macros expanded, " 258 << NumFastMacroExpanded << " on the fast path.\n"; 259 llvm::errs() << (NumFastTokenPaste+NumTokenPaste) 260 << " token paste (##) operations performed, " 261 << NumFastTokenPaste << " on the fast path.\n"; 262 263 llvm::errs() << "\nPreprocessor Memory: " << getTotalMemory() << "B total"; 264 265 llvm::errs() << "\n BumpPtr: " << BP.getTotalMemory(); 266 llvm::errs() << "\n Macro Expanded Tokens: " 267 << llvm::capacity_in_bytes(MacroExpandedTokens); 268 llvm::errs() << "\n Predefines Buffer: " << Predefines.capacity(); 269 llvm::errs() << "\n Macros: " << llvm::capacity_in_bytes(Macros); 270 llvm::errs() << "\n #pragma push_macro Info: " 271 << llvm::capacity_in_bytes(PragmaPushMacroInfo); 272 llvm::errs() << "\n Poison Reasons: " 273 << llvm::capacity_in_bytes(PoisonReasons); 274 llvm::errs() << "\n Comment Handlers: " 275 << llvm::capacity_in_bytes(CommentHandlers) << "\n"; 276 } 277 278 Preprocessor::macro_iterator 279 Preprocessor::macro_begin(bool IncludeExternalMacros) const { 280 if (IncludeExternalMacros && ExternalSource && 281 !ReadMacrosFromExternalSource) { 282 ReadMacrosFromExternalSource = true; 283 ExternalSource->ReadDefinedMacros(); 284 } 285 286 return Macros.begin(); 287 } 288 289 size_t Preprocessor::getTotalMemory() const { 290 return BP.getTotalMemory() 291 + llvm::capacity_in_bytes(MacroExpandedTokens) 292 + Predefines.capacity() /* Predefines buffer. */ 293 + llvm::capacity_in_bytes(Macros) 294 + llvm::capacity_in_bytes(PragmaPushMacroInfo) 295 + llvm::capacity_in_bytes(PoisonReasons) 296 + llvm::capacity_in_bytes(CommentHandlers); 297 } 298 299 Preprocessor::macro_iterator 300 Preprocessor::macro_end(bool IncludeExternalMacros) const { 301 if (IncludeExternalMacros && ExternalSource && 302 !ReadMacrosFromExternalSource) { 303 ReadMacrosFromExternalSource = true; 304 ExternalSource->ReadDefinedMacros(); 305 } 306 307 return Macros.end(); 308 } 309 310 /// \brief Compares macro tokens with a specified token value sequence. 311 static bool MacroDefinitionEquals(const MacroInfo *MI, 312 ArrayRef<TokenValue> Tokens) { 313 return Tokens.size() == MI->getNumTokens() && 314 std::equal(Tokens.begin(), Tokens.end(), MI->tokens_begin()); 315 } 316 317 StringRef Preprocessor::getLastMacroWithSpelling( 318 SourceLocation Loc, 319 ArrayRef<TokenValue> Tokens) const { 320 SourceLocation BestLocation; 321 StringRef BestSpelling; 322 for (Preprocessor::macro_iterator I = macro_begin(), E = macro_end(); 323 I != E; ++I) { 324 if (!I->second->getMacroInfo()->isObjectLike()) 325 continue; 326 const MacroDirective::DefInfo 327 Def = I->second->findDirectiveAtLoc(Loc, SourceMgr); 328 if (!Def) 329 continue; 330 if (!MacroDefinitionEquals(Def.getMacroInfo(), Tokens)) 331 continue; 332 SourceLocation Location = Def.getLocation(); 333 // Choose the macro defined latest. 334 if (BestLocation.isInvalid() || 335 (Location.isValid() && 336 SourceMgr.isBeforeInTranslationUnit(BestLocation, Location))) { 337 BestLocation = Location; 338 BestSpelling = I->first->getName(); 339 } 340 } 341 return BestSpelling; 342 } 343 344 void Preprocessor::recomputeCurLexerKind() { 345 if (CurLexer) 346 CurLexerKind = CLK_Lexer; 347 else if (CurPTHLexer) 348 CurLexerKind = CLK_PTHLexer; 349 else if (CurTokenLexer) 350 CurLexerKind = CLK_TokenLexer; 351 else 352 CurLexerKind = CLK_CachingLexer; 353 } 354 355 bool Preprocessor::SetCodeCompletionPoint(const FileEntry *File, 356 unsigned CompleteLine, 357 unsigned CompleteColumn) { 358 assert(File); 359 assert(CompleteLine && CompleteColumn && "Starts from 1:1"); 360 assert(!CodeCompletionFile && "Already set"); 361 362 using llvm::MemoryBuffer; 363 364 // Load the actual file's contents. 365 bool Invalid = false; 366 const MemoryBuffer *Buffer = SourceMgr.getMemoryBufferForFile(File, &Invalid); 367 if (Invalid) 368 return true; 369 370 // Find the byte position of the truncation point. 371 const char *Position = Buffer->getBufferStart(); 372 for (unsigned Line = 1; Line < CompleteLine; ++Line) { 373 for (; *Position; ++Position) { 374 if (*Position != '\r' && *Position != '\n') 375 continue; 376 377 // Eat \r\n or \n\r as a single line. 378 if ((Position[1] == '\r' || Position[1] == '\n') && 379 Position[0] != Position[1]) 380 ++Position; 381 ++Position; 382 break; 383 } 384 } 385 386 Position += CompleteColumn - 1; 387 388 // Insert '\0' at the code-completion point. 389 if (Position < Buffer->getBufferEnd()) { 390 CodeCompletionFile = File; 391 CodeCompletionOffset = Position - Buffer->getBufferStart(); 392 393 std::unique_ptr<MemoryBuffer> NewBuffer = 394 MemoryBuffer::getNewUninitMemBuffer(Buffer->getBufferSize() + 1, 395 Buffer->getBufferIdentifier()); 396 char *NewBuf = const_cast<char*>(NewBuffer->getBufferStart()); 397 char *NewPos = std::copy(Buffer->getBufferStart(), Position, NewBuf); 398 *NewPos = '\0'; 399 std::copy(Position, Buffer->getBufferEnd(), NewPos+1); 400 SourceMgr.overrideFileContents(File, std::move(NewBuffer)); 401 } 402 403 return false; 404 } 405 406 void Preprocessor::CodeCompleteNaturalLanguage() { 407 if (CodeComplete) 408 CodeComplete->CodeCompleteNaturalLanguage(); 409 setCodeCompletionReached(); 410 } 411 412 /// getSpelling - This method is used to get the spelling of a token into a 413 /// SmallVector. Note that the returned StringRef may not point to the 414 /// supplied buffer if a copy can be avoided. 415 StringRef Preprocessor::getSpelling(const Token &Tok, 416 SmallVectorImpl<char> &Buffer, 417 bool *Invalid) const { 418 // NOTE: this has to be checked *before* testing for an IdentifierInfo. 419 if (Tok.isNot(tok::raw_identifier) && !Tok.hasUCN()) { 420 // Try the fast path. 421 if (const IdentifierInfo *II = Tok.getIdentifierInfo()) 422 return II->getName(); 423 } 424 425 // Resize the buffer if we need to copy into it. 426 if (Tok.needsCleaning()) 427 Buffer.resize(Tok.getLength()); 428 429 const char *Ptr = Buffer.data(); 430 unsigned Len = getSpelling(Tok, Ptr, Invalid); 431 return StringRef(Ptr, Len); 432 } 433 434 /// CreateString - Plop the specified string into a scratch buffer and return a 435 /// location for it. If specified, the source location provides a source 436 /// location for the token. 437 void Preprocessor::CreateString(StringRef Str, Token &Tok, 438 SourceLocation ExpansionLocStart, 439 SourceLocation ExpansionLocEnd) { 440 Tok.setLength(Str.size()); 441 442 const char *DestPtr; 443 SourceLocation Loc = ScratchBuf->getToken(Str.data(), Str.size(), DestPtr); 444 445 if (ExpansionLocStart.isValid()) 446 Loc = SourceMgr.createExpansionLoc(Loc, ExpansionLocStart, 447 ExpansionLocEnd, Str.size()); 448 Tok.setLocation(Loc); 449 450 // If this is a raw identifier or a literal token, set the pointer data. 451 if (Tok.is(tok::raw_identifier)) 452 Tok.setRawIdentifierData(DestPtr); 453 else if (Tok.isLiteral()) 454 Tok.setLiteralData(DestPtr); 455 } 456 457 Module *Preprocessor::getCurrentModule() { 458 if (getLangOpts().CurrentModule.empty()) 459 return nullptr; 460 461 return getHeaderSearchInfo().lookupModule(getLangOpts().CurrentModule); 462 } 463 464 //===----------------------------------------------------------------------===// 465 // Preprocessor Initialization Methods 466 //===----------------------------------------------------------------------===// 467 468 469 /// EnterMainSourceFile - Enter the specified FileID as the main source file, 470 /// which implicitly adds the builtin defines etc. 471 void Preprocessor::EnterMainSourceFile() { 472 // We do not allow the preprocessor to reenter the main file. Doing so will 473 // cause FileID's to accumulate information from both runs (e.g. #line 474 // information) and predefined macros aren't guaranteed to be set properly. 475 assert(NumEnteredSourceFiles == 0 && "Cannot reenter the main file!"); 476 FileID MainFileID = SourceMgr.getMainFileID(); 477 478 // If MainFileID is loaded it means we loaded an AST file, no need to enter 479 // a main file. 480 if (!SourceMgr.isLoadedFileID(MainFileID)) { 481 // Enter the main file source buffer. 482 EnterSourceFile(MainFileID, nullptr, SourceLocation()); 483 484 // If we've been asked to skip bytes in the main file (e.g., as part of a 485 // precompiled preamble), do so now. 486 if (SkipMainFilePreamble.first > 0) 487 CurLexer->SkipBytes(SkipMainFilePreamble.first, 488 SkipMainFilePreamble.second); 489 490 // Tell the header info that the main file was entered. If the file is later 491 // #imported, it won't be re-entered. 492 if (const FileEntry *FE = SourceMgr.getFileEntryForID(MainFileID)) 493 HeaderInfo.IncrementIncludeCount(FE); 494 } 495 496 // Preprocess Predefines to populate the initial preprocessor state. 497 std::unique_ptr<llvm::MemoryBuffer> SB = 498 llvm::MemoryBuffer::getMemBufferCopy(Predefines, "<built-in>"); 499 assert(SB && "Cannot create predefined source buffer"); 500 FileID FID = SourceMgr.createFileID(std::move(SB)); 501 assert(!FID.isInvalid() && "Could not create FileID for predefines?"); 502 setPredefinesFileID(FID); 503 504 // Start parsing the predefines. 505 EnterSourceFile(FID, nullptr, SourceLocation()); 506 } 507 508 void Preprocessor::EndSourceFile() { 509 // Notify the client that we reached the end of the source file. 510 if (Callbacks) 511 Callbacks->EndOfMainFile(); 512 } 513 514 //===----------------------------------------------------------------------===// 515 // Lexer Event Handling. 516 //===----------------------------------------------------------------------===// 517 518 /// LookUpIdentifierInfo - Given a tok::raw_identifier token, look up the 519 /// identifier information for the token and install it into the token, 520 /// updating the token kind accordingly. 521 IdentifierInfo *Preprocessor::LookUpIdentifierInfo(Token &Identifier) const { 522 assert(!Identifier.getRawIdentifier().empty() && "No raw identifier data!"); 523 524 // Look up this token, see if it is a macro, or if it is a language keyword. 525 IdentifierInfo *II; 526 if (!Identifier.needsCleaning() && !Identifier.hasUCN()) { 527 // No cleaning needed, just use the characters from the lexed buffer. 528 II = getIdentifierInfo(Identifier.getRawIdentifier()); 529 } else { 530 // Cleaning needed, alloca a buffer, clean into it, then use the buffer. 531 SmallString<64> IdentifierBuffer; 532 StringRef CleanedStr = getSpelling(Identifier, IdentifierBuffer); 533 534 if (Identifier.hasUCN()) { 535 SmallString<64> UCNIdentifierBuffer; 536 expandUCNs(UCNIdentifierBuffer, CleanedStr); 537 II = getIdentifierInfo(UCNIdentifierBuffer); 538 } else { 539 II = getIdentifierInfo(CleanedStr); 540 } 541 } 542 543 // Update the token info (identifier info and appropriate token kind). 544 Identifier.setIdentifierInfo(II); 545 Identifier.setKind(II->getTokenID()); 546 547 return II; 548 } 549 550 void Preprocessor::SetPoisonReason(IdentifierInfo *II, unsigned DiagID) { 551 PoisonReasons[II] = DiagID; 552 } 553 554 void Preprocessor::PoisonSEHIdentifiers(bool Poison) { 555 assert(Ident__exception_code && Ident__exception_info); 556 assert(Ident___exception_code && Ident___exception_info); 557 Ident__exception_code->setIsPoisoned(Poison); 558 Ident___exception_code->setIsPoisoned(Poison); 559 Ident_GetExceptionCode->setIsPoisoned(Poison); 560 Ident__exception_info->setIsPoisoned(Poison); 561 Ident___exception_info->setIsPoisoned(Poison); 562 Ident_GetExceptionInfo->setIsPoisoned(Poison); 563 Ident__abnormal_termination->setIsPoisoned(Poison); 564 Ident___abnormal_termination->setIsPoisoned(Poison); 565 Ident_AbnormalTermination->setIsPoisoned(Poison); 566 } 567 568 void Preprocessor::HandlePoisonedIdentifier(Token & Identifier) { 569 assert(Identifier.getIdentifierInfo() && 570 "Can't handle identifiers without identifier info!"); 571 llvm::DenseMap<IdentifierInfo*,unsigned>::const_iterator it = 572 PoisonReasons.find(Identifier.getIdentifierInfo()); 573 if(it == PoisonReasons.end()) 574 Diag(Identifier, diag::err_pp_used_poisoned_id); 575 else 576 Diag(Identifier,it->second) << Identifier.getIdentifierInfo(); 577 } 578 579 /// HandleIdentifier - This callback is invoked when the lexer reads an 580 /// identifier. This callback looks up the identifier in the map and/or 581 /// potentially macro expands it or turns it into a named token (like 'for'). 582 /// 583 /// Note that callers of this method are guarded by checking the 584 /// IdentifierInfo's 'isHandleIdentifierCase' bit. If this method changes, the 585 /// IdentifierInfo methods that compute these properties will need to change to 586 /// match. 587 bool Preprocessor::HandleIdentifier(Token &Identifier) { 588 assert(Identifier.getIdentifierInfo() && 589 "Can't handle identifiers without identifier info!"); 590 591 IdentifierInfo &II = *Identifier.getIdentifierInfo(); 592 593 // If the information about this identifier is out of date, update it from 594 // the external source. 595 // We have to treat __VA_ARGS__ in a special way, since it gets 596 // serialized with isPoisoned = true, but our preprocessor may have 597 // unpoisoned it if we're defining a C99 macro. 598 if (II.isOutOfDate()) { 599 bool CurrentIsPoisoned = false; 600 if (&II == Ident__VA_ARGS__) 601 CurrentIsPoisoned = Ident__VA_ARGS__->isPoisoned(); 602 603 ExternalSource->updateOutOfDateIdentifier(II); 604 Identifier.setKind(II.getTokenID()); 605 606 if (&II == Ident__VA_ARGS__) 607 II.setIsPoisoned(CurrentIsPoisoned); 608 } 609 610 // If this identifier was poisoned, and if it was not produced from a macro 611 // expansion, emit an error. 612 if (II.isPoisoned() && CurPPLexer) { 613 HandlePoisonedIdentifier(Identifier); 614 } 615 616 // If this is a macro to be expanded, do it. 617 if (MacroDirective *MD = getMacroDirective(&II)) { 618 MacroInfo *MI = MD->getMacroInfo(); 619 if (!DisableMacroExpansion) { 620 if (!Identifier.isExpandDisabled() && MI->isEnabled()) { 621 // C99 6.10.3p10: If the preprocessing token immediately after the 622 // macro name isn't a '(', this macro should not be expanded. 623 if (!MI->isFunctionLike() || isNextPPTokenLParen()) 624 return HandleMacroExpandedIdentifier(Identifier, MD); 625 } else { 626 // C99 6.10.3.4p2 says that a disabled macro may never again be 627 // expanded, even if it's in a context where it could be expanded in the 628 // future. 629 Identifier.setFlag(Token::DisableExpand); 630 if (MI->isObjectLike() || isNextPPTokenLParen()) 631 Diag(Identifier, diag::pp_disabled_macro_expansion); 632 } 633 } 634 } 635 636 // If this identifier is a keyword in C++11, produce a warning. Don't warn if 637 // we're not considering macro expansion, since this identifier might be the 638 // name of a macro. 639 // FIXME: This warning is disabled in cases where it shouldn't be, like 640 // "#define constexpr constexpr", "int constexpr;" 641 if (II.isCXX11CompatKeyword() && !DisableMacroExpansion) { 642 Diag(Identifier, diag::warn_cxx11_keyword) << II.getName(); 643 // Don't diagnose this keyword again in this translation unit. 644 II.setIsCXX11CompatKeyword(false); 645 } 646 647 // C++ 2.11p2: If this is an alternative representation of a C++ operator, 648 // then we act as if it is the actual operator and not the textual 649 // representation of it. 650 if (II.isCPlusPlusOperatorKeyword()) 651 Identifier.setIdentifierInfo(nullptr); 652 653 // If this is an extension token, diagnose its use. 654 // We avoid diagnosing tokens that originate from macro definitions. 655 // FIXME: This warning is disabled in cases where it shouldn't be, 656 // like "#define TY typeof", "TY(1) x". 657 if (II.isExtensionToken() && !DisableMacroExpansion) 658 Diag(Identifier, diag::ext_token_used); 659 660 // If this is the 'import' contextual keyword following an '@', note 661 // that the next token indicates a module name. 662 // 663 // Note that we do not treat 'import' as a contextual 664 // keyword when we're in a caching lexer, because caching lexers only get 665 // used in contexts where import declarations are disallowed. 666 if (LastTokenWasAt && II.isModulesImport() && !InMacroArgs && 667 !DisableMacroExpansion && getLangOpts().Modules && 668 CurLexerKind != CLK_CachingLexer) { 669 ModuleImportLoc = Identifier.getLocation(); 670 ModuleImportPath.clear(); 671 ModuleImportExpectsIdentifier = true; 672 CurLexerKind = CLK_LexAfterModuleImport; 673 } 674 return true; 675 } 676 677 void Preprocessor::Lex(Token &Result) { 678 // We loop here until a lex function retuns a token; this avoids recursion. 679 bool ReturnedToken; 680 do { 681 switch (CurLexerKind) { 682 case CLK_Lexer: 683 ReturnedToken = CurLexer->Lex(Result); 684 break; 685 case CLK_PTHLexer: 686 ReturnedToken = CurPTHLexer->Lex(Result); 687 break; 688 case CLK_TokenLexer: 689 ReturnedToken = CurTokenLexer->Lex(Result); 690 break; 691 case CLK_CachingLexer: 692 CachingLex(Result); 693 ReturnedToken = true; 694 break; 695 case CLK_LexAfterModuleImport: 696 LexAfterModuleImport(Result); 697 ReturnedToken = true; 698 break; 699 } 700 } while (!ReturnedToken); 701 702 LastTokenWasAt = Result.is(tok::at); 703 } 704 705 706 /// \brief Lex a token following the 'import' contextual keyword. 707 /// 708 void Preprocessor::LexAfterModuleImport(Token &Result) { 709 // Figure out what kind of lexer we actually have. 710 recomputeCurLexerKind(); 711 712 // Lex the next token. 713 Lex(Result); 714 715 // The token sequence 716 // 717 // import identifier (. identifier)* 718 // 719 // indicates a module import directive. We already saw the 'import' 720 // contextual keyword, so now we're looking for the identifiers. 721 if (ModuleImportExpectsIdentifier && Result.getKind() == tok::identifier) { 722 // We expected to see an identifier here, and we did; continue handling 723 // identifiers. 724 ModuleImportPath.push_back(std::make_pair(Result.getIdentifierInfo(), 725 Result.getLocation())); 726 ModuleImportExpectsIdentifier = false; 727 CurLexerKind = CLK_LexAfterModuleImport; 728 return; 729 } 730 731 // If we're expecting a '.' or a ';', and we got a '.', then wait until we 732 // see the next identifier. 733 if (!ModuleImportExpectsIdentifier && Result.getKind() == tok::period) { 734 ModuleImportExpectsIdentifier = true; 735 CurLexerKind = CLK_LexAfterModuleImport; 736 return; 737 } 738 739 // If we have a non-empty module path, load the named module. 740 if (!ModuleImportPath.empty() && getLangOpts().Modules) { 741 Module *Imported = TheModuleLoader.loadModule(ModuleImportLoc, 742 ModuleImportPath, 743 Module::MacrosVisible, 744 /*IsIncludeDirective=*/false); 745 if (Callbacks) 746 Callbacks->moduleImport(ModuleImportLoc, ModuleImportPath, Imported); 747 } 748 } 749 750 bool Preprocessor::FinishLexStringLiteral(Token &Result, std::string &String, 751 const char *DiagnosticTag, 752 bool AllowMacroExpansion) { 753 // We need at least one string literal. 754 if (Result.isNot(tok::string_literal)) { 755 Diag(Result, diag::err_expected_string_literal) 756 << /*Source='in...'*/0 << DiagnosticTag; 757 return false; 758 } 759 760 // Lex string literal tokens, optionally with macro expansion. 761 SmallVector<Token, 4> StrToks; 762 do { 763 StrToks.push_back(Result); 764 765 if (Result.hasUDSuffix()) 766 Diag(Result, diag::err_invalid_string_udl); 767 768 if (AllowMacroExpansion) 769 Lex(Result); 770 else 771 LexUnexpandedToken(Result); 772 } while (Result.is(tok::string_literal)); 773 774 // Concatenate and parse the strings. 775 StringLiteralParser Literal(StrToks, *this); 776 assert(Literal.isAscii() && "Didn't allow wide strings in"); 777 778 if (Literal.hadError) 779 return false; 780 781 if (Literal.Pascal) { 782 Diag(StrToks[0].getLocation(), diag::err_expected_string_literal) 783 << /*Source='in...'*/0 << DiagnosticTag; 784 return false; 785 } 786 787 String = Literal.GetString(); 788 return true; 789 } 790 791 bool Preprocessor::parseSimpleIntegerLiteral(Token &Tok, uint64_t &Value) { 792 assert(Tok.is(tok::numeric_constant)); 793 SmallString<8> IntegerBuffer; 794 bool NumberInvalid = false; 795 StringRef Spelling = getSpelling(Tok, IntegerBuffer, &NumberInvalid); 796 if (NumberInvalid) 797 return false; 798 NumericLiteralParser Literal(Spelling, Tok.getLocation(), *this); 799 if (Literal.hadError || !Literal.isIntegerLiteral() || Literal.hasUDSuffix()) 800 return false; 801 llvm::APInt APVal(64, 0); 802 if (Literal.GetIntegerValue(APVal)) 803 return false; 804 Lex(Tok); 805 Value = APVal.getLimitedValue(); 806 return true; 807 } 808 809 void Preprocessor::addCommentHandler(CommentHandler *Handler) { 810 assert(Handler && "NULL comment handler"); 811 assert(std::find(CommentHandlers.begin(), CommentHandlers.end(), Handler) == 812 CommentHandlers.end() && "Comment handler already registered"); 813 CommentHandlers.push_back(Handler); 814 } 815 816 void Preprocessor::removeCommentHandler(CommentHandler *Handler) { 817 std::vector<CommentHandler *>::iterator Pos 818 = std::find(CommentHandlers.begin(), CommentHandlers.end(), Handler); 819 assert(Pos != CommentHandlers.end() && "Comment handler not registered"); 820 CommentHandlers.erase(Pos); 821 } 822 823 bool Preprocessor::HandleComment(Token &result, SourceRange Comment) { 824 bool AnyPendingTokens = false; 825 for (std::vector<CommentHandler *>::iterator H = CommentHandlers.begin(), 826 HEnd = CommentHandlers.end(); 827 H != HEnd; ++H) { 828 if ((*H)->HandleComment(*this, Comment)) 829 AnyPendingTokens = true; 830 } 831 if (!AnyPendingTokens || getCommentRetentionState()) 832 return false; 833 Lex(result); 834 return true; 835 } 836 837 ModuleLoader::~ModuleLoader() { } 838 839 CommentHandler::~CommentHandler() { } 840 841 CodeCompletionHandler::~CodeCompletionHandler() { } 842 843 void Preprocessor::createPreprocessingRecord() { 844 if (Record) 845 return; 846 847 Record = new PreprocessingRecord(getSourceManager()); 848 addPPCallbacks(std::unique_ptr<PPCallbacks>(Record)); 849 } 850