1//===-- RuntimeDyldCheckerImpl.h -- RuntimeDyld test framework --*- C++ -*-===//
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#ifndef LLVM_LIB_EXECUTIONENGINE_RUNTIMEDYLD_RUNTIMEDYLDCHECKERIMPL_H
11#define LLVM_LIB_EXECUTIONENGINE_RUNTIMEDYLD_RUNTIMEDYLDCHECKERIMPL_H
12
13#include "RuntimeDyldImpl.h"
14
15namespace llvm {
16
17class RuntimeDyldCheckerImpl {
18  friend class RuntimeDyldChecker;
19  friend class RuntimeDyldImpl;
20  friend class RuntimeDyldCheckerExprEval;
21  friend class RuntimeDyldELF;
22
23public:
24  RuntimeDyldCheckerImpl(RuntimeDyld &RTDyld, MCDisassembler *Disassembler,
25                         MCInstPrinter *InstPrinter,
26                         llvm::raw_ostream &ErrStream);
27
28  bool check(StringRef CheckExpr) const;
29  bool checkAllRulesInBuffer(StringRef RulePrefix, MemoryBuffer *MemBuf) const;
30
31private:
32
33  // StubMap typedefs.
34  typedef std::map<std::string, uint64_t> StubOffsetsMap;
35  struct SectionAddressInfo {
36    uint64_t SectionID;
37    StubOffsetsMap StubOffsets;
38  };
39  typedef std::map<std::string, SectionAddressInfo> SectionMap;
40  typedef std::map<std::string, SectionMap> StubMap;
41
42  RuntimeDyldImpl &getRTDyld() const { return *RTDyld.Dyld; }
43
44  bool isSymbolValid(StringRef Symbol) const;
45  uint64_t getSymbolLocalAddr(StringRef Symbol) const;
46  uint64_t getSymbolRemoteAddr(StringRef Symbol) const;
47  uint64_t readMemoryAtAddr(uint64_t Addr, unsigned Size) const;
48
49  std::pair<const SectionAddressInfo*, std::string> findSectionAddrInfo(
50                                                   StringRef FileName,
51                                                   StringRef SectionName) const;
52
53  std::pair<uint64_t, std::string> getSectionAddr(StringRef FileName,
54                                                  StringRef SectionName,
55                                                  bool IsInsideLoad) const;
56
57  std::pair<uint64_t, std::string> getStubAddrFor(StringRef FileName,
58                                                  StringRef SectionName,
59                                                  StringRef Symbol,
60                                                  bool IsInsideLoad) const;
61  StringRef getSubsectionStartingAt(StringRef Name) const;
62
63  void registerSection(StringRef FilePath, unsigned SectionID);
64  void registerStubMap(StringRef FilePath, unsigned SectionID,
65                       const RuntimeDyldImpl::StubMap &RTDyldStubs);
66
67  RuntimeDyld &RTDyld;
68  MCDisassembler *Disassembler;
69  MCInstPrinter *InstPrinter;
70  llvm::raw_ostream &ErrStream;
71
72  StubMap Stubs;
73};
74}
75
76#endif
77