blob: 0391c503d6b5b1c9ff3fb6030e6f8b3c62edb140 [file] [log] [blame]
Reid Spencer5f016e22007-07-11 17:01:13 +00001//===--- Preprocess.cpp - C Language Family Preprocessor Implementation ---===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file was developed by Chris Lattner and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the Preprocessor interface.
11//
12//===----------------------------------------------------------------------===//
13//
14// Options to support:
15// -H - Print the name of each header file used.
16// -d[MDNI] - Dump various things.
17// -fworking-directory - #line's with preprocessor's working dir.
18// -fpreprocessed
19// -dependency-file,-M,-MM,-MF,-MG,-MP,-MT,-MQ,-MD,-MMD
20// -W*
21// -w
22//
23// Messages to emit:
24// "Multiple include guards may be useful for:\n"
25//
26//===----------------------------------------------------------------------===//
27
28#include "clang/Lex/Preprocessor.h"
29#include "clang/Lex/HeaderSearch.h"
30#include "clang/Lex/MacroInfo.h"
31#include "clang/Lex/PPCallbacks.h"
32#include "clang/Lex/Pragma.h"
33#include "clang/Lex/ScratchBuffer.h"
34#include "clang/Basic/Diagnostic.h"
35#include "clang/Basic/FileManager.h"
36#include "clang/Basic/SourceManager.h"
37#include "clang/Basic/TargetInfo.h"
38#include "llvm/ADT/SmallVector.h"
Chris Lattner97ba77c2007-07-16 06:48:38 +000039#include "llvm/Support/MemoryBuffer.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000040#include <iostream>
Chris Lattner77034d32007-09-03 18:30:32 +000041#include <ctime>
Reid Spencer5f016e22007-07-11 17:01:13 +000042using namespace clang;
43
44//===----------------------------------------------------------------------===//
45
46Preprocessor::Preprocessor(Diagnostic &diags, const LangOptions &opts,
47 TargetInfo &target, SourceManager &SM,
48 HeaderSearch &Headers)
49 : Diags(diags), Features(opts), Target(target), FileMgr(Headers.getFileMgr()),
50 SourceMgr(SM), HeaderInfo(Headers), Identifiers(opts),
51 CurLexer(0), CurDirLookup(0), CurMacroExpander(0), Callbacks(0) {
52 ScratchBuf = new ScratchBuffer(SourceMgr);
Chris Lattner9594acf2007-07-15 00:25:26 +000053
Reid Spencer5f016e22007-07-11 17:01:13 +000054 // Clear stats.
55 NumDirectives = NumDefined = NumUndefined = NumPragma = 0;
56 NumIf = NumElse = NumEndif = 0;
57 NumEnteredSourceFiles = 0;
58 NumMacroExpanded = NumFnMacroExpanded = NumBuiltinMacroExpanded = 0;
59 NumFastMacroExpanded = NumTokenPaste = NumFastTokenPaste = 0;
60 MaxIncludeStackDepth = 0;
61 NumSkipped = 0;
62
63 // Default to discarding comments.
64 KeepComments = false;
65 KeepMacroComments = false;
66
67 // Macro expansion is enabled.
68 DisableMacroExpansion = false;
69 InMacroArgs = false;
Chris Lattner9594acf2007-07-15 00:25:26 +000070 NumCachedMacroExpanders = 0;
Reid Spencer5f016e22007-07-11 17:01:13 +000071
72 // "Poison" __VA_ARGS__, which can only appear in the expansion of a macro.
73 // This gets unpoisoned where it is allowed.
74 (Ident__VA_ARGS__ = getIdentifierInfo("__VA_ARGS__"))->setIsPoisoned();
75
76 // Initialize the pragma handlers.
77 PragmaHandlers = new PragmaNamespace(0);
78 RegisterBuiltinPragmas();
79
80 // Initialize builtin macros like __LINE__ and friends.
81 RegisterBuiltinMacros();
82}
83
84Preprocessor::~Preprocessor() {
85 // Free any active lexers.
86 delete CurLexer;
87
88 while (!IncludeMacroStack.empty()) {
89 delete IncludeMacroStack.back().TheLexer;
90 delete IncludeMacroStack.back().TheMacroExpander;
91 IncludeMacroStack.pop_back();
92 }
Chris Lattnercc1a8752007-10-07 08:44:20 +000093
94 // Free any macro definitions.
95 for (llvm::DenseMap<IdentifierInfo*, MacroInfo*>::iterator I =
96 Macros.begin(), E = Macros.end(); I != E; ++I) {
97 // Free the macro definition.
98 delete I->second;
99 I->second = 0;
100 I->first->setHasMacroDefinition(false);
101 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000102
Chris Lattner9594acf2007-07-15 00:25:26 +0000103 // Free any cached macro expanders.
104 for (unsigned i = 0, e = NumCachedMacroExpanders; i != e; ++i)
105 delete MacroExpanderCache[i];
106
Reid Spencer5f016e22007-07-11 17:01:13 +0000107 // Release pragma information.
108 delete PragmaHandlers;
109
110 // Delete the scratch buffer info.
111 delete ScratchBuf;
112}
113
114PPCallbacks::~PPCallbacks() {
115}
116
117/// Diag - Forwarding function for diagnostics. This emits a diagnostic at
Chris Lattnerd2177732007-07-20 16:59:19 +0000118/// the specified Token's location, translating the token's start
Reid Spencer5f016e22007-07-11 17:01:13 +0000119/// position in the current buffer into a SourcePosition object for rendering.
120void Preprocessor::Diag(SourceLocation Loc, unsigned DiagID) {
121 Diags.Report(Loc, DiagID);
122}
123
124void Preprocessor::Diag(SourceLocation Loc, unsigned DiagID,
125 const std::string &Msg) {
126 Diags.Report(Loc, DiagID, &Msg, 1);
127}
128
Chris Lattnerd2177732007-07-20 16:59:19 +0000129void Preprocessor::DumpToken(const Token &Tok, bool DumpFlags) const {
Reid Spencer5f016e22007-07-11 17:01:13 +0000130 std::cerr << tok::getTokenName(Tok.getKind()) << " '"
131 << getSpelling(Tok) << "'";
132
133 if (!DumpFlags) return;
134 std::cerr << "\t";
135 if (Tok.isAtStartOfLine())
136 std::cerr << " [StartOfLine]";
137 if (Tok.hasLeadingSpace())
138 std::cerr << " [LeadingSpace]";
139 if (Tok.isExpandDisabled())
140 std::cerr << " [ExpandDisabled]";
141 if (Tok.needsCleaning()) {
142 const char *Start = SourceMgr.getCharacterData(Tok.getLocation());
143 std::cerr << " [UnClean='" << std::string(Start, Start+Tok.getLength())
144 << "']";
145 }
146}
147
148void Preprocessor::DumpMacro(const MacroInfo &MI) const {
149 std::cerr << "MACRO: ";
150 for (unsigned i = 0, e = MI.getNumTokens(); i != e; ++i) {
151 DumpToken(MI.getReplacementToken(i));
152 std::cerr << " ";
153 }
154 std::cerr << "\n";
155}
156
157void Preprocessor::PrintStats() {
158 std::cerr << "\n*** Preprocessor Stats:\n";
159 std::cerr << NumDirectives << " directives found:\n";
160 std::cerr << " " << NumDefined << " #define.\n";
161 std::cerr << " " << NumUndefined << " #undef.\n";
162 std::cerr << " #include/#include_next/#import:\n";
163 std::cerr << " " << NumEnteredSourceFiles << " source files entered.\n";
164 std::cerr << " " << MaxIncludeStackDepth << " max include stack depth\n";
165 std::cerr << " " << NumIf << " #if/#ifndef/#ifdef.\n";
166 std::cerr << " " << NumElse << " #else/#elif.\n";
167 std::cerr << " " << NumEndif << " #endif.\n";
168 std::cerr << " " << NumPragma << " #pragma.\n";
169 std::cerr << NumSkipped << " #if/#ifndef#ifdef regions skipped\n";
170
171 std::cerr << NumMacroExpanded << "/" << NumFnMacroExpanded << "/"
172 << NumBuiltinMacroExpanded << " obj/fn/builtin macros expanded, "
173 << NumFastMacroExpanded << " on the fast path.\n";
174 std::cerr << (NumFastTokenPaste+NumTokenPaste)
175 << " token paste (##) operations performed, "
176 << NumFastTokenPaste << " on the fast path.\n";
177}
178
179//===----------------------------------------------------------------------===//
180// Token Spelling
181//===----------------------------------------------------------------------===//
182
183
184/// getSpelling() - Return the 'spelling' of this token. The spelling of a
185/// token are the characters used to represent the token in the source file
186/// after trigraph expansion and escaped-newline folding. In particular, this
187/// wants to get the true, uncanonicalized, spelling of things like digraphs
188/// UCNs, etc.
Chris Lattnerd2177732007-07-20 16:59:19 +0000189std::string Preprocessor::getSpelling(const Token &Tok) const {
Reid Spencer5f016e22007-07-11 17:01:13 +0000190 assert((int)Tok.getLength() >= 0 && "Token character range is bogus!");
191
192 // If this token contains nothing interesting, return it directly.
193 const char *TokStart = SourceMgr.getCharacterData(Tok.getLocation());
194 if (!Tok.needsCleaning())
195 return std::string(TokStart, TokStart+Tok.getLength());
196
197 std::string Result;
198 Result.reserve(Tok.getLength());
199
200 // Otherwise, hard case, relex the characters into the string.
201 for (const char *Ptr = TokStart, *End = TokStart+Tok.getLength();
202 Ptr != End; ) {
203 unsigned CharSize;
204 Result.push_back(Lexer::getCharAndSizeNoWarn(Ptr, CharSize, Features));
205 Ptr += CharSize;
206 }
207 assert(Result.size() != unsigned(Tok.getLength()) &&
208 "NeedsCleaning flag set on something that didn't need cleaning!");
209 return Result;
210}
211
212/// getSpelling - This method is used to get the spelling of a token into a
213/// preallocated buffer, instead of as an std::string. The caller is required
214/// to allocate enough space for the token, which is guaranteed to be at least
215/// Tok.getLength() bytes long. The actual length of the token is returned.
216///
217/// Note that this method may do two possible things: it may either fill in
218/// the buffer specified with characters, or it may *change the input pointer*
219/// to point to a constant buffer with the data already in it (avoiding a
220/// copy). The caller is not allowed to modify the returned buffer pointer
221/// if an internal buffer is returned.
Chris Lattnerd2177732007-07-20 16:59:19 +0000222unsigned Preprocessor::getSpelling(const Token &Tok,
Reid Spencer5f016e22007-07-11 17:01:13 +0000223 const char *&Buffer) const {
224 assert((int)Tok.getLength() >= 0 && "Token character range is bogus!");
225
226 // If this token is an identifier, just return the string from the identifier
227 // table, which is very quick.
228 if (const IdentifierInfo *II = Tok.getIdentifierInfo()) {
229 Buffer = II->getName();
Chris Lattner0f670322007-07-22 22:50:09 +0000230
231 // Return the length of the token. If the token needed cleaning, don't
232 // include the size of the newlines or trigraphs in it.
233 if (!Tok.needsCleaning())
234 return Tok.getLength();
235 else
236 return strlen(Buffer);
Reid Spencer5f016e22007-07-11 17:01:13 +0000237 }
238
239 // Otherwise, compute the start of the token in the input lexer buffer.
240 const char *TokStart = SourceMgr.getCharacterData(Tok.getLocation());
241
242 // If this token contains nothing interesting, return it directly.
243 if (!Tok.needsCleaning()) {
244 Buffer = TokStart;
245 return Tok.getLength();
246 }
247 // Otherwise, hard case, relex the characters into the string.
248 char *OutBuf = const_cast<char*>(Buffer);
249 for (const char *Ptr = TokStart, *End = TokStart+Tok.getLength();
250 Ptr != End; ) {
251 unsigned CharSize;
252 *OutBuf++ = Lexer::getCharAndSizeNoWarn(Ptr, CharSize, Features);
253 Ptr += CharSize;
254 }
255 assert(unsigned(OutBuf-Buffer) != Tok.getLength() &&
256 "NeedsCleaning flag set on something that didn't need cleaning!");
257
258 return OutBuf-Buffer;
259}
260
261
262/// CreateString - Plop the specified string into a scratch buffer and return a
263/// location for it. If specified, the source location provides a source
264/// location for the token.
265SourceLocation Preprocessor::
266CreateString(const char *Buf, unsigned Len, SourceLocation SLoc) {
267 if (SLoc.isValid())
268 return ScratchBuf->getToken(Buf, Len, SLoc);
269 return ScratchBuf->getToken(Buf, Len);
270}
271
272
Chris Lattner97ba77c2007-07-16 06:48:38 +0000273/// AdvanceToTokenCharacter - Given a location that specifies the start of a
274/// token, return a new location that specifies a character within the token.
275SourceLocation Preprocessor::AdvanceToTokenCharacter(SourceLocation TokStart,
276 unsigned CharNo) {
Chris Lattner9dc1f532007-07-20 16:37:10 +0000277 // If they request the first char of the token, we're trivially done. If this
278 // is a macro expansion, it doesn't make sense to point to a character within
279 // the instantiation point (the name). We could point to the source
280 // character, but without also pointing to instantiation info, this is
281 // confusing.
282 if (CharNo == 0 || TokStart.isMacroID()) return TokStart;
Chris Lattner97ba77c2007-07-16 06:48:38 +0000283
284 // Figure out how many physical characters away the specified logical
285 // character is. This needs to take into consideration newlines and
286 // trigraphs.
Chris Lattner9dc1f532007-07-20 16:37:10 +0000287 const char *TokPtr = SourceMgr.getCharacterData(TokStart);
288 unsigned PhysOffset = 0;
Chris Lattner97ba77c2007-07-16 06:48:38 +0000289
290 // The usual case is that tokens don't contain anything interesting. Skip
291 // over the uninteresting characters. If a token only consists of simple
292 // chars, this method is extremely fast.
293 while (CharNo && Lexer::isObviouslySimpleCharacter(*TokPtr))
Chris Lattner9dc1f532007-07-20 16:37:10 +0000294 ++TokPtr, --CharNo, ++PhysOffset;
Chris Lattner97ba77c2007-07-16 06:48:38 +0000295
296 // If we have a character that may be a trigraph or escaped newline, create a
297 // lexer to parse it correctly.
Chris Lattner97ba77c2007-07-16 06:48:38 +0000298 if (CharNo != 0) {
299 // Create a lexer starting at this token position.
Chris Lattner25bdb512007-07-20 16:52:03 +0000300 Lexer TheLexer(TokStart, *this, TokPtr);
Chris Lattnerd2177732007-07-20 16:59:19 +0000301 Token Tok;
Chris Lattner97ba77c2007-07-16 06:48:38 +0000302 // Skip over characters the remaining characters.
Chris Lattner9dc1f532007-07-20 16:37:10 +0000303 const char *TokStartPtr = TokPtr;
Chris Lattner97ba77c2007-07-16 06:48:38 +0000304 for (; CharNo; --CharNo)
305 TheLexer.getAndAdvanceChar(TokPtr, Tok);
Chris Lattner9dc1f532007-07-20 16:37:10 +0000306
307 PhysOffset += TokPtr-TokStartPtr;
Chris Lattner97ba77c2007-07-16 06:48:38 +0000308 }
Chris Lattner9dc1f532007-07-20 16:37:10 +0000309
310 return TokStart.getFileLocWithOffset(PhysOffset);
Chris Lattner97ba77c2007-07-16 06:48:38 +0000311}
312
313
314
Reid Spencer5f016e22007-07-11 17:01:13 +0000315//===----------------------------------------------------------------------===//
316// Source File Location Methods.
317//===----------------------------------------------------------------------===//
318
319/// LookupFile - Given a "foo" or <foo> reference, look up the indicated file,
320/// return null on failure. isAngled indicates whether the file reference is
321/// for system #include's or not (i.e. using <> instead of "").
322const FileEntry *Preprocessor::LookupFile(const char *FilenameStart,
323 const char *FilenameEnd,
324 bool isAngled,
325 const DirectoryLookup *FromDir,
326 const DirectoryLookup *&CurDir) {
327 // If the header lookup mechanism may be relative to the current file, pass in
328 // info about where the current file is.
329 const FileEntry *CurFileEnt = 0;
330 if (!FromDir) {
Chris Lattner9dc1f532007-07-20 16:37:10 +0000331 SourceLocation FileLoc = getCurrentFileLexer()->getFileLoc();
332 CurFileEnt = SourceMgr.getFileEntryForLoc(FileLoc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000333 }
334
335 // Do a standard file entry lookup.
336 CurDir = CurDirLookup;
337 const FileEntry *FE =
338 HeaderInfo.LookupFile(FilenameStart, FilenameEnd,
339 isAngled, FromDir, CurDir, CurFileEnt);
340 if (FE) return FE;
341
342 // Otherwise, see if this is a subframework header. If so, this is relative
343 // to one of the headers on the #include stack. Walk the list of the current
344 // headers on the #include stack and pass them to HeaderInfo.
345 if (CurLexer && !CurLexer->Is_PragmaLexer) {
Chris Lattner9dc1f532007-07-20 16:37:10 +0000346 CurFileEnt = SourceMgr.getFileEntryForLoc(CurLexer->getFileLoc());
Reid Spencer5f016e22007-07-11 17:01:13 +0000347 if ((FE = HeaderInfo.LookupSubframeworkHeader(FilenameStart, FilenameEnd,
348 CurFileEnt)))
349 return FE;
350 }
351
352 for (unsigned i = 0, e = IncludeMacroStack.size(); i != e; ++i) {
353 IncludeStackInfo &ISEntry = IncludeMacroStack[e-i-1];
354 if (ISEntry.TheLexer && !ISEntry.TheLexer->Is_PragmaLexer) {
Chris Lattner9dc1f532007-07-20 16:37:10 +0000355 CurFileEnt = SourceMgr.getFileEntryForLoc(ISEntry.TheLexer->getFileLoc());
Reid Spencer5f016e22007-07-11 17:01:13 +0000356 if ((FE = HeaderInfo.LookupSubframeworkHeader(FilenameStart, FilenameEnd,
357 CurFileEnt)))
358 return FE;
359 }
360 }
361
362 // Otherwise, we really couldn't find the file.
363 return 0;
364}
365
366/// isInPrimaryFile - Return true if we're in the top-level file, not in a
367/// #include.
368bool Preprocessor::isInPrimaryFile() const {
369 if (CurLexer && !CurLexer->Is_PragmaLexer)
370 return CurLexer->isMainFile();
371
372 // If there are any stacked lexers, we're in a #include.
373 for (unsigned i = 0, e = IncludeMacroStack.size(); i != e; ++i)
374 if (IncludeMacroStack[i].TheLexer &&
375 !IncludeMacroStack[i].TheLexer->Is_PragmaLexer)
376 return IncludeMacroStack[i].TheLexer->isMainFile();
377 return false;
378}
379
380/// getCurrentLexer - Return the current file lexer being lexed from. Note
381/// that this ignores any potentially active macro expansions and _Pragma
382/// expansions going on at the time.
383Lexer *Preprocessor::getCurrentFileLexer() const {
384 if (CurLexer && !CurLexer->Is_PragmaLexer) return CurLexer;
385
386 // Look for a stacked lexer.
387 for (unsigned i = IncludeMacroStack.size(); i != 0; --i) {
388 Lexer *L = IncludeMacroStack[i-1].TheLexer;
389 if (L && !L->Is_PragmaLexer) // Ignore macro & _Pragma expansions.
390 return L;
391 }
392 return 0;
393}
394
395
396/// EnterSourceFile - Add a source file to the top of the include stack and
397/// start lexing tokens from it instead of the current buffer. Return true
398/// on failure.
399void Preprocessor::EnterSourceFile(unsigned FileID,
400 const DirectoryLookup *CurDir,
401 bool isMainFile) {
402 assert(CurMacroExpander == 0 && "Cannot #include a file inside a macro!");
403 ++NumEnteredSourceFiles;
404
405 if (MaxIncludeStackDepth < IncludeMacroStack.size())
406 MaxIncludeStackDepth = IncludeMacroStack.size();
407
Chris Lattner25bdb512007-07-20 16:52:03 +0000408 Lexer *TheLexer = new Lexer(SourceLocation::getFileLoc(FileID, 0), *this);
Reid Spencer5f016e22007-07-11 17:01:13 +0000409 if (isMainFile) TheLexer->setIsMainFile();
410 EnterSourceFileWithLexer(TheLexer, CurDir);
411}
412
413/// EnterSourceFile - Add a source file to the top of the include stack and
414/// start lexing tokens from it instead of the current buffer.
415void Preprocessor::EnterSourceFileWithLexer(Lexer *TheLexer,
416 const DirectoryLookup *CurDir) {
417
418 // Add the current lexer to the include stack.
419 if (CurLexer || CurMacroExpander)
420 IncludeMacroStack.push_back(IncludeStackInfo(CurLexer, CurDirLookup,
421 CurMacroExpander));
422
423 CurLexer = TheLexer;
424 CurDirLookup = CurDir;
425 CurMacroExpander = 0;
426
427 // Notify the client, if desired, that we are in a new source file.
428 if (Callbacks && !CurLexer->Is_PragmaLexer) {
429 DirectoryLookup::DirType FileType = DirectoryLookup::NormalHeaderDir;
430
431 // Get the file entry for the current file.
432 if (const FileEntry *FE =
Chris Lattner9dc1f532007-07-20 16:37:10 +0000433 SourceMgr.getFileEntryForLoc(CurLexer->getFileLoc()))
Reid Spencer5f016e22007-07-11 17:01:13 +0000434 FileType = HeaderInfo.getFileDirFlavor(FE);
435
Chris Lattner9dc1f532007-07-20 16:37:10 +0000436 Callbacks->FileChanged(CurLexer->getFileLoc(),
Reid Spencer5f016e22007-07-11 17:01:13 +0000437 PPCallbacks::EnterFile, FileType);
438 }
439}
440
441
442
443/// EnterMacro - Add a Macro to the top of the include stack and start lexing
444/// tokens from it instead of the current buffer.
Chris Lattnerd2177732007-07-20 16:59:19 +0000445void Preprocessor::EnterMacro(Token &Tok, MacroArgs *Args) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000446 IncludeMacroStack.push_back(IncludeStackInfo(CurLexer, CurDirLookup,
447 CurMacroExpander));
448 CurLexer = 0;
449 CurDirLookup = 0;
450
Chris Lattner9594acf2007-07-15 00:25:26 +0000451 if (NumCachedMacroExpanders == 0) {
452 CurMacroExpander = new MacroExpander(Tok, Args, *this);
453 } else {
454 CurMacroExpander = MacroExpanderCache[--NumCachedMacroExpanders];
455 CurMacroExpander->Init(Tok, Args);
456 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000457}
458
459/// EnterTokenStream - Add a "macro" context to the top of the include stack,
460/// which will cause the lexer to start returning the specified tokens. Note
461/// that these tokens will be re-macro-expanded when/if expansion is enabled.
462/// This method assumes that the specified stream of tokens has a permanent
463/// owner somewhere, so they do not need to be copied.
Chris Lattnerd2177732007-07-20 16:59:19 +0000464void Preprocessor::EnterTokenStream(const Token *Toks, unsigned NumToks) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000465 // Save our current state.
466 IncludeMacroStack.push_back(IncludeStackInfo(CurLexer, CurDirLookup,
467 CurMacroExpander));
468 CurLexer = 0;
469 CurDirLookup = 0;
470
471 // Create a macro expander to expand from the specified token stream.
Chris Lattner9594acf2007-07-15 00:25:26 +0000472 if (NumCachedMacroExpanders == 0) {
473 CurMacroExpander = new MacroExpander(Toks, NumToks, *this);
474 } else {
475 CurMacroExpander = MacroExpanderCache[--NumCachedMacroExpanders];
476 CurMacroExpander->Init(Toks, NumToks);
477 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000478}
479
480/// RemoveTopOfLexerStack - Pop the current lexer/macro exp off the top of the
481/// lexer stack. This should only be used in situations where the current
482/// state of the top-of-stack lexer is known.
483void Preprocessor::RemoveTopOfLexerStack() {
484 assert(!IncludeMacroStack.empty() && "Ran out of stack entries to load");
Chris Lattner9594acf2007-07-15 00:25:26 +0000485
486 if (CurMacroExpander) {
487 // Delete or cache the now-dead macro expander.
488 if (NumCachedMacroExpanders == MacroExpanderCacheSize)
489 delete CurMacroExpander;
490 else
491 MacroExpanderCache[NumCachedMacroExpanders++] = CurMacroExpander;
492 } else {
493 delete CurLexer;
494 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000495 CurLexer = IncludeMacroStack.back().TheLexer;
496 CurDirLookup = IncludeMacroStack.back().TheDirLookup;
497 CurMacroExpander = IncludeMacroStack.back().TheMacroExpander;
498 IncludeMacroStack.pop_back();
499}
500
501//===----------------------------------------------------------------------===//
502// Macro Expansion Handling.
503//===----------------------------------------------------------------------===//
504
Chris Lattnercc1a8752007-10-07 08:44:20 +0000505/// setMacroInfo - Specify a macro for this identifier.
506///
507void Preprocessor::setMacroInfo(IdentifierInfo *II, MacroInfo *MI) {
508 if (MI == 0) {
509 if (II->hasMacroDefinition()) {
510 Macros.erase(II);
511 II->setHasMacroDefinition(false);
512 }
513 } else {
514 Macros[II] = MI;
515 II->setHasMacroDefinition(true);
516 }
517}
518
Reid Spencer5f016e22007-07-11 17:01:13 +0000519/// RegisterBuiltinMacro - Register the specified identifier in the identifier
520/// table and mark it as a builtin macro to be expanded.
521IdentifierInfo *Preprocessor::RegisterBuiltinMacro(const char *Name) {
522 // Get the identifier.
523 IdentifierInfo *Id = getIdentifierInfo(Name);
524
525 // Mark it as being a macro that is builtin.
526 MacroInfo *MI = new MacroInfo(SourceLocation());
527 MI->setIsBuiltinMacro();
Chris Lattnercc1a8752007-10-07 08:44:20 +0000528 setMacroInfo(Id, MI);
Reid Spencer5f016e22007-07-11 17:01:13 +0000529 return Id;
530}
531
532
533/// RegisterBuiltinMacros - Register builtin macros, such as __LINE__ with the
534/// identifier table.
535void Preprocessor::RegisterBuiltinMacros() {
536 Ident__LINE__ = RegisterBuiltinMacro("__LINE__");
537 Ident__FILE__ = RegisterBuiltinMacro("__FILE__");
538 Ident__DATE__ = RegisterBuiltinMacro("__DATE__");
539 Ident__TIME__ = RegisterBuiltinMacro("__TIME__");
540 Ident_Pragma = RegisterBuiltinMacro("_Pragma");
541
542 // GCC Extensions.
543 Ident__BASE_FILE__ = RegisterBuiltinMacro("__BASE_FILE__");
544 Ident__INCLUDE_LEVEL__ = RegisterBuiltinMacro("__INCLUDE_LEVEL__");
545 Ident__TIMESTAMP__ = RegisterBuiltinMacro("__TIMESTAMP__");
546}
547
548/// isTrivialSingleTokenExpansion - Return true if MI, which has a single token
549/// in its expansion, currently expands to that token literally.
550static bool isTrivialSingleTokenExpansion(const MacroInfo *MI,
Chris Lattnercc1a8752007-10-07 08:44:20 +0000551 const IdentifierInfo *MacroIdent,
552 Preprocessor &PP) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000553 IdentifierInfo *II = MI->getReplacementToken(0).getIdentifierInfo();
554
555 // If the token isn't an identifier, it's always literally expanded.
556 if (II == 0) return true;
557
558 // If the identifier is a macro, and if that macro is enabled, it may be
559 // expanded so it's not a trivial expansion.
Chris Lattnercc1a8752007-10-07 08:44:20 +0000560 if (II->hasMacroDefinition() && PP.getMacroInfo(II)->isEnabled() &&
Reid Spencer5f016e22007-07-11 17:01:13 +0000561 // Fast expanding "#define X X" is ok, because X would be disabled.
562 II != MacroIdent)
563 return false;
564
565 // If this is an object-like macro invocation, it is safe to trivially expand
566 // it.
567 if (MI->isObjectLike()) return true;
568
569 // If this is a function-like macro invocation, it's safe to trivially expand
570 // as long as the identifier is not a macro argument.
571 for (MacroInfo::arg_iterator I = MI->arg_begin(), E = MI->arg_end();
572 I != E; ++I)
573 if (*I == II)
574 return false; // Identifier is a macro argument.
575
576 return true;
577}
578
579
580/// isNextPPTokenLParen - Determine whether the next preprocessor token to be
581/// lexed is a '('. If so, consume the token and return true, if not, this
582/// method should have no observable side-effect on the lexed tokens.
583bool Preprocessor::isNextPPTokenLParen() {
584 // Do some quick tests for rejection cases.
585 unsigned Val;
586 if (CurLexer)
587 Val = CurLexer->isNextPPTokenLParen();
588 else
589 Val = CurMacroExpander->isNextTokenLParen();
590
591 if (Val == 2) {
Chris Lattner0ea793e2007-07-19 00:07:36 +0000592 // We have run off the end. If it's a source file we don't
593 // examine enclosing ones (C99 5.1.1.2p4). Otherwise walk up the
594 // macro stack.
595 if (CurLexer)
596 return false;
597 for (unsigned i = IncludeMacroStack.size(); i != 0; --i) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000598 IncludeStackInfo &Entry = IncludeMacroStack[i-1];
599 if (Entry.TheLexer)
600 Val = Entry.TheLexer->isNextPPTokenLParen();
601 else
602 Val = Entry.TheMacroExpander->isNextTokenLParen();
Chris Lattner0ea793e2007-07-19 00:07:36 +0000603
604 if (Val != 2)
605 break;
606
607 // Ran off the end of a source file?
608 if (Entry.TheLexer)
609 return false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000610 }
611 }
612
613 // Okay, if we know that the token is a '(', lex it and return. Otherwise we
614 // have found something that isn't a '(' or we found the end of the
615 // translation unit. In either case, return false.
616 if (Val != 1)
617 return false;
618
Chris Lattnerd2177732007-07-20 16:59:19 +0000619 Token Tok;
Reid Spencer5f016e22007-07-11 17:01:13 +0000620 LexUnexpandedToken(Tok);
621 assert(Tok.getKind() == tok::l_paren && "Error computing l-paren-ness?");
622 return true;
623}
624
625/// HandleMacroExpandedIdentifier - If an identifier token is read that is to be
626/// expanded as a macro, handle it and return the next token as 'Identifier'.
Chris Lattnerd2177732007-07-20 16:59:19 +0000627bool Preprocessor::HandleMacroExpandedIdentifier(Token &Identifier,
Reid Spencer5f016e22007-07-11 17:01:13 +0000628 MacroInfo *MI) {
629
630 // If this is a builtin macro, like __LINE__ or _Pragma, handle it specially.
631 if (MI->isBuiltinMacro()) {
632 ExpandBuiltinMacro(Identifier);
633 return false;
634 }
635
636 // If this is the first use of a target-specific macro, warn about it.
637 if (MI->isTargetSpecific()) {
638 MI->setIsTargetSpecific(false); // Don't warn on second use.
639 getTargetInfo().DiagnoseNonPortability(Identifier.getLocation(),
640 diag::port_target_macro_use);
641 }
642
643 /// Args - If this is a function-like macro expansion, this contains,
644 /// for each macro argument, the list of tokens that were provided to the
645 /// invocation.
646 MacroArgs *Args = 0;
647
648 // If this is a function-like macro, read the arguments.
649 if (MI->isFunctionLike()) {
650 // C99 6.10.3p10: If the preprocessing token immediately after the the macro
Chris Lattner2b64fdc2007-07-19 16:11:58 +0000651 // name isn't a '(', this macro should not be expanded. Otherwise, consume
652 // it.
Reid Spencer5f016e22007-07-11 17:01:13 +0000653 if (!isNextPPTokenLParen())
654 return true;
655
656 // Remember that we are now parsing the arguments to a macro invocation.
657 // Preprocessor directives used inside macro arguments are not portable, and
658 // this enables the warning.
659 InMacroArgs = true;
660 Args = ReadFunctionLikeMacroArgs(Identifier, MI);
661
662 // Finished parsing args.
663 InMacroArgs = false;
664
665 // If there was an error parsing the arguments, bail out.
666 if (Args == 0) return false;
667
668 ++NumFnMacroExpanded;
669 } else {
670 ++NumMacroExpanded;
671 }
672
673 // Notice that this macro has been used.
674 MI->setIsUsed(true);
675
676 // If we started lexing a macro, enter the macro expansion body.
677
678 // If this macro expands to no tokens, don't bother to push it onto the
679 // expansion stack, only to take it right back off.
680 if (MI->getNumTokens() == 0) {
681 // No need for arg info.
682 if (Args) Args->destroy();
683
684 // Ignore this macro use, just return the next token in the current
685 // buffer.
686 bool HadLeadingSpace = Identifier.hasLeadingSpace();
687 bool IsAtStartOfLine = Identifier.isAtStartOfLine();
688
689 Lex(Identifier);
690
691 // If the identifier isn't on some OTHER line, inherit the leading
692 // whitespace/first-on-a-line property of this token. This handles
693 // stuff like "! XX," -> "! ," and " XX," -> " ,", when XX is
694 // empty.
695 if (!Identifier.isAtStartOfLine()) {
Chris Lattnerd2177732007-07-20 16:59:19 +0000696 if (IsAtStartOfLine) Identifier.setFlag(Token::StartOfLine);
697 if (HadLeadingSpace) Identifier.setFlag(Token::LeadingSpace);
Reid Spencer5f016e22007-07-11 17:01:13 +0000698 }
699 ++NumFastMacroExpanded;
700 return false;
701
702 } else if (MI->getNumTokens() == 1 &&
Chris Lattnercc1a8752007-10-07 08:44:20 +0000703 isTrivialSingleTokenExpansion(MI, Identifier.getIdentifierInfo(),
704 *this)){
Reid Spencer5f016e22007-07-11 17:01:13 +0000705 // Otherwise, if this macro expands into a single trivially-expanded
706 // token: expand it now. This handles common cases like
707 // "#define VAL 42".
708
709 // Propagate the isAtStartOfLine/hasLeadingSpace markers of the macro
710 // identifier to the expanded token.
711 bool isAtStartOfLine = Identifier.isAtStartOfLine();
712 bool hasLeadingSpace = Identifier.hasLeadingSpace();
713
714 // Remember where the token is instantiated.
715 SourceLocation InstantiateLoc = Identifier.getLocation();
716
717 // Replace the result token.
718 Identifier = MI->getReplacementToken(0);
719
720 // Restore the StartOfLine/LeadingSpace markers.
Chris Lattnerd2177732007-07-20 16:59:19 +0000721 Identifier.setFlagValue(Token::StartOfLine , isAtStartOfLine);
722 Identifier.setFlagValue(Token::LeadingSpace, hasLeadingSpace);
Reid Spencer5f016e22007-07-11 17:01:13 +0000723
724 // Update the tokens location to include both its logical and physical
725 // locations.
726 SourceLocation Loc =
727 SourceMgr.getInstantiationLoc(Identifier.getLocation(), InstantiateLoc);
728 Identifier.setLocation(Loc);
729
730 // If this is #define X X, we must mark the result as unexpandible.
731 if (IdentifierInfo *NewII = Identifier.getIdentifierInfo())
Chris Lattnercc1a8752007-10-07 08:44:20 +0000732 if (getMacroInfo(NewII) == MI)
Chris Lattnerd2177732007-07-20 16:59:19 +0000733 Identifier.setFlag(Token::DisableExpand);
Reid Spencer5f016e22007-07-11 17:01:13 +0000734
735 // Since this is not an identifier token, it can't be macro expanded, so
736 // we're done.
737 ++NumFastMacroExpanded;
738 return false;
739 }
740
741 // Start expanding the macro.
742 EnterMacro(Identifier, Args);
743
744 // Now that the macro is at the top of the include stack, ask the
745 // preprocessor to read the next token from it.
746 Lex(Identifier);
747 return false;
748}
749
750/// ReadFunctionLikeMacroArgs - After reading "MACRO(", this method is
751/// invoked to read all of the actual arguments specified for the macro
752/// invocation. This returns null on error.
Chris Lattnerd2177732007-07-20 16:59:19 +0000753MacroArgs *Preprocessor::ReadFunctionLikeMacroArgs(Token &MacroName,
Reid Spencer5f016e22007-07-11 17:01:13 +0000754 MacroInfo *MI) {
755 // The number of fixed arguments to parse.
756 unsigned NumFixedArgsLeft = MI->getNumArgs();
757 bool isVariadic = MI->isVariadic();
758
759 // Outer loop, while there are more arguments, keep reading them.
Chris Lattnerd2177732007-07-20 16:59:19 +0000760 Token Tok;
Reid Spencer5f016e22007-07-11 17:01:13 +0000761 Tok.setKind(tok::comma);
762 --NumFixedArgsLeft; // Start reading the first arg.
763
764 // ArgTokens - Build up a list of tokens that make up each argument. Each
765 // argument is separated by an EOF token. Use a SmallVector so we can avoid
766 // heap allocations in the common case.
Chris Lattnerd2177732007-07-20 16:59:19 +0000767 llvm::SmallVector<Token, 64> ArgTokens;
Reid Spencer5f016e22007-07-11 17:01:13 +0000768
769 unsigned NumActuals = 0;
770 while (Tok.getKind() == tok::comma) {
Chris Lattner2b64fdc2007-07-19 16:11:58 +0000771 // C99 6.10.3p11: Keep track of the number of l_parens we have seen. Note
772 // that we already consumed the first one.
Reid Spencer5f016e22007-07-11 17:01:13 +0000773 unsigned NumParens = 0;
774
775 while (1) {
776 // Read arguments as unexpanded tokens. This avoids issues, e.g., where
777 // an argument value in a macro could expand to ',' or '(' or ')'.
778 LexUnexpandedToken(Tok);
779
780 if (Tok.getKind() == tok::eof) {
781 Diag(MacroName, diag::err_unterm_macro_invoc);
782 // Do not lose the EOF. Return it to the client.
783 MacroName = Tok;
784 return 0;
785 } else if (Tok.getKind() == tok::r_paren) {
786 // If we found the ) token, the macro arg list is done.
787 if (NumParens-- == 0)
788 break;
789 } else if (Tok.getKind() == tok::l_paren) {
790 ++NumParens;
791 } else if (Tok.getKind() == tok::comma && NumParens == 0) {
792 // Comma ends this argument if there are more fixed arguments expected.
793 if (NumFixedArgsLeft)
794 break;
795
796 // If this is not a variadic macro, too many args were specified.
797 if (!isVariadic) {
798 // Emit the diagnostic at the macro name in case there is a missing ).
799 // Emitting it at the , could be far away from the macro name.
800 Diag(MacroName, diag::err_too_many_args_in_macro_invoc);
801 return 0;
802 }
803 // Otherwise, continue to add the tokens to this variable argument.
804 } else if (Tok.getKind() == tok::comment && !KeepMacroComments) {
805 // If this is a comment token in the argument list and we're just in
806 // -C mode (not -CC mode), discard the comment.
807 continue;
808 }
809
810 ArgTokens.push_back(Tok);
811 }
812
813 // Empty arguments are standard in C99 and supported as an extension in
814 // other modes.
815 if (ArgTokens.empty() && !Features.C99)
816 Diag(Tok, diag::ext_empty_fnmacro_arg);
817
818 // Add a marker EOF token to the end of the token list for this argument.
Chris Lattnerd2177732007-07-20 16:59:19 +0000819 Token EOFTok;
Reid Spencer5f016e22007-07-11 17:01:13 +0000820 EOFTok.startToken();
821 EOFTok.setKind(tok::eof);
822 EOFTok.setLocation(Tok.getLocation());
823 EOFTok.setLength(0);
824 ArgTokens.push_back(EOFTok);
825 ++NumActuals;
826 --NumFixedArgsLeft;
827 };
828
829 // Okay, we either found the r_paren. Check to see if we parsed too few
830 // arguments.
831 unsigned MinArgsExpected = MI->getNumArgs();
832
833 // See MacroArgs instance var for description of this.
834 bool isVarargsElided = false;
835
836 if (NumActuals < MinArgsExpected) {
837 // There are several cases where too few arguments is ok, handle them now.
838 if (NumActuals+1 == MinArgsExpected && MI->isVariadic()) {
839 // Varargs where the named vararg parameter is missing: ok as extension.
840 // #define A(x, ...)
841 // A("blah")
842 Diag(Tok, diag::ext_missing_varargs_arg);
843
844 // Remember this occurred if this is a C99 macro invocation with at least
845 // one actual argument.
846 isVarargsElided = MI->isC99Varargs() && MI->getNumArgs() > 1;
847 } else if (MI->getNumArgs() == 1) {
848 // #define A(x)
849 // A()
850 // is ok because it is an empty argument.
851
852 // Empty arguments are standard in C99 and supported as an extension in
853 // other modes.
854 if (ArgTokens.empty() && !Features.C99)
855 Diag(Tok, diag::ext_empty_fnmacro_arg);
856 } else {
857 // Otherwise, emit the error.
858 Diag(Tok, diag::err_too_few_args_in_macro_invoc);
859 return 0;
860 }
861
862 // Add a marker EOF token to the end of the token list for this argument.
863 SourceLocation EndLoc = Tok.getLocation();
864 Tok.startToken();
865 Tok.setKind(tok::eof);
866 Tok.setLocation(EndLoc);
867 Tok.setLength(0);
868 ArgTokens.push_back(Tok);
869 }
870
871 return MacroArgs::create(MI, &ArgTokens[0], ArgTokens.size(),isVarargsElided);
872}
873
874/// ComputeDATE_TIME - Compute the current time, enter it into the specified
875/// scratch buffer, then return DATELoc/TIMELoc locations with the position of
876/// the identifier tokens inserted.
877static void ComputeDATE_TIME(SourceLocation &DATELoc, SourceLocation &TIMELoc,
878 Preprocessor &PP) {
879 time_t TT = time(0);
880 struct tm *TM = localtime(&TT);
881
882 static const char * const Months[] = {
883 "Jan","Feb","Mar","Apr","May","Jun","Jul","Aug","Sep","Oct","Nov","Dec"
884 };
885
886 char TmpBuffer[100];
887 sprintf(TmpBuffer, "\"%s %2d %4d\"", Months[TM->tm_mon], TM->tm_mday,
888 TM->tm_year+1900);
889 DATELoc = PP.CreateString(TmpBuffer, strlen(TmpBuffer));
890
891 sprintf(TmpBuffer, "\"%02d:%02d:%02d\"", TM->tm_hour, TM->tm_min, TM->tm_sec);
892 TIMELoc = PP.CreateString(TmpBuffer, strlen(TmpBuffer));
893}
894
895/// ExpandBuiltinMacro - If an identifier token is read that is to be expanded
896/// as a builtin macro, handle it and return the next token as 'Tok'.
Chris Lattnerd2177732007-07-20 16:59:19 +0000897void Preprocessor::ExpandBuiltinMacro(Token &Tok) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000898 // Figure out which token this is.
899 IdentifierInfo *II = Tok.getIdentifierInfo();
900 assert(II && "Can't be a macro without id info!");
901
902 // If this is an _Pragma directive, expand it, invoke the pragma handler, then
903 // lex the token after it.
904 if (II == Ident_Pragma)
905 return Handle_Pragma(Tok);
906
907 ++NumBuiltinMacroExpanded;
908
909 char TmpBuffer[100];
910
911 // Set up the return result.
912 Tok.setIdentifierInfo(0);
Chris Lattnerd2177732007-07-20 16:59:19 +0000913 Tok.clearFlag(Token::NeedsCleaning);
Reid Spencer5f016e22007-07-11 17:01:13 +0000914
915 if (II == Ident__LINE__) {
916 // __LINE__ expands to a simple numeric value.
Chris Lattner9dc1f532007-07-20 16:37:10 +0000917 sprintf(TmpBuffer, "%u", SourceMgr.getLogicalLineNumber(Tok.getLocation()));
Reid Spencer5f016e22007-07-11 17:01:13 +0000918 unsigned Length = strlen(TmpBuffer);
919 Tok.setKind(tok::numeric_constant);
920 Tok.setLength(Length);
921 Tok.setLocation(CreateString(TmpBuffer, Length, Tok.getLocation()));
922 } else if (II == Ident__FILE__ || II == Ident__BASE_FILE__) {
923 SourceLocation Loc = Tok.getLocation();
924 if (II == Ident__BASE_FILE__) {
925 Diag(Tok, diag::ext_pp_base_file);
Chris Lattner9dc1f532007-07-20 16:37:10 +0000926 SourceLocation NextLoc = SourceMgr.getIncludeLoc(Loc);
927 while (NextLoc.isValid()) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000928 Loc = NextLoc;
Chris Lattner9dc1f532007-07-20 16:37:10 +0000929 NextLoc = SourceMgr.getIncludeLoc(Loc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000930 }
931 }
932
933 // Escape this filename. Turn '\' -> '\\' '"' -> '\"'
Chris Lattner9dc1f532007-07-20 16:37:10 +0000934 std::string FN = SourceMgr.getSourceName(SourceMgr.getLogicalLoc(Loc));
Reid Spencer5f016e22007-07-11 17:01:13 +0000935 FN = '"' + Lexer::Stringify(FN) + '"';
936 Tok.setKind(tok::string_literal);
937 Tok.setLength(FN.size());
938 Tok.setLocation(CreateString(&FN[0], FN.size(), Tok.getLocation()));
939 } else if (II == Ident__DATE__) {
940 if (!DATELoc.isValid())
941 ComputeDATE_TIME(DATELoc, TIMELoc, *this);
942 Tok.setKind(tok::string_literal);
943 Tok.setLength(strlen("\"Mmm dd yyyy\""));
944 Tok.setLocation(SourceMgr.getInstantiationLoc(DATELoc, Tok.getLocation()));
945 } else if (II == Ident__TIME__) {
946 if (!TIMELoc.isValid())
947 ComputeDATE_TIME(DATELoc, TIMELoc, *this);
948 Tok.setKind(tok::string_literal);
949 Tok.setLength(strlen("\"hh:mm:ss\""));
950 Tok.setLocation(SourceMgr.getInstantiationLoc(TIMELoc, Tok.getLocation()));
951 } else if (II == Ident__INCLUDE_LEVEL__) {
952 Diag(Tok, diag::ext_pp_include_level);
953
954 // Compute the include depth of this token.
955 unsigned Depth = 0;
Chris Lattner9dc1f532007-07-20 16:37:10 +0000956 SourceLocation Loc = SourceMgr.getIncludeLoc(Tok.getLocation());
957 for (; Loc.isValid(); ++Depth)
958 Loc = SourceMgr.getIncludeLoc(Loc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000959
960 // __INCLUDE_LEVEL__ expands to a simple numeric value.
961 sprintf(TmpBuffer, "%u", Depth);
962 unsigned Length = strlen(TmpBuffer);
963 Tok.setKind(tok::numeric_constant);
964 Tok.setLength(Length);
965 Tok.setLocation(CreateString(TmpBuffer, Length, Tok.getLocation()));
966 } else if (II == Ident__TIMESTAMP__) {
967 // MSVC, ICC, GCC, VisualAge C++ extension. The generated string should be
968 // of the form "Ddd Mmm dd hh::mm::ss yyyy", which is returned by asctime.
969 Diag(Tok, diag::ext_pp_timestamp);
970
971 // Get the file that we are lexing out of. If we're currently lexing from
972 // a macro, dig into the include stack.
973 const FileEntry *CurFile = 0;
974 Lexer *TheLexer = getCurrentFileLexer();
975
976 if (TheLexer)
Chris Lattner9dc1f532007-07-20 16:37:10 +0000977 CurFile = SourceMgr.getFileEntryForLoc(TheLexer->getFileLoc());
Reid Spencer5f016e22007-07-11 17:01:13 +0000978
979 // If this file is older than the file it depends on, emit a diagnostic.
980 const char *Result;
981 if (CurFile) {
982 time_t TT = CurFile->getModificationTime();
983 struct tm *TM = localtime(&TT);
984 Result = asctime(TM);
985 } else {
986 Result = "??? ??? ?? ??:??:?? ????\n";
987 }
988 TmpBuffer[0] = '"';
989 strcpy(TmpBuffer+1, Result);
990 unsigned Len = strlen(TmpBuffer);
991 TmpBuffer[Len-1] = '"'; // Replace the newline with a quote.
992 Tok.setKind(tok::string_literal);
993 Tok.setLength(Len);
994 Tok.setLocation(CreateString(TmpBuffer, Len, Tok.getLocation()));
995 } else {
996 assert(0 && "Unknown identifier!");
997 }
998}
999
1000//===----------------------------------------------------------------------===//
1001// Lexer Event Handling.
1002//===----------------------------------------------------------------------===//
1003
1004/// LookUpIdentifierInfo - Given a tok::identifier token, look up the
1005/// identifier information for the token and install it into the token.
Chris Lattnerd2177732007-07-20 16:59:19 +00001006IdentifierInfo *Preprocessor::LookUpIdentifierInfo(Token &Identifier,
Reid Spencer5f016e22007-07-11 17:01:13 +00001007 const char *BufPtr) {
1008 assert(Identifier.getKind() == tok::identifier && "Not an identifier!");
1009 assert(Identifier.getIdentifierInfo() == 0 && "Identinfo already exists!");
1010
1011 // Look up this token, see if it is a macro, or if it is a language keyword.
1012 IdentifierInfo *II;
1013 if (BufPtr && !Identifier.needsCleaning()) {
1014 // No cleaning needed, just use the characters from the lexed buffer.
1015 II = getIdentifierInfo(BufPtr, BufPtr+Identifier.getLength());
1016 } else {
1017 // Cleaning needed, alloca a buffer, clean into it, then use the buffer.
Chris Lattnerc35717a2007-07-13 17:10:38 +00001018 llvm::SmallVector<char, 64> IdentifierBuffer;
1019 IdentifierBuffer.resize(Identifier.getLength());
1020 const char *TmpBuf = &IdentifierBuffer[0];
Reid Spencer5f016e22007-07-11 17:01:13 +00001021 unsigned Size = getSpelling(Identifier, TmpBuf);
1022 II = getIdentifierInfo(TmpBuf, TmpBuf+Size);
1023 }
1024 Identifier.setIdentifierInfo(II);
1025 return II;
1026}
1027
1028
1029/// HandleIdentifier - This callback is invoked when the lexer reads an
1030/// identifier. This callback looks up the identifier in the map and/or
1031/// potentially macro expands it or turns it into a named token (like 'for').
Chris Lattnerd2177732007-07-20 16:59:19 +00001032void Preprocessor::HandleIdentifier(Token &Identifier) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001033 assert(Identifier.getIdentifierInfo() &&
1034 "Can't handle identifiers without identifier info!");
1035
1036 IdentifierInfo &II = *Identifier.getIdentifierInfo();
1037
1038 // If this identifier was poisoned, and if it was not produced from a macro
1039 // expansion, emit an error.
1040 if (II.isPoisoned() && CurLexer) {
1041 if (&II != Ident__VA_ARGS__) // We warn about __VA_ARGS__ with poisoning.
1042 Diag(Identifier, diag::err_pp_used_poisoned_id);
1043 else
1044 Diag(Identifier, diag::ext_pp_bad_vaargs_use);
1045 }
1046
1047 // If this is a macro to be expanded, do it.
Chris Lattnercc1a8752007-10-07 08:44:20 +00001048 if (MacroInfo *MI = getMacroInfo(&II)) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001049 if (!DisableMacroExpansion && !Identifier.isExpandDisabled()) {
1050 if (MI->isEnabled()) {
1051 if (!HandleMacroExpandedIdentifier(Identifier, MI))
1052 return;
1053 } else {
1054 // C99 6.10.3.4p2 says that a disabled macro may never again be
1055 // expanded, even if it's in a context where it could be expanded in the
1056 // future.
Chris Lattnerd2177732007-07-20 16:59:19 +00001057 Identifier.setFlag(Token::DisableExpand);
Reid Spencer5f016e22007-07-11 17:01:13 +00001058 }
1059 }
1060 } else if (II.isOtherTargetMacro() && !DisableMacroExpansion) {
1061 // If this identifier is a macro on some other target, emit a diagnostic.
1062 // This diagnosic is only emitted when macro expansion is enabled, because
1063 // the macro would not have been expanded for the other target either.
1064 II.setIsOtherTargetMacro(false); // Don't warn on second use.
1065 getTargetInfo().DiagnoseNonPortability(Identifier.getLocation(),
1066 diag::port_target_macro_use);
1067
1068 }
1069
1070 // C++ 2.11p2: If this is an alternative representation of a C++ operator,
1071 // then we act as if it is the actual operator and not the textual
1072 // representation of it.
1073 if (II.isCPlusPlusOperatorKeyword())
1074 Identifier.setIdentifierInfo(0);
1075
1076 // Change the kind of this identifier to the appropriate token kind, e.g.
1077 // turning "for" into a keyword.
1078 Identifier.setKind(II.getTokenID());
1079
1080 // If this is an extension token, diagnose its use.
1081 // FIXME: tried (unsuccesfully) to shut this up when compiling with gnu99
1082 // For now, I'm just commenting it out (while I work on attributes).
1083 if (II.isExtensionToken() && Features.C99)
1084 Diag(Identifier, diag::ext_token_used);
1085}
1086
1087/// HandleEndOfFile - This callback is invoked when the lexer hits the end of
1088/// the current file. This either returns the EOF token or pops a level off
1089/// the include stack and keeps going.
Chris Lattnerd2177732007-07-20 16:59:19 +00001090bool Preprocessor::HandleEndOfFile(Token &Result, bool isEndOfMacro) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001091 assert(!CurMacroExpander &&
1092 "Ending a file when currently in a macro!");
1093
1094 // See if this file had a controlling macro.
1095 if (CurLexer) { // Not ending a macro, ignore it.
1096 if (const IdentifierInfo *ControllingMacro =
1097 CurLexer->MIOpt.GetControllingMacroAtEndOfFile()) {
1098 // Okay, this has a controlling macro, remember in PerFileInfo.
1099 if (const FileEntry *FE =
Chris Lattner9dc1f532007-07-20 16:37:10 +00001100 SourceMgr.getFileEntryForLoc(CurLexer->getFileLoc()))
Reid Spencer5f016e22007-07-11 17:01:13 +00001101 HeaderInfo.SetFileControllingMacro(FE, ControllingMacro);
1102 }
1103 }
1104
1105 // If this is a #include'd file, pop it off the include stack and continue
1106 // lexing the #includer file.
1107 if (!IncludeMacroStack.empty()) {
1108 // We're done with the #included file.
1109 RemoveTopOfLexerStack();
1110
1111 // Notify the client, if desired, that we are in a new source file.
1112 if (Callbacks && !isEndOfMacro && CurLexer) {
1113 DirectoryLookup::DirType FileType = DirectoryLookup::NormalHeaderDir;
1114
1115 // Get the file entry for the current file.
1116 if (const FileEntry *FE =
Chris Lattner9dc1f532007-07-20 16:37:10 +00001117 SourceMgr.getFileEntryForLoc(CurLexer->getFileLoc()))
Reid Spencer5f016e22007-07-11 17:01:13 +00001118 FileType = HeaderInfo.getFileDirFlavor(FE);
1119
1120 Callbacks->FileChanged(CurLexer->getSourceLocation(CurLexer->BufferPtr),
1121 PPCallbacks::ExitFile, FileType);
1122 }
1123
1124 // Client should lex another token.
1125 return false;
1126 }
1127
1128 Result.startToken();
1129 CurLexer->BufferPtr = CurLexer->BufferEnd;
1130 CurLexer->FormTokenWithChars(Result, CurLexer->BufferEnd);
1131 Result.setKind(tok::eof);
1132
1133 // We're done with the #included file.
1134 delete CurLexer;
1135 CurLexer = 0;
1136
1137 // This is the end of the top-level file. If the diag::pp_macro_not_used
Chris Lattnercc1a8752007-10-07 08:44:20 +00001138 // diagnostic is enabled, look for macros that have not been used.
Reid Spencer5f016e22007-07-11 17:01:13 +00001139 if (Diags.getDiagnosticLevel(diag::pp_macro_not_used) != Diagnostic::Ignored){
Chris Lattnercc1a8752007-10-07 08:44:20 +00001140 for (llvm::DenseMap<IdentifierInfo*, MacroInfo*>::iterator I =
1141 Macros.begin(), E = Macros.end(); I != E; ++I) {
1142 if (!I->second->isUsed())
1143 Diag(I->second->getDefinitionLoc(), diag::pp_macro_not_used);
Reid Spencer5f016e22007-07-11 17:01:13 +00001144 }
1145 }
Reid Spencer5f016e22007-07-11 17:01:13 +00001146 return true;
1147}
1148
1149/// HandleEndOfMacro - This callback is invoked when the lexer hits the end of
1150/// the current macro expansion or token stream expansion.
Chris Lattnerd2177732007-07-20 16:59:19 +00001151bool Preprocessor::HandleEndOfMacro(Token &Result) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001152 assert(CurMacroExpander && !CurLexer &&
1153 "Ending a macro when currently in a #include file!");
1154
Chris Lattner9594acf2007-07-15 00:25:26 +00001155 // Delete or cache the now-dead macro expander.
1156 if (NumCachedMacroExpanders == MacroExpanderCacheSize)
1157 delete CurMacroExpander;
1158 else
1159 MacroExpanderCache[NumCachedMacroExpanders++] = CurMacroExpander;
Reid Spencer5f016e22007-07-11 17:01:13 +00001160
1161 // Handle this like a #include file being popped off the stack.
1162 CurMacroExpander = 0;
1163 return HandleEndOfFile(Result, true);
1164}
1165
1166
1167//===----------------------------------------------------------------------===//
1168// Utility Methods for Preprocessor Directive Handling.
1169//===----------------------------------------------------------------------===//
1170
1171/// DiscardUntilEndOfDirective - Read and discard all tokens remaining on the
1172/// current line until the tok::eom token is found.
1173void Preprocessor::DiscardUntilEndOfDirective() {
Chris Lattnerd2177732007-07-20 16:59:19 +00001174 Token Tmp;
Reid Spencer5f016e22007-07-11 17:01:13 +00001175 do {
1176 LexUnexpandedToken(Tmp);
1177 } while (Tmp.getKind() != tok::eom);
1178}
1179
1180/// isCXXNamedOperator - Returns "true" if the token is a named operator in C++.
1181static bool isCXXNamedOperator(const std::string &Spelling) {
1182 return Spelling == "and" || Spelling == "bitand" || Spelling == "bitor" ||
1183 Spelling == "compl" || Spelling == "not" || Spelling == "not_eq" ||
1184 Spelling == "or" || Spelling == "xor";
1185}
1186
1187/// ReadMacroName - Lex and validate a macro name, which occurs after a
1188/// #define or #undef. This sets the token kind to eom and discards the rest
1189/// of the macro line if the macro name is invalid. isDefineUndef is 1 if
1190/// this is due to a a #define, 2 if #undef directive, 0 if it is something
1191/// else (e.g. #ifdef).
Chris Lattnerd2177732007-07-20 16:59:19 +00001192void Preprocessor::ReadMacroName(Token &MacroNameTok, char isDefineUndef) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001193 // Read the token, don't allow macro expansion on it.
1194 LexUnexpandedToken(MacroNameTok);
1195
1196 // Missing macro name?
1197 if (MacroNameTok.getKind() == tok::eom)
1198 return Diag(MacroNameTok, diag::err_pp_missing_macro_name);
1199
1200 IdentifierInfo *II = MacroNameTok.getIdentifierInfo();
1201 if (II == 0) {
1202 std::string Spelling = getSpelling(MacroNameTok);
1203 if (isCXXNamedOperator(Spelling))
1204 // C++ 2.5p2: Alternative tokens behave the same as its primary token
1205 // except for their spellings.
1206 Diag(MacroNameTok, diag::err_pp_operator_used_as_macro_name, Spelling);
1207 else
1208 Diag(MacroNameTok, diag::err_pp_macro_not_identifier);
1209 // Fall through on error.
1210 } else if (isDefineUndef && II->getPPKeywordID() == tok::pp_defined) {
1211 // Error if defining "defined": C99 6.10.8.4.
1212 Diag(MacroNameTok, diag::err_defined_macro_name);
Chris Lattner0edde552007-10-07 08:04:56 +00001213 } else if (isDefineUndef && II->hasMacroDefinition() &&
Chris Lattnercc1a8752007-10-07 08:44:20 +00001214 getMacroInfo(II)->isBuiltinMacro()) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001215 // Error if defining "__LINE__" and other builtins: C99 6.10.8.4.
1216 if (isDefineUndef == 1)
1217 Diag(MacroNameTok, diag::pp_redef_builtin_macro);
1218 else
1219 Diag(MacroNameTok, diag::pp_undef_builtin_macro);
1220 } else {
1221 // Okay, we got a good identifier node. Return it.
1222 return;
1223 }
1224
1225 // Invalid macro name, read and discard the rest of the line. Then set the
1226 // token kind to tok::eom.
1227 MacroNameTok.setKind(tok::eom);
1228 return DiscardUntilEndOfDirective();
1229}
1230
1231/// CheckEndOfDirective - Ensure that the next token is a tok::eom token. If
1232/// not, emit a diagnostic and consume up until the eom.
1233void Preprocessor::CheckEndOfDirective(const char *DirType) {
Chris Lattnerd2177732007-07-20 16:59:19 +00001234 Token Tmp;
Reid Spencer5f016e22007-07-11 17:01:13 +00001235 Lex(Tmp);
1236 // There should be no tokens after the directive, but we allow them as an
1237 // extension.
1238 while (Tmp.getKind() == tok::comment) // Skip comments in -C mode.
1239 Lex(Tmp);
1240
1241 if (Tmp.getKind() != tok::eom) {
1242 Diag(Tmp, diag::ext_pp_extra_tokens_at_eol, DirType);
1243 DiscardUntilEndOfDirective();
1244 }
1245}
1246
1247
1248
1249/// SkipExcludedConditionalBlock - We just read a #if or related directive and
1250/// decided that the subsequent tokens are in the #if'd out portion of the
1251/// file. Lex the rest of the file, until we see an #endif. If
1252/// FoundNonSkipPortion is true, then we have already emitted code for part of
1253/// this #if directive, so #else/#elif blocks should never be entered. If ElseOk
1254/// is true, then #else directives are ok, if not, then we have already seen one
1255/// so a #else directive is a duplicate. When this returns, the caller can lex
1256/// the first valid token.
1257void Preprocessor::SkipExcludedConditionalBlock(SourceLocation IfTokenLoc,
1258 bool FoundNonSkipPortion,
1259 bool FoundElse) {
1260 ++NumSkipped;
1261 assert(CurMacroExpander == 0 && CurLexer &&
1262 "Lexing a macro, not a file?");
1263
1264 CurLexer->pushConditionalLevel(IfTokenLoc, /*isSkipping*/false,
1265 FoundNonSkipPortion, FoundElse);
1266
1267 // Enter raw mode to disable identifier lookup (and thus macro expansion),
1268 // disabling warnings, etc.
1269 CurLexer->LexingRawMode = true;
Chris Lattnerd2177732007-07-20 16:59:19 +00001270 Token Tok;
Reid Spencer5f016e22007-07-11 17:01:13 +00001271 while (1) {
1272 CurLexer->Lex(Tok);
1273
1274 // If this is the end of the buffer, we have an error.
1275 if (Tok.getKind() == tok::eof) {
1276 // Emit errors for each unterminated conditional on the stack, including
1277 // the current one.
1278 while (!CurLexer->ConditionalStack.empty()) {
1279 Diag(CurLexer->ConditionalStack.back().IfLoc,
1280 diag::err_pp_unterminated_conditional);
1281 CurLexer->ConditionalStack.pop_back();
1282 }
1283
1284 // Just return and let the caller lex after this #include.
1285 break;
1286 }
1287
1288 // If this token is not a preprocessor directive, just skip it.
1289 if (Tok.getKind() != tok::hash || !Tok.isAtStartOfLine())
1290 continue;
1291
1292 // We just parsed a # character at the start of a line, so we're in
1293 // directive mode. Tell the lexer this so any newlines we see will be
1294 // converted into an EOM token (this terminates the macro).
1295 CurLexer->ParsingPreprocessorDirective = true;
1296 CurLexer->KeepCommentMode = false;
1297
1298
1299 // Read the next token, the directive flavor.
1300 LexUnexpandedToken(Tok);
1301
1302 // If this isn't an identifier directive (e.g. is "# 1\n" or "#\n", or
1303 // something bogus), skip it.
1304 if (Tok.getKind() != tok::identifier) {
1305 CurLexer->ParsingPreprocessorDirective = false;
1306 // Restore comment saving mode.
1307 CurLexer->KeepCommentMode = KeepComments;
1308 continue;
1309 }
1310
1311 // If the first letter isn't i or e, it isn't intesting to us. We know that
1312 // this is safe in the face of spelling differences, because there is no way
1313 // to spell an i/e in a strange way that is another letter. Skipping this
1314 // allows us to avoid looking up the identifier info for #define/#undef and
1315 // other common directives.
1316 const char *RawCharData = SourceMgr.getCharacterData(Tok.getLocation());
1317 char FirstChar = RawCharData[0];
1318 if (FirstChar >= 'a' && FirstChar <= 'z' &&
1319 FirstChar != 'i' && FirstChar != 'e') {
1320 CurLexer->ParsingPreprocessorDirective = false;
1321 // Restore comment saving mode.
1322 CurLexer->KeepCommentMode = KeepComments;
1323 continue;
1324 }
1325
1326 // Get the identifier name without trigraphs or embedded newlines. Note
1327 // that we can't use Tok.getIdentifierInfo() because its lookup is disabled
1328 // when skipping.
1329 // TODO: could do this with zero copies in the no-clean case by using
1330 // strncmp below.
1331 char Directive[20];
1332 unsigned IdLen;
1333 if (!Tok.needsCleaning() && Tok.getLength() < 20) {
1334 IdLen = Tok.getLength();
1335 memcpy(Directive, RawCharData, IdLen);
1336 Directive[IdLen] = 0;
1337 } else {
1338 std::string DirectiveStr = getSpelling(Tok);
1339 IdLen = DirectiveStr.size();
1340 if (IdLen >= 20) {
1341 CurLexer->ParsingPreprocessorDirective = false;
1342 // Restore comment saving mode.
1343 CurLexer->KeepCommentMode = KeepComments;
1344 continue;
1345 }
1346 memcpy(Directive, &DirectiveStr[0], IdLen);
1347 Directive[IdLen] = 0;
1348 }
1349
1350 if (FirstChar == 'i' && Directive[1] == 'f') {
1351 if ((IdLen == 2) || // "if"
1352 (IdLen == 5 && !strcmp(Directive+2, "def")) || // "ifdef"
1353 (IdLen == 6 && !strcmp(Directive+2, "ndef"))) { // "ifndef"
1354 // We know the entire #if/#ifdef/#ifndef block will be skipped, don't
1355 // bother parsing the condition.
1356 DiscardUntilEndOfDirective();
1357 CurLexer->pushConditionalLevel(Tok.getLocation(), /*wasskipping*/true,
1358 /*foundnonskip*/false,
1359 /*fnddelse*/false);
1360 }
1361 } else if (FirstChar == 'e') {
1362 if (IdLen == 5 && !strcmp(Directive+1, "ndif")) { // "endif"
1363 CheckEndOfDirective("#endif");
1364 PPConditionalInfo CondInfo;
1365 CondInfo.WasSkipping = true; // Silence bogus warning.
1366 bool InCond = CurLexer->popConditionalLevel(CondInfo);
1367 InCond = InCond; // Silence warning in no-asserts mode.
1368 assert(!InCond && "Can't be skipping if not in a conditional!");
1369
1370 // If we popped the outermost skipping block, we're done skipping!
1371 if (!CondInfo.WasSkipping)
1372 break;
1373 } else if (IdLen == 4 && !strcmp(Directive+1, "lse")) { // "else".
1374 // #else directive in a skipping conditional. If not in some other
1375 // skipping conditional, and if #else hasn't already been seen, enter it
1376 // as a non-skipping conditional.
1377 CheckEndOfDirective("#else");
1378 PPConditionalInfo &CondInfo = CurLexer->peekConditionalLevel();
1379
1380 // If this is a #else with a #else before it, report the error.
1381 if (CondInfo.FoundElse) Diag(Tok, diag::pp_err_else_after_else);
1382
1383 // Note that we've seen a #else in this conditional.
1384 CondInfo.FoundElse = true;
1385
1386 // If the conditional is at the top level, and the #if block wasn't
1387 // entered, enter the #else block now.
1388 if (!CondInfo.WasSkipping && !CondInfo.FoundNonSkip) {
1389 CondInfo.FoundNonSkip = true;
1390 break;
1391 }
1392 } else if (IdLen == 4 && !strcmp(Directive+1, "lif")) { // "elif".
1393 PPConditionalInfo &CondInfo = CurLexer->peekConditionalLevel();
1394
1395 bool ShouldEnter;
1396 // If this is in a skipping block or if we're already handled this #if
1397 // block, don't bother parsing the condition.
1398 if (CondInfo.WasSkipping || CondInfo.FoundNonSkip) {
1399 DiscardUntilEndOfDirective();
1400 ShouldEnter = false;
1401 } else {
1402 // Restore the value of LexingRawMode so that identifiers are
1403 // looked up, etc, inside the #elif expression.
1404 assert(CurLexer->LexingRawMode && "We have to be skipping here!");
1405 CurLexer->LexingRawMode = false;
1406 IdentifierInfo *IfNDefMacro = 0;
1407 ShouldEnter = EvaluateDirectiveExpression(IfNDefMacro);
1408 CurLexer->LexingRawMode = true;
1409 }
1410
1411 // If this is a #elif with a #else before it, report the error.
1412 if (CondInfo.FoundElse) Diag(Tok, diag::pp_err_elif_after_else);
1413
1414 // If this condition is true, enter it!
1415 if (ShouldEnter) {
1416 CondInfo.FoundNonSkip = true;
1417 break;
1418 }
1419 }
1420 }
1421
1422 CurLexer->ParsingPreprocessorDirective = false;
1423 // Restore comment saving mode.
1424 CurLexer->KeepCommentMode = KeepComments;
1425 }
1426
1427 // Finally, if we are out of the conditional (saw an #endif or ran off the end
1428 // of the file, just stop skipping and return to lexing whatever came after
1429 // the #if block.
1430 CurLexer->LexingRawMode = false;
1431}
1432
1433//===----------------------------------------------------------------------===//
1434// Preprocessor Directive Handling.
1435//===----------------------------------------------------------------------===//
1436
1437/// HandleDirective - This callback is invoked when the lexer sees a # token
1438/// at the start of a line. This consumes the directive, modifies the
1439/// lexer/preprocessor state, and advances the lexer(s) so that the next token
1440/// read is the correct one.
Chris Lattnerd2177732007-07-20 16:59:19 +00001441void Preprocessor::HandleDirective(Token &Result) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001442 // FIXME: Traditional: # with whitespace before it not recognized by K&R?
1443
1444 // We just parsed a # character at the start of a line, so we're in directive
1445 // mode. Tell the lexer this so any newlines we see will be converted into an
1446 // EOM token (which terminates the directive).
1447 CurLexer->ParsingPreprocessorDirective = true;
1448
1449 ++NumDirectives;
1450
1451 // We are about to read a token. For the multiple-include optimization FA to
1452 // work, we have to remember if we had read any tokens *before* this
1453 // pp-directive.
1454 bool ReadAnyTokensBeforeDirective = CurLexer->MIOpt.getHasReadAnyTokensVal();
1455
1456 // Read the next token, the directive flavor. This isn't expanded due to
1457 // C99 6.10.3p8.
1458 LexUnexpandedToken(Result);
1459
1460 // C99 6.10.3p11: Is this preprocessor directive in macro invocation? e.g.:
1461 // #define A(x) #x
1462 // A(abc
1463 // #warning blah
1464 // def)
1465 // If so, the user is relying on non-portable behavior, emit a diagnostic.
1466 if (InMacroArgs)
1467 Diag(Result, diag::ext_embedded_directive);
1468
1469TryAgain:
1470 switch (Result.getKind()) {
1471 case tok::eom:
1472 return; // null directive.
1473 case tok::comment:
1474 // Handle stuff like "# /*foo*/ define X" in -E -C mode.
1475 LexUnexpandedToken(Result);
1476 goto TryAgain;
1477
1478 case tok::numeric_constant:
1479 // FIXME: implement # 7 line numbers!
1480 DiscardUntilEndOfDirective();
1481 return;
1482 default:
1483 IdentifierInfo *II = Result.getIdentifierInfo();
1484 if (II == 0) break; // Not an identifier.
1485
1486 // Ask what the preprocessor keyword ID is.
1487 switch (II->getPPKeywordID()) {
1488 default: break;
1489 // C99 6.10.1 - Conditional Inclusion.
1490 case tok::pp_if:
1491 return HandleIfDirective(Result, ReadAnyTokensBeforeDirective);
1492 case tok::pp_ifdef:
1493 return HandleIfdefDirective(Result, false, true/*not valid for miopt*/);
1494 case tok::pp_ifndef:
1495 return HandleIfdefDirective(Result, true, ReadAnyTokensBeforeDirective);
1496 case tok::pp_elif:
1497 return HandleElifDirective(Result);
1498 case tok::pp_else:
1499 return HandleElseDirective(Result);
1500 case tok::pp_endif:
1501 return HandleEndifDirective(Result);
1502
1503 // C99 6.10.2 - Source File Inclusion.
1504 case tok::pp_include:
1505 return HandleIncludeDirective(Result); // Handle #include.
1506
1507 // C99 6.10.3 - Macro Replacement.
1508 case tok::pp_define:
1509 return HandleDefineDirective(Result, false);
1510 case tok::pp_undef:
1511 return HandleUndefDirective(Result);
1512
1513 // C99 6.10.4 - Line Control.
1514 case tok::pp_line:
1515 // FIXME: implement #line
1516 DiscardUntilEndOfDirective();
1517 return;
1518
1519 // C99 6.10.5 - Error Directive.
1520 case tok::pp_error:
1521 return HandleUserDiagnosticDirective(Result, false);
1522
1523 // C99 6.10.6 - Pragma Directive.
1524 case tok::pp_pragma:
1525 return HandlePragmaDirective();
1526
1527 // GNU Extensions.
1528 case tok::pp_import:
1529 return HandleImportDirective(Result);
1530 case tok::pp_include_next:
1531 return HandleIncludeNextDirective(Result);
1532
1533 case tok::pp_warning:
1534 Diag(Result, diag::ext_pp_warning_directive);
1535 return HandleUserDiagnosticDirective(Result, true);
1536 case tok::pp_ident:
1537 return HandleIdentSCCSDirective(Result);
1538 case tok::pp_sccs:
1539 return HandleIdentSCCSDirective(Result);
1540 case tok::pp_assert:
1541 //isExtension = true; // FIXME: implement #assert
1542 break;
1543 case tok::pp_unassert:
1544 //isExtension = true; // FIXME: implement #unassert
1545 break;
1546
1547 // clang extensions.
1548 case tok::pp_define_target:
1549 return HandleDefineDirective(Result, true);
1550 case tok::pp_define_other_target:
1551 return HandleDefineOtherTargetDirective(Result);
1552 }
1553 break;
1554 }
1555
1556 // If we reached here, the preprocessing token is not valid!
1557 Diag(Result, diag::err_pp_invalid_directive);
1558
1559 // Read the rest of the PP line.
1560 DiscardUntilEndOfDirective();
1561
1562 // Okay, we're done parsing the directive.
1563}
1564
Chris Lattnerd2177732007-07-20 16:59:19 +00001565void Preprocessor::HandleUserDiagnosticDirective(Token &Tok,
Reid Spencer5f016e22007-07-11 17:01:13 +00001566 bool isWarning) {
1567 // Read the rest of the line raw. We do this because we don't want macros
1568 // to be expanded and we don't require that the tokens be valid preprocessing
1569 // tokens. For example, this is allowed: "#warning ` 'foo". GCC does
1570 // collapse multiple consequtive white space between tokens, but this isn't
1571 // specified by the standard.
1572 std::string Message = CurLexer->ReadToEndOfLine();
1573
1574 unsigned DiagID = isWarning ? diag::pp_hash_warning : diag::err_pp_hash_error;
1575 return Diag(Tok, DiagID, Message);
1576}
1577
1578/// HandleIdentSCCSDirective - Handle a #ident/#sccs directive.
1579///
Chris Lattnerd2177732007-07-20 16:59:19 +00001580void Preprocessor::HandleIdentSCCSDirective(Token &Tok) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001581 // Yes, this directive is an extension.
1582 Diag(Tok, diag::ext_pp_ident_directive);
1583
1584 // Read the string argument.
Chris Lattnerd2177732007-07-20 16:59:19 +00001585 Token StrTok;
Reid Spencer5f016e22007-07-11 17:01:13 +00001586 Lex(StrTok);
1587
1588 // If the token kind isn't a string, it's a malformed directive.
1589 if (StrTok.getKind() != tok::string_literal &&
1590 StrTok.getKind() != tok::wide_string_literal)
1591 return Diag(StrTok, diag::err_pp_malformed_ident);
1592
1593 // Verify that there is nothing after the string, other than EOM.
1594 CheckEndOfDirective("#ident");
1595
1596 if (Callbacks)
1597 Callbacks->Ident(Tok.getLocation(), getSpelling(StrTok));
1598}
1599
1600//===----------------------------------------------------------------------===//
1601// Preprocessor Include Directive Handling.
1602//===----------------------------------------------------------------------===//
1603
1604/// GetIncludeFilenameSpelling - Turn the specified lexer token into a fully
1605/// checked and spelled filename, e.g. as an operand of #include. This returns
1606/// true if the input filename was in <>'s or false if it were in ""'s. The
1607/// caller is expected to provide a buffer that is large enough to hold the
1608/// spelling of the filename, but is also expected to handle the case when
1609/// this method decides to use a different buffer.
Chris Lattnerf1c99ac2007-07-23 04:15:27 +00001610bool Preprocessor::GetIncludeFilenameSpelling(SourceLocation Loc,
Reid Spencer5f016e22007-07-11 17:01:13 +00001611 const char *&BufStart,
1612 const char *&BufEnd) {
1613 // Get the text form of the filename.
Reid Spencer5f016e22007-07-11 17:01:13 +00001614 assert(BufStart != BufEnd && "Can't have tokens with empty spellings!");
1615
1616 // Make sure the filename is <x> or "x".
1617 bool isAngled;
1618 if (BufStart[0] == '<') {
1619 if (BufEnd[-1] != '>') {
Chris Lattnerf1c99ac2007-07-23 04:15:27 +00001620 Diag(Loc, diag::err_pp_expects_filename);
Reid Spencer5f016e22007-07-11 17:01:13 +00001621 BufStart = 0;
1622 return true;
1623 }
1624 isAngled = true;
1625 } else if (BufStart[0] == '"') {
1626 if (BufEnd[-1] != '"') {
Chris Lattnerf1c99ac2007-07-23 04:15:27 +00001627 Diag(Loc, diag::err_pp_expects_filename);
Reid Spencer5f016e22007-07-11 17:01:13 +00001628 BufStart = 0;
1629 return true;
1630 }
1631 isAngled = false;
1632 } else {
Chris Lattnerf1c99ac2007-07-23 04:15:27 +00001633 Diag(Loc, diag::err_pp_expects_filename);
Reid Spencer5f016e22007-07-11 17:01:13 +00001634 BufStart = 0;
1635 return true;
1636 }
1637
1638 // Diagnose #include "" as invalid.
1639 if (BufEnd-BufStart <= 2) {
Chris Lattnerf1c99ac2007-07-23 04:15:27 +00001640 Diag(Loc, diag::err_pp_empty_filename);
Reid Spencer5f016e22007-07-11 17:01:13 +00001641 BufStart = 0;
1642 return "";
1643 }
1644
1645 // Skip the brackets.
1646 ++BufStart;
1647 --BufEnd;
1648 return isAngled;
1649}
1650
Chris Lattner706ab502007-07-23 04:56:47 +00001651/// ConcatenateIncludeName - Handle cases where the #include name is expanded
1652/// from a macro as multiple tokens, which need to be glued together. This
1653/// occurs for code like:
1654/// #define FOO <a/b.h>
1655/// #include FOO
1656/// because in this case, "<a/b.h>" is returned as 7 tokens, not one.
1657///
1658/// This code concatenates and consumes tokens up to the '>' token. It returns
1659/// false if the > was found, otherwise it returns true if it finds and consumes
1660/// the EOM marker.
1661static bool ConcatenateIncludeName(llvm::SmallVector<char, 128> &FilenameBuffer,
1662 Preprocessor &PP) {
1663 Token CurTok;
1664
1665 PP.Lex(CurTok);
1666 while (CurTok.getKind() != tok::eom) {
1667 // Append the spelling of this token to the buffer. If there was a space
1668 // before it, add it now.
1669 if (CurTok.hasLeadingSpace())
1670 FilenameBuffer.push_back(' ');
1671
1672 // Get the spelling of the token, directly into FilenameBuffer if possible.
1673 unsigned PreAppendSize = FilenameBuffer.size();
1674 FilenameBuffer.resize(PreAppendSize+CurTok.getLength());
1675
1676 const char *BufPtr = &FilenameBuffer[PreAppendSize];
1677 unsigned ActualLen = PP.getSpelling(CurTok, BufPtr);
1678
1679 // If the token was spelled somewhere else, copy it into FilenameBuffer.
1680 if (BufPtr != &FilenameBuffer[PreAppendSize])
1681 memcpy(&FilenameBuffer[PreAppendSize], BufPtr, ActualLen);
1682
1683 // Resize FilenameBuffer to the correct size.
1684 if (CurTok.getLength() != ActualLen)
1685 FilenameBuffer.resize(PreAppendSize+ActualLen);
1686
1687 // If we found the '>' marker, return success.
1688 if (CurTok.getKind() == tok::greater)
1689 return false;
1690
1691 PP.Lex(CurTok);
1692 }
1693
1694 // If we hit the eom marker, emit an error and return true so that the caller
1695 // knows the EOM has been read.
1696 PP.Diag(CurTok.getLocation(), diag::err_pp_expects_filename);
1697 return true;
1698}
1699
Reid Spencer5f016e22007-07-11 17:01:13 +00001700/// HandleIncludeDirective - The "#include" tokens have just been read, read the
1701/// file to be included from the lexer, then include it! This is a common
1702/// routine with functionality shared between #include, #include_next and
1703/// #import.
Chris Lattnerd2177732007-07-20 16:59:19 +00001704void Preprocessor::HandleIncludeDirective(Token &IncludeTok,
Reid Spencer5f016e22007-07-11 17:01:13 +00001705 const DirectoryLookup *LookupFrom,
1706 bool isImport) {
1707
Chris Lattnerd2177732007-07-20 16:59:19 +00001708 Token FilenameTok;
Reid Spencer5f016e22007-07-11 17:01:13 +00001709 CurLexer->LexIncludeFilename(FilenameTok);
1710
Reid Spencer5f016e22007-07-11 17:01:13 +00001711 // Reserve a buffer to get the spelling.
1712 llvm::SmallVector<char, 128> FilenameBuffer;
Chris Lattner706ab502007-07-23 04:56:47 +00001713 const char *FilenameStart, *FilenameEnd;
1714
1715 switch (FilenameTok.getKind()) {
1716 case tok::eom:
1717 // If the token kind is EOM, the error has already been diagnosed.
1718 return;
Reid Spencer5f016e22007-07-11 17:01:13 +00001719
Chris Lattner706ab502007-07-23 04:56:47 +00001720 case tok::angle_string_literal:
Chris Lattnerf11ccfc2007-07-23 22:23:52 +00001721 case tok::string_literal: {
Chris Lattner706ab502007-07-23 04:56:47 +00001722 FilenameBuffer.resize(FilenameTok.getLength());
1723 FilenameStart = &FilenameBuffer[0];
1724 unsigned Len = getSpelling(FilenameTok, FilenameStart);
1725 FilenameEnd = FilenameStart+Len;
1726 break;
Chris Lattnerf11ccfc2007-07-23 22:23:52 +00001727 }
Chris Lattner706ab502007-07-23 04:56:47 +00001728
1729 case tok::less:
1730 // This could be a <foo/bar.h> file coming from a macro expansion. In this
1731 // case, glue the tokens together into FilenameBuffer and interpret those.
1732 FilenameBuffer.push_back('<');
1733 if (ConcatenateIncludeName(FilenameBuffer, *this))
1734 return; // Found <eom> but no ">"? Diagnostic already emitted.
1735 FilenameStart = &FilenameBuffer[0];
1736 FilenameEnd = &FilenameBuffer[FilenameBuffer.size()];
1737 break;
1738 default:
1739 Diag(FilenameTok.getLocation(), diag::err_pp_expects_filename);
1740 DiscardUntilEndOfDirective();
1741 return;
1742 }
1743
Chris Lattnerf1c99ac2007-07-23 04:15:27 +00001744 bool isAngled = GetIncludeFilenameSpelling(FilenameTok.getLocation(),
Reid Spencer5f016e22007-07-11 17:01:13 +00001745 FilenameStart, FilenameEnd);
1746 // If GetIncludeFilenameSpelling set the start ptr to null, there was an
1747 // error.
Chris Lattner706ab502007-07-23 04:56:47 +00001748 if (FilenameStart == 0) {
1749 DiscardUntilEndOfDirective();
Reid Spencer5f016e22007-07-11 17:01:13 +00001750 return;
Chris Lattner706ab502007-07-23 04:56:47 +00001751 }
Reid Spencer5f016e22007-07-11 17:01:13 +00001752
1753 // Verify that there is nothing after the filename, other than EOM. Use the
1754 // preprocessor to lex this in case lexing the filename entered a macro.
1755 CheckEndOfDirective("#include");
1756
1757 // Check that we don't have infinite #include recursion.
1758 if (IncludeMacroStack.size() == MaxAllowedIncludeStackDepth-1)
1759 return Diag(FilenameTok, diag::err_pp_include_too_deep);
1760
1761 // Search include directories.
1762 const DirectoryLookup *CurDir;
1763 const FileEntry *File = LookupFile(FilenameStart, FilenameEnd,
1764 isAngled, LookupFrom, CurDir);
1765 if (File == 0)
1766 return Diag(FilenameTok, diag::err_pp_file_not_found,
1767 std::string(FilenameStart, FilenameEnd));
1768
1769 // Ask HeaderInfo if we should enter this #include file.
1770 if (!HeaderInfo.ShouldEnterIncludeFile(File, isImport)) {
1771 // If it returns true, #including this file will have no effect.
1772 return;
1773 }
1774
1775 // Look up the file, create a File ID for it.
1776 unsigned FileID = SourceMgr.createFileID(File, FilenameTok.getLocation());
1777 if (FileID == 0)
1778 return Diag(FilenameTok, diag::err_pp_file_not_found,
1779 std::string(FilenameStart, FilenameEnd));
1780
1781 // Finally, if all is good, enter the new file!
1782 EnterSourceFile(FileID, CurDir);
1783}
1784
1785/// HandleIncludeNextDirective - Implements #include_next.
1786///
Chris Lattnerd2177732007-07-20 16:59:19 +00001787void Preprocessor::HandleIncludeNextDirective(Token &IncludeNextTok) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001788 Diag(IncludeNextTok, diag::ext_pp_include_next_directive);
1789
1790 // #include_next is like #include, except that we start searching after
1791 // the current found directory. If we can't do this, issue a
1792 // diagnostic.
1793 const DirectoryLookup *Lookup = CurDirLookup;
1794 if (isInPrimaryFile()) {
1795 Lookup = 0;
1796 Diag(IncludeNextTok, diag::pp_include_next_in_primary);
1797 } else if (Lookup == 0) {
1798 Diag(IncludeNextTok, diag::pp_include_next_absolute_path);
1799 } else {
1800 // Start looking up in the next directory.
1801 ++Lookup;
1802 }
1803
1804 return HandleIncludeDirective(IncludeNextTok, Lookup);
1805}
1806
1807/// HandleImportDirective - Implements #import.
1808///
Chris Lattnerd2177732007-07-20 16:59:19 +00001809void Preprocessor::HandleImportDirective(Token &ImportTok) {
Reid Spencer5f016e22007-07-11 17:01:13 +00001810 Diag(ImportTok, diag::ext_pp_import_directive);
1811
1812 return HandleIncludeDirective(ImportTok, 0, true);
1813}
1814
1815//===----------------------------------------------------------------------===//
1816// Preprocessor Macro Directive Handling.
1817//===----------------------------------------------------------------------===//
1818
1819/// ReadMacroDefinitionArgList - The ( starting an argument list of a macro
1820/// definition has just been read. Lex the rest of the arguments and the
1821/// closing ), updating MI with what we learn. Return true if an error occurs
1822/// parsing the arg list.
1823bool Preprocessor::ReadMacroDefinitionArgList(MacroInfo *MI) {
Chris Lattner25c96482007-07-14 22:46:43 +00001824 llvm::SmallVector<IdentifierInfo*, 32> Arguments;
1825
Chris Lattnerd2177732007-07-20 16:59:19 +00001826 Token Tok;
Reid Spencer5f016e22007-07-11 17:01:13 +00001827 while (1) {
1828 LexUnexpandedToken(Tok);
1829 switch (Tok.getKind()) {
1830 case tok::r_paren:
1831 // Found the end of the argument list.
Chris Lattner25c96482007-07-14 22:46:43 +00001832 if (Arguments.empty()) { // #define FOO()
1833 MI->setArgumentList(Arguments.begin(), Arguments.end());
1834 return false;
1835 }
Reid Spencer5f016e22007-07-11 17:01:13 +00001836 // Otherwise we have #define FOO(A,)
1837 Diag(Tok, diag::err_pp_expected_ident_in_arg_list);
1838 return true;
1839 case tok::ellipsis: // #define X(... -> C99 varargs
1840 // Warn if use of C99 feature in non-C99 mode.
1841 if (!Features.C99) Diag(Tok, diag::ext_variadic_macro);
1842
1843 // Lex the token after the identifier.
1844 LexUnexpandedToken(Tok);
1845 if (Tok.getKind() != tok::r_paren) {
1846 Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
1847 return true;
1848 }
1849 // Add the __VA_ARGS__ identifier as an argument.
Chris Lattner25c96482007-07-14 22:46:43 +00001850 Arguments.push_back(Ident__VA_ARGS__);
Reid Spencer5f016e22007-07-11 17:01:13 +00001851 MI->setIsC99Varargs();
Chris Lattner25c96482007-07-14 22:46:43 +00001852 MI->setArgumentList(Arguments.begin(), Arguments.end());
Reid Spencer5f016e22007-07-11 17:01:13 +00001853 return false;
1854 case tok::eom: // #define X(
1855 Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
1856 return true;
1857 default:
1858 // Handle keywords and identifiers here to accept things like
1859 // #define Foo(for) for.
1860 IdentifierInfo *II = Tok.getIdentifierInfo();
1861 if (II == 0) {
1862 // #define X(1
1863 Diag(Tok, diag::err_pp_invalid_tok_in_arg_list);
1864 return true;
1865 }
1866
1867 // If this is already used as an argument, it is used multiple times (e.g.
1868 // #define X(A,A.
Chris Lattner25c96482007-07-14 22:46:43 +00001869 if (std::find(Arguments.begin(), Arguments.end(), II) !=
1870 Arguments.end()) { // C99 6.10.3p6
Reid Spencer5f016e22007-07-11 17:01:13 +00001871 Diag(Tok, diag::err_pp_duplicate_name_in_arg_list, II->getName());
1872 return true;
1873 }
1874
1875 // Add the argument to the macro info.
Chris Lattner25c96482007-07-14 22:46:43 +00001876 Arguments.push_back(II);
Reid Spencer5f016e22007-07-11 17:01:13 +00001877
1878 // Lex the token after the identifier.
1879 LexUnexpandedToken(Tok);
1880
1881 switch (Tok.getKind()) {
1882 default: // #define X(A B
1883 Diag(Tok, diag::err_pp_expected_comma_in_arg_list);
1884 return true;
1885 case tok::r_paren: // #define X(A)
Chris Lattner25c96482007-07-14 22:46:43 +00001886 MI->setArgumentList(Arguments.begin(), Arguments.end());
Reid Spencer5f016e22007-07-11 17:01:13 +00001887 return false;
1888 case tok::comma: // #define X(A,
1889 break;
1890 case tok::ellipsis: // #define X(A... -> GCC extension
1891 // Diagnose extension.
1892 Diag(Tok, diag::ext_named_variadic_macro);
1893
1894 // Lex the token after the identifier.
1895 LexUnexpandedToken(Tok);
1896 if (Tok.getKind() != tok::r_paren) {
1897 Diag(Tok, diag::err_pp_missing_rparen_in_macro_def);
1898 return true;
1899 }
1900
1901 MI->setIsGNUVarargs();
Chris Lattner25c96482007-07-14 22:46:43 +00001902 MI->setArgumentList(Arguments.begin(), Arguments.end());
Reid Spencer5f016e22007-07-11 17:01:13 +00001903 return false;
1904 }
1905 }
1906 }
1907}
1908
1909/// HandleDefineDirective - Implements #define. This consumes the entire macro
1910/// line then lets the caller lex the next real token. If 'isTargetSpecific' is
1911/// true, then this is a "#define_target", otherwise this is a "#define".
1912///
Chris Lattnerd2177732007-07-20 16:59:19 +00001913void Preprocessor::HandleDefineDirective(Token &DefineTok,
Reid Spencer5f016e22007-07-11 17:01:13 +00001914 bool isTargetSpecific) {
1915 ++NumDefined;
1916
Chris Lattnerd2177732007-07-20 16:59:19 +00001917 Token MacroNameTok;
Reid Spencer5f016e22007-07-11 17:01:13 +00001918 ReadMacroName(MacroNameTok, 1);
1919
1920 // Error reading macro name? If so, diagnostic already issued.
1921 if (MacroNameTok.getKind() == tok::eom)
1922 return;
Chris Lattnerc215bd62007-07-14 22:11:41 +00001923
Reid Spencer5f016e22007-07-11 17:01:13 +00001924 // If we are supposed to keep comments in #defines, reenable comment saving
1925 // mode.
1926 CurLexer->KeepCommentMode = KeepMacroComments;
1927
1928 // Create the new macro.
1929 MacroInfo *MI = new MacroInfo(MacroNameTok.getLocation());
1930 if (isTargetSpecific) MI->setIsTargetSpecific();
1931
1932 // If the identifier is an 'other target' macro, clear this bit.
1933 MacroNameTok.getIdentifierInfo()->setIsOtherTargetMacro(false);
1934
1935
Chris Lattnerd2177732007-07-20 16:59:19 +00001936 Token Tok;
Reid Spencer5f016e22007-07-11 17:01:13 +00001937 LexUnexpandedToken(Tok);
1938
1939 // If this is a function-like macro definition, parse the argument list,
1940 // marking each of the identifiers as being used as macro arguments. Also,
1941 // check other constraints on the first token of the macro body.
1942 if (Tok.getKind() == tok::eom) {
1943 // If there is no body to this macro, we have no special handling here.
1944 } else if (Tok.getKind() == tok::l_paren && !Tok.hasLeadingSpace()) {
1945 // This is a function-like macro definition. Read the argument list.
1946 MI->setIsFunctionLike();
1947 if (ReadMacroDefinitionArgList(MI)) {
1948 // Forget about MI.
1949 delete MI;
1950 // Throw away the rest of the line.
1951 if (CurLexer->ParsingPreprocessorDirective)
1952 DiscardUntilEndOfDirective();
1953 return;
1954 }
1955
1956 // Read the first token after the arg list for down below.
1957 LexUnexpandedToken(Tok);
1958 } else if (!Tok.hasLeadingSpace()) {
1959 // C99 requires whitespace between the macro definition and the body. Emit
1960 // a diagnostic for something like "#define X+".
1961 if (Features.C99) {
1962 Diag(Tok, diag::ext_c99_whitespace_required_after_macro_name);
1963 } else {
1964 // FIXME: C90/C++ do not get this diagnostic, but it does get a similar
1965 // one in some cases!
1966 }
1967 } else {
1968 // This is a normal token with leading space. Clear the leading space
1969 // marker on the first token to get proper expansion.
Chris Lattnerd2177732007-07-20 16:59:19 +00001970 Tok.clearFlag(Token::LeadingSpace);
Reid Spencer5f016e22007-07-11 17:01:13 +00001971 }
1972
1973 // If this is a definition of a variadic C99 function-like macro, not using
1974 // the GNU named varargs extension, enabled __VA_ARGS__.
1975
1976 // "Poison" __VA_ARGS__, which can only appear in the expansion of a macro.
1977 // This gets unpoisoned where it is allowed.
1978 assert(Ident__VA_ARGS__->isPoisoned() && "__VA_ARGS__ should be poisoned!");
1979 if (MI->isC99Varargs())
1980 Ident__VA_ARGS__->setIsPoisoned(false);
1981
1982 // Read the rest of the macro body.
Chris Lattnerb5e240f2007-07-14 21:54:03 +00001983 if (MI->isObjectLike()) {
1984 // Object-like macros are very simple, just read their body.
1985 while (Tok.getKind() != tok::eom) {
1986 MI->AddTokenToBody(Tok);
Reid Spencer5f016e22007-07-11 17:01:13 +00001987 // Get the next token of the macro.
1988 LexUnexpandedToken(Tok);
Reid Spencer5f016e22007-07-11 17:01:13 +00001989 }
1990
Chris Lattnerb5e240f2007-07-14 21:54:03 +00001991 } else {
1992 // Otherwise, read the body of a function-like macro. This has to validate
1993 // the # (stringize) operator.
1994 while (Tok.getKind() != tok::eom) {
1995 MI->AddTokenToBody(Tok);
Reid Spencer5f016e22007-07-11 17:01:13 +00001996
Chris Lattnerb5e240f2007-07-14 21:54:03 +00001997 // Check C99 6.10.3.2p1: ensure that # operators are followed by macro
1998 // parameters in function-like macro expansions.
1999 if (Tok.getKind() != tok::hash) {
2000 // Get the next token of the macro.
2001 LexUnexpandedToken(Tok);
2002 continue;
2003 }
2004
2005 // Get the next token of the macro.
2006 LexUnexpandedToken(Tok);
2007
2008 // Not a macro arg identifier?
2009 if (!Tok.getIdentifierInfo() ||
2010 MI->getArgumentNum(Tok.getIdentifierInfo()) == -1) {
2011 Diag(Tok, diag::err_pp_stringize_not_parameter);
2012 delete MI;
2013
2014 // Disable __VA_ARGS__ again.
2015 Ident__VA_ARGS__->setIsPoisoned(true);
2016 return;
2017 }
2018
2019 // Things look ok, add the param name token to the macro.
2020 MI->AddTokenToBody(Tok);
2021
2022 // Get the next token of the macro.
2023 LexUnexpandedToken(Tok);
2024 }
Reid Spencer5f016e22007-07-11 17:01:13 +00002025 }
2026
Chris Lattnerc215bd62007-07-14 22:11:41 +00002027
Reid Spencer5f016e22007-07-11 17:01:13 +00002028 // Disable __VA_ARGS__ again.
2029 Ident__VA_ARGS__->setIsPoisoned(true);
2030
2031 // Check that there is no paste (##) operator at the begining or end of the
2032 // replacement list.
2033 unsigned NumTokens = MI->getNumTokens();
2034 if (NumTokens != 0) {
2035 if (MI->getReplacementToken(0).getKind() == tok::hashhash) {
2036 Diag(MI->getReplacementToken(0), diag::err_paste_at_start);
2037 delete MI;
2038 return;
2039 }
2040 if (MI->getReplacementToken(NumTokens-1).getKind() == tok::hashhash) {
2041 Diag(MI->getReplacementToken(NumTokens-1), diag::err_paste_at_end);
2042 delete MI;
2043 return;
2044 }
2045 }
2046
2047 // If this is the primary source file, remember that this macro hasn't been
2048 // used yet.
2049 if (isInPrimaryFile())
2050 MI->setIsUsed(false);
2051
2052 // Finally, if this identifier already had a macro defined for it, verify that
2053 // the macro bodies are identical and free the old definition.
Chris Lattnercc1a8752007-10-07 08:44:20 +00002054 if (MacroInfo *OtherMI = getMacroInfo(MacroNameTok.getIdentifierInfo())) {
Reid Spencer5f016e22007-07-11 17:01:13 +00002055 if (!OtherMI->isUsed())
2056 Diag(OtherMI->getDefinitionLoc(), diag::pp_macro_not_used);
2057
2058 // Macros must be identical. This means all tokes and whitespace separation
2059 // must be the same. C99 6.10.3.2.
2060 if (!MI->isIdenticalTo(*OtherMI, *this)) {
2061 Diag(MI->getDefinitionLoc(), diag::ext_pp_macro_redef,
2062 MacroNameTok.getIdentifierInfo()->getName());
2063 Diag(OtherMI->getDefinitionLoc(), diag::ext_pp_macro_redef2);
2064 }
2065 delete OtherMI;
2066 }
2067
Chris Lattnercc1a8752007-10-07 08:44:20 +00002068 setMacroInfo(MacroNameTok.getIdentifierInfo(), MI);
Reid Spencer5f016e22007-07-11 17:01:13 +00002069}
2070
2071/// HandleDefineOtherTargetDirective - Implements #define_other_target.
Chris Lattnerd2177732007-07-20 16:59:19 +00002072void Preprocessor::HandleDefineOtherTargetDirective(Token &Tok) {
2073 Token MacroNameTok;
Reid Spencer5f016e22007-07-11 17:01:13 +00002074 ReadMacroName(MacroNameTok, 1);
2075
2076 // Error reading macro name? If so, diagnostic already issued.
2077 if (MacroNameTok.getKind() == tok::eom)
2078 return;
2079
2080 // Check to see if this is the last token on the #undef line.
2081 CheckEndOfDirective("#define_other_target");
2082
2083 // If there is already a macro defined by this name, turn it into a
2084 // target-specific define.
Chris Lattnercc1a8752007-10-07 08:44:20 +00002085 if (MacroInfo *MI = getMacroInfo(MacroNameTok.getIdentifierInfo())) {
Reid Spencer5f016e22007-07-11 17:01:13 +00002086 MI->setIsTargetSpecific(true);
2087 return;
2088 }
2089
2090 // Mark the identifier as being a macro on some other target.
2091 MacroNameTok.getIdentifierInfo()->setIsOtherTargetMacro();
2092}
2093
2094
2095/// HandleUndefDirective - Implements #undef.
2096///
Chris Lattnerd2177732007-07-20 16:59:19 +00002097void Preprocessor::HandleUndefDirective(Token &UndefTok) {
Reid Spencer5f016e22007-07-11 17:01:13 +00002098 ++NumUndefined;
2099
Chris Lattnerd2177732007-07-20 16:59:19 +00002100 Token MacroNameTok;
Reid Spencer5f016e22007-07-11 17:01:13 +00002101 ReadMacroName(MacroNameTok, 2);
2102
2103 // Error reading macro name? If so, diagnostic already issued.
2104 if (MacroNameTok.getKind() == tok::eom)
2105 return;
2106
2107 // Check to see if this is the last token on the #undef line.
2108 CheckEndOfDirective("#undef");
2109
2110 // Okay, we finally have a valid identifier to undef.
Chris Lattnercc1a8752007-10-07 08:44:20 +00002111 MacroInfo *MI = getMacroInfo(MacroNameTok.getIdentifierInfo());
Reid Spencer5f016e22007-07-11 17:01:13 +00002112
2113 // #undef untaints an identifier if it were marked by define_other_target.
2114 MacroNameTok.getIdentifierInfo()->setIsOtherTargetMacro(false);
2115
2116 // If the macro is not defined, this is a noop undef, just return.
2117 if (MI == 0) return;
2118
2119 if (!MI->isUsed())
2120 Diag(MI->getDefinitionLoc(), diag::pp_macro_not_used);
2121
2122 // Free macro definition.
2123 delete MI;
Chris Lattnercc1a8752007-10-07 08:44:20 +00002124 setMacroInfo(MacroNameTok.getIdentifierInfo(), 0);
Reid Spencer5f016e22007-07-11 17:01:13 +00002125}
2126
2127
2128//===----------------------------------------------------------------------===//
2129// Preprocessor Conditional Directive Handling.
2130//===----------------------------------------------------------------------===//
2131
2132/// HandleIfdefDirective - Implements the #ifdef/#ifndef directive. isIfndef is
2133/// true when this is a #ifndef directive. ReadAnyTokensBeforeDirective is true
2134/// if any tokens have been returned or pp-directives activated before this
2135/// #ifndef has been lexed.
2136///
Chris Lattnerd2177732007-07-20 16:59:19 +00002137void Preprocessor::HandleIfdefDirective(Token &Result, bool isIfndef,
Reid Spencer5f016e22007-07-11 17:01:13 +00002138 bool ReadAnyTokensBeforeDirective) {
2139 ++NumIf;
Chris Lattnerd2177732007-07-20 16:59:19 +00002140 Token DirectiveTok = Result;
Reid Spencer5f016e22007-07-11 17:01:13 +00002141
Chris Lattnerd2177732007-07-20 16:59:19 +00002142 Token MacroNameTok;
Reid Spencer5f016e22007-07-11 17:01:13 +00002143 ReadMacroName(MacroNameTok);
2144
2145 // Error reading macro name? If so, diagnostic already issued.
Chris Lattnerf37bb252007-09-24 05:14:57 +00002146 if (MacroNameTok.getKind() == tok::eom) {
2147 // Skip code until we get to #endif. This helps with recovery by not
2148 // emitting an error when the #endif is reached.
2149 SkipExcludedConditionalBlock(DirectiveTok.getLocation(),
2150 /*Foundnonskip*/false, /*FoundElse*/false);
Reid Spencer5f016e22007-07-11 17:01:13 +00002151 return;
Chris Lattnerf37bb252007-09-24 05:14:57 +00002152 }
Reid Spencer5f016e22007-07-11 17:01:13 +00002153
2154 // Check to see if this is the last token on the #if[n]def line.
2155 CheckEndOfDirective(isIfndef ? "#ifndef" : "#ifdef");
2156
2157 // If the start of a top-level #ifdef, inform MIOpt.
2158 if (!ReadAnyTokensBeforeDirective &&
2159 CurLexer->getConditionalStackDepth() == 0) {
2160 assert(isIfndef && "#ifdef shouldn't reach here");
2161 CurLexer->MIOpt.EnterTopLevelIFNDEF(MacroNameTok.getIdentifierInfo());
2162 }
2163
2164 IdentifierInfo *MII = MacroNameTok.getIdentifierInfo();
Chris Lattnercc1a8752007-10-07 08:44:20 +00002165 MacroInfo *MI = getMacroInfo(MII);
Reid Spencer5f016e22007-07-11 17:01:13 +00002166
2167 // If there is a macro, process it.
2168 if (MI) {
2169 // Mark it used.
2170 MI->setIsUsed(true);
2171
2172 // If this is the first use of a target-specific macro, warn about it.
2173 if (MI->isTargetSpecific()) {
2174 MI->setIsTargetSpecific(false); // Don't warn on second use.
2175 getTargetInfo().DiagnoseNonPortability(MacroNameTok.getLocation(),
2176 diag::port_target_macro_use);
2177 }
2178 } else {
2179 // Use of a target-specific macro for some other target? If so, warn.
2180 if (MII->isOtherTargetMacro()) {
2181 MII->setIsOtherTargetMacro(false); // Don't warn on second use.
2182 getTargetInfo().DiagnoseNonPortability(MacroNameTok.getLocation(),
2183 diag::port_target_macro_use);
2184 }
2185 }
2186
2187 // Should we include the stuff contained by this directive?
2188 if (!MI == isIfndef) {
2189 // Yes, remember that we are inside a conditional, then lex the next token.
2190 CurLexer->pushConditionalLevel(DirectiveTok.getLocation(), /*wasskip*/false,
2191 /*foundnonskip*/true, /*foundelse*/false);
2192 } else {
2193 // No, skip the contents of this block and return the first token after it.
2194 SkipExcludedConditionalBlock(DirectiveTok.getLocation(),
2195 /*Foundnonskip*/false,
2196 /*FoundElse*/false);
2197 }
2198}
2199
2200/// HandleIfDirective - Implements the #if directive.
2201///
Chris Lattnerd2177732007-07-20 16:59:19 +00002202void Preprocessor::HandleIfDirective(Token &IfToken,
Reid Spencer5f016e22007-07-11 17:01:13 +00002203 bool ReadAnyTokensBeforeDirective) {
2204 ++NumIf;
2205
2206 // Parse and evaluation the conditional expression.
2207 IdentifierInfo *IfNDefMacro = 0;
2208 bool ConditionalTrue = EvaluateDirectiveExpression(IfNDefMacro);
2209
2210 // Should we include the stuff contained by this directive?
2211 if (ConditionalTrue) {
2212 // If this condition is equivalent to #ifndef X, and if this is the first
2213 // directive seen, handle it for the multiple-include optimization.
2214 if (!ReadAnyTokensBeforeDirective &&
2215 CurLexer->getConditionalStackDepth() == 0 && IfNDefMacro)
2216 CurLexer->MIOpt.EnterTopLevelIFNDEF(IfNDefMacro);
2217
2218 // Yes, remember that we are inside a conditional, then lex the next token.
2219 CurLexer->pushConditionalLevel(IfToken.getLocation(), /*wasskip*/false,
2220 /*foundnonskip*/true, /*foundelse*/false);
2221 } else {
2222 // No, skip the contents of this block and return the first token after it.
2223 SkipExcludedConditionalBlock(IfToken.getLocation(), /*Foundnonskip*/false,
2224 /*FoundElse*/false);
2225 }
2226}
2227
2228/// HandleEndifDirective - Implements the #endif directive.
2229///
Chris Lattnerd2177732007-07-20 16:59:19 +00002230void Preprocessor::HandleEndifDirective(Token &EndifToken) {
Reid Spencer5f016e22007-07-11 17:01:13 +00002231 ++NumEndif;
2232
2233 // Check that this is the whole directive.
2234 CheckEndOfDirective("#endif");
2235
2236 PPConditionalInfo CondInfo;
2237 if (CurLexer->popConditionalLevel(CondInfo)) {
2238 // No conditionals on the stack: this is an #endif without an #if.
2239 return Diag(EndifToken, diag::err_pp_endif_without_if);
2240 }
2241
2242 // If this the end of a top-level #endif, inform MIOpt.
2243 if (CurLexer->getConditionalStackDepth() == 0)
2244 CurLexer->MIOpt.ExitTopLevelConditional();
2245
2246 assert(!CondInfo.WasSkipping && !CurLexer->LexingRawMode &&
2247 "This code should only be reachable in the non-skipping case!");
2248}
2249
2250
Chris Lattnerd2177732007-07-20 16:59:19 +00002251void Preprocessor::HandleElseDirective(Token &Result) {
Reid Spencer5f016e22007-07-11 17:01:13 +00002252 ++NumElse;
2253
2254 // #else directive in a non-skipping conditional... start skipping.
2255 CheckEndOfDirective("#else");
2256
2257 PPConditionalInfo CI;
2258 if (CurLexer->popConditionalLevel(CI))
2259 return Diag(Result, diag::pp_err_else_without_if);
2260
2261 // If this is a top-level #else, inform the MIOpt.
2262 if (CurLexer->getConditionalStackDepth() == 0)
2263 CurLexer->MIOpt.FoundTopLevelElse();
2264
2265 // If this is a #else with a #else before it, report the error.
2266 if (CI.FoundElse) Diag(Result, diag::pp_err_else_after_else);
2267
2268 // Finally, skip the rest of the contents of this block and return the first
2269 // token after it.
2270 return SkipExcludedConditionalBlock(CI.IfLoc, /*Foundnonskip*/true,
2271 /*FoundElse*/true);
2272}
2273
Chris Lattnerd2177732007-07-20 16:59:19 +00002274void Preprocessor::HandleElifDirective(Token &ElifToken) {
Reid Spencer5f016e22007-07-11 17:01:13 +00002275 ++NumElse;
2276
2277 // #elif directive in a non-skipping conditional... start skipping.
2278 // We don't care what the condition is, because we will always skip it (since
2279 // the block immediately before it was included).
2280 DiscardUntilEndOfDirective();
2281
2282 PPConditionalInfo CI;
2283 if (CurLexer->popConditionalLevel(CI))
2284 return Diag(ElifToken, diag::pp_err_elif_without_if);
2285
2286 // If this is a top-level #elif, inform the MIOpt.
2287 if (CurLexer->getConditionalStackDepth() == 0)
2288 CurLexer->MIOpt.FoundTopLevelElse();
2289
2290 // If this is a #elif with a #else before it, report the error.
2291 if (CI.FoundElse) Diag(ElifToken, diag::pp_err_elif_after_else);
2292
2293 // Finally, skip the rest of the contents of this block and return the first
2294 // token after it.
2295 return SkipExcludedConditionalBlock(CI.IfLoc, /*Foundnonskip*/true,
2296 /*FoundElse*/CI.FoundElse);
2297}
2298