1
//===-- X86AsmLexer.cpp - Tokenize X86 assembly to AsmTokens --------------===//
3
// The LLVM Compiler Infrastructure
5
// This file is distributed under the University of Illinois Open Source
6
// License. See LICENSE.TXT for details.
8
//===----------------------------------------------------------------------===//
10
#include "llvm/ADT/SmallVector.h"
11
#include "llvm/ADT/StringExtras.h"
12
#include "llvm/Target/TargetAsmLexer.h"
13
#include "llvm/Target/TargetRegistry.h"
14
#include "llvm/MC/MCAsmInfo.h"
15
#include "llvm/MC/MCParser/MCAsmLexer.h"
16
#include "llvm/MC/MCParser/MCParsedAsmOperand.h"
23
class X86AsmLexer : public TargetAsmLexer {
24
const MCAsmInfo &AsmInfo;
26
bool tentativeIsValid;
27
AsmToken tentativeToken;
29
const AsmToken &lexTentative() {
30
tentativeToken = getLexer()->Lex();
31
tentativeIsValid = true;
32
return tentativeToken;
35
const AsmToken &lexDefinite() {
36
if(tentativeIsValid) {
37
tentativeIsValid = false;
38
return tentativeToken;
41
return getLexer()->Lex();
45
AsmToken LexTokenATT();
46
AsmToken LexTokenIntel();
50
SetError(SMLoc(), "No MCAsmLexer installed");
51
return AsmToken(AsmToken::Error, "", 0);
54
switch (AsmInfo.getAssemblerDialect()) {
56
SetError(SMLoc(), "Unhandled dialect");
57
return AsmToken(AsmToken::Error, "", 0);
61
return LexTokenIntel();
65
X86AsmLexer(const Target &T, const MCAsmInfo &MAI)
66
: TargetAsmLexer(T), AsmInfo(MAI), tentativeIsValid(false) {
72
static unsigned MatchRegisterName(StringRef Name);
74
AsmToken X86AsmLexer::LexTokenATT() {
75
const AsmToken lexedToken = lexDefinite();
77
switch (lexedToken.getKind()) {
79
return AsmToken(lexedToken);
81
SetError(Lexer->getErrLoc(), Lexer->getErr());
82
return AsmToken(lexedToken);
83
case AsmToken::Percent:
85
const AsmToken &nextToken = lexTentative();
86
if (nextToken.getKind() == AsmToken::Identifier) {
87
unsigned regID = MatchRegisterName(nextToken.getString());
92
StringRef regStr(lexedToken.getString().data(),
93
lexedToken.getString().size() +
94
nextToken.getString().size());
96
return AsmToken(AsmToken::Register,
98
static_cast<int64_t>(regID));
101
return AsmToken(lexedToken);
105
return AsmToken(lexedToken);
111
AsmToken X86AsmLexer::LexTokenIntel() {
112
const AsmToken &lexedToken = lexDefinite();
114
switch(lexedToken.getKind()) {
116
return AsmToken(lexedToken);
117
case AsmToken::Error:
118
SetError(Lexer->getErrLoc(), Lexer->getErr());
119
return AsmToken(lexedToken);
120
case AsmToken::Identifier:
122
std::string upperCase = lexedToken.getString().str();
123
std::string lowerCase = LowercaseString(upperCase);
124
StringRef lowerRef(lowerCase);
126
unsigned regID = MatchRegisterName(lowerRef);
129
return AsmToken(AsmToken::Register,
130
lexedToken.getString(),
131
static_cast<int64_t>(regID));
134
return AsmToken(lexedToken);
140
extern "C" void LLVMInitializeX86AsmLexer() {
141
RegisterAsmLexer<X86AsmLexer> X(TheX86_32Target);
142
RegisterAsmLexer<X86AsmLexer> Y(TheX86_64Target);
145
#define REGISTERS_ONLY
146
#include "X86GenAsmMatcher.inc"
147
#undef REGISTERS_ONLY