1 //===-EDToken.cpp - LLVM Enhanced Disassembler ----------------------------===//
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 // This file implements the Enhanced Disassembler library's token class. The
11 // token is responsible for vending information about the token, such as its
12 // type and logical value.
14 //===----------------------------------------------------------------------===//
16 #include "EDDisassembler.h"
19 #include "llvm/ADT/SmallVector.h"
20 #include "llvm/MC/MCParser/MCAsmLexer.h"
21 #include "llvm/MC/MCParser/MCParsedAsmOperand.h"
25 EDToken::EDToken(StringRef str,
28 EDDisassembler &disassembler) :
29 Disassembler(disassembler),
39 void EDToken::makeLiteral(bool sign, uint64_t absoluteValue) {
42 LiteralAbsoluteValue = absoluteValue;
45 void EDToken::makeRegister(unsigned registerID) {
46 Type = kTokenRegister;
47 RegisterID = registerID;
50 void EDToken::setOperandID(int operandID) {
51 OperandID = operandID;
54 enum EDToken::tokenType EDToken::type() const {
58 uint64_t EDToken::localType() const {
62 StringRef EDToken::string() const {
66 int EDToken::operandID() const {
70 int EDToken::literalSign() const {
71 if(Type != kTokenLiteral)
73 return (LiteralSign ? 1 : 0);
76 int EDToken::literalAbsoluteValue(uint64_t &value) const {
77 if(Type != kTokenLiteral)
79 value = LiteralAbsoluteValue;
83 int EDToken::registerID(unsigned ®isterID) const {
84 if(Type != kTokenRegister)
86 registerID = RegisterID;
90 int EDToken::tokenize(std::vector<EDToken*> &tokens,
92 const char *operandOrder,
93 EDDisassembler &disassembler) {
94 SmallVector<MCParsedAsmOperand*, 5> parsedOperands;
95 SmallVector<AsmToken, 10> asmTokens;
97 disassembler.parseInst(parsedOperands, asmTokens, str);
99 SmallVectorImpl<MCParsedAsmOperand*>::iterator operandIterator;
100 unsigned int operandIndex;
101 SmallVectorImpl<AsmToken>::iterator tokenIterator;
103 operandIterator = parsedOperands.begin();
106 bool readOpcode = false;
108 for (tokenIterator = asmTokens.begin();
109 tokenIterator != asmTokens.end();
111 SMLoc tokenLoc = tokenIterator->getLoc();
113 while (operandIterator != parsedOperands.end() &&
114 tokenLoc.getPointer() >
115 (*operandIterator)->getEndLoc().getPointer()) {
122 switch (tokenIterator->getKind()) {
123 case AsmToken::Identifier:
125 token = new EDToken(tokenIterator->getString(),
126 EDToken::kTokenOpcode,
127 (uint64_t)tokenIterator->getKind(),
132 // any identifier that isn't an opcode is mere punctuation; so we fall
135 token = new EDToken(tokenIterator->getString(),
136 EDToken::kTokenPunctuation,
137 (uint64_t)tokenIterator->getKind(),
140 case AsmToken::Integer:
142 token = new EDToken(tokenIterator->getString(),
143 EDToken::kTokenLiteral,
144 (uint64_t)tokenIterator->getKind(),
147 int64_t intVal = tokenIterator->getIntVal();
150 token->makeLiteral(true, -intVal);
152 token->makeLiteral(false, intVal);
155 case AsmToken::Register:
157 token = new EDToken(tokenIterator->getString(),
158 EDToken::kTokenLiteral,
159 (uint64_t)tokenIterator->getKind(),
162 token->makeRegister((unsigned)tokenIterator->getRegVal());
167 if(operandIterator != parsedOperands.end() &&
168 tokenLoc.getPointer() >=
169 (*operandIterator)->getStartLoc().getPointer()) {
170 token->setOperandID(operandOrder[operandIndex]);
173 tokens.push_back(token);
179 int EDToken::getString(const char*& buf) {
180 if(PermStr.length() == 0) {
183 buf = PermStr.c_str();