15188c4b9SDmitri Gribenko #include "clang/AST/CommentLexer.h"
2ca7f80adSDmitri Gribenko #include "clang/AST/CommentCommandTraits.h"
34586df76SDmitri Gribenko #include "clang/Basic/ConvertUTF.h"
45188c4b9SDmitri Gribenko #include "llvm/ADT/StringSwitch.h"
55188c4b9SDmitri Gribenko #include "llvm/Support/ErrorHandling.h"
65188c4b9SDmitri Gribenko 
75188c4b9SDmitri Gribenko namespace clang {
85188c4b9SDmitri Gribenko namespace comments {
95188c4b9SDmitri Gribenko 
105188c4b9SDmitri Gribenko void Token::dump(const Lexer &L, const SourceManager &SM) const {
115188c4b9SDmitri Gribenko   llvm::errs() << "comments::Token Kind=" << Kind << " ";
125188c4b9SDmitri Gribenko   Loc.dump(SM);
135188c4b9SDmitri Gribenko   llvm::errs() << " " << Length << " \"" << L.getSpelling(*this, SM) << "\"\n";
145188c4b9SDmitri Gribenko }
155188c4b9SDmitri Gribenko 
164586df76SDmitri Gribenko namespace {
174586df76SDmitri Gribenko bool isHTMLNamedCharacterReferenceCharacter(char C) {
184586df76SDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
194586df76SDmitri Gribenko          (C >= 'A' && C <= 'Z');
204586df76SDmitri Gribenko }
214586df76SDmitri Gribenko 
224586df76SDmitri Gribenko bool isHTMLDecimalCharacterReferenceCharacter(char C) {
234586df76SDmitri Gribenko   return C >= '0' && C <= '9';
244586df76SDmitri Gribenko }
254586df76SDmitri Gribenko 
264586df76SDmitri Gribenko bool isHTMLHexCharacterReferenceCharacter(char C) {
274586df76SDmitri Gribenko   return (C >= '0' && C <= '9') ||
284586df76SDmitri Gribenko          (C >= 'a' && C <= 'f') ||
294586df76SDmitri Gribenko          (C >= 'A' && C <= 'F');
304586df76SDmitri Gribenko }
31107618a6SDmitri Gribenko 
323ca956f7SDmitri Gribenko #include "clang/AST/CommentHTMLTags.inc"
333ca956f7SDmitri Gribenko 
344586df76SDmitri Gribenko } // unnamed namespace
354586df76SDmitri Gribenko 
364586df76SDmitri Gribenko StringRef Lexer::resolveHTMLNamedCharacterReference(StringRef Name) const {
374586df76SDmitri Gribenko   return llvm::StringSwitch<StringRef>(Name)
384586df76SDmitri Gribenko       .Case("amp", "&")
394586df76SDmitri Gribenko       .Case("lt", "<")
404586df76SDmitri Gribenko       .Case("gt", ">")
414586df76SDmitri Gribenko       .Case("quot", "\"")
424586df76SDmitri Gribenko       .Case("apos", "\'")
434586df76SDmitri Gribenko       .Default("");
444586df76SDmitri Gribenko }
454586df76SDmitri Gribenko 
464586df76SDmitri Gribenko StringRef Lexer::resolveHTMLDecimalCharacterReference(StringRef Name) const {
474586df76SDmitri Gribenko   unsigned CodePoint = 0;
484586df76SDmitri Gribenko   for (unsigned i = 0, e = Name.size(); i != e; ++i) {
494586df76SDmitri Gribenko     assert(isHTMLDecimalCharacterReferenceCharacter(Name[i]));
504586df76SDmitri Gribenko     CodePoint *= 10;
514586df76SDmitri Gribenko     CodePoint += Name[i] - '0';
524586df76SDmitri Gribenko   }
534586df76SDmitri Gribenko 
544586df76SDmitri Gribenko   char *Resolved = Allocator.Allocate<char>(UNI_MAX_UTF8_BYTES_PER_CODE_POINT);
554586df76SDmitri Gribenko   char *ResolvedPtr = Resolved;
564586df76SDmitri Gribenko   if (ConvertCodePointToUTF8(CodePoint, ResolvedPtr))
574586df76SDmitri Gribenko     return StringRef(Resolved, ResolvedPtr - Resolved);
584586df76SDmitri Gribenko   else
594586df76SDmitri Gribenko     return StringRef();
604586df76SDmitri Gribenko }
614586df76SDmitri Gribenko 
624586df76SDmitri Gribenko StringRef Lexer::resolveHTMLHexCharacterReference(StringRef Name) const {
634586df76SDmitri Gribenko   unsigned CodePoint = 0;
644586df76SDmitri Gribenko   for (unsigned i = 0, e = Name.size(); i != e; ++i) {
654586df76SDmitri Gribenko     CodePoint *= 16;
664586df76SDmitri Gribenko     const char C = Name[i];
674586df76SDmitri Gribenko     assert(isHTMLHexCharacterReferenceCharacter(C));
684586df76SDmitri Gribenko     if (C >= '0' && C <= '9')
694586df76SDmitri Gribenko       CodePoint += Name[i] - '0';
704586df76SDmitri Gribenko     else if (C >= 'a' && C <= 'f')
714586df76SDmitri Gribenko       CodePoint += Name[i] - 'a' + 10;
724586df76SDmitri Gribenko     else
734586df76SDmitri Gribenko       CodePoint += Name[i] - 'A' + 10;
744586df76SDmitri Gribenko   }
754586df76SDmitri Gribenko 
764586df76SDmitri Gribenko   char *Resolved = Allocator.Allocate<char>(UNI_MAX_UTF8_BYTES_PER_CODE_POINT);
774586df76SDmitri Gribenko   char *ResolvedPtr = Resolved;
784586df76SDmitri Gribenko   if (ConvertCodePointToUTF8(CodePoint, ResolvedPtr))
794586df76SDmitri Gribenko     return StringRef(Resolved, ResolvedPtr - Resolved);
804586df76SDmitri Gribenko   else
814586df76SDmitri Gribenko     return StringRef();
824586df76SDmitri Gribenko }
834586df76SDmitri Gribenko 
845188c4b9SDmitri Gribenko void Lexer::skipLineStartingDecorations() {
855188c4b9SDmitri Gribenko   // This function should be called only for C comments
865188c4b9SDmitri Gribenko   assert(CommentState == LCS_InsideCComment);
875188c4b9SDmitri Gribenko 
885188c4b9SDmitri Gribenko   if (BufferPtr == CommentEnd)
895188c4b9SDmitri Gribenko     return;
905188c4b9SDmitri Gribenko 
915188c4b9SDmitri Gribenko   switch (*BufferPtr) {
925188c4b9SDmitri Gribenko   case ' ':
935188c4b9SDmitri Gribenko   case '\t':
945188c4b9SDmitri Gribenko   case '\f':
955188c4b9SDmitri Gribenko   case '\v': {
965188c4b9SDmitri Gribenko     const char *NewBufferPtr = BufferPtr;
975188c4b9SDmitri Gribenko     NewBufferPtr++;
985188c4b9SDmitri Gribenko     if (NewBufferPtr == CommentEnd)
995188c4b9SDmitri Gribenko       return;
1005188c4b9SDmitri Gribenko 
1015188c4b9SDmitri Gribenko     char C = *NewBufferPtr;
1025188c4b9SDmitri Gribenko     while (C == ' ' || C == '\t' || C == '\f' || C == '\v') {
1035188c4b9SDmitri Gribenko       NewBufferPtr++;
1045188c4b9SDmitri Gribenko       if (NewBufferPtr == CommentEnd)
1055188c4b9SDmitri Gribenko         return;
1065188c4b9SDmitri Gribenko       C = *NewBufferPtr;
1075188c4b9SDmitri Gribenko     }
1085188c4b9SDmitri Gribenko     if (C == '*')
1095188c4b9SDmitri Gribenko       BufferPtr = NewBufferPtr + 1;
1105188c4b9SDmitri Gribenko     break;
1115188c4b9SDmitri Gribenko   }
1125188c4b9SDmitri Gribenko   case '*':
1135188c4b9SDmitri Gribenko     BufferPtr++;
1145188c4b9SDmitri Gribenko     break;
1155188c4b9SDmitri Gribenko   }
1165188c4b9SDmitri Gribenko }
1175188c4b9SDmitri Gribenko 
1185188c4b9SDmitri Gribenko namespace {
119ec92531cSDmitri Gribenko /// Returns pointer to the first newline character in the string.
1205188c4b9SDmitri Gribenko const char *findNewline(const char *BufferPtr, const char *BufferEnd) {
1215188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1225188c4b9SDmitri Gribenko     const char C = *BufferPtr;
1235188c4b9SDmitri Gribenko     if (C == '\n' || C == '\r')
1245188c4b9SDmitri Gribenko       return BufferPtr;
1255188c4b9SDmitri Gribenko   }
1265188c4b9SDmitri Gribenko   return BufferEnd;
1275188c4b9SDmitri Gribenko }
1285188c4b9SDmitri Gribenko 
1295188c4b9SDmitri Gribenko const char *skipNewline(const char *BufferPtr, const char *BufferEnd) {
1305188c4b9SDmitri Gribenko   if (BufferPtr == BufferEnd)
1315188c4b9SDmitri Gribenko     return BufferPtr;
1325188c4b9SDmitri Gribenko 
1335188c4b9SDmitri Gribenko   if (*BufferPtr == '\n')
1345188c4b9SDmitri Gribenko     BufferPtr++;
1355188c4b9SDmitri Gribenko   else {
1365188c4b9SDmitri Gribenko     assert(*BufferPtr == '\r');
1375188c4b9SDmitri Gribenko     BufferPtr++;
1385188c4b9SDmitri Gribenko     if (BufferPtr != BufferEnd && *BufferPtr == '\n')
1395188c4b9SDmitri Gribenko       BufferPtr++;
1405188c4b9SDmitri Gribenko   }
1415188c4b9SDmitri Gribenko   return BufferPtr;
1425188c4b9SDmitri Gribenko }
1435188c4b9SDmitri Gribenko 
1444586df76SDmitri Gribenko const char *skipNamedCharacterReference(const char *BufferPtr,
1454586df76SDmitri Gribenko                                         const char *BufferEnd) {
1464586df76SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1474586df76SDmitri Gribenko     if (!isHTMLNamedCharacterReferenceCharacter(*BufferPtr))
1484586df76SDmitri Gribenko       return BufferPtr;
1494586df76SDmitri Gribenko   }
1504586df76SDmitri Gribenko   return BufferEnd;
1514586df76SDmitri Gribenko }
1524586df76SDmitri Gribenko 
1534586df76SDmitri Gribenko const char *skipDecimalCharacterReference(const char *BufferPtr,
1544586df76SDmitri Gribenko                                           const char *BufferEnd) {
1554586df76SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1564586df76SDmitri Gribenko     if (!isHTMLDecimalCharacterReferenceCharacter(*BufferPtr))
1574586df76SDmitri Gribenko       return BufferPtr;
1584586df76SDmitri Gribenko   }
1594586df76SDmitri Gribenko   return BufferEnd;
1604586df76SDmitri Gribenko }
1614586df76SDmitri Gribenko 
1624586df76SDmitri Gribenko const char *skipHexCharacterReference(const char *BufferPtr,
1634586df76SDmitri Gribenko                                           const char *BufferEnd) {
1644586df76SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1654586df76SDmitri Gribenko     if (!isHTMLHexCharacterReferenceCharacter(*BufferPtr))
1664586df76SDmitri Gribenko       return BufferPtr;
1674586df76SDmitri Gribenko   }
1684586df76SDmitri Gribenko   return BufferEnd;
1694586df76SDmitri Gribenko }
1704586df76SDmitri Gribenko 
17117709ae8SDmitri Gribenko bool isHTMLIdentifierStartingCharacter(char C) {
17217709ae8SDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
17317709ae8SDmitri Gribenko          (C >= 'A' && C <= 'Z');
17417709ae8SDmitri Gribenko }
17517709ae8SDmitri Gribenko 
1765188c4b9SDmitri Gribenko bool isHTMLIdentifierCharacter(char C) {
1775188c4b9SDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
1785188c4b9SDmitri Gribenko          (C >= 'A' && C <= 'Z') ||
1795188c4b9SDmitri Gribenko          (C >= '0' && C <= '9');
1805188c4b9SDmitri Gribenko }
1815188c4b9SDmitri Gribenko 
1825188c4b9SDmitri Gribenko const char *skipHTMLIdentifier(const char *BufferPtr, const char *BufferEnd) {
1835188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
1845188c4b9SDmitri Gribenko     if (!isHTMLIdentifierCharacter(*BufferPtr))
1855188c4b9SDmitri Gribenko       return BufferPtr;
1865188c4b9SDmitri Gribenko   }
1875188c4b9SDmitri Gribenko   return BufferEnd;
1885188c4b9SDmitri Gribenko }
1895188c4b9SDmitri Gribenko 
1905188c4b9SDmitri Gribenko /// Skip HTML string quoted in single or double quotes.  Escaping quotes inside
1915188c4b9SDmitri Gribenko /// string allowed.
1925188c4b9SDmitri Gribenko ///
1935188c4b9SDmitri Gribenko /// Returns pointer to closing quote.
1945188c4b9SDmitri Gribenko const char *skipHTMLQuotedString(const char *BufferPtr, const char *BufferEnd)
1955188c4b9SDmitri Gribenko {
1965188c4b9SDmitri Gribenko   const char Quote = *BufferPtr;
1975188c4b9SDmitri Gribenko   assert(Quote == '\"' || Quote == '\'');
1985188c4b9SDmitri Gribenko 
1995188c4b9SDmitri Gribenko   BufferPtr++;
2005188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2015188c4b9SDmitri Gribenko     const char C = *BufferPtr;
2025188c4b9SDmitri Gribenko     if (C == Quote && BufferPtr[-1] != '\\')
2035188c4b9SDmitri Gribenko       return BufferPtr;
2045188c4b9SDmitri Gribenko   }
2055188c4b9SDmitri Gribenko   return BufferEnd;
2065188c4b9SDmitri Gribenko }
2075188c4b9SDmitri Gribenko 
2085188c4b9SDmitri Gribenko bool isHorizontalWhitespace(char C) {
2095188c4b9SDmitri Gribenko   return C == ' ' || C == '\t' || C == '\f' || C == '\v';
2105188c4b9SDmitri Gribenko }
2115188c4b9SDmitri Gribenko 
2125188c4b9SDmitri Gribenko bool isWhitespace(char C) {
2135188c4b9SDmitri Gribenko   return C == ' ' || C == '\n' || C == '\r' ||
2145188c4b9SDmitri Gribenko          C == '\t' || C == '\f' || C == '\v';
2155188c4b9SDmitri Gribenko }
2165188c4b9SDmitri Gribenko 
2175188c4b9SDmitri Gribenko const char *skipWhitespace(const char *BufferPtr, const char *BufferEnd) {
2185188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2195188c4b9SDmitri Gribenko     if (!isWhitespace(*BufferPtr))
2205188c4b9SDmitri Gribenko       return BufferPtr;
2215188c4b9SDmitri Gribenko   }
2225188c4b9SDmitri Gribenko   return BufferEnd;
2235188c4b9SDmitri Gribenko }
2245188c4b9SDmitri Gribenko 
225e4a3997dSDmitri Gribenko bool isWhitespace(const char *BufferPtr, const char *BufferEnd) {
226e4a3997dSDmitri Gribenko   return skipWhitespace(BufferPtr, BufferEnd) == BufferEnd;
227e4a3997dSDmitri Gribenko }
228e4a3997dSDmitri Gribenko 
229*ad45ad6fSDmitri Gribenko bool isCommandNameStartCharacter(char C) {
230*ad45ad6fSDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
231*ad45ad6fSDmitri Gribenko          (C >= 'A' && C <= 'Z');
232*ad45ad6fSDmitri Gribenko }
233*ad45ad6fSDmitri Gribenko 
2345188c4b9SDmitri Gribenko bool isCommandNameCharacter(char C) {
2355188c4b9SDmitri Gribenko   return (C >= 'a' && C <= 'z') ||
2365188c4b9SDmitri Gribenko          (C >= 'A' && C <= 'Z') ||
2375188c4b9SDmitri Gribenko          (C >= '0' && C <= '9');
2385188c4b9SDmitri Gribenko }
2395188c4b9SDmitri Gribenko 
2405188c4b9SDmitri Gribenko const char *skipCommandName(const char *BufferPtr, const char *BufferEnd) {
2415188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2425188c4b9SDmitri Gribenko     if (!isCommandNameCharacter(*BufferPtr))
2435188c4b9SDmitri Gribenko       return BufferPtr;
2445188c4b9SDmitri Gribenko   }
2455188c4b9SDmitri Gribenko   return BufferEnd;
2465188c4b9SDmitri Gribenko }
2475188c4b9SDmitri Gribenko 
2485188c4b9SDmitri Gribenko /// Return the one past end pointer for BCPL comments.
2495188c4b9SDmitri Gribenko /// Handles newlines escaped with backslash or trigraph for backslahs.
2505188c4b9SDmitri Gribenko const char *findBCPLCommentEnd(const char *BufferPtr, const char *BufferEnd) {
2515188c4b9SDmitri Gribenko   const char *CurPtr = BufferPtr;
2525188c4b9SDmitri Gribenko   while (CurPtr != BufferEnd) {
2535188c4b9SDmitri Gribenko     char C = *CurPtr;
2545188c4b9SDmitri Gribenko     while (C != '\n' && C != '\r') {
2555188c4b9SDmitri Gribenko       CurPtr++;
2565188c4b9SDmitri Gribenko       if (CurPtr == BufferEnd)
2575188c4b9SDmitri Gribenko         return BufferEnd;
2585188c4b9SDmitri Gribenko       C = *CurPtr;
2595188c4b9SDmitri Gribenko     }
2605188c4b9SDmitri Gribenko     // We found a newline, check if it is escaped.
2615188c4b9SDmitri Gribenko     const char *EscapePtr = CurPtr - 1;
2625188c4b9SDmitri Gribenko     while(isHorizontalWhitespace(*EscapePtr))
2635188c4b9SDmitri Gribenko       EscapePtr--;
2645188c4b9SDmitri Gribenko 
2655188c4b9SDmitri Gribenko     if (*EscapePtr == '\\' ||
2665188c4b9SDmitri Gribenko         (EscapePtr - 2 >= BufferPtr && EscapePtr[0] == '/' &&
2675188c4b9SDmitri Gribenko          EscapePtr[-1] == '?' && EscapePtr[-2] == '?')) {
2685188c4b9SDmitri Gribenko       // We found an escaped newline.
2695188c4b9SDmitri Gribenko       CurPtr = skipNewline(CurPtr, BufferEnd);
2705188c4b9SDmitri Gribenko     } else
2715188c4b9SDmitri Gribenko       return CurPtr; // Not an escaped newline.
2725188c4b9SDmitri Gribenko   }
2735188c4b9SDmitri Gribenko   return BufferEnd;
2745188c4b9SDmitri Gribenko }
2755188c4b9SDmitri Gribenko 
2765188c4b9SDmitri Gribenko /// Return the one past end pointer for C comments.
2775188c4b9SDmitri Gribenko /// Very dumb, does not handle escaped newlines or trigraphs.
2785188c4b9SDmitri Gribenko const char *findCCommentEnd(const char *BufferPtr, const char *BufferEnd) {
2795188c4b9SDmitri Gribenko   for ( ; BufferPtr != BufferEnd; ++BufferPtr) {
2805188c4b9SDmitri Gribenko     if (*BufferPtr == '*') {
2815188c4b9SDmitri Gribenko       assert(BufferPtr + 1 != BufferEnd);
2825188c4b9SDmitri Gribenko       if (*(BufferPtr + 1) == '/')
2835188c4b9SDmitri Gribenko         return BufferPtr;
2845188c4b9SDmitri Gribenko     }
2855188c4b9SDmitri Gribenko   }
2865188c4b9SDmitri Gribenko   llvm_unreachable("buffer end hit before '*/' was seen");
2875188c4b9SDmitri Gribenko }
2885188c4b9SDmitri Gribenko } // unnamed namespace
2895188c4b9SDmitri Gribenko 
2905188c4b9SDmitri Gribenko void Lexer::lexCommentText(Token &T) {
2915188c4b9SDmitri Gribenko   assert(CommentState == LCS_InsideBCPLComment ||
2925188c4b9SDmitri Gribenko          CommentState == LCS_InsideCComment);
2935188c4b9SDmitri Gribenko 
2945188c4b9SDmitri Gribenko   switch (State) {
2955188c4b9SDmitri Gribenko   case LS_Normal:
2965188c4b9SDmitri Gribenko     break;
2975188c4b9SDmitri Gribenko   case LS_VerbatimBlockFirstLine:
2985188c4b9SDmitri Gribenko     lexVerbatimBlockFirstLine(T);
2995188c4b9SDmitri Gribenko     return;
3005188c4b9SDmitri Gribenko   case LS_VerbatimBlockBody:
3015188c4b9SDmitri Gribenko     lexVerbatimBlockBody(T);
3025188c4b9SDmitri Gribenko     return;
3031669f702SDmitri Gribenko   case LS_VerbatimLineText:
3041669f702SDmitri Gribenko     lexVerbatimLineText(T);
3051669f702SDmitri Gribenko     return;
306e00ffc7bSDmitri Gribenko   case LS_HTMLStartTag:
307e00ffc7bSDmitri Gribenko     lexHTMLStartTag(T);
3085188c4b9SDmitri Gribenko     return;
309e00ffc7bSDmitri Gribenko   case LS_HTMLEndTag:
310e00ffc7bSDmitri Gribenko     lexHTMLEndTag(T);
311ec92531cSDmitri Gribenko     return;
3125188c4b9SDmitri Gribenko   }
3135188c4b9SDmitri Gribenko 
3145188c4b9SDmitri Gribenko   assert(State == LS_Normal);
3155188c4b9SDmitri Gribenko 
3165188c4b9SDmitri Gribenko   const char *TokenPtr = BufferPtr;
3175188c4b9SDmitri Gribenko   assert(TokenPtr < CommentEnd);
3185188c4b9SDmitri Gribenko   while (TokenPtr != CommentEnd) {
3195188c4b9SDmitri Gribenko     switch(*TokenPtr) {
3205188c4b9SDmitri Gribenko       case '\\':
3215188c4b9SDmitri Gribenko       case '@': {
3225188c4b9SDmitri Gribenko         TokenPtr++;
3235188c4b9SDmitri Gribenko         if (TokenPtr == CommentEnd) {
3244586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
3255188c4b9SDmitri Gribenko           return;
3265188c4b9SDmitri Gribenko         }
3275188c4b9SDmitri Gribenko         char C = *TokenPtr;
3285188c4b9SDmitri Gribenko         switch (C) {
3295188c4b9SDmitri Gribenko         default:
3305188c4b9SDmitri Gribenko           break;
3315188c4b9SDmitri Gribenko 
3325188c4b9SDmitri Gribenko         case '\\': case '@': case '&': case '$':
3335188c4b9SDmitri Gribenko         case '#':  case '<': case '>': case '%':
3345188c4b9SDmitri Gribenko         case '\"': case '.': case ':':
3355188c4b9SDmitri Gribenko           // This is one of \\ \@ \& \$ etc escape sequences.
3365188c4b9SDmitri Gribenko           TokenPtr++;
3375188c4b9SDmitri Gribenko           if (C == ':' && TokenPtr != CommentEnd && *TokenPtr == ':') {
3385188c4b9SDmitri Gribenko             // This is the \:: escape sequence.
3395188c4b9SDmitri Gribenko             TokenPtr++;
3405188c4b9SDmitri Gribenko           }
34160ddd8a1SDmitri Gribenko           StringRef UnescapedText(BufferPtr + 1, TokenPtr - (BufferPtr + 1));
3425188c4b9SDmitri Gribenko           formTokenWithChars(T, TokenPtr, tok::text);
34360ddd8a1SDmitri Gribenko           T.setText(UnescapedText);
3445188c4b9SDmitri Gribenko           return;
3455188c4b9SDmitri Gribenko         }
3465188c4b9SDmitri Gribenko 
3475188c4b9SDmitri Gribenko         // Don't make zero-length commands.
348*ad45ad6fSDmitri Gribenko         if (!isCommandNameStartCharacter(*TokenPtr)) {
3494586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
3505188c4b9SDmitri Gribenko           return;
3515188c4b9SDmitri Gribenko         }
3525188c4b9SDmitri Gribenko 
3535188c4b9SDmitri Gribenko         TokenPtr = skipCommandName(TokenPtr, CommentEnd);
3545188c4b9SDmitri Gribenko         unsigned Length = TokenPtr - (BufferPtr + 1);
3555188c4b9SDmitri Gribenko 
3565188c4b9SDmitri Gribenko         // Hardcoded support for lexing LaTeX formula commands
3575188c4b9SDmitri Gribenko         // \f$ \f[ \f] \f{ \f} as a single command.
3585188c4b9SDmitri Gribenko         if (Length == 1 && TokenPtr[-1] == 'f' && TokenPtr != CommentEnd) {
3595188c4b9SDmitri Gribenko           C = *TokenPtr;
3605188c4b9SDmitri Gribenko           if (C == '$' || C == '[' || C == ']' || C == '{' || C == '}') {
3615188c4b9SDmitri Gribenko             TokenPtr++;
3625188c4b9SDmitri Gribenko             Length++;
3635188c4b9SDmitri Gribenko           }
3645188c4b9SDmitri Gribenko         }
3655188c4b9SDmitri Gribenko 
3665188c4b9SDmitri Gribenko         const StringRef CommandName(BufferPtr + 1, Length);
3675188c4b9SDmitri Gribenko 
3687acbf00fSDmitri Gribenko         const CommandInfo *Info = Traits.getCommandInfoOrNULL(CommandName);
3697acbf00fSDmitri Gribenko         if (!Info) {
3707acbf00fSDmitri Gribenko           formTokenWithChars(T, TokenPtr, tok::unknown_command);
3717acbf00fSDmitri Gribenko           T.setUnknownCommandName(CommandName);
3725188c4b9SDmitri Gribenko           return;
3735188c4b9SDmitri Gribenko         }
3747acbf00fSDmitri Gribenko         if (Info->IsVerbatimBlockCommand) {
3757acbf00fSDmitri Gribenko           setupAndLexVerbatimBlock(T, TokenPtr, *BufferPtr, Info);
3767acbf00fSDmitri Gribenko           return;
3777acbf00fSDmitri Gribenko         }
3787acbf00fSDmitri Gribenko         if (Info->IsVerbatimLineCommand) {
3797acbf00fSDmitri Gribenko           setupAndLexVerbatimLine(T, TokenPtr, Info);
3805188c4b9SDmitri Gribenko           return;
3815188c4b9SDmitri Gribenko         }
3825188c4b9SDmitri Gribenko         formTokenWithChars(T, TokenPtr, tok::command);
3837acbf00fSDmitri Gribenko         T.setCommandID(Info->getID());
3845188c4b9SDmitri Gribenko         return;
3855188c4b9SDmitri Gribenko       }
3865188c4b9SDmitri Gribenko 
3874586df76SDmitri Gribenko       case '&':
3884586df76SDmitri Gribenko         lexHTMLCharacterReference(T);
3894586df76SDmitri Gribenko         return;
3904586df76SDmitri Gribenko 
3915188c4b9SDmitri Gribenko       case '<': {
3925188c4b9SDmitri Gribenko         TokenPtr++;
3935188c4b9SDmitri Gribenko         if (TokenPtr == CommentEnd) {
3944586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
3955188c4b9SDmitri Gribenko           return;
3965188c4b9SDmitri Gribenko         }
3975188c4b9SDmitri Gribenko         const char C = *TokenPtr;
39817709ae8SDmitri Gribenko         if (isHTMLIdentifierStartingCharacter(C))
399e00ffc7bSDmitri Gribenko           setupAndLexHTMLStartTag(T);
4005188c4b9SDmitri Gribenko         else if (C == '/')
401e00ffc7bSDmitri Gribenko           setupAndLexHTMLEndTag(T);
4024586df76SDmitri Gribenko         else
4034586df76SDmitri Gribenko           formTextToken(T, TokenPtr);
4044586df76SDmitri Gribenko 
4055188c4b9SDmitri Gribenko         return;
4065188c4b9SDmitri Gribenko       }
4075188c4b9SDmitri Gribenko 
4085188c4b9SDmitri Gribenko       case '\n':
4095188c4b9SDmitri Gribenko       case '\r':
4105188c4b9SDmitri Gribenko         TokenPtr = skipNewline(TokenPtr, CommentEnd);
4115188c4b9SDmitri Gribenko         formTokenWithChars(T, TokenPtr, tok::newline);
4125188c4b9SDmitri Gribenko 
4135188c4b9SDmitri Gribenko         if (CommentState == LCS_InsideCComment)
4145188c4b9SDmitri Gribenko           skipLineStartingDecorations();
4155188c4b9SDmitri Gribenko         return;
4165188c4b9SDmitri Gribenko 
4175188c4b9SDmitri Gribenko       default: {
4185188c4b9SDmitri Gribenko         while (true) {
4195188c4b9SDmitri Gribenko           TokenPtr++;
4205188c4b9SDmitri Gribenko           if (TokenPtr == CommentEnd)
4215188c4b9SDmitri Gribenko             break;
42217709ae8SDmitri Gribenko           const char C = *TokenPtr;
4235188c4b9SDmitri Gribenko           if(C == '\n' || C == '\r' ||
4244586df76SDmitri Gribenko              C == '\\' || C == '@' || C == '&' || C == '<')
4255188c4b9SDmitri Gribenko             break;
4265188c4b9SDmitri Gribenko         }
4274586df76SDmitri Gribenko         formTextToken(T, TokenPtr);
4285188c4b9SDmitri Gribenko         return;
4295188c4b9SDmitri Gribenko       }
4305188c4b9SDmitri Gribenko     }
4315188c4b9SDmitri Gribenko   }
4325188c4b9SDmitri Gribenko }
4335188c4b9SDmitri Gribenko 
4345188c4b9SDmitri Gribenko void Lexer::setupAndLexVerbatimBlock(Token &T,
4355188c4b9SDmitri Gribenko                                      const char *TextBegin,
4367acbf00fSDmitri Gribenko                                      char Marker, const CommandInfo *Info) {
4377acbf00fSDmitri Gribenko   assert(Info->IsVerbatimBlockCommand);
4387acbf00fSDmitri Gribenko 
4395188c4b9SDmitri Gribenko   VerbatimBlockEndCommandName.clear();
4405188c4b9SDmitri Gribenko   VerbatimBlockEndCommandName.append(Marker == '\\' ? "\\" : "@");
4417acbf00fSDmitri Gribenko   VerbatimBlockEndCommandName.append(Info->EndCommandName);
4425188c4b9SDmitri Gribenko 
4435188c4b9SDmitri Gribenko   formTokenWithChars(T, TextBegin, tok::verbatim_block_begin);
4447acbf00fSDmitri Gribenko   T.setVerbatimBlockID(Info->getID());
4455188c4b9SDmitri Gribenko 
446ec92531cSDmitri Gribenko   // If there is a newline following the verbatim opening command, skip the
447ec92531cSDmitri Gribenko   // newline so that we don't create an tok::verbatim_block_line with empty
448ec92531cSDmitri Gribenko   // text content.
449ec92531cSDmitri Gribenko   if (BufferPtr != CommentEnd) {
450ec92531cSDmitri Gribenko     const char C = *BufferPtr;
451ec92531cSDmitri Gribenko     if (C == '\n' || C == '\r') {
452ec92531cSDmitri Gribenko       BufferPtr = skipNewline(BufferPtr, CommentEnd);
453ec92531cSDmitri Gribenko       State = LS_VerbatimBlockBody;
454ec92531cSDmitri Gribenko       return;
455ec92531cSDmitri Gribenko     }
456ec92531cSDmitri Gribenko   }
457ec92531cSDmitri Gribenko 
4585188c4b9SDmitri Gribenko   State = LS_VerbatimBlockFirstLine;
4595188c4b9SDmitri Gribenko }
4605188c4b9SDmitri Gribenko 
4615188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockFirstLine(Token &T) {
462e4a3997dSDmitri Gribenko again:
4635188c4b9SDmitri Gribenko   assert(BufferPtr < CommentEnd);
4645188c4b9SDmitri Gribenko 
4655188c4b9SDmitri Gribenko   // FIXME: It would be better to scan the text once, finding either the block
4665188c4b9SDmitri Gribenko   // end command or newline.
4675188c4b9SDmitri Gribenko   //
4685188c4b9SDmitri Gribenko   // Extract current line.
4695188c4b9SDmitri Gribenko   const char *Newline = findNewline(BufferPtr, CommentEnd);
4705188c4b9SDmitri Gribenko   StringRef Line(BufferPtr, Newline - BufferPtr);
4715188c4b9SDmitri Gribenko 
4725188c4b9SDmitri Gribenko   // Look for end command in current line.
4735188c4b9SDmitri Gribenko   size_t Pos = Line.find(VerbatimBlockEndCommandName);
474ec92531cSDmitri Gribenko   const char *TextEnd;
4755188c4b9SDmitri Gribenko   const char *NextLine;
4765188c4b9SDmitri Gribenko   if (Pos == StringRef::npos) {
4775188c4b9SDmitri Gribenko     // Current line is completely verbatim.
478ec92531cSDmitri Gribenko     TextEnd = Newline;
4795188c4b9SDmitri Gribenko     NextLine = skipNewline(Newline, CommentEnd);
4805188c4b9SDmitri Gribenko   } else if (Pos == 0) {
4815188c4b9SDmitri Gribenko     // Current line contains just an end command.
4825188c4b9SDmitri Gribenko     const char *End = BufferPtr + VerbatimBlockEndCommandName.size();
48360ddd8a1SDmitri Gribenko     StringRef Name(BufferPtr + 1, End - (BufferPtr + 1));
4845188c4b9SDmitri Gribenko     formTokenWithChars(T, End, tok::verbatim_block_end);
4857acbf00fSDmitri Gribenko     T.setVerbatimBlockID(Traits.getCommandInfo(Name)->getID());
4865188c4b9SDmitri Gribenko     State = LS_Normal;
4875188c4b9SDmitri Gribenko     return;
4885188c4b9SDmitri Gribenko   } else {
4895188c4b9SDmitri Gribenko     // There is some text, followed by end command.  Extract text first.
490ec92531cSDmitri Gribenko     TextEnd = BufferPtr + Pos;
491ec92531cSDmitri Gribenko     NextLine = TextEnd;
492e4a3997dSDmitri Gribenko     // If there is only whitespace before end command, skip whitespace.
493e4a3997dSDmitri Gribenko     if (isWhitespace(BufferPtr, TextEnd)) {
494e4a3997dSDmitri Gribenko       BufferPtr = TextEnd;
495e4a3997dSDmitri Gribenko       goto again;
496e4a3997dSDmitri Gribenko     }
4975188c4b9SDmitri Gribenko   }
4985188c4b9SDmitri Gribenko 
499ec92531cSDmitri Gribenko   StringRef Text(BufferPtr, TextEnd - BufferPtr);
5005188c4b9SDmitri Gribenko   formTokenWithChars(T, NextLine, tok::verbatim_block_line);
50160ddd8a1SDmitri Gribenko   T.setVerbatimBlockText(Text);
5025188c4b9SDmitri Gribenko 
5035188c4b9SDmitri Gribenko   State = LS_VerbatimBlockBody;
5045188c4b9SDmitri Gribenko }
5055188c4b9SDmitri Gribenko 
5065188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockBody(Token &T) {
5075188c4b9SDmitri Gribenko   assert(State == LS_VerbatimBlockBody);
5085188c4b9SDmitri Gribenko 
5095188c4b9SDmitri Gribenko   if (CommentState == LCS_InsideCComment)
5105188c4b9SDmitri Gribenko     skipLineStartingDecorations();
5115188c4b9SDmitri Gribenko 
5125188c4b9SDmitri Gribenko   lexVerbatimBlockFirstLine(T);
5135188c4b9SDmitri Gribenko }
5145188c4b9SDmitri Gribenko 
5157acbf00fSDmitri Gribenko void Lexer::setupAndLexVerbatimLine(Token &T, const char *TextBegin,
5167acbf00fSDmitri Gribenko                                     const CommandInfo *Info) {
5177acbf00fSDmitri Gribenko   assert(Info->IsVerbatimLineCommand);
5181669f702SDmitri Gribenko   formTokenWithChars(T, TextBegin, tok::verbatim_line_name);
5197acbf00fSDmitri Gribenko   T.setVerbatimLineID(Info->getID());
5201669f702SDmitri Gribenko 
5211669f702SDmitri Gribenko   State = LS_VerbatimLineText;
5221669f702SDmitri Gribenko }
5231669f702SDmitri Gribenko 
5241669f702SDmitri Gribenko void Lexer::lexVerbatimLineText(Token &T) {
5251669f702SDmitri Gribenko   assert(State == LS_VerbatimLineText);
5261669f702SDmitri Gribenko 
5275188c4b9SDmitri Gribenko   // Extract current line.
5285188c4b9SDmitri Gribenko   const char *Newline = findNewline(BufferPtr, CommentEnd);
5291669f702SDmitri Gribenko   const StringRef Text(BufferPtr, Newline - BufferPtr);
5301669f702SDmitri Gribenko   formTokenWithChars(T, Newline, tok::verbatim_line_text);
5315188c4b9SDmitri Gribenko   T.setVerbatimLineText(Text);
5321669f702SDmitri Gribenko 
5331669f702SDmitri Gribenko   State = LS_Normal;
5345188c4b9SDmitri Gribenko }
5355188c4b9SDmitri Gribenko 
5364586df76SDmitri Gribenko void Lexer::lexHTMLCharacterReference(Token &T) {
5374586df76SDmitri Gribenko   const char *TokenPtr = BufferPtr;
5384586df76SDmitri Gribenko   assert(*TokenPtr == '&');
5394586df76SDmitri Gribenko   TokenPtr++;
5404586df76SDmitri Gribenko   if (TokenPtr == CommentEnd) {
5414586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5424586df76SDmitri Gribenko     return;
5434586df76SDmitri Gribenko   }
5444586df76SDmitri Gribenko   const char *NamePtr;
5454586df76SDmitri Gribenko   bool isNamed = false;
5464586df76SDmitri Gribenko   bool isDecimal = false;
5474586df76SDmitri Gribenko   char C = *TokenPtr;
5484586df76SDmitri Gribenko   if (isHTMLNamedCharacterReferenceCharacter(C)) {
5494586df76SDmitri Gribenko     NamePtr = TokenPtr;
5504586df76SDmitri Gribenko     TokenPtr = skipNamedCharacterReference(TokenPtr, CommentEnd);
5514586df76SDmitri Gribenko     isNamed = true;
5524586df76SDmitri Gribenko   } else if (C == '#') {
5534586df76SDmitri Gribenko     TokenPtr++;
5544586df76SDmitri Gribenko     if (TokenPtr == CommentEnd) {
5554586df76SDmitri Gribenko       formTextToken(T, TokenPtr);
5564586df76SDmitri Gribenko       return;
5574586df76SDmitri Gribenko     }
5584586df76SDmitri Gribenko     C = *TokenPtr;
5594586df76SDmitri Gribenko     if (isHTMLDecimalCharacterReferenceCharacter(C)) {
5604586df76SDmitri Gribenko       NamePtr = TokenPtr;
5614586df76SDmitri Gribenko       TokenPtr = skipDecimalCharacterReference(TokenPtr, CommentEnd);
5624586df76SDmitri Gribenko       isDecimal = true;
5634586df76SDmitri Gribenko     } else if (C == 'x' || C == 'X') {
5644586df76SDmitri Gribenko       TokenPtr++;
5654586df76SDmitri Gribenko       NamePtr = TokenPtr;
5664586df76SDmitri Gribenko       TokenPtr = skipHexCharacterReference(TokenPtr, CommentEnd);
5674586df76SDmitri Gribenko     } else {
5684586df76SDmitri Gribenko       formTextToken(T, TokenPtr);
5694586df76SDmitri Gribenko       return;
5704586df76SDmitri Gribenko     }
5714586df76SDmitri Gribenko   } else {
5724586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5734586df76SDmitri Gribenko     return;
5744586df76SDmitri Gribenko   }
5754586df76SDmitri Gribenko   if (NamePtr == TokenPtr || TokenPtr == CommentEnd ||
5764586df76SDmitri Gribenko       *TokenPtr != ';') {
5774586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5784586df76SDmitri Gribenko     return;
5794586df76SDmitri Gribenko   }
5804586df76SDmitri Gribenko   StringRef Name(NamePtr, TokenPtr - NamePtr);
5814586df76SDmitri Gribenko   TokenPtr++; // Skip semicolon.
5824586df76SDmitri Gribenko   StringRef Resolved;
5834586df76SDmitri Gribenko   if (isNamed)
5844586df76SDmitri Gribenko     Resolved = resolveHTMLNamedCharacterReference(Name);
5854586df76SDmitri Gribenko   else if (isDecimal)
5864586df76SDmitri Gribenko     Resolved = resolveHTMLDecimalCharacterReference(Name);
5874586df76SDmitri Gribenko   else
5884586df76SDmitri Gribenko     Resolved = resolveHTMLHexCharacterReference(Name);
5894586df76SDmitri Gribenko 
5904586df76SDmitri Gribenko   if (Resolved.empty()) {
5914586df76SDmitri Gribenko     formTextToken(T, TokenPtr);
5924586df76SDmitri Gribenko     return;
5934586df76SDmitri Gribenko   }
5944586df76SDmitri Gribenko   formTokenWithChars(T, TokenPtr, tok::text);
5954586df76SDmitri Gribenko   T.setText(Resolved);
5964586df76SDmitri Gribenko   return;
5974586df76SDmitri Gribenko }
5984586df76SDmitri Gribenko 
599e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLStartTag(Token &T) {
60017709ae8SDmitri Gribenko   assert(BufferPtr[0] == '<' &&
60117709ae8SDmitri Gribenko          isHTMLIdentifierStartingCharacter(BufferPtr[1]));
6025188c4b9SDmitri Gribenko   const char *TagNameEnd = skipHTMLIdentifier(BufferPtr + 2, CommentEnd);
60360ddd8a1SDmitri Gribenko   StringRef Name(BufferPtr + 1, TagNameEnd - (BufferPtr + 1));
604107618a6SDmitri Gribenko   if (!isHTMLTagName(Name)) {
605107618a6SDmitri Gribenko     formTextToken(T, TagNameEnd);
606107618a6SDmitri Gribenko     return;
607107618a6SDmitri Gribenko   }
608107618a6SDmitri Gribenko 
609e00ffc7bSDmitri Gribenko   formTokenWithChars(T, TagNameEnd, tok::html_start_tag);
610e00ffc7bSDmitri Gribenko   T.setHTMLTagStartName(Name);
6115188c4b9SDmitri Gribenko 
6125188c4b9SDmitri Gribenko   BufferPtr = skipWhitespace(BufferPtr, CommentEnd);
6135188c4b9SDmitri Gribenko 
61417709ae8SDmitri Gribenko   const char C = *BufferPtr;
61517709ae8SDmitri Gribenko   if (BufferPtr != CommentEnd &&
616f26054f0SDmitri Gribenko       (C == '>' || C == '/' || isHTMLIdentifierStartingCharacter(C)))
617e00ffc7bSDmitri Gribenko     State = LS_HTMLStartTag;
6185188c4b9SDmitri Gribenko }
6195188c4b9SDmitri Gribenko 
620e00ffc7bSDmitri Gribenko void Lexer::lexHTMLStartTag(Token &T) {
621e00ffc7bSDmitri Gribenko   assert(State == LS_HTMLStartTag);
6225188c4b9SDmitri Gribenko 
6235188c4b9SDmitri Gribenko   const char *TokenPtr = BufferPtr;
6245188c4b9SDmitri Gribenko   char C = *TokenPtr;
6255188c4b9SDmitri Gribenko   if (isHTMLIdentifierCharacter(C)) {
6265188c4b9SDmitri Gribenko     TokenPtr = skipHTMLIdentifier(TokenPtr, CommentEnd);
62760ddd8a1SDmitri Gribenko     StringRef Ident(BufferPtr, TokenPtr - BufferPtr);
6285188c4b9SDmitri Gribenko     formTokenWithChars(T, TokenPtr, tok::html_ident);
62960ddd8a1SDmitri Gribenko     T.setHTMLIdent(Ident);
6305188c4b9SDmitri Gribenko   } else {
6315188c4b9SDmitri Gribenko     switch (C) {
6325188c4b9SDmitri Gribenko     case '=':
6335188c4b9SDmitri Gribenko       TokenPtr++;
6345188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_equals);
6355188c4b9SDmitri Gribenko       break;
6365188c4b9SDmitri Gribenko     case '\"':
6375188c4b9SDmitri Gribenko     case '\'': {
6385188c4b9SDmitri Gribenko       const char *OpenQuote = TokenPtr;
6395188c4b9SDmitri Gribenko       TokenPtr = skipHTMLQuotedString(TokenPtr, CommentEnd);
6405188c4b9SDmitri Gribenko       const char *ClosingQuote = TokenPtr;
6415188c4b9SDmitri Gribenko       if (TokenPtr != CommentEnd) // Skip closing quote.
6425188c4b9SDmitri Gribenko         TokenPtr++;
6435188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_quoted_string);
6445188c4b9SDmitri Gribenko       T.setHTMLQuotedString(StringRef(OpenQuote + 1,
6455188c4b9SDmitri Gribenko                                       ClosingQuote - (OpenQuote + 1)));
6465188c4b9SDmitri Gribenko       break;
6475188c4b9SDmitri Gribenko     }
6485188c4b9SDmitri Gribenko     case '>':
6495188c4b9SDmitri Gribenko       TokenPtr++;
6505188c4b9SDmitri Gribenko       formTokenWithChars(T, TokenPtr, tok::html_greater);
65117709ae8SDmitri Gribenko       State = LS_Normal;
65217709ae8SDmitri Gribenko       return;
653f26054f0SDmitri Gribenko     case '/':
654f26054f0SDmitri Gribenko       TokenPtr++;
655f26054f0SDmitri Gribenko       if (TokenPtr != CommentEnd && *TokenPtr == '>') {
656f26054f0SDmitri Gribenko         TokenPtr++;
657f26054f0SDmitri Gribenko         formTokenWithChars(T, TokenPtr, tok::html_slash_greater);
6584586df76SDmitri Gribenko       } else
6594586df76SDmitri Gribenko         formTextToken(T, TokenPtr);
6604586df76SDmitri Gribenko 
661f26054f0SDmitri Gribenko       State = LS_Normal;
662f26054f0SDmitri Gribenko       return;
6635188c4b9SDmitri Gribenko     }
6645188c4b9SDmitri Gribenko   }
6655188c4b9SDmitri Gribenko 
6665188c4b9SDmitri Gribenko   // Now look ahead and return to normal state if we don't see any HTML tokens
6675188c4b9SDmitri Gribenko   // ahead.
6685188c4b9SDmitri Gribenko   BufferPtr = skipWhitespace(BufferPtr, CommentEnd);
6695188c4b9SDmitri Gribenko   if (BufferPtr == CommentEnd) {
6705188c4b9SDmitri Gribenko     State = LS_Normal;
6715188c4b9SDmitri Gribenko     return;
6725188c4b9SDmitri Gribenko   }
6735188c4b9SDmitri Gribenko 
6745188c4b9SDmitri Gribenko   C = *BufferPtr;
67517709ae8SDmitri Gribenko   if (!isHTMLIdentifierStartingCharacter(C) &&
6765188c4b9SDmitri Gribenko       C != '=' && C != '\"' && C != '\'' && C != '>') {
6775188c4b9SDmitri Gribenko     State = LS_Normal;
6785188c4b9SDmitri Gribenko     return;
6795188c4b9SDmitri Gribenko   }
6805188c4b9SDmitri Gribenko }
6815188c4b9SDmitri Gribenko 
682e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLEndTag(Token &T) {
6835188c4b9SDmitri Gribenko   assert(BufferPtr[0] == '<' && BufferPtr[1] == '/');
6845188c4b9SDmitri Gribenko 
6855188c4b9SDmitri Gribenko   const char *TagNameBegin = skipWhitespace(BufferPtr + 2, CommentEnd);
6865188c4b9SDmitri Gribenko   const char *TagNameEnd = skipHTMLIdentifier(TagNameBegin, CommentEnd);
687107618a6SDmitri Gribenko   StringRef Name(TagNameBegin, TagNameEnd - TagNameBegin);
688107618a6SDmitri Gribenko   if (!isHTMLTagName(Name)) {
689107618a6SDmitri Gribenko     formTextToken(T, TagNameEnd);
690107618a6SDmitri Gribenko     return;
691107618a6SDmitri Gribenko   }
6925188c4b9SDmitri Gribenko 
6935188c4b9SDmitri Gribenko   const char *End = skipWhitespace(TagNameEnd, CommentEnd);
6945188c4b9SDmitri Gribenko 
695e00ffc7bSDmitri Gribenko   formTokenWithChars(T, End, tok::html_end_tag);
696107618a6SDmitri Gribenko   T.setHTMLTagEndName(Name);
697ec92531cSDmitri Gribenko 
698ec92531cSDmitri Gribenko   if (BufferPtr != CommentEnd && *BufferPtr == '>')
699e00ffc7bSDmitri Gribenko     State = LS_HTMLEndTag;
700ec92531cSDmitri Gribenko }
701ec92531cSDmitri Gribenko 
702e00ffc7bSDmitri Gribenko void Lexer::lexHTMLEndTag(Token &T) {
703ec92531cSDmitri Gribenko   assert(BufferPtr != CommentEnd && *BufferPtr == '>');
704ec92531cSDmitri Gribenko 
705ec92531cSDmitri Gribenko   formTokenWithChars(T, BufferPtr + 1, tok::html_greater);
706ec92531cSDmitri Gribenko   State = LS_Normal;
7075188c4b9SDmitri Gribenko }
7085188c4b9SDmitri Gribenko 
709ca7f80adSDmitri Gribenko Lexer::Lexer(llvm::BumpPtrAllocator &Allocator, const CommandTraits &Traits,
7106bab9113SDmitri Gribenko              SourceLocation FileLoc,
7115188c4b9SDmitri Gribenko              const char *BufferStart, const char *BufferEnd):
712ca7f80adSDmitri Gribenko     Allocator(Allocator), Traits(Traits),
7135188c4b9SDmitri Gribenko     BufferStart(BufferStart), BufferEnd(BufferEnd),
7146bab9113SDmitri Gribenko     FileLoc(FileLoc), BufferPtr(BufferStart),
7155188c4b9SDmitri Gribenko     CommentState(LCS_BeforeComment), State(LS_Normal) {
7165188c4b9SDmitri Gribenko }
7175188c4b9SDmitri Gribenko 
7185188c4b9SDmitri Gribenko void Lexer::lex(Token &T) {
7195188c4b9SDmitri Gribenko again:
7205188c4b9SDmitri Gribenko   switch (CommentState) {
7215188c4b9SDmitri Gribenko   case LCS_BeforeComment:
7225188c4b9SDmitri Gribenko     if (BufferPtr == BufferEnd) {
7235188c4b9SDmitri Gribenko       formTokenWithChars(T, BufferPtr, tok::eof);
7245188c4b9SDmitri Gribenko       return;
7255188c4b9SDmitri Gribenko     }
7265188c4b9SDmitri Gribenko 
7275188c4b9SDmitri Gribenko     assert(*BufferPtr == '/');
7285188c4b9SDmitri Gribenko     BufferPtr++; // Skip first slash.
7295188c4b9SDmitri Gribenko     switch(*BufferPtr) {
7305188c4b9SDmitri Gribenko     case '/': { // BCPL comment.
7315188c4b9SDmitri Gribenko       BufferPtr++; // Skip second slash.
7325188c4b9SDmitri Gribenko 
7335188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd) {
7345188c4b9SDmitri Gribenko         // Skip Doxygen magic marker, if it is present.
7355188c4b9SDmitri Gribenko         // It might be missing because of a typo //< or /*<, or because we
7365188c4b9SDmitri Gribenko         // merged this non-Doxygen comment into a bunch of Doxygen comments
7375188c4b9SDmitri Gribenko         // around it: /** ... */ /* ... */ /** ... */
7385188c4b9SDmitri Gribenko         const char C = *BufferPtr;
7395188c4b9SDmitri Gribenko         if (C == '/' || C == '!')
7405188c4b9SDmitri Gribenko           BufferPtr++;
7415188c4b9SDmitri Gribenko       }
7425188c4b9SDmitri Gribenko 
7435188c4b9SDmitri Gribenko       // Skip less-than symbol that marks trailing comments.
7445188c4b9SDmitri Gribenko       // Skip it even if the comment is not a Doxygen one, because //< and /*<
7455188c4b9SDmitri Gribenko       // are frequent typos.
7465188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd && *BufferPtr == '<')
7475188c4b9SDmitri Gribenko         BufferPtr++;
7485188c4b9SDmitri Gribenko 
7495188c4b9SDmitri Gribenko       CommentState = LCS_InsideBCPLComment;
750ec92531cSDmitri Gribenko       if (State != LS_VerbatimBlockBody && State != LS_VerbatimBlockFirstLine)
7515188c4b9SDmitri Gribenko         State = LS_Normal;
7525188c4b9SDmitri Gribenko       CommentEnd = findBCPLCommentEnd(BufferPtr, BufferEnd);
7535188c4b9SDmitri Gribenko       goto again;
7545188c4b9SDmitri Gribenko     }
7555188c4b9SDmitri Gribenko     case '*': { // C comment.
7565188c4b9SDmitri Gribenko       BufferPtr++; // Skip star.
7575188c4b9SDmitri Gribenko 
7585188c4b9SDmitri Gribenko       // Skip Doxygen magic marker.
7595188c4b9SDmitri Gribenko       const char C = *BufferPtr;
7605188c4b9SDmitri Gribenko       if ((C == '*' && *(BufferPtr + 1) != '/') || C == '!')
7615188c4b9SDmitri Gribenko         BufferPtr++;
7625188c4b9SDmitri Gribenko 
7635188c4b9SDmitri Gribenko       // Skip less-than symbol that marks trailing comments.
7645188c4b9SDmitri Gribenko       if (BufferPtr != BufferEnd && *BufferPtr == '<')
7655188c4b9SDmitri Gribenko         BufferPtr++;
7665188c4b9SDmitri Gribenko 
7675188c4b9SDmitri Gribenko       CommentState = LCS_InsideCComment;
7685188c4b9SDmitri Gribenko       State = LS_Normal;
7695188c4b9SDmitri Gribenko       CommentEnd = findCCommentEnd(BufferPtr, BufferEnd);
7705188c4b9SDmitri Gribenko       goto again;
7715188c4b9SDmitri Gribenko     }
7725188c4b9SDmitri Gribenko     default:
7735188c4b9SDmitri Gribenko       llvm_unreachable("second character of comment should be '/' or '*'");
7745188c4b9SDmitri Gribenko     }
7755188c4b9SDmitri Gribenko 
7765188c4b9SDmitri Gribenko   case LCS_BetweenComments: {
7775188c4b9SDmitri Gribenko     // Consecutive comments are extracted only if there is only whitespace
7785188c4b9SDmitri Gribenko     // between them.  So we can search for the start of the next comment.
7795188c4b9SDmitri Gribenko     const char *EndWhitespace = BufferPtr;
7805188c4b9SDmitri Gribenko     while(EndWhitespace != BufferEnd && *EndWhitespace != '/')
7815188c4b9SDmitri Gribenko       EndWhitespace++;
7825188c4b9SDmitri Gribenko 
7835188c4b9SDmitri Gribenko     // Turn any whitespace between comments (and there is only whitespace
78417709ae8SDmitri Gribenko     // between them -- guaranteed by comment extraction) into a newline.  We
78517709ae8SDmitri Gribenko     // have two newlines between C comments in total (first one was synthesized
78617709ae8SDmitri Gribenko     // after a comment).
7875188c4b9SDmitri Gribenko     formTokenWithChars(T, EndWhitespace, tok::newline);
7885188c4b9SDmitri Gribenko 
7895188c4b9SDmitri Gribenko     CommentState = LCS_BeforeComment;
7905188c4b9SDmitri Gribenko     break;
7915188c4b9SDmitri Gribenko   }
7925188c4b9SDmitri Gribenko 
7935188c4b9SDmitri Gribenko   case LCS_InsideBCPLComment:
7945188c4b9SDmitri Gribenko   case LCS_InsideCComment:
7955188c4b9SDmitri Gribenko     if (BufferPtr != CommentEnd) {
7965188c4b9SDmitri Gribenko       lexCommentText(T);
7975188c4b9SDmitri Gribenko       break;
7985188c4b9SDmitri Gribenko     } else {
7995188c4b9SDmitri Gribenko       // Skip C comment closing sequence.
8005188c4b9SDmitri Gribenko       if (CommentState == LCS_InsideCComment) {
8015188c4b9SDmitri Gribenko         assert(BufferPtr[0] == '*' && BufferPtr[1] == '/');
8025188c4b9SDmitri Gribenko         BufferPtr += 2;
8035188c4b9SDmitri Gribenko         assert(BufferPtr <= BufferEnd);
8045188c4b9SDmitri Gribenko 
8055188c4b9SDmitri Gribenko         // Synthenize newline just after the C comment, regardless if there is
8065188c4b9SDmitri Gribenko         // actually a newline.
8075188c4b9SDmitri Gribenko         formTokenWithChars(T, BufferPtr, tok::newline);
8085188c4b9SDmitri Gribenko 
8095188c4b9SDmitri Gribenko         CommentState = LCS_BetweenComments;
8105188c4b9SDmitri Gribenko         break;
8115188c4b9SDmitri Gribenko       } else {
8125188c4b9SDmitri Gribenko         // Don't synthesized a newline after BCPL comment.
8135188c4b9SDmitri Gribenko         CommentState = LCS_BetweenComments;
8145188c4b9SDmitri Gribenko         goto again;
8155188c4b9SDmitri Gribenko       }
8165188c4b9SDmitri Gribenko     }
8175188c4b9SDmitri Gribenko   }
8185188c4b9SDmitri Gribenko }
8195188c4b9SDmitri Gribenko 
8205188c4b9SDmitri Gribenko StringRef Lexer::getSpelling(const Token &Tok,
8215188c4b9SDmitri Gribenko                              const SourceManager &SourceMgr,
8225188c4b9SDmitri Gribenko                              bool *Invalid) const {
8235188c4b9SDmitri Gribenko   SourceLocation Loc = Tok.getLocation();
8245188c4b9SDmitri Gribenko   std::pair<FileID, unsigned> LocInfo = SourceMgr.getDecomposedLoc(Loc);
8255188c4b9SDmitri Gribenko 
8265188c4b9SDmitri Gribenko   bool InvalidTemp = false;
8275188c4b9SDmitri Gribenko   StringRef File = SourceMgr.getBufferData(LocInfo.first, &InvalidTemp);
8285188c4b9SDmitri Gribenko   if (InvalidTemp) {
8295188c4b9SDmitri Gribenko     *Invalid = true;
8305188c4b9SDmitri Gribenko     return StringRef();
8315188c4b9SDmitri Gribenko   }
8325188c4b9SDmitri Gribenko 
8335188c4b9SDmitri Gribenko   const char *Begin = File.data() + LocInfo.second;
8345188c4b9SDmitri Gribenko   return StringRef(Begin, Tok.getLength());
8355188c4b9SDmitri Gribenko }
8365188c4b9SDmitri Gribenko 
8375188c4b9SDmitri Gribenko } // end namespace comments
8385188c4b9SDmitri Gribenko } // end namespace clang
8395188c4b9SDmitri Gribenko 
840