Home | History | Annotate | Download | only in executionengine 
      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 */
     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 */
     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 */
     64 CAMLprim value llvm_ee_dispose(LLVMExecutionEngineRef EE) {
     65   LLVMDisposeExecutionEngine(EE);
     66   return Val_unit;
     67 }
     68 
     69 /* llmodule -> ExecutionEngine.t -> unit */
     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 */
     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 */
     85 CAMLprim value llvm_ee_run_static_ctors(LLVMExecutionEngineRef EE) {
     86   LLVMRunStaticConstructors(EE);
     87   return Val_unit;
     88 }
     89 
     90 /* ExecutionEngine.t -> unit */
     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 */
     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 */
    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 
    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 
    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