PPDirectives.cpp revision 1a4761edca58c6b559de825b9abfb66f7f1ba94a
1a3b605ef0d535c5c5ac3f00e1e2d46fa7336d26cChris Lattner//===--- PPDirectives.cpp - Directive Handling for Preprocessor -----------===//
2141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//
3141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//                     The LLVM Compiler Infrastructure
4141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//
5141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// This file is distributed under the University of Illinois Open Source
6141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// License. See LICENSE.TXT for details.
7141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//
8141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
9141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//
10141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//  This file implements # directive processing for the Preprocessor.
11141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//
12141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
13141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
14141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner#include "clang/Lex/Preprocessor.h"
15359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner#include "clang/Lex/LiteralSupport.h"
16141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner#include "clang/Lex/HeaderSearch.h"
17141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner#include "clang/Lex/MacroInfo.h"
18500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/Lex/LexDiagnostic.h"
19f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor#include "clang/Lex/CodeCompletionHandler.h"
20fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor#include "clang/Lex/ModuleLoader.h"
2180c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor#include "clang/Lex/Pragma.h"
226e2901407bff59aeb4cc301cc58b034723d0eb49Chris Lattner#include "clang/Basic/FileManager.h"
23141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner#include "clang/Basic/SourceManager.h"
24359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner#include "llvm/ADT/APInt.h"
25e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor#include "llvm/Support/ErrorHandling.h"
26141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnerusing namespace clang;
27141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
28141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
29141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Utility Methods for Preprocessor Directive Handling.
30141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
31141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
32f47724bf78299c7a50f008e0443c5f9f9f279ddcChris LattnerMacroInfo *Preprocessor::AllocateMacroInfo() {
339714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  MacroInfoChain *MIChain;
341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
359714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  if (MICache) {
369714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    MIChain = MICache;
379714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    MICache = MICache->Next;
38af8fa25c0d4e0540952a50bbd06dc1558954ccd9Ted Kremenek  }
399714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  else {
409714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    MIChain = BP.Allocate<MacroInfoChain>();
419714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  }
429714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek
439714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  MIChain->Next = MIChainHead;
449714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  MIChain->Prev = 0;
459714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  if (MIChainHead)
469714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    MIChainHead->Prev = MIChain;
479714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  MIChainHead = MIChain;
489714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek
499714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  return &(MIChain->MI);
50f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner}
51f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner
52f47724bf78299c7a50f008e0443c5f9f9f279ddcChris LattnerMacroInfo *Preprocessor::AllocateMacroInfo(SourceLocation L) {
53f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner  MacroInfo *MI = AllocateMacroInfo();
540ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek  new (MI) MacroInfo(L);
550ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek  return MI;
560ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek}
570ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek
58f47724bf78299c7a50f008e0443c5f9f9f279ddcChris LattnerMacroInfo *Preprocessor::CloneMacroInfo(const MacroInfo &MacroToClone) {
59f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner  MacroInfo *MI = AllocateMacroInfo();
60f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner  new (MI) MacroInfo(MacroToClone, BP);
61f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner  return MI;
62f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner}
63f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner
640301b3ff132a4d986c092d161cb77d74b04cd2a6Chris Lattner/// ReleaseMacroInfo - Release the specified MacroInfo.  This memory will
650301b3ff132a4d986c092d161cb77d74b04cd2a6Chris Lattner///  be reused for allocating new MacroInfo objects.
662c1ab9079cb117dc0470ab423fe0bc5177546339Chris Lattnervoid Preprocessor::ReleaseMacroInfo(MacroInfo *MI) {
679714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  MacroInfoChain *MIChain = (MacroInfoChain*) MI;
689714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  if (MacroInfoChain *Prev = MIChain->Prev) {
699714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    MacroInfoChain *Next = MIChain->Next;
709714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    Prev->Next = Next;
719714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    if (Next)
729714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek      Next->Prev = Prev;
739714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  }
749714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  else {
759714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    assert(MIChainHead == MIChain);
769714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    MIChainHead = MIChain->Next;
779714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek    MIChainHead->Prev = 0;
789714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  }
799714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  MIChain->Next = MICache;
809714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  MICache = MIChain;
810301b3ff132a4d986c092d161cb77d74b04cd2a6Chris Lattner
829714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek  MI->Destroy();
839714a2385cb66b6efa373fc668641de602dd9adbTed Kremenek}
840301b3ff132a4d986c092d161cb77d74b04cd2a6Chris Lattner
85141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// DiscardUntilEndOfDirective - Read and discard all tokens remaining on the
8684021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne/// current line until the tok::eod token is found.
87141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::DiscardUntilEndOfDirective() {
88141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token Tmp;
89141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  do {
90141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    LexUnexpandedToken(Tmp);
91a5ef584fd3d18da0c98342b4b6453948b7eb30d3Peter Collingbourne    assert(Tmp.isNot(tok::eof) && "EOF seen while discarding directive tokens");
9284021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  } while (Tmp.isNot(tok::eod));
93141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
94141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
95141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// ReadMacroName - Lex and validate a macro name, which occurs after a
9684021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne/// #define or #undef.  This sets the token kind to eod and discards the rest
97141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// of the macro line if the macro name is invalid.  isDefineUndef is 1 if
98141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// this is due to a a #define, 2 if #undef directive, 0 if it is something
99141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// else (e.g. #ifdef).
100141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::ReadMacroName(Token &MacroNameTok, char isDefineUndef) {
101141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Read the token, don't allow macro expansion on it.
102141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  LexUnexpandedToken(MacroNameTok);
1031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1041fbb447e9d43c2c676e94081fbfee7eb6cbe933bDouglas Gregor  if (MacroNameTok.is(tok::code_completion)) {
1051fbb447e9d43c2c676e94081fbfee7eb6cbe933bDouglas Gregor    if (CodeComplete)
1061fbb447e9d43c2c676e94081fbfee7eb6cbe933bDouglas Gregor      CodeComplete->CodeCompleteMacroName(isDefineUndef == 1);
1077d100872341f233c81e1d7b72b40457e62c36862Argyrios Kyrtzidis    setCodeCompletionReached();
1081fbb447e9d43c2c676e94081fbfee7eb6cbe933bDouglas Gregor    LexUnexpandedToken(MacroNameTok);
1091fbb447e9d43c2c676e94081fbfee7eb6cbe933bDouglas Gregor  }
1101fbb447e9d43c2c676e94081fbfee7eb6cbe933bDouglas Gregor
111141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Missing macro name?
11284021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (MacroNameTok.is(tok::eod)) {
1133692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    Diag(MacroNameTok, diag::err_pp_missing_macro_name);
1143692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    return;
1153692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner  }
1161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
117141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  IdentifierInfo *II = MacroNameTok.getIdentifierInfo();
118141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (II == 0) {
119453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor    bool Invalid = false;
120453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor    std::string Spelling = getSpelling(MacroNameTok, &Invalid);
121453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor    if (Invalid)
122453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor      return;
123453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor
1249485d2382c1e3833ccbed20a185f6ab96d4dd703Chris Lattner    const IdentifierInfo &Info = Identifiers.get(Spelling);
1259485d2382c1e3833ccbed20a185f6ab96d4dd703Chris Lattner    if (Info.isCPlusPlusOperatorKeyword())
126141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // C++ 2.5p2: Alternative tokens behave the same as its primary token
127141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // except for their spellings.
12856b05c8829efd13b7e5b333f8f587c71d025c67dChris Lattner      Diag(MacroNameTok, diag::err_pp_operator_used_as_macro_name) << Spelling;
129141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    else
130141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(MacroNameTok, diag::err_pp_macro_not_identifier);
131141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Fall through on error.
132141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else if (isDefineUndef && II->getPPKeywordID() == tok::pp_defined) {
133141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Error if defining "defined": C99 6.10.8.4.
134141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Diag(MacroNameTok, diag::err_defined_macro_name);
135141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else if (isDefineUndef && II->hasMacroDefinition() &&
136141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner             getMacroInfo(II)->isBuiltinMacro()) {
137141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Error if defining "__LINE__" and other builtins: C99 6.10.8.4.
138141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (isDefineUndef == 1)
139141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(MacroNameTok, diag::pp_redef_builtin_macro);
140141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    else
141141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(MacroNameTok, diag::pp_undef_builtin_macro);
142141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else {
143141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Okay, we got a good identifier node.  Return it.
144141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return;
145141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
1461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
147141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Invalid macro name, read and discard the rest of the line.  Then set the
14884021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // token kind to tok::eod.
14984021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  MacroNameTok.setKind(tok::eod);
150141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  return DiscardUntilEndOfDirective();
151141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
152141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
15384021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne/// CheckEndOfDirective - Ensure that the next token is a tok::eod token.  If
15484021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne/// not, emit a diagnostic and consume up until the eod.  If EnableMacros is
155ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner/// true, then we consider macros that expand to zero tokens as being ok.
156ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattnervoid Preprocessor::CheckEndOfDirective(const char *DirType, bool EnableMacros) {
157141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token Tmp;
158ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner  // Lex unexpanded tokens for most directives: macros might expand to zero
159ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner  // tokens, causing us to miss diagnosing invalid lines.  Some directives (like
160ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner  // #line) allow empty macros.
161ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner  if (EnableMacros)
162ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner    Lex(Tmp);
163ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner  else
164ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner    LexUnexpandedToken(Tmp);
1651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
166141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // There should be no tokens after the directive, but we allow them as an
167141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // extension.
168141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  while (Tmp.is(tok::comment))  // Skip comments in -C mode.
169141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    LexUnexpandedToken(Tmp);
1701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
17184021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (Tmp.isNot(tok::eod)) {
172959875a66d74d38b8f2d499662afb142d249bf3eChris Lattner    // Add a fixit in GNU/C99/C++ mode.  Don't offer a fixit for strict-C89,
173b2eb53d9fd973a1a02e05e67a3307b3efd12eff2Peter Collingbourne    // or if this is a macro-style preprocessing directive, because it is more
174b2eb53d9fd973a1a02e05e67a3307b3efd12eff2Peter Collingbourne    // trouble than it is worth to insert /**/ and check that there is no /**/
175b2eb53d9fd973a1a02e05e67a3307b3efd12eff2Peter Collingbourne    // in the range also.
176849b243d4065f56742a4677d6dc8277609a151f8Douglas Gregor    FixItHint Hint;
177b2eb53d9fd973a1a02e05e67a3307b3efd12eff2Peter Collingbourne    if ((Features.GNUMode || Features.C99 || Features.CPlusPlus) &&
178b2eb53d9fd973a1a02e05e67a3307b3efd12eff2Peter Collingbourne        !CurTokenLexer)
179849b243d4065f56742a4677d6dc8277609a151f8Douglas Gregor      Hint = FixItHint::CreateInsertion(Tmp.getLocation(),"//");
180849b243d4065f56742a4677d6dc8277609a151f8Douglas Gregor    Diag(Tmp, diag::ext_pp_extra_tokens_at_eol) << DirType << Hint;
181141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    DiscardUntilEndOfDirective();
182141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
183141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
184141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
185141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
186141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
187141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// SkipExcludedConditionalBlock - We just read a #if or related directive and
188141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// decided that the subsequent tokens are in the #if'd out portion of the
189141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// file.  Lex the rest of the file, until we see an #endif.  If
190141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// FoundNonSkipPortion is true, then we have already emitted code for part of
191141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// this #if directive, so #else/#elif blocks should never be entered. If ElseOk
192141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// is true, then #else directives are ok, if not, then we have already seen one
193141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// so a #else directive is a duplicate.  When this returns, the caller can lex
194141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// the first valid token.
195141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::SkipExcludedConditionalBlock(SourceLocation IfTokenLoc,
196141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                                bool FoundNonSkipPortion,
1976b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis                                                bool FoundElse,
1986b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis                                                SourceLocation ElseLoc) {
199141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumSkipped;
200f6452c5421c5db5a7ceff581525f286931d97f1aTed Kremenek  assert(CurTokenLexer == 0 && CurPPLexer && "Lexing a macro, not a file?");
201141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
20260e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  CurPPLexer->pushConditionalLevel(IfTokenLoc, /*isSkipping*/false,
203141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                 FoundNonSkipPortion, FoundElse);
2041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
205268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  if (CurPTHLexer) {
206268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    PTHSkipExcludedConditionalBlock();
207268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    return;
208268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  }
2091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
210141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Enter raw mode to disable identifier lookup (and thus macro expansion),
211141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // disabling warnings, etc.
21260e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  CurPPLexer->LexingRawMode = true;
213141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token Tok;
214141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  while (1) {
2152c6b193d574e0780ff0b8450ce1fde105f4881acChris Lattner    CurLexer->Lex(Tok);
2161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
217f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor    if (Tok.is(tok::code_completion)) {
218f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor      if (CodeComplete)
219f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor        CodeComplete->CodeCompleteInConditionalExclusion();
2207d100872341f233c81e1d7b72b40457e62c36862Argyrios Kyrtzidis      setCodeCompletionReached();
221f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor      continue;
222f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor    }
223f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor
224141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // If this is the end of the buffer, we have an error.
225141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (Tok.is(tok::eof)) {
226141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Emit errors for each unterminated conditional on the stack, including
227141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // the current one.
22860e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek      while (!CurPPLexer->ConditionalStack.empty()) {
2297d100872341f233c81e1d7b72b40457e62c36862Argyrios Kyrtzidis        if (CurLexer->getFileLoc() != CodeCompletionFileLoc)
2302d474ba9e8ae43a1a5a9f72718c0d79092b9453fDouglas Gregor          Diag(CurPPLexer->ConditionalStack.back().IfLoc,
2312d474ba9e8ae43a1a5a9f72718c0d79092b9453fDouglas Gregor               diag::err_pp_unterminated_conditional);
23260e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek        CurPPLexer->ConditionalStack.pop_back();
2331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      }
2341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
235141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Just return and let the caller lex after this #include.
236141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      break;
237141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
2381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
239141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // If this token is not a preprocessor directive, just skip it.
240141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (Tok.isNot(tok::hash) || !Tok.isAtStartOfLine())
241141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      continue;
2421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
243141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // We just parsed a # character at the start of a line, so we're in
244141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // directive mode.  Tell the lexer this so any newlines we see will be
24584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    // converted into an EOD token (this terminates the macro).
24660e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek    CurPPLexer->ParsingPreprocessorDirective = true;
247ac6b06df32e907cac388a92310650b5895c11241Ted Kremenek    if (CurLexer) CurLexer->SetCommentRetentionState(false);
248141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
2491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
250141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Read the next token, the directive flavor.
251141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    LexUnexpandedToken(Tok);
2521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
253141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // If this isn't an identifier directive (e.g. is "# 1\n" or "#\n", or
254141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // something bogus), skip it.
255c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    if (Tok.isNot(tok::raw_identifier)) {
25660e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek      CurPPLexer->ParsingPreprocessorDirective = false;
257141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Restore comment saving mode.
258ac6b06df32e907cac388a92310650b5895c11241Ted Kremenek      if (CurLexer) CurLexer->SetCommentRetentionState(KeepComments);
259141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      continue;
260141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
261141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
262141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // If the first letter isn't i or e, it isn't intesting to us.  We know that
263141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // this is safe in the face of spelling differences, because there is no way
264141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // to spell an i/e in a strange way that is another letter.  Skipping this
265141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // allows us to avoid looking up the identifier info for #define/#undef and
266141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // other common directives.
267c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    const char *RawCharData = Tok.getRawIdentifierData();
268c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara
269141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    char FirstChar = RawCharData[0];
2701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    if (FirstChar >= 'a' && FirstChar <= 'z' &&
271141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        FirstChar != 'i' && FirstChar != 'e') {
27260e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek      CurPPLexer->ParsingPreprocessorDirective = false;
273141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Restore comment saving mode.
274ac6b06df32e907cac388a92310650b5895c11241Ted Kremenek      if (CurLexer) CurLexer->SetCommentRetentionState(KeepComments);
275141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      continue;
276141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
2771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
278141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Get the identifier name without trigraphs or embedded newlines.  Note
279141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // that we can't use Tok.getIdentifierInfo() because its lookup is disabled
280141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // when skipping.
281b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer    char DirectiveBuf[20];
2825f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Directive;
283141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (!Tok.needsCleaning() && Tok.getLength() < 20) {
2845f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      Directive = StringRef(RawCharData, Tok.getLength());
285141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    } else {
286141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      std::string DirectiveStr = getSpelling(Tok);
287b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer      unsigned IdLen = DirectiveStr.size();
288141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      if (IdLen >= 20) {
28960e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek        CurPPLexer->ParsingPreprocessorDirective = false;
290141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // Restore comment saving mode.
291ac6b06df32e907cac388a92310650b5895c11241Ted Kremenek        if (CurLexer) CurLexer->SetCommentRetentionState(KeepComments);
292141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        continue;
293141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
294b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer      memcpy(DirectiveBuf, &DirectiveStr[0], IdLen);
2955f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      Directive = StringRef(DirectiveBuf, IdLen);
296141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
2971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
298b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer    if (Directive.startswith("if")) {
2995f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      StringRef Sub = Directive.substr(2);
300b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer      if (Sub.empty() ||   // "if"
301b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer          Sub == "def" ||   // "ifdef"
302b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer          Sub == "ndef") {  // "ifndef"
303141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // We know the entire #if/#ifdef/#ifndef block will be skipped, don't
304141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // bother parsing the condition.
305141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        DiscardUntilEndOfDirective();
30660e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek        CurPPLexer->pushConditionalLevel(Tok.getLocation(), /*wasskipping*/true,
307141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                       /*foundnonskip*/false,
3083a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth                                       /*foundelse*/false);
3093a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth
3103a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth        if (Callbacks)
3113a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth          Callbacks->Endif();
312141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
313b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer    } else if (Directive[0] == 'e') {
3145f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      StringRef Sub = Directive.substr(1);
315b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer      if (Sub == "ndif") {  // "endif"
31635410d5cbbb18735d76e00496540d416dc49b577Chris Lattner        CheckEndOfDirective("endif");
317141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        PPConditionalInfo CondInfo;
318141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        CondInfo.WasSkipping = true; // Silence bogus warning.
31960e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek        bool InCond = CurPPLexer->popConditionalLevel(CondInfo);
320c6ed729f669044f5072a49d79041f455d971ece3Jeffrey Yasskin        (void)InCond;  // Silence warning in no-asserts mode.
321141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        assert(!InCond && "Can't be skipping if not in a conditional!");
3221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
323141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // If we popped the outermost skipping block, we're done skipping!
324141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        if (!CondInfo.WasSkipping)
325141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          break;
326b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer      } else if (Sub == "lse") { // "else".
327141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // #else directive in a skipping conditional.  If not in some other
328141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // skipping conditional, and if #else hasn't already been seen, enter it
329141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // as a non-skipping conditional.
33060e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek        PPConditionalInfo &CondInfo = CurPPLexer->peekConditionalLevel();
3311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
332141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // If this is a #else with a #else before it, report the error.
333141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        if (CondInfo.FoundElse) Diag(Tok, diag::pp_err_else_after_else);
3341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
335141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // Note that we've seen a #else in this conditional.
336141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        CondInfo.FoundElse = true;
3371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3383a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth        if (Callbacks)
3393a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth          Callbacks->Else();
3403a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth
341141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // If the conditional is at the top level, and the #if block wasn't
342141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // entered, enter the #else block now.
343141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        if (!CondInfo.WasSkipping && !CondInfo.FoundNonSkip) {
344141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          CondInfo.FoundNonSkip = true;
345e26224e61c8a3b607c73c52f2cd0344065d7b412Argyrios Kyrtzidis          CheckEndOfDirective("else");
346141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          break;
347e26224e61c8a3b607c73c52f2cd0344065d7b412Argyrios Kyrtzidis        } else {
348e26224e61c8a3b607c73c52f2cd0344065d7b412Argyrios Kyrtzidis          DiscardUntilEndOfDirective();  // C99 6.10p4.
349141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        }
350b939a4e8b471f6a5e996fa57b2fd9df30b59cb60Benjamin Kramer      } else if (Sub == "lif") {  // "elif".
35160e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek        PPConditionalInfo &CondInfo = CurPPLexer->peekConditionalLevel();
352141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
353141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        bool ShouldEnter;
3543a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth        const SourceLocation ConditionalBegin = CurPPLexer->getSourceLocation();
355141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // If this is in a skipping block or if we're already handled this #if
356141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // block, don't bother parsing the condition.
357141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        if (CondInfo.WasSkipping || CondInfo.FoundNonSkip) {
358141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          DiscardUntilEndOfDirective();
359141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          ShouldEnter = false;
360141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        } else {
361141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          // Restore the value of LexingRawMode so that identifiers are
362141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          // looked up, etc, inside the #elif expression.
36360e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek          assert(CurPPLexer->LexingRawMode && "We have to be skipping here!");
36460e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek          CurPPLexer->LexingRawMode = false;
365141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          IdentifierInfo *IfNDefMacro = 0;
366141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          ShouldEnter = EvaluateDirectiveExpression(IfNDefMacro);
36760e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek          CurPPLexer->LexingRawMode = true;
368141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        }
3693a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth        const SourceLocation ConditionalEnd = CurPPLexer->getSourceLocation();
3701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
371141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // If this is a #elif with a #else before it, report the error.
372141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        if (CondInfo.FoundElse) Diag(Tok, diag::pp_err_elif_after_else);
3731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3743a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth        if (Callbacks)
3753a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth          Callbacks->Elif(SourceRange(ConditionalBegin, ConditionalEnd));
3763a1a8749cf6239f29e834f18a4e08638126678d9Chandler Carruth
377141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // If this condition is true, enter it!
378141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        if (ShouldEnter) {
379141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          CondInfo.FoundNonSkip = true;
380141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          break;
381141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        }
382141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
383141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
3841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
38560e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek    CurPPLexer->ParsingPreprocessorDirective = false;
386141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Restore comment saving mode.
387ac6b06df32e907cac388a92310650b5895c11241Ted Kremenek    if (CurLexer) CurLexer->SetCommentRetentionState(KeepComments);
388141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
389141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
390141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Finally, if we are out of the conditional (saw an #endif or ran off the end
391141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // of the file, just stop skipping and return to lexing whatever came after
392141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // the #if block.
39360e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  CurPPLexer->LexingRawMode = false;
3946b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis
3956b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis  if (Callbacks) {
3966b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis    SourceLocation BeginLoc = ElseLoc.isValid() ? ElseLoc : IfTokenLoc;
3976b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis    Callbacks->SourceRangeSkipped(SourceRange(BeginLoc, Tok.getLocation()));
3986b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis  }
399141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
400141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
401268ee7016a2811803989487c0ad3799486092c63Ted Kremenekvoid Preprocessor::PTHSkipExcludedConditionalBlock() {
4021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  while (1) {
404268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    assert(CurPTHLexer);
405268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    assert(CurPTHLexer->LexingRawMode == false);
4061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
407268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // Skip to the next '#else', '#elif', or #endif.
408268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    if (CurPTHLexer->SkipBlock()) {
409268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      // We have reached an #endif.  Both the '#' and 'endif' tokens
410268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      // have been consumed by the PTHLexer.  Just pop off the condition level.
411268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      PPConditionalInfo CondInfo;
412268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      bool InCond = CurPTHLexer->popConditionalLevel(CondInfo);
413c6ed729f669044f5072a49d79041f455d971ece3Jeffrey Yasskin      (void)InCond;  // Silence warning in no-asserts mode.
414268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      assert(!InCond && "Can't be skipping if not in a conditional!");
415268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      break;
416268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    }
4171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
418268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // We have reached a '#else' or '#elif'.  Lex the next token to get
419268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // the directive flavor.
420268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    Token Tok;
421268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    LexUnexpandedToken(Tok);
4221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
423268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // We can actually look up the IdentifierInfo here since we aren't in
424268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // raw mode.
425268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    tok::PPKeywordKind K = Tok.getIdentifierInfo()->getPPKeywordID();
426268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
427268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    if (K == tok::pp_else) {
428268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      // #else: Enter the else condition.  We aren't in a nested condition
429268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      //  since we skip those. We're always in the one matching the last
430268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      //  blocked we skipped.
431268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      PPConditionalInfo &CondInfo = CurPTHLexer->peekConditionalLevel();
432268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      // Note that we've seen a #else in this conditional.
433268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      CondInfo.FoundElse = true;
4341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
435268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      // If the #if block wasn't entered then enter the #else block now.
436268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      if (!CondInfo.FoundNonSkip) {
437268ee7016a2811803989487c0ad3799486092c63Ted Kremenek        CondInfo.FoundNonSkip = true;
4381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
43984021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne        // Scan until the eod token.
440e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek        CurPTHLexer->ParsingPreprocessorDirective = true;
4418533bd5a32a12c51341e82d2aef083acbac7c04cDaniel Dunbar        DiscardUntilEndOfDirective();
442e5680f3cd678014cf0872d34726dc804b0cbbdd4Ted Kremenek        CurPTHLexer->ParsingPreprocessorDirective = false;
4431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
444268ee7016a2811803989487c0ad3799486092c63Ted Kremenek        break;
445268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      }
4461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
447268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      // Otherwise skip this block.
448268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      continue;
449268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    }
4501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
451268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    assert(K == tok::pp_elif);
452268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    PPConditionalInfo &CondInfo = CurPTHLexer->peekConditionalLevel();
453268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
454268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // If this is a #elif with a #else before it, report the error.
455268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    if (CondInfo.FoundElse)
456268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      Diag(Tok, diag::pp_err_elif_after_else);
4571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
458268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // If this is in a skipping block or if we're already handled this #if
4591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // block, don't bother parsing the condition.  We just skip this block.
460268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    if (CondInfo.FoundNonSkip)
461268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      continue;
462268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
463268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // Evaluate the condition of the #elif.
464268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    IdentifierInfo *IfNDefMacro = 0;
465268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    CurPTHLexer->ParsingPreprocessorDirective = true;
466268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    bool ShouldEnter = EvaluateDirectiveExpression(IfNDefMacro);
467268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    CurPTHLexer->ParsingPreprocessorDirective = false;
468268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
469268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // If this condition is true, enter it!
470268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    if (ShouldEnter) {
471268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      CondInfo.FoundNonSkip = true;
472268ee7016a2811803989487c0ad3799486092c63Ted Kremenek      break;
473268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    }
474268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
475268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    // Otherwise, skip this block and go to the next one.
476268ee7016a2811803989487c0ad3799486092c63Ted Kremenek    continue;
477268ee7016a2811803989487c0ad3799486092c63Ted Kremenek  }
478268ee7016a2811803989487c0ad3799486092c63Ted Kremenek}
479268ee7016a2811803989487c0ad3799486092c63Ted Kremenek
4801072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner/// LookupFile - Given a "foo" or <foo> reference, look up the indicated file,
4811072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner/// return null on failure.  isAngled indicates whether the file reference is
4821072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner/// for system #include's or not (i.e. using <> instead of "").
483b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruthconst FileEntry *Preprocessor::LookupFile(
4845f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Filename,
485b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruth    bool isAngled,
486b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruth    const DirectoryLookup *FromDir,
487b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruth    const DirectoryLookup *&CurDir,
4885f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    SmallVectorImpl<char> *SearchPath,
489fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor    SmallVectorImpl<char> *RelativePath,
4901a4761edca58c6b559de825b9abfb66f7f1ba94aDouglas Gregor    Module **SuggestedModule,
4911c2e9332fa69727425a3a2b912e36e2ab62083f8Douglas Gregor    bool SkipCache) {
4921072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  // If the header lookup mechanism may be relative to the current file, pass in
4931072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  // info about where the current file is.
49410fe93d57c11f068aa4d78eb4ca7f60329818306Douglas Gregor  const FileEntry *CurFileEnt = 0;
4951072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  if (!FromDir) {
4962b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner    FileID FID = getCurrentFileLexer()->getFileID();
49710fe93d57c11f068aa4d78eb4ca7f60329818306Douglas Gregor    CurFileEnt = SourceMgr.getFileEntryForID(FID);
4981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
499be5c64d7a765acaf9454e7a511233771472bf811Chris Lattner    // If there is no file entry associated with this file, it must be the
500be5c64d7a765acaf9454e7a511233771472bf811Chris Lattner    // predefines buffer.  Any other file is not lexed with a normal lexer, so
50110fe93d57c11f068aa4d78eb4ca7f60329818306Douglas Gregor    // it won't be scanned for preprocessor directives.   If we have the
50210fe93d57c11f068aa4d78eb4ca7f60329818306Douglas Gregor    // predefines buffer, resolve #include references (which come from the
50310fe93d57c11f068aa4d78eb4ca7f60329818306Douglas Gregor    // -include command line argument) as if they came from the main file, this
50410fe93d57c11f068aa4d78eb4ca7f60329818306Douglas Gregor    // affects file lookup etc.
50510fe93d57c11f068aa4d78eb4ca7f60329818306Douglas Gregor    if (CurFileEnt == 0) {
506be5c64d7a765acaf9454e7a511233771472bf811Chris Lattner      FID = SourceMgr.getMainFileID();
507be5c64d7a765acaf9454e7a511233771472bf811Chris Lattner      CurFileEnt = SourceMgr.getFileEntryForID(FID);
508be5c64d7a765acaf9454e7a511233771472bf811Chris Lattner    }
5091072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  }
5101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5111072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  // Do a standard file entry lookup.
5121072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  CurDir = CurDirLookup;
513b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruth  const FileEntry *FE = HeaderInfo.LookupFile(
5147412494982c8b50c90961302c3a718633b2c3ab7Manuel Klimek      Filename, isAngled, FromDir, CurDir, CurFileEnt,
5151c2e9332fa69727425a3a2b912e36e2ab62083f8Douglas Gregor      SearchPath, RelativePath, SuggestedModule, SkipCache);
516f45b646244705410866d62f1d8bf017a047ed662Chris Lattner  if (FE) return FE;
5171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5181072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  // Otherwise, see if this is a subframework header.  If so, this is relative
5191072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  // to one of the headers on the #include stack.  Walk the list of the current
5201072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  // headers on the #include stack and pass them to HeaderInfo.
521fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor  // FIXME: SuggestedModule!
52281d24e135c60dcbe421864e6a4334ba99b8fe7ecTed Kremenek  if (IsFileLexer()) {
52341938c8493b4380df738263166b746eacb33c309Ted Kremenek    if ((CurFileEnt = SourceMgr.getFileEntryForID(CurPPLexer->getFileID())))
524b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruth      if ((FE = HeaderInfo.LookupSubframeworkHeader(Filename, CurFileEnt,
5257412494982c8b50c90961302c3a718633b2c3ab7Manuel Klimek                                                    SearchPath, RelativePath)))
5261072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner        return FE;
5271072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  }
5281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5291072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  for (unsigned i = 0, e = IncludeMacroStack.size(); i != e; ++i) {
5301072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner    IncludeStackInfo &ISEntry = IncludeMacroStack[e-i-1];
53181d24e135c60dcbe421864e6a4334ba99b8fe7ecTed Kremenek    if (IsFileLexer(ISEntry)) {
5321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      if ((CurFileEnt =
53341938c8493b4380df738263166b746eacb33c309Ted Kremenek           SourceMgr.getFileEntryForID(ISEntry.ThePPLexer->getFileID())))
5347412494982c8b50c90961302c3a718633b2c3ab7Manuel Klimek        if ((FE = HeaderInfo.LookupSubframeworkHeader(
5357412494982c8b50c90961302c3a718633b2c3ab7Manuel Klimek                Filename, CurFileEnt, SearchPath, RelativePath)))
5361072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner          return FE;
5371072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner    }
5381072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  }
5391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
5401072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  // Otherwise, we really couldn't find the file.
5411072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner  return 0;
5421072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner}
5431072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner
544141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
545141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
546141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Preprocessor Directive Handling.
547141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
548141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
549141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleDirective - This callback is invoked when the lexer sees a # token
5501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// at the start of a line.  This consumes the directive, modifies the
551141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// lexer/preprocessor state, and advances the lexer(s) so that the next token
552141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// read is the correct one.
553141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleDirective(Token &Result) {
554141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // FIXME: Traditional: # with whitespace before it not recognized by K&R?
5551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
556141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // We just parsed a # character at the start of a line, so we're in directive
557141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // mode.  Tell the lexer this so any newlines we see will be converted into an
55884021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // EOD token (which terminates the directive).
55960e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  CurPPLexer->ParsingPreprocessorDirective = true;
5601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
561141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumDirectives;
562193575455e00eca03fd7177f60e3f2e6263cb661Kovarththanan Rajaratnam
563141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // We are about to read a token.  For the multiple-include optimization FA to
5641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // work, we have to remember if we had read any tokens *before* this
565141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // pp-directive.
5661d9c54df56391ac4740db27d551782e81189cb51Chris Lattner  bool ReadAnyTokensBeforeDirective =CurPPLexer->MIOpt.getHasReadAnyTokensVal();
5671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
56842aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner  // Save the '#' token in case we need to return it later.
56942aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner  Token SavedHash = Result;
5701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
571141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Read the next token, the directive flavor.  This isn't expanded due to
572141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // C99 6.10.3p8.
573141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  LexUnexpandedToken(Result);
5741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
575141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // C99 6.10.3p11: Is this preprocessor directive in macro invocation?  e.g.:
576141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  //   #define A(x) #x
577141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  //   A(abc
578141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  //     #warning blah
579141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  //   def)
580141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If so, the user is relying on non-portable behavior, emit a diagnostic.
581141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (InMacroArgs)
582141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Diag(Result, diag::ext_embedded_directive);
5831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
584141e71fec621f65ab401dbe66609f7d3c033c6a5Chris LattnerTryAgain:
585141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  switch (Result.getKind()) {
58684021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  case tok::eod:
587141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return;   // null directive.
588141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  case tok::comment:
589141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Handle stuff like "# /*foo*/ define X" in -E -C mode.
590141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    LexUnexpandedToken(Result);
591141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    goto TryAgain;
592f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor  case tok::code_completion:
593f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor    if (CodeComplete)
594f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor      CodeComplete->CodeCompleteDirective(
595f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor                                    CurPPLexer->getConditionalStackDepth() > 0);
5967d100872341f233c81e1d7b72b40457e62c36862Argyrios Kyrtzidis    setCodeCompletionReached();
597f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor    return;
598478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  case tok::numeric_constant:  // # 7  GNU line marker directive.
5995f607c4b75abeedded11eff079dee73561d98203Chris Lattner    if (getLangOptions().AsmPreprocessor)
6005f607c4b75abeedded11eff079dee73561d98203Chris Lattner      break;  // # 4 is not a preprocessor directive in .S files.
601478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return HandleDigitDirective(Result);
602141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  default:
603141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    IdentifierInfo *II = Result.getIdentifierInfo();
604141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (II == 0) break;  // Not an identifier.
6051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
606141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Ask what the preprocessor keyword ID is.
607141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    switch (II->getPPKeywordID()) {
608141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    default: break;
609141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // C99 6.10.1 - Conditional Inclusion.
610141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_if:
611141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleIfDirective(Result, ReadAnyTokensBeforeDirective);
612141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_ifdef:
613141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleIfdefDirective(Result, false, true/*not valid for miopt*/);
614141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_ifndef:
615141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleIfdefDirective(Result, true, ReadAnyTokensBeforeDirective);
616141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_elif:
617141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleElifDirective(Result);
618141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_else:
619141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleElseDirective(Result);
620141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_endif:
621141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleEndifDirective(Result);
6221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
623141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // C99 6.10.2 - Source File Inclusion.
624141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_include:
625ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      // Handle #include.
626ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      return HandleIncludeDirective(SavedHash.getLocation(), Result);
627b8e240ed8b8f58c7b7fe82776bd6147b437f5d53Chris Lattner    case tok::pp___include_macros:
628ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      // Handle -imacros.
629ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      return HandleIncludeMacrosDirective(SavedHash.getLocation(), Result);
6301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
631141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // C99 6.10.3 - Macro Replacement.
632141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_define:
633141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleDefineDirective(Result);
634141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_undef:
635141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleUndefDirective(Result);
636141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
637141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // C99 6.10.4 - Line Control.
638141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_line:
639359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner      return HandleLineDirective(Result);
6401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
641141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // C99 6.10.5 - Error Directive.
642141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_error:
643141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleUserDiagnosticDirective(Result, false);
6441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
645141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // C99 6.10.6 - Pragma Directive.
646141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_pragma:
64780c60f72848896f867f6b7e664e7060d9e78f019Douglas Gregor      return HandlePragmaDirective(PIK_HashPragma);
6481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
649141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // GNU Extensions.
650141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_import:
651ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      return HandleImportDirective(SavedHash.getLocation(), Result);
652141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_include_next:
653ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      return HandleIncludeNextDirective(SavedHash.getLocation(), Result);
6541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
655141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_warning:
656141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(Result, diag::ext_pp_warning_directive);
657141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleUserDiagnosticDirective(Result, true);
658141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_ident:
659141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleIdentSCCSDirective(Result);
660141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_sccs:
661141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return HandleIdentSCCSDirective(Result);
662141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_assert:
663141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      //isExtension = true;  // FIXME: implement #assert
664141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      break;
665141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::pp_unassert:
666141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      //isExtension = true;  // FIXME: implement #unassert
667141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      break;
6687143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor
6697143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor    case tok::pp___export_macro__:
6707143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor      return HandleMacroExportDirective(Result);
671aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor    case tok::pp___private_macro__:
672aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor      return HandleMacroPrivateDirective(Result);
673141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
674141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    break;
675141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
6761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
67742aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner  // If this is a .S file, treat unknown # directives as non-preprocessor
67842aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner  // directives.  This is important because # may be a comment or introduce
67942aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner  // various pseudo-ops.  Just return the # token and push back the following
68042aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner  // token to be lexed next time.
68142aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner  if (getLangOptions().AsmPreprocessor) {
6823d399a010102966158f6b69b07d636f113dc6ab4Daniel Dunbar    Token *Toks = new Token[2];
68342aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner    // Return the # and the token after it.
6841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    Toks[0] = SavedHash;
68542aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner    Toks[1] = Result;
686ba3ca5271a01eae7cbd6a19ace004d75012c02a8Chris Lattner
687ba3ca5271a01eae7cbd6a19ace004d75012c02a8Chris Lattner    // If the second token is a hashhash token, then we need to translate it to
688ba3ca5271a01eae7cbd6a19ace004d75012c02a8Chris Lattner    // unknown so the token lexer doesn't try to perform token pasting.
689ba3ca5271a01eae7cbd6a19ace004d75012c02a8Chris Lattner    if (Result.is(tok::hashhash))
690ba3ca5271a01eae7cbd6a19ace004d75012c02a8Chris Lattner      Toks[1].setKind(tok::unknown);
691ba3ca5271a01eae7cbd6a19ace004d75012c02a8Chris Lattner
69242aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner    // Enter this token stream so that we re-lex the tokens.  Make sure to
69342aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner    // enable macro expansion, in case the token after the # is an identifier
69442aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner    // that is expanded.
69542aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner    EnterTokenStream(Toks, 2, false, true);
69642aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner    return;
69742aa16cccb3d6cd4195ccff9c32d8cb50a25a094Chris Lattner  }
6981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
699141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If we reached here, the preprocessing token is not valid!
700141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Diag(Result, diag::err_pp_invalid_directive);
7011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
702141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Read the rest of the PP line.
703141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  DiscardUntilEndOfDirective();
7041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
705141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Okay, we're done parsing the directive.
706141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
707141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
708478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner/// GetLineValue - Convert a numeric token into an unsigned value, emitting
709478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner/// Diagnostic DiagID if it is invalid, and returning the value in Val.
710478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattnerstatic bool GetLineValue(Token &DigitTok, unsigned &Val,
711478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner                         unsigned DiagID, Preprocessor &PP) {
712359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  if (DigitTok.isNot(tok::numeric_constant)) {
713478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    PP.Diag(DigitTok, DiagID);
7141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
71584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    if (DigitTok.isNot(tok::eod))
716478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner      PP.DiscardUntilEndOfDirective();
717478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return true;
718359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  }
7191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
720359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  llvm::SmallString<64> IntegerBuffer;
721359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  IntegerBuffer.resize(DigitTok.getLength());
722359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  const char *DigitTokBegin = &IntegerBuffer[0];
723453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor  bool Invalid = false;
724453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor  unsigned ActualLength = PP.getSpelling(DigitTok, DigitTokBegin, &Invalid);
725453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor  if (Invalid)
726453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor    return true;
727453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor
728dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner  // Verify that we have a simple digit-sequence, and compute the value.  This
729dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner  // is always a simple digit string computed in decimal, so we do this manually
730dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner  // here.
731dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner  Val = 0;
732dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner  for (unsigned i = 0; i != ActualLength; ++i) {
733dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner    if (!isdigit(DigitTokBegin[i])) {
734dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(DigitTok.getLocation(), i),
735dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner              diag::err_pp_line_digit_sequence);
736dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner      PP.DiscardUntilEndOfDirective();
737dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner      return true;
738dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner    }
7391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
740dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner    unsigned NextVal = Val*10+(DigitTokBegin[i]-'0');
741dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner    if (NextVal < Val) { // overflow.
742dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner      PP.Diag(DigitTok, DiagID);
743dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner      PP.DiscardUntilEndOfDirective();
744dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner      return true;
745dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner    }
746dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner    Val = NextVal;
747359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  }
7481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Reject 0, this is needed both by #line numbers and flags.
750478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  if (Val == 0) {
751478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    PP.Diag(DigitTok, DiagID);
752478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    PP.DiscardUntilEndOfDirective();
753478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return true;
754359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  }
7551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
756dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner  if (DigitTokBegin[0] == '0')
757dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner    PP.Diag(DigitTok.getLocation(), diag::warn_pp_line_decimal);
7581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
759478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  return false;
760478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner}
761478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
7621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// HandleLineDirective - Handle #line directive: C99 6.10.4.  The two
763478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner/// acceptable forms are:
764478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner///   # line digit-sequence
765478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner///   # line digit-sequence "s-char-sequence"
766478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattnervoid Preprocessor::HandleLineDirective(Token &Tok) {
767478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // Read the line # and string argument.  Per C99 6.10.4p5, these tokens are
768478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // expanded.
769478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  Token DigitTok;
770478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  Lex(DigitTok);
771478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
772478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // Validate the number and convert it to an unsigned.
773478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  unsigned LineNo;
774dc8c90da158030c0d0c62539e2da0adccc09f271Chris Lattner  if (GetLineValue(DigitTok, LineNo, diag::err_pp_line_requires_integer,*this))
775478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return;
776478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
777478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // Enforce C99 6.10.4p3: "The digit sequence shall not specify ... a
778478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // number greater than 2147483647".  C90 requires that the line # be <= 32767.
779158ebfba7545b3a9947f87bed83ea710df95ac3dEli Friedman  unsigned LineLimit = 32768U;
780158ebfba7545b3a9947f87bed83ea710df95ac3dEli Friedman  if (Features.C99 || Features.CPlusPlus0x)
781158ebfba7545b3a9947f87bed83ea710df95ac3dEli Friedman    LineLimit = 2147483648U;
782359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  if (LineNo >= LineLimit)
783359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner    Diag(DigitTok, diag::ext_pp_line_too_big) << LineLimit;
784661a99690bc133bbaa029da925481d4a860dec90Richard Smith  else if (Features.CPlusPlus0x && LineNo >= 32768U)
785661a99690bc133bbaa029da925481d4a860dec90Richard Smith    Diag(DigitTok, diag::warn_cxx98_compat_pp_line_too_big);
7861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7875b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner  int FilenameID = -1;
788359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  Token StrTok;
789359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  Lex(StrTok);
790359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner
79184021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // If the StrTok is "eod", then it wasn't present.  Otherwise, it must be a
79284021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // string followed by eod.
79384021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (StrTok.is(tok::eod))
794359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner    ; // ok
795359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  else if (StrTok.isNot(tok::string_literal)) {
796359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner    Diag(StrTok, diag::err_pp_line_invalid_filename);
797359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner    DiscardUntilEndOfDirective();
798359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner    return;
799359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  } else {
8005b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    // Parse and validate the string, converting it into a unique ID.
8015b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    StringLiteralParser Literal(&StrTok, 1, *this);
8025cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    assert(Literal.isAscii() && "Didn't allow wide strings in");
8035b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    if (Literal.hadError)
8045b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner      return DiscardUntilEndOfDirective();
8055b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    if (Literal.Pascal) {
8065b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner      Diag(StrTok, diag::err_pp_linemarker_invalid_filename);
8075b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner      return DiscardUntilEndOfDirective();
8085b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    }
80965aa6885818d4b4eea2e5a9d12085b2398148662Jay Foad    FilenameID = SourceMgr.getLineTableFilenameID(Literal.GetString());
8101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
81184021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    // Verify that there is nothing after the string, other than EOD.  Because
812ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner    // of C99 6.10.4p5, macros that expand to empty tokens are ok.
813ab82f41b217ce588a9456c0b4411f219d3ed0df8Chris Lattner    CheckEndOfDirective("line", true);
814359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  }
8151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8164c4ea17d7f991516c37a871dfa4bbe5723fa85f0Chris Lattner  SourceMgr.AddLineNote(DigitTok.getLocation(), LineNo, FilenameID);
8171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
81816629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner  if (Callbacks)
81986d0ef74303e9164a0860ce849af597d34f937abChris Lattner    Callbacks->FileChanged(CurPPLexer->getSourceLocation(),
82086d0ef74303e9164a0860ce849af597d34f937abChris Lattner                           PPCallbacks::RenameFile,
82116629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner                           SrcMgr::C_User);
822359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner}
823359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner
824478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner/// ReadLineMarkerFlags - Parse and validate any flags at the end of a GNU line
825478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner/// marker directive.
826478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattnerstatic bool ReadLineMarkerFlags(bool &IsFileEntry, bool &IsFileExit,
827478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner                                bool &IsSystemHeader, bool &IsExternCHeader,
828478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner                                Preprocessor &PP) {
829478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  unsigned FlagVal;
830478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  Token FlagTok;
831478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  PP.Lex(FlagTok);
83284021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (FlagTok.is(tok::eod)) return false;
833478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  if (GetLineValue(FlagTok, FlagVal, diag::err_pp_linemarker_invalid_flag, PP))
834478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return true;
835478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
836478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  if (FlagVal == 1) {
837478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    IsFileEntry = true;
8381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
839478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    PP.Lex(FlagTok);
84084021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    if (FlagTok.is(tok::eod)) return false;
841478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    if (GetLineValue(FlagTok, FlagVal, diag::err_pp_linemarker_invalid_flag,PP))
842478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner      return true;
843478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  } else if (FlagVal == 2) {
844478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    IsFileExit = true;
8451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
846137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    SourceManager &SM = PP.getSourceManager();
847137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    // If we are leaving the current presumed file, check to make sure the
848137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    // presumed include stack isn't empty!
849137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    FileID CurFileID =
850e7b2b6e87dbe5b1207f77b6ff9c210a02f95bb39Chandler Carruth      SM.getDecomposedExpansionLoc(FlagTok.getLocation()).first;
851137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    PresumedLoc PLoc = SM.getPresumedLoc(FlagTok.getLocation());
852cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor    if (PLoc.isInvalid())
853cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor      return true;
854cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor
855137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    // If there is no include loc (main file) or if the include loc is in a
856137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    // different physical file, then we aren't in a "1" line marker flag region.
857137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    SourceLocation IncLoc = PLoc.getIncludeLoc();
858137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    if (IncLoc.isInvalid() ||
859e7b2b6e87dbe5b1207f77b6ff9c210a02f95bb39Chandler Carruth        SM.getDecomposedExpansionLoc(IncLoc).first != CurFileID) {
860137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner      PP.Diag(FlagTok, diag::err_pp_linemarker_invalid_pop);
861137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner      PP.DiscardUntilEndOfDirective();
862137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner      return true;
863137b6a6149c53dbbcb8fba98e524d9ad0f3c8736Chris Lattner    }
8641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
865478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    PP.Lex(FlagTok);
86684021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    if (FlagTok.is(tok::eod)) return false;
867478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    if (GetLineValue(FlagTok, FlagVal, diag::err_pp_linemarker_invalid_flag,PP))
868478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner      return true;
869478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  }
870478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
871478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // We must have 3 if there are still flags.
872478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  if (FlagVal != 3) {
873478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    PP.Diag(FlagTok, diag::err_pp_linemarker_invalid_flag);
8745b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    PP.DiscardUntilEndOfDirective();
875478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return true;
876478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  }
8771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
878478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  IsSystemHeader = true;
8791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
880478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  PP.Lex(FlagTok);
88184021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (FlagTok.is(tok::eod)) return false;
8829d79ebac47ffde6a1cb312f4c09b66b1b9a397fbChris Lattner  if (GetLineValue(FlagTok, FlagVal, diag::err_pp_linemarker_invalid_flag, PP))
883478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return true;
884478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
885478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // We must have 4 if there is yet another flag.
886478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  if (FlagVal != 4) {
887478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    PP.Diag(FlagTok, diag::err_pp_linemarker_invalid_flag);
8885b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    PP.DiscardUntilEndOfDirective();
889478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return true;
890478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  }
8911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
892478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  IsExternCHeader = true;
8931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
894478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  PP.Lex(FlagTok);
89584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (FlagTok.is(tok::eod)) return false;
896478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
897478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // There are no more valid flags here.
898478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  PP.Diag(FlagTok, diag::err_pp_linemarker_invalid_flag);
8995b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner  PP.DiscardUntilEndOfDirective();
900478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  return true;
901478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner}
902478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
903478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner/// HandleDigitDirective - Handle a GNU line marker directive, whose syntax is
904478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner/// one of the following forms:
905478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner///
906478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner///     # 42
9071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///     # 42 "file" ('1' | '2')?
908478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner///     # 42 "file" ('1' | '2')? '3' '4'?
909478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner///
910478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattnervoid Preprocessor::HandleDigitDirective(Token &DigitTok) {
911478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // Validate the number and convert it to an unsigned.  GNU does not have a
912478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  // line # limit other than it fit in 32-bits.
913478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  unsigned LineNo;
914478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  if (GetLineValue(DigitTok, LineNo, diag::err_pp_linemarker_requires_integer,
915478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner                   *this))
916478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    return;
9171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
918478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  Token StrTok;
919478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  Lex(StrTok);
9201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
921478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  bool IsFileEntry = false, IsFileExit = false;
922478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  bool IsSystemHeader = false, IsExternCHeader = false;
9235b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner  int FilenameID = -1;
9245b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner
92584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // If the StrTok is "eod", then it wasn't present.  Otherwise, it must be a
92684021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // string followed by eod.
92784021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (StrTok.is(tok::eod))
928478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    ; // ok
929478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  else if (StrTok.isNot(tok::string_literal)) {
930478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    Diag(StrTok, diag::err_pp_linemarker_invalid_filename);
9315b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    return DiscardUntilEndOfDirective();
932478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  } else {
9335b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    // Parse and validate the string, converting it into a unique ID.
9345b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    StringLiteralParser Literal(&StrTok, 1, *this);
9355cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    assert(Literal.isAscii() && "Didn't allow wide strings in");
9365b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    if (Literal.hadError)
9375b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner      return DiscardUntilEndOfDirective();
9385b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    if (Literal.Pascal) {
9395b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner      Diag(StrTok, diag::err_pp_linemarker_invalid_filename);
9405b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner      return DiscardUntilEndOfDirective();
9415b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner    }
94265aa6885818d4b4eea2e5a9d12085b2398148662Jay Foad    FilenameID = SourceMgr.getLineTableFilenameID(Literal.GetString());
9431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
944478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner    // If a filename was present, read any flags that are present.
9451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    if (ReadLineMarkerFlags(IsFileEntry, IsFileExit,
9465b9a504720fb52594ca3686e10eb6c0cfa2e7d62Chris Lattner                            IsSystemHeader, IsExternCHeader, *this))
947478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner      return;
948478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner  }
9491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9509d79ebac47ffde6a1cb312f4c09b66b1b9a397fbChris Lattner  // Create a line note with this information.
9519d79ebac47ffde6a1cb312f4c09b66b1b9a397fbChris Lattner  SourceMgr.AddLineNote(DigitTok.getLocation(), LineNo, FilenameID,
9521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                        IsFileEntry, IsFileExit,
9539d79ebac47ffde6a1cb312f4c09b66b1b9a397fbChris Lattner                        IsSystemHeader, IsExternCHeader);
9541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
95516629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner  // If the preprocessor has callbacks installed, notify them of the #line
95616629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner  // change.  This is used so that the line marker comes out in -E mode for
95716629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner  // example.
95816629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner  if (Callbacks) {
95916629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner    PPCallbacks::FileChangeReason Reason = PPCallbacks::RenameFile;
96016629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner    if (IsFileEntry)
96116629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner      Reason = PPCallbacks::EnterFile;
96216629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner    else if (IsFileExit)
96316629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner      Reason = PPCallbacks::ExitFile;
96416629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner    SrcMgr::CharacteristicKind FileKind = SrcMgr::C_User;
96516629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner    if (IsExternCHeader)
96616629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner      FileKind = SrcMgr::C_ExternCSystem;
96716629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner    else if (IsSystemHeader)
96816629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner      FileKind = SrcMgr::C_System;
9691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
97086d0ef74303e9164a0860ce849af597d34f937abChris Lattner    Callbacks->FileChanged(CurPPLexer->getSourceLocation(), Reason, FileKind);
97116629386f1326608dc1ff9f4ff785fb3fae0bfb3Chris Lattner  }
972478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner}
973478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
974478a18ec47fdb7e0e580a2635648456e9db9ad4fChris Lattner
975099dd05b579ccefa97bd4f4a3088d4c262f080e4Chris Lattner/// HandleUserDiagnosticDirective - Handle a #warning or #error directive.
976099dd05b579ccefa97bd4f4a3088d4c262f080e4Chris Lattner///
9771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpvoid Preprocessor::HandleUserDiagnosticDirective(Token &Tok,
978141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                                 bool isWarning) {
979099dd05b579ccefa97bd4f4a3088d4c262f080e4Chris Lattner  // PTH doesn't emit #warning or #error directives.
980099dd05b579ccefa97bd4f4a3088d4c262f080e4Chris Lattner  if (CurPTHLexer)
981359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner    return CurPTHLexer->DiscardToEndOfLine();
982359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner
983141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Read the rest of the line raw.  We do this because we don't want macros
984141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // to be expanded and we don't require that the tokens be valid preprocessing
985141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // tokens.  For example, this is allowed: "#warning `   'foo".  GCC does
986141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // collapse multiple consequtive white space between tokens, but this isn't
987141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // specified by the standard.
988359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  std::string Message = CurLexer->ReadToEndOfLine();
989359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  if (isWarning)
990359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner    Diag(Tok, diag::pp_hash_warning) << Message;
991359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner  else
992359cc4475487ea0a660b3e8918334a2866e71c66Chris Lattner    Diag(Tok, diag::err_pp_hash_error) << Message;
993141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
994141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
995141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIdentSCCSDirective - Handle a #ident/#sccs directive.
996141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///
997141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleIdentSCCSDirective(Token &Tok) {
998141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Yes, this directive is an extension.
999141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Diag(Tok, diag::ext_pp_ident_directive);
10001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1001141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Read the string argument.
1002141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token StrTok;
1003141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Lex(StrTok);
10041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1005141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If the token kind isn't a string, it's a malformed directive.
1006141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (StrTok.isNot(tok::string_literal) &&
10073692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner      StrTok.isNot(tok::wide_string_literal)) {
10083692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    Diag(StrTok, diag::err_pp_malformed_ident);
100984021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    if (StrTok.isNot(tok::eod))
1010099dd05b579ccefa97bd4f4a3088d4c262f080e4Chris Lattner      DiscardUntilEndOfDirective();
10113692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    return;
10123692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner  }
10131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
101484021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // Verify that there is nothing after the string, other than EOD.
101535410d5cbbb18735d76e00496540d416dc49b577Chris Lattner  CheckEndOfDirective("ident");
1016141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1017453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor  if (Callbacks) {
1018453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor    bool Invalid = false;
1019453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor    std::string Str = getSpelling(StrTok, &Invalid);
1020453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor    if (!Invalid)
1021453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor      Callbacks->Ident(Tok.getLocation(), Str);
1022453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor  }
1023141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1024141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
10257143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor/// \brief Handle a #__export_macro__ directive.
10267143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregorvoid Preprocessor::HandleMacroExportDirective(Token &Tok) {
10277143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  Token MacroNameTok;
10287143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  ReadMacroName(MacroNameTok, 2);
10297143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor
10307143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  // Error reading macro name?  If so, diagnostic already issued.
10317143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  if (MacroNameTok.is(tok::eod))
10327143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor    return;
10337143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor
10347143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  // Check to see if this is the last token on the #__export_macro__ line.
10357143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  CheckEndOfDirective("__export_macro__");
10367143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor
10377143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  // Okay, we finally have a valid identifier to undef.
10387143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  MacroInfo *MI = getMacroInfo(MacroNameTok.getIdentifierInfo());
10397143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor
10407143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  // If the macro is not defined, this is an error.
10417143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  if (MI == 0) {
1042aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor    Diag(MacroNameTok, diag::err_pp_visibility_non_macro)
10437143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor      << MacroNameTok.getIdentifierInfo();
10447143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor    return;
10457143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  }
10467143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor
10477143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  // Note that this macro has now been exported.
1048aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  MI->setVisibility(/*IsPublic=*/true, MacroNameTok.getLocation());
1049aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor
1050aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  // If this macro definition came from a PCH file, mark it
1051aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  // as having changed since serialization.
1052aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  if (MI->isFromAST())
1053aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor    MI->setChangedAfterLoad();
1054aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor}
1055aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor
1056aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor/// \brief Handle a #__private_macro__ directive.
1057aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregorvoid Preprocessor::HandleMacroPrivateDirective(Token &Tok) {
1058aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  Token MacroNameTok;
1059aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  ReadMacroName(MacroNameTok, 2);
1060aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor
1061aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  // Error reading macro name?  If so, diagnostic already issued.
1062aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  if (MacroNameTok.is(tok::eod))
1063aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor    return;
1064aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor
1065aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  // Check to see if this is the last token on the #__private_macro__ line.
1066aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  CheckEndOfDirective("__private_macro__");
1067aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor
1068aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  // Okay, we finally have a valid identifier to undef.
1069aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  MacroInfo *MI = getMacroInfo(MacroNameTok.getIdentifierInfo());
1070aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor
1071aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  // If the macro is not defined, this is an error.
1072aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  if (MI == 0) {
1073aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor    Diag(MacroNameTok, diag::err_pp_visibility_non_macro)
1074aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor      << MacroNameTok.getIdentifierInfo();
1075aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor    return;
1076aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  }
1077aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor
1078aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  // Note that this macro has now been marked private.
1079aa93a875605536d72a10359a0098396192b7d4ecDouglas Gregor  MI->setVisibility(/*IsPublic=*/false, MacroNameTok.getLocation());
10807143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor
10817143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  // If this macro definition came from a PCH file, mark it
10827143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  // as having changed since serialization.
10837143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor  if (MI->isFromAST())
10847143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor    MI->setChangedAfterLoad();
10857143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor}
10867143aab97c6e849a5a5005b7853b8c7d5af008edDouglas Gregor
1087141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
1088141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Preprocessor Include Directive Handling.
1089141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
1090141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1091141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// GetIncludeFilenameSpelling - Turn the specified lexer token into a fully
1092141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// checked and spelled filename, e.g. as an operand of #include. This returns
1093141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// true if the input filename was in <>'s or false if it were in ""'s.  The
1094141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// caller is expected to provide a buffer that is large enough to hold the
1095141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// spelling of the filename, but is also expected to handle the case when
1096141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// this method decides to use a different buffer.
1097141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnerbool Preprocessor::GetIncludeFilenameSpelling(SourceLocation Loc,
10985f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner                                              StringRef &Buffer) {
1099141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Get the text form of the filename.
1100a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner  assert(!Buffer.empty() && "Can't have tokens with empty spellings!");
11011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1102141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Make sure the filename is <x> or "x".
1103141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  bool isAngled;
1104a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner  if (Buffer[0] == '<') {
1105a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner    if (Buffer.back() != '>') {
1106141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(Loc, diag::err_pp_expects_filename);
11075f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      Buffer = StringRef();
1108141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return true;
1109141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
1110141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    isAngled = true;
1111a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner  } else if (Buffer[0] == '"') {
1112a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner    if (Buffer.back() != '"') {
1113141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(Loc, diag::err_pp_expects_filename);
11145f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      Buffer = StringRef();
1115141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return true;
1116141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
1117141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    isAngled = false;
1118141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else {
1119141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Diag(Loc, diag::err_pp_expects_filename);
11205f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    Buffer = StringRef();
1121141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return true;
1122141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
11231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1124141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Diagnose #include "" as invalid.
1125a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner  if (Buffer.size() <= 2) {
1126141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Diag(Loc, diag::err_pp_empty_filename);
11275f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    Buffer = StringRef();
1128a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner    return true;
1129141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
11301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1131141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Skip the brackets.
1132a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner  Buffer = Buffer.substr(1, Buffer.size()-2);
1133141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  return isAngled;
1134141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1135141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1136141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// ConcatenateIncludeName - Handle cases where the #include name is expanded
1137141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// from a macro as multiple tokens, which need to be glued together.  This
1138141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// occurs for code like:
1139141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///    #define FOO <a/b.h>
1140141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///    #include FOO
1141141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// because in this case, "<a/b.h>" is returned as 7 tokens, not one.
1142141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///
1143141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// This code concatenates and consumes tokens up to the '>' token.  It returns
1144141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// false if the > was found, otherwise it returns true if it finds and consumes
114584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne/// the EOD marker.
1146a28cc090f265d5bdaef9ebb24fb9d579928d3a18John Thompsonbool Preprocessor::ConcatenateIncludeName(
1147ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor                                        llvm::SmallString<128> &FilenameBuffer,
1148ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor                                          SourceLocation &End) {
1149141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token CurTok;
11501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1151a28cc090f265d5bdaef9ebb24fb9d579928d3a18John Thompson  Lex(CurTok);
115284021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  while (CurTok.isNot(tok::eod)) {
1153ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    End = CurTok.getLocation();
1154ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
115525bb03bc0f2561e830d866d5905b20c5475d6762Douglas Gregor    // FIXME: Provide code completion for #includes.
115625bb03bc0f2561e830d866d5905b20c5475d6762Douglas Gregor    if (CurTok.is(tok::code_completion)) {
11577d100872341f233c81e1d7b72b40457e62c36862Argyrios Kyrtzidis      setCodeCompletionReached();
115825bb03bc0f2561e830d866d5905b20c5475d6762Douglas Gregor      Lex(CurTok);
115925bb03bc0f2561e830d866d5905b20c5475d6762Douglas Gregor      continue;
116025bb03bc0f2561e830d866d5905b20c5475d6762Douglas Gregor    }
116125bb03bc0f2561e830d866d5905b20c5475d6762Douglas Gregor
1162141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Append the spelling of this token to the buffer. If there was a space
1163141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // before it, add it now.
1164141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (CurTok.hasLeadingSpace())
1165141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      FilenameBuffer.push_back(' ');
11661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1167141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Get the spelling of the token, directly into FilenameBuffer if possible.
1168141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    unsigned PreAppendSize = FilenameBuffer.size();
1169141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    FilenameBuffer.resize(PreAppendSize+CurTok.getLength());
11701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1171141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    const char *BufPtr = &FilenameBuffer[PreAppendSize];
1172a28cc090f265d5bdaef9ebb24fb9d579928d3a18John Thompson    unsigned ActualLen = getSpelling(CurTok, BufPtr);
11731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1174141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // If the token was spelled somewhere else, copy it into FilenameBuffer.
1175141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (BufPtr != &FilenameBuffer[PreAppendSize])
1176141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      memcpy(&FilenameBuffer[PreAppendSize], BufPtr, ActualLen);
11771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1178141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Resize FilenameBuffer to the correct size.
1179141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (CurTok.getLength() != ActualLen)
1180141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      FilenameBuffer.resize(PreAppendSize+ActualLen);
11811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1182141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // If we found the '>' marker, return success.
1183141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (CurTok.is(tok::greater))
1184141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return false;
11851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1186a28cc090f265d5bdaef9ebb24fb9d579928d3a18John Thompson    Lex(CurTok);
1187141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
1188141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
118984021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // If we hit the eod marker, emit an error and return true so that the caller
119084021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // knows the EOD has been read.
1191a28cc090f265d5bdaef9ebb24fb9d579928d3a18John Thompson  Diag(CurTok.getLocation(), diag::err_pp_expects_filename);
1192141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  return true;
1193141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1194141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1195141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIncludeDirective - The "#include" tokens have just been read, read the
1196141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// file to be included from the lexer, then include it!  This is a common
1197141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// routine with functionality shared between #include, #include_next and
1198721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner/// #import.  LookupFrom is set when this is a #include_next directive, it
11991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// specifies the file to start searching from.
1200ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregorvoid Preprocessor::HandleIncludeDirective(SourceLocation HashLoc,
1201ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor                                          Token &IncludeTok,
1202141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                          const DirectoryLookup *LookupFrom,
1203141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                          bool isImport) {
1204141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1205141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token FilenameTok;
120660e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  CurPPLexer->LexIncludeFilename(FilenameTok);
12071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1208141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Reserve a buffer to get the spelling.
1209a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner  llvm::SmallString<128> FilenameBuffer;
12105f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Filename;
1211ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  SourceLocation End;
1212e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor  SourceLocation CharEnd; // the end of this directive, in characters
1213ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
1214141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  switch (FilenameTok.getKind()) {
121584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  case tok::eod:
121684021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    // If the token kind is EOD, the error has already been diagnosed.
1217141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return;
12181eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1219141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  case tok::angle_string_literal:
1220ddeea5644367c9c153c9fee9e51bdea85ce43cbdBenjamin Kramer  case tok::string_literal:
1221ddeea5644367c9c153c9fee9e51bdea85ce43cbdBenjamin Kramer    Filename = getSpelling(FilenameTok, FilenameBuffer);
1222ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    End = FilenameTok.getLocation();
1223e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    CharEnd = End.getLocWithOffset(Filename.size());
1224141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    break;
12251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1226141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  case tok::less:
1227141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // This could be a <foo/bar.h> file coming from a macro expansion.  In this
1228141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // case, glue the tokens together into FilenameBuffer and interpret those.
1229141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    FilenameBuffer.push_back('<');
1230ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    if (ConcatenateIncludeName(FilenameBuffer, End))
123184021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne      return;   // Found <eod> but no ">"?  Diagnostic already emitted.
1232a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner    Filename = FilenameBuffer.str();
1233e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    CharEnd = getLocForEndOfToken(End);
1234141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    break;
1235141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  default:
1236141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Diag(FilenameTok.getLocation(), diag::err_pp_expects_filename);
1237141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    DiscardUntilEndOfDirective();
1238141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return;
1239141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
12401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1241193575455e00eca03fd7177f60e3f2e6263cb661Kovarththanan Rajaratnam  bool isAngled =
1242a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner    GetIncludeFilenameSpelling(FilenameTok.getLocation(), Filename);
1243141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If GetIncludeFilenameSpelling set the start ptr to null, there was an
1244141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // error.
1245a139481e62fdb209d9d87a54a5733f989d2e8d51Chris Lattner  if (Filename.empty()) {
1246141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    DiscardUntilEndOfDirective();
1247141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return;
1248141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
12491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
125084021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  // Verify that there is nothing after the filename, other than EOD.  Note that
12519cb51ce03caf0755759f7e67aea7d4be684ff738Chris Lattner  // we allow macros that expand to nothing after the filename, because this
12529cb51ce03caf0755759f7e67aea7d4be684ff738Chris Lattner  // falls into the category of "#include pp-tokens new-line" specified in
12539cb51ce03caf0755759f7e67aea7d4be684ff738Chris Lattner  // C99 6.10.2p4.
1254e013d685c6689ac7ae103ee88acf573422d1ed6aDaniel Dunbar  CheckEndOfDirective(IncludeTok.getIdentifierInfo()->getNameStart(), true);
1255141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1256141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Check that we don't have infinite #include recursion.
12573692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner  if (IncludeMacroStack.size() == MaxAllowedIncludeStackDepth-1) {
12583692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    Diag(FilenameTok, diag::err_pp_include_too_deep);
12593692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    return;
12603692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner  }
12611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12628dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall  // Complain about attempts to #include files in an audit pragma.
12638dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall  if (PragmaARCCFCodeAuditedLoc.isValid()) {
12648dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall    Diag(HashLoc, diag::err_pp_include_in_arc_cf_code_audited);
12658dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall    Diag(PragmaARCCFCodeAuditedLoc, diag::note_pragma_entered_here);
12668dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall
12678dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall    // Immediately leave the pragma.
12688dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall    PragmaARCCFCodeAuditedLoc = SourceLocation();
12698dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall  }
12708dfac0baaf0f81d3945bcb306480e358ba8d1f08John McCall
1271141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Search include directories.
1272141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  const DirectoryLookup *CurDir;
12737412494982c8b50c90961302c3a718633b2c3ab7Manuel Klimek  llvm::SmallString<1024> SearchPath;
12747412494982c8b50c90961302c3a718633b2c3ab7Manuel Klimek  llvm::SmallString<1024> RelativePath;
1275b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruth  // We get the raw path only if we have 'Callbacks' to which we later pass
1276b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruth  // the path.
12771a4761edca58c6b559de825b9abfb66f7f1ba94aDouglas Gregor  Module *SuggestedModule = 0;
1278b5142bb7af5c70fffd09f05172a1379a35a9c29aChandler Carruth  const FileEntry *File = LookupFile(
12797412494982c8b50c90961302c3a718633b2c3ab7Manuel Klimek      Filename, isAngled, LookupFrom, CurDir,
1280fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor      Callbacks ? &SearchPath : NULL, Callbacks ? &RelativePath : NULL,
1281fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor      AutoModuleImport? &SuggestedModule : 0);
1282fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor
12838cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor  if (Callbacks) {
12848cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor    if (!File) {
12858cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor      // Give the clients a chance to recover.
12868cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor      llvm::SmallString<128> RecoveryPath;
12878cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor      if (Callbacks->FileNotFound(Filename, RecoveryPath)) {
12888cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor        if (const DirectoryEntry *DE = FileMgr.getDirectory(RecoveryPath)) {
12898cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor          // Add the recovery path to the list of search paths.
12908cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor          DirectoryLookup DL(DE, SrcMgr::C_User, true, false);
12918cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor          HeaderInfo.AddSearchPath(DL, isAngled);
12928cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor
12938cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor          // Try the lookup again, skipping the cache.
12948cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor          File = LookupFile(Filename, isAngled, LookupFrom, CurDir, 0, 0,
12958cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor                            AutoModuleImport ? &SuggestedModule : 0,
12968cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor                            /*SkipCache*/true);
12978cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor        }
12988cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor      }
12998cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor    }
13008cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor
13018cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor    // Notify the callback object that we've seen an inclusion directive.
13028cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor    Callbacks->InclusionDirective(HashLoc, IncludeTok, Filename, isAngled, File,
13038cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor                                  End, SearchPath, RelativePath);
13048cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor  }
13058cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor
13068cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor  if (File == 0) {
13078cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor    if (!SuppressIncludeNotFoundError)
13088cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor      Diag(FilenameTok, diag::err_pp_file_not_found) << Filename;
13098cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor    return;
13108cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor  }
13118cfbe6a31ad51cb19046607ad6b792638e69b88eDouglas Gregor
1312fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor  // If we are supposed to import a module rather than including the header,
1313fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor  // do so now.
1314c69c42e939e6bdaa56d162cc36da4f6b6c53e8dbDouglas Gregor  if (SuggestedModule) {
13153d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor    // Compute the module access path corresponding to this module.
13163d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor    // FIXME: Should we have a second loadModule() overload to avoid this
13173d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor    // extra lookup step?
13183d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor    llvm::SmallVector<std::pair<IdentifierInfo *, SourceLocation>, 2> Path;
13191a4761edca58c6b559de825b9abfb66f7f1ba94aDouglas Gregor    for (Module *Mod = SuggestedModule; Mod; Mod = Mod->Parent)
13203d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor      Path.push_back(std::make_pair(getIdentifierInfo(Mod->Name),
13213d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor                                    FilenameTok.getLocation()));
13223d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor    std::reverse(Path.begin(), Path.end());
13233d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor
1324e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    // Warn that we're replacing the include/import with a module import.
1325e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    llvm::SmallString<128> PathString;
1326e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    for (unsigned I = 0, N = Path.size(); I != N; ++I) {
1327e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      if (I)
1328e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor        PathString += '.';
1329e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      PathString += Path[I].first->getName();
1330e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    }
1331e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    int IncludeKind = 0;
1332e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor
1333e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    switch (IncludeTok.getIdentifierInfo()->getPPKeywordID()) {
1334e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    case tok::pp_include:
1335e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      IncludeKind = 0;
1336e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      break;
1337e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor
1338e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    case tok::pp_import:
1339e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      IncludeKind = 1;
1340e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      break;
1341e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor
1342edee9699964a1f38892c74acc555883bf26b51a3Douglas Gregor    case tok::pp_include_next:
1343edee9699964a1f38892c74acc555883bf26b51a3Douglas Gregor      IncludeKind = 2;
1344edee9699964a1f38892c74acc555883bf26b51a3Douglas Gregor      break;
1345e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor
1346e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    case tok::pp___include_macros:
1347e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      IncludeKind = 3;
1348e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      break;
1349e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor
1350e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    default:
1351e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      llvm_unreachable("unknown include directive kind");
1352e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      break;
1353e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    }
1354e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor
1355e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    CharSourceRange ReplaceRange(SourceRange(HashLoc, CharEnd),
1356e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor                                 /*IsTokenRange=*/false);
1357e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor    Diag(HashLoc, diag::warn_auto_module_import)
1358e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      << IncludeKind << PathString
1359e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor      << FixItHint::CreateReplacement(ReplaceRange,
1360e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor           "__import_module__ " + PathString.str().str() + ";");
1361e3a8256a9c1f2f1037fcb844ed7666759124a852Douglas Gregor
13623d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor    // Load the module.
13634b7cac5d04dba8a6cda46ba05d6a79218557aa94Douglas Gregor    // FIXME: Deal with __include_macros here.
13643d3589db579f7695667b913c5043dd264ebe546fDouglas Gregor    TheModuleLoader.loadModule(IncludeTok.getLocation(), Path);
1365fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor    return;
1366fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor  }
1367fba18aa8f2cd1994dc65e8cb9f4be201c560dc0bDouglas Gregor
1368721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner  // The #included file will be considered to be a system header if either it is
1369721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner  // in a system include directory, or if the #includer is a system include
1370721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner  // header.
13711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  SrcMgr::CharacteristicKind FileCharacter =
13720b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner    std::max(HeaderInfo.getFileDirFlavor(File),
1373693faa6e6918994bc22aacf078a889bb93e0fe25Chris Lattner             SourceMgr.getFileCharacteristic(FilenameTok.getLocation()));
13741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
13756fbe3ebeaef08665a37423f8425314c90b8b5bcfChris Lattner  // Ask HeaderInfo if we should enter this #include file.  If not, #including
13766fbe3ebeaef08665a37423f8425314c90b8b5bcfChris Lattner  // this file will have no effect.
13776fbe3ebeaef08665a37423f8425314c90b8b5bcfChris Lattner  if (!HeaderInfo.ShouldEnterIncludeFile(File, isImport)) {
1378e127a0d80155b45dafe77f2b4380e5fa111a3345Chris Lattner    if (Callbacks)
13796fbe3ebeaef08665a37423f8425314c90b8b5bcfChris Lattner      Callbacks->FileSkipped(*File, FilenameTok, FileCharacter);
13806fbe3ebeaef08665a37423f8425314c90b8b5bcfChris Lattner    return;
13816fbe3ebeaef08665a37423f8425314c90b8b5bcfChris Lattner  }
13826fbe3ebeaef08665a37423f8425314c90b8b5bcfChris Lattner
1383141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Look up the file, create a File ID for it.
13842b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner  FileID FID = SourceMgr.createFileID(File, FilenameTok.getLocation(),
13852b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner                                      FileCharacter);
1386d57b7ff9bebc4c45f325fc1be6f238cfcd4c3732Peter Collingbourne  assert(!FID.isInvalid() && "Expected valid file ID");
1387141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1388141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Finally, if all is good, enter the new file!
1389e127a0d80155b45dafe77f2b4380e5fa111a3345Chris Lattner  EnterSourceFile(FID, CurDir, FilenameTok.getLocation());
1390141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1391141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1392141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIncludeNextDirective - Implements #include_next.
1393141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///
1394ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregorvoid Preprocessor::HandleIncludeNextDirective(SourceLocation HashLoc,
1395ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor                                              Token &IncludeNextTok) {
1396141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Diag(IncludeNextTok, diag::ext_pp_include_next_directive);
13971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1398141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // #include_next is like #include, except that we start searching after
1399141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // the current found directory.  If we can't do this, issue a
1400141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // diagnostic.
1401141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  const DirectoryLookup *Lookup = CurDirLookup;
1402141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (isInPrimaryFile()) {
1403141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Lookup = 0;
1404141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Diag(IncludeNextTok, diag::pp_include_next_in_primary);
1405141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else if (Lookup == 0) {
1406141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Diag(IncludeNextTok, diag::pp_include_next_absolute_path);
1407141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else {
1408141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Start looking up in the next directory.
1409141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    ++Lookup;
1410141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
14111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1412ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return HandleIncludeDirective(HashLoc, IncludeNextTok, Lookup);
1413141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1414141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1415141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleImportDirective - Implements #import.
1416141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///
1417ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregorvoid Preprocessor::HandleImportDirective(SourceLocation HashLoc,
1418ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor                                         Token &ImportTok) {
1419b627c8d683c4d3facd774c432b1fdc2d35819369Chris Lattner  if (!Features.ObjC1)  // #import is standard for ObjC.
1420b627c8d683c4d3facd774c432b1fdc2d35819369Chris Lattner    Diag(ImportTok, diag::ext_pp_import_directive);
14211eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1422ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return HandleIncludeDirective(HashLoc, ImportTok, 0, true);
1423141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1424141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1425de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner/// HandleIncludeMacrosDirective - The -imacros command line option turns into a
1426de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner/// pseudo directive in the predefines buffer.  This handles it by sucking all
1427de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner/// tokens through the preprocessor and discarding them (only keeping the side
1428de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner/// effects on the preprocessor).
1429ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregorvoid Preprocessor::HandleIncludeMacrosDirective(SourceLocation HashLoc,
1430ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor                                                Token &IncludeMacrosTok) {
1431de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner  // This directive should only occur in the predefines buffer.  If not, emit an
1432de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner  // error and reject it.
1433de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner  SourceLocation Loc = IncludeMacrosTok.getLocation();
1434de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner  if (strcmp(SourceMgr.getBufferName(Loc), "<built-in>") != 0) {
1435de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner    Diag(IncludeMacrosTok.getLocation(),
1436de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner         diag::pp_include_macros_out_of_predefines);
1437de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner    DiscardUntilEndOfDirective();
1438de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner    return;
1439de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner  }
14401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1441fd10511b3e686c5b045f8324b0632f3624d9ef05Chris Lattner  // Treat this as a normal #include for checking purposes.  If this is
1442fd10511b3e686c5b045f8324b0632f3624d9ef05Chris Lattner  // successful, it will push a new lexer onto the include stack.
1443ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  HandleIncludeDirective(HashLoc, IncludeMacrosTok, 0, false);
14441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1445fd10511b3e686c5b045f8324b0632f3624d9ef05Chris Lattner  Token TmpTok;
1446fd10511b3e686c5b045f8324b0632f3624d9ef05Chris Lattner  do {
1447fd10511b3e686c5b045f8324b0632f3624d9ef05Chris Lattner    Lex(TmpTok);
1448fd10511b3e686c5b045f8324b0632f3624d9ef05Chris Lattner    assert(TmpTok.isNot(tok::eof) && "Didn't find end of -imacros!");
1449fd10511b3e686c5b045f8324b0632f3624d9ef05Chris Lattner  } while (TmpTok.isNot(tok::hashhash));
1450de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner}
1451de076650fa79a02743b5fe57a72f991ba24779f9Chris Lattner
1452141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
1453141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Preprocessor Macro Directive Handling.
1454141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
1455141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1456141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// ReadMacroDefinitionArgList - The ( starting an argument list of a macro
1457141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// definition has just been read.  Lex the rest of the arguments and the
1458141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// closing ), updating MI with what we learn.  Return true if an error occurs
1459141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// parsing the arg list.
1460141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnerbool Preprocessor::ReadMacroDefinitionArgList(MacroInfo *MI) {
14615f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<IdentifierInfo*, 32> Arguments;
14621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1463141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token Tok;
1464141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  while (1) {
1465141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    LexUnexpandedToken(Tok);
1466141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    switch (Tok.getKind()) {
1467141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::r_paren:
1468141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Found the end of the argument list.
1469cf29e0716bb3ecbbc15b74cd648367d6b075fdf0Chris Lattner      if (Arguments.empty())  // #define FOO()
1470141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        return false;
1471141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Otherwise we have #define FOO(A,)
1472141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(Tok, diag::err_pp_expected_ident_in_arg_list);
1473141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return true;
1474141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    case tok::ellipsis:  // #define X(... -> C99 varargs
1475661a99690bc133bbaa029da925481d4a860dec90Richard Smith      if (!Features.C99)
1476661a99690bc133bbaa029da925481d4a860dec90Richard Smith        Diag(Tok, Features.CPlusPlus0x ?
1477661a99690bc133bbaa029da925481d4a860dec90Richard Smith             diag::warn_cxx98_compat_variadic_macro :
1478661a99690bc133bbaa029da925481d4a860dec90Richard Smith             diag::ext_variadic_macro);
1479141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1480141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Lex the token after the identifier.
1481141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      LexUnexpandedToken(Tok);
1482141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      if (Tok.isNot(tok::r_paren)) {
1483141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
1484141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        return true;
1485141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
1486141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Add the __VA_ARGS__ identifier as an argument.
1487141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Arguments.push_back(Ident__VA_ARGS__);
1488141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      MI->setIsC99Varargs();
1489685befeb5f6472585bae473a6389e47cab9eac67Chris Lattner      MI->setArgumentList(&Arguments[0], Arguments.size(), BP);
1490141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return false;
149184021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    case tok::eod:  // #define X(
1492141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
1493141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return true;
1494141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    default:
1495141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Handle keywords and identifiers here to accept things like
1496141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // #define Foo(for) for.
1497141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      IdentifierInfo *II = Tok.getIdentifierInfo();
1498141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      if (II == 0) {
1499141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // #define X(1
1500141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        Diag(Tok, diag::err_pp_invalid_tok_in_arg_list);
1501141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        return true;
1502141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
1503141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1504141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // If this is already used as an argument, it is used multiple times (e.g.
1505141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // #define X(A,A.
15061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      if (std::find(Arguments.begin(), Arguments.end(), II) !=
1507141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          Arguments.end()) {  // C99 6.10.3p6
15086cf3ed7be70a7a9c269a65e3316884f41c7373d4Chris Lattner        Diag(Tok, diag::err_pp_duplicate_name_in_arg_list) << II;
1509141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        return true;
1510141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
15111eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1512141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Add the argument to the macro info.
1513141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Arguments.push_back(II);
15141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1515141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Lex the token after the identifier.
1516141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      LexUnexpandedToken(Tok);
15171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1518141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      switch (Tok.getKind()) {
1519141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      default:          // #define X(A B
1520141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        Diag(Tok, diag::err_pp_expected_comma_in_arg_list);
1521141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        return true;
1522141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      case tok::r_paren: // #define X(A)
1523685befeb5f6472585bae473a6389e47cab9eac67Chris Lattner        MI->setArgumentList(&Arguments[0], Arguments.size(), BP);
1524141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        return false;
1525141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      case tok::comma:  // #define X(A,
1526141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        break;
1527141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      case tok::ellipsis:  // #define X(A... -> GCC extension
1528141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // Diagnose extension.
1529141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        Diag(Tok, diag::ext_named_variadic_macro);
15301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1531141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // Lex the token after the identifier.
1532141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        LexUnexpandedToken(Tok);
1533141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        if (Tok.isNot(tok::r_paren)) {
1534141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
1535141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner          return true;
1536141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        }
15371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1538141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        MI->setIsGNUVarargs();
1539685befeb5f6472585bae473a6389e47cab9eac67Chris Lattner        MI->setArgumentList(&Arguments[0], Arguments.size(), BP);
1540141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        return false;
1541141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
1542141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
1543141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
1544141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1545141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1546141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleDefineDirective - Implements #define.  This consumes the entire macro
1547141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// line then lets the caller lex the next real token.
1548141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleDefineDirective(Token &DefineTok) {
1549141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumDefined;
1550141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1551141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token MacroNameTok;
1552141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ReadMacroName(MacroNameTok, 1);
15531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1554141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Error reading macro name?  If so, diagnostic already issued.
155584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (MacroNameTok.is(tok::eod))
1556141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return;
1557141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
15582451b528fe114595d0f10ef2c05047928558ab0fChris Lattner  Token LastTok = MacroNameTok;
15592451b528fe114595d0f10ef2c05047928558ab0fChris Lattner
1560141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If we are supposed to keep comments in #defines, reenable comment saving
1561141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // mode.
1562ac6b06df32e907cac388a92310650b5895c11241Ted Kremenek  if (CurLexer) CurLexer->SetCommentRetentionState(KeepMacroComments);
15631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1564141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Create the new macro.
15650ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek  MacroInfo *MI = AllocateMacroInfo(MacroNameTok.getLocation());
15661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1567141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token Tok;
1568141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  LexUnexpandedToken(Tok);
15691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1570141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If this is a function-like macro definition, parse the argument list,
1571141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // marking each of the identifiers as being used as macro arguments.  Also,
1572141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // check other constraints on the first token of the macro body.
157384021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (Tok.is(tok::eod)) {
1574141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // If there is no body to this macro, we have no special handling here.
15756272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner  } else if (Tok.hasLeadingSpace()) {
15766272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    // This is a normal token with leading space.  Clear the leading space
15776272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    // marker on the first token to get proper expansion.
15786272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    Tok.clearFlag(Token::LeadingSpace);
15796272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner  } else if (Tok.is(tok::l_paren)) {
1580141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // This is a function-like macro definition.  Read the argument list.
1581141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    MI->setIsFunctionLike();
1582141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (ReadMacroDefinitionArgList(MI)) {
1583141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Forget about MI.
15840ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek      ReleaseMacroInfo(MI);
1585141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Throw away the rest of the line.
158660e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek      if (CurPPLexer->ParsingPreprocessorDirective)
1587141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        DiscardUntilEndOfDirective();
1588141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return;
1589141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
1590141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
15918fde5978a63bcce5c294a93b2a5dd424d7628971Chris Lattner    // If this is a definition of a variadic C99 function-like macro, not using
15928fde5978a63bcce5c294a93b2a5dd424d7628971Chris Lattner    // the GNU named varargs extension, enabled __VA_ARGS__.
15931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
15948fde5978a63bcce5c294a93b2a5dd424d7628971Chris Lattner    // "Poison" __VA_ARGS__, which can only appear in the expansion of a macro.
15958fde5978a63bcce5c294a93b2a5dd424d7628971Chris Lattner    // This gets unpoisoned where it is allowed.
15968fde5978a63bcce5c294a93b2a5dd424d7628971Chris Lattner    assert(Ident__VA_ARGS__->isPoisoned() && "__VA_ARGS__ should be poisoned!");
15978fde5978a63bcce5c294a93b2a5dd424d7628971Chris Lattner    if (MI->isC99Varargs())
15988fde5978a63bcce5c294a93b2a5dd424d7628971Chris Lattner      Ident__VA_ARGS__->setIsPoisoned(false);
15991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1600141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Read the first token after the arg list for down below.
1601141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    LexUnexpandedToken(Tok);
1602158ebfba7545b3a9947f87bed83ea710df95ac3dEli Friedman  } else if (Features.C99 || Features.CPlusPlus0x) {
1603141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // C99 requires whitespace between the macro definition and the body.  Emit
1604141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // a diagnostic for something like "#define X+".
16056272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    Diag(Tok, diag::ext_c99_whitespace_required_after_macro_name);
1606141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else {
16076272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    // C90 6.8 TC1 says: "In the definition of an object-like macro, if the
16086272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    // first character of a replacement list is not a character required by
16096272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    // subclause 5.2.1, then there shall be white-space separation between the
16106272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    // identifier and the replacement list.".  5.2.1 lists this set:
16116272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    //   "A-Za-z0-9!"#%&'()*+,_./:;<=>?[\]^_{|}~" as well as whitespace, which
16126272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    // is irrelevant here.
16136272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    bool isInvalid = false;
16146272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    if (Tok.is(tok::at)) // @ is not in the list above.
16156272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner      isInvalid = true;
16166272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    else if (Tok.is(tok::unknown)) {
16176272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner      // If we have an unknown token, it is something strange like "`".  Since
16186272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner      // all of valid characters would have lexed into a single character
16196272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner      // token of some sort, we know this is not a valid case.
16206272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner      isInvalid = true;
16216272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    }
16226272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    if (isInvalid)
16236272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner      Diag(Tok, diag::ext_missing_whitespace_after_macro_name);
16246272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner    else
16256272bcfe5dc148d293ccce9db19df9e71c991cdeChris Lattner      Diag(Tok, diag::warn_missing_whitespace_after_macro_name);
1626141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
16272451b528fe114595d0f10ef2c05047928558ab0fChris Lattner
162884021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (!Tok.is(tok::eod))
16292451b528fe114595d0f10ef2c05047928558ab0fChris Lattner    LastTok = Tok;
16302451b528fe114595d0f10ef2c05047928558ab0fChris Lattner
1631141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Read the rest of the macro body.
1632141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (MI->isObjectLike()) {
1633141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Object-like macros are very simple, just read their body.
163484021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    while (Tok.isNot(tok::eod)) {
16352451b528fe114595d0f10ef2c05047928558ab0fChris Lattner      LastTok = Tok;
1636141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      MI->AddTokenToBody(Tok);
1637141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Get the next token of the macro.
1638141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      LexUnexpandedToken(Tok);
1639141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
16401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1641141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else {
16423240469102eca90fa5123ec9636fa0364df3034cChris Lattner    // Otherwise, read the body of a function-like macro.  While we are at it,
16433240469102eca90fa5123ec9636fa0364df3034cChris Lattner    // check C99 6.10.3.2p1: ensure that # operators are followed by macro
16443240469102eca90fa5123ec9636fa0364df3034cChris Lattner    // parameters in function-like macro expansions.
164584021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne    while (Tok.isNot(tok::eod)) {
16462451b528fe114595d0f10ef2c05047928558ab0fChris Lattner      LastTok = Tok;
1647141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1648141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      if (Tok.isNot(tok::hash)) {
16493240469102eca90fa5123ec9636fa0364df3034cChris Lattner        MI->AddTokenToBody(Tok);
16501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1651141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        // Get the next token of the macro.
1652141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        LexUnexpandedToken(Tok);
1653141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner        continue;
1654141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
16551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1656141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Get the next token of the macro.
1657141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      LexUnexpandedToken(Tok);
16581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
16593240469102eca90fa5123ec9636fa0364df3034cChris Lattner      // Check for a valid macro arg identifier.
16603240469102eca90fa5123ec9636fa0364df3034cChris Lattner      if (Tok.getIdentifierInfo() == 0 ||
16613240469102eca90fa5123ec9636fa0364df3034cChris Lattner          MI->getArgumentNum(Tok.getIdentifierInfo()) == -1) {
16623240469102eca90fa5123ec9636fa0364df3034cChris Lattner
16633240469102eca90fa5123ec9636fa0364df3034cChris Lattner        // If this is assembler-with-cpp mode, we accept random gibberish after
16643240469102eca90fa5123ec9636fa0364df3034cChris Lattner        // the '#' because '#' is often a comment character.  However, change
16653240469102eca90fa5123ec9636fa0364df3034cChris Lattner        // the kind of the token to tok::unknown so that the preprocessor isn't
16663240469102eca90fa5123ec9636fa0364df3034cChris Lattner        // confused.
166784021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne        if (getLangOptions().AsmPreprocessor && Tok.isNot(tok::eod)) {
16683240469102eca90fa5123ec9636fa0364df3034cChris Lattner          LastTok.setKind(tok::unknown);
16693240469102eca90fa5123ec9636fa0364df3034cChris Lattner        } else {
16703240469102eca90fa5123ec9636fa0364df3034cChris Lattner          Diag(Tok, diag::err_pp_stringize_not_parameter);
16713240469102eca90fa5123ec9636fa0364df3034cChris Lattner          ReleaseMacroInfo(MI);
16721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
16733240469102eca90fa5123ec9636fa0364df3034cChris Lattner          // Disable __VA_ARGS__ again.
16743240469102eca90fa5123ec9636fa0364df3034cChris Lattner          Ident__VA_ARGS__->setIsPoisoned(true);
16753240469102eca90fa5123ec9636fa0364df3034cChris Lattner          return;
16763240469102eca90fa5123ec9636fa0364df3034cChris Lattner        }
1677141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      }
16781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
16793240469102eca90fa5123ec9636fa0364df3034cChris Lattner      // Things look ok, add the '#' and param name tokens to the macro.
16803240469102eca90fa5123ec9636fa0364df3034cChris Lattner      MI->AddTokenToBody(LastTok);
1681141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      MI->AddTokenToBody(Tok);
16823240469102eca90fa5123ec9636fa0364df3034cChris Lattner      LastTok = Tok;
16831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1684141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      // Get the next token of the macro.
1685141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      LexUnexpandedToken(Tok);
1686141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
1687141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
16881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
16891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1690141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Disable __VA_ARGS__ again.
1691141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Ident__VA_ARGS__->setIsPoisoned(true);
1692141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1693fc8f0e14ad142ed811e90fbd9a30e419e301c717Chris Lattner  // Check that there is no paste (##) operator at the beginning or end of the
1694141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // replacement list.
1695141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  unsigned NumTokens = MI->getNumTokens();
1696141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (NumTokens != 0) {
1697141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (MI->getReplacementToken(0).is(tok::hashhash)) {
1698141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(MI->getReplacementToken(0), diag::err_paste_at_start);
16990ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek      ReleaseMacroInfo(MI);
1700141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return;
1701141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
1702141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    if (MI->getReplacementToken(NumTokens-1).is(tok::hashhash)) {
1703141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      Diag(MI->getReplacementToken(NumTokens-1), diag::err_paste_at_end);
17040ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek      ReleaseMacroInfo(MI);
1705141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      return;
1706141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
1707141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
17081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
17092451b528fe114595d0f10ef2c05047928558ab0fChris Lattner  MI->setDefinitionEndLoc(LastTok.getLocation());
17101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1711141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Finally, if this identifier already had a macro defined for it, verify that
1712141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // the macro bodies are identical and free the old definition.
1713141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (MacroInfo *OtherMI = getMacroInfo(MacroNameTok.getIdentifierInfo())) {
171441c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner    // It is very common for system headers to have tons of macro redefinitions
171541c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner    // and for warnings to be disabled in system headers.  If this is the case,
171641c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner    // then don't bother calling MacroInfo::isIdenticalTo.
17177f549dfd7b1d3b2b09e5878eeb754efae253d557Chris Lattner    if (!getDiagnostics().getSuppressSystemWarnings() ||
171841c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner        !SourceMgr.isInSystemHeader(DefineTok.getLocation())) {
1719a33e0504f61132ac43f4f89af43b12c87e987c20Argyrios Kyrtzidis      if (!OtherMI->isUsed() && OtherMI->isWarnIfUnused())
172041c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner        Diag(OtherMI->getDefinitionLoc(), diag::pp_macro_not_used);
1721141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1722f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner      // Macros must be identical.  This means all tokens and whitespace
172341c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner      // separation must be the same.  C99 6.10.3.2.
1724f47724bf78299c7a50f008e0443c5f9f9f279ddcChris Lattner      if (!OtherMI->isAllowRedefinitionsWithoutWarning() &&
1725a7e6845660f91ec611427e1db842780e1ec12bdbEli Friedman          !MI->isIdenticalTo(*OtherMI, *this)) {
172641c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner        Diag(MI->getDefinitionLoc(), diag::ext_pp_macro_redef)
172741c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner          << MacroNameTok.getIdentifierInfo();
172841c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner        Diag(OtherMI->getDefinitionLoc(), diag::note_previous_definition);
172941c3ae108a54437f27fa99ddcb9bafcf113bde80Chris Lattner      }
1730141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    }
1731a33e0504f61132ac43f4f89af43b12c87e987c20Argyrios Kyrtzidis    if (OtherMI->isWarnIfUnused())
1732a33e0504f61132ac43f4f89af43b12c87e987c20Argyrios Kyrtzidis      WarnUnusedMacroLocs.erase(OtherMI->getDefinitionLoc());
17330ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek    ReleaseMacroInfo(OtherMI);
1734141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
17351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1736141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  setMacroInfo(MacroNameTok.getIdentifierInfo(), MI);
17371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
17380827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis  assert(!MI->isUsed());
17390827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis  // If we need warning for not using the macro, add its location in the
17400827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis  // warn-because-unused-macro set. If it gets used it will be removed from set.
17410827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis  if (isInPrimaryFile() && // don't warn for include'd macros.
17420827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis      Diags->getDiagnosticLevel(diag::pp_macro_not_used,
1743d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie          MI->getDefinitionLoc()) != DiagnosticsEngine::Ignored) {
17440827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis    MI->setIsWarnIfUnused(true);
17450827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis    WarnUnusedMacroLocs.insert(MI->getDefinitionLoc());
17460827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis  }
17470827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis
1748f4a72b07747cd6ae1a9f6974e148ae504c879349Chris Lattner  // If the callbacks want to know, tell them about the macro definition.
1749f4a72b07747cd6ae1a9f6974e148ae504c879349Chris Lattner  if (Callbacks)
17502aa9267cf193e5eceb9fd24a51b51c031b606fe9Craig Silverstein    Callbacks->MacroDefined(MacroNameTok, MI);
1751141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1752141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1753141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleUndefDirective - Implements #undef.
1754141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///
1755141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleUndefDirective(Token &UndefTok) {
1756141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumUndefined;
1757141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1758141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token MacroNameTok;
1759141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ReadMacroName(MacroNameTok, 2);
17601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1761141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Error reading macro name?  If so, diagnostic already issued.
176284021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (MacroNameTok.is(tok::eod))
1763141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return;
17641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1765141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Check to see if this is the last token on the #undef line.
176635410d5cbbb18735d76e00496540d416dc49b577Chris Lattner  CheckEndOfDirective("undef");
17671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1768141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Okay, we finally have a valid identifier to undef.
1769141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  MacroInfo *MI = getMacroInfo(MacroNameTok.getIdentifierInfo());
17701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1771141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If the macro is not defined, this is a noop undef, just return.
1772141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (MI == 0) return;
1773141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
17741f8dcfca6c57606b9f4f96404642c02562289b83Argyrios Kyrtzidis  if (!MI->isUsed() && MI->isWarnIfUnused())
1775141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    Diag(MI->getDefinitionLoc(), diag::pp_macro_not_used);
177641c17473e3ece9e60d97c5d9397866b7730cf7eeChris Lattner
177741c17473e3ece9e60d97c5d9397866b7730cf7eeChris Lattner  // If the callbacks want to know, tell them about the macro #undef.
177841c17473e3ece9e60d97c5d9397866b7730cf7eeChris Lattner  if (Callbacks)
17792aa9267cf193e5eceb9fd24a51b51c031b606fe9Craig Silverstein    Callbacks->MacroUndefined(MacroNameTok, MI);
178041c17473e3ece9e60d97c5d9397866b7730cf7eeChris Lattner
17810827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis  if (MI->isWarnIfUnused())
17820827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis    WarnUnusedMacroLocs.erase(MI->getDefinitionLoc());
17830827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis
1784141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Free macro definition.
17850ea76727ae91bca918a8414ed85b530eddcfedebTed Kremenek  ReleaseMacroInfo(MI);
1786141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  setMacroInfo(MacroNameTok.getIdentifierInfo(), 0);
1787141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1788141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1789141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1790141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
1791141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Preprocessor Conditional Directive Handling.
1792141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===//
1793141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1794141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIfdefDirective - Implements the #ifdef/#ifndef directive.  isIfndef is
1795141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// true when this is a #ifndef directive.  ReadAnyTokensBeforeDirective is true
1796141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// if any tokens have been returned or pp-directives activated before this
1797141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// #ifndef has been lexed.
1798141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///
1799141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleIfdefDirective(Token &Result, bool isIfndef,
1800141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                        bool ReadAnyTokensBeforeDirective) {
1801141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumIf;
1802141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token DirectiveTok = Result;
1803141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1804141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  Token MacroNameTok;
1805141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ReadMacroName(MacroNameTok);
18061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1807141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Error reading macro name?  If so, diagnostic already issued.
180884021556baceb76eedf7d44be8ba71d9b8cfaccePeter Collingbourne  if (MacroNameTok.is(tok::eod)) {
1809141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Skip code until we get to #endif.  This helps with recovery by not
1810141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // emitting an error when the #endif is reached.
1811141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    SkipExcludedConditionalBlock(DirectiveTok.getLocation(),
1812141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                 /*Foundnonskip*/false, /*FoundElse*/false);
1813141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    return;
1814141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
18151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1816141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Check to see if this is the last token on the #if[n]def line.
181735410d5cbbb18735d76e00496540d416dc49b577Chris Lattner  CheckEndOfDirective(isIfndef ? "ifndef" : "ifdef");
1818141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
181913d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner  IdentifierInfo *MII = MacroNameTok.getIdentifierInfo();
182013d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner  MacroInfo *MI = getMacroInfo(MII);
1821193575455e00eca03fd7177f60e3f2e6263cb661Kovarththanan Rajaratnam
182260e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  if (CurPPLexer->getConditionalStackDepth() == 0) {
182313d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner    // If the start of a top-level #ifdef and if the macro is not defined,
182413d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner    // inform MIOpt that this might be the start of a proper include guard.
182513d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner    // Otherwise it is some other form of unknown conditional which we can't
182613d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner    // handle.
182713d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner    if (!ReadAnyTokensBeforeDirective && MI == 0) {
1828141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner      assert(isIfndef && "#ifdef shouldn't reach here");
182913d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner      CurPPLexer->MIOpt.EnterTopLevelIFNDEF(MII);
1830141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    } else
183160e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek      CurPPLexer->MIOpt.EnterTopLevelConditional();
1832141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
1833141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1834141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If there is a macro, process it.
1835141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (MI)  // Mark it used.
18360827408865e32789e0ec4b8113a302ccdc531423Argyrios Kyrtzidis    markMacroAsUsed(MI);
18371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1838141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Should we include the stuff contained by this directive?
1839141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (!MI == isIfndef) {
1840141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Yes, remember that we are inside a conditional, then lex the next token.
18411d9c54df56391ac4740db27d551782e81189cb51Chris Lattner    CurPPLexer->pushConditionalLevel(DirectiveTok.getLocation(),
18421d9c54df56391ac4740db27d551782e81189cb51Chris Lattner                                     /*wasskip*/false, /*foundnonskip*/true,
18431d9c54df56391ac4740db27d551782e81189cb51Chris Lattner                                     /*foundelse*/false);
1844141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else {
184508985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein    // No, skip the contents of this block.
1846141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    SkipExcludedConditionalBlock(DirectiveTok.getLocation(),
18471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                 /*Foundnonskip*/false,
1848141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                 /*FoundElse*/false);
1849141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
185008985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein
185108985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  if (Callbacks) {
185208985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein    if (isIfndef)
18532aa9267cf193e5eceb9fd24a51b51c031b606fe9Craig Silverstein      Callbacks->Ifndef(MacroNameTok);
185408985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein    else
18552aa9267cf193e5eceb9fd24a51b51c031b606fe9Craig Silverstein      Callbacks->Ifdef(MacroNameTok);
185608985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  }
1857141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1858141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1859141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIfDirective - Implements the #if directive.
1860141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///
1861141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleIfDirective(Token &IfToken,
1862141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                     bool ReadAnyTokensBeforeDirective) {
1863141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumIf;
18641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
186508985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  // Parse and evaluate the conditional expression.
1866141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  IdentifierInfo *IfNDefMacro = 0;
186708985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  const SourceLocation ConditionalBegin = CurPPLexer->getSourceLocation();
186808985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  const bool ConditionalTrue = EvaluateDirectiveExpression(IfNDefMacro);
186908985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  const SourceLocation ConditionalEnd = CurPPLexer->getSourceLocation();
18700049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes
18710049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes  // If this condition is equivalent to #ifndef X, and if this is the first
18720049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes  // directive seen, handle it for the multiple-include optimization.
187360e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  if (CurPPLexer->getConditionalStackDepth() == 0) {
187413d283d6a2f50cc4a2b03357191d49672de0eb9bChris Lattner    if (!ReadAnyTokensBeforeDirective && IfNDefMacro && ConditionalTrue)
187560e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek      CurPPLexer->MIOpt.EnterTopLevelIFNDEF(IfNDefMacro);
18760049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes    else
187760e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek      CurPPLexer->MIOpt.EnterTopLevelConditional();
18780049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes  }
18790049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes
1880141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Should we include the stuff contained by this directive?
1881141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (ConditionalTrue) {
1882141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // Yes, remember that we are inside a conditional, then lex the next token.
188360e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek    CurPPLexer->pushConditionalLevel(IfToken.getLocation(), /*wasskip*/false,
1884141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                   /*foundnonskip*/true, /*foundelse*/false);
1885141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  } else {
188608985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein    // No, skip the contents of this block.
18871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    SkipExcludedConditionalBlock(IfToken.getLocation(), /*Foundnonskip*/false,
1888141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner                                 /*FoundElse*/false);
1889141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
189008985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein
189108985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  if (Callbacks)
189208985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein    Callbacks->If(SourceRange(ConditionalBegin, ConditionalEnd));
1893141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1894141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1895141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleEndifDirective - Implements the #endif directive.
1896141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner///
1897141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleEndifDirective(Token &EndifToken) {
1898141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumEndif;
18991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1900141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // Check that this is the whole directive.
190135410d5cbbb18735d76e00496540d416dc49b577Chris Lattner  CheckEndOfDirective("endif");
19021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1903141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  PPConditionalInfo CondInfo;
190460e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  if (CurPPLexer->popConditionalLevel(CondInfo)) {
1905141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner    // No conditionals on the stack: this is an #endif without an #if.
19063692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    Diag(EndifToken, diag::err_pp_endif_without_if);
19073692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    return;
1908141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  }
19091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1910141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If this the end of a top-level #endif, inform MIOpt.
191160e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  if (CurPPLexer->getConditionalStackDepth() == 0)
191260e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek    CurPPLexer->MIOpt.ExitTopLevelConditional();
19131eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
191460e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  assert(!CondInfo.WasSkipping && !CurPPLexer->LexingRawMode &&
1915141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner         "This code should only be reachable in the non-skipping case!");
1916141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
191708985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  if (Callbacks)
191808985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein    Callbacks->Endif();
191908985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein}
1920141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
192108985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein/// HandleElseDirective - Implements the #else directive.
192208985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein///
1923141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleElseDirective(Token &Result) {
1924141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumElse;
19251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1926141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // #else directive in a non-skipping conditional... start skipping.
192735410d5cbbb18735d76e00496540d416dc49b577Chris Lattner  CheckEndOfDirective("else");
19281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1929141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  PPConditionalInfo CI;
19303692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner  if (CurPPLexer->popConditionalLevel(CI)) {
19313692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    Diag(Result, diag::pp_err_else_without_if);
19323692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    return;
19333692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner  }
19341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1935141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If this is a top-level #else, inform the MIOpt.
193660e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  if (CurPPLexer->getConditionalStackDepth() == 0)
193760e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek    CurPPLexer->MIOpt.EnterTopLevelConditional();
1938141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1939141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If this is a #else with a #else before it, report the error.
1940141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (CI.FoundElse) Diag(Result, diag::pp_err_else_after_else);
19411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
194208985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  // Finally, skip the rest of the contents of this block.
194308985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  SkipExcludedConditionalBlock(CI.IfLoc, /*Foundnonskip*/true,
19446b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis                               /*FoundElse*/true, Result.getLocation());
194508985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein
194608985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  if (Callbacks)
194708985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein    Callbacks->Else();
1948141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1949141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
195008985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein/// HandleElifDirective - Implements the #elif directive.
195108985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein///
1952141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleElifDirective(Token &ElifToken) {
1953141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  ++NumElse;
19541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1955141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // #elif directive in a non-skipping conditional... start skipping.
1956141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // We don't care what the condition is, because we will always skip it (since
1957141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // the block immediately before it was included).
195808985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  const SourceLocation ConditionalBegin = CurPPLexer->getSourceLocation();
1959141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  DiscardUntilEndOfDirective();
196008985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  const SourceLocation ConditionalEnd = CurPPLexer->getSourceLocation();
1961141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
1962141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  PPConditionalInfo CI;
19633692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner  if (CurPPLexer->popConditionalLevel(CI)) {
19643692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    Diag(ElifToken, diag::pp_err_elif_without_if);
19653692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner    return;
19663692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner  }
19671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1968141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If this is a top-level #elif, inform the MIOpt.
196960e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek  if (CurPPLexer->getConditionalStackDepth() == 0)
197060e45d4c41479fb58ad14d97efd03418e8549509Ted Kremenek    CurPPLexer->MIOpt.EnterTopLevelConditional();
19711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1972141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  // If this is a #elif with a #else before it, report the error.
1973141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner  if (CI.FoundElse) Diag(ElifToken, diag::pp_err_elif_after_else);
1974141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner
197508985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  // Finally, skip the rest of the contents of this block.
197608985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  SkipExcludedConditionalBlock(CI.IfLoc, /*Foundnonskip*/true,
19776b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis                               /*FoundElse*/CI.FoundElse,
19786b4ff04dd22e7159887ea6c1bb8b1c01fc9c70a9Argyrios Kyrtzidis                               ElifToken.getLocation());
197908985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein
198008985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein  if (Callbacks)
198108985b91e5a87e7731ba24c675f2dc9f9cb645d4Craig Silverstein    Callbacks->Elif(SourceRange(ConditionalBegin, ConditionalEnd));
1982141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner}
1983