98c04172a26eab5ef9120c475dc2531cfa5382b0
[oota-llvm.git] / test / Bindings / Ocaml / bitwriter.ml
1 (* RUN: rm -rf %t.builddir
2  * RUN: mkdir -p %t.builddir
3  * RUN: cp %s %t.builddir
4  * RUN: %ocamlcomp -warn-error A-3 unix.%cma llvm.%cma llvm_bitwriter.%cma %t.builddir/bitwriter.ml -o %t
5  * RUN: %t %t.bc
6  * RUN: llvm-dis < %t.bc
7  * XFAIL: vg_leak
8  *)
9
10 (* Note that this takes a moment to link, so it's best to keep the number of
11    individual tests low. *)
12
13 let context = Llvm.global_context ()
14
15 let test x = if not x then exit 1 else ()
16
17 let read_file name =
18   let ic = open_in_bin name in
19   let len = in_channel_length ic in
20   let buf = String.create len in
21
22   test ((input ic buf 0 len) = len);
23
24   close_in ic;
25
26   buf
27
28 let temp_bitcode ?unbuffered m =
29   let temp_name, temp_oc = Filename.open_temp_file ~mode:[Open_binary] "" "" in
30
31   test (Llvm_bitwriter.output_bitcode ?unbuffered temp_oc m);
32   flush temp_oc;
33
34   let temp_buf = read_file temp_name in
35
36   close_out temp_oc;
37
38   temp_buf
39
40 let _ =
41   let m = Llvm.create_module context "ocaml_test_module" in
42
43   test (Llvm_bitwriter.write_bitcode_file m Sys.argv.(1));
44   let file_buf = read_file Sys.argv.(1) in
45
46   test (file_buf = temp_bitcode m);
47   test (file_buf = temp_bitcode ~unbuffered:false m);
48   test (file_buf = temp_bitcode ~unbuffered:true m);
49   test (file_buf = Llvm.MemoryBuffer.as_string (Llvm_bitwriter.write_bitcode_to_memory_buffer m))