lib/AsmParser/LLLexer.cpp

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