15188c4b9SDmitri Gribenko #include "clang/AST/CommentLexer.h"
2ca7f80adSDmitri Gribenko #include "clang/AST/CommentCommandTraits.h"
3b2e5482bSDmitri Gribenko #include "llvm/ADT/StringExtras.h"
45188c4b9SDmitri Gribenko #include "llvm/ADT/StringSwitch.h"
59feeef40SDmitri Gribenko #include "llvm/Support/ConvertUTF.h"
65188c4b9SDmitri Gribenko #include "llvm/Support/ErrorHandling.h"
75188c4b9SDmitri Gribenko 
85188c4b9SDmitri Gribenko namespace clang {
95188c4b9SDmitri Gribenko namespace comments {
105188c4b9SDmitri Gribenko 
115188c4b9SDmitri Gribenko void Token::dump(const Lexer &L, const SourceManager &SM) const {
125188c4b9SDmitri Gribenko   llvm::errs() << "comments::Token Kind=" << Kind << " ";
135188c4b9SDmitri Gribenko   Loc.dump(SM);
145188c4b9SDmitri Gribenko   llvm::errs() << " " << Length << " \"" << L.getSpelling(*this, SM) << "\"\n";
155188c4b9SDmitri Gribenko }
165188c4b9SDmitri Gribenko 
174586df76SDmitri Gribenko namespace {
184586df76SDmitri Gribenko bool isHTMLNamedCharacterReferenceCharacter(char C) {
194586df76SDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
204586df76SDmitri Gribenko          (C >= 'A' && C <= 'Z');
214586df76SDmitri Gribenko }
224586df76SDmitri Gribenko 
234586df76SDmitri Gribenko bool isHTMLDecimalCharacterReferenceCharacter(char C) {
244586df76SDmitri Gribenko   return C >= '0' && C <= '9';
254586df76SDmitri Gribenko }
264586df76SDmitri Gribenko 
274586df76SDmitri Gribenko bool isHTMLHexCharacterReferenceCharacter(char C) {
284586df76SDmitri Gribenko   return (C >= '0' && C <= '9') ||
294586df76SDmitri Gribenko          (C >= 'a' && C <= 'f') ||
304586df76SDmitri Gribenko          (C >= 'A' && C <= 'F');
314586df76SDmitri Gribenko }
32107618a6SDmitri Gribenko 
33*28800da1SDmitri Gribenko StringRef convertCodePointToUTF8(llvm::BumpPtrAllocator &Allocator,
34*28800da1SDmitri Gribenko                                  unsigned CodePoint) {
354586df76SDmitri Gribenko   char *Resolved = Allocator.Allocate<char>(UNI_MAX_UTF8_BYTES_PER_CODE_POINT);
364586df76SDmitri Gribenko   char *ResolvedPtr = Resolved;
379feeef40SDmitri Gribenko   if (llvm::ConvertCodePointToUTF8(CodePoint, ResolvedPtr))
384586df76SDmitri Gribenko     return StringRef(Resolved, ResolvedPtr - Resolved);
394586df76SDmitri Gribenko   else
404586df76SDmitri Gribenko     return StringRef();
414586df76SDmitri Gribenko }
424586df76SDmitri Gribenko 
43*28800da1SDmitri Gribenko #include "clang/AST/CommentHTMLTags.inc"
44*28800da1SDmitri Gribenko #include "clang/AST/CommentHTMLNamedCharacterReferences.inc"
45*28800da1SDmitri Gribenko 
46*28800da1SDmitri Gribenko } // unnamed namespace
477b3ae190SFariborz Jahanian 
487b3ae190SFariborz Jahanian StringRef Lexer::resolveHTMLNamedCharacterReference(StringRef Name) const {
49*28800da1SDmitri Gribenko   // Fast path, first check a few most widely used named character references.
507b3ae190SFariborz Jahanian   return llvm::StringSwitch<StringRef>(Name)
517b3ae190SFariborz Jahanian       .Case("amp", "&")
527b3ae190SFariborz Jahanian       .Case("lt", "<")
537b3ae190SFariborz Jahanian       .Case("gt", ">")
547b3ae190SFariborz Jahanian       .Case("quot", "\"")
557b3ae190SFariborz Jahanian       .Case("apos", "\'")
56*28800da1SDmitri Gribenko       // Slow path.
57*28800da1SDmitri Gribenko       .Default(translateHTMLNamedCharacterReferenceToUTF8(Name));
587b3ae190SFariborz Jahanian }
597b3ae190SFariborz Jahanian 
607b3ae190SFariborz Jahanian StringRef Lexer::resolveHTMLDecimalCharacterReference(StringRef Name) const {
614586df76SDmitri Gribenko   unsigned CodePoint = 0;
624586df76SDmitri Gribenko   for (unsigned i = 0, e = Name.size(); i != e; ++i) {
637b3ae190SFariborz Jahanian     assert(isHTMLDecimalCharacterReferenceCharacter(Name[i]));
647b3ae190SFariborz Jahanian     CodePoint *= 10;
657b3ae190SFariborz Jahanian     CodePoint += Name[i] - '0';
664586df76SDmitri Gribenko   }
67*28800da1SDmitri Gribenko   return convertCodePointToUTF8(Allocator, CodePoint);
68*28800da1SDmitri Gribenko }
694586df76SDmitri Gribenko 
70*28800da1SDmitri Gribenko StringRef Lexer::resolveHTMLHexCharacterReference(StringRef Name) const {
71*28800da1SDmitri Gribenko   unsigned CodePoint = 0;
72*28800da1SDmitri Gribenko   for (unsigned i = 0, e = Name.size(); i != e; ++i) {
73*28800da1SDmitri Gribenko     CodePoint *= 16;
74*28800da1SDmitri Gribenko     const char C = Name[i];
75*28800da1SDmitri Gribenko     assert(isHTMLHexCharacterReferenceCharacter(C));
76*28800da1SDmitri Gribenko     CodePoint += llvm::hexDigitValue(C);
77*28800da1SDmitri Gribenko   }
78*28800da1SDmitri Gribenko   return convertCodePointToUTF8(Allocator, CodePoint);
794586df76SDmitri Gribenko }
804586df76SDmitri Gribenko 
815188c4b9SDmitri Gribenko void Lexer::skipLineStartingDecorations() {
825188c4b9SDmitri Gribenko   // This function should be called only for C comments
835188c4b9SDmitri Gribenko   assert(CommentState == LCS_InsideCComment);
845188c4b9SDmitri Gribenko 
855188c4b9SDmitri Gribenko   if (BufferPtr == CommentEnd)
865188c4b9SDmitri Gribenko     return;
875188c4b9SDmitri Gribenko 
885188c4b9SDmitri Gribenko   switch (*BufferPtr) {
895188c4b9SDmitri Gribenko   case ' ':
905188c4b9SDmitri Gribenko   case '\t':
915188c4b9SDmitri Gribenko   case '\f':
925188c4b9SDmitri Gribenko   case '\v': {
935188c4b9SDmitri Gribenko     const char *NewBufferPtr = BufferPtr;
945188c4b9SDmitri Gribenko     NewBufferPtr++;
955188c4b9SDmitri Gribenko     if (NewBufferPtr == CommentEnd)
965188c4b9SDmitri Gribenko       return;
975188c4b9SDmitri Gribenko 
985188c4b9SDmitri Gribenko     char C = *NewBufferPtr;
995188c4b9SDmitri Gribenko     while (C == ' ' || C == '\t' || C == '\f' || C == '\v') {
1005188c4b9SDmitri Gribenko       NewBufferPtr++;
1015188c4b9SDmitri Gribenko       if (NewBufferPtr == CommentEnd)
1025188c4b9SDmitri Gribenko         return;
1035188c4b9SDmitri Gribenko       C = *NewBufferPtr;
1045188c4b9SDmitri Gribenko     }
1055188c4b9SDmitri Gribenko     if (C == '*')
1065188c4b9SDmitri Gribenko       BufferPtr = NewBufferPtr + 1;
1075188c4b9SDmitri Gribenko     break;
1085188c4b9SDmitri Gribenko   }
1095188c4b9SDmitri Gribenko   case '*':
1105188c4b9SDmitri Gribenko     BufferPtr++;
1115188c4b9SDmitri Gribenko     break;
1125188c4b9SDmitri Gribenko   }
1135188c4b9SDmitri Gribenko }
1145188c4b9SDmitri Gribenko 
1155188c4b9SDmitri Gribenko namespace {
116ec92531cSDmitri Gribenko /// Returns pointer to the first newline character in the string.
1175188c4b9SDmitri Gribenko const char *findNewline(const char *BufferPtr, const char *BufferEnd) {
1185188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1195188c4b9SDmitri Gribenko     const char C = *BufferPtr;
1205188c4b9SDmitri Gribenko     if (C == '\n' || C == '\r')
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) {
16917709ae8SDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
17017709ae8SDmitri Gribenko          (C >= 'A' && C <= 'Z');
17117709ae8SDmitri Gribenko }
17217709ae8SDmitri Gribenko 
1735188c4b9SDmitri Gribenko bool isHTMLIdentifierCharacter(char C) {
1745188c4b9SDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
1755188c4b9SDmitri Gribenko          (C >= 'A' && C <= 'Z') ||
1765188c4b9SDmitri Gribenko          (C >= '0' && C <= '9');
1775188c4b9SDmitri Gribenko }
1785188c4b9SDmitri Gribenko 
1795188c4b9SDmitri Gribenko const char *skipHTMLIdentifier(const char *BufferPtr, const char *BufferEnd) {
1805188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1815188c4b9SDmitri Gribenko     if (!isHTMLIdentifierCharacter(*BufferPtr))
1825188c4b9SDmitri Gribenko       return BufferPtr;
1835188c4b9SDmitri Gribenko   }
1845188c4b9SDmitri Gribenko   return BufferEnd;
1855188c4b9SDmitri Gribenko }
1865188c4b9SDmitri Gribenko 
1875188c4b9SDmitri Gribenko /// Skip HTML string quoted in single or double quotes.  Escaping quotes inside
1885188c4b9SDmitri Gribenko /// string allowed.
1895188c4b9SDmitri Gribenko ///
1905188c4b9SDmitri Gribenko /// Returns pointer to closing quote.
1915188c4b9SDmitri Gribenko const char *skipHTMLQuotedString(const char *BufferPtr, const char *BufferEnd)
1925188c4b9SDmitri Gribenko {
1935188c4b9SDmitri Gribenko   const char Quote = *BufferPtr;
1945188c4b9SDmitri Gribenko   assert(Quote == '\"' || Quote == '\'');
1955188c4b9SDmitri Gribenko 
1965188c4b9SDmitri Gribenko   BufferPtr++;
1975188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1985188c4b9SDmitri Gribenko     const char C = *BufferPtr;
1995188c4b9SDmitri Gribenko     if (C == Quote && BufferPtr[-1] != '\\')
2005188c4b9SDmitri Gribenko       return BufferPtr;
2015188c4b9SDmitri Gribenko   }
2025188c4b9SDmitri Gribenko   return BufferEnd;
2035188c4b9SDmitri Gribenko }
2045188c4b9SDmitri Gribenko 
2055188c4b9SDmitri Gribenko bool isHorizontalWhitespace(char C) {
2065188c4b9SDmitri Gribenko   return C == ' ' || C == '\t' || C == '\f' || C == '\v';
2075188c4b9SDmitri Gribenko }
2085188c4b9SDmitri Gribenko 
2095188c4b9SDmitri Gribenko bool isWhitespace(char C) {
2105188c4b9SDmitri Gribenko   return C == ' ' || C == '\n' || C == '\r' ||
2115188c4b9SDmitri Gribenko          C == '\t' || C == '\f' || C == '\v';
2125188c4b9SDmitri Gribenko }
2135188c4b9SDmitri Gribenko 
2145188c4b9SDmitri Gribenko const char *skipWhitespace(const char *BufferPtr, const char *BufferEnd) {
2155188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2165188c4b9SDmitri Gribenko     if (!isWhitespace(*BufferPtr))
2175188c4b9SDmitri Gribenko       return BufferPtr;
2185188c4b9SDmitri Gribenko   }
2195188c4b9SDmitri Gribenko   return BufferEnd;
2205188c4b9SDmitri Gribenko }
2215188c4b9SDmitri Gribenko 
222e4a3997dSDmitri Gribenko bool isWhitespace(const char *BufferPtr, const char *BufferEnd) {
223e4a3997dSDmitri Gribenko   return skipWhitespace(BufferPtr, BufferEnd) == BufferEnd;
224e4a3997dSDmitri Gribenko }
225e4a3997dSDmitri Gribenko 
226ad45ad6fSDmitri Gribenko bool isCommandNameStartCharacter(char C) {
227ad45ad6fSDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
228ad45ad6fSDmitri Gribenko          (C >= 'A' && C <= 'Z');
229ad45ad6fSDmitri Gribenko }
230ad45ad6fSDmitri Gribenko 
2315188c4b9SDmitri Gribenko bool isCommandNameCharacter(char C) {
2325188c4b9SDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
2335188c4b9SDmitri Gribenko          (C >= 'A' && C <= 'Z') ||
2345188c4b9SDmitri Gribenko          (C >= '0' && C <= '9');
2355188c4b9SDmitri Gribenko }
2365188c4b9SDmitri Gribenko 
2375188c4b9SDmitri Gribenko const char *skipCommandName(const char *BufferPtr, const char *BufferEnd) {
2385188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2395188c4b9SDmitri Gribenko     if (!isCommandNameCharacter(*BufferPtr))
2405188c4b9SDmitri Gribenko       return BufferPtr;
2415188c4b9SDmitri Gribenko   }
2425188c4b9SDmitri Gribenko   return BufferEnd;
2435188c4b9SDmitri Gribenko }
2445188c4b9SDmitri Gribenko 
2455188c4b9SDmitri Gribenko /// Return the one past end pointer for BCPL comments.
2465188c4b9SDmitri Gribenko /// Handles newlines escaped with backslash or trigraph for backslahs.
2475188c4b9SDmitri Gribenko const char *findBCPLCommentEnd(const char *BufferPtr, const char *BufferEnd) {
2485188c4b9SDmitri Gribenko   const char *CurPtr = BufferPtr;
2495188c4b9SDmitri Gribenko   while (CurPtr != BufferEnd) {
2505188c4b9SDmitri Gribenko     char C = *CurPtr;
2515188c4b9SDmitri Gribenko     while (C != '\n' && C != '\r') {
2525188c4b9SDmitri Gribenko       CurPtr++;
2535188c4b9SDmitri Gribenko       if (CurPtr == BufferEnd)
2545188c4b9SDmitri Gribenko         return BufferEnd;
2555188c4b9SDmitri Gribenko       C = *CurPtr;
2565188c4b9SDmitri Gribenko     }
2575188c4b9SDmitri Gribenko     // We found a newline, check if it is escaped.
2585188c4b9SDmitri Gribenko     const char *EscapePtr = CurPtr - 1;
2595188c4b9SDmitri Gribenko     while(isHorizontalWhitespace(*EscapePtr))
2605188c4b9SDmitri Gribenko       EscapePtr--;
2615188c4b9SDmitri Gribenko 
2625188c4b9SDmitri Gribenko     if (*EscapePtr == '\\' ||
2635188c4b9SDmitri Gribenko         (EscapePtr - 2 >= BufferPtr && EscapePtr[0] == '/' &&
2645188c4b9SDmitri Gribenko          EscapePtr[-1] == '?' && EscapePtr[-2] == '?')) {
2655188c4b9SDmitri Gribenko       // We found an escaped newline.
2665188c4b9SDmitri Gribenko       CurPtr = skipNewline(CurPtr, BufferEnd);
2675188c4b9SDmitri Gribenko     } else
2685188c4b9SDmitri Gribenko       return CurPtr; // Not an escaped newline.
2695188c4b9SDmitri Gribenko   }
2705188c4b9SDmitri Gribenko   return BufferEnd;
2715188c4b9SDmitri Gribenko }
2725188c4b9SDmitri Gribenko 
2735188c4b9SDmitri Gribenko /// Return the one past end pointer for C comments.
2745188c4b9SDmitri Gribenko /// Very dumb, does not handle escaped newlines or trigraphs.
2755188c4b9SDmitri Gribenko const char *findCCommentEnd(const char *BufferPtr, const char *BufferEnd) {
2765188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2775188c4b9SDmitri Gribenko     if (*BufferPtr == '*') {
2785188c4b9SDmitri Gribenko       assert(BufferPtr + 1 != BufferEnd);
2795188c4b9SDmitri Gribenko       if (*(BufferPtr + 1) == '/')
2805188c4b9SDmitri Gribenko         return BufferPtr;
2815188c4b9SDmitri Gribenko     }
2825188c4b9SDmitri Gribenko   }
2835188c4b9SDmitri Gribenko   llvm_unreachable("buffer end hit before '*/' was seen");
2845188c4b9SDmitri Gribenko }
2855188c4b9SDmitri Gribenko } // unnamed namespace
2865188c4b9SDmitri Gribenko 
2875188c4b9SDmitri Gribenko void Lexer::lexCommentText(Token &T) {
2885188c4b9SDmitri Gribenko   assert(CommentState == LCS_InsideBCPLComment ||
2895188c4b9SDmitri Gribenko          CommentState == LCS_InsideCComment);
2905188c4b9SDmitri Gribenko 
2915188c4b9SDmitri Gribenko   switch (State) {
2925188c4b9SDmitri Gribenko   case LS_Normal:
2935188c4b9SDmitri Gribenko     break;
2945188c4b9SDmitri Gribenko   case LS_VerbatimBlockFirstLine:
2955188c4b9SDmitri Gribenko     lexVerbatimBlockFirstLine(T);
2965188c4b9SDmitri Gribenko     return;
2975188c4b9SDmitri Gribenko   case LS_VerbatimBlockBody:
2985188c4b9SDmitri Gribenko     lexVerbatimBlockBody(T);
2995188c4b9SDmitri Gribenko     return;
3001669f702SDmitri Gribenko   case LS_VerbatimLineText:
3011669f702SDmitri Gribenko     lexVerbatimLineText(T);
3021669f702SDmitri Gribenko     return;
303e00ffc7bSDmitri Gribenko   case LS_HTMLStartTag:
304e00ffc7bSDmitri Gribenko     lexHTMLStartTag(T);
3055188c4b9SDmitri Gribenko     return;
306e00ffc7bSDmitri Gribenko   case LS_HTMLEndTag:
307e00ffc7bSDmitri Gribenko     lexHTMLEndTag(T);
308ec92531cSDmitri Gribenko     return;
3095188c4b9SDmitri Gribenko   }
3105188c4b9SDmitri Gribenko 
3115188c4b9SDmitri Gribenko   assert(State == LS_Normal);
3125188c4b9SDmitri Gribenko 
3135188c4b9SDmitri Gribenko   const char *TokenPtr = BufferPtr;
3145188c4b9SDmitri Gribenko   assert(TokenPtr < CommentEnd);
3155188c4b9SDmitri Gribenko   while (TokenPtr != CommentEnd) {
3165188c4b9SDmitri Gribenko     switch(*TokenPtr) {
3175188c4b9SDmitri Gribenko       case '\\':
3185188c4b9SDmitri Gribenko       case '@': {
3195188c4b9SDmitri Gribenko         TokenPtr++;
3205188c4b9SDmitri Gribenko         if (TokenPtr == CommentEnd) {
3214586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
3225188c4b9SDmitri Gribenko           return;
3235188c4b9SDmitri Gribenko         }
3245188c4b9SDmitri Gribenko         char C = *TokenPtr;
3255188c4b9SDmitri Gribenko         switch (C) {
3265188c4b9SDmitri Gribenko         default:
3275188c4b9SDmitri Gribenko           break;
3285188c4b9SDmitri Gribenko 
3295188c4b9SDmitri Gribenko         case '\\': case '@': case '&': case '$':
3305188c4b9SDmitri Gribenko         case '#':  case '<': case '>': case '%':
3315188c4b9SDmitri Gribenko         case '\"': case '.': case ':':
3325188c4b9SDmitri Gribenko           // This is one of \\ \@ \& \$ etc escape sequences.
3335188c4b9SDmitri Gribenko           TokenPtr++;
3345188c4b9SDmitri Gribenko           if (C == ':' && TokenPtr != CommentEnd && *TokenPtr == ':') {
3355188c4b9SDmitri Gribenko             // This is the \:: escape sequence.
3365188c4b9SDmitri Gribenko             TokenPtr++;
3375188c4b9SDmitri Gribenko           }
33860ddd8a1SDmitri Gribenko           StringRef UnescapedText(BufferPtr + 1, TokenPtr - (BufferPtr + 1));
3395188c4b9SDmitri Gribenko           formTokenWithChars(T, TokenPtr, tok::text);
34060ddd8a1SDmitri Gribenko           T.setText(UnescapedText);
3415188c4b9SDmitri Gribenko           return;
3425188c4b9SDmitri Gribenko         }
3435188c4b9SDmitri Gribenko 
3445188c4b9SDmitri Gribenko         // Don't make zero-length commands.
345ad45ad6fSDmitri Gribenko         if (!isCommandNameStartCharacter(*TokenPtr)) {
3464586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
3475188c4b9SDmitri Gribenko           return;
3485188c4b9SDmitri Gribenko         }
3495188c4b9SDmitri Gribenko 
3505188c4b9SDmitri Gribenko         TokenPtr = skipCommandName(TokenPtr, CommentEnd);
3515188c4b9SDmitri Gribenko         unsigned Length = TokenPtr - (BufferPtr + 1);
3525188c4b9SDmitri Gribenko 
3535188c4b9SDmitri Gribenko         // Hardcoded support for lexing LaTeX formula commands
3545188c4b9SDmitri Gribenko         // \f$ \f[ \f] \f{ \f} as a single command.
3555188c4b9SDmitri Gribenko         if (Length == 1 && TokenPtr[-1] == 'f' && TokenPtr != CommentEnd) {
3565188c4b9SDmitri Gribenko           C = *TokenPtr;
3575188c4b9SDmitri Gribenko           if (C == '$' || C == '[' || C == ']' || C == '{' || C == '}') {
3585188c4b9SDmitri Gribenko             TokenPtr++;
3595188c4b9SDmitri Gribenko             Length++;
3605188c4b9SDmitri Gribenko           }
3615188c4b9SDmitri Gribenko         }
3625188c4b9SDmitri Gribenko 
3635188c4b9SDmitri Gribenko         const StringRef CommandName(BufferPtr + 1, Length);
3645188c4b9SDmitri Gribenko 
3657acbf00fSDmitri Gribenko         const CommandInfo *Info = Traits.getCommandInfoOrNULL(CommandName);
3667acbf00fSDmitri Gribenko         if (!Info) {
3677acbf00fSDmitri Gribenko           formTokenWithChars(T, TokenPtr, tok::unknown_command);
3687acbf00fSDmitri Gribenko           T.setUnknownCommandName(CommandName);
3695188c4b9SDmitri Gribenko           return;
3705188c4b9SDmitri Gribenko         }
3717acbf00fSDmitri Gribenko         if (Info->IsVerbatimBlockCommand) {
3727acbf00fSDmitri Gribenko           setupAndLexVerbatimBlock(T, TokenPtr, *BufferPtr, Info);
3737acbf00fSDmitri Gribenko           return;
3747acbf00fSDmitri Gribenko         }
3757acbf00fSDmitri Gribenko         if (Info->IsVerbatimLineCommand) {
3767acbf00fSDmitri Gribenko           setupAndLexVerbatimLine(T, TokenPtr, Info);
3775188c4b9SDmitri Gribenko           return;
3785188c4b9SDmitri Gribenko         }
3795188c4b9SDmitri Gribenko         formTokenWithChars(T, TokenPtr, tok::command);
3807acbf00fSDmitri Gribenko         T.setCommandID(Info->getID());
3815188c4b9SDmitri Gribenko         return;
3825188c4b9SDmitri Gribenko       }
3835188c4b9SDmitri Gribenko 
3844586df76SDmitri Gribenko       case '&':
3854586df76SDmitri Gribenko         lexHTMLCharacterReference(T);
3864586df76SDmitri Gribenko         return;
3874586df76SDmitri Gribenko 
3885188c4b9SDmitri Gribenko       case '<': {
3895188c4b9SDmitri Gribenko         TokenPtr++;
3905188c4b9SDmitri Gribenko         if (TokenPtr == CommentEnd) {
3914586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
3925188c4b9SDmitri Gribenko           return;
3935188c4b9SDmitri Gribenko         }
3945188c4b9SDmitri Gribenko         const char C = *TokenPtr;
39517709ae8SDmitri Gribenko         if (isHTMLIdentifierStartingCharacter(C))
396e00ffc7bSDmitri Gribenko           setupAndLexHTMLStartTag(T);
3975188c4b9SDmitri Gribenko         else if (C == '/')
398e00ffc7bSDmitri Gribenko           setupAndLexHTMLEndTag(T);
3994586df76SDmitri Gribenko         else
4004586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
4014586df76SDmitri Gribenko 
4025188c4b9SDmitri Gribenko         return;
4035188c4b9SDmitri Gribenko       }
4045188c4b9SDmitri Gribenko 
4055188c4b9SDmitri Gribenko       case '\n':
4065188c4b9SDmitri Gribenko       case '\r':
4075188c4b9SDmitri Gribenko         TokenPtr = skipNewline(TokenPtr, CommentEnd);
4085188c4b9SDmitri Gribenko         formTokenWithChars(T, TokenPtr, tok::newline);
4095188c4b9SDmitri Gribenko 
4105188c4b9SDmitri Gribenko         if (CommentState == LCS_InsideCComment)
4115188c4b9SDmitri Gribenko           skipLineStartingDecorations();
4125188c4b9SDmitri Gribenko         return;
4135188c4b9SDmitri Gribenko 
4145188c4b9SDmitri Gribenko       default: {
41510af67a9SDmitri Gribenko         size_t End = StringRef(TokenPtr, CommentEnd - TokenPtr).
41610af67a9SDmitri Gribenko                          find_first_of("\n\r\\@&<");
41710af67a9SDmitri Gribenko         if (End != StringRef::npos)
41810af67a9SDmitri Gribenko           TokenPtr += End;
41910af67a9SDmitri Gribenko         else
42010af67a9SDmitri Gribenko           TokenPtr = CommentEnd;
4214586df76SDmitri Gribenko         formTextToken(T, TokenPtr);
4225188c4b9SDmitri Gribenko         return;
4235188c4b9SDmitri Gribenko       }
4245188c4b9SDmitri Gribenko     }
4255188c4b9SDmitri Gribenko   }
4265188c4b9SDmitri Gribenko }
4275188c4b9SDmitri Gribenko 
4285188c4b9SDmitri Gribenko void Lexer::setupAndLexVerbatimBlock(Token &T,
4295188c4b9SDmitri Gribenko                                      const char *TextBegin,
4307acbf00fSDmitri Gribenko                                      char Marker, const CommandInfo *Info) {
4317acbf00fSDmitri Gribenko   assert(Info->IsVerbatimBlockCommand);
4327acbf00fSDmitri Gribenko 
4335188c4b9SDmitri Gribenko   VerbatimBlockEndCommandName.clear();
4345188c4b9SDmitri Gribenko   VerbatimBlockEndCommandName.append(Marker == '\\' ? "\\" : "@");
4357acbf00fSDmitri Gribenko   VerbatimBlockEndCommandName.append(Info->EndCommandName);
4365188c4b9SDmitri Gribenko 
4375188c4b9SDmitri Gribenko   formTokenWithChars(T, TextBegin, tok::verbatim_block_begin);
4387acbf00fSDmitri Gribenko   T.setVerbatimBlockID(Info->getID());
4395188c4b9SDmitri Gribenko 
440ec92531cSDmitri Gribenko   // If there is a newline following the verbatim opening command, skip the
441ec92531cSDmitri Gribenko   // newline so that we don't create an tok::verbatim_block_line with empty
442ec92531cSDmitri Gribenko   // text content.
443ec92531cSDmitri Gribenko   if (BufferPtr != CommentEnd) {
444ec92531cSDmitri Gribenko     const char C = *BufferPtr;
445ec92531cSDmitri Gribenko     if (C == '\n' || C == '\r') {
446ec92531cSDmitri Gribenko       BufferPtr = skipNewline(BufferPtr, CommentEnd);
447ec92531cSDmitri Gribenko       State = LS_VerbatimBlockBody;
448ec92531cSDmitri Gribenko       return;
449ec92531cSDmitri Gribenko     }
450ec92531cSDmitri Gribenko   }
451ec92531cSDmitri Gribenko 
4525188c4b9SDmitri Gribenko   State = LS_VerbatimBlockFirstLine;
4535188c4b9SDmitri Gribenko }
4545188c4b9SDmitri Gribenko 
4555188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockFirstLine(Token &T) {
456e4a3997dSDmitri Gribenko again:
4575188c4b9SDmitri Gribenko   assert(BufferPtr < CommentEnd);
4585188c4b9SDmitri Gribenko 
4595188c4b9SDmitri Gribenko   // FIXME: It would be better to scan the text once, finding either the block
4605188c4b9SDmitri Gribenko   // end command or newline.
4615188c4b9SDmitri Gribenko   //
4625188c4b9SDmitri Gribenko   // Extract current line.
4635188c4b9SDmitri Gribenko   const char *Newline = findNewline(BufferPtr, CommentEnd);
4645188c4b9SDmitri Gribenko   StringRef Line(BufferPtr, Newline - BufferPtr);
4655188c4b9SDmitri Gribenko 
4665188c4b9SDmitri Gribenko   // Look for end command in current line.
4675188c4b9SDmitri Gribenko   size_t Pos = Line.find(VerbatimBlockEndCommandName);
468ec92531cSDmitri Gribenko   const char *TextEnd;
4695188c4b9SDmitri Gribenko   const char *NextLine;
4705188c4b9SDmitri Gribenko   if (Pos == StringRef::npos) {
4715188c4b9SDmitri Gribenko     // Current line is completely verbatim.
472ec92531cSDmitri Gribenko     TextEnd = Newline;
4735188c4b9SDmitri Gribenko     NextLine = skipNewline(Newline, CommentEnd);
4745188c4b9SDmitri Gribenko   } else if (Pos == 0) {
4755188c4b9SDmitri Gribenko     // Current line contains just an end command.
4765188c4b9SDmitri Gribenko     const char *End = BufferPtr + VerbatimBlockEndCommandName.size();
47760ddd8a1SDmitri Gribenko     StringRef Name(BufferPtr + 1, End - (BufferPtr + 1));
4785188c4b9SDmitri Gribenko     formTokenWithChars(T, End, tok::verbatim_block_end);
4797acbf00fSDmitri Gribenko     T.setVerbatimBlockID(Traits.getCommandInfo(Name)->getID());
4805188c4b9SDmitri Gribenko     State = LS_Normal;
4815188c4b9SDmitri Gribenko     return;
4825188c4b9SDmitri Gribenko   } else {
4835188c4b9SDmitri Gribenko     // There is some text, followed by end command.  Extract text first.
484ec92531cSDmitri Gribenko     TextEnd = BufferPtr + Pos;
485ec92531cSDmitri Gribenko     NextLine = TextEnd;
486e4a3997dSDmitri Gribenko     // If there is only whitespace before end command, skip whitespace.
487e4a3997dSDmitri Gribenko     if (isWhitespace(BufferPtr, TextEnd)) {
488e4a3997dSDmitri Gribenko       BufferPtr = TextEnd;
489e4a3997dSDmitri Gribenko       goto again;
490e4a3997dSDmitri Gribenko     }
4915188c4b9SDmitri Gribenko   }
4925188c4b9SDmitri Gribenko 
493ec92531cSDmitri Gribenko   StringRef Text(BufferPtr, TextEnd - BufferPtr);
4945188c4b9SDmitri Gribenko   formTokenWithChars(T, NextLine, tok::verbatim_block_line);
49560ddd8a1SDmitri Gribenko   T.setVerbatimBlockText(Text);
4965188c4b9SDmitri Gribenko 
4975188c4b9SDmitri Gribenko   State = LS_VerbatimBlockBody;
4985188c4b9SDmitri Gribenko }
4995188c4b9SDmitri Gribenko 
5005188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockBody(Token &T) {
5015188c4b9SDmitri Gribenko   assert(State == LS_VerbatimBlockBody);
5025188c4b9SDmitri Gribenko 
5035188c4b9SDmitri Gribenko   if (CommentState == LCS_InsideCComment)
5045188c4b9SDmitri Gribenko     skipLineStartingDecorations();
5055188c4b9SDmitri Gribenko 
5065188c4b9SDmitri Gribenko   lexVerbatimBlockFirstLine(T);
5075188c4b9SDmitri Gribenko }
5085188c4b9SDmitri Gribenko 
5097acbf00fSDmitri Gribenko void Lexer::setupAndLexVerbatimLine(Token &T, const char *TextBegin,
5107acbf00fSDmitri Gribenko                                     const CommandInfo *Info) {
5117acbf00fSDmitri Gribenko   assert(Info->IsVerbatimLineCommand);
5121669f702SDmitri Gribenko   formTokenWithChars(T, TextBegin, tok::verbatim_line_name);
5137acbf00fSDmitri Gribenko   T.setVerbatimLineID(Info->getID());
5141669f702SDmitri Gribenko 
5151669f702SDmitri Gribenko   State = LS_VerbatimLineText;
5161669f702SDmitri Gribenko }
5171669f702SDmitri Gribenko 
5181669f702SDmitri Gribenko void Lexer::lexVerbatimLineText(Token &T) {
5191669f702SDmitri Gribenko   assert(State == LS_VerbatimLineText);
5201669f702SDmitri Gribenko 
5215188c4b9SDmitri Gribenko   // Extract current line.
5225188c4b9SDmitri Gribenko   const char *Newline = findNewline(BufferPtr, CommentEnd);
5231669f702SDmitri Gribenko   const StringRef Text(BufferPtr, Newline - BufferPtr);
5241669f702SDmitri Gribenko   formTokenWithChars(T, Newline, tok::verbatim_line_text);
5255188c4b9SDmitri Gribenko   T.setVerbatimLineText(Text);
5261669f702SDmitri Gribenko 
5271669f702SDmitri Gribenko   State = LS_Normal;
5285188c4b9SDmitri Gribenko }
5295188c4b9SDmitri Gribenko 
5304586df76SDmitri Gribenko void Lexer::lexHTMLCharacterReference(Token &T) {
5314586df76SDmitri Gribenko   const char *TokenPtr = BufferPtr;
5324586df76SDmitri Gribenko   assert(*TokenPtr == '&');
5334586df76SDmitri Gribenko   TokenPtr++;
5344586df76SDmitri Gribenko   if (TokenPtr == CommentEnd) {
5354586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5364586df76SDmitri Gribenko     return;
5374586df76SDmitri Gribenko   }
5384586df76SDmitri Gribenko   const char *NamePtr;
5394586df76SDmitri Gribenko   bool isNamed = false;
5404586df76SDmitri Gribenko   bool isDecimal = false;
5414586df76SDmitri Gribenko   char C = *TokenPtr;
5424586df76SDmitri Gribenko   if (isHTMLNamedCharacterReferenceCharacter(C)) {
5434586df76SDmitri Gribenko     NamePtr = TokenPtr;
5444586df76SDmitri Gribenko     TokenPtr = skipNamedCharacterReference(TokenPtr, CommentEnd);
5454586df76SDmitri Gribenko     isNamed = true;
5464586df76SDmitri Gribenko   } else if (C == '#') {
5474586df76SDmitri Gribenko     TokenPtr++;
5484586df76SDmitri Gribenko     if (TokenPtr == CommentEnd) {
5494586df76SDmitri Gribenko       formTextToken(T, TokenPtr);
5504586df76SDmitri Gribenko       return;
5514586df76SDmitri Gribenko     }
5524586df76SDmitri Gribenko     C = *TokenPtr;
5534586df76SDmitri Gribenko     if (isHTMLDecimalCharacterReferenceCharacter(C)) {
5544586df76SDmitri Gribenko       NamePtr = TokenPtr;
5554586df76SDmitri Gribenko       TokenPtr = skipDecimalCharacterReference(TokenPtr, CommentEnd);
5564586df76SDmitri Gribenko       isDecimal = true;
5574586df76SDmitri Gribenko     } else if (C == 'x' || C == 'X') {
5584586df76SDmitri Gribenko       TokenPtr++;
5594586df76SDmitri Gribenko       NamePtr = TokenPtr;
5604586df76SDmitri Gribenko       TokenPtr = skipHexCharacterReference(TokenPtr, CommentEnd);
5614586df76SDmitri Gribenko     } else {
5624586df76SDmitri Gribenko       formTextToken(T, TokenPtr);
5634586df76SDmitri Gribenko       return;
5644586df76SDmitri Gribenko     }
5654586df76SDmitri Gribenko   } else {
5664586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5674586df76SDmitri Gribenko     return;
5684586df76SDmitri Gribenko   }
5694586df76SDmitri Gribenko   if (NamePtr == TokenPtr || TokenPtr == CommentEnd ||
5704586df76SDmitri Gribenko       *TokenPtr != ';') {
5714586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5724586df76SDmitri Gribenko     return;
5734586df76SDmitri Gribenko   }
5744586df76SDmitri Gribenko   StringRef Name(NamePtr, TokenPtr - NamePtr);
5754586df76SDmitri Gribenko   TokenPtr++; // Skip semicolon.
5764586df76SDmitri Gribenko   StringRef Resolved;
577*28800da1SDmitri Gribenko   if (isNamed)
5784586df76SDmitri Gribenko     Resolved = resolveHTMLNamedCharacterReference(Name);
5794586df76SDmitri Gribenko   else if (isDecimal)
5804586df76SDmitri Gribenko     Resolved = resolveHTMLDecimalCharacterReference(Name);
5814586df76SDmitri Gribenko   else
5824586df76SDmitri Gribenko     Resolved = resolveHTMLHexCharacterReference(Name);
5834586df76SDmitri Gribenko 
5844586df76SDmitri Gribenko   if (Resolved.empty()) {
5854586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5864586df76SDmitri Gribenko     return;
5874586df76SDmitri Gribenko   }
5884586df76SDmitri Gribenko   formTokenWithChars(T, TokenPtr, tok::text);
5894586df76SDmitri Gribenko   T.setText(Resolved);
5904586df76SDmitri Gribenko   return;
5914586df76SDmitri Gribenko }
5924586df76SDmitri Gribenko 
593e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLStartTag(Token &T) {
59417709ae8SDmitri Gribenko   assert(BufferPtr[0] == '<' &&
59517709ae8SDmitri Gribenko          isHTMLIdentifierStartingCharacter(BufferPtr[1]));
5965188c4b9SDmitri Gribenko   const char *TagNameEnd = skipHTMLIdentifier(BufferPtr + 2, CommentEnd);
59760ddd8a1SDmitri Gribenko   StringRef Name(BufferPtr + 1, TagNameEnd - (BufferPtr + 1));
598107618a6SDmitri Gribenko   if (!isHTMLTagName(Name)) {
599107618a6SDmitri Gribenko     formTextToken(T, TagNameEnd);
600107618a6SDmitri Gribenko     return;
601107618a6SDmitri Gribenko   }
602107618a6SDmitri Gribenko 
603e00ffc7bSDmitri Gribenko   formTokenWithChars(T, TagNameEnd, tok::html_start_tag);
604e00ffc7bSDmitri Gribenko   T.setHTMLTagStartName(Name);
6055188c4b9SDmitri Gribenko 
6065188c4b9SDmitri Gribenko   BufferPtr = skipWhitespace(BufferPtr, CommentEnd);
6075188c4b9SDmitri Gribenko 
60817709ae8SDmitri Gribenko   const char C = *BufferPtr;
60917709ae8SDmitri Gribenko   if (BufferPtr != CommentEnd &&
610f26054f0SDmitri Gribenko       (C == '>' || C == '/' || isHTMLIdentifierStartingCharacter(C)))
611e00ffc7bSDmitri Gribenko     State = LS_HTMLStartTag;
6125188c4b9SDmitri Gribenko }
6135188c4b9SDmitri Gribenko 
614e00ffc7bSDmitri Gribenko void Lexer::lexHTMLStartTag(Token &T) {
615e00ffc7bSDmitri Gribenko   assert(State == LS_HTMLStartTag);
6165188c4b9SDmitri Gribenko 
6175188c4b9SDmitri Gribenko   const char *TokenPtr = BufferPtr;
6185188c4b9SDmitri Gribenko   char C = *TokenPtr;
6195188c4b9SDmitri Gribenko   if (isHTMLIdentifierCharacter(C)) {
6205188c4b9SDmitri Gribenko     TokenPtr = skipHTMLIdentifier(TokenPtr, CommentEnd);
62160ddd8a1SDmitri Gribenko     StringRef Ident(BufferPtr, TokenPtr - BufferPtr);
6225188c4b9SDmitri Gribenko     formTokenWithChars(T, TokenPtr, tok::html_ident);
62360ddd8a1SDmitri Gribenko     T.setHTMLIdent(Ident);
6245188c4b9SDmitri Gribenko   } else {
6255188c4b9SDmitri Gribenko     switch (C) {
6265188c4b9SDmitri Gribenko     case '=':
6275188c4b9SDmitri Gribenko       TokenPtr++;
6285188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_equals);
6295188c4b9SDmitri Gribenko       break;
6305188c4b9SDmitri Gribenko     case '\"':
6315188c4b9SDmitri Gribenko     case '\'': {
6325188c4b9SDmitri Gribenko       const char *OpenQuote = TokenPtr;
6335188c4b9SDmitri Gribenko       TokenPtr = skipHTMLQuotedString(TokenPtr, CommentEnd);
6345188c4b9SDmitri Gribenko       const char *ClosingQuote = TokenPtr;
6355188c4b9SDmitri Gribenko       if (TokenPtr != CommentEnd) // Skip closing quote.
6365188c4b9SDmitri Gribenko         TokenPtr++;
6375188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_quoted_string);
6385188c4b9SDmitri Gribenko       T.setHTMLQuotedString(StringRef(OpenQuote + 1,
6395188c4b9SDmitri Gribenko                                       ClosingQuote - (OpenQuote + 1)));
6405188c4b9SDmitri Gribenko       break;
6415188c4b9SDmitri Gribenko     }
6425188c4b9SDmitri Gribenko     case '>':
6435188c4b9SDmitri Gribenko       TokenPtr++;
6445188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_greater);
64517709ae8SDmitri Gribenko       State = LS_Normal;
64617709ae8SDmitri Gribenko       return;
647f26054f0SDmitri Gribenko     case '/':
648f26054f0SDmitri Gribenko       TokenPtr++;
649f26054f0SDmitri Gribenko       if (TokenPtr != CommentEnd && *TokenPtr == '>') {
650f26054f0SDmitri Gribenko         TokenPtr++;
651f26054f0SDmitri Gribenko         formTokenWithChars(T, TokenPtr, tok::html_slash_greater);
6524586df76SDmitri Gribenko       } else
6534586df76SDmitri Gribenko         formTextToken(T, TokenPtr);
6544586df76SDmitri Gribenko 
655f26054f0SDmitri Gribenko       State = LS_Normal;
656f26054f0SDmitri Gribenko       return;
6575188c4b9SDmitri Gribenko     }
6585188c4b9SDmitri Gribenko   }
6595188c4b9SDmitri Gribenko 
6605188c4b9SDmitri Gribenko   // Now look ahead and return to normal state if we don't see any HTML tokens
6615188c4b9SDmitri Gribenko   // ahead.
6625188c4b9SDmitri Gribenko   BufferPtr = skipWhitespace(BufferPtr, CommentEnd);
6635188c4b9SDmitri Gribenko   if (BufferPtr == CommentEnd) {
6645188c4b9SDmitri Gribenko     State = LS_Normal;
6655188c4b9SDmitri Gribenko     return;
6665188c4b9SDmitri Gribenko   }
6675188c4b9SDmitri Gribenko 
6685188c4b9SDmitri Gribenko   C = *BufferPtr;
66917709ae8SDmitri Gribenko   if (!isHTMLIdentifierStartingCharacter(C) &&
6705188c4b9SDmitri Gribenko       C != '=' && C != '\"' && C != '\'' && C != '>') {
6715188c4b9SDmitri Gribenko     State = LS_Normal;
6725188c4b9SDmitri Gribenko     return;
6735188c4b9SDmitri Gribenko   }
6745188c4b9SDmitri Gribenko }
6755188c4b9SDmitri Gribenko 
676e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLEndTag(Token &T) {
6775188c4b9SDmitri Gribenko   assert(BufferPtr[0] == '<' && BufferPtr[1] == '/');
6785188c4b9SDmitri Gribenko 
6795188c4b9SDmitri Gribenko   const char *TagNameBegin = skipWhitespace(BufferPtr + 2, CommentEnd);
6805188c4b9SDmitri Gribenko   const char *TagNameEnd = skipHTMLIdentifier(TagNameBegin, CommentEnd);
681107618a6SDmitri Gribenko   StringRef Name(TagNameBegin, TagNameEnd - TagNameBegin);
682107618a6SDmitri Gribenko   if (!isHTMLTagName(Name)) {
683107618a6SDmitri Gribenko     formTextToken(T, TagNameEnd);
684107618a6SDmitri Gribenko     return;
685107618a6SDmitri Gribenko   }
6865188c4b9SDmitri Gribenko 
6875188c4b9SDmitri Gribenko   const char *End = skipWhitespace(TagNameEnd, CommentEnd);
6885188c4b9SDmitri Gribenko 
689e00ffc7bSDmitri Gribenko   formTokenWithChars(T, End, tok::html_end_tag);
690107618a6SDmitri Gribenko   T.setHTMLTagEndName(Name);
691ec92531cSDmitri Gribenko 
692ec92531cSDmitri Gribenko   if (BufferPtr != CommentEnd && *BufferPtr == '>')
693e00ffc7bSDmitri Gribenko     State = LS_HTMLEndTag;
694ec92531cSDmitri Gribenko }
695ec92531cSDmitri Gribenko 
696e00ffc7bSDmitri Gribenko void Lexer::lexHTMLEndTag(Token &T) {
697ec92531cSDmitri Gribenko   assert(BufferPtr != CommentEnd && *BufferPtr == '>');
698ec92531cSDmitri Gribenko 
699ec92531cSDmitri Gribenko   formTokenWithChars(T, BufferPtr + 1, tok::html_greater);
700ec92531cSDmitri Gribenko   State = LS_Normal;
7015188c4b9SDmitri Gribenko }
7025188c4b9SDmitri Gribenko 
703ca7f80adSDmitri Gribenko Lexer::Lexer(llvm::BumpPtrAllocator &Allocator, const CommandTraits &Traits,
7046bab9113SDmitri Gribenko              SourceLocation FileLoc,
7055188c4b9SDmitri Gribenko              const char *BufferStart, const char *BufferEnd):
706ca7f80adSDmitri Gribenko     Allocator(Allocator), Traits(Traits),
7075188c4b9SDmitri Gribenko     BufferStart(BufferStart), BufferEnd(BufferEnd),
7086bab9113SDmitri Gribenko     FileLoc(FileLoc), BufferPtr(BufferStart),
7095188c4b9SDmitri Gribenko     CommentState(LCS_BeforeComment), State(LS_Normal) {
7105188c4b9SDmitri Gribenko }
7115188c4b9SDmitri Gribenko 
7125188c4b9SDmitri Gribenko void Lexer::lex(Token &T) {
7135188c4b9SDmitri Gribenko again:
7145188c4b9SDmitri Gribenko   switch (CommentState) {
7155188c4b9SDmitri Gribenko   case LCS_BeforeComment:
7165188c4b9SDmitri Gribenko     if (BufferPtr == BufferEnd) {
7175188c4b9SDmitri Gribenko       formTokenWithChars(T, BufferPtr, tok::eof);
7185188c4b9SDmitri Gribenko       return;
7195188c4b9SDmitri Gribenko     }
7205188c4b9SDmitri Gribenko 
7215188c4b9SDmitri Gribenko     assert(*BufferPtr == '/');
7225188c4b9SDmitri Gribenko     BufferPtr++; // Skip first slash.
7235188c4b9SDmitri Gribenko     switch(*BufferPtr) {
7245188c4b9SDmitri Gribenko     case '/': { // BCPL comment.
7255188c4b9SDmitri Gribenko       BufferPtr++; // Skip second slash.
7265188c4b9SDmitri Gribenko 
7275188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd) {
7285188c4b9SDmitri Gribenko         // Skip Doxygen magic marker, if it is present.
7295188c4b9SDmitri Gribenko         // It might be missing because of a typo //< or /*<, or because we
7305188c4b9SDmitri Gribenko         // merged this non-Doxygen comment into a bunch of Doxygen comments
7315188c4b9SDmitri Gribenko         // around it: /** ... */ /* ... */ /** ... */
7325188c4b9SDmitri Gribenko         const char C = *BufferPtr;
7335188c4b9SDmitri Gribenko         if (C == '/' || C == '!')
7345188c4b9SDmitri Gribenko           BufferPtr++;
7355188c4b9SDmitri Gribenko       }
7365188c4b9SDmitri Gribenko 
7375188c4b9SDmitri Gribenko       // Skip less-than symbol that marks trailing comments.
7385188c4b9SDmitri Gribenko       // Skip it even if the comment is not a Doxygen one, because //< and /*<
7395188c4b9SDmitri Gribenko       // are frequent typos.
7405188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd && *BufferPtr == '<')
7415188c4b9SDmitri Gribenko         BufferPtr++;
7425188c4b9SDmitri Gribenko 
7435188c4b9SDmitri Gribenko       CommentState = LCS_InsideBCPLComment;
744ec92531cSDmitri Gribenko       if (State != LS_VerbatimBlockBody && State != LS_VerbatimBlockFirstLine)
7455188c4b9SDmitri Gribenko         State = LS_Normal;
7465188c4b9SDmitri Gribenko       CommentEnd = findBCPLCommentEnd(BufferPtr, BufferEnd);
7475188c4b9SDmitri Gribenko       goto again;
7485188c4b9SDmitri Gribenko     }
7495188c4b9SDmitri Gribenko     case '*': { // C comment.
7505188c4b9SDmitri Gribenko       BufferPtr++; // Skip star.
7515188c4b9SDmitri Gribenko 
7525188c4b9SDmitri Gribenko       // Skip Doxygen magic marker.
7535188c4b9SDmitri Gribenko       const char C = *BufferPtr;
7545188c4b9SDmitri Gribenko       if ((C == '*' && *(BufferPtr + 1) != '/') || C == '!')
7555188c4b9SDmitri Gribenko         BufferPtr++;
7565188c4b9SDmitri Gribenko 
7575188c4b9SDmitri Gribenko       // Skip less-than symbol that marks trailing comments.
7585188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd && *BufferPtr == '<')
7595188c4b9SDmitri Gribenko         BufferPtr++;
7605188c4b9SDmitri Gribenko 
7615188c4b9SDmitri Gribenko       CommentState = LCS_InsideCComment;
7625188c4b9SDmitri Gribenko       State = LS_Normal;
7635188c4b9SDmitri Gribenko       CommentEnd = findCCommentEnd(BufferPtr, BufferEnd);
7645188c4b9SDmitri Gribenko       goto again;
7655188c4b9SDmitri Gribenko     }
7665188c4b9SDmitri Gribenko     default:
7675188c4b9SDmitri Gribenko       llvm_unreachable("second character of comment should be '/' or '*'");
7685188c4b9SDmitri Gribenko     }
7695188c4b9SDmitri Gribenko 
7705188c4b9SDmitri Gribenko   case LCS_BetweenComments: {
7715188c4b9SDmitri Gribenko     // Consecutive comments are extracted only if there is only whitespace
7725188c4b9SDmitri Gribenko     // between them.  So we can search for the start of the next comment.
7735188c4b9SDmitri Gribenko     const char *EndWhitespace = BufferPtr;
7745188c4b9SDmitri Gribenko     while(EndWhitespace != BufferEnd && *EndWhitespace != '/')
7755188c4b9SDmitri Gribenko       EndWhitespace++;
7765188c4b9SDmitri Gribenko 
7775188c4b9SDmitri Gribenko     // Turn any whitespace between comments (and there is only whitespace
77817709ae8SDmitri Gribenko     // between them -- guaranteed by comment extraction) into a newline.  We
77917709ae8SDmitri Gribenko     // have two newlines between C comments in total (first one was synthesized
78017709ae8SDmitri Gribenko     // after a comment).
7815188c4b9SDmitri Gribenko     formTokenWithChars(T, EndWhitespace, tok::newline);
7825188c4b9SDmitri Gribenko 
7835188c4b9SDmitri Gribenko     CommentState = LCS_BeforeComment;
7845188c4b9SDmitri Gribenko     break;
7855188c4b9SDmitri Gribenko   }
7865188c4b9SDmitri Gribenko 
7875188c4b9SDmitri Gribenko   case LCS_InsideBCPLComment:
7885188c4b9SDmitri Gribenko   case LCS_InsideCComment:
7895188c4b9SDmitri Gribenko     if (BufferPtr != CommentEnd) {
7905188c4b9SDmitri Gribenko       lexCommentText(T);
7915188c4b9SDmitri Gribenko       break;
7925188c4b9SDmitri Gribenko     } else {
7935188c4b9SDmitri Gribenko       // Skip C comment closing sequence.
7945188c4b9SDmitri Gribenko       if (CommentState == LCS_InsideCComment) {
7955188c4b9SDmitri Gribenko         assert(BufferPtr[0] == '*' && BufferPtr[1] == '/');
7965188c4b9SDmitri Gribenko         BufferPtr += 2;
7975188c4b9SDmitri Gribenko         assert(BufferPtr <= BufferEnd);
7985188c4b9SDmitri Gribenko 
7995188c4b9SDmitri Gribenko         // Synthenize newline just after the C comment, regardless if there is
8005188c4b9SDmitri Gribenko         // actually a newline.
8015188c4b9SDmitri Gribenko         formTokenWithChars(T, BufferPtr, tok::newline);
8025188c4b9SDmitri Gribenko 
8035188c4b9SDmitri Gribenko         CommentState = LCS_BetweenComments;
8045188c4b9SDmitri Gribenko         break;
8055188c4b9SDmitri Gribenko       } else {
8065188c4b9SDmitri Gribenko         // Don't synthesized a newline after BCPL comment.
8075188c4b9SDmitri Gribenko         CommentState = LCS_BetweenComments;
8085188c4b9SDmitri Gribenko         goto again;
8095188c4b9SDmitri Gribenko       }
8105188c4b9SDmitri Gribenko     }
8115188c4b9SDmitri Gribenko   }
8125188c4b9SDmitri Gribenko }
8135188c4b9SDmitri Gribenko 
8145188c4b9SDmitri Gribenko StringRef Lexer::getSpelling(const Token &Tok,
8155188c4b9SDmitri Gribenko                              const SourceManager &SourceMgr,
8165188c4b9SDmitri Gribenko                              bool *Invalid) const {
8175188c4b9SDmitri Gribenko   SourceLocation Loc = Tok.getLocation();
8185188c4b9SDmitri Gribenko   std::pair<FileID, unsigned> LocInfo = SourceMgr.getDecomposedLoc(Loc);
8195188c4b9SDmitri Gribenko 
8205188c4b9SDmitri Gribenko   bool InvalidTemp = false;
8215188c4b9SDmitri Gribenko   StringRef File = SourceMgr.getBufferData(LocInfo.first, &InvalidTemp);
8225188c4b9SDmitri Gribenko   if (InvalidTemp) {
8235188c4b9SDmitri Gribenko     *Invalid = true;
8245188c4b9SDmitri Gribenko     return StringRef();
8255188c4b9SDmitri Gribenko   }
8265188c4b9SDmitri Gribenko 
8275188c4b9SDmitri Gribenko   const char *Begin = File.data() + LocInfo.second;
8285188c4b9SDmitri Gribenko   return StringRef(Begin, Tok.getLength());
8295188c4b9SDmitri Gribenko }
8305188c4b9SDmitri Gribenko 
8315188c4b9SDmitri Gribenko } // end namespace comments
8325188c4b9SDmitri Gribenko } // end namespace clang
8335188c4b9SDmitri Gribenko 
834