1 /*===-- executionengine_ocaml.c - LLVM OCaml Glue ---------------*- 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 |* This file glues LLVM's OCaml interface to its C interface. These functions *|
11 |* are by and large transparent wrappers to the corresponding C functions. *|
12 |* *|
13 |* Note that these functions intentionally take liberties with the CAMLparamX *|
14 |* macros, since most of the parameters are not GC heap objects. *|
15 |* *|
16 \*===----------------------------------------------------------------------===*/
17
18 #include <string.h>
19 #include <assert.h>
20 #include "llvm-c/Core.h"
21 #include "llvm-c/ExecutionEngine.h"
22 #include "llvm-c/Target.h"
23 #include "caml/alloc.h"
24 #include "caml/custom.h"
25 #include "caml/fail.h"
26 #include "caml/memory.h"
27 #include "caml/callback.h"
28
29 void llvm_raise(value Prototype, char *Message);
30
31 /* unit -> bool */
llvm_ee_initialize(value Unit)32 CAMLprim value llvm_ee_initialize(value Unit) {
33 LLVMLinkInMCJIT();
34
35 return Val_bool(!LLVMInitializeNativeTarget() &&
36 !LLVMInitializeNativeAsmParser() &&
37 !LLVMInitializeNativeAsmPrinter());
38 }
39
40 /* llmodule -> llcompileroption -> ExecutionEngine.t */
llvm_ee_create(value OptRecordOpt,LLVMModuleRef M)41 CAMLprim LLVMExecutionEngineRef llvm_ee_create(value OptRecordOpt, LLVMModuleRef M) {
42 value OptRecord;
43 LLVMExecutionEngineRef MCJIT;
44 char *Error;
45 struct LLVMMCJITCompilerOptions Options;
46
47 LLVMInitializeMCJITCompilerOptions(&Options, sizeof(Options));
48 if (OptRecordOpt != Val_int(0)) {
49 OptRecord = Field(OptRecordOpt, 0);
50 Options.OptLevel = Int_val(Field(OptRecord, 0));
51 Options.CodeModel = Int_val(Field(OptRecord, 1));
52 Options.NoFramePointerElim = Int_val(Field(OptRecord, 2));
53 Options.EnableFastISel = Int_val(Field(OptRecord, 3));
54 Options.MCJMM = NULL;
55 }
56
57 if (LLVMCreateMCJITCompilerForModule(&MCJIT, M, &Options,
58 sizeof(Options), &Error))
59 llvm_raise(*caml_named_value("Llvm_executionengine.Error"), Error);
60 return MCJIT;
61 }
62
63 /* ExecutionEngine.t -> unit */
llvm_ee_dispose(LLVMExecutionEngineRef EE)64 CAMLprim value llvm_ee_dispose(LLVMExecutionEngineRef EE) {
65 LLVMDisposeExecutionEngine(EE);
66 return Val_unit;
67 }
68
69 /* llmodule -> ExecutionEngine.t -> unit */
llvm_ee_add_module(LLVMModuleRef M,LLVMExecutionEngineRef EE)70 CAMLprim value llvm_ee_add_module(LLVMModuleRef M, LLVMExecutionEngineRef EE) {
71 LLVMAddModule(EE, M);
72 return Val_unit;
73 }
74
75 /* llmodule -> ExecutionEngine.t -> llmodule */
llvm_ee_remove_module(LLVMModuleRef M,LLVMExecutionEngineRef EE)76 CAMLprim value llvm_ee_remove_module(LLVMModuleRef M, LLVMExecutionEngineRef EE) {
77 LLVMModuleRef RemovedModule;
78 char *Error;
79 if (LLVMRemoveModule(EE, M, &RemovedModule, &Error))
80 llvm_raise(*caml_named_value("Llvm_executionengine.Error"), Error);
81 return Val_unit;
82 }
83
84 /* ExecutionEngine.t -> unit */
llvm_ee_run_static_ctors(LLVMExecutionEngineRef EE)85 CAMLprim value llvm_ee_run_static_ctors(LLVMExecutionEngineRef EE) {
86 LLVMRunStaticConstructors(EE);
87 return Val_unit;
88 }
89
90 /* ExecutionEngine.t -> unit */
llvm_ee_run_static_dtors(LLVMExecutionEngineRef EE)91 CAMLprim value llvm_ee_run_static_dtors(LLVMExecutionEngineRef EE) {
92 LLVMRunStaticDestructors(EE);
93 return Val_unit;
94 }
95
96 extern value llvm_alloc_data_layout(LLVMTargetDataRef TargetData);
97
98 /* ExecutionEngine.t -> Llvm_target.DataLayout.t */
llvm_ee_get_data_layout(LLVMExecutionEngineRef EE)99 CAMLprim value llvm_ee_get_data_layout(LLVMExecutionEngineRef EE) {
100 value DataLayout;
101 LLVMTargetDataRef OrigDataLayout;
102 char* TargetDataCStr;
103
104 OrigDataLayout = LLVMGetExecutionEngineTargetData(EE);
105 TargetDataCStr = LLVMCopyStringRepOfTargetData(OrigDataLayout);
106 DataLayout = llvm_alloc_data_layout(LLVMCreateTargetData(TargetDataCStr));
107 LLVMDisposeMessage(TargetDataCStr);
108
109 return DataLayout;
110 }
111
112 /* Llvm.llvalue -> int64 -> llexecutionengine -> unit */
llvm_ee_add_global_mapping(LLVMValueRef Global,value Ptr,LLVMExecutionEngineRef EE)113 CAMLprim value llvm_ee_add_global_mapping(LLVMValueRef Global, value Ptr,
114 LLVMExecutionEngineRef EE) {
115 LLVMAddGlobalMapping(EE, Global, (void*) (Int64_val(Ptr)));
116 return Val_unit;
117 }
118
llvm_ee_get_global_value_address(value Name,LLVMExecutionEngineRef EE)119 CAMLprim value llvm_ee_get_global_value_address(value Name,
120 LLVMExecutionEngineRef EE) {
121 return caml_copy_int64((int64_t) LLVMGetGlobalValueAddress(EE, String_val(Name)));
122 }
123
llvm_ee_get_function_address(value Name,LLVMExecutionEngineRef EE)124 CAMLprim value llvm_ee_get_function_address(value Name,
125 LLVMExecutionEngineRef EE) {
126 return caml_copy_int64((int64_t) LLVMGetFunctionAddress(EE, String_val(Name)));
127 }
128