mirror of
https://github.com/RPCS3/llvm.git
synced 2025-05-14 01:16:26 +00:00
154d0643db
than a pre-allocated slab of stubs. Also add a convenience method for creating a single stub, rather than a whole block a time. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@251658 91177308-0d34-0410-b5e6-96231b3b80d8
88 lines
2.7 KiB
C++
88 lines
2.7 KiB
C++
//===----- CompileOnDemandLayerTest.cpp - Unit tests for the COD layer ----===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "OrcTestCommon.h"
|
|
#include "llvm/ExecutionEngine/Orc/CompileOnDemandLayer.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::orc;
|
|
|
|
namespace {
|
|
|
|
class DummyCallbackManager : public orc::JITCompileCallbackManagerBase {
|
|
public:
|
|
DummyCallbackManager()
|
|
: JITCompileCallbackManagerBase(0, 0), NextStubAddress(0),
|
|
UniversalCompile([]() { return 0; }) {
|
|
}
|
|
|
|
CompileCallbackInfo getCompileCallback(LLVMContext &Context) override {
|
|
return CompileCallbackInfo(++NextStubAddress, UniversalCompile);
|
|
}
|
|
public:
|
|
TargetAddress NextStubAddress;
|
|
CompileFtor UniversalCompile;
|
|
};
|
|
|
|
class DummyStubsManager : public orc::IndirectStubsManagerBase {
|
|
public:
|
|
std::error_code createStub(StringRef StubName, TargetAddress InitAddr,
|
|
JITSymbolFlags Flags) override {
|
|
llvm_unreachable("Not implemented");
|
|
}
|
|
|
|
std::error_code createStubs(const StubInitsMap &StubInits) override {
|
|
llvm_unreachable("Not implemented");
|
|
}
|
|
|
|
JITSymbol findStub(StringRef Name, bool ExportedStubsOnly) override {
|
|
llvm_unreachable("Not implemented");
|
|
}
|
|
|
|
JITSymbol findPointer(StringRef Name) override {
|
|
llvm_unreachable("Not implemented");
|
|
}
|
|
|
|
std::error_code updatePointer(StringRef Name,
|
|
TargetAddress NewAddr) override {
|
|
llvm_unreachable("Not implemented");
|
|
}
|
|
};
|
|
|
|
TEST(CompileOnDemandLayerTest, FindSymbol) {
|
|
auto MockBaseLayer =
|
|
createMockBaseLayer<int>(DoNothingAndReturn<int>(0),
|
|
DoNothingAndReturn<void>(),
|
|
[](const std::string &Name, bool) {
|
|
if (Name == "foo")
|
|
return JITSymbol(1, JITSymbolFlags::Exported);
|
|
return JITSymbol(nullptr);
|
|
},
|
|
DoNothingAndReturn<JITSymbol>(nullptr));
|
|
|
|
typedef decltype(MockBaseLayer) MockBaseLayerT;
|
|
DummyCallbackManager CallbackMgr;
|
|
auto StubsMgrBuilder =
|
|
[]() {
|
|
return llvm::make_unique<DummyStubsManager>();
|
|
};
|
|
|
|
llvm::orc::CompileOnDemandLayer<MockBaseLayerT>
|
|
COD(MockBaseLayer,
|
|
[](Function &F) { std::set<Function*> S; S.insert(&F); return S; },
|
|
CallbackMgr, StubsMgrBuilder, true);
|
|
auto Sym = COD.findSymbol("foo", true);
|
|
|
|
EXPECT_TRUE(!!Sym)
|
|
<< "CompileOnDemand::findSymbol should call findSymbol in the base layer.";
|
|
}
|
|
|
|
}
|