1 //===----- CompileOnDemandLayerTest.cpp - Unit tests for the COD layer ----===//
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 #include "llvm/ExecutionEngine/Orc/CompileOnDemandLayer.h"
11 #include "OrcTestCommon.h"
12 #include "gtest/gtest.h"
13 
14 using namespace llvm;
15 using namespace llvm::orc;
16 
17 namespace {
18 
19 class DummyCallbackManager : public orc::JITCompileCallbackManager {
20 public:
21   DummyCallbackManager(ExecutionSession &ES)
22       : JITCompileCallbackManager(ES, 0) {}
23 
24 public:
25   Error grow() override { llvm_unreachable("not implemented"); }
26 };
27 
28 class DummyStubsManager : public orc::IndirectStubsManager {
29 public:
30   Error createStub(StringRef StubName, JITTargetAddress InitAddr,
31                    JITSymbolFlags Flags) override {
32     llvm_unreachable("Not implemented");
33   }
34 
35   Error createStubs(const StubInitsMap &StubInits) override {
36     llvm_unreachable("Not implemented");
37   }
38 
MCJITTestAPICommon()39   JITEvaluatedSymbol findStub(StringRef Name, bool ExportedStubsOnly) override {
40     llvm_unreachable("Not implemented");
41   }
42 
43   JITEvaluatedSymbol findPointer(StringRef Name) override {
44     llvm_unreachable("Not implemented");
45   }
46 
47   Error updatePointer(StringRef Name, JITTargetAddress NewAddr) override {
48     llvm_unreachable("Not implemented");
49   }
50 };
51 
52 TEST(CompileOnDemandLayerTest, FindSymbol) {
53   MockBaseLayer<int, std::shared_ptr<Module>> TestBaseLayer;
54   TestBaseLayer.findSymbolImpl =
55     [](const std::string &Name, bool) {
56       if (Name == "foo")
ArchSupportsMCJIT()57         return JITSymbol(1, JITSymbolFlags::Exported);
58       return JITSymbol(nullptr);
59     };
60 
61 
62   ExecutionSession ES(std::make_shared<SymbolStringPool>());
63   DummyCallbackManager CallbackMgr(ES);
64 
65   auto GetResolver =
66       [](orc::VModuleKey) -> std::shared_ptr<llvm::orc::SymbolResolver> {
67     llvm_unreachable("Should never be called");
68   };
69 
70   auto SetResolver = [](orc::VModuleKey, std::shared_ptr<orc::SymbolResolver>) {
71     llvm_unreachable("Should never be called");
72   };
73 
74   llvm::orc::CompileOnDemandLayer<decltype(TestBaseLayer)> COD(
75       ES, TestBaseLayer, GetResolver, SetResolver,
76       [](Function &F) { return std::set<Function *>{&F}; }, CallbackMgr,
OSSupportsMCJIT()77       [] { return llvm::make_unique<DummyStubsManager>(); }, true);
78 
79   auto Sym = COD.findSymbol("foo", true);
80 
81   EXPECT_TRUE(!!Sym) << "CompileOnDemand::findSymbol should call findSymbol in "
82                         "the base layer.";
83 }
84 }
85