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
17namespace lld::elf {
18
19class ScriptLexer {
20public:
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  MemoryBufferRef getCurrentMB();
36
37  std::vector<MemoryBufferRef> mbs;
38  std::vector<StringRef> tokens;
39  bool inExpr = false;
40  size_t pos = 0;
41
42  size_t lastLineNumber = 0;
43  size_t lastLineNumberOffset = 0;
44
45private:
46  void maybeSplitExpr();
47  StringRef getLine();
48  size_t getLineNumber();
49  size_t getColumnNumber();
50};
51
52} // namespace lld::elf
53
54#endif
55