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