10b57cec5SDimitry Andric //===- ScriptLexer.h --------------------------------------------*- C++ -*-===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric 90b57cec5SDimitry Andric #ifndef LLD_ELF_SCRIPT_LEXER_H 100b57cec5SDimitry Andric #define LLD_ELF_SCRIPT_LEXER_H 110b57cec5SDimitry Andric 120b57cec5SDimitry Andric #include "lld/Common/LLVM.h" 130b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h" 140b57cec5SDimitry Andric #include "llvm/Support/MemoryBuffer.h" 150b57cec5SDimitry Andric #include <utility> 160b57cec5SDimitry Andric #include <vector> 170b57cec5SDimitry Andric 180b57cec5SDimitry Andric namespace lld { 190b57cec5SDimitry Andric namespace elf { 200b57cec5SDimitry Andric 210b57cec5SDimitry Andric class ScriptLexer { 220b57cec5SDimitry Andric public: 230b57cec5SDimitry Andric explicit ScriptLexer(MemoryBufferRef mb); 240b57cec5SDimitry Andric 250b57cec5SDimitry Andric void setError(const Twine &msg); 260b57cec5SDimitry Andric void tokenize(MemoryBufferRef mb); 27e8d8bef9SDimitry Andric StringRef skipSpace(StringRef s); 280b57cec5SDimitry Andric bool atEOF(); 290b57cec5SDimitry Andric StringRef next(); 300b57cec5SDimitry Andric StringRef peek(); 310b57cec5SDimitry Andric StringRef peek2(); 320b57cec5SDimitry Andric void skip(); 330b57cec5SDimitry Andric bool consume(StringRef tok); 340b57cec5SDimitry Andric void expect(StringRef expect); 350b57cec5SDimitry Andric bool consumeLabel(StringRef tok); 360b57cec5SDimitry Andric std::string getCurrentLocation(); 370b57cec5SDimitry Andric 380b57cec5SDimitry Andric std::vector<MemoryBufferRef> mbs; 390b57cec5SDimitry Andric std::vector<StringRef> tokens; 400b57cec5SDimitry Andric bool inExpr = false; 410b57cec5SDimitry Andric size_t pos = 0; 420b57cec5SDimitry Andric 43*fe6060f1SDimitry Andric size_t lastLineNumber = 0; 44*fe6060f1SDimitry Andric size_t lastLineNumberOffset = 0; 45*fe6060f1SDimitry Andric 465ffd83dbSDimitry Andric protected: 475ffd83dbSDimitry Andric MemoryBufferRef getCurrentMB(); 485ffd83dbSDimitry Andric 490b57cec5SDimitry Andric private: 500b57cec5SDimitry Andric void maybeSplitExpr(); 510b57cec5SDimitry Andric StringRef getLine(); 520b57cec5SDimitry Andric size_t getLineNumber(); 530b57cec5SDimitry Andric size_t getColumnNumber(); 540b57cec5SDimitry Andric }; 550b57cec5SDimitry Andric 560b57cec5SDimitry Andric } // namespace elf 570b57cec5SDimitry Andric } // namespace lld 580b57cec5SDimitry Andric 590b57cec5SDimitry Andric #endif 60