LiteralSupport.cpp revision df9ef1bc8c3780307ab2ed81bf5e31c23310b936
15f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===--- LiteralSupport.cpp - Code to parse and process literals ----------===//
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//
105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// This file implements the NumericLiteralParser, CharLiteralParser, and
115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer// StringLiteralParser interfaces.
125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//
135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer//===----------------------------------------------------------------------===//
145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Lex/LiteralSupport.h"
165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "clang/Lex/Preprocessor.h"
17500d3297d2a21edeac4d46cbcbe21bc2352c2a28Chris Lattner#include "clang/Lex/LexDiagnostic.h"
18136f93a2b660fc780fc2dba82a6f42ca4dc9164aChris Lattner#include "clang/Basic/TargetInfo.h"
19f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman#include "clang/Basic/ConvertUTF.h"
205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "llvm/ADT/StringExtras.h"
219fe8c74a93ac8e92512615c5f83e7a328b3b0544David Blaikie#include "llvm/Support/ErrorHandling.h"
225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerusing namespace clang;
235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// HexDigitValue - Return the value of the specified hex digit, or -1 if it's
255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// not valid.
265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic int HexDigitValue(char C) {
275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= '0' && C <= '9') return C-'0';
285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= 'a' && C <= 'f') return C-'a'+10;
295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= 'A' && C <= 'F') return C-'A'+10;
305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return -1;
315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
335cee1195584fa8672253139c86e922daeda69b9eDouglas Gregorstatic unsigned getCharWidth(tok::TokenKind kind, const TargetInfo &Target) {
345cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  switch (kind) {
35b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie  default: llvm_unreachable("Unknown token type!");
365cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::char_constant:
375cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::string_literal:
385cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::utf8_string_literal:
395cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    return Target.getCharWidth();
405cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::wide_char_constant:
415cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::wide_string_literal:
425cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    return Target.getWCharWidth();
435cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::utf16_char_constant:
445cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::utf16_string_literal:
455cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    return Target.getChar16Width();
465cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::utf32_char_constant:
475cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  case tok::utf32_string_literal:
485cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    return Target.getChar32Width();
495cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  }
505cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor}
515cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor
525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ProcessCharEscape - Parse a standard C escape sequence, which can occur in
535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// either a character or a string literal.
545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic unsigned ProcessCharEscape(const char *&ThisTokBuf,
555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                                  const char *ThisTokEnd, bool &HadError,
565cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor                                  FullSourceLoc Loc, unsigned CharWidth,
57d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie                                  DiagnosticsEngine *Diags) {
585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Skip the '\' char.
595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++ThisTokBuf;
605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // We know that this character can't be off the end of the buffer, because
625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // that would have been \", which would not have been the end of string.
635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  unsigned ResultChar = *ThisTokBuf++;
645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  switch (ResultChar) {
655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // These map to themselves.
665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '\\': case '\'': case '"': case '?': break;
671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // These have fixed mappings.
695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'a':
705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // TODO: K&R: the meaning of '\\a' is different in traditional C
715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 7;
725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'b':
745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 8;
755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'e':
7791f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner    if (Diags)
7891f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner      Diags->Report(Loc, diag::ext_nonstandard_escape) << "e";
795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 27;
805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
813c54801fbc67d8df2fed0711a2e2022db6b1bbcfEli Friedman  case 'E':
8291f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner    if (Diags)
8391f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner      Diags->Report(Loc, diag::ext_nonstandard_escape) << "E";
843c54801fbc67d8df2fed0711a2e2022db6b1bbcfEli Friedman    ResultChar = 27;
853c54801fbc67d8df2fed0711a2e2022db6b1bbcfEli Friedman    break;
865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'f':
875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 12;
885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'n':
905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 10;
915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'r':
935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 13;
945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 't':
965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 9;
975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'v':
995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 11;
1005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'x': { // Hex escape.
1025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 0;
1035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (ThisTokBuf == ThisTokEnd || !isxdigit(*ThisTokBuf)) {
10491f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner      if (Diags)
10591f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner        Diags->Report(Loc, diag::err_hex_escape_no_digits);
1065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      HadError = 1;
1075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      break;
1085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Hex escapes are a maximal series of hex digits.
1115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    bool Overflow = false;
1125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    for (; ThisTokBuf != ThisTokEnd; ++ThisTokBuf) {
1135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      int CharVal = HexDigitValue(ThisTokBuf[0]);
1145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (CharVal == -1) break;
115c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner      // About to shift out a digit?
116c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner      Overflow |= (ResultChar & 0xF0000000) ? true : false;
1175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar <<= 4;
1185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar |= CharVal;
1195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // See if any bits will be truncated when evaluated as a character.
1225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) {
1235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      Overflow = true;
1245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar &= ~0U >> (32-CharWidth);
1255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Check for overflow.
12891f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner    if (Overflow && Diags)   // Too many digits to fit in
12991f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner      Diags->Report(Loc, diag::warn_hex_escape_too_large);
1305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '0': case '1': case '2': case '3':
1335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '4': case '5': case '6': case '7': {
1345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Octal escapes.
1355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    --ThisTokBuf;
1365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 0;
1375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Octal escapes are a series of octal digits with maximum length 3.
1395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // "\0123" is a two digit sequence equal to "\012" "3".
1405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    unsigned NumDigits = 0;
1415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    do {
1425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar <<= 3;
1435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar |= *ThisTokBuf++ - '0';
1445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ++NumDigits;
1455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    } while (ThisTokBuf != ThisTokEnd && NumDigits < 3 &&
1465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer             ThisTokBuf[0] >= '0' && ThisTokBuf[0] <= '7');
1471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Check for overflow.  Reject '\777', but not L'\777'.
1495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) {
15091f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner      if (Diags)
15191f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner        Diags->Report(Loc, diag::warn_octal_escape_too_large);
1525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar &= ~0U >> (32-CharWidth);
1535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Otherwise, these are not valid escapes.
1585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '(': case '{': case '[': case '%':
1595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // GCC accepts these as extensions.  We warn about them as such though.
16091f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner    if (Diags)
16191f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner      Diags->Report(Loc, diag::ext_nonstandard_escape)
162b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor        << std::string()+(char)ResultChar;
163f01fdff97b245caac98100d232c760b4d0531411Eli Friedman    break;
1645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  default:
16591f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner    if (Diags == 0)
166b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      break;
167b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor
16823ef69d197ba3b5e9602f7161fee50990059502aTed Kremenek    if (isgraph(ResultChar))
16991f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner      Diags->Report(Loc, diag::ext_unknown_escape)
17091f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner        << std::string()+(char)ResultChar;
171ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    else
17291f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner      Diags->Report(Loc, diag::ext_unknown_escape)
17391f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner        << "x"+llvm::utohexstr(ResultChar);
1745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return ResultChar;
1785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
1795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1800e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// ProcessUCNEscape - Read the Universal Character Name, check constraints and
18159705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber/// return the UTF32.
18226b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smithstatic bool ProcessUCNEscape(const char *ThisTokBegin, const char *&ThisTokBuf,
18326b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith                             const char *ThisTokEnd,
18459705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber                             uint32_t &UcnVal, unsigned short &UcnLen,
185d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie                             FullSourceLoc Loc, DiagnosticsEngine *Diags,
186be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell                             const LangOptions &Features,
187be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell                             bool in_char_string_literal = false) {
1886c66f07854c1334a1ce9eae1428d61d54182a6e1Chris Lattner  if (!Features.CPlusPlus && !Features.C99 && Diags)
189872a45e91778eb0b706ff57272fe547d4512eb19Chris Lattner    Diags->Report(Loc, diag::warn_ucn_not_valid_in_c89);
1901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19126b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith  const char *UcnBegin = ThisTokBuf;
1921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1930e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Skip the '\u' char's.
1940e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ThisTokBuf += 2;
1950e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
1960e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if (ThisTokBuf == ThisTokEnd || !isxdigit(*ThisTokBuf)) {
1976c66f07854c1334a1ce9eae1428d61d54182a6e1Chris Lattner    if (Diags)
198872a45e91778eb0b706ff57272fe547d4512eb19Chris Lattner      Diags->Report(Loc, diag::err_ucn_escape_no_digits);
19959705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber    return false;
2000e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
20159705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber  UcnLen = (ThisTokBuf[-1] == 'u' ? 4 : 8);
20256bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian  unsigned short UcnLenSave = UcnLen;
20359705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber  for (; ThisTokBuf != ThisTokEnd && UcnLenSave; ++ThisTokBuf, UcnLenSave--) {
2040e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    int CharVal = HexDigitValue(ThisTokBuf[0]);
2050e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    if (CharVal == -1) break;
2060e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    UcnVal <<= 4;
2070e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    UcnVal |= CharVal;
2080e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
2090e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // If we didn't consume the proper number of digits, there is a problem.
21059705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber  if (UcnLenSave) {
211872a45e91778eb0b706ff57272fe547d4512eb19Chris Lattner    if (Diags) {
2127ef5c27eb6e8ebe58b52013246c06753c3613263Chris Lattner      SourceLocation L =
21326b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith        Lexer::AdvanceToTokenCharacter(Loc, UcnBegin - ThisTokBegin,
2147ef5c27eb6e8ebe58b52013246c06753c3613263Chris Lattner                                       Loc.getManager(), Features);
21526b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      Diags->Report(L, diag::err_ucn_escape_incomplete);
216872a45e91778eb0b706ff57272fe547d4512eb19Chris Lattner    }
21759705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber    return false;
2180e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
21926b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith
220be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // Check UCN constraints (C99 6.4.3p2) [C++11 lex.charset p2]
22126b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith  if ((0xD800 <= UcnVal && UcnVal <= 0xDFFF) || // surrogate codepoints
22226b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      UcnVal > 0x10FFFF) {                      // maximum legal UTF32 value
22326b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith    if (Diags)
22426b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      Diags->Report(Loc, diag::err_ucn_escape_invalid);
22526b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith    return false;
22626b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith  }
227be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
228be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // C++11 allows UCNs that refer to control characters and basic source
229be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // characters inside character and string literals
23026b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith  if (UcnVal < 0xa0 &&
23126b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      (UcnVal != 0x24 && UcnVal != 0x40 && UcnVal != 0x60)) {  // $, @, `
23226b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith    bool IsError = (!Features.CPlusPlus0x || !in_char_string_literal);
23326b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith    if (Diags) {
23426b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      SourceLocation UcnBeginLoc =
23526b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith        Lexer::AdvanceToTokenCharacter(Loc, UcnBegin - ThisTokBegin,
23626b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith                                       Loc.getManager(), Features);
23726b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      char BasicSCSChar = UcnVal;
23826b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      if (UcnVal >= 0x20 && UcnVal < 0x7f)
23926b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith        Diags->Report(UcnBeginLoc, IsError ? diag::err_ucn_escape_basic_scs :
24026b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith                      diag::warn_cxx98_compat_literal_ucn_escape_basic_scs)
24126b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith          << StringRef(&BasicSCSChar, 1);
24226b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      else
24326b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith        Diags->Report(UcnBeginLoc, IsError ? diag::err_ucn_control_character :
24426b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith                      diag::warn_cxx98_compat_literal_ucn_control_character);
245be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    }
24626b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith    if (IsError)
24726b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      return false;
248be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  }
249be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
25059705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber  return true;
25159705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber}
25259705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber
253df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith/// MeasureUCNEscape - Determine the number of bytes within the resulting string
254df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith/// which this UCN will occupy.
255df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smithstatic int MeasureUCNEscape(const char *ThisTokBegin, const char *&ThisTokBuf,
256df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith                            const char *ThisTokEnd, unsigned CharByteWidth,
257df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith                            const LangOptions &Features, bool &HadError) {
258df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  // UTF-32: 4 bytes per escape.
259df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  if (CharByteWidth == 4)
260df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    return 4;
261df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith
262df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  uint32_t UcnVal = 0;
263df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  unsigned short UcnLen = 0;
264df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  FullSourceLoc Loc;
265df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith
266df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  if (!ProcessUCNEscape(ThisTokBegin, ThisTokBuf, ThisTokEnd, UcnVal,
267df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith                        UcnLen, Loc, 0, Features, true)) {
268df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    HadError = true;
269df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    return 0;
270df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  }
271df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith
272df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  // UTF-16: 2 bytes for BMP, 4 bytes otherwise.
273df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  if (CharByteWidth == 2)
274df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    return UcnVal <= 0xFFFF ? 2 : 4;
275df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith
276df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  // UTF-8.
277df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  if (UcnVal < 0x80)
278df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    return 1;
279df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  if (UcnVal < 0x800)
280df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    return 2;
281df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  if (UcnVal < 0x10000)
282df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    return 3;
283df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  return 4;
284df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith}
285df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith
28659705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber/// EncodeUCNEscape - Read the Universal Character Name, check constraints and
28759705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber/// convert the UTF32 to UTF8 or UTF16. This is a subroutine of
28859705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber/// StringLiteralParser. When we decide to implement UCN's for identifiers,
28959705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber/// we will likely rework our support for UCN's.
29026b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smithstatic void EncodeUCNEscape(const char *ThisTokBegin, const char *&ThisTokBuf,
29126b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith                            const char *ThisTokEnd,
292a95880d6513c617bb96634bcc1f16c6bdb80dedcChris Lattner                            char *&ResultBuf, bool &HadError,
2935cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor                            FullSourceLoc Loc, unsigned CharByteWidth,
294d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie                            DiagnosticsEngine *Diags,
295d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie                            const LangOptions &Features) {
29659705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber  typedef uint32_t UTF32;
29759705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber  UTF32 UcnVal = 0;
29859705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber  unsigned short UcnLen = 0;
29926b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith  if (!ProcessUCNEscape(ThisTokBegin, ThisTokBuf, ThisTokEnd, UcnVal, UcnLen,
30026b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith                        Loc, Diags, Features, true)) {
301df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    HadError = true;
3020e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    return;
3030e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
30459705aee3fe01aa6fb6962dd11350161b47983d9Nico Weber
3055cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  assert((CharByteWidth == 1 || CharByteWidth == 2 || CharByteWidth) &&
3065cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor         "only character widths of 1, 2, or 4 bytes supported");
307a0f15b0848405ae16d63bd5d78c862a6526b338aNico Weber
3085cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  (void)UcnLen;
3095cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  assert((UcnLen== 4 || UcnLen== 8) && "only ucn length of 4 or 8 supported");
3105cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor
3115cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  if (CharByteWidth == 4) {
312caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    // FIXME: Make the type of the result buffer correct instead of
313caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    // using reinterpret_cast.
314caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    UTF32 *ResultPtr = reinterpret_cast<UTF32*>(ResultBuf);
315caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    *ResultPtr = UcnVal;
316caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    ResultBuf += 4;
3175cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    return;
3185cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  }
319a0f15b0848405ae16d63bd5d78c862a6526b338aNico Weber
3205cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  if (CharByteWidth == 2) {
321caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    // FIXME: Make the type of the result buffer correct instead of
322caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    // using reinterpret_cast.
323caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    UTF16 *ResultPtr = reinterpret_cast<UTF16*>(ResultBuf);
324caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman
325a0f15b0848405ae16d63bd5d78c862a6526b338aNico Weber    if (UcnVal < (UTF32)0xFFFF) {
326caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman      *ResultPtr = UcnVal;
327caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman      ResultBuf += 2;
328a0f15b0848405ae16d63bd5d78c862a6526b338aNico Weber      return;
329a0f15b0848405ae16d63bd5d78c862a6526b338aNico Weber    }
330a0f15b0848405ae16d63bd5d78c862a6526b338aNico Weber
331caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    // Convert to UTF16.
332a0f15b0848405ae16d63bd5d78c862a6526b338aNico Weber    UcnVal -= 0x10000;
333caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    *ResultPtr     = 0xD800 + (UcnVal >> 10);
334caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    *(ResultPtr+1) = 0xDC00 + (UcnVal & 0x3FF);
335caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman    ResultBuf += 4;
33656bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    return;
33756bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian  }
3385cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor
3395cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  assert(CharByteWidth == 1 && "UTF-8 encoding is only for 1 byte characters");
3405cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor
3410e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Now that we've parsed/checked the UCN, we convert from UTF32->UTF8.
3420e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // The conversion below was inspired by:
3430e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  //   http://www.unicode.org/Public/PROGRAMS/CVTUTF/ConvertUTF.c
3441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // First, we determine how many bytes the result will require.
3454e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff  typedef uint8_t UTF8;
3460e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
3470e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  unsigned short bytesToWrite = 0;
3480e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if (UcnVal < (UTF32)0x80)
3490e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 1;
3500e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else if (UcnVal < (UTF32)0x800)
3510e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 2;
3520e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else if (UcnVal < (UTF32)0x10000)
3530e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 3;
3540e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else
3550e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 4;
3561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3570e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  const unsigned byteMask = 0xBF;
3580e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  const unsigned byteMark = 0x80;
3591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3600e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Once the bits are split out into bytes of UTF8, this is a mask OR-ed
3618a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff  // into the first byte, depending on how many bytes follow.
3621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static const UTF8 firstByteMark[5] = {
3638a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff    0x00, 0x00, 0xC0, 0xE0, 0xF0
3640e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  };
3650e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Finally, we write the bytes into ResultBuf.
3660e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ResultBuf += bytesToWrite;
3670e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  switch (bytesToWrite) { // note: everything falls through.
3680e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 4: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
3690e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 3: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
3700e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 2: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
3710e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 1: *--ResultBuf = (UTF8) (UcnVal | firstByteMark[bytesToWrite]);
3720e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
3730e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Update the buffer.
3740e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ResultBuf += bytesToWrite;
3750e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff}
3765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
3775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
3785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       integer-constant: [C99 6.4.4.1]
3795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         decimal-constant integer-suffix
3805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-constant integer-suffix
3815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-constant integer-suffix
38249d517467c3dcd2c67e8a6c740ba5160e37625f7Richard Smith///       user-defined-integer-literal: [C++11 lex.ext]
383b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith///         decimal-literal ud-suffix
384b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith///         octal-literal ud-suffix
385b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith///         hexadecimal-literal ud-suffix
3861eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///       decimal-constant:
3875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         nonzero-digit
3885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         decimal-constant digit
3891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///       octal-constant:
3905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0
3915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-constant octal-digit
3921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///       hexadecimal-constant:
3935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-prefix hexadecimal-digit
3945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-constant hexadecimal-digit
3955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hexadecimal-prefix: one of
3965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0x 0X
3975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       integer-suffix:
3985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         unsigned-suffix [long-suffix]
3995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         unsigned-suffix [long-long-suffix]
4005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         long-suffix [unsigned-suffix]
4015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         long-long-suffix [unsigned-sufix]
4025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       nonzero-digit:
4035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         1 2 3 4 5 6 7 8 9
4045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       octal-digit:
4055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0 1 2 3 4 5 6 7
4065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hexadecimal-digit:
4075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0 1 2 3 4 5 6 7 8 9
4085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         a b c d e f
4095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         A B C D E F
4105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       unsigned-suffix: one of
4115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         u U
4125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       long-suffix: one of
4135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         l L
4141eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///       long-long-suffix: one of
4155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         ll LL
4165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
4175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       floating-constant: [C99 6.4.4.2]
4185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         TODO: add rules...
4195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
4205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerNumericLiteralParser::
4215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerNumericLiteralParser(const char *begin, const char *end,
4225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                     SourceLocation TokLoc, Preprocessor &pp)
4235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  : PP(pp), ThisTokBegin(begin), ThisTokEnd(end) {
4241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
425c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // This routine assumes that the range begin/end matches the regex for integer
426c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // and FP constants (specifically, the 'pp-number' regex), and assumes that
427c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // the byte at "*end" is both valid and not part of the regex.  Because of
428c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // this, it doesn't have to check for 'overscan' in various places.
429c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  assert(!isalnum(*end) && *end != '.' && *end != '_' &&
430c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner         "Lexer didn't maximally munch?");
4311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  s = DigitsBegin = begin;
4335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  saw_exponent = false;
4345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  saw_period = false;
435b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith  saw_ud_suffix = false;
4365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isLong = false;
4375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isUnsigned = false;
4385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isLongLong = false;
4396e400c286b485e28d04a742ea87860ddfefa672eChris Lattner  isFloat = false;
440506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  isImaginary = false;
441b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump  isMicrosoftInteger = false;
4425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  hadError = false;
4431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (*s == '0') { // parse radix
445368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    ParseNumberStartingWithZero(TokLoc);
446368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (hadError)
447368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      return;
4485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  } else { // the first digit is non-zero
4495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    radix = 10;
4505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    s = SkipDigits(s);
4515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (s == ThisTokEnd) {
4525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      // Done.
453016765e3453db2e302efe53905e99cdb25501234Christopher Lamb    } else if (isxdigit(*s) && !(*s == 'e' || *s == 'E')) {
454ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
4555f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner              diag::err_invalid_decimal_digit) << StringRef(s, 1);
456ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
4575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      return;
4585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    } else if (*s == '.') {
4595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s++;
4605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      saw_period = true;
4615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s = SkipDigits(s);
4621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    }
4634411f46050216a139ab2fc7ff145ec384d11ec7fChris Lattner    if ((*s == 'e' || *s == 'E')) { // exponent
46470f66ab053f36ab3df7a778d09bcb2b4b0fec1f8Chris Lattner      const char *Exponent = s;
4655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s++;
4665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      saw_exponent = true;
4675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (*s == '+' || *s == '-')  s++; // sign
4685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      const char *first_non_digit = SkipDigits(s);
4690b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner      if (first_non_digit != s) {
4705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        s = first_non_digit;
4710b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner      } else {
472ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-begin),
473ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                diag::err_exponent_has_no_digits);
474ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        hadError = true;
4750b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner        return;
4765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
4775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
4785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
4795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
4805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SuffixBegin = s;
4811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
482506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // Parse the suffix.  At this point we can classify whether we have an FP or
483506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // integer constant.
484506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  bool isFPConstant = isFloatingLiteral();
4851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
486506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // Loop over all of the characters of the suffix.  If we see something bad,
487506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // we break out of the loop.
488506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  for (; s != ThisTokEnd; ++s) {
489506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    switch (*s) {
490506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'f':      // FP Suffix for "float"
491506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'F':
492506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (!isFPConstant) break;  // Error for integer constant.
4936e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      if (isFloat || isLong) break; // FF, LF invalid.
4946e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      isFloat = true;
495506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
496506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'u':
497506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'U':
498506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isFPConstant) break;  // Error for floating constant.
499506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isUnsigned) break;    // Cannot be repeated.
500506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      isUnsigned = true;
501506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
502506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'l':
503506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'L':
504506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isLong || isLongLong) break;  // Cannot be repeated.
5056e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      if (isFloat) break;               // LF invalid.
5061eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
507506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      // Check for long long.  The L's need to be adjacent and the same case.
508506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (s+1 != ThisTokEnd && s[1] == s[0]) {
509506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        if (isFPConstant) break;        // long long invalid for floats.
510506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        isLongLong = true;
511506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        ++s;  // Eat both of them.
512506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      } else {
5135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        isLong = true;
5145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
515506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
516506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'i':
517c637415a96c16abc7e28ef83c6c105716f7e8936Chris Lattner    case 'I':
5184e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie      if (PP.getLangOpts().MicrosoftExt) {
519a8be02b655b76e4dbe776b0c62bc3c450dc6feabFariborz Jahanian        if (isFPConstant || isLong || isLongLong) break;
5206e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes
5210c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        // Allow i8, i16, i32, i64, and i128.
522b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump        if (s + 1 != ThisTokEnd) {
523b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump          switch (s[1]) {
524b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            case '8':
525b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              s += 2; // i8 suffix
526b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              isMicrosoftInteger = true;
5276e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              break;
528b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            case '1':
5296e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s + 2 == ThisTokEnd) break;
530d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet              if (s[2] == '6') {
531d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                s += 3; // i16 suffix
532d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                isMicrosoftInteger = true;
533d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet              }
5346e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              else if (s[2] == '2') {
5356e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes                if (s + 3 == ThisTokEnd) break;
536d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                if (s[3] == '8') {
537d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                  s += 4; // i128 suffix
538d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                  isMicrosoftInteger = true;
539d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                }
540b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              }
5416e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              break;
542b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            case '3':
5436e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s + 2 == ThisTokEnd) break;
544d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet              if (s[2] == '2') {
545d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                s += 3; // i32 suffix
546d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                isLong = true;
547d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                isMicrosoftInteger = true;
548d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet              }
5496e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              break;
550b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            case '6':
5516e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s + 2 == ThisTokEnd) break;
552d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet              if (s[2] == '4') {
553d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                s += 3; // i64 suffix
554d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                isLongLong = true;
555d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet                isMicrosoftInteger = true;
556d062b604548be6e2f85f6f63a461702e5ea14115Francois Pichet              }
5576e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              break;
558b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            default:
559b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              break;
560b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump          }
561b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump          break;
5620c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        }
5630c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff      }
5640c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff      // fall through.
565506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'j':
566506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'J':
567506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isImaginary) break;   // Cannot be repeated.
568506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
569506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner              diag::ext_imaginary_constant);
570506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      isImaginary = true;
571506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
5725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
573b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith    // If we reached here, there was an error or a ud-suffix.
574506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    break;
575506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  }
5761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
577506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  if (s != ThisTokEnd) {
5784e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie    if (PP.getLangOpts().CPlusPlus0x && s == SuffixBegin && *s == '_') {
579b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith      // We have a ud-suffix! By C++11 [lex.ext]p10, ud-suffixes not starting
580b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith      // with an '_' are ill-formed.
581b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith      saw_ud_suffix = true;
582b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith      return;
583b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith    }
584b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith
585b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith    // Report an error if there are any.
586b453ad3214d00acc51c9aa702c76c58354d84b84Richard Smith    PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, SuffixBegin-begin),
587ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner            isFPConstant ? diag::err_invalid_suffix_float_constant :
588ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                           diag::err_invalid_suffix_integer_constant)
5895f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      << StringRef(SuffixBegin, ThisTokEnd-SuffixBegin);
590ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    hadError = true;
591506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    return;
5925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
5935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
5945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
595368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// ParseNumberStartingWithZero - This method is called when the first character
596368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// of the number is found to be a zero.  This means it is either an octal
597368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// number (like '04') or a hex number ('0x123a') a binary number ('0b1010') or
5981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// a floating point number (01239.123e4).  Eat the prefix, determining the
599368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// radix etc.
600368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattnervoid NumericLiteralParser::ParseNumberStartingWithZero(SourceLocation TokLoc) {
601368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  assert(s[0] == '0' && "Invalid method call");
602368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  s++;
6031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
604368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // Handle a hex number like 0x1234.
605368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if ((*s == 'x' || *s == 'X') && (isxdigit(s[1]) || s[1] == '.')) {
606368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
607368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 16;
608368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    DigitsBegin = s;
609368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s = SkipHexDigits(s);
61066b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman    bool noSignificand = (s == DigitsBegin);
611368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (s == ThisTokEnd) {
612368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      // Done.
613368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (*s == '.') {
614368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s++;
615368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      saw_period = true;
61666b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman      const char *floatDigitsBegin = s;
617368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s = SkipHexDigits(s);
61866b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman      noSignificand &= (floatDigitsBegin == s);
61966b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman    }
62066b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman
62166b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman    if (noSignificand) {
62266b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin), \
62366b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman        diag::err_hexconstant_requires_digits);
62466b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman      hadError = true;
62566b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman      return;
626368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
62766b0ebac276353f3ff7d41eaba3e6d24d48663b7Aaron Ballman
628368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    // A binary exponent can appear with or with a '.'. If dotted, the
6291eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // binary exponent is required.
6301155c42e7b1b4e401bb0a331a6d715d637958c75Douglas Gregor    if (*s == 'p' || *s == 'P') {
631368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      const char *Exponent = s;
632368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s++;
633368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      saw_exponent = true;
634368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      if (*s == '+' || *s == '-')  s++; // sign
635368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      const char *first_non_digit = SkipDigits(s);
6366ea623823f8532670480425b573f35115404b4a0Chris Lattner      if (first_non_digit == s) {
637ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin),
638ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                diag::err_exponent_has_no_digits);
639ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        hadError = true;
6406ea623823f8532670480425b573f35115404b4a0Chris Lattner        return;
641368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      }
6426ea623823f8532670480425b573f35115404b4a0Chris Lattner      s = first_non_digit;
6431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6444e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie      if (!PP.getLangOpts().HexFloats)
645ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(TokLoc, diag::ext_hexconstant_invalid);
646368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (saw_period) {
647ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
648ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner              diag::err_hexconstant_requires_exponent);
649ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
650368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
651368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
652368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
6531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
654368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // Handle simple binary numbers 0b01010
655368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == 'b' || *s == 'B') {
656368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    // 0b101010 is a GCC extension.
657413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    PP.Diag(TokLoc, diag::ext_binary_literal);
658368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    ++s;
659368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 2;
660368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    DigitsBegin = s;
661368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s = SkipBinaryDigits(s);
662368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (s == ThisTokEnd) {
663368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      // Done.
664368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (isxdigit(*s)) {
665ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
6665f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner              diag::err_invalid_binary_digit) << StringRef(s, 1);
667ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
668368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
669413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    // Other suffixes will be diagnosed by the caller.
670368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
671368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
6721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
673368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // For now, the radix is set to 8. If we discover that we have a
674368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // floating point constant, the radix will change to 10. Octal floating
6751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // point constants are not permitted (only decimal and hexadecimal).
676368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  radix = 8;
677368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  DigitsBegin = s;
678368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  s = SkipOctalDigits(s);
679368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (s == ThisTokEnd)
680368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return; // Done, simple octal number like 01234
6811eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
682413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // If we have some other non-octal digit that *is* a decimal digit, see if
683413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // this is part of a floating point number like 094.123 or 09e1.
684413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  if (isdigit(*s)) {
685413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    const char *EndDecimal = SkipDigits(s);
686413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    if (EndDecimal[0] == '.' || EndDecimal[0] == 'e' || EndDecimal[0] == 'E') {
687413d355e38755a71f106dbc0ac900ca989070916Chris Lattner      s = EndDecimal;
688413d355e38755a71f106dbc0ac900ca989070916Chris Lattner      radix = 10;
689413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    }
690413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  }
6911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
692413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // If we have a hex digit other than 'e' (which denotes a FP exponent) then
693413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // the code is using an incorrect base.
694368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (isxdigit(*s) && *s != 'e' && *s != 'E') {
695ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
6965f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner            diag::err_invalid_octal_digit) << StringRef(s, 1);
697ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    hadError = true;
698368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
699368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
7001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
701368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == '.') {
702368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
703368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 10;
704368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    saw_period = true;
705413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    s = SkipDigits(s); // Skip suffix.
706368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
707368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == 'e' || *s == 'E') { // exponent
708368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    const char *Exponent = s;
709368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
710368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 10;
711368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    saw_exponent = true;
712368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (*s == '+' || *s == '-')  s++; // sign
713368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    const char *first_non_digit = SkipDigits(s);
714368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (first_non_digit != s) {
715368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s = first_non_digit;
716368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else {
7171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin),
718ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner              diag::err_exponent_has_no_digits);
719ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
720368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      return;
721368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
722368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
723368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner}
724368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
725368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
7265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// GetIntegerValue - Convert this numeric literal value to an APInt that
7275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// matches Val's input width.  If there is an overflow, set Val to the low bits
7285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// of the result and return true.  Otherwise, return false.
7295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerbool NumericLiteralParser::GetIntegerValue(llvm::APInt &Val) {
730a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // Fast path: Compute a conservative bound on the maximum number of
731a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // bits per digit in this radix. If we can't possibly overflow a
732a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // uint64 based on that bound then do the simple conversion to
733a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // integer. This avoids the expensive overflow checking below, and
734a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // handles the common cases that matter (small decimal integers and
735a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // hex/octal values which don't overflow).
736a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  unsigned MaxBitsPerDigit = 1;
7371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  while ((1U << MaxBitsPerDigit) < radix)
738a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    MaxBitsPerDigit += 1;
739a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  if ((SuffixBegin - DigitsBegin) * MaxBitsPerDigit <= 64) {
740a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    uint64_t N = 0;
741a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    for (s = DigitsBegin; s != SuffixBegin; ++s)
742a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar      N = N*radix + HexDigitValue(*s);
743a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar
744a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    // This will truncate the value to Val's input width. Simply check
745a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    // for overflow by comparing.
746a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    Val = N;
747a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    return Val.getZExtValue() != N;
748a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  }
749a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar
7505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  Val = 0;
7515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  s = DigitsBegin;
7525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt RadixVal(Val.getBitWidth(), radix);
7545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt CharVal(Val.getBitWidth(), 0);
7555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt OldVal = Val;
7561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool OverflowOccurred = false;
7585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  while (s < SuffixBegin) {
7595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    unsigned C = HexDigitValue(*s++);
7601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // If this letter is out of bound for this radix, reject it.
7625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    assert(C < radix && "NumericLiteralParser ctor should have rejected this");
7631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    CharVal = C;
7651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Add the digit to the value in the appropriate radix.  If adding in digits
7675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // made the value smaller, then this overflowed.
7685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OldVal = Val;
7695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Multiply by radix, did overflow occur on the multiply?
7715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Val *= RadixVal;
7725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OverflowOccurred |= Val.udiv(RadixVal) != OldVal;
7735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Add value, did overflow occur on the value?
775d70cb645702bdbb42aee58403306a7c47e0d901cDaniel Dunbar    //   (a + b) ult b  <=> overflow
7765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Val += CharVal;
7775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OverflowOccurred |= Val.ult(CharVal);
7785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
7795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return OverflowOccurred;
7805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
7815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
78294c939dc1d4958b62ea5a89294dd8b2905f3191fJohn McCallllvm::APFloat::opStatus
78394c939dc1d4958b62ea5a89294dd8b2905f3191fJohn McCallNumericLiteralParser::GetFloatValue(llvm::APFloat &Result) {
784427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek  using llvm::APFloat;
7851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
786e9f195f15ffe96d0a220c872ab12d0630a633c44Erick Tryzelaar  unsigned n = std::min(SuffixBegin - ThisTokBegin, ThisTokEnd - ThisTokBegin);
78794c939dc1d4958b62ea5a89294dd8b2905f3191fJohn McCall  return Result.convertFromString(StringRef(ThisTokBegin, n),
78894c939dc1d4958b62ea5a89294dd8b2905f3191fJohn McCall                                  APFloat::rmNearestTiesToEven);
7895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
7905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7925cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith///       user-defined-character-literal: [C++11 lex.ext]
7935cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith///         character-literal ud-suffix
7945cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith///       ud-suffix:
7955cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith///         identifier
7965cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith///       character-literal: [C++11 lex.ccon]
7972fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         ' c-char-sequence '
7982fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         u' c-char-sequence '
7992fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         U' c-char-sequence '
8002fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         L' c-char-sequence '
8012fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       c-char-sequence:
8022fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         c-char
8032fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         c-char-sequence c-char
8042fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       c-char:
8052fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         any member of the source character set except the single-quote ',
8062fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///           backslash \, or new-line character
8072fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         escape-sequence
8082fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         universal-character-name
8095cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith///       escape-sequence:
8102fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         simple-escape-sequence
8112fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         octal-escape-sequence
8122fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         hexadecimal-escape-sequence
8132fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       simple-escape-sequence:
814ddddd48da72bc29d1c3f388ed91ea5549328129eNAKAMURA Takumi///         one of \' \" \? \\ \a \b \f \n \r \t \v
8152fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       octal-escape-sequence:
8162fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \ octal-digit
8172fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \ octal-digit octal-digit
8182fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \ octal-digit octal-digit octal-digit
8192fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       hexadecimal-escape-sequence:
8202fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \x hexadecimal-digit
8212fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         hexadecimal-escape-sequence hexadecimal-digit
8225cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith///       universal-character-name: [C++11 lex.charset]
8232fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \u hex-quad
8242fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \U hex-quad hex-quad
8252fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       hex-quad:
8262fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         hex-digit hex-digit hex-digit hex-digit
8272fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///
8285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerCharLiteralParser::CharLiteralParser(const char *begin, const char *end,
8295cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor                                     SourceLocation Loc, Preprocessor &PP,
8305cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor                                     tok::TokenKind kind) {
831be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // At this point we know that the character matches the regex "(L|u|U)?'.*'".
8325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  HadError = false;
8331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8345cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  Kind = kind;
8355cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor
83626b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith  const char *TokBegin = begin;
83726b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith
838be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // Skip over wide character determinant.
839be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  if (Kind != tok::char_constant) {
8405cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    ++begin;
8415cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  }
8421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Skip over the entry quote.
8445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  assert(begin[0] == '\'' && "Invalid token lexed");
8455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++begin;
8465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8475cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith  // Remove an optional ud-suffix.
8485cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith  if (end[-1] != '\'') {
8495cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    const char *UDSuffixEnd = end;
8505cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    do {
8515cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith      --end;
8525cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    } while (end[-1] != '\'');
8535cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    UDSuffixBuf.assign(end, UDSuffixEnd);
85426b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith    UDSuffixOffset = end - TokBegin;
8555cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith  }
8565cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith
857be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // Trim the ending quote.
8585cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith  assert(end != begin && "Invalid token lexed");
859be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  --end;
860be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
8611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // FIXME: The "Value" is an uint64_t so we can handle char literals of
862fc8f0e14ad142ed811e90fbd9a30e419e301c717Chris Lattner  // up to 64-bits.
8635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // FIXME: This extensively assumes that 'char' is 8-bits.
86498be4943e8dc4f3905629a7102668960873cf863Chris Lattner  assert(PP.getTargetInfo().getCharWidth() == 8 &&
8655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer         "Assumes char is 8 bits");
866e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  assert(PP.getTargetInfo().getIntWidth() <= 64 &&
867e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner         (PP.getTargetInfo().getIntWidth() & 7) == 0 &&
868e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner         "Assumes sizeof(int) on target is <= 64 and a multiple of char");
869e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  assert(PP.getTargetInfo().getWCharWidth() <= 64 &&
870e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner         "Assumes sizeof(wchar) on target is <= 64");
8714bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta
872be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  SmallVector<uint32_t,4> codepoint_buffer;
873be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  codepoint_buffer.resize(end-begin);
874be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  uint32_t *buffer_begin = &codepoint_buffer.front();
875be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  uint32_t *buffer_end = buffer_begin + codepoint_buffer.size();
876be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
877be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // Unicode escapes representing characters that cannot be correctly
878be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // represented in a single code unit are disallowed in character literals
879be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // by this implementation.
880be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  uint32_t largest_character_for_kind;
881be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  if (tok::wide_char_constant == Kind) {
882be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    largest_character_for_kind = 0xFFFFFFFFu >> (32-PP.getTargetInfo().getWCharWidth());
883be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  } else if (tok::utf16_char_constant == Kind) {
884be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    largest_character_for_kind = 0xFFFF;
885be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  } else if (tok::utf32_char_constant == Kind) {
886be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    largest_character_for_kind = 0x10FFFF;
887be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  } else {
888be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    largest_character_for_kind = 0x7Fu;
889be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  }
8905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
891be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  while (begin!=end) {
892be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    // Is this a span of non-escape characters?
893be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    if (begin[0] != '\\') {
894be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      char const *start = begin;
895be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      do {
896be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell        ++begin;
897be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      } while (begin != end && *begin != '\\');
898be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
89991359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman      char const *tmp_in_start = start;
90091359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman      uint32_t *tmp_out_start = buffer_begin;
901be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      ConversionResult res =
902be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      ConvertUTF8toUTF32(reinterpret_cast<UTF8 const **>(&start),
903be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell                         reinterpret_cast<UTF8 const *>(begin),
904be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell                         &buffer_begin,buffer_end,strictConversion);
905be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      if (res!=conversionOK) {
90691359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        // If we see bad encoding for unprefixed character literals, warn and
90791359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        // simply copy the byte values, for compatibility with gcc and
90891359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        // older versions of clang.
90991359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        bool NoErrorOnBadEncoding = isAscii();
91091359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        unsigned Msg = diag::err_bad_character_encoding;
91191359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        if (NoErrorOnBadEncoding)
91291359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman          Msg = diag::warn_bad_character_encoding;
91391359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        PP.Diag(Loc, Msg);
91491359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        if (NoErrorOnBadEncoding) {
91591359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman          start = tmp_in_start;
91691359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman          buffer_begin = tmp_out_start;
91791359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman          for ( ; start != begin; ++start, ++buffer_begin)
91891359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman            *buffer_begin = static_cast<uint8_t>(*start);
91991359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        } else {
92091359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman          HadError = true;
92191359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        }
9225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      } else {
92391359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        for (; tmp_out_start <buffer_begin; ++tmp_out_start) {
92491359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman          if (*tmp_out_start > largest_character_for_kind) {
925be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell            HadError = true;
926be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell            PP.Diag(Loc, diag::err_character_too_large);
927be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell          }
9281c6c64b5181a960c7d4cace4995a938d4dfa6fbfChris Lattner        }
9295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
930be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
931be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      continue;
9325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
933be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    // Is this a Universal Character Name excape?
934be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    if (begin[1] == 'u' || begin[1] == 'U') {
935be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      unsigned short UcnLen = 0;
93626b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith      if (!ProcessUCNEscape(TokBegin, begin, end, *buffer_begin, UcnLen,
937be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell                            FullSourceLoc(Loc, PP.getSourceManager()),
9384e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie                            &PP.getDiagnostics(), PP.getLangOpts(),
939be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell                            true))
940be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      {
941be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell        HadError = true;
942be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      } else if (*buffer_begin > largest_character_for_kind) {
943be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell        HadError = true;
944be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell        PP.Diag(Loc,diag::err_character_too_large);
945be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      }
9461eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
947be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      ++buffer_begin;
948be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      continue;
949be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    }
950be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    unsigned CharWidth = getCharWidth(Kind, PP.getTargetInfo());
951be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    uint64_t result =
952be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    ProcessCharEscape(begin, end, HadError,
953be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell                      FullSourceLoc(Loc,PP.getSourceManager()),
954be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell                      CharWidth, &PP.getDiagnostics());
955be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    *buffer_begin++ = result;
956e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  }
957e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner
958be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  unsigned NumCharsSoFar = buffer_begin-&codepoint_buffer.front();
959be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
960e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  if (NumCharsSoFar > 1) {
961be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    if (isWide())
9625cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor      PP.Diag(Loc, diag::warn_extraneous_char_constant);
963be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    else if (isAscii() && NumCharsSoFar == 4)
964be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      PP.Diag(Loc, diag::ext_four_char_character_literal);
965be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    else if (isAscii())
966e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner      PP.Diag(Loc, diag::ext_multichar_character_literal);
967e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner    else
968be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      PP.Diag(Loc, diag::err_multichar_utf_character_literal);
9692a1c363f38e59a5044fc349aa7e538a50954c244Eli Friedman    IsMultiChar = true;
970930b71a4a7dedf70a73e5fd875bae7df452b80a9Daniel Dunbar  } else
971930b71a4a7dedf70a73e5fd875bae7df452b80a9Daniel Dunbar    IsMultiChar = false;
9724bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta
973be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  llvm::APInt LitVal(PP.getTargetInfo().getIntWidth(), 0);
974be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
975be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // Narrow character literals act as though their value is concatenated
976be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  // in this implementation, but warn on overflow.
977be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  bool multi_char_too_long = false;
978be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  if (isAscii() && isMultiChar()) {
979be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    LitVal = 0;
980be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    for (size_t i=0;i<NumCharsSoFar;++i) {
981be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      // check for enough leading zeros to shift into
982be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      multi_char_too_long |= (LitVal.countLeadingZeros() < 8);
983be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      LitVal <<= 8;
984be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell      LitVal = LitVal + (codepoint_buffer[i] & 0xFF);
985be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    }
986be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  } else if (NumCharsSoFar > 0) {
987be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    // otherwise just take the last character
988be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    LitVal = buffer_begin[-1];
989be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  }
990be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
991be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  if (!HadError && multi_char_too_long) {
992be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell    PP.Diag(Loc,diag::warn_char_constant_too_large);
993be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell  }
994be773526230b5a7121a8b321b05f2e53fa473f5cSeth Cantrell
9954bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta  // Transfer the value from APInt to uint64_t
9964bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta  Value = LitVal.getZExtValue();
9971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // If this is a single narrow character, sign extend it (e.g. '\xFF' is "-1")
9995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // if 'char' is signed for this target (C99 6.4.4.4p10).  Note that multiple
10005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // character constants are not sign extended in the this implementation:
10015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // '\xFF\xFF' = 65536 and '\x0\xFF' = 255, which matches GCC.
10025cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  if (isAscii() && NumCharsSoFar == 1 && (Value & 128) &&
10034e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie      PP.getLangOpts().CharIsSigned)
10045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Value = (signed char)Value;
10055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
10065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10082fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       string-literal: [C++0x lex.string]
10092fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         encoding-prefix " [s-char-sequence] "
10102fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         encoding-prefix R raw-string
10112fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       encoding-prefix:
10122fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         u8
10132fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         u
10142fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         U
10152fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         L
10165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       s-char-sequence:
10175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         s-char
10185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         s-char-sequence s-char
10195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       s-char:
10202fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         any member of the source character set except the double-quote ",
10212fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///           backslash \, or new-line character
10222fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         escape-sequence
10235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         universal-character-name
10242fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       raw-string:
10252fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         " d-char-sequence ( r-char-sequence ) d-char-sequence "
10262fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       r-char-sequence:
10272fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         r-char
10282fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         r-char-sequence r-char
10292fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       r-char:
10302fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         any member of the source character set, except a right parenthesis )
10312fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///           followed by the initial d-char-sequence (which may be empty)
10322fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///           followed by a double quote ".
10332fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       d-char-sequence:
10342fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         d-char
10352fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         d-char-sequence d-char
10362fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       d-char:
10372fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         any member of the basic source character set except:
10382fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///           space, the left parenthesis (, the right parenthesis ),
10392fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///           the backslash \, and the control characters representing horizontal
10402fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///           tab, vertical tab, form feed, and newline.
10412fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       escape-sequence: [C++0x lex.ccon]
10422fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         simple-escape-sequence
10432fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         octal-escape-sequence
10442fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         hexadecimal-escape-sequence
10452fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       simple-escape-sequence:
1046ddddd48da72bc29d1c3f388ed91ea5549328129eNAKAMURA Takumi///         one of \' \" \? \\ \a \b \f \n \r \t \v
10472fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       octal-escape-sequence:
10482fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \ octal-digit
10492fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \ octal-digit octal-digit
10502fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \ octal-digit octal-digit octal-digit
10512fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///       hexadecimal-escape-sequence:
10522fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         \x hexadecimal-digit
10532fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper///         hexadecimal-escape-sequence hexadecimal-digit
10545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       universal-character-name:
10555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \u hex-quad
10565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \U hex-quad hex-quad
10575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hex-quad:
10585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hex-digit hex-digit hex-digit hex-digit
10595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
10605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerStringLiteralParser::
1061d217773f106856a11879ec79dc468efefaf2ee75Chris LattnerStringLiteralParser(const Token *StringToks, unsigned NumStringToks,
10620833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner                    Preprocessor &PP, bool Complain)
10634e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie  : SM(PP.getSourceManager()), Features(PP.getLangOpts()),
1064403de3f932b5d1d3e4e58f69960000911d04dd2aArgyrios Kyrtzidis    Target(PP.getTargetInfo()), Diags(Complain ? &PP.getDiagnostics() : 0),
10655cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    MaxTokenLength(0), SizeBound(0), CharByteWidth(0), Kind(tok::unknown),
10665cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    ResultPtr(ResultBuf.data()), hadError(false), Pascal(false) {
10670833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner  init(StringToks, NumStringToks);
10680833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner}
10690833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner
10700833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattnervoid StringLiteralParser::init(const Token *StringToks, unsigned NumStringToks){
1071403de3f932b5d1d3e4e58f69960000911d04dd2aArgyrios Kyrtzidis  // The literal token may have come from an invalid source location (e.g. due
1072403de3f932b5d1d3e4e58f69960000911d04dd2aArgyrios Kyrtzidis  // to a PCH error), in which case the token length will be 0.
10733144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis  if (NumStringToks == 0 || StringToks[0].getLength() < 2)
10743144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis    return DiagnoseLexingError(SourceLocation());
1075403de3f932b5d1d3e4e58f69960000911d04dd2aArgyrios Kyrtzidis
10765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Scan all of the string portions, remember the max individual token length,
10775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // computing a bound on the concatenated string length, and see whether any
10785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // piece is a wide-string.  If any of the string portions is a wide-string
10795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // literal, the result is a wide-string literal [C99 6.4.5p4].
1080403de3f932b5d1d3e4e58f69960000911d04dd2aArgyrios Kyrtzidis  assert(NumStringToks && "expected at least one token");
10816cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  MaxTokenLength = StringToks[0].getLength();
1082403de3f932b5d1d3e4e58f69960000911d04dd2aArgyrios Kyrtzidis  assert(StringToks[0].getLength() >= 2 && "literal token is invalid!");
10836cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  SizeBound = StringToks[0].getLength()-2;  // -2 for "".
10845cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  Kind = StringToks[0].getKind();
10856cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt
10866cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  hadError = false;
10875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
10885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Implement Translation Phase #6: concatenation of string literals
10895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// (C99 5.1.1.2p1).  The common case is only one string fragment.
10905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  for (unsigned i = 1; i != NumStringToks; ++i) {
10913144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis    if (StringToks[i].getLength() < 2)
10923144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis      return DiagnoseLexingError(StringToks[i].getLocation());
1093403de3f932b5d1d3e4e58f69960000911d04dd2aArgyrios Kyrtzidis
10945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // The string could be shorter than this if it needs cleaning, but this is a
10955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // reasonable bound, which is all we need.
1096403de3f932b5d1d3e4e58f69960000911d04dd2aArgyrios Kyrtzidis    assert(StringToks[i].getLength() >= 2 && "literal token is invalid!");
10976cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt    SizeBound += StringToks[i].getLength()-2;  // -2 for "".
10981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
10995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Remember maximum string piece length.
11006cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt    if (StringToks[i].getLength() > MaxTokenLength)
11016cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt      MaxTokenLength = StringToks[i].getLength();
11021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11035cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    // Remember if we see any wide or utf-8/16/32 strings.
11045cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    // Also check for illegal concatenations.
11055cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    if (StringToks[i].isNot(Kind) && StringToks[i].isNot(tok::string_literal)) {
11065cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor      if (isAscii()) {
11075cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor        Kind = StringToks[i].getKind();
11085cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor      } else {
11095cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor        if (Diags)
11105cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor          Diags->Report(FullSourceLoc(StringToks[i].getLocation(), SM),
11115cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor                        diag::err_unsupported_string_concat);
11125cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor        hadError = true;
11135cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor      }
11145cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    }
11155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1116dbb1ecc32ca122b07b7c98fd0a8f6f53985adaccChris Lattner
11175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Include space for the null terminator.
11185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++SizeBound;
11191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // TODO: K&R warning: "traditional C rejects string constant concatenation"
11211eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11225cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  // Get the width in bytes of char/wchar_t/char16_t/char32_t
11235cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  CharByteWidth = getCharWidth(Kind, Target);
11245cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  assert((CharByteWidth & 7) == 0 && "Assumes character size is byte multiple");
11255cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  CharByteWidth /= 8;
11261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // The output buffer size needs to be large enough to hold wide characters.
11285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // This is a worst-case assumption which basically corresponds to L"" "long".
11295cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  SizeBound *= CharByteWidth;
11301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Size the temporary buffer to hold the result string data.
11325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ResultBuf.resize(SizeBound);
11331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Likewise, but for each string piece.
1135f7ccbad5d9949e7ddd1cbef43d482553b811e026Dylan Noblesmith  SmallString<512> TokenBuf;
11365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  TokenBuf.resize(MaxTokenLength);
11371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Loop over all the strings, getting their spelling, and expanding them to
11395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // wide strings as appropriate.
11405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ResultPtr = &ResultBuf[0];   // Next byte to fill in.
11411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1142ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson  Pascal = false;
11431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11445cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith  SourceLocation UDSuffixTokLoc;
11455cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith
11465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  for (unsigned i = 0, e = NumStringToks; i != e; ++i) {
11475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    const char *ThisTokBuf = &TokenBuf[0];
11485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Get the spelling of the token, which eliminates trigraphs, etc.  We know
11495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // that ThisTokBuf points to a buffer that is big enough for the whole token
11505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // and 'spelled' tokens can only shrink.
115150f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor    bool StringInvalid = false;
11520833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner    unsigned ThisTokLen =
1153b0607279cb98bbf2bbfe0db170aed39ef91e86a2Chris Lattner      Lexer::getSpelling(StringToks[i], ThisTokBuf, SM, Features,
1154b0607279cb98bbf2bbfe0db170aed39ef91e86a2Chris Lattner                         &StringInvalid);
11553144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis    if (StringInvalid)
11563144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis      return DiagnoseLexingError(StringToks[i].getLocation());
115750f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor
115826b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith    const char *ThisTokBegin = ThisTokBuf;
11595cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    const char *ThisTokEnd = ThisTokBuf+ThisTokLen;
11605cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith
11615cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    // Remove an optional ud-suffix.
11625cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    if (ThisTokEnd[-1] != '"') {
11635cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith      const char *UDSuffixEnd = ThisTokEnd;
11645cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith      do {
11655cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        --ThisTokEnd;
11665cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith      } while (ThisTokEnd[-1] != '"');
11675cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith
11685cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith      StringRef UDSuffix(ThisTokEnd, UDSuffixEnd - ThisTokEnd);
11695cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith
11705cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith      if (UDSuffixBuf.empty()) {
11715cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        UDSuffixBuf.assign(UDSuffix);
1172dd66be718f23c8149d74ae8b011b002e11e8d5baRichard Smith        UDSuffixToken = i;
1173dd66be718f23c8149d74ae8b011b002e11e8d5baRichard Smith        UDSuffixOffset = ThisTokEnd - ThisTokBuf;
11745cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        UDSuffixTokLoc = StringToks[i].getLocation();
11755cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith      } else if (!UDSuffixBuf.equals(UDSuffix)) {
11765cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        // C++11 [lex.ext]p8: At the end of phase 6, if a string literal is the
11775cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        // result of a concatenation involving at least one user-defined-string-
11785cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        // literal, all the participating user-defined-string-literals shall
11795cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        // have the same ud-suffix.
11805cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        if (Diags) {
11815cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith          SourceLocation TokLoc = StringToks[i].getLocation();
11825cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith          Diags->Report(TokLoc, diag::err_string_concat_mixed_suffix)
11835cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith            << UDSuffixBuf << UDSuffix
11845cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith            << SourceRange(UDSuffixTokLoc, UDSuffixTokLoc)
11855cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith            << SourceRange(TokLoc, TokLoc);
11865cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        }
11875cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith        hadError = true;
11885cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith      }
11895cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    }
11905cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith
11915cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    // Strip the end quote.
11925cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith    --ThisTokEnd;
11935cc2c6eb67b6e5361bbe96f79b519fd62ec666d6Richard Smith
11945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // TODO: Input character set mapping support.
11951eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
11961661d717563d6a27dec3da69deba2b2efaa45802Craig Topper    // Skip marker for wide or unicode strings.
11975cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor    if (ThisTokBuf[0] == 'L' || ThisTokBuf[0] == 'u' || ThisTokBuf[0] == 'U') {
11985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ++ThisTokBuf;
11995cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor      // Skip 8 of u8 marker for utf8 strings.
12005cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor      if (ThisTokBuf[0] == '8')
12015cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor        ++ThisTokBuf;
120256bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    }
12031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12042fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper    // Check for raw string
12052fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper    if (ThisTokBuf[0] == 'R') {
12062fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      ThisTokBuf += 2; // skip R"
12071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12082fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      const char *Prefix = ThisTokBuf;
12092fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      while (ThisTokBuf[0] != '(')
1210ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        ++ThisTokBuf;
12112fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      ++ThisTokBuf; // skip '('
12122fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper
121349d517467c3dcd2c67e8a6c740ba5160e37625f7Richard Smith      // Remove same number of characters from the end
121449d517467c3dcd2c67e8a6c740ba5160e37625f7Richard Smith      ThisTokEnd -= ThisTokBuf - Prefix;
121549d517467c3dcd2c67e8a6c740ba5160e37625f7Richard Smith      assert(ThisTokEnd >= ThisTokBuf && "malformed raw string literal");
12162fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper
12172fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      // Copy the string over
121849d517467c3dcd2c67e8a6c740ba5160e37625f7Richard Smith      if (CopyStringFragment(StringRef(ThisTokBuf, ThisTokEnd - ThisTokBuf)))
121991359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman        if (DiagnoseBadString(StringToks[i]))
122091359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman          hadError = true;
12212fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper    } else {
122207a075870110a8376ddc1cd09412a0ec00987153Argyrios Kyrtzidis      if (ThisTokBuf[0] != '"') {
122307a075870110a8376ddc1cd09412a0ec00987153Argyrios Kyrtzidis        // The file may have come from PCH and then changed after loading the
122407a075870110a8376ddc1cd09412a0ec00987153Argyrios Kyrtzidis        // PCH; Fail gracefully.
12253144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis        return DiagnoseLexingError(StringToks[i].getLocation());
122607a075870110a8376ddc1cd09412a0ec00987153Argyrios Kyrtzidis      }
12272fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      ++ThisTokBuf; // skip "
12282fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper
12292fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      // Check if this is a pascal string
12302fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      if (Features.PascalStrings && ThisTokBuf + 1 != ThisTokEnd &&
12312fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          ThisTokBuf[0] == '\\' && ThisTokBuf[1] == 'p') {
12321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12332fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        // If the \p sequence is found in the first token, we have a pascal string
12342fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        // Otherwise, if we already have a pascal string, ignore the first \p
12352fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        if (i == 0) {
12365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          ++ThisTokBuf;
12372fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          Pascal = true;
12382fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        } else if (Pascal)
12392fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          ThisTokBuf += 2;
12405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
12411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12422fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      while (ThisTokBuf != ThisTokEnd) {
12432fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        // Is this a span of non-escape characters?
12442fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        if (ThisTokBuf[0] != '\\') {
12452fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          const char *InStart = ThisTokBuf;
12462fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          do {
12472fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper            ++ThisTokBuf;
12482fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          } while (ThisTokBuf != ThisTokEnd && ThisTokBuf[0] != '\\');
12492fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper
12502fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          // Copy the character span over.
125149d517467c3dcd2c67e8a6c740ba5160e37625f7Richard Smith          if (CopyStringFragment(StringRef(InStart, ThisTokBuf - InStart)))
125291359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman            if (DiagnoseBadString(StringToks[i]))
125391359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman              hadError = true;
12542fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          continue;
12552fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        }
12562fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        // Is this a Universal Character Name escape?
12572fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        if (ThisTokBuf[1] == 'u' || ThisTokBuf[1] == 'U') {
125826b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith          EncodeUCNEscape(ThisTokBegin, ThisTokBuf, ThisTokEnd,
125926b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith                          ResultPtr, hadError,
126026b75c07317a3b50a8a00a1623e3ef38af1d8349Richard Smith                          FullSourceLoc(StringToks[i].getLocation(), SM),
12612fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper                          CharByteWidth, Diags, Features);
12622fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          continue;
12632fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        }
12642fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        // Otherwise, this is a non-UCN escape character.  Process it.
12652fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper        unsigned ResultChar =
12662fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper          ProcessCharEscape(ThisTokBuf, ThisTokEnd, hadError,
12672fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper                            FullSourceLoc(StringToks[i].getLocation(), SM),
12682fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper                            CharByteWidth*8, Diags);
12692fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper
1270caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman        if (CharByteWidth == 4) {
1271caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          // FIXME: Make the type of the result buffer correct instead of
1272caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          // using reinterpret_cast.
1273caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          UTF32 *ResultWidePtr = reinterpret_cast<UTF32*>(ResultPtr);
12749b483df983759d51d61d54e8ae34bff423d15403Nico Weber          *ResultWidePtr = ResultChar;
1275caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          ResultPtr += 4;
1276caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman        } else if (CharByteWidth == 2) {
1277caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          // FIXME: Make the type of the result buffer correct instead of
1278caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          // using reinterpret_cast.
1279caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          UTF16 *ResultWidePtr = reinterpret_cast<UTF16*>(ResultPtr);
12809b483df983759d51d61d54e8ae34bff423d15403Nico Weber          *ResultWidePtr = ResultChar & 0xFFFF;
1281caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          ResultPtr += 2;
1282caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman        } else {
1283caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          assert(CharByteWidth == 1 && "Unexpected char width");
1284caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman          *ResultPtr++ = ResultChar & 0xFF;
1285caf1f26777c3adf2556c3af7bf9e01bd8ead17d9Eli Friedman        }
12862fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper      }
12875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
12885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
12891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1290bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner  if (Pascal) {
129122508f410b3d727d5c557af3304c0a1bad94999eEli Friedman    if (CharByteWidth == 4) {
129222508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      // FIXME: Make the type of the result buffer correct instead of
129322508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      // using reinterpret_cast.
129422508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      UTF32 *ResultWidePtr = reinterpret_cast<UTF32*>(ResultBuf.data());
129522508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      ResultWidePtr[0] = GetNumStringChars() - 1;
129622508f410b3d727d5c557af3304c0a1bad94999eEli Friedman    } else if (CharByteWidth == 2) {
129722508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      // FIXME: Make the type of the result buffer correct instead of
129822508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      // using reinterpret_cast.
129922508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      UTF16 *ResultWidePtr = reinterpret_cast<UTF16*>(ResultBuf.data());
130022508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      ResultWidePtr[0] = GetNumStringChars() - 1;
130122508f410b3d727d5c557af3304c0a1bad94999eEli Friedman    } else {
130222508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      assert(CharByteWidth == 1 && "Unexpected char width");
130322508f410b3d727d5c557af3304c0a1bad94999eEli Friedman      ResultBuf[0] = GetNumStringChars() - 1;
130422508f410b3d727d5c557af3304c0a1bad94999eEli Friedman    }
1305bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner
1306bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner    // Verify that pascal strings aren't too large.
13070833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner    if (GetStringLength() > 256) {
13080833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner      if (Diags)
13090833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner        Diags->Report(FullSourceLoc(StringToks[0].getLocation(), SM),
13100833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner                      diag::err_pascal_string_too_long)
13110833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner          << SourceRange(StringToks[0].getLocation(),
13120833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner                         StringToks[NumStringToks-1].getLocation());
13135cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor      hadError = true;
131457d7dde770c67b282e7fb77b1b81e429910937b3Eli Friedman      return;
131557d7dde770c67b282e7fb77b1b81e429910937b3Eli Friedman    }
13160833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner  } else if (Diags) {
1317427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor    // Complain if this string literal has too many characters.
1318a95880d6513c617bb96634bcc1f16c6bdb80dedcChris Lattner    unsigned MaxChars = Features.CPlusPlus? 65536 : Features.C99 ? 4095 : 509;
1319427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor
1320427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor    if (GetNumStringChars() > MaxChars)
13210833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner      Diags->Report(FullSourceLoc(StringToks[0].getLocation(), SM),
13220833dd0675c25cbb35671c7a2006d511d5c77ce3Chris Lattner                    diag::ext_string_too_long)
1323427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor        << GetNumStringChars() << MaxChars
1324a95880d6513c617bb96634bcc1f16c6bdb80dedcChris Lattner        << (Features.CPlusPlus ? 2 : Features.C99 ? 1 : 0)
1325427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor        << SourceRange(StringToks[0].getLocation(),
1326427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor                       StringToks[NumStringToks-1].getLocation());
1327bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner  }
13285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
1329719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
1330719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
13312fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper/// copyStringFragment - This function copies from Start to End into ResultPtr.
13322fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper/// Performs widening for multi-byte characters.
1333f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedmanbool StringLiteralParser::CopyStringFragment(StringRef Fragment) {
1334f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman  assert(CharByteWidth==1 || CharByteWidth==2 || CharByteWidth==4);
1335f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman  ConversionResult result = conversionOK;
13362fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper  // Copy the character span over.
13372fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper  if (CharByteWidth == 1) {
133849d517467c3dcd2c67e8a6c740ba5160e37625f7Richard Smith    if (!isLegalUTF8String(reinterpret_cast<const UTF8*>(Fragment.begin()),
133949d517467c3dcd2c67e8a6c740ba5160e37625f7Richard Smith                           reinterpret_cast<const UTF8*>(Fragment.end())))
134091359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman      result = sourceIllegal;
13412fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper    memcpy(ResultPtr, Fragment.data(), Fragment.size());
13422fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper    ResultPtr += Fragment.size();
1343f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman  } else if (CharByteWidth == 2) {
1344f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    UTF8 const *sourceStart = (UTF8 const *)Fragment.data();
1345f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    // FIXME: Make the type of the result buffer correct instead of
1346f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    // using reinterpret_cast.
1347f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    UTF16 *targetStart = reinterpret_cast<UTF16*>(ResultPtr);
134891359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman    ConversionFlags flags = strictConversion;
1349f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    result = ConvertUTF8toUTF16(
1350f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman	    &sourceStart,sourceStart + Fragment.size(),
1351f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman        &targetStart,targetStart + 2*Fragment.size(),flags);
1352f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    if (result==conversionOK)
1353f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman      ResultPtr = reinterpret_cast<char*>(targetStart);
1354f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman  } else if (CharByteWidth == 4) {
1355f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    UTF8 const *sourceStart = (UTF8 const *)Fragment.data();
1356f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    // FIXME: Make the type of the result buffer correct instead of
1357f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    // using reinterpret_cast.
1358f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    UTF32 *targetStart = reinterpret_cast<UTF32*>(ResultPtr);
135991359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman    ConversionFlags flags = strictConversion;
1360f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    result = ConvertUTF8toUTF32(
1361f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman        &sourceStart,sourceStart + Fragment.size(),
1362f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman        &targetStart,targetStart + 4*Fragment.size(),flags);
1363f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman    if (result==conversionOK)
1364f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman      ResultPtr = reinterpret_cast<char*>(targetStart);
13652fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper  }
1366f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman  assert((result != targetExhausted)
1367f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman         && "ConvertUTF8toUTFXX exhausted target buffer");
1368f74a4587629615ffd13bd0724868f86ba8c8f27bEli Friedman  return result != conversionOK;
13692fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper}
13702fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper
137191359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedmanbool StringLiteralParser::DiagnoseBadString(const Token &Tok) {
137291359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman  // If we see bad encoding for unprefixed string literals, warn and
137391359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman  // simply copy the byte values, for compatibility with gcc and older
137491359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman  // versions of clang.
137591359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman  bool NoErrorOnBadEncoding = isAscii();
137691359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman  unsigned Msg = NoErrorOnBadEncoding ? diag::warn_bad_string_encoding :
137791359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman                                        diag::err_bad_string_encoding;
137891359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman  if (Diags)
137991359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman    Diags->Report(FullSourceLoc(Tok.getLocation(), SM), Msg);
138091359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman  return !NoErrorOnBadEncoding;
138191359302b822d829afa93c0dadf5f7ce6e19fbc6Eli Friedman}
13822fa4e86b4fdada3b9ecbbbd99965b83ed879f69bCraig Topper
13833144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidisvoid StringLiteralParser::DiagnoseLexingError(SourceLocation Loc) {
13843144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis  hadError = true;
13853144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis  if (Diags)
13863144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis    Diags->Report(Loc, diag::err_lexing_string);
13873144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis}
13883144749f8bf9bbf7c027f2161a930bff80ad6f72Argyrios Kyrtzidis
1389719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// getOffsetOfStringByte - This function returns the offset of the
1390719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// specified byte of the string data represented by Token.  This handles
1391719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// advancing over escape sequences in the string.
1392719e61573f27c11057ecfe0dd8f141621602c571Chris Lattnerunsigned StringLiteralParser::getOffsetOfStringByte(const Token &Tok,
13936c66f07854c1334a1ce9eae1428d61d54182a6e1Chris Lattner                                                    unsigned ByteNo) const {
1394719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  // Get the spelling of the token.
1395f7ccbad5d9949e7ddd1cbef43d482553b811e026Dylan Noblesmith  SmallString<32> SpellingBuffer;
13966cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  SpellingBuffer.resize(Tok.getLength());
13971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
139850f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor  bool StringInvalid = false;
1399719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  const char *SpellingPtr = &SpellingBuffer[0];
1400b0607279cb98bbf2bbfe0db170aed39ef91e86a2Chris Lattner  unsigned TokLen = Lexer::getSpelling(Tok, SpellingPtr, SM, Features,
1401b0607279cb98bbf2bbfe0db170aed39ef91e86a2Chris Lattner                                       &StringInvalid);
140291f54ce93bec136fb9e18740b895cf1c1339524bChris Lattner  if (StringInvalid)
140350f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor    return 0;
1404719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
1405df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  const char *SpellingStart = SpellingPtr;
1406df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  const char *SpellingEnd = SpellingPtr+TokLen;
1407df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith
1408df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  // Handle UTF-8 strings just like narrow strings.
1409df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  if (SpellingPtr[0] == 'u' && SpellingPtr[1] == '8')
1410df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    SpellingPtr += 2;
1411df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith
14125cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor  assert(SpellingPtr[0] != 'L' && SpellingPtr[0] != 'u' &&
14135cee1195584fa8672253139c86e922daeda69b9eDouglas Gregor         SpellingPtr[0] != 'U' && "Doesn't handle wide or utf strings yet");
1414719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
1415df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  // For raw string literals, this is easy.
1416df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  if (SpellingPtr[0] == 'R') {
1417df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    assert(SpellingPtr[1] == '"' && "Should be a raw string literal!");
1418df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    // Skip 'R"'.
1419df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    SpellingPtr += 2;
1420df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    while (*SpellingPtr != '(') {
1421df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      ++SpellingPtr;
1422df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      assert(SpellingPtr < SpellingEnd && "Missing ( for raw string literal");
1423df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    }
1424df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    // Skip '('.
1425df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    ++SpellingPtr;
1426df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    return SpellingPtr - SpellingStart + ByteNo;
1427df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  }
14281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1429df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith  // Skip over the leading quote
1430719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  assert(SpellingPtr[0] == '"' && "Should be a string literal!");
1431719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  ++SpellingPtr;
14321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1433719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  // Skip over bytes until we find the offset we're looking for.
1434719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  while (ByteNo) {
1435719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    assert(SpellingPtr < SpellingEnd && "Didn't find byte offset!");
14361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1437719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    // Step over non-escapes simply.
1438719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    if (*SpellingPtr != '\\') {
1439719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      ++SpellingPtr;
1440719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      --ByteNo;
1441719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      continue;
1442719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    }
14431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1444719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    // Otherwise, this is an escape character.  Advance over it.
1445719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    bool HadError = false;
1446df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    if (SpellingPtr[1] == 'u' || SpellingPtr[1] == 'U') {
1447df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      const char *EscapePtr = SpellingPtr;
1448df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      unsigned Len = MeasureUCNEscape(SpellingStart, SpellingPtr, SpellingEnd,
1449df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith                                      1, Features, HadError);
1450df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      if (Len > ByteNo) {
1451df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith        // ByteNo is somewhere within the escape sequence.
1452df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith        SpellingPtr = EscapePtr;
1453df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith        break;
1454df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      }
1455df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      ByteNo -= Len;
1456df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    } else {
1457df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      ProcessCharEscape(SpellingPtr, SpellingEnd, HadError,
1458df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith                        FullSourceLoc(Tok.getLocation(), SM),
1459df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith                        CharByteWidth*8, Diags);
1460df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith      --ByteNo;
1461df9ef1bc8c3780307ab2ed81bf5e31c23310b936Richard Smith    }
1462719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    assert(!HadError && "This method isn't valid on erroneous strings");
1463719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  }
14641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1465719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  return SpellingPtr-SpellingStart;
1466719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner}
1467