LLLexer.cpp revision 59d3ae6cdc4316ad338cd848251f33a236ccb36c
1//===- LLLexer.cpp - Lexer for .ll Files ----------------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// Implement the Lexer for .ll files.
11//
12//===----------------------------------------------------------------------===//
13
14#include "LLLexer.h"
15#include "llvm/ADT/StringExtras.h"
16#include "llvm/ADT/Twine.h"
17#include "llvm/Assembly/Parser.h"
18#include "llvm/IR/DerivedTypes.h"
19#include "llvm/IR/Instruction.h"
20#include "llvm/IR/LLVMContext.h"
21#include "llvm/Support/ErrorHandling.h"
22#include "llvm/Support/MathExtras.h"
23#include "llvm/Support/MemoryBuffer.h"
24#include "llvm/Support/SourceMgr.h"
25#include "llvm/Support/raw_ostream.h"
26#include <cctype>
27#include <cstdio>
28#include <cstdlib>
29#include <cstring>
30using namespace llvm;
31
32bool LLLexer::Error(LocTy ErrorLoc, const Twine &Msg) const {
33  ErrorInfo = SM.GetMessage(ErrorLoc, SourceMgr::DK_Error, Msg);
34  return true;
35}
36
37//===----------------------------------------------------------------------===//
38// Helper functions.
39//===----------------------------------------------------------------------===//
40
41// atoull - Convert an ascii string of decimal digits into the unsigned long
42// long representation... this does not have to do input error checking,
43// because we know that the input will be matched by a suitable regex...
44//
45uint64_t LLLexer::atoull(const char *Buffer, const char *End) {
46  uint64_t Result = 0;
47  for (; Buffer != End; Buffer++) {
48    uint64_t OldRes = Result;
49    Result *= 10;
50    Result += *Buffer-'0';
51    if (Result < OldRes) {  // Uh, oh, overflow detected!!!
52      Error("constant bigger than 64 bits detected!");
53      return 0;
54    }
55  }
56  return Result;
57}
58
59uint64_t LLLexer::HexIntToVal(const char *Buffer, const char *End) {
60  uint64_t Result = 0;
61  for (; Buffer != End; ++Buffer) {
62    uint64_t OldRes = Result;
63    Result *= 16;
64    Result += hexDigitValue(*Buffer);
65
66    if (Result < OldRes) {   // Uh, oh, overflow detected!!!
67      Error("constant bigger than 64 bits detected!");
68      return 0;
69    }
70  }
71  return Result;
72}
73
74void LLLexer::HexToIntPair(const char *Buffer, const char *End,
75                           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    Pair[0] += hexDigitValue(*Buffer);
81  }
82  Pair[1] = 0;
83  for (int i=0; i<16 && Buffer != End; i++, Buffer++) {
84    Pair[1] *= 16;
85    Pair[1] += hexDigitValue(*Buffer);
86  }
87  if (Buffer != End)
88    Error("constant bigger than 128 bits detected!");
89}
90
91/// FP80HexToIntPair - translate an 80 bit FP80 number (20 hexits) into
92/// { low64, high16 } as usual for an APInt.
93void LLLexer::FP80HexToIntPair(const char *Buffer, const char *End,
94                           uint64_t Pair[2]) {
95  Pair[1] = 0;
96  for (int i=0; i<4 && Buffer != End; i++, Buffer++) {
97    assert(Buffer != End);
98    Pair[1] *= 16;
99    Pair[1] += hexDigitValue(*Buffer);
100  }
101  Pair[0] = 0;
102  for (int i=0; i<16; i++, Buffer++) {
103    Pair[0] *= 16;
104    Pair[0] += hexDigitValue(*Buffer);
105  }
106  if (Buffer != End)
107    Error("constant bigger than 128 bits detected!");
108}
109
110// UnEscapeLexed - Run through the specified buffer and change \xx codes to the
111// appropriate character.
112static void UnEscapeLexed(std::string &Str) {
113  if (Str.empty()) return;
114
115  char *Buffer = &Str[0], *EndBuffer = Buffer+Str.size();
116  char *BOut = Buffer;
117  for (char *BIn = Buffer; BIn != EndBuffer; ) {
118    if (BIn[0] == '\\') {
119      if (BIn < EndBuffer-1 && BIn[1] == '\\') {
120        *BOut++ = '\\'; // Two \ becomes one
121        BIn += 2;
122      } else if (BIn < EndBuffer-2 &&
123                 isxdigit(static_cast<unsigned char>(BIn[1])) &&
124                 isxdigit(static_cast<unsigned char>(BIn[2]))) {
125        *BOut = hexDigitValue(BIn[1]) * 16 + hexDigitValue(BIn[2]);
126        BIn += 3;                           // Skip over handled chars
127        ++BOut;
128      } else {
129        *BOut++ = *BIn++;
130      }
131    } else {
132      *BOut++ = *BIn++;
133    }
134  }
135  Str.resize(BOut-Buffer);
136}
137
138/// isLabelChar - Return true for [-a-zA-Z$._0-9].
139static bool isLabelChar(char C) {
140  return isalnum(static_cast<unsigned char>(C)) || C == '-' || C == '$' ||
141         C == '.' || C == '_';
142}
143
144
145/// isLabelTail - Return true if this pointer points to a valid end of a label.
146static const char *isLabelTail(const char *CurPtr) {
147  while (1) {
148    if (CurPtr[0] == ':') return CurPtr+1;
149    if (!isLabelChar(CurPtr[0])) return 0;
150    ++CurPtr;
151  }
152}
153
154
155
156//===----------------------------------------------------------------------===//
157// Lexer definition.
158//===----------------------------------------------------------------------===//
159
160LLLexer::LLLexer(MemoryBuffer *StartBuf, SourceMgr &sm, SMDiagnostic &Err,
161                 LLVMContext &C)
162  : CurBuf(StartBuf), ErrorInfo(Err), SM(sm), Context(C), APFloatVal(0.0) {
163  CurPtr = CurBuf->getBufferStart();
164}
165
166std::string LLLexer::getFilename() const {
167  return CurBuf->getBufferIdentifier();
168}
169
170int LLLexer::getNextChar() {
171  char CurChar = *CurPtr++;
172  switch (CurChar) {
173  default: return (unsigned char)CurChar;
174  case 0:
175    // A nul character in the stream is either the end of the current buffer or
176    // a random nul in the file.  Disambiguate that here.
177    if (CurPtr-1 != CurBuf->getBufferEnd())
178      return 0;  // Just whitespace.
179
180    // Otherwise, return end of file.
181    --CurPtr;  // Another call to lex will return EOF again.
182    return EOF;
183  }
184}
185
186
187lltok::Kind LLLexer::LexToken() {
188  TokStart = CurPtr;
189
190  int CurChar = getNextChar();
191  switch (CurChar) {
192  default:
193    // Handle letters: [a-zA-Z_]
194    if (isalpha(static_cast<unsigned char>(CurChar)) || CurChar == '_')
195      return LexIdentifier();
196
197    return lltok::Error;
198  case EOF: return lltok::Eof;
199  case 0:
200  case ' ':
201  case '\t':
202  case '\n':
203  case '\r':
204    // Ignore whitespace.
205    return LexToken();
206  case '+': return LexPositive();
207  case '@': return LexAt();
208  case '%': return LexPercent();
209  case '"': return LexQuote();
210  case '.':
211    if (const char *Ptr = isLabelTail(CurPtr)) {
212      CurPtr = Ptr;
213      StrVal.assign(TokStart, CurPtr-1);
214      return lltok::LabelStr;
215    }
216    if (CurPtr[0] == '.' && CurPtr[1] == '.') {
217      CurPtr += 2;
218      return lltok::dotdotdot;
219    }
220    return lltok::Error;
221  case '$':
222    if (const char *Ptr = isLabelTail(CurPtr)) {
223      CurPtr = Ptr;
224      StrVal.assign(TokStart, CurPtr-1);
225      return lltok::LabelStr;
226    }
227    return lltok::Error;
228  case ';':
229    SkipLineComment();
230    return LexToken();
231  case '!': return LexExclaim();
232  case '#': return LexHash();
233  case '0': case '1': case '2': case '3': case '4':
234  case '5': case '6': case '7': case '8': case '9':
235  case '-':
236    return LexDigitOrNegative();
237  case '=': return lltok::equal;
238  case '[': return lltok::lsquare;
239  case ']': return lltok::rsquare;
240  case '{': return lltok::lbrace;
241  case '}': return lltok::rbrace;
242  case '<': return lltok::less;
243  case '>': return lltok::greater;
244  case '(': return lltok::lparen;
245  case ')': return lltok::rparen;
246  case ',': return lltok::comma;
247  case '*': return lltok::star;
248  case '\\': return lltok::backslash;
249  }
250}
251
252void LLLexer::SkipLineComment() {
253  while (1) {
254    if (CurPtr[0] == '\n' || CurPtr[0] == '\r' || getNextChar() == EOF)
255      return;
256  }
257}
258
259/// LexAt - Lex all tokens that start with an @ character:
260///   GlobalVar   @\"[^\"]*\"
261///   GlobalVar   @[-a-zA-Z$._][-a-zA-Z$._0-9]*
262///   GlobalVarID @[0-9]+
263lltok::Kind LLLexer::LexAt() {
264  // Handle AtStringConstant: @\"[^\"]*\"
265  if (CurPtr[0] == '"') {
266    ++CurPtr;
267
268    while (1) {
269      int CurChar = getNextChar();
270
271      if (CurChar == EOF) {
272        Error("end of file in global variable name");
273        return lltok::Error;
274      }
275      if (CurChar == '"') {
276        StrVal.assign(TokStart+2, CurPtr-1);
277        UnEscapeLexed(StrVal);
278        return lltok::GlobalVar;
279      }
280    }
281  }
282
283  // Handle GlobalVarName: @[-a-zA-Z$._][-a-zA-Z$._0-9]*
284  if (ReadVarName())
285    return lltok::GlobalVar;
286
287  // Handle GlobalVarID: @[0-9]+
288  if (isdigit(static_cast<unsigned char>(CurPtr[0]))) {
289    for (++CurPtr; isdigit(static_cast<unsigned char>(CurPtr[0])); ++CurPtr)
290      /*empty*/;
291
292    uint64_t Val = atoull(TokStart+1, CurPtr);
293    if ((unsigned)Val != Val)
294      Error("invalid value number (too large)!");
295    UIntVal = unsigned(Val);
296    return lltok::GlobalID;
297  }
298
299  return lltok::Error;
300}
301
302/// ReadString - Read a string until the closing quote.
303lltok::Kind LLLexer::ReadString(lltok::Kind kind) {
304  const char *Start = CurPtr;
305  while (1) {
306    int CurChar = getNextChar();
307
308    if (CurChar == EOF) {
309      Error("end of file in string constant");
310      return lltok::Error;
311    }
312    if (CurChar == '"') {
313      StrVal.assign(Start, CurPtr-1);
314      UnEscapeLexed(StrVal);
315      return kind;
316    }
317  }
318}
319
320/// ReadVarName - Read the rest of a token containing a variable name.
321bool LLLexer::ReadVarName() {
322  const char *NameStart = CurPtr;
323  if (isalpha(static_cast<unsigned char>(CurPtr[0])) ||
324      CurPtr[0] == '-' || CurPtr[0] == '$' ||
325      CurPtr[0] == '.' || CurPtr[0] == '_') {
326    ++CurPtr;
327    while (isalnum(static_cast<unsigned char>(CurPtr[0])) ||
328           CurPtr[0] == '-' || CurPtr[0] == '$' ||
329           CurPtr[0] == '.' || CurPtr[0] == '_')
330      ++CurPtr;
331
332    StrVal.assign(NameStart, CurPtr);
333    return true;
334  }
335  return false;
336}
337
338/// LexPercent - Lex all tokens that start with a % character:
339///   LocalVar   ::= %\"[^\"]*\"
340///   LocalVar   ::= %[-a-zA-Z$._][-a-zA-Z$._0-9]*
341///   LocalVarID ::= %[0-9]+
342lltok::Kind LLLexer::LexPercent() {
343  // Handle LocalVarName: %\"[^\"]*\"
344  if (CurPtr[0] == '"') {
345    ++CurPtr;
346    return ReadString(lltok::LocalVar);
347  }
348
349  // Handle LocalVarName: %[-a-zA-Z$._][-a-zA-Z$._0-9]*
350  if (ReadVarName())
351    return lltok::LocalVar;
352
353  // Handle LocalVarID: %[0-9]+
354  if (isdigit(static_cast<unsigned char>(CurPtr[0]))) {
355    for (++CurPtr; isdigit(static_cast<unsigned char>(CurPtr[0])); ++CurPtr)
356      /*empty*/;
357
358    uint64_t Val = atoull(TokStart+1, CurPtr);
359    if ((unsigned)Val != Val)
360      Error("invalid value number (too large)!");
361    UIntVal = unsigned(Val);
362    return lltok::LocalVarID;
363  }
364
365  return lltok::Error;
366}
367
368/// LexQuote - Lex all tokens that start with a " character:
369///   QuoteLabel        "[^"]+":
370///   StringConstant    "[^"]*"
371lltok::Kind LLLexer::LexQuote() {
372  lltok::Kind kind = ReadString(lltok::StringConstant);
373  if (kind == lltok::Error || kind == lltok::Eof)
374    return kind;
375
376  if (CurPtr[0] == ':') {
377    ++CurPtr;
378    kind = lltok::LabelStr;
379  }
380
381  return kind;
382}
383
384/// LexExclaim:
385///    !foo
386///    !
387lltok::Kind LLLexer::LexExclaim() {
388  // Lex a metadata name as a MetadataVar.
389  if (isalpha(static_cast<unsigned char>(CurPtr[0])) ||
390      CurPtr[0] == '-' || CurPtr[0] == '$' ||
391      CurPtr[0] == '.' || CurPtr[0] == '_' || CurPtr[0] == '\\') {
392    ++CurPtr;
393    while (isalnum(static_cast<unsigned char>(CurPtr[0])) ||
394           CurPtr[0] == '-' || CurPtr[0] == '$' ||
395           CurPtr[0] == '.' || CurPtr[0] == '_' || CurPtr[0] == '\\')
396      ++CurPtr;
397
398    StrVal.assign(TokStart+1, CurPtr);   // Skip !
399    UnEscapeLexed(StrVal);
400    return lltok::MetadataVar;
401  }
402  return lltok::exclaim;
403}
404
405/// LexHash - Lex all tokens that start with a # character:
406///    AttrGrpID ::= #[0-9]+
407lltok::Kind LLLexer::LexHash() {
408  // Handle AttrGrpID: #[0-9]+
409  if (isdigit(static_cast<unsigned char>(CurPtr[0]))) {
410    for (++CurPtr; isdigit(static_cast<unsigned char>(CurPtr[0])); ++CurPtr)
411      /*empty*/;
412
413    uint64_t Val = atoull(TokStart+1, CurPtr);
414    if ((unsigned)Val != Val)
415      Error("invalid value number (too large)!");
416    UIntVal = unsigned(Val);
417    return lltok::AttrGrpID;
418  }
419
420  return lltok::Error;
421}
422
423/// LexIdentifier: Handle several related productions:
424///    Label           [-a-zA-Z$._0-9]+:
425///    IntegerType     i[0-9]+
426///    Keyword         sdiv, float, ...
427///    HexIntConstant  [us]0x[0-9A-Fa-f]+
428lltok::Kind LLLexer::LexIdentifier() {
429  const char *StartChar = CurPtr;
430  const char *IntEnd = CurPtr[-1] == 'i' ? 0 : StartChar;
431  const char *KeywordEnd = 0;
432
433  for (; isLabelChar(*CurPtr); ++CurPtr) {
434    // If we decide this is an integer, remember the end of the sequence.
435    if (!IntEnd && !isdigit(static_cast<unsigned char>(*CurPtr)))
436      IntEnd = CurPtr;
437    if (!KeywordEnd && !isalnum(static_cast<unsigned char>(*CurPtr)) &&
438        *CurPtr != '_')
439      KeywordEnd = CurPtr;
440  }
441
442  // If we stopped due to a colon, this really is a label.
443  if (*CurPtr == ':') {
444    StrVal.assign(StartChar-1, CurPtr++);
445    return lltok::LabelStr;
446  }
447
448  // Otherwise, this wasn't a label.  If this was valid as an integer type,
449  // return it.
450  if (IntEnd == 0) IntEnd = CurPtr;
451  if (IntEnd != StartChar) {
452    CurPtr = IntEnd;
453    uint64_t NumBits = atoull(StartChar, CurPtr);
454    if (NumBits < IntegerType::MIN_INT_BITS ||
455        NumBits > IntegerType::MAX_INT_BITS) {
456      Error("bitwidth for integer type out of range!");
457      return lltok::Error;
458    }
459    TyVal = IntegerType::get(Context, NumBits);
460    return lltok::Type;
461  }
462
463  // Otherwise, this was a letter sequence.  See which keyword this is.
464  if (KeywordEnd == 0) KeywordEnd = CurPtr;
465  CurPtr = KeywordEnd;
466  --StartChar;
467  unsigned Len = CurPtr-StartChar;
468#define KEYWORD(STR)                                                    \
469  do {                                                                  \
470    if (Len == strlen(#STR) && !memcmp(StartChar, #STR, strlen(#STR)))  \
471      return lltok::kw_##STR;                                           \
472  } while (0)
473
474  KEYWORD(true);    KEYWORD(false);
475  KEYWORD(declare); KEYWORD(define);
476  KEYWORD(global);  KEYWORD(constant);
477
478  KEYWORD(private);
479  KEYWORD(linker_private);
480  KEYWORD(linker_private_weak);
481  KEYWORD(internal);
482  KEYWORD(available_externally);
483  KEYWORD(linkonce);
484  KEYWORD(linkonce_odr);
485  KEYWORD(weak);
486  KEYWORD(weak_odr);
487  KEYWORD(appending);
488  KEYWORD(dllimport);
489  KEYWORD(dllexport);
490  KEYWORD(common);
491  KEYWORD(default);
492  KEYWORD(hidden);
493  KEYWORD(protected);
494  KEYWORD(unnamed_addr);
495  KEYWORD(externally_initialized);
496  KEYWORD(extern_weak);
497  KEYWORD(external);
498  KEYWORD(thread_local);
499  KEYWORD(localdynamic);
500  KEYWORD(initialexec);
501  KEYWORD(localexec);
502  KEYWORD(zeroinitializer);
503  KEYWORD(undef);
504  KEYWORD(null);
505  KEYWORD(to);
506  KEYWORD(tail);
507  KEYWORD(target);
508  KEYWORD(triple);
509  KEYWORD(unwind);
510  KEYWORD(deplibs);             // FIXME: Remove in 4.0.
511  KEYWORD(datalayout);
512  KEYWORD(volatile);
513  KEYWORD(atomic);
514  KEYWORD(unordered);
515  KEYWORD(monotonic);
516  KEYWORD(acquire);
517  KEYWORD(release);
518  KEYWORD(acq_rel);
519  KEYWORD(seq_cst);
520  KEYWORD(singlethread);
521
522  KEYWORD(nnan);
523  KEYWORD(ninf);
524  KEYWORD(nsz);
525  KEYWORD(arcp);
526  KEYWORD(fast);
527  KEYWORD(nuw);
528  KEYWORD(nsw);
529  KEYWORD(exact);
530  KEYWORD(inbounds);
531  KEYWORD(align);
532  KEYWORD(addrspace);
533  KEYWORD(section);
534  KEYWORD(alias);
535  KEYWORD(module);
536  KEYWORD(asm);
537  KEYWORD(sideeffect);
538  KEYWORD(alignstack);
539  KEYWORD(inteldialect);
540  KEYWORD(gc);
541  KEYWORD(prefix);
542
543  KEYWORD(ccc);
544  KEYWORD(fastcc);
545  KEYWORD(coldcc);
546  KEYWORD(x86_stdcallcc);
547  KEYWORD(x86_fastcallcc);
548  KEYWORD(x86_thiscallcc);
549  KEYWORD(arm_apcscc);
550  KEYWORD(arm_aapcscc);
551  KEYWORD(arm_aapcs_vfpcc);
552  KEYWORD(msp430_intrcc);
553  KEYWORD(ptx_kernel);
554  KEYWORD(ptx_device);
555  KEYWORD(spir_kernel);
556  KEYWORD(spir_func);
557  KEYWORD(intel_ocl_bicc);
558  KEYWORD(x86_64_sysvcc);
559  KEYWORD(x86_64_win64cc);
560  KEYWORD(webkit_jscc);
561  KEYWORD(anyregcc);
562
563  KEYWORD(cc);
564  KEYWORD(c);
565
566  KEYWORD(attributes);
567
568  KEYWORD(alwaysinline);
569  KEYWORD(builtin);
570  KEYWORD(byval);
571  KEYWORD(cold);
572  KEYWORD(inlinehint);
573  KEYWORD(inreg);
574  KEYWORD(minsize);
575  KEYWORD(naked);
576  KEYWORD(nest);
577  KEYWORD(noalias);
578  KEYWORD(nobuiltin);
579  KEYWORD(nocapture);
580  KEYWORD(noduplicate);
581  KEYWORD(noimplicitfloat);
582  KEYWORD(noinline);
583  KEYWORD(nonlazybind);
584  KEYWORD(noredzone);
585  KEYWORD(noreturn);
586  KEYWORD(nounwind);
587  KEYWORD(optnone);
588  KEYWORD(optsize);
589  KEYWORD(readnone);
590  KEYWORD(readonly);
591  KEYWORD(returned);
592  KEYWORD(returns_twice);
593  KEYWORD(signext);
594  KEYWORD(sret);
595  KEYWORD(ssp);
596  KEYWORD(sspreq);
597  KEYWORD(sspstrong);
598  KEYWORD(sanitize_address);
599  KEYWORD(sanitize_thread);
600  KEYWORD(sanitize_memory);
601  KEYWORD(uwtable);
602  KEYWORD(zeroext);
603
604  KEYWORD(type);
605  KEYWORD(opaque);
606
607  KEYWORD(eq); KEYWORD(ne); KEYWORD(slt); KEYWORD(sgt); KEYWORD(sle);
608  KEYWORD(sge); KEYWORD(ult); KEYWORD(ugt); KEYWORD(ule); KEYWORD(uge);
609  KEYWORD(oeq); KEYWORD(one); KEYWORD(olt); KEYWORD(ogt); KEYWORD(ole);
610  KEYWORD(oge); KEYWORD(ord); KEYWORD(uno); KEYWORD(ueq); KEYWORD(une);
611
612  KEYWORD(xchg); KEYWORD(nand); KEYWORD(max); KEYWORD(min); KEYWORD(umax);
613  KEYWORD(umin);
614
615  KEYWORD(x);
616  KEYWORD(blockaddress);
617
618  KEYWORD(personality);
619  KEYWORD(cleanup);
620  KEYWORD(catch);
621  KEYWORD(filter);
622#undef KEYWORD
623
624  // Keywords for types.
625#define TYPEKEYWORD(STR, LLVMTY) \
626  if (Len == strlen(STR) && !memcmp(StartChar, STR, strlen(STR))) { \
627    TyVal = LLVMTY; return lltok::Type; }
628  TYPEKEYWORD("void",      Type::getVoidTy(Context));
629  TYPEKEYWORD("half",      Type::getHalfTy(Context));
630  TYPEKEYWORD("float",     Type::getFloatTy(Context));
631  TYPEKEYWORD("double",    Type::getDoubleTy(Context));
632  TYPEKEYWORD("x86_fp80",  Type::getX86_FP80Ty(Context));
633  TYPEKEYWORD("fp128",     Type::getFP128Ty(Context));
634  TYPEKEYWORD("ppc_fp128", Type::getPPC_FP128Ty(Context));
635  TYPEKEYWORD("label",     Type::getLabelTy(Context));
636  TYPEKEYWORD("metadata",  Type::getMetadataTy(Context));
637  TYPEKEYWORD("x86_mmx",   Type::getX86_MMXTy(Context));
638#undef TYPEKEYWORD
639
640  // Keywords for instructions.
641#define INSTKEYWORD(STR, Enum) \
642  if (Len == strlen(#STR) && !memcmp(StartChar, #STR, strlen(#STR))) { \
643    UIntVal = Instruction::Enum; return lltok::kw_##STR; }
644
645  INSTKEYWORD(add,   Add);  INSTKEYWORD(fadd,   FAdd);
646  INSTKEYWORD(sub,   Sub);  INSTKEYWORD(fsub,   FSub);
647  INSTKEYWORD(mul,   Mul);  INSTKEYWORD(fmul,   FMul);
648  INSTKEYWORD(udiv,  UDiv); INSTKEYWORD(sdiv,  SDiv); INSTKEYWORD(fdiv,  FDiv);
649  INSTKEYWORD(urem,  URem); INSTKEYWORD(srem,  SRem); INSTKEYWORD(frem,  FRem);
650  INSTKEYWORD(shl,   Shl);  INSTKEYWORD(lshr,  LShr); INSTKEYWORD(ashr,  AShr);
651  INSTKEYWORD(and,   And);  INSTKEYWORD(or,    Or);   INSTKEYWORD(xor,   Xor);
652  INSTKEYWORD(icmp,  ICmp); INSTKEYWORD(fcmp,  FCmp);
653
654  INSTKEYWORD(phi,         PHI);
655  INSTKEYWORD(call,        Call);
656  INSTKEYWORD(trunc,       Trunc);
657  INSTKEYWORD(zext,        ZExt);
658  INSTKEYWORD(sext,        SExt);
659  INSTKEYWORD(fptrunc,     FPTrunc);
660  INSTKEYWORD(fpext,       FPExt);
661  INSTKEYWORD(uitofp,      UIToFP);
662  INSTKEYWORD(sitofp,      SIToFP);
663  INSTKEYWORD(fptoui,      FPToUI);
664  INSTKEYWORD(fptosi,      FPToSI);
665  INSTKEYWORD(inttoptr,    IntToPtr);
666  INSTKEYWORD(ptrtoint,    PtrToInt);
667  INSTKEYWORD(bitcast,     BitCast);
668  INSTKEYWORD(addrspacecast, AddrSpaceCast);
669  INSTKEYWORD(select,      Select);
670  INSTKEYWORD(va_arg,      VAArg);
671  INSTKEYWORD(ret,         Ret);
672  INSTKEYWORD(br,          Br);
673  INSTKEYWORD(switch,      Switch);
674  INSTKEYWORD(indirectbr,  IndirectBr);
675  INSTKEYWORD(invoke,      Invoke);
676  INSTKEYWORD(resume,      Resume);
677  INSTKEYWORD(unreachable, Unreachable);
678
679  INSTKEYWORD(alloca,      Alloca);
680  INSTKEYWORD(load,        Load);
681  INSTKEYWORD(store,       Store);
682  INSTKEYWORD(cmpxchg,     AtomicCmpXchg);
683  INSTKEYWORD(atomicrmw,   AtomicRMW);
684  INSTKEYWORD(fence,       Fence);
685  INSTKEYWORD(getelementptr, GetElementPtr);
686
687  INSTKEYWORD(extractelement, ExtractElement);
688  INSTKEYWORD(insertelement,  InsertElement);
689  INSTKEYWORD(shufflevector,  ShuffleVector);
690  INSTKEYWORD(extractvalue,   ExtractValue);
691  INSTKEYWORD(insertvalue,    InsertValue);
692  INSTKEYWORD(landingpad,     LandingPad);
693#undef INSTKEYWORD
694
695  // Check for [us]0x[0-9A-Fa-f]+ which are Hexadecimal constant generated by
696  // the CFE to avoid forcing it to deal with 64-bit numbers.
697  if ((TokStart[0] == 'u' || TokStart[0] == 's') &&
698      TokStart[1] == '0' && TokStart[2] == 'x' &&
699      isxdigit(static_cast<unsigned char>(TokStart[3]))) {
700    int len = CurPtr-TokStart-3;
701    uint32_t bits = len * 4;
702    APInt Tmp(bits, StringRef(TokStart+3, len), 16);
703    uint32_t activeBits = Tmp.getActiveBits();
704    if (activeBits > 0 && activeBits < bits)
705      Tmp = Tmp.trunc(activeBits);
706    APSIntVal = APSInt(Tmp, TokStart[0] == 'u');
707    return lltok::APSInt;
708  }
709
710  // If this is "cc1234", return this as just "cc".
711  if (TokStart[0] == 'c' && TokStart[1] == 'c') {
712    CurPtr = TokStart+2;
713    return lltok::kw_cc;
714  }
715
716  // Finally, if this isn't known, return an error.
717  CurPtr = TokStart+1;
718  return lltok::Error;
719}
720
721
722/// Lex0x: Handle productions that start with 0x, knowing that it matches and
723/// that this is not a label:
724///    HexFPConstant     0x[0-9A-Fa-f]+
725///    HexFP80Constant   0xK[0-9A-Fa-f]+
726///    HexFP128Constant  0xL[0-9A-Fa-f]+
727///    HexPPC128Constant 0xM[0-9A-Fa-f]+
728///    HexHalfConstant   0xH[0-9A-Fa-f]+
729lltok::Kind LLLexer::Lex0x() {
730  CurPtr = TokStart + 2;
731
732  char Kind;
733  if ((CurPtr[0] >= 'K' && CurPtr[0] <= 'M') || CurPtr[0] == 'H') {
734    Kind = *CurPtr++;
735  } else {
736    Kind = 'J';
737  }
738
739  if (!isxdigit(static_cast<unsigned char>(CurPtr[0]))) {
740    // Bad token, return it as an error.
741    CurPtr = TokStart+1;
742    return lltok::Error;
743  }
744
745  while (isxdigit(static_cast<unsigned char>(CurPtr[0])))
746    ++CurPtr;
747
748  if (Kind == 'J') {
749    // HexFPConstant - Floating point constant represented in IEEE format as a
750    // hexadecimal number for when exponential notation is not precise enough.
751    // Half, Float, and double only.
752    APFloatVal = APFloat(BitsToDouble(HexIntToVal(TokStart+2, CurPtr)));
753    return lltok::APFloat;
754  }
755
756  uint64_t Pair[2];
757  switch (Kind) {
758  default: llvm_unreachable("Unknown kind!");
759  case 'K':
760    // F80HexFPConstant - x87 long double in hexadecimal format (10 bytes)
761    FP80HexToIntPair(TokStart+3, CurPtr, Pair);
762    APFloatVal = APFloat(APFloat::x87DoubleExtended, APInt(80, Pair));
763    return lltok::APFloat;
764  case 'L':
765    // F128HexFPConstant - IEEE 128-bit in hexadecimal format (16 bytes)
766    HexToIntPair(TokStart+3, CurPtr, Pair);
767    APFloatVal = APFloat(APFloat::IEEEquad, APInt(128, Pair));
768    return lltok::APFloat;
769  case 'M':
770    // PPC128HexFPConstant - PowerPC 128-bit in hexadecimal format (16 bytes)
771    HexToIntPair(TokStart+3, CurPtr, Pair);
772    APFloatVal = APFloat(APFloat::PPCDoubleDouble, APInt(128, Pair));
773    return lltok::APFloat;
774  case 'H':
775    APFloatVal = APFloat(APFloat::IEEEhalf,
776                         APInt(16,HexIntToVal(TokStart+3, CurPtr)));
777    return lltok::APFloat;
778  }
779}
780
781/// LexIdentifier: Handle several related productions:
782///    Label             [-a-zA-Z$._0-9]+:
783///    NInteger          -[0-9]+
784///    FPConstant        [-+]?[0-9]+[.][0-9]*([eE][-+]?[0-9]+)?
785///    PInteger          [0-9]+
786///    HexFPConstant     0x[0-9A-Fa-f]+
787///    HexFP80Constant   0xK[0-9A-Fa-f]+
788///    HexFP128Constant  0xL[0-9A-Fa-f]+
789///    HexPPC128Constant 0xM[0-9A-Fa-f]+
790lltok::Kind LLLexer::LexDigitOrNegative() {
791  // If the letter after the negative is not a number, this is probably a label.
792  if (!isdigit(static_cast<unsigned char>(TokStart[0])) &&
793      !isdigit(static_cast<unsigned char>(CurPtr[0]))) {
794    // Okay, this is not a number after the -, it's probably a label.
795    if (const char *End = isLabelTail(CurPtr)) {
796      StrVal.assign(TokStart, End-1);
797      CurPtr = End;
798      return lltok::LabelStr;
799    }
800
801    return lltok::Error;
802  }
803
804  // At this point, it is either a label, int or fp constant.
805
806  // Skip digits, we have at least one.
807  for (; isdigit(static_cast<unsigned char>(CurPtr[0])); ++CurPtr)
808    /*empty*/;
809
810  // Check to see if this really is a label afterall, e.g. "-1:".
811  if (isLabelChar(CurPtr[0]) || CurPtr[0] == ':') {
812    if (const char *End = isLabelTail(CurPtr)) {
813      StrVal.assign(TokStart, End-1);
814      CurPtr = End;
815      return lltok::LabelStr;
816    }
817  }
818
819  // If the next character is a '.', then it is a fp value, otherwise its
820  // integer.
821  if (CurPtr[0] != '.') {
822    if (TokStart[0] == '0' && TokStart[1] == 'x')
823      return Lex0x();
824    unsigned Len = CurPtr-TokStart;
825    uint32_t numBits = ((Len * 64) / 19) + 2;
826    APInt Tmp(numBits, StringRef(TokStart, Len), 10);
827    if (TokStart[0] == '-') {
828      uint32_t minBits = Tmp.getMinSignedBits();
829      if (minBits > 0 && minBits < numBits)
830        Tmp = Tmp.trunc(minBits);
831      APSIntVal = APSInt(Tmp, false);
832    } else {
833      uint32_t activeBits = Tmp.getActiveBits();
834      if (activeBits > 0 && activeBits < numBits)
835        Tmp = Tmp.trunc(activeBits);
836      APSIntVal = APSInt(Tmp, true);
837    }
838    return lltok::APSInt;
839  }
840
841  ++CurPtr;
842
843  // Skip over [0-9]*([eE][-+]?[0-9]+)?
844  while (isdigit(static_cast<unsigned char>(CurPtr[0]))) ++CurPtr;
845
846  if (CurPtr[0] == 'e' || CurPtr[0] == 'E') {
847    if (isdigit(static_cast<unsigned char>(CurPtr[1])) ||
848        ((CurPtr[1] == '-' || CurPtr[1] == '+') &&
849          isdigit(static_cast<unsigned char>(CurPtr[2])))) {
850      CurPtr += 2;
851      while (isdigit(static_cast<unsigned char>(CurPtr[0]))) ++CurPtr;
852    }
853  }
854
855  APFloatVal = APFloat(std::atof(TokStart));
856  return lltok::APFloat;
857}
858
859///    FPConstant  [-+]?[0-9]+[.][0-9]*([eE][-+]?[0-9]+)?
860lltok::Kind LLLexer::LexPositive() {
861  // If the letter after the negative is a number, this is probably not a
862  // label.
863  if (!isdigit(static_cast<unsigned char>(CurPtr[0])))
864    return lltok::Error;
865
866  // Skip digits.
867  for (++CurPtr; isdigit(static_cast<unsigned char>(CurPtr[0])); ++CurPtr)
868    /*empty*/;
869
870  // At this point, we need a '.'.
871  if (CurPtr[0] != '.') {
872    CurPtr = TokStart+1;
873    return lltok::Error;
874  }
875
876  ++CurPtr;
877
878  // Skip over [0-9]*([eE][-+]?[0-9]+)?
879  while (isdigit(static_cast<unsigned char>(CurPtr[0]))) ++CurPtr;
880
881  if (CurPtr[0] == 'e' || CurPtr[0] == 'E') {
882    if (isdigit(static_cast<unsigned char>(CurPtr[1])) ||
883        ((CurPtr[1] == '-' || CurPtr[1] == '+') &&
884        isdigit(static_cast<unsigned char>(CurPtr[2])))) {
885      CurPtr += 2;
886      while (isdigit(static_cast<unsigned char>(CurPtr[0]))) ++CurPtr;
887    }
888  }
889
890  APFloatVal = APFloat(std::atof(TokStart));
891  return lltok::APFloat;
892}
893