LiteralSupport.cpp revision 56bedefe92ae8f604d14bea75cc3040ab32337c2
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"
19e9f195f15ffe96d0a220c872ab12d0630a633c44Erick Tryzelaar#include "llvm/ADT/StringRef.h"
205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "llvm/ADT/StringExtras.h"
215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerusing namespace clang;
225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// HexDigitValue - Return the value of the specified hex digit, or -1 if it's
245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// not valid.
255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic int HexDigitValue(char C) {
265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= '0' && C <= '9') return C-'0';
275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= 'a' && C <= 'f') return C-'a'+10;
285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= 'A' && C <= 'F') return C-'A'+10;
295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return -1;
305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ProcessCharEscape - Parse a standard C escape sequence, which can occur in
335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// either a character or a string literal.
345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic unsigned ProcessCharEscape(const char *&ThisTokBuf,
355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                                  const char *ThisTokEnd, bool &HadError,
365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                                  SourceLocation Loc, bool IsWide,
37b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor                                  Preprocessor &PP, bool Complain) {
385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Skip the '\' char.
395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++ThisTokBuf;
405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // We know that this character can't be off the end of the buffer, because
425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // that would have been \", which would not have been the end of string.
435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  unsigned ResultChar = *ThisTokBuf++;
445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  switch (ResultChar) {
455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // These map to themselves.
465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '\\': case '\'': case '"': case '?': break;
471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // These have fixed mappings.
495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'a':
505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // TODO: K&R: the meaning of '\\a' is different in traditional C
515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 7;
525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'b':
545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 8;
555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'e':
57b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (Complain)
58b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      PP.Diag(Loc, diag::ext_nonstandard_escape) << "e";
595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 27;
605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
613c54801fbc67d8df2fed0711a2e2022db6b1bbcfEli Friedman  case 'E':
62b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (Complain)
63b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      PP.Diag(Loc, diag::ext_nonstandard_escape) << "E";
643c54801fbc67d8df2fed0711a2e2022db6b1bbcfEli Friedman    ResultChar = 27;
653c54801fbc67d8df2fed0711a2e2022db6b1bbcfEli Friedman    break;
665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'f':
675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 12;
685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'n':
705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 10;
715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'r':
735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 13;
745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 't':
765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 9;
775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'v':
795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 11;
805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'x': { // Hex escape.
825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 0;
835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (ThisTokBuf == ThisTokEnd || !isxdigit(*ThisTokBuf)) {
84b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      if (Complain)
85b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor        PP.Diag(Loc, diag::err_hex_escape_no_digits);
865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      HadError = 1;
875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      break;
885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Hex escapes are a maximal series of hex digits.
915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    bool Overflow = false;
925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    for (; ThisTokBuf != ThisTokEnd; ++ThisTokBuf) {
935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      int CharVal = HexDigitValue(ThisTokBuf[0]);
945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (CharVal == -1) break;
95c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner      // About to shift out a digit?
96c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner      Overflow |= (ResultChar & 0xF0000000) ? true : false;
975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar <<= 4;
985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar |= CharVal;
995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // See if any bits will be truncated when evaluated as a character.
1021a75ee270941dd8b5c7fdd23dffb5e81a0fd7290Alisdair Meredith    unsigned CharWidth = IsWide
1031a75ee270941dd8b5c7fdd23dffb5e81a0fd7290Alisdair Meredith                       ? PP.getTargetInfo().getWCharWidth()
1041a75ee270941dd8b5c7fdd23dffb5e81a0fd7290Alisdair Meredith                       : PP.getTargetInfo().getCharWidth();
1051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) {
1075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      Overflow = true;
1085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar &= ~0U >> (32-CharWidth);
1095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Check for overflow.
112b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (Overflow && Complain)   // Too many digits to fit in
1135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      PP.Diag(Loc, diag::warn_hex_escape_too_large);
1145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '0': case '1': case '2': case '3':
1175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '4': case '5': case '6': case '7': {
1185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Octal escapes.
1195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    --ThisTokBuf;
1205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 0;
1215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Octal escapes are a series of octal digits with maximum length 3.
1235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // "\0123" is a two digit sequence equal to "\012" "3".
1245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    unsigned NumDigits = 0;
1255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    do {
1265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar <<= 3;
1275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar |= *ThisTokBuf++ - '0';
1285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ++NumDigits;
1295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    } while (ThisTokBuf != ThisTokEnd && NumDigits < 3 &&
1305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer             ThisTokBuf[0] >= '0' && ThisTokBuf[0] <= '7');
1311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Check for overflow.  Reject '\777', but not L'\777'.
1331a75ee270941dd8b5c7fdd23dffb5e81a0fd7290Alisdair Meredith    unsigned CharWidth = IsWide
1341a75ee270941dd8b5c7fdd23dffb5e81a0fd7290Alisdair Meredith                       ? PP.getTargetInfo().getWCharWidth()
1351a75ee270941dd8b5c7fdd23dffb5e81a0fd7290Alisdair Meredith                       : PP.getTargetInfo().getCharWidth();
1361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) {
138b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      if (Complain)
139b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor        PP.Diag(Loc, diag::warn_octal_escape_too_large);
1405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar &= ~0U >> (32-CharWidth);
1415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Otherwise, these are not valid escapes.
1465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '(': case '{': case '[': case '%':
1475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // GCC accepts these as extensions.  We warn about them as such though.
148b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (Complain)
149b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      PP.Diag(Loc, diag::ext_nonstandard_escape)
150b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor        << std::string()+(char)ResultChar;
151f01fdff97b245caac98100d232c760b4d0531411Eli Friedman    break;
1525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  default:
153b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (!Complain)
154b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      break;
155b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor
156ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    if (isgraph(ThisTokBuf[0]))
157204b2fed909b1eabea7aeb6caadd7cff718edee5Chris Lattner      PP.Diag(Loc, diag::ext_unknown_escape) << std::string()+(char)ResultChar;
158ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    else
159204b2fed909b1eabea7aeb6caadd7cff718edee5Chris Lattner      PP.Diag(Loc, diag::ext_unknown_escape) << "x"+llvm::utohexstr(ResultChar);
1605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return ResultChar;
1645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
1655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1660e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// ProcessUCNEscape - Read the Universal Character Name, check constraints and
1670e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// convert the UTF32 to UTF8. This is a subroutine of StringLiteralParser.
1680e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// When we decide to implement UCN's for character constants and identifiers,
1690e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// we will likely rework our support for UCN's.
1701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpstatic void ProcessUCNEscape(const char *&ThisTokBuf, const char *ThisTokEnd,
1711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                             char *&ResultBuf, bool &HadError,
17237dd3ecbfc555c3ee0a1922f4bd5ab999c1e5461Chris Lattner                             SourceLocation Loc, Preprocessor &PP,
17356bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian                             bool wide,
17437dd3ecbfc555c3ee0a1922f4bd5ab999c1e5461Chris Lattner                             bool Complain) {
1750e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // FIXME: Add a warning - UCN's are only valid in C++ & C99.
1768a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff  // FIXME: Handle wide strings.
1771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1784e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff  // Save the beginning of the string (for error diagnostics).
1794e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff  const char *ThisTokBegin = ThisTokBuf;
1801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1810e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Skip the '\u' char's.
1820e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ThisTokBuf += 2;
1830e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
1840e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if (ThisTokBuf == ThisTokEnd || !isxdigit(*ThisTokBuf)) {
185b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (Complain)
186b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      PP.Diag(Loc, diag::err_ucn_escape_no_digits);
1870e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    HadError = 1;
1880e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    return;
1890e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
1904e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff  typedef uint32_t UTF32;
1911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1920e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  UTF32 UcnVal = 0;
1930e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  unsigned short UcnLen = (ThisTokBuf[-1] == 'u' ? 4 : 8);
19456bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian  unsigned short UcnLenSave = UcnLen;
1950e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  for (; ThisTokBuf != ThisTokEnd && UcnLen; ++ThisTokBuf, UcnLen--) {
1960e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    int CharVal = HexDigitValue(ThisTokBuf[0]);
1970e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    if (CharVal == -1) break;
1980e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    UcnVal <<= 4;
1990e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    UcnVal |= CharVal;
2000e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
2010e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // If we didn't consume the proper number of digits, there is a problem.
2020e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if (UcnLen) {
203b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (Complain)
204b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      PP.Diag(PP.AdvanceToTokenCharacter(Loc, ThisTokBuf-ThisTokBegin),
205b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor              diag::err_ucn_escape_incomplete);
2060e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    HadError = 1;
2070e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    return;
2080e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
2091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Check UCN constraints (C99 6.4.3p2).
2100e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if ((UcnVal < 0xa0 &&
2110e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff      (UcnVal != 0x24 && UcnVal != 0x40 && UcnVal != 0x60 )) // $, @, `
2121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      || (UcnVal >= 0xD800 && UcnVal <= 0xDFFF)
2138a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff      || (UcnVal > 0x10FFFF)) /* the maximum legal UTF32 value */ {
214b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (Complain)
215b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      PP.Diag(Loc, diag::err_ucn_escape_invalid);
2160e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    HadError = 1;
2170e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    return;
2180e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
21956bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian  if (wide) {
22056bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    assert(UcnLenSave == 4 &&
22156bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian           "ProcessUCNEscape - only ucn length of 4 supported");
22256bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    // little endian assumed.
22356bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    *ResultBuf++ = (UcnVal & 0x000000FF);
22456bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    *ResultBuf++ = (UcnVal & 0x0000FF00) >> 8;
22556bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    *ResultBuf++ = (UcnVal & 0x00FF0000) >> 16;
22656bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    *ResultBuf++ = (UcnVal & 0xFF000000) >> 24;
22756bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    return;
22856bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian  }
2290e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Now that we've parsed/checked the UCN, we convert from UTF32->UTF8.
2300e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // The conversion below was inspired by:
2310e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  //   http://www.unicode.org/Public/PROGRAMS/CVTUTF/ConvertUTF.c
2321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // First, we determine how many bytes the result will require.
2334e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff  typedef uint8_t UTF8;
2340e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
2350e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  unsigned short bytesToWrite = 0;
2360e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if (UcnVal < (UTF32)0x80)
2370e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 1;
2380e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else if (UcnVal < (UTF32)0x800)
2390e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 2;
2400e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else if (UcnVal < (UTF32)0x10000)
2410e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 3;
2420e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else
2430e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 4;
2441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2450e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  const unsigned byteMask = 0xBF;
2460e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  const unsigned byteMark = 0x80;
2471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
2480e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Once the bits are split out into bytes of UTF8, this is a mask OR-ed
2498a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff  // into the first byte, depending on how many bytes follow.
2501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  static const UTF8 firstByteMark[5] = {
2518a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff    0x00, 0x00, 0xC0, 0xE0, 0xF0
2520e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  };
2530e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Finally, we write the bytes into ResultBuf.
2540e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ResultBuf += bytesToWrite;
2550e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  switch (bytesToWrite) { // note: everything falls through.
2560e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 4: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
2570e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 3: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
2580e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 2: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
2590e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 1: *--ResultBuf = (UTF8) (UcnVal | firstByteMark[bytesToWrite]);
2600e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
2610e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Update the buffer.
2620e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ResultBuf += bytesToWrite;
2630e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff}
2645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       integer-constant: [C99 6.4.4.1]
2675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         decimal-constant integer-suffix
2685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-constant integer-suffix
2695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-constant integer-suffix
2701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///       decimal-constant:
2715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         nonzero-digit
2725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         decimal-constant digit
2731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///       octal-constant:
2745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0
2755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-constant octal-digit
2761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///       hexadecimal-constant:
2775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-prefix hexadecimal-digit
2785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-constant hexadecimal-digit
2795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hexadecimal-prefix: one of
2805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0x 0X
2815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       integer-suffix:
2825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         unsigned-suffix [long-suffix]
2835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         unsigned-suffix [long-long-suffix]
2845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         long-suffix [unsigned-suffix]
2855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         long-long-suffix [unsigned-sufix]
2865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       nonzero-digit:
2875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         1 2 3 4 5 6 7 8 9
2885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       octal-digit:
2895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0 1 2 3 4 5 6 7
2905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hexadecimal-digit:
2915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0 1 2 3 4 5 6 7 8 9
2925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         a b c d e f
2935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         A B C D E F
2945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       unsigned-suffix: one of
2955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         u U
2965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       long-suffix: one of
2975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         l L
2981eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump///       long-long-suffix: one of
2995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         ll LL
3005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
3015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       floating-constant: [C99 6.4.4.2]
3025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         TODO: add rules...
3035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
3045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerNumericLiteralParser::
3055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerNumericLiteralParser(const char *begin, const char *end,
3065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                     SourceLocation TokLoc, Preprocessor &pp)
3075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  : PP(pp), ThisTokBegin(begin), ThisTokEnd(end) {
3081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
309c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // This routine assumes that the range begin/end matches the regex for integer
310c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // and FP constants (specifically, the 'pp-number' regex), and assumes that
311c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // the byte at "*end" is both valid and not part of the regex.  Because of
312c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // this, it doesn't have to check for 'overscan' in various places.
313c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  assert(!isalnum(*end) && *end != '.' && *end != '_' &&
314c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner         "Lexer didn't maximally munch?");
3151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  s = DigitsBegin = begin;
3175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  saw_exponent = false;
3185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  saw_period = false;
3195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isLong = false;
3205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isUnsigned = false;
3215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isLongLong = false;
3226e400c286b485e28d04a742ea87860ddfefa672eChris Lattner  isFloat = false;
323506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  isImaginary = false;
324b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump  isMicrosoftInteger = false;
3255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  hadError = false;
3261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (*s == '0') { // parse radix
328368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    ParseNumberStartingWithZero(TokLoc);
329368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (hadError)
330368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      return;
3315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  } else { // the first digit is non-zero
3325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    radix = 10;
3335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    s = SkipDigits(s);
3345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (s == ThisTokEnd) {
3355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      // Done.
336016765e3453db2e302efe53905e99cdb25501234Christopher Lamb    } else if (isxdigit(*s) && !(*s == 'e' || *s == 'E')) {
337ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
338476d8b863cb65b2b5833235d97315cdb46e6f5aaBenjamin Kramer              diag::err_invalid_decimal_digit) << llvm::StringRef(s, 1);
339ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
3405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      return;
3415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    } else if (*s == '.') {
3425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s++;
3435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      saw_period = true;
3445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s = SkipDigits(s);
3451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    }
3464411f46050216a139ab2fc7ff145ec384d11ec7fChris Lattner    if ((*s == 'e' || *s == 'E')) { // exponent
34770f66ab053f36ab3df7a778d09bcb2b4b0fec1f8Chris Lattner      const char *Exponent = s;
3485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s++;
3495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      saw_exponent = true;
3505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (*s == '+' || *s == '-')  s++; // sign
3515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      const char *first_non_digit = SkipDigits(s);
3520b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner      if (first_non_digit != s) {
3535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        s = first_non_digit;
3540b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner      } else {
355ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-begin),
356ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                diag::err_exponent_has_no_digits);
357ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        hadError = true;
3580b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner        return;
3595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
3605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
3615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
3625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
3635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SuffixBegin = s;
3641eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
365506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // Parse the suffix.  At this point we can classify whether we have an FP or
366506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // integer constant.
367506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  bool isFPConstant = isFloatingLiteral();
3681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
369506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // Loop over all of the characters of the suffix.  If we see something bad,
370506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // we break out of the loop.
371506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  for (; s != ThisTokEnd; ++s) {
372506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    switch (*s) {
373506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'f':      // FP Suffix for "float"
374506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'F':
375506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (!isFPConstant) break;  // Error for integer constant.
3766e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      if (isFloat || isLong) break; // FF, LF invalid.
3776e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      isFloat = true;
378506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
379506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'u':
380506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'U':
381506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isFPConstant) break;  // Error for floating constant.
382506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isUnsigned) break;    // Cannot be repeated.
383506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      isUnsigned = true;
384506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
385506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'l':
386506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'L':
387506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isLong || isLongLong) break;  // Cannot be repeated.
3886e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      if (isFloat) break;               // LF invalid.
3891eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
390506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      // Check for long long.  The L's need to be adjacent and the same case.
391506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (s+1 != ThisTokEnd && s[1] == s[0]) {
392506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        if (isFPConstant) break;        // long long invalid for floats.
393506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        isLongLong = true;
394506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        ++s;  // Eat both of them.
395506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      } else {
3965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        isLong = true;
3975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
398506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
399506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'i':
4000c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff      if (PP.getLangOptions().Microsoft) {
401a8be02b655b76e4dbe776b0c62bc3c450dc6feabFariborz Jahanian        if (isFPConstant || isLong || isLongLong) break;
4026e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes
4030c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        // Allow i8, i16, i32, i64, and i128.
404b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump        if (s + 1 != ThisTokEnd) {
405b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump          switch (s[1]) {
406b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            case '8':
407b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              s += 2; // i8 suffix
408b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              isMicrosoftInteger = true;
4096e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              break;
410b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            case '1':
4116e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s + 2 == ThisTokEnd) break;
4126e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s[2] == '6') s += 3; // i16 suffix
4136e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              else if (s[2] == '2') {
4146e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes                if (s + 3 == ThisTokEnd) break;
4156e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes                if (s[3] == '8') s += 4; // i128 suffix
416b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              }
417b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              isMicrosoftInteger = true;
4186e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              break;
419b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            case '3':
4206e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s + 2 == ThisTokEnd) break;
4216e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s[2] == '2') s += 3; // i32 suffix
422b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              isMicrosoftInteger = true;
4236e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              break;
424b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            case '6':
4256e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s + 2 == ThisTokEnd) break;
4266e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              if (s[2] == '4') s += 3; // i64 suffix
427b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              isMicrosoftInteger = true;
4286e8c7acb61b2c7f421d6e1aba8a7a84e96ab6981Nuno Lopes              break;
429b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump            default:
430b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump              break;
431b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump          }
432b79fe2d28777652a4df4f49dc876cbec060ca90eMike Stump          break;
4330c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        }
4340c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff      }
4350c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff      // fall through.
436506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'I':
437506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'j':
438506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'J':
439506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isImaginary) break;   // Cannot be repeated.
440506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
441506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner              diag::ext_imaginary_constant);
442506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      isImaginary = true;
443506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
4445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
445506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    // If we reached here, there was an error.
446506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    break;
447506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  }
4481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
449506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // Report an error if there are any.
450506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  if (s != ThisTokEnd) {
451ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
452ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner            isFPConstant ? diag::err_invalid_suffix_float_constant :
453ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                           diag::err_invalid_suffix_integer_constant)
454476d8b863cb65b2b5833235d97315cdb46e6f5aaBenjamin Kramer      << llvm::StringRef(SuffixBegin, ThisTokEnd-SuffixBegin);
455ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    hadError = true;
456506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    return;
4575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
4585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
4595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
460368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// ParseNumberStartingWithZero - This method is called when the first character
461368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// of the number is found to be a zero.  This means it is either an octal
462368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// number (like '04') or a hex number ('0x123a') a binary number ('0b1010') or
4631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// a floating point number (01239.123e4).  Eat the prefix, determining the
464368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// radix etc.
465368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattnervoid NumericLiteralParser::ParseNumberStartingWithZero(SourceLocation TokLoc) {
466368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  assert(s[0] == '0' && "Invalid method call");
467368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  s++;
4681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
469368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // Handle a hex number like 0x1234.
470368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if ((*s == 'x' || *s == 'X') && (isxdigit(s[1]) || s[1] == '.')) {
471368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
472368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 16;
473368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    DigitsBegin = s;
474368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s = SkipHexDigits(s);
475368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (s == ThisTokEnd) {
476368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      // Done.
477368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (*s == '.') {
478368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s++;
479368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      saw_period = true;
480368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s = SkipHexDigits(s);
481368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
482368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    // A binary exponent can appear with or with a '.'. If dotted, the
4831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // binary exponent is required.
4848c723404c34e9347fae473489194e8c200c4ac83Sean Hunt    if ((*s == 'p' || *s == 'P') && !PP.getLangOptions().CPlusPlus0x) {
485368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      const char *Exponent = s;
486368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s++;
487368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      saw_exponent = true;
488368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      if (*s == '+' || *s == '-')  s++; // sign
489368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      const char *first_non_digit = SkipDigits(s);
4906ea623823f8532670480425b573f35115404b4a0Chris Lattner      if (first_non_digit == s) {
491ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin),
492ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                diag::err_exponent_has_no_digits);
493ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        hadError = true;
4946ea623823f8532670480425b573f35115404b4a0Chris Lattner        return;
495368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      }
4966ea623823f8532670480425b573f35115404b4a0Chris Lattner      s = first_non_digit;
4971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
4988c723404c34e9347fae473489194e8c200c4ac83Sean Hunt      // In C++0x, we cannot support hexadecmial floating literals because
4998c723404c34e9347fae473489194e8c200c4ac83Sean Hunt      // they conflict with user-defined literals, so we warn in previous
5008c723404c34e9347fae473489194e8c200c4ac83Sean Hunt      // versions of C++ by default.
5018c723404c34e9347fae473489194e8c200c4ac83Sean Hunt      if (PP.getLangOptions().CPlusPlus)
5028c723404c34e9347fae473489194e8c200c4ac83Sean Hunt        PP.Diag(TokLoc, diag::ext_hexconstant_cplusplus);
5038c723404c34e9347fae473489194e8c200c4ac83Sean Hunt      else if (!PP.getLangOptions().HexFloats)
504ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(TokLoc, diag::ext_hexconstant_invalid);
505368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (saw_period) {
506ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
507ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner              diag::err_hexconstant_requires_exponent);
508ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
509368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
510368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
511368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
5121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
513368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // Handle simple binary numbers 0b01010
514368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == 'b' || *s == 'B') {
515368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    // 0b101010 is a GCC extension.
516413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    PP.Diag(TokLoc, diag::ext_binary_literal);
517368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    ++s;
518368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 2;
519368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    DigitsBegin = s;
520368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s = SkipBinaryDigits(s);
521368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (s == ThisTokEnd) {
522368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      // Done.
523368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (isxdigit(*s)) {
524ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
525476d8b863cb65b2b5833235d97315cdb46e6f5aaBenjamin Kramer              diag::err_invalid_binary_digit) << llvm::StringRef(s, 1);
526ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
527368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
528413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    // Other suffixes will be diagnosed by the caller.
529368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
530368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
5311eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
532368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // For now, the radix is set to 8. If we discover that we have a
533368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // floating point constant, the radix will change to 10. Octal floating
5341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // point constants are not permitted (only decimal and hexadecimal).
535368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  radix = 8;
536368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  DigitsBegin = s;
537368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  s = SkipOctalDigits(s);
538368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (s == ThisTokEnd)
539368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return; // Done, simple octal number like 01234
5401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
541413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // If we have some other non-octal digit that *is* a decimal digit, see if
542413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // this is part of a floating point number like 094.123 or 09e1.
543413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  if (isdigit(*s)) {
544413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    const char *EndDecimal = SkipDigits(s);
545413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    if (EndDecimal[0] == '.' || EndDecimal[0] == 'e' || EndDecimal[0] == 'E') {
546413d355e38755a71f106dbc0ac900ca989070916Chris Lattner      s = EndDecimal;
547413d355e38755a71f106dbc0ac900ca989070916Chris Lattner      radix = 10;
548413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    }
549413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  }
5501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
551413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // If we have a hex digit other than 'e' (which denotes a FP exponent) then
552413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // the code is using an incorrect base.
553368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (isxdigit(*s) && *s != 'e' && *s != 'E') {
554ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
555476d8b863cb65b2b5833235d97315cdb46e6f5aaBenjamin Kramer            diag::err_invalid_octal_digit) << llvm::StringRef(s, 1);
556ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    hadError = true;
557368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
558368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
5591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
560368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == '.') {
561368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
562368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 10;
563368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    saw_period = true;
564413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    s = SkipDigits(s); // Skip suffix.
565368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
566368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == 'e' || *s == 'E') { // exponent
567368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    const char *Exponent = s;
568368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
569368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 10;
570368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    saw_exponent = true;
571368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (*s == '+' || *s == '-')  s++; // sign
572368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    const char *first_non_digit = SkipDigits(s);
573368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (first_non_digit != s) {
574368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s = first_non_digit;
575368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else {
5761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin),
577ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner              diag::err_exponent_has_no_digits);
578ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
579368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      return;
580368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
581368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
582368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner}
583368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
584368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
5855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// GetIntegerValue - Convert this numeric literal value to an APInt that
5865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// matches Val's input width.  If there is an overflow, set Val to the low bits
5875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// of the result and return true.  Otherwise, return false.
5885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerbool NumericLiteralParser::GetIntegerValue(llvm::APInt &Val) {
589a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // Fast path: Compute a conservative bound on the maximum number of
590a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // bits per digit in this radix. If we can't possibly overflow a
591a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // uint64 based on that bound then do the simple conversion to
592a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // integer. This avoids the expensive overflow checking below, and
593a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // handles the common cases that matter (small decimal integers and
594a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // hex/octal values which don't overflow).
595a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  unsigned MaxBitsPerDigit = 1;
5961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  while ((1U << MaxBitsPerDigit) < radix)
597a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    MaxBitsPerDigit += 1;
598a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  if ((SuffixBegin - DigitsBegin) * MaxBitsPerDigit <= 64) {
599a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    uint64_t N = 0;
600a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    for (s = DigitsBegin; s != SuffixBegin; ++s)
601a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar      N = N*radix + HexDigitValue(*s);
602a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar
603a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    // This will truncate the value to Val's input width. Simply check
604a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    // for overflow by comparing.
605a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    Val = N;
606a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    return Val.getZExtValue() != N;
607a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  }
608a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar
6095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  Val = 0;
6105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  s = DigitsBegin;
6115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt RadixVal(Val.getBitWidth(), radix);
6135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt CharVal(Val.getBitWidth(), 0);
6145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt OldVal = Val;
6151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool OverflowOccurred = false;
6175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  while (s < SuffixBegin) {
6185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    unsigned C = HexDigitValue(*s++);
6191eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // If this letter is out of bound for this radix, reject it.
6215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    assert(C < radix && "NumericLiteralParser ctor should have rejected this");
6221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    CharVal = C;
6241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Add the digit to the value in the appropriate radix.  If adding in digits
6265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // made the value smaller, then this overflowed.
6275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OldVal = Val;
6285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Multiply by radix, did overflow occur on the multiply?
6305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Val *= RadixVal;
6315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OverflowOccurred |= Val.udiv(RadixVal) != OldVal;
6325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Add value, did overflow occur on the value?
634d70cb645702bdbb42aee58403306a7c47e0d901cDaniel Dunbar    //   (a + b) ult b  <=> overflow
6355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Val += CharVal;
6365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OverflowOccurred |= Val.ult(CharVal);
6375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
6385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return OverflowOccurred;
6395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
6405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
64194c939dc1d4958b62ea5a89294dd8b2905f3191fJohn McCallllvm::APFloat::opStatus
64294c939dc1d4958b62ea5a89294dd8b2905f3191fJohn McCallNumericLiteralParser::GetFloatValue(llvm::APFloat &Result) {
643427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek  using llvm::APFloat;
644e9f195f15ffe96d0a220c872ab12d0630a633c44Erick Tryzelaar  using llvm::StringRef;
6451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
646e9f195f15ffe96d0a220c872ab12d0630a633c44Erick Tryzelaar  unsigned n = std::min(SuffixBegin - ThisTokBegin, ThisTokEnd - ThisTokBegin);
64794c939dc1d4958b62ea5a89294dd8b2905f3191fJohn McCall  return Result.convertFromString(StringRef(ThisTokBegin, n),
64894c939dc1d4958b62ea5a89294dd8b2905f3191fJohn McCall                                  APFloat::rmNearestTiesToEven);
6495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
6505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerCharLiteralParser::CharLiteralParser(const char *begin, const char *end,
6535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                                     SourceLocation Loc, Preprocessor &PP) {
6545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // At this point we know that the character matches the regex "L?'.*'".
6555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  HadError = false;
6561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Determine if this is a wide character.
6585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  IsWide = begin[0] == 'L';
6595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (IsWide) ++begin;
6601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
6615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Skip over the entry quote.
6625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  assert(begin[0] == '\'' && "Invalid token lexed");
6635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++begin;
6645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // FIXME: The "Value" is an uint64_t so we can handle char literals of
6664bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta  // upto 64-bits.
6675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // FIXME: This extensively assumes that 'char' is 8-bits.
66898be4943e8dc4f3905629a7102668960873cf863Chris Lattner  assert(PP.getTargetInfo().getCharWidth() == 8 &&
6695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer         "Assumes char is 8 bits");
670e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  assert(PP.getTargetInfo().getIntWidth() <= 64 &&
671e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner         (PP.getTargetInfo().getIntWidth() & 7) == 0 &&
672e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner         "Assumes sizeof(int) on target is <= 64 and a multiple of char");
673e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  assert(PP.getTargetInfo().getWCharWidth() <= 64 &&
674e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner         "Assumes sizeof(wchar) on target is <= 64");
6754bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta
6761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // This is what we will use for overflow detection
6774bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta  llvm::APInt LitVal(PP.getTargetInfo().getIntWidth(), 0);
6781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
679e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  unsigned NumCharsSoFar = 0;
6801c6c64b5181a960c7d4cace4995a938d4dfa6fbfChris Lattner  bool Warned = false;
6815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  while (begin[0] != '\'') {
6824bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta    uint64_t ResultChar;
6835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (begin[0] != '\\')     // If this is a normal character, consume it.
6845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar = *begin++;
6855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    else                      // Otherwise, this is an escape character.
686b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor      ResultChar = ProcessCharEscape(begin, end, HadError, Loc, IsWide, PP,
687b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor                                     /*Complain=*/true);
6885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // If this is a multi-character constant (e.g. 'abc'), handle it.  These are
6905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // implementation defined (C99 6.4.4.4p10).
691e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner    if (NumCharsSoFar) {
6925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (IsWide) {
6935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // Emulate GCC's (unintentional?) behavior: L'ab' -> L'b'.
6944bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta        LitVal = 0;
6955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      } else {
6965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // Narrow character literals act as though their value is concatenated
697e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner        // in this implementation, but warn on overflow.
6981c6c64b5181a960c7d4cace4995a938d4dfa6fbfChris Lattner        if (LitVal.countLeadingZeros() < 8 && !Warned) {
6995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          PP.Diag(Loc, diag::warn_char_constant_too_large);
7001c6c64b5181a960c7d4cace4995a938d4dfa6fbfChris Lattner          Warned = true;
7011c6c64b5181a960c7d4cace4995a938d4dfa6fbfChris Lattner        }
7024bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta        LitVal <<= 8;
7035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
7045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
7051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7064bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta    LitVal = LitVal + ResultChar;
707e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner    ++NumCharsSoFar;
708e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  }
709e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner
710e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  // If this is the second character being processed, do special handling.
711e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  if (NumCharsSoFar > 1) {
712e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner    // Warn about discarding the top bits for multi-char wide-character
713e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner    // constants (L'abcd').
714e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner    if (IsWide)
715e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner      PP.Diag(Loc, diag::warn_extraneous_wide_char_constant);
716e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner    else if (NumCharsSoFar != 4)
717e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner      PP.Diag(Loc, diag::ext_multichar_character_literal);
718e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner    else
719e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner      PP.Diag(Loc, diag::ext_four_char_character_literal);
7202a1c363f38e59a5044fc349aa7e538a50954c244Eli Friedman    IsMultiChar = true;
721930b71a4a7dedf70a73e5fd875bae7df452b80a9Daniel Dunbar  } else
722930b71a4a7dedf70a73e5fd875bae7df452b80a9Daniel Dunbar    IsMultiChar = false;
7234bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta
7244bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta  // Transfer the value from APInt to uint64_t
7254bc11af9bed1d4a247e3db1fcb754d410ad99099Sanjiv Gupta  Value = LitVal.getZExtValue();
7261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // If this is a single narrow character, sign extend it (e.g. '\xFF' is "-1")
7285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // if 'char' is signed for this target (C99 6.4.4.4p10).  Note that multiple
7295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // character constants are not sign extended in the this implementation:
7305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // '\xFF\xFF' = 65536 and '\x0\xFF' = 255, which matches GCC.
731e3ad881e4e9620e941dabd4e78dacdb028b85682Chris Lattner  if (!IsWide && NumCharsSoFar == 1 && (Value & 128) &&
73215b91764d08e886391c865c4a444d7b51141c284Eli Friedman      PP.getLangOptions().CharIsSigned)
7335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Value = (signed char)Value;
7345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
7355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       string-literal: [C99 6.4.5]
7385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///          " [s-char-sequence] "
7395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         L" [s-char-sequence] "
7405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       s-char-sequence:
7415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         s-char
7425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         s-char-sequence s-char
7435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       s-char:
7445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         any source character except the double quote ",
7455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///           backslash \, or newline character
7465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         escape-character
7475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         universal-character-name
7485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       escape-character: [C99 6.4.4.4]
7495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \ escape-code
7505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         universal-character-name
7515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       escape-code:
7525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         character-escape-code
7535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-escape-code
7545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hex-escape-code
7555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       character-escape-code: one of
7565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         n t b r f v a
7575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \ ' " ?
7585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       octal-escape-code:
7595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-digit
7605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-digit octal-digit
7615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-digit octal-digit octal-digit
7625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hex-escape-code:
7635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         x hex-digit
7645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hex-escape-code hex-digit
7655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       universal-character-name:
7665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \u hex-quad
7675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \U hex-quad hex-quad
7685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hex-quad:
7695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hex-digit hex-digit hex-digit hex-digit
7705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
7715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerStringLiteralParser::
772d217773f106856a11879ec79dc468efefaf2ee75Chris LattnerStringLiteralParser(const Token *StringToks, unsigned NumStringToks,
7736cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt                    Preprocessor &pp, bool Complain) : PP(pp) {
7745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Scan all of the string portions, remember the max individual token length,
7755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // computing a bound on the concatenated string length, and see whether any
7765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // piece is a wide-string.  If any of the string portions is a wide-string
7775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // literal, the result is a wide-string literal [C99 6.4.5p4].
7786cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  MaxTokenLength = StringToks[0].getLength();
7796cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  SizeBound = StringToks[0].getLength()-2;  // -2 for "".
78022f6bbcafa8871f4f20c4402d9cbc5c024fee99aChris Lattner  AnyWide = StringToks[0].is(tok::wide_string_literal);
7816cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt
7826cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  hadError = false;
7835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Implement Translation Phase #6: concatenation of string literals
7855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// (C99 5.1.1.2p1).  The common case is only one string fragment.
7865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  for (unsigned i = 1; i != NumStringToks; ++i) {
7875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // The string could be shorter than this if it needs cleaning, but this is a
7885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // reasonable bound, which is all we need.
7896cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt    SizeBound += StringToks[i].getLength()-2;  // -2 for "".
7901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Remember maximum string piece length.
7926cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt    if (StringToks[i].getLength() > MaxTokenLength)
7936cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt      MaxTokenLength = StringToks[i].getLength();
7941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
7955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Remember if we see any wide strings.
79622f6bbcafa8871f4f20c4402d9cbc5c024fee99aChris Lattner    AnyWide |= StringToks[i].is(tok::wide_string_literal);
7975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
798dbb1ecc32ca122b07b7c98fd0a8f6f53985adaccChris Lattner
7995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Include space for the null terminator.
8005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++SizeBound;
8011eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // TODO: K&R warning: "traditional C rejects string constant concatenation"
8031eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Get the width in bytes of wchar_t.  If no wchar_t strings are used, do not
8055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // query the target.  As such, wchar_tByteWidth is only valid if AnyWide=true.
8065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  wchar_tByteWidth = ~0U;
8075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (AnyWide) {
808bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner    wchar_tByteWidth = PP.getTargetInfo().getWCharWidth();
8095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    assert((wchar_tByteWidth & 7) == 0 && "Assumes wchar_t is byte multiple!");
8105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    wchar_tByteWidth /= 8;
8115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
8121eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // The output buffer size needs to be large enough to hold wide characters.
8145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // This is a worst-case assumption which basically corresponds to L"" "long".
8155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (AnyWide)
8165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    SizeBound *= wchar_tByteWidth;
8171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Size the temporary buffer to hold the result string data.
8195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ResultBuf.resize(SizeBound);
8201eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Likewise, but for each string piece.
8225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::SmallString<512> TokenBuf;
8235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  TokenBuf.resize(MaxTokenLength);
8241eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Loop over all the strings, getting their spelling, and expanding them to
8265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // wide strings as appropriate.
8275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ResultPtr = &ResultBuf[0];   // Next byte to fill in.
8281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
829ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson  Pascal = false;
8301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  for (unsigned i = 0, e = NumStringToks; i != e; ++i) {
8325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    const char *ThisTokBuf = &TokenBuf[0];
8335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Get the spelling of the token, which eliminates trigraphs, etc.  We know
8345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // that ThisTokBuf points to a buffer that is big enough for the whole token
8355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // and 'spelled' tokens can only shrink.
83650f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor    bool StringInvalid = false;
83750f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor    unsigned ThisTokLen = PP.getSpelling(StringToks[i], ThisTokBuf,
8386cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt                                         &StringInvalid);
83950f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor    if (StringInvalid) {
84050f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor      hadError = 1;
84150f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor      continue;
84250f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor    }
84350f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor
8445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    const char *ThisTokEnd = ThisTokBuf+ThisTokLen-1;  // Skip end quote.
84556bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    bool wide = false;
8465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // TODO: Input character set mapping support.
8471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Skip L marker for wide strings.
84956bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    if (ThisTokBuf[0] == 'L') {
85056bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian      wide = true;
8515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ++ThisTokBuf;
85256bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian    }
8531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    assert(ThisTokBuf[0] == '"' && "Expected quote, lexer broken?");
8555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ++ThisTokBuf;
8561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
857ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson    // Check if this is a pascal string
858ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson    if (pp.getLangOptions().PascalStrings && ThisTokBuf + 1 != ThisTokEnd &&
859ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        ThisTokBuf[0] == '\\' && ThisTokBuf[1] == 'p') {
8601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
861ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson      // If the \p sequence is found in the first token, we have a pascal string
862ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson      // Otherwise, if we already have a pascal string, ignore the first \p
863ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson      if (i == 0) {
864ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        ++ThisTokBuf;
865ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        Pascal = true;
866ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson      } else if (Pascal)
867ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        ThisTokBuf += 2;
868ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson    }
8691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    while (ThisTokBuf != ThisTokEnd) {
8715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      // Is this a span of non-escape characters?
8725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (ThisTokBuf[0] != '\\') {
8735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        const char *InStart = ThisTokBuf;
8745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        do {
8755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          ++ThisTokBuf;
8765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        } while (ThisTokBuf != ThisTokEnd && ThisTokBuf[0] != '\\');
8771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
8785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // Copy the character span over.
8795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        unsigned Len = ThisTokBuf-InStart;
8805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        if (!AnyWide) {
8815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          memcpy(ResultPtr, InStart, Len);
8825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          ResultPtr += Len;
8835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        } else {
8845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          // Note: our internal rep of wide char tokens is always little-endian.
8855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          for (; Len; --Len, ++InStart) {
8865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer            *ResultPtr++ = InStart[0];
8875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer            // Add zeros at the end.
8885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer            for (unsigned i = 1, e = wchar_tByteWidth; i != e; ++i)
8890e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff              *ResultPtr++ = 0;
8905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          }
8915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        }
8925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        continue;
8935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
8944e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff      // Is this a Universal Character Name escape?
8950e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff      if (ThisTokBuf[1] == 'u' || ThisTokBuf[1] == 'U') {
8961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        ProcessUCNEscape(ThisTokBuf, ThisTokEnd, ResultPtr,
89756bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian                         hadError, StringToks[i].getLocation(), PP, wide,
89856bedefe92ae8f604d14bea75cc3040ab32337c2Fariborz Jahanian                         Complain);
8994e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff        continue;
9004e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff      }
9014e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff      // Otherwise, this is a non-UCN escape character.  Process it.
9024e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff      unsigned ResultChar = ProcessCharEscape(ThisTokBuf, ThisTokEnd, hadError,
9034e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff                                              StringToks[i].getLocation(),
90437dd3ecbfc555c3ee0a1922f4bd5ab999c1e5461Chris Lattner                                              AnyWide, PP, Complain);
9051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9064e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff      // Note: our internal rep of wide char tokens is always little-endian.
9074e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff      *ResultPtr++ = ResultChar & 0xFF;
9081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
9094e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff      if (AnyWide) {
9104e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff        for (unsigned i = 1, e = wchar_tByteWidth; i != e; ++i)
9114e93b34fdb798abfa0534062a139f2c37cbf876eSteve Naroff          *ResultPtr++ = ResultChar >> i*8;
9125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
9135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
9145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
9151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
916bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner  if (Pascal) {
917ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson    ResultBuf[0] = ResultPtr-&ResultBuf[0]-1;
91864a8034226e6529a7ba65c005c846a70e17a1664Fariborz Jahanian    if (AnyWide)
91964a8034226e6529a7ba65c005c846a70e17a1664Fariborz Jahanian      ResultBuf[0] /= wchar_tByteWidth;
920bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner
921bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner    // Verify that pascal strings aren't too large.
922b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor    if (GetStringLength() > 256 && Complain) {
923bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner      PP.Diag(StringToks[0].getLocation(), diag::err_pascal_string_too_long)
924bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner        << SourceRange(StringToks[0].getLocation(),
925bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner                       StringToks[NumStringToks-1].getLocation());
92657d7dde770c67b282e7fb77b1b81e429910937b3Eli Friedman      hadError = 1;
92757d7dde770c67b282e7fb77b1b81e429910937b3Eli Friedman      return;
92857d7dde770c67b282e7fb77b1b81e429910937b3Eli Friedman    }
929427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor  } else if (Complain) {
930427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor    // Complain if this string literal has too many characters.
931427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor    unsigned MaxChars = PP.getLangOptions().CPlusPlus? 65536
932427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor                      : PP.getLangOptions().C99 ? 4095
933427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor                      : 509;
934427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor
935427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor    if (GetNumStringChars() > MaxChars)
936427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor      PP.Diag(StringToks[0].getLocation(), diag::ext_string_too_long)
937427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor        << GetNumStringChars() << MaxChars
938427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor        << (PP.getLangOptions().CPlusPlus? 2
939427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor            : PP.getLangOptions().C99 ? 1
940427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor            : 0)
941427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor        << SourceRange(StringToks[0].getLocation(),
942427c492d368d6ecf409fa8053eecb5cd0e779c5bDouglas Gregor                       StringToks[NumStringToks-1].getLocation());
943bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner  }
9445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
945719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
946719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
947719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// getOffsetOfStringByte - This function returns the offset of the
948719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// specified byte of the string data represented by Token.  This handles
949719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// advancing over escape sequences in the string.
950719e61573f27c11057ecfe0dd8f141621602c571Chris Lattnerunsigned StringLiteralParser::getOffsetOfStringByte(const Token &Tok,
951719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner                                                    unsigned ByteNo,
952b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor                                                    Preprocessor &PP,
953b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor                                                    bool Complain) {
954719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  // Get the spelling of the token.
955719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  llvm::SmallString<16> SpellingBuffer;
9566cf750298d3621d8a10a6dd07fcee8e274b9d94dSean Hunt  SpellingBuffer.resize(Tok.getLength());
9571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
95850f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor  bool StringInvalid = false;
959719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  const char *SpellingPtr = &SpellingBuffer[0];
96050f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor  unsigned TokLen = PP.getSpelling(Tok, SpellingPtr, &StringInvalid);
96150f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor  if (StringInvalid) {
96250f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor    return 0;
96350f6af7a6d6951a63f3da7d4c5a7d3965bf73b63Douglas Gregor  }
964719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
965719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  assert(SpellingPtr[0] != 'L' && "Doesn't handle wide strings yet");
966719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
9671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
968719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  const char *SpellingStart = SpellingPtr;
969719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  const char *SpellingEnd = SpellingPtr+TokLen;
970719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
971719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  // Skip over the leading quote.
972719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  assert(SpellingPtr[0] == '"' && "Should be a string literal!");
973719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  ++SpellingPtr;
9741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
975719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  // Skip over bytes until we find the offset we're looking for.
976719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  while (ByteNo) {
977719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    assert(SpellingPtr < SpellingEnd && "Didn't find byte offset!");
9781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
979719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    // Step over non-escapes simply.
980719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    if (*SpellingPtr != '\\') {
981719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      ++SpellingPtr;
982719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      --ByteNo;
983719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      continue;
984719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    }
9851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
986719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    // Otherwise, this is an escape character.  Advance over it.
987719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    bool HadError = false;
988719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    ProcessCharEscape(SpellingPtr, SpellingEnd, HadError,
989b90f4b3fb94056609da9cca5eef7358d95a363b2Douglas Gregor                      Tok.getLocation(), false, PP, Complain);
990719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    assert(!HadError && "This method isn't valid on erroneous strings");
991719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    --ByteNo;
992719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  }
9931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
994719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  return SpellingPtr-SpellingStart;
995719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner}
996