blob: 4a23355642d4f0382e36d1875778ee34f5ec1f6e [file] [log] [blame]
Reid Spencer5f016e22007-07-11 17:01:13 +00001//===--- LiteralSupport.cpp - Code to parse and process literals ----------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Reid Spencer5f016e22007-07-11 17:01:13 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the NumericLiteralParser, CharLiteralParser, and
11// StringLiteralParser interfaces.
12//
13//===----------------------------------------------------------------------===//
14
15#include "clang/Lex/LiteralSupport.h"
16#include "clang/Lex/Preprocessor.h"
Chris Lattner500d3292009-01-29 05:15:15 +000017#include "clang/Lex/LexDiagnostic.h"
Chris Lattner136f93a2007-07-16 06:55:01 +000018#include "clang/Basic/TargetInfo.h"
Erick Tryzelaare9f195f2009-08-16 23:36:28 +000019#include "llvm/ADT/StringRef.h"
Reid Spencer5f016e22007-07-11 17:01:13 +000020#include "llvm/ADT/StringExtras.h"
21using namespace clang;
22
23/// HexDigitValue - Return the value of the specified hex digit, or -1 if it's
24/// not valid.
25static int HexDigitValue(char C) {
26 if (C >= '0' && C <= '9') return C-'0';
27 if (C >= 'a' && C <= 'f') return C-'a'+10;
28 if (C >= 'A' && C <= 'F') return C-'A'+10;
29 return -1;
30}
31
32/// ProcessCharEscape - Parse a standard C escape sequence, which can occur in
33/// either a character or a string literal.
34static unsigned ProcessCharEscape(const char *&ThisTokBuf,
35 const char *ThisTokEnd, bool &HadError,
Chris Lattner91f54ce2010-11-17 06:26:08 +000036 FullSourceLoc Loc, bool IsWide,
37 Diagnostic *Diags, const TargetInfo &Target) {
Reid Spencer5f016e22007-07-11 17:01:13 +000038 // Skip the '\' char.
39 ++ThisTokBuf;
40
41 // We know that this character can't be off the end of the buffer, because
42 // that would have been \", which would not have been the end of string.
43 unsigned ResultChar = *ThisTokBuf++;
44 switch (ResultChar) {
45 // These map to themselves.
46 case '\\': case '\'': case '"': case '?': break;
Mike Stump1eb44332009-09-09 15:08:12 +000047
Reid Spencer5f016e22007-07-11 17:01:13 +000048 // These have fixed mappings.
49 case 'a':
50 // TODO: K&R: the meaning of '\\a' is different in traditional C
51 ResultChar = 7;
52 break;
53 case 'b':
54 ResultChar = 8;
55 break;
56 case 'e':
Chris Lattner91f54ce2010-11-17 06:26:08 +000057 if (Diags)
58 Diags->Report(Loc, diag::ext_nonstandard_escape) << "e";
Reid Spencer5f016e22007-07-11 17:01:13 +000059 ResultChar = 27;
60 break;
Eli Friedman3c548012009-06-10 01:32:39 +000061 case 'E':
Chris Lattner91f54ce2010-11-17 06:26:08 +000062 if (Diags)
63 Diags->Report(Loc, diag::ext_nonstandard_escape) << "E";
Eli Friedman3c548012009-06-10 01:32:39 +000064 ResultChar = 27;
65 break;
Reid Spencer5f016e22007-07-11 17:01:13 +000066 case 'f':
67 ResultChar = 12;
68 break;
69 case 'n':
70 ResultChar = 10;
71 break;
72 case 'r':
73 ResultChar = 13;
74 break;
75 case 't':
76 ResultChar = 9;
77 break;
78 case 'v':
79 ResultChar = 11;
80 break;
Reid Spencer5f016e22007-07-11 17:01:13 +000081 case 'x': { // Hex escape.
82 ResultChar = 0;
83 if (ThisTokBuf == ThisTokEnd || !isxdigit(*ThisTokBuf)) {
Chris Lattner91f54ce2010-11-17 06:26:08 +000084 if (Diags)
85 Diags->Report(Loc, diag::err_hex_escape_no_digits);
Reid Spencer5f016e22007-07-11 17:01:13 +000086 HadError = 1;
87 break;
88 }
Mike Stump1eb44332009-09-09 15:08:12 +000089
Reid Spencer5f016e22007-07-11 17:01:13 +000090 // Hex escapes are a maximal series of hex digits.
91 bool Overflow = false;
92 for (; ThisTokBuf != ThisTokEnd; ++ThisTokBuf) {
93 int CharVal = HexDigitValue(ThisTokBuf[0]);
94 if (CharVal == -1) break;
Chris Lattnerc29bbde2008-09-30 20:45:40 +000095 // About to shift out a digit?
96 Overflow |= (ResultChar & 0xF0000000) ? true : false;
Reid Spencer5f016e22007-07-11 17:01:13 +000097 ResultChar <<= 4;
98 ResultChar |= CharVal;
99 }
100
101 // See if any bits will be truncated when evaluated as a character.
Chris Lattner91f54ce2010-11-17 06:26:08 +0000102 unsigned CharWidth =
103 IsWide ? Target.getWCharWidth() : Target.getCharWidth();
Mike Stump1eb44332009-09-09 15:08:12 +0000104
Reid Spencer5f016e22007-07-11 17:01:13 +0000105 if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) {
106 Overflow = true;
107 ResultChar &= ~0U >> (32-CharWidth);
108 }
Mike Stump1eb44332009-09-09 15:08:12 +0000109
Reid Spencer5f016e22007-07-11 17:01:13 +0000110 // Check for overflow.
Chris Lattner91f54ce2010-11-17 06:26:08 +0000111 if (Overflow && Diags) // Too many digits to fit in
112 Diags->Report(Loc, diag::warn_hex_escape_too_large);
Reid Spencer5f016e22007-07-11 17:01:13 +0000113 break;
114 }
115 case '0': case '1': case '2': case '3':
116 case '4': case '5': case '6': case '7': {
117 // Octal escapes.
118 --ThisTokBuf;
119 ResultChar = 0;
120
121 // Octal escapes are a series of octal digits with maximum length 3.
122 // "\0123" is a two digit sequence equal to "\012" "3".
123 unsigned NumDigits = 0;
124 do {
125 ResultChar <<= 3;
126 ResultChar |= *ThisTokBuf++ - '0';
127 ++NumDigits;
128 } while (ThisTokBuf != ThisTokEnd && NumDigits < 3 &&
129 ThisTokBuf[0] >= '0' && ThisTokBuf[0] <= '7');
Mike Stump1eb44332009-09-09 15:08:12 +0000130
Reid Spencer5f016e22007-07-11 17:01:13 +0000131 // Check for overflow. Reject '\777', but not L'\777'.
Chris Lattner91f54ce2010-11-17 06:26:08 +0000132 unsigned CharWidth =
133 IsWide ? Target.getWCharWidth() : Target.getCharWidth();
Mike Stump1eb44332009-09-09 15:08:12 +0000134
Reid Spencer5f016e22007-07-11 17:01:13 +0000135 if (CharWidth != 32 && (ResultChar >> CharWidth) != 0) {
Chris Lattner91f54ce2010-11-17 06:26:08 +0000136 if (Diags)
137 Diags->Report(Loc, diag::warn_octal_escape_too_large);
Reid Spencer5f016e22007-07-11 17:01:13 +0000138 ResultChar &= ~0U >> (32-CharWidth);
139 }
140 break;
141 }
Mike Stump1eb44332009-09-09 15:08:12 +0000142
Reid Spencer5f016e22007-07-11 17:01:13 +0000143 // Otherwise, these are not valid escapes.
144 case '(': case '{': case '[': case '%':
145 // GCC accepts these as extensions. We warn about them as such though.
Chris Lattner91f54ce2010-11-17 06:26:08 +0000146 if (Diags)
147 Diags->Report(Loc, diag::ext_nonstandard_escape)
Douglas Gregorb90f4b32010-05-26 05:35:51 +0000148 << std::string()+(char)ResultChar;
Eli Friedmanf01fdff2009-04-28 00:51:18 +0000149 break;
Reid Spencer5f016e22007-07-11 17:01:13 +0000150 default:
Chris Lattner91f54ce2010-11-17 06:26:08 +0000151 if (Diags == 0)
Douglas Gregorb90f4b32010-05-26 05:35:51 +0000152 break;
153
Chris Lattnerac92d822008-11-22 07:23:31 +0000154 if (isgraph(ThisTokBuf[0]))
Chris Lattner91f54ce2010-11-17 06:26:08 +0000155 Diags->Report(Loc, diag::ext_unknown_escape)
156 << std::string()+(char)ResultChar;
Chris Lattnerac92d822008-11-22 07:23:31 +0000157 else
Chris Lattner91f54ce2010-11-17 06:26:08 +0000158 Diags->Report(Loc, diag::ext_unknown_escape)
159 << "x"+llvm::utohexstr(ResultChar);
Reid Spencer5f016e22007-07-11 17:01:13 +0000160 break;
161 }
Mike Stump1eb44332009-09-09 15:08:12 +0000162
Reid Spencer5f016e22007-07-11 17:01:13 +0000163 return ResultChar;
164}
165
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000166/// ProcessUCNEscape - Read the Universal Character Name, check constraints and
Nico Weber59705ae2010-10-09 00:27:47 +0000167/// return the UTF32.
168static bool ProcessUCNEscape(const char *&ThisTokBuf, const char *ThisTokEnd,
169 uint32_t &UcnVal, unsigned short &UcnLen,
Chris Lattner872a45e2010-11-17 06:55:10 +0000170 FullSourceLoc Loc, Diagnostic *Diags,
Chris Lattner6c66f072010-11-17 06:46:14 +0000171 const LangOptions &Features) {
172 if (!Features.CPlusPlus && !Features.C99 && Diags)
Chris Lattner872a45e2010-11-17 06:55:10 +0000173 Diags->Report(Loc, diag::warn_ucn_not_valid_in_c89);
Mike Stump1eb44332009-09-09 15:08:12 +0000174
Steve Naroff4e93b342009-04-01 11:09:15 +0000175 // Save the beginning of the string (for error diagnostics).
176 const char *ThisTokBegin = ThisTokBuf;
Mike Stump1eb44332009-09-09 15:08:12 +0000177
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000178 // Skip the '\u' char's.
179 ThisTokBuf += 2;
Reid Spencer5f016e22007-07-11 17:01:13 +0000180
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000181 if (ThisTokBuf == ThisTokEnd || !isxdigit(*ThisTokBuf)) {
Chris Lattner6c66f072010-11-17 06:46:14 +0000182 if (Diags)
Chris Lattner872a45e2010-11-17 06:55:10 +0000183 Diags->Report(Loc, diag::err_ucn_escape_no_digits);
Nico Weber59705ae2010-10-09 00:27:47 +0000184 return false;
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000185 }
Nico Weber59705ae2010-10-09 00:27:47 +0000186 UcnLen = (ThisTokBuf[-1] == 'u' ? 4 : 8);
Fariborz Jahanian56bedef2010-08-31 23:34:27 +0000187 unsigned short UcnLenSave = UcnLen;
Nico Weber59705ae2010-10-09 00:27:47 +0000188 for (; ThisTokBuf != ThisTokEnd && UcnLenSave; ++ThisTokBuf, UcnLenSave--) {
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000189 int CharVal = HexDigitValue(ThisTokBuf[0]);
190 if (CharVal == -1) break;
191 UcnVal <<= 4;
192 UcnVal |= CharVal;
193 }
194 // If we didn't consume the proper number of digits, there is a problem.
Nico Weber59705ae2010-10-09 00:27:47 +0000195 if (UcnLenSave) {
Chris Lattner872a45e2010-11-17 06:55:10 +0000196 if (Diags) {
Chris Lattner7ef5c272010-11-17 07:05:50 +0000197 SourceLocation L =
198 Lexer::AdvanceToTokenCharacter(Loc, ThisTokBuf-ThisTokBegin,
199 Loc.getManager(), Features);
200 Diags->Report(FullSourceLoc(L, Loc.getManager()),
201 diag::err_ucn_escape_incomplete);
Chris Lattner872a45e2010-11-17 06:55:10 +0000202 }
Nico Weber59705ae2010-10-09 00:27:47 +0000203 return false;
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000204 }
Mike Stump1eb44332009-09-09 15:08:12 +0000205 // Check UCN constraints (C99 6.4.3p2).
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000206 if ((UcnVal < 0xa0 &&
207 (UcnVal != 0x24 && UcnVal != 0x40 && UcnVal != 0x60 )) // $, @, `
Mike Stump1eb44332009-09-09 15:08:12 +0000208 || (UcnVal >= 0xD800 && UcnVal <= 0xDFFF)
Steve Naroff8a5c0cd2009-03-31 10:29:45 +0000209 || (UcnVal > 0x10FFFF)) /* the maximum legal UTF32 value */ {
Chris Lattner6c66f072010-11-17 06:46:14 +0000210 if (Diags)
Chris Lattner872a45e2010-11-17 06:55:10 +0000211 Diags->Report(Loc, diag::err_ucn_escape_invalid);
Nico Weber59705ae2010-10-09 00:27:47 +0000212 return false;
213 }
214 return true;
215}
216
217/// EncodeUCNEscape - Read the Universal Character Name, check constraints and
218/// convert the UTF32 to UTF8 or UTF16. This is a subroutine of
219/// StringLiteralParser. When we decide to implement UCN's for identifiers,
220/// we will likely rework our support for UCN's.
221static void EncodeUCNEscape(const char *&ThisTokBuf, const char *ThisTokEnd,
222 char *&ResultBuf, bool &HadError,
223 SourceLocation Loc, Preprocessor &PP,
Chris Lattner6c66f072010-11-17 06:46:14 +0000224 bool wide, bool Complain) {
Nico Weber59705ae2010-10-09 00:27:47 +0000225 typedef uint32_t UTF32;
226 UTF32 UcnVal = 0;
227 unsigned short UcnLen = 0;
Chris Lattner872a45e2010-11-17 06:55:10 +0000228 if (!ProcessUCNEscape(ThisTokBuf, ThisTokEnd, UcnVal, UcnLen,
229 FullSourceLoc(Loc, PP.getSourceManager()),
Chris Lattner6c66f072010-11-17 06:46:14 +0000230 Complain ? &PP.getDiagnostics() : 0,
231 PP.getLangOptions())){
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000232 HadError = 1;
233 return;
234 }
Nico Weber59705ae2010-10-09 00:27:47 +0000235
Fariborz Jahanian56bedef2010-08-31 23:34:27 +0000236 if (wide) {
Nico Weber59705ae2010-10-09 00:27:47 +0000237 (void)UcnLen;
238 assert((UcnLen== 4 || UcnLen== 8) &&
239 "EncodeUCNEscape - only ucn length of 4 or 8 supported");
Nico Webera0f15b02010-10-06 04:57:26 +0000240
241 if (!PP.getLangOptions().ShortWChar) {
242 // Note: our internal rep of wide char tokens is always little-endian.
243 *ResultBuf++ = (UcnVal & 0x000000FF);
244 *ResultBuf++ = (UcnVal & 0x0000FF00) >> 8;
245 *ResultBuf++ = (UcnVal & 0x00FF0000) >> 16;
246 *ResultBuf++ = (UcnVal & 0xFF000000) >> 24;
247 return;
248 }
249
250 // Convert to UTF16.
251 if (UcnVal < (UTF32)0xFFFF) {
252 *ResultBuf++ = (UcnVal & 0x000000FF);
253 *ResultBuf++ = (UcnVal & 0x0000FF00) >> 8;
254 return;
255 }
256 PP.Diag(Loc, diag::warn_ucn_escape_too_large);
257
258 typedef uint16_t UTF16;
259 UcnVal -= 0x10000;
260 UTF16 surrogate1 = 0xD800 + (UcnVal >> 10);
261 UTF16 surrogate2 = 0xDC00 + (UcnVal & 0x3FF);
262 *ResultBuf++ = (surrogate1 & 0x000000FF);
263 *ResultBuf++ = (surrogate1 & 0x0000FF00) >> 8;
264 *ResultBuf++ = (surrogate2 & 0x000000FF);
265 *ResultBuf++ = (surrogate2 & 0x0000FF00) >> 8;
Fariborz Jahanian56bedef2010-08-31 23:34:27 +0000266 return;
267 }
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000268 // Now that we've parsed/checked the UCN, we convert from UTF32->UTF8.
269 // The conversion below was inspired by:
270 // http://www.unicode.org/Public/PROGRAMS/CVTUTF/ConvertUTF.c
Mike Stump1eb44332009-09-09 15:08:12 +0000271 // First, we determine how many bytes the result will require.
Steve Naroff4e93b342009-04-01 11:09:15 +0000272 typedef uint8_t UTF8;
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000273
274 unsigned short bytesToWrite = 0;
275 if (UcnVal < (UTF32)0x80)
276 bytesToWrite = 1;
277 else if (UcnVal < (UTF32)0x800)
278 bytesToWrite = 2;
279 else if (UcnVal < (UTF32)0x10000)
280 bytesToWrite = 3;
281 else
282 bytesToWrite = 4;
Mike Stump1eb44332009-09-09 15:08:12 +0000283
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000284 const unsigned byteMask = 0xBF;
285 const unsigned byteMark = 0x80;
Mike Stump1eb44332009-09-09 15:08:12 +0000286
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000287 // Once the bits are split out into bytes of UTF8, this is a mask OR-ed
Steve Naroff8a5c0cd2009-03-31 10:29:45 +0000288 // into the first byte, depending on how many bytes follow.
Mike Stump1eb44332009-09-09 15:08:12 +0000289 static const UTF8 firstByteMark[5] = {
Steve Naroff8a5c0cd2009-03-31 10:29:45 +0000290 0x00, 0x00, 0xC0, 0xE0, 0xF0
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000291 };
292 // Finally, we write the bytes into ResultBuf.
293 ResultBuf += bytesToWrite;
294 switch (bytesToWrite) { // note: everything falls through.
295 case 4: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
296 case 3: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
297 case 2: *--ResultBuf = (UTF8)((UcnVal | byteMark) & byteMask); UcnVal >>= 6;
298 case 1: *--ResultBuf = (UTF8) (UcnVal | firstByteMark[bytesToWrite]);
299 }
300 // Update the buffer.
301 ResultBuf += bytesToWrite;
302}
Reid Spencer5f016e22007-07-11 17:01:13 +0000303
304
305/// integer-constant: [C99 6.4.4.1]
306/// decimal-constant integer-suffix
307/// octal-constant integer-suffix
308/// hexadecimal-constant integer-suffix
Mike Stump1eb44332009-09-09 15:08:12 +0000309/// decimal-constant:
Reid Spencer5f016e22007-07-11 17:01:13 +0000310/// nonzero-digit
311/// decimal-constant digit
Mike Stump1eb44332009-09-09 15:08:12 +0000312/// octal-constant:
Reid Spencer5f016e22007-07-11 17:01:13 +0000313/// 0
314/// octal-constant octal-digit
Mike Stump1eb44332009-09-09 15:08:12 +0000315/// hexadecimal-constant:
Reid Spencer5f016e22007-07-11 17:01:13 +0000316/// hexadecimal-prefix hexadecimal-digit
317/// hexadecimal-constant hexadecimal-digit
318/// hexadecimal-prefix: one of
319/// 0x 0X
320/// integer-suffix:
321/// unsigned-suffix [long-suffix]
322/// unsigned-suffix [long-long-suffix]
323/// long-suffix [unsigned-suffix]
324/// long-long-suffix [unsigned-sufix]
325/// nonzero-digit:
326/// 1 2 3 4 5 6 7 8 9
327/// octal-digit:
328/// 0 1 2 3 4 5 6 7
329/// hexadecimal-digit:
330/// 0 1 2 3 4 5 6 7 8 9
331/// a b c d e f
332/// A B C D E F
333/// unsigned-suffix: one of
334/// u U
335/// long-suffix: one of
336/// l L
Mike Stump1eb44332009-09-09 15:08:12 +0000337/// long-long-suffix: one of
Reid Spencer5f016e22007-07-11 17:01:13 +0000338/// ll LL
339///
340/// floating-constant: [C99 6.4.4.2]
341/// TODO: add rules...
342///
Reid Spencer5f016e22007-07-11 17:01:13 +0000343NumericLiteralParser::
344NumericLiteralParser(const char *begin, const char *end,
345 SourceLocation TokLoc, Preprocessor &pp)
346 : PP(pp), ThisTokBegin(begin), ThisTokEnd(end) {
Mike Stump1eb44332009-09-09 15:08:12 +0000347
Chris Lattnerc29bbde2008-09-30 20:45:40 +0000348 // This routine assumes that the range begin/end matches the regex for integer
349 // and FP constants (specifically, the 'pp-number' regex), and assumes that
350 // the byte at "*end" is both valid and not part of the regex. Because of
351 // this, it doesn't have to check for 'overscan' in various places.
352 assert(!isalnum(*end) && *end != '.' && *end != '_' &&
353 "Lexer didn't maximally munch?");
Mike Stump1eb44332009-09-09 15:08:12 +0000354
Reid Spencer5f016e22007-07-11 17:01:13 +0000355 s = DigitsBegin = begin;
356 saw_exponent = false;
357 saw_period = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000358 isLong = false;
359 isUnsigned = false;
360 isLongLong = false;
Chris Lattner6e400c22007-08-26 03:29:23 +0000361 isFloat = false;
Chris Lattner506b8de2007-08-26 01:58:14 +0000362 isImaginary = false;
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000363 isMicrosoftInteger = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000364 hadError = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000365
Reid Spencer5f016e22007-07-11 17:01:13 +0000366 if (*s == '0') { // parse radix
Chris Lattner368328c2008-06-30 06:39:54 +0000367 ParseNumberStartingWithZero(TokLoc);
368 if (hadError)
369 return;
Reid Spencer5f016e22007-07-11 17:01:13 +0000370 } else { // the first digit is non-zero
371 radix = 10;
372 s = SkipDigits(s);
373 if (s == ThisTokEnd) {
374 // Done.
Christopher Lamb016765e2007-11-29 06:06:27 +0000375 } else if (isxdigit(*s) && !(*s == 'e' || *s == 'E')) {
Chris Lattnerac92d822008-11-22 07:23:31 +0000376 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
Benjamin Kramer476d8b82010-08-11 14:47:12 +0000377 diag::err_invalid_decimal_digit) << llvm::StringRef(s, 1);
Chris Lattnerac92d822008-11-22 07:23:31 +0000378 hadError = true;
Reid Spencer5f016e22007-07-11 17:01:13 +0000379 return;
380 } else if (*s == '.') {
381 s++;
382 saw_period = true;
383 s = SkipDigits(s);
Mike Stump1eb44332009-09-09 15:08:12 +0000384 }
Chris Lattner4411f462008-09-29 23:12:31 +0000385 if ((*s == 'e' || *s == 'E')) { // exponent
Chris Lattner70f66ab2008-04-20 18:47:55 +0000386 const char *Exponent = s;
Reid Spencer5f016e22007-07-11 17:01:13 +0000387 s++;
388 saw_exponent = true;
389 if (*s == '+' || *s == '-') s++; // sign
390 const char *first_non_digit = SkipDigits(s);
Chris Lattner0b7f69d2008-04-20 18:41:46 +0000391 if (first_non_digit != s) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000392 s = first_non_digit;
Chris Lattner0b7f69d2008-04-20 18:41:46 +0000393 } else {
Chris Lattnerac92d822008-11-22 07:23:31 +0000394 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-begin),
395 diag::err_exponent_has_no_digits);
396 hadError = true;
Chris Lattner0b7f69d2008-04-20 18:41:46 +0000397 return;
Reid Spencer5f016e22007-07-11 17:01:13 +0000398 }
399 }
400 }
401
402 SuffixBegin = s;
Mike Stump1eb44332009-09-09 15:08:12 +0000403
Chris Lattner506b8de2007-08-26 01:58:14 +0000404 // Parse the suffix. At this point we can classify whether we have an FP or
405 // integer constant.
406 bool isFPConstant = isFloatingLiteral();
Mike Stump1eb44332009-09-09 15:08:12 +0000407
Chris Lattner506b8de2007-08-26 01:58:14 +0000408 // Loop over all of the characters of the suffix. If we see something bad,
409 // we break out of the loop.
410 for (; s != ThisTokEnd; ++s) {
411 switch (*s) {
412 case 'f': // FP Suffix for "float"
413 case 'F':
414 if (!isFPConstant) break; // Error for integer constant.
Chris Lattner6e400c22007-08-26 03:29:23 +0000415 if (isFloat || isLong) break; // FF, LF invalid.
416 isFloat = true;
Chris Lattner506b8de2007-08-26 01:58:14 +0000417 continue; // Success.
418 case 'u':
419 case 'U':
420 if (isFPConstant) break; // Error for floating constant.
421 if (isUnsigned) break; // Cannot be repeated.
422 isUnsigned = true;
423 continue; // Success.
424 case 'l':
425 case 'L':
426 if (isLong || isLongLong) break; // Cannot be repeated.
Chris Lattner6e400c22007-08-26 03:29:23 +0000427 if (isFloat) break; // LF invalid.
Mike Stump1eb44332009-09-09 15:08:12 +0000428
Chris Lattner506b8de2007-08-26 01:58:14 +0000429 // Check for long long. The L's need to be adjacent and the same case.
430 if (s+1 != ThisTokEnd && s[1] == s[0]) {
431 if (isFPConstant) break; // long long invalid for floats.
432 isLongLong = true;
433 ++s; // Eat both of them.
434 } else {
Reid Spencer5f016e22007-07-11 17:01:13 +0000435 isLong = true;
Reid Spencer5f016e22007-07-11 17:01:13 +0000436 }
Chris Lattner506b8de2007-08-26 01:58:14 +0000437 continue; // Success.
438 case 'i':
Chris Lattnerc6374152010-10-14 00:24:10 +0000439 case 'I':
Steve Naroff0c29b222008-04-04 21:02:54 +0000440 if (PP.getLangOptions().Microsoft) {
Fariborz Jahaniana8be02b2010-01-22 21:36:53 +0000441 if (isFPConstant || isLong || isLongLong) break;
Nuno Lopes6e8c7ac2009-11-28 13:37:52 +0000442
Steve Naroff0c29b222008-04-04 21:02:54 +0000443 // Allow i8, i16, i32, i64, and i128.
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000444 if (s + 1 != ThisTokEnd) {
445 switch (s[1]) {
446 case '8':
447 s += 2; // i8 suffix
448 isMicrosoftInteger = true;
Nuno Lopes6e8c7ac2009-11-28 13:37:52 +0000449 break;
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000450 case '1':
Nuno Lopes6e8c7ac2009-11-28 13:37:52 +0000451 if (s + 2 == ThisTokEnd) break;
452 if (s[2] == '6') s += 3; // i16 suffix
453 else if (s[2] == '2') {
454 if (s + 3 == ThisTokEnd) break;
455 if (s[3] == '8') s += 4; // i128 suffix
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000456 }
457 isMicrosoftInteger = true;
Nuno Lopes6e8c7ac2009-11-28 13:37:52 +0000458 break;
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000459 case '3':
Nuno Lopes6e8c7ac2009-11-28 13:37:52 +0000460 if (s + 2 == ThisTokEnd) break;
461 if (s[2] == '2') s += 3; // i32 suffix
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000462 isMicrosoftInteger = true;
Nuno Lopes6e8c7ac2009-11-28 13:37:52 +0000463 break;
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000464 case '6':
Nuno Lopes6e8c7ac2009-11-28 13:37:52 +0000465 if (s + 2 == ThisTokEnd) break;
466 if (s[2] == '4') s += 3; // i64 suffix
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000467 isMicrosoftInteger = true;
Nuno Lopes6e8c7ac2009-11-28 13:37:52 +0000468 break;
Mike Stumpb79fe2d2009-10-08 22:55:36 +0000469 default:
470 break;
471 }
472 break;
Steve Naroff0c29b222008-04-04 21:02:54 +0000473 }
Steve Naroff0c29b222008-04-04 21:02:54 +0000474 }
475 // fall through.
Chris Lattner506b8de2007-08-26 01:58:14 +0000476 case 'j':
477 case 'J':
478 if (isImaginary) break; // Cannot be repeated.
479 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
480 diag::ext_imaginary_constant);
481 isImaginary = true;
482 continue; // Success.
Reid Spencer5f016e22007-07-11 17:01:13 +0000483 }
Chris Lattner506b8de2007-08-26 01:58:14 +0000484 // If we reached here, there was an error.
485 break;
486 }
Mike Stump1eb44332009-09-09 15:08:12 +0000487
Chris Lattner506b8de2007-08-26 01:58:14 +0000488 // Report an error if there are any.
489 if (s != ThisTokEnd) {
Chris Lattnerac92d822008-11-22 07:23:31 +0000490 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-begin),
491 isFPConstant ? diag::err_invalid_suffix_float_constant :
492 diag::err_invalid_suffix_integer_constant)
Benjamin Kramer476d8b82010-08-11 14:47:12 +0000493 << llvm::StringRef(SuffixBegin, ThisTokEnd-SuffixBegin);
Chris Lattnerac92d822008-11-22 07:23:31 +0000494 hadError = true;
Chris Lattner506b8de2007-08-26 01:58:14 +0000495 return;
Reid Spencer5f016e22007-07-11 17:01:13 +0000496 }
497}
498
Chris Lattner368328c2008-06-30 06:39:54 +0000499/// ParseNumberStartingWithZero - This method is called when the first character
500/// of the number is found to be a zero. This means it is either an octal
501/// number (like '04') or a hex number ('0x123a') a binary number ('0b1010') or
Mike Stump1eb44332009-09-09 15:08:12 +0000502/// a floating point number (01239.123e4). Eat the prefix, determining the
Chris Lattner368328c2008-06-30 06:39:54 +0000503/// radix etc.
504void NumericLiteralParser::ParseNumberStartingWithZero(SourceLocation TokLoc) {
505 assert(s[0] == '0' && "Invalid method call");
506 s++;
Mike Stump1eb44332009-09-09 15:08:12 +0000507
Chris Lattner368328c2008-06-30 06:39:54 +0000508 // Handle a hex number like 0x1234.
509 if ((*s == 'x' || *s == 'X') && (isxdigit(s[1]) || s[1] == '.')) {
510 s++;
511 radix = 16;
512 DigitsBegin = s;
513 s = SkipHexDigits(s);
514 if (s == ThisTokEnd) {
515 // Done.
516 } else if (*s == '.') {
517 s++;
518 saw_period = true;
519 s = SkipHexDigits(s);
520 }
521 // A binary exponent can appear with or with a '.'. If dotted, the
Mike Stump1eb44332009-09-09 15:08:12 +0000522 // binary exponent is required.
Sean Hunt8c723402010-01-10 23:37:56 +0000523 if ((*s == 'p' || *s == 'P') && !PP.getLangOptions().CPlusPlus0x) {
Chris Lattner368328c2008-06-30 06:39:54 +0000524 const char *Exponent = s;
525 s++;
526 saw_exponent = true;
527 if (*s == '+' || *s == '-') s++; // sign
528 const char *first_non_digit = SkipDigits(s);
Chris Lattner6ea62382008-07-25 18:18:34 +0000529 if (first_non_digit == s) {
Chris Lattnerac92d822008-11-22 07:23:31 +0000530 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin),
531 diag::err_exponent_has_no_digits);
532 hadError = true;
Chris Lattner6ea62382008-07-25 18:18:34 +0000533 return;
Chris Lattner368328c2008-06-30 06:39:54 +0000534 }
Chris Lattner6ea62382008-07-25 18:18:34 +0000535 s = first_non_digit;
Mike Stump1eb44332009-09-09 15:08:12 +0000536
Sean Hunt8c723402010-01-10 23:37:56 +0000537 // In C++0x, we cannot support hexadecmial floating literals because
538 // they conflict with user-defined literals, so we warn in previous
539 // versions of C++ by default.
540 if (PP.getLangOptions().CPlusPlus)
541 PP.Diag(TokLoc, diag::ext_hexconstant_cplusplus);
542 else if (!PP.getLangOptions().HexFloats)
Chris Lattnerac92d822008-11-22 07:23:31 +0000543 PP.Diag(TokLoc, diag::ext_hexconstant_invalid);
Chris Lattner368328c2008-06-30 06:39:54 +0000544 } else if (saw_period) {
Chris Lattnerac92d822008-11-22 07:23:31 +0000545 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
546 diag::err_hexconstant_requires_exponent);
547 hadError = true;
Chris Lattner368328c2008-06-30 06:39:54 +0000548 }
549 return;
550 }
Mike Stump1eb44332009-09-09 15:08:12 +0000551
Chris Lattner368328c2008-06-30 06:39:54 +0000552 // Handle simple binary numbers 0b01010
553 if (*s == 'b' || *s == 'B') {
554 // 0b101010 is a GCC extension.
Chris Lattner413d3552008-06-30 06:44:49 +0000555 PP.Diag(TokLoc, diag::ext_binary_literal);
Chris Lattner368328c2008-06-30 06:39:54 +0000556 ++s;
557 radix = 2;
558 DigitsBegin = s;
559 s = SkipBinaryDigits(s);
560 if (s == ThisTokEnd) {
561 // Done.
562 } else if (isxdigit(*s)) {
Chris Lattnerac92d822008-11-22 07:23:31 +0000563 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
Benjamin Kramer476d8b82010-08-11 14:47:12 +0000564 diag::err_invalid_binary_digit) << llvm::StringRef(s, 1);
Chris Lattnerac92d822008-11-22 07:23:31 +0000565 hadError = true;
Chris Lattner368328c2008-06-30 06:39:54 +0000566 }
Chris Lattner413d3552008-06-30 06:44:49 +0000567 // Other suffixes will be diagnosed by the caller.
Chris Lattner368328c2008-06-30 06:39:54 +0000568 return;
569 }
Mike Stump1eb44332009-09-09 15:08:12 +0000570
Chris Lattner368328c2008-06-30 06:39:54 +0000571 // For now, the radix is set to 8. If we discover that we have a
572 // floating point constant, the radix will change to 10. Octal floating
Mike Stump1eb44332009-09-09 15:08:12 +0000573 // point constants are not permitted (only decimal and hexadecimal).
Chris Lattner368328c2008-06-30 06:39:54 +0000574 radix = 8;
575 DigitsBegin = s;
576 s = SkipOctalDigits(s);
577 if (s == ThisTokEnd)
578 return; // Done, simple octal number like 01234
Mike Stump1eb44332009-09-09 15:08:12 +0000579
Chris Lattner413d3552008-06-30 06:44:49 +0000580 // If we have some other non-octal digit that *is* a decimal digit, see if
581 // this is part of a floating point number like 094.123 or 09e1.
582 if (isdigit(*s)) {
583 const char *EndDecimal = SkipDigits(s);
584 if (EndDecimal[0] == '.' || EndDecimal[0] == 'e' || EndDecimal[0] == 'E') {
585 s = EndDecimal;
586 radix = 10;
587 }
588 }
Mike Stump1eb44332009-09-09 15:08:12 +0000589
Chris Lattner413d3552008-06-30 06:44:49 +0000590 // If we have a hex digit other than 'e' (which denotes a FP exponent) then
591 // the code is using an incorrect base.
Chris Lattner368328c2008-06-30 06:39:54 +0000592 if (isxdigit(*s) && *s != 'e' && *s != 'E') {
Chris Lattnerac92d822008-11-22 07:23:31 +0000593 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, s-ThisTokBegin),
Benjamin Kramer476d8b82010-08-11 14:47:12 +0000594 diag::err_invalid_octal_digit) << llvm::StringRef(s, 1);
Chris Lattnerac92d822008-11-22 07:23:31 +0000595 hadError = true;
Chris Lattner368328c2008-06-30 06:39:54 +0000596 return;
597 }
Mike Stump1eb44332009-09-09 15:08:12 +0000598
Chris Lattner368328c2008-06-30 06:39:54 +0000599 if (*s == '.') {
600 s++;
601 radix = 10;
602 saw_period = true;
Chris Lattner413d3552008-06-30 06:44:49 +0000603 s = SkipDigits(s); // Skip suffix.
Chris Lattner368328c2008-06-30 06:39:54 +0000604 }
605 if (*s == 'e' || *s == 'E') { // exponent
606 const char *Exponent = s;
607 s++;
608 radix = 10;
609 saw_exponent = true;
610 if (*s == '+' || *s == '-') s++; // sign
611 const char *first_non_digit = SkipDigits(s);
612 if (first_non_digit != s) {
613 s = first_non_digit;
614 } else {
Mike Stump1eb44332009-09-09 15:08:12 +0000615 PP.Diag(PP.AdvanceToTokenCharacter(TokLoc, Exponent-ThisTokBegin),
Chris Lattnerac92d822008-11-22 07:23:31 +0000616 diag::err_exponent_has_no_digits);
617 hadError = true;
Chris Lattner368328c2008-06-30 06:39:54 +0000618 return;
619 }
620 }
621}
622
623
Reid Spencer5f016e22007-07-11 17:01:13 +0000624/// GetIntegerValue - Convert this numeric literal value to an APInt that
625/// matches Val's input width. If there is an overflow, set Val to the low bits
626/// of the result and return true. Otherwise, return false.
627bool NumericLiteralParser::GetIntegerValue(llvm::APInt &Val) {
Daniel Dunbara179be32008-10-16 07:32:01 +0000628 // Fast path: Compute a conservative bound on the maximum number of
629 // bits per digit in this radix. If we can't possibly overflow a
630 // uint64 based on that bound then do the simple conversion to
631 // integer. This avoids the expensive overflow checking below, and
632 // handles the common cases that matter (small decimal integers and
633 // hex/octal values which don't overflow).
634 unsigned MaxBitsPerDigit = 1;
Mike Stump1eb44332009-09-09 15:08:12 +0000635 while ((1U << MaxBitsPerDigit) < radix)
Daniel Dunbara179be32008-10-16 07:32:01 +0000636 MaxBitsPerDigit += 1;
637 if ((SuffixBegin - DigitsBegin) * MaxBitsPerDigit <= 64) {
638 uint64_t N = 0;
639 for (s = DigitsBegin; s != SuffixBegin; ++s)
640 N = N*radix + HexDigitValue(*s);
641
642 // This will truncate the value to Val's input width. Simply check
643 // for overflow by comparing.
644 Val = N;
645 return Val.getZExtValue() != N;
646 }
647
Reid Spencer5f016e22007-07-11 17:01:13 +0000648 Val = 0;
649 s = DigitsBegin;
650
651 llvm::APInt RadixVal(Val.getBitWidth(), radix);
652 llvm::APInt CharVal(Val.getBitWidth(), 0);
653 llvm::APInt OldVal = Val;
Mike Stump1eb44332009-09-09 15:08:12 +0000654
Reid Spencer5f016e22007-07-11 17:01:13 +0000655 bool OverflowOccurred = false;
656 while (s < SuffixBegin) {
657 unsigned C = HexDigitValue(*s++);
Mike Stump1eb44332009-09-09 15:08:12 +0000658
Reid Spencer5f016e22007-07-11 17:01:13 +0000659 // If this letter is out of bound for this radix, reject it.
660 assert(C < radix && "NumericLiteralParser ctor should have rejected this");
Mike Stump1eb44332009-09-09 15:08:12 +0000661
Reid Spencer5f016e22007-07-11 17:01:13 +0000662 CharVal = C;
Mike Stump1eb44332009-09-09 15:08:12 +0000663
Reid Spencer5f016e22007-07-11 17:01:13 +0000664 // Add the digit to the value in the appropriate radix. If adding in digits
665 // made the value smaller, then this overflowed.
666 OldVal = Val;
667
668 // Multiply by radix, did overflow occur on the multiply?
669 Val *= RadixVal;
670 OverflowOccurred |= Val.udiv(RadixVal) != OldVal;
671
Reid Spencer5f016e22007-07-11 17:01:13 +0000672 // Add value, did overflow occur on the value?
Daniel Dunbard70cb642008-10-16 06:39:30 +0000673 // (a + b) ult b <=> overflow
Reid Spencer5f016e22007-07-11 17:01:13 +0000674 Val += CharVal;
Reid Spencer5f016e22007-07-11 17:01:13 +0000675 OverflowOccurred |= Val.ult(CharVal);
676 }
677 return OverflowOccurred;
678}
679
John McCall94c939d2009-12-24 09:08:04 +0000680llvm::APFloat::opStatus
681NumericLiteralParser::GetFloatValue(llvm::APFloat &Result) {
Ted Kremenek427d5af2007-11-26 23:12:30 +0000682 using llvm::APFloat;
Erick Tryzelaare9f195f2009-08-16 23:36:28 +0000683 using llvm::StringRef;
Mike Stump1eb44332009-09-09 15:08:12 +0000684
Erick Tryzelaare9f195f2009-08-16 23:36:28 +0000685 unsigned n = std::min(SuffixBegin - ThisTokBegin, ThisTokEnd - ThisTokBegin);
John McCall94c939d2009-12-24 09:08:04 +0000686 return Result.convertFromString(StringRef(ThisTokBegin, n),
687 APFloat::rmNearestTiesToEven);
Reid Spencer5f016e22007-07-11 17:01:13 +0000688}
689
Reid Spencer5f016e22007-07-11 17:01:13 +0000690
691CharLiteralParser::CharLiteralParser(const char *begin, const char *end,
692 SourceLocation Loc, Preprocessor &PP) {
693 // At this point we know that the character matches the regex "L?'.*'".
694 HadError = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000695
Reid Spencer5f016e22007-07-11 17:01:13 +0000696 // Determine if this is a wide character.
697 IsWide = begin[0] == 'L';
698 if (IsWide) ++begin;
Mike Stump1eb44332009-09-09 15:08:12 +0000699
Reid Spencer5f016e22007-07-11 17:01:13 +0000700 // Skip over the entry quote.
701 assert(begin[0] == '\'' && "Invalid token lexed");
702 ++begin;
703
Mike Stump1eb44332009-09-09 15:08:12 +0000704 // FIXME: The "Value" is an uint64_t so we can handle char literals of
Sanjiv Gupta4bc11af2009-04-21 02:21:29 +0000705 // upto 64-bits.
Reid Spencer5f016e22007-07-11 17:01:13 +0000706 // FIXME: This extensively assumes that 'char' is 8-bits.
Chris Lattner98be4942008-03-05 18:54:05 +0000707 assert(PP.getTargetInfo().getCharWidth() == 8 &&
Reid Spencer5f016e22007-07-11 17:01:13 +0000708 "Assumes char is 8 bits");
Chris Lattnere3ad8812009-04-28 21:51:46 +0000709 assert(PP.getTargetInfo().getIntWidth() <= 64 &&
710 (PP.getTargetInfo().getIntWidth() & 7) == 0 &&
711 "Assumes sizeof(int) on target is <= 64 and a multiple of char");
712 assert(PP.getTargetInfo().getWCharWidth() <= 64 &&
713 "Assumes sizeof(wchar) on target is <= 64");
Sanjiv Gupta4bc11af2009-04-21 02:21:29 +0000714
Mike Stump1eb44332009-09-09 15:08:12 +0000715 // This is what we will use for overflow detection
Sanjiv Gupta4bc11af2009-04-21 02:21:29 +0000716 llvm::APInt LitVal(PP.getTargetInfo().getIntWidth(), 0);
Mike Stump1eb44332009-09-09 15:08:12 +0000717
Chris Lattnere3ad8812009-04-28 21:51:46 +0000718 unsigned NumCharsSoFar = 0;
Chris Lattner1c6c64b2010-04-16 23:44:05 +0000719 bool Warned = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000720 while (begin[0] != '\'') {
Sanjiv Gupta4bc11af2009-04-21 02:21:29 +0000721 uint64_t ResultChar;
Nico Weber59705ae2010-10-09 00:27:47 +0000722
723 // Is this a Universal Character Name escape?
Reid Spencer5f016e22007-07-11 17:01:13 +0000724 if (begin[0] != '\\') // If this is a normal character, consume it.
725 ResultChar = *begin++;
Nico Weber59705ae2010-10-09 00:27:47 +0000726 else { // Otherwise, this is an escape character.
727 // Check for UCN.
728 if (begin[1] == 'u' || begin[1] == 'U') {
729 uint32_t utf32 = 0;
730 unsigned short UcnLen = 0;
Chris Lattner872a45e2010-11-17 06:55:10 +0000731 if (!ProcessUCNEscape(begin, end, utf32, UcnLen,
732 FullSourceLoc(Loc, PP.getSourceManager()),
Chris Lattner6c66f072010-11-17 06:46:14 +0000733 &PP.getDiagnostics(), PP.getLangOptions())) {
Nico Weber59705ae2010-10-09 00:27:47 +0000734 HadError = 1;
735 }
736 ResultChar = utf32;
737 } else {
738 // Otherwise, this is a non-UCN escape character. Process it.
Chris Lattner91f54ce2010-11-17 06:26:08 +0000739 ResultChar = ProcessCharEscape(begin, end, HadError,
740 FullSourceLoc(Loc,PP.getSourceManager()),
741 IsWide,
742 &PP.getDiagnostics(), PP.getTargetInfo());
Nico Weber59705ae2010-10-09 00:27:47 +0000743 }
744 }
Reid Spencer5f016e22007-07-11 17:01:13 +0000745
746 // If this is a multi-character constant (e.g. 'abc'), handle it. These are
747 // implementation defined (C99 6.4.4.4p10).
Chris Lattnere3ad8812009-04-28 21:51:46 +0000748 if (NumCharsSoFar) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000749 if (IsWide) {
750 // Emulate GCC's (unintentional?) behavior: L'ab' -> L'b'.
Sanjiv Gupta4bc11af2009-04-21 02:21:29 +0000751 LitVal = 0;
Reid Spencer5f016e22007-07-11 17:01:13 +0000752 } else {
753 // Narrow character literals act as though their value is concatenated
Chris Lattnere3ad8812009-04-28 21:51:46 +0000754 // in this implementation, but warn on overflow.
Chris Lattner1c6c64b2010-04-16 23:44:05 +0000755 if (LitVal.countLeadingZeros() < 8 && !Warned) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000756 PP.Diag(Loc, diag::warn_char_constant_too_large);
Chris Lattner1c6c64b2010-04-16 23:44:05 +0000757 Warned = true;
758 }
Sanjiv Gupta4bc11af2009-04-21 02:21:29 +0000759 LitVal <<= 8;
Reid Spencer5f016e22007-07-11 17:01:13 +0000760 }
761 }
Mike Stump1eb44332009-09-09 15:08:12 +0000762
Sanjiv Gupta4bc11af2009-04-21 02:21:29 +0000763 LitVal = LitVal + ResultChar;
Chris Lattnere3ad8812009-04-28 21:51:46 +0000764 ++NumCharsSoFar;
765 }
766
767 // If this is the second character being processed, do special handling.
768 if (NumCharsSoFar > 1) {
769 // Warn about discarding the top bits for multi-char wide-character
770 // constants (L'abcd').
771 if (IsWide)
772 PP.Diag(Loc, diag::warn_extraneous_wide_char_constant);
773 else if (NumCharsSoFar != 4)
774 PP.Diag(Loc, diag::ext_multichar_character_literal);
775 else
776 PP.Diag(Loc, diag::ext_four_char_character_literal);
Eli Friedman2a1c3632009-06-01 05:25:02 +0000777 IsMultiChar = true;
Daniel Dunbar930b71a2009-07-29 01:46:05 +0000778 } else
779 IsMultiChar = false;
Sanjiv Gupta4bc11af2009-04-21 02:21:29 +0000780
781 // Transfer the value from APInt to uint64_t
782 Value = LitVal.getZExtValue();
Mike Stump1eb44332009-09-09 15:08:12 +0000783
Nico Weber59705ae2010-10-09 00:27:47 +0000784 if (IsWide && PP.getLangOptions().ShortWChar && Value > 0xFFFF)
785 PP.Diag(Loc, diag::warn_ucn_escape_too_large);
786
Reid Spencer5f016e22007-07-11 17:01:13 +0000787 // If this is a single narrow character, sign extend it (e.g. '\xFF' is "-1")
788 // if 'char' is signed for this target (C99 6.4.4.4p10). Note that multiple
789 // character constants are not sign extended in the this implementation:
790 // '\xFF\xFF' = 65536 and '\x0\xFF' = 255, which matches GCC.
Chris Lattnere3ad8812009-04-28 21:51:46 +0000791 if (!IsWide && NumCharsSoFar == 1 && (Value & 128) &&
Eli Friedman15b91762009-06-05 07:05:05 +0000792 PP.getLangOptions().CharIsSigned)
Reid Spencer5f016e22007-07-11 17:01:13 +0000793 Value = (signed char)Value;
794}
795
796
797/// string-literal: [C99 6.4.5]
798/// " [s-char-sequence] "
799/// L" [s-char-sequence] "
800/// s-char-sequence:
801/// s-char
802/// s-char-sequence s-char
803/// s-char:
804/// any source character except the double quote ",
805/// backslash \, or newline character
806/// escape-character
807/// universal-character-name
808/// escape-character: [C99 6.4.4.4]
809/// \ escape-code
810/// universal-character-name
811/// escape-code:
812/// character-escape-code
813/// octal-escape-code
814/// hex-escape-code
815/// character-escape-code: one of
816/// n t b r f v a
817/// \ ' " ?
818/// octal-escape-code:
819/// octal-digit
820/// octal-digit octal-digit
821/// octal-digit octal-digit octal-digit
822/// hex-escape-code:
823/// x hex-digit
824/// hex-escape-code hex-digit
825/// universal-character-name:
826/// \u hex-quad
827/// \U hex-quad hex-quad
828/// hex-quad:
829/// hex-digit hex-digit hex-digit hex-digit
830///
831StringLiteralParser::
Chris Lattnerd2177732007-07-20 16:59:19 +0000832StringLiteralParser(const Token *StringToks, unsigned NumStringToks,
Chris Lattner6c66f072010-11-17 06:46:14 +0000833 Preprocessor &pp, bool Complain)
834 : PP(pp), SM(PP.getSourceManager()), Features(PP.getLangOptions()),
835 Target(PP.getTargetInfo()), Diags(Complain ? &PP.getDiagnostics() : 0) {
Reid Spencer5f016e22007-07-11 17:01:13 +0000836 // Scan all of the string portions, remember the max individual token length,
837 // computing a bound on the concatenated string length, and see whether any
838 // piece is a wide-string. If any of the string portions is a wide-string
839 // literal, the result is a wide-string literal [C99 6.4.5p4].
Sean Hunt6cf75022010-08-30 17:47:05 +0000840 MaxTokenLength = StringToks[0].getLength();
841 SizeBound = StringToks[0].getLength()-2; // -2 for "".
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000842 AnyWide = StringToks[0].is(tok::wide_string_literal);
Sean Hunt6cf75022010-08-30 17:47:05 +0000843
844 hadError = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000845
846 // Implement Translation Phase #6: concatenation of string literals
847 /// (C99 5.1.1.2p1). The common case is only one string fragment.
848 for (unsigned i = 1; i != NumStringToks; ++i) {
849 // The string could be shorter than this if it needs cleaning, but this is a
850 // reasonable bound, which is all we need.
Sean Hunt6cf75022010-08-30 17:47:05 +0000851 SizeBound += StringToks[i].getLength()-2; // -2 for "".
Mike Stump1eb44332009-09-09 15:08:12 +0000852
Reid Spencer5f016e22007-07-11 17:01:13 +0000853 // Remember maximum string piece length.
Sean Hunt6cf75022010-08-30 17:47:05 +0000854 if (StringToks[i].getLength() > MaxTokenLength)
855 MaxTokenLength = StringToks[i].getLength();
Mike Stump1eb44332009-09-09 15:08:12 +0000856
Reid Spencer5f016e22007-07-11 17:01:13 +0000857 // Remember if we see any wide strings.
Chris Lattner22f6bbc2007-10-09 18:02:16 +0000858 AnyWide |= StringToks[i].is(tok::wide_string_literal);
Reid Spencer5f016e22007-07-11 17:01:13 +0000859 }
Chris Lattnerdbb1ecc2009-02-26 23:01:51 +0000860
Reid Spencer5f016e22007-07-11 17:01:13 +0000861 // Include space for the null terminator.
862 ++SizeBound;
Mike Stump1eb44332009-09-09 15:08:12 +0000863
Reid Spencer5f016e22007-07-11 17:01:13 +0000864 // TODO: K&R warning: "traditional C rejects string constant concatenation"
Mike Stump1eb44332009-09-09 15:08:12 +0000865
Reid Spencer5f016e22007-07-11 17:01:13 +0000866 // Get the width in bytes of wchar_t. If no wchar_t strings are used, do not
867 // query the target. As such, wchar_tByteWidth is only valid if AnyWide=true.
868 wchar_tByteWidth = ~0U;
869 if (AnyWide) {
Chris Lattnerbbee00b2009-01-16 18:51:42 +0000870 wchar_tByteWidth = PP.getTargetInfo().getWCharWidth();
Reid Spencer5f016e22007-07-11 17:01:13 +0000871 assert((wchar_tByteWidth & 7) == 0 && "Assumes wchar_t is byte multiple!");
872 wchar_tByteWidth /= 8;
873 }
Mike Stump1eb44332009-09-09 15:08:12 +0000874
Reid Spencer5f016e22007-07-11 17:01:13 +0000875 // The output buffer size needs to be large enough to hold wide characters.
876 // This is a worst-case assumption which basically corresponds to L"" "long".
877 if (AnyWide)
878 SizeBound *= wchar_tByteWidth;
Mike Stump1eb44332009-09-09 15:08:12 +0000879
Reid Spencer5f016e22007-07-11 17:01:13 +0000880 // Size the temporary buffer to hold the result string data.
881 ResultBuf.resize(SizeBound);
Mike Stump1eb44332009-09-09 15:08:12 +0000882
Reid Spencer5f016e22007-07-11 17:01:13 +0000883 // Likewise, but for each string piece.
884 llvm::SmallString<512> TokenBuf;
885 TokenBuf.resize(MaxTokenLength);
Mike Stump1eb44332009-09-09 15:08:12 +0000886
Reid Spencer5f016e22007-07-11 17:01:13 +0000887 // Loop over all the strings, getting their spelling, and expanding them to
888 // wide strings as appropriate.
889 ResultPtr = &ResultBuf[0]; // Next byte to fill in.
Mike Stump1eb44332009-09-09 15:08:12 +0000890
Anders Carlssonee98ac52007-10-15 02:50:23 +0000891 Pascal = false;
Mike Stump1eb44332009-09-09 15:08:12 +0000892
Reid Spencer5f016e22007-07-11 17:01:13 +0000893 for (unsigned i = 0, e = NumStringToks; i != e; ++i) {
894 const char *ThisTokBuf = &TokenBuf[0];
895 // Get the spelling of the token, which eliminates trigraphs, etc. We know
896 // that ThisTokBuf points to a buffer that is big enough for the whole token
897 // and 'spelled' tokens can only shrink.
Douglas Gregor50f6af72010-03-16 05:20:39 +0000898 bool StringInvalid = false;
899 unsigned ThisTokLen = PP.getSpelling(StringToks[i], ThisTokBuf,
Sean Hunt6cf75022010-08-30 17:47:05 +0000900 &StringInvalid);
Douglas Gregor50f6af72010-03-16 05:20:39 +0000901 if (StringInvalid) {
902 hadError = 1;
903 continue;
904 }
905
Reid Spencer5f016e22007-07-11 17:01:13 +0000906 const char *ThisTokEnd = ThisTokBuf+ThisTokLen-1; // Skip end quote.
Fariborz Jahanian56bedef2010-08-31 23:34:27 +0000907 bool wide = false;
Reid Spencer5f016e22007-07-11 17:01:13 +0000908 // TODO: Input character set mapping support.
Mike Stump1eb44332009-09-09 15:08:12 +0000909
Reid Spencer5f016e22007-07-11 17:01:13 +0000910 // Skip L marker for wide strings.
Fariborz Jahanian56bedef2010-08-31 23:34:27 +0000911 if (ThisTokBuf[0] == 'L') {
912 wide = true;
Reid Spencer5f016e22007-07-11 17:01:13 +0000913 ++ThisTokBuf;
Fariborz Jahanian56bedef2010-08-31 23:34:27 +0000914 }
Mike Stump1eb44332009-09-09 15:08:12 +0000915
Reid Spencer5f016e22007-07-11 17:01:13 +0000916 assert(ThisTokBuf[0] == '"' && "Expected quote, lexer broken?");
917 ++ThisTokBuf;
Mike Stump1eb44332009-09-09 15:08:12 +0000918
Anders Carlssonee98ac52007-10-15 02:50:23 +0000919 // Check if this is a pascal string
920 if (pp.getLangOptions().PascalStrings && ThisTokBuf + 1 != ThisTokEnd &&
921 ThisTokBuf[0] == '\\' && ThisTokBuf[1] == 'p') {
Mike Stump1eb44332009-09-09 15:08:12 +0000922
Anders Carlssonee98ac52007-10-15 02:50:23 +0000923 // If the \p sequence is found in the first token, we have a pascal string
924 // Otherwise, if we already have a pascal string, ignore the first \p
925 if (i == 0) {
926 ++ThisTokBuf;
927 Pascal = true;
928 } else if (Pascal)
929 ThisTokBuf += 2;
930 }
Mike Stump1eb44332009-09-09 15:08:12 +0000931
Reid Spencer5f016e22007-07-11 17:01:13 +0000932 while (ThisTokBuf != ThisTokEnd) {
933 // Is this a span of non-escape characters?
934 if (ThisTokBuf[0] != '\\') {
935 const char *InStart = ThisTokBuf;
936 do {
937 ++ThisTokBuf;
938 } while (ThisTokBuf != ThisTokEnd && ThisTokBuf[0] != '\\');
Mike Stump1eb44332009-09-09 15:08:12 +0000939
Reid Spencer5f016e22007-07-11 17:01:13 +0000940 // Copy the character span over.
941 unsigned Len = ThisTokBuf-InStart;
942 if (!AnyWide) {
943 memcpy(ResultPtr, InStart, Len);
944 ResultPtr += Len;
945 } else {
946 // Note: our internal rep of wide char tokens is always little-endian.
947 for (; Len; --Len, ++InStart) {
948 *ResultPtr++ = InStart[0];
949 // Add zeros at the end.
950 for (unsigned i = 1, e = wchar_tByteWidth; i != e; ++i)
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000951 *ResultPtr++ = 0;
Reid Spencer5f016e22007-07-11 17:01:13 +0000952 }
953 }
954 continue;
955 }
Steve Naroff4e93b342009-04-01 11:09:15 +0000956 // Is this a Universal Character Name escape?
Steve Naroff0e3e3eb2009-03-30 23:46:03 +0000957 if (ThisTokBuf[1] == 'u' || ThisTokBuf[1] == 'U') {
Nico Weber59705ae2010-10-09 00:27:47 +0000958 EncodeUCNEscape(ThisTokBuf, ThisTokEnd, ResultPtr,
959 hadError, StringToks[i].getLocation(), PP, wide,
960 Complain);
Steve Naroff4e93b342009-04-01 11:09:15 +0000961 continue;
962 }
963 // Otherwise, this is a non-UCN escape character. Process it.
Chris Lattner91f54ce2010-11-17 06:26:08 +0000964 unsigned ResultChar =
965 ProcessCharEscape(ThisTokBuf, ThisTokEnd, hadError,
Chris Lattner6c66f072010-11-17 06:46:14 +0000966 FullSourceLoc(StringToks[i].getLocation(), SM),
967 AnyWide, Diags, Target);
Mike Stump1eb44332009-09-09 15:08:12 +0000968
Steve Naroff4e93b342009-04-01 11:09:15 +0000969 // Note: our internal rep of wide char tokens is always little-endian.
970 *ResultPtr++ = ResultChar & 0xFF;
Mike Stump1eb44332009-09-09 15:08:12 +0000971
Steve Naroff4e93b342009-04-01 11:09:15 +0000972 if (AnyWide) {
973 for (unsigned i = 1, e = wchar_tByteWidth; i != e; ++i)
974 *ResultPtr++ = ResultChar >> i*8;
Reid Spencer5f016e22007-07-11 17:01:13 +0000975 }
976 }
977 }
Mike Stump1eb44332009-09-09 15:08:12 +0000978
Chris Lattnerbbee00b2009-01-16 18:51:42 +0000979 if (Pascal) {
Anders Carlssonee98ac52007-10-15 02:50:23 +0000980 ResultBuf[0] = ResultPtr-&ResultBuf[0]-1;
Fariborz Jahanian64a80342010-05-28 19:40:48 +0000981 if (AnyWide)
982 ResultBuf[0] /= wchar_tByteWidth;
Chris Lattnerbbee00b2009-01-16 18:51:42 +0000983
984 // Verify that pascal strings aren't too large.
Douglas Gregorb90f4b32010-05-26 05:35:51 +0000985 if (GetStringLength() > 256 && Complain) {
Chris Lattnerbbee00b2009-01-16 18:51:42 +0000986 PP.Diag(StringToks[0].getLocation(), diag::err_pascal_string_too_long)
987 << SourceRange(StringToks[0].getLocation(),
988 StringToks[NumStringToks-1].getLocation());
Eli Friedman57d7dde2009-04-01 03:17:08 +0000989 hadError = 1;
990 return;
991 }
Douglas Gregor427c4922010-07-20 14:33:20 +0000992 } else if (Complain) {
993 // Complain if this string literal has too many characters.
994 unsigned MaxChars = PP.getLangOptions().CPlusPlus? 65536
995 : PP.getLangOptions().C99 ? 4095
996 : 509;
997
998 if (GetNumStringChars() > MaxChars)
999 PP.Diag(StringToks[0].getLocation(), diag::ext_string_too_long)
1000 << GetNumStringChars() << MaxChars
1001 << (PP.getLangOptions().CPlusPlus? 2
1002 : PP.getLangOptions().C99 ? 1
1003 : 0)
1004 << SourceRange(StringToks[0].getLocation(),
1005 StringToks[NumStringToks-1].getLocation());
Chris Lattnerbbee00b2009-01-16 18:51:42 +00001006 }
Reid Spencer5f016e22007-07-11 17:01:13 +00001007}
Chris Lattner719e6152009-02-18 19:21:10 +00001008
1009
1010/// getOffsetOfStringByte - This function returns the offset of the
1011/// specified byte of the string data represented by Token. This handles
1012/// advancing over escape sequences in the string.
1013unsigned StringLiteralParser::getOffsetOfStringByte(const Token &Tok,
Chris Lattner6c66f072010-11-17 06:46:14 +00001014 unsigned ByteNo) const {
Chris Lattner719e6152009-02-18 19:21:10 +00001015 // Get the spelling of the token.
Chris Lattnerca1475e2010-11-17 06:35:43 +00001016 llvm::SmallString<32> SpellingBuffer;
Sean Hunt6cf75022010-08-30 17:47:05 +00001017 SpellingBuffer.resize(Tok.getLength());
Mike Stump1eb44332009-09-09 15:08:12 +00001018
Douglas Gregor50f6af72010-03-16 05:20:39 +00001019 bool StringInvalid = false;
Chris Lattner719e6152009-02-18 19:21:10 +00001020 const char *SpellingPtr = &SpellingBuffer[0];
Chris Lattnerca1475e2010-11-17 06:35:43 +00001021 unsigned TokLen = Preprocessor::getSpelling(Tok, SpellingPtr, SM, Features,
Chris Lattner48cf9822010-11-17 06:31:48 +00001022 &StringInvalid);
Chris Lattner91f54ce2010-11-17 06:26:08 +00001023 if (StringInvalid)
Douglas Gregor50f6af72010-03-16 05:20:39 +00001024 return 0;
Chris Lattner719e6152009-02-18 19:21:10 +00001025
1026 assert(SpellingPtr[0] != 'L' && "Doesn't handle wide strings yet");
1027
Mike Stump1eb44332009-09-09 15:08:12 +00001028
Chris Lattner719e6152009-02-18 19:21:10 +00001029 const char *SpellingStart = SpellingPtr;
1030 const char *SpellingEnd = SpellingPtr+TokLen;
1031
1032 // Skip over the leading quote.
1033 assert(SpellingPtr[0] == '"' && "Should be a string literal!");
1034 ++SpellingPtr;
Mike Stump1eb44332009-09-09 15:08:12 +00001035
Chris Lattner719e6152009-02-18 19:21:10 +00001036 // Skip over bytes until we find the offset we're looking for.
1037 while (ByteNo) {
1038 assert(SpellingPtr < SpellingEnd && "Didn't find byte offset!");
Mike Stump1eb44332009-09-09 15:08:12 +00001039
Chris Lattner719e6152009-02-18 19:21:10 +00001040 // Step over non-escapes simply.
1041 if (*SpellingPtr != '\\') {
1042 ++SpellingPtr;
1043 --ByteNo;
1044 continue;
1045 }
Mike Stump1eb44332009-09-09 15:08:12 +00001046
Chris Lattner719e6152009-02-18 19:21:10 +00001047 // Otherwise, this is an escape character. Advance over it.
1048 bool HadError = false;
1049 ProcessCharEscape(SpellingPtr, SpellingEnd, HadError,
Chris Lattnerca1475e2010-11-17 06:35:43 +00001050 FullSourceLoc(Tok.getLocation(), SM),
Chris Lattner91f54ce2010-11-17 06:26:08 +00001051 false, Diags, Target);
Chris Lattner719e6152009-02-18 19:21:10 +00001052 assert(!HadError && "This method isn't valid on erroneous strings");
1053 --ByteNo;
1054 }
Mike Stump1eb44332009-09-09 15:08:12 +00001055
Chris Lattner719e6152009-02-18 19:21:10 +00001056 return SpellingPtr-SpellingStart;
1057}