1 //===--------------- OrcCAPITest.cpp - Unit tests Orc C API ---------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "OrcTestCommon.h"
11 #include "gtest/gtest.h"
12 #include "llvm-c/OrcBindings.h"
13 #include "llvm-c/Target.h"
14 #include "llvm-c/TargetMachine.h"
22 DEFINE_SIMPLE_CONVERSION_FUNCTIONS(TargetMachine, LLVMTargetMachineRef);
24 class OrcCAPIExecutionTest : public testing::Test, public OrcExecutionTest {
27 std::unique_ptr<Module> createTestModule(const Triple &TT) {
28 ModuleBuilder MB(getGlobalContext(), TT.str(), "");
29 Function *TestFunc = MB.createFunctionDecl<int()>("testFunc");
30 Function *Main = MB.createFunctionDecl<int(int, char*[])>("main");
32 Main->getBasicBlockList().push_back(BasicBlock::Create(getGlobalContext()));
33 IRBuilder<> B(&Main->back());
34 Value* Result = B.CreateCall(TestFunc);
37 return MB.takeModule();
40 typedef int (*MainFnTy)(void);
42 static int myTestFuncImpl(void) {
46 static char *testFuncName;
48 static uint64_t myResolver(const char *Name, void *Ctx) {
49 if (!strncmp(Name, testFuncName, 8))
50 return (uint64_t)&myTestFuncImpl;
54 struct CompileContext {
55 CompileContext() : Compiled(false) { }
57 OrcCAPIExecutionTest* APIExecTest;
58 std::unique_ptr<Module> M;
59 LLVMOrcModuleHandle H;
63 static LLVMOrcTargetAddress myCompileCallback(LLVMOrcJITStackRef JITStack,
65 CompileContext *CCtx = static_cast<CompileContext*>(Ctx);
66 auto *ET = CCtx->APIExecTest;
67 CCtx->M = ET->createTestModule(ET->TM->getTargetTriple());
68 CCtx->H = LLVMOrcAddEagerlyCompiledIR(JITStack, wrap(CCtx->M.get()),
70 CCtx->Compiled = true;
71 LLVMOrcTargetAddress MainAddr = LLVMOrcGetSymbolAddress(JITStack, "main");
72 LLVMOrcSetIndirectStubPointer(JITStack, "foo", MainAddr);
78 char *OrcCAPIExecutionTest::testFuncName = 0;
80 TEST_F(OrcCAPIExecutionTest, TestEagerIRCompilation) {
84 LLVMOrcJITStackRef JIT =
85 LLVMOrcCreateInstance(wrap(TM.get()), LLVMGetGlobalContext());
87 std::unique_ptr<Module> M = createTestModule(TM->getTargetTriple());
89 LLVMOrcGetMangledSymbol(JIT, &testFuncName, "testFunc");
91 LLVMOrcModuleHandle H =
92 LLVMOrcAddEagerlyCompiledIR(JIT, wrap(M.get()), myResolver, 0);
93 MainFnTy MainFn = (MainFnTy)LLVMOrcGetSymbolAddress(JIT, "main");
94 int Result = MainFn();
96 << "Eagerly JIT'd code did not return expected result";
98 LLVMOrcRemoveModule(JIT, H);
100 LLVMOrcDisposeMangledSymbol(testFuncName);
101 LLVMOrcDisposeInstance(JIT);
104 TEST_F(OrcCAPIExecutionTest, TestLazyIRCompilation) {
108 LLVMOrcJITStackRef JIT =
109 LLVMOrcCreateInstance(wrap(TM.get()), LLVMGetGlobalContext());
111 std::unique_ptr<Module> M = createTestModule(TM->getTargetTriple());
113 LLVMOrcGetMangledSymbol(JIT, &testFuncName, "testFunc");
115 LLVMOrcModuleHandle H =
116 LLVMOrcAddLazilyCompiledIR(JIT, wrap(M.get()), myResolver, 0);
117 MainFnTy MainFn = (MainFnTy)LLVMOrcGetSymbolAddress(JIT, "main");
118 int Result = MainFn();
119 EXPECT_EQ(Result, 42)
120 << "Lazily JIT'd code did not return expected result";
122 LLVMOrcRemoveModule(JIT, H);
124 LLVMOrcDisposeMangledSymbol(testFuncName);
125 LLVMOrcDisposeInstance(JIT);
128 TEST_F(OrcCAPIExecutionTest, TestDirectCallbacksAPI) {
132 LLVMOrcJITStackRef JIT =
133 LLVMOrcCreateInstance(wrap(TM.get()), LLVMGetGlobalContext());
135 LLVMOrcGetMangledSymbol(JIT, &testFuncName, "testFunc");
138 C.APIExecTest = this;
139 LLVMOrcCreateIndirectStub(JIT, "foo",
140 LLVMOrcCreateLazyCompileCallback(JIT,
143 MainFnTy FooFn = (MainFnTy)LLVMOrcGetSymbolAddress(JIT, "foo");
144 int Result = FooFn();
145 EXPECT_TRUE(C.Compiled)
146 << "Function wasn't lazily compiled";
147 EXPECT_EQ(Result, 42)
148 << "Direct-callback JIT'd code did not return expected result";
152 EXPECT_FALSE(C.Compiled)
153 << "Direct-callback JIT'd code was JIT'd twice";
155 LLVMOrcRemoveModule(JIT, C.H);
157 LLVMOrcDisposeMangledSymbol(testFuncName);
158 LLVMOrcDisposeInstance(JIT);