xref: /freebsd/contrib/llvm-project/llvm/lib/ExecutionEngine/RuntimeDyld/RuntimeDyldCheckerImpl.h (revision 6966ac055c3b7a39266fb982493330df7a097997)
1 //===-- RuntimeDyldCheckerImpl.h -- RuntimeDyld test framework --*- 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 LLVM_LIB_EXECUTIONENGINE_RUNTIMEDYLD_RUNTIMEDYLDCHECKERIMPL_H
10 #define LLVM_LIB_EXECUTIONENGINE_RUNTIMEDYLD_RUNTIMEDYLDCHECKERIMPL_H
11 
12 #include "RuntimeDyldImpl.h"
13 
14 namespace llvm {
15 
16 class RuntimeDyldCheckerImpl {
17   friend class RuntimeDyldChecker;
18   friend class RuntimeDyldCheckerExprEval;
19 
20   using IsSymbolValidFunction =
21     RuntimeDyldChecker::IsSymbolValidFunction;
22   using GetSymbolInfoFunction = RuntimeDyldChecker::GetSymbolInfoFunction;
23   using GetSectionInfoFunction = RuntimeDyldChecker::GetSectionInfoFunction;
24   using GetStubInfoFunction = RuntimeDyldChecker::GetStubInfoFunction;
25   using GetGOTInfoFunction = RuntimeDyldChecker::GetGOTInfoFunction;
26 
27 public:
28   RuntimeDyldCheckerImpl(
29       IsSymbolValidFunction IsSymbolValid, GetSymbolInfoFunction GetSymbolInfo,
30       GetSectionInfoFunction GetSectionInfo, GetStubInfoFunction GetStubInfo,
31       GetGOTInfoFunction GetGOTInfo, support::endianness Endianness,
32       MCDisassembler *Disassembler, MCInstPrinter *InstPrinter,
33       llvm::raw_ostream &ErrStream);
34 
35   bool check(StringRef CheckExpr) const;
36   bool checkAllRulesInBuffer(StringRef RulePrefix, MemoryBuffer *MemBuf) const;
37 
38 private:
39 
40   // StubMap typedefs.
41 
42   Expected<JITSymbolResolver::LookupResult>
43   lookup(const JITSymbolResolver::LookupSet &Symbols) const;
44 
45   bool isSymbolValid(StringRef Symbol) const;
46   uint64_t getSymbolLocalAddr(StringRef Symbol) const;
47   uint64_t getSymbolRemoteAddr(StringRef Symbol) const;
48   uint64_t readMemoryAtAddr(uint64_t Addr, unsigned Size) const;
49 
50   StringRef getSymbolContent(StringRef Symbol) const;
51 
52   std::pair<uint64_t, std::string> getSectionAddr(StringRef FileName,
53                                                   StringRef SectionName,
54                                                   bool IsInsideLoad) const;
55 
56   std::pair<uint64_t, std::string>
57   getStubOrGOTAddrFor(StringRef StubContainerName, StringRef Symbol,
58                       bool IsInsideLoad, bool IsStubAddr) const;
59 
60   Optional<uint64_t> getSectionLoadAddress(void *LocalAddr) const;
61 
62   IsSymbolValidFunction IsSymbolValid;
63   GetSymbolInfoFunction GetSymbolInfo;
64   GetSectionInfoFunction GetSectionInfo;
65   GetStubInfoFunction GetStubInfo;
66   GetGOTInfoFunction GetGOTInfo;
67   support::endianness Endianness;
68   MCDisassembler *Disassembler;
69   MCInstPrinter *InstPrinter;
70   llvm::raw_ostream &ErrStream;
71 };
72 }
73 
74 #endif
75