blob: 39d62bf71cff10bcfcf1ea07cb1965c0115a5f85 [file] [log] [blame]
Chris Lattner8e3a8e02007-11-18 08:46:26 +00001//===- LLLexer.cpp - Lexer for .ll Files ----------------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattner8e3a8e02007-11-18 08:46:26 +00007//
8//===----------------------------------------------------------------------===//
9//
10// Implement the Lexer for .ll files.
11//
12//===----------------------------------------------------------------------===//
13
14#include "LLLexer.h"
15#include "ParserInternals.h"
16#include "llvm/Support/MemoryBuffer.h"
Chris Lattnerd185f642007-12-08 19:03:30 +000017#include "llvm/Support/MathExtras.h"
Chris Lattner8e3a8e02007-11-18 08:46:26 +000018
19#include <list>
20#include "llvmAsmParser.h"
Anton Korobeynikovae9f3a32008-02-20 11:08:44 +000021
22#include <cstring>
Chris Lattner8e3a8e02007-11-18 08:46:26 +000023using namespace llvm;
24
25//===----------------------------------------------------------------------===//
26// Helper functions.
27//===----------------------------------------------------------------------===//
28
29// atoull - Convert an ascii string of decimal digits into the unsigned long
30// long representation... this does not have to do input error checking,
31// because we know that the input will be matched by a suitable regex...
32//
33static uint64_t atoull(const char *Buffer, const char *End) {
34 uint64_t Result = 0;
35 for (; Buffer != End; Buffer++) {
36 uint64_t OldRes = Result;
37 Result *= 10;
38 Result += *Buffer-'0';
39 if (Result < OldRes) { // Uh, oh, overflow detected!!!
40 GenerateError("constant bigger than 64 bits detected!");
41 return 0;
42 }
43 }
44 return Result;
45}
46
47static uint64_t HexIntToVal(const char *Buffer, const char *End) {
48 uint64_t Result = 0;
49 for (; Buffer != End; ++Buffer) {
50 uint64_t OldRes = Result;
51 Result *= 16;
52 char C = *Buffer;
53 if (C >= '0' && C <= '9')
54 Result += C-'0';
55 else if (C >= 'A' && C <= 'F')
56 Result += C-'A'+10;
57 else if (C >= 'a' && C <= 'f')
58 Result += C-'a'+10;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +000059
Chris Lattner8e3a8e02007-11-18 08:46:26 +000060 if (Result < OldRes) { // Uh, oh, overflow detected!!!
61 GenerateError("constant bigger than 64 bits detected!");
62 return 0;
63 }
64 }
65 return Result;
66}
67
68// HexToFP - Convert the ascii string in hexadecimal format to the floating
69// point representation of it.
70//
71static double HexToFP(const char *Buffer, const char *End) {
72 return BitsToDouble(HexIntToVal(Buffer, End)); // Cast Hex constant to double
73}
74
75static void HexToIntPair(const char *Buffer, const char *End, uint64_t Pair[2]){
76 Pair[0] = 0;
77 for (int i=0; i<16; i++, Buffer++) {
78 assert(Buffer != End);
79 Pair[0] *= 16;
80 char C = *Buffer;
81 if (C >= '0' && C <= '9')
82 Pair[0] += C-'0';
83 else if (C >= 'A' && C <= 'F')
84 Pair[0] += C-'A'+10;
85 else if (C >= 'a' && C <= 'f')
86 Pair[0] += C-'a'+10;
87 }
88 Pair[1] = 0;
89 for (int i=0; i<16 && Buffer != End; i++, Buffer++) {
90 Pair[1] *= 16;
91 char C = *Buffer;
92 if (C >= '0' && C <= '9')
93 Pair[1] += C-'0';
94 else if (C >= 'A' && C <= 'F')
95 Pair[1] += C-'A'+10;
96 else if (C >= 'a' && C <= 'f')
97 Pair[1] += C-'a'+10;
98 }
Chris Lattnerd343c6b2007-11-18 18:25:18 +000099 if (Buffer != End)
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000100 GenerateError("constant bigger than 128 bits detected!");
101}
102
103// UnEscapeLexed - Run through the specified buffer and change \xx codes to the
104// appropriate character.
105static void UnEscapeLexed(std::string &Str) {
106 if (Str.empty()) return;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000107
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000108 char *Buffer = &Str[0], *EndBuffer = Buffer+Str.size();
109 char *BOut = Buffer;
110 for (char *BIn = Buffer; BIn != EndBuffer; ) {
111 if (BIn[0] == '\\') {
112 if (BIn < EndBuffer-1 && BIn[1] == '\\') {
113 *BOut++ = '\\'; // Two \ becomes one
114 BIn += 2;
115 } else if (BIn < EndBuffer-2 && isxdigit(BIn[1]) && isxdigit(BIn[2])) {
116 char Tmp = BIn[3]; BIn[3] = 0; // Terminate string
117 *BOut = (char)strtol(BIn+1, 0, 16); // Convert to number
118 BIn[3] = Tmp; // Restore character
119 BIn += 3; // Skip over handled chars
120 ++BOut;
121 } else {
122 *BOut++ = *BIn++;
123 }
124 } else {
125 *BOut++ = *BIn++;
126 }
127 }
128 Str.resize(BOut-Buffer);
129}
130
131/// isLabelChar - Return true for [-a-zA-Z$._0-9].
132static bool isLabelChar(char C) {
133 return isalnum(C) || C == '-' || C == '$' || C == '.' || C == '_';
134}
135
136
137/// isLabelTail - Return true if this pointer points to a valid end of a label.
138static const char *isLabelTail(const char *CurPtr) {
139 while (1) {
140 if (CurPtr[0] == ':') return CurPtr+1;
141 if (!isLabelChar(CurPtr[0])) return 0;
142 ++CurPtr;
143 }
144}
145
146
147
148//===----------------------------------------------------------------------===//
149// Lexer definition.
150//===----------------------------------------------------------------------===//
151
152// FIXME: REMOVE THIS.
153#define YYEOF 0
154#define YYERROR -2
155
156LLLexer::LLLexer(MemoryBuffer *StartBuf) : CurLineNo(1), CurBuf(StartBuf) {
157 CurPtr = CurBuf->getBufferStart();
158}
159
160std::string LLLexer::getFilename() const {
161 return CurBuf->getBufferIdentifier();
162}
163
164int LLLexer::getNextChar() {
165 char CurChar = *CurPtr++;
166 switch (CurChar) {
167 default: return (unsigned char)CurChar;
168 case 0:
169 // A nul character in the stream is either the end of the current buffer or
170 // a random nul in the file. Disambiguate that here.
171 if (CurPtr-1 != CurBuf->getBufferEnd())
172 return 0; // Just whitespace.
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000173
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000174 // Otherwise, return end of file.
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000175 --CurPtr; // Another call to lex will return EOF again.
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000176 return EOF;
177 case '\n':
178 case '\r':
179 // Handle the newline character by ignoring it and incrementing the line
180 // count. However, be careful about 'dos style' files with \n\r in them.
181 // Only treat a \n\r or \r\n as a single line.
182 if ((*CurPtr == '\n' || (*CurPtr == '\r')) &&
183 *CurPtr != CurChar)
184 ++CurPtr; // Eat the two char newline sequence.
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000185
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000186 ++CurLineNo;
187 return '\n';
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000188 }
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000189}
190
191
192int LLLexer::LexToken() {
193 TokStart = CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000194
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000195 int CurChar = getNextChar();
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000196
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000197 switch (CurChar) {
198 default:
199 // Handle letters: [a-zA-Z_]
200 if (isalpha(CurChar) || CurChar == '_')
201 return LexIdentifier();
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000202
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000203 return CurChar;
204 case EOF: return YYEOF;
205 case 0:
206 case ' ':
207 case '\t':
208 case '\n':
209 case '\r':
210 // Ignore whitespace.
211 return LexToken();
212 case '+': return LexPositive();
213 case '@': return LexAt();
214 case '%': return LexPercent();
215 case '"': return LexQuote();
216 case '.':
217 if (const char *Ptr = isLabelTail(CurPtr)) {
218 CurPtr = Ptr;
219 llvmAsmlval.StrVal = new std::string(TokStart, CurPtr-1);
220 return LABELSTR;
221 }
222 if (CurPtr[0] == '.' && CurPtr[1] == '.') {
223 CurPtr += 2;
224 return DOTDOTDOT;
225 }
226 return '.';
227 case '$':
228 if (const char *Ptr = isLabelTail(CurPtr)) {
229 CurPtr = Ptr;
230 llvmAsmlval.StrVal = new std::string(TokStart, CurPtr-1);
231 return LABELSTR;
232 }
233 return '$';
234 case ';':
235 SkipLineComment();
236 return LexToken();
237 case '0': case '1': case '2': case '3': case '4':
238 case '5': case '6': case '7': case '8': case '9':
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000239 case '-':
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000240 return LexDigitOrNegative();
241 }
242}
243
244void LLLexer::SkipLineComment() {
245 while (1) {
246 if (CurPtr[0] == '\n' || CurPtr[0] == '\r' || getNextChar() == EOF)
247 return;
248 }
249}
250
251/// LexAt - Lex all tokens that start with an @ character:
252/// AtStringConstant @\"[^\"]*\"
253/// GlobalVarName @[-a-zA-Z$._][-a-zA-Z$._0-9]*
254/// GlobalVarID @[0-9]+
255int LLLexer::LexAt() {
256 // Handle AtStringConstant: @\"[^\"]*\"
257 if (CurPtr[0] == '"') {
258 ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000259
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000260 while (1) {
261 int CurChar = getNextChar();
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000262
263 if (CurChar == EOF) {
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000264 GenerateError("End of file in global variable name");
265 return YYERROR;
266 }
267 if (CurChar == '"') {
268 llvmAsmlval.StrVal = new std::string(TokStart+2, CurPtr-1);
269 UnEscapeLexed(*llvmAsmlval.StrVal);
270 return ATSTRINGCONSTANT;
271 }
272 }
273 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000274
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000275 // Handle GlobalVarName: @[-a-zA-Z$._][-a-zA-Z$._0-9]*
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000276 if (isalpha(CurPtr[0]) || CurPtr[0] == '-' || CurPtr[0] == '$' ||
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000277 CurPtr[0] == '.' || CurPtr[0] == '_') {
278 ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000279 while (isalnum(CurPtr[0]) || CurPtr[0] == '-' || CurPtr[0] == '$' ||
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000280 CurPtr[0] == '.' || CurPtr[0] == '_')
281 ++CurPtr;
282
283 llvmAsmlval.StrVal = new std::string(TokStart+1, CurPtr); // Skip @
284 return GLOBALVAR;
285 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000286
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000287 // Handle GlobalVarID: @[0-9]+
288 if (isdigit(CurPtr[0])) {
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000289 for (++CurPtr; isdigit(CurPtr[0]); ++CurPtr)
290 /*empty*/;
291
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000292 uint64_t Val = atoull(TokStart+1, CurPtr);
293 if ((unsigned)Val != Val)
294 GenerateError("Invalid value number (too large)!");
295 llvmAsmlval.UIntVal = unsigned(Val);
296 return GLOBALVAL_ID;
297 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000298
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000299 return '@';
300}
301
302
303/// LexPercent - Lex all tokens that start with a % character:
304/// PctStringConstant %\"[^\"]*\"
305/// LocalVarName %[-a-zA-Z$._][-a-zA-Z$._0-9]*
306/// LocalVarID %[0-9]+
307int LLLexer::LexPercent() {
308 // Handle PctStringConstant: %\"[^\"]*\"
309 if (CurPtr[0] == '"') {
310 ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000311
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000312 while (1) {
313 int CurChar = getNextChar();
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000314
315 if (CurChar == EOF) {
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000316 GenerateError("End of file in local variable name");
317 return YYERROR;
318 }
319 if (CurChar == '"') {
320 llvmAsmlval.StrVal = new std::string(TokStart+2, CurPtr-1);
321 UnEscapeLexed(*llvmAsmlval.StrVal);
322 return PCTSTRINGCONSTANT;
323 }
324 }
325 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000326
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000327 // Handle LocalVarName: %[-a-zA-Z$._][-a-zA-Z$._0-9]*
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000328 if (isalpha(CurPtr[0]) || CurPtr[0] == '-' || CurPtr[0] == '$' ||
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000329 CurPtr[0] == '.' || CurPtr[0] == '_') {
330 ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000331 while (isalnum(CurPtr[0]) || CurPtr[0] == '-' || CurPtr[0] == '$' ||
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000332 CurPtr[0] == '.' || CurPtr[0] == '_')
333 ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000334
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000335 llvmAsmlval.StrVal = new std::string(TokStart+1, CurPtr); // Skip %
336 return LOCALVAR;
337 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000338
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000339 // Handle LocalVarID: %[0-9]+
340 if (isdigit(CurPtr[0])) {
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000341 for (++CurPtr; isdigit(CurPtr[0]); ++CurPtr)
342 /*empty*/;
343
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000344 uint64_t Val = atoull(TokStart+1, CurPtr);
345 if ((unsigned)Val != Val)
346 GenerateError("Invalid value number (too large)!");
347 llvmAsmlval.UIntVal = unsigned(Val);
348 return LOCALVAL_ID;
349 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000350
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000351 return '%';
352}
353
354/// LexQuote - Lex all tokens that start with a " character:
355/// QuoteLabel "[^"]+":
356/// StringConstant "[^"]*"
357int LLLexer::LexQuote() {
358 while (1) {
359 int CurChar = getNextChar();
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000360
361 if (CurChar == EOF) {
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000362 GenerateError("End of file in quoted string");
363 return YYERROR;
364 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000365
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000366 if (CurChar != '"') continue;
367
368 if (CurPtr[0] != ':') {
369 llvmAsmlval.StrVal = new std::string(TokStart+1, CurPtr-1);
370 UnEscapeLexed(*llvmAsmlval.StrVal);
371 return STRINGCONSTANT;
372 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000373
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000374 ++CurPtr;
375 llvmAsmlval.StrVal = new std::string(TokStart+1, CurPtr-2);
376 UnEscapeLexed(*llvmAsmlval.StrVal);
377 return LABELSTR;
378 }
379}
380
381static bool JustWhitespaceNewLine(const char *&Ptr) {
382 const char *ThisPtr = Ptr;
383 while (*ThisPtr == ' ' || *ThisPtr == '\t')
384 ++ThisPtr;
385 if (*ThisPtr == '\n' || *ThisPtr == '\r') {
386 Ptr = ThisPtr;
387 return true;
388 }
389 return false;
390}
391
392
393/// LexIdentifier: Handle several related productions:
394/// Label [-a-zA-Z$._0-9]+:
395/// IntegerType i[0-9]+
396/// Keyword sdiv, float, ...
397/// HexIntConstant [us]0x[0-9A-Fa-f]+
398int LLLexer::LexIdentifier() {
399 const char *StartChar = CurPtr;
400 const char *IntEnd = CurPtr[-1] == 'i' ? 0 : StartChar;
401 const char *KeywordEnd = 0;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000402
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000403 for (; isLabelChar(*CurPtr); ++CurPtr) {
404 // If we decide this is an integer, remember the end of the sequence.
405 if (!IntEnd && !isdigit(*CurPtr)) IntEnd = CurPtr;
406 if (!KeywordEnd && !isalnum(*CurPtr) && *CurPtr != '_') KeywordEnd = CurPtr;
407 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000408
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000409 // If we stopped due to a colon, this really is a label.
410 if (*CurPtr == ':') {
411 llvmAsmlval.StrVal = new std::string(StartChar-1, CurPtr++);
412 return LABELSTR;
413 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000414
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000415 // Otherwise, this wasn't a label. If this was valid as an integer type,
416 // return it.
417 if (IntEnd == 0) IntEnd = CurPtr;
418 if (IntEnd != StartChar) {
419 CurPtr = IntEnd;
420 uint64_t NumBits = atoull(StartChar, CurPtr);
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000421 if (NumBits < IntegerType::MIN_INT_BITS ||
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000422 NumBits > IntegerType::MAX_INT_BITS) {
423 GenerateError("Bitwidth for integer type out of range!");
424 return YYERROR;
425 }
426 const Type* Ty = IntegerType::get(NumBits);
427 llvmAsmlval.PrimType = Ty;
428 return INTTYPE;
429 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000430
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000431 // Otherwise, this was a letter sequence. See which keyword this is.
432 if (KeywordEnd == 0) KeywordEnd = CurPtr;
433 CurPtr = KeywordEnd;
434 --StartChar;
435 unsigned Len = CurPtr-StartChar;
436#define KEYWORD(STR, TOK) \
437 if (Len == strlen(STR) && !memcmp(StartChar, STR, strlen(STR))) return TOK;
438
439 KEYWORD("begin", BEGINTOK);
440 KEYWORD("end", ENDTOK);
441 KEYWORD("true", TRUETOK);
442 KEYWORD("false", FALSETOK);
443 KEYWORD("declare", DECLARE);
444 KEYWORD("define", DEFINE);
445 KEYWORD("global", GLOBAL);
446 KEYWORD("constant", CONSTANT);
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000447
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000448 KEYWORD("internal", INTERNAL);
449 KEYWORD("linkonce", LINKONCE);
450 KEYWORD("weak", WEAK);
451 KEYWORD("appending", APPENDING);
452 KEYWORD("dllimport", DLLIMPORT);
453 KEYWORD("dllexport", DLLEXPORT);
Dale Johannesenaafce772008-05-14 20:12:51 +0000454 KEYWORD("common", COMMON);
Dan Gohmanfdfef0d2008-05-22 22:30:09 +0000455 KEYWORD("default", DEFAULT);
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000456 KEYWORD("hidden", HIDDEN);
457 KEYWORD("protected", PROTECTED);
458 KEYWORD("extern_weak", EXTERN_WEAK);
459 KEYWORD("external", EXTERNAL);
460 KEYWORD("thread_local", THREAD_LOCAL);
461 KEYWORD("zeroinitializer", ZEROINITIALIZER);
462 KEYWORD("undef", UNDEF);
463 KEYWORD("null", NULL_TOK);
464 KEYWORD("to", TO);
465 KEYWORD("tail", TAIL);
466 KEYWORD("target", TARGET);
467 KEYWORD("triple", TRIPLE);
468 KEYWORD("deplibs", DEPLIBS);
469 KEYWORD("datalayout", DATALAYOUT);
470 KEYWORD("volatile", VOLATILE);
471 KEYWORD("align", ALIGN);
Christopher Lambfe63fb92007-12-11 08:59:05 +0000472 KEYWORD("addrspace", ADDRSPACE);
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000473 KEYWORD("section", SECTION);
474 KEYWORD("alias", ALIAS);
475 KEYWORD("module", MODULE);
476 KEYWORD("asm", ASM_TOK);
477 KEYWORD("sideeffect", SIDEEFFECT);
Gordon Henriksen80a75bf2007-12-10 03:18:06 +0000478 KEYWORD("gc", GC);
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000479
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000480 KEYWORD("cc", CC_TOK);
481 KEYWORD("ccc", CCC_TOK);
482 KEYWORD("fastcc", FASTCC_TOK);
483 KEYWORD("coldcc", COLDCC_TOK);
484 KEYWORD("x86_stdcallcc", X86_STDCALLCC_TOK);
485 KEYWORD("x86_fastcallcc", X86_FASTCALLCC_TOK);
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000486
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000487 KEYWORD("signext", SIGNEXT);
488 KEYWORD("zeroext", ZEROEXT);
489 KEYWORD("inreg", INREG);
490 KEYWORD("sret", SRET);
491 KEYWORD("nounwind", NOUNWIND);
492 KEYWORD("noreturn", NORETURN);
493 KEYWORD("noalias", NOALIAS);
494 KEYWORD("byval", BYVAL);
495 KEYWORD("nest", NEST);
Duncan Sandsed4a2f12007-11-22 20:23:04 +0000496 KEYWORD("readnone", READNONE);
497 KEYWORD("readonly", READONLY);
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000498
Devang Patel2c9c3e72008-09-26 23:51:19 +0000499 KEYWORD("noinline", NOINLINE);
500 KEYWORD("alwaysinline", ALWAYSINLINE);
501 KEYWORD("optsize", OPTSIZE);
Devang Pateld4980812008-09-02 20:52:40 +0000502
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000503 KEYWORD("type", TYPE);
504 KEYWORD("opaque", OPAQUE);
505
506 KEYWORD("eq" , EQ);
507 KEYWORD("ne" , NE);
508 KEYWORD("slt", SLT);
509 KEYWORD("sgt", SGT);
510 KEYWORD("sle", SLE);
511 KEYWORD("sge", SGE);
512 KEYWORD("ult", ULT);
513 KEYWORD("ugt", UGT);
514 KEYWORD("ule", ULE);
515 KEYWORD("uge", UGE);
516 KEYWORD("oeq", OEQ);
517 KEYWORD("one", ONE);
518 KEYWORD("olt", OLT);
519 KEYWORD("ogt", OGT);
520 KEYWORD("ole", OLE);
521 KEYWORD("oge", OGE);
522 KEYWORD("ord", ORD);
523 KEYWORD("uno", UNO);
524 KEYWORD("ueq", UEQ);
525 KEYWORD("une", UNE);
526#undef KEYWORD
527
528 // Keywords for types.
529#define TYPEKEYWORD(STR, LLVMTY, TOK) \
530 if (Len == strlen(STR) && !memcmp(StartChar, STR, strlen(STR))) { \
531 llvmAsmlval.PrimType = LLVMTY; return TOK; }
532 TYPEKEYWORD("void", Type::VoidTy, VOID);
533 TYPEKEYWORD("float", Type::FloatTy, FLOAT);
534 TYPEKEYWORD("double", Type::DoubleTy, DOUBLE);
535 TYPEKEYWORD("x86_fp80", Type::X86_FP80Ty, X86_FP80);
536 TYPEKEYWORD("fp128", Type::FP128Ty, FP128);
537 TYPEKEYWORD("ppc_fp128", Type::PPC_FP128Ty, PPC_FP128);
538 TYPEKEYWORD("label", Type::LabelTy, LABEL);
539#undef TYPEKEYWORD
540
541 // Handle special forms for autoupgrading. Drop these in LLVM 3.0. This is
542 // to avoid conflicting with the sext/zext instructions, below.
543 if (Len == 4 && !memcmp(StartChar, "sext", 4)) {
544 // Scan CurPtr ahead, seeing if there is just whitespace before the newline.
545 if (JustWhitespaceNewLine(CurPtr))
546 return SIGNEXT;
547 } else if (Len == 4 && !memcmp(StartChar, "zext", 4)) {
548 // Scan CurPtr ahead, seeing if there is just whitespace before the newline.
549 if (JustWhitespaceNewLine(CurPtr))
550 return ZEROEXT;
551 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000552
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000553 // Keywords for instructions.
554#define INSTKEYWORD(STR, type, Enum, TOK) \
555 if (Len == strlen(STR) && !memcmp(StartChar, STR, strlen(STR))) { \
556 llvmAsmlval.type = Instruction::Enum; return TOK; }
557
558 INSTKEYWORD("add", BinaryOpVal, Add, ADD);
559 INSTKEYWORD("sub", BinaryOpVal, Sub, SUB);
560 INSTKEYWORD("mul", BinaryOpVal, Mul, MUL);
561 INSTKEYWORD("udiv", BinaryOpVal, UDiv, UDIV);
562 INSTKEYWORD("sdiv", BinaryOpVal, SDiv, SDIV);
563 INSTKEYWORD("fdiv", BinaryOpVal, FDiv, FDIV);
564 INSTKEYWORD("urem", BinaryOpVal, URem, UREM);
565 INSTKEYWORD("srem", BinaryOpVal, SRem, SREM);
566 INSTKEYWORD("frem", BinaryOpVal, FRem, FREM);
567 INSTKEYWORD("shl", BinaryOpVal, Shl, SHL);
568 INSTKEYWORD("lshr", BinaryOpVal, LShr, LSHR);
569 INSTKEYWORD("ashr", BinaryOpVal, AShr, ASHR);
570 INSTKEYWORD("and", BinaryOpVal, And, AND);
571 INSTKEYWORD("or", BinaryOpVal, Or , OR );
572 INSTKEYWORD("xor", BinaryOpVal, Xor, XOR);
573 INSTKEYWORD("icmp", OtherOpVal, ICmp, ICMP);
574 INSTKEYWORD("fcmp", OtherOpVal, FCmp, FCMP);
Nate Begemanac80ade2008-05-12 19:01:56 +0000575 INSTKEYWORD("vicmp", OtherOpVal, VICmp, VICMP);
576 INSTKEYWORD("vfcmp", OtherOpVal, VFCmp, VFCMP);
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000577
578 INSTKEYWORD("phi", OtherOpVal, PHI, PHI_TOK);
579 INSTKEYWORD("call", OtherOpVal, Call, CALL);
580 INSTKEYWORD("trunc", CastOpVal, Trunc, TRUNC);
581 INSTKEYWORD("zext", CastOpVal, ZExt, ZEXT);
582 INSTKEYWORD("sext", CastOpVal, SExt, SEXT);
583 INSTKEYWORD("fptrunc", CastOpVal, FPTrunc, FPTRUNC);
584 INSTKEYWORD("fpext", CastOpVal, FPExt, FPEXT);
585 INSTKEYWORD("uitofp", CastOpVal, UIToFP, UITOFP);
586 INSTKEYWORD("sitofp", CastOpVal, SIToFP, SITOFP);
587 INSTKEYWORD("fptoui", CastOpVal, FPToUI, FPTOUI);
588 INSTKEYWORD("fptosi", CastOpVal, FPToSI, FPTOSI);
589 INSTKEYWORD("inttoptr", CastOpVal, IntToPtr, INTTOPTR);
590 INSTKEYWORD("ptrtoint", CastOpVal, PtrToInt, PTRTOINT);
591 INSTKEYWORD("bitcast", CastOpVal, BitCast, BITCAST);
592 INSTKEYWORD("select", OtherOpVal, Select, SELECT);
593 INSTKEYWORD("va_arg", OtherOpVal, VAArg , VAARG);
594 INSTKEYWORD("ret", TermOpVal, Ret, RET);
595 INSTKEYWORD("br", TermOpVal, Br, BR);
596 INSTKEYWORD("switch", TermOpVal, Switch, SWITCH);
597 INSTKEYWORD("invoke", TermOpVal, Invoke, INVOKE);
598 INSTKEYWORD("unwind", TermOpVal, Unwind, UNWIND);
599 INSTKEYWORD("unreachable", TermOpVal, Unreachable, UNREACHABLE);
600
601 INSTKEYWORD("malloc", MemOpVal, Malloc, MALLOC);
602 INSTKEYWORD("alloca", MemOpVal, Alloca, ALLOCA);
603 INSTKEYWORD("free", MemOpVal, Free, FREE);
604 INSTKEYWORD("load", MemOpVal, Load, LOAD);
605 INSTKEYWORD("store", MemOpVal, Store, STORE);
606 INSTKEYWORD("getelementptr", MemOpVal, GetElementPtr, GETELEMENTPTR);
607
608 INSTKEYWORD("extractelement", OtherOpVal, ExtractElement, EXTRACTELEMENT);
609 INSTKEYWORD("insertelement", OtherOpVal, InsertElement, INSERTELEMENT);
610 INSTKEYWORD("shufflevector", OtherOpVal, ShuffleVector, SHUFFLEVECTOR);
Dan Gohmanfc74abf2008-07-23 00:34:11 +0000611 INSTKEYWORD("getresult", OtherOpVal, ExtractValue, GETRESULT);
Dan Gohmane4977cf2008-05-23 01:55:30 +0000612 INSTKEYWORD("extractvalue", OtherOpVal, ExtractValue, EXTRACTVALUE);
613 INSTKEYWORD("insertvalue", OtherOpVal, InsertValue, INSERTVALUE);
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000614#undef INSTKEYWORD
615
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000616 // Check for [us]0x[0-9A-Fa-f]+ which are Hexadecimal constant generated by
617 // the CFE to avoid forcing it to deal with 64-bit numbers.
618 if ((TokStart[0] == 'u' || TokStart[0] == 's') &&
619 TokStart[1] == '0' && TokStart[2] == 'x' && isxdigit(TokStart[3])) {
620 int len = CurPtr-TokStart-3;
621 uint32_t bits = len * 4;
622 APInt Tmp(bits, TokStart+3, len, 16);
623 uint32_t activeBits = Tmp.getActiveBits();
624 if (activeBits > 0 && activeBits < bits)
625 Tmp.trunc(activeBits);
626 if (Tmp.getBitWidth() > 64) {
627 llvmAsmlval.APIntVal = new APInt(Tmp);
628 return TokStart[0] == 's' ? ESAPINTVAL : EUAPINTVAL;
629 } else if (TokStart[0] == 's') {
630 llvmAsmlval.SInt64Val = Tmp.getSExtValue();
631 return ESINT64VAL;
632 } else {
633 llvmAsmlval.UInt64Val = Tmp.getZExtValue();
634 return EUINT64VAL;
635 }
636 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000637
Chris Lattner4ce0df62007-11-18 18:43:24 +0000638 // If this is "cc1234", return this as just "cc".
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000639 if (TokStart[0] == 'c' && TokStart[1] == 'c') {
640 CurPtr = TokStart+2;
641 return CC_TOK;
642 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000643
Chris Lattner4ce0df62007-11-18 18:43:24 +0000644 // If this starts with "call", return it as CALL. This is to support old
645 // broken .ll files. FIXME: remove this with LLVM 3.0.
646 if (CurPtr-TokStart > 4 && !memcmp(TokStart, "call", 4)) {
647 CurPtr = TokStart+4;
648 llvmAsmlval.OtherOpVal = Instruction::Call;
649 return CALL;
650 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000651
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000652 // Finally, if this isn't known, return just a single character.
653 CurPtr = TokStart+1;
654 return TokStart[0];
655}
656
657
658/// Lex0x: Handle productions that start with 0x, knowing that it matches and
659/// that this is not a label:
660/// HexFPConstant 0x[0-9A-Fa-f]+
661/// HexFP80Constant 0xK[0-9A-Fa-f]+
662/// HexFP128Constant 0xL[0-9A-Fa-f]+
663/// HexPPC128Constant 0xM[0-9A-Fa-f]+
664int LLLexer::Lex0x() {
665 CurPtr = TokStart + 2;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000666
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000667 char Kind;
668 if (CurPtr[0] >= 'K' && CurPtr[0] <= 'M') {
669 Kind = *CurPtr++;
670 } else {
671 Kind = 'J';
672 }
673
674 if (!isxdigit(CurPtr[0])) {
675 // Bad token, return it as just zero.
676 CurPtr = TokStart+1;
677 return '0';
678 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000679
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000680 while (isxdigit(CurPtr[0]))
681 ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000682
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000683 if (Kind == 'J') {
684 // HexFPConstant - Floating point constant represented in IEEE format as a
685 // hexadecimal number for when exponential notation is not precise enough.
686 // Float and double only.
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000687 llvmAsmlval.FPVal = new APFloat(HexToFP(TokStart+2, CurPtr));
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000688 return FPVAL;
689 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000690
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000691 uint64_t Pair[2];
692 HexToIntPair(TokStart+3, CurPtr, Pair);
693 switch (Kind) {
694 default: assert(0 && "Unknown kind!");
695 case 'K':
696 // F80HexFPConstant - x87 long double in hexadecimal format (10 bytes)
697 llvmAsmlval.FPVal = new APFloat(APInt(80, 2, Pair));
698 return FPVAL;
699 case 'L':
700 // F128HexFPConstant - IEEE 128-bit in hexadecimal format (16 bytes)
701 llvmAsmlval.FPVal = new APFloat(APInt(128, 2, Pair), true);
702 return FPVAL;
703 case 'M':
704 // PPC128HexFPConstant - PowerPC 128-bit in hexadecimal format (16 bytes)
705 llvmAsmlval.FPVal = new APFloat(APInt(128, 2, Pair));
706 return FPVAL;
707 }
708}
709
710/// LexIdentifier: Handle several related productions:
711/// Label [-a-zA-Z$._0-9]+:
712/// NInteger -[0-9]+
713/// FPConstant [-+]?[0-9]+[.][0-9]*([eE][-+]?[0-9]+)?
714/// PInteger [0-9]+
715/// HexFPConstant 0x[0-9A-Fa-f]+
716/// HexFP80Constant 0xK[0-9A-Fa-f]+
717/// HexFP128Constant 0xL[0-9A-Fa-f]+
718/// HexPPC128Constant 0xM[0-9A-Fa-f]+
719int LLLexer::LexDigitOrNegative() {
720 // If the letter after the negative is a number, this is probably a label.
721 if (!isdigit(TokStart[0]) && !isdigit(CurPtr[0])) {
722 // Okay, this is not a number after the -, it's probably a label.
723 if (const char *End = isLabelTail(CurPtr)) {
724 llvmAsmlval.StrVal = new std::string(TokStart, End-1);
725 CurPtr = End;
726 return LABELSTR;
727 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000728
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000729 return CurPtr[-1];
730 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000731
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000732 // At this point, it is either a label, int or fp constant.
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000733
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000734 // Skip digits, we have at least one.
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000735 for (; isdigit(CurPtr[0]); ++CurPtr)
736 /*empty*/;
737
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000738 // Check to see if this really is a label afterall, e.g. "-1:".
739 if (isLabelChar(CurPtr[0]) || CurPtr[0] == ':') {
740 if (const char *End = isLabelTail(CurPtr)) {
741 llvmAsmlval.StrVal = new std::string(TokStart, End-1);
742 CurPtr = End;
743 return LABELSTR;
744 }
745 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000746
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000747 // If the next character is a '.', then it is a fp value, otherwise its
748 // integer.
749 if (CurPtr[0] != '.') {
750 if (TokStart[0] == '0' && TokStart[1] == 'x')
751 return Lex0x();
752 unsigned Len = CurPtr-TokStart;
753 uint32_t numBits = ((Len * 64) / 19) + 2;
754 APInt Tmp(numBits, TokStart, Len, 10);
755 if (TokStart[0] == '-') {
756 uint32_t minBits = Tmp.getMinSignedBits();
757 if (minBits > 0 && minBits < numBits)
758 Tmp.trunc(minBits);
759 if (Tmp.getBitWidth() > 64) {
760 llvmAsmlval.APIntVal = new APInt(Tmp);
761 return ESAPINTVAL;
762 } else {
763 llvmAsmlval.SInt64Val = Tmp.getSExtValue();
764 return ESINT64VAL;
765 }
766 } else {
767 uint32_t activeBits = Tmp.getActiveBits();
768 if (activeBits > 0 && activeBits < numBits)
769 Tmp.trunc(activeBits);
770 if (Tmp.getBitWidth() > 64) {
771 llvmAsmlval.APIntVal = new APInt(Tmp);
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000772 return EUAPINTVAL;
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000773 } else {
774 llvmAsmlval.UInt64Val = Tmp.getZExtValue();
775 return EUINT64VAL;
776 }
777 }
778 }
779
780 ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000781
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000782 // Skip over [0-9]*([eE][-+]?[0-9]+)?
783 while (isdigit(CurPtr[0])) ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000784
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000785 if (CurPtr[0] == 'e' || CurPtr[0] == 'E') {
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000786 if (isdigit(CurPtr[1]) ||
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000787 ((CurPtr[1] == '-' || CurPtr[1] == '+') && isdigit(CurPtr[2]))) {
788 CurPtr += 2;
789 while (isdigit(CurPtr[0])) ++CurPtr;
790 }
791 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000792
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000793 llvmAsmlval.FPVal = new APFloat(atof(TokStart));
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000794 return FPVAL;
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000795}
796
797/// FPConstant [-+]?[0-9]+[.][0-9]*([eE][-+]?[0-9]+)?
798int LLLexer::LexPositive() {
799 // If the letter after the negative is a number, this is probably not a
800 // label.
801 if (!isdigit(CurPtr[0]))
802 return CurPtr[-1];
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000803
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000804 // Skip digits.
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000805 for (++CurPtr; isdigit(CurPtr[0]); ++CurPtr)
806 /*empty*/;
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000807
808 // At this point, we need a '.'.
809 if (CurPtr[0] != '.') {
810 CurPtr = TokStart+1;
811 return TokStart[0];
812 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000813
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000814 ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000815
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000816 // Skip over [0-9]*([eE][-+]?[0-9]+)?
817 while (isdigit(CurPtr[0])) ++CurPtr;
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000818
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000819 if (CurPtr[0] == 'e' || CurPtr[0] == 'E') {
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000820 if (isdigit(CurPtr[1]) ||
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000821 ((CurPtr[1] == '-' || CurPtr[1] == '+') && isdigit(CurPtr[2]))) {
822 CurPtr += 2;
823 while (isdigit(CurPtr[0])) ++CurPtr;
824 }
825 }
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000826
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000827 llvmAsmlval.FPVal = new APFloat(atof(TokStart));
Bill Wendling2c6fd8c2007-12-16 09:16:12 +0000828 return FPVAL;
Chris Lattner8e3a8e02007-11-18 08:46:26 +0000829}
830
831
832//===----------------------------------------------------------------------===//
833// Define the interface to this file.
834//===----------------------------------------------------------------------===//
835
836static LLLexer *TheLexer;
837
838void InitLLLexer(llvm::MemoryBuffer *MB) {
839 assert(TheLexer == 0 && "LL Lexer isn't reentrant yet");
840 TheLexer = new LLLexer(MB);
841}
842
843int llvmAsmlex() {
844 return TheLexer->LexToken();
845}
846const char *LLLgetTokenStart() { return TheLexer->getTokStart(); }
847unsigned LLLgetTokenLength() { return TheLexer->getTokLength(); }
848std::string LLLgetFilename() { return TheLexer->getFilename(); }
849unsigned LLLgetLineNo() { return TheLexer->getLineNo(); }
850
851void FreeLexer() {
852 delete TheLexer;
853 TheLexer = 0;
854}