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