blob: 586f26fd2bcce6c12c873f5540de53c93f422b37 [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() {
Argyrios Kyrtzidis0fd9c482011-07-07 03:40:37 +0000124 SourceManager &SM = PP.getSourceManager();
125
Chris Lattnerd2177732007-07-20 16:59:19 +0000126 llvm::SmallVector<Token, 128> ResultToks;
Mike Stump1eb44332009-09-09 15:08:12 +0000127
Chris Lattner8d896432008-03-09 02:07:49 +0000128 // Loop through 'Tokens', expanding them into ResultToks. Keep
Reid Spencer5f016e22007-07-11 17:01:13 +0000129 // track of whether we change anything. If not, no need to keep them. If so,
Chris Lattner8d896432008-03-09 02:07:49 +0000130 // we install the newly expanded sequence as the new 'Tokens' list.
Reid Spencer5f016e22007-07-11 17:01:13 +0000131 bool MadeChange = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000132
Reid Spencer5f016e22007-07-11 17:01:13 +0000133 // NextTokGetsSpace - When this is true, the next token appended to the
134 // output list will get a leading space, regardless of whether it had one to
135 // begin with or not. This is used for placemarker support.
136 bool NextTokGetsSpace = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000137
Chris Lattner8d896432008-03-09 02:07:49 +0000138 for (unsigned i = 0, e = NumTokens; i != e; ++i) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000139 // If we found the stringify operator, get the argument stringified. The
140 // preprocessor already verified that the following token is a macro name
141 // when the #define was parsed.
Chris Lattner8d896432008-03-09 02:07:49 +0000142 const Token &CurTok = Tokens[i];
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000143 if (CurTok.is(tok::hash) || CurTok.is(tok::hashat)) {
Chris Lattner8d896432008-03-09 02:07:49 +0000144 int ArgNo = Macro->getArgumentNum(Tokens[i+1].getIdentifierInfo());
Reid Spencer5f016e22007-07-11 17:01:13 +0000145 assert(ArgNo != -1 && "Token following # is not an argument?");
Mike Stump1eb44332009-09-09 15:08:12 +0000146
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000147 SourceLocation hashInstLoc;
148 if(InstantiateLocStart.isValid()) {
149 hashInstLoc = getMacroExpansionLocation(CurTok.getLocation());
150 assert(hashInstLoc.isValid() && "Expected '#' to come from definition");
151 }
152
Chris Lattnerd2177732007-07-20 16:59:19 +0000153 Token Res;
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000154 if (CurTok.is(tok::hash)) // Stringify
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000155 Res = ActualArgs->getStringifiedArgument(ArgNo, PP, hashInstLoc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000156 else {
157 // 'charify': don't bother caching these.
Chris Lattnere5c8ffe2008-03-09 02:55:12 +0000158 Res = MacroArgs::StringifyArgument(ActualArgs->getUnexpArgument(ArgNo),
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000159 PP, true, hashInstLoc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000160 }
Mike Stump1eb44332009-09-09 15:08:12 +0000161
Reid Spencer5f016e22007-07-11 17:01:13 +0000162 // The stringified/charified string leading space flag gets set to match
163 // the #/#@ operator.
164 if (CurTok.hasLeadingSpace() || NextTokGetsSpace)
Chris Lattnerd2177732007-07-20 16:59:19 +0000165 Res.setFlag(Token::LeadingSpace);
Mike Stump1eb44332009-09-09 15:08:12 +0000166
Reid Spencer5f016e22007-07-11 17:01:13 +0000167 ResultToks.push_back(Res);
168 MadeChange = true;
169 ++i; // Skip arg name.
170 NextTokGetsSpace = false;
171 continue;
172 }
Mike Stump1eb44332009-09-09 15:08:12 +0000173
Reid Spencer5f016e22007-07-11 17:01:13 +0000174 // Otherwise, if this is not an argument token, just add the token to the
175 // output buffer.
176 IdentifierInfo *II = CurTok.getIdentifierInfo();
177 int ArgNo = II ? Macro->getArgumentNum(II) : -1;
178 if (ArgNo == -1) {
179 // This isn't an argument, just add it.
180 ResultToks.push_back(CurTok);
181
182 if (NextTokGetsSpace) {
Chris Lattnerd2177732007-07-20 16:59:19 +0000183 ResultToks.back().setFlag(Token::LeadingSpace);
Reid Spencer5f016e22007-07-11 17:01:13 +0000184 NextTokGetsSpace = false;
185 }
186 continue;
187 }
Mike Stump1eb44332009-09-09 15:08:12 +0000188
Reid Spencer5f016e22007-07-11 17:01:13 +0000189 // An argument is expanded somehow, the result is different than the
190 // input.
191 MadeChange = true;
192
193 // Otherwise, this is a use of the argument. Find out if there is a paste
194 // (##) operator before or after the argument.
Mike Stump1eb44332009-09-09 15:08:12 +0000195 bool PasteBefore =
Argyrios Kyrtzidis0fd9c482011-07-07 03:40:37 +0000196 !ResultToks.empty() && ResultToks.back().is(tok::hashhash) &&
197 // If the '##' came from expanding an argument,treat it as a normal token.
198 SM.isBeforeInSourceLocationOffset(ResultToks.back().getLocation(),
199 MacroStartSLocOffset);
Chris Lattner8d896432008-03-09 02:07:49 +0000200 bool PasteAfter = i+1 != e && Tokens[i+1].is(tok::hashhash);
Mike Stump1eb44332009-09-09 15:08:12 +0000201
Reid Spencer5f016e22007-07-11 17:01:13 +0000202 // If it is not the LHS/RHS of a ## operator, we must pre-expand the
203 // argument and substitute the expanded tokens into the result. This is
204 // C99 6.10.3.1p1.
205 if (!PasteBefore && !PasteAfter) {
Chris Lattnerd2177732007-07-20 16:59:19 +0000206 const Token *ResultArgToks;
Reid Spencer5f016e22007-07-11 17:01:13 +0000207
208 // Only preexpand the argument if it could possibly need it. This
209 // avoids some work in common cases.
Chris Lattnerd2177732007-07-20 16:59:19 +0000210 const Token *ArgTok = ActualArgs->getUnexpArgument(ArgNo);
Chris Lattnercc1a8752007-10-07 08:44:20 +0000211 if (ActualArgs->ArgNeedsPreexpansion(ArgTok, PP))
Chris Lattnerf5809a72009-12-28 06:17:16 +0000212 ResultArgToks = &ActualArgs->getPreExpArgument(ArgNo, Macro, PP)[0];
Reid Spencer5f016e22007-07-11 17:01:13 +0000213 else
214 ResultArgToks = ArgTok; // Use non-preexpanded tokens.
Mike Stump1eb44332009-09-09 15:08:12 +0000215
Reid Spencer5f016e22007-07-11 17:01:13 +0000216 // If the arg token expanded into anything, append it.
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000217 if (ResultArgToks->isNot(tok::eof)) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000218 unsigned FirstResult = ResultToks.size();
219 unsigned NumToks = MacroArgs::getArgLength(ResultArgToks);
220 ResultToks.append(ResultArgToks, ResultArgToks+NumToks);
Mike Stump1eb44332009-09-09 15:08:12 +0000221
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000222 if(InstantiateLocStart.isValid()) {
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000223 SourceLocation curInst =
224 getMacroExpansionLocation(CurTok.getLocation());
225 assert(curInst.isValid() &&
226 "Expected arg identifier to come from definition");
227 for (unsigned i = FirstResult, e = ResultToks.size(); i != e; ++i) {
228 Token &Tok = ResultToks[i];
229 Tok.setLocation(SM.createInstantiationLoc(Tok.getLocation(),
230 curInst, curInst,
231 Tok.getLength()));
232 }
233 }
234
Reid Spencer5f016e22007-07-11 17:01:13 +0000235 // If any tokens were substituted from the argument, the whitespace
236 // before the first token should match the whitespace of the arg
237 // identifier.
Chris Lattnerd2177732007-07-20 16:59:19 +0000238 ResultToks[FirstResult].setFlagValue(Token::LeadingSpace,
Reid Spencer5f016e22007-07-11 17:01:13 +0000239 CurTok.hasLeadingSpace() ||
240 NextTokGetsSpace);
241 NextTokGetsSpace = false;
242 } else {
243 // If this is an empty argument, and if there was whitespace before the
244 // formal token, make sure the next token gets whitespace before it.
245 NextTokGetsSpace = CurTok.hasLeadingSpace();
246 }
247 continue;
248 }
Mike Stump1eb44332009-09-09 15:08:12 +0000249
Reid Spencer5f016e22007-07-11 17:01:13 +0000250 // Okay, we have a token that is either the LHS or RHS of a paste (##)
251 // argument. It gets substituted as its non-pre-expanded tokens.
Chris Lattnerd2177732007-07-20 16:59:19 +0000252 const Token *ArgToks = ActualArgs->getUnexpArgument(ArgNo);
Reid Spencer5f016e22007-07-11 17:01:13 +0000253 unsigned NumToks = MacroArgs::getArgLength(ArgToks);
254 if (NumToks) { // Not an empty argument?
Chris Lattner71a3a8d2008-01-29 07:54:23 +0000255 // If this is the GNU ", ## __VA_ARG__" extension, and we just learned
256 // that __VA_ARG__ expands to multiple tokens, avoid a pasting error when
257 // the expander trys to paste ',' with the first token of the __VA_ARG__
258 // expansion.
259 if (PasteBefore && ResultToks.size() >= 2 &&
260 ResultToks[ResultToks.size()-2].is(tok::comma) &&
261 (unsigned)ArgNo == Macro->getNumArgs()-1 &&
262 Macro->isVariadic()) {
263 // Remove the paste operator, report use of the extension.
264 PP.Diag(ResultToks.back().getLocation(), diag::ext_paste_comma);
265 ResultToks.pop_back();
266 }
Mike Stump1eb44332009-09-09 15:08:12 +0000267
Reid Spencer5f016e22007-07-11 17:01:13 +0000268 ResultToks.append(ArgToks, ArgToks+NumToks);
Mike Stump1eb44332009-09-09 15:08:12 +0000269
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000270 if(InstantiateLocStart.isValid()) {
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000271 SourceLocation curInst =
272 getMacroExpansionLocation(CurTok.getLocation());
273 assert(curInst.isValid() &&
274 "Expected arg identifier to come from definition");
275 for (unsigned i = ResultToks.size() - NumToks, e = ResultToks.size();
276 i != e; ++i) {
277 Token &Tok = ResultToks[i];
278 Tok.setLocation(SM.createInstantiationLoc(Tok.getLocation(),
279 curInst, curInst,
280 Tok.getLength()));
281 }
282 }
283
Chris Lattner442a6612009-05-25 16:23:08 +0000284 // If this token (the macro argument) was supposed to get leading
285 // whitespace, transfer this information onto the first token of the
286 // expansion.
287 //
288 // Do not do this if the paste operator occurs before the macro argument,
289 // as in "A ## MACROARG". In valid code, the first token will get
290 // smooshed onto the preceding one anyway (forming AMACROARG). In
291 // assembler-with-cpp mode, invalid pastes are allowed through: in this
292 // case, we do not want the extra whitespace to be added. For example,
293 // we want ". ## foo" -> ".foo" not ". foo".
294 if ((CurTok.hasLeadingSpace() || NextTokGetsSpace) &&
295 !PasteBefore)
296 ResultToks[ResultToks.size()-NumToks].setFlag(Token::LeadingSpace);
Mike Stump1eb44332009-09-09 15:08:12 +0000297
Chris Lattner442a6612009-05-25 16:23:08 +0000298 NextTokGetsSpace = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000299 continue;
300 }
Mike Stump1eb44332009-09-09 15:08:12 +0000301
Reid Spencer5f016e22007-07-11 17:01:13 +0000302 // If an empty argument is on the LHS or RHS of a paste, the standard (C99
303 // 6.10.3.3p2,3) calls for a bunch of placemarker stuff to occur. We
304 // implement this by eating ## operators when a LHS or RHS expands to
305 // empty.
306 NextTokGetsSpace |= CurTok.hasLeadingSpace();
307 if (PasteAfter) {
308 // Discard the argument token and skip (don't copy to the expansion
309 // buffer) the paste operator after it.
Chris Lattner8d896432008-03-09 02:07:49 +0000310 NextTokGetsSpace |= Tokens[i+1].hasLeadingSpace();
Reid Spencer5f016e22007-07-11 17:01:13 +0000311 ++i;
312 continue;
313 }
Mike Stump1eb44332009-09-09 15:08:12 +0000314
Reid Spencer5f016e22007-07-11 17:01:13 +0000315 // If this is on the RHS of a paste operator, we've already copied the
316 // paste operator to the ResultToks list. Remove it.
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000317 assert(PasteBefore && ResultToks.back().is(tok::hashhash));
Reid Spencer5f016e22007-07-11 17:01:13 +0000318 NextTokGetsSpace |= ResultToks.back().hasLeadingSpace();
319 ResultToks.pop_back();
Mike Stump1eb44332009-09-09 15:08:12 +0000320
Reid Spencer5f016e22007-07-11 17:01:13 +0000321 // If this is the __VA_ARGS__ token, and if the argument wasn't provided,
322 // and if the macro had at least one real argument, and if the token before
323 // the ## was a comma, remove the comma.
324 if ((unsigned)ArgNo == Macro->getNumArgs()-1 && // is __VA_ARGS__
325 ActualArgs->isVarargsElidedUse() && // Argument elided.
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000326 !ResultToks.empty() && ResultToks.back().is(tok::comma)) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000327 // Never add a space, even if the comma, ##, or arg had a space.
328 NextTokGetsSpace = false;
Chris Lattner71a3a8d2008-01-29 07:54:23 +0000329 // Remove the paste operator, report use of the extension.
330 PP.Diag(ResultToks.back().getLocation(), diag::ext_paste_comma);
Reid Spencer5f016e22007-07-11 17:01:13 +0000331 ResultToks.pop_back();
Chris Lattner496af572010-08-21 00:27:00 +0000332
333 // If the comma was right after another paste (e.g. "X##,##__VA_ARGS__"),
334 // then removal of the comma should produce a placemarker token (in C99
335 // terms) which we model by popping off the previous ##, giving us a plain
336 // "X" when __VA_ARGS__ is empty.
337 if (!ResultToks.empty() && ResultToks.back().is(tok::hashhash))
338 ResultToks.pop_back();
Reid Spencer5f016e22007-07-11 17:01:13 +0000339 }
340 continue;
341 }
Mike Stump1eb44332009-09-09 15:08:12 +0000342
Chris Lattner8d896432008-03-09 02:07:49 +0000343 // If anything changed, install this as the new Tokens list.
Reid Spencer5f016e22007-07-11 17:01:13 +0000344 if (MadeChange) {
Chris Lattner32fca722009-03-04 06:50:57 +0000345 assert(!OwnsTokens && "This would leak if we already own the token list");
Reid Spencer5f016e22007-07-11 17:01:13 +0000346 // This is deleted in the dtor.
Chris Lattner8d896432008-03-09 02:07:49 +0000347 NumTokens = ResultToks.size();
Argyrios Kyrtzidis5b3284a2011-06-29 22:20:11 +0000348 // The tokens will be added to Preprocessor's cache and will be removed
349 // when this TokenLexer finishes lexing them.
350 Tokens = PP.cacheMacroExpandedTokens(this, ResultToks);
Mike Stump1eb44332009-09-09 15:08:12 +0000351
Argyrios Kyrtzidis5b3284a2011-06-29 22:20:11 +0000352 // The preprocessor cache of macro expanded tokens owns these tokens,not us.
Chris Lattner32fca722009-03-04 06:50:57 +0000353 OwnsTokens = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000354 }
355}
356
357/// Lex - Lex and return a token from this macro stream.
358///
Chris Lattner1543e9c2008-03-09 02:18:51 +0000359void TokenLexer::Lex(Token &Tok) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000360 // Lexing off the end of the macro, pop this macro off the expansion stack.
361 if (isAtEnd()) {
362 // If this is a macro (not a token stream), mark the macro enabled now
363 // that it is no longer being expanded.
364 if (Macro) Macro->EnableMacro();
365
366 // Pop this context off the preprocessors lexer stack and get the next
367 // token. This will delete "this" so remember the PP instance var.
368 Preprocessor &PPCache = PP;
Chris Lattnerfde2bf92008-03-09 03:04:16 +0000369 if (PP.HandleEndOfTokenLexer(Tok))
Reid Spencer5f016e22007-07-11 17:01:13 +0000370 return;
371
Chris Lattnerfde2bf92008-03-09 03:04:16 +0000372 // HandleEndOfTokenLexer may not return a token. If it doesn't, lex
373 // whatever is next.
Reid Spencer5f016e22007-07-11 17:01:13 +0000374 return PPCache.Lex(Tok);
375 }
Mike Stump1eb44332009-09-09 15:08:12 +0000376
Argyrios Kyrtzidis0fd9c482011-07-07 03:40:37 +0000377 SourceManager &SM = PP.getSourceManager();
378
Reid Spencer5f016e22007-07-11 17:01:13 +0000379 // If this is the first token of the expanded result, we inherit spacing
380 // properties later.
381 bool isFirstToken = CurToken == 0;
Mike Stump1eb44332009-09-09 15:08:12 +0000382
Reid Spencer5f016e22007-07-11 17:01:13 +0000383 // Get the next token to return.
Chris Lattner8d896432008-03-09 02:07:49 +0000384 Tok = Tokens[CurToken++];
Mike Stump1eb44332009-09-09 15:08:12 +0000385
Chris Lattner332ee082009-04-19 20:29:42 +0000386 bool TokenIsFromPaste = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000387
Reid Spencer5f016e22007-07-11 17:01:13 +0000388 // If this token is followed by a token paste (##) operator, paste the tokens!
Chris Lattnerb11e43c2011-06-14 18:19:37 +0000389 // Note that ## is a normal token when not expanding a macro.
Argyrios Kyrtzidis0fd9c482011-07-07 03:40:37 +0000390 if (!isAtEnd() && Tokens[CurToken].is(tok::hashhash) && Macro &&
391 // If the '##' came from expanding an argument,treat it as a normal token.
392 SM.isBeforeInSourceLocationOffset(Tokens[CurToken].getLocation(),
393 MacroStartSLocOffset)) {
Chris Lattner04f2bd82009-12-04 06:14:03 +0000394 // When handling the microsoft /##/ extension, the final token is
395 // returned by PasteTokens, not the pasted token.
396 if (PasteTokens(Tok))
Chris Lattner3f1cc832008-02-07 06:03:59 +0000397 return;
Kovarththanan Rajaratnam65cc1e82010-03-13 08:53:33 +0000398
Chris Lattner04f2bd82009-12-04 06:14:03 +0000399 TokenIsFromPaste = true;
Mike Stump1eb44332009-09-09 15:08:12 +0000400 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000401
402 // The token's current location indicate where the token was lexed from. We
403 // need this information to compute the spelling of the token, but any
404 // diagnostics for the expanded token should appear as if they came from
405 // InstantiationLoc. Pull this information together into a new SourceLocation
406 // that captures all of this.
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000407 if (InstantiateLocStart.isValid() && // Don't do this for token streams.
408 // Check that the token's location was not already set properly.
409 SM.isBeforeInSourceLocationOffset(Tok.getLocation(),
410 MacroStartSLocOffset)) {
411 SourceLocation instLoc;
412 if (Tok.is(tok::comment)) {
413 instLoc = SM.createInstantiationLoc(Tok.getLocation(),
414 InstantiateLocStart,
415 InstantiateLocEnd,
416 Tok.getLength());
417 } else {
418 instLoc = getMacroExpansionLocation(Tok.getLocation());
419 assert(instLoc.isValid() &&
420 "Location for token not coming from definition was not set!");
421 }
422
423 Tok.setLocation(instLoc);
Reid Spencer5f016e22007-07-11 17:01:13 +0000424 }
Mike Stump1eb44332009-09-09 15:08:12 +0000425
Reid Spencer5f016e22007-07-11 17:01:13 +0000426 // If this is the first token, set the lexical properties of the token to
427 // match the lexical properties of the macro identifier.
428 if (isFirstToken) {
Chris Lattnerd2177732007-07-20 16:59:19 +0000429 Tok.setFlagValue(Token::StartOfLine , AtStartOfLine);
430 Tok.setFlagValue(Token::LeadingSpace, HasLeadingSpace);
Reid Spencer5f016e22007-07-11 17:01:13 +0000431 }
Mike Stump1eb44332009-09-09 15:08:12 +0000432
Reid Spencer5f016e22007-07-11 17:01:13 +0000433 // Handle recursive expansion!
Argyrios Kyrtzidis6f421f92009-05-22 21:09:31 +0000434 if (!Tok.isAnnotation() && Tok.getIdentifierInfo() != 0) {
Chris Lattner863c4862009-01-23 18:35:48 +0000435 // Change the kind of this identifier to the appropriate token kind, e.g.
436 // turning "for" into a keyword.
Argyrios Kyrtzidis6f421f92009-05-22 21:09:31 +0000437 IdentifierInfo *II = Tok.getIdentifierInfo();
Chris Lattner863c4862009-01-23 18:35:48 +0000438 Tok.setKind(II->getTokenID());
Mike Stump1eb44332009-09-09 15:08:12 +0000439
Chris Lattner332ee082009-04-19 20:29:42 +0000440 // If this identifier was poisoned and from a paste, emit an error. This
441 // won't be handled by Preprocessor::HandleIdentifier because this is coming
442 // from a macro expansion.
443 if (II->isPoisoned() && TokenIsFromPaste) {
John Wiegley28bbe4b2011-04-28 01:08:34 +0000444 PP.HandlePoisonedIdentifier(Tok);
Chris Lattner332ee082009-04-19 20:29:42 +0000445 }
Mike Stump1eb44332009-09-09 15:08:12 +0000446
Chris Lattner863c4862009-01-23 18:35:48 +0000447 if (!DisableMacroExpansion && II->isHandleIdentifierCase())
448 PP.HandleIdentifier(Tok);
449 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000450
451 // Otherwise, return a normal token.
452}
453
454/// PasteTokens - Tok is the LHS of a ## operator, and CurToken is the ##
455/// operator. Read the ## and RHS, and paste the LHS/RHS together. If there
Nico Weber48002c82008-09-29 00:25:48 +0000456/// are more ## after it, chomp them iteratively. Return the result as Tok.
Chris Lattner3f1cc832008-02-07 06:03:59 +0000457/// If this returns true, the caller should immediately return the token.
Chris Lattner1543e9c2008-03-09 02:18:51 +0000458bool TokenLexer::PasteTokens(Token &Tok) {
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000459 llvm::SmallString<128> Buffer;
Chris Lattner47246be2009-01-26 19:29:26 +0000460 const char *ResultTokStrPtr = 0;
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000461 SourceLocation PasteOpLoc;
Reid Spencer5f016e22007-07-11 17:01:13 +0000462 do {
463 // Consume the ## operator.
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000464 PasteOpLoc = Tokens[CurToken].getLocation();
Reid Spencer5f016e22007-07-11 17:01:13 +0000465 ++CurToken;
466 assert(!isAtEnd() && "No token on the RHS of a paste operator!");
Mike Stump1eb44332009-09-09 15:08:12 +0000467
Reid Spencer5f016e22007-07-11 17:01:13 +0000468 // Get the RHS token.
Chris Lattner8d896432008-03-09 02:07:49 +0000469 const Token &RHS = Tokens[CurToken];
Mike Stump1eb44332009-09-09 15:08:12 +0000470
Reid Spencer5f016e22007-07-11 17:01:13 +0000471 // Allocate space for the result token. This is guaranteed to be enough for
Chris Lattner1fa49532009-03-08 08:08:45 +0000472 // the two tokens.
473 Buffer.resize(Tok.getLength() + RHS.getLength());
Mike Stump1eb44332009-09-09 15:08:12 +0000474
Reid Spencer5f016e22007-07-11 17:01:13 +0000475 // Get the spelling of the LHS token in Buffer.
476 const char *BufPtr = &Buffer[0];
Douglas Gregor453091c2010-03-16 22:30:13 +0000477 bool Invalid = false;
478 unsigned LHSLen = PP.getSpelling(Tok, BufPtr, &Invalid);
Reid Spencer5f016e22007-07-11 17:01:13 +0000479 if (BufPtr != &Buffer[0]) // Really, we want the chars in Buffer!
480 memcpy(&Buffer[0], BufPtr, LHSLen);
Douglas Gregor453091c2010-03-16 22:30:13 +0000481 if (Invalid)
482 return true;
483
Reid Spencer5f016e22007-07-11 17:01:13 +0000484 BufPtr = &Buffer[LHSLen];
Douglas Gregor453091c2010-03-16 22:30:13 +0000485 unsigned RHSLen = PP.getSpelling(RHS, BufPtr, &Invalid);
486 if (Invalid)
487 return true;
Reid Spencer5f016e22007-07-11 17:01:13 +0000488 if (BufPtr != &Buffer[LHSLen]) // Really, we want the chars in Buffer!
489 memcpy(&Buffer[LHSLen], BufPtr, RHSLen);
Mike Stump1eb44332009-09-09 15:08:12 +0000490
Reid Spencer5f016e22007-07-11 17:01:13 +0000491 // Trim excess space.
Chris Lattner1fa49532009-03-08 08:08:45 +0000492 Buffer.resize(LHSLen+RHSLen);
Mike Stump1eb44332009-09-09 15:08:12 +0000493
Reid Spencer5f016e22007-07-11 17:01:13 +0000494 // Plop the pasted result (including the trailing newline and null) into a
495 // scratch buffer where we can lex it.
Chris Lattner47246be2009-01-26 19:29:26 +0000496 Token ResultTokTmp;
497 ResultTokTmp.startToken();
Mike Stump1eb44332009-09-09 15:08:12 +0000498
Chris Lattner47246be2009-01-26 19:29:26 +0000499 // Claim that the tmp token is a string_literal so that we can get the
Chris Lattner917f1432009-12-23 21:29:53 +0000500 // character pointer back from CreateString in getLiteralData().
Chris Lattner47246be2009-01-26 19:29:26 +0000501 ResultTokTmp.setKind(tok::string_literal);
502 PP.CreateString(&Buffer[0], Buffer.size(), ResultTokTmp);
503 SourceLocation ResultTokLoc = ResultTokTmp.getLocation();
504 ResultTokStrPtr = ResultTokTmp.getLiteralData();
505
Reid Spencer5f016e22007-07-11 17:01:13 +0000506 // Lex the resultant pasted token into Result.
Chris Lattnerd2177732007-07-20 16:59:19 +0000507 Token Result;
Mike Stump1eb44332009-09-09 15:08:12 +0000508
Abramo Bagnarac4bf2b92010-12-22 08:23:18 +0000509 if (Tok.isAnyIdentifier() && RHS.isAnyIdentifier()) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000510 // Common paste case: identifier+identifier = identifier. Avoid creating
511 // a lexer and other overhead.
512 PP.IncrementPasteCounter(true);
513 Result.startToken();
Abramo Bagnarac4bf2b92010-12-22 08:23:18 +0000514 Result.setKind(tok::raw_identifier);
515 Result.setRawIdentifierData(ResultTokStrPtr);
Reid Spencer5f016e22007-07-11 17:01:13 +0000516 Result.setLocation(ResultTokLoc);
517 Result.setLength(LHSLen+RHSLen);
518 } else {
519 PP.IncrementPasteCounter(false);
Mike Stump1eb44332009-09-09 15:08:12 +0000520
Chris Lattnerbcc2a672009-01-19 06:46:35 +0000521 assert(ResultTokLoc.isFileID() &&
522 "Should be a raw location into scratch buffer");
Reid Spencer5f016e22007-07-11 17:01:13 +0000523 SourceManager &SourceMgr = PP.getSourceManager();
Chris Lattner47246be2009-01-26 19:29:26 +0000524 FileID LocFileID = SourceMgr.getFileID(ResultTokLoc);
Mike Stump1eb44332009-09-09 15:08:12 +0000525
Douglas Gregorf715ca12010-03-16 00:06:06 +0000526 bool Invalid = false;
Douglas Gregoraea67db2010-03-15 22:54:52 +0000527 const char *ScratchBufStart
Benjamin Kramerf6ac97b2010-03-16 14:14:31 +0000528 = SourceMgr.getBufferData(LocFileID, &Invalid).data();
Douglas Gregorf715ca12010-03-16 00:06:06 +0000529 if (Invalid)
Douglas Gregoraea67db2010-03-15 22:54:52 +0000530 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000531
Chris Lattnerbcc2a672009-01-19 06:46:35 +0000532 // Make a lexer to lex this string from. Lex just this one token.
Reid Spencer5f016e22007-07-11 17:01:13 +0000533 // Make a lexer object so that we lex and expand the paste result.
Chris Lattner47246be2009-01-26 19:29:26 +0000534 Lexer TL(SourceMgr.getLocForStartOfFile(LocFileID),
535 PP.getLangOptions(), ScratchBufStart,
Chris Lattner1fa49532009-03-08 08:08:45 +0000536 ResultTokStrPtr, ResultTokStrPtr+LHSLen+RHSLen);
Mike Stump1eb44332009-09-09 15:08:12 +0000537
Reid Spencer5f016e22007-07-11 17:01:13 +0000538 // Lex a token in raw mode. This way it won't look up identifiers
539 // automatically, lexing off the end will return an eof token, and
540 // warnings are disabled. This returns true if the result token is the
541 // entire buffer.
Chris Lattner1fa49532009-03-08 08:08:45 +0000542 bool isInvalid = !TL.LexFromRawLexer(Result);
Mike Stump1eb44332009-09-09 15:08:12 +0000543
Reid Spencer5f016e22007-07-11 17:01:13 +0000544 // If we got an EOF token, we didn't form even ONE token. For example, we
545 // did "/ ## /" to get "//".
Chris Lattner1fa49532009-03-08 08:08:45 +0000546 isInvalid |= Result.is(tok::eof);
Mike Stump1eb44332009-09-09 15:08:12 +0000547
Chris Lattner1fa49532009-03-08 08:08:45 +0000548 // If pasting the two tokens didn't form a full new token, this is an
549 // error. This occurs with "x ## +" and other stuff. Return with Tok
550 // unmodified and with RHS as the next token to lex.
551 if (isInvalid) {
552 // Test for the Microsoft extension of /##/ turning into // here on the
553 // error path.
Mike Stump1eb44332009-09-09 15:08:12 +0000554 if (PP.getLangOptions().Microsoft && Tok.is(tok::slash) &&
Chris Lattner1fa49532009-03-08 08:08:45 +0000555 RHS.is(tok::slash)) {
556 HandleMicrosoftCommentPaste(Tok);
557 return true;
558 }
Mike Stump1eb44332009-09-09 15:08:12 +0000559
Chris Lattner920bf452010-07-17 16:24:30 +0000560 // Do not emit the error when preprocessing assembler code.
Chris Lattner3c17b8a2009-05-28 05:39:39 +0000561 if (!PP.getLangOptions().AsmPreprocessor) {
562 // Explicitly convert the token location to have proper instantiation
563 // information so that the user knows where it came from.
564 SourceManager &SM = PP.getSourceManager();
Chris Lattnerb11e43c2011-06-14 18:19:37 +0000565 SourceLocation Loc =
566 SM.createInstantiationLoc(PasteOpLoc, InstantiateLocStart,
567 InstantiateLocEnd, 2);
Chris Lattner920bf452010-07-17 16:24:30 +0000568 // If we're in microsoft extensions mode, downgrade this from a hard
569 // error to a warning that defaults to an error. This allows
570 // disabling it.
571 PP.Diag(Loc,
572 PP.getLangOptions().Microsoft ? diag::err_pp_bad_paste_ms
573 : diag::err_pp_bad_paste)
Benjamin Kramer476d8b82010-08-11 14:47:12 +0000574 << Buffer.str();
Chris Lattner3c17b8a2009-05-28 05:39:39 +0000575 }
Mike Stump1eb44332009-09-09 15:08:12 +0000576
Chris Lattnerf1538de2009-03-23 17:32:45 +0000577 // Do not consume the RHS.
578 --CurToken;
Chris Lattner1fa49532009-03-08 08:08:45 +0000579 }
Mike Stump1eb44332009-09-09 15:08:12 +0000580
Chris Lattner1fa49532009-03-08 08:08:45 +0000581 // Turn ## into 'unknown' to avoid # ## # from looking like a paste
582 // operator.
583 if (Result.is(tok::hashhash))
584 Result.setKind(tok::unknown);
Reid Spencer5f016e22007-07-11 17:01:13 +0000585 }
Mike Stump1eb44332009-09-09 15:08:12 +0000586
Reid Spencer5f016e22007-07-11 17:01:13 +0000587 // Transfer properties of the LHS over the the Result.
Chris Lattnerd2177732007-07-20 16:59:19 +0000588 Result.setFlagValue(Token::StartOfLine , Tok.isAtStartOfLine());
589 Result.setFlagValue(Token::LeadingSpace, Tok.hasLeadingSpace());
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000590
Reid Spencer5f016e22007-07-11 17:01:13 +0000591 // Finally, replace LHS with the result, consume the RHS, and iterate.
592 ++CurToken;
593 Tok = Result;
Chris Lattner8d896432008-03-09 02:07:49 +0000594 } while (!isAtEnd() && Tokens[CurToken].is(tok::hashhash));
Mike Stump1eb44332009-09-09 15:08:12 +0000595
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000596 // The token's current location indicate where the token was lexed from. We
597 // need this information to compute the spelling of the token, but any
598 // diagnostics for the expanded token should appear as if the token was
599 // instantiated from the (##) operator. Pull this information together into
600 // a new SourceLocation that captures all of this.
601 if (InstantiateLocStart.isValid()) {
602 SourceManager &SM = PP.getSourceManager();
603 SourceLocation pasteLocInst =
604 getMacroExpansionLocation(PasteOpLoc);
605 assert(pasteLocInst.isValid() &&
606 "Expected '##' to come from definition");
607
608 Tok.setLocation(SM.createInstantiationLoc(Tok.getLocation(),
609 pasteLocInst,
610 pasteLocInst,
611 Tok.getLength()));
612 }
613
Reid Spencer5f016e22007-07-11 17:01:13 +0000614 // Now that we got the result token, it will be subject to expansion. Since
615 // token pasting re-lexes the result token in raw mode, identifier information
616 // isn't looked up. As such, if the result is an identifier, look up id info.
Abramo Bagnarac4bf2b92010-12-22 08:23:18 +0000617 if (Tok.is(tok::raw_identifier)) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000618 // Look up the identifier info for the token. We disabled identifier lookup
619 // by saying we're skipping contents, so we need to do this manually.
Abramo Bagnarac4bf2b92010-12-22 08:23:18 +0000620 PP.LookUpIdentifierInfo(Tok);
Reid Spencer5f016e22007-07-11 17:01:13 +0000621 }
Chris Lattner3f1cc832008-02-07 06:03:59 +0000622 return false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000623}
624
625/// isNextTokenLParen - If the next token lexed will pop this macro off the
626/// expansion stack, return 2. If the next unexpanded token is a '(', return
627/// 1, otherwise return 0.
Chris Lattner1543e9c2008-03-09 02:18:51 +0000628unsigned TokenLexer::isNextTokenLParen() const {
Reid Spencer5f016e22007-07-11 17:01:13 +0000629 // Out of tokens?
630 if (isAtEnd())
631 return 2;
Chris Lattner8d896432008-03-09 02:07:49 +0000632 return Tokens[CurToken].is(tok::l_paren);
Reid Spencer5f016e22007-07-11 17:01:13 +0000633}
Chris Lattner3f1cc832008-02-07 06:03:59 +0000634
Peter Collingbourneb2eb53d2011-02-22 13:49:00 +0000635/// isParsingPreprocessorDirective - Return true if we are in the middle of a
636/// preprocessor directive.
637bool TokenLexer::isParsingPreprocessorDirective() const {
Peter Collingbourne84021552011-02-28 02:37:51 +0000638 return Tokens[NumTokens-1].is(tok::eod) && !isAtEnd();
Peter Collingbourneb2eb53d2011-02-22 13:49:00 +0000639}
Chris Lattner3f1cc832008-02-07 06:03:59 +0000640
641/// HandleMicrosoftCommentPaste - In microsoft compatibility mode, /##/ pastes
642/// together to form a comment that comments out everything in the current
643/// macro, other active macros, and anything left on the current physical
644/// source line of the instantiated buffer. Handle this by returning the
645/// first token on the next line.
Chris Lattner1543e9c2008-03-09 02:18:51 +0000646void TokenLexer::HandleMicrosoftCommentPaste(Token &Tok) {
Chris Lattner3f1cc832008-02-07 06:03:59 +0000647 // We 'comment out' the rest of this macro by just ignoring the rest of the
648 // tokens that have not been lexed yet, if any.
Mike Stump1eb44332009-09-09 15:08:12 +0000649
Chris Lattner3f1cc832008-02-07 06:03:59 +0000650 // Since this must be a macro, mark the macro enabled now that it is no longer
651 // being expanded.
652 assert(Macro && "Token streams can't paste comments");
653 Macro->EnableMacro();
Mike Stump1eb44332009-09-09 15:08:12 +0000654
Chris Lattner3f1cc832008-02-07 06:03:59 +0000655 PP.HandleMicrosoftCommentPaste(Tok);
656}
Argyrios Kyrtzidisb73377e2011-07-07 03:40:34 +0000657
658/// \brief If \arg loc is a FileID and points inside the current macro
659/// definition, returns the appropriate source location pointing at the
660/// macro expansion source location entry.
661SourceLocation TokenLexer::getMacroExpansionLocation(SourceLocation loc) const {
662 assert(InstantiateLocStart.isValid() && MacroExpansionStart.isValid() &&
663 "Not appropriate for token streams");
664 assert(loc.isValid());
665
666 SourceManager &SM = PP.getSourceManager();
667 unsigned relativeOffset;
668 if (loc.isFileID() &&
669 SM.isInFileID(loc,
670 MacroDefStartInfo.first, MacroDefStartInfo.second,
671 Macro->getDefinitionLength(SM), &relativeOffset)) {
672 return MacroExpansionStart.getFileLocWithOffset(relativeOffset);
673 }
674
675 return SourceLocation();
676}