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 32*3ca956f7SDmitri Gribenko #include "clang/AST/CommentHTMLTags.inc" 33*3ca956f7SDmitri 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 2295188c4b9SDmitri Gribenko bool isCommandNameCharacter(char C) { 2305188c4b9SDmitri Gribenko return (C >= 'a' && C <= 'z') || 2315188c4b9SDmitri Gribenko (C >= 'A' && C <= 'Z') || 2325188c4b9SDmitri Gribenko (C >= '0' && C <= '9'); 2335188c4b9SDmitri Gribenko } 2345188c4b9SDmitri Gribenko 2355188c4b9SDmitri Gribenko const char *skipCommandName(const char *BufferPtr, const char *BufferEnd) { 2365188c4b9SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 2375188c4b9SDmitri Gribenko if (!isCommandNameCharacter(*BufferPtr)) 2385188c4b9SDmitri Gribenko return BufferPtr; 2395188c4b9SDmitri Gribenko } 2405188c4b9SDmitri Gribenko return BufferEnd; 2415188c4b9SDmitri Gribenko } 2425188c4b9SDmitri Gribenko 2435188c4b9SDmitri Gribenko /// Return the one past end pointer for BCPL comments. 2445188c4b9SDmitri Gribenko /// Handles newlines escaped with backslash or trigraph for backslahs. 2455188c4b9SDmitri Gribenko const char *findBCPLCommentEnd(const char *BufferPtr, const char *BufferEnd) { 2465188c4b9SDmitri Gribenko const char *CurPtr = BufferPtr; 2475188c4b9SDmitri Gribenko while (CurPtr != BufferEnd) { 2485188c4b9SDmitri Gribenko char C = *CurPtr; 2495188c4b9SDmitri Gribenko while (C != '\n' && C != '\r') { 2505188c4b9SDmitri Gribenko CurPtr++; 2515188c4b9SDmitri Gribenko if (CurPtr == BufferEnd) 2525188c4b9SDmitri Gribenko return BufferEnd; 2535188c4b9SDmitri Gribenko C = *CurPtr; 2545188c4b9SDmitri Gribenko } 2555188c4b9SDmitri Gribenko // We found a newline, check if it is escaped. 2565188c4b9SDmitri Gribenko const char *EscapePtr = CurPtr - 1; 2575188c4b9SDmitri Gribenko while(isHorizontalWhitespace(*EscapePtr)) 2585188c4b9SDmitri Gribenko EscapePtr--; 2595188c4b9SDmitri Gribenko 2605188c4b9SDmitri Gribenko if (*EscapePtr == '\\' || 2615188c4b9SDmitri Gribenko (EscapePtr - 2 >= BufferPtr && EscapePtr[0] == '/' && 2625188c4b9SDmitri Gribenko EscapePtr[-1] == '?' && EscapePtr[-2] == '?')) { 2635188c4b9SDmitri Gribenko // We found an escaped newline. 2645188c4b9SDmitri Gribenko CurPtr = skipNewline(CurPtr, BufferEnd); 2655188c4b9SDmitri Gribenko } else 2665188c4b9SDmitri Gribenko return CurPtr; // Not an escaped newline. 2675188c4b9SDmitri Gribenko } 2685188c4b9SDmitri Gribenko return BufferEnd; 2695188c4b9SDmitri Gribenko } 2705188c4b9SDmitri Gribenko 2715188c4b9SDmitri Gribenko /// Return the one past end pointer for C comments. 2725188c4b9SDmitri Gribenko /// Very dumb, does not handle escaped newlines or trigraphs. 2735188c4b9SDmitri Gribenko const char *findCCommentEnd(const char *BufferPtr, const char *BufferEnd) { 2745188c4b9SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 2755188c4b9SDmitri Gribenko if (*BufferPtr == '*') { 2765188c4b9SDmitri Gribenko assert(BufferPtr + 1 != BufferEnd); 2775188c4b9SDmitri Gribenko if (*(BufferPtr + 1) == '/') 2785188c4b9SDmitri Gribenko return BufferPtr; 2795188c4b9SDmitri Gribenko } 2805188c4b9SDmitri Gribenko } 2815188c4b9SDmitri Gribenko llvm_unreachable("buffer end hit before '*/' was seen"); 2825188c4b9SDmitri Gribenko } 2835188c4b9SDmitri Gribenko } // unnamed namespace 2845188c4b9SDmitri Gribenko 2855188c4b9SDmitri Gribenko void Lexer::lexCommentText(Token &T) { 2865188c4b9SDmitri Gribenko assert(CommentState == LCS_InsideBCPLComment || 2875188c4b9SDmitri Gribenko CommentState == LCS_InsideCComment); 2885188c4b9SDmitri Gribenko 2895188c4b9SDmitri Gribenko switch (State) { 2905188c4b9SDmitri Gribenko case LS_Normal: 2915188c4b9SDmitri Gribenko break; 2925188c4b9SDmitri Gribenko case LS_VerbatimBlockFirstLine: 2935188c4b9SDmitri Gribenko lexVerbatimBlockFirstLine(T); 2945188c4b9SDmitri Gribenko return; 2955188c4b9SDmitri Gribenko case LS_VerbatimBlockBody: 2965188c4b9SDmitri Gribenko lexVerbatimBlockBody(T); 2975188c4b9SDmitri Gribenko return; 2981669f702SDmitri Gribenko case LS_VerbatimLineText: 2991669f702SDmitri Gribenko lexVerbatimLineText(T); 3001669f702SDmitri Gribenko return; 301e00ffc7bSDmitri Gribenko case LS_HTMLStartTag: 302e00ffc7bSDmitri Gribenko lexHTMLStartTag(T); 3035188c4b9SDmitri Gribenko return; 304e00ffc7bSDmitri Gribenko case LS_HTMLEndTag: 305e00ffc7bSDmitri Gribenko lexHTMLEndTag(T); 306ec92531cSDmitri Gribenko return; 3075188c4b9SDmitri Gribenko } 3085188c4b9SDmitri Gribenko 3095188c4b9SDmitri Gribenko assert(State == LS_Normal); 3105188c4b9SDmitri Gribenko 3115188c4b9SDmitri Gribenko const char *TokenPtr = BufferPtr; 3125188c4b9SDmitri Gribenko assert(TokenPtr < CommentEnd); 3135188c4b9SDmitri Gribenko while (TokenPtr != CommentEnd) { 3145188c4b9SDmitri Gribenko switch(*TokenPtr) { 3155188c4b9SDmitri Gribenko case '\\': 3165188c4b9SDmitri Gribenko case '@': { 3175188c4b9SDmitri Gribenko TokenPtr++; 3185188c4b9SDmitri Gribenko if (TokenPtr == CommentEnd) { 3194586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3205188c4b9SDmitri Gribenko return; 3215188c4b9SDmitri Gribenko } 3225188c4b9SDmitri Gribenko char C = *TokenPtr; 3235188c4b9SDmitri Gribenko switch (C) { 3245188c4b9SDmitri Gribenko default: 3255188c4b9SDmitri Gribenko break; 3265188c4b9SDmitri Gribenko 3275188c4b9SDmitri Gribenko case '\\': case '@': case '&': case '$': 3285188c4b9SDmitri Gribenko case '#': case '<': case '>': case '%': 3295188c4b9SDmitri Gribenko case '\"': case '.': case ':': 3305188c4b9SDmitri Gribenko // This is one of \\ \@ \& \$ etc escape sequences. 3315188c4b9SDmitri Gribenko TokenPtr++; 3325188c4b9SDmitri Gribenko if (C == ':' && TokenPtr != CommentEnd && *TokenPtr == ':') { 3335188c4b9SDmitri Gribenko // This is the \:: escape sequence. 3345188c4b9SDmitri Gribenko TokenPtr++; 3355188c4b9SDmitri Gribenko } 33660ddd8a1SDmitri Gribenko StringRef UnescapedText(BufferPtr + 1, TokenPtr - (BufferPtr + 1)); 3375188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::text); 33860ddd8a1SDmitri Gribenko T.setText(UnescapedText); 3395188c4b9SDmitri Gribenko return; 3405188c4b9SDmitri Gribenko } 3415188c4b9SDmitri Gribenko 3425188c4b9SDmitri Gribenko // Don't make zero-length commands. 3435188c4b9SDmitri Gribenko if (!isCommandNameCharacter(*TokenPtr)) { 3444586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3455188c4b9SDmitri Gribenko return; 3465188c4b9SDmitri Gribenko } 3475188c4b9SDmitri Gribenko 3485188c4b9SDmitri Gribenko TokenPtr = skipCommandName(TokenPtr, CommentEnd); 3495188c4b9SDmitri Gribenko unsigned Length = TokenPtr - (BufferPtr + 1); 3505188c4b9SDmitri Gribenko 3515188c4b9SDmitri Gribenko // Hardcoded support for lexing LaTeX formula commands 3525188c4b9SDmitri Gribenko // \f$ \f[ \f] \f{ \f} as a single command. 3535188c4b9SDmitri Gribenko if (Length == 1 && TokenPtr[-1] == 'f' && TokenPtr != CommentEnd) { 3545188c4b9SDmitri Gribenko C = *TokenPtr; 3555188c4b9SDmitri Gribenko if (C == '$' || C == '[' || C == ']' || C == '{' || C == '}') { 3565188c4b9SDmitri Gribenko TokenPtr++; 3575188c4b9SDmitri Gribenko Length++; 3585188c4b9SDmitri Gribenko } 3595188c4b9SDmitri Gribenko } 3605188c4b9SDmitri Gribenko 3615188c4b9SDmitri Gribenko const StringRef CommandName(BufferPtr + 1, Length); 3625188c4b9SDmitri Gribenko StringRef EndName; 3635188c4b9SDmitri Gribenko 364ca7f80adSDmitri Gribenko if (Traits.isVerbatimBlockCommand(CommandName, EndName)) { 3655188c4b9SDmitri Gribenko setupAndLexVerbatimBlock(T, TokenPtr, *BufferPtr, EndName); 3665188c4b9SDmitri Gribenko return; 3675188c4b9SDmitri Gribenko } 368ca7f80adSDmitri Gribenko if (Traits.isVerbatimLineCommand(CommandName)) { 3691669f702SDmitri Gribenko setupAndLexVerbatimLine(T, TokenPtr); 3705188c4b9SDmitri Gribenko return; 3715188c4b9SDmitri Gribenko } 3725188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::command); 3735188c4b9SDmitri Gribenko T.setCommandName(CommandName); 3745188c4b9SDmitri Gribenko return; 3755188c4b9SDmitri Gribenko } 3765188c4b9SDmitri Gribenko 3774586df76SDmitri Gribenko case '&': 3784586df76SDmitri Gribenko lexHTMLCharacterReference(T); 3794586df76SDmitri Gribenko return; 3804586df76SDmitri Gribenko 3815188c4b9SDmitri Gribenko case '<': { 3825188c4b9SDmitri Gribenko TokenPtr++; 3835188c4b9SDmitri Gribenko if (TokenPtr == CommentEnd) { 3844586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3855188c4b9SDmitri Gribenko return; 3865188c4b9SDmitri Gribenko } 3875188c4b9SDmitri Gribenko const char C = *TokenPtr; 38817709ae8SDmitri Gribenko if (isHTMLIdentifierStartingCharacter(C)) 389e00ffc7bSDmitri Gribenko setupAndLexHTMLStartTag(T); 3905188c4b9SDmitri Gribenko else if (C == '/') 391e00ffc7bSDmitri Gribenko setupAndLexHTMLEndTag(T); 3924586df76SDmitri Gribenko else 3934586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3944586df76SDmitri Gribenko 3955188c4b9SDmitri Gribenko return; 3965188c4b9SDmitri Gribenko } 3975188c4b9SDmitri Gribenko 3985188c4b9SDmitri Gribenko case '\n': 3995188c4b9SDmitri Gribenko case '\r': 4005188c4b9SDmitri Gribenko TokenPtr = skipNewline(TokenPtr, CommentEnd); 4015188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::newline); 4025188c4b9SDmitri Gribenko 4035188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) 4045188c4b9SDmitri Gribenko skipLineStartingDecorations(); 4055188c4b9SDmitri Gribenko return; 4065188c4b9SDmitri Gribenko 4075188c4b9SDmitri Gribenko default: { 4085188c4b9SDmitri Gribenko while (true) { 4095188c4b9SDmitri Gribenko TokenPtr++; 4105188c4b9SDmitri Gribenko if (TokenPtr == CommentEnd) 4115188c4b9SDmitri Gribenko break; 41217709ae8SDmitri Gribenko const char C = *TokenPtr; 4135188c4b9SDmitri Gribenko if(C == '\n' || C == '\r' || 4144586df76SDmitri Gribenko C == '\\' || C == '@' || C == '&' || C == '<') 4155188c4b9SDmitri Gribenko break; 4165188c4b9SDmitri Gribenko } 4174586df76SDmitri Gribenko formTextToken(T, TokenPtr); 4185188c4b9SDmitri Gribenko return; 4195188c4b9SDmitri Gribenko } 4205188c4b9SDmitri Gribenko } 4215188c4b9SDmitri Gribenko } 4225188c4b9SDmitri Gribenko } 4235188c4b9SDmitri Gribenko 4245188c4b9SDmitri Gribenko void Lexer::setupAndLexVerbatimBlock(Token &T, 4255188c4b9SDmitri Gribenko const char *TextBegin, 4265188c4b9SDmitri Gribenko char Marker, StringRef EndName) { 4275188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.clear(); 4285188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.append(Marker == '\\' ? "\\" : "@"); 4295188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.append(EndName); 4305188c4b9SDmitri Gribenko 43160ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, TextBegin - (BufferPtr + 1)); 4325188c4b9SDmitri Gribenko formTokenWithChars(T, TextBegin, tok::verbatim_block_begin); 43360ddd8a1SDmitri Gribenko T.setVerbatimBlockName(Name); 4345188c4b9SDmitri Gribenko 435ec92531cSDmitri Gribenko // If there is a newline following the verbatim opening command, skip the 436ec92531cSDmitri Gribenko // newline so that we don't create an tok::verbatim_block_line with empty 437ec92531cSDmitri Gribenko // text content. 438ec92531cSDmitri Gribenko if (BufferPtr != CommentEnd) { 439ec92531cSDmitri Gribenko const char C = *BufferPtr; 440ec92531cSDmitri Gribenko if (C == '\n' || C == '\r') { 441ec92531cSDmitri Gribenko BufferPtr = skipNewline(BufferPtr, CommentEnd); 442ec92531cSDmitri Gribenko State = LS_VerbatimBlockBody; 443ec92531cSDmitri Gribenko return; 444ec92531cSDmitri Gribenko } 445ec92531cSDmitri Gribenko } 446ec92531cSDmitri Gribenko 4475188c4b9SDmitri Gribenko State = LS_VerbatimBlockFirstLine; 4485188c4b9SDmitri Gribenko } 4495188c4b9SDmitri Gribenko 4505188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockFirstLine(Token &T) { 451e4a3997dSDmitri Gribenko again: 4525188c4b9SDmitri Gribenko assert(BufferPtr < CommentEnd); 4535188c4b9SDmitri Gribenko 4545188c4b9SDmitri Gribenko // FIXME: It would be better to scan the text once, finding either the block 4555188c4b9SDmitri Gribenko // end command or newline. 4565188c4b9SDmitri Gribenko // 4575188c4b9SDmitri Gribenko // Extract current line. 4585188c4b9SDmitri Gribenko const char *Newline = findNewline(BufferPtr, CommentEnd); 4595188c4b9SDmitri Gribenko StringRef Line(BufferPtr, Newline - BufferPtr); 4605188c4b9SDmitri Gribenko 4615188c4b9SDmitri Gribenko // Look for end command in current line. 4625188c4b9SDmitri Gribenko size_t Pos = Line.find(VerbatimBlockEndCommandName); 463ec92531cSDmitri Gribenko const char *TextEnd; 4645188c4b9SDmitri Gribenko const char *NextLine; 4655188c4b9SDmitri Gribenko if (Pos == StringRef::npos) { 4665188c4b9SDmitri Gribenko // Current line is completely verbatim. 467ec92531cSDmitri Gribenko TextEnd = Newline; 4685188c4b9SDmitri Gribenko NextLine = skipNewline(Newline, CommentEnd); 4695188c4b9SDmitri Gribenko } else if (Pos == 0) { 4705188c4b9SDmitri Gribenko // Current line contains just an end command. 4715188c4b9SDmitri Gribenko const char *End = BufferPtr + VerbatimBlockEndCommandName.size(); 47260ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, End - (BufferPtr + 1)); 4735188c4b9SDmitri Gribenko formTokenWithChars(T, End, tok::verbatim_block_end); 47460ddd8a1SDmitri Gribenko T.setVerbatimBlockName(Name); 4755188c4b9SDmitri Gribenko State = LS_Normal; 4765188c4b9SDmitri Gribenko return; 4775188c4b9SDmitri Gribenko } else { 4785188c4b9SDmitri Gribenko // There is some text, followed by end command. Extract text first. 479ec92531cSDmitri Gribenko TextEnd = BufferPtr + Pos; 480ec92531cSDmitri Gribenko NextLine = TextEnd; 481e4a3997dSDmitri Gribenko // If there is only whitespace before end command, skip whitespace. 482e4a3997dSDmitri Gribenko if (isWhitespace(BufferPtr, TextEnd)) { 483e4a3997dSDmitri Gribenko BufferPtr = TextEnd; 484e4a3997dSDmitri Gribenko goto again; 485e4a3997dSDmitri Gribenko } 4865188c4b9SDmitri Gribenko } 4875188c4b9SDmitri Gribenko 488ec92531cSDmitri Gribenko StringRef Text(BufferPtr, TextEnd - BufferPtr); 4895188c4b9SDmitri Gribenko formTokenWithChars(T, NextLine, tok::verbatim_block_line); 49060ddd8a1SDmitri Gribenko T.setVerbatimBlockText(Text); 4915188c4b9SDmitri Gribenko 4925188c4b9SDmitri Gribenko State = LS_VerbatimBlockBody; 4935188c4b9SDmitri Gribenko } 4945188c4b9SDmitri Gribenko 4955188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockBody(Token &T) { 4965188c4b9SDmitri Gribenko assert(State == LS_VerbatimBlockBody); 4975188c4b9SDmitri Gribenko 4985188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) 4995188c4b9SDmitri Gribenko skipLineStartingDecorations(); 5005188c4b9SDmitri Gribenko 5015188c4b9SDmitri Gribenko lexVerbatimBlockFirstLine(T); 5025188c4b9SDmitri Gribenko } 5035188c4b9SDmitri Gribenko 5041669f702SDmitri Gribenko void Lexer::setupAndLexVerbatimLine(Token &T, const char *TextBegin) { 5051669f702SDmitri Gribenko const StringRef Name(BufferPtr + 1, TextBegin - BufferPtr - 1); 5061669f702SDmitri Gribenko formTokenWithChars(T, TextBegin, tok::verbatim_line_name); 5071669f702SDmitri Gribenko T.setVerbatimLineName(Name); 5081669f702SDmitri Gribenko 5091669f702SDmitri Gribenko State = LS_VerbatimLineText; 5101669f702SDmitri Gribenko } 5111669f702SDmitri Gribenko 5121669f702SDmitri Gribenko void Lexer::lexVerbatimLineText(Token &T) { 5131669f702SDmitri Gribenko assert(State == LS_VerbatimLineText); 5141669f702SDmitri Gribenko 5155188c4b9SDmitri Gribenko // Extract current line. 5165188c4b9SDmitri Gribenko const char *Newline = findNewline(BufferPtr, CommentEnd); 5171669f702SDmitri Gribenko const StringRef Text(BufferPtr, Newline - BufferPtr); 5181669f702SDmitri Gribenko formTokenWithChars(T, Newline, tok::verbatim_line_text); 5195188c4b9SDmitri Gribenko T.setVerbatimLineText(Text); 5201669f702SDmitri Gribenko 5211669f702SDmitri Gribenko State = LS_Normal; 5225188c4b9SDmitri Gribenko } 5235188c4b9SDmitri Gribenko 5244586df76SDmitri Gribenko void Lexer::lexHTMLCharacterReference(Token &T) { 5254586df76SDmitri Gribenko const char *TokenPtr = BufferPtr; 5264586df76SDmitri Gribenko assert(*TokenPtr == '&'); 5274586df76SDmitri Gribenko TokenPtr++; 5284586df76SDmitri Gribenko if (TokenPtr == CommentEnd) { 5294586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5304586df76SDmitri Gribenko return; 5314586df76SDmitri Gribenko } 5324586df76SDmitri Gribenko const char *NamePtr; 5334586df76SDmitri Gribenko bool isNamed = false; 5344586df76SDmitri Gribenko bool isDecimal = false; 5354586df76SDmitri Gribenko char C = *TokenPtr; 5364586df76SDmitri Gribenko if (isHTMLNamedCharacterReferenceCharacter(C)) { 5374586df76SDmitri Gribenko NamePtr = TokenPtr; 5384586df76SDmitri Gribenko TokenPtr = skipNamedCharacterReference(TokenPtr, CommentEnd); 5394586df76SDmitri Gribenko isNamed = true; 5404586df76SDmitri Gribenko } else if (C == '#') { 5414586df76SDmitri Gribenko TokenPtr++; 5424586df76SDmitri Gribenko if (TokenPtr == CommentEnd) { 5434586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5444586df76SDmitri Gribenko return; 5454586df76SDmitri Gribenko } 5464586df76SDmitri Gribenko C = *TokenPtr; 5474586df76SDmitri Gribenko if (isHTMLDecimalCharacterReferenceCharacter(C)) { 5484586df76SDmitri Gribenko NamePtr = TokenPtr; 5494586df76SDmitri Gribenko TokenPtr = skipDecimalCharacterReference(TokenPtr, CommentEnd); 5504586df76SDmitri Gribenko isDecimal = true; 5514586df76SDmitri Gribenko } else if (C == 'x' || C == 'X') { 5524586df76SDmitri Gribenko TokenPtr++; 5534586df76SDmitri Gribenko NamePtr = TokenPtr; 5544586df76SDmitri Gribenko TokenPtr = skipHexCharacterReference(TokenPtr, CommentEnd); 5554586df76SDmitri Gribenko } else { 5564586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5574586df76SDmitri Gribenko return; 5584586df76SDmitri Gribenko } 5594586df76SDmitri Gribenko } else { 5604586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5614586df76SDmitri Gribenko return; 5624586df76SDmitri Gribenko } 5634586df76SDmitri Gribenko if (NamePtr == TokenPtr || TokenPtr == CommentEnd || 5644586df76SDmitri Gribenko *TokenPtr != ';') { 5654586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5664586df76SDmitri Gribenko return; 5674586df76SDmitri Gribenko } 5684586df76SDmitri Gribenko StringRef Name(NamePtr, TokenPtr - NamePtr); 5694586df76SDmitri Gribenko TokenPtr++; // Skip semicolon. 5704586df76SDmitri Gribenko StringRef Resolved; 5714586df76SDmitri Gribenko if (isNamed) 5724586df76SDmitri Gribenko Resolved = resolveHTMLNamedCharacterReference(Name); 5734586df76SDmitri Gribenko else if (isDecimal) 5744586df76SDmitri Gribenko Resolved = resolveHTMLDecimalCharacterReference(Name); 5754586df76SDmitri Gribenko else 5764586df76SDmitri Gribenko Resolved = resolveHTMLHexCharacterReference(Name); 5774586df76SDmitri Gribenko 5784586df76SDmitri Gribenko if (Resolved.empty()) { 5794586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5804586df76SDmitri Gribenko return; 5814586df76SDmitri Gribenko } 5824586df76SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::text); 5834586df76SDmitri Gribenko T.setText(Resolved); 5844586df76SDmitri Gribenko return; 5854586df76SDmitri Gribenko } 5864586df76SDmitri Gribenko 587e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLStartTag(Token &T) { 58817709ae8SDmitri Gribenko assert(BufferPtr[0] == '<' && 58917709ae8SDmitri Gribenko isHTMLIdentifierStartingCharacter(BufferPtr[1])); 5905188c4b9SDmitri Gribenko const char *TagNameEnd = skipHTMLIdentifier(BufferPtr + 2, CommentEnd); 59160ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, TagNameEnd - (BufferPtr + 1)); 592107618a6SDmitri Gribenko if (!isHTMLTagName(Name)) { 593107618a6SDmitri Gribenko formTextToken(T, TagNameEnd); 594107618a6SDmitri Gribenko return; 595107618a6SDmitri Gribenko } 596107618a6SDmitri Gribenko 597e00ffc7bSDmitri Gribenko formTokenWithChars(T, TagNameEnd, tok::html_start_tag); 598e00ffc7bSDmitri Gribenko T.setHTMLTagStartName(Name); 5995188c4b9SDmitri Gribenko 6005188c4b9SDmitri Gribenko BufferPtr = skipWhitespace(BufferPtr, CommentEnd); 6015188c4b9SDmitri Gribenko 60217709ae8SDmitri Gribenko const char C = *BufferPtr; 60317709ae8SDmitri Gribenko if (BufferPtr != CommentEnd && 604f26054f0SDmitri Gribenko (C == '>' || C == '/' || isHTMLIdentifierStartingCharacter(C))) 605e00ffc7bSDmitri Gribenko State = LS_HTMLStartTag; 6065188c4b9SDmitri Gribenko } 6075188c4b9SDmitri Gribenko 608e00ffc7bSDmitri Gribenko void Lexer::lexHTMLStartTag(Token &T) { 609e00ffc7bSDmitri Gribenko assert(State == LS_HTMLStartTag); 6105188c4b9SDmitri Gribenko 6115188c4b9SDmitri Gribenko const char *TokenPtr = BufferPtr; 6125188c4b9SDmitri Gribenko char C = *TokenPtr; 6135188c4b9SDmitri Gribenko if (isHTMLIdentifierCharacter(C)) { 6145188c4b9SDmitri Gribenko TokenPtr = skipHTMLIdentifier(TokenPtr, CommentEnd); 61560ddd8a1SDmitri Gribenko StringRef Ident(BufferPtr, TokenPtr - BufferPtr); 6165188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_ident); 61760ddd8a1SDmitri Gribenko T.setHTMLIdent(Ident); 6185188c4b9SDmitri Gribenko } else { 6195188c4b9SDmitri Gribenko switch (C) { 6205188c4b9SDmitri Gribenko case '=': 6215188c4b9SDmitri Gribenko TokenPtr++; 6225188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_equals); 6235188c4b9SDmitri Gribenko break; 6245188c4b9SDmitri Gribenko case '\"': 6255188c4b9SDmitri Gribenko case '\'': { 6265188c4b9SDmitri Gribenko const char *OpenQuote = TokenPtr; 6275188c4b9SDmitri Gribenko TokenPtr = skipHTMLQuotedString(TokenPtr, CommentEnd); 6285188c4b9SDmitri Gribenko const char *ClosingQuote = TokenPtr; 6295188c4b9SDmitri Gribenko if (TokenPtr != CommentEnd) // Skip closing quote. 6305188c4b9SDmitri Gribenko TokenPtr++; 6315188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_quoted_string); 6325188c4b9SDmitri Gribenko T.setHTMLQuotedString(StringRef(OpenQuote + 1, 6335188c4b9SDmitri Gribenko ClosingQuote - (OpenQuote + 1))); 6345188c4b9SDmitri Gribenko break; 6355188c4b9SDmitri Gribenko } 6365188c4b9SDmitri Gribenko case '>': 6375188c4b9SDmitri Gribenko TokenPtr++; 6385188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_greater); 63917709ae8SDmitri Gribenko State = LS_Normal; 64017709ae8SDmitri Gribenko return; 641f26054f0SDmitri Gribenko case '/': 642f26054f0SDmitri Gribenko TokenPtr++; 643f26054f0SDmitri Gribenko if (TokenPtr != CommentEnd && *TokenPtr == '>') { 644f26054f0SDmitri Gribenko TokenPtr++; 645f26054f0SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_slash_greater); 6464586df76SDmitri Gribenko } else 6474586df76SDmitri Gribenko formTextToken(T, TokenPtr); 6484586df76SDmitri Gribenko 649f26054f0SDmitri Gribenko State = LS_Normal; 650f26054f0SDmitri Gribenko return; 6515188c4b9SDmitri Gribenko } 6525188c4b9SDmitri Gribenko } 6535188c4b9SDmitri Gribenko 6545188c4b9SDmitri Gribenko // Now look ahead and return to normal state if we don't see any HTML tokens 6555188c4b9SDmitri Gribenko // ahead. 6565188c4b9SDmitri Gribenko BufferPtr = skipWhitespace(BufferPtr, CommentEnd); 6575188c4b9SDmitri Gribenko if (BufferPtr == CommentEnd) { 6585188c4b9SDmitri Gribenko State = LS_Normal; 6595188c4b9SDmitri Gribenko return; 6605188c4b9SDmitri Gribenko } 6615188c4b9SDmitri Gribenko 6625188c4b9SDmitri Gribenko C = *BufferPtr; 66317709ae8SDmitri Gribenko if (!isHTMLIdentifierStartingCharacter(C) && 6645188c4b9SDmitri Gribenko C != '=' && C != '\"' && C != '\'' && C != '>') { 6655188c4b9SDmitri Gribenko State = LS_Normal; 6665188c4b9SDmitri Gribenko return; 6675188c4b9SDmitri Gribenko } 6685188c4b9SDmitri Gribenko } 6695188c4b9SDmitri Gribenko 670e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLEndTag(Token &T) { 6715188c4b9SDmitri Gribenko assert(BufferPtr[0] == '<' && BufferPtr[1] == '/'); 6725188c4b9SDmitri Gribenko 6735188c4b9SDmitri Gribenko const char *TagNameBegin = skipWhitespace(BufferPtr + 2, CommentEnd); 6745188c4b9SDmitri Gribenko const char *TagNameEnd = skipHTMLIdentifier(TagNameBegin, CommentEnd); 675107618a6SDmitri Gribenko StringRef Name(TagNameBegin, TagNameEnd - TagNameBegin); 676107618a6SDmitri Gribenko if (!isHTMLTagName(Name)) { 677107618a6SDmitri Gribenko formTextToken(T, TagNameEnd); 678107618a6SDmitri Gribenko return; 679107618a6SDmitri Gribenko } 6805188c4b9SDmitri Gribenko 6815188c4b9SDmitri Gribenko const char *End = skipWhitespace(TagNameEnd, CommentEnd); 6825188c4b9SDmitri Gribenko 683e00ffc7bSDmitri Gribenko formTokenWithChars(T, End, tok::html_end_tag); 684107618a6SDmitri Gribenko T.setHTMLTagEndName(Name); 685ec92531cSDmitri Gribenko 686ec92531cSDmitri Gribenko if (BufferPtr != CommentEnd && *BufferPtr == '>') 687e00ffc7bSDmitri Gribenko State = LS_HTMLEndTag; 688ec92531cSDmitri Gribenko } 689ec92531cSDmitri Gribenko 690e00ffc7bSDmitri Gribenko void Lexer::lexHTMLEndTag(Token &T) { 691ec92531cSDmitri Gribenko assert(BufferPtr != CommentEnd && *BufferPtr == '>'); 692ec92531cSDmitri Gribenko 693ec92531cSDmitri Gribenko formTokenWithChars(T, BufferPtr + 1, tok::html_greater); 694ec92531cSDmitri Gribenko State = LS_Normal; 6955188c4b9SDmitri Gribenko } 6965188c4b9SDmitri Gribenko 697ca7f80adSDmitri Gribenko Lexer::Lexer(llvm::BumpPtrAllocator &Allocator, const CommandTraits &Traits, 6984586df76SDmitri Gribenko SourceLocation FileLoc, const CommentOptions &CommOpts, 6995188c4b9SDmitri Gribenko const char *BufferStart, const char *BufferEnd): 700ca7f80adSDmitri Gribenko Allocator(Allocator), Traits(Traits), 7015188c4b9SDmitri Gribenko BufferStart(BufferStart), BufferEnd(BufferEnd), 7025188c4b9SDmitri Gribenko FileLoc(FileLoc), CommOpts(CommOpts), BufferPtr(BufferStart), 7035188c4b9SDmitri Gribenko CommentState(LCS_BeforeComment), State(LS_Normal) { 7045188c4b9SDmitri Gribenko } 7055188c4b9SDmitri Gribenko 7065188c4b9SDmitri Gribenko void Lexer::lex(Token &T) { 7075188c4b9SDmitri Gribenko again: 7085188c4b9SDmitri Gribenko switch (CommentState) { 7095188c4b9SDmitri Gribenko case LCS_BeforeComment: 7105188c4b9SDmitri Gribenko if (BufferPtr == BufferEnd) { 7115188c4b9SDmitri Gribenko formTokenWithChars(T, BufferPtr, tok::eof); 7125188c4b9SDmitri Gribenko return; 7135188c4b9SDmitri Gribenko } 7145188c4b9SDmitri Gribenko 7155188c4b9SDmitri Gribenko assert(*BufferPtr == '/'); 7165188c4b9SDmitri Gribenko BufferPtr++; // Skip first slash. 7175188c4b9SDmitri Gribenko switch(*BufferPtr) { 7185188c4b9SDmitri Gribenko case '/': { // BCPL comment. 7195188c4b9SDmitri Gribenko BufferPtr++; // Skip second slash. 7205188c4b9SDmitri Gribenko 7215188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd) { 7225188c4b9SDmitri Gribenko // Skip Doxygen magic marker, if it is present. 7235188c4b9SDmitri Gribenko // It might be missing because of a typo //< or /*<, or because we 7245188c4b9SDmitri Gribenko // merged this non-Doxygen comment into a bunch of Doxygen comments 7255188c4b9SDmitri Gribenko // around it: /** ... */ /* ... */ /** ... */ 7265188c4b9SDmitri Gribenko const char C = *BufferPtr; 7275188c4b9SDmitri Gribenko if (C == '/' || C == '!') 7285188c4b9SDmitri Gribenko BufferPtr++; 7295188c4b9SDmitri Gribenko } 7305188c4b9SDmitri Gribenko 7315188c4b9SDmitri Gribenko // Skip less-than symbol that marks trailing comments. 7325188c4b9SDmitri Gribenko // Skip it even if the comment is not a Doxygen one, because //< and /*< 7335188c4b9SDmitri Gribenko // are frequent typos. 7345188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '<') 7355188c4b9SDmitri Gribenko BufferPtr++; 7365188c4b9SDmitri Gribenko 7375188c4b9SDmitri Gribenko CommentState = LCS_InsideBCPLComment; 738ec92531cSDmitri Gribenko if (State != LS_VerbatimBlockBody && State != LS_VerbatimBlockFirstLine) 7395188c4b9SDmitri Gribenko State = LS_Normal; 7405188c4b9SDmitri Gribenko CommentEnd = findBCPLCommentEnd(BufferPtr, BufferEnd); 7415188c4b9SDmitri Gribenko goto again; 7425188c4b9SDmitri Gribenko } 7435188c4b9SDmitri Gribenko case '*': { // C comment. 7445188c4b9SDmitri Gribenko BufferPtr++; // Skip star. 7455188c4b9SDmitri Gribenko 7465188c4b9SDmitri Gribenko // Skip Doxygen magic marker. 7475188c4b9SDmitri Gribenko const char C = *BufferPtr; 7485188c4b9SDmitri Gribenko if ((C == '*' && *(BufferPtr + 1) != '/') || C == '!') 7495188c4b9SDmitri Gribenko BufferPtr++; 7505188c4b9SDmitri Gribenko 7515188c4b9SDmitri Gribenko // Skip less-than symbol that marks trailing comments. 7525188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '<') 7535188c4b9SDmitri Gribenko BufferPtr++; 7545188c4b9SDmitri Gribenko 7555188c4b9SDmitri Gribenko CommentState = LCS_InsideCComment; 7565188c4b9SDmitri Gribenko State = LS_Normal; 7575188c4b9SDmitri Gribenko CommentEnd = findCCommentEnd(BufferPtr, BufferEnd); 7585188c4b9SDmitri Gribenko goto again; 7595188c4b9SDmitri Gribenko } 7605188c4b9SDmitri Gribenko default: 7615188c4b9SDmitri Gribenko llvm_unreachable("second character of comment should be '/' or '*'"); 7625188c4b9SDmitri Gribenko } 7635188c4b9SDmitri Gribenko 7645188c4b9SDmitri Gribenko case LCS_BetweenComments: { 7655188c4b9SDmitri Gribenko // Consecutive comments are extracted only if there is only whitespace 7665188c4b9SDmitri Gribenko // between them. So we can search for the start of the next comment. 7675188c4b9SDmitri Gribenko const char *EndWhitespace = BufferPtr; 7685188c4b9SDmitri Gribenko while(EndWhitespace != BufferEnd && *EndWhitespace != '/') 7695188c4b9SDmitri Gribenko EndWhitespace++; 7705188c4b9SDmitri Gribenko 7715188c4b9SDmitri Gribenko // Turn any whitespace between comments (and there is only whitespace 77217709ae8SDmitri Gribenko // between them -- guaranteed by comment extraction) into a newline. We 77317709ae8SDmitri Gribenko // have two newlines between C comments in total (first one was synthesized 77417709ae8SDmitri Gribenko // after a comment). 7755188c4b9SDmitri Gribenko formTokenWithChars(T, EndWhitespace, tok::newline); 7765188c4b9SDmitri Gribenko 7775188c4b9SDmitri Gribenko CommentState = LCS_BeforeComment; 7785188c4b9SDmitri Gribenko break; 7795188c4b9SDmitri Gribenko } 7805188c4b9SDmitri Gribenko 7815188c4b9SDmitri Gribenko case LCS_InsideBCPLComment: 7825188c4b9SDmitri Gribenko case LCS_InsideCComment: 7835188c4b9SDmitri Gribenko if (BufferPtr != CommentEnd) { 7845188c4b9SDmitri Gribenko lexCommentText(T); 7855188c4b9SDmitri Gribenko break; 7865188c4b9SDmitri Gribenko } else { 7875188c4b9SDmitri Gribenko // Skip C comment closing sequence. 7885188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) { 7895188c4b9SDmitri Gribenko assert(BufferPtr[0] == '*' && BufferPtr[1] == '/'); 7905188c4b9SDmitri Gribenko BufferPtr += 2; 7915188c4b9SDmitri Gribenko assert(BufferPtr <= BufferEnd); 7925188c4b9SDmitri Gribenko 7935188c4b9SDmitri Gribenko // Synthenize newline just after the C comment, regardless if there is 7945188c4b9SDmitri Gribenko // actually a newline. 7955188c4b9SDmitri Gribenko formTokenWithChars(T, BufferPtr, tok::newline); 7965188c4b9SDmitri Gribenko 7975188c4b9SDmitri Gribenko CommentState = LCS_BetweenComments; 7985188c4b9SDmitri Gribenko break; 7995188c4b9SDmitri Gribenko } else { 8005188c4b9SDmitri Gribenko // Don't synthesized a newline after BCPL comment. 8015188c4b9SDmitri Gribenko CommentState = LCS_BetweenComments; 8025188c4b9SDmitri Gribenko goto again; 8035188c4b9SDmitri Gribenko } 8045188c4b9SDmitri Gribenko } 8055188c4b9SDmitri Gribenko } 8065188c4b9SDmitri Gribenko } 8075188c4b9SDmitri Gribenko 8085188c4b9SDmitri Gribenko StringRef Lexer::getSpelling(const Token &Tok, 8095188c4b9SDmitri Gribenko const SourceManager &SourceMgr, 8105188c4b9SDmitri Gribenko bool *Invalid) const { 8115188c4b9SDmitri Gribenko SourceLocation Loc = Tok.getLocation(); 8125188c4b9SDmitri Gribenko std::pair<FileID, unsigned> LocInfo = SourceMgr.getDecomposedLoc(Loc); 8135188c4b9SDmitri Gribenko 8145188c4b9SDmitri Gribenko bool InvalidTemp = false; 8155188c4b9SDmitri Gribenko StringRef File = SourceMgr.getBufferData(LocInfo.first, &InvalidTemp); 8165188c4b9SDmitri Gribenko if (InvalidTemp) { 8175188c4b9SDmitri Gribenko *Invalid = true; 8185188c4b9SDmitri Gribenko return StringRef(); 8195188c4b9SDmitri Gribenko } 8205188c4b9SDmitri Gribenko 8215188c4b9SDmitri Gribenko const char *Begin = File.data() + LocInfo.second; 8225188c4b9SDmitri Gribenko return StringRef(Begin, Tok.getLength()); 8235188c4b9SDmitri Gribenko } 8245188c4b9SDmitri Gribenko 8255188c4b9SDmitri Gribenko } // end namespace comments 8265188c4b9SDmitri Gribenko } // end namespace clang 8275188c4b9SDmitri Gribenko 828