Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 1 | //===-- ARMAsmLexer.cpp - Tokenize ARM assembly to AsmTokens --------------===// |
| 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 | #include "ARM.h" |
| 11 | #include "ARMTargetMachine.h" |
| 12 | |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 13 | #include "llvm/MC/MCAsmInfo.h" |
| 14 | #include "llvm/MC/MCParser/MCAsmLexer.h" |
| 15 | #include "llvm/MC/MCParser/MCParsedAsmOperand.h" |
| 16 | |
| 17 | #include "llvm/Target/TargetAsmLexer.h" |
| 18 | #include "llvm/Target/TargetMachine.h" |
| 19 | #include "llvm/Target/TargetRegistry.h" |
| 20 | |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 21 | #include "llvm/ADT/OwningPtr.h" |
| 22 | #include "llvm/ADT/SmallVector.h" |
| 23 | #include "llvm/ADT/StringExtras.h" |
| 24 | #include "llvm/ADT/StringSwitch.h" |
| 25 | |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 26 | #include <string> |
| 27 | #include <map> |
| 28 | |
| 29 | using namespace llvm; |
| 30 | |
| 31 | namespace { |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 32 | |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 33 | class ARMBaseAsmLexer : public TargetAsmLexer { |
| 34 | const MCAsmInfo &AsmInfo; |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 35 | |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 36 | const AsmToken &lexDefinite() { |
| 37 | return getLexer()->Lex(); |
| 38 | } |
| 39 | |
| 40 | AsmToken LexTokenUAL(); |
| 41 | protected: |
| 42 | typedef std::map <std::string, unsigned> rmap_ty; |
| 43 | |
| 44 | rmap_ty RegisterMap; |
| 45 | |
| 46 | void InitRegisterMap(const TargetRegisterInfo *info) { |
| 47 | unsigned numRegs = info->getNumRegs(); |
| 48 | |
| 49 | for (unsigned i = 0; i < numRegs; ++i) { |
| 50 | const char *regName = info->getName(i); |
| 51 | if (regName) |
| 52 | RegisterMap[regName] = i; |
| 53 | } |
| 54 | } |
| 55 | |
| 56 | unsigned MatchRegisterName(StringRef Name) { |
| 57 | rmap_ty::iterator iter = RegisterMap.find(Name.str()); |
| 58 | if (iter != RegisterMap.end()) |
| 59 | return iter->second; |
| 60 | else |
| 61 | return 0; |
| 62 | } |
| 63 | |
| 64 | AsmToken LexToken() { |
| 65 | if (!Lexer) { |
| 66 | SetError(SMLoc(), "No MCAsmLexer installed"); |
| 67 | return AsmToken(AsmToken::Error, "", 0); |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 68 | } |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 69 | |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 70 | switch (AsmInfo.getAssemblerDialect()) { |
| 71 | default: |
| 72 | SetError(SMLoc(), "Unhandled dialect"); |
| 73 | return AsmToken(AsmToken::Error, "", 0); |
| 74 | case 0: |
| 75 | return LexTokenUAL(); |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 76 | } |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 77 | } |
| 78 | public: |
| 79 | ARMBaseAsmLexer(const Target &T, const MCAsmInfo &MAI) |
| 80 | : TargetAsmLexer(T), AsmInfo(MAI) { |
| 81 | } |
| 82 | }; |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 83 | |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 84 | class ARMAsmLexer : public ARMBaseAsmLexer { |
| 85 | public: |
| 86 | ARMAsmLexer(const Target &T, const MCAsmInfo &MAI) |
| 87 | : ARMBaseAsmLexer(T, MAI) { |
| 88 | std::string tripleString("arm-unknown-unknown"); |
| 89 | std::string featureString; |
| 90 | OwningPtr<const TargetMachine> |
| 91 | targetMachine(T.createTargetMachine(tripleString, featureString)); |
| 92 | InitRegisterMap(targetMachine->getRegisterInfo()); |
| 93 | } |
| 94 | }; |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 95 | |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 96 | class ThumbAsmLexer : public ARMBaseAsmLexer { |
| 97 | public: |
| 98 | ThumbAsmLexer(const Target &T, const MCAsmInfo &MAI) |
| 99 | : ARMBaseAsmLexer(T, MAI) { |
| 100 | std::string tripleString("thumb-unknown-unknown"); |
| 101 | std::string featureString; |
| 102 | OwningPtr<const TargetMachine> |
| 103 | targetMachine(T.createTargetMachine(tripleString, featureString)); |
| 104 | InitRegisterMap(targetMachine->getRegisterInfo()); |
| 105 | } |
| 106 | }; |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 107 | |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 108 | } // end anonymous namespace |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 109 | |
| 110 | AsmToken ARMBaseAsmLexer::LexTokenUAL() { |
| 111 | const AsmToken &lexedToken = lexDefinite(); |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 112 | |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 113 | switch (lexedToken.getKind()) { |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 114 | default: break; |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 115 | case AsmToken::Error: |
| 116 | SetError(Lexer->getErrLoc(), Lexer->getErr()); |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 117 | break; |
| 118 | case AsmToken::Identifier: { |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 119 | std::string upperCase = lexedToken.getString().str(); |
| 120 | std::string lowerCase = LowercaseString(upperCase); |
| 121 | StringRef lowerRef(lowerCase); |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 122 | |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 123 | unsigned regID = MatchRegisterName(lowerRef); |
Jim Grosbach | 940c8e5 | 2010-12-23 23:19:54 +0000 | [diff] [blame] | 124 | // Check for register aliases. |
| 125 | // r13 -> sp |
| 126 | // r14 -> lr |
| 127 | // r15 -> pc |
| 128 | // ip -> r12 |
| 129 | // FIXME: Some assemblers support lots of others. Do we want them all? |
| 130 | if (!regID) { |
Jim Grosbach | f12eee7 | 2010-12-24 00:03:39 +0000 | [diff] [blame] | 131 | regID = StringSwitch<unsigned>(lowerCase) |
| 132 | .Case("r13", ARM::SP) |
| 133 | .Case("r14", ARM::LR) |
| 134 | .Case("r15", ARM::PC) |
| 135 | .Case("ip", ARM::R12) |
| 136 | .Default(0); |
Jim Grosbach | 940c8e5 | 2010-12-23 23:19:54 +0000 | [diff] [blame] | 137 | } |
Jim Grosbach | 9584bd8 | 2010-12-22 23:26:02 +0000 | [diff] [blame] | 138 | |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 139 | if (regID) |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 140 | return AsmToken(AsmToken::Register, |
| 141 | lexedToken.getString(), |
| 142 | static_cast<int64_t>(regID)); |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 143 | } |
| 144 | } |
Bill Wendling | 2c502f9 | 2011-01-03 23:59:05 +0000 | [diff] [blame] | 145 | |
| 146 | return AsmToken(lexedToken); |
Sean Callanan | 90b7097 | 2010-04-07 20:29:34 +0000 | [diff] [blame] | 147 | } |
| 148 | |
| 149 | extern "C" void LLVMInitializeARMAsmLexer() { |
| 150 | RegisterAsmLexer<ARMAsmLexer> X(TheARMTarget); |
| 151 | RegisterAsmLexer<ThumbAsmLexer> Y(TheThumbTarget); |
| 152 | } |