PPDirectives.cpp revision 9d72851fec9e9c62570a027d42701562bbf29751
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" 15141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner#include "clang/Lex/HeaderSearch.h" 16141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner#include "clang/Lex/MacroInfo.h" 17141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner#include "clang/Basic/Diagnostic.h" 18141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner#include "clang/Basic/SourceManager.h" 19141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnerusing namespace clang; 20141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 21141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 22141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Utility Methods for Preprocessor Directive Handling. 23141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 24141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 25141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// DiscardUntilEndOfDirective - Read and discard all tokens remaining on the 26141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// current line until the tok::eom token is found. 27141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::DiscardUntilEndOfDirective() { 28141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token Tmp; 29141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner do { 30141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tmp); 31141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } while (Tmp.isNot(tok::eom)); 32141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 33141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 34141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// isCXXNamedOperator - Returns "true" if the token is a named operator in C++. 35141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnerstatic bool isCXXNamedOperator(const std::string &Spelling) { 36141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Spelling == "and" || Spelling == "bitand" || Spelling == "bitor" || 37141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Spelling == "compl" || Spelling == "not" || Spelling == "not_eq" || 38141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Spelling == "or" || Spelling == "xor"; 39141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 40141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 41141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// ReadMacroName - Lex and validate a macro name, which occurs after a 42141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// #define or #undef. This sets the token kind to eom and discards the rest 43141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// of the macro line if the macro name is invalid. isDefineUndef is 1 if 44141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// this is due to a a #define, 2 if #undef directive, 0 if it is something 45141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// else (e.g. #ifdef). 46141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::ReadMacroName(Token &MacroNameTok, char isDefineUndef) { 47141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Read the token, don't allow macro expansion on it. 48141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(MacroNameTok); 49141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 50141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Missing macro name? 51141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MacroNameTok.is(tok::eom)) 52141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(MacroNameTok, diag::err_pp_missing_macro_name); 53141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 54141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner IdentifierInfo *II = MacroNameTok.getIdentifierInfo(); 55141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (II == 0) { 56141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner std::string Spelling = getSpelling(MacroNameTok); 57141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (isCXXNamedOperator(Spelling)) 58141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C++ 2.5p2: Alternative tokens behave the same as its primary token 59141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // except for their spellings. 60141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MacroNameTok, diag::err_pp_operator_used_as_macro_name, Spelling); 61141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner else 62141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MacroNameTok, diag::err_pp_macro_not_identifier); 63141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Fall through on error. 64141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (isDefineUndef && II->getPPKeywordID() == tok::pp_defined) { 65141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Error if defining "defined": C99 6.10.8.4. 66141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MacroNameTok, diag::err_defined_macro_name); 67141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (isDefineUndef && II->hasMacroDefinition() && 68141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner getMacroInfo(II)->isBuiltinMacro()) { 69141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Error if defining "__LINE__" and other builtins: C99 6.10.8.4. 70141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (isDefineUndef == 1) 71141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MacroNameTok, diag::pp_redef_builtin_macro); 72141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner else 73141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MacroNameTok, diag::pp_undef_builtin_macro); 74141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 75141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Okay, we got a good identifier node. Return it. 76141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 77141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 78141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 79141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Invalid macro name, read and discard the rest of the line. Then set the 80141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // token kind to tok::eom. 81141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MacroNameTok.setKind(tok::eom); 82141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return DiscardUntilEndOfDirective(); 83141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 84141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 85141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// CheckEndOfDirective - Ensure that the next token is a tok::eom token. If 86141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// not, emit a diagnostic and consume up until the eom. 87141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::CheckEndOfDirective(const char *DirType) { 88141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token Tmp; 89141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Lex unexpanded tokens: macros might expand to zero tokens, causing us to 90141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // miss diagnosing invalid lines. 91141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tmp); 92141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 93141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // There should be no tokens after the directive, but we allow them as an 94141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // extension. 95141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner while (Tmp.is(tok::comment)) // Skip comments in -C mode. 96141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tmp); 97141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 98141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Tmp.isNot(tok::eom)) { 99141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tmp, diag::ext_pp_extra_tokens_at_eol, DirType); 100141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 101141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 102141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 103141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 104141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 105141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 106141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// SkipExcludedConditionalBlock - We just read a #if or related directive and 107141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// decided that the subsequent tokens are in the #if'd out portion of the 108141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// file. Lex the rest of the file, until we see an #endif. If 109141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// FoundNonSkipPortion is true, then we have already emitted code for part of 110141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// this #if directive, so #else/#elif blocks should never be entered. If ElseOk 111141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// is true, then #else directives are ok, if not, then we have already seen one 112141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// so a #else directive is a duplicate. When this returns, the caller can lex 113141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// the first valid token. 114141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::SkipExcludedConditionalBlock(SourceLocation IfTokenLoc, 115141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool FoundNonSkipPortion, 116141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool FoundElse) { 117141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumSkipped; 1186cfe7594a46b5d270142cfcb688a9c1a3a487a48Chris Lattner assert(CurTokenLexer == 0 && CurLexer && 119141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner "Lexing a macro, not a file?"); 120141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 121141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->pushConditionalLevel(IfTokenLoc, /*isSkipping*/false, 122141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FoundNonSkipPortion, FoundElse); 123141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 124141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Enter raw mode to disable identifier lookup (and thus macro expansion), 125141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // disabling warnings, etc. 126141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->LexingRawMode = true; 127141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token Tok; 128141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner while (1) { 129141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->Lex(Tok); 130141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 131141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is the end of the buffer, we have an error. 132141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Tok.is(tok::eof)) { 133141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Emit errors for each unterminated conditional on the stack, including 134141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // the current one. 135141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner while (!CurLexer->ConditionalStack.empty()) { 136141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(CurLexer->ConditionalStack.back().IfLoc, 137141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner diag::err_pp_unterminated_conditional); 138141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->ConditionalStack.pop_back(); 139141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 140141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 141141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Just return and let the caller lex after this #include. 142141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 143141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 144141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 145141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this token is not a preprocessor directive, just skip it. 146141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Tok.isNot(tok::hash) || !Tok.isAtStartOfLine()) 147141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner continue; 148141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 149141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // We just parsed a # character at the start of a line, so we're in 150141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // directive mode. Tell the lexer this so any newlines we see will be 151141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // converted into an EOM token (this terminates the macro). 152141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->ParsingPreprocessorDirective = true; 153aea3ad073793f7764aa914bd9e551f858a0b5739Chris Lattner CurLexer->SetCommentRetentionState(false); 154141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 155141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 156141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Read the next token, the directive flavor. 157141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 158141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 159141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this isn't an identifier directive (e.g. is "# 1\n" or "#\n", or 160141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // something bogus), skip it. 161141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Tok.isNot(tok::identifier)) { 162141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->ParsingPreprocessorDirective = false; 163141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Restore comment saving mode. 164aea3ad073793f7764aa914bd9e551f858a0b5739Chris Lattner CurLexer->SetCommentRetentionState(KeepComments); 165141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner continue; 166141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 167141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 168141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If the first letter isn't i or e, it isn't intesting to us. We know that 169141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // this is safe in the face of spelling differences, because there is no way 170141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // to spell an i/e in a strange way that is another letter. Skipping this 171141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // allows us to avoid looking up the identifier info for #define/#undef and 172141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // other common directives. 173141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const char *RawCharData = SourceMgr.getCharacterData(Tok.getLocation()); 174141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner char FirstChar = RawCharData[0]; 175141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (FirstChar >= 'a' && FirstChar <= 'z' && 176141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FirstChar != 'i' && FirstChar != 'e') { 177141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->ParsingPreprocessorDirective = false; 178141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Restore comment saving mode. 179aea3ad073793f7764aa914bd9e551f858a0b5739Chris Lattner CurLexer->SetCommentRetentionState(KeepComments); 180141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner continue; 181141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 182141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 183141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Get the identifier name without trigraphs or embedded newlines. Note 184141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // that we can't use Tok.getIdentifierInfo() because its lookup is disabled 185141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // when skipping. 186141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // TODO: could do this with zero copies in the no-clean case by using 187141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // strncmp below. 188141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner char Directive[20]; 189141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner unsigned IdLen; 190141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!Tok.needsCleaning() && Tok.getLength() < 20) { 191141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner IdLen = Tok.getLength(); 192141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner memcpy(Directive, RawCharData, IdLen); 193141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Directive[IdLen] = 0; 194141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 195141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner std::string DirectiveStr = getSpelling(Tok); 196141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner IdLen = DirectiveStr.size(); 197141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (IdLen >= 20) { 198141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->ParsingPreprocessorDirective = false; 199141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Restore comment saving mode. 200aea3ad073793f7764aa914bd9e551f858a0b5739Chris Lattner CurLexer->SetCommentRetentionState(KeepComments); 201141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner continue; 202141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 203141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner memcpy(Directive, &DirectiveStr[0], IdLen); 204141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Directive[IdLen] = 0; 205141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 206141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 207141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (FirstChar == 'i' && Directive[1] == 'f') { 208141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if ((IdLen == 2) || // "if" 209141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner (IdLen == 5 && !strcmp(Directive+2, "def")) || // "ifdef" 210141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner (IdLen == 6 && !strcmp(Directive+2, "ndef"))) { // "ifndef" 211141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // We know the entire #if/#ifdef/#ifndef block will be skipped, don't 212141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // bother parsing the condition. 213141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 214141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->pushConditionalLevel(Tok.getLocation(), /*wasskipping*/true, 215141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*foundnonskip*/false, 216141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*fnddelse*/false); 217141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 218141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (FirstChar == 'e') { 219141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (IdLen == 5 && !strcmp(Directive+1, "ndif")) { // "endif" 220141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CheckEndOfDirective("#endif"); 221141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PPConditionalInfo CondInfo; 222141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CondInfo.WasSkipping = true; // Silence bogus warning. 223141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool InCond = CurLexer->popConditionalLevel(CondInfo); 224141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner InCond = InCond; // Silence warning in no-asserts mode. 225141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner assert(!InCond && "Can't be skipping if not in a conditional!"); 226141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 227141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If we popped the outermost skipping block, we're done skipping! 228141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!CondInfo.WasSkipping) 229141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 230141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (IdLen == 4 && !strcmp(Directive+1, "lse")) { // "else". 231141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #else directive in a skipping conditional. If not in some other 232141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // skipping conditional, and if #else hasn't already been seen, enter it 233141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // as a non-skipping conditional. 234141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CheckEndOfDirective("#else"); 235141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PPConditionalInfo &CondInfo = CurLexer->peekConditionalLevel(); 236141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 237141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is a #else with a #else before it, report the error. 238141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CondInfo.FoundElse) Diag(Tok, diag::pp_err_else_after_else); 239141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 240141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Note that we've seen a #else in this conditional. 241141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CondInfo.FoundElse = true; 242141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 243141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If the conditional is at the top level, and the #if block wasn't 244141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // entered, enter the #else block now. 245141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!CondInfo.WasSkipping && !CondInfo.FoundNonSkip) { 246141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CondInfo.FoundNonSkip = true; 247141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 248141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 249141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (IdLen == 4 && !strcmp(Directive+1, "lif")) { // "elif". 250141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PPConditionalInfo &CondInfo = CurLexer->peekConditionalLevel(); 251141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 252141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool ShouldEnter; 253141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is in a skipping block or if we're already handled this #if 254141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // block, don't bother parsing the condition. 255141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CondInfo.WasSkipping || CondInfo.FoundNonSkip) { 256141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 257141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ShouldEnter = false; 258141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 259141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Restore the value of LexingRawMode so that identifiers are 260141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // looked up, etc, inside the #elif expression. 261141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner assert(CurLexer->LexingRawMode && "We have to be skipping here!"); 262141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->LexingRawMode = false; 263141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner IdentifierInfo *IfNDefMacro = 0; 264141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ShouldEnter = EvaluateDirectiveExpression(IfNDefMacro); 265141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->LexingRawMode = true; 266141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 267141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 268141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is a #elif with a #else before it, report the error. 269141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CondInfo.FoundElse) Diag(Tok, diag::pp_err_elif_after_else); 270141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 271141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this condition is true, enter it! 272141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (ShouldEnter) { 273141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CondInfo.FoundNonSkip = true; 274141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 275141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 276141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 277141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 278141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 279141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->ParsingPreprocessorDirective = false; 280141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Restore comment saving mode. 281aea3ad073793f7764aa914bd9e551f858a0b5739Chris Lattner CurLexer->SetCommentRetentionState(KeepComments); 282141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 283141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 284141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Finally, if we are out of the conditional (saw an #endif or ran off the end 285141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // of the file, just stop skipping and return to lexing whatever came after 286141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // the #if block. 287141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->LexingRawMode = false; 288141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 289141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 2901072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner/// LookupFile - Given a "foo" or <foo> reference, look up the indicated file, 2911072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner/// return null on failure. isAngled indicates whether the file reference is 2921072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner/// for system #include's or not (i.e. using <> instead of ""). 2931072509e126e70c85652ab0056bcabd72edff7e5Chris Lattnerconst FileEntry *Preprocessor::LookupFile(const char *FilenameStart, 2941072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner const char *FilenameEnd, 2951072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner bool isAngled, 2961072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner const DirectoryLookup *FromDir, 2971072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner const DirectoryLookup *&CurDir) { 2981072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner // If the header lookup mechanism may be relative to the current file, pass in 2991072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner // info about where the current file is. 3001072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner const FileEntry *CurFileEnt = 0; 3011072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner if (!FromDir) { 3021072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner SourceLocation FileLoc = getCurrentFileLexer()->getFileLoc(); 3031072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner CurFileEnt = SourceMgr.getFileEntryForLoc(FileLoc); 3041072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner } 3051072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner 3061072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner // Do a standard file entry lookup. 3071072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner CurDir = CurDirLookup; 3081072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner const FileEntry *FE = 3091072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner HeaderInfo.LookupFile(FilenameStart, FilenameEnd, 3101072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner isAngled, FromDir, CurDir, CurFileEnt); 3111072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner if (FE) return FE; 3121072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner 3131072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner // Otherwise, see if this is a subframework header. If so, this is relative 3141072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner // to one of the headers on the #include stack. Walk the list of the current 3151072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner // headers on the #include stack and pass them to HeaderInfo. 3161072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner if (CurLexer && !CurLexer->Is_PragmaLexer) { 3171072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner if ((CurFileEnt = SourceMgr.getFileEntryForLoc(CurLexer->getFileLoc()))) 3181072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner if ((FE = HeaderInfo.LookupSubframeworkHeader(FilenameStart, FilenameEnd, 3191072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner CurFileEnt))) 3201072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner return FE; 3211072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner } 3221072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner 3231072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner for (unsigned i = 0, e = IncludeMacroStack.size(); i != e; ++i) { 3241072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner IncludeStackInfo &ISEntry = IncludeMacroStack[e-i-1]; 3251072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner if (ISEntry.TheLexer && !ISEntry.TheLexer->Is_PragmaLexer) { 3261072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner if ((CurFileEnt = 3271072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner SourceMgr.getFileEntryForLoc(ISEntry.TheLexer->getFileLoc()))) 3281072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner if ((FE = HeaderInfo.LookupSubframeworkHeader(FilenameStart, 3291072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner FilenameEnd, CurFileEnt))) 3301072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner return FE; 3311072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner } 3321072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner } 3331072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner 3341072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner // Otherwise, we really couldn't find the file. 3351072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner return 0; 3361072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner} 3371072509e126e70c85652ab0056bcabd72edff7e5Chris Lattner 338141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 339141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 340141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Preprocessor Directive Handling. 341141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 342141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 343141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleDirective - This callback is invoked when the lexer sees a # token 344141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// at the start of a line. This consumes the directive, modifies the 345141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// lexer/preprocessor state, and advances the lexer(s) so that the next token 346141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// read is the correct one. 347141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleDirective(Token &Result) { 348141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // FIXME: Traditional: # with whitespace before it not recognized by K&R? 349141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 350141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // We just parsed a # character at the start of a line, so we're in directive 351141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // mode. Tell the lexer this so any newlines we see will be converted into an 352141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // EOM token (which terminates the directive). 353141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->ParsingPreprocessorDirective = true; 354141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 355141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumDirectives; 356141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 357141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // We are about to read a token. For the multiple-include optimization FA to 358141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // work, we have to remember if we had read any tokens *before* this 359141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // pp-directive. 360141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool ReadAnyTokensBeforeDirective = CurLexer->MIOpt.getHasReadAnyTokensVal(); 361141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 362141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Read the next token, the directive flavor. This isn't expanded due to 363141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 6.10.3p8. 364141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Result); 365141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 366141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 6.10.3p11: Is this preprocessor directive in macro invocation? e.g.: 367141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #define A(x) #x 368141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // A(abc 369141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #warning blah 370141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // def) 371141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If so, the user is relying on non-portable behavior, emit a diagnostic. 372141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (InMacroArgs) 373141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Result, diag::ext_embedded_directive); 374141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 375141e71fec621f65ab401dbe66609f7d3c033c6a5Chris LattnerTryAgain: 376141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner switch (Result.getKind()) { 377141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::eom: 378141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; // null directive. 379141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::comment: 380141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Handle stuff like "# /*foo*/ define X" in -E -C mode. 381141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Result); 382141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner goto TryAgain; 383141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 384141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::numeric_constant: 385141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // FIXME: implement # 7 line numbers! 386141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 387141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 388141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner default: 389141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner IdentifierInfo *II = Result.getIdentifierInfo(); 390141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (II == 0) break; // Not an identifier. 391141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 392141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Ask what the preprocessor keyword ID is. 393141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner switch (II->getPPKeywordID()) { 394141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner default: break; 395141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 6.10.1 - Conditional Inclusion. 396141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_if: 397141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIfDirective(Result, ReadAnyTokensBeforeDirective); 398141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_ifdef: 399141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIfdefDirective(Result, false, true/*not valid for miopt*/); 400141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_ifndef: 401141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIfdefDirective(Result, true, ReadAnyTokensBeforeDirective); 402141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_elif: 403141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleElifDirective(Result); 404141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_else: 405141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleElseDirective(Result); 406141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_endif: 407141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleEndifDirective(Result); 408141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 409141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 6.10.2 - Source File Inclusion. 410141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_include: 411141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIncludeDirective(Result); // Handle #include. 412141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 413141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 6.10.3 - Macro Replacement. 414141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_define: 415141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleDefineDirective(Result); 416141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_undef: 417141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleUndefDirective(Result); 418141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 419141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 6.10.4 - Line Control. 420141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_line: 421141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // FIXME: implement #line 422141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 423141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 424141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 425141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 6.10.5 - Error Directive. 426141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_error: 427141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleUserDiagnosticDirective(Result, false); 428141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 429141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 6.10.6 - Pragma Directive. 430141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_pragma: 431141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandlePragmaDirective(); 432141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 433141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // GNU Extensions. 434141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_import: 435141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleImportDirective(Result); 436141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_include_next: 437141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIncludeNextDirective(Result); 438141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 439141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_warning: 440141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Result, diag::ext_pp_warning_directive); 441141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleUserDiagnosticDirective(Result, true); 442141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_ident: 443141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIdentSCCSDirective(Result); 444141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_sccs: 445141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIdentSCCSDirective(Result); 446141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_assert: 447141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner //isExtension = true; // FIXME: implement #assert 448141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 449141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::pp_unassert: 450141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner //isExtension = true; // FIXME: implement #unassert 451141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 452141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 453141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 454141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 455141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 456141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If we reached here, the preprocessing token is not valid! 457141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Result, diag::err_pp_invalid_directive); 458141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 459141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Read the rest of the PP line. 460141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 461141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 462141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Okay, we're done parsing the directive. 463141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 464141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 465141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleUserDiagnosticDirective(Token &Tok, 466141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool isWarning) { 467141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Read the rest of the line raw. We do this because we don't want macros 468141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // to be expanded and we don't require that the tokens be valid preprocessing 469141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // tokens. For example, this is allowed: "#warning ` 'foo". GCC does 470141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // collapse multiple consequtive white space between tokens, but this isn't 471141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // specified by the standard. 472141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner std::string Message = CurLexer->ReadToEndOfLine(); 473141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 474141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner unsigned DiagID = isWarning ? diag::pp_hash_warning : diag::err_pp_hash_error; 475141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(Tok, DiagID, Message); 476141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 477141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 478141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIdentSCCSDirective - Handle a #ident/#sccs directive. 479141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// 480141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleIdentSCCSDirective(Token &Tok) { 481141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Yes, this directive is an extension. 482141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::ext_pp_ident_directive); 483141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 484141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Read the string argument. 485141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token StrTok; 486141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Lex(StrTok); 487141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 488141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If the token kind isn't a string, it's a malformed directive. 489141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (StrTok.isNot(tok::string_literal) && 490141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner StrTok.isNot(tok::wide_string_literal)) 491141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(StrTok, diag::err_pp_malformed_ident); 492141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 493141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Verify that there is nothing after the string, other than EOM. 494141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CheckEndOfDirective("#ident"); 495141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 496141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Callbacks) 497141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Callbacks->Ident(Tok.getLocation(), getSpelling(StrTok)); 498141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 499141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 500141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 501141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Preprocessor Include Directive Handling. 502141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 503141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 504141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// GetIncludeFilenameSpelling - Turn the specified lexer token into a fully 505141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// checked and spelled filename, e.g. as an operand of #include. This returns 506141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// true if the input filename was in <>'s or false if it were in ""'s. The 507141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// caller is expected to provide a buffer that is large enough to hold the 508141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// spelling of the filename, but is also expected to handle the case when 509141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// this method decides to use a different buffer. 510141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnerbool Preprocessor::GetIncludeFilenameSpelling(SourceLocation Loc, 511141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const char *&BufStart, 512141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const char *&BufEnd) { 513141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Get the text form of the filename. 514141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner assert(BufStart != BufEnd && "Can't have tokens with empty spellings!"); 515141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 516141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Make sure the filename is <x> or "x". 517141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool isAngled; 518141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (BufStart[0] == '<') { 519141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (BufEnd[-1] != '>') { 520141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Loc, diag::err_pp_expects_filename); 521141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner BufStart = 0; 522141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 523141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 524141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner isAngled = true; 525141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (BufStart[0] == '"') { 526141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (BufEnd[-1] != '"') { 527141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Loc, diag::err_pp_expects_filename); 528141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner BufStart = 0; 529141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 530141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 531141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner isAngled = false; 532141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 533141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Loc, diag::err_pp_expects_filename); 534141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner BufStart = 0; 535141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 536141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 537141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 538141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Diagnose #include "" as invalid. 539141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (BufEnd-BufStart <= 2) { 540141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Loc, diag::err_pp_empty_filename); 541141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner BufStart = 0; 542141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return ""; 543141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 544141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 545141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Skip the brackets. 546141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++BufStart; 547141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner --BufEnd; 548141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return isAngled; 549141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 550141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 551141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// ConcatenateIncludeName - Handle cases where the #include name is expanded 552141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// from a macro as multiple tokens, which need to be glued together. This 553141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// occurs for code like: 554141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// #define FOO <a/b.h> 555141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// #include FOO 556141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// because in this case, "<a/b.h>" is returned as 7 tokens, not one. 557141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// 558141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// This code concatenates and consumes tokens up to the '>' token. It returns 559141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// false if the > was found, otherwise it returns true if it finds and consumes 560141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// the EOM marker. 561141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnerstatic bool ConcatenateIncludeName(llvm::SmallVector<char, 128> &FilenameBuffer, 562141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Preprocessor &PP) { 563141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token CurTok; 564141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 565141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PP.Lex(CurTok); 566141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner while (CurTok.isNot(tok::eom)) { 567141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Append the spelling of this token to the buffer. If there was a space 568141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // before it, add it now. 569141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurTok.hasLeadingSpace()) 570141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameBuffer.push_back(' '); 571141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 572141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Get the spelling of the token, directly into FilenameBuffer if possible. 573141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner unsigned PreAppendSize = FilenameBuffer.size(); 574141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameBuffer.resize(PreAppendSize+CurTok.getLength()); 575141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 576141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const char *BufPtr = &FilenameBuffer[PreAppendSize]; 577141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner unsigned ActualLen = PP.getSpelling(CurTok, BufPtr); 578141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 579141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If the token was spelled somewhere else, copy it into FilenameBuffer. 580141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (BufPtr != &FilenameBuffer[PreAppendSize]) 581141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner memcpy(&FilenameBuffer[PreAppendSize], BufPtr, ActualLen); 582141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 583141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Resize FilenameBuffer to the correct size. 584141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurTok.getLength() != ActualLen) 585141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameBuffer.resize(PreAppendSize+ActualLen); 586141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 587141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If we found the '>' marker, return success. 588141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurTok.is(tok::greater)) 589141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return false; 590141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 591141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PP.Lex(CurTok); 592141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 593141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 594141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If we hit the eom marker, emit an error and return true so that the caller 595141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // knows the EOM has been read. 596141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PP.Diag(CurTok.getLocation(), diag::err_pp_expects_filename); 597141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 598141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 599141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 600141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIncludeDirective - The "#include" tokens have just been read, read the 601141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// file to be included from the lexer, then include it! This is a common 602141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// routine with functionality shared between #include, #include_next and 603721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner/// #import. LookupFrom is set when this is a #include_next directive, it 604721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner/// specifies the file to start searching from. 605141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleIncludeDirective(Token &IncludeTok, 606141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const DirectoryLookup *LookupFrom, 607141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool isImport) { 608141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 609141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token FilenameTok; 610141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->LexIncludeFilename(FilenameTok); 611141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 612141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Reserve a buffer to get the spelling. 613141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner llvm::SmallVector<char, 128> FilenameBuffer; 614141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const char *FilenameStart, *FilenameEnd; 615141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 616141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner switch (FilenameTok.getKind()) { 617141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::eom: 618141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If the token kind is EOM, the error has already been diagnosed. 619141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 620141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 621141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::angle_string_literal: 622141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::string_literal: { 623141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameBuffer.resize(FilenameTok.getLength()); 624141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameStart = &FilenameBuffer[0]; 625141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner unsigned Len = getSpelling(FilenameTok, FilenameStart); 626141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameEnd = FilenameStart+Len; 627141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 628141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 629141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 630141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::less: 631141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // This could be a <foo/bar.h> file coming from a macro expansion. In this 632141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // case, glue the tokens together into FilenameBuffer and interpret those. 633141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameBuffer.push_back('<'); 634141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (ConcatenateIncludeName(FilenameBuffer, *this)) 635141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; // Found <eom> but no ">"? Diagnostic already emitted. 636141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameStart = &FilenameBuffer[0]; 637141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameEnd = &FilenameBuffer[FilenameBuffer.size()]; 638141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 639141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner default: 640141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(FilenameTok.getLocation(), diag::err_pp_expects_filename); 641141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 642141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 643141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 644141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 645141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool isAngled = GetIncludeFilenameSpelling(FilenameTok.getLocation(), 646141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner FilenameStart, FilenameEnd); 647141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If GetIncludeFilenameSpelling set the start ptr to null, there was an 648141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // error. 649141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (FilenameStart == 0) { 650141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 651141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 652141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 653141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 654141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Verify that there is nothing after the filename, other than EOM. Use the 655141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // preprocessor to lex this in case lexing the filename entered a macro. 656141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CheckEndOfDirective("#include"); 657141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 658141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Check that we don't have infinite #include recursion. 659141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (IncludeMacroStack.size() == MaxAllowedIncludeStackDepth-1) 660141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(FilenameTok, diag::err_pp_include_too_deep); 661141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 662141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Search include directories. 663141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const DirectoryLookup *CurDir; 664141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const FileEntry *File = LookupFile(FilenameStart, FilenameEnd, 665141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner isAngled, LookupFrom, CurDir); 666141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (File == 0) 667141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(FilenameTok, diag::err_pp_file_not_found, 668141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner std::string(FilenameStart, FilenameEnd)); 669141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 670721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner // Ask HeaderInfo if we should enter this #include file. If not, #including 671721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner // this file will have no effect. 672721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner if (!HeaderInfo.ShouldEnterIncludeFile(File, isImport)) 673141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 674721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner 675721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner // The #included file will be considered to be a system header if either it is 676721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner // in a system include directory, or if the #includer is a system include 677721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner // header. 6789d72851fec9e9c62570a027d42701562bbf29751Chris Lattner SrcMgr::CharacteristicKind FileCharacter = 6790b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner std::max(HeaderInfo.getFileDirFlavor(File), 6800b9e736308af5397f558ffc8e780c438c2fdb563Chris Lattner SourceMgr.getFileCharacteristic(getCurrentFileLexer()->getFileLoc())); 681721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner 682141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Look up the file, create a File ID for it. 6837bfaaaecb3113f955db31e8d8a51acffd1bc0c27Nico Weber unsigned FileID = SourceMgr.createFileID(File, FilenameTok.getLocation(), 684721818304ac462d8c6ce05eecd02884033db78f1Chris Lattner FileCharacter); 685141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (FileID == 0) 686141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(FilenameTok, diag::err_pp_file_not_found, 687141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner std::string(FilenameStart, FilenameEnd)); 688141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 689141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Finally, if all is good, enter the new file! 690141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner EnterSourceFile(FileID, CurDir); 691141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 692141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 693141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIncludeNextDirective - Implements #include_next. 694141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// 695141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleIncludeNextDirective(Token &IncludeNextTok) { 696141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(IncludeNextTok, diag::ext_pp_include_next_directive); 697141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 698141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #include_next is like #include, except that we start searching after 699141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // the current found directory. If we can't do this, issue a 700141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // diagnostic. 701141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner const DirectoryLookup *Lookup = CurDirLookup; 702141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (isInPrimaryFile()) { 703141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Lookup = 0; 704141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(IncludeNextTok, diag::pp_include_next_in_primary); 705141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (Lookup == 0) { 706141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(IncludeNextTok, diag::pp_include_next_absolute_path); 707141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 708141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Start looking up in the next directory. 709141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++Lookup; 710141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 711141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 712141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIncludeDirective(IncludeNextTok, Lookup); 713141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 714141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 715141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleImportDirective - Implements #import. 716141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// 717141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleImportDirective(Token &ImportTok) { 718141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(ImportTok, diag::ext_pp_import_directive); 719141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 720141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return HandleIncludeDirective(ImportTok, 0, true); 721141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 722141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 723141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 724141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Preprocessor Macro Directive Handling. 725141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 726141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 727141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// ReadMacroDefinitionArgList - The ( starting an argument list of a macro 728141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// definition has just been read. Lex the rest of the arguments and the 729141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// closing ), updating MI with what we learn. Return true if an error occurs 730141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// parsing the arg list. 731141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnerbool Preprocessor::ReadMacroDefinitionArgList(MacroInfo *MI) { 732141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner llvm::SmallVector<IdentifierInfo*, 32> Arguments; 733141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 734141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token Tok; 735141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner while (1) { 736141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 737141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner switch (Tok.getKind()) { 738141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::r_paren: 739141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Found the end of the argument list. 740141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Arguments.empty()) { // #define FOO() 741141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setArgumentList(Arguments.begin(), Arguments.end()); 742141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return false; 743141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 744141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Otherwise we have #define FOO(A,) 745141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::err_pp_expected_ident_in_arg_list); 746141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 747141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::ellipsis: // #define X(... -> C99 varargs 748141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Warn if use of C99 feature in non-C99 mode. 749141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!Features.C99) Diag(Tok, diag::ext_variadic_macro); 750141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 751141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Lex the token after the identifier. 752141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 753141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Tok.isNot(tok::r_paren)) { 754141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::err_pp_missing_rparen_in_macro_def); 755141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 756141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 757141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Add the __VA_ARGS__ identifier as an argument. 758141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Arguments.push_back(Ident__VA_ARGS__); 759141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setIsC99Varargs(); 760141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setArgumentList(Arguments.begin(), Arguments.end()); 761141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return false; 762141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::eom: // #define X( 763141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::err_pp_missing_rparen_in_macro_def); 764141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 765141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner default: 766141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Handle keywords and identifiers here to accept things like 767141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #define Foo(for) for. 768141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner IdentifierInfo *II = Tok.getIdentifierInfo(); 769141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (II == 0) { 770141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #define X(1 771141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::err_pp_invalid_tok_in_arg_list); 772141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 773141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 774141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 775141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is already used as an argument, it is used multiple times (e.g. 776141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #define X(A,A. 777141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (std::find(Arguments.begin(), Arguments.end(), II) != 778141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Arguments.end()) { // C99 6.10.3p6 779141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::err_pp_duplicate_name_in_arg_list, II->getName()); 780141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 781141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 782141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 783141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Add the argument to the macro info. 784141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Arguments.push_back(II); 785141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 786141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Lex the token after the identifier. 787141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 788141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 789141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner switch (Tok.getKind()) { 790141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner default: // #define X(A B 791141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::err_pp_expected_comma_in_arg_list); 792141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 793141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::r_paren: // #define X(A) 794141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setArgumentList(Arguments.begin(), Arguments.end()); 795141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return false; 796141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::comma: // #define X(A, 797141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner break; 798141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner case tok::ellipsis: // #define X(A... -> GCC extension 799141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Diagnose extension. 800141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::ext_named_variadic_macro); 801141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 802141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Lex the token after the identifier. 803141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 804141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Tok.isNot(tok::r_paren)) { 805141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::err_pp_missing_rparen_in_macro_def); 806141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return true; 807141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 808141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 809141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setIsGNUVarargs(); 810141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setArgumentList(Arguments.begin(), Arguments.end()); 811141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return false; 812141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 813141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 814141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 815141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 816141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 817141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleDefineDirective - Implements #define. This consumes the entire macro 818141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// line then lets the caller lex the next real token. 819141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleDefineDirective(Token &DefineTok) { 820141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumDefined; 821141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 822141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token MacroNameTok; 823141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ReadMacroName(MacroNameTok, 1); 824141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 825141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Error reading macro name? If so, diagnostic already issued. 826141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MacroNameTok.is(tok::eom)) 827141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 828141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 829141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If we are supposed to keep comments in #defines, reenable comment saving 830141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // mode. 831aea3ad073793f7764aa914bd9e551f858a0b5739Chris Lattner CurLexer->SetCommentRetentionState(KeepMacroComments); 832141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 833141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Create the new macro. 834141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MacroInfo *MI = new MacroInfo(MacroNameTok.getLocation()); 835141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 836141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token Tok; 837141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 838141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 839141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is a function-like macro definition, parse the argument list, 840141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // marking each of the identifiers as being used as macro arguments. Also, 841141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // check other constraints on the first token of the macro body. 842141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Tok.is(tok::eom)) { 843141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If there is no body to this macro, we have no special handling here. 844141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (Tok.is(tok::l_paren) && !Tok.hasLeadingSpace()) { 845141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // This is a function-like macro definition. Read the argument list. 846141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setIsFunctionLike(); 847141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (ReadMacroDefinitionArgList(MI)) { 848141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Forget about MI. 849141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner delete MI; 850141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Throw away the rest of the line. 851141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurLexer->ParsingPreprocessorDirective) 852141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 853141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 854141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 855141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 856141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Read the first token after the arg list for down below. 857141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 858141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else if (!Tok.hasLeadingSpace()) { 859141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // C99 requires whitespace between the macro definition and the body. Emit 860141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // a diagnostic for something like "#define X+". 861141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Features.C99) { 862141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::ext_c99_whitespace_required_after_macro_name); 863141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 864141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // FIXME: C90/C++ do not get this diagnostic, but it does get a similar 865141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // one in some cases! 866141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 867141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 868141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // This is a normal token with leading space. Clear the leading space 869141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // marker on the first token to get proper expansion. 870141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Tok.clearFlag(Token::LeadingSpace); 871141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 872141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 873141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is a definition of a variadic C99 function-like macro, not using 874141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // the GNU named varargs extension, enabled __VA_ARGS__. 875141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 876141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // "Poison" __VA_ARGS__, which can only appear in the expansion of a macro. 877141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // This gets unpoisoned where it is allowed. 878141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner assert(Ident__VA_ARGS__->isPoisoned() && "__VA_ARGS__ should be poisoned!"); 879141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MI->isC99Varargs()) 880141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Ident__VA_ARGS__->setIsPoisoned(false); 881141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 882141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Read the rest of the macro body. 883141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MI->isObjectLike()) { 884141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Object-like macros are very simple, just read their body. 885141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner while (Tok.isNot(tok::eom)) { 886141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->AddTokenToBody(Tok); 887141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Get the next token of the macro. 888141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 889141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 890141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 891141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 892141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Otherwise, read the body of a function-like macro. This has to validate 893141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // the # (stringize) operator. 894141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner while (Tok.isNot(tok::eom)) { 895141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->AddTokenToBody(Tok); 896141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 897141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Check C99 6.10.3.2p1: ensure that # operators are followed by macro 898141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // parameters in function-like macro expansions. 899141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (Tok.isNot(tok::hash)) { 900141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Get the next token of the macro. 901141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 902141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner continue; 903141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 904141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 905141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Get the next token of the macro. 906141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 907141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 908141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Not a macro arg identifier? 909141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!Tok.getIdentifierInfo() || 910141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->getArgumentNum(Tok.getIdentifierInfo()) == -1) { 911141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(Tok, diag::err_pp_stringize_not_parameter); 912141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner delete MI; 913141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 914141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Disable __VA_ARGS__ again. 915141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Ident__VA_ARGS__->setIsPoisoned(true); 916141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 917141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 918141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 919141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Things look ok, add the param name token to the macro. 920141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->AddTokenToBody(Tok); 921141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 922141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Get the next token of the macro. 923141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner LexUnexpandedToken(Tok); 924141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 925141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 926141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 927141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 928141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Disable __VA_ARGS__ again. 929141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Ident__VA_ARGS__->setIsPoisoned(true); 930141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 931141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Check that there is no paste (##) operator at the begining or end of the 932141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // replacement list. 933141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner unsigned NumTokens = MI->getNumTokens(); 934141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (NumTokens != 0) { 935141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MI->getReplacementToken(0).is(tok::hashhash)) { 936141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MI->getReplacementToken(0), diag::err_paste_at_start); 937141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner delete MI; 938141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 939141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 940141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MI->getReplacementToken(NumTokens-1).is(tok::hashhash)) { 941141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MI->getReplacementToken(NumTokens-1), diag::err_paste_at_end); 942141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner delete MI; 943141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 944141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 945141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 946141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 947141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is the primary source file, remember that this macro hasn't been 948141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // used yet. 949141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (isInPrimaryFile()) 950141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setIsUsed(false); 951141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 952141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Finally, if this identifier already had a macro defined for it, verify that 953141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // the macro bodies are identical and free the old definition. 954141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MacroInfo *OtherMI = getMacroInfo(MacroNameTok.getIdentifierInfo())) { 955141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!OtherMI->isUsed()) 956141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(OtherMI->getDefinitionLoc(), diag::pp_macro_not_used); 957141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 958141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Macros must be identical. This means all tokes and whitespace separation 959141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // must be the same. C99 6.10.3.2. 960141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!MI->isIdenticalTo(*OtherMI, *this)) { 961141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MI->getDefinitionLoc(), diag::ext_pp_macro_redef, 962141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MacroNameTok.getIdentifierInfo()->getName()); 963141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(OtherMI->getDefinitionLoc(), diag::ext_pp_macro_redef2); 964141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 965141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner delete OtherMI; 966141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 967141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 968141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner setMacroInfo(MacroNameTok.getIdentifierInfo(), MI); 969141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 970141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 971141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleUndefDirective - Implements #undef. 972141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// 973141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleUndefDirective(Token &UndefTok) { 974141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumUndefined; 975141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 976141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token MacroNameTok; 977141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ReadMacroName(MacroNameTok, 2); 978141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 979141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Error reading macro name? If so, diagnostic already issued. 980141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MacroNameTok.is(tok::eom)) 981141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 982141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 983141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Check to see if this is the last token on the #undef line. 984141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CheckEndOfDirective("#undef"); 985141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 986141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Okay, we finally have a valid identifier to undef. 987141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MacroInfo *MI = getMacroInfo(MacroNameTok.getIdentifierInfo()); 988141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 989141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If the macro is not defined, this is a noop undef, just return. 990141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MI == 0) return; 991141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 992141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!MI->isUsed()) 993141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Diag(MI->getDefinitionLoc(), diag::pp_macro_not_used); 994141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 995141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Free macro definition. 996141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner delete MI; 997141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner setMacroInfo(MacroNameTok.getIdentifierInfo(), 0); 998141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 999141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1000141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1001141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 1002141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner// Preprocessor Conditional Directive Handling. 1003141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner//===----------------------------------------------------------------------===// 1004141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1005141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIfdefDirective - Implements the #ifdef/#ifndef directive. isIfndef is 1006141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// true when this is a #ifndef directive. ReadAnyTokensBeforeDirective is true 1007141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// if any tokens have been returned or pp-directives activated before this 1008141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// #ifndef has been lexed. 1009141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// 1010141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleIfdefDirective(Token &Result, bool isIfndef, 1011141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool ReadAnyTokensBeforeDirective) { 1012141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumIf; 1013141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token DirectiveTok = Result; 1014141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1015141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner Token MacroNameTok; 1016141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ReadMacroName(MacroNameTok); 1017141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1018141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Error reading macro name? If so, diagnostic already issued. 1019141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MacroNameTok.is(tok::eom)) { 1020141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Skip code until we get to #endif. This helps with recovery by not 1021141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // emitting an error when the #endif is reached. 1022141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner SkipExcludedConditionalBlock(DirectiveTok.getLocation(), 1023141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*Foundnonskip*/false, /*FoundElse*/false); 1024141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return; 1025141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 1026141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1027141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Check to see if this is the last token on the #if[n]def line. 1028141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CheckEndOfDirective(isIfndef ? "#ifndef" : "#ifdef"); 1029141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1030141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurLexer->getConditionalStackDepth() == 0) { 1031141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If the start of a top-level #ifdef, inform MIOpt. 1032141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!ReadAnyTokensBeforeDirective) { 1033141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner assert(isIfndef && "#ifdef shouldn't reach here"); 1034141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->MIOpt.EnterTopLevelIFNDEF(MacroNameTok.getIdentifierInfo()); 1035141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else 1036141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->MIOpt.EnterTopLevelConditional(); 1037141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 1038141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1039141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner IdentifierInfo *MII = MacroNameTok.getIdentifierInfo(); 1040141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MacroInfo *MI = getMacroInfo(MII); 1041141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1042141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If there is a macro, process it. 1043141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (MI) // Mark it used. 1044141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner MI->setIsUsed(true); 1045141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1046141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Should we include the stuff contained by this directive? 1047141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (!MI == isIfndef) { 1048141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Yes, remember that we are inside a conditional, then lex the next token. 1049141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->pushConditionalLevel(DirectiveTok.getLocation(), /*wasskip*/false, 1050141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*foundnonskip*/true, /*foundelse*/false); 1051141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 1052141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // No, skip the contents of this block and return the first token after it. 1053141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner SkipExcludedConditionalBlock(DirectiveTok.getLocation(), 1054141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*Foundnonskip*/false, 1055141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*FoundElse*/false); 1056141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 1057141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 1058141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1059141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleIfDirective - Implements the #if directive. 1060141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// 1061141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleIfDirective(Token &IfToken, 1062141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool ReadAnyTokensBeforeDirective) { 1063141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumIf; 1064141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1065141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Parse and evaluation the conditional expression. 1066141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner IdentifierInfo *IfNDefMacro = 0; 1067141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner bool ConditionalTrue = EvaluateDirectiveExpression(IfNDefMacro); 1068141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 10690049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes 10700049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes // If this condition is equivalent to #ifndef X, and if this is the first 10710049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes // directive seen, handle it for the multiple-include optimization. 10720049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes if (CurLexer->getConditionalStackDepth() == 0) { 10730049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes if (!ReadAnyTokensBeforeDirective && IfNDefMacro) 10740049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes CurLexer->MIOpt.EnterTopLevelIFNDEF(IfNDefMacro); 10750049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes else 10760049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes CurLexer->MIOpt.EnterTopLevelConditional(); 10770049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes } 10780049db6ce127cbf4671a6a81ad4dfe61fdc4aa5cNuno Lopes 1079141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Should we include the stuff contained by this directive? 1080141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (ConditionalTrue) { 1081141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Yes, remember that we are inside a conditional, then lex the next token. 1082141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->pushConditionalLevel(IfToken.getLocation(), /*wasskip*/false, 1083141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*foundnonskip*/true, /*foundelse*/false); 1084141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } else { 1085141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // No, skip the contents of this block and return the first token after it. 1086141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner SkipExcludedConditionalBlock(IfToken.getLocation(), /*Foundnonskip*/false, 1087141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*FoundElse*/false); 1088141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 1089141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 1090141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1091141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// HandleEndifDirective - Implements the #endif directive. 1092141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner/// 1093141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleEndifDirective(Token &EndifToken) { 1094141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumEndif; 1095141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1096141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Check that this is the whole directive. 1097141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CheckEndOfDirective("#endif"); 1098141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1099141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PPConditionalInfo CondInfo; 1100141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurLexer->popConditionalLevel(CondInfo)) { 1101141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // No conditionals on the stack: this is an #endif without an #if. 1102141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(EndifToken, diag::err_pp_endif_without_if); 1103141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner } 1104141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1105141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this the end of a top-level #endif, inform MIOpt. 1106141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurLexer->getConditionalStackDepth() == 0) 1107141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->MIOpt.ExitTopLevelConditional(); 1108141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1109141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner assert(!CondInfo.WasSkipping && !CurLexer->LexingRawMode && 1110141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner "This code should only be reachable in the non-skipping case!"); 1111141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 1112141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1113141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1114141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleElseDirective(Token &Result) { 1115141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumElse; 1116141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1117141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #else directive in a non-skipping conditional... start skipping. 1118141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CheckEndOfDirective("#else"); 1119141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1120141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PPConditionalInfo CI; 1121141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurLexer->popConditionalLevel(CI)) 1122141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(Result, diag::pp_err_else_without_if); 1123141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1124141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is a top-level #else, inform the MIOpt. 1125141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurLexer->getConditionalStackDepth() == 0) 1126141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->MIOpt.EnterTopLevelConditional(); 1127141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1128141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is a #else with a #else before it, report the error. 1129141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CI.FoundElse) Diag(Result, diag::pp_err_else_after_else); 1130141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1131141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Finally, skip the rest of the contents of this block and return the first 1132141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // token after it. 1133141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return SkipExcludedConditionalBlock(CI.IfLoc, /*Foundnonskip*/true, 1134141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*FoundElse*/true); 1135141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 1136141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1137141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattnervoid Preprocessor::HandleElifDirective(Token &ElifToken) { 1138141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner ++NumElse; 1139141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1140141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // #elif directive in a non-skipping conditional... start skipping. 1141141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // We don't care what the condition is, because we will always skip it (since 1142141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // the block immediately before it was included). 1143141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner DiscardUntilEndOfDirective(); 1144141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1145141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner PPConditionalInfo CI; 1146141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurLexer->popConditionalLevel(CI)) 1147141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return Diag(ElifToken, diag::pp_err_elif_without_if); 1148141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1149141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is a top-level #elif, inform the MIOpt. 1150141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CurLexer->getConditionalStackDepth() == 0) 1151141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner CurLexer->MIOpt.EnterTopLevelConditional(); 1152141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1153141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // If this is a #elif with a #else before it, report the error. 1154141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner if (CI.FoundElse) Diag(ElifToken, diag::pp_err_elif_after_else); 1155141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1156141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // Finally, skip the rest of the contents of this block and return the first 1157141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner // token after it. 1158141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner return SkipExcludedConditionalBlock(CI.IfLoc, /*Foundnonskip*/true, 1159141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner /*FoundElse*/CI.FoundElse); 1160141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner} 1161141e71fec621f65ab401dbe66609f7d3c033c6a5Chris Lattner 1162