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 229ad45ad6fSDmitri Gribenko bool isCommandNameStartCharacter(char C) { 230ad45ad6fSDmitri Gribenko return (C >= 'a' && C <= 'z') || 231ad45ad6fSDmitri Gribenko (C >= 'A' && C <= 'Z'); 232ad45ad6fSDmitri Gribenko } 233ad45ad6fSDmitri 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. 348ad45ad6fSDmitri 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: { 418*10af67a9SDmitri Gribenko size_t End = StringRef(TokenPtr, CommentEnd - TokenPtr). 419*10af67a9SDmitri Gribenko find_first_of("\n\r\\@&<"); 420*10af67a9SDmitri Gribenko if (End != StringRef::npos) 421*10af67a9SDmitri Gribenko TokenPtr += End; 422*10af67a9SDmitri Gribenko else 423*10af67a9SDmitri Gribenko TokenPtr = CommentEnd; 4244586df76SDmitri Gribenko formTextToken(T, TokenPtr); 4255188c4b9SDmitri Gribenko return; 4265188c4b9SDmitri Gribenko } 4275188c4b9SDmitri Gribenko } 4285188c4b9SDmitri Gribenko } 4295188c4b9SDmitri Gribenko } 4305188c4b9SDmitri Gribenko 4315188c4b9SDmitri Gribenko void Lexer::setupAndLexVerbatimBlock(Token &T, 4325188c4b9SDmitri Gribenko const char *TextBegin, 4337acbf00fSDmitri Gribenko char Marker, const CommandInfo *Info) { 4347acbf00fSDmitri Gribenko assert(Info->IsVerbatimBlockCommand); 4357acbf00fSDmitri Gribenko 4365188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.clear(); 4375188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.append(Marker == '\\' ? "\\" : "@"); 4387acbf00fSDmitri Gribenko VerbatimBlockEndCommandName.append(Info->EndCommandName); 4395188c4b9SDmitri Gribenko 4405188c4b9SDmitri Gribenko formTokenWithChars(T, TextBegin, tok::verbatim_block_begin); 4417acbf00fSDmitri Gribenko T.setVerbatimBlockID(Info->getID()); 4425188c4b9SDmitri Gribenko 443ec92531cSDmitri Gribenko // If there is a newline following the verbatim opening command, skip the 444ec92531cSDmitri Gribenko // newline so that we don't create an tok::verbatim_block_line with empty 445ec92531cSDmitri Gribenko // text content. 446ec92531cSDmitri Gribenko if (BufferPtr != CommentEnd) { 447ec92531cSDmitri Gribenko const char C = *BufferPtr; 448ec92531cSDmitri Gribenko if (C == '\n' || C == '\r') { 449ec92531cSDmitri Gribenko BufferPtr = skipNewline(BufferPtr, CommentEnd); 450ec92531cSDmitri Gribenko State = LS_VerbatimBlockBody; 451ec92531cSDmitri Gribenko return; 452ec92531cSDmitri Gribenko } 453ec92531cSDmitri Gribenko } 454ec92531cSDmitri Gribenko 4555188c4b9SDmitri Gribenko State = LS_VerbatimBlockFirstLine; 4565188c4b9SDmitri Gribenko } 4575188c4b9SDmitri Gribenko 4585188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockFirstLine(Token &T) { 459e4a3997dSDmitri Gribenko again: 4605188c4b9SDmitri Gribenko assert(BufferPtr < CommentEnd); 4615188c4b9SDmitri Gribenko 4625188c4b9SDmitri Gribenko // FIXME: It would be better to scan the text once, finding either the block 4635188c4b9SDmitri Gribenko // end command or newline. 4645188c4b9SDmitri Gribenko // 4655188c4b9SDmitri Gribenko // Extract current line. 4665188c4b9SDmitri Gribenko const char *Newline = findNewline(BufferPtr, CommentEnd); 4675188c4b9SDmitri Gribenko StringRef Line(BufferPtr, Newline - BufferPtr); 4685188c4b9SDmitri Gribenko 4695188c4b9SDmitri Gribenko // Look for end command in current line. 4705188c4b9SDmitri Gribenko size_t Pos = Line.find(VerbatimBlockEndCommandName); 471ec92531cSDmitri Gribenko const char *TextEnd; 4725188c4b9SDmitri Gribenko const char *NextLine; 4735188c4b9SDmitri Gribenko if (Pos == StringRef::npos) { 4745188c4b9SDmitri Gribenko // Current line is completely verbatim. 475ec92531cSDmitri Gribenko TextEnd = Newline; 4765188c4b9SDmitri Gribenko NextLine = skipNewline(Newline, CommentEnd); 4775188c4b9SDmitri Gribenko } else if (Pos == 0) { 4785188c4b9SDmitri Gribenko // Current line contains just an end command. 4795188c4b9SDmitri Gribenko const char *End = BufferPtr + VerbatimBlockEndCommandName.size(); 48060ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, End - (BufferPtr + 1)); 4815188c4b9SDmitri Gribenko formTokenWithChars(T, End, tok::verbatim_block_end); 4827acbf00fSDmitri Gribenko T.setVerbatimBlockID(Traits.getCommandInfo(Name)->getID()); 4835188c4b9SDmitri Gribenko State = LS_Normal; 4845188c4b9SDmitri Gribenko return; 4855188c4b9SDmitri Gribenko } else { 4865188c4b9SDmitri Gribenko // There is some text, followed by end command. Extract text first. 487ec92531cSDmitri Gribenko TextEnd = BufferPtr + Pos; 488ec92531cSDmitri Gribenko NextLine = TextEnd; 489e4a3997dSDmitri Gribenko // If there is only whitespace before end command, skip whitespace. 490e4a3997dSDmitri Gribenko if (isWhitespace(BufferPtr, TextEnd)) { 491e4a3997dSDmitri Gribenko BufferPtr = TextEnd; 492e4a3997dSDmitri Gribenko goto again; 493e4a3997dSDmitri Gribenko } 4945188c4b9SDmitri Gribenko } 4955188c4b9SDmitri Gribenko 496ec92531cSDmitri Gribenko StringRef Text(BufferPtr, TextEnd - BufferPtr); 4975188c4b9SDmitri Gribenko formTokenWithChars(T, NextLine, tok::verbatim_block_line); 49860ddd8a1SDmitri Gribenko T.setVerbatimBlockText(Text); 4995188c4b9SDmitri Gribenko 5005188c4b9SDmitri Gribenko State = LS_VerbatimBlockBody; 5015188c4b9SDmitri Gribenko } 5025188c4b9SDmitri Gribenko 5035188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockBody(Token &T) { 5045188c4b9SDmitri Gribenko assert(State == LS_VerbatimBlockBody); 5055188c4b9SDmitri Gribenko 5065188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) 5075188c4b9SDmitri Gribenko skipLineStartingDecorations(); 5085188c4b9SDmitri Gribenko 5095188c4b9SDmitri Gribenko lexVerbatimBlockFirstLine(T); 5105188c4b9SDmitri Gribenko } 5115188c4b9SDmitri Gribenko 5127acbf00fSDmitri Gribenko void Lexer::setupAndLexVerbatimLine(Token &T, const char *TextBegin, 5137acbf00fSDmitri Gribenko const CommandInfo *Info) { 5147acbf00fSDmitri Gribenko assert(Info->IsVerbatimLineCommand); 5151669f702SDmitri Gribenko formTokenWithChars(T, TextBegin, tok::verbatim_line_name); 5167acbf00fSDmitri Gribenko T.setVerbatimLineID(Info->getID()); 5171669f702SDmitri Gribenko 5181669f702SDmitri Gribenko State = LS_VerbatimLineText; 5191669f702SDmitri Gribenko } 5201669f702SDmitri Gribenko 5211669f702SDmitri Gribenko void Lexer::lexVerbatimLineText(Token &T) { 5221669f702SDmitri Gribenko assert(State == LS_VerbatimLineText); 5231669f702SDmitri Gribenko 5245188c4b9SDmitri Gribenko // Extract current line. 5255188c4b9SDmitri Gribenko const char *Newline = findNewline(BufferPtr, CommentEnd); 5261669f702SDmitri Gribenko const StringRef Text(BufferPtr, Newline - BufferPtr); 5271669f702SDmitri Gribenko formTokenWithChars(T, Newline, tok::verbatim_line_text); 5285188c4b9SDmitri Gribenko T.setVerbatimLineText(Text); 5291669f702SDmitri Gribenko 5301669f702SDmitri Gribenko State = LS_Normal; 5315188c4b9SDmitri Gribenko } 5325188c4b9SDmitri Gribenko 5334586df76SDmitri Gribenko void Lexer::lexHTMLCharacterReference(Token &T) { 5344586df76SDmitri Gribenko const char *TokenPtr = BufferPtr; 5354586df76SDmitri Gribenko assert(*TokenPtr == '&'); 5364586df76SDmitri Gribenko TokenPtr++; 5374586df76SDmitri Gribenko if (TokenPtr == CommentEnd) { 5384586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5394586df76SDmitri Gribenko return; 5404586df76SDmitri Gribenko } 5414586df76SDmitri Gribenko const char *NamePtr; 5424586df76SDmitri Gribenko bool isNamed = false; 5434586df76SDmitri Gribenko bool isDecimal = false; 5444586df76SDmitri Gribenko char C = *TokenPtr; 5454586df76SDmitri Gribenko if (isHTMLNamedCharacterReferenceCharacter(C)) { 5464586df76SDmitri Gribenko NamePtr = TokenPtr; 5474586df76SDmitri Gribenko TokenPtr = skipNamedCharacterReference(TokenPtr, CommentEnd); 5484586df76SDmitri Gribenko isNamed = true; 5494586df76SDmitri Gribenko } else if (C == '#') { 5504586df76SDmitri Gribenko TokenPtr++; 5514586df76SDmitri Gribenko if (TokenPtr == CommentEnd) { 5524586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5534586df76SDmitri Gribenko return; 5544586df76SDmitri Gribenko } 5554586df76SDmitri Gribenko C = *TokenPtr; 5564586df76SDmitri Gribenko if (isHTMLDecimalCharacterReferenceCharacter(C)) { 5574586df76SDmitri Gribenko NamePtr = TokenPtr; 5584586df76SDmitri Gribenko TokenPtr = skipDecimalCharacterReference(TokenPtr, CommentEnd); 5594586df76SDmitri Gribenko isDecimal = true; 5604586df76SDmitri Gribenko } else if (C == 'x' || C == 'X') { 5614586df76SDmitri Gribenko TokenPtr++; 5624586df76SDmitri Gribenko NamePtr = TokenPtr; 5634586df76SDmitri Gribenko TokenPtr = skipHexCharacterReference(TokenPtr, CommentEnd); 5644586df76SDmitri Gribenko } else { 5654586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5664586df76SDmitri Gribenko return; 5674586df76SDmitri Gribenko } 5684586df76SDmitri Gribenko } else { 5694586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5704586df76SDmitri Gribenko return; 5714586df76SDmitri Gribenko } 5724586df76SDmitri Gribenko if (NamePtr == TokenPtr || TokenPtr == CommentEnd || 5734586df76SDmitri Gribenko *TokenPtr != ';') { 5744586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5754586df76SDmitri Gribenko return; 5764586df76SDmitri Gribenko } 5774586df76SDmitri Gribenko StringRef Name(NamePtr, TokenPtr - NamePtr); 5784586df76SDmitri Gribenko TokenPtr++; // Skip semicolon. 5794586df76SDmitri Gribenko StringRef Resolved; 5804586df76SDmitri Gribenko if (isNamed) 5814586df76SDmitri Gribenko Resolved = resolveHTMLNamedCharacterReference(Name); 5824586df76SDmitri Gribenko else if (isDecimal) 5834586df76SDmitri Gribenko Resolved = resolveHTMLDecimalCharacterReference(Name); 5844586df76SDmitri Gribenko else 5854586df76SDmitri Gribenko Resolved = resolveHTMLHexCharacterReference(Name); 5864586df76SDmitri Gribenko 5874586df76SDmitri Gribenko if (Resolved.empty()) { 5884586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5894586df76SDmitri Gribenko return; 5904586df76SDmitri Gribenko } 5914586df76SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::text); 5924586df76SDmitri Gribenko T.setText(Resolved); 5934586df76SDmitri Gribenko return; 5944586df76SDmitri Gribenko } 5954586df76SDmitri Gribenko 596e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLStartTag(Token &T) { 59717709ae8SDmitri Gribenko assert(BufferPtr[0] == '<' && 59817709ae8SDmitri Gribenko isHTMLIdentifierStartingCharacter(BufferPtr[1])); 5995188c4b9SDmitri Gribenko const char *TagNameEnd = skipHTMLIdentifier(BufferPtr + 2, CommentEnd); 60060ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, TagNameEnd - (BufferPtr + 1)); 601107618a6SDmitri Gribenko if (!isHTMLTagName(Name)) { 602107618a6SDmitri Gribenko formTextToken(T, TagNameEnd); 603107618a6SDmitri Gribenko return; 604107618a6SDmitri Gribenko } 605107618a6SDmitri Gribenko 606e00ffc7bSDmitri Gribenko formTokenWithChars(T, TagNameEnd, tok::html_start_tag); 607e00ffc7bSDmitri Gribenko T.setHTMLTagStartName(Name); 6085188c4b9SDmitri Gribenko 6095188c4b9SDmitri Gribenko BufferPtr = skipWhitespace(BufferPtr, CommentEnd); 6105188c4b9SDmitri Gribenko 61117709ae8SDmitri Gribenko const char C = *BufferPtr; 61217709ae8SDmitri Gribenko if (BufferPtr != CommentEnd && 613f26054f0SDmitri Gribenko (C == '>' || C == '/' || isHTMLIdentifierStartingCharacter(C))) 614e00ffc7bSDmitri Gribenko State = LS_HTMLStartTag; 6155188c4b9SDmitri Gribenko } 6165188c4b9SDmitri Gribenko 617e00ffc7bSDmitri Gribenko void Lexer::lexHTMLStartTag(Token &T) { 618e00ffc7bSDmitri Gribenko assert(State == LS_HTMLStartTag); 6195188c4b9SDmitri Gribenko 6205188c4b9SDmitri Gribenko const char *TokenPtr = BufferPtr; 6215188c4b9SDmitri Gribenko char C = *TokenPtr; 6225188c4b9SDmitri Gribenko if (isHTMLIdentifierCharacter(C)) { 6235188c4b9SDmitri Gribenko TokenPtr = skipHTMLIdentifier(TokenPtr, CommentEnd); 62460ddd8a1SDmitri Gribenko StringRef Ident(BufferPtr, TokenPtr - BufferPtr); 6255188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_ident); 62660ddd8a1SDmitri Gribenko T.setHTMLIdent(Ident); 6275188c4b9SDmitri Gribenko } else { 6285188c4b9SDmitri Gribenko switch (C) { 6295188c4b9SDmitri Gribenko case '=': 6305188c4b9SDmitri Gribenko TokenPtr++; 6315188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_equals); 6325188c4b9SDmitri Gribenko break; 6335188c4b9SDmitri Gribenko case '\"': 6345188c4b9SDmitri Gribenko case '\'': { 6355188c4b9SDmitri Gribenko const char *OpenQuote = TokenPtr; 6365188c4b9SDmitri Gribenko TokenPtr = skipHTMLQuotedString(TokenPtr, CommentEnd); 6375188c4b9SDmitri Gribenko const char *ClosingQuote = TokenPtr; 6385188c4b9SDmitri Gribenko if (TokenPtr != CommentEnd) // Skip closing quote. 6395188c4b9SDmitri Gribenko TokenPtr++; 6405188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_quoted_string); 6415188c4b9SDmitri Gribenko T.setHTMLQuotedString(StringRef(OpenQuote + 1, 6425188c4b9SDmitri Gribenko ClosingQuote - (OpenQuote + 1))); 6435188c4b9SDmitri Gribenko break; 6445188c4b9SDmitri Gribenko } 6455188c4b9SDmitri Gribenko case '>': 6465188c4b9SDmitri Gribenko TokenPtr++; 6475188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_greater); 64817709ae8SDmitri Gribenko State = LS_Normal; 64917709ae8SDmitri Gribenko return; 650f26054f0SDmitri Gribenko case '/': 651f26054f0SDmitri Gribenko TokenPtr++; 652f26054f0SDmitri Gribenko if (TokenPtr != CommentEnd && *TokenPtr == '>') { 653f26054f0SDmitri Gribenko TokenPtr++; 654f26054f0SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_slash_greater); 6554586df76SDmitri Gribenko } else 6564586df76SDmitri Gribenko formTextToken(T, TokenPtr); 6574586df76SDmitri Gribenko 658f26054f0SDmitri Gribenko State = LS_Normal; 659f26054f0SDmitri Gribenko return; 6605188c4b9SDmitri Gribenko } 6615188c4b9SDmitri Gribenko } 6625188c4b9SDmitri Gribenko 6635188c4b9SDmitri Gribenko // Now look ahead and return to normal state if we don't see any HTML tokens 6645188c4b9SDmitri Gribenko // ahead. 6655188c4b9SDmitri Gribenko BufferPtr = skipWhitespace(BufferPtr, CommentEnd); 6665188c4b9SDmitri Gribenko if (BufferPtr == CommentEnd) { 6675188c4b9SDmitri Gribenko State = LS_Normal; 6685188c4b9SDmitri Gribenko return; 6695188c4b9SDmitri Gribenko } 6705188c4b9SDmitri Gribenko 6715188c4b9SDmitri Gribenko C = *BufferPtr; 67217709ae8SDmitri Gribenko if (!isHTMLIdentifierStartingCharacter(C) && 6735188c4b9SDmitri Gribenko C != '=' && C != '\"' && C != '\'' && C != '>') { 6745188c4b9SDmitri Gribenko State = LS_Normal; 6755188c4b9SDmitri Gribenko return; 6765188c4b9SDmitri Gribenko } 6775188c4b9SDmitri Gribenko } 6785188c4b9SDmitri Gribenko 679e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLEndTag(Token &T) { 6805188c4b9SDmitri Gribenko assert(BufferPtr[0] == '<' && BufferPtr[1] == '/'); 6815188c4b9SDmitri Gribenko 6825188c4b9SDmitri Gribenko const char *TagNameBegin = skipWhitespace(BufferPtr + 2, CommentEnd); 6835188c4b9SDmitri Gribenko const char *TagNameEnd = skipHTMLIdentifier(TagNameBegin, CommentEnd); 684107618a6SDmitri Gribenko StringRef Name(TagNameBegin, TagNameEnd - TagNameBegin); 685107618a6SDmitri Gribenko if (!isHTMLTagName(Name)) { 686107618a6SDmitri Gribenko formTextToken(T, TagNameEnd); 687107618a6SDmitri Gribenko return; 688107618a6SDmitri Gribenko } 6895188c4b9SDmitri Gribenko 6905188c4b9SDmitri Gribenko const char *End = skipWhitespace(TagNameEnd, CommentEnd); 6915188c4b9SDmitri Gribenko 692e00ffc7bSDmitri Gribenko formTokenWithChars(T, End, tok::html_end_tag); 693107618a6SDmitri Gribenko T.setHTMLTagEndName(Name); 694ec92531cSDmitri Gribenko 695ec92531cSDmitri Gribenko if (BufferPtr != CommentEnd && *BufferPtr == '>') 696e00ffc7bSDmitri Gribenko State = LS_HTMLEndTag; 697ec92531cSDmitri Gribenko } 698ec92531cSDmitri Gribenko 699e00ffc7bSDmitri Gribenko void Lexer::lexHTMLEndTag(Token &T) { 700ec92531cSDmitri Gribenko assert(BufferPtr != CommentEnd && *BufferPtr == '>'); 701ec92531cSDmitri Gribenko 702ec92531cSDmitri Gribenko formTokenWithChars(T, BufferPtr + 1, tok::html_greater); 703ec92531cSDmitri Gribenko State = LS_Normal; 7045188c4b9SDmitri Gribenko } 7055188c4b9SDmitri Gribenko 706ca7f80adSDmitri Gribenko Lexer::Lexer(llvm::BumpPtrAllocator &Allocator, const CommandTraits &Traits, 7076bab9113SDmitri Gribenko SourceLocation FileLoc, 7085188c4b9SDmitri Gribenko const char *BufferStart, const char *BufferEnd): 709ca7f80adSDmitri Gribenko Allocator(Allocator), Traits(Traits), 7105188c4b9SDmitri Gribenko BufferStart(BufferStart), BufferEnd(BufferEnd), 7116bab9113SDmitri Gribenko FileLoc(FileLoc), BufferPtr(BufferStart), 7125188c4b9SDmitri Gribenko CommentState(LCS_BeforeComment), State(LS_Normal) { 7135188c4b9SDmitri Gribenko } 7145188c4b9SDmitri Gribenko 7155188c4b9SDmitri Gribenko void Lexer::lex(Token &T) { 7165188c4b9SDmitri Gribenko again: 7175188c4b9SDmitri Gribenko switch (CommentState) { 7185188c4b9SDmitri Gribenko case LCS_BeforeComment: 7195188c4b9SDmitri Gribenko if (BufferPtr == BufferEnd) { 7205188c4b9SDmitri Gribenko formTokenWithChars(T, BufferPtr, tok::eof); 7215188c4b9SDmitri Gribenko return; 7225188c4b9SDmitri Gribenko } 7235188c4b9SDmitri Gribenko 7245188c4b9SDmitri Gribenko assert(*BufferPtr == '/'); 7255188c4b9SDmitri Gribenko BufferPtr++; // Skip first slash. 7265188c4b9SDmitri Gribenko switch(*BufferPtr) { 7275188c4b9SDmitri Gribenko case '/': { // BCPL comment. 7285188c4b9SDmitri Gribenko BufferPtr++; // Skip second slash. 7295188c4b9SDmitri Gribenko 7305188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd) { 7315188c4b9SDmitri Gribenko // Skip Doxygen magic marker, if it is present. 7325188c4b9SDmitri Gribenko // It might be missing because of a typo //< or /*<, or because we 7335188c4b9SDmitri Gribenko // merged this non-Doxygen comment into a bunch of Doxygen comments 7345188c4b9SDmitri Gribenko // around it: /** ... */ /* ... */ /** ... */ 7355188c4b9SDmitri Gribenko const char C = *BufferPtr; 7365188c4b9SDmitri Gribenko if (C == '/' || C == '!') 7375188c4b9SDmitri Gribenko BufferPtr++; 7385188c4b9SDmitri Gribenko } 7395188c4b9SDmitri Gribenko 7405188c4b9SDmitri Gribenko // Skip less-than symbol that marks trailing comments. 7415188c4b9SDmitri Gribenko // Skip it even if the comment is not a Doxygen one, because //< and /*< 7425188c4b9SDmitri Gribenko // are frequent typos. 7435188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '<') 7445188c4b9SDmitri Gribenko BufferPtr++; 7455188c4b9SDmitri Gribenko 7465188c4b9SDmitri Gribenko CommentState = LCS_InsideBCPLComment; 747ec92531cSDmitri Gribenko if (State != LS_VerbatimBlockBody && State != LS_VerbatimBlockFirstLine) 7485188c4b9SDmitri Gribenko State = LS_Normal; 7495188c4b9SDmitri Gribenko CommentEnd = findBCPLCommentEnd(BufferPtr, BufferEnd); 7505188c4b9SDmitri Gribenko goto again; 7515188c4b9SDmitri Gribenko } 7525188c4b9SDmitri Gribenko case '*': { // C comment. 7535188c4b9SDmitri Gribenko BufferPtr++; // Skip star. 7545188c4b9SDmitri Gribenko 7555188c4b9SDmitri Gribenko // Skip Doxygen magic marker. 7565188c4b9SDmitri Gribenko const char C = *BufferPtr; 7575188c4b9SDmitri Gribenko if ((C == '*' && *(BufferPtr + 1) != '/') || C == '!') 7585188c4b9SDmitri Gribenko BufferPtr++; 7595188c4b9SDmitri Gribenko 7605188c4b9SDmitri Gribenko // Skip less-than symbol that marks trailing comments. 7615188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '<') 7625188c4b9SDmitri Gribenko BufferPtr++; 7635188c4b9SDmitri Gribenko 7645188c4b9SDmitri Gribenko CommentState = LCS_InsideCComment; 7655188c4b9SDmitri Gribenko State = LS_Normal; 7665188c4b9SDmitri Gribenko CommentEnd = findCCommentEnd(BufferPtr, BufferEnd); 7675188c4b9SDmitri Gribenko goto again; 7685188c4b9SDmitri Gribenko } 7695188c4b9SDmitri Gribenko default: 7705188c4b9SDmitri Gribenko llvm_unreachable("second character of comment should be '/' or '*'"); 7715188c4b9SDmitri Gribenko } 7725188c4b9SDmitri Gribenko 7735188c4b9SDmitri Gribenko case LCS_BetweenComments: { 7745188c4b9SDmitri Gribenko // Consecutive comments are extracted only if there is only whitespace 7755188c4b9SDmitri Gribenko // between them. So we can search for the start of the next comment. 7765188c4b9SDmitri Gribenko const char *EndWhitespace = BufferPtr; 7775188c4b9SDmitri Gribenko while(EndWhitespace != BufferEnd && *EndWhitespace != '/') 7785188c4b9SDmitri Gribenko EndWhitespace++; 7795188c4b9SDmitri Gribenko 7805188c4b9SDmitri Gribenko // Turn any whitespace between comments (and there is only whitespace 78117709ae8SDmitri Gribenko // between them -- guaranteed by comment extraction) into a newline. We 78217709ae8SDmitri Gribenko // have two newlines between C comments in total (first one was synthesized 78317709ae8SDmitri Gribenko // after a comment). 7845188c4b9SDmitri Gribenko formTokenWithChars(T, EndWhitespace, tok::newline); 7855188c4b9SDmitri Gribenko 7865188c4b9SDmitri Gribenko CommentState = LCS_BeforeComment; 7875188c4b9SDmitri Gribenko break; 7885188c4b9SDmitri Gribenko } 7895188c4b9SDmitri Gribenko 7905188c4b9SDmitri Gribenko case LCS_InsideBCPLComment: 7915188c4b9SDmitri Gribenko case LCS_InsideCComment: 7925188c4b9SDmitri Gribenko if (BufferPtr != CommentEnd) { 7935188c4b9SDmitri Gribenko lexCommentText(T); 7945188c4b9SDmitri Gribenko break; 7955188c4b9SDmitri Gribenko } else { 7965188c4b9SDmitri Gribenko // Skip C comment closing sequence. 7975188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) { 7985188c4b9SDmitri Gribenko assert(BufferPtr[0] == '*' && BufferPtr[1] == '/'); 7995188c4b9SDmitri Gribenko BufferPtr += 2; 8005188c4b9SDmitri Gribenko assert(BufferPtr <= BufferEnd); 8015188c4b9SDmitri Gribenko 8025188c4b9SDmitri Gribenko // Synthenize newline just after the C comment, regardless if there is 8035188c4b9SDmitri Gribenko // actually a newline. 8045188c4b9SDmitri Gribenko formTokenWithChars(T, BufferPtr, tok::newline); 8055188c4b9SDmitri Gribenko 8065188c4b9SDmitri Gribenko CommentState = LCS_BetweenComments; 8075188c4b9SDmitri Gribenko break; 8085188c4b9SDmitri Gribenko } else { 8095188c4b9SDmitri Gribenko // Don't synthesized a newline after BCPL comment. 8105188c4b9SDmitri Gribenko CommentState = LCS_BetweenComments; 8115188c4b9SDmitri Gribenko goto again; 8125188c4b9SDmitri Gribenko } 8135188c4b9SDmitri Gribenko } 8145188c4b9SDmitri Gribenko } 8155188c4b9SDmitri Gribenko } 8165188c4b9SDmitri Gribenko 8175188c4b9SDmitri Gribenko StringRef Lexer::getSpelling(const Token &Tok, 8185188c4b9SDmitri Gribenko const SourceManager &SourceMgr, 8195188c4b9SDmitri Gribenko bool *Invalid) const { 8205188c4b9SDmitri Gribenko SourceLocation Loc = Tok.getLocation(); 8215188c4b9SDmitri Gribenko std::pair<FileID, unsigned> LocInfo = SourceMgr.getDecomposedLoc(Loc); 8225188c4b9SDmitri Gribenko 8235188c4b9SDmitri Gribenko bool InvalidTemp = false; 8245188c4b9SDmitri Gribenko StringRef File = SourceMgr.getBufferData(LocInfo.first, &InvalidTemp); 8255188c4b9SDmitri Gribenko if (InvalidTemp) { 8265188c4b9SDmitri Gribenko *Invalid = true; 8275188c4b9SDmitri Gribenko return StringRef(); 8285188c4b9SDmitri Gribenko } 8295188c4b9SDmitri Gribenko 8305188c4b9SDmitri Gribenko const char *Begin = File.data() + LocInfo.second; 8315188c4b9SDmitri Gribenko return StringRef(Begin, Tok.getLength()); 8325188c4b9SDmitri Gribenko } 8335188c4b9SDmitri Gribenko 8345188c4b9SDmitri Gribenko } // end namespace comments 8355188c4b9SDmitri Gribenko } // end namespace clang 8365188c4b9SDmitri Gribenko 837