1(* RUN: cp %s %T/target.ml
2 * RUN: %ocamlc -g -warn-error A -package llvm.target -package llvm.all_backends -linkpkg %T/target.ml -o %t
3 * RUN: %ocamlopt -g -warn-error A -package llvm.target -package llvm.all_backends -linkpkg %T/target.ml -o %t
4 * RUN: %t %t.bc
5 * XFAIL: vg_leak
6 *)
7
8(* Note: It takes several seconds for ocamlopt to link an executable with
9         libLLVMCore.a, so it's better to write a big test than a bunch of
10         little ones. *)
11
12open Llvm
13open Llvm_target
14
15let () = Llvm_all_backends.initialize ()
16
17let context = global_context ()
18let i32_type = Llvm.i32_type context
19let i64_type = Llvm.i64_type context
20
21(* Tiny unit test framework - really just to help find which line is busted *)
22let print_checkpoints = false
23
24let _ =
25  Printexc.record_backtrace true
26
27let assert_equal a b =
28  if a <> b then failwith "assert_equal"
29
30
31(*===-- Fixture -----------------------------------------------------------===*)
32
33let filename = Sys.argv.(1)
34let m = create_module context filename
35
36let target = Target.by_triple (Target.default_triple ())
37
38let machine = TargetMachine.create (Target.default_triple ()) target
39
40(*===-- Data Layout -------------------------------------------------------===*)
41
42let test_target_data () =
43  let module DL = DataLayout in
44  let layout = "e-p:32:32-f64:32:64-v64:32:64-v128:32:128-n32-S32" in
45  let dl     = DL.of_string layout in
46  let sty    = struct_type context [| i32_type; i64_type |] in
47
48  assert_equal (DL.as_string dl) layout;
49  assert_equal (DL.byte_order dl) Endian.Little;
50  assert_equal (DL.pointer_size dl) 4;
51  assert_equal (DL.intptr_type context dl) i32_type;
52  assert_equal (DL.qualified_pointer_size 0 dl) 4;
53  assert_equal (DL.qualified_intptr_type context 0 dl) i32_type;
54  assert_equal (DL.size_in_bits sty dl) (Int64.of_int 96);
55  assert_equal (DL.store_size sty dl) (Int64.of_int 12);
56  assert_equal (DL.abi_size sty dl) (Int64.of_int 12);
57  assert_equal (DL.stack_align sty dl) 4;
58  assert_equal (DL.preferred_align sty dl) 8;
59  assert_equal (DL.preferred_align_of_global (declare_global sty "g" m) dl) 8;
60  assert_equal (DL.element_at_offset sty (Int64.of_int 1) dl) 0;
61  assert_equal (DL.offset_of_element sty 1 dl) (Int64.of_int 4);
62
63  let pm = PassManager.create () in
64  ignore (DL.add_to_pass_manager pm dl)
65
66
67(*===-- Target ------------------------------------------------------------===*)
68
69let test_target () =
70  let module T = Target in
71  ignore (T.succ target);
72  ignore (T.name target);
73  ignore (T.description target);
74  ignore (T.has_jit target);
75  ignore (T.has_target_machine target);
76  ignore (T.has_asm_backend target)
77
78
79(*===-- Target Machine ----------------------------------------------------===*)
80
81let test_target_machine () =
82  let module TM = TargetMachine in
83  assert_equal (TM.target machine) target;
84  assert_equal (TM.triple machine) (Target.default_triple ());
85  assert_equal (TM.cpu machine) "";
86  assert_equal (TM.features machine) "";
87  ignore (TM.data_layout machine);
88  TM.set_verbose_asm true machine;
89  let pm = PassManager.create () in
90  TM.add_analysis_passes pm machine
91
92
93(*===-- Code Emission -----------------------------------------------------===*)
94
95let test_code_emission () =
96  TargetMachine.emit_to_file m CodeGenFileType.ObjectFile filename machine;
97  try
98    TargetMachine.emit_to_file m CodeGenFileType.ObjectFile
99                               "/nonexistent/file" machine;
100    failwith "must raise"
101  with Llvm_target.Error _ ->
102    ();
103
104  let buf = TargetMachine.emit_to_memory_buffer m CodeGenFileType.ObjectFile
105                                                machine in
106  Llvm.MemoryBuffer.dispose buf
107
108
109(*===-- Driver ------------------------------------------------------------===*)
110
111let _ =
112  test_target_data ();
113  test_target ();
114  test_target_machine ();
115  test_code_emission ();
116  dispose_module m
117