lld/ELF/ScriptLexer.h

   1 //===- ScriptLexer.h --------------------------------------------*- 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 #ifndef LLD_ELF_SCRIPT_LEXER_H
  10 #define LLD_ELF_SCRIPT_LEXER_H
  11
  12 #include "lld/Common/LLVM.h"
  13 #include "llvm/ADT/StringRef.h"
  14 #include "llvm/Support/MemoryBufferRef.h"
  15 #include <vector>
  16
  17 namespace lld::elf {
  18
  19 class ScriptLexer {
  20 public:
  21   explicit ScriptLexer(MemoryBufferRef mb);
  22
  23   void setError(const Twine &msg);
  24   void tokenize(MemoryBufferRef mb);
  25   StringRef skipSpace(StringRef s);
  26   bool atEOF();
  27   StringRef next();
  28   StringRef peek();
  29   StringRef peek2();
  30   void skip();
  31   bool consume(StringRef tok);
  32   void expect(StringRef expect);
  33   bool consumeLabel(StringRef tok);
  34   std::string getCurrentLocation();
  35
  36   std::vector<MemoryBufferRef> mbs;
  37   std::vector<StringRef> tokens;
  38   bool inExpr = false;
  39   size_t pos = 0;
  40
  41   size_t lastLineNumber = 0;
  42   size_t lastLineNumberOffset = 0;
  43
  44 protected:
  45   MemoryBufferRef getCurrentMB();
  46
  47 private:
  48   void maybeSplitExpr();
  49   StringRef getLine();
  50   size_t getLineNumber();
  51   size_t getColumnNumber();
  52 };
  53
  54 } // namespace lld::elf
  55
  56 #endif