Tests of the ocaml (and thus C) bindings for constants.
[oota-llvm.git] / test / Bindings / Ocaml / vmcore.ml
1 (* RUN: %ocamlc llvm.cma llvm_bitwriter.cma %s -o %t
2  * RUN: ./%t %t.bc
3  * RUN: llvm-dis < %t.bc > %t.ll
4  *)
5
6 (* Note: It takes several seconds for ocamlc to link an executable with
7          libLLVMCore.a, so it's better to write a big test than a bunch of
8          little ones. *)
9
10 open Llvm
11 open Llvm_bitwriter
12
13
14 (* Tiny unit test framework - really just to help find which line is busted *)
15 let exit_status = ref 0
16 let case_num = ref 0
17
18 let group name =
19   case_num := 0;
20   prerr_endline ("  " ^ name ^ "...")
21
22 let insist cond =
23   incr case_num;
24   let msg = if cond then "    pass " else begin
25     exit_status := 10;
26     "    FAIL "
27   end in
28   prerr_endline (msg ^ (string_of_int !case_num))
29
30 let suite name f =
31   prerr_endline (name ^ ":");
32   f ()
33
34
35 (*===-- Fixture -----------------------------------------------------------===*)
36
37 let filename = Sys.argv.(1)
38 let m = create_module filename
39
40
41 (*===-- Types -------------------------------------------------------------===*)
42
43 let test_types () =
44   (* RUN: grep {Ty01.*void} < %t.ll
45    *)
46   group "void";
47   insist (add_type_name "Ty01" void_type m);
48   insist (Void_type == classify_type void_type);
49
50   (* RUN: grep {Ty02.*i1} < %t.ll
51    *)
52   group "i1";
53   insist (add_type_name "Ty02" i1_type m);
54   insist (Integer_type == classify_type i1_type);
55
56   (* RUN: grep {Ty03.*i32} < %t.ll
57    *)
58   group "i32";
59   insist (add_type_name "Ty03" i32_type m);
60
61   (* RUN: grep {Ty04.*i42} < %t.ll
62    *)
63   group "i42";
64   let ty = make_integer_type 42 in
65   insist (add_type_name "Ty04" ty m);
66
67   (* RUN: grep {Ty05.*float} < %t.ll
68    *)
69   group "float";
70   insist (add_type_name "Ty05" float_type m);
71   insist (Float_type == classify_type float_type);
72
73   (* RUN: grep {Ty06.*double} < %t.ll
74    *)
75   group "double";
76   insist (add_type_name "Ty06" double_type m);
77   insist (Double_type == classify_type double_type);
78
79   (* RUN: grep {Ty07.*i32.*i1, double} < %t.ll
80    *)
81   group "function";
82   let ty = make_function_type i32_type [| i1_type; double_type |] false in
83   insist (add_type_name "Ty07" ty m);
84   insist (Function_type = classify_type ty);
85   insist (not (is_var_arg ty));
86   insist (i32_type == return_type ty);
87   insist (double_type == (param_types ty).(1));
88   
89   (* RUN: grep {Ty08.*\.\.\.} < %t.ll
90    *)
91   group "vararg";
92   let ty = make_function_type void_type [| i32_type |] true in
93   insist (add_type_name "Ty08" ty m);
94   insist (is_var_arg ty);
95   
96   (* RUN: grep {Ty09.*\\\[7 x i8\\\]} < %t.ll
97    *)
98   group "array";
99   let ty = make_array_type i8_type 7 in
100   insist (add_type_name "Ty09" ty m);
101   insist (7 = array_length ty);
102   insist (i8_type == element_type ty);
103   insist (Array_type == classify_type ty);
104   
105   (* RUN: grep {Ty10.*float\*} < %t.ll
106    *)
107   group "pointer";
108   let ty = make_pointer_type float_type in
109   insist (add_type_name "Ty10" ty m);
110   insist (float_type == element_type ty);
111   insist (Pointer_type == classify_type ty);
112   
113   (* RUN: grep {Ty11.*\<4 x i16\>} < %t.ll
114    *)
115   group "vector";
116   let ty = make_vector_type i16_type 4 in
117   insist (add_type_name "Ty11" ty m);
118   insist (i16_type == element_type ty);
119   insist (4 = vector_size ty);
120   
121   (* RUN: grep {Ty12.*opaque} < %t.ll
122    *)
123   group "opaque";
124   let ty = make_opaque_type () in
125   insist (add_type_name "Ty12" ty m);
126   insist (ty == ty);
127   insist (ty <> make_opaque_type ())
128
129
130 (*===-- Constants ---------------------------------------------------------===*)
131
132 let test_constants () =
133   (* RUN: grep {Const01.*i32.*-1} < %t.ll
134    *)
135   group "int";
136   let c = make_int_constant i32_type (-1) true in
137   ignore (define_global "Const01" c m);
138   insist (i32_type = type_of c);
139   insist (is_constant c);
140
141   (* RUN: grep {Const02.*i64.*-1} < %t.ll
142    *)
143   group "sext int";
144   let c = make_int_constant i64_type (-1) true in
145   ignore (define_global "Const02" c m);
146   insist (i64_type = type_of c);
147
148   (* RUN: grep {Const03.*i64.*4294967295} < %t.ll
149    *)
150   group "zext int64";
151   let c = make_int64_constant i64_type (Int64.of_string "4294967295") false in
152   ignore (define_global "Const03" c m);
153   insist (i64_type = type_of c);
154
155   (* RUN: grep {Const04.*"cruel\\\\00world"} < %t.ll
156    *)
157   group "string";
158   let c = make_string_constant "cruel\x00world" false in
159   ignore (define_global "Const04" c m);
160   insist ((make_array_type i8_type 11) = type_of c);
161
162   (* RUN: grep {Const05.*"hi\\\\00again\\\\00"} < %t.ll
163    *)
164   group "string w/ null";
165   let c = make_string_constant "hi\x00again" true in
166   ignore (define_global "Const05" c m);
167   insist ((make_array_type i8_type 9) = type_of c);
168
169   (* RUN: grep {Const06.*3.1459} < %t.ll
170    *)
171   group "real";
172   let c = make_real_constant double_type 3.1459 in
173   ignore (define_global "Const06" c m);
174   insist (double_type = type_of c);
175   
176   let one = make_int_constant i16_type 1 true in
177   let two = make_int_constant i16_type 2 true in
178   let three = make_int_constant i32_type 3 true in
179   let four = make_int_constant i32_type 4 true in
180   
181   (* RUN: grep {Const07.*\\\[ i32 3, i32 4 \\\]} < %t.ll
182    *)
183   group "array";
184   let c = make_array_constant i32_type [| three; four |] in
185   ignore (define_global "Const07" c m);
186   insist ((make_array_type i32_type 2) = (type_of c));
187   
188   (* RUN: grep {Const08.*< i16 1, i16 2.* >} < %t.ll
189    *)
190   group "vector";
191   let c = make_vector_constant [| one; two; one; two;
192                                   one; two; one; two |] in
193   ignore (define_global "Const08" c m);
194   insist ((make_vector_type i16_type 8) = (type_of c));
195   
196   (* RUN: grep {Const09.*\{ i16, i16, i32, i32 \} \{} < %t.ll
197    *)
198   group "structure";
199   let c = make_struct_constant [| one; two; three; four |] false in
200   ignore (define_global "Const09" c m);
201   insist ((make_struct_type [| i16_type; i16_type; i32_type; i32_type |] false)
202         = (type_of c));
203   
204   (* RUN: grep {Const10.*zeroinit} < %t.ll
205    *)
206   group "null";
207   let c = make_null (make_struct_type [| i1_type; i8_type;
208                                          i64_type; double_type |] true) in
209   ignore (define_global "Const10" c m);
210   
211   (* RUN: grep {Const11.*-1} < %t.ll
212    *)
213   group "all ones";
214   let c = make_all_ones i64_type in
215   ignore (define_global "Const11" c m);
216   
217   (* RUN: grep {Const12.*undef} < %t.ll
218    *)
219   group "undef";
220   let c = make_undef i1_type in
221   ignore (define_global "Const12" c m);
222   insist (i1_type = type_of c);
223   insist (is_undef c)
224
225
226 (*===-- Global Values -----------------------------------------------------===*)
227
228 let test_global_values () =
229   let (++) x f = f x; x in
230   let zero32 = make_null i32_type in
231
232   (* RUN: grep {GVal01} < %t.ll
233    *)
234   group "naming";
235   let g = define_global "TEMPORARY" zero32 m in
236   insist ("TEMPORARY" = value_name g);
237   set_value_name "GVal01" g;
238   insist ("GVal01" = value_name g);
239
240   (* RUN: grep {GVal02.*linkonce} < %t.ll
241    *)
242   group "linkage";
243   let g = define_global "GVal02" zero32 m ++
244           set_linkage Link_once_linkage in
245   insist (Link_once_linkage = linkage g);
246
247   (* RUN: grep {GVal03.*Hanalei} < %t.ll
248    *)
249   group "section";
250   let g = define_global "GVal03" zero32 m ++
251           set_section "Hanalei" in
252   insist ("Hanalei" = section g);
253   
254   (* RUN: grep {GVal04.*hidden} < %t.ll
255    *)
256   group "visibility";
257   let g = define_global "GVal04" zero32 m ++
258           set_visibility Hidden_visibility in
259   insist (Hidden_visibility = visibility g);
260   
261   (* RUN: grep {GVal05.*align 128} < %t.ll
262    *)
263   group "alignment";
264   let g = define_global "GVal05" zero32 m ++
265           set_alignment 128 in
266   insist (128 = alignment g)
267
268
269 (*===-- Global Variables --------------------------------------------------===*)
270
271 let test_global_variables () =
272   let (++) x f = f x; x in
273   let fourty_two32 = make_int_constant i32_type 42 false in
274
275   (* RUN: grep {GVar01.*external} < %t.ll
276    *)
277   group "declarations";
278   let g = declare_global i32_type "GVar01" m in
279   insist (is_declaration g);
280   
281   (* RUN: grep {GVar02.*42} < %t.ll
282    * RUN: grep {GVar03.*42} < %t.ll
283    *)
284   group "definitions";
285   let g = define_global "GVar02" fourty_two32 m in
286   let g2 = declare_global i32_type "GVar03" m ++
287            set_initializer fourty_two32 in
288   insist (not (is_declaration g));
289   insist (not (is_declaration g2));
290   insist ((global_initializer g) == (global_initializer g2));
291
292   (* RUN: grep {GVar04.*thread_local} < %t.ll
293    *)
294   group "threadlocal";
295   let g = define_global "GVar04" fourty_two32 m ++
296           set_thread_local true in
297   insist (is_thread_local g);
298
299   (* RUN: grep -v {GVar05} < %t.ll
300    *)
301   group "delete";
302   let g = define_global "GVar05" fourty_two32 m in
303   delete_global g
304
305
306 (*===-- Writer ------------------------------------------------------------===*)
307
308 let test_writer () =
309   group "writer";
310   insist (write_bitcode_file m filename);
311   
312   dispose_module m
313
314
315 (*===-- Driver ------------------------------------------------------------===*)
316
317 let _ =
318   suite "types"            test_types;
319   suite "constants"        test_constants;
320   suite "global values"    test_global_values;
321   suite "global variables" test_global_variables;
322   suite "writer"           test_writer;
323   exit !exit_status