1 //===--- LiteralSupport.cpp - Code to parse and process literals ----------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This file implements the NumericLiteralParser, CharLiteralParser, and 11 // StringLiteralParser interfaces. 12 // 13 //===----------------------------------------------------------------------===// 14 15 #include "clang/Lex/LiteralSupport.h" 16 #include "clang/Basic/CharInfo.h" 17 #include "clang/Basic/TargetInfo.h" 18 #include "clang/Lex/LexDiagnostic.h" 19 #include "clang/Lex/Preprocessor.h" 20 #include "llvm/ADT/StringExtras.h" 21 #include "llvm/Support/ConvertUTF.h" 22 #include "llvm/Support/ErrorHandling.h" 23 24 using namespace clang; 25 26 static unsigned getCharWidth(tok::TokenKind kind, const TargetInfo &Target) { 27 switch (kind) { 28 default: llvm_unreachable("Unknown token type!"); 29 case tok::char_constant: 30 case tok::string_literal: 31 case tok::utf8_string_literal: 32 return Target.getCharWidth(); 33 case tok::wide_char_constant: 34 case tok::wide_string_literal: 35 return Target.getWCharWidth(); 36 case tok::utf16_char_constant: 37 case tok::utf16_string_literal: 38 return Target.getChar16Width(); 39 case tok::utf32_char_constant: 40 case tok::utf32_string_literal: 41 return Target.getChar32Width(); 42 } 43 } 44 45 static CharSourceRange MakeCharSourceRange(const LangOptions &Features, 46 FullSourceLoc TokLoc, 47 const char *TokBegin, 48 const char *TokRangeBegin, 49 const char *TokRangeEnd) { 50 SourceLocation Begin = 51 Lexer::AdvanceToTokenCharacter(TokLoc, TokRangeBegin - TokBegin, 52 TokLoc.getManager(), Features); 53 SourceLocation End = 54 Lexer::AdvanceToTokenCharacter(Begin, TokRangeEnd - TokRangeBegin, 55 TokLoc.getManager(), Features); 56 return CharSourceRange::getCharRange(Begin, End); 57 } 58 59 /// \brief Produce a diagnostic highlighting some portion of a literal. 60 /// 61 /// Emits the diagnostic \p DiagID, highlighting the range of characters from 62 /// \p TokRangeBegin (inclusive) to \p TokRangeEnd (exclusive), which must be 63 /// a substring of a spelling buffer for the token beginning at \p TokBegin. 64 static DiagnosticBuilder Diag(DiagnosticsEngine *Diags, 65 const LangOptions &Features, FullSourceLoc TokLoc, 66 const char *TokBegin, const char *TokRangeBegin, 67 const char *TokRangeEnd, unsigned DiagID) { 68 SourceLocation Begin = 69 Lexer::AdvanceToTokenCharacter(TokLoc, TokRangeBegin - TokBegin, 70 TokLoc.getManager(), Features); 71 return Diags->Report(Begin, DiagID) << 72 MakeCharSourceRange(Features, TokLoc, TokBegin, TokRangeBegin, TokRangeEnd); 73 } 74 75 /// ProcessCharEscape - Parse a standard C escape sequence, which can occur in 76 /// either a character or a string literal. 77 static unsigned ProcessCharEscape(const char *ThisTokBegin, 78 const char *&ThisTokBuf, 79 const char *ThisTokEnd, bool &HadError, 80 FullSourceLoc Loc, unsigned CharWidth, 81 DiagnosticsEngine *Diags, 82 const LangOptions &Features) { 83 const char *EscapeBegin = ThisTokBuf; 84 85 // Skip the '\' char. 86 ++ThisTokBuf; 87 88 // We know that this character can't be off the end of the buffer, because 89 // that would have been \", which would not have been the end of string. 90 unsigned ResultChar = *ThisTokBuf++; 91 switch (ResultChar) { 92 // These map to themselves. 93 case '\\': case '\'': case '"': case '?': break; 94 95 // These have fixed mappings. 96 case 'a': 97 // TODO: K&R: the meaning of '\\a' is different in traditional C 98 ResultChar = 7; 99 break; 100 case 'b': 101 ResultChar = 8; 102 break; 103 case 'e': 104 if (Diags) 105 Diag(Diags, Features, Loc, ThisTokBegin, EscapeBegin, ThisTokBuf, 106 diag::ext_nonstandard_escape) << "e"; 107 ResultChar = 27; 108 break; 109 case 'E': 110 if (Diags) 111 Diag(Diags, Features, Loc, ThisTokBegin, EscapeBegin, ThisTokBuf, 112 diag::ext_nonstandard_escape) << "E"; 113 ResultChar = 27; 114 break; 115 case 'f': 116 ResultChar = 12; 117 break; 118 case 'n': 119 ResultChar = 10; 120 break; 121 case 'r': 122 ResultChar = 13; 123 break; 124 case 't': 125 ResultChar = 9; 126 break; 127 case 'v': 128 ResultChar = 11; 129 break; 130 case 'x': { // Hex escape. 131 ResultChar = 0; 132 if (ThisTokBuf == ThisTokEnd || !isHexDigit(*ThisTokBuf)) { 133 if (Diags) 134 Diag(Diags, Features, Loc, ThisTokBegin, EscapeBegin, ThisTokBuf, 135 diag::err_hex_escape_no_digits) << "x"; 136 HadError = 1; 137 break; 138 } 139 140 // Hex escapes are a maximal series of hex digits. 141 bool Overflow = false; 142 for (; ThisTokBuf != ThisTokEnd; ++ThisTokBuf) { 143 int CharVal = llvm::hexDigitValue(ThisTokBuf[0]); 144 if (CharVal == -1) break; 145 // About to shift out a digit? 146 Overflow |= (ResultChar & 0xF0000000) ? true : false; 147 ResultChar <<= 4; 148 ResultChar |= CharVal; 149 } 150 151 // See if any bits will be truncated when evaluated as a character. 152 if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) { 153 Overflow = true; 154 ResultChar &= ~0U >> (32-CharWidth); 155 } 156 157 // Check for overflow. 158 if (Overflow && Diags) // Too many digits to fit in 159 Diag(Diags, Features, Loc, ThisTokBegin, EscapeBegin, ThisTokBuf, 160 diag::err_hex_escape_too_large); 161 break; 162 } 163 case '0': case '1': case '2': case '3': 164 case '4': case '5': case '6': case '7': { 165 // Octal escapes. 166 --ThisTokBuf; 167 ResultChar = 0; 168 169 // Octal escapes are a series of octal digits with maximum length 3. 170 // "\0123" is a two digit sequence equal to "\012" "3". 171 unsigned NumDigits = 0; 172 do { 173 ResultChar <<= 3; 174 ResultChar |= *ThisTokBuf++ - '0'; 175 ++NumDigits; 176 } while (ThisTokBuf != ThisTokEnd && NumDigits < 3 && 177 ThisTokBuf[0] >= '0' && ThisTokBuf[0] <= '7'); 178 179 // Check for overflow. Reject '\777', but not L'\777'. 180 if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) { 181 if (Diags) 182 Diag(Diags, Features, Loc, ThisTokBegin, EscapeBegin, ThisTokBuf, 183 diag::err_octal_escape_too_large); 184 ResultChar &= ~0U >> (32-CharWidth); 185 } 186 break; 187 } 188 189 // Otherwise, these are not valid escapes. 190 case '(': case '{': case '[': case '%': 191 // GCC accepts these as extensions. We warn about them as such though. 192 if (Diags) 193 Diag(Diags, Features, Loc, ThisTokBegin, EscapeBegin, ThisTokBuf, 194 diag::ext_nonstandard_escape) 195 << std::string(1, ResultChar); 196 break; 197 default: 198 if (Diags == 0) 199 break; 200 201 if (isPrintable(ResultChar)) 202 Diag(Diags, Features, Loc, ThisTokBegin, EscapeBegin, ThisTokBuf, 203 diag::ext_unknown_escape) 204 << std::string(1, ResultChar); 205 else 206 Diag(Diags, Features, Loc, ThisTokBegin, EscapeBegin, ThisTokBuf, 207 diag::ext_unknown_escape) 208 << "x" + llvm::utohexstr(ResultChar); 209 break; 210 } 211 212 return ResultChar; 213 } 214 215 static void appendCodePoint(unsigned Codepoint, 216 llvm::SmallVectorImpl<char> &Str) { 217 char ResultBuf[4]; 218 char *ResultPtr = ResultBuf; 219 bool Res = llvm::ConvertCodePointToUTF8(Codepoint, ResultPtr); 220 (void)Res; 221 assert(Res && "Unexpected conversion failure"); 222 Str.append(ResultBuf, ResultPtr); 223 } 224 225 void clang::expandUCNs(SmallVectorImpl<char> &Buf, StringRef Input) { 226 for (StringRef::iterator I = Input.begin(), E = Input.end(); I != E; ++I) { 227 if (*I != '\\') { 228 Buf.push_back(*I); 229 continue; 230 } 231 232 ++I; 233 assert(*I == 'u' || *I == 'U'); 234 235 unsigned NumHexDigits; 236 if (*I == 'u') 237 NumHexDigits = 4; 238 else 239 NumHexDigits = 8; 240 241 assert(I + NumHexDigits <= E); 242 243 uint32_t CodePoint = 0; 244 for (++I; NumHexDigits != 0; ++I, --NumHexDigits) { 245 unsigned Value = llvm::hexDigitValue(*I); 246 assert(Value != -1U); 247 248 CodePoint <<= 4; 249 CodePoint += Value; 250 } 251 252 appendCodePoint(CodePoint, Buf); 253 --I; 254 } 255 } 256 257 /// ProcessUCNEscape - Read the Universal Character Name, check constraints and 258 /// return the UTF32. 259 static bool ProcessUCNEscape(const char *ThisTokBegin, const char *&ThisTokBuf, 260 const char *ThisTokEnd, 261 uint32_t &UcnVal, unsigned short &UcnLen, 262 FullSourceLoc Loc, DiagnosticsEngine *Diags, 263 const LangOptions &Features, 264 bool in_char_string_literal = false) { 265 const char *UcnBegin = ThisTokBuf; 266 267 // Skip the '\u' char's. 268 ThisTokBuf += 2; 269 270 if (ThisTokBuf == ThisTokEnd || !isHexDigit(*ThisTokBuf)) { 271 if (Diags) 272 Diag(Diags, Features, Loc, ThisTokBegin, UcnBegin, ThisTokBuf, 273 diag::err_hex_escape_no_digits) << StringRef(&ThisTokBuf[-1], 1); 274 return false; 275 } 276 UcnLen = (ThisTokBuf[-1] == 'u' ? 4 : 8); 277 unsigned short UcnLenSave = UcnLen; 278 for (; ThisTokBuf != ThisTokEnd && UcnLenSave; ++ThisTokBuf, UcnLenSave--) { 279 int CharVal = llvm::hexDigitValue(ThisTokBuf[0]); 280 if (CharVal == -1) break; 281 UcnVal <<= 4; 282 UcnVal |= CharVal; 283 } 284 // If we didn't consume the proper number of digits, there is a problem. 285 if (UcnLenSave) { 286 if (Diags) 287 Diag(Diags, Features, Loc, ThisTokBegin, UcnBegin, ThisTokBuf, 288 diag::err_ucn_escape_incomplete); 289 return false; 290 } 291 292 // Check UCN constraints (C99 6.4.3p2) [C++11 lex.charset p2] 293 if ((0xD800 <= UcnVal && UcnVal <= 0xDFFF) || // surrogate codepoints 294 UcnVal > 0x10FFFF) { // maximum legal UTF32 value 295 if (Diags) 296 Diag(Diags, Features, Loc, ThisTokBegin, UcnBegin, ThisTokBuf, 297 diag::err_ucn_escape_invalid); 298 return false; 299 } 300 301 // C++11 allows UCNs that refer to control characters and basic source 302 // characters inside character and string literals 303 if (UcnVal < 0xa0 && 304 (UcnVal != 0x24 && UcnVal != 0x40 && UcnVal != 0x60)) { // $, @, ` 305 bool IsError = (!Features.CPlusPlus11 || !in_char_string_literal); 306 if (Diags) { 307 char BasicSCSChar = UcnVal; 308 if (UcnVal >= 0x20 && UcnVal < 0x7f) 309 Diag(Diags, Features, Loc, ThisTokBegin, UcnBegin, ThisTokBuf, 310 IsError ? diag::err_ucn_escape_basic_scs : 311 diag::warn_cxx98_compat_literal_ucn_escape_basic_scs) 312 << StringRef(&BasicSCSChar, 1); 313 else 314 Diag(Diags, Features, Loc, ThisTokBegin, UcnBegin, ThisTokBuf, 315 IsError ? diag::err_ucn_control_character : 316 diag::warn_cxx98_compat_literal_ucn_control_character); 317 } 318 if (IsError) 319 return false; 320 } 321 322 if (!Features.CPlusPlus && !Features.C99 && Diags) 323 Diag(Diags, Features, Loc, ThisTokBegin, UcnBegin, ThisTokBuf, 324 diag::warn_ucn_not_valid_in_c89_literal); 325 326 return true; 327 } 328 329 /// MeasureUCNEscape - Determine the number of bytes within the resulting string 330 /// which this UCN will occupy. 331 static int MeasureUCNEscape(const char *ThisTokBegin, const char *&ThisTokBuf, 332 const char *ThisTokEnd, unsigned CharByteWidth, 333 const LangOptions &Features, bool &HadError) { 334 // UTF-32: 4 bytes per escape. 335 if (CharByteWidth == 4) 336 return 4; 337 338 uint32_t UcnVal = 0; 339 unsigned short UcnLen = 0; 340 FullSourceLoc Loc; 341 342 if (!ProcessUCNEscape(ThisTokBegin, ThisTokBuf, ThisTokEnd, UcnVal, 343 UcnLen, Loc, 0, Features, true)) { 344 HadError = true; 345 return 0; 346 } 347 348 // UTF-16: 2 bytes for BMP, 4 bytes otherwise. 349 if (CharByteWidth == 2) 350 return UcnVal <= 0xFFFF ? 2 : 4; 351 352 // UTF-8. 353 if (UcnVal < 0x80) 354 return 1; 355 if (UcnVal < 0x800) 356 return 2; 357 if (UcnVal < 0x10000) 358 return 3; 359 return 4; 360 } 361 362 /// EncodeUCNEscape - Read the Universal Character Name, check constraints and 363 /// convert the UTF32 to UTF8 or UTF16. This is a subroutine of 364 /// StringLiteralParser. When we decide to implement UCN's for identifiers, 365 /// we will likely rework our support for UCN's. 366 static void EncodeUCNEscape(const char *ThisTokBegin, const char *&ThisTokBuf, 367 const char *ThisTokEnd, 368 char *&ResultBuf, bool &HadError, 369 FullSourceLoc Loc, unsigned CharByteWidth, 370 DiagnosticsEngine *Diags, 371 const LangOptions &Features) { 372 typedef uint32_t UTF32; 373 UTF32 UcnVal = 0; 374 unsigned short UcnLen = 0; 375 if (!ProcessUCNEscape(ThisTokBegin, ThisTokBuf, ThisTokEnd, UcnVal, UcnLen, 376 Loc, Diags, Features, true)) { 377 HadError = true; 378 return; 379 } 380 381 assert((CharByteWidth == 1 || CharByteWidth == 2 || CharByteWidth == 4) && 382 "only character widths of 1, 2, or 4 bytes supported"); 383 384 (void)UcnLen; 385 assert((UcnLen== 4 || UcnLen== 8) && "only ucn length of 4 or 8 supported"); 386 387 if (CharByteWidth == 4) { 388 // FIXME: Make the type of the result buffer correct instead of 389 // using reinterpret_cast. 390 UTF32 *ResultPtr = reinterpret_cast<UTF32*>(ResultBuf); 391 *ResultPtr = UcnVal; 392 ResultBuf += 4; 393 return; 394 } 395 396 if (CharByteWidth == 2) { 397 // FIXME: Make the type of the result buffer correct instead of 398 // using reinterpret_cast. 399 UTF16 *ResultPtr = reinterpret_cast<UTF16*>(ResultBuf); 400 401 if (UcnVal <= (UTF32)0xFFFF) { 402 *ResultPtr = UcnVal; 403 ResultBuf += 2; 404 return; 405 } 406 407 // Convert to UTF16. 408 UcnVal -= 0x10000; 409 *ResultPtr = 0xD800 + (UcnVal >> 10); 410 *(ResultPtr+1) = 0xDC00 + (UcnVal & 0x3FF); 411 ResultBuf += 4; 412 return; 413 } 414 415 assert(CharByteWidth == 1 && "UTF-8 encoding is only for 1 byte characters"); 416 417 // Now that we've parsed/checked the UCN, we convert from UTF32->UTF8. 418 // The conversion below was inspired by: 419 // http://www.unicode.org/Public/PROGRAMS/CVTUTF/ConvertUTF.c 420 // First, we determine how many bytes the result will require. 421 typedef uint8_t UTF8; 422 423 unsigned short bytesToWrite = 0; 424 if (UcnVal < (UTF32)0x80) 425 bytesToWrite = 1; 426 else if (UcnVal < (UTF32)0x800) 427 bytesToWrite = 2; 428 else if (UcnVal < (UTF32)0x10000) 429 bytesToWrite = 3; 430 else 431 bytesToWrite = 4; 432 433 const unsigned byteMask = 0xBF; 434 const unsigned byteMark = 0x80; 435 436 // Once the bits are split out into bytes of UTF8, this is a mask OR-ed 437 // into the first byte, depending on how many bytes follow. 438 static const UTF8 firstByteMark[5] = { 439 0x00, 0x00, 0xC0, 0xE0, 0xF0 440 }; 441 // Finally, we write the bytes into ResultBuf. 442 ResultBuf += bytesToWrite; 443 switch (bytesToWrite) { // note: everything falls through. 444 case 4: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6; 445 case 3: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6; 446 case 2: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6; 447 case 1: *--ResultBuf = (UTF8) (UcnVal | firstByteMark[bytesToWrite]); 448 } 449 // Update the buffer. 450 ResultBuf += bytesToWrite; 451 } 452 453 454 /// integer-constant: [C99 6.4.4.1] 455 /// decimal-constant integer-suffix 456 /// octal-constant integer-suffix 457 /// hexadecimal-constant integer-suffix 458 /// binary-literal integer-suffix [GNU, C++1y] 459 /// user-defined-integer-literal: [C++11 lex.ext] 460 /// decimal-literal ud-suffix 461 /// octal-literal ud-suffix 462 /// hexadecimal-literal ud-suffix 463 /// binary-literal ud-suffix [GNU, C++1y] 464 /// decimal-constant: 465 /// nonzero-digit 466 /// decimal-constant digit 467 /// octal-constant: 468 /// 0 469 /// octal-constant octal-digit 470 /// hexadecimal-constant: 471 /// hexadecimal-prefix hexadecimal-digit 472 /// hexadecimal-constant hexadecimal-digit 473 /// hexadecimal-prefix: one of 474 /// 0x 0X 475 /// binary-literal: 476 /// 0b binary-digit 477 /// 0B binary-digit 478 /// binary-literal binary-digit 479 /// integer-suffix: 480 /// unsigned-suffix [long-suffix] 481 /// unsigned-suffix [long-long-suffix] 482 /// long-suffix [unsigned-suffix] 483 /// long-long-suffix [unsigned-sufix] 484 /// nonzero-digit: 485 /// 1 2 3 4 5 6 7 8 9 486 /// octal-digit: 487 /// 0 1 2 3 4 5 6 7 488 /// hexadecimal-digit: 489 /// 0 1 2 3 4 5 6 7 8 9 490 /// a b c d e f 491 /// A B C D E F 492 /// binary-digit: 493 /// 0 494 /// 1 495 /// unsigned-suffix: one of 496 /// u U 497 /// long-suffix: one of 498 /// l L 499 /// long-long-suffix: one of 500 /// ll LL 501 /// 502 /// floating-constant: [C99 6.4.4.2] 503 /// TODO: add rules... 504 /// 505 NumericLiteralParser::NumericLiteralParser(StringRef TokSpelling, 506 SourceLocation TokLoc, 507 Preprocessor &PP) 508 : PP(PP), ThisTokBegin(TokSpelling.begin()), ThisTokEnd(TokSpelling.end()) { 509 510 // This routine assumes that the range begin/end matches the regex for integer 511 // and FP constants (specifically, the 'pp-number' regex), and assumes that 512 // the byte at "*end" is both valid and not part of the regex. Because of 513 // this, it doesn't have to check for 'overscan' in various places. 514 assert(!isPreprocessingNumberBody(*ThisTokEnd) && "didn't maximally munch?"); 515 516 s = DigitsBegin = ThisTokBegin; 517 saw_exponent = false; 518 saw_period = false; 519 saw_ud_suffix = false; 520 isLong = false; 521 isUnsigned = false; 522 isLongLong = false; 523 isFloat = false; 524 isImaginary = false; 525 isMicrosoftInteger = false; 526 hadError = false; 527 528 if (*s == '0') { // parse radix 529 ParseNumberStartingWithZero(TokLoc); 530 if (hadError) 531 return; 532 } else { // the first digit is non-zero 533 radix = 10; 534 s = SkipDigits(s); 535 if (s == ThisTokEnd) { 536 // Done. 537 } else if (isHexDigit(*s) && !(*s == 'e' || *s == 'E')) { 538 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s - ThisTokBegin), 539 diag::err_invalid_decimal_digit) << StringRef(s, 1); 540 hadError = true; 541 return; 542 } else if (*s == '.') { 543 checkSeparator(TokLoc, s, CSK_AfterDigits); 544 s++; 545 saw_period = true; 546 checkSeparator(TokLoc, s, CSK_BeforeDigits); 547 s = SkipDigits(s); 548 } 549 if ((*s == 'e' || *s == 'E')) { // exponent 550 checkSeparator(TokLoc, s, CSK_AfterDigits); 551 const char *Exponent = s; 552 s++; 553 saw_exponent = true; 554 if (*s == '+' || *s == '-') s++; // sign 555 checkSeparator(TokLoc, s, CSK_BeforeDigits); 556 const char *first_non_digit = SkipDigits(s); 557 if (first_non_digit != s) { 558 s = first_non_digit; 559 } else { 560 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent - ThisTokBegin), 561 diag::err_exponent_has_no_digits); 562 hadError = true; 563 return; 564 } 565 } 566 } 567 568 SuffixBegin = s; 569 checkSeparator(TokLoc, s, CSK_AfterDigits); 570 571 // Parse the suffix. At this point we can classify whether we have an FP or 572 // integer constant. 573 bool isFPConstant = isFloatingLiteral(); 574 const char *ImaginarySuffixLoc = 0; 575 576 // Loop over all of the characters of the suffix. If we see something bad, 577 // we break out of the loop. 578 for (; s != ThisTokEnd; ++s) { 579 switch (*s) { 580 case 'f': // FP Suffix for "float" 581 case 'F': 582 if (!isFPConstant) break; // Error for integer constant. 583 if (isFloat || isLong) break; // FF, LF invalid. 584 isFloat = true; 585 continue; // Success. 586 case 'u': 587 case 'U': 588 if (isFPConstant) break; // Error for floating constant. 589 if (isUnsigned) break; // Cannot be repeated. 590 isUnsigned = true; 591 continue; // Success. 592 case 'l': 593 case 'L': 594 if (isLong || isLongLong) break; // Cannot be repeated. 595 if (isFloat) break; // LF invalid. 596 597 // Check for long long. The L's need to be adjacent and the same case. 598 if (s+1 != ThisTokEnd && s[1] == s[0]) { 599 if (isFPConstant) break; // long long invalid for floats. 600 isLongLong = true; 601 ++s; // Eat both of them. 602 } else { 603 isLong = true; 604 } 605 continue; // Success. 606 case 'i': 607 case 'I': 608 if (PP.getLangOpts().MicrosoftExt) { 609 if (isFPConstant || isLong || isLongLong) break; 610 611 // Allow i8, i16, i32, i64, and i128. 612 if (s + 1 != ThisTokEnd) { 613 switch (s[1]) { 614 case '8': 615 s += 2; // i8 suffix 616 isMicrosoftInteger = true; 617 break; 618 case '1': 619 if (s + 2 == ThisTokEnd) break; 620 if (s[2] == '6') { 621 s += 3; // i16 suffix 622 isMicrosoftInteger = true; 623 } 624 else if (s[2] == '2') { 625 if (s + 3 == ThisTokEnd) break; 626 if (s[3] == '8') { 627 s += 4; // i128 suffix 628 isMicrosoftInteger = true; 629 } 630 } 631 break; 632 case '3': 633 if (s + 2 == ThisTokEnd) break; 634 if (s[2] == '2') { 635 s += 3; // i32 suffix 636 isLong = true; 637 isMicrosoftInteger = true; 638 } 639 break; 640 case '6': 641 if (s + 2 == ThisTokEnd) break; 642 if (s[2] == '4') { 643 s += 3; // i64 suffix 644 isLongLong = true; 645 isMicrosoftInteger = true; 646 } 647 break; 648 default: 649 break; 650 } 651 break; 652 } 653 } 654 // "i", "if", and "il" are user-defined suffixes in C++1y. 655 if (PP.getLangOpts().CPlusPlus1y && *s == 'i') 656 break; 657 // fall through. 658 case 'j': 659 case 'J': 660 if (isImaginary) break; // Cannot be repeated. 661 isImaginary = true; 662 ImaginarySuffixLoc = s; 663 continue; // Success. 664 } 665 // If we reached here, there was an error or a ud-suffix. 666 break; 667 } 668 669 if (s != ThisTokEnd) { 670 // FIXME: Don't bother expanding UCNs if !tok.hasUCN(). 671 expandUCNs(UDSuffixBuf, StringRef(SuffixBegin, ThisTokEnd - SuffixBegin)); 672 if (isValidUDSuffix(PP.getLangOpts(), UDSuffixBuf)) { 673 // Any suffix pieces we might have parsed are actually part of the 674 // ud-suffix. 675 isLong = false; 676 isUnsigned = false; 677 isLongLong = false; 678 isFloat = false; 679 isImaginary = false; 680 isMicrosoftInteger = false; 681 682 saw_ud_suffix = true; 683 return; 684 } 685 686 // Report an error if there are any. 687 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, SuffixBegin - ThisTokBegin), 688 isFPConstant ? diag::err_invalid_suffix_float_constant : 689 diag::err_invalid_suffix_integer_constant) 690 << StringRef(SuffixBegin, ThisTokEnd-SuffixBegin); 691 hadError = true; 692 return; 693 } 694 695 if (isImaginary) { 696 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, 697 ImaginarySuffixLoc - ThisTokBegin), 698 diag::ext_imaginary_constant); 699 } 700 } 701 702 /// Determine whether a suffix is a valid ud-suffix. We avoid treating reserved 703 /// suffixes as ud-suffixes, because the diagnostic experience is better if we 704 /// treat it as an invalid suffix. 705 bool NumericLiteralParser::isValidUDSuffix(const LangOptions &LangOpts, 706 StringRef Suffix) { 707 if (!LangOpts.CPlusPlus11 || Suffix.empty()) 708 return false; 709 710 // By C++11 [lex.ext]p10, ud-suffixes starting with an '_' are always valid. 711 if (Suffix[0] == '_') 712 return true; 713 714 // In C++11, there are no library suffixes. 715 if (!LangOpts.CPlusPlus1y) 716 return false; 717 718 // In C++1y, "s", "h", "min", "ms", "us", and "ns" are used in the library. 719 // Per tweaked N3660, "il", "i", and "if" are also used in the library. 720 return llvm::StringSwitch<bool>(Suffix) 721 .Cases("h", "min", "s", true) 722 .Cases("ms", "us", "ns", true) 723 .Cases("il", "i", "if", true) 724 .Default(false); 725 } 726 727 void NumericLiteralParser::checkSeparator(SourceLocation TokLoc, 728 const char *Pos, 729 CheckSeparatorKind IsAfterDigits) { 730 if (IsAfterDigits == CSK_AfterDigits) { 731 if (Pos == ThisTokBegin) 732 return; 733 --Pos; 734 } else if (Pos == ThisTokEnd) 735 return; 736 737 if (isDigitSeparator(*Pos)) 738 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Pos - ThisTokBegin), 739 diag::err_digit_separator_not_between_digits) 740 << IsAfterDigits; 741 } 742 743 /// ParseNumberStartingWithZero - This method is called when the first character 744 /// of the number is found to be a zero. This means it is either an octal 745 /// number (like '04') or a hex number ('0x123a') a binary number ('0b1010') or 746 /// a floating point number (01239.123e4). Eat the prefix, determining the 747 /// radix etc. 748 void NumericLiteralParser::ParseNumberStartingWithZero(SourceLocation TokLoc) { 749 assert(s[0] == '0' && "Invalid method call"); 750 s++; 751 752 int c1 = s[0]; 753 int c2 = s[1]; 754 755 // Handle a hex number like 0x1234. 756 if ((c1 == 'x' || c1 == 'X') && (isHexDigit(c2) || c2 == '.')) { 757 s++; 758 radix = 16; 759 DigitsBegin = s; 760 s = SkipHexDigits(s); 761 bool noSignificand = (s == DigitsBegin); 762 if (s == ThisTokEnd) { 763 // Done. 764 } else if (*s == '.') { 765 s++; 766 saw_period = true; 767 const char *floatDigitsBegin = s; 768 checkSeparator(TokLoc, s, CSK_BeforeDigits); 769 s = SkipHexDigits(s); 770 noSignificand &= (floatDigitsBegin == s); 771 } 772 773 if (noSignificand) { 774 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s - ThisTokBegin), 775 diag::err_hexconstant_requires_digits); 776 hadError = true; 777 return; 778 } 779 780 // A binary exponent can appear with or with a '.'. If dotted, the 781 // binary exponent is required. 782 if (*s == 'p' || *s == 'P') { 783 checkSeparator(TokLoc, s, CSK_AfterDigits); 784 const char *Exponent = s; 785 s++; 786 saw_exponent = true; 787 if (*s == '+' || *s == '-') s++; // sign 788 const char *first_non_digit = SkipDigits(s); 789 if (first_non_digit == s) { 790 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin), 791 diag::err_exponent_has_no_digits); 792 hadError = true; 793 return; 794 } 795 checkSeparator(TokLoc, s, CSK_BeforeDigits); 796 s = first_non_digit; 797 798 if (!PP.getLangOpts().HexFloats) 799 PP.Diag(TokLoc, diag::ext_hexconstant_invalid); 800 } else if (saw_period) { 801 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin), 802 diag::err_hexconstant_requires_exponent); 803 hadError = true; 804 } 805 return; 806 } 807 808 // Handle simple binary numbers 0b01010 809 if ((c1 == 'b' || c1 == 'B') && (c2 == '0' || c2 == '1')) { 810 // 0b101010 is a C++1y / GCC extension. 811 PP.Diag(TokLoc, 812 PP.getLangOpts().CPlusPlus1y 813 ? diag::warn_cxx11_compat_binary_literal 814 : PP.getLangOpts().CPlusPlus 815 ? diag::ext_binary_literal_cxx1y 816 : diag::ext_binary_literal); 817 ++s; 818 radix = 2; 819 DigitsBegin = s; 820 s = SkipBinaryDigits(s); 821 if (s == ThisTokEnd) { 822 // Done. 823 } else if (isHexDigit(*s)) { 824 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin), 825 diag::err_invalid_binary_digit) << StringRef(s, 1); 826 hadError = true; 827 } 828 // Other suffixes will be diagnosed by the caller. 829 return; 830 } 831 832 // For now, the radix is set to 8. If we discover that we have a 833 // floating point constant, the radix will change to 10. Octal floating 834 // point constants are not permitted (only decimal and hexadecimal). 835 radix = 8; 836 DigitsBegin = s; 837 s = SkipOctalDigits(s); 838 if (s == ThisTokEnd) 839 return; // Done, simple octal number like 01234 840 841 // If we have some other non-octal digit that *is* a decimal digit, see if 842 // this is part of a floating point number like 094.123 or 09e1. 843 if (isDigit(*s)) { 844 const char *EndDecimal = SkipDigits(s); 845 if (EndDecimal[0] == '.' || EndDecimal[0] == 'e' || EndDecimal[0] == 'E') { 846 s = EndDecimal; 847 radix = 10; 848 } 849 } 850 851 // If we have a hex digit other than 'e' (which denotes a FP exponent) then 852 // the code is using an incorrect base. 853 if (isHexDigit(*s) && *s != 'e' && *s != 'E') { 854 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin), 855 diag::err_invalid_octal_digit) << StringRef(s, 1); 856 hadError = true; 857 return; 858 } 859 860 if (*s == '.') { 861 s++; 862 radix = 10; 863 saw_period = true; 864 checkSeparator(TokLoc, s, CSK_BeforeDigits); 865 s = SkipDigits(s); // Skip suffix. 866 } 867 if (*s == 'e' || *s == 'E') { // exponent 868 checkSeparator(TokLoc, s, CSK_AfterDigits); 869 const char *Exponent = s; 870 s++; 871 radix = 10; 872 saw_exponent = true; 873 if (*s == '+' || *s == '-') s++; // sign 874 const char *first_non_digit = SkipDigits(s); 875 if (first_non_digit != s) { 876 checkSeparator(TokLoc, s, CSK_BeforeDigits); 877 s = first_non_digit; 878 } else { 879 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin), 880 diag::err_exponent_has_no_digits); 881 hadError = true; 882 return; 883 } 884 } 885 } 886 887 static bool alwaysFitsInto64Bits(unsigned Radix, unsigned NumDigits) { 888 switch (Radix) { 889 case 2: 890 return NumDigits <= 64; 891 case 8: 892 return NumDigits <= 64 / 3; // Digits are groups of 3 bits. 893 case 10: 894 return NumDigits <= 19; // floor(log10(2^64)) 895 case 16: 896 return NumDigits <= 64 / 4; // Digits are groups of 4 bits. 897 default: 898 llvm_unreachable("impossible Radix"); 899 } 900 } 901 902 /// GetIntegerValue - Convert this numeric literal value to an APInt that 903 /// matches Val's input width. If there is an overflow, set Val to the low bits 904 /// of the result and return true. Otherwise, return false. 905 bool NumericLiteralParser::GetIntegerValue(llvm::APInt &Val) { 906 // Fast path: Compute a conservative bound on the maximum number of 907 // bits per digit in this radix. If we can't possibly overflow a 908 // uint64 based on that bound then do the simple conversion to 909 // integer. This avoids the expensive overflow checking below, and 910 // handles the common cases that matter (small decimal integers and 911 // hex/octal values which don't overflow). 912 const unsigned NumDigits = SuffixBegin - DigitsBegin; 913 if (alwaysFitsInto64Bits(radix, NumDigits)) { 914 uint64_t N = 0; 915 for (const char *Ptr = DigitsBegin; Ptr != SuffixBegin; ++Ptr) 916 if (!isDigitSeparator(*Ptr)) 917 N = N * radix + llvm::hexDigitValue(*Ptr); 918 919 // This will truncate the value to Val's input width. Simply check 920 // for overflow by comparing. 921 Val = N; 922 return Val.getZExtValue() != N; 923 } 924 925 Val = 0; 926 const char *Ptr = DigitsBegin; 927 928 llvm::APInt RadixVal(Val.getBitWidth(), radix); 929 llvm::APInt CharVal(Val.getBitWidth(), 0); 930 llvm::APInt OldVal = Val; 931 932 bool OverflowOccurred = false; 933 while (Ptr < SuffixBegin) { 934 if (isDigitSeparator(*Ptr)) { 935 ++Ptr; 936 continue; 937 } 938 939 unsigned C = llvm::hexDigitValue(*Ptr++); 940 941 // If this letter is out of bound for this radix, reject it. 942 assert(C < radix && "NumericLiteralParser ctor should have rejected this"); 943 944 CharVal = C; 945 946 // Add the digit to the value in the appropriate radix. If adding in digits 947 // made the value smaller, then this overflowed. 948 OldVal = Val; 949 950 // Multiply by radix, did overflow occur on the multiply? 951 Val *= RadixVal; 952 OverflowOccurred |= Val.udiv(RadixVal) != OldVal; 953 954 // Add value, did overflow occur on the value? 955 // (a + b) ult b <=> overflow 956 Val += CharVal; 957 OverflowOccurred |= Val.ult(CharVal); 958 } 959 return OverflowOccurred; 960 } 961 962 llvm::APFloat::opStatus 963 NumericLiteralParser::GetFloatValue(llvm::APFloat &Result) { 964 using llvm::APFloat; 965 966 unsigned n = std::min(SuffixBegin - ThisTokBegin, ThisTokEnd - ThisTokBegin); 967 968 llvm::SmallString<16> Buffer; 969 StringRef Str(ThisTokBegin, n); 970 if (Str.find('\'') != StringRef::npos) { 971 Buffer.reserve(n); 972 std::remove_copy_if(Str.begin(), Str.end(), std::back_inserter(Buffer), 973 &isDigitSeparator); 974 Str = Buffer; 975 } 976 977 return Result.convertFromString(Str, APFloat::rmNearestTiesToEven); 978 } 979 980 981 /// \verbatim 982 /// user-defined-character-literal: [C++11 lex.ext] 983 /// character-literal ud-suffix 984 /// ud-suffix: 985 /// identifier 986 /// character-literal: [C++11 lex.ccon] 987 /// ' c-char-sequence ' 988 /// u' c-char-sequence ' 989 /// U' c-char-sequence ' 990 /// L' c-char-sequence ' 991 /// c-char-sequence: 992 /// c-char 993 /// c-char-sequence c-char 994 /// c-char: 995 /// any member of the source character set except the single-quote ', 996 /// backslash \, or new-line character 997 /// escape-sequence 998 /// universal-character-name 999 /// escape-sequence: 1000 /// simple-escape-sequence 1001 /// octal-escape-sequence 1002 /// hexadecimal-escape-sequence 1003 /// simple-escape-sequence: 1004 /// one of \' \" \? \\ \a \b \f \n \r \t \v 1005 /// octal-escape-sequence: 1006 /// \ octal-digit 1007 /// \ octal-digit octal-digit 1008 /// \ octal-digit octal-digit octal-digit 1009 /// hexadecimal-escape-sequence: 1010 /// \x hexadecimal-digit 1011 /// hexadecimal-escape-sequence hexadecimal-digit 1012 /// universal-character-name: [C++11 lex.charset] 1013 /// \u hex-quad 1014 /// \U hex-quad hex-quad 1015 /// hex-quad: 1016 /// hex-digit hex-digit hex-digit hex-digit 1017 /// \endverbatim 1018 /// 1019 CharLiteralParser::CharLiteralParser(const char *begin, const char *end, 1020 SourceLocation Loc, Preprocessor &PP, 1021 tok::TokenKind kind) { 1022 // At this point we know that the character matches the regex "(L|u|U)?'.*'". 1023 HadError = false; 1024 1025 Kind = kind; 1026 1027 const char *TokBegin = begin; 1028 1029 // Skip over wide character determinant. 1030 if (Kind != tok::char_constant) { 1031 ++begin; 1032 } 1033 1034 // Skip over the entry quote. 1035 assert(begin[0] == '\'' && "Invalid token lexed"); 1036 ++begin; 1037 1038 // Remove an optional ud-suffix. 1039 if (end[-1] != '\'') { 1040 const char *UDSuffixEnd = end; 1041 do { 1042 --end; 1043 } while (end[-1] != '\''); 1044 // FIXME: Don't bother with this if !tok.hasUCN(). 1045 expandUCNs(UDSuffixBuf, StringRef(end, UDSuffixEnd - end)); 1046 UDSuffixOffset = end - TokBegin; 1047 } 1048 1049 // Trim the ending quote. 1050 assert(end != begin && "Invalid token lexed"); 1051 --end; 1052 1053 // FIXME: The "Value" is an uint64_t so we can handle char literals of 1054 // up to 64-bits. 1055 // FIXME: This extensively assumes that 'char' is 8-bits. 1056 assert(PP.getTargetInfo().getCharWidth() == 8 && 1057 "Assumes char is 8 bits"); 1058 assert(PP.getTargetInfo().getIntWidth() <= 64 && 1059 (PP.getTargetInfo().getIntWidth() & 7) == 0 && 1060 "Assumes sizeof(int) on target is <= 64 and a multiple of char"); 1061 assert(PP.getTargetInfo().getWCharWidth() <= 64 && 1062 "Assumes sizeof(wchar) on target is <= 64"); 1063 1064 SmallVector<uint32_t, 4> codepoint_buffer; 1065 codepoint_buffer.resize(end - begin); 1066 uint32_t *buffer_begin = &codepoint_buffer.front(); 1067 uint32_t *buffer_end = buffer_begin + codepoint_buffer.size(); 1068 1069 // Unicode escapes representing characters that cannot be correctly 1070 // represented in a single code unit are disallowed in character literals 1071 // by this implementation. 1072 uint32_t largest_character_for_kind; 1073 if (tok::wide_char_constant == Kind) { 1074 largest_character_for_kind = 1075 0xFFFFFFFFu >> (32-PP.getTargetInfo().getWCharWidth()); 1076 } else if (tok::utf16_char_constant == Kind) { 1077 largest_character_for_kind = 0xFFFF; 1078 } else if (tok::utf32_char_constant == Kind) { 1079 largest_character_for_kind = 0x10FFFF; 1080 } else { 1081 largest_character_for_kind = 0x7Fu; 1082 } 1083 1084 while (begin != end) { 1085 // Is this a span of non-escape characters? 1086 if (begin[0] != '\\') { 1087 char const *start = begin; 1088 do { 1089 ++begin; 1090 } while (begin != end && *begin != '\\'); 1091 1092 char const *tmp_in_start = start; 1093 uint32_t *tmp_out_start = buffer_begin; 1094 ConversionResult res = 1095 ConvertUTF8toUTF32(reinterpret_cast<UTF8 const **>(&start), 1096 reinterpret_cast<UTF8 const *>(begin), 1097 &buffer_begin, buffer_end, strictConversion); 1098 if (res != conversionOK) { 1099 // If we see bad encoding for unprefixed character literals, warn and 1100 // simply copy the byte values, for compatibility with gcc and 1101 // older versions of clang. 1102 bool NoErrorOnBadEncoding = isAscii(); 1103 unsigned Msg = diag::err_bad_character_encoding; 1104 if (NoErrorOnBadEncoding) 1105 Msg = diag::warn_bad_character_encoding; 1106 PP.Diag(Loc, Msg); 1107 if (NoErrorOnBadEncoding) { 1108 start = tmp_in_start; 1109 buffer_begin = tmp_out_start; 1110 for (; start != begin; ++start, ++buffer_begin) 1111 *buffer_begin = static_cast<uint8_t>(*start); 1112 } else { 1113 HadError = true; 1114 } 1115 } else { 1116 for (; tmp_out_start < buffer_begin; ++tmp_out_start) { 1117 if (*tmp_out_start > largest_character_for_kind) { 1118 HadError = true; 1119 PP.Diag(Loc, diag::err_character_too_large); 1120 } 1121 } 1122 } 1123 1124 continue; 1125 } 1126 // Is this a Universal Character Name escape? 1127 if (begin[1] == 'u' || begin[1] == 'U') { 1128 unsigned short UcnLen = 0; 1129 if (!ProcessUCNEscape(TokBegin, begin, end, *buffer_begin, UcnLen, 1130 FullSourceLoc(Loc, PP.getSourceManager()), 1131 &PP.getDiagnostics(), PP.getLangOpts(), true)) { 1132 HadError = true; 1133 } else if (*buffer_begin > largest_character_for_kind) { 1134 HadError = true; 1135 PP.Diag(Loc, diag::err_character_too_large); 1136 } 1137 1138 ++buffer_begin; 1139 continue; 1140 } 1141 unsigned CharWidth = getCharWidth(Kind, PP.getTargetInfo()); 1142 uint64_t result = 1143 ProcessCharEscape(TokBegin, begin, end, HadError, 1144 FullSourceLoc(Loc,PP.getSourceManager()), 1145 CharWidth, &PP.getDiagnostics(), PP.getLangOpts()); 1146 *buffer_begin++ = result; 1147 } 1148 1149 unsigned NumCharsSoFar = buffer_begin - &codepoint_buffer.front(); 1150 1151 if (NumCharsSoFar > 1) { 1152 if (isWide()) 1153 PP.Diag(Loc, diag::warn_extraneous_char_constant); 1154 else if (isAscii() && NumCharsSoFar == 4) 1155 PP.Diag(Loc, diag::ext_four_char_character_literal); 1156 else if (isAscii()) 1157 PP.Diag(Loc, diag::ext_multichar_character_literal); 1158 else 1159 PP.Diag(Loc, diag::err_multichar_utf_character_literal); 1160 IsMultiChar = true; 1161 } else { 1162 IsMultiChar = false; 1163 } 1164 1165 llvm::APInt LitVal(PP.getTargetInfo().getIntWidth(), 0); 1166 1167 // Narrow character literals act as though their value is concatenated 1168 // in this implementation, but warn on overflow. 1169 bool multi_char_too_long = false; 1170 if (isAscii() && isMultiChar()) { 1171 LitVal = 0; 1172 for (size_t i = 0; i < NumCharsSoFar; ++i) { 1173 // check for enough leading zeros to shift into 1174 multi_char_too_long |= (LitVal.countLeadingZeros() < 8); 1175 LitVal <<= 8; 1176 LitVal = LitVal + (codepoint_buffer[i] & 0xFF); 1177 } 1178 } else if (NumCharsSoFar > 0) { 1179 // otherwise just take the last character 1180 LitVal = buffer_begin[-1]; 1181 } 1182 1183 if (!HadError && multi_char_too_long) { 1184 PP.Diag(Loc, diag::warn_char_constant_too_large); 1185 } 1186 1187 // Transfer the value from APInt to uint64_t 1188 Value = LitVal.getZExtValue(); 1189 1190 // If this is a single narrow character, sign extend it (e.g. '\xFF' is "-1") 1191 // if 'char' is signed for this target (C99 6.4.4.4p10). Note that multiple 1192 // character constants are not sign extended in the this implementation: 1193 // '\xFF\xFF' = 65536 and '\x0\xFF' = 255, which matches GCC. 1194 if (isAscii() && NumCharsSoFar == 1 && (Value & 128) && 1195 PP.getLangOpts().CharIsSigned) 1196 Value = (signed char)Value; 1197 } 1198 1199 /// \verbatim 1200 /// string-literal: [C++0x lex.string] 1201 /// encoding-prefix " [s-char-sequence] " 1202 /// encoding-prefix R raw-string 1203 /// encoding-prefix: 1204 /// u8 1205 /// u 1206 /// U 1207 /// L 1208 /// s-char-sequence: 1209 /// s-char 1210 /// s-char-sequence s-char 1211 /// s-char: 1212 /// any member of the source character set except the double-quote ", 1213 /// backslash \, or new-line character 1214 /// escape-sequence 1215 /// universal-character-name 1216 /// raw-string: 1217 /// " d-char-sequence ( r-char-sequence ) d-char-sequence " 1218 /// r-char-sequence: 1219 /// r-char 1220 /// r-char-sequence r-char 1221 /// r-char: 1222 /// any member of the source character set, except a right parenthesis ) 1223 /// followed by the initial d-char-sequence (which may be empty) 1224 /// followed by a double quote ". 1225 /// d-char-sequence: 1226 /// d-char 1227 /// d-char-sequence d-char 1228 /// d-char: 1229 /// any member of the basic source character set except: 1230 /// space, the left parenthesis (, the right parenthesis ), 1231 /// the backslash \, and the control characters representing horizontal 1232 /// tab, vertical tab, form feed, and newline. 1233 /// escape-sequence: [C++0x lex.ccon] 1234 /// simple-escape-sequence 1235 /// octal-escape-sequence 1236 /// hexadecimal-escape-sequence 1237 /// simple-escape-sequence: 1238 /// one of \' \" \? \\ \a \b \f \n \r \t \v 1239 /// octal-escape-sequence: 1240 /// \ octal-digit 1241 /// \ octal-digit octal-digit 1242 /// \ octal-digit octal-digit octal-digit 1243 /// hexadecimal-escape-sequence: 1244 /// \x hexadecimal-digit 1245 /// hexadecimal-escape-sequence hexadecimal-digit 1246 /// universal-character-name: 1247 /// \u hex-quad 1248 /// \U hex-quad hex-quad 1249 /// hex-quad: 1250 /// hex-digit hex-digit hex-digit hex-digit 1251 /// \endverbatim 1252 /// 1253 StringLiteralParser:: 1254 StringLiteralParser(const Token *StringToks, unsigned NumStringToks, 1255 Preprocessor &PP, bool Complain) 1256 : SM(PP.getSourceManager()), Features(PP.getLangOpts()), 1257 Target(PP.getTargetInfo()), Diags(Complain ? &PP.getDiagnostics() : 0), 1258 MaxTokenLength(0), SizeBound(0), CharByteWidth(0), Kind(tok::unknown), 1259 ResultPtr(ResultBuf.data()), hadError(false), Pascal(false) { 1260 init(StringToks, NumStringToks); 1261 } 1262 1263 void StringLiteralParser::init(const Token *StringToks, unsigned NumStringToks){ 1264 // The literal token may have come from an invalid source location (e.g. due 1265 // to a PCH error), in which case the token length will be 0. 1266 if (NumStringToks == 0 || StringToks[0].getLength() < 2) 1267 return DiagnoseLexingError(SourceLocation()); 1268 1269 // Scan all of the string portions, remember the max individual token length, 1270 // computing a bound on the concatenated string length, and see whether any 1271 // piece is a wide-string. If any of the string portions is a wide-string 1272 // literal, the result is a wide-string literal [C99 6.4.5p4]. 1273 assert(NumStringToks && "expected at least one token"); 1274 MaxTokenLength = StringToks[0].getLength(); 1275 assert(StringToks[0].getLength() >= 2 && "literal token is invalid!"); 1276 SizeBound = StringToks[0].getLength()-2; // -2 for "". 1277 Kind = StringToks[0].getKind(); 1278 1279 hadError = false; 1280 1281 // Implement Translation Phase #6: concatenation of string literals 1282 /// (C99 5.1.1.2p1). The common case is only one string fragment. 1283 for (unsigned i = 1; i != NumStringToks; ++i) { 1284 if (StringToks[i].getLength() < 2) 1285 return DiagnoseLexingError(StringToks[i].getLocation()); 1286 1287 // The string could be shorter than this if it needs cleaning, but this is a 1288 // reasonable bound, which is all we need. 1289 assert(StringToks[i].getLength() >= 2 && "literal token is invalid!"); 1290 SizeBound += StringToks[i].getLength()-2; // -2 for "". 1291 1292 // Remember maximum string piece length. 1293 if (StringToks[i].getLength() > MaxTokenLength) 1294 MaxTokenLength = StringToks[i].getLength(); 1295 1296 // Remember if we see any wide or utf-8/16/32 strings. 1297 // Also check for illegal concatenations. 1298 if (StringToks[i].isNot(Kind) && StringToks[i].isNot(tok::string_literal)) { 1299 if (isAscii()) { 1300 Kind = StringToks[i].getKind(); 1301 } else { 1302 if (Diags) 1303 Diags->Report(StringToks[i].getLocation(), 1304 diag::err_unsupported_string_concat); 1305 hadError = true; 1306 } 1307 } 1308 } 1309 1310 // Include space for the null terminator. 1311 ++SizeBound; 1312 1313 // TODO: K&R warning: "traditional C rejects string constant concatenation" 1314 1315 // Get the width in bytes of char/wchar_t/char16_t/char32_t 1316 CharByteWidth = getCharWidth(Kind, Target); 1317 assert((CharByteWidth & 7) == 0 && "Assumes character size is byte multiple"); 1318 CharByteWidth /= 8; 1319 1320 // The output buffer size needs to be large enough to hold wide characters. 1321 // This is a worst-case assumption which basically corresponds to L"" "long". 1322 SizeBound *= CharByteWidth; 1323 1324 // Size the temporary buffer to hold the result string data. 1325 ResultBuf.resize(SizeBound); 1326 1327 // Likewise, but for each string piece. 1328 SmallString<512> TokenBuf; 1329 TokenBuf.resize(MaxTokenLength); 1330 1331 // Loop over all the strings, getting their spelling, and expanding them to 1332 // wide strings as appropriate. 1333 ResultPtr = &ResultBuf[0]; // Next byte to fill in. 1334 1335 Pascal = false; 1336 1337 SourceLocation UDSuffixTokLoc; 1338 1339 for (unsigned i = 0, e = NumStringToks; i != e; ++i) { 1340 const char *ThisTokBuf = &TokenBuf[0]; 1341 // Get the spelling of the token, which eliminates trigraphs, etc. We know 1342 // that ThisTokBuf points to a buffer that is big enough for the whole token 1343 // and 'spelled' tokens can only shrink. 1344 bool StringInvalid = false; 1345 unsigned ThisTokLen = 1346 Lexer::getSpelling(StringToks[i], ThisTokBuf, SM, Features, 1347 &StringInvalid); 1348 if (StringInvalid) 1349 return DiagnoseLexingError(StringToks[i].getLocation()); 1350 1351 const char *ThisTokBegin = ThisTokBuf; 1352 const char *ThisTokEnd = ThisTokBuf+ThisTokLen; 1353 1354 // Remove an optional ud-suffix. 1355 if (ThisTokEnd[-1] != '"') { 1356 const char *UDSuffixEnd = ThisTokEnd; 1357 do { 1358 --ThisTokEnd; 1359 } while (ThisTokEnd[-1] != '"'); 1360 1361 StringRef UDSuffix(ThisTokEnd, UDSuffixEnd - ThisTokEnd); 1362 1363 if (UDSuffixBuf.empty()) { 1364 if (StringToks[i].hasUCN()) 1365 expandUCNs(UDSuffixBuf, UDSuffix); 1366 else 1367 UDSuffixBuf.assign(UDSuffix); 1368 UDSuffixToken = i; 1369 UDSuffixOffset = ThisTokEnd - ThisTokBuf; 1370 UDSuffixTokLoc = StringToks[i].getLocation(); 1371 } else { 1372 SmallString<32> ExpandedUDSuffix; 1373 if (StringToks[i].hasUCN()) { 1374 expandUCNs(ExpandedUDSuffix, UDSuffix); 1375 UDSuffix = ExpandedUDSuffix; 1376 } 1377 1378 // C++11 [lex.ext]p8: At the end of phase 6, if a string literal is the 1379 // result of a concatenation involving at least one user-defined-string- 1380 // literal, all the participating user-defined-string-literals shall 1381 // have the same ud-suffix. 1382 if (UDSuffixBuf != UDSuffix) { 1383 if (Diags) { 1384 SourceLocation TokLoc = StringToks[i].getLocation(); 1385 Diags->Report(TokLoc, diag::err_string_concat_mixed_suffix) 1386 << UDSuffixBuf << UDSuffix 1387 << SourceRange(UDSuffixTokLoc, UDSuffixTokLoc) 1388 << SourceRange(TokLoc, TokLoc); 1389 } 1390 hadError = true; 1391 } 1392 } 1393 } 1394 1395 // Strip the end quote. 1396 --ThisTokEnd; 1397 1398 // TODO: Input character set mapping support. 1399 1400 // Skip marker for wide or unicode strings. 1401 if (ThisTokBuf[0] == 'L' || ThisTokBuf[0] == 'u' || ThisTokBuf[0] == 'U') { 1402 ++ThisTokBuf; 1403 // Skip 8 of u8 marker for utf8 strings. 1404 if (ThisTokBuf[0] == '8') 1405 ++ThisTokBuf; 1406 } 1407 1408 // Check for raw string 1409 if (ThisTokBuf[0] == 'R') { 1410 ThisTokBuf += 2; // skip R" 1411 1412 const char *Prefix = ThisTokBuf; 1413 while (ThisTokBuf[0] != '(') 1414 ++ThisTokBuf; 1415 ++ThisTokBuf; // skip '(' 1416 1417 // Remove same number of characters from the end 1418 ThisTokEnd -= ThisTokBuf - Prefix; 1419 assert(ThisTokEnd >= ThisTokBuf && "malformed raw string literal"); 1420 1421 // Copy the string over 1422 if (CopyStringFragment(StringToks[i], ThisTokBegin, 1423 StringRef(ThisTokBuf, ThisTokEnd - ThisTokBuf))) 1424 hadError = true; 1425 } else { 1426 if (ThisTokBuf[0] != '"') { 1427 // The file may have come from PCH and then changed after loading the 1428 // PCH; Fail gracefully. 1429 return DiagnoseLexingError(StringToks[i].getLocation()); 1430 } 1431 ++ThisTokBuf; // skip " 1432 1433 // Check if this is a pascal string 1434 if (Features.PascalStrings && ThisTokBuf + 1 != ThisTokEnd && 1435 ThisTokBuf[0] == '\\' && ThisTokBuf[1] == 'p') { 1436 1437 // If the \p sequence is found in the first token, we have a pascal string 1438 // Otherwise, if we already have a pascal string, ignore the first \p 1439 if (i == 0) { 1440 ++ThisTokBuf; 1441 Pascal = true; 1442 } else if (Pascal) 1443 ThisTokBuf += 2; 1444 } 1445 1446 while (ThisTokBuf != ThisTokEnd) { 1447 // Is this a span of non-escape characters? 1448 if (ThisTokBuf[0] != '\\') { 1449 const char *InStart = ThisTokBuf; 1450 do { 1451 ++ThisTokBuf; 1452 } while (ThisTokBuf != ThisTokEnd && ThisTokBuf[0] != '\\'); 1453 1454 // Copy the character span over. 1455 if (CopyStringFragment(StringToks[i], ThisTokBegin, 1456 StringRef(InStart, ThisTokBuf - InStart))) 1457 hadError = true; 1458 continue; 1459 } 1460 // Is this a Universal Character Name escape? 1461 if (ThisTokBuf[1] == 'u' || ThisTokBuf[1] == 'U') { 1462 EncodeUCNEscape(ThisTokBegin, ThisTokBuf, ThisTokEnd, 1463 ResultPtr, hadError, 1464 FullSourceLoc(StringToks[i].getLocation(), SM), 1465 CharByteWidth, Diags, Features); 1466 continue; 1467 } 1468 // Otherwise, this is a non-UCN escape character. Process it. 1469 unsigned ResultChar = 1470 ProcessCharEscape(ThisTokBegin, ThisTokBuf, ThisTokEnd, hadError, 1471 FullSourceLoc(StringToks[i].getLocation(), SM), 1472 CharByteWidth*8, Diags, Features); 1473 1474 if (CharByteWidth == 4) { 1475 // FIXME: Make the type of the result buffer correct instead of 1476 // using reinterpret_cast. 1477 UTF32 *ResultWidePtr = reinterpret_cast<UTF32*>(ResultPtr); 1478 *ResultWidePtr = ResultChar; 1479 ResultPtr += 4; 1480 } else if (CharByteWidth == 2) { 1481 // FIXME: Make the type of the result buffer correct instead of 1482 // using reinterpret_cast. 1483 UTF16 *ResultWidePtr = reinterpret_cast<UTF16*>(ResultPtr); 1484 *ResultWidePtr = ResultChar & 0xFFFF; 1485 ResultPtr += 2; 1486 } else { 1487 assert(CharByteWidth == 1 && "Unexpected char width"); 1488 *ResultPtr++ = ResultChar & 0xFF; 1489 } 1490 } 1491 } 1492 } 1493 1494 if (Pascal) { 1495 if (CharByteWidth == 4) { 1496 // FIXME: Make the type of the result buffer correct instead of 1497 // using reinterpret_cast. 1498 UTF32 *ResultWidePtr = reinterpret_cast<UTF32*>(ResultBuf.data()); 1499 ResultWidePtr[0] = GetNumStringChars() - 1; 1500 } else if (CharByteWidth == 2) { 1501 // FIXME: Make the type of the result buffer correct instead of 1502 // using reinterpret_cast. 1503 UTF16 *ResultWidePtr = reinterpret_cast<UTF16*>(ResultBuf.data()); 1504 ResultWidePtr[0] = GetNumStringChars() - 1; 1505 } else { 1506 assert(CharByteWidth == 1 && "Unexpected char width"); 1507 ResultBuf[0] = GetNumStringChars() - 1; 1508 } 1509 1510 // Verify that pascal strings aren't too large. 1511 if (GetStringLength() > 256) { 1512 if (Diags) 1513 Diags->Report(StringToks[0].getLocation(), 1514 diag::err_pascal_string_too_long) 1515 << SourceRange(StringToks[0].getLocation(), 1516 StringToks[NumStringToks-1].getLocation()); 1517 hadError = true; 1518 return; 1519 } 1520 } else if (Diags) { 1521 // Complain if this string literal has too many characters. 1522 unsigned MaxChars = Features.CPlusPlus? 65536 : Features.C99 ? 4095 : 509; 1523 1524 if (GetNumStringChars() > MaxChars) 1525 Diags->Report(StringToks[0].getLocation(), 1526 diag::ext_string_too_long) 1527 << GetNumStringChars() << MaxChars 1528 << (Features.CPlusPlus ? 2 : Features.C99 ? 1 : 0) 1529 << SourceRange(StringToks[0].getLocation(), 1530 StringToks[NumStringToks-1].getLocation()); 1531 } 1532 } 1533 1534 static const char *resyncUTF8(const char *Err, const char *End) { 1535 if (Err == End) 1536 return End; 1537 End = Err + std::min<unsigned>(getNumBytesForUTF8(*Err), End-Err); 1538 while (++Err != End && (*Err & 0xC0) == 0x80) 1539 ; 1540 return Err; 1541 } 1542 1543 /// \brief This function copies from Fragment, which is a sequence of bytes 1544 /// within Tok's contents (which begin at TokBegin) into ResultPtr. 1545 /// Performs widening for multi-byte characters. 1546 bool StringLiteralParser::CopyStringFragment(const Token &Tok, 1547 const char *TokBegin, 1548 StringRef Fragment) { 1549 const UTF8 *ErrorPtrTmp; 1550 if (ConvertUTF8toWide(CharByteWidth, Fragment, ResultPtr, ErrorPtrTmp)) 1551 return false; 1552 1553 // If we see bad encoding for unprefixed string literals, warn and 1554 // simply copy the byte values, for compatibility with gcc and older 1555 // versions of clang. 1556 bool NoErrorOnBadEncoding = isAscii(); 1557 if (NoErrorOnBadEncoding) { 1558 memcpy(ResultPtr, Fragment.data(), Fragment.size()); 1559 ResultPtr += Fragment.size(); 1560 } 1561 1562 if (Diags) { 1563 const char *ErrorPtr = reinterpret_cast<const char *>(ErrorPtrTmp); 1564 1565 FullSourceLoc SourceLoc(Tok.getLocation(), SM); 1566 const DiagnosticBuilder &Builder = 1567 Diag(Diags, Features, SourceLoc, TokBegin, 1568 ErrorPtr, resyncUTF8(ErrorPtr, Fragment.end()), 1569 NoErrorOnBadEncoding ? diag::warn_bad_string_encoding 1570 : diag::err_bad_string_encoding); 1571 1572 const char *NextStart = resyncUTF8(ErrorPtr, Fragment.end()); 1573 StringRef NextFragment(NextStart, Fragment.end()-NextStart); 1574 1575 // Decode into a dummy buffer. 1576 SmallString<512> Dummy; 1577 Dummy.reserve(Fragment.size() * CharByteWidth); 1578 char *Ptr = Dummy.data(); 1579 1580 while (!Builder.hasMaxRanges() && 1581 !ConvertUTF8toWide(CharByteWidth, NextFragment, Ptr, ErrorPtrTmp)) { 1582 const char *ErrorPtr = reinterpret_cast<const char *>(ErrorPtrTmp); 1583 NextStart = resyncUTF8(ErrorPtr, Fragment.end()); 1584 Builder << MakeCharSourceRange(Features, SourceLoc, TokBegin, 1585 ErrorPtr, NextStart); 1586 NextFragment = StringRef(NextStart, Fragment.end()-NextStart); 1587 } 1588 } 1589 return !NoErrorOnBadEncoding; 1590 } 1591 1592 void StringLiteralParser::DiagnoseLexingError(SourceLocation Loc) { 1593 hadError = true; 1594 if (Diags) 1595 Diags->Report(Loc, diag::err_lexing_string); 1596 } 1597 1598 /// getOffsetOfStringByte - This function returns the offset of the 1599 /// specified byte of the string data represented by Token. This handles 1600 /// advancing over escape sequences in the string. 1601 unsigned StringLiteralParser::getOffsetOfStringByte(const Token &Tok, 1602 unsigned ByteNo) const { 1603 // Get the spelling of the token. 1604 SmallString<32> SpellingBuffer; 1605 SpellingBuffer.resize(Tok.getLength()); 1606 1607 bool StringInvalid = false; 1608 const char *SpellingPtr = &SpellingBuffer[0]; 1609 unsigned TokLen = Lexer::getSpelling(Tok, SpellingPtr, SM, Features, 1610 &StringInvalid); 1611 if (StringInvalid) 1612 return 0; 1613 1614 const char *SpellingStart = SpellingPtr; 1615 const char *SpellingEnd = SpellingPtr+TokLen; 1616 1617 // Handle UTF-8 strings just like narrow strings. 1618 if (SpellingPtr[0] == 'u' && SpellingPtr[1] == '8') 1619 SpellingPtr += 2; 1620 1621 assert(SpellingPtr[0] != 'L' && SpellingPtr[0] != 'u' && 1622 SpellingPtr[0] != 'U' && "Doesn't handle wide or utf strings yet"); 1623 1624 // For raw string literals, this is easy. 1625 if (SpellingPtr[0] == 'R') { 1626 assert(SpellingPtr[1] == '"' && "Should be a raw string literal!"); 1627 // Skip 'R"'. 1628 SpellingPtr += 2; 1629 while (*SpellingPtr != '(') { 1630 ++SpellingPtr; 1631 assert(SpellingPtr < SpellingEnd && "Missing ( for raw string literal"); 1632 } 1633 // Skip '('. 1634 ++SpellingPtr; 1635 return SpellingPtr - SpellingStart + ByteNo; 1636 } 1637 1638 // Skip over the leading quote 1639 assert(SpellingPtr[0] == '"' && "Should be a string literal!"); 1640 ++SpellingPtr; 1641 1642 // Skip over bytes until we find the offset we're looking for. 1643 while (ByteNo) { 1644 assert(SpellingPtr < SpellingEnd && "Didn't find byte offset!"); 1645 1646 // Step over non-escapes simply. 1647 if (*SpellingPtr != '\\') { 1648 ++SpellingPtr; 1649 --ByteNo; 1650 continue; 1651 } 1652 1653 // Otherwise, this is an escape character. Advance over it. 1654 bool HadError = false; 1655 if (SpellingPtr[1] == 'u' || SpellingPtr[1] == 'U') { 1656 const char *EscapePtr = SpellingPtr; 1657 unsigned Len = MeasureUCNEscape(SpellingStart, SpellingPtr, SpellingEnd, 1658 1, Features, HadError); 1659 if (Len > ByteNo) { 1660 // ByteNo is somewhere within the escape sequence. 1661 SpellingPtr = EscapePtr; 1662 break; 1663 } 1664 ByteNo -= Len; 1665 } else { 1666 ProcessCharEscape(SpellingStart, SpellingPtr, SpellingEnd, HadError, 1667 FullSourceLoc(Tok.getLocation(), SM), 1668 CharByteWidth*8, Diags, Features); 1669 --ByteNo; 1670 } 1671 assert(!HadError && "This method isn't valid on erroneous strings"); 1672 } 1673 1674 return SpellingPtr-SpellingStart; 1675 } 1676