15188c4b9SDmitri Gribenko #include "clang/AST/CommentLexer.h" 2ca7f80adSDmitri Gribenko #include "clang/AST/CommentCommandTraits.h" 36738e430SFariborz Jahanian #include "clang/AST/CommentDiagnostic.h" 4bcef3411SDmitri Gribenko #include "clang/Basic/CharInfo.h" 5b2e5482bSDmitri Gribenko #include "llvm/ADT/StringExtras.h" 65188c4b9SDmitri Gribenko #include "llvm/ADT/StringSwitch.h" 79feeef40SDmitri Gribenko #include "llvm/Support/ConvertUTF.h" 85188c4b9SDmitri Gribenko #include "llvm/Support/ErrorHandling.h" 95188c4b9SDmitri Gribenko 105188c4b9SDmitri Gribenko namespace clang { 115188c4b9SDmitri Gribenko namespace comments { 125188c4b9SDmitri Gribenko 135188c4b9SDmitri Gribenko void Token::dump(const Lexer &L, const SourceManager &SM) const { 145188c4b9SDmitri Gribenko llvm::errs() << "comments::Token Kind=" << Kind << " "; 155188c4b9SDmitri Gribenko Loc.dump(SM); 165188c4b9SDmitri Gribenko llvm::errs() << " " << Length << " \"" << L.getSpelling(*this, SM) << "\"\n"; 175188c4b9SDmitri Gribenko } 185188c4b9SDmitri Gribenko 1974f4d028SDmitri Gribenko static inline bool isHTMLNamedCharacterReferenceCharacter(char C) { 20bcef3411SDmitri Gribenko return isLetter(C); 214586df76SDmitri Gribenko } 224586df76SDmitri Gribenko 2374f4d028SDmitri Gribenko static inline bool isHTMLDecimalCharacterReferenceCharacter(char C) { 24bcef3411SDmitri Gribenko return isDigit(C); 254586df76SDmitri Gribenko } 264586df76SDmitri Gribenko 2774f4d028SDmitri Gribenko static inline bool isHTMLHexCharacterReferenceCharacter(char C) { 28bcef3411SDmitri Gribenko return isHexDigit(C); 294586df76SDmitri Gribenko } 30107618a6SDmitri Gribenko 3174f4d028SDmitri Gribenko static inline StringRef convertCodePointToUTF8( 3274f4d028SDmitri Gribenko llvm::BumpPtrAllocator &Allocator, 3328800da1SDmitri Gribenko unsigned CodePoint) { 344586df76SDmitri Gribenko char *Resolved = Allocator.Allocate<char>(UNI_MAX_UTF8_BYTES_PER_CODE_POINT); 354586df76SDmitri Gribenko char *ResolvedPtr = Resolved; 369feeef40SDmitri Gribenko if (llvm::ConvertCodePointToUTF8(CodePoint, ResolvedPtr)) 374586df76SDmitri Gribenko return StringRef(Resolved, ResolvedPtr - Resolved); 384586df76SDmitri Gribenko else 394586df76SDmitri Gribenko return StringRef(); 404586df76SDmitri Gribenko } 414586df76SDmitri Gribenko 4274f4d028SDmitri Gribenko namespace { 4374f4d028SDmitri Gribenko 4428800da1SDmitri Gribenko #include "clang/AST/CommentHTMLTags.inc" 4528800da1SDmitri Gribenko #include "clang/AST/CommentHTMLNamedCharacterReferences.inc" 4628800da1SDmitri Gribenko 4728800da1SDmitri Gribenko } // unnamed namespace 487b3ae190SFariborz Jahanian 497b3ae190SFariborz Jahanian StringRef Lexer::resolveHTMLNamedCharacterReference(StringRef Name) const { 5028800da1SDmitri Gribenko // Fast path, first check a few most widely used named character references. 517b3ae190SFariborz Jahanian return llvm::StringSwitch<StringRef>(Name) 527b3ae190SFariborz Jahanian .Case("amp", "&") 537b3ae190SFariborz Jahanian .Case("lt", "<") 547b3ae190SFariborz Jahanian .Case("gt", ">") 557b3ae190SFariborz Jahanian .Case("quot", "\"") 567b3ae190SFariborz Jahanian .Case("apos", "\'") 5728800da1SDmitri Gribenko // Slow path. 5828800da1SDmitri Gribenko .Default(translateHTMLNamedCharacterReferenceToUTF8(Name)); 597b3ae190SFariborz Jahanian } 607b3ae190SFariborz Jahanian 617b3ae190SFariborz Jahanian StringRef Lexer::resolveHTMLDecimalCharacterReference(StringRef Name) const { 624586df76SDmitri Gribenko unsigned CodePoint = 0; 634586df76SDmitri Gribenko for (unsigned i = 0, e = Name.size(); i != e; ++i) { 647b3ae190SFariborz Jahanian assert(isHTMLDecimalCharacterReferenceCharacter(Name[i])); 657b3ae190SFariborz Jahanian CodePoint *= 10; 667b3ae190SFariborz Jahanian CodePoint += Name[i] - '0'; 674586df76SDmitri Gribenko } 6828800da1SDmitri Gribenko return convertCodePointToUTF8(Allocator, CodePoint); 6928800da1SDmitri Gribenko } 704586df76SDmitri Gribenko 7128800da1SDmitri Gribenko StringRef Lexer::resolveHTMLHexCharacterReference(StringRef Name) const { 7228800da1SDmitri Gribenko unsigned CodePoint = 0; 7328800da1SDmitri Gribenko for (unsigned i = 0, e = Name.size(); i != e; ++i) { 7428800da1SDmitri Gribenko CodePoint *= 16; 7528800da1SDmitri Gribenko const char C = Name[i]; 7628800da1SDmitri Gribenko assert(isHTMLHexCharacterReferenceCharacter(C)); 7728800da1SDmitri Gribenko CodePoint += llvm::hexDigitValue(C); 7828800da1SDmitri Gribenko } 7928800da1SDmitri Gribenko return convertCodePointToUTF8(Allocator, CodePoint); 804586df76SDmitri Gribenko } 814586df76SDmitri Gribenko 825188c4b9SDmitri Gribenko void Lexer::skipLineStartingDecorations() { 835188c4b9SDmitri Gribenko // This function should be called only for C comments 845188c4b9SDmitri Gribenko assert(CommentState == LCS_InsideCComment); 855188c4b9SDmitri Gribenko 865188c4b9SDmitri Gribenko if (BufferPtr == CommentEnd) 875188c4b9SDmitri Gribenko return; 885188c4b9SDmitri Gribenko 895188c4b9SDmitri Gribenko switch (*BufferPtr) { 905188c4b9SDmitri Gribenko case ' ': 915188c4b9SDmitri Gribenko case '\t': 925188c4b9SDmitri Gribenko case '\f': 935188c4b9SDmitri Gribenko case '\v': { 945188c4b9SDmitri Gribenko const char *NewBufferPtr = BufferPtr; 955188c4b9SDmitri Gribenko NewBufferPtr++; 965188c4b9SDmitri Gribenko if (NewBufferPtr == CommentEnd) 975188c4b9SDmitri Gribenko return; 985188c4b9SDmitri Gribenko 995188c4b9SDmitri Gribenko char C = *NewBufferPtr; 100bcef3411SDmitri Gribenko while (isHorizontalWhitespace(C)) { 1015188c4b9SDmitri Gribenko NewBufferPtr++; 1025188c4b9SDmitri Gribenko if (NewBufferPtr == CommentEnd) 1035188c4b9SDmitri Gribenko return; 1045188c4b9SDmitri Gribenko C = *NewBufferPtr; 1055188c4b9SDmitri Gribenko } 1065188c4b9SDmitri Gribenko if (C == '*') 1075188c4b9SDmitri Gribenko BufferPtr = NewBufferPtr + 1; 1085188c4b9SDmitri Gribenko break; 1095188c4b9SDmitri Gribenko } 1105188c4b9SDmitri Gribenko case '*': 1115188c4b9SDmitri Gribenko BufferPtr++; 1125188c4b9SDmitri Gribenko break; 1135188c4b9SDmitri Gribenko } 1145188c4b9SDmitri Gribenko } 1155188c4b9SDmitri Gribenko 1165188c4b9SDmitri Gribenko namespace { 117ec92531cSDmitri Gribenko /// Returns pointer to the first newline character in the string. 1185188c4b9SDmitri Gribenko const char *findNewline(const char *BufferPtr, const char *BufferEnd) { 1195188c4b9SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 120bcef3411SDmitri Gribenko if (isVerticalWhitespace(*BufferPtr)) 1215188c4b9SDmitri Gribenko return BufferPtr; 1225188c4b9SDmitri Gribenko } 1235188c4b9SDmitri Gribenko return BufferEnd; 1245188c4b9SDmitri Gribenko } 1255188c4b9SDmitri Gribenko 1265188c4b9SDmitri Gribenko const char *skipNewline(const char *BufferPtr, const char *BufferEnd) { 1275188c4b9SDmitri Gribenko if (BufferPtr == BufferEnd) 1285188c4b9SDmitri Gribenko return BufferPtr; 1295188c4b9SDmitri Gribenko 1305188c4b9SDmitri Gribenko if (*BufferPtr == '\n') 1315188c4b9SDmitri Gribenko BufferPtr++; 1325188c4b9SDmitri Gribenko else { 1335188c4b9SDmitri Gribenko assert(*BufferPtr == '\r'); 1345188c4b9SDmitri Gribenko BufferPtr++; 1355188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '\n') 1365188c4b9SDmitri Gribenko BufferPtr++; 1375188c4b9SDmitri Gribenko } 1385188c4b9SDmitri Gribenko return BufferPtr; 1395188c4b9SDmitri Gribenko } 1405188c4b9SDmitri Gribenko 1414586df76SDmitri Gribenko const char *skipNamedCharacterReference(const char *BufferPtr, 1424586df76SDmitri Gribenko const char *BufferEnd) { 1434586df76SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 1444586df76SDmitri Gribenko if (!isHTMLNamedCharacterReferenceCharacter(*BufferPtr)) 1454586df76SDmitri Gribenko return BufferPtr; 1464586df76SDmitri Gribenko } 1474586df76SDmitri Gribenko return BufferEnd; 1484586df76SDmitri Gribenko } 1494586df76SDmitri Gribenko 1504586df76SDmitri Gribenko const char *skipDecimalCharacterReference(const char *BufferPtr, 1514586df76SDmitri Gribenko const char *BufferEnd) { 1524586df76SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 1534586df76SDmitri Gribenko if (!isHTMLDecimalCharacterReferenceCharacter(*BufferPtr)) 1544586df76SDmitri Gribenko return BufferPtr; 1554586df76SDmitri Gribenko } 1564586df76SDmitri Gribenko return BufferEnd; 1574586df76SDmitri Gribenko } 1584586df76SDmitri Gribenko 1594586df76SDmitri Gribenko const char *skipHexCharacterReference(const char *BufferPtr, 1604586df76SDmitri Gribenko const char *BufferEnd) { 1614586df76SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 1624586df76SDmitri Gribenko if (!isHTMLHexCharacterReferenceCharacter(*BufferPtr)) 1634586df76SDmitri Gribenko return BufferPtr; 1644586df76SDmitri Gribenko } 1654586df76SDmitri Gribenko return BufferEnd; 1664586df76SDmitri Gribenko } 1674586df76SDmitri Gribenko 16817709ae8SDmitri Gribenko bool isHTMLIdentifierStartingCharacter(char C) { 169bcef3411SDmitri Gribenko return isLetter(C); 17017709ae8SDmitri Gribenko } 17117709ae8SDmitri Gribenko 1725188c4b9SDmitri Gribenko bool isHTMLIdentifierCharacter(char C) { 173bcef3411SDmitri Gribenko return isAlphanumeric(C); 1745188c4b9SDmitri Gribenko } 1755188c4b9SDmitri Gribenko 1765188c4b9SDmitri Gribenko const char *skipHTMLIdentifier(const char *BufferPtr, const char *BufferEnd) { 1775188c4b9SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 1785188c4b9SDmitri Gribenko if (!isHTMLIdentifierCharacter(*BufferPtr)) 1795188c4b9SDmitri Gribenko return BufferPtr; 1805188c4b9SDmitri Gribenko } 1815188c4b9SDmitri Gribenko return BufferEnd; 1825188c4b9SDmitri Gribenko } 1835188c4b9SDmitri Gribenko 1845188c4b9SDmitri Gribenko /// Skip HTML string quoted in single or double quotes. Escaping quotes inside 1855188c4b9SDmitri Gribenko /// string allowed. 1865188c4b9SDmitri Gribenko /// 1875188c4b9SDmitri Gribenko /// Returns pointer to closing quote. 1885188c4b9SDmitri Gribenko const char *skipHTMLQuotedString(const char *BufferPtr, const char *BufferEnd) 1895188c4b9SDmitri Gribenko { 1905188c4b9SDmitri Gribenko const char Quote = *BufferPtr; 1915188c4b9SDmitri Gribenko assert(Quote == '\"' || Quote == '\''); 1925188c4b9SDmitri Gribenko 1935188c4b9SDmitri Gribenko BufferPtr++; 1945188c4b9SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 1955188c4b9SDmitri Gribenko const char C = *BufferPtr; 1965188c4b9SDmitri Gribenko if (C == Quote && BufferPtr[-1] != '\\') 1975188c4b9SDmitri Gribenko return BufferPtr; 1985188c4b9SDmitri Gribenko } 1995188c4b9SDmitri Gribenko return BufferEnd; 2005188c4b9SDmitri Gribenko } 2015188c4b9SDmitri Gribenko 2025188c4b9SDmitri Gribenko const char *skipWhitespace(const char *BufferPtr, const char *BufferEnd) { 2035188c4b9SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 2045188c4b9SDmitri Gribenko if (!isWhitespace(*BufferPtr)) 2055188c4b9SDmitri Gribenko return BufferPtr; 2065188c4b9SDmitri Gribenko } 2075188c4b9SDmitri Gribenko return BufferEnd; 2085188c4b9SDmitri Gribenko } 2095188c4b9SDmitri Gribenko 210e4a3997dSDmitri Gribenko bool isWhitespace(const char *BufferPtr, const char *BufferEnd) { 211e4a3997dSDmitri Gribenko return skipWhitespace(BufferPtr, BufferEnd) == BufferEnd; 212e4a3997dSDmitri Gribenko } 213e4a3997dSDmitri Gribenko 214ad45ad6fSDmitri Gribenko bool isCommandNameStartCharacter(char C) { 215bcef3411SDmitri Gribenko return isLetter(C); 216ad45ad6fSDmitri Gribenko } 217ad45ad6fSDmitri Gribenko 2185188c4b9SDmitri Gribenko bool isCommandNameCharacter(char C) { 219bcef3411SDmitri Gribenko return isAlphanumeric(C); 2205188c4b9SDmitri Gribenko } 2215188c4b9SDmitri Gribenko 2225188c4b9SDmitri Gribenko const char *skipCommandName(const char *BufferPtr, const char *BufferEnd) { 2235188c4b9SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 2245188c4b9SDmitri Gribenko if (!isCommandNameCharacter(*BufferPtr)) 2255188c4b9SDmitri Gribenko return BufferPtr; 2265188c4b9SDmitri Gribenko } 2275188c4b9SDmitri Gribenko return BufferEnd; 2285188c4b9SDmitri Gribenko } 2295188c4b9SDmitri Gribenko 2305188c4b9SDmitri Gribenko /// Return the one past end pointer for BCPL comments. 2315188c4b9SDmitri Gribenko /// Handles newlines escaped with backslash or trigraph for backslahs. 2325188c4b9SDmitri Gribenko const char *findBCPLCommentEnd(const char *BufferPtr, const char *BufferEnd) { 2335188c4b9SDmitri Gribenko const char *CurPtr = BufferPtr; 2345188c4b9SDmitri Gribenko while (CurPtr != BufferEnd) { 235bcef3411SDmitri Gribenko while (!isVerticalWhitespace(*CurPtr)) { 2365188c4b9SDmitri Gribenko CurPtr++; 2375188c4b9SDmitri Gribenko if (CurPtr == BufferEnd) 2385188c4b9SDmitri Gribenko return BufferEnd; 2395188c4b9SDmitri Gribenko } 2405188c4b9SDmitri Gribenko // We found a newline, check if it is escaped. 2415188c4b9SDmitri Gribenko const char *EscapePtr = CurPtr - 1; 2425188c4b9SDmitri Gribenko while(isHorizontalWhitespace(*EscapePtr)) 2435188c4b9SDmitri Gribenko EscapePtr--; 2445188c4b9SDmitri Gribenko 2455188c4b9SDmitri Gribenko if (*EscapePtr == '\\' || 2465188c4b9SDmitri Gribenko (EscapePtr - 2 >= BufferPtr && EscapePtr[0] == '/' && 2475188c4b9SDmitri Gribenko EscapePtr[-1] == '?' && EscapePtr[-2] == '?')) { 2485188c4b9SDmitri Gribenko // We found an escaped newline. 2495188c4b9SDmitri Gribenko CurPtr = skipNewline(CurPtr, BufferEnd); 2505188c4b9SDmitri Gribenko } else 2515188c4b9SDmitri Gribenko return CurPtr; // Not an escaped newline. 2525188c4b9SDmitri Gribenko } 2535188c4b9SDmitri Gribenko return BufferEnd; 2545188c4b9SDmitri Gribenko } 2555188c4b9SDmitri Gribenko 2565188c4b9SDmitri Gribenko /// Return the one past end pointer for C comments. 2575188c4b9SDmitri Gribenko /// Very dumb, does not handle escaped newlines or trigraphs. 2585188c4b9SDmitri Gribenko const char *findCCommentEnd(const char *BufferPtr, const char *BufferEnd) { 2595188c4b9SDmitri Gribenko for ( ; BufferPtr != BufferEnd; ++BufferPtr) { 2605188c4b9SDmitri Gribenko if (*BufferPtr == '*') { 2615188c4b9SDmitri Gribenko assert(BufferPtr + 1 != BufferEnd); 2625188c4b9SDmitri Gribenko if (*(BufferPtr + 1) == '/') 2635188c4b9SDmitri Gribenko return BufferPtr; 2645188c4b9SDmitri Gribenko } 2655188c4b9SDmitri Gribenko } 2665188c4b9SDmitri Gribenko llvm_unreachable("buffer end hit before '*/' was seen"); 2675188c4b9SDmitri Gribenko } 2686c7a1666SFariborz Jahanian 2695188c4b9SDmitri Gribenko } // unnamed namespace 2705188c4b9SDmitri Gribenko 2715188c4b9SDmitri Gribenko void Lexer::lexCommentText(Token &T) { 2725188c4b9SDmitri Gribenko assert(CommentState == LCS_InsideBCPLComment || 2735188c4b9SDmitri Gribenko CommentState == LCS_InsideCComment); 2745188c4b9SDmitri Gribenko 2755188c4b9SDmitri Gribenko switch (State) { 2765188c4b9SDmitri Gribenko case LS_Normal: 2775188c4b9SDmitri Gribenko break; 2785188c4b9SDmitri Gribenko case LS_VerbatimBlockFirstLine: 2795188c4b9SDmitri Gribenko lexVerbatimBlockFirstLine(T); 2805188c4b9SDmitri Gribenko return; 2815188c4b9SDmitri Gribenko case LS_VerbatimBlockBody: 2825188c4b9SDmitri Gribenko lexVerbatimBlockBody(T); 2835188c4b9SDmitri Gribenko return; 2841669f702SDmitri Gribenko case LS_VerbatimLineText: 2851669f702SDmitri Gribenko lexVerbatimLineText(T); 2861669f702SDmitri Gribenko return; 287e00ffc7bSDmitri Gribenko case LS_HTMLStartTag: 288e00ffc7bSDmitri Gribenko lexHTMLStartTag(T); 2895188c4b9SDmitri Gribenko return; 290e00ffc7bSDmitri Gribenko case LS_HTMLEndTag: 291e00ffc7bSDmitri Gribenko lexHTMLEndTag(T); 292ec92531cSDmitri Gribenko return; 2935188c4b9SDmitri Gribenko } 2945188c4b9SDmitri Gribenko 2955188c4b9SDmitri Gribenko assert(State == LS_Normal); 2965188c4b9SDmitri Gribenko 2975188c4b9SDmitri Gribenko const char *TokenPtr = BufferPtr; 2985188c4b9SDmitri Gribenko assert(TokenPtr < CommentEnd); 2995188c4b9SDmitri Gribenko while (TokenPtr != CommentEnd) { 3005188c4b9SDmitri Gribenko switch(*TokenPtr) { 3015188c4b9SDmitri Gribenko case '\\': 3025188c4b9SDmitri Gribenko case '@': { 303bcf7f4d3SDmitri Gribenko // Commands that start with a backslash and commands that start with 304bcf7f4d3SDmitri Gribenko // 'at' have equivalent semantics. But we keep information about the 305bcf7f4d3SDmitri Gribenko // exact syntax in AST for comments. 306bcf7f4d3SDmitri Gribenko tok::TokenKind CommandKind = 307bcf7f4d3SDmitri Gribenko (*TokenPtr == '@') ? tok::at_command : tok::backslash_command; 3085188c4b9SDmitri Gribenko TokenPtr++; 3095188c4b9SDmitri Gribenko if (TokenPtr == CommentEnd) { 3104586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3115188c4b9SDmitri Gribenko return; 3125188c4b9SDmitri Gribenko } 3135188c4b9SDmitri Gribenko char C = *TokenPtr; 3145188c4b9SDmitri Gribenko switch (C) { 3155188c4b9SDmitri Gribenko default: 3165188c4b9SDmitri Gribenko break; 3175188c4b9SDmitri Gribenko 3185188c4b9SDmitri Gribenko case '\\': case '@': case '&': case '$': 3195188c4b9SDmitri Gribenko case '#': case '<': case '>': case '%': 3205188c4b9SDmitri Gribenko case '\"': case '.': case ':': 3215188c4b9SDmitri Gribenko // This is one of \\ \@ \& \$ etc escape sequences. 3225188c4b9SDmitri Gribenko TokenPtr++; 3235188c4b9SDmitri Gribenko if (C == ':' && TokenPtr != CommentEnd && *TokenPtr == ':') { 3245188c4b9SDmitri Gribenko // This is the \:: escape sequence. 3255188c4b9SDmitri Gribenko TokenPtr++; 3265188c4b9SDmitri Gribenko } 32760ddd8a1SDmitri Gribenko StringRef UnescapedText(BufferPtr + 1, TokenPtr - (BufferPtr + 1)); 3285188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::text); 32960ddd8a1SDmitri Gribenko T.setText(UnescapedText); 3305188c4b9SDmitri Gribenko return; 3315188c4b9SDmitri Gribenko } 3325188c4b9SDmitri Gribenko 3335188c4b9SDmitri Gribenko // Don't make zero-length commands. 334ad45ad6fSDmitri Gribenko if (!isCommandNameStartCharacter(*TokenPtr)) { 3354586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3365188c4b9SDmitri Gribenko return; 3375188c4b9SDmitri Gribenko } 3385188c4b9SDmitri Gribenko 3395188c4b9SDmitri Gribenko TokenPtr = skipCommandName(TokenPtr, CommentEnd); 3405188c4b9SDmitri Gribenko unsigned Length = TokenPtr - (BufferPtr + 1); 3415188c4b9SDmitri Gribenko 3425188c4b9SDmitri Gribenko // Hardcoded support for lexing LaTeX formula commands 3435188c4b9SDmitri Gribenko // \f$ \f[ \f] \f{ \f} as a single command. 3445188c4b9SDmitri Gribenko if (Length == 1 && TokenPtr[-1] == 'f' && TokenPtr != CommentEnd) { 3455188c4b9SDmitri Gribenko C = *TokenPtr; 3465188c4b9SDmitri Gribenko if (C == '$' || C == '[' || C == ']' || C == '{' || C == '}') { 3475188c4b9SDmitri Gribenko TokenPtr++; 3485188c4b9SDmitri Gribenko Length++; 3495188c4b9SDmitri Gribenko } 3505188c4b9SDmitri Gribenko } 3515188c4b9SDmitri Gribenko 3525188c4b9SDmitri Gribenko const StringRef CommandName(BufferPtr + 1, Length); 3535188c4b9SDmitri Gribenko 3547acbf00fSDmitri Gribenko const CommandInfo *Info = Traits.getCommandInfoOrNULL(CommandName); 3557acbf00fSDmitri Gribenko if (!Info) { 356ebb262f7SFariborz Jahanian if ((Info = Traits.getTypoCorrectCommandInfo(CommandName))) { 3576c7a1666SFariborz Jahanian StringRef CorrectedName = Info->Name; 358*2dece574SBenjamin Kramer SourceLocation Loc = getSourceLocation(BufferPtr); 359*2dece574SBenjamin Kramer SourceRange CommandRange(Loc.getLocWithOffset(1), 360*2dece574SBenjamin Kramer getSourceLocation(TokenPtr)); 361*2dece574SBenjamin Kramer Diag(Loc, diag::warn_correct_comment_command_name) 3626c7a1666SFariborz Jahanian << CommandName << CorrectedName 3636c7a1666SFariborz Jahanian << FixItHint::CreateReplacement(CommandRange, CorrectedName); 3646c7a1666SFariborz Jahanian } else { 365*2dece574SBenjamin Kramer formTokenWithChars(T, TokenPtr, tok::unknown_command); 366*2dece574SBenjamin Kramer T.setUnknownCommandName(CommandName); 3676738e430SFariborz Jahanian Diag(T.getLocation(), diag::warn_unknown_comment_command_name); 3685188c4b9SDmitri Gribenko return; 3695188c4b9SDmitri Gribenko } 3706c7a1666SFariborz Jahanian } 3717acbf00fSDmitri Gribenko if (Info->IsVerbatimBlockCommand) { 3727acbf00fSDmitri Gribenko setupAndLexVerbatimBlock(T, TokenPtr, *BufferPtr, Info); 3737acbf00fSDmitri Gribenko return; 3747acbf00fSDmitri Gribenko } 3757acbf00fSDmitri Gribenko if (Info->IsVerbatimLineCommand) { 3767acbf00fSDmitri Gribenko setupAndLexVerbatimLine(T, TokenPtr, Info); 3775188c4b9SDmitri Gribenko return; 3785188c4b9SDmitri Gribenko } 379bcf7f4d3SDmitri Gribenko formTokenWithChars(T, TokenPtr, CommandKind); 3807acbf00fSDmitri Gribenko T.setCommandID(Info->getID()); 3815188c4b9SDmitri Gribenko return; 3825188c4b9SDmitri Gribenko } 3835188c4b9SDmitri Gribenko 3844586df76SDmitri Gribenko case '&': 3854586df76SDmitri Gribenko lexHTMLCharacterReference(T); 3864586df76SDmitri Gribenko return; 3874586df76SDmitri Gribenko 3885188c4b9SDmitri Gribenko case '<': { 3895188c4b9SDmitri Gribenko TokenPtr++; 3905188c4b9SDmitri Gribenko if (TokenPtr == CommentEnd) { 3914586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3925188c4b9SDmitri Gribenko return; 3935188c4b9SDmitri Gribenko } 3945188c4b9SDmitri Gribenko const char C = *TokenPtr; 39517709ae8SDmitri Gribenko if (isHTMLIdentifierStartingCharacter(C)) 396e00ffc7bSDmitri Gribenko setupAndLexHTMLStartTag(T); 3975188c4b9SDmitri Gribenko else if (C == '/') 398e00ffc7bSDmitri Gribenko setupAndLexHTMLEndTag(T); 3994586df76SDmitri Gribenko else 4004586df76SDmitri Gribenko formTextToken(T, TokenPtr); 4014586df76SDmitri Gribenko 4025188c4b9SDmitri Gribenko return; 4035188c4b9SDmitri Gribenko } 4045188c4b9SDmitri Gribenko 4055188c4b9SDmitri Gribenko case '\n': 4065188c4b9SDmitri Gribenko case '\r': 4075188c4b9SDmitri Gribenko TokenPtr = skipNewline(TokenPtr, CommentEnd); 4085188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::newline); 4095188c4b9SDmitri Gribenko 4105188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) 4115188c4b9SDmitri Gribenko skipLineStartingDecorations(); 4125188c4b9SDmitri Gribenko return; 4135188c4b9SDmitri Gribenko 4145188c4b9SDmitri Gribenko default: { 41510af67a9SDmitri Gribenko size_t End = StringRef(TokenPtr, CommentEnd - TokenPtr). 41610af67a9SDmitri Gribenko find_first_of("\n\r\\@&<"); 41710af67a9SDmitri Gribenko if (End != StringRef::npos) 41810af67a9SDmitri Gribenko TokenPtr += End; 41910af67a9SDmitri Gribenko else 42010af67a9SDmitri Gribenko TokenPtr = CommentEnd; 4214586df76SDmitri Gribenko formTextToken(T, TokenPtr); 4225188c4b9SDmitri Gribenko return; 4235188c4b9SDmitri Gribenko } 4245188c4b9SDmitri Gribenko } 4255188c4b9SDmitri Gribenko } 4265188c4b9SDmitri Gribenko } 4275188c4b9SDmitri Gribenko 4285188c4b9SDmitri Gribenko void Lexer::setupAndLexVerbatimBlock(Token &T, 4295188c4b9SDmitri Gribenko const char *TextBegin, 4307acbf00fSDmitri Gribenko char Marker, const CommandInfo *Info) { 4317acbf00fSDmitri Gribenko assert(Info->IsVerbatimBlockCommand); 4327acbf00fSDmitri Gribenko 4335188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.clear(); 4345188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.append(Marker == '\\' ? "\\" : "@"); 4357acbf00fSDmitri Gribenko VerbatimBlockEndCommandName.append(Info->EndCommandName); 4365188c4b9SDmitri Gribenko 4375188c4b9SDmitri Gribenko formTokenWithChars(T, TextBegin, tok::verbatim_block_begin); 4387acbf00fSDmitri Gribenko T.setVerbatimBlockID(Info->getID()); 4395188c4b9SDmitri Gribenko 440ec92531cSDmitri Gribenko // If there is a newline following the verbatim opening command, skip the 441ec92531cSDmitri Gribenko // newline so that we don't create an tok::verbatim_block_line with empty 442ec92531cSDmitri Gribenko // text content. 443bcef3411SDmitri Gribenko if (BufferPtr != CommentEnd && 444bcef3411SDmitri Gribenko isVerticalWhitespace(*BufferPtr)) { 445ec92531cSDmitri Gribenko BufferPtr = skipNewline(BufferPtr, CommentEnd); 446ec92531cSDmitri Gribenko State = LS_VerbatimBlockBody; 447ec92531cSDmitri Gribenko return; 448ec92531cSDmitri Gribenko } 449ec92531cSDmitri Gribenko 4505188c4b9SDmitri Gribenko State = LS_VerbatimBlockFirstLine; 4515188c4b9SDmitri Gribenko } 4525188c4b9SDmitri Gribenko 4535188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockFirstLine(Token &T) { 454e4a3997dSDmitri Gribenko again: 4555188c4b9SDmitri Gribenko assert(BufferPtr < CommentEnd); 4565188c4b9SDmitri Gribenko 4575188c4b9SDmitri Gribenko // FIXME: It would be better to scan the text once, finding either the block 4585188c4b9SDmitri Gribenko // end command or newline. 4595188c4b9SDmitri Gribenko // 4605188c4b9SDmitri Gribenko // Extract current line. 4615188c4b9SDmitri Gribenko const char *Newline = findNewline(BufferPtr, CommentEnd); 4625188c4b9SDmitri Gribenko StringRef Line(BufferPtr, Newline - BufferPtr); 4635188c4b9SDmitri Gribenko 4645188c4b9SDmitri Gribenko // Look for end command in current line. 4655188c4b9SDmitri Gribenko size_t Pos = Line.find(VerbatimBlockEndCommandName); 466ec92531cSDmitri Gribenko const char *TextEnd; 4675188c4b9SDmitri Gribenko const char *NextLine; 4685188c4b9SDmitri Gribenko if (Pos == StringRef::npos) { 4695188c4b9SDmitri Gribenko // Current line is completely verbatim. 470ec92531cSDmitri Gribenko TextEnd = Newline; 4715188c4b9SDmitri Gribenko NextLine = skipNewline(Newline, CommentEnd); 4725188c4b9SDmitri Gribenko } else if (Pos == 0) { 4735188c4b9SDmitri Gribenko // Current line contains just an end command. 4745188c4b9SDmitri Gribenko const char *End = BufferPtr + VerbatimBlockEndCommandName.size(); 47560ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, End - (BufferPtr + 1)); 4765188c4b9SDmitri Gribenko formTokenWithChars(T, End, tok::verbatim_block_end); 4777acbf00fSDmitri Gribenko T.setVerbatimBlockID(Traits.getCommandInfo(Name)->getID()); 4785188c4b9SDmitri Gribenko State = LS_Normal; 4795188c4b9SDmitri Gribenko return; 4805188c4b9SDmitri Gribenko } else { 4815188c4b9SDmitri Gribenko // There is some text, followed by end command. Extract text first. 482ec92531cSDmitri Gribenko TextEnd = BufferPtr + Pos; 483ec92531cSDmitri Gribenko NextLine = TextEnd; 484e4a3997dSDmitri Gribenko // If there is only whitespace before end command, skip whitespace. 485e4a3997dSDmitri Gribenko if (isWhitespace(BufferPtr, TextEnd)) { 486e4a3997dSDmitri Gribenko BufferPtr = TextEnd; 487e4a3997dSDmitri Gribenko goto again; 488e4a3997dSDmitri Gribenko } 4895188c4b9SDmitri Gribenko } 4905188c4b9SDmitri Gribenko 491ec92531cSDmitri Gribenko StringRef Text(BufferPtr, TextEnd - BufferPtr); 4925188c4b9SDmitri Gribenko formTokenWithChars(T, NextLine, tok::verbatim_block_line); 49360ddd8a1SDmitri Gribenko T.setVerbatimBlockText(Text); 4945188c4b9SDmitri Gribenko 4955188c4b9SDmitri Gribenko State = LS_VerbatimBlockBody; 4965188c4b9SDmitri Gribenko } 4975188c4b9SDmitri Gribenko 4985188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockBody(Token &T) { 4995188c4b9SDmitri Gribenko assert(State == LS_VerbatimBlockBody); 5005188c4b9SDmitri Gribenko 5015188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) 5025188c4b9SDmitri Gribenko skipLineStartingDecorations(); 5035188c4b9SDmitri Gribenko 5045188c4b9SDmitri Gribenko lexVerbatimBlockFirstLine(T); 5055188c4b9SDmitri Gribenko } 5065188c4b9SDmitri Gribenko 5077acbf00fSDmitri Gribenko void Lexer::setupAndLexVerbatimLine(Token &T, const char *TextBegin, 5087acbf00fSDmitri Gribenko const CommandInfo *Info) { 5097acbf00fSDmitri Gribenko assert(Info->IsVerbatimLineCommand); 5101669f702SDmitri Gribenko formTokenWithChars(T, TextBegin, tok::verbatim_line_name); 5117acbf00fSDmitri Gribenko T.setVerbatimLineID(Info->getID()); 5121669f702SDmitri Gribenko 5131669f702SDmitri Gribenko State = LS_VerbatimLineText; 5141669f702SDmitri Gribenko } 5151669f702SDmitri Gribenko 5161669f702SDmitri Gribenko void Lexer::lexVerbatimLineText(Token &T) { 5171669f702SDmitri Gribenko assert(State == LS_VerbatimLineText); 5181669f702SDmitri Gribenko 5195188c4b9SDmitri Gribenko // Extract current line. 5205188c4b9SDmitri Gribenko const char *Newline = findNewline(BufferPtr, CommentEnd); 5211669f702SDmitri Gribenko const StringRef Text(BufferPtr, Newline - BufferPtr); 5221669f702SDmitri Gribenko formTokenWithChars(T, Newline, tok::verbatim_line_text); 5235188c4b9SDmitri Gribenko T.setVerbatimLineText(Text); 5241669f702SDmitri Gribenko 5251669f702SDmitri Gribenko State = LS_Normal; 5265188c4b9SDmitri Gribenko } 5275188c4b9SDmitri Gribenko 5284586df76SDmitri Gribenko void Lexer::lexHTMLCharacterReference(Token &T) { 5294586df76SDmitri Gribenko const char *TokenPtr = BufferPtr; 5304586df76SDmitri Gribenko assert(*TokenPtr == '&'); 5314586df76SDmitri Gribenko TokenPtr++; 5324586df76SDmitri Gribenko if (TokenPtr == CommentEnd) { 5334586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5344586df76SDmitri Gribenko return; 5354586df76SDmitri Gribenko } 5364586df76SDmitri Gribenko const char *NamePtr; 5374586df76SDmitri Gribenko bool isNamed = false; 5384586df76SDmitri Gribenko bool isDecimal = false; 5394586df76SDmitri Gribenko char C = *TokenPtr; 5404586df76SDmitri Gribenko if (isHTMLNamedCharacterReferenceCharacter(C)) { 5414586df76SDmitri Gribenko NamePtr = TokenPtr; 5424586df76SDmitri Gribenko TokenPtr = skipNamedCharacterReference(TokenPtr, CommentEnd); 5434586df76SDmitri Gribenko isNamed = true; 5444586df76SDmitri Gribenko } else if (C == '#') { 5454586df76SDmitri Gribenko TokenPtr++; 5464586df76SDmitri Gribenko if (TokenPtr == CommentEnd) { 5474586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5484586df76SDmitri Gribenko return; 5494586df76SDmitri Gribenko } 5504586df76SDmitri Gribenko C = *TokenPtr; 5514586df76SDmitri Gribenko if (isHTMLDecimalCharacterReferenceCharacter(C)) { 5524586df76SDmitri Gribenko NamePtr = TokenPtr; 5534586df76SDmitri Gribenko TokenPtr = skipDecimalCharacterReference(TokenPtr, CommentEnd); 5544586df76SDmitri Gribenko isDecimal = true; 5554586df76SDmitri Gribenko } else if (C == 'x' || C == 'X') { 5564586df76SDmitri Gribenko TokenPtr++; 5574586df76SDmitri Gribenko NamePtr = TokenPtr; 5584586df76SDmitri Gribenko TokenPtr = skipHexCharacterReference(TokenPtr, CommentEnd); 5594586df76SDmitri Gribenko } else { 5604586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5614586df76SDmitri Gribenko return; 5624586df76SDmitri Gribenko } 5634586df76SDmitri Gribenko } else { 5644586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5654586df76SDmitri Gribenko return; 5664586df76SDmitri Gribenko } 5674586df76SDmitri Gribenko if (NamePtr == TokenPtr || TokenPtr == CommentEnd || 5684586df76SDmitri Gribenko *TokenPtr != ';') { 5694586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5704586df76SDmitri Gribenko return; 5714586df76SDmitri Gribenko } 5724586df76SDmitri Gribenko StringRef Name(NamePtr, TokenPtr - NamePtr); 5734586df76SDmitri Gribenko TokenPtr++; // Skip semicolon. 5744586df76SDmitri Gribenko StringRef Resolved; 57528800da1SDmitri Gribenko if (isNamed) 5764586df76SDmitri Gribenko Resolved = resolveHTMLNamedCharacterReference(Name); 5774586df76SDmitri Gribenko else if (isDecimal) 5784586df76SDmitri Gribenko Resolved = resolveHTMLDecimalCharacterReference(Name); 5794586df76SDmitri Gribenko else 5804586df76SDmitri Gribenko Resolved = resolveHTMLHexCharacterReference(Name); 5814586df76SDmitri Gribenko 5824586df76SDmitri Gribenko if (Resolved.empty()) { 5834586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5844586df76SDmitri Gribenko return; 5854586df76SDmitri Gribenko } 5864586df76SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::text); 5874586df76SDmitri Gribenko T.setText(Resolved); 5884586df76SDmitri Gribenko return; 5894586df76SDmitri Gribenko } 5904586df76SDmitri Gribenko 591e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLStartTag(Token &T) { 59217709ae8SDmitri Gribenko assert(BufferPtr[0] == '<' && 59317709ae8SDmitri Gribenko isHTMLIdentifierStartingCharacter(BufferPtr[1])); 5945188c4b9SDmitri Gribenko const char *TagNameEnd = skipHTMLIdentifier(BufferPtr + 2, CommentEnd); 59560ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, TagNameEnd - (BufferPtr + 1)); 596107618a6SDmitri Gribenko if (!isHTMLTagName(Name)) { 597107618a6SDmitri Gribenko formTextToken(T, TagNameEnd); 598107618a6SDmitri Gribenko return; 599107618a6SDmitri Gribenko } 600107618a6SDmitri Gribenko 601e00ffc7bSDmitri Gribenko formTokenWithChars(T, TagNameEnd, tok::html_start_tag); 602e00ffc7bSDmitri Gribenko T.setHTMLTagStartName(Name); 6035188c4b9SDmitri Gribenko 6045188c4b9SDmitri Gribenko BufferPtr = skipWhitespace(BufferPtr, CommentEnd); 6055188c4b9SDmitri Gribenko 60617709ae8SDmitri Gribenko const char C = *BufferPtr; 60717709ae8SDmitri Gribenko if (BufferPtr != CommentEnd && 608f26054f0SDmitri Gribenko (C == '>' || C == '/' || isHTMLIdentifierStartingCharacter(C))) 609e00ffc7bSDmitri Gribenko State = LS_HTMLStartTag; 6105188c4b9SDmitri Gribenko } 6115188c4b9SDmitri Gribenko 612e00ffc7bSDmitri Gribenko void Lexer::lexHTMLStartTag(Token &T) { 613e00ffc7bSDmitri Gribenko assert(State == LS_HTMLStartTag); 6145188c4b9SDmitri Gribenko 6155188c4b9SDmitri Gribenko const char *TokenPtr = BufferPtr; 6165188c4b9SDmitri Gribenko char C = *TokenPtr; 6175188c4b9SDmitri Gribenko if (isHTMLIdentifierCharacter(C)) { 6185188c4b9SDmitri Gribenko TokenPtr = skipHTMLIdentifier(TokenPtr, CommentEnd); 61960ddd8a1SDmitri Gribenko StringRef Ident(BufferPtr, TokenPtr - BufferPtr); 6205188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_ident); 62160ddd8a1SDmitri Gribenko T.setHTMLIdent(Ident); 6225188c4b9SDmitri Gribenko } else { 6235188c4b9SDmitri Gribenko switch (C) { 6245188c4b9SDmitri Gribenko case '=': 6255188c4b9SDmitri Gribenko TokenPtr++; 6265188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_equals); 6275188c4b9SDmitri Gribenko break; 6285188c4b9SDmitri Gribenko case '\"': 6295188c4b9SDmitri Gribenko case '\'': { 6305188c4b9SDmitri Gribenko const char *OpenQuote = TokenPtr; 6315188c4b9SDmitri Gribenko TokenPtr = skipHTMLQuotedString(TokenPtr, CommentEnd); 6325188c4b9SDmitri Gribenko const char *ClosingQuote = TokenPtr; 6335188c4b9SDmitri Gribenko if (TokenPtr != CommentEnd) // Skip closing quote. 6345188c4b9SDmitri Gribenko TokenPtr++; 6355188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_quoted_string); 6365188c4b9SDmitri Gribenko T.setHTMLQuotedString(StringRef(OpenQuote + 1, 6375188c4b9SDmitri Gribenko ClosingQuote - (OpenQuote + 1))); 6385188c4b9SDmitri Gribenko break; 6395188c4b9SDmitri Gribenko } 6405188c4b9SDmitri Gribenko case '>': 6415188c4b9SDmitri Gribenko TokenPtr++; 6425188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_greater); 64317709ae8SDmitri Gribenko State = LS_Normal; 64417709ae8SDmitri Gribenko return; 645f26054f0SDmitri Gribenko case '/': 646f26054f0SDmitri Gribenko TokenPtr++; 647f26054f0SDmitri Gribenko if (TokenPtr != CommentEnd && *TokenPtr == '>') { 648f26054f0SDmitri Gribenko TokenPtr++; 649f26054f0SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_slash_greater); 6504586df76SDmitri Gribenko } else 6514586df76SDmitri Gribenko formTextToken(T, TokenPtr); 6524586df76SDmitri Gribenko 653f26054f0SDmitri Gribenko State = LS_Normal; 654f26054f0SDmitri Gribenko return; 6555188c4b9SDmitri Gribenko } 6565188c4b9SDmitri Gribenko } 6575188c4b9SDmitri Gribenko 6585188c4b9SDmitri Gribenko // Now look ahead and return to normal state if we don't see any HTML tokens 6595188c4b9SDmitri Gribenko // ahead. 6605188c4b9SDmitri Gribenko BufferPtr = skipWhitespace(BufferPtr, CommentEnd); 6615188c4b9SDmitri Gribenko if (BufferPtr == CommentEnd) { 6625188c4b9SDmitri Gribenko State = LS_Normal; 6635188c4b9SDmitri Gribenko return; 6645188c4b9SDmitri Gribenko } 6655188c4b9SDmitri Gribenko 6665188c4b9SDmitri Gribenko C = *BufferPtr; 66717709ae8SDmitri Gribenko if (!isHTMLIdentifierStartingCharacter(C) && 6685188c4b9SDmitri Gribenko C != '=' && C != '\"' && C != '\'' && C != '>') { 6695188c4b9SDmitri Gribenko State = LS_Normal; 6705188c4b9SDmitri Gribenko return; 6715188c4b9SDmitri Gribenko } 6725188c4b9SDmitri Gribenko } 6735188c4b9SDmitri Gribenko 674e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLEndTag(Token &T) { 6755188c4b9SDmitri Gribenko assert(BufferPtr[0] == '<' && BufferPtr[1] == '/'); 6765188c4b9SDmitri Gribenko 6775188c4b9SDmitri Gribenko const char *TagNameBegin = skipWhitespace(BufferPtr + 2, CommentEnd); 6785188c4b9SDmitri Gribenko const char *TagNameEnd = skipHTMLIdentifier(TagNameBegin, CommentEnd); 679107618a6SDmitri Gribenko StringRef Name(TagNameBegin, TagNameEnd - TagNameBegin); 680107618a6SDmitri Gribenko if (!isHTMLTagName(Name)) { 681107618a6SDmitri Gribenko formTextToken(T, TagNameEnd); 682107618a6SDmitri Gribenko return; 683107618a6SDmitri Gribenko } 6845188c4b9SDmitri Gribenko 6855188c4b9SDmitri Gribenko const char *End = skipWhitespace(TagNameEnd, CommentEnd); 6865188c4b9SDmitri Gribenko 687e00ffc7bSDmitri Gribenko formTokenWithChars(T, End, tok::html_end_tag); 688107618a6SDmitri Gribenko T.setHTMLTagEndName(Name); 689ec92531cSDmitri Gribenko 690ec92531cSDmitri Gribenko if (BufferPtr != CommentEnd && *BufferPtr == '>') 691e00ffc7bSDmitri Gribenko State = LS_HTMLEndTag; 692ec92531cSDmitri Gribenko } 693ec92531cSDmitri Gribenko 694e00ffc7bSDmitri Gribenko void Lexer::lexHTMLEndTag(Token &T) { 695ec92531cSDmitri Gribenko assert(BufferPtr != CommentEnd && *BufferPtr == '>'); 696ec92531cSDmitri Gribenko 697ec92531cSDmitri Gribenko formTokenWithChars(T, BufferPtr + 1, tok::html_greater); 698ec92531cSDmitri Gribenko State = LS_Normal; 6995188c4b9SDmitri Gribenko } 7005188c4b9SDmitri Gribenko 7015b637078SFariborz Jahanian Lexer::Lexer(llvm::BumpPtrAllocator &Allocator, DiagnosticsEngine &Diags, 7025b637078SFariborz Jahanian const CommandTraits &Traits, 7036bab9113SDmitri Gribenko SourceLocation FileLoc, 7045188c4b9SDmitri Gribenko const char *BufferStart, const char *BufferEnd): 7055b637078SFariborz Jahanian Allocator(Allocator), Diags(Diags), Traits(Traits), 7065188c4b9SDmitri Gribenko BufferStart(BufferStart), BufferEnd(BufferEnd), 7076bab9113SDmitri Gribenko FileLoc(FileLoc), BufferPtr(BufferStart), 7085188c4b9SDmitri Gribenko CommentState(LCS_BeforeComment), State(LS_Normal) { 7095188c4b9SDmitri Gribenko } 7105188c4b9SDmitri Gribenko 7115188c4b9SDmitri Gribenko void Lexer::lex(Token &T) { 7125188c4b9SDmitri Gribenko again: 7135188c4b9SDmitri Gribenko switch (CommentState) { 7145188c4b9SDmitri Gribenko case LCS_BeforeComment: 7155188c4b9SDmitri Gribenko if (BufferPtr == BufferEnd) { 7165188c4b9SDmitri Gribenko formTokenWithChars(T, BufferPtr, tok::eof); 7175188c4b9SDmitri Gribenko return; 7185188c4b9SDmitri Gribenko } 7195188c4b9SDmitri Gribenko 7205188c4b9SDmitri Gribenko assert(*BufferPtr == '/'); 7215188c4b9SDmitri Gribenko BufferPtr++; // Skip first slash. 7225188c4b9SDmitri Gribenko switch(*BufferPtr) { 7235188c4b9SDmitri Gribenko case '/': { // BCPL comment. 7245188c4b9SDmitri Gribenko BufferPtr++; // Skip second slash. 7255188c4b9SDmitri Gribenko 7265188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd) { 7275188c4b9SDmitri Gribenko // Skip Doxygen magic marker, if it is present. 7285188c4b9SDmitri Gribenko // It might be missing because of a typo //< or /*<, or because we 7295188c4b9SDmitri Gribenko // merged this non-Doxygen comment into a bunch of Doxygen comments 7305188c4b9SDmitri Gribenko // around it: /** ... */ /* ... */ /** ... */ 7315188c4b9SDmitri Gribenko const char C = *BufferPtr; 7325188c4b9SDmitri Gribenko if (C == '/' || C == '!') 7335188c4b9SDmitri Gribenko BufferPtr++; 7345188c4b9SDmitri Gribenko } 7355188c4b9SDmitri Gribenko 7365188c4b9SDmitri Gribenko // Skip less-than symbol that marks trailing comments. 7375188c4b9SDmitri Gribenko // Skip it even if the comment is not a Doxygen one, because //< and /*< 7385188c4b9SDmitri Gribenko // are frequent typos. 7395188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '<') 7405188c4b9SDmitri Gribenko BufferPtr++; 7415188c4b9SDmitri Gribenko 7425188c4b9SDmitri Gribenko CommentState = LCS_InsideBCPLComment; 743ec92531cSDmitri Gribenko if (State != LS_VerbatimBlockBody && State != LS_VerbatimBlockFirstLine) 7445188c4b9SDmitri Gribenko State = LS_Normal; 7455188c4b9SDmitri Gribenko CommentEnd = findBCPLCommentEnd(BufferPtr, BufferEnd); 7465188c4b9SDmitri Gribenko goto again; 7475188c4b9SDmitri Gribenko } 7485188c4b9SDmitri Gribenko case '*': { // C comment. 7495188c4b9SDmitri Gribenko BufferPtr++; // Skip star. 7505188c4b9SDmitri Gribenko 7515188c4b9SDmitri Gribenko // Skip Doxygen magic marker. 7525188c4b9SDmitri Gribenko const char C = *BufferPtr; 7535188c4b9SDmitri Gribenko if ((C == '*' && *(BufferPtr + 1) != '/') || C == '!') 7545188c4b9SDmitri Gribenko BufferPtr++; 7555188c4b9SDmitri Gribenko 7565188c4b9SDmitri Gribenko // Skip less-than symbol that marks trailing comments. 7575188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '<') 7585188c4b9SDmitri Gribenko BufferPtr++; 7595188c4b9SDmitri Gribenko 7605188c4b9SDmitri Gribenko CommentState = LCS_InsideCComment; 7615188c4b9SDmitri Gribenko State = LS_Normal; 7625188c4b9SDmitri Gribenko CommentEnd = findCCommentEnd(BufferPtr, BufferEnd); 7635188c4b9SDmitri Gribenko goto again; 7645188c4b9SDmitri Gribenko } 7655188c4b9SDmitri Gribenko default: 7665188c4b9SDmitri Gribenko llvm_unreachable("second character of comment should be '/' or '*'"); 7675188c4b9SDmitri Gribenko } 7685188c4b9SDmitri Gribenko 7695188c4b9SDmitri Gribenko case LCS_BetweenComments: { 7705188c4b9SDmitri Gribenko // Consecutive comments are extracted only if there is only whitespace 7715188c4b9SDmitri Gribenko // between them. So we can search for the start of the next comment. 7725188c4b9SDmitri Gribenko const char *EndWhitespace = BufferPtr; 7735188c4b9SDmitri Gribenko while(EndWhitespace != BufferEnd && *EndWhitespace != '/') 7745188c4b9SDmitri Gribenko EndWhitespace++; 7755188c4b9SDmitri Gribenko 7765188c4b9SDmitri Gribenko // Turn any whitespace between comments (and there is only whitespace 77717709ae8SDmitri Gribenko // between them -- guaranteed by comment extraction) into a newline. We 77817709ae8SDmitri Gribenko // have two newlines between C comments in total (first one was synthesized 77917709ae8SDmitri Gribenko // after a comment). 7805188c4b9SDmitri Gribenko formTokenWithChars(T, EndWhitespace, tok::newline); 7815188c4b9SDmitri Gribenko 7825188c4b9SDmitri Gribenko CommentState = LCS_BeforeComment; 7835188c4b9SDmitri Gribenko break; 7845188c4b9SDmitri Gribenko } 7855188c4b9SDmitri Gribenko 7865188c4b9SDmitri Gribenko case LCS_InsideBCPLComment: 7875188c4b9SDmitri Gribenko case LCS_InsideCComment: 7885188c4b9SDmitri Gribenko if (BufferPtr != CommentEnd) { 7895188c4b9SDmitri Gribenko lexCommentText(T); 7905188c4b9SDmitri Gribenko break; 7915188c4b9SDmitri Gribenko } else { 7925188c4b9SDmitri Gribenko // Skip C comment closing sequence. 7935188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) { 7945188c4b9SDmitri Gribenko assert(BufferPtr[0] == '*' && BufferPtr[1] == '/'); 7955188c4b9SDmitri Gribenko BufferPtr += 2; 7965188c4b9SDmitri Gribenko assert(BufferPtr <= BufferEnd); 7975188c4b9SDmitri Gribenko 7985188c4b9SDmitri Gribenko // Synthenize newline just after the C comment, regardless if there is 7995188c4b9SDmitri Gribenko // actually a newline. 8005188c4b9SDmitri Gribenko formTokenWithChars(T, BufferPtr, tok::newline); 8015188c4b9SDmitri Gribenko 8025188c4b9SDmitri Gribenko CommentState = LCS_BetweenComments; 8035188c4b9SDmitri Gribenko break; 8045188c4b9SDmitri Gribenko } else { 8055188c4b9SDmitri Gribenko // Don't synthesized a newline after BCPL comment. 8065188c4b9SDmitri Gribenko CommentState = LCS_BetweenComments; 8075188c4b9SDmitri Gribenko goto again; 8085188c4b9SDmitri Gribenko } 8095188c4b9SDmitri Gribenko } 8105188c4b9SDmitri Gribenko } 8115188c4b9SDmitri Gribenko } 8125188c4b9SDmitri Gribenko 8135188c4b9SDmitri Gribenko StringRef Lexer::getSpelling(const Token &Tok, 8145188c4b9SDmitri Gribenko const SourceManager &SourceMgr, 8155188c4b9SDmitri Gribenko bool *Invalid) const { 8165188c4b9SDmitri Gribenko SourceLocation Loc = Tok.getLocation(); 8175188c4b9SDmitri Gribenko std::pair<FileID, unsigned> LocInfo = SourceMgr.getDecomposedLoc(Loc); 8185188c4b9SDmitri Gribenko 8195188c4b9SDmitri Gribenko bool InvalidTemp = false; 8205188c4b9SDmitri Gribenko StringRef File = SourceMgr.getBufferData(LocInfo.first, &InvalidTemp); 8215188c4b9SDmitri Gribenko if (InvalidTemp) { 8225188c4b9SDmitri Gribenko *Invalid = true; 8235188c4b9SDmitri Gribenko return StringRef(); 8245188c4b9SDmitri Gribenko } 8255188c4b9SDmitri Gribenko 8265188c4b9SDmitri Gribenko const char *Begin = File.data() + LocInfo.second; 8275188c4b9SDmitri Gribenko return StringRef(Begin, Tok.getLength()); 8285188c4b9SDmitri Gribenko } 8295188c4b9SDmitri Gribenko 8305188c4b9SDmitri Gribenko } // end namespace comments 8315188c4b9SDmitri Gribenko } // end namespace clang 8325188c4b9SDmitri Gribenko 833