Lexer.cpp revision 14bd96571ef6f0e97dc79ec4d01b547d60e8fa68
15f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===--- Lexer.cpp - C Language Family Lexer ------------------------------===// 25f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 35f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// The LLVM Compiler Infrastructure 45f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 50bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// This file is distributed under the University of Illinois Open Source 60bc735ffcfb223c0186419547abaa5c84482663eChris Lattner// License. See LICENSE.TXT for details. 75f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 85f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 95f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 10d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner// This file implements the Lexer and Token interfaces. 115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// TODO: GCC Diagnostics emitted by the lexer: 155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// PEDWARN: (form feed|vertical tab) in preprocessing directive 165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Universal characters, unicode, char mapping: 185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// WARNING: `%.*s' is not in NFKC 195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// WARNING: `%.*s' is not in NFC 205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Other: 225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// TODO: Options to support: 235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// -fexec-charset,-fwide-exec-charset 245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// 255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Lex/Lexer.h" 285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Lex/Preprocessor.h" 29500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/Lex/LexDiagnostic.h" 3055817afdf9d453a443262a733f6caf6692dca118Douglas Gregor#include "clang/Lex/CodeCompletionHandler.h" 319dc1f530c086d2c16f8cba758b0f59a5bf41323aChris Lattner#include "clang/Basic/SourceManager.h" 32f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor#include "llvm/ADT/StringSwitch.h" 33409a03671224d4e5bdab1594c43baf070148f830Chris Lattner#include "llvm/Support/Compiler.h" 345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "llvm/Support/MemoryBuffer.h" 355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include <cctype> 365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerusing namespace clang; 375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 38a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattnerstatic void InitCharacterInfo(); 395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 40dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner//===----------------------------------------------------------------------===// 41dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner// Token Class Implementation 42dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner//===----------------------------------------------------------------------===// 43dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner 441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// isObjCAtKeyword - Return true if we have an ObjC keyword identifier. 45dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattnerbool Token::isObjCAtKeyword(tok::ObjCKeywordKind objcKey) const { 46bec1c9d6f3feeec112cc8eeae90c1be29c6aaf13Douglas Gregor if (IdentifierInfo *II = getIdentifierInfo()) 47bec1c9d6f3feeec112cc8eeae90c1be29c6aaf13Douglas Gregor return II->getObjCKeywordID() == objcKey; 48bec1c9d6f3feeec112cc8eeae90c1be29c6aaf13Douglas Gregor return false; 49dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner} 50dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner 51dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner/// getObjCKeywordID - Return the ObjC keyword kind. 52dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattnertok::ObjCKeywordKind Token::getObjCKeywordID() const { 53dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner IdentifierInfo *specId = getIdentifierInfo(); 54dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner return specId ? specId->getObjCKeywordID() : tok::objc_not_keyword; 55dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner} 56dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner 5753702cd401b8fdca985aede7732c2f6a82ad9b1cChris Lattner 58dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner//===----------------------------------------------------------------------===// 59dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner// Lexer Class Implementation 60dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner//===----------------------------------------------------------------------===// 61dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner 621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpvoid Lexer::InitLexer(const char *BufStart, const char *BufPtr, 6322d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner const char *BufEnd) { 64a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner InitCharacterInfo(); 651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 6622d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner BufferStart = BufStart; 6722d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner BufferPtr = BufPtr; 6822d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner BufferEnd = BufEnd; 691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 7022d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner assert(BufEnd[0] == 0 && 7122d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner "We assume that the input buffer has a null character at the end" 7222d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner " to simplify lexing!"); 731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 7422d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner Is_PragmaLexer = false; 7534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner IsInConflictMarker = false; 7681b747b7fcc91c2fba9a3183d8fac80adbfc1d3eDouglas Gregor 7722d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner // Start of the file is a start of line. 7822d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner IsAtStartOfLine = true; 791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8022d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner // We are not after parsing a #. 8122d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner ParsingPreprocessorDirective = false; 821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8322d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner // We are not after parsing #include. 8422d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner ParsingFilename = false; 851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8622d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner // We are not in raw mode. Raw mode disables diagnostics and interpretation 8722d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner // of tokens (e.g. identifiers, thus disabling macro expansion). It is used 8822d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner // to quickly lex the tokens of the buffer, e.g. when handling a "#if 0" block 8922d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner // or otherwise skipping over tokens. 9022d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner LexingRawMode = false; 911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 9222d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner // Default to not keeping comments. 9322d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner ExtendedTokenMode = 0; 9422d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner} 9522d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner 960770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner/// Lexer constructor - Create a new lexer object for the specified buffer 970770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner/// with the specified preprocessor managing the lexing process. This lexer 980770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner/// assumes that the associated file buffer and Preprocessor objects will 990770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner/// outlive it, so it doesn't take ownership of either of them. 1006e2901407bff59aeb4cc301cc58b034723d0eb49Chris LattnerLexer::Lexer(FileID FID, const llvm::MemoryBuffer *InputFile, Preprocessor &PP) 10188d3ac1341aa016cabd966c5b113a95ac05ea43fChris Lattner : PreprocessorLexer(&PP, FID), 10288d3ac1341aa016cabd966c5b113a95ac05ea43fChris Lattner FileLoc(PP.getSourceManager().getLocForStartOfFile(FID)), 10388d3ac1341aa016cabd966c5b113a95ac05ea43fChris Lattner Features(PP.getLangOptions()) { 1041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1050770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner InitLexer(InputFile->getBufferStart(), InputFile->getBufferStart(), 1060770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner InputFile->getBufferEnd()); 1071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1080770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner // Default to keeping comments if the preprocessor wants them. 1090770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner SetCommentRetentionState(PP.getCommentRetentionState()); 1100770dabb1ae81a2a9c2e7199262067103062a0b3Chris Lattner} 111dbf388b54bb1688ee4470c61235494cbc34b8f3cChris Lattner 112168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner/// Lexer constructor - Create a new raw lexer object. This object is only 113590f0cc643274267d4d41125b62557e1d87886c3Chris Lattner/// suitable for calls to 'LexRawToken'. This lexer assumes that the text 114590f0cc643274267d4d41125b62557e1d87886c3Chris Lattner/// range will outlive it, so it doesn't take ownership of it. 115168ae2d44a443da75ea85db5f3b5081eb0bce113Chris LattnerLexer::Lexer(SourceLocation fileloc, const LangOptions &features, 116de96c0f29c4cacabe6ea577c61db87c2a85aea6cChris Lattner const char *BufStart, const char *BufPtr, const char *BufEnd) 117c6fe32a91c7372caf09152ee31a24c4b5d24deedChris Lattner : FileLoc(fileloc), Features(features) { 11822d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner 11922d91ca8d7c134eac5cc6a4869e6a84c461ad624Chris Lattner InitLexer(BufStart, BufPtr, BufEnd); 1201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 121168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner // We *are* in raw mode. 122168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner LexingRawMode = true; 1235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 1245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 125025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner/// Lexer constructor - Create a new raw lexer object. This object is only 126025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner/// suitable for calls to 'LexRawToken'. This lexer assumes that the text 127025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner/// range will outlive it, so it doesn't take ownership of it. 1286e2901407bff59aeb4cc301cc58b034723d0eb49Chris LattnerLexer::Lexer(FileID FID, const llvm::MemoryBuffer *FromFile, 1296e2901407bff59aeb4cc301cc58b034723d0eb49Chris Lattner const SourceManager &SM, const LangOptions &features) 130025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner : FileLoc(SM.getLocForStartOfFile(FID)), Features(features) { 131025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner 1321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump InitLexer(FromFile->getBufferStart(), FromFile->getBufferStart(), 133025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner FromFile->getBufferEnd()); 1341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 135025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner // We *are* in raw mode. 136025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner LexingRawMode = true; 137025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner} 138025c3a66402fb713c2d9bf5dc174ff264765379aChris Lattner 13942e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// Create_PragmaLexer: Lexer constructor - Create a new lexer object for 14042e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// _Pragma expansion. This has a variety of magic semantics that this method 14142e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// sets up. It returns a new'd Lexer that must be delete'd when done. 14242e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// 14342e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// On entrance to this routine, TokStartLoc is a macro location which has a 14442e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// spelling loc that indicates the bytes to be lexed for the token and an 14542e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// instantiation location that indicates where all lexed tokens should be 14642e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// "expanded from". 14742e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// 14842e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// FIXME: It would really be nice to make _Pragma just be a wrapper around a 14942e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// normal lexer that remaps tokens as they fly by. This would require making 15042e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// Preprocessor::Lex virtual. Given that, we could just dump in a magic lexer 15142e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// interface that could handle this stuff. This would pull GetMappedTokenLoc 15242e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// out of the critical path of the lexer! 15342e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner/// 1541eb4433ac451dc16f4133a88af2d002ac26c58efMike StumpLexer *Lexer::Create_PragmaLexer(SourceLocation SpellingLoc, 155e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner SourceLocation InstantiationLocStart, 156e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner SourceLocation InstantiationLocEnd, 157bcc2a67e5180612417727cbdd8afd0f79fdf726dChris Lattner unsigned TokLen, Preprocessor &PP) { 15842e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner SourceManager &SM = PP.getSourceManager(); 15942e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner 16042e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // Create the lexer as if we were going to lex the file normally. 161a11d61793341fea195c29a0dab3fbd74f2b39a8cChris Lattner FileID SpellingFID = SM.getFileID(SpellingLoc); 1626e2901407bff59aeb4cc301cc58b034723d0eb49Chris Lattner const llvm::MemoryBuffer *InputFile = SM.getBuffer(SpellingFID); 1636e2901407bff59aeb4cc301cc58b034723d0eb49Chris Lattner Lexer *L = new Lexer(SpellingFID, InputFile, PP); 1641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 16542e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // Now that the lexer is created, change the start/end locations so that we 16642e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // just lex the subsection of the file that we want. This is lexing from a 16742e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // scratch buffer. 16842e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner const char *StrData = SM.getCharacterData(SpellingLoc); 1691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 17042e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner L->BufferPtr = StrData; 17142e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner L->BufferEnd = StrData+TokLen; 1721fa495304c81e03f07f278a47b5efe9317104aabChris Lattner assert(L->BufferEnd[0] == 0 && "Buffer is not nul terminated!"); 17342e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner 17442e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // Set the SourceLocation with the remapping information. This ensures that 17542e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // GetMappedTokenLoc will remap the tokens as they are lexed. 176de7aeefc5573d669ed476d7bda7a8940d3bcadb7Chris Lattner L->FileLoc = SM.createInstantiationLoc(SM.getLocForStartOfFile(SpellingFID), 177e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner InstantiationLocStart, 178e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner InstantiationLocEnd, TokLen); 1791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18042e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // Ensure that the lexer thinks it is inside a directive, so that end \n will 18142e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // return an EOM token. 18242e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner L->ParsingPreprocessorDirective = true; 1831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18442e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner // This lexer really is for _Pragma. 18542e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner L->Is_PragmaLexer = true; 18642e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner return L; 18742e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner} 18842e00d19b0dac64732eb5449d52a076282fcbf77Chris Lattner 189168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner 1905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// Stringify - Convert the specified string into a C string, with surrounding 1915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ""'s, and with escaped \ and " characters. 1925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstd::string Lexer::Stringify(const std::string &Str, bool Charify) { 1935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer std::string Result = Str; 1945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char Quote = Charify ? '\'' : '"'; 1955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer for (unsigned i = 0, e = Result.size(); i != e; ++i) { 1965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Result[i] == '\\' || Result[i] == Quote) { 1975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Result.insert(Result.begin()+i, '\\'); 1985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++i; ++e; 1995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 2005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 2015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return Result; 2025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 2035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 204d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner/// Stringify - Convert the specified string into a C string by escaping '\' 205d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner/// and " characters. This does not add surrounding ""'s to the string. 206d8e3083840fef752d11ca183f42786470ed061e3Chris Lattnervoid Lexer::Stringify(llvm::SmallVectorImpl<char> &Str) { 207d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner for (unsigned i = 0, e = Str.size(); i != e; ++i) { 208d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner if (Str[i] == '\\' || Str[i] == '"') { 209d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner Str.insert(Str.begin()+i, '\\'); 210d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner ++i; ++e; 211d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner } 212d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner } 213d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner} 214d8e3083840fef752d11ca183f42786470ed061e3Chris Lattner 21533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorstatic bool isWhitespace(unsigned char c); 2165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 2179a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner/// MeasureTokenLength - Relex the token at the specified location and return 2189a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner/// its length in bytes in the input file. If the token needs cleaning (e.g. 2199a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner/// includes a trigraph or an escaped newline) then this count includes bytes 2209a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner/// that are part of that. 2219a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattnerunsigned Lexer::MeasureTokenLength(SourceLocation Loc, 2222c78b873f4f3823ae859c15674cb3d76c8554113Chris Lattner const SourceManager &SM, 2232c78b873f4f3823ae859c15674cb3d76c8554113Chris Lattner const LangOptions &LangOpts) { 2249a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner // TODO: this could be special cased for common tokens like identifiers, ')', 2259a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner // etc to make this faster, if it mattered. Just look at StrData[0] to handle 2261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump // all obviously single-char tokens. This could use 2279a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner // Lexer::isObviouslySimpleCharacter for example to handle identifiers or 2289a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner // something. 229de7aeefc5573d669ed476d7bda7a8940d3bcadb7Chris Lattner 230de7aeefc5573d669ed476d7bda7a8940d3bcadb7Chris Lattner // If this comes from a macro expansion, we really do want the macro name, not 231de7aeefc5573d669ed476d7bda7a8940d3bcadb7Chris Lattner // the token this macro expanded to. 232363fdc29656cc03c1817268888f95e6343470aa8Chris Lattner Loc = SM.getInstantiationLoc(Loc); 233363fdc29656cc03c1817268888f95e6343470aa8Chris Lattner std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(Loc); 234f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor bool Invalid = false; 235f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer llvm::StringRef Buffer = SM.getBufferData(LocInfo.first, &Invalid); 236f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor if (Invalid) 237aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor return 0; 238f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer 239f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer const char *StrData = Buffer.data()+LocInfo.second; 2408350394c65b81bba3986dfe44ae17423873741deChris Lattner 24133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor if (isWhitespace(StrData[0])) 24233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor return 0; 24333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor 2449a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner // Create a lexer starting at the beginning of this token. 245c3526d89ef9c31639ec8b25180cfb22354344241Sebastian Redl Lexer TheLexer(SM.getLocForStartOfFile(LocInfo.first), LangOpts, 246c3526d89ef9c31639ec8b25180cfb22354344241Sebastian Redl Buffer.begin(), StrData, Buffer.end()); 24739de7409bffb6b725a8aa64f0ba77ab51e8c9eb3Chris Lattner TheLexer.SetCommentRetentionState(true); 2489a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner Token TheTok; 249590f0cc643274267d4d41125b62557e1d87886c3Chris Lattner TheLexer.LexFromRawLexer(TheTok); 2509a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner return TheTok.getLength(); 2519a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner} 2529a6119437672f42be5f50c3fe89fe843b1bfa5b5Chris Lattner 253a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorSourceLocation Lexer::GetBeginningOfToken(SourceLocation Loc, 254a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor const SourceManager &SM, 255a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor const LangOptions &LangOpts) { 256a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(Loc); 257a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor bool Invalid = false; 258a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor llvm::StringRef Buffer = SM.getBufferData(LocInfo.first, &Invalid); 259a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor if (Invalid) 260a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor return Loc; 261a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 262a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // Back up from the current location until we hit the beginning of a line 263a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // (or the buffer). We'll relex from that point. 264a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor const char *BufStart = Buffer.data(); 265a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor const char *StrData = BufStart+LocInfo.second; 266a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor if (StrData[0] == '\n' || StrData[0] == '\r') 267a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor return Loc; 268a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 269a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor const char *LexStart = StrData; 270a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor while (LexStart != BufStart) { 271a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor if (LexStart[0] == '\n' || LexStart[0] == '\r') { 272a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor ++LexStart; 273a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor break; 274a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor } 275a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 276a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor --LexStart; 277a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor } 278a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 279a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // Create a lexer starting at the beginning of this token. 280a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor SourceLocation LexerStartLoc = Loc.getFileLocWithOffset(-LocInfo.second); 281a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor Lexer TheLexer(LexerStartLoc, LangOpts, BufStart, LexStart, Buffer.end()); 282a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor TheLexer.SetCommentRetentionState(true); 283a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 284a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // Lex tokens until we find the token that contains the source location. 285a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor Token TheTok; 286a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor do { 287a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor TheLexer.LexFromRawLexer(TheTok); 288a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 289a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor if (TheLexer.getBufferLocation() > StrData) { 290a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // Lexing this token has taken the lexer past the source location we're 291a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // looking for. If the current token encompasses our source location, 292a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // return the beginning of that token. 293a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor if (TheLexer.getBufferLocation() - TheTok.getLength() <= StrData) 294a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor return TheTok.getLocation(); 295a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 296a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // We ended up skipping over the source location entirely, which means 297a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // that it points into whitespace. We're done here. 298a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor break; 299a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor } 300a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor } while (TheTok.getKind() != tok::eof); 301a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 302a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor // We've passed our source location; just return the original source location. 303a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor return Loc; 304a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} 305a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor 306f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregornamespace { 307f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor enum PreambleDirectiveKind { 308f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor PDK_Skipped, 309f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor PDK_StartIf, 310f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor PDK_EndIf, 311f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor PDK_Unknown 312f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor }; 313f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor} 314f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 315f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregorstd::pair<unsigned, bool> 316df95a13ec73d2cdaea79555cb412d767f4963120Douglas GregorLexer::ComputePreamble(const llvm::MemoryBuffer *Buffer, unsigned MaxLines) { 317f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // Create a lexer starting at the beginning of the file. Note that we use a 318f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // "fake" file source location at offset 1 so that the lexer will track our 319f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // position within the file. 320f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor const unsigned StartOffset = 1; 321f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor SourceLocation StartLoc = SourceLocation::getFromRawEncoding(StartOffset); 322f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor LangOptions LangOpts; 323f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor Lexer TheLexer(StartLoc, LangOpts, Buffer->getBufferStart(), 324f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor Buffer->getBufferStart(), Buffer->getBufferEnd()); 325f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 326f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor bool InPreprocessorDirective = false; 327f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor Token TheTok; 328f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor Token IfStartTok; 329f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor unsigned IfCount = 0; 330df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor unsigned Line = 0; 331df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor 332f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor do { 333f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor TheLexer.LexFromRawLexer(TheTok); 334f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 335f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor if (InPreprocessorDirective) { 336f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // If we've hit the end of the file, we're done. 337f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor if (TheTok.getKind() == tok::eof) { 338f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor InPreprocessorDirective = false; 339f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor break; 340f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor } 341f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 342f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // If we haven't hit the end of the preprocessor directive, skip this 343f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // token. 344f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor if (!TheTok.isAtStartOfLine()) 345f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor continue; 346f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 347f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // We've passed the end of the preprocessor directive, and will look 348f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // at this token again below. 349f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor InPreprocessorDirective = false; 350f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor } 351f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 352df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor // Keep track of the # of lines in the preamble. 353df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor if (TheTok.isAtStartOfLine()) { 354df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor ++Line; 355df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor 356df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor // If we were asked to limit the number of lines in the preamble, 357df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor // and we're about to exceed that limit, we're done. 358df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor if (MaxLines && Line >= MaxLines) 359df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor break; 360df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor } 361df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor 362f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // Comments are okay; skip over them. 363f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor if (TheTok.getKind() == tok::comment) 364f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor continue; 365f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 366f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor if (TheTok.isAtStartOfLine() && TheTok.getKind() == tok::hash) { 367f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // This is the start of a preprocessor directive. 368f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor Token HashTok = TheTok; 369f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor InPreprocessorDirective = true; 370f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 371f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // Figure out which direective this is. Since we're lexing raw tokens, 372f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // we don't have an identifier table available. Instead, just look at 373f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // the raw identifier to recognize and categorize preprocessor directives. 374f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor TheLexer.LexFromRawLexer(TheTok); 375f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor if (TheTok.getKind() == tok::identifier && !TheTok.needsCleaning()) { 376f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor const char *IdStart = Buffer->getBufferStart() 377f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor + TheTok.getLocation().getRawEncoding() - 1; 378f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor llvm::StringRef Keyword(IdStart, TheTok.getLength()); 379f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor PreambleDirectiveKind PDK 380f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor = llvm::StringSwitch<PreambleDirectiveKind>(Keyword) 381f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("include", PDK_Skipped) 382f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("__include_macros", PDK_Skipped) 383f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("define", PDK_Skipped) 384f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("undef", PDK_Skipped) 385f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("line", PDK_Skipped) 386f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("error", PDK_Skipped) 387f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("pragma", PDK_Skipped) 388f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("import", PDK_Skipped) 389f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("include_next", PDK_Skipped) 390f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("warning", PDK_Skipped) 391f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("ident", PDK_Skipped) 392f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("sccs", PDK_Skipped) 393f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("assert", PDK_Skipped) 394f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("unassert", PDK_Skipped) 395f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("if", PDK_StartIf) 396f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("ifdef", PDK_StartIf) 397f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("ifndef", PDK_StartIf) 398f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("elif", PDK_Skipped) 399f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("else", PDK_Skipped) 400f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Case("endif", PDK_EndIf) 401f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor .Default(PDK_Unknown); 402f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 403f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor switch (PDK) { 404f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor case PDK_Skipped: 405f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor continue; 406f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 407f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor case PDK_StartIf: 408f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor if (IfCount == 0) 409f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor IfStartTok = HashTok; 410f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 411f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor ++IfCount; 412f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor continue; 413f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 414f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor case PDK_EndIf: 415f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // Mismatched #endif. The preamble ends here. 416f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor if (IfCount == 0) 417f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor break; 418f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 419f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor --IfCount; 420f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor continue; 421f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 422f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor case PDK_Unknown: 423f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // We don't know what this directive is; stop at the '#'. 424f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor break; 425f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor } 426f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor } 427f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 428f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // We only end up here if we didn't recognize the preprocessor 429f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // directive or it was one that can't occur in the preamble at this 430f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor // point. Roll back the current token to the location of the '#'. 431f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor InPreprocessorDirective = false; 432f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor TheTok = HashTok; 433f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor } 434f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 435df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor // We hit a token that we don't recognize as being in the 436df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor // "preprocessing only" part of the file, so we're no longer in 437df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor // the preamble. 438f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor break; 439f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor } while (true); 440f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 441f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor SourceLocation End = IfCount? IfStartTok.getLocation() : TheTok.getLocation(); 442f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor return std::make_pair(End.getRawEncoding() - StartLoc.getRawEncoding(), 443f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor IfCount? IfStartTok.isAtStartOfLine() 444f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor : TheTok.isAtStartOfLine()); 445f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor} 446f033f1da4a34f8df6e95e9929dc04ff54bb8fb01Douglas Gregor 4475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 4485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Character information. 4495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 4505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 4515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerenum { 4525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CHAR_HORZ_WS = 0x01, // ' ', '\t', '\f', '\v'. Note, no '\0' 4535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CHAR_VERT_WS = 0x02, // '\r', '\n' 4545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CHAR_LETTER = 0x04, // a-z,A-Z 4555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CHAR_NUMBER = 0x08, // 0-9 4565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CHAR_UNDER = 0x10, // _ 4575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CHAR_PERIOD = 0x20 // . 4585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}; 4595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 46003b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner// Statically initialize CharInfo table based on ASCII character set 46103b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner// Reference: FreeBSD 7.2 /usr/share/misc/ascii 462a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattnerstatic const unsigned char CharInfo[256] = 46303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner{ 46403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner// 0 NUL 1 SOH 2 STX 3 ETX 46503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner// 4 EOT 5 ENQ 6 ACK 7 BEL 46603b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 46703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 46803b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner// 8 BS 9 HT 10 NL 11 VT 46903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//12 NP 13 CR 14 SO 15 SI 47003b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , CHAR_HORZ_WS, CHAR_VERT_WS, CHAR_HORZ_WS, 47103b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_HORZ_WS, CHAR_VERT_WS, 0 , 0 , 47203b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//16 DLE 17 DC1 18 DC2 19 DC3 47303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//20 DC4 21 NAK 22 SYN 23 ETB 47403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 47503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 47603b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//24 CAN 25 EM 26 SUB 27 ESC 47703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//28 FS 29 GS 30 RS 31 US 47803b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 47903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 48003b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//32 SP 33 ! 34 " 35 # 48103b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//36 $ 37 % 38 & 39 ' 48203b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_HORZ_WS, 0 , 0 , 0 , 48303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 48403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//40 ( 41 ) 42 * 43 + 48503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//44 , 45 - 46 . 47 / 48603b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 48703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , CHAR_PERIOD , 0 , 48803b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//48 0 49 1 50 2 51 3 48903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//52 4 53 5 54 6 55 7 49003b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_NUMBER , CHAR_NUMBER , CHAR_NUMBER , CHAR_NUMBER , 49103b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_NUMBER , CHAR_NUMBER , CHAR_NUMBER , CHAR_NUMBER , 49203b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//56 8 57 9 58 : 59 ; 49303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//60 < 61 = 62 > 63 ? 49403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_NUMBER , CHAR_NUMBER , 0 , 0 , 49503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 , 49603b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//64 @ 65 A 66 B 67 C 49703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//68 D 69 E 70 F 71 G 49803b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 49903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 50003b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//72 H 73 I 74 J 75 K 50103b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//76 L 77 M 78 N 79 O 50203b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 50303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 50403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//80 P 81 Q 82 R 83 S 50503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//84 T 85 U 86 V 87 W 50603b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 50703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 50803b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//88 X 89 Y 90 Z 91 [ 50903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//92 \ 93 ] 94 ^ 95 _ 51003b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 0 , 51103b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , CHAR_UNDER , 51203b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//96 ` 97 a 98 b 99 c 51303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//100 d 101 e 102 f 103 g 51403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 51503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 51603b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//104 h 105 i 106 j 107 k 51703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//108 l 109 m 110 n 111 o 51803b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 51903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 52003b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//112 p 113 q 114 r 115 s 52103b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//116 t 117 u 118 v 119 w 52203b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 52303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 52403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//120 x 121 y 122 z 123 { 52503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner//124 | 125 } 126 ~ 127 DEL 52603b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner CHAR_LETTER , CHAR_LETTER , CHAR_LETTER , 0 , 52703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 0 , 0 , 0 , 0 52803b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner}; 52903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 530a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattnerstatic void InitCharacterInfo() { 5315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer static bool isInited = false; 5325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (isInited) return; 53303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner // check the statically-initialized CharInfo table 53403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_HORZ_WS == CharInfo[(int)' ']); 53503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_HORZ_WS == CharInfo[(int)'\t']); 53603b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_HORZ_WS == CharInfo[(int)'\f']); 53703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_HORZ_WS == CharInfo[(int)'\v']); 53803b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_VERT_WS == CharInfo[(int)'\n']); 53903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_VERT_WS == CharInfo[(int)'\r']); 54003b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_UNDER == CharInfo[(int)'_']); 54103b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_PERIOD == CharInfo[(int)'.']); 54203b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner for (unsigned i = 'a'; i <= 'z'; ++i) { 54303b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_LETTER == CharInfo[i]); 54403b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_LETTER == CharInfo[i+'A'-'a']); 54503b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner } 5465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer for (unsigned i = '0'; i <= '9'; ++i) 54703b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner assert(CHAR_NUMBER == CharInfo[i]); 5487b682656310b0d7e94a98d982444bc7ddedc653fSteve Naroff 54903b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner isInited = true; 5505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 5515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 55203b986660cc9ae7a08588f99913f2d91e359a65fChris Lattner 5535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// isIdentifierBody - Return true if this is the body character of an 5545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// identifier, which is [a-zA-Z0-9_]. 5555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic inline bool isIdentifierBody(unsigned char c) { 55695c062b94e2af5380617cf5dc115497ae7a5f817Hartmut Kaiser return (CharInfo[c] & (CHAR_LETTER|CHAR_NUMBER|CHAR_UNDER)) ? true : false; 5575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 5585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 5595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// isHorizontalWhitespace - Return true if this character is horizontal 5605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// whitespace: ' ', '\t', '\f', '\v'. Note that this returns false for '\0'. 5615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic inline bool isHorizontalWhitespace(unsigned char c) { 56295c062b94e2af5380617cf5dc115497ae7a5f817Hartmut Kaiser return (CharInfo[c] & CHAR_HORZ_WS) ? true : false; 5635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 5645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 5655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// isWhitespace - Return true if this character is horizontal or vertical 5665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// whitespace: ' ', '\t', '\f', '\v', '\n', '\r'. Note that this returns false 5675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// for '\0'. 5685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic inline bool isWhitespace(unsigned char c) { 56995c062b94e2af5380617cf5dc115497ae7a5f817Hartmut Kaiser return (CharInfo[c] & (CHAR_HORZ_WS|CHAR_VERT_WS)) ? true : false; 5705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 5715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 5725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// isNumberBody - Return true if this is the body character of an 5735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// preprocessing number, which is [a-zA-Z0-9_.]. 5745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic inline bool isNumberBody(unsigned char c) { 5751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump return (CharInfo[c] & (CHAR_LETTER|CHAR_NUMBER|CHAR_UNDER|CHAR_PERIOD)) ? 57695c062b94e2af5380617cf5dc115497ae7a5f817Hartmut Kaiser true : false; 5775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 5785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 5795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 5805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 5815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Diagnostics forwarding code. 5825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 5835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 584409a03671224d4e5bdab1594c43baf070148f830Chris Lattner/// GetMappedTokenLoc - If lexing out of a 'mapped buffer', where we pretend the 585409a03671224d4e5bdab1594c43baf070148f830Chris Lattner/// lexer buffer was all instantiated at a single point, perform the mapping. 586409a03671224d4e5bdab1594c43baf070148f830Chris Lattner/// This is currently only used for _Pragma implementation, so it is the slow 587409a03671224d4e5bdab1594c43baf070148f830Chris Lattner/// path of the hot getSourceLocation method. Do not allow it to be inlined. 58814bd96571ef6f0e97dc79ec4d01b547d60e8fa68Chandler Carruthstatic LLVM_ATTRIBUTE_NOINLINE SourceLocation GetMappedTokenLoc( 58914bd96571ef6f0e97dc79ec4d01b547d60e8fa68Chandler Carruth Preprocessor &PP, SourceLocation FileLoc, unsigned CharNo, unsigned TokLen); 590409a03671224d4e5bdab1594c43baf070148f830Chris Lattnerstatic SourceLocation GetMappedTokenLoc(Preprocessor &PP, 591409a03671224d4e5bdab1594c43baf070148f830Chris Lattner SourceLocation FileLoc, 592de7aeefc5573d669ed476d7bda7a8940d3bcadb7Chris Lattner unsigned CharNo, unsigned TokLen) { 593e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner assert(FileLoc.isMacroID() && "Must be an instantiation"); 5941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 595409a03671224d4e5bdab1594c43baf070148f830Chris Lattner // Otherwise, we're lexing "mapped tokens". This is used for things like 596409a03671224d4e5bdab1594c43baf070148f830Chris Lattner // _Pragma handling. Combine the instantiation location of FileLoc with the 597df7c17a8d02fe09a3466786bae3e40fc3252687aChris Lattner // spelling location. 598e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner SourceManager &SM = PP.getSourceManager(); 5991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 600f7cf85b330bedd2877e1371fb0a83e99751ae162Chris Lattner // Create a new SLoc which is expanded from Instantiation(FileLoc) but whose 601df7c17a8d02fe09a3466786bae3e40fc3252687aChris Lattner // characters come from spelling(FileLoc)+Offset. 602e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner SourceLocation SpellingLoc = SM.getSpellingLoc(FileLoc); 603bcc2a67e5180612417727cbdd8afd0f79fdf726dChris Lattner SpellingLoc = SpellingLoc.getFileLocWithOffset(CharNo); 6041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 605e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner // Figure out the expansion loc range, which is the range covered by the 606e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner // original _Pragma(...) sequence. 607e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner std::pair<SourceLocation,SourceLocation> II = 608e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner SM.getImmediateInstantiationRange(FileLoc); 6091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 610e7fb48466afcbf2c4ccdfa658824282fdc3c512cChris Lattner return SM.createInstantiationLoc(SpellingLoc, II.first, II.second, TokLen); 611409a03671224d4e5bdab1594c43baf070148f830Chris Lattner} 612409a03671224d4e5bdab1594c43baf070148f830Chris Lattner 6135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// getSourceLocation - Return a source location identifier for the specified 6145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// offset in the current file. 615de7aeefc5573d669ed476d7bda7a8940d3bcadb7Chris LattnerSourceLocation Lexer::getSourceLocation(const char *Loc, 616de7aeefc5573d669ed476d7bda7a8940d3bcadb7Chris Lattner unsigned TokLen) const { 617448cec4c1c3705f6f49ffdefb58a7329942a2dd8Chris Lattner assert(Loc >= BufferStart && Loc <= BufferEnd && 6185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer "Location out of range for this buffer!"); 6199dc1f530c086d2c16f8cba758b0f59a5bf41323aChris Lattner 6209dc1f530c086d2c16f8cba758b0f59a5bf41323aChris Lattner // In the normal case, we're just lexing from a simple file buffer, return 6219dc1f530c086d2c16f8cba758b0f59a5bf41323aChris Lattner // the file id from FileLoc with the offset specified. 622448cec4c1c3705f6f49ffdefb58a7329942a2dd8Chris Lattner unsigned CharNo = Loc-BufferStart; 6239dc1f530c086d2c16f8cba758b0f59a5bf41323aChris Lattner if (FileLoc.isFileID()) 624bcc2a67e5180612417727cbdd8afd0f79fdf726dChris Lattner return FileLoc.getFileLocWithOffset(CharNo); 6251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 6262b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner // Otherwise, this is the _Pragma lexer case, which pretends that all of the 6272b2453a7d8fe732561795431f39ceb2b2a832d84Chris Lattner // tokens are lexed from where the _Pragma was defined. 628168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner assert(PP && "This doesn't work on raw lexers"); 629de7aeefc5573d669ed476d7bda7a8940d3bcadb7Chris Lattner return GetMappedTokenLoc(*PP, FileLoc, CharNo, TokLen); 6305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 6315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 6325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// Diag - Forwarding function for diagnostics. This translate a source 6335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// position in the current buffer into a SourceLocation object for rendering. 6343cbfe2c4159e0a219ae660d50625c013aa4afbd0Chris LattnerDiagnosticBuilder Lexer::Diag(const char *Loc, unsigned DiagID) const { 6353692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner return PP->Diag(getSourceLocation(Loc), DiagID); 6365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 6375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 6385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 6395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Trigraph and Escaped Newline Handling Code. 6405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 6415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 6425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// GetTrigraphCharForLetter - Given a character that occurs after a ?? pair, 6435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// return the decoded trigraph letter it corresponds to, or '\0' if nothing. 6445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic char GetTrigraphCharForLetter(char Letter) { 6455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer switch (Letter) { 6465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer default: return 0; 6475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '=': return '#'; 6485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case ')': return ']'; 6495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '(': return '['; 6505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '!': return '|'; 6515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\'': return '^'; 6525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '>': return '}'; 6535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '/': return '\\'; 6545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '<': return '{'; 6555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '-': return '~'; 6565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 6575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 6585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 6595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// DecodeTrigraphChar - If the specified character is a legal trigraph when 6605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// prefixed with ??, emit a trigraph warning. If trigraphs are enabled, 6615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// return the result character. Finally, emit a warning about trigraph use 6625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// whether trigraphs are enabled or not. 6635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic char DecodeTrigraphChar(const char *CP, Lexer *L) { 6645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char Res = GetTrigraphCharForLetter(*CP); 6653692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner if (!Res || !L) return Res; 6661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 6673692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner if (!L->getFeatures().Trigraphs) { 66874d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!L->isLexingRawMode()) 66974d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner L->Diag(CP-2, diag::trigraph_ignored); 6703692b09faa9fe346f39bc922db6dce48cdcc3f63Chris Lattner return 0; 6715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 6721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 67374d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!L->isLexingRawMode()) 674476d8b863cb65b2b5833235d97315cdb46e6f5aaBenjamin Kramer L->Diag(CP-2, diag::trigraph_converted) << llvm::StringRef(&Res, 1); 6755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return Res; 6765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 6775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 67824f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner/// getEscapedNewLineSize - Return the size of the specified escaped newline, 67924f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner/// or 0 if it is not an escaped newline. P[-1] is known to be a "\" or a 6801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// trigraph equivalent on entry to this function. 68124f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattnerunsigned Lexer::getEscapedNewLineSize(const char *Ptr) { 68224f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner unsigned Size = 0; 68324f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner while (isWhitespace(Ptr[Size])) { 68424f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner ++Size; 6851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 68624f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner if (Ptr[Size-1] != '\n' && Ptr[Size-1] != '\r') 68724f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner continue; 68824f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner 68924f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner // If this is a \r\n or \n\r, skip the other half. 69024f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner if ((Ptr[Size] == '\r' || Ptr[Size] == '\n') && 69124f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner Ptr[Size-1] != Ptr[Size]) 69224f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner ++Size; 6931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 69424f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner return Size; 6951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump } 6961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 69724f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner // Not an escaped newline, must be a \t or something else. 69824f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner return 0; 69924f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner} 70024f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner 701033749571f8d4c804eeb357c70b06424aa24503bChris Lattner/// SkipEscapedNewLines - If P points to an escaped newline (or a series of 702033749571f8d4c804eeb357c70b06424aa24503bChris Lattner/// them), skip over them and return the first non-escaped-newline found, 703033749571f8d4c804eeb357c70b06424aa24503bChris Lattner/// otherwise return P. 704033749571f8d4c804eeb357c70b06424aa24503bChris Lattnerconst char *Lexer::SkipEscapedNewLines(const char *P) { 705033749571f8d4c804eeb357c70b06424aa24503bChris Lattner while (1) { 706033749571f8d4c804eeb357c70b06424aa24503bChris Lattner const char *AfterEscape; 707033749571f8d4c804eeb357c70b06424aa24503bChris Lattner if (*P == '\\') { 708033749571f8d4c804eeb357c70b06424aa24503bChris Lattner AfterEscape = P+1; 709033749571f8d4c804eeb357c70b06424aa24503bChris Lattner } else if (*P == '?') { 710033749571f8d4c804eeb357c70b06424aa24503bChris Lattner // If not a trigraph for escape, bail out. 711033749571f8d4c804eeb357c70b06424aa24503bChris Lattner if (P[1] != '?' || P[2] != '/') 712033749571f8d4c804eeb357c70b06424aa24503bChris Lattner return P; 713033749571f8d4c804eeb357c70b06424aa24503bChris Lattner AfterEscape = P+3; 714033749571f8d4c804eeb357c70b06424aa24503bChris Lattner } else { 715033749571f8d4c804eeb357c70b06424aa24503bChris Lattner return P; 716033749571f8d4c804eeb357c70b06424aa24503bChris Lattner } 7171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 718033749571f8d4c804eeb357c70b06424aa24503bChris Lattner unsigned NewLineSize = Lexer::getEscapedNewLineSize(AfterEscape); 719033749571f8d4c804eeb357c70b06424aa24503bChris Lattner if (NewLineSize == 0) return P; 720033749571f8d4c804eeb357c70b06424aa24503bChris Lattner P = AfterEscape+NewLineSize; 721033749571f8d4c804eeb357c70b06424aa24503bChris Lattner } 722033749571f8d4c804eeb357c70b06424aa24503bChris Lattner} 723033749571f8d4c804eeb357c70b06424aa24503bChris Lattner 72424f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner 7255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// getCharAndSizeSlow - Peek a single 'character' from the specified buffer, 7265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// get its size, and return it. This is tricky in several cases: 7275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 1. If currently at the start of a trigraph, we warn about the trigraph, 7285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// then either return the trigraph (skipping 3 chars) or the '?', 7295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// depending on whether trigraphs are enabled or not. 7305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 2. If this is an escaped newline (potentially with whitespace between 7315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// the backslash and newline), implicitly skip the newline and return 7325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// the char after it. 7335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 3. If this is a UCN, return it. FIXME: C++ UCN's? 7345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 7355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// This handles the slow/uncommon case of the getCharAndSize method. Here we 7365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// know that we can accumulate into Size, and that we have already incremented 7375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// Ptr by Size bytes. 7385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 7395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// NOTE: When this method is updated, getCharAndSizeSlowNoWarn (below) should 7405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// be updated to match. 7415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 7425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerchar Lexer::getCharAndSizeSlow(const char *Ptr, unsigned &Size, 743d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Token *Tok) { 7445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we have a slash, look for an escaped newline. 7455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Ptr[0] == '\\') { 7465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++Size; 7475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++Ptr; 7485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerSlash: 7495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Common case, backslash-char where the char is not whitespace. 7505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (!isWhitespace(Ptr[0])) return '\\'; 7511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 7525636a3b6ece2c1f413464b72545e08eb0b7f06e4Chris Lattner // See if we have optional whitespace characters between the slash and 7535636a3b6ece2c1f413464b72545e08eb0b7f06e4Chris Lattner // newline. 75424f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner if (unsigned EscapedNewLineSize = getEscapedNewLineSize(Ptr)) { 75524f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner // Remember that this token needs to be cleaned. 75624f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner if (Tok) Tok->setFlag(Token::NeedsCleaning); 75724f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner 75824f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner // Warn if there was whitespace between the backslash and newline. 7595636a3b6ece2c1f413464b72545e08eb0b7f06e4Chris Lattner if (Ptr[0] != '\n' && Ptr[0] != '\r' && Tok && !isLexingRawMode()) 76024f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner Diag(Ptr, diag::backslash_newline_space); 7611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 76224f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner // Found backslash<whitespace><newline>. Parse the char after it. 76324f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner Size += EscapedNewLineSize; 76424f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner Ptr += EscapedNewLineSize; 76524f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner // Use slow version to accumulate a correct size field. 76624f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner return getCharAndSizeSlow(Ptr, Size, Tok); 76724f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner } 7681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 7695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, this is not an escaped newline, just return the slash. 7705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return '\\'; 7715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 7721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 7735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is a trigraph, process it. 7745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Ptr[0] == '?' && Ptr[1] == '?') { 7755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is actually a legal trigraph (not something like "??x"), emit 7765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // a trigraph warning. If so, and if trigraphs are enabled, return it. 7775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (char C = DecodeTrigraphChar(Ptr+2, Tok ? this : 0)) { 7785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Remember that this token needs to be cleaned. 779d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner if (Tok) Tok->setFlag(Token::NeedsCleaning); 7805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 7815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Ptr += 3; 7825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Size += 3; 7835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == '\\') goto Slash; 7845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return C; 7855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 7865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 7871eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 7885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is neither, return a single character. 7895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++Size; 7905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return *Ptr; 7915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 7925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 7935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 7945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// getCharAndSizeSlowNoWarn - Handle the slow/uncommon case of the 7955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// getCharAndSizeNoWarn method. Here we know that we can accumulate into Size, 7965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// and that we have already incremented Ptr by Size bytes. 7975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// 7985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// NOTE: When this method is updated, getCharAndSizeSlow (above) should 7995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// be updated to match. 8005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerchar Lexer::getCharAndSizeSlowNoWarn(const char *Ptr, unsigned &Size, 8015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const LangOptions &Features) { 8025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we have a slash, look for an escaped newline. 8035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Ptr[0] == '\\') { 8045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++Size; 8055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++Ptr; 8065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerSlash: 8075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Common case, backslash-char where the char is not whitespace. 8085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (!isWhitespace(Ptr[0])) return '\\'; 8091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // See if we have optional whitespace characters followed by a newline. 81124f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner if (unsigned EscapedNewLineSize = getEscapedNewLineSize(Ptr)) { 81224f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner // Found backslash<whitespace><newline>. Parse the char after it. 81324f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner Size += EscapedNewLineSize; 81424f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner Ptr += EscapedNewLineSize; 8151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 81624f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner // Use slow version to accumulate a correct size field. 81724f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner return getCharAndSizeSlowNoWarn(Ptr, Size, Features); 81824f0e48c0aa62f2268e061aad70f9b19a59e7b52Chris Lattner } 8191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, this is not an escaped newline, just return the slash. 8215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return '\\'; 8225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 8231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is a trigraph, process it. 8255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Features.Trigraphs && Ptr[0] == '?' && Ptr[1] == '?') { 8265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is actually a legal trigraph (not something like "??x"), return 8275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // it. 8285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (char C = GetTrigraphCharForLetter(Ptr[2])) { 8295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Ptr += 3; 8305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Size += 3; 8315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == '\\') goto Slash; 8325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return C; 8335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 8345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 8351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is neither, return a single character. 8375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++Size; 8385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return *Ptr; 8395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 8405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 8415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 8425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Helper methods for lexing. 8435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 8445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 845f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor/// \brief Routine that indiscriminately skips bytes in the source file. 846f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregorvoid Lexer::SkipBytes(unsigned Bytes, bool StartOfLine) { 847f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor BufferPtr += Bytes; 848f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor if (BufferPtr > BufferEnd) 849f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor BufferPtr = BufferEnd; 850f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor IsAtStartOfLine = StartOfLine; 851f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor} 852f4f6c9db68465b886ec2e596feaa6ecc782395a4Douglas Gregor 853d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Lexer::LexIdentifier(Token &Result, const char *CurPtr) { 8545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Match [_A-Za-z0-9]*, we have already matched [_A-Za-z$] 8555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer unsigned Size; 8565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer unsigned char C = *CurPtr++; 857cd991dbb12f24368753cef086c7ad3ec203c9ea6Chris Lattner while (isIdentifierBody(C)) 8585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = *CurPtr++; 859cd991dbb12f24368753cef086c7ad3ec203c9ea6Chris Lattner 8605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer --CurPtr; // Back up over the skipped character. 8615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 8625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Fast path, no $,\,? in identifier found. '\' might be an escaped newline 8635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // or UCN, and ? might be a trigraph for '\', an escaped newline or UCN. 8645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // FIXME: UCNs. 865cd991dbb12f24368753cef086c7ad3ec203c9ea6Chris Lattner // 866cd991dbb12f24368753cef086c7ad3ec203c9ea6Chris Lattner // TODO: Could merge these checks into a CharInfo flag to make the comparison 867cd991dbb12f24368753cef086c7ad3ec203c9ea6Chris Lattner // cheaper 8685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C != '\\' && C != '?' && (C != '$' || !Features.DollarIdents)) { 8695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerFinishIdentifier: 8705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *IdStart = BufferPtr; 8719e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::identifier); 8721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we are in raw mode, return this identifier raw. There is no need to 8745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // look up identifier information or attempt to macro expand it. 8755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (LexingRawMode) return; 8761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Fill in Result.IdentifierInfo, looking up the identifier in the 8785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // identifier table. 879d1186fa38166a581b51975f0382a45fc3a0733d0Chris Lattner IdentifierInfo *II = PP->LookUpIdentifierInfo(Result, IdStart); 8801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 881863c486fcb6162495a94fddf7ac8409de2638995Chris Lattner // Change the kind of this identifier to the appropriate token kind, e.g. 882863c486fcb6162495a94fddf7ac8409de2638995Chris Lattner // turning "for" into a keyword. 883863c486fcb6162495a94fddf7ac8409de2638995Chris Lattner Result.setKind(II->getTokenID()); 8841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Finally, now that we know we have an identifier, pass this off to the 8865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // preprocessor, which may macro expand it or something. 887d1186fa38166a581b51975f0382a45fc3a0733d0Chris Lattner if (II->isHandleIdentifierCase()) 8886a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner PP->HandleIdentifier(Result); 8896a170eb3ea6d6319277becabef68eb1a26bf8766Chris Lattner return; 8905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 8911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, $,\,? in identifier found. Enter slower path. 8931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 8945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getCharAndSize(CurPtr, Size); 8955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (1) { 8965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == '$') { 8975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we hit a $ and they are not supported in identifiers, we are done. 8985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (!Features.DollarIdents) goto FinishIdentifier; 8991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 9005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, emit a diagnostic and continue. 90174d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!isLexingRawMode()) 90274d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(CurPtr, diag::ext_dollar_in_identifier); 9035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, Size, Result); 9045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getCharAndSize(CurPtr, Size); 9055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer continue; 9065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (!isIdentifierBody(C)) { // FIXME: UCNs. 9075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Found end of identifier. 9085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer goto FinishIdentifier; 9095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 9115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, this character is good, consume it. 9125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, Size, Result); 9135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 9145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getCharAndSize(CurPtr, Size); 9155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (isIdentifierBody(C)) { // FIXME: UCNs. 9165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, Size, Result); 9175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getCharAndSize(CurPtr, Size); 9185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 9215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 922a75ec43d625753b4439b0d6f70bd988444c74617Douglas Gregor/// isHexaLiteral - Return true if Start points to a hex constant. 9234a551000bee716ac8b1bbe16134a53f0ad221a5aChris Lattner/// in microsoft mode (where this is supposed to be several different tokens). 9246ab55ebab20086f725c4017b48ef8d7691ef870fChris Lattnerstatic bool isHexaLiteral(const char *Start, const LangOptions &Features) { 9256ab55ebab20086f725c4017b48ef8d7691ef870fChris Lattner unsigned Size; 9266ab55ebab20086f725c4017b48ef8d7691ef870fChris Lattner char C1 = Lexer::getCharAndSizeNoWarn(Start, Size, Features); 9276ab55ebab20086f725c4017b48ef8d7691ef870fChris Lattner if (C1 != '0') 9286ab55ebab20086f725c4017b48ef8d7691ef870fChris Lattner return false; 9296ab55ebab20086f725c4017b48ef8d7691ef870fChris Lattner char C2 = Lexer::getCharAndSizeNoWarn(Start + Size, Size, Features); 9306ab55ebab20086f725c4017b48ef8d7691ef870fChris Lattner return (C2 == 'x' || C2 == 'X'); 931a75ec43d625753b4439b0d6f70bd988444c74617Douglas Gregor} 9325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 9335253c7ff266ae79308050c9f43d60dd1a67c5fb9Nate Begeman/// LexNumericConstant - Lex the remainder of a integer or floating point 9345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// constant. From[-1] is the first character lexed. Return the end of the 9355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// constant. 936d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Lexer::LexNumericConstant(Token &Result, const char *CurPtr) { 9375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer unsigned Size; 9385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char C = getCharAndSize(CurPtr, Size); 9395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char PrevCh = 0; 9405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (isNumberBody(C)) { // FIXME: UCNs? 9415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, Size, Result); 9425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer PrevCh = C; 9435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getCharAndSize(CurPtr, Size); 9445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 9465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we fell out, check for a sign, due to 1e+12. If we have one, continue. 947b2f4a20ddc281b194caa00e850ad74759e0d50baChris Lattner if ((C == '-' || C == '+') && (PrevCh == 'E' || PrevCh == 'e')) { 948b2f4a20ddc281b194caa00e850ad74759e0d50baChris Lattner // If we are in Microsoft mode, don't continue if the constant is hex. 949b2f4a20ddc281b194caa00e850ad74759e0d50baChris Lattner // For example, MSVC will accept the following as 3 tokens: 0x1234567e+1 9506ab55ebab20086f725c4017b48ef8d7691ef870fChris Lattner if (!Features.Microsoft || !isHexaLiteral(BufferPtr, Features)) 951b2f4a20ddc281b194caa00e850ad74759e0d50baChris Lattner return LexNumericConstant(Result, ConsumeChar(CurPtr, Size, Result)); 952b2f4a20ddc281b194caa00e850ad74759e0d50baChris Lattner } 9535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 9545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we have a hex FP constant, continue. 9558c723404c34e9347fae473489194e8c200c4ac83Sean Hunt if ((C == '-' || C == '+') && (PrevCh == 'P' || PrevCh == 'p') && 956b2f4a20ddc281b194caa00e850ad74759e0d50baChris Lattner !Features.CPlusPlus0x) 9575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return LexNumericConstant(Result, ConsumeChar(CurPtr, Size, Result)); 9581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 9595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Update the location of token as well as BufferPtr. 96047246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner const char *TokStart = BufferPtr; 9619e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::numeric_constant); 96247246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner Result.setLiteralData(TokStart); 9635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 9645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 9655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// LexStringLiteral - Lex the remainder of a string literal, after having lexed 9665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// either " or L". 967d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattnervoid Lexer::LexStringLiteral(Token &Result, const char *CurPtr, bool Wide) { 9685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *NulCharacter = 0; // Does this string contain the \0 character? 9691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 9705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char C = getAndAdvanceChar(CurPtr, Result); 9715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (C != '"') { 972571339c215dc3fa44acccb89c7c6b1d5b962b7f2Chris Lattner // Skip escaped characters. Escaped newlines will already be processed by 973571339c215dc3fa44acccb89c7c6b1d5b962b7f2Chris Lattner // getAndAdvanceChar. 974571339c215dc3fa44acccb89c7c6b1d5b962b7f2Chris Lattner if (C == '\\') 9755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getAndAdvanceChar(CurPtr, Result); 97633611e0d5ab1372608a7649b1877cd4300621c71Douglas Gregor 977571339c215dc3fa44acccb89c7c6b1d5b962b7f2Chris Lattner if (C == '\n' || C == '\r' || // Newline. 97833611e0d5ab1372608a7649b1877cd4300621c71Douglas Gregor (C == 0 && CurPtr-1 == BufferEnd)) { // End of file. 97955817afdf9d453a443262a733f6caf6692dca118Douglas Gregor if (C == 0 && PP && PP->isCodeCompletionFile(FileLoc)) 98055817afdf9d453a443262a733f6caf6692dca118Douglas Gregor PP->CodeCompleteNaturalLanguage(); 98155817afdf9d453a443262a733f6caf6692dca118Douglas Gregor else if (!isLexingRawMode() && !Features.AsmPreprocessor) 98274d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(BufferPtr, diag::err_unterminated_string); 9839e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr-1, tok::unknown); 9845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; 9855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 986571339c215dc3fa44acccb89c7c6b1d5b962b7f2Chris Lattner 987571339c215dc3fa44acccb89c7c6b1d5b962b7f2Chris Lattner if (C == 0) 988571339c215dc3fa44acccb89c7c6b1d5b962b7f2Chris Lattner NulCharacter = CurPtr-1; 9895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getAndAdvanceChar(CurPtr, Result); 9905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 9911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 9925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If a nul character existed in the string, warn about it. 99374d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (NulCharacter && !isLexingRawMode()) 99474d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(NulCharacter, diag::null_in_string); 9955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 9965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Update the location of the token as well as the BufferPtr instance var. 99747246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner const char *TokStart = BufferPtr; 9986cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt FormTokenWithChars(Result, CurPtr, 9996cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt Wide ? tok::wide_string_literal : tok::string_literal); 100047246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner Result.setLiteralData(TokStart); 10015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 10025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 10035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// LexAngledStringLiteral - Lex the remainder of an angled string literal, 10045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// after having lexed the '<' character. This is used for #include filenames. 1005d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Lexer::LexAngledStringLiteral(Token &Result, const char *CurPtr) { 10065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *NulCharacter = 0; // Does this string contain the \0 character? 10079cb51ce03caf0755759f7e67aea7d4be684ff738Chris Lattner const char *AfterLessPos = CurPtr; 10085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char C = getAndAdvanceChar(CurPtr, Result); 10095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (C != '>') { 10105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Skip escaped characters. 10115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == '\\') { 10125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Skip the escaped character. 10135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getAndAdvanceChar(CurPtr, Result); 10145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (C == '\n' || C == '\r' || // Newline. 10155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer (C == 0 && CurPtr-1 == BufferEnd)) { // End of file. 10169cb51ce03caf0755759f7e67aea7d4be684ff738Chris Lattner // If the filename is unterminated, then it must just be a lone < 10179cb51ce03caf0755759f7e67aea7d4be684ff738Chris Lattner // character. Return this as such. 10189cb51ce03caf0755759f7e67aea7d4be684ff738Chris Lattner FormTokenWithChars(Result, AfterLessPos, tok::less); 10195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; 10205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (C == 0) { 10215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer NulCharacter = CurPtr-1; 10225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 10235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getAndAdvanceChar(CurPtr, Result); 10245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 10251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 10265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If a nul character existed in the string, warn about it. 102774d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (NulCharacter && !isLexingRawMode()) 102874d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(NulCharacter, diag::null_in_string); 10291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 10305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Update the location of token as well as BufferPtr. 103147246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner const char *TokStart = BufferPtr; 10329e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::angle_string_literal); 103347246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner Result.setLiteralData(TokStart); 10345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 10355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 10365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 10375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// LexCharConstant - Lex the remainder of a character constant, after having 10385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// lexed either ' or L'. 1039d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Lexer::LexCharConstant(Token &Result, const char *CurPtr) { 10405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *NulCharacter = 0; // Does this character contain the \0 character? 10415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 10425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char C = getAndAdvanceChar(CurPtr, Result); 10435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == '\'') { 104433ab3f6c7e3065550de202088f8f335ecfa16ae1Chris Lattner if (!isLexingRawMode() && !Features.AsmPreprocessor) 104574d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(BufferPtr, diag::err_empty_character); 10469e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::unknown); 10475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; 10485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 10491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1050d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner while (C != '\'') { 1051d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner // Skip escaped characters. 1052d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner if (C == '\\') { 1053d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner // Skip the escaped character. 1054d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner // FIXME: UCN's 10555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getAndAdvanceChar(CurPtr, Result); 1056d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner } else if (C == '\n' || C == '\r' || // Newline. 1057d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner (C == 0 && CurPtr-1 == BufferEnd)) { // End of file. 105855817afdf9d453a443262a733f6caf6692dca118Douglas Gregor if (C == 0 && PP && PP->isCodeCompletionFile(FileLoc)) 105955817afdf9d453a443262a733f6caf6692dca118Douglas Gregor PP->CodeCompleteNaturalLanguage(); 106055817afdf9d453a443262a733f6caf6692dca118Douglas Gregor else if (!isLexingRawMode() && !Features.AsmPreprocessor) 1061d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner Diag(BufferPtr, diag::err_unterminated_char); 1062d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner FormTokenWithChars(Result, CurPtr-1, tok::unknown); 1063d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner return; 1064d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner } else if (C == 0) { 1065d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner NulCharacter = CurPtr-1; 1066d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner } 1067d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner C = getAndAdvanceChar(CurPtr, Result); 10685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 10691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1070d80f786689d608e5c22d6e1045884de7aff76c40Chris Lattner // If a nul character existed in the character, warn about it. 107174d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (NulCharacter && !isLexingRawMode()) 107274d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(NulCharacter, diag::null_in_char); 10735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 10745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Update the location of token as well as BufferPtr. 107547246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner const char *TokStart = BufferPtr; 10769e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::char_constant); 107747246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner Result.setLiteralData(TokStart); 10785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 10795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 10805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// SkipWhitespace - Efficiently skip over a series of whitespace characters. 10815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// Update BufferPtr to point to the next non-whitespace character and return. 1082d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner/// 1083d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner/// This method forms a token and returns true if KeepWhitespaceMode is enabled. 1084d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner/// 1085d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattnerbool Lexer::SkipWhitespace(Token &Result, const char *CurPtr) { 10865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Whitespace - Skip it, then return the token after the whitespace. 10875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer unsigned char Char = *CurPtr; // Skip consequtive spaces efficiently. 10885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (1) { 10895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Skip horizontal whitespace very aggressively. 10905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (isHorizontalWhitespace(Char)) 10915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = *++CurPtr; 10921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1093ddd3e8b90a92c39f620bda7a0945320c0b9e60dbDaniel Dunbar // Otherwise if we have something other than whitespace, we're done. 10945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char != '\n' && Char != '\r') 10955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 10961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 10975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (ParsingPreprocessorDirective) { 10985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // End of preprocessor directive line, let LexTokenInternal handle this. 10995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = CurPtr; 1100d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner return false; 11015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 11021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 11035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // ok, but handle newline. 11045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // The returned token is at the start of the line. 1105d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::StartOfLine); 11065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // No leading whitespace seen so far. 1107d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.clearFlag(Token::LeadingSpace); 11085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = *++CurPtr; 11095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 11105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 11115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this isn't immediately after a newline, there is leading space. 11125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char PrevChar = CurPtr[-1]; 11135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (PrevChar != '\n' && PrevChar != '\r') 1114d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::LeadingSpace); 11155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 1116d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner // If the client wants us to return whitespace, return it now. 1117d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner if (isKeepWhitespaceMode()) { 11189e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::unknown); 1119d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner return true; 1120d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner } 11211eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 11225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = CurPtr; 1123d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner return false; 11245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 11255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 11265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// SkipBCPLComment - We have just read the // characters from input. Skip until 11275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// we find the newline character thats terminate the comment. Then update 1128046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner/// BufferPtr and return. 1129046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner/// 1130046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner/// If we're in KeepCommentMode or any CommentHandler has inserted 1131046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner/// some tokens, this will store the first token and return true. 1132d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnerbool Lexer::SkipBCPLComment(Token &Result, const char *CurPtr) { 11335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If BCPL comments aren't explicitly enabled for this language, emit an 11345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // extension warning. 113574d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!Features.BCPLComment && !isLexingRawMode()) { 11365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Diag(BufferPtr, diag::ext_bcpl_comment); 11371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 11385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Mark them enabled so we only emit one warning for this translation 11395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // unit. 11405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Features.BCPLComment = true; 11415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 11421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 11435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Scan over the body of the comment. The common case, when scanning, is that 11445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // the comment contains normal ascii characters with nothing interesting in 11455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // them. As such, optimize for this case with the inner loop. 11465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char C; 11475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer do { 11485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = *CurPtr; 11495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // FIXME: Speedup BCPL comment lexing. Just scan for a \n or \r character. 11505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we find a \n character, scan backwards, checking to see if it's an 11515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // escaped newline, like we do for block comments. 11521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 11535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Skip over characters in the fast loop. 11545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (C != 0 && // Potentially EOF. 11555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C != '\\' && // Potentially escaped newline. 11565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C != '?' && // Potentially trigraph. 11575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C != '\n' && C != '\r') // Newline or DOS-style newline. 11585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = *++CurPtr; 11595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 11605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is a newline, we're done. 11615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == '\n' || C == '\r') 11625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; // Found the newline? Break out! 11631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 11645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, this is a hard case. Fall back on getAndAdvanceChar to 1165bc3e9842212b78bcf49682641307e4f3419ecc5eChris Lattner // properly decode the character. Read it in raw mode to avoid emitting 1166bc3e9842212b78bcf49682641307e4f3419ecc5eChris Lattner // diagnostics about things like trigraphs. If we see an escaped newline, 1167bc3e9842212b78bcf49682641307e4f3419ecc5eChris Lattner // we'll handle it below. 11685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *OldPtr = CurPtr; 1169bc3e9842212b78bcf49682641307e4f3419ecc5eChris Lattner bool OldRawMode = isLexingRawMode(); 1170bc3e9842212b78bcf49682641307e4f3419ecc5eChris Lattner LexingRawMode = true; 11715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = getAndAdvanceChar(CurPtr, Result); 1172bc3e9842212b78bcf49682641307e4f3419ecc5eChris Lattner LexingRawMode = OldRawMode; 1173ead616c5d8681a41b11273327813e61bda01907aChris Lattner 1174ead616c5d8681a41b11273327813e61bda01907aChris Lattner // If the char that we finally got was a \n, then we must have had something 1175ead616c5d8681a41b11273327813e61bda01907aChris Lattner // like \<newline><newline>. We don't want to have consumed the second 1176ead616c5d8681a41b11273327813e61bda01907aChris Lattner // newline, we want CurPtr, to end up pointing to it down below. 1177ead616c5d8681a41b11273327813e61bda01907aChris Lattner if (C == '\n' || C == '\r') { 1178ead616c5d8681a41b11273327813e61bda01907aChris Lattner --CurPtr; 1179ead616c5d8681a41b11273327813e61bda01907aChris Lattner C = 'x'; // doesn't matter what this is. 1180ead616c5d8681a41b11273327813e61bda01907aChris Lattner } 11811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 11825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we read multiple characters, and one of those characters was a \r or 11835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // \n, then we had an escaped newline within the comment. Emit diagnostic 11845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // unless the next line is also a // comment. 11855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr != OldPtr+1 && C != '/' && CurPtr[0] != '/') { 11865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer for (; OldPtr != CurPtr; ++OldPtr) 11875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (OldPtr[0] == '\n' || OldPtr[0] == '\r') { 11885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Okay, we found a // comment that ends in a newline, if the next 11895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // line is also a // comment, but has spaces, don't emit a diagnostic. 11905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (isspace(C)) { 11915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *ForwardPtr = CurPtr; 11925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (isspace(*ForwardPtr)) // Skip whitespace. 11935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++ForwardPtr; 11945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (ForwardPtr[0] == '/' && ForwardPtr[1] == '/') 11955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 11965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 11971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 119874d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!isLexingRawMode()) 119974d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(OldPtr-1, diag::ext_multi_line_bcpl_comment); 12005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 12015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 12025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 12031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 120455817afdf9d453a443262a733f6caf6692dca118Douglas Gregor if (CurPtr == BufferEnd+1) { 120555817afdf9d453a443262a733f6caf6692dca118Douglas Gregor if (PP && PP->isCodeCompletionFile(FileLoc)) 120655817afdf9d453a443262a733f6caf6692dca118Douglas Gregor PP->CodeCompleteNaturalLanguage(); 120755817afdf9d453a443262a733f6caf6692dca118Douglas Gregor 120855817afdf9d453a443262a733f6caf6692dca118Douglas Gregor --CurPtr; 120955817afdf9d453a443262a733f6caf6692dca118Douglas Gregor break; 121055817afdf9d453a443262a733f6caf6692dca118Douglas Gregor } 12115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } while (C != '\n' && C != '\r'); 12125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 12133d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner // Found but did not consume the newline. Notify comment handlers about the 12143d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner // comment unless we're in a #if 0 block. 12153d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner if (PP && !isLexingRawMode() && 12163d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner PP->HandleComment(Result, SourceRange(getSourceLocation(BufferPtr), 12173d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner getSourceLocation(CurPtr)))) { 1218046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner BufferPtr = CurPtr; 1219046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner return true; // A token has to be returned. 1220046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner } 12211eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we are returning comments as tokens, return this comment as a token. 1223fa95a019da00b926d64ff83358ba73bbc6ae1e37Chris Lattner if (inKeepCommentMode()) 12245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return SaveBCPLComment(Result, CurPtr); 12255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 12265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we are inside a preprocessor directive and we see the end of line, 12275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // return immediately, so that the lexer can return this as an EOM token. 12285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (ParsingPreprocessorDirective || CurPtr == BufferEnd) { 12295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = CurPtr; 12302d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner return false; 12315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 12321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, eat the \n character. We don't care if this is a \n\r or 12347a4f004a77421af876152281b92c4f8ea24afb2dChris Lattner // \r\n sequence. This is an efficiency hack (because we know the \n can't 1235d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner // contribute to another token), it isn't needed for correctness. Note that 1236d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner // this is ok even in KeepWhitespaceMode, because we would have returned the 1237d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner /// comment above in that mode. 12385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++CurPtr; 12391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // The next returned token is at the start of the line. 1241d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::StartOfLine); 12425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // No leading whitespace seen so far. 1243d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.clearFlag(Token::LeadingSpace); 12445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = CurPtr; 12452d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner return false; 12465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 12475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 12485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// SaveBCPLComment - If in save-comment mode, package up this BCPL comment in 12495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// an appropriate way and return it. 1250d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnerbool Lexer::SaveBCPLComment(Token &Result, const char *CurPtr) { 12519e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner // If we're not in a preprocessor directive, just return the // comment 12529e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner // directly. 12539e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::comment); 12541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12559e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner if (!ParsingPreprocessorDirective) 12569e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner return true; 12571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12589e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner // If this BCPL-style comment is in a macro definition, transmogrify it into 12599e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner // a C-style block comment. 1260453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor bool Invalid = false; 1261453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor std::string Spelling = PP->getSpelling(Result, &Invalid); 1262453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor if (Invalid) 1263453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor return true; 1264453091cc2082e207ea2c2dda645a9bc01b37fb0cDouglas Gregor 12659e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner assert(Spelling[0] == '/' && Spelling[1] == '/' && "Not bcpl comment?"); 12669e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Spelling[1] = '*'; // Change prefix to "/*". 12679e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Spelling += "*/"; // add suffix. 12681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12699e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Result.setKind(tok::comment); 127047246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner PP->CreateString(&Spelling[0], Spelling.size(), Result, 127147246be8ac5b0ddde6c402b8fc6946b6135487b5Chris Lattner Result.getLocation()); 12722d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner return true; 12735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 12745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 12755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// isBlockCommentEndOfEscapedNewLine - Return true if the specified newline 12765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// character (either \n or \r) is part of an escaped newline sequence. Issue a 127747a2b40b7a3d9c89d6c3d12e5a81e2498c207750Chris Lattner/// diagnostic if so. We know that the newline is inside of a block comment. 12781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic bool isEndOfBlockCommentWithEscapedNewLine(const char *CurPtr, 12795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Lexer *L) { 12805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer assert(CurPtr[0] == '\n' || CurPtr[0] == '\r'); 12811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Back up off the newline. 12835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer --CurPtr; 12841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is a two-character newline sequence, skip the other character. 12865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr[0] == '\n' || CurPtr[0] == '\r') { 12875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // \n\n or \r\r -> not escaped newline. 12885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr[0] == CurPtr[1]) 12895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return false; 12905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // \n\r or \r\n -> skip the newline. 12915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer --CurPtr; 12925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 12931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 12945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we have horizontal whitespace, skip over it. We allow whitespace 12955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // between the slash and newline. 12965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer bool HasSpace = false; 12975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (isHorizontalWhitespace(*CurPtr) || *CurPtr == 0) { 12985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer --CurPtr; 12995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer HasSpace = true; 13005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 13011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 13025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we have a slash, we know this is an escaped newline. 13035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (*CurPtr == '\\') { 13045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr[-1] != '*') return false; 13055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 13065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // It isn't a slash, is it the ?? / trigraph? 13075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr[0] != '/' || CurPtr[-1] != '?' || CurPtr[-2] != '?' || 13085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr[-3] != '*') 13095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return false; 13101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 13115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // This is the trigraph ending the comment. Emit a stern warning! 13125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr -= 2; 13135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 13145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If no trigraphs are enabled, warn that we ignored this trigraph and 13155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // ignore this * character. 13165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (!L->getFeatures().Trigraphs) { 131774d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!L->isLexingRawMode()) 131874d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner L->Diag(CurPtr, diag::trigraph_ignored_block_comment); 13195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return false; 13205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 132174d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!L->isLexingRawMode()) 132274d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner L->Diag(CurPtr, diag::trigraph_ends_block_comment); 13235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 13241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 13255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Warn about having an escaped newline between the */ characters. 132674d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!L->isLexingRawMode()) 132774d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner L->Diag(CurPtr, diag::escaped_newline_block_comment_end); 13281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 13295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If there was space between the backslash and newline, warn about it. 133074d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (HasSpace && !L->isLexingRawMode()) 133174d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner L->Diag(CurPtr, diag::backslash_newline_space); 13321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 13335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return true; 13345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 13355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 13365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#ifdef __SSE2__ 13375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include <emmintrin.h> 13385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#elif __ALTIVEC__ 13395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include <altivec.h> 13405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#undef bool 13415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#endif 13425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 13435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// SkipBlockComment - We have just read the /* characters from input. Read 13445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// until we find the */ characters that terminate the comment. Note that we 13455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// don't bother decoding trigraphs or escaped newlines in block comments, 13465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// because they cannot cause the comment to end. The only thing that can 13475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// happen is the comment could end with an escaped newline between the */ end 13485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// of comment. 13492d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner/// 1350046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner/// If we're in KeepCommentMode or any CommentHandler has inserted 1351046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner/// some tokens, this will store the first token and return true. 1352d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnerbool Lexer::SkipBlockComment(Token &Result, const char *CurPtr) { 13535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Scan one character past where we should, looking for a '/' character. Once 13545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // we find it, check to see if it was preceeded by a *. This common 13555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // optimization helps people who like to put a lot of * characters in their 13565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // comments. 13578146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner 13588146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner // The first character we get with newlines and trigraphs skipped to handle 13598146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner // the degenerate /*/ case below correctly if the * has an escaped newline 13608146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner // after it. 13618146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner unsigned CharSize; 13628146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner unsigned char C = getCharAndSize(CurPtr, CharSize); 13638146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner CurPtr += CharSize; 13645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == 0 && CurPtr == BufferEnd+1) { 1365150fcd5dcb79ad0a9cc6824c5589000e3669c7a4Chris Lattner if (!isLexingRawMode() && 1366150fcd5dcb79ad0a9cc6824c5589000e3669c7a4Chris Lattner !PP->isCodeCompletionFile(FileLoc)) 13670af574270d3be2b0e73a3379dfaa633746f8fc6fChris Lattner Diag(BufferPtr, diag::err_unterminated_block_comment); 136831f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner --CurPtr; 13691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 137031f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner // KeepWhitespaceMode should return this broken comment as a token. Since 137131f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner // it isn't a well formed comment, just return it as an 'unknown' token. 137231f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner if (isKeepWhitespaceMode()) { 13739e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::unknown); 137431f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner return true; 137531f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner } 13761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 137731f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner BufferPtr = CurPtr; 13782d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner return false; 13795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 13801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 13818146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner // Check to see if the first character after the '/*' is another /. If so, 13828146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner // then this slash does not end the block comment, it is part of it. 13838146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner if (C == '/') 13848146b6851e873eab71341f1b1d3198894bc4c0acChris Lattner C = *CurPtr++; 13851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 13865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (1) { 13875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Skip over all non-interesting characters until we find end of buffer or a 13885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // (probably ending) '/' character. 13895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr + 24 < BufferEnd) { 13905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // While not aligned to a 16-byte boundary. 13915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (C != '/' && ((intptr_t)CurPtr & 0x0F) != 0) 13925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = *CurPtr++; 13931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 13945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == '/') goto FoundSlash; 13955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 13965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#ifdef __SSE2__ 13975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer __m128i Slashes = _mm_set_epi8('/', '/', '/', '/', '/', '/', '/', '/', 13985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer '/', '/', '/', '/', '/', '/', '/', '/'); 13995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (CurPtr+16 <= BufferEnd && 14005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer _mm_movemask_epi8(_mm_cmpeq_epi8(*(__m128i*)CurPtr, Slashes)) == 0) 14015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr += 16; 14025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#elif __ALTIVEC__ 14035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer __vector unsigned char Slashes = { 14041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump '/', '/', '/', '/', '/', '/', '/', '/', 14055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer '/', '/', '/', '/', '/', '/', '/', '/' 14065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer }; 14075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (CurPtr+16 <= BufferEnd && 14085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer !vec_any_eq(*(vector unsigned char*)CurPtr, Slashes)) 14095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr += 16; 14101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump#else 14115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Scan for '/' quickly. Many block comments are very large. 14125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (CurPtr[0] != '/' && 14135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr[1] != '/' && 14145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr[2] != '/' && 14155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr[3] != '/' && 14165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr+4 < BufferEnd) { 14175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr += 4; 14185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#endif 14201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 14215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // It has to be one of the bytes scanned, increment to it and read one. 14225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = *CurPtr++; 14235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 14255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Loop to scan the remainder. 14265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (C != '/' && C != '\0') 14275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = *CurPtr++; 14281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 14295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer FoundSlash: 14305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (C == '/') { 14315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr[-2] == '*') // We found the final */. We're done! 14325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 14331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 14345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if ((CurPtr[-2] == '\n' || CurPtr[-2] == '\r')) { 14355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (isEndOfBlockCommentWithEscapedNewLine(CurPtr-2, this)) { 14365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // We found the final */, though it had an escaped newline between the 14375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // * and /. We're done! 14385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 14395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr[0] == '*' && CurPtr[1] != '/') { 14425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If this is a /* inside of the comment, emit a warning. Don't do this 14435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // if this is a /*/, which will end the comment. This misses cases with 14445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // embedded escaped newlines, but oh well. 144574d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!isLexingRawMode()) 144674d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(CurPtr-1, diag::warn_nested_block_comment); 14475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (C == 0 && CurPtr == BufferEnd+1) { 144955817afdf9d453a443262a733f6caf6692dca118Douglas Gregor if (PP && PP->isCodeCompletionFile(FileLoc)) 145055817afdf9d453a443262a733f6caf6692dca118Douglas Gregor PP->CodeCompleteNaturalLanguage(); 145155817afdf9d453a443262a733f6caf6692dca118Douglas Gregor else if (!isLexingRawMode()) 145274d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(BufferPtr, diag::err_unterminated_block_comment); 14535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Note: the user probably forgot a */. We could continue immediately 14545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // after the /*, but this would involve lexing a lot of what really is the 14555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // comment, which surely would confuse the parser. 145631f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner --CurPtr; 14571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 145831f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner // KeepWhitespaceMode should return this broken comment as a token. Since 145931f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner // it isn't a well formed comment, just return it as an 'unknown' token. 146031f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner if (isKeepWhitespaceMode()) { 14619e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::unknown); 146231f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner return true; 146331f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner } 14641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 146531f0ecae4b2d4d268d04c562a43d9fbfee96f189Chris Lattner BufferPtr = CurPtr; 14662d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner return false; 14675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer C = *CurPtr++; 14695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 14713d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner // Notify comment handlers about the comment unless we're in a #if 0 block. 14723d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner if (PP && !isLexingRawMode() && 14733d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner PP->HandleComment(Result, SourceRange(getSourceLocation(BufferPtr), 14743d0ad58b28e0d50fca7f21c6a078b05370510288Chris Lattner getSourceLocation(CurPtr)))) { 1475046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner BufferPtr = CurPtr; 1476046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner return true; // A token has to be returned. 1477046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner } 14782e22253e03e175144aeb9d13350a12fd83f858beDouglas Gregor 14795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we are returning comments as tokens, return this comment as a token. 1480fa95a019da00b926d64ff83358ba73bbc6ae1e37Chris Lattner if (inKeepCommentMode()) { 14819e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::comment); 14822d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner return true; 14835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 14855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // It is common for the tokens immediately after a /**/ comment to be 14865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // whitespace. Instead of going through the big switch, handle it 1487d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner // efficiently now. This is safe even in KeepWhitespaceMode because we would 1488d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner // have already returned above with the comment as a token. 14895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (isHorizontalWhitespace(*CurPtr)) { 1490d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::LeadingSpace); 14915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer SkipWhitespace(Result, CurPtr+1); 14922d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner return false; 14935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 14945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 14955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Otherwise, just return so that the next character will be lexed as a token. 14965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = CurPtr; 1497d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::LeadingSpace); 14982d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner return false; 14995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 15005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 15015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 15025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// Primary Lexing Entry Points 15035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===// 15045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 15055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ReadToEndOfLine - Read the rest of the current preprocessor line as an 15065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// uninterpreted string. This switches the lexer out of directive mode. 15075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstd::string Lexer::ReadToEndOfLine() { 15085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer assert(ParsingPreprocessorDirective && ParsingFilename == false && 15095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer "Must be in a preprocessing directive!"); 15105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer std::string Result; 1511d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Token Tmp; 15125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 15135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // CurPtr - Cache BufferPtr in an automatic variable. 15145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *CurPtr = BufferPtr; 15155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (1) { 15165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char Char = getAndAdvanceChar(CurPtr, Tmp); 15175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer switch (Char) { 15185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer default: 15195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Result += Char; 15205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 15215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 0: // Null. 15225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Found end of file? 15235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr-1 != BufferEnd) { 15245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Nope, normal character, continue. 15255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Result += Char; 15265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 15275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 15285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // FALL THROUGH. 15295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\r': 15305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\n': 15315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Okay, we found the end of the line. First, back up past the \0, \r, \n. 15325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer assert(CurPtr[-1] == Char && "Trigraphs for newline?"); 15335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = CurPtr-1; 15341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 15355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Next, lex the character, which should handle the EOM transition. 15365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Lex(Tmp); 153755817afdf9d453a443262a733f6caf6692dca118Douglas Gregor if (Tmp.is(tok::code_completion)) { 153855817afdf9d453a443262a733f6caf6692dca118Douglas Gregor if (PP && PP->getCodeCompletionHandler()) 153955817afdf9d453a443262a733f6caf6692dca118Douglas Gregor PP->getCodeCompletionHandler()->CodeCompleteNaturalLanguage(); 154055817afdf9d453a443262a733f6caf6692dca118Douglas Gregor Lex(Tmp); 154155817afdf9d453a443262a733f6caf6692dca118Douglas Gregor } 154222f6bbcafa8871f4f20c4402d9cbc5c024fee99aChris Lattner assert(Tmp.is(tok::eom) && "Unexpected token!"); 15431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 15445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Finally, we're done, return the string we found. 15455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return Result; 15465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 15475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 15485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 15495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 15505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// LexEndOfFile - CurPtr points to the end of this file. Handle this 15515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// condition, reporting diagnostics and handling other edge cases as required. 15525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// This returns true if Result contains a token, false if PP.Lex should be 15535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// called again. 1554d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnerbool Lexer::LexEndOfFile(Token &Result, const char *CurPtr) { 1555f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor // Check if we are performing code completion. 1556f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor if (PP && PP->isCodeCompletionFile(FileLoc)) { 1557f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor // We're at the end of the file, but we've been asked to consider the 1558f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor // end of the file to be a code-completion token. Return the 1559f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor // code-completion token. 1560f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor Result.startToken(); 1561f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor FormTokenWithChars(Result, CurPtr, tok::code_completion); 1562f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor 1563f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor // Only do the eof -> code_completion translation once. 1564f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor PP->SetCodeCompletionPoint(0, 0, 0); 1565f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor 1566f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor // Silence any diagnostics that occur once we hit the code-completion point. 1567f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor PP->getDiagnostics().setSuppressAllDiagnostics(true); 1568f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor return true; 1569f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor } 1570f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor 15715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we hit the end of the file while parsing a preprocessor directive, 15725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // end the preprocessor directive first. The next token returned will 15735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // then be the end of file. 15745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (ParsingPreprocessorDirective) { 15755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Done parsing the "line". 15765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ParsingPreprocessorDirective = false; 15775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Update the location of token as well as BufferPtr. 15789e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::eom); 15791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 15805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Restore comment saving mode, in case it was disabled for directive. 1581f744d133e8d36eec65828d046ee695bda238125cChris Lattner SetCommentRetentionState(PP->getCommentRetentionState()); 15825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return true; // Have a token. 15831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump } 158486d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor 158586d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor // If we are in raw mode, return this event as an EOF token. Let the caller 158686d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor // that put us in raw mode handle the event. 158786d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor if (isLexingRawMode()) { 158886d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor Result.startToken(); 158986d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor BufferPtr = BufferEnd; 159086d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor FormTokenWithChars(Result, BufferEnd, tok::eof); 159186d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor return true; 159286d9a52c24d390631a888d4ff812e1b15445e0a0Douglas Gregor } 159381b747b7fcc91c2fba9a3183d8fac80adbfc1d3eDouglas Gregor 1594f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor // Issue diagnostics for unterminated #if and missing newline. 1595f44e854ed1e3aa86d2ed6d615ccd109d50ddcff9Douglas Gregor 15965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we are in a #if directive, emit an error. 15975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while (!ConditionalStack.empty()) { 15982d474ba9e8ae43a1a5a9f72718c0d79092b9453fDouglas Gregor if (!PP->isCodeCompletionFile(FileLoc)) 15992d474ba9e8ae43a1a5a9f72718c0d79092b9453fDouglas Gregor PP->Diag(ConditionalStack.back().IfLoc, 16002d474ba9e8ae43a1a5a9f72718c0d79092b9453fDouglas Gregor diag::err_pp_unterminated_conditional); 16015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ConditionalStack.pop_back(); 16025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 16031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1604b25e5d79d9d1967df058a242e96a62d0d0ace074Chris Lattner // C99 5.1.1.2p2: If the file is non-empty and didn't end in a newline, issue 1605b25e5d79d9d1967df058a242e96a62d0d0ace074Chris Lattner // a pedwarn. 1606b25e5d79d9d1967df058a242e96a62d0d0ace074Chris Lattner if (CurPtr != BufferStart && (CurPtr[-1] != '\n' && CurPtr[-1] != '\r')) 160720d0ee5d23c0f0e879ce1711d11df3118b3fb033Mike Stump Diag(BufferEnd, diag::ext_no_newline_eof) 1608849b243d4065f56742a4677d6dc8277609a151f8Douglas Gregor << FixItHint::CreateInsertion(getSourceLocation(BufferEnd), "\n"); 16091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 16105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = CurPtr; 16115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 16125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Finally, let the preprocessor handle this. 1613168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner return PP->HandleEndOfFile(Result); 16145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 16155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 16165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// isNextPPTokenLParen - Return 1 if the next unexpanded token lexed from 16175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// the specified lexer will return a tok::l_paren token, 0 if it is something 16185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// else and 2 if there are no more tokens in the buffer controlled by the 16195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// lexer. 16205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerunsigned Lexer::isNextPPTokenLParen() { 16215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer assert(!LexingRawMode && "How can we expand a macro from a skipping buffer?"); 16221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 16235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Switch to 'skipping' mode. This will ensure that we can lex a token 16245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // without emitting diagnostics, disables macro expansion, and will cause EOF 16255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // to return an EOF token instead of popping the include stack. 16265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer LexingRawMode = true; 16271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 16285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Save state that can be changed while lexing so that we can restore it. 16295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *TmpBufferPtr = BufferPtr; 1630a864cf7c1d774a0f790bfc46befc87d9dbf1f65cChris Lattner bool inPPDirectiveMode = ParsingPreprocessorDirective; 16311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1632d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Token Tok; 16335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Tok.startToken(); 16345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer LexTokenInternal(Tok); 16351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 16365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Restore state that may have changed. 16375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = TmpBufferPtr; 1638a864cf7c1d774a0f790bfc46befc87d9dbf1f65cChris Lattner ParsingPreprocessorDirective = inPPDirectiveMode; 16391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 16405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Restore the lexer back to non-skipping mode. 16415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer LexingRawMode = false; 16421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 164322f6bbcafa8871f4f20c4402d9cbc5c024fee99aChris Lattner if (Tok.is(tok::eof)) 16445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return 2; 164522f6bbcafa8871f4f20c4402d9cbc5c024fee99aChris Lattner return Tok.is(tok::l_paren); 16465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 16475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 164834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// FindConflictEnd - Find the end of a version control conflict marker. 164934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattnerstatic const char *FindConflictEnd(const char *CurPtr, const char *BufferEnd) { 165034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner llvm::StringRef RestOfBuffer(CurPtr+7, BufferEnd-CurPtr-7); 165134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner size_t Pos = RestOfBuffer.find(">>>>>>>"); 165234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner while (Pos != llvm::StringRef::npos) { 165334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Must occur at start of line. 165434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (RestOfBuffer[Pos-1] != '\r' && 165534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner RestOfBuffer[Pos-1] != '\n') { 165634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner RestOfBuffer = RestOfBuffer.substr(Pos+7); 16573d488990e13835cb35538b2f0cd53dbda92e36beChris Lattner Pos = RestOfBuffer.find(">>>>>>>"); 165834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner continue; 165934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } 166034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return RestOfBuffer.data()+Pos; 166134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } 166234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return 0; 166334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner} 166434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 166534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// IsStartOfConflictMarker - If the specified pointer is the start of a version 166634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// control conflict marker like '<<<<<<<', recognize it as such, emit an error 166734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// and recover nicely. This returns true if it is a conflict marker and false 166834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// if not. 166934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattnerbool Lexer::IsStartOfConflictMarker(const char *CurPtr) { 167034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Only a conflict marker if it starts at the beginning of a line. 167134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (CurPtr != BufferStart && 167234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner CurPtr[-1] != '\n' && CurPtr[-1] != '\r') 167334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return false; 167434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 167534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Check to see if we have <<<<<<<. 167634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (BufferEnd-CurPtr < 8 || 167734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner llvm::StringRef(CurPtr, 7) != "<<<<<<<") 167834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return false; 167934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 168034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // If we have a situation where we don't care about conflict markers, ignore 168134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // it. 168234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (IsInConflictMarker || isLexingRawMode()) 168334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return false; 168434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 168534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Check to see if there is a >>>>>>> somewhere in the buffer at the start of 168634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // a line to terminate this conflict marker. 16873d488990e13835cb35538b2f0cd53dbda92e36beChris Lattner if (FindConflictEnd(CurPtr, BufferEnd)) { 168834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // We found a match. We are really in a conflict marker. 168934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Diagnose this, and ignore to the end of line. 169034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner Diag(CurPtr, diag::err_conflict_marker); 169134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner IsInConflictMarker = true; 169234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 169334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Skip ahead to the end of line. We know this exists because the 169434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // end-of-conflict marker starts with \r or \n. 169534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner while (*CurPtr != '\r' && *CurPtr != '\n') { 169634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner assert(CurPtr != BufferEnd && "Didn't find end of line"); 169734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner ++CurPtr; 169834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } 169934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner BufferPtr = CurPtr; 170034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return true; 170134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } 170234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 170334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // No end of conflict marker found. 170434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return false; 170534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner} 170634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 170734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 170834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// HandleEndOfConflictMarker - If this is a '=======' or '|||||||' or '>>>>>>>' 170934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// marker, then it is the end of a conflict marker. Handle it by ignoring up 171034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// until the end of the line. This returns true if it is a conflict marker and 171134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner/// false if not. 171234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattnerbool Lexer::HandleEndOfConflictMarker(const char *CurPtr) { 171334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Only a conflict marker if it starts at the beginning of a line. 171434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (CurPtr != BufferStart && 171534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner CurPtr[-1] != '\n' && CurPtr[-1] != '\r') 171634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return false; 171734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 171834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // If we have a situation where we don't care about conflict markers, ignore 171934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // it. 172034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (!IsInConflictMarker || isLexingRawMode()) 172134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return false; 172234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 172334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Check to see if we have the marker (7 characters in a row). 172434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner for (unsigned i = 1; i != 7; ++i) 172534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (CurPtr[i] != CurPtr[0]) 172634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return false; 172734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 172834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // If we do have it, search for the end of the conflict marker. This could 172934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // fail if it got skipped with a '#if 0' or something. Note that CurPtr might 173034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // be the end of conflict marker. 173134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (const char *End = FindConflictEnd(CurPtr, BufferEnd)) { 173234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner CurPtr = End; 173334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 173434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // Skip ahead to the end of line. 173534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner while (CurPtr != BufferEnd && *CurPtr != '\r' && *CurPtr != '\n') 173634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner ++CurPtr; 173734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 173834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner BufferPtr = CurPtr; 173934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 174034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // No longer in the conflict marker. 174134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner IsInConflictMarker = false; 174234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return true; 174334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } 174434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 174534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner return false; 174634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner} 174734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 17485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 17495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// LexTokenInternal - This implements a simple C family lexer. It is an 17505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// extremely performance critical piece of code. This assumes that the buffer 1751efb173ddd95325c7fd3c15070392b27c07a49a85Chris Lattner/// has a null character at the end of the file. This returns a preprocessing 1752efb173ddd95325c7fd3c15070392b27c07a49a85Chris Lattner/// token, not a normal token, as such, it is an internal interface. It assumes 1753efb173ddd95325c7fd3c15070392b27c07a49a85Chris Lattner/// that the Flags of result have been cleared before calling this. 1754d217773f106856a11879ec79dc468efefaf2ee75Chris Lattnervoid Lexer::LexTokenInternal(Token &Result) { 17555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerLexNextToken: 17565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // New token, can't need cleaning yet. 1757d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.clearFlag(Token::NeedsCleaning); 17585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Result.setIdentifierInfo(0); 17591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 17605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // CurPtr - Cache BufferPtr in an automatic variable. 17615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer const char *CurPtr = BufferPtr; 17625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 17635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Small amounts of horizontal whitespace is very common between tokens. 17645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if ((*CurPtr == ' ') || (*CurPtr == '\t')) { 17655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++CurPtr; 17665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer while ((*CurPtr == ' ') || (*CurPtr == '\t')) 17675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ++CurPtr; 17681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1769d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner // If we are keeping whitespace and other tokens, just return what we just 1770d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner // skipped. The next lexer invocation will return the token after the 1771d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner // whitespace. 1772d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner if (isKeepWhitespaceMode()) { 17739e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, tok::unknown); 1774d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner return; 1775d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner } 17761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 17775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer BufferPtr = CurPtr; 1778d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::LeadingSpace); 17795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 17801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 17815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer unsigned SizeTmp, SizeTmp2; // Temporaries for use in cases below. 17821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 17835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Read a character, advancing over it. 17845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer char Char = getAndAdvanceChar(CurPtr, Result); 17859e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner tok::TokenKind Kind; 17861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 17875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer switch (Char) { 17885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 0: // Null. 17895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Found end of file? 17905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (CurPtr-1 == BufferEnd) { 17915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Read the PP instance variable into an automatic variable, because 17925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // LexEndOfFile will often delete 'this'. 1793168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner Preprocessor *PPCache = PP; 17945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (LexEndOfFile(Result, CurPtr-1)) // Retreat back into the file. 17955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return; // Got a token to return. 1796168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner assert(PPCache && "Raw buffer::LexEndOfFile should return a token"); 1797168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner return PPCache->Lex(Result); 17985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 17991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 180074d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!isLexingRawMode()) 180174d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(CurPtr-1, diag::null_in_file); 1802d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::LeadingSpace); 1803d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner if (SkipWhitespace(Result, CurPtr)) 1804d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner return; // KeepWhitespaceMode 18051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer goto LexNextToken; // GCC isn't tail call eliminating. 1807a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner 1808a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner case 26: // DOS & CP/M EOF: "^Z". 1809a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner // If we're in Microsoft extensions mode, treat this as end of file. 1810a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner if (Features.Microsoft) { 1811a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner // Read the PP instance variable into an automatic variable, because 1812a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner // LexEndOfFile will often delete 'this'. 1813a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner Preprocessor *PPCache = PP; 1814a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner if (LexEndOfFile(Result, CurPtr-1)) // Retreat back into the file. 1815a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner return; // Got a token to return. 1816a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner assert(PPCache && "Raw buffer::LexEndOfFile should return a token"); 1817a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner return PPCache->Lex(Result); 1818a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner } 1819a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner // If Microsoft extensions are disabled, this is just random garbage. 1820a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner Kind = tok::unknown; 1821a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner break; 1822a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner 18235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\n': 18245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\r': 18255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // If we are inside a preprocessor directive and we see the end of line, 18265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // we know we are done with the directive, so return an EOM token. 18275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (ParsingPreprocessorDirective) { 18285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Done parsing the "line". 18295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer ParsingPreprocessorDirective = false; 18301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Restore comment saving mode, in case it was disabled for directive. 1832f744d133e8d36eec65828d046ee695bda238125cChris Lattner SetCommentRetentionState(PP->getCommentRetentionState()); 18331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Since we consumed a newline, we are back at the start of a line. 18355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer IsAtStartOfLine = true; 18361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18379e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::eom; 18385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 18395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 18405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // The returned token is at the start of the line. 1841d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::StartOfLine); 18425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // No leading whitespace seen so far. 1843d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.clearFlag(Token::LeadingSpace); 18441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1845d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner if (SkipWhitespace(Result, CurPtr)) 1846d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner return; // KeepWhitespaceMode 18475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer goto LexNextToken; // GCC isn't tail call eliminating. 18485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case ' ': 18495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\t': 18505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\f': 18515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\v': 18528133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner SkipHorizontalWhitespace: 1853d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::LeadingSpace); 1854d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner if (SkipWhitespace(Result, CurPtr)) 1855d88dc48e33d71732708960170e57a3d1bdc8f847Chris Lattner return; // KeepWhitespaceMode 18568133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner 18578133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner SkipIgnoredUnits: 18588133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner CurPtr = BufferPtr; 18591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18608133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner // If the next token is obviously a // or /* */ comment, skip it efficiently 18618133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner // too (without going through the big switch stmt). 18628402c73dd880e8af46c826d873681820aebe32ebChris Lattner if (CurPtr[0] == '/' && CurPtr[1] == '/' && !inKeepCommentMode() && 18638402c73dd880e8af46c826d873681820aebe32ebChris Lattner Features.BCPLComment) { 1864046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner if (SkipBCPLComment(Result, CurPtr+2)) 1865046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner return; // There is a token to return. 18668133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner goto SkipIgnoredUnits; 1867fa95a019da00b926d64ff83358ba73bbc6ae1e37Chris Lattner } else if (CurPtr[0] == '/' && CurPtr[1] == '*' && !inKeepCommentMode()) { 1868046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner if (SkipBlockComment(Result, CurPtr+2)) 1869046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner return; // There is a token to return. 18708133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner goto SkipIgnoredUnits; 18718133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner } else if (isHorizontalWhitespace(*CurPtr)) { 18728133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner goto SkipHorizontalWhitespace; 18738133cfcc0b672cc86d1ef73ce0c61eccecdb3690Chris Lattner } 18745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer goto LexNextToken; // GCC isn't tail call eliminating. 1875a2bf105739cef7aea8e4d219629c627fcd11742eChris Lattner 18763a5707766850f9ee9daa35299794328b5caf96dcChris Lattner // C99 6.4.4.1: Integer Constants. 18773a5707766850f9ee9daa35299794328b5caf96dcChris Lattner // C99 6.4.4.2: Floating Constants. 18783a5707766850f9ee9daa35299794328b5caf96dcChris Lattner case '0': case '1': case '2': case '3': case '4': 18793a5707766850f9ee9daa35299794328b5caf96dcChris Lattner case '5': case '6': case '7': case '8': case '9': 18803a5707766850f9ee9daa35299794328b5caf96dcChris Lattner // Notify MIOpt that we read a non-whitespace/non-comment token. 18813a5707766850f9ee9daa35299794328b5caf96dcChris Lattner MIOpt.ReadToken(); 18823a5707766850f9ee9daa35299794328b5caf96dcChris Lattner return LexNumericConstant(Result, CurPtr); 18831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18843a5707766850f9ee9daa35299794328b5caf96dcChris Lattner case 'L': // Identifier (Loony) or wide literal (L'x' or L"xyz"). 18855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Notify MIOpt that we read a non-whitespace/non-comment token. 18865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer MIOpt.ReadToken(); 18875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 18885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 18895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Wide string literal. 18905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '"') 18915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return LexStringLiteral(Result, ConsumeChar(CurPtr, SizeTmp, Result), 18925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer true); 18935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 18945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Wide character constant. 18955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '\'') 18965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return LexCharConstant(Result, ConsumeChar(CurPtr, SizeTmp, Result)); 18975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // FALL THROUGH, treating L like the start of an identifier. 18981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 18995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // C99 6.4.2: Identifiers. 19005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 'A': case 'B': case 'C': case 'D': case 'E': case 'F': case 'G': 19015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 'H': case 'I': case 'J': case 'K': /*'L'*/case 'M': case 'N': 19025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 'O': case 'P': case 'Q': case 'R': case 'S': case 'T': case 'U': 19035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 'V': case 'W': case 'X': case 'Y': case 'Z': 19045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 'a': case 'b': case 'c': case 'd': case 'e': case 'f': case 'g': 19055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 'h': case 'i': case 'j': case 'k': case 'l': case 'm': case 'n': 19065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 'o': case 'p': case 'q': case 'r': case 's': case 't': case 'u': 19075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case 'v': case 'w': case 'x': case 'y': case 'z': 19085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '_': 19095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Notify MIOpt that we read a non-whitespace/non-comment token. 19105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer MIOpt.ReadToken(); 19115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return LexIdentifier(Result, CurPtr); 19123a5707766850f9ee9daa35299794328b5caf96dcChris Lattner 19133a5707766850f9ee9daa35299794328b5caf96dcChris Lattner case '$': // $ in identifiers. 19143a5707766850f9ee9daa35299794328b5caf96dcChris Lattner if (Features.DollarIdents) { 191574d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!isLexingRawMode()) 191674d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(CurPtr-1, diag::ext_dollar_in_identifier); 19173a5707766850f9ee9daa35299794328b5caf96dcChris Lattner // Notify MIOpt that we read a non-whitespace/non-comment token. 19183a5707766850f9ee9daa35299794328b5caf96dcChris Lattner MIOpt.ReadToken(); 19193a5707766850f9ee9daa35299794328b5caf96dcChris Lattner return LexIdentifier(Result, CurPtr); 19203a5707766850f9ee9daa35299794328b5caf96dcChris Lattner } 19211eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 19229e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::unknown; 19233a5707766850f9ee9daa35299794328b5caf96dcChris Lattner break; 19241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 19255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // C99 6.4.4: Character Constants. 19265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\'': 19275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Notify MIOpt that we read a non-whitespace/non-comment token. 19285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer MIOpt.ReadToken(); 19295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return LexCharConstant(Result, CurPtr); 19305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 19315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // C99 6.4.5: String Literals. 19325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '"': 19335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Notify MIOpt that we read a non-whitespace/non-comment token. 19345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer MIOpt.ReadToken(); 19355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return LexStringLiteral(Result, CurPtr, false); 19365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 19375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // C99 6.4.6: Punctuators. 19385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '?': 19399e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::question; 19405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '[': 19429e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::l_square; 19435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case ']': 19459e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::r_square; 19465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '(': 19489e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::l_paren; 19495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case ')': 19519e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::r_paren; 19525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '{': 19549e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::l_brace; 19555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '}': 19579e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::r_brace; 19585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '.': 19605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 19615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char >= '0' && Char <= '9') { 19625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Notify MIOpt that we read a non-whitespace/non-comment token. 19635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer MIOpt.ReadToken(); 19645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 19655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer return LexNumericConstant(Result, ConsumeChar(CurPtr, SizeTmp, Result)); 19665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Features.CPlusPlus && Char == '*') { 19679e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::periodstar; 19685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr += SizeTmp; 19695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '.' && 19705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer getCharAndSize(CurPtr+SizeTmp, SizeTmp2) == '.') { 19719e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::ellipsis; 19725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(ConsumeChar(CurPtr, SizeTmp, Result), 19735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer SizeTmp2, Result); 19745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 19759e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::period; 19765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 19775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '&': 19795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 19805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '&') { 19819e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::ampamp; 19825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 19835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '=') { 19849e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::ampequal; 19855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 19865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 19879e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::amp; 19885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 19895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump case '*': 19915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (getCharAndSize(CurPtr, SizeTmp) == '=') { 19929e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::starequal; 19935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 19945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 19959e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::star; 19965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 19975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 19985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '+': 19995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 20005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '+') { 20015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20029e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::plusplus; 20035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '=') { 20045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20059e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::plusequal; 20065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 20079e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::plus; 20085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 20095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 20105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '-': 20115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 20129e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner if (Char == '-') { // -- 20135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20149e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::minusminus; 20151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump } else if (Char == '>' && Features.CPlusPlus && 20169e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner getCharAndSize(CurPtr+SizeTmp, SizeTmp2) == '*') { // C++ ->* 20175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(ConsumeChar(CurPtr, SizeTmp, Result), 20185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer SizeTmp2, Result); 20199e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::arrowstar; 20209e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner } else if (Char == '>') { // -> 20215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20229e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::arrow; 20239e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner } else if (Char == '=') { // -= 20245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20259e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::minusequal; 20265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 20279e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::minus; 20285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 20295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 20305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '~': 20319e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::tilde; 20325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 20335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '!': 20345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (getCharAndSize(CurPtr, SizeTmp) == '=') { 20359e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::exclaimequal; 20365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 20389e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::exclaim; 20395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 20405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 20415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '/': 20425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // 6.4.9: Comments 20435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 20445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '/') { // BCPL comment. 20458402c73dd880e8af46c826d873681820aebe32ebChris Lattner // Even if BCPL comments are disabled (e.g. in C89 mode), we generally 20468402c73dd880e8af46c826d873681820aebe32ebChris Lattner // want to lex this as a comment. There is one problem with this though, 20478402c73dd880e8af46c826d873681820aebe32ebChris Lattner // that in one particular corner case, this can change the behavior of the 20488402c73dd880e8af46c826d873681820aebe32ebChris Lattner // resultant program. For example, In "foo //**/ bar", C89 would lex 20498402c73dd880e8af46c826d873681820aebe32ebChris Lattner // this as "foo / bar" and langauges with BCPL comments would lex it as 20508402c73dd880e8af46c826d873681820aebe32ebChris Lattner // "foo". Check to see if the character after the second slash is a '*'. 20518402c73dd880e8af46c826d873681820aebe32ebChris Lattner // If so, we will lex that as a "/" instead of the start of a comment. 20528402c73dd880e8af46c826d873681820aebe32ebChris Lattner if (Features.BCPLComment || 20538402c73dd880e8af46c826d873681820aebe32ebChris Lattner getCharAndSize(CurPtr+SizeTmp, SizeTmp2) != '*') { 20548402c73dd880e8af46c826d873681820aebe32ebChris Lattner if (SkipBCPLComment(Result, ConsumeChar(CurPtr, SizeTmp, Result))) 2055046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner return; // There is a token to return. 20561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 20578402c73dd880e8af46c826d873681820aebe32ebChris Lattner // It is common for the tokens immediately after a // comment to be 20588402c73dd880e8af46c826d873681820aebe32ebChris Lattner // whitespace (indentation for the next line). Instead of going through 20598402c73dd880e8af46c826d873681820aebe32ebChris Lattner // the big switch, handle it efficiently now. 20608402c73dd880e8af46c826d873681820aebe32ebChris Lattner goto SkipIgnoredUnits; 20618402c73dd880e8af46c826d873681820aebe32ebChris Lattner } 20628402c73dd880e8af46c826d873681820aebe32ebChris Lattner } 20631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 20648402c73dd880e8af46c826d873681820aebe32ebChris Lattner if (Char == '*') { // /**/ comment. 20655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (SkipBlockComment(Result, ConsumeChar(CurPtr, SizeTmp, Result))) 2066046c2277dcbcc8eb89dbb5b1b8c5226b7cb81635Chris Lattner return; // There is a token to return. 20672d3818959df7b71bfcf8c2f2b573856bfc4c3a21Chris Lattner goto LexNextToken; // GCC isn't tail call eliminating. 20688402c73dd880e8af46c826d873681820aebe32ebChris Lattner } 20691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 20708402c73dd880e8af46c826d873681820aebe32ebChris Lattner if (Char == '=') { 20715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20729e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::slashequal; 20735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 20749e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::slash; 20755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 20765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 20775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '%': 20785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 20795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '=') { 20809e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::percentequal; 20815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Features.Digraphs && Char == '>') { 20839e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::r_brace; // '%>' -> '}' 20845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Features.Digraphs && Char == ':') { 20865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 20875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 20885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '%' && getCharAndSize(CurPtr+SizeTmp, SizeTmp2) == ':') { 20899e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::hashhash; // '%:%:' -> '##' 20905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(ConsumeChar(CurPtr, SizeTmp, Result), 20915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer SizeTmp2, Result); 20925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '@' && Features.Microsoft) { // %:@ -> #@ -> Charize 20935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 209474d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!isLexingRawMode()) 209574d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(BufferPtr, diag::charize_microsoft_ext); 20969e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::hashat; 2097e91e93225db2e66906878513c6ef4dd6a7ee2b6aChris Lattner } else { // '%:' -> '#' 20985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // We parsed a # character. If this occurs at the start of the line, 20995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // it's actually the start of a preprocessing directive. Callback to 21005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // the preprocessor to handle it. 21015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // FIXME: -fpreprocessed mode?? 2102766703b76dc1eb0c49d11cafc2a17c7407876ebcChris Lattner if (Result.isAtStartOfLine() && !LexingRawMode && !Is_PragmaLexer) { 2103e91e93225db2e66906878513c6ef4dd6a7ee2b6aChris Lattner FormTokenWithChars(Result, CurPtr, tok::hash); 2104168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner PP->HandleDirective(Result); 21051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 21065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // As an optimization, if the preprocessor didn't switch lexers, tail 21075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // recurse. 2108168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner if (PP->isCurrentLexer(this)) { 21095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Start a new token. If this is a #include or something, the PP may 21105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // want us starting at the beginning of the line again. If so, set 2111515f43f9f23de50d155b481b8774ec40bdfd7ff2Chris Lattner // the StartOfLine flag and clear LeadingSpace. 21125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (IsAtStartOfLine) { 2113d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::StartOfLine); 2114515f43f9f23de50d155b481b8774ec40bdfd7ff2Chris Lattner Result.clearFlag(Token::LeadingSpace); 21155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer IsAtStartOfLine = false; 21165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 21175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer goto LexNextToken; // GCC isn't tail call eliminating. 21185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 21191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2120168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner return PP->Lex(Result); 21215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 21221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2123e91e93225db2e66906878513c6ef4dd6a7ee2b6aChris Lattner Kind = tok::hash; 21245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 21255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 21269e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::percent; 21275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 21285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 21295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '<': 21305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 21315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (ParsingFilename) { 21329cb51ce03caf0755759f7e67aea7d4be684ff738Chris Lattner return LexAngledStringLiteral(Result, CurPtr); 21335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '<') { 213434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner char After = getCharAndSize(CurPtr+SizeTmp, SizeTmp2); 213534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (After == '=') { 213634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner Kind = tok::lesslessequal; 213734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner CurPtr = ConsumeChar(ConsumeChar(CurPtr, SizeTmp, Result), 213834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner SizeTmp2, Result); 213934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } else if (After == '<' && IsStartOfConflictMarker(CurPtr-1)) { 214034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // If this is actually a '<<<<<<<' version control conflict marker, 214134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // recognize it as such and recover nicely. 214234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner goto LexNextToken; 214334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } else { 214434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 214534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner Kind = tok::lessless; 214634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } 21475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '=') { 21485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 21499e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::lessequal; 21509e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner } else if (Features.Digraphs && Char == ':') { // '<:' -> '[' 21515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 21529e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::l_square; 21539e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner } else if (Features.Digraphs && Char == '%') { // '<%' -> '{' 21545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 21559e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::l_brace; 21565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 21579e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::less; 21585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 21595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 21605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '>': 21615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 21625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '=') { 21635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 21649e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::greaterequal; 21655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '>') { 216634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner char After = getCharAndSize(CurPtr+SizeTmp, SizeTmp2); 216734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (After == '=') { 216834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner CurPtr = ConsumeChar(ConsumeChar(CurPtr, SizeTmp, Result), 216934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner SizeTmp2, Result); 217034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner Kind = tok::greatergreaterequal; 217134f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } else if (After == '>' && HandleEndOfConflictMarker(CurPtr-1)) { 217234f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // If this is '>>>>>>>' and we're in a conflict marker, ignore it. 217334f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner goto LexNextToken; 217434f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } else { 217534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 217634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner Kind = tok::greatergreater; 217734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner } 217834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 21795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 21809e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::greater; 21815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 21825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 21835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '^': 21845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 21855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '=') { 21865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 21879e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::caretequal; 21885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 21899e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::caret; 21905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 21915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 21925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '|': 21935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 21945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '=') { 21959e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::pipeequal; 21965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 21975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '|') { 219834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // If this is '|||||||' and we're in a conflict marker, ignore it. 219934f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (CurPtr[1] == '|' && HandleEndOfConflictMarker(CurPtr-1)) 220034f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner goto LexNextToken; 22019e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::pipepipe; 22025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 22035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 22049e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::pipe; 22055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 22065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 22075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case ':': 22085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 22095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Features.Digraphs && Char == '>') { 22109e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::r_square; // ':>' -> ']' 22115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 22125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Features.CPlusPlus && Char == ':') { 22139e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::coloncolon; 22145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 22151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump } else { 22169e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::colon; 22175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 22185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 22195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case ';': 22209e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::semi; 22215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 22225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '=': 22235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 22245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '=') { 222534f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner // If this is '=======' and we're in a conflict marker, ignore it. 222634f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner if (CurPtr[1] == '=' && HandleEndOfConflictMarker(CurPtr-1)) 222734f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner goto LexNextToken; 222834f349da38a7bd99103e12d8ea6c73bc8d025193Chris Lattner 22299e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::equalequal; 22305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 22311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump } else { 22329e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::equal; 22335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 22345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 22355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case ',': 22369e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::comma; 22375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 22385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '#': 22395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer Char = getCharAndSize(CurPtr, SizeTmp); 22405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (Char == '#') { 22419e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::hashhash; 22425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 22435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else if (Char == '@' && Features.Microsoft) { // #@ -> Charize 22449e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::hashat; 224574d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner if (!isLexingRawMode()) 224674d15dfd183b2082e8a5d4dfbf66bd861b220901Chris Lattner Diag(BufferPtr, diag::charize_microsoft_ext); 22475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer CurPtr = ConsumeChar(CurPtr, SizeTmp, Result); 22485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } else { 22495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // We parsed a # character. If this occurs at the start of the line, 22505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // it's actually the start of a preprocessing directive. Callback to 22515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // the preprocessor to handle it. 22525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // FIXME: -fpreprocessed mode?? 2253766703b76dc1eb0c49d11cafc2a17c7407876ebcChris Lattner if (Result.isAtStartOfLine() && !LexingRawMode && !Is_PragmaLexer) { 2254e91e93225db2e66906878513c6ef4dd6a7ee2b6aChris Lattner FormTokenWithChars(Result, CurPtr, tok::hash); 2255168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner PP->HandleDirective(Result); 22561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 22575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // As an optimization, if the preprocessor didn't switch lexers, tail 22585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // recurse. 2259168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner if (PP->isCurrentLexer(this)) { 22605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Start a new token. If this is a #include or something, the PP may 22615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // want us starting at the beginning of the line again. If so, set 2262515f43f9f23de50d155b481b8774ec40bdfd7ff2Chris Lattner // the StartOfLine flag and clear LeadingSpace. 22635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer if (IsAtStartOfLine) { 2264d217773f106856a11879ec79dc468efefaf2ee75Chris Lattner Result.setFlag(Token::StartOfLine); 2265515f43f9f23de50d155b481b8774ec40bdfd7ff2Chris Lattner Result.clearFlag(Token::LeadingSpace); 22665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer IsAtStartOfLine = false; 22675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 22685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer goto LexNextToken; // GCC isn't tail call eliminating. 22695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 2270168ae2d44a443da75ea85db5f3b5081eb0bce113Chris Lattner return PP->Lex(Result); 22715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 22721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 2273e91e93225db2e66906878513c6ef4dd6a7ee2b6aChris Lattner Kind = tok::hash; 22745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 22755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 22765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 22773a5707766850f9ee9daa35299794328b5caf96dcChris Lattner case '@': 22783a5707766850f9ee9daa35299794328b5caf96dcChris Lattner // Objective C support. 22793a5707766850f9ee9daa35299794328b5caf96dcChris Lattner if (CurPtr[-1] == '@' && Features.ObjC1) 22809e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::at; 22813a5707766850f9ee9daa35299794328b5caf96dcChris Lattner else 22829e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::unknown; 22833a5707766850f9ee9daa35299794328b5caf96dcChris Lattner break; 22841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 22855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer case '\\': 22865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // FIXME: UCN's. 22875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // FALL THROUGH. 22885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer default: 22899e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner Kind = tok::unknown; 22905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer break; 22915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer } 22921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 22935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Notify MIOpt that we read a non-whitespace/non-comment token. 22945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer MIOpt.ReadToken(); 22955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer 22965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer // Update the location of token as well as BufferPtr. 22979e6293d4dfd688429f77ee3b6edba9dfd7ada3a2Chris Lattner FormTokenWithChars(Result, CurPtr, Kind); 22985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer} 2299