1 //===- Lexer.h - C Language Family Lexer ------------------------*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file defines the Lexer interface. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #ifndef LLVM_CLANG_LEX_LEXER_H 14 #define LLVM_CLANG_LEX_LEXER_H 15 16 #include "clang/Basic/LangOptions.h" 17 #include "clang/Basic/SourceLocation.h" 18 #include "clang/Basic/TokenKinds.h" 19 #include "clang/Lex/PreprocessorLexer.h" 20 #include "clang/Lex/Token.h" 21 #include "llvm/ADT/Optional.h" 22 #include "llvm/ADT/SmallVector.h" 23 #include "llvm/ADT/StringRef.h" 24 #include <cassert> 25 #include <cstdint> 26 #include <string> 27 28 namespace llvm { 29 30 class MemoryBufferRef; 31 32 } // namespace llvm 33 34 namespace clang { 35 36 class DiagnosticBuilder; 37 class Preprocessor; 38 class SourceManager; 39 40 /// ConflictMarkerKind - Kinds of conflict marker which the lexer might be 41 /// recovering from. 42 enum ConflictMarkerKind { 43 /// Not within a conflict marker. 44 CMK_None, 45 46 /// A normal or diff3 conflict marker, initiated by at least 7 "<"s, 47 /// separated by at least 7 "="s or "|"s, and terminated by at least 7 ">"s. 48 CMK_Normal, 49 50 /// A Perforce-style conflict marker, initiated by 4 ">"s, 51 /// separated by 4 "="s, and terminated by 4 "<"s. 52 CMK_Perforce 53 }; 54 55 /// Describes the bounds (start, size) of the preamble and a flag required by 56 /// PreprocessorOptions::PrecompiledPreambleBytes. 57 /// The preamble includes the BOM, if any. 58 struct PreambleBounds { 59 /// Size of the preamble in bytes. 60 unsigned Size; 61 62 /// Whether the preamble ends at the start of a new line. 63 /// 64 /// Used to inform the lexer as to whether it's starting at the beginning of 65 /// a line after skipping the preamble. 66 bool PreambleEndsAtStartOfLine; 67 PreambleBoundsPreambleBounds68 PreambleBounds(unsigned Size, bool PreambleEndsAtStartOfLine) 69 : Size(Size), PreambleEndsAtStartOfLine(PreambleEndsAtStartOfLine) {} 70 }; 71 72 /// Lexer - This provides a simple interface that turns a text buffer into a 73 /// stream of tokens. This provides no support for file reading or buffering, 74 /// or buffering/seeking of tokens, only forward lexing is supported. It relies 75 /// on the specified Preprocessor object to handle preprocessor directives, etc. 76 class Lexer : public PreprocessorLexer { 77 friend class Preprocessor; 78 79 void anchor() override; 80 81 //===--------------------------------------------------------------------===// 82 // Constant configuration values for this lexer. 83 84 // Start of the buffer. 85 const char *BufferStart; 86 87 // End of the buffer. 88 const char *BufferEnd; 89 90 // Location for start of file. 91 SourceLocation FileLoc; 92 93 // LangOpts enabled by this language (cache). 94 LangOptions LangOpts; 95 96 // True if lexer for _Pragma handling. 97 bool Is_PragmaLexer; 98 99 //===--------------------------------------------------------------------===// 100 // Context-specific lexing flags set by the preprocessor. 101 // 102 103 /// ExtendedTokenMode - The lexer can optionally keep comments and whitespace 104 /// and return them as tokens. This is used for -C and -CC modes, and 105 /// whitespace preservation can be useful for some clients that want to lex 106 /// the file in raw mode and get every character from the file. 107 /// 108 /// When this is set to 2 it returns comments and whitespace. When set to 1 109 /// it returns comments, when it is set to 0 it returns normal tokens only. 110 unsigned char ExtendedTokenMode; 111 112 //===--------------------------------------------------------------------===// 113 // Context that changes as the file is lexed. 114 // NOTE: any state that mutates when in raw mode must have save/restore code 115 // in Lexer::isNextPPTokenLParen. 116 117 // BufferPtr - Current pointer into the buffer. This is the next character 118 // to be lexed. 119 const char *BufferPtr; 120 121 // IsAtStartOfLine - True if the next lexed token should get the "start of 122 // line" flag set on it. 123 bool IsAtStartOfLine; 124 125 bool IsAtPhysicalStartOfLine; 126 127 bool HasLeadingSpace; 128 129 bool HasLeadingEmptyMacro; 130 131 // NewLinePtr - A pointer to new line character '\n' being lexed. For '\r\n', 132 // it also points to '\n.' 133 const char *NewLinePtr; 134 135 // CurrentConflictMarkerState - The kind of conflict marker we are handling. 136 ConflictMarkerKind CurrentConflictMarkerState; 137 138 void InitLexer(const char *BufStart, const char *BufPtr, const char *BufEnd); 139 140 public: 141 /// Lexer constructor - Create a new lexer object for the specified buffer 142 /// with the specified preprocessor managing the lexing process. This lexer 143 /// assumes that the associated file buffer and Preprocessor objects will 144 /// outlive it, so it doesn't take ownership of either of them. 145 Lexer(FileID FID, const llvm::MemoryBufferRef &InputFile, Preprocessor &PP); 146 147 /// Lexer constructor - Create a new raw lexer object. This object is only 148 /// suitable for calls to 'LexFromRawLexer'. This lexer assumes that the 149 /// text range will outlive it, so it doesn't take ownership of it. 150 Lexer(SourceLocation FileLoc, const LangOptions &LangOpts, 151 const char *BufStart, const char *BufPtr, const char *BufEnd); 152 153 /// Lexer constructor - Create a new raw lexer object. This object is only 154 /// suitable for calls to 'LexFromRawLexer'. This lexer assumes that the 155 /// text range will outlive it, so it doesn't take ownership of it. 156 Lexer(FileID FID, const llvm::MemoryBufferRef &FromFile, 157 const SourceManager &SM, const LangOptions &LangOpts); 158 159 Lexer(const Lexer &) = delete; 160 Lexer &operator=(const Lexer &) = delete; 161 162 /// Create_PragmaLexer: Lexer constructor - Create a new lexer object for 163 /// _Pragma expansion. This has a variety of magic semantics that this method 164 /// sets up. It returns a new'd Lexer that must be delete'd when done. 165 static Lexer *Create_PragmaLexer(SourceLocation SpellingLoc, 166 SourceLocation ExpansionLocStart, 167 SourceLocation ExpansionLocEnd, 168 unsigned TokLen, Preprocessor &PP); 169 170 /// getLangOpts - Return the language features currently enabled. 171 /// NOTE: this lexer modifies features as a file is parsed! getLangOpts()172 const LangOptions &getLangOpts() const { return LangOpts; } 173 174 /// getFileLoc - Return the File Location for the file we are lexing out of. 175 /// The physical location encodes the location where the characters come from, 176 /// the virtual location encodes where we should *claim* the characters came 177 /// from. Currently this is only used by _Pragma handling. getFileLoc()178 SourceLocation getFileLoc() const { return FileLoc; } 179 180 private: 181 /// Lex - Return the next token in the file. If this is the end of file, it 182 /// return the tok::eof token. This implicitly involves the preprocessor. 183 bool Lex(Token &Result); 184 185 public: 186 /// isPragmaLexer - Returns true if this Lexer is being used to lex a pragma. isPragmaLexer()187 bool isPragmaLexer() const { return Is_PragmaLexer; } 188 189 private: 190 /// IndirectLex - An indirect call to 'Lex' that can be invoked via 191 /// the PreprocessorLexer interface. IndirectLex(Token & Result)192 void IndirectLex(Token &Result) override { Lex(Result); } 193 194 public: 195 /// LexFromRawLexer - Lex a token from a designated raw lexer (one with no 196 /// associated preprocessor object. Return true if the 'next character to 197 /// read' pointer points at the end of the lexer buffer, false otherwise. LexFromRawLexer(Token & Result)198 bool LexFromRawLexer(Token &Result) { 199 assert(LexingRawMode && "Not already in raw mode!"); 200 Lex(Result); 201 // Note that lexing to the end of the buffer doesn't implicitly delete the 202 // lexer when in raw mode. 203 return BufferPtr == BufferEnd; 204 } 205 206 /// isKeepWhitespaceMode - Return true if the lexer should return tokens for 207 /// every character in the file, including whitespace and comments. This 208 /// should only be used in raw mode, as the preprocessor is not prepared to 209 /// deal with the excess tokens. isKeepWhitespaceMode()210 bool isKeepWhitespaceMode() const { 211 return ExtendedTokenMode > 1; 212 } 213 214 /// SetKeepWhitespaceMode - This method lets clients enable or disable 215 /// whitespace retention mode. SetKeepWhitespaceMode(bool Val)216 void SetKeepWhitespaceMode(bool Val) { 217 assert((!Val || LexingRawMode || LangOpts.TraditionalCPP) && 218 "Can only retain whitespace in raw mode or -traditional-cpp"); 219 ExtendedTokenMode = Val ? 2 : 0; 220 } 221 222 /// inKeepCommentMode - Return true if the lexer should return comments as 223 /// tokens. inKeepCommentMode()224 bool inKeepCommentMode() const { 225 return ExtendedTokenMode > 0; 226 } 227 228 /// SetCommentRetentionMode - Change the comment retention mode of the lexer 229 /// to the specified mode. This is really only useful when lexing in raw 230 /// mode, because otherwise the lexer needs to manage this. SetCommentRetentionState(bool Mode)231 void SetCommentRetentionState(bool Mode) { 232 assert(!isKeepWhitespaceMode() && 233 "Can't play with comment retention state when retaining whitespace"); 234 ExtendedTokenMode = Mode ? 1 : 0; 235 } 236 237 /// Sets the extended token mode back to its initial value, according to the 238 /// language options and preprocessor. This controls whether the lexer 239 /// produces comment and whitespace tokens. 240 /// 241 /// This requires the lexer to have an associated preprocessor. A standalone 242 /// lexer has nothing to reset to. 243 void resetExtendedTokenMode(); 244 245 /// Gets source code buffer. getBuffer()246 StringRef getBuffer() const { 247 return StringRef(BufferStart, BufferEnd - BufferStart); 248 } 249 250 /// ReadToEndOfLine - Read the rest of the current preprocessor line as an 251 /// uninterpreted string. This switches the lexer out of directive mode. 252 void ReadToEndOfLine(SmallVectorImpl<char> *Result = nullptr); 253 254 255 /// Diag - Forwarding function for diagnostics. This translate a source 256 /// position in the current buffer into a SourceLocation object for rendering. 257 DiagnosticBuilder Diag(const char *Loc, unsigned DiagID) const; 258 259 /// getSourceLocation - Return a source location identifier for the specified 260 /// offset in the current file. 261 SourceLocation getSourceLocation(const char *Loc, unsigned TokLen = 1) const; 262 263 /// getSourceLocation - Return a source location for the next character in 264 /// the current file. getSourceLocation()265 SourceLocation getSourceLocation() override { 266 return getSourceLocation(BufferPtr); 267 } 268 269 /// Return the current location in the buffer. getBufferLocation()270 const char *getBufferLocation() const { return BufferPtr; } 271 272 /// Returns the current lexing offset. getCurrentBufferOffset()273 unsigned getCurrentBufferOffset() { 274 assert(BufferPtr >= BufferStart && "Invalid buffer state"); 275 return BufferPtr - BufferStart; 276 } 277 278 /// Skip over \p NumBytes bytes. 279 /// 280 /// If the skip is successful, the next token will be lexed from the new 281 /// offset. The lexer also assumes that we skipped to the start of the line. 282 /// 283 /// \returns true if the skip failed (new offset would have been past the 284 /// end of the buffer), false otherwise. 285 bool skipOver(unsigned NumBytes); 286 287 /// Stringify - Convert the specified string into a C string by i) escaping 288 /// '\\' and " characters and ii) replacing newline character(s) with "\\n". 289 /// If Charify is true, this escapes the ' character instead of ". 290 static std::string Stringify(StringRef Str, bool Charify = false); 291 292 /// Stringify - Convert the specified string into a C string by i) escaping 293 /// '\\' and " characters and ii) replacing newline character(s) with "\\n". 294 static void Stringify(SmallVectorImpl<char> &Str); 295 296 /// getSpelling - This method is used to get the spelling of a token into a 297 /// preallocated buffer, instead of as an std::string. The caller is required 298 /// to allocate enough space for the token, which is guaranteed to be at least 299 /// Tok.getLength() bytes long. The length of the actual result is returned. 300 /// 301 /// Note that this method may do two possible things: it may either fill in 302 /// the buffer specified with characters, or it may *change the input pointer* 303 /// to point to a constant buffer with the data already in it (avoiding a 304 /// copy). The caller is not allowed to modify the returned buffer pointer 305 /// if an internal buffer is returned. 306 static unsigned getSpelling(const Token &Tok, const char *&Buffer, 307 const SourceManager &SourceMgr, 308 const LangOptions &LangOpts, 309 bool *Invalid = nullptr); 310 311 /// getSpelling() - Return the 'spelling' of the Tok token. The spelling of a 312 /// token is the characters used to represent the token in the source file 313 /// after trigraph expansion and escaped-newline folding. In particular, this 314 /// wants to get the true, uncanonicalized, spelling of things like digraphs 315 /// UCNs, etc. 316 static std::string getSpelling(const Token &Tok, 317 const SourceManager &SourceMgr, 318 const LangOptions &LangOpts, 319 bool *Invalid = nullptr); 320 321 /// getSpelling - This method is used to get the spelling of the 322 /// token at the given source location. If, as is usually true, it 323 /// is not necessary to copy any data, then the returned string may 324 /// not point into the provided buffer. 325 /// 326 /// This method lexes at the expansion depth of the given 327 /// location and does not jump to the expansion or spelling 328 /// location. 329 static StringRef getSpelling(SourceLocation loc, 330 SmallVectorImpl<char> &buffer, 331 const SourceManager &SM, 332 const LangOptions &options, 333 bool *invalid = nullptr); 334 335 /// MeasureTokenLength - Relex the token at the specified location and return 336 /// its length in bytes in the input file. If the token needs cleaning (e.g. 337 /// includes a trigraph or an escaped newline) then this count includes bytes 338 /// that are part of that. 339 static unsigned MeasureTokenLength(SourceLocation Loc, 340 const SourceManager &SM, 341 const LangOptions &LangOpts); 342 343 /// Relex the token at the specified location. 344 /// \returns true if there was a failure, false on success. 345 static bool getRawToken(SourceLocation Loc, Token &Result, 346 const SourceManager &SM, 347 const LangOptions &LangOpts, 348 bool IgnoreWhiteSpace = false); 349 350 /// Given a location any where in a source buffer, find the location 351 /// that corresponds to the beginning of the token in which the original 352 /// source location lands. 353 static SourceLocation GetBeginningOfToken(SourceLocation Loc, 354 const SourceManager &SM, 355 const LangOptions &LangOpts); 356 357 /// Get the physical length (including trigraphs and escaped newlines) of the 358 /// first \p Characters characters of the token starting at TokStart. 359 static unsigned getTokenPrefixLength(SourceLocation TokStart, 360 unsigned CharNo, 361 const SourceManager &SM, 362 const LangOptions &LangOpts); 363 364 /// AdvanceToTokenCharacter - If the current SourceLocation specifies a 365 /// location at the start of a token, return a new location that specifies a 366 /// character within the token. This handles trigraphs and escaped newlines. AdvanceToTokenCharacter(SourceLocation TokStart,unsigned Characters,const SourceManager & SM,const LangOptions & LangOpts)367 static SourceLocation AdvanceToTokenCharacter(SourceLocation TokStart, 368 unsigned Characters, 369 const SourceManager &SM, 370 const LangOptions &LangOpts) { 371 return TokStart.getLocWithOffset( 372 getTokenPrefixLength(TokStart, Characters, SM, LangOpts)); 373 } 374 375 /// Computes the source location just past the end of the 376 /// token at this source location. 377 /// 378 /// This routine can be used to produce a source location that 379 /// points just past the end of the token referenced by \p Loc, and 380 /// is generally used when a diagnostic needs to point just after a 381 /// token where it expected something different that it received. If 382 /// the returned source location would not be meaningful (e.g., if 383 /// it points into a macro), this routine returns an invalid 384 /// source location. 385 /// 386 /// \param Offset an offset from the end of the token, where the source 387 /// location should refer to. The default offset (0) produces a source 388 /// location pointing just past the end of the token; an offset of 1 produces 389 /// a source location pointing to the last character in the token, etc. 390 static SourceLocation getLocForEndOfToken(SourceLocation Loc, unsigned Offset, 391 const SourceManager &SM, 392 const LangOptions &LangOpts); 393 394 /// Given a token range, produce a corresponding CharSourceRange that 395 /// is not a token range. This allows the source range to be used by 396 /// components that don't have access to the lexer and thus can't find the 397 /// end of the range for themselves. getAsCharRange(SourceRange Range,const SourceManager & SM,const LangOptions & LangOpts)398 static CharSourceRange getAsCharRange(SourceRange Range, 399 const SourceManager &SM, 400 const LangOptions &LangOpts) { 401 SourceLocation End = getLocForEndOfToken(Range.getEnd(), 0, SM, LangOpts); 402 return End.isInvalid() ? CharSourceRange() 403 : CharSourceRange::getCharRange( 404 Range.getBegin(), End); 405 } getAsCharRange(CharSourceRange Range,const SourceManager & SM,const LangOptions & LangOpts)406 static CharSourceRange getAsCharRange(CharSourceRange Range, 407 const SourceManager &SM, 408 const LangOptions &LangOpts) { 409 return Range.isTokenRange() 410 ? getAsCharRange(Range.getAsRange(), SM, LangOpts) 411 : Range; 412 } 413 414 /// Returns true if the given MacroID location points at the first 415 /// token of the macro expansion. 416 /// 417 /// \param MacroBegin If non-null and function returns true, it is set to 418 /// begin location of the macro. 419 static bool isAtStartOfMacroExpansion(SourceLocation loc, 420 const SourceManager &SM, 421 const LangOptions &LangOpts, 422 SourceLocation *MacroBegin = nullptr); 423 424 /// Returns true if the given MacroID location points at the last 425 /// token of the macro expansion. 426 /// 427 /// \param MacroEnd If non-null and function returns true, it is set to 428 /// end location of the macro. 429 static bool isAtEndOfMacroExpansion(SourceLocation loc, 430 const SourceManager &SM, 431 const LangOptions &LangOpts, 432 SourceLocation *MacroEnd = nullptr); 433 434 /// Accepts a range and returns a character range with file locations. 435 /// 436 /// Returns a null range if a part of the range resides inside a macro 437 /// expansion or the range does not reside on the same FileID. 438 /// 439 /// This function is trying to deal with macros and return a range based on 440 /// file locations. The cases where it can successfully handle macros are: 441 /// 442 /// -begin or end range lies at the start or end of a macro expansion, in 443 /// which case the location will be set to the expansion point, e.g: 444 /// \#define M 1 2 445 /// a M 446 /// If you have a range [a, 2] (where 2 came from the macro), the function 447 /// will return a range for "a M" 448 /// if you have range [a, 1], the function will fail because the range 449 /// overlaps with only a part of the macro 450 /// 451 /// -The macro is a function macro and the range can be mapped to the macro 452 /// arguments, e.g: 453 /// \#define M 1 2 454 /// \#define FM(x) x 455 /// FM(a b M) 456 /// if you have range [b, 2], the function will return the file range "b M" 457 /// inside the macro arguments. 458 /// if you have range [a, 2], the function will return the file range 459 /// "FM(a b M)" since the range includes all of the macro expansion. 460 static CharSourceRange makeFileCharRange(CharSourceRange Range, 461 const SourceManager &SM, 462 const LangOptions &LangOpts); 463 464 /// Returns a string for the source that the range encompasses. 465 static StringRef getSourceText(CharSourceRange Range, 466 const SourceManager &SM, 467 const LangOptions &LangOpts, 468 bool *Invalid = nullptr); 469 470 /// Retrieve the name of the immediate macro expansion. 471 /// 472 /// This routine starts from a source location, and finds the name of the macro 473 /// responsible for its immediate expansion. It looks through any intervening 474 /// macro argument expansions to compute this. It returns a StringRef which 475 /// refers to the SourceManager-owned buffer of the source where that macro 476 /// name is spelled. Thus, the result shouldn't out-live that SourceManager. 477 static StringRef getImmediateMacroName(SourceLocation Loc, 478 const SourceManager &SM, 479 const LangOptions &LangOpts); 480 481 /// Retrieve the name of the immediate macro expansion. 482 /// 483 /// This routine starts from a source location, and finds the name of the 484 /// macro responsible for its immediate expansion. It looks through any 485 /// intervening macro argument expansions to compute this. It returns a 486 /// StringRef which refers to the SourceManager-owned buffer of the source 487 /// where that macro name is spelled. Thus, the result shouldn't out-live 488 /// that SourceManager. 489 /// 490 /// This differs from Lexer::getImmediateMacroName in that any macro argument 491 /// location will result in the topmost function macro that accepted it. 492 /// e.g. 493 /// \code 494 /// MAC1( MAC2(foo) ) 495 /// \endcode 496 /// for location of 'foo' token, this function will return "MAC1" while 497 /// Lexer::getImmediateMacroName will return "MAC2". 498 static StringRef getImmediateMacroNameForDiagnostics( 499 SourceLocation Loc, const SourceManager &SM, const LangOptions &LangOpts); 500 501 /// Compute the preamble of the given file. 502 /// 503 /// The preamble of a file contains the initial comments, include directives, 504 /// and other preprocessor directives that occur before the code in this 505 /// particular file actually begins. The preamble of the main source file is 506 /// a potential prefix header. 507 /// 508 /// \param Buffer The memory buffer containing the file's contents. 509 /// 510 /// \param MaxLines If non-zero, restrict the length of the preamble 511 /// to fewer than this number of lines. 512 /// 513 /// \returns The offset into the file where the preamble ends and the rest 514 /// of the file begins along with a boolean value indicating whether 515 /// the preamble ends at the beginning of a new line. 516 static PreambleBounds ComputePreamble(StringRef Buffer, 517 const LangOptions &LangOpts, 518 unsigned MaxLines = 0); 519 520 /// Finds the token that comes right after the given location. 521 /// 522 /// Returns the next token, or none if the location is inside a macro. 523 static Optional<Token> findNextToken(SourceLocation Loc, 524 const SourceManager &SM, 525 const LangOptions &LangOpts); 526 527 /// Checks that the given token is the first token that occurs after 528 /// the given location (this excludes comments and whitespace). Returns the 529 /// location immediately after the specified token. If the token is not found 530 /// or the location is inside a macro, the returned source location will be 531 /// invalid. 532 static SourceLocation findLocationAfterToken(SourceLocation loc, 533 tok::TokenKind TKind, 534 const SourceManager &SM, 535 const LangOptions &LangOpts, 536 bool SkipTrailingWhitespaceAndNewLine); 537 538 /// Returns true if the given character could appear in an identifier. 539 static bool isIdentifierBodyChar(char c, const LangOptions &LangOpts); 540 541 /// Checks whether new line pointed by Str is preceded by escape 542 /// sequence. 543 static bool isNewLineEscaped(const char *BufferStart, const char *Str); 544 545 /// getCharAndSizeNoWarn - Like the getCharAndSize method, but does not ever 546 /// emit a warning. getCharAndSizeNoWarn(const char * Ptr,unsigned & Size,const LangOptions & LangOpts)547 static inline char getCharAndSizeNoWarn(const char *Ptr, unsigned &Size, 548 const LangOptions &LangOpts) { 549 // If this is not a trigraph and not a UCN or escaped newline, return 550 // quickly. 551 if (isObviouslySimpleCharacter(Ptr[0])) { 552 Size = 1; 553 return *Ptr; 554 } 555 556 Size = 0; 557 return getCharAndSizeSlowNoWarn(Ptr, Size, LangOpts); 558 } 559 560 /// Returns the leading whitespace for line that corresponds to the given 561 /// location \p Loc. 562 static StringRef getIndentationForLine(SourceLocation Loc, 563 const SourceManager &SM); 564 565 private: 566 //===--------------------------------------------------------------------===// 567 // Internal implementation interfaces. 568 569 /// LexTokenInternal - Internal interface to lex a preprocessing token. Called 570 /// by Lex. 571 /// 572 bool LexTokenInternal(Token &Result, bool TokAtPhysicalStartOfLine); 573 574 bool CheckUnicodeWhitespace(Token &Result, uint32_t C, const char *CurPtr); 575 576 /// Given that a token begins with the Unicode character \p C, figure out 577 /// what kind of token it is and dispatch to the appropriate lexing helper 578 /// function. 579 bool LexUnicode(Token &Result, uint32_t C, const char *CurPtr); 580 581 /// FormTokenWithChars - When we lex a token, we have identified a span 582 /// starting at BufferPtr, going to TokEnd that forms the token. This method 583 /// takes that range and assigns it to the token as its location and size. In 584 /// addition, since tokens cannot overlap, this also updates BufferPtr to be 585 /// TokEnd. FormTokenWithChars(Token & Result,const char * TokEnd,tok::TokenKind Kind)586 void FormTokenWithChars(Token &Result, const char *TokEnd, 587 tok::TokenKind Kind) { 588 unsigned TokLen = TokEnd-BufferPtr; 589 Result.setLength(TokLen); 590 Result.setLocation(getSourceLocation(BufferPtr, TokLen)); 591 Result.setKind(Kind); 592 BufferPtr = TokEnd; 593 } 594 595 /// isNextPPTokenLParen - Return 1 if the next unexpanded token will return a 596 /// tok::l_paren token, 0 if it is something else and 2 if there are no more 597 /// tokens in the buffer controlled by this lexer. 598 unsigned isNextPPTokenLParen(); 599 600 //===--------------------------------------------------------------------===// 601 // Lexer character reading interfaces. 602 603 // This lexer is built on two interfaces for reading characters, both of which 604 // automatically provide phase 1/2 translation. getAndAdvanceChar is used 605 // when we know that we will be reading a character from the input buffer and 606 // that this character will be part of the result token. This occurs in (f.e.) 607 // string processing, because we know we need to read until we find the 608 // closing '"' character. 609 // 610 // The second interface is the combination of getCharAndSize with 611 // ConsumeChar. getCharAndSize reads a phase 1/2 translated character, 612 // returning it and its size. If the lexer decides that this character is 613 // part of the current token, it calls ConsumeChar on it. This two stage 614 // approach allows us to emit diagnostics for characters (e.g. warnings about 615 // trigraphs), knowing that they only are emitted if the character is 616 // consumed. 617 618 /// isObviouslySimpleCharacter - Return true if the specified character is 619 /// obviously the same in translation phase 1 and translation phase 3. This 620 /// can return false for characters that end up being the same, but it will 621 /// never return true for something that needs to be mapped. isObviouslySimpleCharacter(char C)622 static bool isObviouslySimpleCharacter(char C) { 623 return C != '?' && C != '\\'; 624 } 625 626 /// getAndAdvanceChar - Read a single 'character' from the specified buffer, 627 /// advance over it, and return it. This is tricky in several cases. Here we 628 /// just handle the trivial case and fall-back to the non-inlined 629 /// getCharAndSizeSlow method to handle the hard case. getAndAdvanceChar(const char * & Ptr,Token & Tok)630 inline char getAndAdvanceChar(const char *&Ptr, Token &Tok) { 631 // If this is not a trigraph and not a UCN or escaped newline, return 632 // quickly. 633 if (isObviouslySimpleCharacter(Ptr[0])) return *Ptr++; 634 635 unsigned Size = 0; 636 char C = getCharAndSizeSlow(Ptr, Size, &Tok); 637 Ptr += Size; 638 return C; 639 } 640 641 /// ConsumeChar - When a character (identified by getCharAndSize) is consumed 642 /// and added to a given token, check to see if there are diagnostics that 643 /// need to be emitted or flags that need to be set on the token. If so, do 644 /// it. ConsumeChar(const char * Ptr,unsigned Size,Token & Tok)645 const char *ConsumeChar(const char *Ptr, unsigned Size, Token &Tok) { 646 // Normal case, we consumed exactly one token. Just return it. 647 if (Size == 1) 648 return Ptr+Size; 649 650 // Otherwise, re-lex the character with a current token, allowing 651 // diagnostics to be emitted and flags to be set. 652 Size = 0; 653 getCharAndSizeSlow(Ptr, Size, &Tok); 654 return Ptr+Size; 655 } 656 657 /// getCharAndSize - Peek a single 'character' from the specified buffer, 658 /// get its size, and return it. This is tricky in several cases. Here we 659 /// just handle the trivial case and fall-back to the non-inlined 660 /// getCharAndSizeSlow method to handle the hard case. getCharAndSize(const char * Ptr,unsigned & Size)661 inline char getCharAndSize(const char *Ptr, unsigned &Size) { 662 // If this is not a trigraph and not a UCN or escaped newline, return 663 // quickly. 664 if (isObviouslySimpleCharacter(Ptr[0])) { 665 Size = 1; 666 return *Ptr; 667 } 668 669 Size = 0; 670 return getCharAndSizeSlow(Ptr, Size); 671 } 672 673 /// getCharAndSizeSlow - Handle the slow/uncommon case of the getCharAndSize 674 /// method. 675 char getCharAndSizeSlow(const char *Ptr, unsigned &Size, 676 Token *Tok = nullptr); 677 678 /// getEscapedNewLineSize - Return the size of the specified escaped newline, 679 /// or 0 if it is not an escaped newline. P[-1] is known to be a "\" on entry 680 /// to this function. 681 static unsigned getEscapedNewLineSize(const char *P); 682 683 /// SkipEscapedNewLines - If P points to an escaped newline (or a series of 684 /// them), skip over them and return the first non-escaped-newline found, 685 /// otherwise return P. 686 static const char *SkipEscapedNewLines(const char *P); 687 688 /// getCharAndSizeSlowNoWarn - Same as getCharAndSizeSlow, but never emits a 689 /// diagnostic. 690 static char getCharAndSizeSlowNoWarn(const char *Ptr, unsigned &Size, 691 const LangOptions &LangOpts); 692 693 //===--------------------------------------------------------------------===// 694 // Other lexer functions. 695 696 void SetByteOffset(unsigned Offset, bool StartOfLine); 697 698 void PropagateLineStartLeadingSpaceInfo(Token &Result); 699 700 const char *LexUDSuffix(Token &Result, const char *CurPtr, 701 bool IsStringLiteral); 702 703 // Helper functions to lex the remainder of a token of the specific type. 704 bool LexIdentifier (Token &Result, const char *CurPtr); 705 bool LexNumericConstant (Token &Result, const char *CurPtr); 706 bool LexStringLiteral (Token &Result, const char *CurPtr, 707 tok::TokenKind Kind); 708 bool LexRawStringLiteral (Token &Result, const char *CurPtr, 709 tok::TokenKind Kind); 710 bool LexAngledStringLiteral(Token &Result, const char *CurPtr); 711 bool LexCharConstant (Token &Result, const char *CurPtr, 712 tok::TokenKind Kind); 713 bool LexEndOfFile (Token &Result, const char *CurPtr); 714 bool SkipWhitespace (Token &Result, const char *CurPtr, 715 bool &TokAtPhysicalStartOfLine); 716 bool SkipLineComment (Token &Result, const char *CurPtr, 717 bool &TokAtPhysicalStartOfLine); 718 bool SkipBlockComment (Token &Result, const char *CurPtr, 719 bool &TokAtPhysicalStartOfLine); 720 bool SaveLineComment (Token &Result, const char *CurPtr); 721 722 bool IsStartOfConflictMarker(const char *CurPtr); 723 bool HandleEndOfConflictMarker(const char *CurPtr); 724 725 bool lexEditorPlaceholder(Token &Result, const char *CurPtr); 726 727 bool isCodeCompletionPoint(const char *CurPtr) const; cutOffLexing()728 void cutOffLexing() { BufferPtr = BufferEnd; } 729 730 bool isHexaLiteral(const char *Start, const LangOptions &LangOpts); 731 732 void codeCompleteIncludedFile(const char *PathStart, 733 const char *CompletionPoint, bool IsAngled); 734 735 /// Read a universal character name. 736 /// 737 /// \param StartPtr The position in the source buffer after the initial '\'. 738 /// If the UCN is syntactically well-formed (but not 739 /// necessarily valid), this parameter will be updated to 740 /// point to the character after the UCN. 741 /// \param SlashLoc The position in the source buffer of the '\'. 742 /// \param Result The token being formed. Pass \c nullptr to suppress 743 /// diagnostics and handle token formation in the caller. 744 /// 745 /// \return The Unicode codepoint specified by the UCN, or 0 if the UCN is 746 /// invalid. 747 uint32_t tryReadUCN(const char *&StartPtr, const char *SlashLoc, Token *Result); 748 749 /// Try to consume a UCN as part of an identifier at the current 750 /// location. 751 /// \param CurPtr Initially points to the range of characters in the source 752 /// buffer containing the '\'. Updated to point past the end of 753 /// the UCN on success. 754 /// \param Size The number of characters occupied by the '\' (including 755 /// trigraphs and escaped newlines). 756 /// \param Result The token being produced. Marked as containing a UCN on 757 /// success. 758 /// \return \c true if a UCN was lexed and it produced an acceptable 759 /// identifier character, \c false otherwise. 760 bool tryConsumeIdentifierUCN(const char *&CurPtr, unsigned Size, 761 Token &Result); 762 763 /// Try to consume an identifier character encoded in UTF-8. 764 /// \param CurPtr Points to the start of the (potential) UTF-8 code unit 765 /// sequence. On success, updated to point past the end of it. 766 /// \return \c true if a UTF-8 sequence mapping to an acceptable identifier 767 /// character was lexed, \c false otherwise. 768 bool tryConsumeIdentifierUTF8Char(const char *&CurPtr); 769 }; 770 771 } // namespace clang 772 773 #endif // LLVM_CLANG_LEX_LEXER_H 774