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 2713ffab05fSAlp Toker void Lexer::formTokenWithChars(Token &Result, const char *TokEnd, 2723ffab05fSAlp Toker tok::TokenKind Kind) { 2733ffab05fSAlp Toker const unsigned TokLen = TokEnd - BufferPtr; 2743ffab05fSAlp Toker Result.setLocation(getSourceLocation(BufferPtr)); 2753ffab05fSAlp Toker Result.setKind(Kind); 2763ffab05fSAlp Toker Result.setLength(TokLen); 2773ffab05fSAlp Toker #ifndef NDEBUG 2783ffab05fSAlp Toker Result.TextPtr = "<UNSET>"; 2793ffab05fSAlp Toker Result.IntVal = 7; 2803ffab05fSAlp Toker #endif 2813ffab05fSAlp Toker BufferPtr = TokEnd; 2823ffab05fSAlp Toker } 2833ffab05fSAlp Toker 2845188c4b9SDmitri Gribenko void Lexer::lexCommentText(Token &T) { 2855188c4b9SDmitri Gribenko assert(CommentState == LCS_InsideBCPLComment || 2865188c4b9SDmitri Gribenko CommentState == LCS_InsideCComment); 2875188c4b9SDmitri Gribenko 2885188c4b9SDmitri Gribenko switch (State) { 2895188c4b9SDmitri Gribenko case LS_Normal: 2905188c4b9SDmitri Gribenko break; 2915188c4b9SDmitri Gribenko case LS_VerbatimBlockFirstLine: 2925188c4b9SDmitri Gribenko lexVerbatimBlockFirstLine(T); 2935188c4b9SDmitri Gribenko return; 2945188c4b9SDmitri Gribenko case LS_VerbatimBlockBody: 2955188c4b9SDmitri Gribenko lexVerbatimBlockBody(T); 2965188c4b9SDmitri Gribenko return; 2971669f702SDmitri Gribenko case LS_VerbatimLineText: 2981669f702SDmitri Gribenko lexVerbatimLineText(T); 2991669f702SDmitri Gribenko return; 300e00ffc7bSDmitri Gribenko case LS_HTMLStartTag: 301e00ffc7bSDmitri Gribenko lexHTMLStartTag(T); 3025188c4b9SDmitri Gribenko return; 303e00ffc7bSDmitri Gribenko case LS_HTMLEndTag: 304e00ffc7bSDmitri Gribenko lexHTMLEndTag(T); 305ec92531cSDmitri Gribenko return; 3065188c4b9SDmitri Gribenko } 3075188c4b9SDmitri Gribenko 3085188c4b9SDmitri Gribenko assert(State == LS_Normal); 3095188c4b9SDmitri Gribenko 3105188c4b9SDmitri Gribenko const char *TokenPtr = BufferPtr; 3115188c4b9SDmitri Gribenko assert(TokenPtr < CommentEnd); 3125188c4b9SDmitri Gribenko while (TokenPtr != CommentEnd) { 3135188c4b9SDmitri Gribenko switch(*TokenPtr) { 3145188c4b9SDmitri Gribenko case '\\': 3155188c4b9SDmitri Gribenko case '@': { 316bcf7f4d3SDmitri Gribenko // Commands that start with a backslash and commands that start with 317bcf7f4d3SDmitri Gribenko // 'at' have equivalent semantics. But we keep information about the 318bcf7f4d3SDmitri Gribenko // exact syntax in AST for comments. 319bcf7f4d3SDmitri Gribenko tok::TokenKind CommandKind = 320bcf7f4d3SDmitri Gribenko (*TokenPtr == '@') ? tok::at_command : tok::backslash_command; 3215188c4b9SDmitri Gribenko TokenPtr++; 3225188c4b9SDmitri Gribenko if (TokenPtr == CommentEnd) { 3234586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3245188c4b9SDmitri Gribenko return; 3255188c4b9SDmitri Gribenko } 3265188c4b9SDmitri Gribenko char C = *TokenPtr; 3275188c4b9SDmitri Gribenko switch (C) { 3285188c4b9SDmitri Gribenko default: 3295188c4b9SDmitri Gribenko break; 3305188c4b9SDmitri Gribenko 3315188c4b9SDmitri Gribenko case '\\': case '@': case '&': case '$': 3325188c4b9SDmitri Gribenko case '#': case '<': case '>': case '%': 3335188c4b9SDmitri Gribenko case '\"': case '.': case ':': 3345188c4b9SDmitri Gribenko // This is one of \\ \@ \& \$ etc escape sequences. 3355188c4b9SDmitri Gribenko TokenPtr++; 3365188c4b9SDmitri Gribenko if (C == ':' && TokenPtr != CommentEnd && *TokenPtr == ':') { 3375188c4b9SDmitri Gribenko // This is the \:: escape sequence. 3385188c4b9SDmitri Gribenko TokenPtr++; 3395188c4b9SDmitri Gribenko } 34060ddd8a1SDmitri Gribenko StringRef UnescapedText(BufferPtr + 1, TokenPtr - (BufferPtr + 1)); 3415188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::text); 34260ddd8a1SDmitri Gribenko T.setText(UnescapedText); 3435188c4b9SDmitri Gribenko return; 3445188c4b9SDmitri Gribenko } 3455188c4b9SDmitri Gribenko 3465188c4b9SDmitri Gribenko // Don't make zero-length commands. 347ad45ad6fSDmitri Gribenko if (!isCommandNameStartCharacter(*TokenPtr)) { 3484586df76SDmitri Gribenko formTextToken(T, TokenPtr); 3495188c4b9SDmitri Gribenko return; 3505188c4b9SDmitri Gribenko } 3515188c4b9SDmitri Gribenko 3525188c4b9SDmitri Gribenko TokenPtr = skipCommandName(TokenPtr, CommentEnd); 3535188c4b9SDmitri Gribenko unsigned Length = TokenPtr - (BufferPtr + 1); 3545188c4b9SDmitri Gribenko 3555188c4b9SDmitri Gribenko // Hardcoded support for lexing LaTeX formula commands 3565188c4b9SDmitri Gribenko // \f$ \f[ \f] \f{ \f} as a single command. 3575188c4b9SDmitri Gribenko if (Length == 1 && TokenPtr[-1] == 'f' && TokenPtr != CommentEnd) { 3585188c4b9SDmitri Gribenko C = *TokenPtr; 3595188c4b9SDmitri Gribenko if (C == '$' || C == '[' || C == ']' || C == '{' || C == '}') { 3605188c4b9SDmitri Gribenko TokenPtr++; 3615188c4b9SDmitri Gribenko Length++; 3625188c4b9SDmitri Gribenko } 3635188c4b9SDmitri Gribenko } 3645188c4b9SDmitri Gribenko 365*bf3e3270SCraig Topper StringRef CommandName(BufferPtr + 1, Length); 3665188c4b9SDmitri Gribenko 3677acbf00fSDmitri Gribenko const CommandInfo *Info = Traits.getCommandInfoOrNULL(CommandName); 3687acbf00fSDmitri Gribenko if (!Info) { 369ebb262f7SFariborz Jahanian if ((Info = Traits.getTypoCorrectCommandInfo(CommandName))) { 3706c7a1666SFariborz Jahanian StringRef CorrectedName = Info->Name; 3712dece574SBenjamin Kramer SourceLocation Loc = getSourceLocation(BufferPtr); 3722dece574SBenjamin Kramer SourceRange CommandRange(Loc.getLocWithOffset(1), 3732dece574SBenjamin Kramer getSourceLocation(TokenPtr)); 3742dece574SBenjamin Kramer Diag(Loc, diag::warn_correct_comment_command_name) 3756c7a1666SFariborz Jahanian << CommandName << CorrectedName 3766c7a1666SFariborz Jahanian << FixItHint::CreateReplacement(CommandRange, CorrectedName); 3776c7a1666SFariborz Jahanian } else { 3782dece574SBenjamin Kramer formTokenWithChars(T, TokenPtr, tok::unknown_command); 3792dece574SBenjamin Kramer T.setUnknownCommandName(CommandName); 3806738e430SFariborz Jahanian Diag(T.getLocation(), diag::warn_unknown_comment_command_name); 3815188c4b9SDmitri Gribenko return; 3825188c4b9SDmitri Gribenko } 3836c7a1666SFariborz Jahanian } 3847acbf00fSDmitri Gribenko if (Info->IsVerbatimBlockCommand) { 3857acbf00fSDmitri Gribenko setupAndLexVerbatimBlock(T, TokenPtr, *BufferPtr, Info); 3867acbf00fSDmitri Gribenko return; 3877acbf00fSDmitri Gribenko } 3887acbf00fSDmitri Gribenko if (Info->IsVerbatimLineCommand) { 3897acbf00fSDmitri Gribenko setupAndLexVerbatimLine(T, TokenPtr, Info); 3905188c4b9SDmitri Gribenko return; 3915188c4b9SDmitri Gribenko } 392bcf7f4d3SDmitri Gribenko formTokenWithChars(T, TokenPtr, CommandKind); 3937acbf00fSDmitri Gribenko T.setCommandID(Info->getID()); 3945188c4b9SDmitri Gribenko return; 3955188c4b9SDmitri Gribenko } 3965188c4b9SDmitri Gribenko 3974586df76SDmitri Gribenko case '&': 3984586df76SDmitri Gribenko lexHTMLCharacterReference(T); 3994586df76SDmitri Gribenko return; 4004586df76SDmitri Gribenko 4015188c4b9SDmitri Gribenko case '<': { 4025188c4b9SDmitri Gribenko TokenPtr++; 4035188c4b9SDmitri Gribenko if (TokenPtr == CommentEnd) { 4044586df76SDmitri Gribenko formTextToken(T, TokenPtr); 4055188c4b9SDmitri Gribenko return; 4065188c4b9SDmitri Gribenko } 4075188c4b9SDmitri Gribenko const char C = *TokenPtr; 40817709ae8SDmitri Gribenko if (isHTMLIdentifierStartingCharacter(C)) 409e00ffc7bSDmitri Gribenko setupAndLexHTMLStartTag(T); 4105188c4b9SDmitri Gribenko else if (C == '/') 411e00ffc7bSDmitri Gribenko setupAndLexHTMLEndTag(T); 4124586df76SDmitri Gribenko else 4134586df76SDmitri Gribenko formTextToken(T, TokenPtr); 4144586df76SDmitri Gribenko 4155188c4b9SDmitri Gribenko return; 4165188c4b9SDmitri Gribenko } 4175188c4b9SDmitri Gribenko 4185188c4b9SDmitri Gribenko case '\n': 4195188c4b9SDmitri Gribenko case '\r': 4205188c4b9SDmitri Gribenko TokenPtr = skipNewline(TokenPtr, CommentEnd); 4215188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::newline); 4225188c4b9SDmitri Gribenko 4235188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) 4245188c4b9SDmitri Gribenko skipLineStartingDecorations(); 4255188c4b9SDmitri Gribenko return; 4265188c4b9SDmitri Gribenko 4275188c4b9SDmitri Gribenko default: { 42810af67a9SDmitri Gribenko size_t End = StringRef(TokenPtr, CommentEnd - TokenPtr). 42910af67a9SDmitri Gribenko find_first_of("\n\r\\@&<"); 43010af67a9SDmitri Gribenko if (End != StringRef::npos) 43110af67a9SDmitri Gribenko TokenPtr += End; 43210af67a9SDmitri Gribenko else 43310af67a9SDmitri Gribenko TokenPtr = CommentEnd; 4344586df76SDmitri Gribenko formTextToken(T, TokenPtr); 4355188c4b9SDmitri Gribenko return; 4365188c4b9SDmitri Gribenko } 4375188c4b9SDmitri Gribenko } 4385188c4b9SDmitri Gribenko } 4395188c4b9SDmitri Gribenko } 4405188c4b9SDmitri Gribenko 4415188c4b9SDmitri Gribenko void Lexer::setupAndLexVerbatimBlock(Token &T, 4425188c4b9SDmitri Gribenko const char *TextBegin, 4437acbf00fSDmitri Gribenko char Marker, const CommandInfo *Info) { 4447acbf00fSDmitri Gribenko assert(Info->IsVerbatimBlockCommand); 4457acbf00fSDmitri Gribenko 4465188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.clear(); 4475188c4b9SDmitri Gribenko VerbatimBlockEndCommandName.append(Marker == '\\' ? "\\" : "@"); 4487acbf00fSDmitri Gribenko VerbatimBlockEndCommandName.append(Info->EndCommandName); 4495188c4b9SDmitri Gribenko 4505188c4b9SDmitri Gribenko formTokenWithChars(T, TextBegin, tok::verbatim_block_begin); 4517acbf00fSDmitri Gribenko T.setVerbatimBlockID(Info->getID()); 4525188c4b9SDmitri Gribenko 453ec92531cSDmitri Gribenko // If there is a newline following the verbatim opening command, skip the 454ec92531cSDmitri Gribenko // newline so that we don't create an tok::verbatim_block_line with empty 455ec92531cSDmitri Gribenko // text content. 456bcef3411SDmitri Gribenko if (BufferPtr != CommentEnd && 457bcef3411SDmitri Gribenko isVerticalWhitespace(*BufferPtr)) { 458ec92531cSDmitri Gribenko BufferPtr = skipNewline(BufferPtr, CommentEnd); 459ec92531cSDmitri Gribenko State = LS_VerbatimBlockBody; 460ec92531cSDmitri Gribenko return; 461ec92531cSDmitri Gribenko } 462ec92531cSDmitri Gribenko 4635188c4b9SDmitri Gribenko State = LS_VerbatimBlockFirstLine; 4645188c4b9SDmitri Gribenko } 4655188c4b9SDmitri Gribenko 4665188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockFirstLine(Token &T) { 467e4a3997dSDmitri Gribenko again: 4685188c4b9SDmitri Gribenko assert(BufferPtr < CommentEnd); 4695188c4b9SDmitri Gribenko 4705188c4b9SDmitri Gribenko // FIXME: It would be better to scan the text once, finding either the block 4715188c4b9SDmitri Gribenko // end command or newline. 4725188c4b9SDmitri Gribenko // 4735188c4b9SDmitri Gribenko // Extract current line. 4745188c4b9SDmitri Gribenko const char *Newline = findNewline(BufferPtr, CommentEnd); 4755188c4b9SDmitri Gribenko StringRef Line(BufferPtr, Newline - BufferPtr); 4765188c4b9SDmitri Gribenko 4775188c4b9SDmitri Gribenko // Look for end command in current line. 4785188c4b9SDmitri Gribenko size_t Pos = Line.find(VerbatimBlockEndCommandName); 479ec92531cSDmitri Gribenko const char *TextEnd; 4805188c4b9SDmitri Gribenko const char *NextLine; 4815188c4b9SDmitri Gribenko if (Pos == StringRef::npos) { 4825188c4b9SDmitri Gribenko // Current line is completely verbatim. 483ec92531cSDmitri Gribenko TextEnd = Newline; 4845188c4b9SDmitri Gribenko NextLine = skipNewline(Newline, CommentEnd); 4855188c4b9SDmitri Gribenko } else if (Pos == 0) { 4865188c4b9SDmitri Gribenko // Current line contains just an end command. 4875188c4b9SDmitri Gribenko const char *End = BufferPtr + VerbatimBlockEndCommandName.size(); 48860ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, End - (BufferPtr + 1)); 4895188c4b9SDmitri Gribenko formTokenWithChars(T, End, tok::verbatim_block_end); 4907acbf00fSDmitri Gribenko T.setVerbatimBlockID(Traits.getCommandInfo(Name)->getID()); 4915188c4b9SDmitri Gribenko State = LS_Normal; 4925188c4b9SDmitri Gribenko return; 4935188c4b9SDmitri Gribenko } else { 4945188c4b9SDmitri Gribenko // There is some text, followed by end command. Extract text first. 495ec92531cSDmitri Gribenko TextEnd = BufferPtr + Pos; 496ec92531cSDmitri Gribenko NextLine = TextEnd; 497e4a3997dSDmitri Gribenko // If there is only whitespace before end command, skip whitespace. 498e4a3997dSDmitri Gribenko if (isWhitespace(BufferPtr, TextEnd)) { 499e4a3997dSDmitri Gribenko BufferPtr = TextEnd; 500e4a3997dSDmitri Gribenko goto again; 501e4a3997dSDmitri Gribenko } 5025188c4b9SDmitri Gribenko } 5035188c4b9SDmitri Gribenko 504ec92531cSDmitri Gribenko StringRef Text(BufferPtr, TextEnd - BufferPtr); 5055188c4b9SDmitri Gribenko formTokenWithChars(T, NextLine, tok::verbatim_block_line); 50660ddd8a1SDmitri Gribenko T.setVerbatimBlockText(Text); 5075188c4b9SDmitri Gribenko 5085188c4b9SDmitri Gribenko State = LS_VerbatimBlockBody; 5095188c4b9SDmitri Gribenko } 5105188c4b9SDmitri Gribenko 5115188c4b9SDmitri Gribenko void Lexer::lexVerbatimBlockBody(Token &T) { 5125188c4b9SDmitri Gribenko assert(State == LS_VerbatimBlockBody); 5135188c4b9SDmitri Gribenko 5145188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) 5155188c4b9SDmitri Gribenko skipLineStartingDecorations(); 5165188c4b9SDmitri Gribenko 5175188c4b9SDmitri Gribenko lexVerbatimBlockFirstLine(T); 5185188c4b9SDmitri Gribenko } 5195188c4b9SDmitri Gribenko 5207acbf00fSDmitri Gribenko void Lexer::setupAndLexVerbatimLine(Token &T, const char *TextBegin, 5217acbf00fSDmitri Gribenko const CommandInfo *Info) { 5227acbf00fSDmitri Gribenko assert(Info->IsVerbatimLineCommand); 5231669f702SDmitri Gribenko formTokenWithChars(T, TextBegin, tok::verbatim_line_name); 5247acbf00fSDmitri Gribenko T.setVerbatimLineID(Info->getID()); 5251669f702SDmitri Gribenko 5261669f702SDmitri Gribenko State = LS_VerbatimLineText; 5271669f702SDmitri Gribenko } 5281669f702SDmitri Gribenko 5291669f702SDmitri Gribenko void Lexer::lexVerbatimLineText(Token &T) { 5301669f702SDmitri Gribenko assert(State == LS_VerbatimLineText); 5311669f702SDmitri Gribenko 5325188c4b9SDmitri Gribenko // Extract current line. 5335188c4b9SDmitri Gribenko const char *Newline = findNewline(BufferPtr, CommentEnd); 534*bf3e3270SCraig Topper StringRef Text(BufferPtr, Newline - BufferPtr); 5351669f702SDmitri Gribenko formTokenWithChars(T, Newline, tok::verbatim_line_text); 5365188c4b9SDmitri Gribenko T.setVerbatimLineText(Text); 5371669f702SDmitri Gribenko 5381669f702SDmitri Gribenko State = LS_Normal; 5395188c4b9SDmitri Gribenko } 5405188c4b9SDmitri Gribenko 5414586df76SDmitri Gribenko void Lexer::lexHTMLCharacterReference(Token &T) { 5424586df76SDmitri Gribenko const char *TokenPtr = BufferPtr; 5434586df76SDmitri Gribenko assert(*TokenPtr == '&'); 5444586df76SDmitri Gribenko TokenPtr++; 5454586df76SDmitri Gribenko if (TokenPtr == CommentEnd) { 5464586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5474586df76SDmitri Gribenko return; 5484586df76SDmitri Gribenko } 5494586df76SDmitri Gribenko const char *NamePtr; 5504586df76SDmitri Gribenko bool isNamed = false; 5514586df76SDmitri Gribenko bool isDecimal = false; 5524586df76SDmitri Gribenko char C = *TokenPtr; 5534586df76SDmitri Gribenko if (isHTMLNamedCharacterReferenceCharacter(C)) { 5544586df76SDmitri Gribenko NamePtr = TokenPtr; 5554586df76SDmitri Gribenko TokenPtr = skipNamedCharacterReference(TokenPtr, CommentEnd); 5564586df76SDmitri Gribenko isNamed = true; 5574586df76SDmitri Gribenko } else if (C == '#') { 5584586df76SDmitri Gribenko TokenPtr++; 5594586df76SDmitri Gribenko if (TokenPtr == CommentEnd) { 5604586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5614586df76SDmitri Gribenko return; 5624586df76SDmitri Gribenko } 5634586df76SDmitri Gribenko C = *TokenPtr; 5644586df76SDmitri Gribenko if (isHTMLDecimalCharacterReferenceCharacter(C)) { 5654586df76SDmitri Gribenko NamePtr = TokenPtr; 5664586df76SDmitri Gribenko TokenPtr = skipDecimalCharacterReference(TokenPtr, CommentEnd); 5674586df76SDmitri Gribenko isDecimal = true; 5684586df76SDmitri Gribenko } else if (C == 'x' || C == 'X') { 5694586df76SDmitri Gribenko TokenPtr++; 5704586df76SDmitri Gribenko NamePtr = TokenPtr; 5714586df76SDmitri Gribenko TokenPtr = skipHexCharacterReference(TokenPtr, CommentEnd); 5724586df76SDmitri Gribenko } else { 5734586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5744586df76SDmitri Gribenko return; 5754586df76SDmitri Gribenko } 5764586df76SDmitri Gribenko } else { 5774586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5784586df76SDmitri Gribenko return; 5794586df76SDmitri Gribenko } 5804586df76SDmitri Gribenko if (NamePtr == TokenPtr || TokenPtr == CommentEnd || 5814586df76SDmitri Gribenko *TokenPtr != ';') { 5824586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5834586df76SDmitri Gribenko return; 5844586df76SDmitri Gribenko } 5854586df76SDmitri Gribenko StringRef Name(NamePtr, TokenPtr - NamePtr); 5864586df76SDmitri Gribenko TokenPtr++; // Skip semicolon. 5874586df76SDmitri Gribenko StringRef Resolved; 58828800da1SDmitri Gribenko if (isNamed) 5894586df76SDmitri Gribenko Resolved = resolveHTMLNamedCharacterReference(Name); 5904586df76SDmitri Gribenko else if (isDecimal) 5914586df76SDmitri Gribenko Resolved = resolveHTMLDecimalCharacterReference(Name); 5924586df76SDmitri Gribenko else 5934586df76SDmitri Gribenko Resolved = resolveHTMLHexCharacterReference(Name); 5944586df76SDmitri Gribenko 5954586df76SDmitri Gribenko if (Resolved.empty()) { 5964586df76SDmitri Gribenko formTextToken(T, TokenPtr); 5974586df76SDmitri Gribenko return; 5984586df76SDmitri Gribenko } 5994586df76SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::text); 6004586df76SDmitri Gribenko T.setText(Resolved); 6014586df76SDmitri Gribenko return; 6024586df76SDmitri Gribenko } 6034586df76SDmitri Gribenko 604e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLStartTag(Token &T) { 60517709ae8SDmitri Gribenko assert(BufferPtr[0] == '<' && 60617709ae8SDmitri Gribenko isHTMLIdentifierStartingCharacter(BufferPtr[1])); 6075188c4b9SDmitri Gribenko const char *TagNameEnd = skipHTMLIdentifier(BufferPtr + 2, CommentEnd); 60860ddd8a1SDmitri Gribenko StringRef Name(BufferPtr + 1, TagNameEnd - (BufferPtr + 1)); 609107618a6SDmitri Gribenko if (!isHTMLTagName(Name)) { 610107618a6SDmitri Gribenko formTextToken(T, TagNameEnd); 611107618a6SDmitri Gribenko return; 612107618a6SDmitri Gribenko } 613107618a6SDmitri Gribenko 614e00ffc7bSDmitri Gribenko formTokenWithChars(T, TagNameEnd, tok::html_start_tag); 615e00ffc7bSDmitri Gribenko T.setHTMLTagStartName(Name); 6165188c4b9SDmitri Gribenko 6175188c4b9SDmitri Gribenko BufferPtr = skipWhitespace(BufferPtr, CommentEnd); 6185188c4b9SDmitri Gribenko 61917709ae8SDmitri Gribenko const char C = *BufferPtr; 62017709ae8SDmitri Gribenko if (BufferPtr != CommentEnd && 621f26054f0SDmitri Gribenko (C == '>' || C == '/' || isHTMLIdentifierStartingCharacter(C))) 622e00ffc7bSDmitri Gribenko State = LS_HTMLStartTag; 6235188c4b9SDmitri Gribenko } 6245188c4b9SDmitri Gribenko 625e00ffc7bSDmitri Gribenko void Lexer::lexHTMLStartTag(Token &T) { 626e00ffc7bSDmitri Gribenko assert(State == LS_HTMLStartTag); 6275188c4b9SDmitri Gribenko 6285188c4b9SDmitri Gribenko const char *TokenPtr = BufferPtr; 6295188c4b9SDmitri Gribenko char C = *TokenPtr; 6305188c4b9SDmitri Gribenko if (isHTMLIdentifierCharacter(C)) { 6315188c4b9SDmitri Gribenko TokenPtr = skipHTMLIdentifier(TokenPtr, CommentEnd); 63260ddd8a1SDmitri Gribenko StringRef Ident(BufferPtr, TokenPtr - BufferPtr); 6335188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_ident); 63460ddd8a1SDmitri Gribenko T.setHTMLIdent(Ident); 6355188c4b9SDmitri Gribenko } else { 6365188c4b9SDmitri Gribenko switch (C) { 6375188c4b9SDmitri Gribenko case '=': 6385188c4b9SDmitri Gribenko TokenPtr++; 6395188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_equals); 6405188c4b9SDmitri Gribenko break; 6415188c4b9SDmitri Gribenko case '\"': 6425188c4b9SDmitri Gribenko case '\'': { 6435188c4b9SDmitri Gribenko const char *OpenQuote = TokenPtr; 6445188c4b9SDmitri Gribenko TokenPtr = skipHTMLQuotedString(TokenPtr, CommentEnd); 6455188c4b9SDmitri Gribenko const char *ClosingQuote = TokenPtr; 6465188c4b9SDmitri Gribenko if (TokenPtr != CommentEnd) // Skip closing quote. 6475188c4b9SDmitri Gribenko TokenPtr++; 6485188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_quoted_string); 6495188c4b9SDmitri Gribenko T.setHTMLQuotedString(StringRef(OpenQuote + 1, 6505188c4b9SDmitri Gribenko ClosingQuote - (OpenQuote + 1))); 6515188c4b9SDmitri Gribenko break; 6525188c4b9SDmitri Gribenko } 6535188c4b9SDmitri Gribenko case '>': 6545188c4b9SDmitri Gribenko TokenPtr++; 6555188c4b9SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_greater); 65617709ae8SDmitri Gribenko State = LS_Normal; 65717709ae8SDmitri Gribenko return; 658f26054f0SDmitri Gribenko case '/': 659f26054f0SDmitri Gribenko TokenPtr++; 660f26054f0SDmitri Gribenko if (TokenPtr != CommentEnd && *TokenPtr == '>') { 661f26054f0SDmitri Gribenko TokenPtr++; 662f26054f0SDmitri Gribenko formTokenWithChars(T, TokenPtr, tok::html_slash_greater); 6634586df76SDmitri Gribenko } else 6644586df76SDmitri Gribenko formTextToken(T, TokenPtr); 6654586df76SDmitri Gribenko 666f26054f0SDmitri Gribenko State = LS_Normal; 667f26054f0SDmitri Gribenko return; 6685188c4b9SDmitri Gribenko } 6695188c4b9SDmitri Gribenko } 6705188c4b9SDmitri Gribenko 6715188c4b9SDmitri Gribenko // Now look ahead and return to normal state if we don't see any HTML tokens 6725188c4b9SDmitri Gribenko // ahead. 6735188c4b9SDmitri Gribenko BufferPtr = skipWhitespace(BufferPtr, CommentEnd); 6745188c4b9SDmitri Gribenko if (BufferPtr == CommentEnd) { 6755188c4b9SDmitri Gribenko State = LS_Normal; 6765188c4b9SDmitri Gribenko return; 6775188c4b9SDmitri Gribenko } 6785188c4b9SDmitri Gribenko 6795188c4b9SDmitri Gribenko C = *BufferPtr; 68017709ae8SDmitri Gribenko if (!isHTMLIdentifierStartingCharacter(C) && 6815188c4b9SDmitri Gribenko C != '=' && C != '\"' && C != '\'' && C != '>') { 6825188c4b9SDmitri Gribenko State = LS_Normal; 6835188c4b9SDmitri Gribenko return; 6845188c4b9SDmitri Gribenko } 6855188c4b9SDmitri Gribenko } 6865188c4b9SDmitri Gribenko 687e00ffc7bSDmitri Gribenko void Lexer::setupAndLexHTMLEndTag(Token &T) { 6885188c4b9SDmitri Gribenko assert(BufferPtr[0] == '<' && BufferPtr[1] == '/'); 6895188c4b9SDmitri Gribenko 6905188c4b9SDmitri Gribenko const char *TagNameBegin = skipWhitespace(BufferPtr + 2, CommentEnd); 6915188c4b9SDmitri Gribenko const char *TagNameEnd = skipHTMLIdentifier(TagNameBegin, CommentEnd); 692107618a6SDmitri Gribenko StringRef Name(TagNameBegin, TagNameEnd - TagNameBegin); 693107618a6SDmitri Gribenko if (!isHTMLTagName(Name)) { 694107618a6SDmitri Gribenko formTextToken(T, TagNameEnd); 695107618a6SDmitri Gribenko return; 696107618a6SDmitri Gribenko } 6975188c4b9SDmitri Gribenko 6985188c4b9SDmitri Gribenko const char *End = skipWhitespace(TagNameEnd, CommentEnd); 6995188c4b9SDmitri Gribenko 700e00ffc7bSDmitri Gribenko formTokenWithChars(T, End, tok::html_end_tag); 701107618a6SDmitri Gribenko T.setHTMLTagEndName(Name); 702ec92531cSDmitri Gribenko 703ec92531cSDmitri Gribenko if (BufferPtr != CommentEnd && *BufferPtr == '>') 704e00ffc7bSDmitri Gribenko State = LS_HTMLEndTag; 705ec92531cSDmitri Gribenko } 706ec92531cSDmitri Gribenko 707e00ffc7bSDmitri Gribenko void Lexer::lexHTMLEndTag(Token &T) { 708ec92531cSDmitri Gribenko assert(BufferPtr != CommentEnd && *BufferPtr == '>'); 709ec92531cSDmitri Gribenko 710ec92531cSDmitri Gribenko formTokenWithChars(T, BufferPtr + 1, tok::html_greater); 711ec92531cSDmitri Gribenko State = LS_Normal; 7125188c4b9SDmitri Gribenko } 7135188c4b9SDmitri Gribenko 7145b637078SFariborz Jahanian Lexer::Lexer(llvm::BumpPtrAllocator &Allocator, DiagnosticsEngine &Diags, 7155b637078SFariborz Jahanian const CommandTraits &Traits, 7166bab9113SDmitri Gribenko SourceLocation FileLoc, 7175188c4b9SDmitri Gribenko const char *BufferStart, const char *BufferEnd): 7185b637078SFariborz Jahanian Allocator(Allocator), Diags(Diags), Traits(Traits), 7195188c4b9SDmitri Gribenko BufferStart(BufferStart), BufferEnd(BufferEnd), 7206bab9113SDmitri Gribenko FileLoc(FileLoc), BufferPtr(BufferStart), 7215188c4b9SDmitri Gribenko CommentState(LCS_BeforeComment), State(LS_Normal) { 7225188c4b9SDmitri Gribenko } 7235188c4b9SDmitri Gribenko 7245188c4b9SDmitri Gribenko void Lexer::lex(Token &T) { 7255188c4b9SDmitri Gribenko again: 7265188c4b9SDmitri Gribenko switch (CommentState) { 7275188c4b9SDmitri Gribenko case LCS_BeforeComment: 7285188c4b9SDmitri Gribenko if (BufferPtr == BufferEnd) { 7295188c4b9SDmitri Gribenko formTokenWithChars(T, BufferPtr, tok::eof); 7305188c4b9SDmitri Gribenko return; 7315188c4b9SDmitri Gribenko } 7325188c4b9SDmitri Gribenko 7335188c4b9SDmitri Gribenko assert(*BufferPtr == '/'); 7345188c4b9SDmitri Gribenko BufferPtr++; // Skip first slash. 7355188c4b9SDmitri Gribenko switch(*BufferPtr) { 7365188c4b9SDmitri Gribenko case '/': { // BCPL comment. 7375188c4b9SDmitri Gribenko BufferPtr++; // Skip second slash. 7385188c4b9SDmitri Gribenko 7395188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd) { 7405188c4b9SDmitri Gribenko // Skip Doxygen magic marker, if it is present. 7415188c4b9SDmitri Gribenko // It might be missing because of a typo //< or /*<, or because we 7425188c4b9SDmitri Gribenko // merged this non-Doxygen comment into a bunch of Doxygen comments 7435188c4b9SDmitri Gribenko // around it: /** ... */ /* ... */ /** ... */ 7445188c4b9SDmitri Gribenko const char C = *BufferPtr; 7455188c4b9SDmitri Gribenko if (C == '/' || C == '!') 7465188c4b9SDmitri Gribenko BufferPtr++; 7475188c4b9SDmitri Gribenko } 7485188c4b9SDmitri Gribenko 7495188c4b9SDmitri Gribenko // Skip less-than symbol that marks trailing comments. 7505188c4b9SDmitri Gribenko // Skip it even if the comment is not a Doxygen one, because //< and /*< 7515188c4b9SDmitri Gribenko // are frequent typos. 7525188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '<') 7535188c4b9SDmitri Gribenko BufferPtr++; 7545188c4b9SDmitri Gribenko 7555188c4b9SDmitri Gribenko CommentState = LCS_InsideBCPLComment; 756ec92531cSDmitri Gribenko if (State != LS_VerbatimBlockBody && State != LS_VerbatimBlockFirstLine) 7575188c4b9SDmitri Gribenko State = LS_Normal; 7585188c4b9SDmitri Gribenko CommentEnd = findBCPLCommentEnd(BufferPtr, BufferEnd); 7595188c4b9SDmitri Gribenko goto again; 7605188c4b9SDmitri Gribenko } 7615188c4b9SDmitri Gribenko case '*': { // C comment. 7625188c4b9SDmitri Gribenko BufferPtr++; // Skip star. 7635188c4b9SDmitri Gribenko 7645188c4b9SDmitri Gribenko // Skip Doxygen magic marker. 7655188c4b9SDmitri Gribenko const char C = *BufferPtr; 7665188c4b9SDmitri Gribenko if ((C == '*' && *(BufferPtr + 1) != '/') || C == '!') 7675188c4b9SDmitri Gribenko BufferPtr++; 7685188c4b9SDmitri Gribenko 7695188c4b9SDmitri Gribenko // Skip less-than symbol that marks trailing comments. 7705188c4b9SDmitri Gribenko if (BufferPtr != BufferEnd && *BufferPtr == '<') 7715188c4b9SDmitri Gribenko BufferPtr++; 7725188c4b9SDmitri Gribenko 7735188c4b9SDmitri Gribenko CommentState = LCS_InsideCComment; 7745188c4b9SDmitri Gribenko State = LS_Normal; 7755188c4b9SDmitri Gribenko CommentEnd = findCCommentEnd(BufferPtr, BufferEnd); 7765188c4b9SDmitri Gribenko goto again; 7775188c4b9SDmitri Gribenko } 7785188c4b9SDmitri Gribenko default: 7795188c4b9SDmitri Gribenko llvm_unreachable("second character of comment should be '/' or '*'"); 7805188c4b9SDmitri Gribenko } 7815188c4b9SDmitri Gribenko 7825188c4b9SDmitri Gribenko case LCS_BetweenComments: { 7835188c4b9SDmitri Gribenko // Consecutive comments are extracted only if there is only whitespace 7845188c4b9SDmitri Gribenko // between them. So we can search for the start of the next comment. 7855188c4b9SDmitri Gribenko const char *EndWhitespace = BufferPtr; 7865188c4b9SDmitri Gribenko while(EndWhitespace != BufferEnd && *EndWhitespace != '/') 7875188c4b9SDmitri Gribenko EndWhitespace++; 7885188c4b9SDmitri Gribenko 7895188c4b9SDmitri Gribenko // Turn any whitespace between comments (and there is only whitespace 79017709ae8SDmitri Gribenko // between them -- guaranteed by comment extraction) into a newline. We 79117709ae8SDmitri Gribenko // have two newlines between C comments in total (first one was synthesized 79217709ae8SDmitri Gribenko // after a comment). 7935188c4b9SDmitri Gribenko formTokenWithChars(T, EndWhitespace, tok::newline); 7945188c4b9SDmitri Gribenko 7955188c4b9SDmitri Gribenko CommentState = LCS_BeforeComment; 7965188c4b9SDmitri Gribenko break; 7975188c4b9SDmitri Gribenko } 7985188c4b9SDmitri Gribenko 7995188c4b9SDmitri Gribenko case LCS_InsideBCPLComment: 8005188c4b9SDmitri Gribenko case LCS_InsideCComment: 8015188c4b9SDmitri Gribenko if (BufferPtr != CommentEnd) { 8025188c4b9SDmitri Gribenko lexCommentText(T); 8035188c4b9SDmitri Gribenko break; 8045188c4b9SDmitri Gribenko } else { 8055188c4b9SDmitri Gribenko // Skip C comment closing sequence. 8065188c4b9SDmitri Gribenko if (CommentState == LCS_InsideCComment) { 8075188c4b9SDmitri Gribenko assert(BufferPtr[0] == '*' && BufferPtr[1] == '/'); 8085188c4b9SDmitri Gribenko BufferPtr += 2; 8095188c4b9SDmitri Gribenko assert(BufferPtr <= BufferEnd); 8105188c4b9SDmitri Gribenko 8115188c4b9SDmitri Gribenko // Synthenize newline just after the C comment, regardless if there is 8125188c4b9SDmitri Gribenko // actually a newline. 8135188c4b9SDmitri Gribenko formTokenWithChars(T, BufferPtr, tok::newline); 8145188c4b9SDmitri Gribenko 8155188c4b9SDmitri Gribenko CommentState = LCS_BetweenComments; 8165188c4b9SDmitri Gribenko break; 8175188c4b9SDmitri Gribenko } else { 8185188c4b9SDmitri Gribenko // Don't synthesized a newline after BCPL comment. 8195188c4b9SDmitri Gribenko CommentState = LCS_BetweenComments; 8205188c4b9SDmitri Gribenko goto again; 8215188c4b9SDmitri Gribenko } 8225188c4b9SDmitri Gribenko } 8235188c4b9SDmitri Gribenko } 8245188c4b9SDmitri Gribenko } 8255188c4b9SDmitri Gribenko 8265188c4b9SDmitri Gribenko StringRef Lexer::getSpelling(const Token &Tok, 8275188c4b9SDmitri Gribenko const SourceManager &SourceMgr, 8285188c4b9SDmitri Gribenko bool *Invalid) const { 8295188c4b9SDmitri Gribenko SourceLocation Loc = Tok.getLocation(); 8305188c4b9SDmitri Gribenko std::pair<FileID, unsigned> LocInfo = SourceMgr.getDecomposedLoc(Loc); 8315188c4b9SDmitri Gribenko 8325188c4b9SDmitri Gribenko bool InvalidTemp = false; 8335188c4b9SDmitri Gribenko StringRef File = SourceMgr.getBufferData(LocInfo.first, &InvalidTemp); 8345188c4b9SDmitri Gribenko if (InvalidTemp) { 8355188c4b9SDmitri Gribenko *Invalid = true; 8365188c4b9SDmitri Gribenko return StringRef(); 8375188c4b9SDmitri Gribenko } 8385188c4b9SDmitri Gribenko 8395188c4b9SDmitri Gribenko const char *Begin = File.data() + LocInfo.second; 8405188c4b9SDmitri Gribenko return StringRef(Begin, Tok.getLength()); 8415188c4b9SDmitri Gribenko } 8425188c4b9SDmitri Gribenko 8435188c4b9SDmitri Gribenko } // end namespace comments 8445188c4b9SDmitri Gribenko } // end namespace clang 8455188c4b9SDmitri Gribenko 846