1 //===--- PPDirectives.cpp - Directive Handling for Preprocessor -----------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 ///
9 /// \file
10 /// Implements # directive processing for the Preprocessor.
11 ///
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/Basic/CharInfo.h"
15 #include "clang/Basic/FileManager.h"
16 #include "clang/Basic/IdentifierTable.h"
17 #include "clang/Basic/LangOptions.h"
18 #include "clang/Basic/Module.h"
19 #include "clang/Basic/SourceLocation.h"
20 #include "clang/Basic/SourceManager.h"
21 #include "clang/Basic/TokenKinds.h"
22 #include "clang/Lex/CodeCompletionHandler.h"
23 #include "clang/Lex/HeaderSearch.h"
24 #include "clang/Lex/LexDiagnostic.h"
25 #include "clang/Lex/LiteralSupport.h"
26 #include "clang/Lex/MacroInfo.h"
27 #include "clang/Lex/ModuleLoader.h"
28 #include "clang/Lex/ModuleMap.h"
29 #include "clang/Lex/PPCallbacks.h"
30 #include "clang/Lex/Pragma.h"
31 #include "clang/Lex/Preprocessor.h"
32 #include "clang/Lex/PreprocessorOptions.h"
33 #include "clang/Lex/Token.h"
34 #include "clang/Lex/VariadicMacroSupport.h"
35 #include "llvm/ADT/ArrayRef.h"
36 #include "llvm/ADT/ScopeExit.h"
37 #include "llvm/ADT/SmallString.h"
38 #include "llvm/ADT/SmallVector.h"
39 #include "llvm/ADT/STLExtras.h"
40 #include "llvm/ADT/StringSwitch.h"
41 #include "llvm/ADT/StringRef.h"
42 #include "llvm/Support/AlignOf.h"
43 #include "llvm/Support/ErrorHandling.h"
44 #include "llvm/Support/Path.h"
45 #include <algorithm>
46 #include <cassert>
47 #include <cstring>
48 #include <new>
49 #include <string>
50 #include <utility>
51 
52 using namespace clang;
53 
54 //===----------------------------------------------------------------------===//
55 // Utility Methods for Preprocessor Directive Handling.
56 //===----------------------------------------------------------------------===//
57 
58 MacroInfo *Preprocessor::AllocateMacroInfo(SourceLocation L) {
59   auto *MIChain = new (BP) MacroInfoChain{L, MIChainHead};
60   MIChainHead = MIChain;
61   return &MIChain->MI;
62 }
63 
64 DefMacroDirective *Preprocessor::AllocateDefMacroDirective(MacroInfo *MI,
65                                                            SourceLocation Loc) {
66   return new (BP) DefMacroDirective(MI, Loc);
67 }
68 
69 UndefMacroDirective *
70 Preprocessor::AllocateUndefMacroDirective(SourceLocation UndefLoc) {
71   return new (BP) UndefMacroDirective(UndefLoc);
72 }
73 
74 VisibilityMacroDirective *
75 Preprocessor::AllocateVisibilityMacroDirective(SourceLocation Loc,
76                                                bool isPublic) {
77   return new (BP) VisibilityMacroDirective(Loc, isPublic);
78 }
79 
80 /// Read and discard all tokens remaining on the current line until
81 /// the tok::eod token is found.
82 SourceRange Preprocessor::DiscardUntilEndOfDirective() {
83   Token Tmp;
84   SourceRange Res;
85 
86   LexUnexpandedToken(Tmp);
87   Res.setBegin(Tmp.getLocation());
88   while (Tmp.isNot(tok::eod)) {
89     assert(Tmp.isNot(tok::eof) && "EOF seen while discarding directive tokens");
90     LexUnexpandedToken(Tmp);
91   }
92   Res.setEnd(Tmp.getLocation());
93   return Res;
94 }
95 
96 /// Enumerates possible cases of #define/#undef a reserved identifier.
97 enum MacroDiag {
98   MD_NoWarn,        //> Not a reserved identifier
99   MD_KeywordDef,    //> Macro hides keyword, enabled by default
100   MD_ReservedMacro  //> #define of #undef reserved id, disabled by default
101 };
102 
103 /// Checks if the specified identifier is reserved in the specified
104 /// language.
105 /// This function does not check if the identifier is a keyword.
106 static bool isReservedId(StringRef Text, const LangOptions &Lang) {
107   // C++ [macro.names], C11 7.1.3:
108   // All identifiers that begin with an underscore and either an uppercase
109   // letter or another underscore are always reserved for any use.
110   if (Text.size() >= 2 && Text[0] == '_' &&
111       (isUppercase(Text[1]) || Text[1] == '_'))
112       return true;
113   // C++ [global.names]
114   // Each name that contains a double underscore ... is reserved to the
115   // implementation for any use.
116   if (Lang.CPlusPlus) {
117     if (Text.find("__") != StringRef::npos)
118       return true;
119   }
120   return false;
121 }
122 
123 // The -fmodule-name option tells the compiler to textually include headers in
124 // the specified module, meaning clang won't build the specified module. This is
125 // useful in a number of situations, for instance, when building a library that
126 // vends a module map, one might want to avoid hitting intermediate build
127 // products containimg the the module map or avoid finding the system installed
128 // modulemap for that library.
129 static bool isForModuleBuilding(Module *M, StringRef CurrentModule,
130                                 StringRef ModuleName) {
131   StringRef TopLevelName = M->getTopLevelModuleName();
132 
133   // When building framework Foo, we wanna make sure that Foo *and* Foo_Private
134   // are textually included and no modules are built for both.
135   if (M->getTopLevelModule()->IsFramework && CurrentModule == ModuleName &&
136       !CurrentModule.endswith("_Private") && TopLevelName.endswith("_Private"))
137     TopLevelName = TopLevelName.drop_back(8);
138 
139   return TopLevelName == CurrentModule;
140 }
141 
142 static MacroDiag shouldWarnOnMacroDef(Preprocessor &PP, IdentifierInfo *II) {
143   const LangOptions &Lang = PP.getLangOpts();
144   StringRef Text = II->getName();
145   if (isReservedId(Text, Lang))
146     return MD_ReservedMacro;
147   if (II->isKeyword(Lang))
148     return MD_KeywordDef;
149   if (Lang.CPlusPlus11 && (Text.equals("override") || Text.equals("final")))
150     return MD_KeywordDef;
151   return MD_NoWarn;
152 }
153 
154 static MacroDiag shouldWarnOnMacroUndef(Preprocessor &PP, IdentifierInfo *II) {
155   const LangOptions &Lang = PP.getLangOpts();
156   StringRef Text = II->getName();
157   // Do not warn on keyword undef.  It is generally harmless and widely used.
158   if (isReservedId(Text, Lang))
159     return MD_ReservedMacro;
160   return MD_NoWarn;
161 }
162 
163 // Return true if we want to issue a diagnostic by default if we
164 // encounter this name in a #include with the wrong case. For now,
165 // this includes the standard C and C++ headers, Posix headers,
166 // and Boost headers. Improper case for these #includes is a
167 // potential portability issue.
168 static bool warnByDefaultOnWrongCase(StringRef Include) {
169   // If the first component of the path is "boost", treat this like a standard header
170   // for the purposes of diagnostics.
171   if (::llvm::sys::path::begin(Include)->equals_lower("boost"))
172     return true;
173 
174   // "condition_variable" is the longest standard header name at 18 characters.
175   // If the include file name is longer than that, it can't be a standard header.
176   static const size_t MaxStdHeaderNameLen = 18u;
177   if (Include.size() > MaxStdHeaderNameLen)
178     return false;
179 
180   // Lowercase and normalize the search string.
181   SmallString<32> LowerInclude{Include};
182   for (char &Ch : LowerInclude) {
183     // In the ASCII range?
184     if (static_cast<unsigned char>(Ch) > 0x7f)
185       return false; // Can't be a standard header
186     // ASCII lowercase:
187     if (Ch >= 'A' && Ch <= 'Z')
188       Ch += 'a' - 'A';
189     // Normalize path separators for comparison purposes.
190     else if (::llvm::sys::path::is_separator(Ch))
191       Ch = '/';
192   }
193 
194   // The standard C/C++ and Posix headers
195   return llvm::StringSwitch<bool>(LowerInclude)
196     // C library headers
197     .Cases("assert.h", "complex.h", "ctype.h", "errno.h", "fenv.h", true)
198     .Cases("float.h", "inttypes.h", "iso646.h", "limits.h", "locale.h", true)
199     .Cases("math.h", "setjmp.h", "signal.h", "stdalign.h", "stdarg.h", true)
200     .Cases("stdatomic.h", "stdbool.h", "stddef.h", "stdint.h", "stdio.h", true)
201     .Cases("stdlib.h", "stdnoreturn.h", "string.h", "tgmath.h", "threads.h", true)
202     .Cases("time.h", "uchar.h", "wchar.h", "wctype.h", true)
203 
204     // C++ headers for C library facilities
205     .Cases("cassert", "ccomplex", "cctype", "cerrno", "cfenv", true)
206     .Cases("cfloat", "cinttypes", "ciso646", "climits", "clocale", true)
207     .Cases("cmath", "csetjmp", "csignal", "cstdalign", "cstdarg", true)
208     .Cases("cstdbool", "cstddef", "cstdint", "cstdio", "cstdlib", true)
209     .Cases("cstring", "ctgmath", "ctime", "cuchar", "cwchar", true)
210     .Case("cwctype", true)
211 
212     // C++ library headers
213     .Cases("algorithm", "fstream", "list", "regex", "thread", true)
214     .Cases("array", "functional", "locale", "scoped_allocator", "tuple", true)
215     .Cases("atomic", "future", "map", "set", "type_traits", true)
216     .Cases("bitset", "initializer_list", "memory", "shared_mutex", "typeindex", true)
217     .Cases("chrono", "iomanip", "mutex", "sstream", "typeinfo", true)
218     .Cases("codecvt", "ios", "new", "stack", "unordered_map", true)
219     .Cases("complex", "iosfwd", "numeric", "stdexcept", "unordered_set", true)
220     .Cases("condition_variable", "iostream", "ostream", "streambuf", "utility", true)
221     .Cases("deque", "istream", "queue", "string", "valarray", true)
222     .Cases("exception", "iterator", "random", "strstream", "vector", true)
223     .Cases("forward_list", "limits", "ratio", "system_error", true)
224 
225     // POSIX headers (which aren't also C headers)
226     .Cases("aio.h", "arpa/inet.h", "cpio.h", "dirent.h", "dlfcn.h", true)
227     .Cases("fcntl.h", "fmtmsg.h", "fnmatch.h", "ftw.h", "glob.h", true)
228     .Cases("grp.h", "iconv.h", "langinfo.h", "libgen.h", "monetary.h", true)
229     .Cases("mqueue.h", "ndbm.h", "net/if.h", "netdb.h", "netinet/in.h", true)
230     .Cases("netinet/tcp.h", "nl_types.h", "poll.h", "pthread.h", "pwd.h", true)
231     .Cases("regex.h", "sched.h", "search.h", "semaphore.h", "spawn.h", true)
232     .Cases("strings.h", "stropts.h", "sys/ipc.h", "sys/mman.h", "sys/msg.h", true)
233     .Cases("sys/resource.h", "sys/select.h",  "sys/sem.h", "sys/shm.h", "sys/socket.h", true)
234     .Cases("sys/stat.h", "sys/statvfs.h", "sys/time.h", "sys/times.h", "sys/types.h", true)
235     .Cases("sys/uio.h", "sys/un.h", "sys/utsname.h", "sys/wait.h", "syslog.h", true)
236     .Cases("tar.h", "termios.h", "trace.h", "ulimit.h", true)
237     .Cases("unistd.h", "utime.h", "utmpx.h", "wordexp.h", true)
238     .Default(false);
239 }
240 
241 bool Preprocessor::CheckMacroName(Token &MacroNameTok, MacroUse isDefineUndef,
242                                   bool *ShadowFlag) {
243   // Missing macro name?
244   if (MacroNameTok.is(tok::eod))
245     return Diag(MacroNameTok, diag::err_pp_missing_macro_name);
246 
247   IdentifierInfo *II = MacroNameTok.getIdentifierInfo();
248   if (!II)
249     return Diag(MacroNameTok, diag::err_pp_macro_not_identifier);
250 
251   if (II->isCPlusPlusOperatorKeyword()) {
252     // C++ 2.5p2: Alternative tokens behave the same as its primary token
253     // except for their spellings.
254     Diag(MacroNameTok, getLangOpts().MicrosoftExt
255                            ? diag::ext_pp_operator_used_as_macro_name
256                            : diag::err_pp_operator_used_as_macro_name)
257         << II << MacroNameTok.getKind();
258     // Allow #defining |and| and friends for Microsoft compatibility or
259     // recovery when legacy C headers are included in C++.
260   }
261 
262   if ((isDefineUndef != MU_Other) && II->getPPKeywordID() == tok::pp_defined) {
263     // Error if defining "defined": C99 6.10.8/4, C++ [cpp.predefined]p4.
264     return Diag(MacroNameTok, diag::err_defined_macro_name);
265   }
266 
267   if (isDefineUndef == MU_Undef) {
268     auto *MI = getMacroInfo(II);
269     if (MI && MI->isBuiltinMacro()) {
270       // Warn if undefining "__LINE__" and other builtins, per C99 6.10.8/4
271       // and C++ [cpp.predefined]p4], but allow it as an extension.
272       Diag(MacroNameTok, diag::ext_pp_undef_builtin_macro);
273     }
274   }
275 
276   // If defining/undefining reserved identifier or a keyword, we need to issue
277   // a warning.
278   SourceLocation MacroNameLoc = MacroNameTok.getLocation();
279   if (ShadowFlag)
280     *ShadowFlag = false;
281   if (!SourceMgr.isInSystemHeader(MacroNameLoc) &&
282       (SourceMgr.getBufferName(MacroNameLoc) != "<built-in>")) {
283     MacroDiag D = MD_NoWarn;
284     if (isDefineUndef == MU_Define) {
285       D = shouldWarnOnMacroDef(*this, II);
286     }
287     else if (isDefineUndef == MU_Undef)
288       D = shouldWarnOnMacroUndef(*this, II);
289     if (D == MD_KeywordDef) {
290       // We do not want to warn on some patterns widely used in configuration
291       // scripts.  This requires analyzing next tokens, so do not issue warnings
292       // now, only inform caller.
293       if (ShadowFlag)
294         *ShadowFlag = true;
295     }
296     if (D == MD_ReservedMacro)
297       Diag(MacroNameTok, diag::warn_pp_macro_is_reserved_id);
298   }
299 
300   // Okay, we got a good identifier.
301   return false;
302 }
303 
304 /// Lex and validate a macro name, which occurs after a
305 /// \#define or \#undef.
306 ///
307 /// This sets the token kind to eod and discards the rest of the macro line if
308 /// the macro name is invalid.
309 ///
310 /// \param MacroNameTok Token that is expected to be a macro name.
311 /// \param isDefineUndef Context in which macro is used.
312 /// \param ShadowFlag Points to a flag that is set if macro shadows a keyword.
313 void Preprocessor::ReadMacroName(Token &MacroNameTok, MacroUse isDefineUndef,
314                                  bool *ShadowFlag) {
315   // Read the token, don't allow macro expansion on it.
316   LexUnexpandedToken(MacroNameTok);
317 
318   if (MacroNameTok.is(tok::code_completion)) {
319     if (CodeComplete)
320       CodeComplete->CodeCompleteMacroName(isDefineUndef == MU_Define);
321     setCodeCompletionReached();
322     LexUnexpandedToken(MacroNameTok);
323   }
324 
325   if (!CheckMacroName(MacroNameTok, isDefineUndef, ShadowFlag))
326     return;
327 
328   // Invalid macro name, read and discard the rest of the line and set the
329   // token kind to tok::eod if necessary.
330   if (MacroNameTok.isNot(tok::eod)) {
331     MacroNameTok.setKind(tok::eod);
332     DiscardUntilEndOfDirective();
333   }
334 }
335 
336 /// Ensure that the next token is a tok::eod token.
337 ///
338 /// If not, emit a diagnostic and consume up until the eod.  If EnableMacros is
339 /// true, then we consider macros that expand to zero tokens as being ok.
340 ///
341 /// Returns the location of the end of the directive.
342 SourceLocation Preprocessor::CheckEndOfDirective(const char *DirType,
343                                                  bool EnableMacros) {
344   Token Tmp;
345   // Lex unexpanded tokens for most directives: macros might expand to zero
346   // tokens, causing us to miss diagnosing invalid lines.  Some directives (like
347   // #line) allow empty macros.
348   if (EnableMacros)
349     Lex(Tmp);
350   else
351     LexUnexpandedToken(Tmp);
352 
353   // There should be no tokens after the directive, but we allow them as an
354   // extension.
355   while (Tmp.is(tok::comment))  // Skip comments in -C mode.
356     LexUnexpandedToken(Tmp);
357 
358   if (Tmp.is(tok::eod))
359     return Tmp.getLocation();
360 
361   // Add a fixit in GNU/C99/C++ mode.  Don't offer a fixit for strict-C89,
362   // or if this is a macro-style preprocessing directive, because it is more
363   // trouble than it is worth to insert /**/ and check that there is no /**/
364   // in the range also.
365   FixItHint Hint;
366   if ((LangOpts.GNUMode || LangOpts.C99 || LangOpts.CPlusPlus) &&
367       !CurTokenLexer)
368     Hint = FixItHint::CreateInsertion(Tmp.getLocation(),"//");
369   Diag(Tmp, diag::ext_pp_extra_tokens_at_eol) << DirType << Hint;
370   return DiscardUntilEndOfDirective().getEnd();
371 }
372 
373 Optional<unsigned> Preprocessor::getSkippedRangeForExcludedConditionalBlock(
374     SourceLocation HashLoc) {
375   if (!ExcludedConditionalDirectiveSkipMappings)
376     return None;
377   if (!HashLoc.isFileID())
378     return None;
379 
380   std::pair<FileID, unsigned> HashFileOffset =
381       SourceMgr.getDecomposedLoc(HashLoc);
382   const llvm::MemoryBuffer *Buf = SourceMgr.getBuffer(HashFileOffset.first);
383   auto It = ExcludedConditionalDirectiveSkipMappings->find(Buf);
384   if (It == ExcludedConditionalDirectiveSkipMappings->end())
385     return None;
386 
387   const PreprocessorSkippedRangeMapping &SkippedRanges = *It->getSecond();
388   // Check if the offset of '#' is mapped in the skipped ranges.
389   auto MappingIt = SkippedRanges.find(HashFileOffset.second);
390   if (MappingIt == SkippedRanges.end())
391     return None;
392 
393   unsigned BytesToSkip = MappingIt->getSecond();
394   unsigned CurLexerBufferOffset = CurLexer->getCurrentBufferOffset();
395   assert(CurLexerBufferOffset >= HashFileOffset.second &&
396          "lexer is before the hash?");
397   // Take into account the fact that the lexer has already advanced, so the
398   // number of bytes to skip must be adjusted.
399   unsigned LengthDiff = CurLexerBufferOffset - HashFileOffset.second;
400   assert(BytesToSkip >= LengthDiff && "lexer is after the skipped range?");
401   return BytesToSkip - LengthDiff;
402 }
403 
404 /// SkipExcludedConditionalBlock - We just read a \#if or related directive and
405 /// decided that the subsequent tokens are in the \#if'd out portion of the
406 /// file.  Lex the rest of the file, until we see an \#endif.  If
407 /// FoundNonSkipPortion is true, then we have already emitted code for part of
408 /// this \#if directive, so \#else/\#elif blocks should never be entered.
409 /// If ElseOk is true, then \#else directives are ok, if not, then we have
410 /// already seen one so a \#else directive is a duplicate.  When this returns,
411 /// the caller can lex the first valid token.
412 void Preprocessor::SkipExcludedConditionalBlock(SourceLocation HashTokenLoc,
413                                                 SourceLocation IfTokenLoc,
414                                                 bool FoundNonSkipPortion,
415                                                 bool FoundElse,
416                                                 SourceLocation ElseLoc) {
417   ++NumSkipped;
418   assert(!CurTokenLexer && CurPPLexer && "Lexing a macro, not a file?");
419 
420   if (PreambleConditionalStack.reachedEOFWhileSkipping())
421     PreambleConditionalStack.clearSkipInfo();
422   else
423     CurPPLexer->pushConditionalLevel(IfTokenLoc, /*isSkipping*/ false,
424                                      FoundNonSkipPortion, FoundElse);
425 
426   // Enter raw mode to disable identifier lookup (and thus macro expansion),
427   // disabling warnings, etc.
428   CurPPLexer->LexingRawMode = true;
429   Token Tok;
430   if (auto SkipLength =
431           getSkippedRangeForExcludedConditionalBlock(HashTokenLoc)) {
432     // Skip to the next '#endif' / '#else' / '#elif'.
433     CurLexer->skipOver(*SkipLength);
434   }
435   while (true) {
436     CurLexer->Lex(Tok);
437 
438     if (Tok.is(tok::code_completion)) {
439       if (CodeComplete)
440         CodeComplete->CodeCompleteInConditionalExclusion();
441       setCodeCompletionReached();
442       continue;
443     }
444 
445     // If this is the end of the buffer, we have an error.
446     if (Tok.is(tok::eof)) {
447       // We don't emit errors for unterminated conditionals here,
448       // Lexer::LexEndOfFile can do that properly.
449       // Just return and let the caller lex after this #include.
450       if (PreambleConditionalStack.isRecording())
451         PreambleConditionalStack.SkipInfo.emplace(
452             HashTokenLoc, IfTokenLoc, FoundNonSkipPortion, FoundElse, ElseLoc);
453       break;
454     }
455 
456     // If this token is not a preprocessor directive, just skip it.
457     if (Tok.isNot(tok::hash) || !Tok.isAtStartOfLine())
458       continue;
459 
460     // We just parsed a # character at the start of a line, so we're in
461     // directive mode.  Tell the lexer this so any newlines we see will be
462     // converted into an EOD token (this terminates the macro).
463     CurPPLexer->ParsingPreprocessorDirective = true;
464     if (CurLexer) CurLexer->SetKeepWhitespaceMode(false);
465 
466 
467     // Read the next token, the directive flavor.
468     LexUnexpandedToken(Tok);
469 
470     // If this isn't an identifier directive (e.g. is "# 1\n" or "#\n", or
471     // something bogus), skip it.
472     if (Tok.isNot(tok::raw_identifier)) {
473       CurPPLexer->ParsingPreprocessorDirective = false;
474       // Restore comment saving mode.
475       if (CurLexer) CurLexer->resetExtendedTokenMode();
476       continue;
477     }
478 
479     // If the first letter isn't i or e, it isn't intesting to us.  We know that
480     // this is safe in the face of spelling differences, because there is no way
481     // to spell an i/e in a strange way that is another letter.  Skipping this
482     // allows us to avoid looking up the identifier info for #define/#undef and
483     // other common directives.
484     StringRef RI = Tok.getRawIdentifier();
485 
486     char FirstChar = RI[0];
487     if (FirstChar >= 'a' && FirstChar <= 'z' &&
488         FirstChar != 'i' && FirstChar != 'e') {
489       CurPPLexer->ParsingPreprocessorDirective = false;
490       // Restore comment saving mode.
491       if (CurLexer) CurLexer->resetExtendedTokenMode();
492       continue;
493     }
494 
495     // Get the identifier name without trigraphs or embedded newlines.  Note
496     // that we can't use Tok.getIdentifierInfo() because its lookup is disabled
497     // when skipping.
498     char DirectiveBuf[20];
499     StringRef Directive;
500     if (!Tok.needsCleaning() && RI.size() < 20) {
501       Directive = RI;
502     } else {
503       std::string DirectiveStr = getSpelling(Tok);
504       size_t IdLen = DirectiveStr.size();
505       if (IdLen >= 20) {
506         CurPPLexer->ParsingPreprocessorDirective = false;
507         // Restore comment saving mode.
508         if (CurLexer) CurLexer->resetExtendedTokenMode();
509         continue;
510       }
511       memcpy(DirectiveBuf, &DirectiveStr[0], IdLen);
512       Directive = StringRef(DirectiveBuf, IdLen);
513     }
514 
515     if (Directive.startswith("if")) {
516       StringRef Sub = Directive.substr(2);
517       if (Sub.empty() ||   // "if"
518           Sub == "def" ||   // "ifdef"
519           Sub == "ndef") {  // "ifndef"
520         // We know the entire #if/#ifdef/#ifndef block will be skipped, don't
521         // bother parsing the condition.
522         DiscardUntilEndOfDirective();
523         CurPPLexer->pushConditionalLevel(Tok.getLocation(), /*wasskipping*/true,
524                                        /*foundnonskip*/false,
525                                        /*foundelse*/false);
526       }
527     } else if (Directive[0] == 'e') {
528       StringRef Sub = Directive.substr(1);
529       if (Sub == "ndif") {  // "endif"
530         PPConditionalInfo CondInfo;
531         CondInfo.WasSkipping = true; // Silence bogus warning.
532         bool InCond = CurPPLexer->popConditionalLevel(CondInfo);
533         (void)InCond;  // Silence warning in no-asserts mode.
534         assert(!InCond && "Can't be skipping if not in a conditional!");
535 
536         // If we popped the outermost skipping block, we're done skipping!
537         if (!CondInfo.WasSkipping) {
538           // Restore the value of LexingRawMode so that trailing comments
539           // are handled correctly, if we've reached the outermost block.
540           CurPPLexer->LexingRawMode = false;
541           CheckEndOfDirective("endif");
542           CurPPLexer->LexingRawMode = true;
543           if (Callbacks)
544             Callbacks->Endif(Tok.getLocation(), CondInfo.IfLoc);
545           break;
546         } else {
547           DiscardUntilEndOfDirective();
548         }
549       } else if (Sub == "lse") { // "else".
550         // #else directive in a skipping conditional.  If not in some other
551         // skipping conditional, and if #else hasn't already been seen, enter it
552         // as a non-skipping conditional.
553         PPConditionalInfo &CondInfo = CurPPLexer->peekConditionalLevel();
554 
555         // If this is a #else with a #else before it, report the error.
556         if (CondInfo.FoundElse) Diag(Tok, diag::pp_err_else_after_else);
557 
558         // Note that we've seen a #else in this conditional.
559         CondInfo.FoundElse = true;
560 
561         // If the conditional is at the top level, and the #if block wasn't
562         // entered, enter the #else block now.
563         if (!CondInfo.WasSkipping && !CondInfo.FoundNonSkip) {
564           CondInfo.FoundNonSkip = true;
565           // Restore the value of LexingRawMode so that trailing comments
566           // are handled correctly.
567           CurPPLexer->LexingRawMode = false;
568           CheckEndOfDirective("else");
569           CurPPLexer->LexingRawMode = true;
570           if (Callbacks)
571             Callbacks->Else(Tok.getLocation(), CondInfo.IfLoc);
572           break;
573         } else {
574           DiscardUntilEndOfDirective();  // C99 6.10p4.
575         }
576       } else if (Sub == "lif") {  // "elif".
577         PPConditionalInfo &CondInfo = CurPPLexer->peekConditionalLevel();
578 
579         // If this is a #elif with a #else before it, report the error.
580         if (CondInfo.FoundElse) Diag(Tok, diag::pp_err_elif_after_else);
581 
582         // If this is in a skipping block or if we're already handled this #if
583         // block, don't bother parsing the condition.
584         if (CondInfo.WasSkipping || CondInfo.FoundNonSkip) {
585           DiscardUntilEndOfDirective();
586         } else {
587           // Restore the value of LexingRawMode so that identifiers are
588           // looked up, etc, inside the #elif expression.
589           assert(CurPPLexer->LexingRawMode && "We have to be skipping here!");
590           CurPPLexer->LexingRawMode = false;
591           IdentifierInfo *IfNDefMacro = nullptr;
592           DirectiveEvalResult DER = EvaluateDirectiveExpression(IfNDefMacro);
593           const bool CondValue = DER.Conditional;
594           CurPPLexer->LexingRawMode = true;
595           if (Callbacks) {
596             Callbacks->Elif(
597                 Tok.getLocation(), DER.ExprRange,
598                 (CondValue ? PPCallbacks::CVK_True : PPCallbacks::CVK_False),
599                 CondInfo.IfLoc);
600           }
601           // If this condition is true, enter it!
602           if (CondValue) {
603             CondInfo.FoundNonSkip = true;
604             break;
605           }
606         }
607       }
608     }
609 
610     CurPPLexer->ParsingPreprocessorDirective = false;
611     // Restore comment saving mode.
612     if (CurLexer) CurLexer->resetExtendedTokenMode();
613   }
614 
615   // Finally, if we are out of the conditional (saw an #endif or ran off the end
616   // of the file, just stop skipping and return to lexing whatever came after
617   // the #if block.
618   CurPPLexer->LexingRawMode = false;
619 
620   // The last skipped range isn't actually skipped yet if it's truncated
621   // by the end of the preamble; we'll resume parsing after the preamble.
622   if (Callbacks && (Tok.isNot(tok::eof) || !isRecordingPreamble()))
623     Callbacks->SourceRangeSkipped(
624         SourceRange(HashTokenLoc, CurPPLexer->getSourceLocation()),
625         Tok.getLocation());
626 }
627 
628 Module *Preprocessor::getModuleForLocation(SourceLocation Loc) {
629   if (!SourceMgr.isInMainFile(Loc)) {
630     // Try to determine the module of the include directive.
631     // FIXME: Look into directly passing the FileEntry from LookupFile instead.
632     FileID IDOfIncl = SourceMgr.getFileID(SourceMgr.getExpansionLoc(Loc));
633     if (const FileEntry *EntryOfIncl = SourceMgr.getFileEntryForID(IDOfIncl)) {
634       // The include comes from an included file.
635       return HeaderInfo.getModuleMap()
636           .findModuleForHeader(EntryOfIncl)
637           .getModule();
638     }
639   }
640 
641   // This is either in the main file or not in a file at all. It belongs
642   // to the current module, if there is one.
643   return getLangOpts().CurrentModule.empty()
644              ? nullptr
645              : HeaderInfo.lookupModule(getLangOpts().CurrentModule);
646 }
647 
648 const FileEntry *
649 Preprocessor::getModuleHeaderToIncludeForDiagnostics(SourceLocation IncLoc,
650                                                      Module *M,
651                                                      SourceLocation Loc) {
652   assert(M && "no module to include");
653 
654   // If the context is the global module fragment of some module, we never
655   // want to return that file; instead, we want the innermost include-guarded
656   // header that it included.
657   bool InGlobalModuleFragment = M->Kind == Module::GlobalModuleFragment;
658 
659   // If we have a module import syntax, we shouldn't include a header to
660   // make a particular module visible.
661   if ((getLangOpts().ObjC || getLangOpts().CPlusPlusModules ||
662        getLangOpts().ModulesTS) &&
663       !InGlobalModuleFragment)
664     return nullptr;
665 
666   Module *TopM = M->getTopLevelModule();
667   Module *IncM = getModuleForLocation(IncLoc);
668 
669   // Walk up through the include stack, looking through textual headers of M
670   // until we hit a non-textual header that we can #include. (We assume textual
671   // headers of a module with non-textual headers aren't meant to be used to
672   // import entities from the module.)
673   auto &SM = getSourceManager();
674   while (!Loc.isInvalid() && !SM.isInMainFile(Loc)) {
675     auto ID = SM.getFileID(SM.getExpansionLoc(Loc));
676     auto *FE = SM.getFileEntryForID(ID);
677     if (!FE)
678       break;
679 
680     if (InGlobalModuleFragment) {
681       if (getHeaderSearchInfo().isFileMultipleIncludeGuarded(FE))
682         return FE;
683       Loc = SM.getIncludeLoc(ID);
684       continue;
685     }
686 
687     bool InTextualHeader = false;
688     for (auto Header : HeaderInfo.getModuleMap().findAllModulesForHeader(FE)) {
689       if (!Header.getModule()->isSubModuleOf(TopM))
690         continue;
691 
692       if (!(Header.getRole() & ModuleMap::TextualHeader)) {
693         // If this is an accessible, non-textual header of M's top-level module
694         // that transitively includes the given location and makes the
695         // corresponding module visible, this is the thing to #include.
696         if (Header.isAccessibleFrom(IncM))
697           return FE;
698 
699         // It's in a private header; we can't #include it.
700         // FIXME: If there's a public header in some module that re-exports it,
701         // then we could suggest including that, but it's not clear that's the
702         // expected way to make this entity visible.
703         continue;
704       }
705 
706       InTextualHeader = true;
707     }
708 
709     if (!InTextualHeader)
710       break;
711 
712     Loc = SM.getIncludeLoc(ID);
713   }
714 
715   return nullptr;
716 }
717 
718 Optional<FileEntryRef> Preprocessor::LookupFile(
719     SourceLocation FilenameLoc, StringRef Filename, bool isAngled,
720     const DirectoryLookup *FromDir, const FileEntry *FromFile,
721     const DirectoryLookup *&CurDir, SmallVectorImpl<char> *SearchPath,
722     SmallVectorImpl<char> *RelativePath,
723     ModuleMap::KnownHeader *SuggestedModule, bool *IsMapped,
724     bool *IsFrameworkFound, bool SkipCache) {
725   Module *RequestingModule = getModuleForLocation(FilenameLoc);
726   bool RequestingModuleIsModuleInterface = !SourceMgr.isInMainFile(FilenameLoc);
727 
728   // If the header lookup mechanism may be relative to the current inclusion
729   // stack, record the parent #includes.
730   SmallVector<std::pair<const FileEntry *, const DirectoryEntry *>, 16>
731       Includers;
732   bool BuildSystemModule = false;
733   if (!FromDir && !FromFile) {
734     FileID FID = getCurrentFileLexer()->getFileID();
735     const FileEntry *FileEnt = SourceMgr.getFileEntryForID(FID);
736 
737     // If there is no file entry associated with this file, it must be the
738     // predefines buffer or the module includes buffer. Any other file is not
739     // lexed with a normal lexer, so it won't be scanned for preprocessor
740     // directives.
741     //
742     // If we have the predefines buffer, resolve #include references (which come
743     // from the -include command line argument) from the current working
744     // directory instead of relative to the main file.
745     //
746     // If we have the module includes buffer, resolve #include references (which
747     // come from header declarations in the module map) relative to the module
748     // map file.
749     if (!FileEnt) {
750       if (FID == SourceMgr.getMainFileID() && MainFileDir) {
751         Includers.push_back(std::make_pair(nullptr, MainFileDir));
752         BuildSystemModule = getCurrentModule()->IsSystem;
753       } else if ((FileEnt =
754                     SourceMgr.getFileEntryForID(SourceMgr.getMainFileID())))
755         Includers.push_back(std::make_pair(FileEnt, *FileMgr.getDirectory(".")));
756     } else {
757       Includers.push_back(std::make_pair(FileEnt, FileEnt->getDir()));
758     }
759 
760     // MSVC searches the current include stack from top to bottom for
761     // headers included by quoted include directives.
762     // See: http://msdn.microsoft.com/en-us/library/36k2cdd4.aspx
763     if (LangOpts.MSVCCompat && !isAngled) {
764       for (IncludeStackInfo &ISEntry : llvm::reverse(IncludeMacroStack)) {
765         if (IsFileLexer(ISEntry))
766           if ((FileEnt = ISEntry.ThePPLexer->getFileEntry()))
767             Includers.push_back(std::make_pair(FileEnt, FileEnt->getDir()));
768       }
769     }
770   }
771 
772   CurDir = CurDirLookup;
773 
774   if (FromFile) {
775     // We're supposed to start looking from after a particular file. Search
776     // the include path until we find that file or run out of files.
777     const DirectoryLookup *TmpCurDir = CurDir;
778     const DirectoryLookup *TmpFromDir = nullptr;
779     while (Optional<FileEntryRef> FE = HeaderInfo.LookupFile(
780                Filename, FilenameLoc, isAngled, TmpFromDir, TmpCurDir,
781                Includers, SearchPath, RelativePath, RequestingModule,
782                SuggestedModule, /*IsMapped=*/nullptr,
783                /*IsFrameworkFound=*/nullptr, SkipCache)) {
784       // Keep looking as if this file did a #include_next.
785       TmpFromDir = TmpCurDir;
786       ++TmpFromDir;
787       if (&FE->getFileEntry() == FromFile) {
788         // Found it.
789         FromDir = TmpFromDir;
790         CurDir = TmpCurDir;
791         break;
792       }
793     }
794   }
795 
796   // Do a standard file entry lookup.
797   Optional<FileEntryRef> FE = HeaderInfo.LookupFile(
798       Filename, FilenameLoc, isAngled, FromDir, CurDir, Includers, SearchPath,
799       RelativePath, RequestingModule, SuggestedModule, IsMapped,
800       IsFrameworkFound, SkipCache, BuildSystemModule);
801   if (FE) {
802     if (SuggestedModule && !LangOpts.AsmPreprocessor)
803       HeaderInfo.getModuleMap().diagnoseHeaderInclusion(
804           RequestingModule, RequestingModuleIsModuleInterface, FilenameLoc,
805           Filename, &FE->getFileEntry());
806     return FE;
807   }
808 
809   const FileEntry *CurFileEnt;
810   // Otherwise, see if this is a subframework header.  If so, this is relative
811   // to one of the headers on the #include stack.  Walk the list of the current
812   // headers on the #include stack and pass them to HeaderInfo.
813   if (IsFileLexer()) {
814     if ((CurFileEnt = CurPPLexer->getFileEntry())) {
815       if (Optional<FileEntryRef> FE = HeaderInfo.LookupSubframeworkHeader(
816               Filename, CurFileEnt, SearchPath, RelativePath, RequestingModule,
817               SuggestedModule)) {
818         if (SuggestedModule && !LangOpts.AsmPreprocessor)
819           HeaderInfo.getModuleMap().diagnoseHeaderInclusion(
820               RequestingModule, RequestingModuleIsModuleInterface, FilenameLoc,
821               Filename, &FE->getFileEntry());
822         return FE;
823       }
824     }
825   }
826 
827   for (IncludeStackInfo &ISEntry : llvm::reverse(IncludeMacroStack)) {
828     if (IsFileLexer(ISEntry)) {
829       if ((CurFileEnt = ISEntry.ThePPLexer->getFileEntry())) {
830         if (Optional<FileEntryRef> FE = HeaderInfo.LookupSubframeworkHeader(
831                 Filename, CurFileEnt, SearchPath, RelativePath,
832                 RequestingModule, SuggestedModule)) {
833           if (SuggestedModule && !LangOpts.AsmPreprocessor)
834             HeaderInfo.getModuleMap().diagnoseHeaderInclusion(
835                 RequestingModule, RequestingModuleIsModuleInterface,
836                 FilenameLoc, Filename, &FE->getFileEntry());
837           return FE;
838         }
839       }
840     }
841   }
842 
843   // Otherwise, we really couldn't find the file.
844   return None;
845 }
846 
847 //===----------------------------------------------------------------------===//
848 // Preprocessor Directive Handling.
849 //===----------------------------------------------------------------------===//
850 
851 class Preprocessor::ResetMacroExpansionHelper {
852 public:
853   ResetMacroExpansionHelper(Preprocessor *pp)
854     : PP(pp), save(pp->DisableMacroExpansion) {
855     if (pp->MacroExpansionInDirectivesOverride)
856       pp->DisableMacroExpansion = false;
857   }
858 
859   ~ResetMacroExpansionHelper() {
860     PP->DisableMacroExpansion = save;
861   }
862 
863 private:
864   Preprocessor *PP;
865   bool save;
866 };
867 
868 /// Process a directive while looking for the through header or a #pragma
869 /// hdrstop. The following directives are handled:
870 /// #include (to check if it is the through header)
871 /// #define (to warn about macros that don't match the PCH)
872 /// #pragma (to check for pragma hdrstop).
873 /// All other directives are completely discarded.
874 void Preprocessor::HandleSkippedDirectiveWhileUsingPCH(Token &Result,
875                                                        SourceLocation HashLoc) {
876   if (const IdentifierInfo *II = Result.getIdentifierInfo()) {
877     if (II->getPPKeywordID() == tok::pp_define) {
878       return HandleDefineDirective(Result,
879                                    /*ImmediatelyAfterHeaderGuard=*/false);
880     }
881     if (SkippingUntilPCHThroughHeader &&
882         II->getPPKeywordID() == tok::pp_include) {
883       return HandleIncludeDirective(HashLoc, Result);
884     }
885     if (SkippingUntilPragmaHdrStop && II->getPPKeywordID() == tok::pp_pragma) {
886       Lex(Result);
887       auto *II = Result.getIdentifierInfo();
888       if (II && II->getName() == "hdrstop")
889         return HandlePragmaHdrstop(Result);
890     }
891   }
892   DiscardUntilEndOfDirective();
893 }
894 
895 /// HandleDirective - This callback is invoked when the lexer sees a # token
896 /// at the start of a line.  This consumes the directive, modifies the
897 /// lexer/preprocessor state, and advances the lexer(s) so that the next token
898 /// read is the correct one.
899 void Preprocessor::HandleDirective(Token &Result) {
900   // FIXME: Traditional: # with whitespace before it not recognized by K&R?
901 
902   // We just parsed a # character at the start of a line, so we're in directive
903   // mode.  Tell the lexer this so any newlines we see will be converted into an
904   // EOD token (which terminates the directive).
905   CurPPLexer->ParsingPreprocessorDirective = true;
906   if (CurLexer) CurLexer->SetKeepWhitespaceMode(false);
907 
908   bool ImmediatelyAfterTopLevelIfndef =
909       CurPPLexer->MIOpt.getImmediatelyAfterTopLevelIfndef();
910   CurPPLexer->MIOpt.resetImmediatelyAfterTopLevelIfndef();
911 
912   ++NumDirectives;
913 
914   // We are about to read a token.  For the multiple-include optimization FA to
915   // work, we have to remember if we had read any tokens *before* this
916   // pp-directive.
917   bool ReadAnyTokensBeforeDirective =CurPPLexer->MIOpt.getHasReadAnyTokensVal();
918 
919   // Save the '#' token in case we need to return it later.
920   Token SavedHash = Result;
921 
922   // Read the next token, the directive flavor.  This isn't expanded due to
923   // C99 6.10.3p8.
924   LexUnexpandedToken(Result);
925 
926   // C99 6.10.3p11: Is this preprocessor directive in macro invocation?  e.g.:
927   //   #define A(x) #x
928   //   A(abc
929   //     #warning blah
930   //   def)
931   // If so, the user is relying on undefined behavior, emit a diagnostic. Do
932   // not support this for #include-like directives, since that can result in
933   // terrible diagnostics, and does not work in GCC.
934   if (InMacroArgs) {
935     if (IdentifierInfo *II = Result.getIdentifierInfo()) {
936       switch (II->getPPKeywordID()) {
937       case tok::pp_include:
938       case tok::pp_import:
939       case tok::pp_include_next:
940       case tok::pp___include_macros:
941       case tok::pp_pragma:
942         Diag(Result, diag::err_embedded_directive) << II->getName();
943         Diag(*ArgMacro, diag::note_macro_expansion_here)
944             << ArgMacro->getIdentifierInfo();
945         DiscardUntilEndOfDirective();
946         return;
947       default:
948         break;
949       }
950     }
951     Diag(Result, diag::ext_embedded_directive);
952   }
953 
954   // Temporarily enable macro expansion if set so
955   // and reset to previous state when returning from this function.
956   ResetMacroExpansionHelper helper(this);
957 
958   if (SkippingUntilPCHThroughHeader || SkippingUntilPragmaHdrStop)
959     return HandleSkippedDirectiveWhileUsingPCH(Result, SavedHash.getLocation());
960 
961   switch (Result.getKind()) {
962   case tok::eod:
963     return;   // null directive.
964   case tok::code_completion:
965     if (CodeComplete)
966       CodeComplete->CodeCompleteDirective(
967                                     CurPPLexer->getConditionalStackDepth() > 0);
968     setCodeCompletionReached();
969     return;
970   case tok::numeric_constant:  // # 7  GNU line marker directive.
971     if (getLangOpts().AsmPreprocessor)
972       break;  // # 4 is not a preprocessor directive in .S files.
973     return HandleDigitDirective(Result);
974   default:
975     IdentifierInfo *II = Result.getIdentifierInfo();
976     if (!II) break; // Not an identifier.
977 
978     // Ask what the preprocessor keyword ID is.
979     switch (II->getPPKeywordID()) {
980     default: break;
981     // C99 6.10.1 - Conditional Inclusion.
982     case tok::pp_if:
983       return HandleIfDirective(Result, SavedHash, ReadAnyTokensBeforeDirective);
984     case tok::pp_ifdef:
985       return HandleIfdefDirective(Result, SavedHash, false,
986                                   true /*not valid for miopt*/);
987     case tok::pp_ifndef:
988       return HandleIfdefDirective(Result, SavedHash, true,
989                                   ReadAnyTokensBeforeDirective);
990     case tok::pp_elif:
991       return HandleElifDirective(Result, SavedHash);
992     case tok::pp_else:
993       return HandleElseDirective(Result, SavedHash);
994     case tok::pp_endif:
995       return HandleEndifDirective(Result);
996 
997     // C99 6.10.2 - Source File Inclusion.
998     case tok::pp_include:
999       // Handle #include.
1000       return HandleIncludeDirective(SavedHash.getLocation(), Result);
1001     case tok::pp___include_macros:
1002       // Handle -imacros.
1003       return HandleIncludeMacrosDirective(SavedHash.getLocation(), Result);
1004 
1005     // C99 6.10.3 - Macro Replacement.
1006     case tok::pp_define:
1007       return HandleDefineDirective(Result, ImmediatelyAfterTopLevelIfndef);
1008     case tok::pp_undef:
1009       return HandleUndefDirective();
1010 
1011     // C99 6.10.4 - Line Control.
1012     case tok::pp_line:
1013       return HandleLineDirective();
1014 
1015     // C99 6.10.5 - Error Directive.
1016     case tok::pp_error:
1017       return HandleUserDiagnosticDirective(Result, false);
1018 
1019     // C99 6.10.6 - Pragma Directive.
1020     case tok::pp_pragma:
1021       return HandlePragmaDirective({PIK_HashPragma, SavedHash.getLocation()});
1022 
1023     // GNU Extensions.
1024     case tok::pp_import:
1025       return HandleImportDirective(SavedHash.getLocation(), Result);
1026     case tok::pp_include_next:
1027       return HandleIncludeNextDirective(SavedHash.getLocation(), Result);
1028 
1029     case tok::pp_warning:
1030       Diag(Result, diag::ext_pp_warning_directive);
1031       return HandleUserDiagnosticDirective(Result, true);
1032     case tok::pp_ident:
1033       return HandleIdentSCCSDirective(Result);
1034     case tok::pp_sccs:
1035       return HandleIdentSCCSDirective(Result);
1036     case tok::pp_assert:
1037       //isExtension = true;  // FIXME: implement #assert
1038       break;
1039     case tok::pp_unassert:
1040       //isExtension = true;  // FIXME: implement #unassert
1041       break;
1042 
1043     case tok::pp___public_macro:
1044       if (getLangOpts().Modules)
1045         return HandleMacroPublicDirective(Result);
1046       break;
1047 
1048     case tok::pp___private_macro:
1049       if (getLangOpts().Modules)
1050         return HandleMacroPrivateDirective();
1051       break;
1052     }
1053     break;
1054   }
1055 
1056   // If this is a .S file, treat unknown # directives as non-preprocessor
1057   // directives.  This is important because # may be a comment or introduce
1058   // various pseudo-ops.  Just return the # token and push back the following
1059   // token to be lexed next time.
1060   if (getLangOpts().AsmPreprocessor) {
1061     auto Toks = std::make_unique<Token[]>(2);
1062     // Return the # and the token after it.
1063     Toks[0] = SavedHash;
1064     Toks[1] = Result;
1065 
1066     // If the second token is a hashhash token, then we need to translate it to
1067     // unknown so the token lexer doesn't try to perform token pasting.
1068     if (Result.is(tok::hashhash))
1069       Toks[1].setKind(tok::unknown);
1070 
1071     // Enter this token stream so that we re-lex the tokens.  Make sure to
1072     // enable macro expansion, in case the token after the # is an identifier
1073     // that is expanded.
1074     EnterTokenStream(std::move(Toks), 2, false, /*IsReinject*/false);
1075     return;
1076   }
1077 
1078   // If we reached here, the preprocessing token is not valid!
1079   Diag(Result, diag::err_pp_invalid_directive);
1080 
1081   // Read the rest of the PP line.
1082   DiscardUntilEndOfDirective();
1083 
1084   // Okay, we're done parsing the directive.
1085 }
1086 
1087 /// GetLineValue - Convert a numeric token into an unsigned value, emitting
1088 /// Diagnostic DiagID if it is invalid, and returning the value in Val.
1089 static bool GetLineValue(Token &DigitTok, unsigned &Val,
1090                          unsigned DiagID, Preprocessor &PP,
1091                          bool IsGNULineDirective=false) {
1092   if (DigitTok.isNot(tok::numeric_constant)) {
1093     PP.Diag(DigitTok, DiagID);
1094 
1095     if (DigitTok.isNot(tok::eod))
1096       PP.DiscardUntilEndOfDirective();
1097     return true;
1098   }
1099 
1100   SmallString<64> IntegerBuffer;
1101   IntegerBuffer.resize(DigitTok.getLength());
1102   const char *DigitTokBegin = &IntegerBuffer[0];
1103   bool Invalid = false;
1104   unsigned ActualLength = PP.getSpelling(DigitTok, DigitTokBegin, &Invalid);
1105   if (Invalid)
1106     return true;
1107 
1108   // Verify that we have a simple digit-sequence, and compute the value.  This
1109   // is always a simple digit string computed in decimal, so we do this manually
1110   // here.
1111   Val = 0;
1112   for (unsigned i = 0; i != ActualLength; ++i) {
1113     // C++1y [lex.fcon]p1:
1114     //   Optional separating single quotes in a digit-sequence are ignored
1115     if (DigitTokBegin[i] == '\'')
1116       continue;
1117 
1118     if (!isDigit(DigitTokBegin[i])) {
1119       PP.Diag(PP.AdvanceToTokenCharacter(DigitTok.getLocation(), i),
1120               diag::err_pp_line_digit_sequence) << IsGNULineDirective;
1121       PP.DiscardUntilEndOfDirective();
1122       return true;
1123     }
1124 
1125     unsigned NextVal = Val*10+(DigitTokBegin[i]-'0');
1126     if (NextVal < Val) { // overflow.
1127       PP.Diag(DigitTok, DiagID);
1128       PP.DiscardUntilEndOfDirective();
1129       return true;
1130     }
1131     Val = NextVal;
1132   }
1133 
1134   if (DigitTokBegin[0] == '0' && Val)
1135     PP.Diag(DigitTok.getLocation(), diag::warn_pp_line_decimal)
1136       << IsGNULineDirective;
1137 
1138   return false;
1139 }
1140 
1141 /// Handle a \#line directive: C99 6.10.4.
1142 ///
1143 /// The two acceptable forms are:
1144 /// \verbatim
1145 ///   # line digit-sequence
1146 ///   # line digit-sequence "s-char-sequence"
1147 /// \endverbatim
1148 void Preprocessor::HandleLineDirective() {
1149   // Read the line # and string argument.  Per C99 6.10.4p5, these tokens are
1150   // expanded.
1151   Token DigitTok;
1152   Lex(DigitTok);
1153 
1154   // Validate the number and convert it to an unsigned.
1155   unsigned LineNo;
1156   if (GetLineValue(DigitTok, LineNo, diag::err_pp_line_requires_integer,*this))
1157     return;
1158 
1159   if (LineNo == 0)
1160     Diag(DigitTok, diag::ext_pp_line_zero);
1161 
1162   // Enforce C99 6.10.4p3: "The digit sequence shall not specify ... a
1163   // number greater than 2147483647".  C90 requires that the line # be <= 32767.
1164   unsigned LineLimit = 32768U;
1165   if (LangOpts.C99 || LangOpts.CPlusPlus11)
1166     LineLimit = 2147483648U;
1167   if (LineNo >= LineLimit)
1168     Diag(DigitTok, diag::ext_pp_line_too_big) << LineLimit;
1169   else if (LangOpts.CPlusPlus11 && LineNo >= 32768U)
1170     Diag(DigitTok, diag::warn_cxx98_compat_pp_line_too_big);
1171 
1172   int FilenameID = -1;
1173   Token StrTok;
1174   Lex(StrTok);
1175 
1176   // If the StrTok is "eod", then it wasn't present.  Otherwise, it must be a
1177   // string followed by eod.
1178   if (StrTok.is(tok::eod))
1179     ; // ok
1180   else if (StrTok.isNot(tok::string_literal)) {
1181     Diag(StrTok, diag::err_pp_line_invalid_filename);
1182     DiscardUntilEndOfDirective();
1183     return;
1184   } else if (StrTok.hasUDSuffix()) {
1185     Diag(StrTok, diag::err_invalid_string_udl);
1186     DiscardUntilEndOfDirective();
1187     return;
1188   } else {
1189     // Parse and validate the string, converting it into a unique ID.
1190     StringLiteralParser Literal(StrTok, *this);
1191     assert(Literal.isAscii() && "Didn't allow wide strings in");
1192     if (Literal.hadError) {
1193       DiscardUntilEndOfDirective();
1194       return;
1195     }
1196     if (Literal.Pascal) {
1197       Diag(StrTok, diag::err_pp_linemarker_invalid_filename);
1198       DiscardUntilEndOfDirective();
1199       return;
1200     }
1201     FilenameID = SourceMgr.getLineTableFilenameID(Literal.GetString());
1202 
1203     // Verify that there is nothing after the string, other than EOD.  Because
1204     // of C99 6.10.4p5, macros that expand to empty tokens are ok.
1205     CheckEndOfDirective("line", true);
1206   }
1207 
1208   // Take the file kind of the file containing the #line directive. #line
1209   // directives are often used for generated sources from the same codebase, so
1210   // the new file should generally be classified the same way as the current
1211   // file. This is visible in GCC's pre-processed output, which rewrites #line
1212   // to GNU line markers.
1213   SrcMgr::CharacteristicKind FileKind =
1214       SourceMgr.getFileCharacteristic(DigitTok.getLocation());
1215 
1216   SourceMgr.AddLineNote(DigitTok.getLocation(), LineNo, FilenameID, false,
1217                         false, FileKind);
1218 
1219   if (Callbacks)
1220     Callbacks->FileChanged(CurPPLexer->getSourceLocation(),
1221                            PPCallbacks::RenameFile, FileKind);
1222 }
1223 
1224 /// ReadLineMarkerFlags - Parse and validate any flags at the end of a GNU line
1225 /// marker directive.
1226 static bool ReadLineMarkerFlags(bool &IsFileEntry, bool &IsFileExit,
1227                                 SrcMgr::CharacteristicKind &FileKind,
1228                                 Preprocessor &PP) {
1229   unsigned FlagVal;
1230   Token FlagTok;
1231   PP.Lex(FlagTok);
1232   if (FlagTok.is(tok::eod)) return false;
1233   if (GetLineValue(FlagTok, FlagVal, diag::err_pp_linemarker_invalid_flag, PP))
1234     return true;
1235 
1236   if (FlagVal == 1) {
1237     IsFileEntry = true;
1238 
1239     PP.Lex(FlagTok);
1240     if (FlagTok.is(tok::eod)) return false;
1241     if (GetLineValue(FlagTok, FlagVal, diag::err_pp_linemarker_invalid_flag,PP))
1242       return true;
1243   } else if (FlagVal == 2) {
1244     IsFileExit = true;
1245 
1246     SourceManager &SM = PP.getSourceManager();
1247     // If we are leaving the current presumed file, check to make sure the
1248     // presumed include stack isn't empty!
1249     FileID CurFileID =
1250       SM.getDecomposedExpansionLoc(FlagTok.getLocation()).first;
1251     PresumedLoc PLoc = SM.getPresumedLoc(FlagTok.getLocation());
1252     if (PLoc.isInvalid())
1253       return true;
1254 
1255     // If there is no include loc (main file) or if the include loc is in a
1256     // different physical file, then we aren't in a "1" line marker flag region.
1257     SourceLocation IncLoc = PLoc.getIncludeLoc();
1258     if (IncLoc.isInvalid() ||
1259         SM.getDecomposedExpansionLoc(IncLoc).first != CurFileID) {
1260       PP.Diag(FlagTok, diag::err_pp_linemarker_invalid_pop);
1261       PP.DiscardUntilEndOfDirective();
1262       return true;
1263     }
1264 
1265     PP.Lex(FlagTok);
1266     if (FlagTok.is(tok::eod)) return false;
1267     if (GetLineValue(FlagTok, FlagVal, diag::err_pp_linemarker_invalid_flag,PP))
1268       return true;
1269   }
1270 
1271   // We must have 3 if there are still flags.
1272   if (FlagVal != 3) {
1273     PP.Diag(FlagTok, diag::err_pp_linemarker_invalid_flag);
1274     PP.DiscardUntilEndOfDirective();
1275     return true;
1276   }
1277 
1278   FileKind = SrcMgr::C_System;
1279 
1280   PP.Lex(FlagTok);
1281   if (FlagTok.is(tok::eod)) return false;
1282   if (GetLineValue(FlagTok, FlagVal, diag::err_pp_linemarker_invalid_flag, PP))
1283     return true;
1284 
1285   // We must have 4 if there is yet another flag.
1286   if (FlagVal != 4) {
1287     PP.Diag(FlagTok, diag::err_pp_linemarker_invalid_flag);
1288     PP.DiscardUntilEndOfDirective();
1289     return true;
1290   }
1291 
1292   FileKind = SrcMgr::C_ExternCSystem;
1293 
1294   PP.Lex(FlagTok);
1295   if (FlagTok.is(tok::eod)) return false;
1296 
1297   // There are no more valid flags here.
1298   PP.Diag(FlagTok, diag::err_pp_linemarker_invalid_flag);
1299   PP.DiscardUntilEndOfDirective();
1300   return true;
1301 }
1302 
1303 /// HandleDigitDirective - Handle a GNU line marker directive, whose syntax is
1304 /// one of the following forms:
1305 ///
1306 ///     # 42
1307 ///     # 42 "file" ('1' | '2')?
1308 ///     # 42 "file" ('1' | '2')? '3' '4'?
1309 ///
1310 void Preprocessor::HandleDigitDirective(Token &DigitTok) {
1311   // Validate the number and convert it to an unsigned.  GNU does not have a
1312   // line # limit other than it fit in 32-bits.
1313   unsigned LineNo;
1314   if (GetLineValue(DigitTok, LineNo, diag::err_pp_linemarker_requires_integer,
1315                    *this, true))
1316     return;
1317 
1318   Token StrTok;
1319   Lex(StrTok);
1320 
1321   bool IsFileEntry = false, IsFileExit = false;
1322   int FilenameID = -1;
1323   SrcMgr::CharacteristicKind FileKind = SrcMgr::C_User;
1324 
1325   // If the StrTok is "eod", then it wasn't present.  Otherwise, it must be a
1326   // string followed by eod.
1327   if (StrTok.is(tok::eod)) {
1328     // Treat this like "#line NN", which doesn't change file characteristics.
1329     FileKind = SourceMgr.getFileCharacteristic(DigitTok.getLocation());
1330   } else if (StrTok.isNot(tok::string_literal)) {
1331     Diag(StrTok, diag::err_pp_linemarker_invalid_filename);
1332     DiscardUntilEndOfDirective();
1333     return;
1334   } else if (StrTok.hasUDSuffix()) {
1335     Diag(StrTok, diag::err_invalid_string_udl);
1336     DiscardUntilEndOfDirective();
1337     return;
1338   } else {
1339     // Parse and validate the string, converting it into a unique ID.
1340     StringLiteralParser Literal(StrTok, *this);
1341     assert(Literal.isAscii() && "Didn't allow wide strings in");
1342     if (Literal.hadError) {
1343       DiscardUntilEndOfDirective();
1344       return;
1345     }
1346     if (Literal.Pascal) {
1347       Diag(StrTok, diag::err_pp_linemarker_invalid_filename);
1348       DiscardUntilEndOfDirective();
1349       return;
1350     }
1351     FilenameID = SourceMgr.getLineTableFilenameID(Literal.GetString());
1352 
1353     // If a filename was present, read any flags that are present.
1354     if (ReadLineMarkerFlags(IsFileEntry, IsFileExit, FileKind, *this))
1355       return;
1356   }
1357 
1358   // Create a line note with this information.
1359   SourceMgr.AddLineNote(DigitTok.getLocation(), LineNo, FilenameID, IsFileEntry,
1360                         IsFileExit, FileKind);
1361 
1362   // If the preprocessor has callbacks installed, notify them of the #line
1363   // change.  This is used so that the line marker comes out in -E mode for
1364   // example.
1365   if (Callbacks) {
1366     PPCallbacks::FileChangeReason Reason = PPCallbacks::RenameFile;
1367     if (IsFileEntry)
1368       Reason = PPCallbacks::EnterFile;
1369     else if (IsFileExit)
1370       Reason = PPCallbacks::ExitFile;
1371 
1372     Callbacks->FileChanged(CurPPLexer->getSourceLocation(), Reason, FileKind);
1373   }
1374 }
1375 
1376 /// HandleUserDiagnosticDirective - Handle a #warning or #error directive.
1377 ///
1378 void Preprocessor::HandleUserDiagnosticDirective(Token &Tok,
1379                                                  bool isWarning) {
1380   // Read the rest of the line raw.  We do this because we don't want macros
1381   // to be expanded and we don't require that the tokens be valid preprocessing
1382   // tokens.  For example, this is allowed: "#warning `   'foo".  GCC does
1383   // collapse multiple consecutive white space between tokens, but this isn't
1384   // specified by the standard.
1385   SmallString<128> Message;
1386   CurLexer->ReadToEndOfLine(&Message);
1387 
1388   // Find the first non-whitespace character, so that we can make the
1389   // diagnostic more succinct.
1390   StringRef Msg = StringRef(Message).ltrim(' ');
1391 
1392   if (isWarning)
1393     Diag(Tok, diag::pp_hash_warning) << Msg;
1394   else
1395     Diag(Tok, diag::err_pp_hash_error) << Msg;
1396 }
1397 
1398 /// HandleIdentSCCSDirective - Handle a #ident/#sccs directive.
1399 ///
1400 void Preprocessor::HandleIdentSCCSDirective(Token &Tok) {
1401   // Yes, this directive is an extension.
1402   Diag(Tok, diag::ext_pp_ident_directive);
1403 
1404   // Read the string argument.
1405   Token StrTok;
1406   Lex(StrTok);
1407 
1408   // If the token kind isn't a string, it's a malformed directive.
1409   if (StrTok.isNot(tok::string_literal) &&
1410       StrTok.isNot(tok::wide_string_literal)) {
1411     Diag(StrTok, diag::err_pp_malformed_ident);
1412     if (StrTok.isNot(tok::eod))
1413       DiscardUntilEndOfDirective();
1414     return;
1415   }
1416 
1417   if (StrTok.hasUDSuffix()) {
1418     Diag(StrTok, diag::err_invalid_string_udl);
1419     DiscardUntilEndOfDirective();
1420     return;
1421   }
1422 
1423   // Verify that there is nothing after the string, other than EOD.
1424   CheckEndOfDirective("ident");
1425 
1426   if (Callbacks) {
1427     bool Invalid = false;
1428     std::string Str = getSpelling(StrTok, &Invalid);
1429     if (!Invalid)
1430       Callbacks->Ident(Tok.getLocation(), Str);
1431   }
1432 }
1433 
1434 /// Handle a #public directive.
1435 void Preprocessor::HandleMacroPublicDirective(Token &Tok) {
1436   Token MacroNameTok;
1437   ReadMacroName(MacroNameTok, MU_Undef);
1438 
1439   // Error reading macro name?  If so, diagnostic already issued.
1440   if (MacroNameTok.is(tok::eod))
1441     return;
1442 
1443   // Check to see if this is the last token on the #__public_macro line.
1444   CheckEndOfDirective("__public_macro");
1445 
1446   IdentifierInfo *II = MacroNameTok.getIdentifierInfo();
1447   // Okay, we finally have a valid identifier to undef.
1448   MacroDirective *MD = getLocalMacroDirective(II);
1449 
1450   // If the macro is not defined, this is an error.
1451   if (!MD) {
1452     Diag(MacroNameTok, diag::err_pp_visibility_non_macro) << II;
1453     return;
1454   }
1455 
1456   // Note that this macro has now been exported.
1457   appendMacroDirective(II, AllocateVisibilityMacroDirective(
1458                                 MacroNameTok.getLocation(), /*isPublic=*/true));
1459 }
1460 
1461 /// Handle a #private directive.
1462 void Preprocessor::HandleMacroPrivateDirective() {
1463   Token MacroNameTok;
1464   ReadMacroName(MacroNameTok, MU_Undef);
1465 
1466   // Error reading macro name?  If so, diagnostic already issued.
1467   if (MacroNameTok.is(tok::eod))
1468     return;
1469 
1470   // Check to see if this is the last token on the #__private_macro line.
1471   CheckEndOfDirective("__private_macro");
1472 
1473   IdentifierInfo *II = MacroNameTok.getIdentifierInfo();
1474   // Okay, we finally have a valid identifier to undef.
1475   MacroDirective *MD = getLocalMacroDirective(II);
1476 
1477   // If the macro is not defined, this is an error.
1478   if (!MD) {
1479     Diag(MacroNameTok, diag::err_pp_visibility_non_macro) << II;
1480     return;
1481   }
1482 
1483   // Note that this macro has now been marked private.
1484   appendMacroDirective(II, AllocateVisibilityMacroDirective(
1485                                MacroNameTok.getLocation(), /*isPublic=*/false));
1486 }
1487 
1488 //===----------------------------------------------------------------------===//
1489 // Preprocessor Include Directive Handling.
1490 //===----------------------------------------------------------------------===//
1491 
1492 /// GetIncludeFilenameSpelling - Turn the specified lexer token into a fully
1493 /// checked and spelled filename, e.g. as an operand of \#include. This returns
1494 /// true if the input filename was in <>'s or false if it were in ""'s.  The
1495 /// caller is expected to provide a buffer that is large enough to hold the
1496 /// spelling of the filename, but is also expected to handle the case when
1497 /// this method decides to use a different buffer.
1498 bool Preprocessor::GetIncludeFilenameSpelling(SourceLocation Loc,
1499                                               StringRef &Buffer) {
1500   // Get the text form of the filename.
1501   assert(!Buffer.empty() && "Can't have tokens with empty spellings!");
1502 
1503   // FIXME: Consider warning on some of the cases described in C11 6.4.7/3 and
1504   // C++20 [lex.header]/2:
1505   //
1506   // If `"`, `'`, `\`, `/*`, or `//` appears in a header-name, then
1507   //   in C: behavior is undefined
1508   //   in C++: program is conditionally-supported with implementation-defined
1509   //           semantics
1510 
1511   // Make sure the filename is <x> or "x".
1512   bool isAngled;
1513   if (Buffer[0] == '<') {
1514     if (Buffer.back() != '>') {
1515       Diag(Loc, diag::err_pp_expects_filename);
1516       Buffer = StringRef();
1517       return true;
1518     }
1519     isAngled = true;
1520   } else if (Buffer[0] == '"') {
1521     if (Buffer.back() != '"') {
1522       Diag(Loc, diag::err_pp_expects_filename);
1523       Buffer = StringRef();
1524       return true;
1525     }
1526     isAngled = false;
1527   } else {
1528     Diag(Loc, diag::err_pp_expects_filename);
1529     Buffer = StringRef();
1530     return true;
1531   }
1532 
1533   // Diagnose #include "" as invalid.
1534   if (Buffer.size() <= 2) {
1535     Diag(Loc, diag::err_pp_empty_filename);
1536     Buffer = StringRef();
1537     return true;
1538   }
1539 
1540   // Skip the brackets.
1541   Buffer = Buffer.substr(1, Buffer.size()-2);
1542   return isAngled;
1543 }
1544 
1545 /// Push a token onto the token stream containing an annotation.
1546 void Preprocessor::EnterAnnotationToken(SourceRange Range,
1547                                         tok::TokenKind Kind,
1548                                         void *AnnotationVal) {
1549   // FIXME: Produce this as the current token directly, rather than
1550   // allocating a new token for it.
1551   auto Tok = std::make_unique<Token[]>(1);
1552   Tok[0].startToken();
1553   Tok[0].setKind(Kind);
1554   Tok[0].setLocation(Range.getBegin());
1555   Tok[0].setAnnotationEndLoc(Range.getEnd());
1556   Tok[0].setAnnotationValue(AnnotationVal);
1557   EnterTokenStream(std::move(Tok), 1, true, /*IsReinject*/ false);
1558 }
1559 
1560 /// Produce a diagnostic informing the user that a #include or similar
1561 /// was implicitly treated as a module import.
1562 static void diagnoseAutoModuleImport(
1563     Preprocessor &PP, SourceLocation HashLoc, Token &IncludeTok,
1564     ArrayRef<std::pair<IdentifierInfo *, SourceLocation>> Path,
1565     SourceLocation PathEnd) {
1566   StringRef ImportKeyword;
1567   if (PP.getLangOpts().ObjC)
1568     ImportKeyword = "@import";
1569   else if (PP.getLangOpts().ModulesTS || PP.getLangOpts().CPlusPlusModules)
1570     ImportKeyword = "import";
1571   else
1572     return; // no import syntax available
1573 
1574   SmallString<128> PathString;
1575   for (size_t I = 0, N = Path.size(); I != N; ++I) {
1576     if (I)
1577       PathString += '.';
1578     PathString += Path[I].first->getName();
1579   }
1580   int IncludeKind = 0;
1581 
1582   switch (IncludeTok.getIdentifierInfo()->getPPKeywordID()) {
1583   case tok::pp_include:
1584     IncludeKind = 0;
1585     break;
1586 
1587   case tok::pp_import:
1588     IncludeKind = 1;
1589     break;
1590 
1591   case tok::pp_include_next:
1592     IncludeKind = 2;
1593     break;
1594 
1595   case tok::pp___include_macros:
1596     IncludeKind = 3;
1597     break;
1598 
1599   default:
1600     llvm_unreachable("unknown include directive kind");
1601   }
1602 
1603   CharSourceRange ReplaceRange(SourceRange(HashLoc, PathEnd),
1604                                /*IsTokenRange=*/false);
1605   PP.Diag(HashLoc, diag::warn_auto_module_import)
1606       << IncludeKind << PathString
1607       << FixItHint::CreateReplacement(
1608              ReplaceRange, (ImportKeyword + " " + PathString + ";").str());
1609 }
1610 
1611 // Given a vector of path components and a string containing the real
1612 // path to the file, build a properly-cased replacement in the vector,
1613 // and return true if the replacement should be suggested.
1614 static bool trySimplifyPath(SmallVectorImpl<StringRef> &Components,
1615                             StringRef RealPathName) {
1616   auto RealPathComponentIter = llvm::sys::path::rbegin(RealPathName);
1617   auto RealPathComponentEnd = llvm::sys::path::rend(RealPathName);
1618   int Cnt = 0;
1619   bool SuggestReplacement = false;
1620   // Below is a best-effort to handle ".." in paths. It is admittedly
1621   // not 100% correct in the presence of symlinks.
1622   for (auto &Component : llvm::reverse(Components)) {
1623     if ("." == Component) {
1624     } else if (".." == Component) {
1625       ++Cnt;
1626     } else if (Cnt) {
1627       --Cnt;
1628     } else if (RealPathComponentIter != RealPathComponentEnd) {
1629       if (Component != *RealPathComponentIter) {
1630         // If these path components differ by more than just case, then we
1631         // may be looking at symlinked paths. Bail on this diagnostic to avoid
1632         // noisy false positives.
1633         SuggestReplacement = RealPathComponentIter->equals_lower(Component);
1634         if (!SuggestReplacement)
1635           break;
1636         Component = *RealPathComponentIter;
1637       }
1638       ++RealPathComponentIter;
1639     }
1640   }
1641   return SuggestReplacement;
1642 }
1643 
1644 bool Preprocessor::checkModuleIsAvailable(const LangOptions &LangOpts,
1645                                           const TargetInfo &TargetInfo,
1646                                           DiagnosticsEngine &Diags, Module *M) {
1647   Module::Requirement Requirement;
1648   Module::UnresolvedHeaderDirective MissingHeader;
1649   Module *ShadowingModule = nullptr;
1650   if (M->isAvailable(LangOpts, TargetInfo, Requirement, MissingHeader,
1651                      ShadowingModule))
1652     return false;
1653 
1654   if (MissingHeader.FileNameLoc.isValid()) {
1655     Diags.Report(MissingHeader.FileNameLoc, diag::err_module_header_missing)
1656         << MissingHeader.IsUmbrella << MissingHeader.FileName;
1657   } else if (ShadowingModule) {
1658     Diags.Report(M->DefinitionLoc, diag::err_module_shadowed) << M->Name;
1659     Diags.Report(ShadowingModule->DefinitionLoc,
1660                  diag::note_previous_definition);
1661   } else {
1662     // FIXME: Track the location at which the requirement was specified, and
1663     // use it here.
1664     Diags.Report(M->DefinitionLoc, diag::err_module_unavailable)
1665         << M->getFullModuleName() << Requirement.second << Requirement.first;
1666   }
1667   return true;
1668 }
1669 
1670 /// HandleIncludeDirective - The "\#include" tokens have just been read, read
1671 /// the file to be included from the lexer, then include it!  This is a common
1672 /// routine with functionality shared between \#include, \#include_next and
1673 /// \#import.  LookupFrom is set when this is a \#include_next directive, it
1674 /// specifies the file to start searching from.
1675 void Preprocessor::HandleIncludeDirective(SourceLocation HashLoc,
1676                                           Token &IncludeTok,
1677                                           const DirectoryLookup *LookupFrom,
1678                                           const FileEntry *LookupFromFile) {
1679   Token FilenameTok;
1680   if (LexHeaderName(FilenameTok))
1681     return;
1682 
1683   if (FilenameTok.isNot(tok::header_name)) {
1684     Diag(FilenameTok.getLocation(), diag::err_pp_expects_filename);
1685     if (FilenameTok.isNot(tok::eod))
1686       DiscardUntilEndOfDirective();
1687     return;
1688   }
1689 
1690   // Verify that there is nothing after the filename, other than EOD.  Note
1691   // that we allow macros that expand to nothing after the filename, because
1692   // this falls into the category of "#include pp-tokens new-line" specified
1693   // in C99 6.10.2p4.
1694   SourceLocation EndLoc =
1695       CheckEndOfDirective(IncludeTok.getIdentifierInfo()->getNameStart(), true);
1696 
1697   auto Action = HandleHeaderIncludeOrImport(HashLoc, IncludeTok, FilenameTok,
1698                                             EndLoc, LookupFrom, LookupFromFile);
1699   switch (Action.Kind) {
1700   case ImportAction::None:
1701   case ImportAction::SkippedModuleImport:
1702     break;
1703   case ImportAction::ModuleBegin:
1704     EnterAnnotationToken(SourceRange(HashLoc, EndLoc),
1705                          tok::annot_module_begin, Action.ModuleForHeader);
1706     break;
1707   case ImportAction::ModuleImport:
1708     EnterAnnotationToken(SourceRange(HashLoc, EndLoc),
1709                          tok::annot_module_include, Action.ModuleForHeader);
1710     break;
1711   }
1712 }
1713 
1714 Optional<FileEntryRef> Preprocessor::LookupHeaderIncludeOrImport(
1715     const DirectoryLookup *&CurDir, StringRef Filename,
1716     SourceLocation FilenameLoc, CharSourceRange FilenameRange,
1717     const Token &FilenameTok, bool &IsFrameworkFound, bool IsImportDecl,
1718     bool &IsMapped, const DirectoryLookup *LookupFrom,
1719     const FileEntry *LookupFromFile, StringRef LookupFilename,
1720     SmallVectorImpl<char> &RelativePath, SmallVectorImpl<char> &SearchPath,
1721     ModuleMap::KnownHeader &SuggestedModule, bool isAngled) {
1722   Optional<FileEntryRef> File = LookupFile(
1723       FilenameLoc, LookupFilename,
1724       isAngled, LookupFrom, LookupFromFile, CurDir,
1725       Callbacks ? &SearchPath : nullptr, Callbacks ? &RelativePath : nullptr,
1726       &SuggestedModule, &IsMapped, &IsFrameworkFound);
1727   if (File)
1728     return File;
1729 
1730   if (Callbacks) {
1731     // Give the clients a chance to recover.
1732     SmallString<128> RecoveryPath;
1733     if (Callbacks->FileNotFound(Filename, RecoveryPath)) {
1734       if (auto DE = FileMgr.getOptionalDirectoryRef(RecoveryPath)) {
1735         // Add the recovery path to the list of search paths.
1736         DirectoryLookup DL(*DE, SrcMgr::C_User, false);
1737         HeaderInfo.AddSearchPath(DL, isAngled);
1738 
1739         // Try the lookup again, skipping the cache.
1740         Optional<FileEntryRef> File = LookupFile(
1741             FilenameLoc,
1742             LookupFilename, isAngled,
1743             LookupFrom, LookupFromFile, CurDir, nullptr, nullptr,
1744             &SuggestedModule, &IsMapped, /*IsFrameworkFound=*/nullptr,
1745             /*SkipCache*/ true);
1746         if (File)
1747           return File;
1748       }
1749     }
1750   }
1751 
1752   if (SuppressIncludeNotFoundError)
1753     return None;
1754 
1755   // If the file could not be located and it was included via angle
1756   // brackets, we can attempt a lookup as though it were a quoted path to
1757   // provide the user with a possible fixit.
1758   if (isAngled) {
1759     Optional<FileEntryRef> File = LookupFile(
1760         FilenameLoc, LookupFilename,
1761         false, LookupFrom, LookupFromFile, CurDir,
1762         Callbacks ? &SearchPath : nullptr, Callbacks ? &RelativePath : nullptr,
1763         &SuggestedModule, &IsMapped,
1764         /*IsFrameworkFound=*/nullptr);
1765     if (File) {
1766       Diag(FilenameTok, diag::err_pp_file_not_found_angled_include_not_fatal)
1767           << Filename << IsImportDecl
1768           << FixItHint::CreateReplacement(FilenameRange,
1769                                           "\"" + Filename.str() + "\"");
1770       return File;
1771     }
1772   }
1773 
1774   // Check for likely typos due to leading or trailing non-isAlphanumeric
1775   // characters
1776   StringRef OriginalFilename = Filename;
1777   if (LangOpts.SpellChecking) {
1778     // A heuristic to correct a typo file name by removing leading and
1779     // trailing non-isAlphanumeric characters.
1780     auto CorrectTypoFilename = [](llvm::StringRef Filename) {
1781       Filename = Filename.drop_until(isAlphanumeric);
1782       while (!Filename.empty() && !isAlphanumeric(Filename.back())) {
1783         Filename = Filename.drop_back();
1784       }
1785       return Filename;
1786     };
1787     StringRef TypoCorrectionName = CorrectTypoFilename(Filename);
1788 
1789 #ifndef _WIN32
1790     // Normalize slashes when compiling with -fms-extensions on non-Windows.
1791     // This is unnecessary on Windows since the filesystem there handles
1792     // backslashes.
1793     SmallString<128> NormalizedTypoCorrectionPath;
1794     if (LangOpts.MicrosoftExt) {
1795       NormalizedTypoCorrectionPath = TypoCorrectionName;
1796       llvm::sys::path::native(NormalizedTypoCorrectionPath);
1797       TypoCorrectionName = NormalizedTypoCorrectionPath;
1798     }
1799 #endif
1800 
1801     Optional<FileEntryRef> File = LookupFile(
1802         FilenameLoc, TypoCorrectionName, isAngled, LookupFrom, LookupFromFile,
1803         CurDir, Callbacks ? &SearchPath : nullptr,
1804         Callbacks ? &RelativePath : nullptr, &SuggestedModule, &IsMapped,
1805         /*IsFrameworkFound=*/nullptr);
1806     if (File) {
1807       auto Hint =
1808           isAngled ? FixItHint::CreateReplacement(
1809                          FilenameRange, "<" + TypoCorrectionName.str() + ">")
1810                    : FixItHint::CreateReplacement(
1811                          FilenameRange, "\"" + TypoCorrectionName.str() + "\"");
1812       Diag(FilenameTok, diag::err_pp_file_not_found_typo_not_fatal)
1813           << OriginalFilename << TypoCorrectionName << Hint;
1814       // We found the file, so set the Filename to the name after typo
1815       // correction.
1816       Filename = TypoCorrectionName;
1817       return File;
1818     }
1819   }
1820 
1821   // If the file is still not found, just go with the vanilla diagnostic
1822   assert(!File.hasValue() && "expected missing file");
1823   Diag(FilenameTok, diag::err_pp_file_not_found)
1824       << OriginalFilename << FilenameRange;
1825   if (IsFrameworkFound) {
1826     size_t SlashPos = OriginalFilename.find('/');
1827     assert(SlashPos != StringRef::npos &&
1828            "Include with framework name should have '/' in the filename");
1829     StringRef FrameworkName = OriginalFilename.substr(0, SlashPos);
1830     FrameworkCacheEntry &CacheEntry =
1831         HeaderInfo.LookupFrameworkCache(FrameworkName);
1832     assert(CacheEntry.Directory && "Found framework should be in cache");
1833     Diag(FilenameTok, diag::note_pp_framework_without_header)
1834         << OriginalFilename.substr(SlashPos + 1) << FrameworkName
1835         << CacheEntry.Directory->getName();
1836   }
1837 
1838   return None;
1839 }
1840 
1841 /// Handle either a #include-like directive or an import declaration that names
1842 /// a header file.
1843 ///
1844 /// \param HashLoc The location of the '#' token for an include, or
1845 ///        SourceLocation() for an import declaration.
1846 /// \param IncludeTok The include / include_next / import token.
1847 /// \param FilenameTok The header-name token.
1848 /// \param EndLoc The location at which any imported macros become visible.
1849 /// \param LookupFrom For #include_next, the starting directory for the
1850 ///        directory lookup.
1851 /// \param LookupFromFile For #include_next, the starting file for the directory
1852 ///        lookup.
1853 Preprocessor::ImportAction Preprocessor::HandleHeaderIncludeOrImport(
1854     SourceLocation HashLoc, Token &IncludeTok, Token &FilenameTok,
1855     SourceLocation EndLoc, const DirectoryLookup *LookupFrom,
1856     const FileEntry *LookupFromFile) {
1857   SmallString<128> FilenameBuffer;
1858   StringRef Filename = getSpelling(FilenameTok, FilenameBuffer);
1859   SourceLocation CharEnd = FilenameTok.getEndLoc();
1860 
1861   CharSourceRange FilenameRange
1862     = CharSourceRange::getCharRange(FilenameTok.getLocation(), CharEnd);
1863   StringRef OriginalFilename = Filename;
1864   bool isAngled =
1865     GetIncludeFilenameSpelling(FilenameTok.getLocation(), Filename);
1866 
1867   // If GetIncludeFilenameSpelling set the start ptr to null, there was an
1868   // error.
1869   if (Filename.empty())
1870     return {ImportAction::None};
1871 
1872   bool IsImportDecl = HashLoc.isInvalid();
1873   SourceLocation StartLoc = IsImportDecl ? IncludeTok.getLocation() : HashLoc;
1874 
1875   // Complain about attempts to #include files in an audit pragma.
1876   if (PragmaARCCFCodeAuditedInfo.second.isValid()) {
1877     Diag(StartLoc, diag::err_pp_include_in_arc_cf_code_audited) << IsImportDecl;
1878     Diag(PragmaARCCFCodeAuditedInfo.second, diag::note_pragma_entered_here);
1879 
1880     // Immediately leave the pragma.
1881     PragmaARCCFCodeAuditedInfo = {nullptr, SourceLocation()};
1882   }
1883 
1884   // Complain about attempts to #include files in an assume-nonnull pragma.
1885   if (PragmaAssumeNonNullLoc.isValid()) {
1886     Diag(StartLoc, diag::err_pp_include_in_assume_nonnull) << IsImportDecl;
1887     Diag(PragmaAssumeNonNullLoc, diag::note_pragma_entered_here);
1888 
1889     // Immediately leave the pragma.
1890     PragmaAssumeNonNullLoc = SourceLocation();
1891   }
1892 
1893   if (HeaderInfo.HasIncludeAliasMap()) {
1894     // Map the filename with the brackets still attached.  If the name doesn't
1895     // map to anything, fall back on the filename we've already gotten the
1896     // spelling for.
1897     StringRef NewName = HeaderInfo.MapHeaderToIncludeAlias(OriginalFilename);
1898     if (!NewName.empty())
1899       Filename = NewName;
1900   }
1901 
1902   // Search include directories.
1903   bool IsMapped = false;
1904   bool IsFrameworkFound = false;
1905   const DirectoryLookup *CurDir;
1906   SmallString<1024> SearchPath;
1907   SmallString<1024> RelativePath;
1908   // We get the raw path only if we have 'Callbacks' to which we later pass
1909   // the path.
1910   ModuleMap::KnownHeader SuggestedModule;
1911   SourceLocation FilenameLoc = FilenameTok.getLocation();
1912   StringRef LookupFilename = Filename;
1913 
1914 #ifndef _WIN32
1915   // Normalize slashes when compiling with -fms-extensions on non-Windows. This
1916   // is unnecessary on Windows since the filesystem there handles backslashes.
1917   SmallString<128> NormalizedPath;
1918   if (LangOpts.MicrosoftExt) {
1919     NormalizedPath = Filename.str();
1920     llvm::sys::path::native(NormalizedPath);
1921     LookupFilename = NormalizedPath;
1922   }
1923 #endif
1924 
1925   Optional<FileEntryRef> File = LookupHeaderIncludeOrImport(
1926       CurDir, Filename, FilenameLoc, FilenameRange, FilenameTok,
1927       IsFrameworkFound, IsImportDecl, IsMapped, LookupFrom, LookupFromFile,
1928       LookupFilename, RelativePath, SearchPath, SuggestedModule, isAngled);
1929 
1930   if (usingPCHWithThroughHeader() && SkippingUntilPCHThroughHeader) {
1931     if (File && isPCHThroughHeader(&File->getFileEntry()))
1932       SkippingUntilPCHThroughHeader = false;
1933     return {ImportAction::None};
1934   }
1935 
1936   // Check for circular inclusion of the main file.
1937   // We can't generate a consistent preamble with regard to the conditional
1938   // stack if the main file is included again as due to the preamble bounds
1939   // some directives (e.g. #endif of a header guard) will never be seen.
1940   // Since this will lead to confusing errors, avoid the inclusion.
1941   if (File && PreambleConditionalStack.isRecording() &&
1942       SourceMgr.translateFile(&File->getFileEntry()) ==
1943           SourceMgr.getMainFileID()) {
1944     Diag(FilenameTok.getLocation(),
1945          diag::err_pp_including_mainfile_in_preamble);
1946     return {ImportAction::None};
1947   }
1948 
1949   // Should we enter the source file? Set to Skip if either the source file is
1950   // known to have no effect beyond its effect on module visibility -- that is,
1951   // if it's got an include guard that is already defined, set to Import if it
1952   // is a modular header we've already built and should import.
1953   enum { Enter, Import, Skip, IncludeLimitReached } Action = Enter;
1954 
1955   if (PPOpts->SingleFileParseMode)
1956     Action = IncludeLimitReached;
1957 
1958   // If we've reached the max allowed include depth, it is usually due to an
1959   // include cycle. Don't enter already processed files again as it can lead to
1960   // reaching the max allowed include depth again.
1961   if (Action == Enter && HasReachedMaxIncludeDepth && File &&
1962       HeaderInfo.getFileInfo(&File->getFileEntry()).NumIncludes)
1963     Action = IncludeLimitReached;
1964 
1965   // Determine whether we should try to import the module for this #include, if
1966   // there is one. Don't do so if precompiled module support is disabled or we
1967   // are processing this module textually (because we're building the module).
1968   if (Action == Enter && File && SuggestedModule && getLangOpts().Modules &&
1969       !isForModuleBuilding(SuggestedModule.getModule(),
1970                            getLangOpts().CurrentModule,
1971                            getLangOpts().ModuleName)) {
1972     // If this include corresponds to a module but that module is
1973     // unavailable, diagnose the situation and bail out.
1974     // FIXME: Remove this; loadModule does the same check (but produces
1975     // slightly worse diagnostics).
1976     if (checkModuleIsAvailable(getLangOpts(), getTargetInfo(), getDiagnostics(),
1977                                SuggestedModule.getModule())) {
1978       Diag(FilenameTok.getLocation(),
1979            diag::note_implicit_top_level_module_import_here)
1980           << SuggestedModule.getModule()->getTopLevelModuleName();
1981       return {ImportAction::None};
1982     }
1983 
1984     // Compute the module access path corresponding to this module.
1985     // FIXME: Should we have a second loadModule() overload to avoid this
1986     // extra lookup step?
1987     SmallVector<std::pair<IdentifierInfo *, SourceLocation>, 2> Path;
1988     for (Module *Mod = SuggestedModule.getModule(); Mod; Mod = Mod->Parent)
1989       Path.push_back(std::make_pair(getIdentifierInfo(Mod->Name),
1990                                     FilenameTok.getLocation()));
1991     std::reverse(Path.begin(), Path.end());
1992 
1993     // Warn that we're replacing the include/import with a module import.
1994     if (!IsImportDecl)
1995       diagnoseAutoModuleImport(*this, StartLoc, IncludeTok, Path, CharEnd);
1996 
1997     // Load the module to import its macros. We'll make the declarations
1998     // visible when the parser gets here.
1999     // FIXME: Pass SuggestedModule in here rather than converting it to a path
2000     // and making the module loader convert it back again.
2001     ModuleLoadResult Imported = TheModuleLoader.loadModule(
2002         IncludeTok.getLocation(), Path, Module::Hidden,
2003         /*IsInclusionDirective=*/true);
2004     assert((Imported == nullptr || Imported == SuggestedModule.getModule()) &&
2005            "the imported module is different than the suggested one");
2006 
2007     if (Imported) {
2008       Action = Import;
2009     } else if (Imported.isMissingExpected()) {
2010       // We failed to find a submodule that we assumed would exist (because it
2011       // was in the directory of an umbrella header, for instance), but no
2012       // actual module containing it exists (because the umbrella header is
2013       // incomplete).  Treat this as a textual inclusion.
2014       SuggestedModule = ModuleMap::KnownHeader();
2015     } else if (Imported.isConfigMismatch()) {
2016       // On a configuration mismatch, enter the header textually. We still know
2017       // that it's part of the corresponding module.
2018     } else {
2019       // We hit an error processing the import. Bail out.
2020       if (hadModuleLoaderFatalFailure()) {
2021         // With a fatal failure in the module loader, we abort parsing.
2022         Token &Result = IncludeTok;
2023         assert(CurLexer && "#include but no current lexer set!");
2024         Result.startToken();
2025         CurLexer->FormTokenWithChars(Result, CurLexer->BufferEnd, tok::eof);
2026         CurLexer->cutOffLexing();
2027       }
2028       return {ImportAction::None};
2029     }
2030   }
2031 
2032   // The #included file will be considered to be a system header if either it is
2033   // in a system include directory, or if the #includer is a system include
2034   // header.
2035   SrcMgr::CharacteristicKind FileCharacter =
2036       SourceMgr.getFileCharacteristic(FilenameTok.getLocation());
2037   if (File)
2038     FileCharacter = std::max(HeaderInfo.getFileDirFlavor(&File->getFileEntry()),
2039                              FileCharacter);
2040 
2041   // If this is a '#import' or an import-declaration, don't re-enter the file.
2042   //
2043   // FIXME: If we have a suggested module for a '#include', and we've already
2044   // visited this file, don't bother entering it again. We know it has no
2045   // further effect.
2046   bool EnterOnce =
2047       IsImportDecl ||
2048       IncludeTok.getIdentifierInfo()->getPPKeywordID() == tok::pp_import;
2049 
2050   // Ask HeaderInfo if we should enter this #include file.  If not, #including
2051   // this file will have no effect.
2052   if (Action == Enter && File &&
2053       !HeaderInfo.ShouldEnterIncludeFile(*this, &File->getFileEntry(),
2054                                          EnterOnce, getLangOpts().Modules,
2055                                          SuggestedModule.getModule())) {
2056     // Even if we've already preprocessed this header once and know that we
2057     // don't need to see its contents again, we still need to import it if it's
2058     // modular because we might not have imported it from this submodule before.
2059     //
2060     // FIXME: We don't do this when compiling a PCH because the AST
2061     // serialization layer can't cope with it. This means we get local
2062     // submodule visibility semantics wrong in that case.
2063     Action = (SuggestedModule && !getLangOpts().CompilingPCH) ? Import : Skip;
2064   }
2065 
2066   if (Callbacks && !IsImportDecl) {
2067     // Notify the callback object that we've seen an inclusion directive.
2068     // FIXME: Use a different callback for a pp-import?
2069     Callbacks->InclusionDirective(
2070         HashLoc, IncludeTok, LookupFilename, isAngled, FilenameRange,
2071         File ? &File->getFileEntry() : nullptr, SearchPath, RelativePath,
2072         Action == Import ? SuggestedModule.getModule() : nullptr,
2073         FileCharacter);
2074     if (Action == Skip && File)
2075       Callbacks->FileSkipped(*File, FilenameTok, FileCharacter);
2076   }
2077 
2078   if (!File)
2079     return {ImportAction::None};
2080 
2081   // If this is a C++20 pp-import declaration, diagnose if we didn't find any
2082   // module corresponding to the named header.
2083   if (IsImportDecl && !SuggestedModule) {
2084     Diag(FilenameTok, diag::err_header_import_not_header_unit)
2085       << OriginalFilename << File->getName();
2086     return {ImportAction::None};
2087   }
2088 
2089   // Issue a diagnostic if the name of the file on disk has a different case
2090   // than the one we're about to open.
2091   const bool CheckIncludePathPortability =
2092       !IsMapped && !File->getFileEntry().tryGetRealPathName().empty();
2093 
2094   if (CheckIncludePathPortability) {
2095     StringRef Name = LookupFilename;
2096     StringRef RealPathName = File->getFileEntry().tryGetRealPathName();
2097     SmallVector<StringRef, 16> Components(llvm::sys::path::begin(Name),
2098                                           llvm::sys::path::end(Name));
2099 
2100     if (trySimplifyPath(Components, RealPathName)) {
2101       SmallString<128> Path;
2102       Path.reserve(Name.size()+2);
2103       Path.push_back(isAngled ? '<' : '"');
2104       bool isLeadingSeparator = llvm::sys::path::is_absolute(Name);
2105       for (auto Component : Components) {
2106         if (isLeadingSeparator)
2107           isLeadingSeparator = false;
2108         else
2109           Path.append(Component);
2110         // Append the separator the user used, or the close quote
2111         Path.push_back(
2112           Path.size() <= Filename.size() ? Filename[Path.size()-1] :
2113             (isAngled ? '>' : '"'));
2114       }
2115       // For user files and known standard headers, by default we issue a diagnostic.
2116       // For other system headers, we don't. They can be controlled separately.
2117       auto DiagId = (FileCharacter == SrcMgr::C_User || warnByDefaultOnWrongCase(Name)) ?
2118           diag::pp_nonportable_path : diag::pp_nonportable_system_path;
2119       Diag(FilenameTok, DiagId) << Path <<
2120         FixItHint::CreateReplacement(FilenameRange, Path);
2121     }
2122   }
2123 
2124   switch (Action) {
2125   case Skip:
2126     // If we don't need to enter the file, stop now.
2127     if (Module *M = SuggestedModule.getModule())
2128       return {ImportAction::SkippedModuleImport, M};
2129     return {ImportAction::None};
2130 
2131   case IncludeLimitReached:
2132     // If we reached our include limit and don't want to enter any more files,
2133     // don't go any further.
2134     return {ImportAction::None};
2135 
2136   case Import: {
2137     // If this is a module import, make it visible if needed.
2138     Module *M = SuggestedModule.getModule();
2139     assert(M && "no module to import");
2140 
2141     makeModuleVisible(M, EndLoc);
2142 
2143     if (IncludeTok.getIdentifierInfo()->getPPKeywordID() ==
2144         tok::pp___include_macros)
2145       return {ImportAction::None};
2146 
2147     return {ImportAction::ModuleImport, M};
2148   }
2149 
2150   case Enter:
2151     break;
2152   }
2153 
2154   // Check that we don't have infinite #include recursion.
2155   if (IncludeMacroStack.size() == MaxAllowedIncludeStackDepth-1) {
2156     Diag(FilenameTok, diag::err_pp_include_too_deep);
2157     HasReachedMaxIncludeDepth = true;
2158     return {ImportAction::None};
2159   }
2160 
2161   // Look up the file, create a File ID for it.
2162   SourceLocation IncludePos = FilenameTok.getLocation();
2163   // If the filename string was the result of macro expansions, set the include
2164   // position on the file where it will be included and after the expansions.
2165   if (IncludePos.isMacroID())
2166     IncludePos = SourceMgr.getExpansionRange(IncludePos).getEnd();
2167   FileID FID = SourceMgr.createFileID(*File, IncludePos, FileCharacter);
2168   assert(FID.isValid() && "Expected valid file ID");
2169 
2170   // If all is good, enter the new file!
2171   if (EnterSourceFile(FID, CurDir, FilenameTok.getLocation()))
2172     return {ImportAction::None};
2173 
2174   // Determine if we're switching to building a new submodule, and which one.
2175   if (auto *M = SuggestedModule.getModule()) {
2176     if (M->getTopLevelModule()->ShadowingModule) {
2177       // We are building a submodule that belongs to a shadowed module. This
2178       // means we find header files in the shadowed module.
2179       Diag(M->DefinitionLoc, diag::err_module_build_shadowed_submodule)
2180         << M->getFullModuleName();
2181       Diag(M->getTopLevelModule()->ShadowingModule->DefinitionLoc,
2182            diag::note_previous_definition);
2183       return {ImportAction::None};
2184     }
2185     // When building a pch, -fmodule-name tells the compiler to textually
2186     // include headers in the specified module. We are not building the
2187     // specified module.
2188     //
2189     // FIXME: This is the wrong way to handle this. We should produce a PCH
2190     // that behaves the same as the header would behave in a compilation using
2191     // that PCH, which means we should enter the submodule. We need to teach
2192     // the AST serialization layer to deal with the resulting AST.
2193     if (getLangOpts().CompilingPCH &&
2194         isForModuleBuilding(M, getLangOpts().CurrentModule,
2195                             getLangOpts().ModuleName))
2196       return {ImportAction::None};
2197 
2198     assert(!CurLexerSubmodule && "should not have marked this as a module yet");
2199     CurLexerSubmodule = M;
2200 
2201     // Let the macro handling code know that any future macros are within
2202     // the new submodule.
2203     EnterSubmodule(M, EndLoc, /*ForPragma*/false);
2204 
2205     // Let the parser know that any future declarations are within the new
2206     // submodule.
2207     // FIXME: There's no point doing this if we're handling a #__include_macros
2208     // directive.
2209     return {ImportAction::ModuleBegin, M};
2210   }
2211 
2212   assert(!IsImportDecl && "failed to diagnose missing module for import decl");
2213   return {ImportAction::None};
2214 }
2215 
2216 /// HandleIncludeNextDirective - Implements \#include_next.
2217 ///
2218 void Preprocessor::HandleIncludeNextDirective(SourceLocation HashLoc,
2219                                               Token &IncludeNextTok) {
2220   Diag(IncludeNextTok, diag::ext_pp_include_next_directive);
2221 
2222   // #include_next is like #include, except that we start searching after
2223   // the current found directory.  If we can't do this, issue a
2224   // diagnostic.
2225   const DirectoryLookup *Lookup = CurDirLookup;
2226   const FileEntry *LookupFromFile = nullptr;
2227   if (isInPrimaryFile() && LangOpts.IsHeaderFile) {
2228     // If the main file is a header, then it's either for PCH/AST generation,
2229     // or libclang opened it. Either way, handle it as a normal include below
2230     // and do not complain about include_next.
2231   } else if (isInPrimaryFile()) {
2232     Lookup = nullptr;
2233     Diag(IncludeNextTok, diag::pp_include_next_in_primary);
2234   } else if (CurLexerSubmodule) {
2235     // Start looking up in the directory *after* the one in which the current
2236     // file would be found, if any.
2237     assert(CurPPLexer && "#include_next directive in macro?");
2238     LookupFromFile = CurPPLexer->getFileEntry();
2239     Lookup = nullptr;
2240   } else if (!Lookup) {
2241     // The current file was not found by walking the include path. Either it
2242     // is the primary file (handled above), or it was found by absolute path,
2243     // or it was found relative to such a file.
2244     // FIXME: Track enough information so we know which case we're in.
2245     Diag(IncludeNextTok, diag::pp_include_next_absolute_path);
2246   } else {
2247     // Start looking up in the next directory.
2248     ++Lookup;
2249   }
2250 
2251   return HandleIncludeDirective(HashLoc, IncludeNextTok, Lookup,
2252                                 LookupFromFile);
2253 }
2254 
2255 /// HandleMicrosoftImportDirective - Implements \#import for Microsoft Mode
2256 void Preprocessor::HandleMicrosoftImportDirective(Token &Tok) {
2257   // The Microsoft #import directive takes a type library and generates header
2258   // files from it, and includes those.  This is beyond the scope of what clang
2259   // does, so we ignore it and error out.  However, #import can optionally have
2260   // trailing attributes that span multiple lines.  We're going to eat those
2261   // so we can continue processing from there.
2262   Diag(Tok, diag::err_pp_import_directive_ms );
2263 
2264   // Read tokens until we get to the end of the directive.  Note that the
2265   // directive can be split over multiple lines using the backslash character.
2266   DiscardUntilEndOfDirective();
2267 }
2268 
2269 /// HandleImportDirective - Implements \#import.
2270 ///
2271 void Preprocessor::HandleImportDirective(SourceLocation HashLoc,
2272                                          Token &ImportTok) {
2273   if (!LangOpts.ObjC) {  // #import is standard for ObjC.
2274     if (LangOpts.MSVCCompat)
2275       return HandleMicrosoftImportDirective(ImportTok);
2276     Diag(ImportTok, diag::ext_pp_import_directive);
2277   }
2278   return HandleIncludeDirective(HashLoc, ImportTok);
2279 }
2280 
2281 /// HandleIncludeMacrosDirective - The -imacros command line option turns into a
2282 /// pseudo directive in the predefines buffer.  This handles it by sucking all
2283 /// tokens through the preprocessor and discarding them (only keeping the side
2284 /// effects on the preprocessor).
2285 void Preprocessor::HandleIncludeMacrosDirective(SourceLocation HashLoc,
2286                                                 Token &IncludeMacrosTok) {
2287   // This directive should only occur in the predefines buffer.  If not, emit an
2288   // error and reject it.
2289   SourceLocation Loc = IncludeMacrosTok.getLocation();
2290   if (SourceMgr.getBufferName(Loc) != "<built-in>") {
2291     Diag(IncludeMacrosTok.getLocation(),
2292          diag::pp_include_macros_out_of_predefines);
2293     DiscardUntilEndOfDirective();
2294     return;
2295   }
2296 
2297   // Treat this as a normal #include for checking purposes.  If this is
2298   // successful, it will push a new lexer onto the include stack.
2299   HandleIncludeDirective(HashLoc, IncludeMacrosTok);
2300 
2301   Token TmpTok;
2302   do {
2303     Lex(TmpTok);
2304     assert(TmpTok.isNot(tok::eof) && "Didn't find end of -imacros!");
2305   } while (TmpTok.isNot(tok::hashhash));
2306 }
2307 
2308 //===----------------------------------------------------------------------===//
2309 // Preprocessor Macro Directive Handling.
2310 //===----------------------------------------------------------------------===//
2311 
2312 /// ReadMacroParameterList - The ( starting a parameter list of a macro
2313 /// definition has just been read.  Lex the rest of the parameters and the
2314 /// closing ), updating MI with what we learn.  Return true if an error occurs
2315 /// parsing the param list.
2316 bool Preprocessor::ReadMacroParameterList(MacroInfo *MI, Token &Tok) {
2317   SmallVector<IdentifierInfo*, 32> Parameters;
2318 
2319   while (true) {
2320     LexUnexpandedToken(Tok);
2321     switch (Tok.getKind()) {
2322     case tok::r_paren:
2323       // Found the end of the parameter list.
2324       if (Parameters.empty())  // #define FOO()
2325         return false;
2326       // Otherwise we have #define FOO(A,)
2327       Diag(Tok, diag::err_pp_expected_ident_in_arg_list);
2328       return true;
2329     case tok::ellipsis:  // #define X(... -> C99 varargs
2330       if (!LangOpts.C99)
2331         Diag(Tok, LangOpts.CPlusPlus11 ?
2332              diag::warn_cxx98_compat_variadic_macro :
2333              diag::ext_variadic_macro);
2334 
2335       // OpenCL v1.2 s6.9.e: variadic macros are not supported.
2336       if (LangOpts.OpenCL) {
2337         Diag(Tok, diag::ext_pp_opencl_variadic_macros);
2338       }
2339 
2340       // Lex the token after the identifier.
2341       LexUnexpandedToken(Tok);
2342       if (Tok.isNot(tok::r_paren)) {
2343         Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
2344         return true;
2345       }
2346       // Add the __VA_ARGS__ identifier as a parameter.
2347       Parameters.push_back(Ident__VA_ARGS__);
2348       MI->setIsC99Varargs();
2349       MI->setParameterList(Parameters, BP);
2350       return false;
2351     case tok::eod:  // #define X(
2352       Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
2353       return true;
2354     default:
2355       // Handle keywords and identifiers here to accept things like
2356       // #define Foo(for) for.
2357       IdentifierInfo *II = Tok.getIdentifierInfo();
2358       if (!II) {
2359         // #define X(1
2360         Diag(Tok, diag::err_pp_invalid_tok_in_arg_list);
2361         return true;
2362       }
2363 
2364       // If this is already used as a parameter, it is used multiple times (e.g.
2365       // #define X(A,A.
2366       if (llvm::find(Parameters, II) != Parameters.end()) { // C99 6.10.3p6
2367         Diag(Tok, diag::err_pp_duplicate_name_in_arg_list) << II;
2368         return true;
2369       }
2370 
2371       // Add the parameter to the macro info.
2372       Parameters.push_back(II);
2373 
2374       // Lex the token after the identifier.
2375       LexUnexpandedToken(Tok);
2376 
2377       switch (Tok.getKind()) {
2378       default:          // #define X(A B
2379         Diag(Tok, diag::err_pp_expected_comma_in_arg_list);
2380         return true;
2381       case tok::r_paren: // #define X(A)
2382         MI->setParameterList(Parameters, BP);
2383         return false;
2384       case tok::comma:  // #define X(A,
2385         break;
2386       case tok::ellipsis:  // #define X(A... -> GCC extension
2387         // Diagnose extension.
2388         Diag(Tok, diag::ext_named_variadic_macro);
2389 
2390         // Lex the token after the identifier.
2391         LexUnexpandedToken(Tok);
2392         if (Tok.isNot(tok::r_paren)) {
2393           Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
2394           return true;
2395         }
2396 
2397         MI->setIsGNUVarargs();
2398         MI->setParameterList(Parameters, BP);
2399         return false;
2400       }
2401     }
2402   }
2403 }
2404 
2405 static bool isConfigurationPattern(Token &MacroName, MacroInfo *MI,
2406                                    const LangOptions &LOptions) {
2407   if (MI->getNumTokens() == 1) {
2408     const Token &Value = MI->getReplacementToken(0);
2409 
2410     // Macro that is identity, like '#define inline inline' is a valid pattern.
2411     if (MacroName.getKind() == Value.getKind())
2412       return true;
2413 
2414     // Macro that maps a keyword to the same keyword decorated with leading/
2415     // trailing underscores is a valid pattern:
2416     //    #define inline __inline
2417     //    #define inline __inline__
2418     //    #define inline _inline (in MS compatibility mode)
2419     StringRef MacroText = MacroName.getIdentifierInfo()->getName();
2420     if (IdentifierInfo *II = Value.getIdentifierInfo()) {
2421       if (!II->isKeyword(LOptions))
2422         return false;
2423       StringRef ValueText = II->getName();
2424       StringRef TrimmedValue = ValueText;
2425       if (!ValueText.startswith("__")) {
2426         if (ValueText.startswith("_"))
2427           TrimmedValue = TrimmedValue.drop_front(1);
2428         else
2429           return false;
2430       } else {
2431         TrimmedValue = TrimmedValue.drop_front(2);
2432         if (TrimmedValue.endswith("__"))
2433           TrimmedValue = TrimmedValue.drop_back(2);
2434       }
2435       return TrimmedValue.equals(MacroText);
2436     } else {
2437       return false;
2438     }
2439   }
2440 
2441   // #define inline
2442   return MacroName.isOneOf(tok::kw_extern, tok::kw_inline, tok::kw_static,
2443                            tok::kw_const) &&
2444          MI->getNumTokens() == 0;
2445 }
2446 
2447 // ReadOptionalMacroParameterListAndBody - This consumes all (i.e. the
2448 // entire line) of the macro's tokens and adds them to MacroInfo, and while
2449 // doing so performs certain validity checks including (but not limited to):
2450 //   - # (stringization) is followed by a macro parameter
2451 //
2452 //  Returns a nullptr if an invalid sequence of tokens is encountered or returns
2453 //  a pointer to a MacroInfo object.
2454 
2455 MacroInfo *Preprocessor::ReadOptionalMacroParameterListAndBody(
2456     const Token &MacroNameTok, const bool ImmediatelyAfterHeaderGuard) {
2457 
2458   Token LastTok = MacroNameTok;
2459   // Create the new macro.
2460   MacroInfo *const MI = AllocateMacroInfo(MacroNameTok.getLocation());
2461 
2462   Token Tok;
2463   LexUnexpandedToken(Tok);
2464 
2465   // Ensure we consume the rest of the macro body if errors occur.
2466   auto _ = llvm::make_scope_exit([&]() {
2467     // The flag indicates if we are still waiting for 'eod'.
2468     if (CurLexer->ParsingPreprocessorDirective)
2469       DiscardUntilEndOfDirective();
2470   });
2471 
2472   // Used to un-poison and then re-poison identifiers of the __VA_ARGS__ ilk
2473   // within their appropriate context.
2474   VariadicMacroScopeGuard VariadicMacroScopeGuard(*this);
2475 
2476   // If this is a function-like macro definition, parse the argument list,
2477   // marking each of the identifiers as being used as macro arguments.  Also,
2478   // check other constraints on the first token of the macro body.
2479   if (Tok.is(tok::eod)) {
2480     if (ImmediatelyAfterHeaderGuard) {
2481       // Save this macro information since it may part of a header guard.
2482       CurPPLexer->MIOpt.SetDefinedMacro(MacroNameTok.getIdentifierInfo(),
2483                                         MacroNameTok.getLocation());
2484     }
2485     // If there is no body to this macro, we have no special handling here.
2486   } else if (Tok.hasLeadingSpace()) {
2487     // This is a normal token with leading space.  Clear the leading space
2488     // marker on the first token to get proper expansion.
2489     Tok.clearFlag(Token::LeadingSpace);
2490   } else if (Tok.is(tok::l_paren)) {
2491     // This is a function-like macro definition.  Read the argument list.
2492     MI->setIsFunctionLike();
2493     if (ReadMacroParameterList(MI, LastTok))
2494       return nullptr;
2495 
2496     // If this is a definition of an ISO C/C++ variadic function-like macro (not
2497     // using the GNU named varargs extension) inform our variadic scope guard
2498     // which un-poisons and re-poisons certain identifiers (e.g. __VA_ARGS__)
2499     // allowed only within the definition of a variadic macro.
2500 
2501     if (MI->isC99Varargs()) {
2502       VariadicMacroScopeGuard.enterScope();
2503     }
2504 
2505     // Read the first token after the arg list for down below.
2506     LexUnexpandedToken(Tok);
2507   } else if (LangOpts.C99 || LangOpts.CPlusPlus11) {
2508     // C99 requires whitespace between the macro definition and the body.  Emit
2509     // a diagnostic for something like "#define X+".
2510     Diag(Tok, diag::ext_c99_whitespace_required_after_macro_name);
2511   } else {
2512     // C90 6.8 TC1 says: "In the definition of an object-like macro, if the
2513     // first character of a replacement list is not a character required by
2514     // subclause 5.2.1, then there shall be white-space separation between the
2515     // identifier and the replacement list.".  5.2.1 lists this set:
2516     //   "A-Za-z0-9!"#%&'()*+,_./:;<=>?[\]^_{|}~" as well as whitespace, which
2517     // is irrelevant here.
2518     bool isInvalid = false;
2519     if (Tok.is(tok::at)) // @ is not in the list above.
2520       isInvalid = true;
2521     else if (Tok.is(tok::unknown)) {
2522       // If we have an unknown token, it is something strange like "`".  Since
2523       // all of valid characters would have lexed into a single character
2524       // token of some sort, we know this is not a valid case.
2525       isInvalid = true;
2526     }
2527     if (isInvalid)
2528       Diag(Tok, diag::ext_missing_whitespace_after_macro_name);
2529     else
2530       Diag(Tok, diag::warn_missing_whitespace_after_macro_name);
2531   }
2532 
2533   if (!Tok.is(tok::eod))
2534     LastTok = Tok;
2535 
2536   // Read the rest of the macro body.
2537   if (MI->isObjectLike()) {
2538     // Object-like macros are very simple, just read their body.
2539     while (Tok.isNot(tok::eod)) {
2540       LastTok = Tok;
2541       MI->AddTokenToBody(Tok);
2542       // Get the next token of the macro.
2543       LexUnexpandedToken(Tok);
2544     }
2545   } else {
2546     // Otherwise, read the body of a function-like macro.  While we are at it,
2547     // check C99 6.10.3.2p1: ensure that # operators are followed by macro
2548     // parameters in function-like macro expansions.
2549 
2550     VAOptDefinitionContext VAOCtx(*this);
2551 
2552     while (Tok.isNot(tok::eod)) {
2553       LastTok = Tok;
2554 
2555       if (!Tok.isOneOf(tok::hash, tok::hashat, tok::hashhash)) {
2556         MI->AddTokenToBody(Tok);
2557 
2558         if (VAOCtx.isVAOptToken(Tok)) {
2559           // If we're already within a VAOPT, emit an error.
2560           if (VAOCtx.isInVAOpt()) {
2561             Diag(Tok, diag::err_pp_vaopt_nested_use);
2562             return nullptr;
2563           }
2564           // Ensure VAOPT is followed by a '(' .
2565           LexUnexpandedToken(Tok);
2566           if (Tok.isNot(tok::l_paren)) {
2567             Diag(Tok, diag::err_pp_missing_lparen_in_vaopt_use);
2568             return nullptr;
2569           }
2570           MI->AddTokenToBody(Tok);
2571           VAOCtx.sawVAOptFollowedByOpeningParens(Tok.getLocation());
2572           LexUnexpandedToken(Tok);
2573           if (Tok.is(tok::hashhash)) {
2574             Diag(Tok, diag::err_vaopt_paste_at_start);
2575             return nullptr;
2576           }
2577           continue;
2578         } else if (VAOCtx.isInVAOpt()) {
2579           if (Tok.is(tok::r_paren)) {
2580             if (VAOCtx.sawClosingParen()) {
2581               const unsigned NumTokens = MI->getNumTokens();
2582               assert(NumTokens >= 3 && "Must have seen at least __VA_OPT__( "
2583                                        "and a subsequent tok::r_paren");
2584               if (MI->getReplacementToken(NumTokens - 2).is(tok::hashhash)) {
2585                 Diag(Tok, diag::err_vaopt_paste_at_end);
2586                 return nullptr;
2587               }
2588             }
2589           } else if (Tok.is(tok::l_paren)) {
2590             VAOCtx.sawOpeningParen(Tok.getLocation());
2591           }
2592         }
2593         // Get the next token of the macro.
2594         LexUnexpandedToken(Tok);
2595         continue;
2596       }
2597 
2598       // If we're in -traditional mode, then we should ignore stringification
2599       // and token pasting. Mark the tokens as unknown so as not to confuse
2600       // things.
2601       if (getLangOpts().TraditionalCPP) {
2602         Tok.setKind(tok::unknown);
2603         MI->AddTokenToBody(Tok);
2604 
2605         // Get the next token of the macro.
2606         LexUnexpandedToken(Tok);
2607         continue;
2608       }
2609 
2610       if (Tok.is(tok::hashhash)) {
2611         // If we see token pasting, check if it looks like the gcc comma
2612         // pasting extension.  We'll use this information to suppress
2613         // diagnostics later on.
2614 
2615         // Get the next token of the macro.
2616         LexUnexpandedToken(Tok);
2617 
2618         if (Tok.is(tok::eod)) {
2619           MI->AddTokenToBody(LastTok);
2620           break;
2621         }
2622 
2623         unsigned NumTokens = MI->getNumTokens();
2624         if (NumTokens && Tok.getIdentifierInfo() == Ident__VA_ARGS__ &&
2625             MI->getReplacementToken(NumTokens-1).is(tok::comma))
2626           MI->setHasCommaPasting();
2627 
2628         // Things look ok, add the '##' token to the macro.
2629         MI->AddTokenToBody(LastTok);
2630         continue;
2631       }
2632 
2633       // Our Token is a stringization operator.
2634       // Get the next token of the macro.
2635       LexUnexpandedToken(Tok);
2636 
2637       // Check for a valid macro arg identifier or __VA_OPT__.
2638       if (!VAOCtx.isVAOptToken(Tok) &&
2639           (Tok.getIdentifierInfo() == nullptr ||
2640            MI->getParameterNum(Tok.getIdentifierInfo()) == -1)) {
2641 
2642         // If this is assembler-with-cpp mode, we accept random gibberish after
2643         // the '#' because '#' is often a comment character.  However, change
2644         // the kind of the token to tok::unknown so that the preprocessor isn't
2645         // confused.
2646         if (getLangOpts().AsmPreprocessor && Tok.isNot(tok::eod)) {
2647           LastTok.setKind(tok::unknown);
2648           MI->AddTokenToBody(LastTok);
2649           continue;
2650         } else {
2651           Diag(Tok, diag::err_pp_stringize_not_parameter)
2652             << LastTok.is(tok::hashat);
2653           return nullptr;
2654         }
2655       }
2656 
2657       // Things look ok, add the '#' and param name tokens to the macro.
2658       MI->AddTokenToBody(LastTok);
2659 
2660       // If the token following '#' is VAOPT, let the next iteration handle it
2661       // and check it for correctness, otherwise add the token and prime the
2662       // loop with the next one.
2663       if (!VAOCtx.isVAOptToken(Tok)) {
2664         MI->AddTokenToBody(Tok);
2665         LastTok = Tok;
2666 
2667         // Get the next token of the macro.
2668         LexUnexpandedToken(Tok);
2669       }
2670     }
2671     if (VAOCtx.isInVAOpt()) {
2672       assert(Tok.is(tok::eod) && "Must be at End Of preprocessing Directive");
2673       Diag(Tok, diag::err_pp_expected_after)
2674         << LastTok.getKind() << tok::r_paren;
2675       Diag(VAOCtx.getUnmatchedOpeningParenLoc(), diag::note_matching) << tok::l_paren;
2676       return nullptr;
2677     }
2678   }
2679   MI->setDefinitionEndLoc(LastTok.getLocation());
2680   return MI;
2681 }
2682 /// HandleDefineDirective - Implements \#define.  This consumes the entire macro
2683 /// line then lets the caller lex the next real token.
2684 void Preprocessor::HandleDefineDirective(
2685     Token &DefineTok, const bool ImmediatelyAfterHeaderGuard) {
2686   ++NumDefined;
2687 
2688   Token MacroNameTok;
2689   bool MacroShadowsKeyword;
2690   ReadMacroName(MacroNameTok, MU_Define, &MacroShadowsKeyword);
2691 
2692   // Error reading macro name?  If so, diagnostic already issued.
2693   if (MacroNameTok.is(tok::eod))
2694     return;
2695 
2696   // If we are supposed to keep comments in #defines, reenable comment saving
2697   // mode.
2698   if (CurLexer) CurLexer->SetCommentRetentionState(KeepMacroComments);
2699 
2700   MacroInfo *const MI = ReadOptionalMacroParameterListAndBody(
2701       MacroNameTok, ImmediatelyAfterHeaderGuard);
2702 
2703   if (!MI) return;
2704 
2705   if (MacroShadowsKeyword &&
2706       !isConfigurationPattern(MacroNameTok, MI, getLangOpts())) {
2707     Diag(MacroNameTok, diag::warn_pp_macro_hides_keyword);
2708   }
2709   // Check that there is no paste (##) operator at the beginning or end of the
2710   // replacement list.
2711   unsigned NumTokens = MI->getNumTokens();
2712   if (NumTokens != 0) {
2713     if (MI->getReplacementToken(0).is(tok::hashhash)) {
2714       Diag(MI->getReplacementToken(0), diag::err_paste_at_start);
2715       return;
2716     }
2717     if (MI->getReplacementToken(NumTokens-1).is(tok::hashhash)) {
2718       Diag(MI->getReplacementToken(NumTokens-1), diag::err_paste_at_end);
2719       return;
2720     }
2721   }
2722 
2723   // When skipping just warn about macros that do not match.
2724   if (SkippingUntilPCHThroughHeader) {
2725     const MacroInfo *OtherMI = getMacroInfo(MacroNameTok.getIdentifierInfo());
2726     if (!OtherMI || !MI->isIdenticalTo(*OtherMI, *this,
2727                              /*Syntactic=*/LangOpts.MicrosoftExt))
2728       Diag(MI->getDefinitionLoc(), diag::warn_pp_macro_def_mismatch_with_pch)
2729           << MacroNameTok.getIdentifierInfo();
2730     // Issue the diagnostic but allow the change if msvc extensions are enabled
2731     if (!LangOpts.MicrosoftExt)
2732       return;
2733   }
2734 
2735   // Finally, if this identifier already had a macro defined for it, verify that
2736   // the macro bodies are identical, and issue diagnostics if they are not.
2737   if (const MacroInfo *OtherMI=getMacroInfo(MacroNameTok.getIdentifierInfo())) {
2738     // In Objective-C, ignore attempts to directly redefine the builtin
2739     // definitions of the ownership qualifiers.  It's still possible to
2740     // #undef them.
2741     auto isObjCProtectedMacro = [](const IdentifierInfo *II) -> bool {
2742       return II->isStr("__strong") ||
2743              II->isStr("__weak") ||
2744              II->isStr("__unsafe_unretained") ||
2745              II->isStr("__autoreleasing");
2746     };
2747    if (getLangOpts().ObjC &&
2748         SourceMgr.getFileID(OtherMI->getDefinitionLoc())
2749           == getPredefinesFileID() &&
2750         isObjCProtectedMacro(MacroNameTok.getIdentifierInfo())) {
2751       // Warn if it changes the tokens.
2752       if ((!getDiagnostics().getSuppressSystemWarnings() ||
2753            !SourceMgr.isInSystemHeader(DefineTok.getLocation())) &&
2754           !MI->isIdenticalTo(*OtherMI, *this,
2755                              /*Syntactic=*/LangOpts.MicrosoftExt)) {
2756         Diag(MI->getDefinitionLoc(), diag::warn_pp_objc_macro_redef_ignored);
2757       }
2758       assert(!OtherMI->isWarnIfUnused());
2759       return;
2760     }
2761 
2762     // It is very common for system headers to have tons of macro redefinitions
2763     // and for warnings to be disabled in system headers.  If this is the case,
2764     // then don't bother calling MacroInfo::isIdenticalTo.
2765     if (!getDiagnostics().getSuppressSystemWarnings() ||
2766         !SourceMgr.isInSystemHeader(DefineTok.getLocation())) {
2767       if (!OtherMI->isUsed() && OtherMI->isWarnIfUnused())
2768         Diag(OtherMI->getDefinitionLoc(), diag::pp_macro_not_used);
2769 
2770       // Warn if defining "__LINE__" and other builtins, per C99 6.10.8/4 and
2771       // C++ [cpp.predefined]p4, but allow it as an extension.
2772       if (OtherMI->isBuiltinMacro())
2773         Diag(MacroNameTok, diag::ext_pp_redef_builtin_macro);
2774       // Macros must be identical.  This means all tokens and whitespace
2775       // separation must be the same.  C99 6.10.3p2.
2776       else if (!OtherMI->isAllowRedefinitionsWithoutWarning() &&
2777                !MI->isIdenticalTo(*OtherMI, *this, /*Syntactic=*/LangOpts.MicrosoftExt)) {
2778         Diag(MI->getDefinitionLoc(), diag::ext_pp_macro_redef)
2779           << MacroNameTok.getIdentifierInfo();
2780         Diag(OtherMI->getDefinitionLoc(), diag::note_previous_definition);
2781       }
2782     }
2783     if (OtherMI->isWarnIfUnused())
2784       WarnUnusedMacroLocs.erase(OtherMI->getDefinitionLoc());
2785   }
2786 
2787   DefMacroDirective *MD =
2788       appendDefMacroDirective(MacroNameTok.getIdentifierInfo(), MI);
2789 
2790   assert(!MI->isUsed());
2791   // If we need warning for not using the macro, add its location in the
2792   // warn-because-unused-macro set. If it gets used it will be removed from set.
2793   if (getSourceManager().isInMainFile(MI->getDefinitionLoc()) &&
2794       !Diags->isIgnored(diag::pp_macro_not_used, MI->getDefinitionLoc()) &&
2795       !MacroExpansionInDirectivesOverride) {
2796     MI->setIsWarnIfUnused(true);
2797     WarnUnusedMacroLocs.insert(MI->getDefinitionLoc());
2798   }
2799 
2800   // If the callbacks want to know, tell them about the macro definition.
2801   if (Callbacks)
2802     Callbacks->MacroDefined(MacroNameTok, MD);
2803 }
2804 
2805 /// HandleUndefDirective - Implements \#undef.
2806 ///
2807 void Preprocessor::HandleUndefDirective() {
2808   ++NumUndefined;
2809 
2810   Token MacroNameTok;
2811   ReadMacroName(MacroNameTok, MU_Undef);
2812 
2813   // Error reading macro name?  If so, diagnostic already issued.
2814   if (MacroNameTok.is(tok::eod))
2815     return;
2816 
2817   // Check to see if this is the last token on the #undef line.
2818   CheckEndOfDirective("undef");
2819 
2820   // Okay, we have a valid identifier to undef.
2821   auto *II = MacroNameTok.getIdentifierInfo();
2822   auto MD = getMacroDefinition(II);
2823   UndefMacroDirective *Undef = nullptr;
2824 
2825   // If the macro is not defined, this is a noop undef.
2826   if (const MacroInfo *MI = MD.getMacroInfo()) {
2827     if (!MI->isUsed() && MI->isWarnIfUnused())
2828       Diag(MI->getDefinitionLoc(), diag::pp_macro_not_used);
2829 
2830     if (MI->isWarnIfUnused())
2831       WarnUnusedMacroLocs.erase(MI->getDefinitionLoc());
2832 
2833     Undef = AllocateUndefMacroDirective(MacroNameTok.getLocation());
2834   }
2835 
2836   // If the callbacks want to know, tell them about the macro #undef.
2837   // Note: no matter if the macro was defined or not.
2838   if (Callbacks)
2839     Callbacks->MacroUndefined(MacroNameTok, MD, Undef);
2840 
2841   if (Undef)
2842     appendMacroDirective(II, Undef);
2843 }
2844 
2845 //===----------------------------------------------------------------------===//
2846 // Preprocessor Conditional Directive Handling.
2847 //===----------------------------------------------------------------------===//
2848 
2849 /// HandleIfdefDirective - Implements the \#ifdef/\#ifndef directive.  isIfndef
2850 /// is true when this is a \#ifndef directive.  ReadAnyTokensBeforeDirective is
2851 /// true if any tokens have been returned or pp-directives activated before this
2852 /// \#ifndef has been lexed.
2853 ///
2854 void Preprocessor::HandleIfdefDirective(Token &Result,
2855                                         const Token &HashToken,
2856                                         bool isIfndef,
2857                                         bool ReadAnyTokensBeforeDirective) {
2858   ++NumIf;
2859   Token DirectiveTok = Result;
2860 
2861   Token MacroNameTok;
2862   ReadMacroName(MacroNameTok);
2863 
2864   // Error reading macro name?  If so, diagnostic already issued.
2865   if (MacroNameTok.is(tok::eod)) {
2866     // Skip code until we get to #endif.  This helps with recovery by not
2867     // emitting an error when the #endif is reached.
2868     SkipExcludedConditionalBlock(HashToken.getLocation(),
2869                                  DirectiveTok.getLocation(),
2870                                  /*Foundnonskip*/ false, /*FoundElse*/ false);
2871     return;
2872   }
2873 
2874   // Check to see if this is the last token on the #if[n]def line.
2875   CheckEndOfDirective(isIfndef ? "ifndef" : "ifdef");
2876 
2877   IdentifierInfo *MII = MacroNameTok.getIdentifierInfo();
2878   auto MD = getMacroDefinition(MII);
2879   MacroInfo *MI = MD.getMacroInfo();
2880 
2881   if (CurPPLexer->getConditionalStackDepth() == 0) {
2882     // If the start of a top-level #ifdef and if the macro is not defined,
2883     // inform MIOpt that this might be the start of a proper include guard.
2884     // Otherwise it is some other form of unknown conditional which we can't
2885     // handle.
2886     if (!ReadAnyTokensBeforeDirective && !MI) {
2887       assert(isIfndef && "#ifdef shouldn't reach here");
2888       CurPPLexer->MIOpt.EnterTopLevelIfndef(MII, MacroNameTok.getLocation());
2889     } else
2890       CurPPLexer->MIOpt.EnterTopLevelConditional();
2891   }
2892 
2893   // If there is a macro, process it.
2894   if (MI)  // Mark it used.
2895     markMacroAsUsed(MI);
2896 
2897   if (Callbacks) {
2898     if (isIfndef)
2899       Callbacks->Ifndef(DirectiveTok.getLocation(), MacroNameTok, MD);
2900     else
2901       Callbacks->Ifdef(DirectiveTok.getLocation(), MacroNameTok, MD);
2902   }
2903 
2904   bool RetainExcludedCB = PPOpts->RetainExcludedConditionalBlocks &&
2905     getSourceManager().isInMainFile(DirectiveTok.getLocation());
2906 
2907   // Should we include the stuff contained by this directive?
2908   if (PPOpts->SingleFileParseMode && !MI) {
2909     // In 'single-file-parse mode' undefined identifiers trigger parsing of all
2910     // the directive blocks.
2911     CurPPLexer->pushConditionalLevel(DirectiveTok.getLocation(),
2912                                      /*wasskip*/false, /*foundnonskip*/false,
2913                                      /*foundelse*/false);
2914   } else if (!MI == isIfndef || RetainExcludedCB) {
2915     // Yes, remember that we are inside a conditional, then lex the next token.
2916     CurPPLexer->pushConditionalLevel(DirectiveTok.getLocation(),
2917                                      /*wasskip*/false, /*foundnonskip*/true,
2918                                      /*foundelse*/false);
2919   } else {
2920     // No, skip the contents of this block.
2921     SkipExcludedConditionalBlock(HashToken.getLocation(),
2922                                  DirectiveTok.getLocation(),
2923                                  /*Foundnonskip*/ false,
2924                                  /*FoundElse*/ false);
2925   }
2926 }
2927 
2928 /// HandleIfDirective - Implements the \#if directive.
2929 ///
2930 void Preprocessor::HandleIfDirective(Token &IfToken,
2931                                      const Token &HashToken,
2932                                      bool ReadAnyTokensBeforeDirective) {
2933   ++NumIf;
2934 
2935   // Parse and evaluate the conditional expression.
2936   IdentifierInfo *IfNDefMacro = nullptr;
2937   const DirectiveEvalResult DER = EvaluateDirectiveExpression(IfNDefMacro);
2938   const bool ConditionalTrue = DER.Conditional;
2939 
2940   // If this condition is equivalent to #ifndef X, and if this is the first
2941   // directive seen, handle it for the multiple-include optimization.
2942   if (CurPPLexer->getConditionalStackDepth() == 0) {
2943     if (!ReadAnyTokensBeforeDirective && IfNDefMacro && ConditionalTrue)
2944       // FIXME: Pass in the location of the macro name, not the 'if' token.
2945       CurPPLexer->MIOpt.EnterTopLevelIfndef(IfNDefMacro, IfToken.getLocation());
2946     else
2947       CurPPLexer->MIOpt.EnterTopLevelConditional();
2948   }
2949 
2950   if (Callbacks)
2951     Callbacks->If(
2952         IfToken.getLocation(), DER.ExprRange,
2953         (ConditionalTrue ? PPCallbacks::CVK_True : PPCallbacks::CVK_False));
2954 
2955   bool RetainExcludedCB = PPOpts->RetainExcludedConditionalBlocks &&
2956     getSourceManager().isInMainFile(IfToken.getLocation());
2957 
2958   // Should we include the stuff contained by this directive?
2959   if (PPOpts->SingleFileParseMode && DER.IncludedUndefinedIds) {
2960     // In 'single-file-parse mode' undefined identifiers trigger parsing of all
2961     // the directive blocks.
2962     CurPPLexer->pushConditionalLevel(IfToken.getLocation(), /*wasskip*/false,
2963                                      /*foundnonskip*/false, /*foundelse*/false);
2964   } else if (ConditionalTrue || RetainExcludedCB) {
2965     // Yes, remember that we are inside a conditional, then lex the next token.
2966     CurPPLexer->pushConditionalLevel(IfToken.getLocation(), /*wasskip*/false,
2967                                    /*foundnonskip*/true, /*foundelse*/false);
2968   } else {
2969     // No, skip the contents of this block.
2970     SkipExcludedConditionalBlock(HashToken.getLocation(), IfToken.getLocation(),
2971                                  /*Foundnonskip*/ false,
2972                                  /*FoundElse*/ false);
2973   }
2974 }
2975 
2976 /// HandleEndifDirective - Implements the \#endif directive.
2977 ///
2978 void Preprocessor::HandleEndifDirective(Token &EndifToken) {
2979   ++NumEndif;
2980 
2981   // Check that this is the whole directive.
2982   CheckEndOfDirective("endif");
2983 
2984   PPConditionalInfo CondInfo;
2985   if (CurPPLexer->popConditionalLevel(CondInfo)) {
2986     // No conditionals on the stack: this is an #endif without an #if.
2987     Diag(EndifToken, diag::err_pp_endif_without_if);
2988     return;
2989   }
2990 
2991   // If this the end of a top-level #endif, inform MIOpt.
2992   if (CurPPLexer->getConditionalStackDepth() == 0)
2993     CurPPLexer->MIOpt.ExitTopLevelConditional();
2994 
2995   assert(!CondInfo.WasSkipping && !CurPPLexer->LexingRawMode &&
2996          "This code should only be reachable in the non-skipping case!");
2997 
2998   if (Callbacks)
2999     Callbacks->Endif(EndifToken.getLocation(), CondInfo.IfLoc);
3000 }
3001 
3002 /// HandleElseDirective - Implements the \#else directive.
3003 ///
3004 void Preprocessor::HandleElseDirective(Token &Result, const Token &HashToken) {
3005   ++NumElse;
3006 
3007   // #else directive in a non-skipping conditional... start skipping.
3008   CheckEndOfDirective("else");
3009 
3010   PPConditionalInfo CI;
3011   if (CurPPLexer->popConditionalLevel(CI)) {
3012     Diag(Result, diag::pp_err_else_without_if);
3013     return;
3014   }
3015 
3016   // If this is a top-level #else, inform the MIOpt.
3017   if (CurPPLexer->getConditionalStackDepth() == 0)
3018     CurPPLexer->MIOpt.EnterTopLevelConditional();
3019 
3020   // If this is a #else with a #else before it, report the error.
3021   if (CI.FoundElse) Diag(Result, diag::pp_err_else_after_else);
3022 
3023   if (Callbacks)
3024     Callbacks->Else(Result.getLocation(), CI.IfLoc);
3025 
3026   bool RetainExcludedCB = PPOpts->RetainExcludedConditionalBlocks &&
3027     getSourceManager().isInMainFile(Result.getLocation());
3028 
3029   if ((PPOpts->SingleFileParseMode && !CI.FoundNonSkip) || RetainExcludedCB) {
3030     // In 'single-file-parse mode' undefined identifiers trigger parsing of all
3031     // the directive blocks.
3032     CurPPLexer->pushConditionalLevel(CI.IfLoc, /*wasskip*/false,
3033                                      /*foundnonskip*/false, /*foundelse*/true);
3034     return;
3035   }
3036 
3037   // Finally, skip the rest of the contents of this block.
3038   SkipExcludedConditionalBlock(HashToken.getLocation(), CI.IfLoc,
3039                                /*Foundnonskip*/ true,
3040                                /*FoundElse*/ true, Result.getLocation());
3041 }
3042 
3043 /// HandleElifDirective - Implements the \#elif directive.
3044 ///
3045 void Preprocessor::HandleElifDirective(Token &ElifToken,
3046                                        const Token &HashToken) {
3047   ++NumElse;
3048 
3049   // #elif directive in a non-skipping conditional... start skipping.
3050   // We don't care what the condition is, because we will always skip it (since
3051   // the block immediately before it was included).
3052   SourceRange ConditionRange = DiscardUntilEndOfDirective();
3053 
3054   PPConditionalInfo CI;
3055   if (CurPPLexer->popConditionalLevel(CI)) {
3056     Diag(ElifToken, diag::pp_err_elif_without_if);
3057     return;
3058   }
3059 
3060   // If this is a top-level #elif, inform the MIOpt.
3061   if (CurPPLexer->getConditionalStackDepth() == 0)
3062     CurPPLexer->MIOpt.EnterTopLevelConditional();
3063 
3064   // If this is a #elif with a #else before it, report the error.
3065   if (CI.FoundElse) Diag(ElifToken, diag::pp_err_elif_after_else);
3066 
3067   if (Callbacks)
3068     Callbacks->Elif(ElifToken.getLocation(), ConditionRange,
3069                     PPCallbacks::CVK_NotEvaluated, CI.IfLoc);
3070 
3071   bool RetainExcludedCB = PPOpts->RetainExcludedConditionalBlocks &&
3072     getSourceManager().isInMainFile(ElifToken.getLocation());
3073 
3074   if ((PPOpts->SingleFileParseMode && !CI.FoundNonSkip) || RetainExcludedCB) {
3075     // In 'single-file-parse mode' undefined identifiers trigger parsing of all
3076     // the directive blocks.
3077     CurPPLexer->pushConditionalLevel(ElifToken.getLocation(), /*wasskip*/false,
3078                                      /*foundnonskip*/false, /*foundelse*/false);
3079     return;
3080   }
3081 
3082   // Finally, skip the rest of the contents of this block.
3083   SkipExcludedConditionalBlock(
3084       HashToken.getLocation(), CI.IfLoc, /*Foundnonskip*/ true,
3085       /*FoundElse*/ CI.FoundElse, ElifToken.getLocation());
3086 }
3087