73b5b286bef27018b437b346dd83e23067cd2894
[oota-llvm.git] / test / Bindings / Ocaml / vmcore.ml
1 (* RUN: %ocamlc -warn-error A llvm.cma llvm_analysis.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   if not cond then exit_status := 10;
25   prerr_endline ("    " ^ (string_of_int !case_num) ^ if cond then ""
26                                                               else " FAIL")
27
28 let suite name f =
29   prerr_endline (name ^ ":");
30   f ()
31
32
33 (*===-- Fixture -----------------------------------------------------------===*)
34
35 let filename = Sys.argv.(1)
36 let m = create_module filename
37 let mp = ModuleProvider.create m
38
39
40 (*===-- Target ------------------------------------------------------------===*)
41
42 let test_target () =
43   begin group "triple";
44     (* RUN: grep "i686-apple-darwin8" < %t.ll
45      *)
46     let trip = "i686-apple-darwin8" in
47     set_target_triple trip m;
48     insist (trip = target_triple m)
49   end;
50   
51   begin group "layout";
52     (* RUN: grep "bogus" < %t.ll
53      *)
54     let layout = "bogus" in
55     set_data_layout layout m;
56     insist (layout = data_layout m)
57   end
58
59 (*===-- Types -------------------------------------------------------------===*)
60
61 let test_types () =
62   (* RUN: grep {Ty01.*void} < %t.ll
63    *)
64   group "void";
65   insist (define_type_name "Ty01" void_type m);
66   insist (TypeKind.Void == classify_type void_type);
67
68   (* RUN: grep {Ty02.*i1} < %t.ll
69    *)
70   group "i1";
71   insist (define_type_name "Ty02" i1_type m);
72   insist (TypeKind.Integer == classify_type i1_type);
73
74   (* RUN: grep {Ty03.*i32} < %t.ll
75    *)
76   group "i32";
77   insist (define_type_name "Ty03" i32_type m);
78
79   (* RUN: grep {Ty04.*i42} < %t.ll
80    *)
81   group "i42";
82   let ty = integer_type 42 in
83   insist (define_type_name "Ty04" ty m);
84
85   (* RUN: grep {Ty05.*float} < %t.ll
86    *)
87   group "float";
88   insist (define_type_name "Ty05" float_type m);
89   insist (TypeKind.Float == classify_type float_type);
90
91   (* RUN: grep {Ty06.*double} < %t.ll
92    *)
93   group "double";
94   insist (define_type_name "Ty06" double_type m);
95   insist (TypeKind.Double == classify_type double_type);
96
97   (* RUN: grep {Ty07.*i32.*i1, double} < %t.ll
98    *)
99   group "function";
100   let ty = function_type i32_type [| i1_type; double_type |] in
101   insist (define_type_name "Ty07" ty m);
102   insist (TypeKind.Function = classify_type ty);
103   insist (not (is_var_arg ty));
104   insist (i32_type == return_type ty);
105   insist (double_type == (param_types ty).(1));
106   
107   (* RUN: grep {Ty08.*\.\.\.} < %t.ll
108    *)
109   group "var arg function";
110   let ty = var_arg_function_type void_type [| i32_type |] in
111   insist (define_type_name "Ty08" ty m);
112   insist (is_var_arg ty);
113   
114   (* RUN: grep {Ty09.*\\\[7 x i8\\\]} < %t.ll
115    *)
116   group "array";
117   let ty = array_type i8_type 7 in
118   insist (define_type_name "Ty09" ty m);
119   insist (7 = array_length ty);
120   insist (i8_type == element_type ty);
121   insist (TypeKind.Array == classify_type ty);
122   
123   begin group "pointer";
124     (* RUN: grep {UnqualPtrTy.*float\*} < %t.ll
125      *)
126     let ty = pointer_type float_type in
127     insist (define_type_name "UnqualPtrTy" ty m);
128     insist (float_type == element_type ty);
129     insist (0 == address_space ty);
130     insist (TypeKind.Pointer == classify_type ty)
131   end;
132   
133   begin group "qualified_pointer";
134     (* RUN: grep {QualPtrTy.*i8.*3.*\*} < %t.ll
135      *)
136     let ty = qualified_pointer_type i8_type 3 in
137     insist (define_type_name "QualPtrTy" ty m);
138     insist (i8_type == element_type ty);
139     insist (3 == address_space ty)
140   end;
141   
142   (* RUN: grep {Ty11.*\<4 x i16\>} < %t.ll
143    *)
144   group "vector";
145   let ty = vector_type i16_type 4 in
146   insist (define_type_name "Ty11" ty m);
147   insist (i16_type == element_type ty);
148   insist (4 = vector_size ty);
149   
150   (* RUN: grep {Ty12.*opaque} < %t.ll
151    *)
152   group "opaque";
153   let ty = opaque_type () in
154   insist (define_type_name "Ty12" ty m);
155   insist (ty == ty);
156   insist (ty <> opaque_type ());
157   
158   (* RUN: grep -v {Ty13} < %t.ll
159    *)
160   group "delete";
161   let ty = opaque_type () in
162   insist (define_type_name "Ty13" ty m);
163   delete_type_name "Ty13" m;
164   
165   (* RUN: grep -v {RecursiveTy.*RecursiveTy} < %t.ll
166    *)
167   group "recursive";
168   let ty = opaque_type () in
169   let th = handle_to_type ty in
170   refine_type ty (pointer_type ty);
171   let ty = type_of_handle th in
172   insist (define_type_name "RecursiveTy" ty m);
173   insist (ty == element_type ty)
174
175
176 (*===-- Constants ---------------------------------------------------------===*)
177
178 let test_constants () =
179   (* RUN: grep {Const01.*i32.*-1} < %t.ll
180    *)
181   group "int";
182   let c = const_int i32_type (-1) in
183   ignore (define_global "Const01" c m);
184   insist (i32_type = type_of c);
185   insist (is_constant c);
186
187   (* RUN: grep {Const02.*i64.*-1} < %t.ll
188    *)
189   group "sext int";
190   let c = const_int i64_type (-1) in
191   ignore (define_global "Const02" c m);
192   insist (i64_type = type_of c);
193
194   (* RUN: grep {Const03.*i64.*4294967295} < %t.ll
195    *)
196   group "zext int64";
197   let c = const_of_int64 i64_type (Int64.of_string "4294967295") false in
198   ignore (define_global "Const03" c m);
199   insist (i64_type = type_of c);
200
201   (* RUN: grep {Const04.*"cruel\\\\00world"} < %t.ll
202    *)
203   group "string";
204   let c = const_string "cruel\000world" in
205   ignore (define_global "Const04" c m);
206   insist ((array_type i8_type 11) = type_of c);
207
208   (* RUN: grep {Const05.*"hi\\\\00again\\\\00"} < %t.ll
209    *)
210   group "stringz";
211   let c = const_stringz "hi\000again" in
212   ignore (define_global "Const05" c m);
213   insist ((array_type i8_type 9) = type_of c);
214
215   (* RUN: grep {ConstSingle.*2.75} < %t.ll
216    * RUN: grep {ConstDouble.*3.1459} < %t.ll
217    *)
218   begin group "real";
219     let cs = const_float float_type 2.75 in
220     ignore (define_global "ConstSingle" cs m);
221     insist (float_type = type_of cs);
222     
223     let cd = const_float double_type 3.1459 in
224     ignore (define_global "ConstDouble" cd m);
225     insist (double_type = type_of cd)
226   end;
227   
228   let one = const_int i16_type 1 in
229   let two = const_int i16_type 2 in
230   let three = const_int i32_type 3 in
231   let four = const_int i32_type 4 in
232   
233   (* RUN: grep {Const07.*\\\[ i32 3, i32 4 \\\]} < %t.ll
234    *)
235   group "array";
236   let c = const_array i32_type [| three; four |] in
237   ignore (define_global "Const07" c m);
238   insist ((array_type i32_type 2) = (type_of c));
239   
240   (* RUN: grep {Const08.*< i16 1, i16 2.* >} < %t.ll
241    *)
242   group "vector";
243   let c = const_vector [| one; two; one; two;
244                           one; two; one; two |] in
245   ignore (define_global "Const08" c m);
246   insist ((vector_type i16_type 8) = (type_of c));
247   
248   (* RUN: grep {Const09.*\{ i16, i16, i32, i32 \} \{} < %t.ll
249    *)
250   group "structure";
251   let c = const_struct [| one; two; three; four |] in
252   ignore (define_global "Const09" c m);
253   insist ((struct_type [| i16_type; i16_type; i32_type; i32_type |])
254         = (type_of c));
255   
256   (* RUN: grep {Const10.*zeroinit} < %t.ll
257    *)
258   group "null";
259   let c = const_null (packed_struct_type [| i1_type; i8_type;
260                                             i64_type; double_type |]) in
261   ignore (define_global "Const10" c m);
262   
263   (* RUN: grep {Const11.*-1} < %t.ll
264    *)
265   group "all ones";
266   let c = const_all_ones i64_type in
267   ignore (define_global "Const11" c m);
268   
269   (* RUN: grep {Const12.*undef} < %t.ll
270    *)
271   group "undef";
272   let c = undef i1_type in
273   ignore (define_global "Const12" c m);
274   insist (i1_type = type_of c);
275   insist (is_undef c);
276   
277   group "constant arithmetic";
278   (* RUN: grep {ConstNeg.*sub} < %t.ll
279    * RUN: grep {ConstNot.*xor} < %t.ll
280    * RUN: grep {ConstAdd.*add} < %t.ll
281    * RUN: grep {ConstSub.*sub} < %t.ll
282    * RUN: grep {ConstMul.*mul} < %t.ll
283    * RUN: grep {ConstUDiv.*udiv} < %t.ll
284    * RUN: grep {ConstSDiv.*sdiv} < %t.ll
285    * RUN: grep {ConstFDiv.*fdiv} < %t.ll
286    * RUN: grep {ConstURem.*urem} < %t.ll
287    * RUN: grep {ConstSRem.*srem} < %t.ll
288    * RUN: grep {ConstFRem.*frem} < %t.ll
289    * RUN: grep {ConstAnd.*and} < %t.ll
290    * RUN: grep {ConstOr.*or} < %t.ll
291    * RUN: grep {ConstXor.*xor} < %t.ll
292    * RUN: grep {ConstICmp.*icmp} < %t.ll
293    * RUN: grep {ConstFCmp.*fcmp} < %t.ll
294    *)
295   let void_ptr = pointer_type i8_type in
296   let five = const_int i64_type 5 in
297   let ffive = const_uitofp five double_type in
298   let foldbomb_gv = define_global "FoldBomb" (const_null i8_type) m in
299   let foldbomb = const_ptrtoint foldbomb_gv i64_type in
300   let ffoldbomb = const_uitofp foldbomb double_type in
301   ignore (define_global "ConstNeg" (const_neg foldbomb) m);
302   ignore (define_global "ConstNot" (const_not foldbomb) m);
303   ignore (define_global "ConstAdd" (const_add foldbomb five) m);
304   ignore (define_global "ConstSub" (const_sub foldbomb five) m);
305   ignore (define_global "ConstMul" (const_mul foldbomb five) m);
306   ignore (define_global "ConstUDiv" (const_udiv foldbomb five) m);
307   ignore (define_global "ConstSDiv" (const_sdiv foldbomb five) m);
308   ignore (define_global "ConstFDiv" (const_fdiv ffoldbomb ffive) m);
309   ignore (define_global "ConstURem" (const_urem foldbomb five) m);
310   ignore (define_global "ConstSRem" (const_srem foldbomb five) m);
311   ignore (define_global "ConstFRem" (const_frem ffoldbomb ffive) m);
312   ignore (define_global "ConstAnd" (const_and foldbomb five) m);
313   ignore (define_global "ConstOr" (const_or foldbomb five) m);
314   ignore (define_global "ConstXor" (const_xor foldbomb five) m);
315   ignore (define_global "ConstICmp" (const_icmp Icmp.Sle foldbomb five) m);
316   ignore (define_global "ConstFCmp" (const_fcmp Fcmp.Ole ffoldbomb ffive) m);
317   
318   group "constant casts";
319   (* RUN: grep {ConstTrunc.*trunc} < %t.ll
320    * RUN: grep {ConstSExt.*sext} < %t.ll
321    * RUN: grep {ConstZExt.*zext} < %t.ll
322    * RUN: grep {ConstFPTrunc.*fptrunc} < %t.ll
323    * RUN: grep {ConstFPExt.*fpext} < %t.ll
324    * RUN: grep {ConstUIToFP.*uitofp} < %t.ll
325    * RUN: grep {ConstSIToFP.*sitofp} < %t.ll
326    * RUN: grep {ConstFPToUI.*fptoui} < %t.ll
327    * RUN: grep {ConstFPToSI.*fptosi} < %t.ll
328    * RUN: grep {ConstPtrToInt.*ptrtoint} < %t.ll
329    * RUN: grep {ConstIntToPtr.*inttoptr} < %t.ll
330    * RUN: grep {ConstBitCast.*bitcast} < %t.ll
331    *)
332   let i128_type = integer_type 128 in
333   ignore (define_global "ConstTrunc" (const_trunc (const_add foldbomb five)
334                                                i8_type) m);
335   ignore (define_global "ConstSExt" (const_sext foldbomb i128_type) m);
336   ignore (define_global "ConstZExt" (const_zext foldbomb i128_type) m);
337   ignore (define_global "ConstFPTrunc" (const_fptrunc ffoldbomb float_type) m);
338   ignore (define_global "ConstFPExt" (const_fpext ffoldbomb fp128_type) m);
339   ignore (define_global "ConstUIToFP" (const_uitofp foldbomb double_type) m);
340   ignore (define_global "ConstSIToFP" (const_sitofp foldbomb double_type) m);
341   ignore (define_global "ConstFPToUI" (const_fptoui ffoldbomb i32_type) m);
342   ignore (define_global "ConstFPToSI" (const_fptosi ffoldbomb i32_type) m);
343   ignore (define_global "ConstPtrToInt" (const_ptrtoint 
344     (const_gep (const_null (pointer_type i8_type))
345                [| const_int i32_type 1 |])
346     i32_type) m);
347   ignore (define_global "ConstIntToPtr" (const_inttoptr (const_add foldbomb five)
348                                                   void_ptr) m);
349   ignore (define_global "ConstBitCast" (const_bitcast ffoldbomb i64_type) m);
350   
351   group "misc constants";
352   (* RUN: grep {ConstSizeOf.*getelementptr.*null} < %t.ll
353    * RUN: grep {ConstGEP.*getelementptr} < %t.ll
354    * RUN: grep {ConstSelect.*select} < %t.ll
355    * RUN: grep {ConstExtractElement.*extractelement} < %t.ll
356    * RUN: grep {ConstInsertElement.*insertelement} < %t.ll
357    * RUN: grep {ConstShuffleVector.*shufflevector} < %t.ll
358    *)
359   ignore (define_global "ConstSizeOf" (size_of (pointer_type i8_type)) m);
360   ignore (define_global "ConstGEP" (const_gep foldbomb_gv [| five |]) m);
361   ignore (define_global "ConstSelect" (const_select
362     (const_icmp Icmp.Sle foldbomb five)
363     (const_int i8_type (-1))
364     (const_int i8_type 0)) m);
365   let zero = const_int i32_type 0 in
366   let one  = const_int i32_type 1 in
367   ignore (define_global "ConstExtractElement" (const_extractelement
368     (const_vector [| zero; one; zero; one |])
369     (const_trunc foldbomb i32_type)) m);
370   ignore (define_global "ConstInsertElement" (const_insertelement
371     (const_vector [| zero; one; zero; one |])
372     zero (const_trunc foldbomb i32_type)) m);
373   ignore (define_global "ConstShuffleVector" (const_shufflevector
374     (const_vector [| zero; one |])
375     (const_vector [| one; zero |])
376     (const_bitcast foldbomb (vector_type i32_type 2))) m)
377
378
379 (*===-- Global Values -----------------------------------------------------===*)
380
381 let test_global_values () =
382   let (++) x f = f x; x in
383   let zero32 = const_null i32_type in
384
385   (* RUN: grep {GVal01} < %t.ll
386    *)
387   group "naming";
388   let g = define_global "TEMPORARY" zero32 m in
389   insist ("TEMPORARY" = value_name g);
390   set_value_name "GVal01" g;
391   insist ("GVal01" = value_name g);
392
393   (* RUN: grep {GVal02.*linkonce} < %t.ll
394    *)
395   group "linkage";
396   let g = define_global "GVal02" zero32 m ++
397           set_linkage Linkage.Link_once in
398   insist (Linkage.Link_once = linkage g);
399
400   (* RUN: grep {GVal03.*Hanalei} < %t.ll
401    *)
402   group "section";
403   let g = define_global "GVal03" zero32 m ++
404           set_section "Hanalei" in
405   insist ("Hanalei" = section g);
406   
407   (* RUN: grep {GVal04.*hidden} < %t.ll
408    *)
409   group "visibility";
410   let g = define_global "GVal04" zero32 m ++
411           set_visibility Visibility.Hidden in
412   insist (Visibility.Hidden = visibility g);
413   
414   (* RUN: grep {GVal05.*align 128} < %t.ll
415    *)
416   group "alignment";
417   let g = define_global "GVal05" zero32 m ++
418           set_alignment 128 in
419   insist (128 = alignment g)
420
421
422 (*===-- Global Variables --------------------------------------------------===*)
423
424 let test_global_variables () =
425   let (++) x f = f x; x in
426   let fourty_two32 = const_int i32_type 42 in
427
428   (* RUN: grep {GVar01.*external} < %t.ll
429    *)
430   group "declarations";
431   insist (None == lookup_global "GVar01" m);
432   let g = declare_global i32_type "GVar01" m in
433   insist (is_declaration g);
434   insist (pointer_type float_type ==
435             type_of (declare_global float_type "GVar01" m));
436   insist (g == declare_global i32_type "GVar01" m);
437   insist (match lookup_global "GVar01" m with Some x -> x = g
438                                             | None -> false);
439   
440   (* RUN: grep {GVar02.*42} < %t.ll
441    * RUN: grep {GVar03.*42} < %t.ll
442    *)
443   group "definitions";
444   let g = define_global "GVar02" fourty_two32 m in
445   let g2 = declare_global i32_type "GVar03" m ++
446            set_initializer fourty_two32 in
447   insist (not (is_declaration g));
448   insist (not (is_declaration g2));
449   insist ((global_initializer g) == (global_initializer g2));
450
451   (* RUN: grep {GVar04.*thread_local} < %t.ll
452    *)
453   group "threadlocal";
454   let g = define_global "GVar04" fourty_two32 m ++
455           set_thread_local true in
456   insist (is_thread_local g);
457
458   (* RUN: grep -v {GVar05} < %t.ll
459    *)
460   group "delete";
461   let g = define_global "GVar05" fourty_two32 m in
462   delete_global g;
463
464   (* RUN: grep -v {ConstGlobalVar.*constant} < %t.ll
465    *)
466   group "constant";
467   let g = define_global "ConstGlobalVar" fourty_two32 m in
468   insist (not (is_global_constant g));
469   set_global_constant true g;
470   insist (is_global_constant g)
471
472
473 (*===-- Functions ---------------------------------------------------------===*)
474
475 let test_functions () =
476   let ty = function_type i32_type [| i32_type; i64_type |] in
477   let ty2 = function_type i8_type [| i8_type; i64_type |] in
478   
479   (* RUN: grep {declare i32 @Fn1\(i32, i64\)} < %t.ll
480    *)
481   begin group "declare";
482     insist (None = lookup_function "Fn1" m);
483     let fn = declare_function "Fn1" ty m in
484     insist (pointer_type ty = type_of fn);
485     insist (is_declaration fn);
486     insist (0 = Array.length (basic_blocks fn));
487     insist (pointer_type ty2 == type_of (declare_function "Fn1" ty2 m));
488     insist (fn == declare_function "Fn1" ty m);
489     insist (None <> lookup_function "Fn1" m);
490     insist (match lookup_function "Fn1" m with Some x -> x = fn
491                                              | None -> false);
492     insist (m == global_parent fn)
493   end;
494   
495   (* RUN: grep -v {Fn2} < %t.ll
496    *)
497   group "delete";
498   let fn = declare_function "Fn2" ty m in
499   delete_function fn;
500   
501   (* RUN: grep {define.*Fn3} < %t.ll
502    *)
503   group "define";
504   let fn = define_function "Fn3" ty m in
505   insist (not (is_declaration fn));
506   insist (1 = Array.length (basic_blocks fn));
507   ignore (build_unreachable (builder_at_end (entry_block fn)));
508   
509   (* RUN: grep {define.*Fn4.*Param1.*Param2} < %t.ll
510    *)
511   group "params";
512   let fn = define_function "Fn4" ty m in
513   let params = params fn in
514   insist (2 = Array.length params);
515   insist (params.(0) = param fn 0);
516   insist (params.(1) = param fn 1);
517   insist (i32_type = type_of params.(0));
518   insist (i64_type = type_of params.(1));
519   set_value_name "Param1" params.(0);
520   set_value_name "Param2" params.(1);
521   ignore (build_unreachable (builder_at_end (entry_block fn)));
522   
523   (* RUN: grep {fastcc.*Fn5} < %t.ll
524    *)
525   group "callconv";
526   let fn = define_function "Fn5" ty m in
527   insist (CallConv.c = function_call_conv fn);
528   set_function_call_conv CallConv.fast fn;
529   insist (CallConv.fast = function_call_conv fn);
530   ignore (build_unreachable (builder_at_end (entry_block fn)));
531   
532   begin group "collector";
533     (* RUN: grep {Fn6.*gc.*shadowstack} < %t.ll
534      *)
535     let fn = define_function "Fn6" ty m in
536     insist (None = collector fn);
537     set_collector (Some "ocaml") fn;
538     insist (Some "ocaml" = collector fn);
539     set_collector None fn;
540     insist (None = collector fn);
541     set_collector (Some "shadowstack") fn;
542     ignore (build_unreachable (builder_at_end (entry_block fn)));
543   end
544
545
546 (*===-- Basic Blocks ------------------------------------------------------===*)
547
548 let test_basic_blocks () =
549   let ty = function_type void_type [| |] in
550   
551   (* RUN: grep {Bb1} < %t.ll
552    *)
553   group "entry";
554   let fn = declare_function "X" ty m in
555   let bb = append_block "Bb1" fn in
556   insist (bb = entry_block fn);
557   ignore (build_unreachable (builder_at_end bb));
558   
559   (* RUN: grep -v Bb2 < %t.ll
560    *)
561   group "delete";
562   let fn = declare_function "X2" ty m in
563   let bb = append_block "Bb2" fn in
564   delete_block bb;
565   
566   group "insert";
567   let fn = declare_function "X3" ty m in
568   let bbb = append_block "b" fn in
569   let bba = insert_block "a" bbb in
570   insist ([| bba; bbb |] = basic_blocks fn);
571   ignore (build_unreachable (builder_at_end bba));
572   ignore (build_unreachable (builder_at_end bbb));
573   
574   (* RUN: grep Bb3 < %t.ll
575    *)
576   group "name/value";
577   let fn = define_function "X4" ty m in
578   let bb = entry_block fn in
579   ignore (build_unreachable (builder_at_end bb));
580   let bbv = value_of_block bb in
581   set_value_name "Bb3" bbv;
582   insist ("Bb3" = value_name bbv);
583   
584   group "casts";
585   let fn = define_function "X5" ty m in
586   let bb = entry_block fn in
587   ignore (build_unreachable (builder_at_end bb));
588   insist (bb = block_of_value (value_of_block bb));
589   insist (value_is_block (value_of_block bb));
590   insist (not (value_is_block (const_null i32_type)))
591
592
593 (*===-- Builder -----------------------------------------------------------===*)
594
595 let test_builder () =
596   let (++) x f = f x; x in
597   
598   begin group "parent";
599     insist (try
600               ignore (insertion_block (builder ()));
601               false
602             with Not_found ->
603               true);
604     
605     let fty = function_type void_type [| i32_type |] in
606     let fn = define_function "BuilderParent" fty m in
607     let bb = entry_block fn in
608     let b = builder_at_end bb in
609     let p = param fn 0 in
610     let sum = build_add p p "sum" b in
611     ignore (build_ret_void b);
612     
613     insist (fn = block_parent bb);
614     insist (fn = param_parent p);
615     insist (bb = instr_parent sum);
616     insist (bb = insertion_block b)
617   end;
618   
619   group "ret void";
620   begin
621     (* RUN: grep {ret void} < %t.ll
622      *)
623     let fty = function_type void_type [| |] in
624     let fn = declare_function "X6" fty m in
625     let b = builder_at_end (append_block "Bb01" fn) in
626     ignore (build_ret_void b)
627   end;
628   
629   (* The rest of the tests will use one big function. *)
630   let fty = function_type i32_type [| i32_type; i32_type |] in
631   let fn = define_function "X7" fty m in
632   let atentry = builder_at_end (entry_block fn) in
633   let p1 = param fn 0 ++ set_value_name "P1" in
634   let p2 = param fn 1 ++ set_value_name "P2" in
635   let f1 = build_uitofp p1 float_type "F1" atentry in
636   let f2 = build_uitofp p2 float_type "F2" atentry in
637   
638   let bb00 = append_block "Bb00" fn in
639   ignore (build_unreachable (builder_at_end bb00));
640   
641   group "ret"; begin
642     (* RUN: grep {ret.*P1} < %t.ll
643      *)
644     let ret = build_ret p1 atentry in
645     position_before ret atentry
646   end;
647   
648   group "br"; begin
649     (* RUN: grep {br.*Bb02} < %t.ll
650      *)
651     let bb02 = append_block "Bb02" fn in
652     let b = builder_at_end bb02 in
653     ignore (build_br bb02 b)
654   end;
655   
656   group "cond_br"; begin
657     (* RUN: grep {br.*Inst01.*Bb03.*Bb00} < %t.ll
658      *)
659     let bb03 = append_block "Bb03" fn in
660     let b = builder_at_end bb03 in
661     let cond = build_trunc p1 i1_type "Inst01" b in
662     ignore (build_cond_br cond bb03 bb00 b)
663   end;
664   
665   (* TODO: Switch *)
666   
667   group "invoke"; begin
668     (* RUN: grep {Inst02.*invoke.*P1.*P2} < %t.ll
669      * RUN: grep {to.*Bb04.*unwind.*Bb00} < %t.ll
670      *)
671     let bb04 = append_block "Bb04" fn in
672     let b = builder_at_end bb04 in
673     ignore (build_invoke fn [| p1; p2 |] bb04 bb00 "Inst02" b)
674   end;
675   
676   group "unwind"; begin
677     (* RUN: grep {unwind} < %t.ll
678      *)
679     let bb05 = append_block "Bb05" fn in
680     let b = builder_at_end bb05 in
681     ignore (build_unwind b)
682   end;
683   
684   group "unreachable"; begin
685     (* RUN: grep {unreachable} < %t.ll
686      *)
687     let bb06 = append_block "Bb06" fn in
688     let b = builder_at_end bb06 in
689     ignore (build_unreachable b)
690   end;
691   
692   group "arithmetic"; begin
693     let bb07 = append_block "Bb07" fn in
694     let b = builder_at_end bb07 in
695     
696     (* RUN: grep {Inst03.*add.*P1.*P2} < %t.ll
697      * RUN: grep {Inst04.*sub.*P1.*Inst03} < %t.ll
698      * RUN: grep {Inst05.*mul.*P1.*Inst04} < %t.ll
699      * RUN: grep {Inst06.*udiv.*P1.*Inst05} < %t.ll
700      * RUN: grep {Inst07.*sdiv.*P1.*Inst06} < %t.ll
701      * RUN: grep {Inst08.*fdiv.*F1.*F2} < %t.ll
702      * RUN: grep {Inst09.*urem.*P1.*Inst07} < %t.ll
703      * RUN: grep {Inst10.*srem.*P1.*Inst09} < %t.ll
704      * RUN: grep {Inst11.*frem.*F1.*Inst08} < %t.ll
705      * RUN: grep {Inst12.*shl.*P1.*Inst10} < %t.ll
706      * RUN: grep {Inst13.*lshr.*P1.*Inst12} < %t.ll
707      * RUN: grep {Inst14.*ashr.*P1.*Inst13} < %t.ll
708      * RUN: grep {Inst15.*and.*P1.*Inst14} < %t.ll
709      * RUN: grep {Inst16.*or.*P1.*Inst15} < %t.ll
710      * RUN: grep {Inst17.*xor.*P1.*Inst16} < %t.ll
711      * RUN: grep {Inst18.*sub.*0.*Inst17} < %t.ll
712      * RUN: grep {Inst19.*xor.*Inst18.*-1} < %t.ll
713      *)
714     let inst03 = build_add  p1 p2     "Inst03" b in
715     let inst04 = build_sub  p1 inst03 "Inst04" b in
716     let inst05 = build_mul  p1 inst04 "Inst05" b in
717     let inst06 = build_udiv p1 inst05 "Inst06" b in
718     let inst07 = build_sdiv p1 inst06 "Inst07" b in
719     let inst08 = build_fdiv f1 f2     "Inst08" b in
720     let inst09 = build_urem p1 inst07 "Inst09" b in
721     let inst10 = build_srem p1 inst09 "Inst10" b in
722           ignore(build_frem f1 inst08 "Inst11" b);
723     let inst12 = build_shl  p1 inst10 "Inst12" b in
724     let inst13 = build_lshr p1 inst12 "Inst13" b in
725     let inst14 = build_ashr p1 inst13 "Inst14" b in
726     let inst15 = build_and  p1 inst14 "Inst15" b in
727     let inst16 = build_or   p1 inst15 "Inst16" b in
728     let inst17 = build_xor  p1 inst16 "Inst17" b in
729     let inst18 = build_neg  inst17    "Inst18" b in
730          ignore (build_not  inst18    "Inst19" b);
731          ignore (build_unreachable b)
732   end;
733   
734   group "memory"; begin
735     let bb08 = append_block "Bb08" fn in
736     let b = builder_at_end bb08 in
737     
738     (* RUN: grep {Inst20.*malloc.*i8    } < %t.ll
739      * RUN: grep {Inst21.*malloc.*i8.*P1} < %t.ll
740      * RUN: grep {Inst22.*alloca.*i32   } < %t.ll
741      * RUN: grep {Inst23.*alloca.*i32.*P2} < %t.ll
742      * RUN: grep {free.*Inst20} < %t.ll
743      * RUN: grep {Inst25.*load.*Inst21} < %t.ll
744      * RUN: grep {store.*P2.*Inst22} < %t.ll
745      * RUN: grep {Inst27.*getelementptr.*Inst23.*P2} < %t.ll
746      *)
747     let inst20 = build_malloc i8_type "Inst20" b in
748     let inst21 = build_array_malloc i8_type p1 "Inst21" b in
749     let inst22 = build_alloca i32_type "Inst22" b in
750     let inst23 = build_array_alloca i32_type p2 "Inst23" b in
751           ignore(build_free inst20 b);
752           ignore(build_load inst21 "Inst25" b);
753           ignore(build_store p2 inst22 b);
754           ignore(build_gep inst23 [| p2 |] "Inst27" b);
755           ignore(build_unreachable b)
756   end;
757   
758   group "casts"; begin
759     let void_ptr = pointer_type i8_type in
760     
761     (* RUN: grep {Inst28.*trunc.*P1.*i8} < %t.ll
762      * RUN: grep {Inst29.*zext.*Inst28.*i32} < %t.ll
763      * RUN: grep {Inst30.*sext.*Inst29.*i64} < %t.ll
764      * RUN: grep {Inst31.*uitofp.*Inst30.*float} < %t.ll
765      * RUN: grep {Inst32.*sitofp.*Inst29.*double} < %t.ll
766      * RUN: grep {Inst33.*fptoui.*Inst31.*i32} < %t.ll
767      * RUN: grep {Inst34.*fptosi.*Inst32.*i64} < %t.ll
768      * RUN: grep {Inst35.*fptrunc.*Inst32.*float} < %t.ll
769      * RUN: grep {Inst36.*fpext.*Inst35.*double} < %t.ll
770      * RUN: grep {Inst37.*inttoptr.*P1.*i8\*} < %t.ll
771      * RUN: grep {Inst38.*ptrtoint.*Inst37.*i64} < %t.ll
772      * RUN: grep {Inst39.*bitcast.*Inst38.*double} < %t.ll
773      *)
774     let inst28 = build_trunc p1 i8_type "Inst28" atentry in
775     let inst29 = build_zext inst28 i32_type "Inst29" atentry in
776     let inst30 = build_sext inst29 i64_type "Inst30" atentry in
777     let inst31 = build_uitofp inst30 float_type "Inst31" atentry in
778     let inst32 = build_sitofp inst29 double_type "Inst32" atentry in
779           ignore(build_fptoui inst31 i32_type "Inst33" atentry);
780           ignore(build_fptosi inst32 i64_type "Inst34" atentry);
781     let inst35 = build_fptrunc inst32 float_type "Inst35" atentry in
782           ignore(build_fpext inst35 double_type "Inst36" atentry);
783     let inst37 = build_inttoptr p1 void_ptr "Inst37" atentry in
784     let inst38 = build_ptrtoint inst37 i64_type "Inst38" atentry in
785           ignore(build_bitcast inst38 double_type "Inst39" atentry)
786   end;
787   
788   group "comparisons"; begin
789     (* RUN: grep {Inst40.*icmp.*ne.*P1.*P2} < %t.ll
790      * RUN: grep {Inst41.*icmp.*sle.*P2.*P1} < %t.ll
791      * RUN: grep {Inst42.*fcmp.*false.*F1.*F2} < %t.ll
792      * RUN: grep {Inst43.*fcmp.*true.*F2.*F1} < %t.ll
793      *)
794     ignore (build_icmp Icmp.Ne    p1 p2 "Inst40" atentry);
795     ignore (build_icmp Icmp.Sle   p2 p1 "Inst41" atentry);
796     ignore (build_fcmp Fcmp.False f1 f2 "Inst42" atentry);
797     ignore (build_fcmp Fcmp.True  f2 f1 "Inst43" atentry)
798   end;
799   
800   group "miscellaneous"; begin
801     (* RUN: grep {CallInst.*call.*P2.*P1} < %t.ll
802      * RUN: grep {CallInst.*cc63} < %t.ll
803      * RUN: grep {Inst47.*select.*Inst46.*P1.*P2} < %t.ll
804      * RUN: grep {Inst48.*va_arg.*null.*i32} < %t.ll
805      * RUN: grep {Inst49.*extractelement.*Vec1.*P2} < %t.ll
806      * RUN: grep {Inst50.*insertelement.*Vec1.*P1.*P2} < %t.ll
807      * RUN: grep {Inst51.*shufflevector.*Vec1.*Vec2.*1.*1.*0.*0} < %t.ll
808      *)
809     let ci = build_call fn [| p2; p1 |] "CallInst" atentry in
810     insist (CallConv.c = instruction_call_conv ci);
811     set_instruction_call_conv 63 ci;
812     insist (63 = instruction_call_conv ci);
813     
814     let inst46 = build_icmp Icmp.Eq p1 p2 "Inst46" atentry in
815          ignore (build_select inst46 p1 p2 "Inst47" atentry);
816          ignore (build_va_arg
817                   (const_null (pointer_type (pointer_type i8_type)))
818                   i32_type "Inst48" atentry);
819     
820     (* Set up some vector vregs. *)
821     let one  = const_int i32_type 1 in
822     let zero = const_int i32_type 0 in
823     let t1 = const_vector [| one; zero; one; zero |] in
824     let t2 = const_vector [| zero; one; zero; one |] in
825     let t3 = const_vector [| one; one; zero; zero |] in
826     let vec1 = build_insertelement t1 p1 p2 "Vec1" atentry in
827     let vec2 = build_insertelement t2 p1 p2 "Vec2" atentry in
828     
829     ignore (build_extractelement vec1 p2 "Inst49" atentry);
830     ignore (build_insertelement vec1 p1 p2 "Inst50" atentry);
831     ignore (build_shufflevector vec1 vec2 t3 "Inst51" atentry);
832   end;
833   
834   group "phi"; begin
835     (* RUN: grep {PhiNode.*P1.*PhiBlock1.*P2.*PhiBlock2} < %t.ll
836      *)
837     let b1 = append_block "PhiBlock1" fn in
838     let b2 = append_block "PhiBlock2" fn in
839     
840     let jb = append_block "PhiJoinBlock" fn in
841     ignore (build_br jb (builder_at_end b1));
842     ignore (build_br jb (builder_at_end b2));
843     let at_jb = builder_at_end jb in
844     
845     let phi = build_phi [(p1, b1)] "PhiNode" at_jb in
846     insist ([(p1, b1)] = incoming phi);
847     
848     add_incoming (p2, b2) phi;
849     insist ([(p1, b1); (p2, b2)] = incoming phi);
850     
851     ignore (build_unreachable at_jb);
852   end
853
854
855 (*===-- Module Provider ---------------------------------------------------===*)
856
857 let test_module_provider () =
858   let m = create_module "test" in
859   let mp = ModuleProvider.create m in
860   ModuleProvider.dispose mp
861
862
863 (*===-- Pass Managers -----------------------------------------------------===*)
864
865 let test_pass_manager () =
866   let (++) x f = ignore (f x); x in
867
868   begin group "module pass manager";
869     ignore (PassManager.create ()
870              ++ PassManager.run_module m
871              ++ PassManager.dispose)
872   end;
873   
874   begin group "function pass manager";
875     let fty = function_type void_type [| |] in
876     let fn = define_function "FunctionPassManager" fty m in
877     ignore (build_ret_void (builder_at_end (entry_block fn)));
878     
879     ignore (PassManager.create_function mp
880              ++ PassManager.initialize
881              ++ PassManager.run_function fn
882              ++ PassManager.finalize
883              ++ PassManager.dispose)
884   end
885
886
887 (*===-- Writer ------------------------------------------------------------===*)
888
889 let test_writer () =
890   group "valid";
891   insist (match Llvm_analysis.verify_module m with
892           | None -> true
893           | Some msg -> prerr_string msg; false);
894
895   group "writer";
896   insist (write_bitcode_file m filename);
897   
898   ModuleProvider.dispose mp
899
900
901 (*===-- Driver ------------------------------------------------------------===*)
902
903 let _ =
904   suite "target"           test_target;
905   suite "types"            test_types;
906   suite "constants"        test_constants;
907   suite "global values"    test_global_values;
908   suite "global variables" test_global_variables;
909   suite "functions"        test_functions;
910   suite "basic blocks"     test_basic_blocks;
911   suite "builder"          test_builder;
912   suite "module provider"  test_module_provider;
913   suite "pass manager"     test_pass_manager;
914   suite "writer"           test_writer; (* Keep this last; it disposes m. *)
915   exit !exit_status