summaryrefslogtreecommitdiffstats
path: root/llvm/lib/ExecutionEngine/RuntimeDyld/RuntimeDyldCheckerImpl.h
blob: 0debd29cfd3c96b7859c50676a655c7b55a4cabb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
//===-- RuntimeDyldCheckerImpl.h -- RuntimeDyld test framework --*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_LIB_EXECUTIONENGINE_RUNTIMEDYLD_RUNTIMEDYLDCHECKERIMPL_H
#define LLVM_LIB_EXECUTIONENGINE_RUNTIMEDYLD_RUNTIMEDYLDCHECKERIMPL_H

#include "RuntimeDyldImpl.h"

namespace llvm {

class RuntimeDyldCheckerImpl {
  friend class RuntimeDyldChecker;
  friend class RuntimeDyldCheckerExprEval;

  using IsSymbolValidFunction =
    RuntimeDyldChecker::IsSymbolValidFunction;
  using GetSymbolAddressFunction =
    RuntimeDyldChecker::GetSymbolAddressFunction;
  using GetSymbolContentFunction =
    RuntimeDyldChecker::GetSymbolContentFunction;

  using GetSectionLoadAddressFunction =
    RuntimeDyldChecker::GetSectionLoadAddressFunction;
  using GetSectionContentFunction =
    RuntimeDyldChecker::GetSectionContentFunction;

  using GetStubOffsetInSectionFunction =
    RuntimeDyldChecker::GetStubOffsetInSectionFunction;

public:
  RuntimeDyldCheckerImpl(
                      IsSymbolValidFunction IsSymbolValid,
                      GetSymbolAddressFunction GetSymbolAddress,
                      GetSymbolContentFunction GetSymbolContent,
                      GetSectionLoadAddressFunction GetSectionLoadAddress,
                      GetSectionContentFunction GetSectionContent,
                      GetStubOffsetInSectionFunction GetStubOffsetInSection,
                      support::endianness Endianness,
                      MCDisassembler *Disassembler,
                      MCInstPrinter *InstPrinter,
                      llvm::raw_ostream &ErrStream);

  bool check(StringRef CheckExpr) const;
  bool checkAllRulesInBuffer(StringRef RulePrefix, MemoryBuffer *MemBuf) const;

private:

  // StubMap typedefs.

  Expected<JITSymbolResolver::LookupResult>
  lookup(const JITSymbolResolver::LookupSet &Symbols) const;

  bool isSymbolValid(StringRef Symbol) const;
  uint64_t getSymbolLocalAddr(StringRef Symbol) const;
  uint64_t getSymbolRemoteAddr(StringRef Symbol) const;
  uint64_t readMemoryAtAddr(uint64_t Addr, unsigned Size) const;

  StringRef getSymbolContent(StringRef Symbol) const;

  std::pair<uint64_t, std::string> getSectionAddr(StringRef FileName,
                                                  StringRef SectionName,
                                                  bool IsInsideLoad) const;

  std::pair<uint64_t, std::string> getStubAddrFor(StringRef FileName,
                                                  StringRef SectionName,
                                                  StringRef Symbol,
                                                  bool IsInsideLoad) const;

  Optional<uint64_t> getSectionLoadAddress(void *LocalAddr) const;

  IsSymbolValidFunction IsSymbolValid;
  GetSymbolAddressFunction GetSymbolAddress;
  GetSymbolContentFunction GetSymbolContent;
  GetSectionLoadAddressFunction GetSectionLoadAddress;
  GetSectionContentFunction GetSectionContent;
  GetStubOffsetInSectionFunction GetStubOffsetInSection;
  support::endianness Endianness;
  MCDisassembler *Disassembler;
  MCInstPrinter *InstPrinter;
  llvm::raw_ostream &ErrStream;
};
}

#endif
OpenPOWER on IntegriCloud