1 //===- AsmLexer.h - Lexer for Assembly Files --------------------*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This class declares the lexer for assembly files. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #ifndef LLVM_MC_MCPARSER_ASMLEXER_H 14 #define LLVM_MC_MCPARSER_ASMLEXER_H 15 16 #include "llvm/ADT/StringRef.h" 17 #include "llvm/MC/MCParser/MCAsmLexer.h" 18 #include <string> 19 20 namespace llvm { 21 22 class MCAsmInfo; 23 24 /// AsmLexer - Lexer class for assembly files. 25 class AsmLexer : public MCAsmLexer { 26 const MCAsmInfo &MAI; 27 28 const char *CurPtr = nullptr; 29 StringRef CurBuf; 30 bool IsAtStartOfLine = true; 31 bool IsAtStartOfStatement = true; 32 bool IsPeeking = false; 33 34 protected: 35 /// LexToken - Read the next token and return its code. 36 AsmToken LexToken() override; 37 38 public: 39 AsmLexer(const MCAsmInfo &MAI); 40 AsmLexer(const AsmLexer &) = delete; 41 AsmLexer &operator=(const AsmLexer &) = delete; 42 ~AsmLexer() override; 43 44 void setBuffer(StringRef Buf, const char *ptr = nullptr); 45 46 StringRef LexUntilEndOfStatement() override; 47 48 size_t peekTokens(MutableArrayRef<AsmToken> Buf, 49 bool ShouldSkipSpace = true) override; 50 51 const MCAsmInfo &getMAI() const { return MAI; } 52 53 private: 54 bool isAtStartOfComment(const char *Ptr); 55 bool isAtStatementSeparator(const char *Ptr); 56 int getNextChar(); 57 AsmToken ReturnError(const char *Loc, const std::string &Msg); 58 59 AsmToken LexIdentifier(); 60 AsmToken LexSlash(); 61 AsmToken LexLineComment(); 62 AsmToken LexDigit(); 63 AsmToken LexSingleQuote(); 64 AsmToken LexQuote(); 65 AsmToken LexFloatLiteral(); 66 AsmToken LexHexFloatLiteral(bool NoIntDigits); 67 68 StringRef LexUntilEndOfLine(); 69 }; 70 71 } // end namespace llvm 72 73 #endif // LLVM_MC_MCPARSER_ASMLEXER_H 74