blob: ff78e8a4fa456f9e7a649988b6ed230ee97014b5 [file] [log] [blame]
Dmitri Gribenko2d44d772012-06-26 20:39:18 +00001#include "clang/AST/CommentLexer.h"
Dmitri Gribenkoaa580812012-08-09 00:03:17 +00002#include "clang/AST/CommentCommandTraits.h"
Dmitri Gribenko477a9f52012-07-27 20:37:06 +00003#include "clang/Basic/ConvertUTF.h"
Dmitri Gribenkoc934dfe2013-01-19 22:06:05 +00004#include "llvm/ADT/StringExtras.h"
Dmitri Gribenko2d44d772012-06-26 20:39:18 +00005#include "llvm/ADT/StringSwitch.h"
6#include "llvm/Support/ErrorHandling.h"
7
8namespace clang {
9namespace comments {
10
11void Token::dump(const Lexer &L, const SourceManager &SM) const {
12 llvm::errs() << "comments::Token Kind=" << Kind << " ";
13 Loc.dump(SM);
14 llvm::errs() << " " << Length << " \"" << L.getSpelling(*this, SM) << "\"\n";
15}
16
Dmitri Gribenko477a9f52012-07-27 20:37:06 +000017namespace {
18bool isHTMLNamedCharacterReferenceCharacter(char C) {
19 return (C >= 'a' && C <= 'z') ||
20 (C >= 'A' && C <= 'Z');
21}
22
23bool isHTMLDecimalCharacterReferenceCharacter(char C) {
24 return C >= '0' && C <= '9';
25}
26
27bool isHTMLHexCharacterReferenceCharacter(char C) {
28 return (C >= '0' && C <= '9') ||
29 (C >= 'a' && C <= 'f') ||
30 (C >= 'A' && C <= 'F');
31}
Dmitri Gribenko834a5bd2012-08-22 22:56:08 +000032
Dmitri Gribenkoc24a76e2012-08-31 02:21:44 +000033#include "clang/AST/CommentHTMLTags.inc"
34
Dmitri Gribenko477a9f52012-07-27 20:37:06 +000035} // unnamed namespace
36
37StringRef Lexer::resolveHTMLNamedCharacterReference(StringRef Name) const {
38 return llvm::StringSwitch<StringRef>(Name)
39 .Case("amp", "&")
40 .Case("lt", "<")
41 .Case("gt", ">")
42 .Case("quot", "\"")
43 .Case("apos", "\'")
44 .Default("");
45}
46
47StringRef Lexer::resolveHTMLDecimalCharacterReference(StringRef Name) const {
48 unsigned CodePoint = 0;
49 for (unsigned i = 0, e = Name.size(); i != e; ++i) {
50 assert(isHTMLDecimalCharacterReferenceCharacter(Name[i]));
51 CodePoint *= 10;
52 CodePoint += Name[i] - '0';
53 }
54
55 char *Resolved = Allocator.Allocate<char>(UNI_MAX_UTF8_BYTES_PER_CODE_POINT);
56 char *ResolvedPtr = Resolved;
57 if (ConvertCodePointToUTF8(CodePoint, ResolvedPtr))
58 return StringRef(Resolved, ResolvedPtr - Resolved);
59 else
60 return StringRef();
61}
62
63StringRef Lexer::resolveHTMLHexCharacterReference(StringRef Name) const {
64 unsigned CodePoint = 0;
65 for (unsigned i = 0, e = Name.size(); i != e; ++i) {
66 CodePoint *= 16;
67 const char C = Name[i];
68 assert(isHTMLHexCharacterReferenceCharacter(C));
Dmitri Gribenkoc934dfe2013-01-19 22:06:05 +000069 CodePoint += llvm::hexDigitValue(C);
Dmitri Gribenko477a9f52012-07-27 20:37:06 +000070 }
71
72 char *Resolved = Allocator.Allocate<char>(UNI_MAX_UTF8_BYTES_PER_CODE_POINT);
73 char *ResolvedPtr = Resolved;
74 if (ConvertCodePointToUTF8(CodePoint, ResolvedPtr))
75 return StringRef(Resolved, ResolvedPtr - Resolved);
76 else
77 return StringRef();
78}
79
Dmitri Gribenko2d44d772012-06-26 20:39:18 +000080void Lexer::skipLineStartingDecorations() {
81 // This function should be called only for C comments
82 assert(CommentState == LCS_InsideCComment);
83
84 if (BufferPtr == CommentEnd)
85 return;
86
87 switch (*BufferPtr) {
88 case ' ':
89 case '\t':
90 case '\f':
91 case '\v': {
92 const char *NewBufferPtr = BufferPtr;
93 NewBufferPtr++;
94 if (NewBufferPtr == CommentEnd)
95 return;
96
97 char C = *NewBufferPtr;
98 while (C == ' ' || C == '\t' || C == '\f' || C == '\v') {
99 NewBufferPtr++;
100 if (NewBufferPtr == CommentEnd)
101 return;
102 C = *NewBufferPtr;
103 }
104 if (C == '*')
105 BufferPtr = NewBufferPtr + 1;
106 break;
107 }
108 case '*':
109 BufferPtr++;
110 break;
111 }
112}
113
114namespace {
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000115/// Returns pointer to the first newline character in the string.
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000116const char *findNewline(const char *BufferPtr, const char *BufferEnd) {
117 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
118 const char C = *BufferPtr;
119 if (C == '\n' || C == '\r')
120 return BufferPtr;
121 }
122 return BufferEnd;
123}
124
125const char *skipNewline(const char *BufferPtr, const char *BufferEnd) {
126 if (BufferPtr == BufferEnd)
127 return BufferPtr;
128
129 if (*BufferPtr == '\n')
130 BufferPtr++;
131 else {
132 assert(*BufferPtr == '\r');
133 BufferPtr++;
134 if (BufferPtr != BufferEnd && *BufferPtr == '\n')
135 BufferPtr++;
136 }
137 return BufferPtr;
138}
139
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000140const char *skipNamedCharacterReference(const char *BufferPtr,
141 const char *BufferEnd) {
142 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
143 if (!isHTMLNamedCharacterReferenceCharacter(*BufferPtr))
144 return BufferPtr;
145 }
146 return BufferEnd;
147}
148
149const char *skipDecimalCharacterReference(const char *BufferPtr,
150 const char *BufferEnd) {
151 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
152 if (!isHTMLDecimalCharacterReferenceCharacter(*BufferPtr))
153 return BufferPtr;
154 }
155 return BufferEnd;
156}
157
158const char *skipHexCharacterReference(const char *BufferPtr,
159 const char *BufferEnd) {
160 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
161 if (!isHTMLHexCharacterReferenceCharacter(*BufferPtr))
162 return BufferPtr;
163 }
164 return BufferEnd;
165}
166
Dmitri Gribenkoa99ec102012-07-09 21:32:40 +0000167bool isHTMLIdentifierStartingCharacter(char C) {
168 return (C >= 'a' && C <= 'z') ||
169 (C >= 'A' && C <= 'Z');
170}
171
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000172bool isHTMLIdentifierCharacter(char C) {
173 return (C >= 'a' && C <= 'z') ||
174 (C >= 'A' && C <= 'Z') ||
175 (C >= '0' && C <= '9');
176}
177
178const char *skipHTMLIdentifier(const char *BufferPtr, const char *BufferEnd) {
179 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
180 if (!isHTMLIdentifierCharacter(*BufferPtr))
181 return BufferPtr;
182 }
183 return BufferEnd;
184}
185
186/// Skip HTML string quoted in single or double quotes. Escaping quotes inside
187/// string allowed.
188///
189/// Returns pointer to closing quote.
190const char *skipHTMLQuotedString(const char *BufferPtr, const char *BufferEnd)
191{
192 const char Quote = *BufferPtr;
193 assert(Quote == '\"' || Quote == '\'');
194
195 BufferPtr++;
196 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
197 const char C = *BufferPtr;
198 if (C == Quote && BufferPtr[-1] != '\\')
199 return BufferPtr;
200 }
201 return BufferEnd;
202}
203
204bool isHorizontalWhitespace(char C) {
205 return C == ' ' || C == '\t' || C == '\f' || C == '\v';
206}
207
208bool isWhitespace(char C) {
209 return C == ' ' || C == '\n' || C == '\r' ||
210 C == '\t' || C == '\f' || C == '\v';
211}
212
213const char *skipWhitespace(const char *BufferPtr, const char *BufferEnd) {
214 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
215 if (!isWhitespace(*BufferPtr))
216 return BufferPtr;
217 }
218 return BufferEnd;
219}
220
Dmitri Gribenko64da4e52012-07-18 23:01:58 +0000221bool isWhitespace(const char *BufferPtr, const char *BufferEnd) {
222 return skipWhitespace(BufferPtr, BufferEnd) == BufferEnd;
223}
224
Dmitri Gribenko8c05da32012-09-14 16:35:35 +0000225bool isCommandNameStartCharacter(char C) {
226 return (C >= 'a' && C <= 'z') ||
227 (C >= 'A' && C <= 'Z');
228}
229
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000230bool isCommandNameCharacter(char C) {
231 return (C >= 'a' && C <= 'z') ||
232 (C >= 'A' && C <= 'Z') ||
233 (C >= '0' && C <= '9');
234}
235
236const char *skipCommandName(const char *BufferPtr, const char *BufferEnd) {
237 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
238 if (!isCommandNameCharacter(*BufferPtr))
239 return BufferPtr;
240 }
241 return BufferEnd;
242}
243
244/// Return the one past end pointer for BCPL comments.
245/// Handles newlines escaped with backslash or trigraph for backslahs.
246const char *findBCPLCommentEnd(const char *BufferPtr, const char *BufferEnd) {
247 const char *CurPtr = BufferPtr;
248 while (CurPtr != BufferEnd) {
249 char C = *CurPtr;
250 while (C != '\n' && C != '\r') {
251 CurPtr++;
252 if (CurPtr == BufferEnd)
253 return BufferEnd;
254 C = *CurPtr;
255 }
256 // We found a newline, check if it is escaped.
257 const char *EscapePtr = CurPtr - 1;
258 while(isHorizontalWhitespace(*EscapePtr))
259 EscapePtr--;
260
261 if (*EscapePtr == '\\' ||
262 (EscapePtr - 2 >= BufferPtr && EscapePtr[0] == '/' &&
263 EscapePtr[-1] == '?' && EscapePtr[-2] == '?')) {
264 // We found an escaped newline.
265 CurPtr = skipNewline(CurPtr, BufferEnd);
266 } else
267 return CurPtr; // Not an escaped newline.
268 }
269 return BufferEnd;
270}
271
272/// Return the one past end pointer for C comments.
273/// Very dumb, does not handle escaped newlines or trigraphs.
274const char *findCCommentEnd(const char *BufferPtr, const char *BufferEnd) {
275 for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
276 if (*BufferPtr == '*') {
277 assert(BufferPtr + 1 != BufferEnd);
278 if (*(BufferPtr + 1) == '/')
279 return BufferPtr;
280 }
281 }
282 llvm_unreachable("buffer end hit before '*/' was seen");
283}
284} // unnamed namespace
285
286void Lexer::lexCommentText(Token &T) {
287 assert(CommentState == LCS_InsideBCPLComment ||
288 CommentState == LCS_InsideCComment);
289
290 switch (State) {
291 case LS_Normal:
292 break;
293 case LS_VerbatimBlockFirstLine:
294 lexVerbatimBlockFirstLine(T);
295 return;
296 case LS_VerbatimBlockBody:
297 lexVerbatimBlockBody(T);
298 return;
Dmitri Gribenko962668d2012-06-27 16:53:58 +0000299 case LS_VerbatimLineText:
300 lexVerbatimLineText(T);
301 return;
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000302 case LS_HTMLStartTag:
303 lexHTMLStartTag(T);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000304 return;
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000305 case LS_HTMLEndTag:
306 lexHTMLEndTag(T);
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000307 return;
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000308 }
309
310 assert(State == LS_Normal);
311
312 const char *TokenPtr = BufferPtr;
313 assert(TokenPtr < CommentEnd);
314 while (TokenPtr != CommentEnd) {
315 switch(*TokenPtr) {
316 case '\\':
317 case '@': {
318 TokenPtr++;
319 if (TokenPtr == CommentEnd) {
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000320 formTextToken(T, TokenPtr);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000321 return;
322 }
323 char C = *TokenPtr;
324 switch (C) {
325 default:
326 break;
327
328 case '\\': case '@': case '&': case '$':
329 case '#': case '<': case '>': case '%':
330 case '\"': case '.': case ':':
331 // This is one of \\ \@ \& \$ etc escape sequences.
332 TokenPtr++;
333 if (C == ':' && TokenPtr != CommentEnd && *TokenPtr == ':') {
334 // This is the \:: escape sequence.
335 TokenPtr++;
336 }
Dmitri Gribenkof5e0aea2012-06-27 16:30:35 +0000337 StringRef UnescapedText(BufferPtr + 1, TokenPtr - (BufferPtr + 1));
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000338 formTokenWithChars(T, TokenPtr, tok::text);
Dmitri Gribenkof5e0aea2012-06-27 16:30:35 +0000339 T.setText(UnescapedText);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000340 return;
341 }
342
343 // Don't make zero-length commands.
Dmitri Gribenko8c05da32012-09-14 16:35:35 +0000344 if (!isCommandNameStartCharacter(*TokenPtr)) {
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000345 formTextToken(T, TokenPtr);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000346 return;
347 }
348
349 TokenPtr = skipCommandName(TokenPtr, CommentEnd);
350 unsigned Length = TokenPtr - (BufferPtr + 1);
351
352 // Hardcoded support for lexing LaTeX formula commands
353 // \f$ \f[ \f] \f{ \f} as a single command.
354 if (Length == 1 && TokenPtr[-1] == 'f' && TokenPtr != CommentEnd) {
355 C = *TokenPtr;
356 if (C == '$' || C == '[' || C == ']' || C == '{' || C == '}') {
357 TokenPtr++;
358 Length++;
359 }
360 }
361
362 const StringRef CommandName(BufferPtr + 1, Length);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000363
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000364 const CommandInfo *Info = Traits.getCommandInfoOrNULL(CommandName);
365 if (!Info) {
366 formTokenWithChars(T, TokenPtr, tok::unknown_command);
367 T.setUnknownCommandName(CommandName);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000368 return;
369 }
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000370 if (Info->IsVerbatimBlockCommand) {
371 setupAndLexVerbatimBlock(T, TokenPtr, *BufferPtr, Info);
372 return;
373 }
374 if (Info->IsVerbatimLineCommand) {
375 setupAndLexVerbatimLine(T, TokenPtr, Info);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000376 return;
377 }
378 formTokenWithChars(T, TokenPtr, tok::command);
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000379 T.setCommandID(Info->getID());
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000380 return;
381 }
382
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000383 case '&':
384 lexHTMLCharacterReference(T);
385 return;
386
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000387 case '<': {
388 TokenPtr++;
389 if (TokenPtr == CommentEnd) {
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000390 formTextToken(T, TokenPtr);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000391 return;
392 }
393 const char C = *TokenPtr;
Dmitri Gribenkoa99ec102012-07-09 21:32:40 +0000394 if (isHTMLIdentifierStartingCharacter(C))
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000395 setupAndLexHTMLStartTag(T);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000396 else if (C == '/')
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000397 setupAndLexHTMLEndTag(T);
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000398 else
399 formTextToken(T, TokenPtr);
400
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000401 return;
402 }
403
404 case '\n':
405 case '\r':
406 TokenPtr = skipNewline(TokenPtr, CommentEnd);
407 formTokenWithChars(T, TokenPtr, tok::newline);
408
409 if (CommentState == LCS_InsideCComment)
410 skipLineStartingDecorations();
411 return;
412
413 default: {
Dmitri Gribenkoaa7dbaf2012-12-30 19:45:46 +0000414 size_t End = StringRef(TokenPtr, CommentEnd - TokenPtr).
415 find_first_of("\n\r\\@&<");
416 if (End != StringRef::npos)
417 TokenPtr += End;
418 else
419 TokenPtr = CommentEnd;
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000420 formTextToken(T, TokenPtr);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000421 return;
422 }
423 }
424 }
425}
426
427void Lexer::setupAndLexVerbatimBlock(Token &T,
428 const char *TextBegin,
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000429 char Marker, const CommandInfo *Info) {
430 assert(Info->IsVerbatimBlockCommand);
431
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000432 VerbatimBlockEndCommandName.clear();
433 VerbatimBlockEndCommandName.append(Marker == '\\' ? "\\" : "@");
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000434 VerbatimBlockEndCommandName.append(Info->EndCommandName);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000435
436 formTokenWithChars(T, TextBegin, tok::verbatim_block_begin);
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000437 T.setVerbatimBlockID(Info->getID());
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000438
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000439 // If there is a newline following the verbatim opening command, skip the
440 // newline so that we don't create an tok::verbatim_block_line with empty
441 // text content.
442 if (BufferPtr != CommentEnd) {
443 const char C = *BufferPtr;
444 if (C == '\n' || C == '\r') {
445 BufferPtr = skipNewline(BufferPtr, CommentEnd);
446 State = LS_VerbatimBlockBody;
447 return;
448 }
449 }
450
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000451 State = LS_VerbatimBlockFirstLine;
452}
453
454void Lexer::lexVerbatimBlockFirstLine(Token &T) {
Dmitri Gribenko64da4e52012-07-18 23:01:58 +0000455again:
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000456 assert(BufferPtr < CommentEnd);
457
458 // FIXME: It would be better to scan the text once, finding either the block
459 // end command or newline.
460 //
461 // Extract current line.
462 const char *Newline = findNewline(BufferPtr, CommentEnd);
463 StringRef Line(BufferPtr, Newline - BufferPtr);
464
465 // Look for end command in current line.
466 size_t Pos = Line.find(VerbatimBlockEndCommandName);
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000467 const char *TextEnd;
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000468 const char *NextLine;
469 if (Pos == StringRef::npos) {
470 // Current line is completely verbatim.
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000471 TextEnd = Newline;
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000472 NextLine = skipNewline(Newline, CommentEnd);
473 } else if (Pos == 0) {
474 // Current line contains just an end command.
475 const char *End = BufferPtr + VerbatimBlockEndCommandName.size();
Dmitri Gribenkof5e0aea2012-06-27 16:30:35 +0000476 StringRef Name(BufferPtr + 1, End - (BufferPtr + 1));
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000477 formTokenWithChars(T, End, tok::verbatim_block_end);
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000478 T.setVerbatimBlockID(Traits.getCommandInfo(Name)->getID());
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000479 State = LS_Normal;
480 return;
481 } else {
482 // There is some text, followed by end command. Extract text first.
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000483 TextEnd = BufferPtr + Pos;
484 NextLine = TextEnd;
Dmitri Gribenko64da4e52012-07-18 23:01:58 +0000485 // If there is only whitespace before end command, skip whitespace.
486 if (isWhitespace(BufferPtr, TextEnd)) {
487 BufferPtr = TextEnd;
488 goto again;
489 }
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000490 }
491
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000492 StringRef Text(BufferPtr, TextEnd - BufferPtr);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000493 formTokenWithChars(T, NextLine, tok::verbatim_block_line);
Dmitri Gribenkof5e0aea2012-06-27 16:30:35 +0000494 T.setVerbatimBlockText(Text);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000495
496 State = LS_VerbatimBlockBody;
497}
498
499void Lexer::lexVerbatimBlockBody(Token &T) {
500 assert(State == LS_VerbatimBlockBody);
501
502 if (CommentState == LCS_InsideCComment)
503 skipLineStartingDecorations();
504
505 lexVerbatimBlockFirstLine(T);
506}
507
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000508void Lexer::setupAndLexVerbatimLine(Token &T, const char *TextBegin,
509 const CommandInfo *Info) {
510 assert(Info->IsVerbatimLineCommand);
Dmitri Gribenko962668d2012-06-27 16:53:58 +0000511 formTokenWithChars(T, TextBegin, tok::verbatim_line_name);
Dmitri Gribenkoe4330a32012-09-10 20:32:42 +0000512 T.setVerbatimLineID(Info->getID());
Dmitri Gribenko962668d2012-06-27 16:53:58 +0000513
514 State = LS_VerbatimLineText;
515}
516
517void Lexer::lexVerbatimLineText(Token &T) {
518 assert(State == LS_VerbatimLineText);
519
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000520 // Extract current line.
521 const char *Newline = findNewline(BufferPtr, CommentEnd);
Dmitri Gribenko962668d2012-06-27 16:53:58 +0000522 const StringRef Text(BufferPtr, Newline - BufferPtr);
523 formTokenWithChars(T, Newline, tok::verbatim_line_text);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000524 T.setVerbatimLineText(Text);
Dmitri Gribenko962668d2012-06-27 16:53:58 +0000525
526 State = LS_Normal;
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000527}
528
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000529void Lexer::lexHTMLCharacterReference(Token &T) {
530 const char *TokenPtr = BufferPtr;
531 assert(*TokenPtr == '&');
532 TokenPtr++;
533 if (TokenPtr == CommentEnd) {
534 formTextToken(T, TokenPtr);
535 return;
536 }
537 const char *NamePtr;
538 bool isNamed = false;
539 bool isDecimal = false;
540 char C = *TokenPtr;
541 if (isHTMLNamedCharacterReferenceCharacter(C)) {
542 NamePtr = TokenPtr;
543 TokenPtr = skipNamedCharacterReference(TokenPtr, CommentEnd);
544 isNamed = true;
545 } else if (C == '#') {
546 TokenPtr++;
547 if (TokenPtr == CommentEnd) {
548 formTextToken(T, TokenPtr);
549 return;
550 }
551 C = *TokenPtr;
552 if (isHTMLDecimalCharacterReferenceCharacter(C)) {
553 NamePtr = TokenPtr;
554 TokenPtr = skipDecimalCharacterReference(TokenPtr, CommentEnd);
555 isDecimal = true;
556 } else if (C == 'x' || C == 'X') {
557 TokenPtr++;
558 NamePtr = TokenPtr;
559 TokenPtr = skipHexCharacterReference(TokenPtr, CommentEnd);
560 } else {
561 formTextToken(T, TokenPtr);
562 return;
563 }
564 } else {
565 formTextToken(T, TokenPtr);
566 return;
567 }
568 if (NamePtr == TokenPtr || TokenPtr == CommentEnd ||
569 *TokenPtr != ';') {
570 formTextToken(T, TokenPtr);
571 return;
572 }
573 StringRef Name(NamePtr, TokenPtr - NamePtr);
574 TokenPtr++; // Skip semicolon.
575 StringRef Resolved;
576 if (isNamed)
577 Resolved = resolveHTMLNamedCharacterReference(Name);
578 else if (isDecimal)
579 Resolved = resolveHTMLDecimalCharacterReference(Name);
580 else
581 Resolved = resolveHTMLHexCharacterReference(Name);
582
583 if (Resolved.empty()) {
584 formTextToken(T, TokenPtr);
585 return;
586 }
587 formTokenWithChars(T, TokenPtr, tok::text);
588 T.setText(Resolved);
589 return;
590}
591
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000592void Lexer::setupAndLexHTMLStartTag(Token &T) {
Dmitri Gribenkoa99ec102012-07-09 21:32:40 +0000593 assert(BufferPtr[0] == '<' &&
594 isHTMLIdentifierStartingCharacter(BufferPtr[1]));
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000595 const char *TagNameEnd = skipHTMLIdentifier(BufferPtr + 2, CommentEnd);
Dmitri Gribenkof5e0aea2012-06-27 16:30:35 +0000596 StringRef Name(BufferPtr + 1, TagNameEnd - (BufferPtr + 1));
Dmitri Gribenko834a5bd2012-08-22 22:56:08 +0000597 if (!isHTMLTagName(Name)) {
598 formTextToken(T, TagNameEnd);
599 return;
600 }
601
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000602 formTokenWithChars(T, TagNameEnd, tok::html_start_tag);
603 T.setHTMLTagStartName(Name);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000604
605 BufferPtr = skipWhitespace(BufferPtr, CommentEnd);
606
Dmitri Gribenkoa99ec102012-07-09 21:32:40 +0000607 const char C = *BufferPtr;
608 if (BufferPtr != CommentEnd &&
Dmitri Gribenkoa5ef44f2012-07-11 21:38:39 +0000609 (C == '>' || C == '/' || isHTMLIdentifierStartingCharacter(C)))
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000610 State = LS_HTMLStartTag;
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000611}
612
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000613void Lexer::lexHTMLStartTag(Token &T) {
614 assert(State == LS_HTMLStartTag);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000615
616 const char *TokenPtr = BufferPtr;
617 char C = *TokenPtr;
618 if (isHTMLIdentifierCharacter(C)) {
619 TokenPtr = skipHTMLIdentifier(TokenPtr, CommentEnd);
Dmitri Gribenkof5e0aea2012-06-27 16:30:35 +0000620 StringRef Ident(BufferPtr, TokenPtr - BufferPtr);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000621 formTokenWithChars(T, TokenPtr, tok::html_ident);
Dmitri Gribenkof5e0aea2012-06-27 16:30:35 +0000622 T.setHTMLIdent(Ident);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000623 } else {
624 switch (C) {
625 case '=':
626 TokenPtr++;
627 formTokenWithChars(T, TokenPtr, tok::html_equals);
628 break;
629 case '\"':
630 case '\'': {
631 const char *OpenQuote = TokenPtr;
632 TokenPtr = skipHTMLQuotedString(TokenPtr, CommentEnd);
633 const char *ClosingQuote = TokenPtr;
634 if (TokenPtr != CommentEnd) // Skip closing quote.
635 TokenPtr++;
636 formTokenWithChars(T, TokenPtr, tok::html_quoted_string);
637 T.setHTMLQuotedString(StringRef(OpenQuote + 1,
638 ClosingQuote - (OpenQuote + 1)));
639 break;
640 }
641 case '>':
642 TokenPtr++;
643 formTokenWithChars(T, TokenPtr, tok::html_greater);
Dmitri Gribenkoa99ec102012-07-09 21:32:40 +0000644 State = LS_Normal;
645 return;
Dmitri Gribenkoa5ef44f2012-07-11 21:38:39 +0000646 case '/':
647 TokenPtr++;
648 if (TokenPtr != CommentEnd && *TokenPtr == '>') {
649 TokenPtr++;
650 formTokenWithChars(T, TokenPtr, tok::html_slash_greater);
Dmitri Gribenko477a9f52012-07-27 20:37:06 +0000651 } else
652 formTextToken(T, TokenPtr);
653
Dmitri Gribenkoa5ef44f2012-07-11 21:38:39 +0000654 State = LS_Normal;
655 return;
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000656 }
657 }
658
659 // Now look ahead and return to normal state if we don't see any HTML tokens
660 // ahead.
661 BufferPtr = skipWhitespace(BufferPtr, CommentEnd);
662 if (BufferPtr == CommentEnd) {
663 State = LS_Normal;
664 return;
665 }
666
667 C = *BufferPtr;
Dmitri Gribenkoa99ec102012-07-09 21:32:40 +0000668 if (!isHTMLIdentifierStartingCharacter(C) &&
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000669 C != '=' && C != '\"' && C != '\'' && C != '>') {
670 State = LS_Normal;
671 return;
672 }
673}
674
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000675void Lexer::setupAndLexHTMLEndTag(Token &T) {
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000676 assert(BufferPtr[0] == '<' && BufferPtr[1] == '/');
677
678 const char *TagNameBegin = skipWhitespace(BufferPtr + 2, CommentEnd);
679 const char *TagNameEnd = skipHTMLIdentifier(TagNameBegin, CommentEnd);
Dmitri Gribenko834a5bd2012-08-22 22:56:08 +0000680 StringRef Name(TagNameBegin, TagNameEnd - TagNameBegin);
681 if (!isHTMLTagName(Name)) {
682 formTextToken(T, TagNameEnd);
683 return;
684 }
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000685
686 const char *End = skipWhitespace(TagNameEnd, CommentEnd);
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000687
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000688 formTokenWithChars(T, End, tok::html_end_tag);
Dmitri Gribenko834a5bd2012-08-22 22:56:08 +0000689 T.setHTMLTagEndName(Name);
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000690
691 if (BufferPtr != CommentEnd && *BufferPtr == '>')
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000692 State = LS_HTMLEndTag;
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000693}
694
Dmitri Gribenko3f38bf22012-07-13 00:44:24 +0000695void Lexer::lexHTMLEndTag(Token &T) {
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000696 assert(BufferPtr != CommentEnd && *BufferPtr == '>');
697
698 formTokenWithChars(T, BufferPtr + 1, tok::html_greater);
699 State = LS_Normal;
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000700}
701
Dmitri Gribenkoaa580812012-08-09 00:03:17 +0000702Lexer::Lexer(llvm::BumpPtrAllocator &Allocator, const CommandTraits &Traits,
Dmitri Gribenkoaf503a62012-08-31 10:35:30 +0000703 SourceLocation FileLoc,
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000704 const char *BufferStart, const char *BufferEnd):
Dmitri Gribenkoaa580812012-08-09 00:03:17 +0000705 Allocator(Allocator), Traits(Traits),
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000706 BufferStart(BufferStart), BufferEnd(BufferEnd),
Dmitri Gribenkoaf503a62012-08-31 10:35:30 +0000707 FileLoc(FileLoc), BufferPtr(BufferStart),
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000708 CommentState(LCS_BeforeComment), State(LS_Normal) {
709}
710
711void Lexer::lex(Token &T) {
712again:
713 switch (CommentState) {
714 case LCS_BeforeComment:
715 if (BufferPtr == BufferEnd) {
716 formTokenWithChars(T, BufferPtr, tok::eof);
717 return;
718 }
719
720 assert(*BufferPtr == '/');
721 BufferPtr++; // Skip first slash.
722 switch(*BufferPtr) {
723 case '/': { // BCPL comment.
724 BufferPtr++; // Skip second slash.
725
726 if (BufferPtr != BufferEnd) {
727 // Skip Doxygen magic marker, if it is present.
728 // It might be missing because of a typo //< or /*<, or because we
729 // merged this non-Doxygen comment into a bunch of Doxygen comments
730 // around it: /** ... */ /* ... */ /** ... */
731 const char C = *BufferPtr;
732 if (C == '/' || C == '!')
733 BufferPtr++;
734 }
735
736 // Skip less-than symbol that marks trailing comments.
737 // Skip it even if the comment is not a Doxygen one, because //< and /*<
738 // are frequent typos.
739 if (BufferPtr != BufferEnd && *BufferPtr == '<')
740 BufferPtr++;
741
742 CommentState = LCS_InsideBCPLComment;
Dmitri Gribenko8d3ba232012-07-06 00:28:32 +0000743 if (State != LS_VerbatimBlockBody && State != LS_VerbatimBlockFirstLine)
744 State = LS_Normal;
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000745 CommentEnd = findBCPLCommentEnd(BufferPtr, BufferEnd);
746 goto again;
747 }
748 case '*': { // C comment.
749 BufferPtr++; // Skip star.
750
751 // Skip Doxygen magic marker.
752 const char C = *BufferPtr;
753 if ((C == '*' && *(BufferPtr + 1) != '/') || C == '!')
754 BufferPtr++;
755
756 // Skip less-than symbol that marks trailing comments.
757 if (BufferPtr != BufferEnd && *BufferPtr == '<')
758 BufferPtr++;
759
760 CommentState = LCS_InsideCComment;
761 State = LS_Normal;
762 CommentEnd = findCCommentEnd(BufferPtr, BufferEnd);
763 goto again;
764 }
765 default:
766 llvm_unreachable("second character of comment should be '/' or '*'");
767 }
768
769 case LCS_BetweenComments: {
770 // Consecutive comments are extracted only if there is only whitespace
771 // between them. So we can search for the start of the next comment.
772 const char *EndWhitespace = BufferPtr;
773 while(EndWhitespace != BufferEnd && *EndWhitespace != '/')
774 EndWhitespace++;
775
776 // Turn any whitespace between comments (and there is only whitespace
Dmitri Gribenkoa99ec102012-07-09 21:32:40 +0000777 // between them -- guaranteed by comment extraction) into a newline. We
778 // have two newlines between C comments in total (first one was synthesized
779 // after a comment).
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000780 formTokenWithChars(T, EndWhitespace, tok::newline);
781
782 CommentState = LCS_BeforeComment;
783 break;
784 }
785
786 case LCS_InsideBCPLComment:
787 case LCS_InsideCComment:
788 if (BufferPtr != CommentEnd) {
789 lexCommentText(T);
790 break;
791 } else {
792 // Skip C comment closing sequence.
793 if (CommentState == LCS_InsideCComment) {
794 assert(BufferPtr[0] == '*' && BufferPtr[1] == '/');
795 BufferPtr += 2;
796 assert(BufferPtr <= BufferEnd);
797
798 // Synthenize newline just after the C comment, regardless if there is
799 // actually a newline.
800 formTokenWithChars(T, BufferPtr, tok::newline);
801
802 CommentState = LCS_BetweenComments;
803 break;
804 } else {
805 // Don't synthesized a newline after BCPL comment.
806 CommentState = LCS_BetweenComments;
807 goto again;
808 }
809 }
810 }
811}
812
813StringRef Lexer::getSpelling(const Token &Tok,
814 const SourceManager &SourceMgr,
815 bool *Invalid) const {
816 SourceLocation Loc = Tok.getLocation();
817 std::pair<FileID, unsigned> LocInfo = SourceMgr.getDecomposedLoc(Loc);
818
819 bool InvalidTemp = false;
820 StringRef File = SourceMgr.getBufferData(LocInfo.first, &InvalidTemp);
821 if (InvalidTemp) {
822 *Invalid = true;
823 return StringRef();
824 }
825
826 const char *Begin = File.data() + LocInfo.second;
827 return StringRef(Begin, Tok.getLength());
828}
829
Dmitri Gribenko2d44d772012-06-26 20:39:18 +0000830} // end namespace comments
831} // end namespace clang
832