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