66 lines
1.7 KiB
OCaml
66 lines
1.7 KiB
OCaml
|
(* RUN: rm -rf %t && mkdir -p %t && cp %s %t/linker.ml
|
||
|
|
* RUN: %ocamlc -g -w +A -package llvm.linker -linkpkg %t/linker.ml -o %t/executable
|
||
|
|
* RUN: %t/executable
|
||
|
|
* RUN: %ocamlopt -g -w +A -package llvm.linker -linkpkg %t/linker.ml -o %t/executable
|
||
|
|
* RUN: %t/executable
|
||
|
|
* XFAIL: vg_leak
|
||
|
|
*)
|
||
|
|
|
||
|
|
(* Note: It takes several seconds for ocamlopt to link an executable with
|
||
|
|
libLLVMCore.a, so it's better to write a big test than a bunch of
|
||
|
|
little ones. *)
|
||
|
|
|
||
|
|
open Llvm
|
||
|
|
open Llvm_linker
|
||
|
|
|
||
|
|
let context = global_context ()
|
||
|
|
let void_type = Llvm.void_type context
|
||
|
|
|
||
|
|
let diagnostic_handler _ = ()
|
||
|
|
|
||
|
|
(* Tiny unit test framework - really just to help find which line is busted *)
|
||
|
|
let print_checkpoints = false
|
||
|
|
|
||
|
|
let suite name f =
|
||
|
|
if print_checkpoints then
|
||
|
|
prerr_endline (name ^ ":");
|
||
|
|
f ()
|
||
|
|
|
||
|
|
|
||
|
|
(*===-- Linker -----------------------------------------------------------===*)
|
||
|
|
|
||
|
|
let test_linker () =
|
||
|
|
set_diagnostic_handler context (Some diagnostic_handler);
|
||
|
|
|
||
|
|
let fty = function_type void_type [| |] in
|
||
|
|
|
||
|
|
let make_module name =
|
||
|
|
let m = create_module context name in
|
||
|
|
let fn = define_function ("fn_" ^ name) fty m in
|
||
|
|
ignore (build_ret_void (builder_at_end context (entry_block fn)));
|
||
|
|
m
|
||
|
|
in
|
||
|
|
|
||
|
|
let m1 = make_module "one"
|
||
|
|
and m2 = make_module "two" in
|
||
|
|
link_modules' m1 m2;
|
||
|
|
dispose_module m1;
|
||
|
|
|
||
|
|
let m1 = make_module "one"
|
||
|
|
and m2 = make_module "two" in
|
||
|
|
link_modules' m1 m2;
|
||
|
|
dispose_module m1;
|
||
|
|
|
||
|
|
let m1 = make_module "one"
|
||
|
|
and m2 = make_module "one" in
|
||
|
|
try
|
||
|
|
link_modules' m1 m2;
|
||
|
|
failwith "must raise"
|
||
|
|
with Error _ ->
|
||
|
|
dispose_module m1
|
||
|
|
|
||
|
|
(*===-- Driver ------------------------------------------------------------===*)
|
||
|
|
|
||
|
|
let _ =
|
||
|
|
suite "linker" test_linker
|