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