blob: db37fe15f384c0f599b0e219f37f58d8c4c80a2c [file] [log] [blame]
Chris Lattner1543e9c2008-03-09 02:18:51 +00001//===--- TokenLexer.cpp - Lex from a token stream -------------------------===//
Reid Spencer5f016e22007-07-11 17:01:13 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Reid Spencer5f016e22007-07-11 17:01:13 +00007//
8//===----------------------------------------------------------------------===//
9//
Chris Lattner1543e9c2008-03-09 02:18:51 +000010// This file implements the TokenLexer interface.
Reid Spencer5f016e22007-07-11 17:01:13 +000011//
12//===----------------------------------------------------------------------===//
13
Chris Lattner5d75de02008-03-09 02:22:57 +000014#include "clang/Lex/TokenLexer.h"
Chris Lattnere5c8ffe2008-03-09 02:55:12 +000015#include "MacroArgs.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000016#include "clang/Lex/MacroInfo.h"
17#include "clang/Lex/Preprocessor.h"
18#include "clang/Basic/SourceManager.h"
Chris Lattner500d3292009-01-29 05:15:15 +000019#include "clang/Lex/LexDiagnostic.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000020#include "llvm/ADT/SmallVector.h"
21using namespace clang;
22
Reid Spencer5f016e22007-07-11 17:01:13 +000023
Chris Lattner1543e9c2008-03-09 02:18:51 +000024/// Create a TokenLexer for the specified macro with the specified actual
Reid Spencer5f016e22007-07-11 17:01:13 +000025/// arguments. Note that this ctor takes ownership of the ActualArgs pointer.
Chris Lattnere7fb4842009-02-15 20:52:18 +000026void TokenLexer::Init(Token &Tok, SourceLocation ILEnd, MacroArgs *Actuals) {
Chris Lattner1543e9c2008-03-09 02:18:51 +000027 // If the client is reusing a TokenLexer, make sure to free any memory
Chris Lattner9594acf2007-07-15 00:25:26 +000028 // associated with it.
29 destroy();
Mike Stump1eb44332009-09-09 15:08:12 +000030
Chris Lattnercc1a8752007-10-07 08:44:20 +000031 Macro = PP.getMacroInfo(Tok.getIdentifierInfo());
Chris Lattner9594acf2007-07-15 00:25:26 +000032 ActualArgs = Actuals;
33 CurToken = 0;
Mike Stump1eb44332009-09-09 15:08:12 +000034
Chris Lattnere7fb4842009-02-15 20:52:18 +000035 InstantiateLocStart = Tok.getLocation();
36 InstantiateLocEnd = ILEnd;
Chris Lattner9594acf2007-07-15 00:25:26 +000037 AtStartOfLine = Tok.isAtStartOfLine();
38 HasLeadingSpace = Tok.hasLeadingSpace();
Chris Lattner8d896432008-03-09 02:07:49 +000039 Tokens = &*Macro->tokens_begin();
40 OwnsTokens = false;
Chris Lattner6b884502008-03-10 06:06:04 +000041 DisableMacroExpansion = false;
Chris Lattner8d896432008-03-09 02:07:49 +000042 NumTokens = Macro->tokens_end()-Macro->tokens_begin();
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +000043 MacroExpansionStart = SourceLocation();
44
45 SourceManager &SM = PP.getSourceManager();
46 MacroStartSLocOffset = SM.getNextOffset();
47
48 if (NumTokens > 0) {
49 assert(Tokens[0].getLocation().isValid());
50 assert((Tokens[0].getLocation().isFileID() || Tokens[0].is(tok::comment)) &&
51 "Macro defined in macro?");
52 assert(InstantiateLocStart.isValid());
53
54 // Reserve a source location entry chunk for the length of the macro
55 // definition. Tokens that get lexed directly from the definition will
56 // have their locations pointing inside this chunk. This is to avoid
57 // creating separate source location entries for each token.
58 SourceLocation macroStart = SM.getInstantiationLoc(Tokens[0].getLocation());
59 MacroDefStartInfo = SM.getDecomposedLoc(macroStart);
60 MacroExpansionStart = SM.createInstantiationLoc(macroStart,
61 InstantiateLocStart,
62 InstantiateLocEnd,
63 Macro->getDefinitionLength(SM));
64 }
Reid Spencer5f016e22007-07-11 17:01:13 +000065
66 // If this is a function-like macro, expand the arguments and change
Chris Lattner8d896432008-03-09 02:07:49 +000067 // Tokens to point to the expanded tokens.
Reid Spencer5f016e22007-07-11 17:01:13 +000068 if (Macro->isFunctionLike() && Macro->getNumArgs())
69 ExpandFunctionArguments();
Mike Stump1eb44332009-09-09 15:08:12 +000070
Reid Spencer5f016e22007-07-11 17:01:13 +000071 // Mark the macro as currently disabled, so that it is not recursively
72 // expanded. The macro must be disabled only after argument pre-expansion of
73 // function-like macro arguments occurs.
74 Macro->DisableMacro();
75}
76
Chris Lattner9594acf2007-07-15 00:25:26 +000077
78
Chris Lattner1543e9c2008-03-09 02:18:51 +000079/// Create a TokenLexer for the specified token stream. This does not
Reid Spencer5f016e22007-07-11 17:01:13 +000080/// take ownership of the specified token vector.
Chris Lattner6b884502008-03-10 06:06:04 +000081void TokenLexer::Init(const Token *TokArray, unsigned NumToks,
82 bool disableMacroExpansion, bool ownsTokens) {
Chris Lattner1543e9c2008-03-09 02:18:51 +000083 // If the client is reusing a TokenLexer, make sure to free any memory
Chris Lattner9594acf2007-07-15 00:25:26 +000084 // associated with it.
85 destroy();
Mike Stump1eb44332009-09-09 15:08:12 +000086
Chris Lattner9594acf2007-07-15 00:25:26 +000087 Macro = 0;
88 ActualArgs = 0;
Chris Lattner8d896432008-03-09 02:07:49 +000089 Tokens = TokArray;
Chris Lattner6b884502008-03-10 06:06:04 +000090 OwnsTokens = ownsTokens;
91 DisableMacroExpansion = disableMacroExpansion;
Chris Lattner8d896432008-03-09 02:07:49 +000092 NumTokens = NumToks;
Chris Lattner9594acf2007-07-15 00:25:26 +000093 CurToken = 0;
Chris Lattnere7fb4842009-02-15 20:52:18 +000094 InstantiateLocStart = InstantiateLocEnd = SourceLocation();
Chris Lattner9594acf2007-07-15 00:25:26 +000095 AtStartOfLine = false;
96 HasLeadingSpace = false;
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +000097 MacroExpansionStart = SourceLocation();
Mike Stump1eb44332009-09-09 15:08:12 +000098
Reid Spencer5f016e22007-07-11 17:01:13 +000099 // Set HasLeadingSpace/AtStartOfLine so that the first token will be
100 // returned unmodified.
101 if (NumToks != 0) {
102 AtStartOfLine = TokArray[0].isAtStartOfLine();
103 HasLeadingSpace = TokArray[0].hasLeadingSpace();
104 }
105}
106
107
Chris Lattner1543e9c2008-03-09 02:18:51 +0000108void TokenLexer::destroy() {
Reid Spencer5f016e22007-07-11 17:01:13 +0000109 // If this was a function-like macro that actually uses its arguments, delete
110 // the expanded tokens.
Chris Lattner8d896432008-03-09 02:07:49 +0000111 if (OwnsTokens) {
112 delete [] Tokens;
113 Tokens = 0;
Chris Lattner32fca722009-03-04 06:50:57 +0000114 OwnsTokens = false;
Chris Lattner9c683062007-07-22 01:16:55 +0000115 }
Mike Stump1eb44332009-09-09 15:08:12 +0000116
Chris Lattner1543e9c2008-03-09 02:18:51 +0000117 // TokenLexer owns its formal arguments.
Chris Lattner561395b2009-12-14 22:12:52 +0000118 if (ActualArgs) ActualArgs->destroy(PP);
Reid Spencer5f016e22007-07-11 17:01:13 +0000119}
120
121/// Expand the arguments of a function-like macro so that we can quickly
Chris Lattner8d896432008-03-09 02:07:49 +0000122/// return preexpanded tokens from Tokens.
Chris Lattner1543e9c2008-03-09 02:18:51 +0000123void TokenLexer::ExpandFunctionArguments() {
Chris Lattnerd2177732007-07-20 16:59:19 +0000124 llvm::SmallVector<Token, 128> ResultToks;
Mike Stump1eb44332009-09-09 15:08:12 +0000125
Chris Lattner8d896432008-03-09 02:07:49 +0000126 // Loop through 'Tokens', expanding them into ResultToks. Keep
Reid Spencer5f016e22007-07-11 17:01:13 +0000127 // track of whether we change anything. If not, no need to keep them. If so,
Chris Lattner8d896432008-03-09 02:07:49 +0000128 // we install the newly expanded sequence as the new 'Tokens' list.
Reid Spencer5f016e22007-07-11 17:01:13 +0000129 bool MadeChange = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000130
Reid Spencer5f016e22007-07-11 17:01:13 +0000131 // NextTokGetsSpace - When this is true, the next token appended to the
132 // output list will get a leading space, regardless of whether it had one to
133 // begin with or not. This is used for placemarker support.
134 bool NextTokGetsSpace = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000135
Chris Lattner8d896432008-03-09 02:07:49 +0000136 for (unsigned i = 0, e = NumTokens; i != e; ++i) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000137 // If we found the stringify operator, get the argument stringified. The
138 // preprocessor already verified that the following token is a macro name
139 // when the #define was parsed.
Chris Lattner8d896432008-03-09 02:07:49 +0000140 const Token &CurTok = Tokens[i];
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000141 if (CurTok.is(tok::hash) || CurTok.is(tok::hashat)) {
Chris Lattner8d896432008-03-09 02:07:49 +0000142 int ArgNo = Macro->getArgumentNum(Tokens[i+1].getIdentifierInfo());
Reid Spencer5f016e22007-07-11 17:01:13 +0000143 assert(ArgNo != -1 && "Token following # is not an argument?");
Mike Stump1eb44332009-09-09 15:08:12 +0000144
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000145 SourceLocation hashInstLoc;
146 if(InstantiateLocStart.isValid()) {
147 hashInstLoc = getMacroExpansionLocation(CurTok.getLocation());
148 assert(hashInstLoc.isValid() && "Expected '#' to come from definition");
149 }
150
Chris Lattnerd2177732007-07-20 16:59:19 +0000151 Token Res;
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000152 if (CurTok.is(tok::hash)) // Stringify
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000153 Res = ActualArgs->getStringifiedArgument(ArgNo, PP, hashInstLoc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000154 else {
155 // 'charify': don't bother caching these.
Chris Lattnere5c8ffe2008-03-09 02:55:12 +0000156 Res = MacroArgs::StringifyArgument(ActualArgs->getUnexpArgument(ArgNo),
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000157 PP, true, hashInstLoc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000158 }
Mike Stump1eb44332009-09-09 15:08:12 +0000159
Reid Spencer5f016e22007-07-11 17:01:13 +0000160 // The stringified/charified string leading space flag gets set to match
161 // the #/#@ operator.
162 if (CurTok.hasLeadingSpace() || NextTokGetsSpace)
Chris Lattnerd2177732007-07-20 16:59:19 +0000163 Res.setFlag(Token::LeadingSpace);
Mike Stump1eb44332009-09-09 15:08:12 +0000164
Reid Spencer5f016e22007-07-11 17:01:13 +0000165 ResultToks.push_back(Res);
166 MadeChange = true;
167 ++i; // Skip arg name.
168 NextTokGetsSpace = false;
169 continue;
170 }
Mike Stump1eb44332009-09-09 15:08:12 +0000171
Reid Spencer5f016e22007-07-11 17:01:13 +0000172 // Otherwise, if this is not an argument token, just add the token to the
173 // output buffer.
174 IdentifierInfo *II = CurTok.getIdentifierInfo();
175 int ArgNo = II ? Macro->getArgumentNum(II) : -1;
176 if (ArgNo == -1) {
177 // This isn't an argument, just add it.
178 ResultToks.push_back(CurTok);
179
180 if (NextTokGetsSpace) {
Chris Lattnerd2177732007-07-20 16:59:19 +0000181 ResultToks.back().setFlag(Token::LeadingSpace);
Reid Spencer5f016e22007-07-11 17:01:13 +0000182 NextTokGetsSpace = false;
183 }
184 continue;
185 }
Mike Stump1eb44332009-09-09 15:08:12 +0000186
Reid Spencer5f016e22007-07-11 17:01:13 +0000187 // An argument is expanded somehow, the result is different than the
188 // input.
189 MadeChange = true;
190
191 // Otherwise, this is a use of the argument. Find out if there is a paste
192 // (##) operator before or after the argument.
Mike Stump1eb44332009-09-09 15:08:12 +0000193 bool PasteBefore =
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000194 !ResultToks.empty() && ResultToks.back().is(tok::hashhash);
Chris Lattner8d896432008-03-09 02:07:49 +0000195 bool PasteAfter = i+1 != e && Tokens[i+1].is(tok::hashhash);
Mike Stump1eb44332009-09-09 15:08:12 +0000196
Reid Spencer5f016e22007-07-11 17:01:13 +0000197 // If it is not the LHS/RHS of a ## operator, we must pre-expand the
198 // argument and substitute the expanded tokens into the result. This is
199 // C99 6.10.3.1p1.
200 if (!PasteBefore && !PasteAfter) {
Chris Lattnerd2177732007-07-20 16:59:19 +0000201 const Token *ResultArgToks;
Reid Spencer5f016e22007-07-11 17:01:13 +0000202
203 // Only preexpand the argument if it could possibly need it. This
204 // avoids some work in common cases.
Chris Lattnerd2177732007-07-20 16:59:19 +0000205 const Token *ArgTok = ActualArgs->getUnexpArgument(ArgNo);
Chris Lattnercc1a8752007-10-07 08:44:20 +0000206 if (ActualArgs->ArgNeedsPreexpansion(ArgTok, PP))
Chris Lattnerf5809a72009-12-28 06:17:16 +0000207 ResultArgToks = &ActualArgs->getPreExpArgument(ArgNo, Macro, PP)[0];
Reid Spencer5f016e22007-07-11 17:01:13 +0000208 else
209 ResultArgToks = ArgTok; // Use non-preexpanded tokens.
Mike Stump1eb44332009-09-09 15:08:12 +0000210
Reid Spencer5f016e22007-07-11 17:01:13 +0000211 // If the arg token expanded into anything, append it.
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000212 if (ResultArgToks->isNot(tok::eof)) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000213 unsigned FirstResult = ResultToks.size();
214 unsigned NumToks = MacroArgs::getArgLength(ResultArgToks);
215 ResultToks.append(ResultArgToks, ResultArgToks+NumToks);
Mike Stump1eb44332009-09-09 15:08:12 +0000216
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000217 if(InstantiateLocStart.isValid()) {
218 SourceManager &SM = PP.getSourceManager();
219 SourceLocation curInst =
220 getMacroExpansionLocation(CurTok.getLocation());
221 assert(curInst.isValid() &&
222 "Expected arg identifier to come from definition");
223 for (unsigned i = FirstResult, e = ResultToks.size(); i != e; ++i) {
224 Token &Tok = ResultToks[i];
225 Tok.setLocation(SM.createInstantiationLoc(Tok.getLocation(),
226 curInst, curInst,
227 Tok.getLength()));
228 }
229 }
230
Reid Spencer5f016e22007-07-11 17:01:13 +0000231 // If any tokens were substituted from the argument, the whitespace
232 // before the first token should match the whitespace of the arg
233 // identifier.
Chris Lattnerd2177732007-07-20 16:59:19 +0000234 ResultToks[FirstResult].setFlagValue(Token::LeadingSpace,
Reid Spencer5f016e22007-07-11 17:01:13 +0000235 CurTok.hasLeadingSpace() ||
236 NextTokGetsSpace);
237 NextTokGetsSpace = false;
238 } else {
239 // If this is an empty argument, and if there was whitespace before the
240 // formal token, make sure the next token gets whitespace before it.
241 NextTokGetsSpace = CurTok.hasLeadingSpace();
242 }
243 continue;
244 }
Mike Stump1eb44332009-09-09 15:08:12 +0000245
Reid Spencer5f016e22007-07-11 17:01:13 +0000246 // Okay, we have a token that is either the LHS or RHS of a paste (##)
247 // argument. It gets substituted as its non-pre-expanded tokens.
Chris Lattnerd2177732007-07-20 16:59:19 +0000248 const Token *ArgToks = ActualArgs->getUnexpArgument(ArgNo);
Reid Spencer5f016e22007-07-11 17:01:13 +0000249 unsigned NumToks = MacroArgs::getArgLength(ArgToks);
250 if (NumToks) { // Not an empty argument?
Chris Lattner71a3a8d2008-01-29 07:54:23 +0000251 // If this is the GNU ", ## __VA_ARG__" extension, and we just learned
252 // that __VA_ARG__ expands to multiple tokens, avoid a pasting error when
253 // the expander trys to paste ',' with the first token of the __VA_ARG__
254 // expansion.
255 if (PasteBefore && ResultToks.size() >= 2 &&
256 ResultToks[ResultToks.size()-2].is(tok::comma) &&
257 (unsigned)ArgNo == Macro->getNumArgs()-1 &&
258 Macro->isVariadic()) {
259 // Remove the paste operator, report use of the extension.
260 PP.Diag(ResultToks.back().getLocation(), diag::ext_paste_comma);
261 ResultToks.pop_back();
262 }
Mike Stump1eb44332009-09-09 15:08:12 +0000263
Reid Spencer5f016e22007-07-11 17:01:13 +0000264 ResultToks.append(ArgToks, ArgToks+NumToks);
Mike Stump1eb44332009-09-09 15:08:12 +0000265
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000266 if(InstantiateLocStart.isValid()) {
267 SourceManager &SM = PP.getSourceManager();
268 SourceLocation curInst =
269 getMacroExpansionLocation(CurTok.getLocation());
270 assert(curInst.isValid() &&
271 "Expected arg identifier to come from definition");
272 for (unsigned i = ResultToks.size() - NumToks, e = ResultToks.size();
273 i != e; ++i) {
274 Token &Tok = ResultToks[i];
275 Tok.setLocation(SM.createInstantiationLoc(Tok.getLocation(),
276 curInst, curInst,
277 Tok.getLength()));
278 }
279 }
280
Chris Lattner442a6612009-05-25 16:23:08 +0000281 // If this token (the macro argument) was supposed to get leading
282 // whitespace, transfer this information onto the first token of the
283 // expansion.
284 //
285 // Do not do this if the paste operator occurs before the macro argument,
286 // as in "A ## MACROARG". In valid code, the first token will get
287 // smooshed onto the preceding one anyway (forming AMACROARG). In
288 // assembler-with-cpp mode, invalid pastes are allowed through: in this
289 // case, we do not want the extra whitespace to be added. For example,
290 // we want ". ## foo" -> ".foo" not ". foo".
291 if ((CurTok.hasLeadingSpace() || NextTokGetsSpace) &&
292 !PasteBefore)
293 ResultToks[ResultToks.size()-NumToks].setFlag(Token::LeadingSpace);
Mike Stump1eb44332009-09-09 15:08:12 +0000294
Chris Lattner442a6612009-05-25 16:23:08 +0000295 NextTokGetsSpace = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000296 continue;
297 }
Mike Stump1eb44332009-09-09 15:08:12 +0000298
Reid Spencer5f016e22007-07-11 17:01:13 +0000299 // If an empty argument is on the LHS or RHS of a paste, the standard (C99
300 // 6.10.3.3p2,3) calls for a bunch of placemarker stuff to occur. We
301 // implement this by eating ## operators when a LHS or RHS expands to
302 // empty.
303 NextTokGetsSpace |= CurTok.hasLeadingSpace();
304 if (PasteAfter) {
305 // Discard the argument token and skip (don't copy to the expansion
306 // buffer) the paste operator after it.
Chris Lattner8d896432008-03-09 02:07:49 +0000307 NextTokGetsSpace |= Tokens[i+1].hasLeadingSpace();
Reid Spencer5f016e22007-07-11 17:01:13 +0000308 ++i;
309 continue;
310 }
Mike Stump1eb44332009-09-09 15:08:12 +0000311
Reid Spencer5f016e22007-07-11 17:01:13 +0000312 // If this is on the RHS of a paste operator, we've already copied the
313 // paste operator to the ResultToks list. Remove it.
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000314 assert(PasteBefore && ResultToks.back().is(tok::hashhash));
Reid Spencer5f016e22007-07-11 17:01:13 +0000315 NextTokGetsSpace |= ResultToks.back().hasLeadingSpace();
316 ResultToks.pop_back();
Mike Stump1eb44332009-09-09 15:08:12 +0000317
Reid Spencer5f016e22007-07-11 17:01:13 +0000318 // If this is the __VA_ARGS__ token, and if the argument wasn't provided,
319 // and if the macro had at least one real argument, and if the token before
320 // the ## was a comma, remove the comma.
321 if ((unsigned)ArgNo == Macro->getNumArgs()-1 && // is __VA_ARGS__
322 ActualArgs->isVarargsElidedUse() && // Argument elided.
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000323 !ResultToks.empty() && ResultToks.back().is(tok::comma)) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000324 // Never add a space, even if the comma, ##, or arg had a space.
325 NextTokGetsSpace = false;
Chris Lattner71a3a8d2008-01-29 07:54:23 +0000326 // Remove the paste operator, report use of the extension.
327 PP.Diag(ResultToks.back().getLocation(), diag::ext_paste_comma);
Reid Spencer5f016e22007-07-11 17:01:13 +0000328 ResultToks.pop_back();
Chris Lattner496af572010-08-21 00:27:00 +0000329
330 // If the comma was right after another paste (e.g. "X##,##__VA_ARGS__"),
331 // then removal of the comma should produce a placemarker token (in C99
332 // terms) which we model by popping off the previous ##, giving us a plain
333 // "X" when __VA_ARGS__ is empty.
334 if (!ResultToks.empty() && ResultToks.back().is(tok::hashhash))
335 ResultToks.pop_back();
Reid Spencer5f016e22007-07-11 17:01:13 +0000336 }
337 continue;
338 }
Mike Stump1eb44332009-09-09 15:08:12 +0000339
Chris Lattner8d896432008-03-09 02:07:49 +0000340 // If anything changed, install this as the new Tokens list.
Reid Spencer5f016e22007-07-11 17:01:13 +0000341 if (MadeChange) {
Chris Lattner32fca722009-03-04 06:50:57 +0000342 assert(!OwnsTokens && "This would leak if we already own the token list");
Reid Spencer5f016e22007-07-11 17:01:13 +0000343 // This is deleted in the dtor.
Chris Lattner8d896432008-03-09 02:07:49 +0000344 NumTokens = ResultToks.size();
Argyrios Kyrtzidis5b3284a2011-06-29 22:20:11 +0000345 // The tokens will be added to Preprocessor's cache and will be removed
346 // when this TokenLexer finishes lexing them.
347 Tokens = PP.cacheMacroExpandedTokens(this, ResultToks);
Mike Stump1eb44332009-09-09 15:08:12 +0000348
Argyrios Kyrtzidis5b3284a2011-06-29 22:20:11 +0000349 // The preprocessor cache of macro expanded tokens owns these tokens,not us.
Chris Lattner32fca722009-03-04 06:50:57 +0000350 OwnsTokens = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000351 }
352}
353
354/// Lex - Lex and return a token from this macro stream.
355///
Chris Lattner1543e9c2008-03-09 02:18:51 +0000356void TokenLexer::Lex(Token &Tok) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000357 // Lexing off the end of the macro, pop this macro off the expansion stack.
358 if (isAtEnd()) {
359 // If this is a macro (not a token stream), mark the macro enabled now
360 // that it is no longer being expanded.
361 if (Macro) Macro->EnableMacro();
362
363 // Pop this context off the preprocessors lexer stack and get the next
364 // token. This will delete "this" so remember the PP instance var.
365 Preprocessor &PPCache = PP;
Chris Lattnerfde2bf92008-03-09 03:04:16 +0000366 if (PP.HandleEndOfTokenLexer(Tok))
Reid Spencer5f016e22007-07-11 17:01:13 +0000367 return;
368
Chris Lattnerfde2bf92008-03-09 03:04:16 +0000369 // HandleEndOfTokenLexer may not return a token. If it doesn't, lex
370 // whatever is next.
Reid Spencer5f016e22007-07-11 17:01:13 +0000371 return PPCache.Lex(Tok);
372 }
Mike Stump1eb44332009-09-09 15:08:12 +0000373
Reid Spencer5f016e22007-07-11 17:01:13 +0000374 // If this is the first token of the expanded result, we inherit spacing
375 // properties later.
376 bool isFirstToken = CurToken == 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000377
Reid Spencer5f016e22007-07-11 17:01:13 +0000378 // Get the next token to return.
Chris Lattner8d896432008-03-09 02:07:49 +0000379 Tok = Tokens[CurToken++];
Mike Stump1eb44332009-09-09 15:08:12 +0000380
Chris Lattner332ee082009-04-19 20:29:42 +0000381 bool TokenIsFromPaste = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000382
Reid Spencer5f016e22007-07-11 17:01:13 +0000383 // If this token is followed by a token paste (##) operator, paste the tokens!
Chris Lattnerb11e43c2011-06-14 18:19:37 +0000384 // Note that ## is a normal token when not expanding a macro.
385 if (!isAtEnd() && Tokens[CurToken].is(tok::hashhash) && Macro) {
Chris Lattner04f2bd82009-12-04 06:14:03 +0000386 // When handling the microsoft /##/ extension, the final token is
387 // returned by PasteTokens, not the pasted token.
388 if (PasteTokens(Tok))
Chris Lattner3f1cc832008-02-07 06:03:59 +0000389 return;
Kovarththanan Rajaratnam65cc1e82010-03-13 08:53:33 +0000390
Chris Lattner04f2bd82009-12-04 06:14:03 +0000391 TokenIsFromPaste = true;
Mike Stump1eb44332009-09-09 15:08:12 +0000392 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000393
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000394 SourceManager &SM = PP.getSourceManager();
Reid Spencer5f016e22007-07-11 17:01:13 +0000395 // The token's current location indicate where the token was lexed from. We
396 // need this information to compute the spelling of the token, but any
397 // diagnostics for the expanded token should appear as if they came from
398 // InstantiationLoc. Pull this information together into a new SourceLocation
399 // that captures all of this.
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000400 if (InstantiateLocStart.isValid() && // Don't do this for token streams.
401 // Check that the token's location was not already set properly.
402 SM.isBeforeInSourceLocationOffset(Tok.getLocation(),
403 MacroStartSLocOffset)) {
404 SourceLocation instLoc;
405 if (Tok.is(tok::comment)) {
406 instLoc = SM.createInstantiationLoc(Tok.getLocation(),
407 InstantiateLocStart,
408 InstantiateLocEnd,
409 Tok.getLength());
410 } else {
411 instLoc = getMacroExpansionLocation(Tok.getLocation());
412 assert(instLoc.isValid() &&
413 "Location for token not coming from definition was not set!");
414 }
415
416 Tok.setLocation(instLoc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000417 }
Mike Stump1eb44332009-09-09 15:08:12 +0000418
Reid Spencer5f016e22007-07-11 17:01:13 +0000419 // If this is the first token, set the lexical properties of the token to
420 // match the lexical properties of the macro identifier.
421 if (isFirstToken) {
Chris Lattnerd2177732007-07-20 16:59:19 +0000422 Tok.setFlagValue(Token::StartOfLine , AtStartOfLine);
423 Tok.setFlagValue(Token::LeadingSpace, HasLeadingSpace);
Reid Spencer5f016e22007-07-11 17:01:13 +0000424 }
Mike Stump1eb44332009-09-09 15:08:12 +0000425
Reid Spencer5f016e22007-07-11 17:01:13 +0000426 // Handle recursive expansion!
Argyrios Kyrtzidis6f421f92009-05-22 21:09:31 +0000427 if (!Tok.isAnnotation() && Tok.getIdentifierInfo() != 0) {
Chris Lattner863c4862009-01-23 18:35:48 +0000428 // Change the kind of this identifier to the appropriate token kind, e.g.
429 // turning "for" into a keyword.
Argyrios Kyrtzidis6f421f92009-05-22 21:09:31 +0000430 IdentifierInfo *II = Tok.getIdentifierInfo();
Chris Lattner863c4862009-01-23 18:35:48 +0000431 Tok.setKind(II->getTokenID());
Mike Stump1eb44332009-09-09 15:08:12 +0000432
Chris Lattner332ee082009-04-19 20:29:42 +0000433 // If this identifier was poisoned and from a paste, emit an error. This
434 // won't be handled by Preprocessor::HandleIdentifier because this is coming
435 // from a macro expansion.
436 if (II->isPoisoned() && TokenIsFromPaste) {
John Wiegley28bbe4b2011-04-28 01:08:34 +0000437 PP.HandlePoisonedIdentifier(Tok);
Chris Lattner332ee082009-04-19 20:29:42 +0000438 }
Mike Stump1eb44332009-09-09 15:08:12 +0000439
Chris Lattner863c4862009-01-23 18:35:48 +0000440 if (!DisableMacroExpansion && II->isHandleIdentifierCase())
441 PP.HandleIdentifier(Tok);
442 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000443
444 // Otherwise, return a normal token.
445}
446
447/// PasteTokens - Tok is the LHS of a ## operator, and CurToken is the ##
448/// operator. Read the ## and RHS, and paste the LHS/RHS together. If there
Nico Weber48002c82008-09-29 00:25:48 +0000449/// are more ## after it, chomp them iteratively. Return the result as Tok.
Chris Lattner3f1cc832008-02-07 06:03:59 +0000450/// If this returns true, the caller should immediately return the token.
Chris Lattner1543e9c2008-03-09 02:18:51 +0000451bool TokenLexer::PasteTokens(Token &Tok) {
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000452 llvm::SmallString<128> Buffer;
Chris Lattner47246be2009-01-26 19:29:26 +0000453 const char *ResultTokStrPtr = 0;
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000454 SourceLocation PasteOpLoc;
Reid Spencer5f016e22007-07-11 17:01:13 +0000455 do {
456 // Consume the ## operator.
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000457 PasteOpLoc = Tokens[CurToken].getLocation();
Reid Spencer5f016e22007-07-11 17:01:13 +0000458 ++CurToken;
459 assert(!isAtEnd() && "No token on the RHS of a paste operator!");
Mike Stump1eb44332009-09-09 15:08:12 +0000460
Reid Spencer5f016e22007-07-11 17:01:13 +0000461 // Get the RHS token.
Chris Lattner8d896432008-03-09 02:07:49 +0000462 const Token &RHS = Tokens[CurToken];
Mike Stump1eb44332009-09-09 15:08:12 +0000463
Reid Spencer5f016e22007-07-11 17:01:13 +0000464 // Allocate space for the result token. This is guaranteed to be enough for
Chris Lattner1fa49532009-03-08 08:08:45 +0000465 // the two tokens.
466 Buffer.resize(Tok.getLength() + RHS.getLength());
Mike Stump1eb44332009-09-09 15:08:12 +0000467
Reid Spencer5f016e22007-07-11 17:01:13 +0000468 // Get the spelling of the LHS token in Buffer.
469 const char *BufPtr = &Buffer[0];
Douglas Gregor453091c2010-03-16 22:30:13 +0000470 bool Invalid = false;
471 unsigned LHSLen = PP.getSpelling(Tok, BufPtr, &Invalid);
Reid Spencer5f016e22007-07-11 17:01:13 +0000472 if (BufPtr != &Buffer[0]) // Really, we want the chars in Buffer!
473 memcpy(&Buffer[0], BufPtr, LHSLen);
Douglas Gregor453091c2010-03-16 22:30:13 +0000474 if (Invalid)
475 return true;
476
Reid Spencer5f016e22007-07-11 17:01:13 +0000477 BufPtr = &Buffer[LHSLen];
Douglas Gregor453091c2010-03-16 22:30:13 +0000478 unsigned RHSLen = PP.getSpelling(RHS, BufPtr, &Invalid);
479 if (Invalid)
480 return true;
Reid Spencer5f016e22007-07-11 17:01:13 +0000481 if (BufPtr != &Buffer[LHSLen]) // Really, we want the chars in Buffer!
482 memcpy(&Buffer[LHSLen], BufPtr, RHSLen);
Mike Stump1eb44332009-09-09 15:08:12 +0000483
Reid Spencer5f016e22007-07-11 17:01:13 +0000484 // Trim excess space.
Chris Lattner1fa49532009-03-08 08:08:45 +0000485 Buffer.resize(LHSLen+RHSLen);
Mike Stump1eb44332009-09-09 15:08:12 +0000486
Reid Spencer5f016e22007-07-11 17:01:13 +0000487 // Plop the pasted result (including the trailing newline and null) into a
488 // scratch buffer where we can lex it.
Chris Lattner47246be2009-01-26 19:29:26 +0000489 Token ResultTokTmp;
490 ResultTokTmp.startToken();
Mike Stump1eb44332009-09-09 15:08:12 +0000491
Chris Lattner47246be2009-01-26 19:29:26 +0000492 // Claim that the tmp token is a string_literal so that we can get the
Chris Lattner917f1432009-12-23 21:29:53 +0000493 // character pointer back from CreateString in getLiteralData().
Chris Lattner47246be2009-01-26 19:29:26 +0000494 ResultTokTmp.setKind(tok::string_literal);
495 PP.CreateString(&Buffer[0], Buffer.size(), ResultTokTmp);
496 SourceLocation ResultTokLoc = ResultTokTmp.getLocation();
497 ResultTokStrPtr = ResultTokTmp.getLiteralData();
498
Reid Spencer5f016e22007-07-11 17:01:13 +0000499 // Lex the resultant pasted token into Result.
Chris Lattnerd2177732007-07-20 16:59:19 +0000500 Token Result;
Mike Stump1eb44332009-09-09 15:08:12 +0000501
Abramo Bagnarac4bf2b92010-12-22 08:23:18 +0000502 if (Tok.isAnyIdentifier() && RHS.isAnyIdentifier()) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000503 // Common paste case: identifier+identifier = identifier. Avoid creating
504 // a lexer and other overhead.
505 PP.IncrementPasteCounter(true);
506 Result.startToken();
Abramo Bagnarac4bf2b92010-12-22 08:23:18 +0000507 Result.setKind(tok::raw_identifier);
508 Result.setRawIdentifierData(ResultTokStrPtr);
Reid Spencer5f016e22007-07-11 17:01:13 +0000509 Result.setLocation(ResultTokLoc);
510 Result.setLength(LHSLen+RHSLen);
511 } else {
512 PP.IncrementPasteCounter(false);
Mike Stump1eb44332009-09-09 15:08:12 +0000513
Chris Lattnerbcc2a672009-01-19 06:46:35 +0000514 assert(ResultTokLoc.isFileID() &&
515 "Should be a raw location into scratch buffer");
Reid Spencer5f016e22007-07-11 17:01:13 +0000516 SourceManager &SourceMgr = PP.getSourceManager();
Chris Lattner47246be2009-01-26 19:29:26 +0000517 FileID LocFileID = SourceMgr.getFileID(ResultTokLoc);
Mike Stump1eb44332009-09-09 15:08:12 +0000518
Douglas Gregorf715ca12010-03-16 00:06:06 +0000519 bool Invalid = false;
Douglas Gregoraea67db2010-03-15 22:54:52 +0000520 const char *ScratchBufStart
Benjamin Kramerf6ac97b2010-03-16 14:14:31 +0000521 = SourceMgr.getBufferData(LocFileID, &Invalid).data();
Douglas Gregorf715ca12010-03-16 00:06:06 +0000522 if (Invalid)
Douglas Gregoraea67db2010-03-15 22:54:52 +0000523 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000524
Chris Lattnerbcc2a672009-01-19 06:46:35 +0000525 // Make a lexer to lex this string from. Lex just this one token.
Reid Spencer5f016e22007-07-11 17:01:13 +0000526 // Make a lexer object so that we lex and expand the paste result.
Chris Lattner47246be2009-01-26 19:29:26 +0000527 Lexer TL(SourceMgr.getLocForStartOfFile(LocFileID),
528 PP.getLangOptions(), ScratchBufStart,
Chris Lattner1fa49532009-03-08 08:08:45 +0000529 ResultTokStrPtr, ResultTokStrPtr+LHSLen+RHSLen);
Mike Stump1eb44332009-09-09 15:08:12 +0000530
Reid Spencer5f016e22007-07-11 17:01:13 +0000531 // Lex a token in raw mode. This way it won't look up identifiers
532 // automatically, lexing off the end will return an eof token, and
533 // warnings are disabled. This returns true if the result token is the
534 // entire buffer.
Chris Lattner1fa49532009-03-08 08:08:45 +0000535 bool isInvalid = !TL.LexFromRawLexer(Result);
Mike Stump1eb44332009-09-09 15:08:12 +0000536
Reid Spencer5f016e22007-07-11 17:01:13 +0000537 // If we got an EOF token, we didn't form even ONE token. For example, we
538 // did "/ ## /" to get "//".
Chris Lattner1fa49532009-03-08 08:08:45 +0000539 isInvalid |= Result.is(tok::eof);
Mike Stump1eb44332009-09-09 15:08:12 +0000540
Chris Lattner1fa49532009-03-08 08:08:45 +0000541 // If pasting the two tokens didn't form a full new token, this is an
542 // error. This occurs with "x ## +" and other stuff. Return with Tok
543 // unmodified and with RHS as the next token to lex.
544 if (isInvalid) {
545 // Test for the Microsoft extension of /##/ turning into // here on the
546 // error path.
Mike Stump1eb44332009-09-09 15:08:12 +0000547 if (PP.getLangOptions().Microsoft && Tok.is(tok::slash) &&
Chris Lattner1fa49532009-03-08 08:08:45 +0000548 RHS.is(tok::slash)) {
549 HandleMicrosoftCommentPaste(Tok);
550 return true;
551 }
Mike Stump1eb44332009-09-09 15:08:12 +0000552
Chris Lattner920bf452010-07-17 16:24:30 +0000553 // Do not emit the error when preprocessing assembler code.
Chris Lattner3c17b8a2009-05-28 05:39:39 +0000554 if (!PP.getLangOptions().AsmPreprocessor) {
555 // Explicitly convert the token location to have proper instantiation
556 // information so that the user knows where it came from.
557 SourceManager &SM = PP.getSourceManager();
Chris Lattnerb11e43c2011-06-14 18:19:37 +0000558 SourceLocation Loc =
559 SM.createInstantiationLoc(PasteOpLoc, InstantiateLocStart,
560 InstantiateLocEnd, 2);
Chris Lattner920bf452010-07-17 16:24:30 +0000561 // If we're in microsoft extensions mode, downgrade this from a hard
562 // error to a warning that defaults to an error. This allows
563 // disabling it.
564 PP.Diag(Loc,
565 PP.getLangOptions().Microsoft ? diag::err_pp_bad_paste_ms
566 : diag::err_pp_bad_paste)
Benjamin Kramer476d8b82010-08-11 14:47:12 +0000567 << Buffer.str();
Chris Lattner3c17b8a2009-05-28 05:39:39 +0000568 }
Mike Stump1eb44332009-09-09 15:08:12 +0000569
Chris Lattnerf1538de2009-03-23 17:32:45 +0000570 // Do not consume the RHS.
571 --CurToken;
Chris Lattner1fa49532009-03-08 08:08:45 +0000572 }
Mike Stump1eb44332009-09-09 15:08:12 +0000573
Chris Lattner1fa49532009-03-08 08:08:45 +0000574 // Turn ## into 'unknown' to avoid # ## # from looking like a paste
575 // operator.
576 if (Result.is(tok::hashhash))
577 Result.setKind(tok::unknown);
Reid Spencer5f016e22007-07-11 17:01:13 +0000578 }
Mike Stump1eb44332009-09-09 15:08:12 +0000579
Reid Spencer5f016e22007-07-11 17:01:13 +0000580 // Transfer properties of the LHS over the the Result.
Chris Lattnerd2177732007-07-20 16:59:19 +0000581 Result.setFlagValue(Token::StartOfLine , Tok.isAtStartOfLine());
582 Result.setFlagValue(Token::LeadingSpace, Tok.hasLeadingSpace());
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000583
Reid Spencer5f016e22007-07-11 17:01:13 +0000584 // Finally, replace LHS with the result, consume the RHS, and iterate.
585 ++CurToken;
586 Tok = Result;
Chris Lattner8d896432008-03-09 02:07:49 +0000587 } while (!isAtEnd() && Tokens[CurToken].is(tok::hashhash));
Mike Stump1eb44332009-09-09 15:08:12 +0000588
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000589 // The token's current location indicate where the token was lexed from. We
590 // need this information to compute the spelling of the token, but any
591 // diagnostics for the expanded token should appear as if the token was
592 // instantiated from the (##) operator. Pull this information together into
593 // a new SourceLocation that captures all of this.
594 if (InstantiateLocStart.isValid()) {
595 SourceManager &SM = PP.getSourceManager();
596 SourceLocation pasteLocInst =
597 getMacroExpansionLocation(PasteOpLoc);
598 assert(pasteLocInst.isValid() &&
599 "Expected '##' to come from definition");
600
601 Tok.setLocation(SM.createInstantiationLoc(Tok.getLocation(),
602 pasteLocInst,
603 pasteLocInst,
604 Tok.getLength()));
605 }
606
Reid Spencer5f016e22007-07-11 17:01:13 +0000607 // Now that we got the result token, it will be subject to expansion. Since
608 // token pasting re-lexes the result token in raw mode, identifier information
609 // isn't looked up. As such, if the result is an identifier, look up id info.
Abramo Bagnarac4bf2b92010-12-22 08:23:18 +0000610 if (Tok.is(tok::raw_identifier)) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000611 // Look up the identifier info for the token. We disabled identifier lookup
612 // by saying we're skipping contents, so we need to do this manually.
Abramo Bagnarac4bf2b92010-12-22 08:23:18 +0000613 PP.LookUpIdentifierInfo(Tok);
Reid Spencer5f016e22007-07-11 17:01:13 +0000614 }
Chris Lattner3f1cc832008-02-07 06:03:59 +0000615 return false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000616}
617
618/// isNextTokenLParen - If the next token lexed will pop this macro off the
619/// expansion stack, return 2. If the next unexpanded token is a '(', return
620/// 1, otherwise return 0.
Chris Lattner1543e9c2008-03-09 02:18:51 +0000621unsigned TokenLexer::isNextTokenLParen() const {
Reid Spencer5f016e22007-07-11 17:01:13 +0000622 // Out of tokens?
623 if (isAtEnd())
624 return 2;
Chris Lattner8d896432008-03-09 02:07:49 +0000625 return Tokens[CurToken].is(tok::l_paren);
Reid Spencer5f016e22007-07-11 17:01:13 +0000626}
Chris Lattner3f1cc832008-02-07 06:03:59 +0000627
Peter Collingbourneb2eb53d2011-02-22 13:49:00 +0000628/// isParsingPreprocessorDirective - Return true if we are in the middle of a
629/// preprocessor directive.
630bool TokenLexer::isParsingPreprocessorDirective() const {
Peter Collingbourne84021552011-02-28 02:37:51 +0000631 return Tokens[NumTokens-1].is(tok::eod) && !isAtEnd();
Peter Collingbourneb2eb53d2011-02-22 13:49:00 +0000632}
Chris Lattner3f1cc832008-02-07 06:03:59 +0000633
634/// HandleMicrosoftCommentPaste - In microsoft compatibility mode, /##/ pastes
635/// together to form a comment that comments out everything in the current
636/// macro, other active macros, and anything left on the current physical
637/// source line of the instantiated buffer. Handle this by returning the
638/// first token on the next line.
Chris Lattner1543e9c2008-03-09 02:18:51 +0000639void TokenLexer::HandleMicrosoftCommentPaste(Token &Tok) {
Chris Lattner3f1cc832008-02-07 06:03:59 +0000640 // We 'comment out' the rest of this macro by just ignoring the rest of the
641 // tokens that have not been lexed yet, if any.
Mike Stump1eb44332009-09-09 15:08:12 +0000642
Chris Lattner3f1cc832008-02-07 06:03:59 +0000643 // Since this must be a macro, mark the macro enabled now that it is no longer
644 // being expanded.
645 assert(Macro && "Token streams can't paste comments");
646 Macro->EnableMacro();
Mike Stump1eb44332009-09-09 15:08:12 +0000647
Chris Lattner3f1cc832008-02-07 06:03:59 +0000648 PP.HandleMicrosoftCommentPaste(Tok);
649}
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000650
651/// \brief If \arg loc is a FileID and points inside the current macro
652/// definition, returns the appropriate source location pointing at the
653/// macro expansion source location entry.
654SourceLocation TokenLexer::getMacroExpansionLocation(SourceLocation loc) const {
655 assert(InstantiateLocStart.isValid() && MacroExpansionStart.isValid() &&
656 "Not appropriate for token streams");
657 assert(loc.isValid());
658
659 SourceManager &SM = PP.getSourceManager();
660 unsigned relativeOffset;
661 if (loc.isFileID() &&
662 SM.isInFileID(loc,
663 MacroDefStartInfo.first, MacroDefStartInfo.second,
664 Macro->getDefinitionLength(SM), &relativeOffset)) {
665 return MacroExpansionStart.getFileLocWithOffset(relativeOffset);
666 }
667
668 return SourceLocation();
669}