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