LiteralSupport.cpp revision 8a5c0cd90b8d607ca284274000ed8716b836d253
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"
195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer#include "llvm/ADT/StringExtras.h"
205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerusing namespace clang;
215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// HexDigitValue - Return the value of the specified hex digit, or -1 if it's
235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// not valid.
245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic int HexDigitValue(char C) {
255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= '0' && C <= '9') return C-'0';
265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= 'a' && C <= 'f') return C-'a'+10;
275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (C >= 'A' && C <= 'F') return C-'A'+10;
285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return -1;
295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// ProcessCharEscape - Parse a standard C escape sequence, which can occur in
325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// either a character or a string literal.
335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerstatic unsigned ProcessCharEscape(const char *&ThisTokBuf,
345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                                  const char *ThisTokEnd, bool &HadError,
355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                                  SourceLocation Loc, bool IsWide,
365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                                  Preprocessor &PP) {
375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Skip the '\' char.
385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++ThisTokBuf;
395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // We know that this character can't be off the end of the buffer, because
415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // that would have been \", which would not have been the end of string.
425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  unsigned ResultChar = *ThisTokBuf++;
435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  switch (ResultChar) {
445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // These map to themselves.
455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '\\': case '\'': case '"': case '?': break;
465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // These have fixed mappings.
485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'a':
495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // TODO: K&R: the meaning of '\\a' is different in traditional C
505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 7;
515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'b':
535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 8;
545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'e':
56204b2fed909b1eabea7aeb6caadd7cff718edee5Chris Lattner    PP.Diag(Loc, diag::ext_nonstandard_escape) << "e";
575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 27;
585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'f':
605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 12;
615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'n':
635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 10;
645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'r':
665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 13;
675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 't':
695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 9;
705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'v':
725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 11;
735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case 'x': { // Hex escape.
755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 0;
765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (ThisTokBuf == ThisTokEnd || !isxdigit(*ThisTokBuf)) {
775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      PP.Diag(Loc, diag::err_hex_escape_no_digits);
785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      HadError = 1;
795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      break;
805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Hex escapes are a maximal series of hex digits.
835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    bool Overflow = false;
845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    for (; ThisTokBuf != ThisTokEnd; ++ThisTokBuf) {
855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      int CharVal = HexDigitValue(ThisTokBuf[0]);
865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (CharVal == -1) break;
87c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner      // About to shift out a digit?
88c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner      Overflow |= (ResultChar & 0xF0000000) ? true : false;
895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar <<= 4;
905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar |= CharVal;
915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // See if any bits will be truncated when evaluated as a character.
9498be4943e8dc4f3905629a7102668960873cf863Chris Lattner    unsigned CharWidth = PP.getTargetInfo().getCharWidth(IsWide);
959c728dc4d8da89c73fcae74c9e72d7a83ffd7b6dTed Kremenek
965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) {
975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      Overflow = true;
985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar &= ~0U >> (32-CharWidth);
995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Check for overflow.
1025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (Overflow)   // Too many digits to fit in
1035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      PP.Diag(Loc, diag::warn_hex_escape_too_large);
1045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '0': case '1': case '2': case '3':
1075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '4': case '5': case '6': case '7': {
1085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Octal escapes.
1095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    --ThisTokBuf;
1105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ResultChar = 0;
1115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Octal escapes are a series of octal digits with maximum length 3.
1135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // "\0123" is a two digit sequence equal to "\012" "3".
1145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    unsigned NumDigits = 0;
1155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    do {
1165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar <<= 3;
1175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar |= *ThisTokBuf++ - '0';
1185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ++NumDigits;
1195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    } while (ThisTokBuf != ThisTokEnd && NumDigits < 3 &&
1205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer             ThisTokBuf[0] >= '0' && ThisTokBuf[0] <= '7');
1215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Check for overflow.  Reject '\777', but not L'\777'.
12398be4943e8dc4f3905629a7102668960873cf863Chris Lattner    unsigned CharWidth = PP.getTargetInfo().getCharWidth(IsWide);
1249c728dc4d8da89c73fcae74c9e72d7a83ffd7b6dTed Kremenek
1255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) {
1265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      PP.Diag(Loc, diag::warn_octal_escape_too_large);
1275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar &= ~0U >> (32-CharWidth);
1285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Otherwise, these are not valid escapes.
1335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  case '(': case '{': case '[': case '%':
1345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // GCC accepts these as extensions.  We warn about them as such though.
1355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (!PP.getLangOptions().NoExtensions) {
136204b2fed909b1eabea7aeb6caadd7cff718edee5Chris Lattner      PP.Diag(Loc, diag::ext_nonstandard_escape)
137204b2fed909b1eabea7aeb6caadd7cff718edee5Chris Lattner        << std::string()+(char)ResultChar;
1385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      break;
1395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
1405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // FALL THROUGH.
1415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  default:
142ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    if (isgraph(ThisTokBuf[0]))
143204b2fed909b1eabea7aeb6caadd7cff718edee5Chris Lattner      PP.Diag(Loc, diag::ext_unknown_escape) << std::string()+(char)ResultChar;
144ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    else
145204b2fed909b1eabea7aeb6caadd7cff718edee5Chris Lattner      PP.Diag(Loc, diag::ext_unknown_escape) << "x"+llvm::utohexstr(ResultChar);
1465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    break;
1475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
1485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return ResultChar;
1505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
1515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
1520e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// ProcessUCNEscape - Read the Universal Character Name, check constraints and
1530e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// convert the UTF32 to UTF8. This is a subroutine of StringLiteralParser.
1540e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// When we decide to implement UCN's for character constants and identifiers,
1550e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff/// we will likely rework our support for UCN's.
1560e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroffstatic void ProcessUCNEscape(const char *&ThisTokBuf, const char *ThisTokEnd,
1578a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff                             char *&ResultBuf, bool &HadError,
1588a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff                             SourceLocation Loc, bool IsWide, Preprocessor &PP)
1598a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff{
1600e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // FIXME: Add a warning - UCN's are only valid in C++ & C99.
1618a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff  // FIXME: Handle wide strings.
1620e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
1630e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Skip the '\u' char's.
1640e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ThisTokBuf += 2;
1650e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
1660e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if (ThisTokBuf == ThisTokEnd || !isxdigit(*ThisTokBuf)) {
1670e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    PP.Diag(Loc, diag::err_ucn_escape_no_digits);
1680e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    HadError = 1;
1690e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    return;
1700e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
1710e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  typedef unsigned int UTF32;
1720e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
1730e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  UTF32 UcnVal = 0;
1740e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  unsigned short UcnLen = (ThisTokBuf[-1] == 'u' ? 4 : 8);
1750e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  for (; ThisTokBuf != ThisTokEnd && UcnLen; ++ThisTokBuf, UcnLen--) {
1760e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    int CharVal = HexDigitValue(ThisTokBuf[0]);
1770e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    if (CharVal == -1) break;
1780e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    UcnVal <<= 4;
1790e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    UcnVal |= CharVal;
1800e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
1810e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // If we didn't consume the proper number of digits, there is a problem.
1820e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if (UcnLen) {
1830e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    PP.Diag(Loc, diag::err_ucn_escape_incomplete);
1840e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    HadError = 1;
1850e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    return;
1860e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
1878a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff  // Check UCN constraints (C99 6.4.3p2).
1880e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if ((UcnVal < 0xa0 &&
1890e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff      (UcnVal != 0x24 && UcnVal != 0x40 && UcnVal != 0x60 )) // $, @, `
1908a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff      || (UcnVal >= 0xD800 && UcnVal <= 0xDFFF)
1918a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff      || (UcnVal > 0x10FFFF)) /* the maximum legal UTF32 value */ {
1920e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    PP.Diag(Loc, diag::err_ucn_escape_invalid);
1930e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    HadError = 1;
1940e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    return;
1950e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
1960e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Now that we've parsed/checked the UCN, we convert from UTF32->UTF8.
1970e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // The conversion below was inspired by:
1980e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  //   http://www.unicode.org/Public/PROGRAMS/CVTUTF/ConvertUTF.c
1990e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // First, we determine how many bytes the result will require.
2000e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  typedef unsigned char UTF8;
2010e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
2020e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  unsigned short bytesToWrite = 0;
2030e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  if (UcnVal < (UTF32)0x80)
2040e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 1;
2050e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else if (UcnVal < (UTF32)0x800)
2060e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 2;
2070e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else if (UcnVal < (UTF32)0x10000)
2080e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 3;
2090e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  else
2100e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    bytesToWrite = 4;
2110e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
2120e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  const unsigned byteMask = 0xBF;
2130e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  const unsigned byteMark = 0x80;
2140e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
2150e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Once the bits are split out into bytes of UTF8, this is a mask OR-ed
2168a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff  // into the first byte, depending on how many bytes follow.
2178a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff  static const UTF8 firstByteMark[5] = {
2188a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff    0x00, 0x00, 0xC0, 0xE0, 0xF0
2190e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  };
2200e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Finally, we write the bytes into ResultBuf.
2210e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ResultBuf += bytesToWrite;
2220e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  switch (bytesToWrite) { // note: everything falls through.
2230e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 4: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
2240e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 3: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
2250e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 2: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
2260e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff    case 1: *--ResultBuf = (UTF8) (UcnVal | firstByteMark[bytesToWrite]);
2270e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  }
2280e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  // Update the buffer.
2290e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff  ResultBuf += bytesToWrite;
2300e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff}
2315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       integer-constant: [C99 6.4.4.1]
2345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         decimal-constant integer-suffix
2355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-constant integer-suffix
2365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-constant integer-suffix
2375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       decimal-constant:
2385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         nonzero-digit
2395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         decimal-constant digit
2405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       octal-constant:
2415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0
2425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-constant octal-digit
2435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hexadecimal-constant:
2445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-prefix hexadecimal-digit
2455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hexadecimal-constant hexadecimal-digit
2465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hexadecimal-prefix: one of
2475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0x 0X
2485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       integer-suffix:
2495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         unsigned-suffix [long-suffix]
2505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         unsigned-suffix [long-long-suffix]
2515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         long-suffix [unsigned-suffix]
2525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         long-long-suffix [unsigned-sufix]
2535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       nonzero-digit:
2545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         1 2 3 4 5 6 7 8 9
2555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       octal-digit:
2565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0 1 2 3 4 5 6 7
2575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hexadecimal-digit:
2585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         0 1 2 3 4 5 6 7 8 9
2595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         a b c d e f
2605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         A B C D E F
2615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       unsigned-suffix: one of
2625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         u U
2635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       long-suffix: one of
2645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         l L
2655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       long-long-suffix: one of
2665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         ll LL
2675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
2685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       floating-constant: [C99 6.4.4.2]
2695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         TODO: add rules...
2705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
2715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerNumericLiteralParser::
2725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerNumericLiteralParser(const char *begin, const char *end,
2735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                     SourceLocation TokLoc, Preprocessor &pp)
2745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  : PP(pp), ThisTokBegin(begin), ThisTokEnd(end) {
275c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner
276c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // This routine assumes that the range begin/end matches the regex for integer
277c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // and FP constants (specifically, the 'pp-number' regex), and assumes that
278c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // the byte at "*end" is both valid and not part of the regex.  Because of
279c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  // this, it doesn't have to check for 'overscan' in various places.
280c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner  assert(!isalnum(*end) && *end != '.' && *end != '_' &&
281c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner         "Lexer didn't maximally munch?");
282c29bbde0a14a664d6843b21d3791478d1f4d2833Chris Lattner
2835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  s = DigitsBegin = begin;
2845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  saw_exponent = false;
2855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  saw_period = false;
2865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isLong = false;
2875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isUnsigned = false;
2885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  isLongLong = false;
2896e400c286b485e28d04a742ea87860ddfefa672eChris Lattner  isFloat = false;
290506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  isImaginary = false;
2915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  hadError = false;
2925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
2935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (*s == '0') { // parse radix
294368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    ParseNumberStartingWithZero(TokLoc);
295368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (hadError)
296368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      return;
2975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  } else { // the first digit is non-zero
2985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    radix = 10;
2995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    s = SkipDigits(s);
3005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (s == ThisTokEnd) {
3015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      // Done.
302016765e3453db2e302efe53905e99cdb25501234Christopher Lamb    } else if (isxdigit(*s) && !(*s == 'e' || *s == 'E')) {
303ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
304ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner              diag::err_invalid_decimal_digit) << std::string(s, s+1);
305ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
3065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      return;
3075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    } else if (*s == '.') {
3085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s++;
3095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      saw_period = true;
3105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s = SkipDigits(s);
3115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
3124411f46050216a139ab2fc7ff145ec384d11ec7fChris Lattner    if ((*s == 'e' || *s == 'E')) { // exponent
31370f66ab053f36ab3df7a778d09bcb2b4b0fec1f8Chris Lattner      const char *Exponent = s;
3145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      s++;
3155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      saw_exponent = true;
3165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (*s == '+' || *s == '-')  s++; // sign
3175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      const char *first_non_digit = SkipDigits(s);
3180b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner      if (first_non_digit != s) {
3195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        s = first_non_digit;
3200b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner      } else {
321ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-begin),
322ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                diag::err_exponent_has_no_digits);
323ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        hadError = true;
3240b7f69d789ca1f76582ee9a336e25861fd0c1416Chris Lattner        return;
3255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
3265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
3275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
3285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
3295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SuffixBegin = s;
330506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner
331506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // Parse the suffix.  At this point we can classify whether we have an FP or
332506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // integer constant.
333506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  bool isFPConstant = isFloatingLiteral();
334506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner
335506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // Loop over all of the characters of the suffix.  If we see something bad,
336506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // we break out of the loop.
337506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  for (; s != ThisTokEnd; ++s) {
338506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    switch (*s) {
339506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'f':      // FP Suffix for "float"
340506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'F':
341506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (!isFPConstant) break;  // Error for integer constant.
3426e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      if (isFloat || isLong) break; // FF, LF invalid.
3436e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      isFloat = true;
344506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
345506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'u':
346506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'U':
347506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isFPConstant) break;  // Error for floating constant.
348506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isUnsigned) break;    // Cannot be repeated.
349506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      isUnsigned = true;
350506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
351506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'l':
352506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'L':
353506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isLong || isLongLong) break;  // Cannot be repeated.
3546e400c286b485e28d04a742ea87860ddfefa672eChris Lattner      if (isFloat) break;               // LF invalid.
355506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner
356506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      // Check for long long.  The L's need to be adjacent and the same case.
357506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (s+1 != ThisTokEnd && s[1] == s[0]) {
358506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        if (isFPConstant) break;        // long long invalid for floats.
359506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        isLongLong = true;
360506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner        ++s;  // Eat both of them.
361506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      } else {
3625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        isLong = true;
3635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
364506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
365506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'i':
3660c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff      if (PP.getLangOptions().Microsoft) {
3670c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        // Allow i8, i16, i32, i64, and i128.
3680c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        if (++s == ThisTokEnd) break;
3690c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        switch (*s) {
3700c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff          case '8':
3710c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            s++; // i8 suffix
3720c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            break;
3730c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff          case '1':
3740c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            if (++s == ThisTokEnd) break;
3750c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            if (*s == '6') s++; // i16 suffix
3760c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            else if (*s == '2') {
3770c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff              if (++s == ThisTokEnd) break;
3780c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff              if (*s == '8') s++; // i128 suffix
3790c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            }
3800c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            break;
3810c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff          case '3':
3820c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            if (++s == ThisTokEnd) break;
3830c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            if (*s == '2') s++; // i32 suffix
3840c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            break;
3850c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff          case '6':
3860c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            if (++s == ThisTokEnd) break;
3870c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            if (*s == '4') s++; // i64 suffix
3880c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            break;
3890c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff          default:
3900c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff            break;
3910c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        }
3920c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff        break;
3930c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff      }
3940c29b22f4384500cc0d04f3072cc5d5d58d10d6cSteve Naroff      // fall through.
395506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'I':
396506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'j':
397506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    case 'J':
398506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      if (isImaginary) break;   // Cannot be repeated.
399506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
400506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner              diag::ext_imaginary_constant);
401506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      isImaginary = true;
402506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner      continue;  // Success.
4035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
404506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    // If we reached here, there was an error.
405506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    break;
406506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  }
407506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner
408506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  // Report an error if there are any.
409506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner  if (s != ThisTokEnd) {
410ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
411ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner            isFPConstant ? diag::err_invalid_suffix_float_constant :
412ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                           diag::err_invalid_suffix_integer_constant)
413ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      << std::string(SuffixBegin, ThisTokEnd);
414ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    hadError = true;
415506b8dec4ed3db3c60bf9e0dd37901f0cf3d6749Chris Lattner    return;
4165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
4175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
4185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
419368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// ParseNumberStartingWithZero - This method is called when the first character
420368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// of the number is found to be a zero.  This means it is either an octal
421368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// number (like '04') or a hex number ('0x123a') a binary number ('0b1010') or
422368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// a floating point number (01239.123e4).  Eat the prefix, determining the
423368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner/// radix etc.
424368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattnervoid NumericLiteralParser::ParseNumberStartingWithZero(SourceLocation TokLoc) {
425368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  assert(s[0] == '0' && "Invalid method call");
426368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  s++;
427368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
428368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // Handle a hex number like 0x1234.
429368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if ((*s == 'x' || *s == 'X') && (isxdigit(s[1]) || s[1] == '.')) {
430368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
431368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 16;
432368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    DigitsBegin = s;
433368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s = SkipHexDigits(s);
434368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (s == ThisTokEnd) {
435368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      // Done.
436368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (*s == '.') {
437368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s++;
438368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      saw_period = true;
439368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s = SkipHexDigits(s);
440368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
441368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    // A binary exponent can appear with or with a '.'. If dotted, the
442368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    // binary exponent is required.
4436ea623823f8532670480425b573f35115404b4a0Chris Lattner    if (*s == 'p' || *s == 'P') {
444368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      const char *Exponent = s;
445368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s++;
446368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      saw_exponent = true;
447368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      if (*s == '+' || *s == '-')  s++; // sign
448368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      const char *first_non_digit = SkipDigits(s);
4496ea623823f8532670480425b573f35115404b4a0Chris Lattner      if (first_non_digit == s) {
450ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin),
451ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner                diag::err_exponent_has_no_digits);
452ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        hadError = true;
4536ea623823f8532670480425b573f35115404b4a0Chris Lattner        return;
454368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      }
4556ea623823f8532670480425b573f35115404b4a0Chris Lattner      s = first_non_digit;
4566ea623823f8532670480425b573f35115404b4a0Chris Lattner
4574984212fced6d9c6538ebb7319839105e3880b45Chris Lattner      if (!PP.getLangOptions().HexFloats)
458ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner        PP.Diag(TokLoc, diag::ext_hexconstant_invalid);
459368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (saw_period) {
460ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
461ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner              diag::err_hexconstant_requires_exponent);
462ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
463368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
464368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
465368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
466368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
467368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // Handle simple binary numbers 0b01010
468368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == 'b' || *s == 'B') {
469368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    // 0b101010 is a GCC extension.
470413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    PP.Diag(TokLoc, diag::ext_binary_literal);
471368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    ++s;
472368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 2;
473368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    DigitsBegin = s;
474368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s = SkipBinaryDigits(s);
475368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (s == ThisTokEnd) {
476368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      // Done.
477368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else if (isxdigit(*s)) {
478ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
479ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner              diag::err_invalid_binary_digit) << std::string(s, s+1);
480ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
481368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
482413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    // Other suffixes will be diagnosed by the caller.
483368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
484368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
485368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
486368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // For now, the radix is set to 8. If we discover that we have a
487368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // floating point constant, the radix will change to 10. Octal floating
488368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  // point constants are not permitted (only decimal and hexadecimal).
489368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  radix = 8;
490368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  DigitsBegin = s;
491368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  s = SkipOctalDigits(s);
492368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (s == ThisTokEnd)
493368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return; // Done, simple octal number like 01234
494368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
495413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // If we have some other non-octal digit that *is* a decimal digit, see if
496413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // this is part of a floating point number like 094.123 or 09e1.
497413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  if (isdigit(*s)) {
498413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    const char *EndDecimal = SkipDigits(s);
499413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    if (EndDecimal[0] == '.' || EndDecimal[0] == 'e' || EndDecimal[0] == 'E') {
500413d355e38755a71f106dbc0ac900ca989070916Chris Lattner      s = EndDecimal;
501413d355e38755a71f106dbc0ac900ca989070916Chris Lattner      radix = 10;
502413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    }
503413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  }
504413d355e38755a71f106dbc0ac900ca989070916Chris Lattner
505413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // If we have a hex digit other than 'e' (which denotes a FP exponent) then
506413d355e38755a71f106dbc0ac900ca989070916Chris Lattner  // the code is using an incorrect base.
507368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (isxdigit(*s) && *s != 'e' && *s != 'E') {
508ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
509ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner            diag::err_invalid_octal_digit) << std::string(s, s+1);
510ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner    hadError = true;
511368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    return;
512368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
513368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
514368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == '.') {
515368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
516368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 10;
517368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    saw_period = true;
518413d355e38755a71f106dbc0ac900ca989070916Chris Lattner    s = SkipDigits(s); // Skip suffix.
519368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
520368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  if (*s == 'e' || *s == 'E') { // exponent
521368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    const char *Exponent = s;
522368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    s++;
523368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    radix = 10;
524368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    saw_exponent = true;
525368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (*s == '+' || *s == '-')  s++; // sign
526368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    const char *first_non_digit = SkipDigits(s);
527368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    if (first_non_digit != s) {
528368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      s = first_non_digit;
529368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    } else {
530ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin),
531ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner              diag::err_exponent_has_no_digits);
532ac92d829111bc19d1cc97cd85c3c04bc39b969d1Chris Lattner      hadError = true;
533368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner      return;
534368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner    }
535368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner  }
536368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner}
537368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
538368328c88bd46f471bbf85f05438b4f2eb95df5bChris Lattner
5395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// GetIntegerValue - Convert this numeric literal value to an APInt that
5405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// matches Val's input width.  If there is an overflow, set Val to the low bits
5415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer/// of the result and return true.  Otherwise, return false.
5425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencerbool NumericLiteralParser::GetIntegerValue(llvm::APInt &Val) {
543a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // Fast path: Compute a conservative bound on the maximum number of
544a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // bits per digit in this radix. If we can't possibly overflow a
545a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // uint64 based on that bound then do the simple conversion to
546a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // integer. This avoids the expensive overflow checking below, and
547a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // handles the common cases that matter (small decimal integers and
548a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  // hex/octal values which don't overflow).
549a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  unsigned MaxBitsPerDigit = 1;
550a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  while ((1U << MaxBitsPerDigit) < radix)
551a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    MaxBitsPerDigit += 1;
552a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  if ((SuffixBegin - DigitsBegin) * MaxBitsPerDigit <= 64) {
553a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    uint64_t N = 0;
554a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    for (s = DigitsBegin; s != SuffixBegin; ++s)
555a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar      N = N*radix + HexDigitValue(*s);
556a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar
557a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    // This will truncate the value to Val's input width. Simply check
558a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    // for overflow by comparing.
559a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    Val = N;
560a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar    return Val.getZExtValue() != N;
561a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar  }
562a179be34c1a3c0190e6b9e39dee2197651f44a5dDaniel Dunbar
5635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  Val = 0;
5645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  s = DigitsBegin;
5655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt RadixVal(Val.getBitWidth(), radix);
5675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt CharVal(Val.getBitWidth(), 0);
5685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::APInt OldVal = Val;
5695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool OverflowOccurred = false;
5715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  while (s < SuffixBegin) {
5725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    unsigned C = HexDigitValue(*s++);
5735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // If this letter is out of bound for this radix, reject it.
5755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    assert(C < radix && "NumericLiteralParser ctor should have rejected this");
5765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    CharVal = C;
5785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Add the digit to the value in the appropriate radix.  If adding in digits
5805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // made the value smaller, then this overflowed.
5815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OldVal = Val;
5825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Multiply by radix, did overflow occur on the multiply?
5845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Val *= RadixVal;
5855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OverflowOccurred |= Val.udiv(RadixVal) != OldVal;
5865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
5875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Add value, did overflow occur on the value?
588d70cb645702bdbb42aee58403306a7c47e0d901cDaniel Dunbar    //   (a + b) ult b  <=> overflow
5895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Val += CharVal;
5905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    OverflowOccurred |= Val.ult(CharVal);
5915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
5925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  return OverflowOccurred;
5935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
5945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
595525a05093a4816af961fe2bc6b8a81c17e2e26c2Chris Lattnerllvm::APFloat NumericLiteralParser::
596427d5af5b601985093b6b4b33ba1e30fc24d86dcTed KremenekGetFloatValue(const llvm::fltSemantics &Format, bool* isExact) {
597427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek  using llvm::APFloat;
598427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek
59932e61bf91fba3c7946b9f71918202e5e55859a5cTed Kremenek  llvm::SmallVector<char,256> floatChars;
60032e61bf91fba3c7946b9f71918202e5e55859a5cTed Kremenek  for (unsigned i = 0, n = ThisTokEnd-ThisTokBegin; i != n; ++i)
60132e61bf91fba3c7946b9f71918202e5e55859a5cTed Kremenek    floatChars.push_back(ThisTokBegin[i]);
60232e61bf91fba3c7946b9f71918202e5e55859a5cTed Kremenek
60332e61bf91fba3c7946b9f71918202e5e55859a5cTed Kremenek  floatChars.push_back('\0');
60432e61bf91fba3c7946b9f71918202e5e55859a5cTed Kremenek
605427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek  APFloat V (Format, APFloat::fcZero, false);
606427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek  APFloat::opStatus status;
60732e61bf91fba3c7946b9f71918202e5e55859a5cTed Kremenek
60832e61bf91fba3c7946b9f71918202e5e55859a5cTed Kremenek  status = V.convertFromString(&floatChars[0],APFloat::rmNearestTiesToEven);
609427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek
610427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek  if (isExact)
611427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek    *isExact = status == APFloat::opOK;
612427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek
613427d5af5b601985093b6b4b33ba1e30fc24d86dcTed Kremenek  return V;
6145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
6155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerCharLiteralParser::CharLiteralParser(const char *begin, const char *end,
6185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer                                     SourceLocation Loc, Preprocessor &PP) {
6195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // At this point we know that the character matches the regex "L?'.*'".
6205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  HadError = false;
6215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  Value = 0;
6225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Determine if this is a wide character.
6245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  IsWide = begin[0] == 'L';
6255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (IsWide) ++begin;
6265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Skip over the entry quote.
6285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  assert(begin[0] == '\'' && "Invalid token lexed");
6295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++begin;
6305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // FIXME: This assumes that 'int' is 32-bits in overflow calculation, and the
6325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // size of "value".
63398be4943e8dc4f3905629a7102668960873cf863Chris Lattner  assert(PP.getTargetInfo().getIntWidth() == 32 &&
6345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer         "Assumes sizeof(int) == 4 for now");
6355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // FIXME: This assumes that wchar_t is 32-bits for now.
63698be4943e8dc4f3905629a7102668960873cf863Chris Lattner  assert(PP.getTargetInfo().getWCharWidth() == 32 &&
6375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer         "Assumes sizeof(wchar_t) == 4 for now");
6385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // FIXME: This extensively assumes that 'char' is 8-bits.
63998be4943e8dc4f3905629a7102668960873cf863Chris Lattner  assert(PP.getTargetInfo().getCharWidth() == 8 &&
6405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer         "Assumes char is 8 bits");
6415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool isFirstChar = true;
6435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  bool isMultiChar = false;
6445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  while (begin[0] != '\'') {
6455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    unsigned ResultChar;
6465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (begin[0] != '\\')     // If this is a normal character, consume it.
6475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar = *begin++;
6485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    else                      // Otherwise, this is an escape character.
6495f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ResultChar = ProcessCharEscape(begin, end, HadError, Loc, IsWide, PP);
6505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6515f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // If this is a multi-character constant (e.g. 'abc'), handle it.  These are
6525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // implementation defined (C99 6.4.4.4p10).
6535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (!isFirstChar) {
6545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      // If this is the second character being processed, do special handling.
6555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (!isMultiChar) {
6565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        isMultiChar = true;
6575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // Warn about discarding the top bits for multi-char wide-character
6595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // constants (L'abcd').
6605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        if (IsWide)
6615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          PP.Diag(Loc, diag::warn_extraneous_wide_char_constant);
6625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
6635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (IsWide) {
6655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // Emulate GCC's (unintentional?) behavior: L'ab' -> L'b'.
6665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        Value = 0;
6675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      } else {
6685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // Narrow character literals act as though their value is concatenated
6695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // in this implementation.
6705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        if (((Value << 8) >> 8) != Value)
6715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          PP.Diag(Loc, diag::warn_char_constant_too_large);
6725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        Value <<= 8;
6735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
6745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
6755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Value += ResultChar;
6775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    isFirstChar = false;
6785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
6795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // If this is a single narrow character, sign extend it (e.g. '\xFF' is "-1")
6815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // if 'char' is signed for this target (C99 6.4.4.4p10).  Note that multiple
6825f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // character constants are not sign extended in the this implementation:
6835f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // '\xFF\xFF' = 65536 and '\x0\xFF' = 255, which matches GCC.
6845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (!IsWide && !isMultiChar && (Value & 128) &&
68598be4943e8dc4f3905629a7102668960873cf863Chris Lattner      PP.getTargetInfo().isCharSigned())
6865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    Value = (signed char)Value;
6875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
6885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
6905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       string-literal: [C99 6.4.5]
6915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///          " [s-char-sequence] "
6925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         L" [s-char-sequence] "
6935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       s-char-sequence:
6945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         s-char
6955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         s-char-sequence s-char
6965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       s-char:
6975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         any source character except the double quote ",
6985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///           backslash \, or newline character
6995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         escape-character
7005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         universal-character-name
7015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       escape-character: [C99 6.4.4.4]
7025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \ escape-code
7035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         universal-character-name
7045f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       escape-code:
7055f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         character-escape-code
7065f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-escape-code
7075f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hex-escape-code
7085f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       character-escape-code: one of
7095f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         n t b r f v a
7105f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \ ' " ?
7115f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       octal-escape-code:
7125f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-digit
7135f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-digit octal-digit
7145f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         octal-digit octal-digit octal-digit
7155f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hex-escape-code:
7165f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         x hex-digit
7175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hex-escape-code hex-digit
7185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       universal-character-name:
7195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \u hex-quad
7205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         \U hex-quad hex-quad
7215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///       hex-quad:
7225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///         hex-digit hex-digit hex-digit hex-digit
7235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer///
7245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid SpencerStringLiteralParser::
725d217773f106856a11879ec79dc468efefaf2ee75Chris LattnerStringLiteralParser(const Token *StringToks, unsigned NumStringToks,
726bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner                    Preprocessor &pp) : PP(pp) {
7275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Scan all of the string portions, remember the max individual token length,
7285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // computing a bound on the concatenated string length, and see whether any
7295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // piece is a wide-string.  If any of the string portions is a wide-string
7305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // literal, the result is a wide-string literal [C99 6.4.5p4].
7315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  MaxTokenLength = StringToks[0].getLength();
7325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  SizeBound = StringToks[0].getLength()-2;  // -2 for "".
73322f6bbcafa8871f4f20c4402d9cbc5c024fee99aChris Lattner  AnyWide = StringToks[0].is(tok::wide_string_literal);
7345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  hadError = false;
7365f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Implement Translation Phase #6: concatenation of string literals
7385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  /// (C99 5.1.1.2p1).  The common case is only one string fragment.
7395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  for (unsigned i = 1; i != NumStringToks; ++i) {
7405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // The string could be shorter than this if it needs cleaning, but this is a
7415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // reasonable bound, which is all we need.
7425f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    SizeBound += StringToks[i].getLength()-2;  // -2 for "".
7435f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7445f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Remember maximum string piece length.
7455f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (StringToks[i].getLength() > MaxTokenLength)
7465f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      MaxTokenLength = StringToks[i].getLength();
7475f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7485f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Remember if we see any wide strings.
74922f6bbcafa8871f4f20c4402d9cbc5c024fee99aChris Lattner    AnyWide |= StringToks[i].is(tok::wide_string_literal);
7505f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
751dbb1ecc32ca122b07b7c98fd0a8f6f53985adaccChris Lattner
7525f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Include space for the null terminator.
7535f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ++SizeBound;
7545f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7555f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // TODO: K&R warning: "traditional C rejects string constant concatenation"
7565f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7575f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Get the width in bytes of wchar_t.  If no wchar_t strings are used, do not
7585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // query the target.  As such, wchar_tByteWidth is only valid if AnyWide=true.
7595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  wchar_tByteWidth = ~0U;
7605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (AnyWide) {
761bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner    wchar_tByteWidth = PP.getTargetInfo().getWCharWidth();
7625f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    assert((wchar_tByteWidth & 7) == 0 && "Assumes wchar_t is byte multiple!");
7635f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    wchar_tByteWidth /= 8;
7645f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
7655f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7665f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // The output buffer size needs to be large enough to hold wide characters.
7675f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // This is a worst-case assumption which basically corresponds to L"" "long".
7685f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  if (AnyWide)
7695f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    SizeBound *= wchar_tByteWidth;
7705f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7715f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Size the temporary buffer to hold the result string data.
7725f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ResultBuf.resize(SizeBound);
7735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7745f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Likewise, but for each string piece.
7755f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  llvm::SmallString<512> TokenBuf;
7765f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  TokenBuf.resize(MaxTokenLength);
7775f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7785f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // Loop over all the strings, getting their spelling, and expanding them to
7795f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  // wide strings as appropriate.
7805f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  ResultPtr = &ResultBuf[0];   // Next byte to fill in.
7815f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
782ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson  Pascal = false;
783ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson
7845f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  for (unsigned i = 0, e = NumStringToks; i != e; ++i) {
7855f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    const char *ThisTokBuf = &TokenBuf[0];
7865f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Get the spelling of the token, which eliminates trigraphs, etc.  We know
7875f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // that ThisTokBuf points to a buffer that is big enough for the whole token
7885f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // and 'spelled' tokens can only shrink.
7895f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    unsigned ThisTokLen = PP.getSpelling(StringToks[i], ThisTokBuf);
7905f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    const char *ThisTokEnd = ThisTokBuf+ThisTokLen-1;  // Skip end quote.
7915f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7925f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // TODO: Input character set mapping support.
7935f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
7945f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    // Skip L marker for wide strings.
7955f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    bool ThisIsWide = false;
7965f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    if (ThisTokBuf[0] == 'L') {
7975f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ++ThisTokBuf;
7985f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      ThisIsWide = true;
7995f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
8005f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8015f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    assert(ThisTokBuf[0] == '"' && "Expected quote, lexer broken?");
8025f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    ++ThisTokBuf;
8035f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
804ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson    // Check if this is a pascal string
805ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson    if (pp.getLangOptions().PascalStrings && ThisTokBuf + 1 != ThisTokEnd &&
806ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        ThisTokBuf[0] == '\\' && ThisTokBuf[1] == 'p') {
807ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson
808ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson      // If the \p sequence is found in the first token, we have a pascal string
809ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson      // Otherwise, if we already have a pascal string, ignore the first \p
810ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson      if (i == 0) {
811ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        ++ThisTokBuf;
812ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        Pascal = true;
813ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson      } else if (Pascal)
814ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson        ThisTokBuf += 2;
815ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson    }
816ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson
8175f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    while (ThisTokBuf != ThisTokEnd) {
8185f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      // Is this a span of non-escape characters?
8195f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      if (ThisTokBuf[0] != '\\') {
8205f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        const char *InStart = ThisTokBuf;
8215f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        do {
8225f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          ++ThisTokBuf;
8235f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        } while (ThisTokBuf != ThisTokEnd && ThisTokBuf[0] != '\\');
8245f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8255f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        // Copy the character span over.
8265f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        unsigned Len = ThisTokBuf-InStart;
8275f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        if (!AnyWide) {
8285f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          memcpy(ResultPtr, InStart, Len);
8295f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          ResultPtr += Len;
8305f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        } else {
8315f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          // Note: our internal rep of wide char tokens is always little-endian.
8325f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          for (; Len; --Len, ++InStart) {
8335f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer            *ResultPtr++ = InStart[0];
8345f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer            // Add zeros at the end.
8355f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer            for (unsigned i = 1, e = wchar_tByteWidth; i != e; ++i)
8360e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff              *ResultPtr++ = 0;
8375f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer          }
8385f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        }
8395f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer        continue;
8405f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
8415f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
8420e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff      if (ThisTokBuf[1] == 'u' || ThisTokBuf[1] == 'U') {
8430e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff        ProcessUCNEscape(ThisTokBuf, ThisTokEnd, ResultPtr,
8448a5c0cd90b8d607ca284274000ed8716b836d253Steve Naroff                         hadError, StringToks[i].getLocation(), ThisIsWide, PP);
8450e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff      } else {
8460e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff        // Otherwise, this is a non-UCN escape character.  Process it.
8470e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff        unsigned ResultChar = ProcessCharEscape(ThisTokBuf, ThisTokEnd, hadError,
8480e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff                                                StringToks[i].getLocation(),
8490e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff                                                ThisIsWide, PP);
8500e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
8510e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff        // Note: our internal rep of wide char tokens is always little-endian.
8520e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff        *ResultPtr++ = ResultChar & 0xFF;
8530e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff
8540e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff        if (AnyWide) {
8550e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff          for (unsigned i = 1, e = wchar_tByteWidth; i != e; ++i)
8560e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff            *ResultPtr++ = ResultChar >> i*8;
8570e3e3eb3879d5a7aaca4a393706149ddef8544f1Steve Naroff        }
8585f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer      }
8595f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer    }
8605f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer  }
8615f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer
862bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner  if (Pascal) {
863ee98ac5a1330db432b188dd2d38b6631aac47bf1Anders Carlsson    ResultBuf[0] = ResultPtr-&ResultBuf[0]-1;
864bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner
865bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner    // Verify that pascal strings aren't too large.
866bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner    if (GetStringLength() > 256)
867bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner      PP.Diag(StringToks[0].getLocation(), diag::err_pascal_string_too_long)
868bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner        << SourceRange(StringToks[0].getLocation(),
869bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner                       StringToks[NumStringToks-1].getLocation());
870bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner    hadError = 1;
871bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner    return;
872bbee00b6456e90a09f63c83c20233e6c5ad6000aChris Lattner  }
8735f016e2cb5d11daeb237544de1c5d59f20fe1a6eReid Spencer}
874719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
875719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
876719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// getOffsetOfStringByte - This function returns the offset of the
877719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// specified byte of the string data represented by Token.  This handles
878719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner/// advancing over escape sequences in the string.
879719e61573f27c11057ecfe0dd8f141621602c571Chris Lattnerunsigned StringLiteralParser::getOffsetOfStringByte(const Token &Tok,
880719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner                                                    unsigned ByteNo,
881719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner                                                    Preprocessor &PP) {
882719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  // Get the spelling of the token.
883719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  llvm::SmallString<16> SpellingBuffer;
884719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  SpellingBuffer.resize(Tok.getLength());
885719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
886719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  const char *SpellingPtr = &SpellingBuffer[0];
887719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  unsigned TokLen = PP.getSpelling(Tok, SpellingPtr);
888719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
889719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  assert(SpellingPtr[0] != 'L' && "Doesn't handle wide strings yet");
890719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
891719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
892719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  const char *SpellingStart = SpellingPtr;
893719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  const char *SpellingEnd = SpellingPtr+TokLen;
894719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
895719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  // Skip over the leading quote.
896719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  assert(SpellingPtr[0] == '"' && "Should be a string literal!");
897719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  ++SpellingPtr;
898719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
899719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  // Skip over bytes until we find the offset we're looking for.
900719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  while (ByteNo) {
901719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    assert(SpellingPtr < SpellingEnd && "Didn't find byte offset!");
902719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
903719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    // Step over non-escapes simply.
904719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    if (*SpellingPtr != '\\') {
905719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      ++SpellingPtr;
906719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      --ByteNo;
907719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner      continue;
908719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    }
909719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
910719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    // Otherwise, this is an escape character.  Advance over it.
911719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    bool HadError = false;
912719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    ProcessCharEscape(SpellingPtr, SpellingEnd, HadError,
913719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner                      Tok.getLocation(), false, PP);
914719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    assert(!HadError && "This method isn't valid on erroneous strings");
915719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner    --ByteNo;
916719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  }
917719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner
918719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner  return SpellingPtr-SpellingStart;
919719e61573f27c11057ecfe0dd8f141621602c571Chris Lattner}
920