15188c4b9SDmitri Gribenko #include "clang/AST/CommentLexer.h"
2ca7f80adSDmitri Gribenko #include "clang/AST/CommentCommandTraits.h"
36738e430SFariborz Jahanian #include "clang/AST/CommentDiagnostic.h"
4bcef3411SDmitri Gribenko #include "clang/Basic/CharInfo.h"
5b2e5482bSDmitri Gribenko #include "llvm/ADT/StringExtras.h"
65188c4b9SDmitri Gribenko #include "llvm/ADT/StringSwitch.h"
79feeef40SDmitri Gribenko #include "llvm/Support/ConvertUTF.h"
85188c4b9SDmitri Gribenko #include "llvm/Support/ErrorHandling.h"
95188c4b9SDmitri Gribenko 
105188c4b9SDmitri Gribenko namespace clang {
115188c4b9SDmitri Gribenko namespace comments {
125188c4b9SDmitri Gribenko 
135188c4b9SDmitri Gribenko void Token::dump(const Lexer &L, const SourceManager &SM) const {
145188c4b9SDmitri Gribenko   llvm::errs() << "comments::Token Kind=" << Kind << " ";
155188c4b9SDmitri Gribenko   Loc.dump(SM);
165188c4b9SDmitri Gribenko   llvm::errs() << " " << Length << " \"" << L.getSpelling(*this, SM) << "\"\n";
175188c4b9SDmitri Gribenko }
185188c4b9SDmitri Gribenko 
1974f4d028SDmitri Gribenko static inline bool isHTMLNamedCharacterReferenceCharacter(char C) {
20bcef3411SDmitri Gribenko   return isLetter(C);
214586df76SDmitri Gribenko }
224586df76SDmitri Gribenko 
2374f4d028SDmitri Gribenko static inline bool isHTMLDecimalCharacterReferenceCharacter(char C) {
24bcef3411SDmitri Gribenko   return isDigit(C);
254586df76SDmitri Gribenko }
264586df76SDmitri Gribenko 
2774f4d028SDmitri Gribenko static inline bool isHTMLHexCharacterReferenceCharacter(char C) {
28bcef3411SDmitri Gribenko   return isHexDigit(C);
294586df76SDmitri Gribenko }
30107618a6SDmitri Gribenko 
3174f4d028SDmitri Gribenko static inline StringRef convertCodePointToUTF8(
3274f4d028SDmitri Gribenko                                       llvm::BumpPtrAllocator &Allocator,
3328800da1SDmitri Gribenko                                       unsigned CodePoint) {
344586df76SDmitri Gribenko   char *Resolved = Allocator.Allocate<char>(UNI_MAX_UTF8_BYTES_PER_CODE_POINT);
354586df76SDmitri Gribenko   char *ResolvedPtr = Resolved;
369feeef40SDmitri Gribenko   if (llvm::ConvertCodePointToUTF8(CodePoint, ResolvedPtr))
374586df76SDmitri Gribenko     return StringRef(Resolved, ResolvedPtr - Resolved);
384586df76SDmitri Gribenko   else
394586df76SDmitri Gribenko     return StringRef();
404586df76SDmitri Gribenko }
414586df76SDmitri Gribenko 
4274f4d028SDmitri Gribenko namespace {
4374f4d028SDmitri Gribenko 
4428800da1SDmitri Gribenko #include "clang/AST/CommentHTMLTags.inc"
4528800da1SDmitri Gribenko #include "clang/AST/CommentHTMLNamedCharacterReferences.inc"
4628800da1SDmitri Gribenko 
4728800da1SDmitri Gribenko } // unnamed namespace
487b3ae190SFariborz Jahanian 
497b3ae190SFariborz Jahanian StringRef Lexer::resolveHTMLNamedCharacterReference(StringRef Name) const {
5028800da1SDmitri Gribenko   // Fast path, first check a few most widely used named character references.
517b3ae190SFariborz Jahanian   return llvm::StringSwitch<StringRef>(Name)
527b3ae190SFariborz Jahanian       .Case("amp", "&")
537b3ae190SFariborz Jahanian       .Case("lt", "<")
547b3ae190SFariborz Jahanian       .Case("gt", ">")
557b3ae190SFariborz Jahanian       .Case("quot", "\"")
567b3ae190SFariborz Jahanian       .Case("apos", "\'")
5728800da1SDmitri Gribenko       // Slow path.
5828800da1SDmitri Gribenko       .Default(translateHTMLNamedCharacterReferenceToUTF8(Name));
597b3ae190SFariborz Jahanian }
607b3ae190SFariborz Jahanian 
617b3ae190SFariborz Jahanian StringRef Lexer::resolveHTMLDecimalCharacterReference(StringRef Name) const {
624586df76SDmitri Gribenko   unsigned CodePoint = 0;
634586df76SDmitri Gribenko   for (unsigned i = 0, e = Name.size(); i != e; ++i) {
647b3ae190SFariborz Jahanian     assert(isHTMLDecimalCharacterReferenceCharacter(Name[i]));
657b3ae190SFariborz Jahanian     CodePoint *= 10;
667b3ae190SFariborz Jahanian     CodePoint += Name[i] - '0';
674586df76SDmitri Gribenko   }
6828800da1SDmitri Gribenko   return convertCodePointToUTF8(Allocator, CodePoint);
6928800da1SDmitri Gribenko }
704586df76SDmitri Gribenko 
7128800da1SDmitri Gribenko StringRef Lexer::resolveHTMLHexCharacterReference(StringRef Name) const {
7228800da1SDmitri Gribenko   unsigned CodePoint = 0;
7328800da1SDmitri Gribenko   for (unsigned i = 0, e = Name.size(); i != e; ++i) {
7428800da1SDmitri Gribenko     CodePoint *= 16;
7528800da1SDmitri Gribenko     const char C = Name[i];
7628800da1SDmitri Gribenko     assert(isHTMLHexCharacterReferenceCharacter(C));
7728800da1SDmitri Gribenko     CodePoint += llvm::hexDigitValue(C);
7828800da1SDmitri Gribenko   }
7928800da1SDmitri Gribenko   return convertCodePointToUTF8(Allocator, CodePoint);
804586df76SDmitri Gribenko }
814586df76SDmitri Gribenko 
825188c4b9SDmitri Gribenko void Lexer::skipLineStartingDecorations() {
835188c4b9SDmitri Gribenko   // This function should be called only for C comments
845188c4b9SDmitri Gribenko   assert(CommentState == LCS_InsideCComment);
855188c4b9SDmitri Gribenko 
865188c4b9SDmitri Gribenko   if (BufferPtr == CommentEnd)
875188c4b9SDmitri Gribenko     return;
885188c4b9SDmitri Gribenko 
895188c4b9SDmitri Gribenko   switch (*BufferPtr) {
905188c4b9SDmitri Gribenko   case ' ':
915188c4b9SDmitri Gribenko   case '\t':
925188c4b9SDmitri Gribenko   case '\f':
935188c4b9SDmitri Gribenko   case '\v': {
945188c4b9SDmitri Gribenko     const char *NewBufferPtr = BufferPtr;
955188c4b9SDmitri Gribenko     NewBufferPtr++;
965188c4b9SDmitri Gribenko     if (NewBufferPtr == CommentEnd)
975188c4b9SDmitri Gribenko       return;
985188c4b9SDmitri Gribenko 
995188c4b9SDmitri Gribenko     char C = *NewBufferPtr;
100bcef3411SDmitri Gribenko     while (isHorizontalWhitespace(C)) {
1015188c4b9SDmitri Gribenko       NewBufferPtr++;
1025188c4b9SDmitri Gribenko       if (NewBufferPtr == CommentEnd)
1035188c4b9SDmitri Gribenko         return;
1045188c4b9SDmitri Gribenko       C = *NewBufferPtr;
1055188c4b9SDmitri Gribenko     }
1065188c4b9SDmitri Gribenko     if (C == '*')
1075188c4b9SDmitri Gribenko       BufferPtr = NewBufferPtr + 1;
1085188c4b9SDmitri Gribenko     break;
1095188c4b9SDmitri Gribenko   }
1105188c4b9SDmitri Gribenko   case '*':
1115188c4b9SDmitri Gribenko     BufferPtr++;
1125188c4b9SDmitri Gribenko     break;
1135188c4b9SDmitri Gribenko   }
1145188c4b9SDmitri Gribenko }
1155188c4b9SDmitri Gribenko 
1165188c4b9SDmitri Gribenko namespace {
117ec92531cSDmitri Gribenko /// Returns pointer to the first newline character in the string.
1185188c4b9SDmitri Gribenko const char *findNewline(const char *BufferPtr, const char *BufferEnd) {
1195188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
120bcef3411SDmitri Gribenko     if (isVerticalWhitespace(*BufferPtr))
1215188c4b9SDmitri Gribenko       return BufferPtr;
1225188c4b9SDmitri Gribenko   }
1235188c4b9SDmitri Gribenko   return BufferEnd;
1245188c4b9SDmitri Gribenko }
1255188c4b9SDmitri Gribenko 
1265188c4b9SDmitri Gribenko const char *skipNewline(const char *BufferPtr, const char *BufferEnd) {
1275188c4b9SDmitri Gribenko   if (BufferPtr == BufferEnd)
1285188c4b9SDmitri Gribenko     return BufferPtr;
1295188c4b9SDmitri Gribenko 
1305188c4b9SDmitri Gribenko   if (*BufferPtr == '\n')
1315188c4b9SDmitri Gribenko     BufferPtr++;
1325188c4b9SDmitri Gribenko   else {
1335188c4b9SDmitri Gribenko     assert(*BufferPtr == '\r');
1345188c4b9SDmitri Gribenko     BufferPtr++;
1355188c4b9SDmitri Gribenko     if (BufferPtr != BufferEnd && *BufferPtr == '\n')
1365188c4b9SDmitri Gribenko       BufferPtr++;
1375188c4b9SDmitri Gribenko   }
1385188c4b9SDmitri Gribenko   return BufferPtr;
1395188c4b9SDmitri Gribenko }
1405188c4b9SDmitri Gribenko 
1414586df76SDmitri Gribenko const char *skipNamedCharacterReference(const char *BufferPtr,
1424586df76SDmitri Gribenko                                         const char *BufferEnd) {
1434586df76SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1444586df76SDmitri Gribenko     if (!isHTMLNamedCharacterReferenceCharacter(*BufferPtr))
1454586df76SDmitri Gribenko       return BufferPtr;
1464586df76SDmitri Gribenko   }
1474586df76SDmitri Gribenko   return BufferEnd;
1484586df76SDmitri Gribenko }
1494586df76SDmitri Gribenko 
1504586df76SDmitri Gribenko const char *skipDecimalCharacterReference(const char *BufferPtr,
1514586df76SDmitri Gribenko                                           const char *BufferEnd) {
1524586df76SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1534586df76SDmitri Gribenko     if (!isHTMLDecimalCharacterReferenceCharacter(*BufferPtr))
1544586df76SDmitri Gribenko       return BufferPtr;
1554586df76SDmitri Gribenko   }
1564586df76SDmitri Gribenko   return BufferEnd;
1574586df76SDmitri Gribenko }
1584586df76SDmitri Gribenko 
1594586df76SDmitri Gribenko const char *skipHexCharacterReference(const char *BufferPtr,
1604586df76SDmitri Gribenko                                       const char *BufferEnd) {
1614586df76SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1624586df76SDmitri Gribenko     if (!isHTMLHexCharacterReferenceCharacter(*BufferPtr))
1634586df76SDmitri Gribenko       return BufferPtr;
1644586df76SDmitri Gribenko   }
1654586df76SDmitri Gribenko   return BufferEnd;
1664586df76SDmitri Gribenko }
1674586df76SDmitri Gribenko 
16817709ae8SDmitri Gribenko bool isHTMLIdentifierStartingCharacter(char C) {
169bcef3411SDmitri Gribenko   return isLetter(C);
17017709ae8SDmitri Gribenko }
17117709ae8SDmitri Gribenko 
1725188c4b9SDmitri Gribenko bool isHTMLIdentifierCharacter(char C) {
173bcef3411SDmitri Gribenko   return isAlphanumeric(C);
1745188c4b9SDmitri Gribenko }
1755188c4b9SDmitri Gribenko 
1765188c4b9SDmitri Gribenko const char *skipHTMLIdentifier(const char *BufferPtr, const char *BufferEnd) {
1775188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1785188c4b9SDmitri Gribenko     if (!isHTMLIdentifierCharacter(*BufferPtr))
1795188c4b9SDmitri Gribenko       return BufferPtr;
1805188c4b9SDmitri Gribenko   }
1815188c4b9SDmitri Gribenko   return BufferEnd;
1825188c4b9SDmitri Gribenko }
1835188c4b9SDmitri Gribenko 
1845188c4b9SDmitri Gribenko /// Skip HTML string quoted in single or double quotes.  Escaping quotes inside
1855188c4b9SDmitri Gribenko /// string allowed.
1865188c4b9SDmitri Gribenko ///
1875188c4b9SDmitri Gribenko /// Returns pointer to closing quote.
1885188c4b9SDmitri Gribenko const char *skipHTMLQuotedString(const char *BufferPtr, const char *BufferEnd)
1895188c4b9SDmitri Gribenko {
1905188c4b9SDmitri Gribenko   const char Quote = *BufferPtr;
1915188c4b9SDmitri Gribenko   assert(Quote == '\"' || Quote == '\'');
1925188c4b9SDmitri Gribenko 
1935188c4b9SDmitri Gribenko   BufferPtr++;
1945188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1955188c4b9SDmitri Gribenko     const char C = *BufferPtr;
1965188c4b9SDmitri Gribenko     if (C == Quote && BufferPtr[-1] != '\\')
1975188c4b9SDmitri Gribenko       return BufferPtr;
1985188c4b9SDmitri Gribenko   }
1995188c4b9SDmitri Gribenko   return BufferEnd;
2005188c4b9SDmitri Gribenko }
2015188c4b9SDmitri Gribenko 
2025188c4b9SDmitri Gribenko const char *skipWhitespace(const char *BufferPtr, const char *BufferEnd) {
2035188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2045188c4b9SDmitri Gribenko     if (!isWhitespace(*BufferPtr))
2055188c4b9SDmitri Gribenko       return BufferPtr;
2065188c4b9SDmitri Gribenko   }
2075188c4b9SDmitri Gribenko   return BufferEnd;
2085188c4b9SDmitri Gribenko }
2095188c4b9SDmitri Gribenko 
210e4a3997dSDmitri Gribenko bool isWhitespace(const char *BufferPtr, const char *BufferEnd) {
211e4a3997dSDmitri Gribenko   return skipWhitespace(BufferPtr, BufferEnd) == BufferEnd;
212e4a3997dSDmitri Gribenko }
213e4a3997dSDmitri Gribenko 
214ad45ad6fSDmitri Gribenko bool isCommandNameStartCharacter(char C) {
215bcef3411SDmitri Gribenko   return isLetter(C);
216ad45ad6fSDmitri Gribenko }
217ad45ad6fSDmitri Gribenko 
2185188c4b9SDmitri Gribenko bool isCommandNameCharacter(char C) {
219bcef3411SDmitri Gribenko   return isAlphanumeric(C);
2205188c4b9SDmitri Gribenko }
2215188c4b9SDmitri Gribenko 
2225188c4b9SDmitri Gribenko const char *skipCommandName(const char *BufferPtr, const char *BufferEnd) {
2235188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2245188c4b9SDmitri Gribenko     if (!isCommandNameCharacter(*BufferPtr))
2255188c4b9SDmitri Gribenko       return BufferPtr;
2265188c4b9SDmitri Gribenko   }
2275188c4b9SDmitri Gribenko   return BufferEnd;
2285188c4b9SDmitri Gribenko }
2295188c4b9SDmitri Gribenko 
2305188c4b9SDmitri Gribenko /// Return the one past end pointer for BCPL comments.
2315188c4b9SDmitri Gribenko /// Handles newlines escaped with backslash or trigraph for backslahs.
2325188c4b9SDmitri Gribenko const char *findBCPLCommentEnd(const char *BufferPtr, const char *BufferEnd) {
2335188c4b9SDmitri Gribenko   const char *CurPtr = BufferPtr;
2345188c4b9SDmitri Gribenko   while (CurPtr != BufferEnd) {
235bcef3411SDmitri Gribenko     while (!isVerticalWhitespace(*CurPtr)) {
2365188c4b9SDmitri Gribenko       CurPtr++;
2375188c4b9SDmitri Gribenko       if (CurPtr == BufferEnd)
2385188c4b9SDmitri Gribenko         return BufferEnd;
2395188c4b9SDmitri Gribenko     }
2405188c4b9SDmitri Gribenko     // We found a newline, check if it is escaped.
2415188c4b9SDmitri Gribenko     const char *EscapePtr = CurPtr - 1;
2425188c4b9SDmitri Gribenko     while(isHorizontalWhitespace(*EscapePtr))
2435188c4b9SDmitri Gribenko       EscapePtr--;
2445188c4b9SDmitri Gribenko 
2455188c4b9SDmitri Gribenko     if (*EscapePtr == '\\' ||
2465188c4b9SDmitri Gribenko         (EscapePtr - 2 >= BufferPtr && EscapePtr[0] == '/' &&
2475188c4b9SDmitri Gribenko          EscapePtr[-1] == '?' && EscapePtr[-2] == '?')) {
2485188c4b9SDmitri Gribenko       // We found an escaped newline.
2495188c4b9SDmitri Gribenko       CurPtr = skipNewline(CurPtr, BufferEnd);
2505188c4b9SDmitri Gribenko     } else
2515188c4b9SDmitri Gribenko       return CurPtr; // Not an escaped newline.
2525188c4b9SDmitri Gribenko   }
2535188c4b9SDmitri Gribenko   return BufferEnd;
2545188c4b9SDmitri Gribenko }
2555188c4b9SDmitri Gribenko 
2565188c4b9SDmitri Gribenko /// Return the one past end pointer for C comments.
2575188c4b9SDmitri Gribenko /// Very dumb, does not handle escaped newlines or trigraphs.
2585188c4b9SDmitri Gribenko const char *findCCommentEnd(const char *BufferPtr, const char *BufferEnd) {
2595188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2605188c4b9SDmitri Gribenko     if (*BufferPtr == '*') {
2615188c4b9SDmitri Gribenko       assert(BufferPtr + 1 != BufferEnd);
2625188c4b9SDmitri Gribenko       if (*(BufferPtr + 1) == '/')
2635188c4b9SDmitri Gribenko         return BufferPtr;
2645188c4b9SDmitri Gribenko     }
2655188c4b9SDmitri Gribenko   }
2665188c4b9SDmitri Gribenko   llvm_unreachable("buffer end hit before '*/' was seen");
2675188c4b9SDmitri Gribenko }
2686c7a1666SFariborz Jahanian 
2695188c4b9SDmitri Gribenko } // unnamed namespace
2705188c4b9SDmitri Gribenko 
2713ffab05fSAlp Toker void Lexer::formTokenWithChars(Token &Result, const char *TokEnd,
2723ffab05fSAlp Toker                                tok::TokenKind Kind) {
2733ffab05fSAlp Toker   const unsigned TokLen = TokEnd - BufferPtr;
2743ffab05fSAlp Toker   Result.setLocation(getSourceLocation(BufferPtr));
2753ffab05fSAlp Toker   Result.setKind(Kind);
2763ffab05fSAlp Toker   Result.setLength(TokLen);
2773ffab05fSAlp Toker #ifndef NDEBUG
2783ffab05fSAlp Toker   Result.TextPtr = "<UNSET>";
2793ffab05fSAlp Toker   Result.IntVal = 7;
2803ffab05fSAlp Toker #endif
2813ffab05fSAlp Toker   BufferPtr = TokEnd;
2823ffab05fSAlp Toker }
2833ffab05fSAlp Toker 
2845188c4b9SDmitri Gribenko void Lexer::lexCommentText(Token &T) {
2855188c4b9SDmitri Gribenko   assert(CommentState == LCS_InsideBCPLComment ||
2865188c4b9SDmitri Gribenko          CommentState == LCS_InsideCComment);
2875188c4b9SDmitri Gribenko 
2885188c4b9SDmitri Gribenko   switch (State) {
2895188c4b9SDmitri Gribenko   case LS_Normal:
2905188c4b9SDmitri Gribenko     break;
2915188c4b9SDmitri Gribenko   case LS_VerbatimBlockFirstLine:
2925188c4b9SDmitri Gribenko     lexVerbatimBlockFirstLine(T);
2935188c4b9SDmitri Gribenko     return;
2945188c4b9SDmitri Gribenko   case LS_VerbatimBlockBody:
2955188c4b9SDmitri Gribenko     lexVerbatimBlockBody(T);
2965188c4b9SDmitri Gribenko     return;
2971669f702SDmitri Gribenko   case LS_VerbatimLineText:
2981669f702SDmitri Gribenko     lexVerbatimLineText(T);
2991669f702SDmitri Gribenko     return;
300e00ffc7bSDmitri Gribenko   case LS_HTMLStartTag:
301e00ffc7bSDmitri Gribenko     lexHTMLStartTag(T);
3025188c4b9SDmitri Gribenko     return;
303e00ffc7bSDmitri Gribenko   case LS_HTMLEndTag:
304e00ffc7bSDmitri Gribenko     lexHTMLEndTag(T);
305ec92531cSDmitri Gribenko     return;
3065188c4b9SDmitri Gribenko   }
3075188c4b9SDmitri Gribenko 
3085188c4b9SDmitri Gribenko   assert(State == LS_Normal);
3095188c4b9SDmitri Gribenko 
3105188c4b9SDmitri Gribenko   const char *TokenPtr = BufferPtr;
3115188c4b9SDmitri Gribenko   assert(TokenPtr < CommentEnd);
3125188c4b9SDmitri Gribenko   while (TokenPtr != CommentEnd) {
3135188c4b9SDmitri Gribenko     switch(*TokenPtr) {
3145188c4b9SDmitri Gribenko       case '\\':
3155188c4b9SDmitri Gribenko       case '@': {
316bcf7f4d3SDmitri Gribenko         // Commands that start with a backslash and commands that start with
317bcf7f4d3SDmitri Gribenko         // 'at' have equivalent semantics.  But we keep information about the
318bcf7f4d3SDmitri Gribenko         // exact syntax in AST for comments.
319bcf7f4d3SDmitri Gribenko         tok::TokenKind CommandKind =
320bcf7f4d3SDmitri Gribenko             (*TokenPtr == '@') ? tok::at_command : tok::backslash_command;
3215188c4b9SDmitri Gribenko         TokenPtr++;
3225188c4b9SDmitri Gribenko         if (TokenPtr == CommentEnd) {
3234586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
3245188c4b9SDmitri Gribenko           return;
3255188c4b9SDmitri Gribenko         }
3265188c4b9SDmitri Gribenko         char C = *TokenPtr;
3275188c4b9SDmitri Gribenko         switch (C) {
3285188c4b9SDmitri Gribenko         default:
3295188c4b9SDmitri Gribenko           break;
3305188c4b9SDmitri Gribenko 
3315188c4b9SDmitri Gribenko         case '\\': case '@': case '&': case '$':
3325188c4b9SDmitri Gribenko         case '#':  case '<': case '>': case '%':
3335188c4b9SDmitri Gribenko         case '\"': case '.': case ':':
3345188c4b9SDmitri Gribenko           // This is one of \\ \@ \& \$ etc escape sequences.
3355188c4b9SDmitri Gribenko           TokenPtr++;
3365188c4b9SDmitri Gribenko           if (C == ':' && TokenPtr != CommentEnd && *TokenPtr == ':') {
3375188c4b9SDmitri Gribenko             // This is the \:: escape sequence.
3385188c4b9SDmitri Gribenko             TokenPtr++;
3395188c4b9SDmitri Gribenko           }
34060ddd8a1SDmitri Gribenko           StringRef UnescapedText(BufferPtr + 1, TokenPtr - (BufferPtr + 1));
3415188c4b9SDmitri Gribenko           formTokenWithChars(T, TokenPtr, tok::text);
34260ddd8a1SDmitri Gribenko           T.setText(UnescapedText);
3435188c4b9SDmitri Gribenko           return;
3445188c4b9SDmitri Gribenko         }
3455188c4b9SDmitri Gribenko 
3465188c4b9SDmitri Gribenko         // Don't make zero-length commands.
347ad45ad6fSDmitri Gribenko         if (!isCommandNameStartCharacter(*TokenPtr)) {
3484586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
3495188c4b9SDmitri Gribenko           return;
3505188c4b9SDmitri Gribenko         }
3515188c4b9SDmitri Gribenko 
3525188c4b9SDmitri Gribenko         TokenPtr = skipCommandName(TokenPtr, CommentEnd);
3535188c4b9SDmitri Gribenko         unsigned Length = TokenPtr - (BufferPtr + 1);
3545188c4b9SDmitri Gribenko 
3555188c4b9SDmitri Gribenko         // Hardcoded support for lexing LaTeX formula commands
3565188c4b9SDmitri Gribenko         // \f$ \f[ \f] \f{ \f} as a single command.
3575188c4b9SDmitri Gribenko         if (Length == 1 && TokenPtr[-1] == 'f' && TokenPtr != CommentEnd) {
3585188c4b9SDmitri Gribenko           C = *TokenPtr;
3595188c4b9SDmitri Gribenko           if (C == '$' || C == '[' || C == ']' || C == '{' || C == '}') {
3605188c4b9SDmitri Gribenko             TokenPtr++;
3615188c4b9SDmitri Gribenko             Length++;
3625188c4b9SDmitri Gribenko           }
3635188c4b9SDmitri Gribenko         }
3645188c4b9SDmitri Gribenko 
365*bf3e3270SCraig Topper         StringRef CommandName(BufferPtr + 1, Length);
3665188c4b9SDmitri Gribenko 
3677acbf00fSDmitri Gribenko         const CommandInfo *Info = Traits.getCommandInfoOrNULL(CommandName);
3687acbf00fSDmitri Gribenko         if (!Info) {
369ebb262f7SFariborz Jahanian           if ((Info = Traits.getTypoCorrectCommandInfo(CommandName))) {
3706c7a1666SFariborz Jahanian             StringRef CorrectedName = Info->Name;
3712dece574SBenjamin Kramer             SourceLocation Loc = getSourceLocation(BufferPtr);
3722dece574SBenjamin Kramer             SourceRange CommandRange(Loc.getLocWithOffset(1),
3732dece574SBenjamin Kramer                                      getSourceLocation(TokenPtr));
3742dece574SBenjamin Kramer             Diag(Loc, diag::warn_correct_comment_command_name)
3756c7a1666SFariborz Jahanian               << CommandName << CorrectedName
3766c7a1666SFariborz Jahanian               << FixItHint::CreateReplacement(CommandRange, CorrectedName);
3776c7a1666SFariborz Jahanian           } else {
3782dece574SBenjamin Kramer             formTokenWithChars(T, TokenPtr, tok::unknown_command);
3792dece574SBenjamin Kramer             T.setUnknownCommandName(CommandName);
3806738e430SFariborz Jahanian             Diag(T.getLocation(), diag::warn_unknown_comment_command_name);
3815188c4b9SDmitri Gribenko             return;
3825188c4b9SDmitri Gribenko           }
3836c7a1666SFariborz Jahanian         }
3847acbf00fSDmitri Gribenko         if (Info->IsVerbatimBlockCommand) {
3857acbf00fSDmitri Gribenko           setupAndLexVerbatimBlock(T, TokenPtr, *BufferPtr, Info);
3867acbf00fSDmitri Gribenko           return;
3877acbf00fSDmitri Gribenko         }
3887acbf00fSDmitri Gribenko         if (Info->IsVerbatimLineCommand) {
3897acbf00fSDmitri Gribenko           setupAndLexVerbatimLine(T, TokenPtr, Info);
3905188c4b9SDmitri Gribenko           return;
3915188c4b9SDmitri Gribenko         }
392bcf7f4d3SDmitri Gribenko         formTokenWithChars(T, TokenPtr, CommandKind);
3937acbf00fSDmitri Gribenko         T.setCommandID(Info->getID());
3945188c4b9SDmitri Gribenko         return;
3955188c4b9SDmitri Gribenko       }
3965188c4b9SDmitri Gribenko 
3974586df76SDmitri Gribenko       case '&':
3984586df76SDmitri Gribenko         lexHTMLCharacterReference(T);
3994586df76SDmitri Gribenko         return;
4004586df76SDmitri Gribenko 
4015188c4b9SDmitri Gribenko       case '<': {
4025188c4b9SDmitri Gribenko         TokenPtr++;
4035188c4b9SDmitri Gribenko         if (TokenPtr == CommentEnd) {
4044586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
4055188c4b9SDmitri Gribenko           return;
4065188c4b9SDmitri Gribenko         }
4075188c4b9SDmitri Gribenko         const char C = *TokenPtr;
40817709ae8SDmitri Gribenko         if (isHTMLIdentifierStartingCharacter(C))
409e00ffc7bSDmitri Gribenko           setupAndLexHTMLStartTag(T);
4105188c4b9SDmitri Gribenko         else if (C == '/')
411e00ffc7bSDmitri Gribenko           setupAndLexHTMLEndTag(T);
4124586df76SDmitri Gribenko         else
4134586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
4144586df76SDmitri Gribenko 
4155188c4b9SDmitri Gribenko         return;
4165188c4b9SDmitri Gribenko       }
4175188c4b9SDmitri Gribenko 
4185188c4b9SDmitri Gribenko       case '\n':
4195188c4b9SDmitri Gribenko       case '\r':
4205188c4b9SDmitri Gribenko         TokenPtr = skipNewline(TokenPtr, CommentEnd);
4215188c4b9SDmitri Gribenko         formTokenWithChars(T, TokenPtr, tok::newline);
4225188c4b9SDmitri Gribenko 
4235188c4b9SDmitri Gribenko         if (CommentState == LCS_InsideCComment)
4245188c4b9SDmitri Gribenko           skipLineStartingDecorations();
4255188c4b9SDmitri Gribenko         return;
4265188c4b9SDmitri Gribenko 
4275188c4b9SDmitri Gribenko       default: {
42810af67a9SDmitri Gribenko         size_t End = StringRef(TokenPtr, CommentEnd - TokenPtr).
42910af67a9SDmitri Gribenko                          find_first_of("\n\r\\@&<");
43010af67a9SDmitri Gribenko         if (End != StringRef::npos)
43110af67a9SDmitri Gribenko           TokenPtr += End;
43210af67a9SDmitri Gribenko         else
43310af67a9SDmitri Gribenko           TokenPtr = CommentEnd;
4344586df76SDmitri Gribenko         formTextToken(T, TokenPtr);
4355188c4b9SDmitri Gribenko         return;
4365188c4b9SDmitri Gribenko       }
4375188c4b9SDmitri Gribenko     }
4385188c4b9SDmitri Gribenko   }
4395188c4b9SDmitri Gribenko }
4405188c4b9SDmitri Gribenko 
4415188c4b9SDmitri Gribenko void Lexer::setupAndLexVerbatimBlock(Token &T,
4425188c4b9SDmitri Gribenko                                      const char *TextBegin,
4437acbf00fSDmitri Gribenko                                      char Marker, const CommandInfo *Info) {
4447acbf00fSDmitri Gribenko   assert(Info->IsVerbatimBlockCommand);
4457acbf00fSDmitri Gribenko 
4465188c4b9SDmitri Gribenko   VerbatimBlockEndCommandName.clear();
4475188c4b9SDmitri Gribenko   VerbatimBlockEndCommandName.append(Marker == '\\' ? "\\" : "@");
4487acbf00fSDmitri Gribenko   VerbatimBlockEndCommandName.append(Info->EndCommandName);
4495188c4b9SDmitri Gribenko 
4505188c4b9SDmitri Gribenko   formTokenWithChars(T, TextBegin, tok::verbatim_block_begin);
4517acbf00fSDmitri Gribenko   T.setVerbatimBlockID(Info->getID());
4525188c4b9SDmitri Gribenko 
453ec92531cSDmitri Gribenko   // If there is a newline following the verbatim opening command, skip the
454ec92531cSDmitri Gribenko   // newline so that we don't create an tok::verbatim_block_line with empty
455ec92531cSDmitri Gribenko   // text content.
456bcef3411SDmitri Gribenko   if (BufferPtr != CommentEnd &&
457bcef3411SDmitri Gribenko       isVerticalWhitespace(*BufferPtr)) {
458ec92531cSDmitri Gribenko     BufferPtr = skipNewline(BufferPtr, CommentEnd);
459ec92531cSDmitri Gribenko     State = LS_VerbatimBlockBody;
460ec92531cSDmitri Gribenko     return;
461ec92531cSDmitri Gribenko   }
462ec92531cSDmitri Gribenko 
4635188c4b9SDmitri Gribenko   State = LS_VerbatimBlockFirstLine;
4645188c4b9SDmitri Gribenko }
4655188c4b9SDmitri Gribenko 
4665188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockFirstLine(Token &T) {
467e4a3997dSDmitri Gribenko again:
4685188c4b9SDmitri Gribenko   assert(BufferPtr < CommentEnd);
4695188c4b9SDmitri Gribenko 
4705188c4b9SDmitri Gribenko   // FIXME: It would be better to scan the text once, finding either the block
4715188c4b9SDmitri Gribenko   // end command or newline.
4725188c4b9SDmitri Gribenko   //
4735188c4b9SDmitri Gribenko   // Extract current line.
4745188c4b9SDmitri Gribenko   const char *Newline = findNewline(BufferPtr, CommentEnd);
4755188c4b9SDmitri Gribenko   StringRef Line(BufferPtr, Newline - BufferPtr);
4765188c4b9SDmitri Gribenko 
4775188c4b9SDmitri Gribenko   // Look for end command in current line.
4785188c4b9SDmitri Gribenko   size_t Pos = Line.find(VerbatimBlockEndCommandName);
479ec92531cSDmitri Gribenko   const char *TextEnd;
4805188c4b9SDmitri Gribenko   const char *NextLine;
4815188c4b9SDmitri Gribenko   if (Pos == StringRef::npos) {
4825188c4b9SDmitri Gribenko     // Current line is completely verbatim.
483ec92531cSDmitri Gribenko     TextEnd = Newline;
4845188c4b9SDmitri Gribenko     NextLine = skipNewline(Newline, CommentEnd);
4855188c4b9SDmitri Gribenko   } else if (Pos == 0) {
4865188c4b9SDmitri Gribenko     // Current line contains just an end command.
4875188c4b9SDmitri Gribenko     const char *End = BufferPtr + VerbatimBlockEndCommandName.size();
48860ddd8a1SDmitri Gribenko     StringRef Name(BufferPtr + 1, End - (BufferPtr + 1));
4895188c4b9SDmitri Gribenko     formTokenWithChars(T, End, tok::verbatim_block_end);
4907acbf00fSDmitri Gribenko     T.setVerbatimBlockID(Traits.getCommandInfo(Name)->getID());
4915188c4b9SDmitri Gribenko     State = LS_Normal;
4925188c4b9SDmitri Gribenko     return;
4935188c4b9SDmitri Gribenko   } else {
4945188c4b9SDmitri Gribenko     // There is some text, followed by end command.  Extract text first.
495ec92531cSDmitri Gribenko     TextEnd = BufferPtr + Pos;
496ec92531cSDmitri Gribenko     NextLine = TextEnd;
497e4a3997dSDmitri Gribenko     // If there is only whitespace before end command, skip whitespace.
498e4a3997dSDmitri Gribenko     if (isWhitespace(BufferPtr, TextEnd)) {
499e4a3997dSDmitri Gribenko       BufferPtr = TextEnd;
500e4a3997dSDmitri Gribenko       goto again;
501e4a3997dSDmitri Gribenko     }
5025188c4b9SDmitri Gribenko   }
5035188c4b9SDmitri Gribenko 
504ec92531cSDmitri Gribenko   StringRef Text(BufferPtr, TextEnd - BufferPtr);
5055188c4b9SDmitri Gribenko   formTokenWithChars(T, NextLine, tok::verbatim_block_line);
50660ddd8a1SDmitri Gribenko   T.setVerbatimBlockText(Text);
5075188c4b9SDmitri Gribenko 
5085188c4b9SDmitri Gribenko   State = LS_VerbatimBlockBody;
5095188c4b9SDmitri Gribenko }
5105188c4b9SDmitri Gribenko 
5115188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockBody(Token &T) {
5125188c4b9SDmitri Gribenko   assert(State == LS_VerbatimBlockBody);
5135188c4b9SDmitri Gribenko 
5145188c4b9SDmitri Gribenko   if (CommentState == LCS_InsideCComment)
5155188c4b9SDmitri Gribenko     skipLineStartingDecorations();
5165188c4b9SDmitri Gribenko 
5175188c4b9SDmitri Gribenko   lexVerbatimBlockFirstLine(T);
5185188c4b9SDmitri Gribenko }
5195188c4b9SDmitri Gribenko 
5207acbf00fSDmitri Gribenko void Lexer::setupAndLexVerbatimLine(Token &T, const char *TextBegin,
5217acbf00fSDmitri Gribenko                                     const CommandInfo *Info) {
5227acbf00fSDmitri Gribenko   assert(Info->IsVerbatimLineCommand);
5231669f702SDmitri Gribenko   formTokenWithChars(T, TextBegin, tok::verbatim_line_name);
5247acbf00fSDmitri Gribenko   T.setVerbatimLineID(Info->getID());
5251669f702SDmitri Gribenko 
5261669f702SDmitri Gribenko   State = LS_VerbatimLineText;
5271669f702SDmitri Gribenko }
5281669f702SDmitri Gribenko 
5291669f702SDmitri Gribenko void Lexer::lexVerbatimLineText(Token &T) {
5301669f702SDmitri Gribenko   assert(State == LS_VerbatimLineText);
5311669f702SDmitri Gribenko 
5325188c4b9SDmitri Gribenko   // Extract current line.
5335188c4b9SDmitri Gribenko   const char *Newline = findNewline(BufferPtr, CommentEnd);
534*bf3e3270SCraig Topper   StringRef Text(BufferPtr, Newline - BufferPtr);
5351669f702SDmitri Gribenko   formTokenWithChars(T, Newline, tok::verbatim_line_text);
5365188c4b9SDmitri Gribenko   T.setVerbatimLineText(Text);
5371669f702SDmitri Gribenko 
5381669f702SDmitri Gribenko   State = LS_Normal;
5395188c4b9SDmitri Gribenko }
5405188c4b9SDmitri Gribenko 
5414586df76SDmitri Gribenko void Lexer::lexHTMLCharacterReference(Token &T) {
5424586df76SDmitri Gribenko   const char *TokenPtr = BufferPtr;
5434586df76SDmitri Gribenko   assert(*TokenPtr == '&');
5444586df76SDmitri Gribenko   TokenPtr++;
5454586df76SDmitri Gribenko   if (TokenPtr == CommentEnd) {
5464586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5474586df76SDmitri Gribenko     return;
5484586df76SDmitri Gribenko   }
5494586df76SDmitri Gribenko   const char *NamePtr;
5504586df76SDmitri Gribenko   bool isNamed = false;
5514586df76SDmitri Gribenko   bool isDecimal = false;
5524586df76SDmitri Gribenko   char C = *TokenPtr;
5534586df76SDmitri Gribenko   if (isHTMLNamedCharacterReferenceCharacter(C)) {
5544586df76SDmitri Gribenko     NamePtr = TokenPtr;
5554586df76SDmitri Gribenko     TokenPtr = skipNamedCharacterReference(TokenPtr, CommentEnd);
5564586df76SDmitri Gribenko     isNamed = true;
5574586df76SDmitri Gribenko   } else if (C == '#') {
5584586df76SDmitri Gribenko     TokenPtr++;
5594586df76SDmitri Gribenko     if (TokenPtr == CommentEnd) {
5604586df76SDmitri Gribenko       formTextToken(T, TokenPtr);
5614586df76SDmitri Gribenko       return;
5624586df76SDmitri Gribenko     }
5634586df76SDmitri Gribenko     C = *TokenPtr;
5644586df76SDmitri Gribenko     if (isHTMLDecimalCharacterReferenceCharacter(C)) {
5654586df76SDmitri Gribenko       NamePtr = TokenPtr;
5664586df76SDmitri Gribenko       TokenPtr = skipDecimalCharacterReference(TokenPtr, CommentEnd);
5674586df76SDmitri Gribenko       isDecimal = true;
5684586df76SDmitri Gribenko     } else if (C == 'x' || C == 'X') {
5694586df76SDmitri Gribenko       TokenPtr++;
5704586df76SDmitri Gribenko       NamePtr = TokenPtr;
5714586df76SDmitri Gribenko       TokenPtr = skipHexCharacterReference(TokenPtr, CommentEnd);
5724586df76SDmitri Gribenko     } else {
5734586df76SDmitri Gribenko       formTextToken(T, TokenPtr);
5744586df76SDmitri Gribenko       return;
5754586df76SDmitri Gribenko     }
5764586df76SDmitri Gribenko   } else {
5774586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5784586df76SDmitri Gribenko     return;
5794586df76SDmitri Gribenko   }
5804586df76SDmitri Gribenko   if (NamePtr == TokenPtr || TokenPtr == CommentEnd ||
5814586df76SDmitri Gribenko       *TokenPtr != ';') {
5824586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5834586df76SDmitri Gribenko     return;
5844586df76SDmitri Gribenko   }
5854586df76SDmitri Gribenko   StringRef Name(NamePtr, TokenPtr - NamePtr);
5864586df76SDmitri Gribenko   TokenPtr++; // Skip semicolon.
5874586df76SDmitri Gribenko   StringRef Resolved;
58828800da1SDmitri Gribenko   if (isNamed)
5894586df76SDmitri Gribenko     Resolved = resolveHTMLNamedCharacterReference(Name);
5904586df76SDmitri Gribenko   else if (isDecimal)
5914586df76SDmitri Gribenko     Resolved = resolveHTMLDecimalCharacterReference(Name);
5924586df76SDmitri Gribenko   else
5934586df76SDmitri Gribenko     Resolved = resolveHTMLHexCharacterReference(Name);
5944586df76SDmitri Gribenko 
5954586df76SDmitri Gribenko   if (Resolved.empty()) {
5964586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5974586df76SDmitri Gribenko     return;
5984586df76SDmitri Gribenko   }
5994586df76SDmitri Gribenko   formTokenWithChars(T, TokenPtr, tok::text);
6004586df76SDmitri Gribenko   T.setText(Resolved);
6014586df76SDmitri Gribenko   return;
6024586df76SDmitri Gribenko }
6034586df76SDmitri Gribenko 
604e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLStartTag(Token &T) {
60517709ae8SDmitri Gribenko   assert(BufferPtr[0] == '<' &&
60617709ae8SDmitri Gribenko          isHTMLIdentifierStartingCharacter(BufferPtr[1]));
6075188c4b9SDmitri Gribenko   const char *TagNameEnd = skipHTMLIdentifier(BufferPtr + 2, CommentEnd);
60860ddd8a1SDmitri Gribenko   StringRef Name(BufferPtr + 1, TagNameEnd - (BufferPtr + 1));
609107618a6SDmitri Gribenko   if (!isHTMLTagName(Name)) {
610107618a6SDmitri Gribenko     formTextToken(T, TagNameEnd);
611107618a6SDmitri Gribenko     return;
612107618a6SDmitri Gribenko   }
613107618a6SDmitri Gribenko 
614e00ffc7bSDmitri Gribenko   formTokenWithChars(T, TagNameEnd, tok::html_start_tag);
615e00ffc7bSDmitri Gribenko   T.setHTMLTagStartName(Name);
6165188c4b9SDmitri Gribenko 
6175188c4b9SDmitri Gribenko   BufferPtr = skipWhitespace(BufferPtr, CommentEnd);
6185188c4b9SDmitri Gribenko 
61917709ae8SDmitri Gribenko   const char C = *BufferPtr;
62017709ae8SDmitri Gribenko   if (BufferPtr != CommentEnd &&
621f26054f0SDmitri Gribenko       (C == '>' || C == '/' || isHTMLIdentifierStartingCharacter(C)))
622e00ffc7bSDmitri Gribenko     State = LS_HTMLStartTag;
6235188c4b9SDmitri Gribenko }
6245188c4b9SDmitri Gribenko 
625e00ffc7bSDmitri Gribenko void Lexer::lexHTMLStartTag(Token &T) {
626e00ffc7bSDmitri Gribenko   assert(State == LS_HTMLStartTag);
6275188c4b9SDmitri Gribenko 
6285188c4b9SDmitri Gribenko   const char *TokenPtr = BufferPtr;
6295188c4b9SDmitri Gribenko   char C = *TokenPtr;
6305188c4b9SDmitri Gribenko   if (isHTMLIdentifierCharacter(C)) {
6315188c4b9SDmitri Gribenko     TokenPtr = skipHTMLIdentifier(TokenPtr, CommentEnd);
63260ddd8a1SDmitri Gribenko     StringRef Ident(BufferPtr, TokenPtr - BufferPtr);
6335188c4b9SDmitri Gribenko     formTokenWithChars(T, TokenPtr, tok::html_ident);
63460ddd8a1SDmitri Gribenko     T.setHTMLIdent(Ident);
6355188c4b9SDmitri Gribenko   } else {
6365188c4b9SDmitri Gribenko     switch (C) {
6375188c4b9SDmitri Gribenko     case '=':
6385188c4b9SDmitri Gribenko       TokenPtr++;
6395188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_equals);
6405188c4b9SDmitri Gribenko       break;
6415188c4b9SDmitri Gribenko     case '\"':
6425188c4b9SDmitri Gribenko     case '\'': {
6435188c4b9SDmitri Gribenko       const char *OpenQuote = TokenPtr;
6445188c4b9SDmitri Gribenko       TokenPtr = skipHTMLQuotedString(TokenPtr, CommentEnd);
6455188c4b9SDmitri Gribenko       const char *ClosingQuote = TokenPtr;
6465188c4b9SDmitri Gribenko       if (TokenPtr != CommentEnd) // Skip closing quote.
6475188c4b9SDmitri Gribenko         TokenPtr++;
6485188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_quoted_string);
6495188c4b9SDmitri Gribenko       T.setHTMLQuotedString(StringRef(OpenQuote + 1,
6505188c4b9SDmitri Gribenko                                       ClosingQuote - (OpenQuote + 1)));
6515188c4b9SDmitri Gribenko       break;
6525188c4b9SDmitri Gribenko     }
6535188c4b9SDmitri Gribenko     case '>':
6545188c4b9SDmitri Gribenko       TokenPtr++;
6555188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_greater);
65617709ae8SDmitri Gribenko       State = LS_Normal;
65717709ae8SDmitri Gribenko       return;
658f26054f0SDmitri Gribenko     case '/':
659f26054f0SDmitri Gribenko       TokenPtr++;
660f26054f0SDmitri Gribenko       if (TokenPtr != CommentEnd && *TokenPtr == '>') {
661f26054f0SDmitri Gribenko         TokenPtr++;
662f26054f0SDmitri Gribenko         formTokenWithChars(T, TokenPtr, tok::html_slash_greater);
6634586df76SDmitri Gribenko       } else
6644586df76SDmitri Gribenko         formTextToken(T, TokenPtr);
6654586df76SDmitri Gribenko 
666f26054f0SDmitri Gribenko       State = LS_Normal;
667f26054f0SDmitri Gribenko       return;
6685188c4b9SDmitri Gribenko     }
6695188c4b9SDmitri Gribenko   }
6705188c4b9SDmitri Gribenko 
6715188c4b9SDmitri Gribenko   // Now look ahead and return to normal state if we don't see any HTML tokens
6725188c4b9SDmitri Gribenko   // ahead.
6735188c4b9SDmitri Gribenko   BufferPtr = skipWhitespace(BufferPtr, CommentEnd);
6745188c4b9SDmitri Gribenko   if (BufferPtr == CommentEnd) {
6755188c4b9SDmitri Gribenko     State = LS_Normal;
6765188c4b9SDmitri Gribenko     return;
6775188c4b9SDmitri Gribenko   }
6785188c4b9SDmitri Gribenko 
6795188c4b9SDmitri Gribenko   C = *BufferPtr;
68017709ae8SDmitri Gribenko   if (!isHTMLIdentifierStartingCharacter(C) &&
6815188c4b9SDmitri Gribenko       C != '=' && C != '\"' && C != '\'' && C != '>') {
6825188c4b9SDmitri Gribenko     State = LS_Normal;
6835188c4b9SDmitri Gribenko     return;
6845188c4b9SDmitri Gribenko   }
6855188c4b9SDmitri Gribenko }
6865188c4b9SDmitri Gribenko 
687e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLEndTag(Token &T) {
6885188c4b9SDmitri Gribenko   assert(BufferPtr[0] == '<' && BufferPtr[1] == '/');
6895188c4b9SDmitri Gribenko 
6905188c4b9SDmitri Gribenko   const char *TagNameBegin = skipWhitespace(BufferPtr + 2, CommentEnd);
6915188c4b9SDmitri Gribenko   const char *TagNameEnd = skipHTMLIdentifier(TagNameBegin, CommentEnd);
692107618a6SDmitri Gribenko   StringRef Name(TagNameBegin, TagNameEnd - TagNameBegin);
693107618a6SDmitri Gribenko   if (!isHTMLTagName(Name)) {
694107618a6SDmitri Gribenko     formTextToken(T, TagNameEnd);
695107618a6SDmitri Gribenko     return;
696107618a6SDmitri Gribenko   }
6975188c4b9SDmitri Gribenko 
6985188c4b9SDmitri Gribenko   const char *End = skipWhitespace(TagNameEnd, CommentEnd);
6995188c4b9SDmitri Gribenko 
700e00ffc7bSDmitri Gribenko   formTokenWithChars(T, End, tok::html_end_tag);
701107618a6SDmitri Gribenko   T.setHTMLTagEndName(Name);
702ec92531cSDmitri Gribenko 
703ec92531cSDmitri Gribenko   if (BufferPtr != CommentEnd && *BufferPtr == '>')
704e00ffc7bSDmitri Gribenko     State = LS_HTMLEndTag;
705ec92531cSDmitri Gribenko }
706ec92531cSDmitri Gribenko 
707e00ffc7bSDmitri Gribenko void Lexer::lexHTMLEndTag(Token &T) {
708ec92531cSDmitri Gribenko   assert(BufferPtr != CommentEnd && *BufferPtr == '>');
709ec92531cSDmitri Gribenko 
710ec92531cSDmitri Gribenko   formTokenWithChars(T, BufferPtr + 1, tok::html_greater);
711ec92531cSDmitri Gribenko   State = LS_Normal;
7125188c4b9SDmitri Gribenko }
7135188c4b9SDmitri Gribenko 
7145b637078SFariborz Jahanian Lexer::Lexer(llvm::BumpPtrAllocator &Allocator, DiagnosticsEngine &Diags,
7155b637078SFariborz Jahanian              const CommandTraits &Traits,
7166bab9113SDmitri Gribenko              SourceLocation FileLoc,
7175188c4b9SDmitri Gribenko              const char *BufferStart, const char *BufferEnd):
7185b637078SFariborz Jahanian     Allocator(Allocator), Diags(Diags), Traits(Traits),
7195188c4b9SDmitri Gribenko     BufferStart(BufferStart), BufferEnd(BufferEnd),
7206bab9113SDmitri Gribenko     FileLoc(FileLoc), BufferPtr(BufferStart),
7215188c4b9SDmitri Gribenko     CommentState(LCS_BeforeComment), State(LS_Normal) {
7225188c4b9SDmitri Gribenko }
7235188c4b9SDmitri Gribenko 
7245188c4b9SDmitri Gribenko void Lexer::lex(Token &T) {
7255188c4b9SDmitri Gribenko again:
7265188c4b9SDmitri Gribenko   switch (CommentState) {
7275188c4b9SDmitri Gribenko   case LCS_BeforeComment:
7285188c4b9SDmitri Gribenko     if (BufferPtr == BufferEnd) {
7295188c4b9SDmitri Gribenko       formTokenWithChars(T, BufferPtr, tok::eof);
7305188c4b9SDmitri Gribenko       return;
7315188c4b9SDmitri Gribenko     }
7325188c4b9SDmitri Gribenko 
7335188c4b9SDmitri Gribenko     assert(*BufferPtr == '/');
7345188c4b9SDmitri Gribenko     BufferPtr++; // Skip first slash.
7355188c4b9SDmitri Gribenko     switch(*BufferPtr) {
7365188c4b9SDmitri Gribenko     case '/': { // BCPL comment.
7375188c4b9SDmitri Gribenko       BufferPtr++; // Skip second slash.
7385188c4b9SDmitri Gribenko 
7395188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd) {
7405188c4b9SDmitri Gribenko         // Skip Doxygen magic marker, if it is present.
7415188c4b9SDmitri Gribenko         // It might be missing because of a typo //< or /*<, or because we
7425188c4b9SDmitri Gribenko         // merged this non-Doxygen comment into a bunch of Doxygen comments
7435188c4b9SDmitri Gribenko         // around it: /** ... */ /* ... */ /** ... */
7445188c4b9SDmitri Gribenko         const char C = *BufferPtr;
7455188c4b9SDmitri Gribenko         if (C == '/' || C == '!')
7465188c4b9SDmitri Gribenko           BufferPtr++;
7475188c4b9SDmitri Gribenko       }
7485188c4b9SDmitri Gribenko 
7495188c4b9SDmitri Gribenko       // Skip less-than symbol that marks trailing comments.
7505188c4b9SDmitri Gribenko       // Skip it even if the comment is not a Doxygen one, because //< and /*<
7515188c4b9SDmitri Gribenko       // are frequent typos.
7525188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd && *BufferPtr == '<')
7535188c4b9SDmitri Gribenko         BufferPtr++;
7545188c4b9SDmitri Gribenko 
7555188c4b9SDmitri Gribenko       CommentState = LCS_InsideBCPLComment;
756ec92531cSDmitri Gribenko       if (State != LS_VerbatimBlockBody && State != LS_VerbatimBlockFirstLine)
7575188c4b9SDmitri Gribenko         State = LS_Normal;
7585188c4b9SDmitri Gribenko       CommentEnd = findBCPLCommentEnd(BufferPtr, BufferEnd);
7595188c4b9SDmitri Gribenko       goto again;
7605188c4b9SDmitri Gribenko     }
7615188c4b9SDmitri Gribenko     case '*': { // C comment.
7625188c4b9SDmitri Gribenko       BufferPtr++; // Skip star.
7635188c4b9SDmitri Gribenko 
7645188c4b9SDmitri Gribenko       // Skip Doxygen magic marker.
7655188c4b9SDmitri Gribenko       const char C = *BufferPtr;
7665188c4b9SDmitri Gribenko       if ((C == '*' && *(BufferPtr + 1) != '/') || C == '!')
7675188c4b9SDmitri Gribenko         BufferPtr++;
7685188c4b9SDmitri Gribenko 
7695188c4b9SDmitri Gribenko       // Skip less-than symbol that marks trailing comments.
7705188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd && *BufferPtr == '<')
7715188c4b9SDmitri Gribenko         BufferPtr++;
7725188c4b9SDmitri Gribenko 
7735188c4b9SDmitri Gribenko       CommentState = LCS_InsideCComment;
7745188c4b9SDmitri Gribenko       State = LS_Normal;
7755188c4b9SDmitri Gribenko       CommentEnd = findCCommentEnd(BufferPtr, BufferEnd);
7765188c4b9SDmitri Gribenko       goto again;
7775188c4b9SDmitri Gribenko     }
7785188c4b9SDmitri Gribenko     default:
7795188c4b9SDmitri Gribenko       llvm_unreachable("second character of comment should be '/' or '*'");
7805188c4b9SDmitri Gribenko     }
7815188c4b9SDmitri Gribenko 
7825188c4b9SDmitri Gribenko   case LCS_BetweenComments: {
7835188c4b9SDmitri Gribenko     // Consecutive comments are extracted only if there is only whitespace
7845188c4b9SDmitri Gribenko     // between them.  So we can search for the start of the next comment.
7855188c4b9SDmitri Gribenko     const char *EndWhitespace = BufferPtr;
7865188c4b9SDmitri Gribenko     while(EndWhitespace != BufferEnd && *EndWhitespace != '/')
7875188c4b9SDmitri Gribenko       EndWhitespace++;
7885188c4b9SDmitri Gribenko 
7895188c4b9SDmitri Gribenko     // Turn any whitespace between comments (and there is only whitespace
79017709ae8SDmitri Gribenko     // between them -- guaranteed by comment extraction) into a newline.  We
79117709ae8SDmitri Gribenko     // have two newlines between C comments in total (first one was synthesized
79217709ae8SDmitri Gribenko     // after a comment).
7935188c4b9SDmitri Gribenko     formTokenWithChars(T, EndWhitespace, tok::newline);
7945188c4b9SDmitri Gribenko 
7955188c4b9SDmitri Gribenko     CommentState = LCS_BeforeComment;
7965188c4b9SDmitri Gribenko     break;
7975188c4b9SDmitri Gribenko   }
7985188c4b9SDmitri Gribenko 
7995188c4b9SDmitri Gribenko   case LCS_InsideBCPLComment:
8005188c4b9SDmitri Gribenko   case LCS_InsideCComment:
8015188c4b9SDmitri Gribenko     if (BufferPtr != CommentEnd) {
8025188c4b9SDmitri Gribenko       lexCommentText(T);
8035188c4b9SDmitri Gribenko       break;
8045188c4b9SDmitri Gribenko     } else {
8055188c4b9SDmitri Gribenko       // Skip C comment closing sequence.
8065188c4b9SDmitri Gribenko       if (CommentState == LCS_InsideCComment) {
8075188c4b9SDmitri Gribenko         assert(BufferPtr[0] == '*' && BufferPtr[1] == '/');
8085188c4b9SDmitri Gribenko         BufferPtr += 2;
8095188c4b9SDmitri Gribenko         assert(BufferPtr <= BufferEnd);
8105188c4b9SDmitri Gribenko 
8115188c4b9SDmitri Gribenko         // Synthenize newline just after the C comment, regardless if there is
8125188c4b9SDmitri Gribenko         // actually a newline.
8135188c4b9SDmitri Gribenko         formTokenWithChars(T, BufferPtr, tok::newline);
8145188c4b9SDmitri Gribenko 
8155188c4b9SDmitri Gribenko         CommentState = LCS_BetweenComments;
8165188c4b9SDmitri Gribenko         break;
8175188c4b9SDmitri Gribenko       } else {
8185188c4b9SDmitri Gribenko         // Don't synthesized a newline after BCPL comment.
8195188c4b9SDmitri Gribenko         CommentState = LCS_BetweenComments;
8205188c4b9SDmitri Gribenko         goto again;
8215188c4b9SDmitri Gribenko       }
8225188c4b9SDmitri Gribenko     }
8235188c4b9SDmitri Gribenko   }
8245188c4b9SDmitri Gribenko }
8255188c4b9SDmitri Gribenko 
8265188c4b9SDmitri Gribenko StringRef Lexer::getSpelling(const Token &Tok,
8275188c4b9SDmitri Gribenko                              const SourceManager &SourceMgr,
8285188c4b9SDmitri Gribenko                              bool *Invalid) const {
8295188c4b9SDmitri Gribenko   SourceLocation Loc = Tok.getLocation();
8305188c4b9SDmitri Gribenko   std::pair<FileID, unsigned> LocInfo = SourceMgr.getDecomposedLoc(Loc);
8315188c4b9SDmitri Gribenko 
8325188c4b9SDmitri Gribenko   bool InvalidTemp = false;
8335188c4b9SDmitri Gribenko   StringRef File = SourceMgr.getBufferData(LocInfo.first, &InvalidTemp);
8345188c4b9SDmitri Gribenko   if (InvalidTemp) {
8355188c4b9SDmitri Gribenko     *Invalid = true;
8365188c4b9SDmitri Gribenko     return StringRef();
8375188c4b9SDmitri Gribenko   }
8385188c4b9SDmitri Gribenko 
8395188c4b9SDmitri Gribenko   const char *Begin = File.data() + LocInfo.second;
8405188c4b9SDmitri Gribenko   return StringRef(Begin, Tok.getLength());
8415188c4b9SDmitri Gribenko }
8425188c4b9SDmitri Gribenko 
8435188c4b9SDmitri Gribenko } // end namespace comments
8445188c4b9SDmitri Gribenko } // end namespace clang
8455188c4b9SDmitri Gribenko 
846