test/Bindings/Ocaml/bitreader.ml

   1 (* RUN: cp %s %T/bitreader.ml
   2  * RUN: %ocamlcomp -warn-error A -package llvm.bitreader -package llvm.bitwriter -linkpkg %T/bitreader.ml -o %t
   3  * RUN: %t %t.bc
   4  * RUN: llvm-dis < %t.bc
   5  * XFAIL: vg_leak
   6  *)
   7
   8 (* Note that this takes a moment to link, so it's best to keep the number of
   9    individual tests low. *)
  10
  11 let context = Llvm.global_context ()
  12
  13 let test x = if not x then exit 1 else ()
  14
  15 let _ =
  16   let fn = Sys.argv.(1) in
  17   let m = Llvm.create_module context "ocaml_test_module" in
  18
  19   test (Llvm_bitwriter.write_bitcode_file m fn);
  20
  21   Llvm.dispose_module m;
  22
  23   (* parse_bitcode *)
  24   begin
  25     let mb = Llvm.MemoryBuffer.of_file fn in
  26     begin try
  27       let m = Llvm_bitreader.parse_bitcode context mb in
  28       Llvm.dispose_module m
  29     with x ->
  30       Llvm.MemoryBuffer.dispose mb;
  31       raise x
  32     end
  33   end;
  34
  35   (* MemoryBuffer.of_file *)
  36   test begin try
  37     let mb = Llvm.MemoryBuffer.of_file (fn ^ ".bogus") in
  38     Llvm.MemoryBuffer.dispose mb;
  39     false
  40   with Llvm.IoError _ ->
  41     true
  42   end;
  43
  44   (* get_module *)
  45   begin
  46     let mb = Llvm.MemoryBuffer.of_file fn in
  47     let m = begin try
  48       Llvm_bitreader.get_module context mb
  49     with x ->
  50       Llvm.MemoryBuffer.dispose mb;
  51       raise x
  52     end in
  53     Llvm.dispose_module m
  54   end;
  55
  56   (* corrupt the bitcode *)
  57   let fn = fn ^ ".txt" in
  58   begin let oc = open_out fn in
  59     output_string oc "not a bitcode file\n";
  60     close_out oc
  61   end;
  62
  63   (* test get_module exceptions *)
  64   test begin
  65     try
  66       let mb = Llvm.MemoryBuffer.of_file fn in
  67       let m = begin try
  68         Llvm_bitreader.get_module context mb
  69       with x ->
  70         Llvm.MemoryBuffer.dispose mb;
  71         raise x
  72       end in
  73       Llvm.dispose_module m;
  74       false
  75     with Llvm_bitreader.Error _ ->
  76       true
  77   end