bindings/ocaml/executionengine/executionengine_ocaml.c

   1 /*===-- analysis_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 "llvm-c/ExecutionEngine.h"
  19 #include "caml/alloc.h"
  20 #include "caml/custom.h"
  21 #include "caml/fail.h"
  22 #include "caml/memory.h"
  23 #include <string.h>
  24 #include <assert.h>
  25
  26
  27 /* Can't use the recommended caml_named_value mechanism for backwards
  28    compatibility reasons. This is largely equivalent. */
  29 static value llvm_ee_error_exn;
  30
  31 CAMLprim value llvm_register_ee_exns(value Error) {
  32   llvm_ee_error_exn = Field(Error, 0);
  33   register_global_root(&llvm_ee_error_exn);
  34   return Val_unit;
  35 }
  36
  37 static void llvm_raise(value Prototype, char *Message) {
  38   CAMLparam1(Prototype);
  39   CAMLlocal1(CamlMessage);
  40
  41   CamlMessage = copy_string(Message);
  42   LLVMDisposeMessage(Message);
  43
  44   raise_with_arg(Prototype, CamlMessage);
  45   abort(); /* NOTREACHED */
  46 #ifdef CAMLnoreturn
  47   CAMLnoreturn; /* Silences warnings, but is missing in some versions. */
  48 #endif
  49 }
  50
  51
  52 /*--... Operations on generic values .......................................--*/
  53
  54 #define Genericvalue_val(v)  (*(LLVMGenericValueRef *)(Data_custom_val(v)))
  55
  56 static void llvm_finalize_generic_value(value GenVal) {
  57   LLVMDisposeGenericValue(Genericvalue_val(GenVal));
  58 }
  59
  60 static struct custom_operations generic_value_ops = {
  61   (char *) "LLVMGenericValue",
  62   llvm_finalize_generic_value,
  63   custom_compare_default,
  64   custom_hash_default,
  65   custom_serialize_default,
  66   custom_deserialize_default
  67 };
  68
  69 static value alloc_generic_value(LLVMGenericValueRef Ref) {
  70   value Val = alloc_custom(&generic_value_ops, sizeof(LLVMGenericValueRef), 0, 1);
  71   Genericvalue_val(Val) = Ref;
  72   return Val;
  73 }
  74
  75 /* Llvm.lltype -> float -> t */
  76 CAMLprim value llvm_genericvalue_of_float(LLVMTypeRef Ty, value N) {
  77   CAMLparam1(N);
  78   CAMLreturn(alloc_generic_value(
  79     LLVMCreateGenericValueOfFloat(Ty, Double_val(N))));
  80 }
  81
  82 /* 'a -> t */
  83 CAMLprim value llvm_genericvalue_of_value(value V) {
  84   CAMLparam1(V);
  85   CAMLreturn(alloc_generic_value(LLVMCreateGenericValueOfPointer(Op_val(V))));
  86 }
  87
  88 /* Llvm.lltype -> int -> t */
  89 CAMLprim value llvm_genericvalue_of_int(LLVMTypeRef Ty, value Int) {
  90   return alloc_generic_value(LLVMCreateGenericValueOfInt(Ty, Int_val(Int), 1));
  91 }
  92
  93 /* Llvm.lltype -> int32 -> t */
  94 CAMLprim value llvm_genericvalue_of_int32(LLVMTypeRef Ty, value Int32) {
  95   CAMLparam1(Int32);
  96   CAMLreturn(alloc_generic_value(
  97     LLVMCreateGenericValueOfInt(Ty, Int32_val(Int32), 1)));
  98 }
  99
 100 /* Llvm.lltype -> nativeint -> t */
 101 CAMLprim value llvm_genericvalue_of_nativeint(LLVMTypeRef Ty, value NatInt) {
 102   CAMLparam1(NatInt);
 103   CAMLreturn(alloc_generic_value(
 104     LLVMCreateGenericValueOfInt(Ty, Nativeint_val(NatInt), 1)));
 105 }
 106
 107 /* Llvm.lltype -> int64 -> t */
 108 CAMLprim value llvm_genericvalue_of_int64(LLVMTypeRef Ty, value Int64) {
 109   CAMLparam1(Int64);
 110   CAMLreturn(alloc_generic_value(
 111     LLVMCreateGenericValueOfInt(Ty, Int64_val(Int64), 1)));
 112 }
 113
 114 /* Llvm.lltype -> t -> float */
 115 CAMLprim value llvm_genericvalue_as_float(LLVMTypeRef Ty, value GenVal) {
 116   CAMLparam1(GenVal);
 117   CAMLreturn(copy_double(
 118     LLVMGenericValueToFloat(Ty, Genericvalue_val(GenVal))));
 119 }
 120
 121 /* t -> 'a */
 122 CAMLprim value llvm_genericvalue_as_value(value GenVal) {
 123   return Val_op(LLVMGenericValueToPointer(Genericvalue_val(GenVal)));
 124 }
 125
 126 /* t -> int */
 127 CAMLprim value llvm_genericvalue_as_int(value GenVal) {
 128   assert(LLVMGenericValueIntWidth(Genericvalue_val(GenVal)) <= 8 * sizeof(value)
 129          && "Generic value too wide to treat as an int!");
 130   return Val_int(LLVMGenericValueToInt(Genericvalue_val(GenVal), 1));
 131 }
 132
 133 /* t -> int32 */
 134 CAMLprim value llvm_genericvalue_as_int32(value GenVal) {
 135   CAMLparam1(GenVal);
 136   assert(LLVMGenericValueIntWidth(Genericvalue_val(GenVal)) <= 32
 137          && "Generic value too wide to treat as an int32!");
 138   CAMLreturn(copy_int32(LLVMGenericValueToInt(Genericvalue_val(GenVal), 1)));
 139 }
 140
 141 /* t -> int64 */
 142 CAMLprim value llvm_genericvalue_as_int64(value GenVal) {
 143   CAMLparam1(GenVal);
 144   assert(LLVMGenericValueIntWidth(Genericvalue_val(GenVal)) <= 64
 145          && "Generic value too wide to treat as an int64!");
 146   CAMLreturn(copy_int64(LLVMGenericValueToInt(Genericvalue_val(GenVal), 1)));
 147 }
 148
 149 /* t -> nativeint */
 150 CAMLprim value llvm_genericvalue_as_nativeint(value GenVal) {
 151   CAMLparam1(GenVal);
 152   assert(LLVMGenericValueIntWidth(Genericvalue_val(GenVal)) <= 8 * sizeof(value)
 153          && "Generic value too wide to treat as a nativeint!");
 154   CAMLreturn(copy_nativeint(LLVMGenericValueToInt(Genericvalue_val(GenVal),1)));
 155 }
 156
 157
 158 /*--... Operations on execution engines ....................................--*/
 159
 160 /* llmoduleprovider -> ExecutionEngine.t */
 161 CAMLprim LLVMExecutionEngineRef llvm_ee_create(LLVMModuleProviderRef MP) {
 162   LLVMExecutionEngineRef Interp;
 163   char *Error;
 164   if (LLVMCreateExecutionEngine(&Interp, MP, &Error))
 165     llvm_raise(llvm_ee_error_exn, Error);
 166   return Interp;
 167 }
 168
 169 /* llmoduleprovider -> ExecutionEngine.t */
 170 CAMLprim LLVMExecutionEngineRef
 171 llvm_ee_create_interpreter(LLVMModuleProviderRef MP) {
 172   LLVMExecutionEngineRef Interp;
 173   char *Error;
 174   if (LLVMCreateInterpreter(&Interp, MP, &Error))
 175     llvm_raise(llvm_ee_error_exn, Error);
 176   return Interp;
 177 }
 178
 179 /* llmoduleprovider -> ExecutionEngine.t */
 180 CAMLprim LLVMExecutionEngineRef
 181 llvm_ee_create_jit(LLVMModuleProviderRef MP) {
 182   LLVMExecutionEngineRef JIT;
 183   char *Error;
 184   if (LLVMCreateJITCompiler(&JIT, MP, &Error))
 185     llvm_raise(llvm_ee_error_exn, Error);
 186   return JIT;
 187 }
 188
 189 /* ExecutionEngine.t -> unit */
 190 CAMLprim value llvm_ee_dispose(LLVMExecutionEngineRef EE) {
 191   LLVMDisposeExecutionEngine(EE);
 192   return Val_unit;
 193 }
 194
 195 /* llmoduleprovider -> ExecutionEngine.t -> unit */
 196 CAMLprim value llvm_ee_add_mp(LLVMModuleProviderRef MP,
 197                               LLVMExecutionEngineRef EE) {
 198   LLVMAddModuleProvider(EE, MP);
 199   return Val_unit;
 200 }
 201
 202 /* llmoduleprovider -> ExecutionEngine.t -> llmodule */
 203 CAMLprim LLVMModuleRef llvm_ee_remove_mp(LLVMModuleProviderRef MP,
 204                                          LLVMExecutionEngineRef EE) {
 205   LLVMModuleRef RemovedModule;
 206   char *Error;
 207   if (LLVMRemoveModuleProvider(EE, MP, &RemovedModule, &Error))
 208     llvm_raise(llvm_ee_error_exn, Error);
 209   return RemovedModule;
 210 }
 211
 212 /* string -> ExecutionEngine.t -> llvalue option */
 213 CAMLprim value llvm_ee_find_function(value Name, LLVMExecutionEngineRef EE) {
 214   CAMLparam1(Name);
 215   CAMLlocal1(Option);
 216   LLVMValueRef Found;
 217   if (LLVMFindFunction(EE, String_val(Name), &Found))
 218     CAMLreturn(Val_unit);
 219   Option = alloc(1, 1);
 220   Field(Option, 0) = Val_op(Found);
 221   CAMLreturn(Option);
 222 }
 223
 224 /* llvalue -> GenericValue.t array -> ExecutionEngine.t -> GenericValue.t */
 225 CAMLprim value llvm_ee_run_function(LLVMValueRef F, value Args,
 226                                     LLVMExecutionEngineRef EE) {
 227   unsigned NumArgs;
 228   LLVMGenericValueRef Result, *GVArgs;
 229   unsigned I;
 230
 231   NumArgs = Wosize_val(Args);
 232   GVArgs = (LLVMGenericValueRef*) malloc(NumArgs * sizeof(LLVMGenericValueRef));
 233   for (I = 0; I != NumArgs; ++I)
 234     GVArgs[I] = Genericvalue_val(Field(Args, I));
 235
 236   Result = LLVMRunFunction(EE, F, NumArgs, GVArgs);
 237
 238   free(GVArgs);
 239   return alloc_generic_value(Result);
 240 }
 241
 242 /* ExecutionEngine.t -> unit */
 243 CAMLprim value llvm_ee_run_static_ctors(LLVMExecutionEngineRef EE) {
 244   LLVMRunStaticConstructors(EE);
 245   return Val_unit;
 246 }
 247
 248 /* ExecutionEngine.t -> unit */
 249 CAMLprim value llvm_ee_run_static_dtors(LLVMExecutionEngineRef EE) {
 250   LLVMRunStaticDestructors(EE);
 251   return Val_unit;
 252 }
 253
 254 /* llvalue -> string array -> (string * string) array -> ExecutionEngine.t ->
 255    int */
 256 CAMLprim value llvm_ee_run_function_as_main(LLVMValueRef F,
 257                                             value Args, value Env,
 258                                             LLVMExecutionEngineRef EE) {
 259   CAMLparam2(Args, Env);
 260   int I, NumArgs, NumEnv, EnvSize, Result;
 261   const char **CArgs, **CEnv;
 262   char *CEnvBuf, *Pos;
 263
 264   NumArgs = Wosize_val(Args);
 265   NumEnv = Wosize_val(Env);
 266
 267   /* Build the environment. */
 268   CArgs = (const char **) malloc(NumArgs * sizeof(char*));
 269   for (I = 0; I != NumArgs; ++I)
 270     CArgs[I] = String_val(Field(Args, I));
 271
 272   /* Compute the size of the environment string buffer. */
 273   for (I = 0, EnvSize = 0; I != NumEnv; ++I) {
 274     EnvSize += strlen(String_val(Field(Field(Env, I), 0))) + 1;
 275     EnvSize += strlen(String_val(Field(Field(Env, I), 1))) + 1;
 276   }
 277
 278   /* Build the environment. */
 279   CEnv = (const char **) malloc((NumEnv + 1) * sizeof(char*));
 280   CEnvBuf = (char*) malloc(EnvSize);
 281   Pos = CEnvBuf;
 282   for (I = 0; I != NumEnv; ++I) {
 283     char *Name  = String_val(Field(Field(Env, I), 0)),
 284          *Value = String_val(Field(Field(Env, I), 1));
 285     int NameLen  = strlen(Name),
 286         ValueLen = strlen(Value);
 287
 288     CEnv[I] = Pos;
 289     memcpy(Pos, Name, NameLen);
 290     Pos += NameLen;
 291     *Pos++ = '=';
 292     memcpy(Pos, Value, ValueLen);
 293     Pos += ValueLen;
 294     *Pos++ = '\0';
 295   }
 296   CEnv[NumEnv] = NULL;
 297
 298   Result = LLVMRunFunctionAsMain(EE, F, NumArgs, CArgs, CEnv);
 299
 300   free(CArgs);
 301   free(CEnv);
 302   free(CEnvBuf);
 303
 304   CAMLreturn(Val_int(Result));
 305 }
 306
 307 /* llvalue -> ExecutionEngine.t -> unit */
 308 CAMLprim value llvm_ee_free_machine_code(LLVMValueRef F,
 309                                          LLVMExecutionEngineRef EE) {
 310   LLVMFreeMachineCodeForFunction(EE, F);
 311   return Val_unit;
 312 }
 313