vmcore.ml revision 782ff3b70086bd3580f12228082ea998f152f9cd
1(* RUN: rm -rf %t.builddir
2 * RUN: mkdir -p %t.builddir
3 * RUN: cp %s %t.builddir
4 * RUN: %ocamlopt -warn-error A llvm.cmxa llvm_analysis.cmxa llvm_bitwriter.cmxa %t.builddir/vmcore.ml -o %t
5 * RUN: %t %t.bc
6 * RUN: llvm-dis < %t.bc > %t.ll
7 * RUN: FileCheck %s < %t.ll
8 * Do a second pass for things that shouldn't be anywhere.
9 * RUN: FileCheck -check-prefix=CHECK-NOWHERE %s < %t.ll
10 * XFAIL: vg_leak
11 *)
12
13(* Note: It takes several seconds for ocamlopt to link an executable with
14         libLLVMCore.a, so it's better to write a big test than a bunch of
15         little ones. *)
16
17open Llvm
18open Llvm_bitwriter
19
20
21(* Tiny unit test framework - really just to help find which line is busted *)
22let exit_status = ref 0
23let suite_name = ref ""
24let group_name = ref ""
25let case_num = ref 0
26let print_checkpoints = false
27let context = global_context ()
28let i1_type = Llvm.i1_type context
29let i8_type = Llvm.i8_type context
30let i16_type = Llvm.i16_type context
31let i32_type = Llvm.i32_type context
32let i64_type = Llvm.i64_type context
33let void_type = Llvm.void_type context
34let float_type = Llvm.float_type context
35let double_type = Llvm.double_type context
36let fp128_type = Llvm.fp128_type context
37
38let group name =
39  group_name := !suite_name ^ "/" ^ name;
40  case_num := 0;
41  if print_checkpoints then
42    prerr_endline ("  " ^ name ^ "...")
43
44let insist cond =
45  incr case_num;
46  if not cond then
47    exit_status := 10;
48  match print_checkpoints, cond with
49  | false, true -> ()
50  | false, false ->
51      prerr_endline ("FAILED: " ^ !suite_name ^ "/" ^ !group_name ^ " #" ^ (string_of_int !case_num))
52  | true, true ->
53      prerr_endline ("    " ^ (string_of_int !case_num))
54  | true, false ->
55      prerr_endline ("    " ^ (string_of_int !case_num) ^ " FAIL")
56
57let suite name f =
58  suite_name := name;
59  if print_checkpoints then
60    prerr_endline (name ^ ":");
61  f ()
62
63
64(*===-- Fixture -----------------------------------------------------------===*)
65
66let filename = Sys.argv.(1)
67let m = create_module context filename
68
69
70(*===-- Conversion --------------------------------------------------------===*)
71
72let test_conversion () =
73  insist ("i32" = (string_of_lltype i32_type));
74  let c = const_int i32_type 42 in
75  insist ("i32 42" = (string_of_llvalue c))
76
77
78(*===-- Target ------------------------------------------------------------===*)
79
80let test_target () =
81  begin group "triple";
82    let trip = "i686-apple-darwin8" in
83    set_target_triple trip m;
84    insist (trip = target_triple m)
85  end;
86  
87  begin group "layout";
88    let layout = "bogus" in
89    set_data_layout layout m;
90    insist (layout = data_layout m)
91  end
92  (* CHECK: target datalayout = "bogus"
93   * CHECK: target triple = "i686-apple-darwin8"
94   *)
95
96
97(*===-- Constants ---------------------------------------------------------===*)
98
99let test_constants () =
100  (* CHECK: const_int{{.*}}i32{{.*}}-1
101   *)
102  group "int";
103  let c = const_int i32_type (-1) in
104  ignore (define_global "const_int" c m);
105  insist (i32_type = type_of c);
106  insist (is_constant c);
107
108  (* CHECK: const_sext_int{{.*}}i64{{.*}}-1
109   *)
110  group "sext int";
111  let c = const_int i64_type (-1) in
112  ignore (define_global "const_sext_int" c m);
113  insist (i64_type = type_of c);
114
115  (* CHECK: const_zext_int64{{.*}}i64{{.*}}4294967295
116   *)
117  group "zext int64";
118  let c = const_of_int64 i64_type (Int64.of_string "4294967295") false in
119  ignore (define_global "const_zext_int64" c m);
120  insist (i64_type = type_of c);
121
122  (* CHECK: const_int_string{{.*}}i32{{.*}}-1
123   *)
124  group "int string";
125  let c = const_int_of_string i32_type "-1" 10 in
126  ignore (define_global "const_int_string" c m);
127  insist (i32_type = type_of c);
128
129  (* CHECK: @const_string = global {{.*}}c"cruel\00world"
130   *)
131  group "string";
132  let c = const_string context "cruel\000world" in
133  ignore (define_global "const_string" c m);
134  insist ((array_type i8_type 11) = type_of c);
135
136  (* CHECK: const_stringz{{.*}}"hi\00again\00"
137   *)
138  group "stringz";
139  let c = const_stringz context "hi\000again" in
140  ignore (define_global "const_stringz" c m);
141  insist ((array_type i8_type 9) = type_of c);
142
143  (* CHECK: const_single{{.*}}2.75
144   * CHECK: const_double{{.*}}3.1459
145   * CHECK: const_double_string{{.*}}1.25
146   *)
147  begin group "real";
148    let cs = const_float float_type 2.75 in
149    ignore (define_global "const_single" cs m);
150    insist (float_type = type_of cs);
151    
152    let cd = const_float double_type 3.1459 in
153    ignore (define_global "const_double" cd m);
154    insist (double_type = type_of cd);
155
156    let cd = const_float_of_string double_type "1.25" in
157    ignore (define_global "const_double_string" cd m);
158    insist (double_type = type_of cd)
159  end;
160  
161  let one = const_int i16_type 1 in
162  let two = const_int i16_type 2 in
163  let three = const_int i32_type 3 in
164  let four = const_int i32_type 4 in
165  
166  (* CHECK: const_array{{.*}}[i32 3, i32 4]
167   *)
168  group "array";
169  let c = const_array i32_type [| three; four |] in
170  ignore (define_global "const_array" c m);
171  insist ((array_type i32_type 2) = (type_of c));
172  
173  (* CHECK: const_vector{{.*}}<i16 1, i16 2{{.*}}>
174   *)
175  group "vector";
176  let c = const_vector [| one; two; one; two;
177                          one; two; one; two |] in
178  ignore (define_global "const_vector" c m);
179  insist ((vector_type i16_type 8) = (type_of c));
180
181  (* CHECK: const_structure{{.*.}}i16 1, i16 2, i32 3, i32 4
182   *)
183  group "structure";
184  let c = const_struct context [| one; two; three; four |] in
185  ignore (define_global "const_structure" c m);
186  insist ((struct_type context [| i16_type; i16_type; i32_type; i32_type |])
187        = (type_of c));
188
189  (* CHECK: const_null{{.*}}zeroinit
190   *)
191  group "null";
192  let c = const_null (packed_struct_type context [| i1_type; i8_type; i64_type;
193                                                    double_type |]) in
194  ignore (define_global "const_null" c m);
195  
196  (* CHECK: const_all_ones{{.*}}-1
197   *)
198  group "all ones";
199  let c = const_all_ones i64_type in
200  ignore (define_global "const_all_ones" c m);
201
202  group "pointer null"; begin
203    (* CHECK: const_pointer_null = global i64* null
204     *)
205    let c = const_pointer_null (pointer_type i64_type) in
206    ignore (define_global "const_pointer_null" c m);
207  end;
208  
209  (* CHECK: const_undef{{.*}}undef
210   *)
211  group "undef";
212  let c = undef i1_type in
213  ignore (define_global "const_undef" c m);
214  insist (i1_type = type_of c);
215  insist (is_undef c);
216  
217  group "constant arithmetic";
218  (* CHECK: @const_neg = global i64 sub
219   * CHECK: @const_nsw_neg = global i64 sub nsw
220   * CHECK: @const_nuw_neg = global i64 sub nuw
221   * CHECK: @const_fneg = global double fsub
222   * CHECK: @const_not = global i64 xor
223   * CHECK: @const_add = global i64 add
224   * CHECK: @const_nsw_add = global i64 add nsw
225   * CHECK: @const_nuw_add = global i64 add nuw
226   * CHECK: @const_fadd = global double fadd
227   * CHECK: @const_sub = global i64 sub
228   * CHECK: @const_nsw_sub = global i64 sub nsw
229   * CHECK: @const_nuw_sub = global i64 sub nuw
230   * CHECK: @const_fsub = global double fsub
231   * CHECK: @const_mul = global i64 mul
232   * CHECK: @const_nsw_mul = global i64 mul nsw
233   * CHECK: @const_nuw_mul = global i64 mul nuw
234   * CHECK: @const_fmul = global double fmul
235   * CHECK: @const_udiv = global i64 udiv
236   * CHECK: @const_sdiv = global i64 sdiv
237   * CHECK: @const_exact_sdiv = global i64 sdiv exact
238   * CHECK: @const_fdiv = global double fdiv
239   * CHECK: @const_urem = global i64 urem
240   * CHECK: @const_srem = global i64 srem
241   * CHECK: @const_frem = global double frem
242   * CHECK: @const_and = global i64 and
243   * CHECK: @const_or = global i64 or
244   * CHECK: @const_xor = global i64 xor
245   * CHECK: @const_icmp = global i1 icmp sle
246   * CHECK: @const_fcmp = global i1 fcmp ole
247   *)
248  let void_ptr = pointer_type i8_type in
249  let five = const_int i64_type 5 in
250  let ffive = const_uitofp five double_type in
251  let foldbomb_gv = define_global "FoldBomb" (const_null i8_type) m in
252  let foldbomb = const_ptrtoint foldbomb_gv i64_type in
253  let ffoldbomb = const_uitofp foldbomb double_type in
254  ignore (define_global "const_neg" (const_neg foldbomb) m);
255  ignore (define_global "const_nsw_neg" (const_nsw_neg foldbomb) m);
256  ignore (define_global "const_nuw_neg" (const_nuw_neg foldbomb) m);
257  ignore (define_global "const_fneg" (const_fneg ffoldbomb) m);
258  ignore (define_global "const_not" (const_not foldbomb) m);
259  ignore (define_global "const_add" (const_add foldbomb five) m);
260  ignore (define_global "const_nsw_add" (const_nsw_add foldbomb five) m);
261  ignore (define_global "const_nuw_add" (const_nuw_add foldbomb five) m);
262  ignore (define_global "const_fadd" (const_fadd ffoldbomb ffive) m);
263  ignore (define_global "const_sub" (const_sub foldbomb five) m);
264  ignore (define_global "const_nsw_sub" (const_nsw_sub foldbomb five) m);
265  ignore (define_global "const_nuw_sub" (const_nuw_sub foldbomb five) m);
266  ignore (define_global "const_fsub" (const_fsub ffoldbomb ffive) m);
267  ignore (define_global "const_mul" (const_mul foldbomb five) m);
268  ignore (define_global "const_nsw_mul" (const_nsw_mul foldbomb five) m);
269  ignore (define_global "const_nuw_mul" (const_nuw_mul foldbomb five) m);
270  ignore (define_global "const_fmul" (const_fmul ffoldbomb ffive) m);
271  ignore (define_global "const_udiv" (const_udiv foldbomb five) m);
272  ignore (define_global "const_sdiv" (const_sdiv foldbomb five) m);
273  ignore (define_global "const_exact_sdiv" (const_exact_sdiv foldbomb five) m);
274  ignore (define_global "const_fdiv" (const_fdiv ffoldbomb ffive) m);
275  ignore (define_global "const_urem" (const_urem foldbomb five) m);
276  ignore (define_global "const_srem" (const_srem foldbomb five) m);
277  ignore (define_global "const_frem" (const_frem ffoldbomb ffive) m);
278  ignore (define_global "const_and" (const_and foldbomb five) m);
279  ignore (define_global "const_or" (const_or foldbomb five) m);
280  ignore (define_global "const_xor" (const_xor foldbomb five) m);
281  ignore (define_global "const_icmp" (const_icmp Icmp.Sle foldbomb five) m);
282  ignore (define_global "const_fcmp" (const_fcmp Fcmp.Ole ffoldbomb ffive) m);
283  
284  group "constant casts";
285  (* CHECK: const_trunc{{.*}}trunc
286   * CHECK: const_sext{{.*}}sext
287   * CHECK: const_zext{{.*}}zext
288   * CHECK: const_fptrunc{{.*}}fptrunc
289   * CHECK: const_fpext{{.*}}fpext
290   * CHECK: const_uitofp{{.*}}uitofp
291   * CHECK: const_sitofp{{.*}}sitofp
292   * CHECK: const_fptoui{{.*}}fptoui
293   * CHECK: const_fptosi{{.*}}fptosi
294   * CHECK: const_ptrtoint{{.*}}ptrtoint
295   * CHECK: const_inttoptr{{.*}}inttoptr
296   * CHECK: const_bitcast{{.*}}bitcast
297   * CHECK: const_intcast{{.*}}zext
298   *)
299  let i128_type = integer_type context 128 in
300  ignore (define_global "const_trunc" (const_trunc (const_add foldbomb five)
301                                               i8_type) m);
302  ignore (define_global "const_sext" (const_sext foldbomb i128_type) m);
303  ignore (define_global "const_zext" (const_zext foldbomb i128_type) m);
304  ignore (define_global "const_fptrunc" (const_fptrunc ffoldbomb float_type) m);
305  ignore (define_global "const_fpext" (const_fpext ffoldbomb fp128_type) m);
306  ignore (define_global "const_uitofp" (const_uitofp foldbomb double_type) m);
307  ignore (define_global "const_sitofp" (const_sitofp foldbomb double_type) m);
308  ignore (define_global "const_fptoui" (const_fptoui ffoldbomb i32_type) m);
309  ignore (define_global "const_fptosi" (const_fptosi ffoldbomb i32_type) m);
310  ignore (define_global "const_ptrtoint" (const_ptrtoint 
311    (const_gep (const_null (pointer_type i8_type))
312               [| const_int i32_type 1 |])
313    i32_type) m);
314  ignore (define_global "const_inttoptr" (const_inttoptr (const_add foldbomb five)
315                                                  void_ptr) m);
316  ignore (define_global "const_bitcast" (const_bitcast ffoldbomb i64_type) m);
317  ignore (define_global "const_intcast"
318          (const_intcast foldbomb i128_type ~is_signed:false) m);
319  
320  group "misc constants";
321  (* CHECK: const_size_of{{.*}}getelementptr{{.*}}null
322   * CHECK: const_gep{{.*}}getelementptr
323   * CHECK: const_select{{.*}}select
324   * CHECK: const_extractelement{{.*}}extractelement
325   * CHECK: const_insertelement{{.*}}insertelement
326   * CHECK: const_shufflevector = global <4 x i32> <i32 0, i32 1, i32 1, i32 0>
327   *)
328  ignore (define_global "const_size_of" (size_of (pointer_type i8_type)) m);
329  ignore (define_global "const_gep" (const_gep foldbomb_gv [| five |]) m);
330  ignore (define_global "const_select" (const_select
331    (const_icmp Icmp.Sle foldbomb five)
332    (const_int i8_type (-1))
333    (const_int i8_type 0)) m);
334  let zero = const_int i32_type 0 in
335  let one  = const_int i32_type 1 in
336  ignore (define_global "const_extractelement" (const_extractelement
337    (const_vector [| zero; one; zero; one |])
338    (const_trunc foldbomb i32_type)) m);
339  ignore (define_global "const_insertelement" (const_insertelement
340    (const_vector [| zero; one; zero; one |])
341    zero (const_trunc foldbomb i32_type)) m);
342  ignore (define_global "const_shufflevector" (const_shufflevector
343    (const_vector [| zero; one |])
344    (const_vector [| one; zero |])
345    (const_vector [| const_int i32_type 0; const_int i32_type 1;
346                     const_int i32_type 2; const_int i32_type 3 |])) m);
347
348  group "asm"; begin
349    let ft = function_type void_type [| i32_type; i32_type; i32_type |] in
350    ignore (const_inline_asm
351      ft
352      ""
353      "{cx},{ax},{di},~{dirflag},~{fpsr},~{flags},~{edi},~{ecx}"
354      true
355      false)
356  end;
357
358  group "recursive struct"; begin
359      let nsty = named_struct_type context "rec" in
360      let pty = pointer_type nsty in
361      struct_set_body nsty [| i32_type; pty |] false;
362      let elts = [| const_int i32_type 4; const_pointer_null pty |] in
363      let grec_init = const_named_struct nsty elts in
364      ignore (define_global "grec" grec_init m);
365      ignore (string_of_lltype nsty);
366  end
367
368
369(*===-- Global Values -----------------------------------------------------===*)
370
371let test_global_values () =
372  let (++) x f = f x; x in
373  let zero32 = const_null i32_type in
374
375  (* CHECK: GVal01
376   *)
377  group "naming";
378  let g = define_global "TEMPORARY" zero32 m in
379  insist ("TEMPORARY" = value_name g);
380  set_value_name "GVal01" g;
381  insist ("GVal01" = value_name g);
382
383  (* CHECK: GVal02{{.*}}linkonce
384   *)
385  group "linkage";
386  let g = define_global "GVal02" zero32 m ++
387          set_linkage Linkage.Link_once in
388  insist (Linkage.Link_once = linkage g);
389
390  (* CHECK: GVal03{{.*}}Hanalei
391   *)
392  group "section";
393  let g = define_global "GVal03" zero32 m ++
394          set_section "Hanalei" in
395  insist ("Hanalei" = section g);
396  
397  (* CHECK: GVal04{{.*}}hidden
398   *)
399  group "visibility";
400  let g = define_global "GVal04" zero32 m ++
401          set_visibility Visibility.Hidden in
402  insist (Visibility.Hidden = visibility g);
403  
404  (* CHECK: GVal05{{.*}}align 128
405   *)
406  group "alignment";
407  let g = define_global "GVal05" zero32 m ++
408          set_alignment 128 in
409  insist (128 = alignment g)
410
411
412(*===-- Global Variables --------------------------------------------------===*)
413
414let test_global_variables () =
415  let (++) x f = f x; x in
416  let fourty_two32 = const_int i32_type 42 in
417
418  group "declarations"; begin
419    (* CHECK: @GVar01 = external global i32
420     * CHECK: @QGVar01 = external addrspace(3) global i32
421     *)
422    insist (None == lookup_global "GVar01" m);
423    let g = declare_global i32_type "GVar01" m in
424    insist (is_declaration g);
425    insist (pointer_type float_type ==
426              type_of (declare_global float_type "GVar01" m));
427    insist (g == declare_global i32_type "GVar01" m);
428    insist (match lookup_global "GVar01" m with Some x -> x = g
429                                              | None -> false);
430
431    insist (None == lookup_global "QGVar01" m);
432    let g = declare_qualified_global i32_type "QGVar01" 3 m in
433    insist (is_declaration g);
434    insist (qualified_pointer_type float_type 3 ==
435              type_of (declare_qualified_global float_type "QGVar01" 3 m));
436    insist (g == declare_qualified_global i32_type "QGVar01" 3 m);
437    insist (match lookup_global "QGVar01" m with Some x -> x = g
438                                              | None -> false);
439  end;
440  
441  group "definitions"; begin
442    (* CHECK: @GVar02 = global i32 42
443     * CHECK: @GVar03 = global i32 42
444     * CHECK: @QGVar02 = addrspace(3) global i32 42
445     * CHECK: @QGVar03 = addrspace(3) global i32 42
446     *)
447    let g = define_global "GVar02" fourty_two32 m in
448    let g2 = declare_global i32_type "GVar03" m ++
449           set_initializer fourty_two32 in
450    insist (not (is_declaration g));
451    insist (not (is_declaration g2));
452    insist ((global_initializer g) == (global_initializer g2));
453
454    let g = define_qualified_global "QGVar02" fourty_two32 3 m in
455    let g2 = declare_qualified_global i32_type "QGVar03" 3 m ++
456           set_initializer fourty_two32 in
457    insist (not (is_declaration g));
458    insist (not (is_declaration g2));
459    insist ((global_initializer g) == (global_initializer g2));
460  end;
461
462  (* CHECK: GVar04{{.*}}thread_local
463   *)
464  group "threadlocal";
465  let g = define_global "GVar04" fourty_two32 m ++
466          set_thread_local true in
467  insist (is_thread_local g);
468
469  (* CHECK: GVar05{{.*}}thread_local(initialexec)
470   *)
471  group "threadlocal_mode";
472  let g = define_global "GVar05" fourty_two32 m ++
473          set_thread_local_mode ThreadLocalMode.InitialExec in
474  insist ((thread_local_mode g) = ThreadLocalMode.InitialExec);
475
476  (* CHECK: GVar06{{.*}}externally_initialized
477   *)
478  group "externally_initialized";
479  let g = define_global "GVar06" fourty_two32 m ++
480          set_externally_initialized true in
481  insist (is_externally_initialized g);
482
483  (* CHECK-NOWHERE-NOT: GVar07
484   *)
485  group "delete";
486  let g = define_global "GVar07" fourty_two32 m in
487  delete_global g;
488
489  (* CHECK: ConstGlobalVar{{.*}}constant
490   *)
491  group "constant";
492  let g = define_global "ConstGlobalVar" fourty_two32 m in
493  insist (not (is_global_constant g));
494  set_global_constant true g;
495  insist (is_global_constant g);
496  
497  begin group "iteration";
498    let m = create_module context "temp" in
499    
500    insist (At_end m = global_begin m);
501    insist (At_start m = global_end m);
502    
503    let g1 = declare_global i32_type "One" m in
504    let g2 = declare_global i32_type "Two" m in
505    
506    insist (Before g1 = global_begin m);
507    insist (Before g2 = global_succ g1);
508    insist (At_end m = global_succ g2);
509    
510    insist (After g2 = global_end m);
511    insist (After g1 = global_pred g2);
512    insist (At_start m = global_pred g1);
513    
514    let lf s x = s ^ "->" ^ value_name x in
515    insist ("->One->Two" = fold_left_globals lf "" m);
516    
517    let rf x s = value_name x ^ "<-" ^ s in
518    insist ("One<-Two<-" = fold_right_globals rf m "");
519    
520    dispose_module m
521  end
522
523(* String globals built below are emitted here.
524 * CHECK: build_global_string{{.*}}stringval
525 *)
526
527
528(*===-- Uses --------------------------------------------------------------===*)
529
530let test_uses () =
531  let ty = function_type i32_type [| i32_type; i32_type |] in
532  let fn = define_function "use_function" ty m in
533  let b = builder_at_end context (entry_block fn) in
534
535  let p1 = param fn 0 in
536  let p2 = param fn 1 in
537  let v1 = build_add p1 p2 "v1" b in
538  let v2 = build_add p1 v1 "v2" b in
539  let _ = build_add v1 v2 "v3" b in
540
541  let lf s u = value_name (user u) ^ "->" ^ s in
542  insist ("v2->v3->" = fold_left_uses lf "" v1);
543  let rf u s = value_name (user u) ^ "<-" ^ s in
544  insist ("v3<-v2<-" = fold_right_uses rf v1 "");
545
546  let lf s u = value_name (used_value u) ^ "->" ^ s in
547  insist ("v1->v1->" = fold_left_uses lf "" v1);
548
549  let rf u s = value_name (used_value u) ^ "<-" ^ s in
550  insist ("v1<-v1<-" = fold_right_uses rf v1 "");
551
552  ignore (build_unreachable b)
553
554
555(*===-- Users -------------------------------------------------------------===*)
556
557let test_users () =
558  let ty = function_type i32_type [| i32_type; i32_type |] in
559  let fn = define_function "user_function" ty m in
560  let b = builder_at_end context (entry_block fn) in
561
562  let p1 = param fn 0 in
563  let p2 = param fn 1 in
564  let a3 = build_alloca i32_type "user_alloca" b in
565  let p3 = build_load a3 "user_load" b in
566  let i = build_add p1 p2 "sum" b in
567
568  insist ((num_operands i) = 2);
569  insist ((operand i 0) = p1);
570  insist ((operand i 1) = p2);
571
572  set_operand i 1 p3;
573  insist ((operand i 1) != p2);
574  insist ((operand i 1) = p3);
575
576  ignore (build_unreachable b)
577
578
579(*===-- Aliases -----------------------------------------------------------===*)
580
581let test_aliases () =
582  (* CHECK: @alias = alias i32* @aliasee
583   *)
584  let v = declare_global i32_type "aliasee" m in
585  ignore (add_alias m (pointer_type i32_type) v "alias")
586
587
588(*===-- Functions ---------------------------------------------------------===*)
589
590let test_functions () =
591  let ty = function_type i32_type [| i32_type; i64_type |] in
592  let ty2 = function_type i8_type [| i8_type; i64_type |] in
593  
594  (* CHECK: declare i32 @Fn1(i32, i64)
595   *)
596  begin group "declare";
597    insist (None = lookup_function "Fn1" m);
598    let fn = declare_function "Fn1" ty m in
599    insist (pointer_type ty = type_of fn);
600    insist (is_declaration fn);
601    insist (0 = Array.length (basic_blocks fn));
602    insist (pointer_type ty2 == type_of (declare_function "Fn1" ty2 m));
603    insist (fn == declare_function "Fn1" ty m);
604    insist (None <> lookup_function "Fn1" m);
605    insist (match lookup_function "Fn1" m with Some x -> x = fn
606                                             | None -> false);
607    insist (m == global_parent fn)
608  end;
609  
610  (* CHECK-NOWHERE-NOT: Fn2
611   *)
612  group "delete";
613  let fn = declare_function "Fn2" ty m in
614  delete_function fn;
615  
616  (* CHECK: define{{.*}}Fn3
617   *)
618  group "define";
619  let fn = define_function "Fn3" ty m in
620  insist (not (is_declaration fn));
621  insist (1 = Array.length (basic_blocks fn));
622  ignore (build_unreachable (builder_at_end context (entry_block fn)));
623  
624  (* CHECK: define{{.*}}Fn4{{.*}}Param1{{.*}}Param2
625   *)
626  group "params";
627  let fn = define_function "Fn4" ty m in
628  let params = params fn in
629  insist (2 = Array.length params);
630  insist (params.(0) = param fn 0);
631  insist (params.(1) = param fn 1);
632  insist (i32_type = type_of params.(0));
633  insist (i64_type = type_of params.(1));
634  set_value_name "Param1" params.(0);
635  set_value_name "Param2" params.(1);
636  ignore (build_unreachable (builder_at_end context (entry_block fn)));
637  
638  (* CHECK: fastcc{{.*}}Fn5
639   *)
640  group "callconv";
641  let fn = define_function "Fn5" ty m in
642  insist (CallConv.c = function_call_conv fn);
643  set_function_call_conv CallConv.fast fn;
644  insist (CallConv.fast = function_call_conv fn);
645  ignore (build_unreachable (builder_at_end context (entry_block fn)));
646  
647  begin group "gc";
648    (* CHECK: Fn6{{.*}}gc{{.*}}shadowstack
649     *)
650    let fn = define_function "Fn6" ty m in
651    insist (None = gc fn);
652    set_gc (Some "ocaml") fn;
653    insist (Some "ocaml" = gc fn);
654    set_gc None fn;
655    insist (None = gc fn);
656    set_gc (Some "shadowstack") fn;
657    ignore (build_unreachable (builder_at_end context (entry_block fn)));
658  end;
659  
660  begin group "iteration";
661    let m = create_module context "temp" in
662    
663    insist (At_end m = function_begin m);
664    insist (At_start m = function_end m);
665    
666    let f1 = define_function "One" ty m in
667    let f2 = define_function "Two" ty m in
668    
669    insist (Before f1 = function_begin m);
670    insist (Before f2 = function_succ f1);
671    insist (At_end m = function_succ f2);
672    
673    insist (After f2 = function_end m);
674    insist (After f1 = function_pred f2);
675    insist (At_start m = function_pred f1);
676    
677    let lf s x = s ^ "->" ^ value_name x in
678    insist ("->One->Two" = fold_left_functions lf "" m);
679    
680    let rf x s = value_name x ^ "<-" ^ s in
681    insist ("One<-Two<-" = fold_right_functions rf m "");
682    
683    dispose_module m
684  end
685
686
687(*===-- Params ------------------------------------------------------------===*)
688
689let test_params () =
690  begin group "iteration";
691    let m = create_module context "temp" in
692    
693    let vf = define_function "void" (function_type void_type [| |]) m in
694    
695    insist (At_end vf = param_begin vf);
696    insist (At_start vf = param_end vf);
697    
698    let ty = function_type void_type [| i32_type; i32_type |] in
699    let f = define_function "f" ty m in
700    let p1 = param f 0 in
701    let p2 = param f 1 in
702    set_value_name "One" p1;
703    set_value_name "Two" p2;
704    add_param_attr p1 Attribute.Sext;
705    add_param_attr p2 Attribute.Noalias;
706    remove_param_attr p2 Attribute.Noalias;
707    add_function_attr f Attribute.Nounwind;
708    add_function_attr f Attribute.Noreturn;
709    remove_function_attr f Attribute.Noreturn;
710
711    insist (Before p1 = param_begin f);
712    insist (Before p2 = param_succ p1);
713    insist (At_end f = param_succ p2);
714    
715    insist (After p2 = param_end f);
716    insist (After p1 = param_pred p2);
717    insist (At_start f = param_pred p1);
718    
719    let lf s x = s ^ "->" ^ value_name x in
720    insist ("->One->Two" = fold_left_params lf "" f);
721    
722    let rf x s = value_name x ^ "<-" ^ s in
723    insist ("One<-Two<-" = fold_right_params rf f "");
724    
725    dispose_module m
726  end
727
728
729(*===-- Basic Blocks ------------------------------------------------------===*)
730
731let test_basic_blocks () =
732  let ty = function_type void_type [| |] in
733  
734  (* CHECK: Bb1
735   *)
736  group "entry";
737  let fn = declare_function "X" ty m in
738  let bb = append_block context "Bb1" fn in
739  insist (bb = entry_block fn);
740  ignore (build_unreachable (builder_at_end context bb));
741  
742  (* CHECK-NOWHERE-NOT: Bb2
743   *)
744  group "delete";
745  let fn = declare_function "X2" ty m in
746  let bb = append_block context "Bb2" fn in
747  delete_block bb;
748  
749  group "insert";
750  let fn = declare_function "X3" ty m in
751  let bbb = append_block context "b" fn in
752  let bba = insert_block context "a" bbb in
753  insist ([| bba; bbb |] = basic_blocks fn);
754  ignore (build_unreachable (builder_at_end context bba));
755  ignore (build_unreachable (builder_at_end context bbb));
756  
757  (* CHECK: Bb3
758   *)
759  group "name/value";
760  let fn = define_function "X4" ty m in
761  let bb = entry_block fn in
762  ignore (build_unreachable (builder_at_end context bb));
763  let bbv = value_of_block bb in
764  set_value_name "Bb3" bbv;
765  insist ("Bb3" = value_name bbv);
766  
767  group "casts";
768  let fn = define_function "X5" ty m in
769  let bb = entry_block fn in
770  ignore (build_unreachable (builder_at_end context bb));
771  insist (bb = block_of_value (value_of_block bb));
772  insist (value_is_block (value_of_block bb));
773  insist (not (value_is_block (const_null i32_type)));
774  
775  begin group "iteration";
776    let m = create_module context "temp" in
777    let f = declare_function "Temp" (function_type i32_type [| |]) m in
778    
779    insist (At_end f = block_begin f);
780    insist (At_start f = block_end f);
781    
782    let b1 = append_block context "One" f in
783    let b2 = append_block context "Two" f in
784    
785    insist (Before b1 = block_begin f);
786    insist (Before b2 = block_succ b1);
787    insist (At_end f = block_succ b2);
788    
789    insist (After b2 = block_end f);
790    insist (After b1 = block_pred b2);
791    insist (At_start f = block_pred b1);
792    
793    let lf s x = s ^ "->" ^ value_name (value_of_block x) in
794    insist ("->One->Two" = fold_left_blocks lf "" f);
795    
796    let rf x s = value_name (value_of_block x) ^ "<-" ^ s in
797    insist ("One<-Two<-" = fold_right_blocks rf f "");
798    
799    dispose_module m
800  end
801
802
803(*===-- Instructions ------------------------------------------------------===*)
804
805let test_instructions () =
806  begin group "iteration";
807    let m = create_module context "temp" in
808    let fty = function_type void_type [| i32_type; i32_type |] in
809    let f = define_function "f" fty m in
810    let bb = entry_block f in
811    let b = builder_at context (At_end bb) in
812    
813    insist (At_end bb = instr_begin bb);
814    insist (At_start bb = instr_end bb);
815    
816    let i1 = build_add (param f 0) (param f 1) "One" b in
817    let i2 = build_sub (param f 0) (param f 1) "Two" b in
818    
819    insist (Before i1 = instr_begin bb);
820    insist (Before i2 = instr_succ i1);
821    insist (At_end bb = instr_succ i2);
822    
823    insist (After i2 = instr_end bb);
824    insist (After i1 = instr_pred i2);
825    insist (At_start bb = instr_pred i1);
826    
827    let lf s x = s ^ "->" ^ value_name x in
828    insist ("->One->Two" = fold_left_instrs lf "" bb);
829    
830    let rf x s = value_name x ^ "<-" ^ s in
831    insist ("One<-Two<-" = fold_right_instrs rf bb "");
832    
833    dispose_module m
834  end
835
836
837(*===-- Builder -----------------------------------------------------------===*)
838
839let test_builder () =
840  let (++) x f = f x; x in
841  
842  begin group "parent";
843    insist (try
844              ignore (insertion_block (builder context));
845              false
846            with Not_found ->
847              true);
848    
849    let fty = function_type void_type [| i32_type |] in
850    let fn = define_function "BuilderParent" fty m in
851    let bb = entry_block fn in
852    let b = builder_at_end context bb in
853    let p = param fn 0 in
854    let sum = build_add p p "sum" b in
855    ignore (build_ret_void b);
856    
857    insist (fn = block_parent bb);
858    insist (fn = param_parent p);
859    insist (bb = instr_parent sum);
860    insist (bb = insertion_block b)
861  end;
862  
863  group "ret void";
864  begin
865    (* CHECK: ret void
866     *)
867    let fty = function_type void_type [| |] in
868    let fn = declare_function "X6" fty m in
869    let b = builder_at_end context (append_block context "Bb01" fn) in
870    ignore (build_ret_void b)
871  end;
872
873  group "ret aggregate";
874  begin
875      (* CHECK: ret { i8, i64 } { i8 4, i64 5 }
876       *)
877      let sty = struct_type context [| i8_type; i64_type |] in
878      let fty = function_type sty [| |] in
879      let fn = declare_function "XA6" fty m in
880      let b = builder_at_end context (append_block context "Bb01" fn) in
881      let agg = [| const_int i8_type 4; const_int i64_type 5 |] in
882      ignore (build_aggregate_ret agg b)
883  end;
884  
885  (* The rest of the tests will use one big function. *)
886  let fty = function_type i32_type [| i32_type; i32_type |] in
887  let fn = define_function "X7" fty m in
888  let atentry = builder_at_end context (entry_block fn) in
889  let p1 = param fn 0 ++ set_value_name "P1" in
890  let p2 = param fn 1 ++ set_value_name "P2" in
891  let f1 = build_uitofp p1 float_type "F1" atentry in
892  let f2 = build_uitofp p2 float_type "F2" atentry in
893  
894  let bb00 = append_block context "Bb00" fn in
895  ignore (build_unreachable (builder_at_end context bb00));
896
897  group "function attribute";
898  begin
899      ignore (add_function_attr fn Attribute.UWTable);
900      (* CHECK: X7{{.*}}#0
901       * #0 is uwtable, defined at EOF.
902       *)
903      insist ([Attribute.UWTable] = function_attr fn);
904  end;
905
906  group "casts"; begin
907    let void_ptr = pointer_type i8_type in
908
909    (* CHECK-DAG: %build_trunc = trunc i32 %P1 to i8
910     * CHECK-DAG: %build_trunc2 = trunc i32 %P1 to i8
911     * CHECK-DAG: %build_trunc3 = trunc i32 %P1 to i8
912     * CHECK-DAG: %build_zext = zext i8 %build_trunc to i32
913     * CHECK-DAG: %build_zext2 = zext i8 %build_trunc to i32
914     * CHECK-DAG: %build_sext = sext i32 %build_zext to i64
915     * CHECK-DAG: %build_sext2 = sext i32 %build_zext to i64
916     * CHECK-DAG: %build_sext3 = sext i32 %build_zext to i64
917     * CHECK-DAG: %build_uitofp = uitofp i64 %build_sext to float
918     * CHECK-DAG: %build_sitofp = sitofp i32 %build_zext to double
919     * CHECK-DAG: %build_fptoui = fptoui float %build_uitofp to i32
920     * CHECK-DAG: %build_fptosi = fptosi double %build_sitofp to i64
921     * CHECK-DAG: %build_fptrunc = fptrunc double %build_sitofp to float
922     * CHECK-DAG: %build_fptrunc2 = fptrunc double %build_sitofp to float
923     * CHECK-DAG: %build_fpext = fpext float %build_fptrunc to double
924     * CHECK-DAG: %build_fpext2 = fpext float %build_fptrunc to double
925     * CHECK-DAG: %build_inttoptr = inttoptr i32 %P1 to i8*
926     * CHECK-DAG: %build_ptrtoint = ptrtoint i8* %build_inttoptr to i64
927     * CHECK-DAG: %build_ptrtoint2 = ptrtoint i8* %build_inttoptr to i64
928     * CHECK-DAG: %build_bitcast = bitcast i64 %build_ptrtoint to double
929     * CHECK-DAG: %build_bitcast2 = bitcast i64 %build_ptrtoint to double
930     * CHECK-DAG: %build_bitcast3 = bitcast i64 %build_ptrtoint to double
931     * CHECK-DAG: %build_bitcast4 = bitcast i64 %build_ptrtoint to double
932     * CHECK-DAG: %build_pointercast = bitcast i8* %build_inttoptr to i16*
933     *)
934    let inst28 = build_trunc p1 i8_type "build_trunc" atentry in
935    let inst29 = build_zext inst28 i32_type "build_zext" atentry in
936    let inst30 = build_sext inst29 i64_type "build_sext" atentry in
937    let inst31 = build_uitofp inst30 float_type "build_uitofp" atentry in
938    let inst32 = build_sitofp inst29 double_type "build_sitofp" atentry in
939    ignore(build_fptoui inst31 i32_type "build_fptoui" atentry);
940    ignore(build_fptosi inst32 i64_type "build_fptosi" atentry);
941    let inst35 = build_fptrunc inst32 float_type "build_fptrunc" atentry in
942    ignore(build_fpext inst35 double_type "build_fpext" atentry);
943    let inst37 = build_inttoptr p1 void_ptr "build_inttoptr" atentry in
944    let inst38 = build_ptrtoint inst37 i64_type "build_ptrtoint" atentry in
945    ignore(build_bitcast inst38 double_type "build_bitcast" atentry);
946    ignore(build_zext_or_bitcast inst38 double_type "build_bitcast2" atentry);
947    ignore(build_sext_or_bitcast inst38 double_type "build_bitcast3" atentry);
948    ignore(build_trunc_or_bitcast inst38 double_type "build_bitcast4" atentry);
949    ignore(build_pointercast inst37 (pointer_type i16_type) "build_pointercast" atentry);
950
951    ignore(build_zext_or_bitcast inst28 i32_type "build_zext2" atentry);
952    ignore(build_sext_or_bitcast inst29 i64_type "build_sext2" atentry);
953    ignore(build_trunc_or_bitcast p1 i8_type "build_trunc2" atentry);
954    ignore(build_pointercast inst37 i64_type "build_ptrtoint2" atentry);
955    ignore(build_intcast inst29 i64_type "build_sext3" atentry);
956    ignore(build_intcast p1 i8_type "build_trunc3" atentry);
957    ignore(build_fpcast inst35 double_type "build_fpext2" atentry);
958    ignore(build_fpcast inst32 float_type "build_fptrunc2" atentry);
959  end;
960
961  group "comparisons"; begin
962    (* CHECK: %build_icmp_ne = icmp ne i32 %P1, %P2
963     * CHECK: %build_icmp_sle = icmp sle i32 %P2, %P1
964     * CHECK: %build_fcmp_false = fcmp false float %F1, %F2
965     * CHECK: %build_fcmp_true = fcmp true float %F2, %F1
966     * CHECK: %build_is_null{{.*}}= icmp eq{{.*}}%X0,{{.*}}null
967     * CHECK: %build_is_not_null = icmp ne i8* %X1, null
968     * CHECK: %build_ptrdiff
969     *)
970    ignore (build_icmp Icmp.Ne    p1 p2 "build_icmp_ne" atentry);
971    ignore (build_icmp Icmp.Sle   p2 p1 "build_icmp_sle" atentry);
972    ignore (build_fcmp Fcmp.False f1 f2 "build_fcmp_false" atentry);
973    ignore (build_fcmp Fcmp.True  f2 f1 "build_fcmp_true" atentry);
974    let g0 = declare_global (pointer_type i8_type) "g0" m in
975    let g1 = declare_global (pointer_type i8_type) "g1" m in
976    let p0 = build_load g0 "X0" atentry in
977    let p1 = build_load g1 "X1" atentry in
978    ignore (build_is_null p0 "build_is_null" atentry);
979    ignore (build_is_not_null p1 "build_is_not_null" atentry);
980    ignore (build_ptrdiff p1 p0 "build_ptrdiff" atentry);
981  end;
982
983  group "miscellaneous"; begin
984    (* CHECK: %build_call = tail call cc63 i32 @{{.*}}(i32 signext %P2, i32 %P1)
985     * CHECK: %build_select = select i1 %build_icmp, i32 %P1, i32 %P2
986     * CHECK: %build_va_arg = va_arg i8** null, i32
987     * CHECK: %build_extractelement = extractelement <4 x i32> %Vec1, i32 %P2
988     * CHECK: %build_insertelement = insertelement <4 x i32> %Vec1, i32 %P1, i32 %P2
989     * CHECK: %build_shufflevector = shufflevector <4 x i32> %Vec1, <4 x i32> %Vec2, <4 x i32> <i32 1, i32 1, i32 0, i32 0>
990     * CHECK: %build_insertvalue0 = insertvalue{{.*}}%bl, i32 1, 0
991     * CHECK: %build_extractvalue = extractvalue{{.*}}%build_insertvalue1, 1
992     *)
993    let ci = build_call fn [| p2; p1 |] "build_call" atentry in
994    insist (CallConv.c = instruction_call_conv ci);
995    set_instruction_call_conv 63 ci;
996    insist (63 = instruction_call_conv ci);
997    insist (not (is_tail_call ci));
998    set_tail_call true ci;
999    insist (is_tail_call ci);
1000    add_instruction_param_attr ci 1 Attribute.Sext;
1001    add_instruction_param_attr ci 2 Attribute.Noalias;
1002    remove_instruction_param_attr ci 2 Attribute.Noalias;
1003
1004    let inst46 = build_icmp Icmp.Eq p1 p2 "build_icmp" atentry in
1005    ignore (build_select inst46 p1 p2 "build_select" atentry);
1006    ignore (build_va_arg
1007      (const_null (pointer_type (pointer_type i8_type)))
1008      i32_type "build_va_arg" atentry);
1009
1010    (* Set up some vector vregs. *)
1011    let one  = const_int i32_type 1 in
1012    let zero = const_int i32_type 0 in
1013    let t1 = const_vector [| one; zero; one; zero |] in
1014    let t2 = const_vector [| zero; one; zero; one |] in
1015    let t3 = const_vector [| one; one; zero; zero |] in
1016    let vec1 = build_insertelement t1 p1 p2 "Vec1" atentry in
1017    let vec2 = build_insertelement t2 p1 p2 "Vec2" atentry in
1018    let sty = struct_type context [| i32_type; i8_type |] in
1019
1020    ignore (build_extractelement vec1 p2 "build_extractelement" atentry);
1021    ignore (build_insertelement vec1 p1 p2 "build_insertelement" atentry);
1022    ignore (build_shufflevector vec1 vec2 t3 "build_shufflevector" atentry);
1023
1024    let p = build_alloca sty "ba" atentry in
1025    let agg = build_load p "bl" atentry in
1026    let agg0 = build_insertvalue agg (const_int i32_type 1) 0
1027                 "build_insertvalue0" atentry in
1028    let agg1 = build_insertvalue agg0 (const_int i8_type 2) 1
1029                 "build_insertvalue1" atentry in
1030    ignore (build_extractvalue agg1 1 "build_extractvalue" atentry)
1031  end;
1032
1033  group "metadata"; begin
1034    (* CHECK: %metadata = add i32 %P1, %P2, !test !1
1035     * !1 is metadata emitted at EOF.
1036     *)
1037    let i = build_add p1 p2 "metadata" atentry in
1038    insist ((has_metadata i) = false);
1039
1040    let m1 = const_int i32_type 1 in
1041    let m2 = mdstring context "metadata test" in
1042    let md = mdnode context [| m1; m2 |] in
1043
1044    let kind = mdkind_id context "test" in
1045    set_metadata i kind md;
1046
1047    insist ((has_metadata i) = true);
1048    insist ((metadata i kind) = Some md);
1049
1050    clear_metadata i kind;
1051
1052    insist ((has_metadata i) = false);
1053    insist ((metadata i kind) = None);
1054
1055    set_metadata i kind md
1056  end;
1057
1058  group "named metadata"; begin
1059    (* !llvm.module.flags is emitted at EOF. *)
1060    let n1 = const_int i32_type 1 in
1061    let n2 = mdstring context "Debug Info Version" in
1062    let md = mdnode context [| n1; n2; n1 |] in
1063    add_named_metadata_operand m "llvm.module.flags" md;
1064
1065    insist ((get_named_metadata m "llvm.module.flags") = [| md |])
1066  end;
1067
1068  group "dbg"; begin
1069    (* CHECK: %dbg = add i32 %P1, %P2, !dbg !2
1070     * !2 is metadata emitted at EOF.
1071     *)
1072    insist ((current_debug_location atentry) = None);
1073
1074    let m_line = const_int i32_type 2 in
1075    let m_col = const_int i32_type 3 in
1076    let m_scope = mdnode context [| |] in
1077    let m_inlined = mdnode context [| |] in
1078    let md = mdnode context [| m_line; m_col; m_scope; m_inlined |] in
1079    set_current_debug_location atentry md;
1080
1081    insist ((current_debug_location atentry) = Some md);
1082
1083    let i = build_add p1 p2 "dbg" atentry in
1084    insist ((has_metadata i) = true);
1085
1086    clear_current_debug_location atentry
1087  end;
1088
1089  group "ret"; begin
1090    (* CHECK: ret{{.*}}P1
1091     *)
1092    let ret = build_ret p1 atentry in
1093    position_before ret atentry
1094  end;
1095
1096  (* see test/Feature/exception.ll *)
1097  let bblpad = append_block context "Bblpad" fn in
1098  let rt = struct_type context [| pointer_type i8_type; i32_type |] in
1099  let ft = var_arg_function_type i32_type  [||] in
1100  let personality = declare_function "__gxx_personality_v0" ft m in
1101  let ztic = declare_global (pointer_type i8_type) "_ZTIc" m in
1102  let ztid = declare_global (pointer_type i8_type) "_ZTId" m in
1103  let ztipkc = declare_global (pointer_type i8_type) "_ZTIPKc" m in
1104  begin
1105      set_global_constant true ztic;
1106      set_global_constant true ztid;
1107      set_global_constant true ztipkc;
1108      let lp = build_landingpad rt personality 0 "lpad"
1109       (builder_at_end context bblpad) in begin
1110           set_cleanup lp true;
1111           add_clause lp ztic;
1112           insist((pointer_type (pointer_type i8_type)) = type_of ztid);
1113           let ety = pointer_type (pointer_type i8_type) in
1114           add_clause lp (const_array ety [| ztipkc; ztid |]);
1115           ignore (build_resume lp (builder_at_end context bblpad));
1116      end;
1117      (* CHECK: landingpad{{.*}}personality{{.*}}__gxx_personality_v0
1118       * CHECK: cleanup
1119       * CHECK: catch{{.*}}i8**{{.*}}@_ZTIc
1120       * CHECK: filter{{.*}}@_ZTIPKc{{.*}}@_ZTId
1121       * CHECK: resume
1122       * *)
1123  end;
1124
1125  group "br"; begin
1126    (* CHECK: br{{.*}}Bb02
1127     *)
1128    let bb02 = append_block context "Bb02" fn in
1129    let b = builder_at_end context bb02 in
1130    ignore (build_br bb02 b)
1131  end;
1132  
1133  group "cond_br"; begin
1134    (* CHECK: br{{.*}}build_br{{.*}}Bb03{{.*}}Bb00
1135     *)
1136    let bb03 = append_block context "Bb03" fn in
1137    let b = builder_at_end context bb03 in
1138    let cond = build_trunc p1 i1_type "build_br" b in
1139    ignore (build_cond_br cond bb03 bb00 b)
1140  end;
1141  
1142  group "switch"; begin
1143    (* CHECK: switch{{.*}}P1{{.*}}SwiBlock3
1144     * CHECK: 2,{{.*}}SwiBlock2
1145     *)
1146    let bb1 = append_block context "SwiBlock1" fn in
1147    let bb2 = append_block context "SwiBlock2" fn in
1148    ignore (build_unreachable (builder_at_end context bb2));
1149    let bb3 = append_block context "SwiBlock3" fn in
1150    ignore (build_unreachable (builder_at_end context bb3));
1151    let si = build_switch p1 bb3 1 (builder_at_end context bb1) in begin
1152        ignore (add_case si (const_int i32_type 2) bb2);
1153        insist (switch_default_dest si = bb3);
1154    end;
1155  end;
1156
1157  group "malloc/free"; begin
1158      (* CHECK: call{{.*}}@malloc(i32 ptrtoint
1159       * CHECK: call{{.*}}@free(i8*
1160       * CHECK: call{{.*}}@malloc(i32 %
1161       *)
1162      let bb1 = append_block context "MallocBlock1" fn in
1163      let m1 = (build_malloc (pointer_type i32_type) "m1"
1164      (builder_at_end context bb1)) in
1165      ignore (build_free m1 (builder_at_end context bb1));
1166      ignore (build_array_malloc i32_type p1 "m2" (builder_at_end context bb1));
1167      ignore (build_unreachable (builder_at_end context bb1));
1168  end;
1169
1170  group "indirectbr"; begin
1171    (* CHECK: indirectbr i8* blockaddress(@X7, %IBRBlock2), [label %IBRBlock2, label %IBRBlock3]
1172     *)
1173    let bb1 = append_block context "IBRBlock1" fn in
1174
1175    let bb2 = append_block context "IBRBlock2" fn in
1176    ignore (build_unreachable (builder_at_end context bb2));
1177
1178    let bb3 = append_block context "IBRBlock3" fn in
1179    ignore (build_unreachable (builder_at_end context bb3));
1180
1181    let addr = block_address fn bb2 in
1182    let ibr = build_indirect_br addr 2 (builder_at_end context bb1) in
1183    ignore (add_destination ibr bb2);
1184    ignore (add_destination ibr bb3)
1185  end;
1186  
1187  group "invoke"; begin
1188    (* CHECK: build_invoke{{.*}}invoke{{.*}}P1{{.*}}P2
1189     * CHECK: to{{.*}}Bb04{{.*}}unwind{{.*}}Bblpad
1190     *)
1191    let bb04 = append_block context "Bb04" fn in
1192    let b = builder_at_end context bb04 in
1193    ignore (build_invoke fn [| p1; p2 |] bb04 bblpad "build_invoke" b)
1194  end;
1195  
1196  group "unreachable"; begin
1197    (* CHECK: unreachable
1198     *)
1199    let bb06 = append_block context "Bb06" fn in
1200    let b = builder_at_end context bb06 in
1201    ignore (build_unreachable b)
1202  end;
1203  
1204  group "arithmetic"; begin
1205    let bb07 = append_block context "Bb07" fn in
1206    let b = builder_at_end context bb07 in
1207    
1208    (* CHECK: %build_add = add i32 %P1, %P2
1209     * CHECK: %build_nsw_add = add nsw i32 %P1, %P2
1210     * CHECK: %build_nuw_add = add nuw i32 %P1, %P2
1211     * CHECK: %build_fadd = fadd float %F1, %F2
1212     * CHECK: %build_sub = sub i32 %P1, %P2
1213     * CHECK: %build_nsw_sub = sub nsw i32 %P1, %P2
1214     * CHECK: %build_nuw_sub = sub nuw i32 %P1, %P2
1215     * CHECK: %build_fsub = fsub float %F1, %F2
1216     * CHECK: %build_mul = mul i32 %P1, %P2
1217     * CHECK: %build_nsw_mul = mul nsw i32 %P1, %P2
1218     * CHECK: %build_nuw_mul = mul nuw i32 %P1, %P2
1219     * CHECK: %build_fmul = fmul float %F1, %F2
1220     * CHECK: %build_udiv = udiv i32 %P1, %P2
1221     * CHECK: %build_sdiv = sdiv i32 %P1, %P2
1222     * CHECK: %build_exact_sdiv = sdiv exact i32 %P1, %P2
1223     * CHECK: %build_fdiv = fdiv float %F1, %F2
1224     * CHECK: %build_urem = urem i32 %P1, %P2
1225     * CHECK: %build_srem = srem i32 %P1, %P2
1226     * CHECK: %build_frem = frem float %F1, %F2
1227     * CHECK: %build_shl = shl i32 %P1, %P2
1228     * CHECK: %build_lshl = lshr i32 %P1, %P2
1229     * CHECK: %build_ashl = ashr i32 %P1, %P2
1230     * CHECK: %build_and = and i32 %P1, %P2
1231     * CHECK: %build_or = or i32 %P1, %P2
1232     * CHECK: %build_xor = xor i32 %P1, %P2
1233     * CHECK: %build_neg = sub i32 0, %P1
1234     * CHECK: %build_nsw_neg = sub nsw i32 0, %P1
1235     * CHECK: %build_nuw_neg = sub nuw i32 0, %P1
1236     * CHECK: %build_fneg = fsub float {{.*}}0{{.*}}, %F1
1237     * CHECK: %build_not = xor i32 %P1, -1
1238     *)
1239    ignore (build_add p1 p2 "build_add" b);
1240    ignore (build_nsw_add p1 p2 "build_nsw_add" b);
1241    ignore (build_nuw_add p1 p2 "build_nuw_add" b);
1242    ignore (build_fadd f1 f2 "build_fadd" b);
1243    ignore (build_sub p1 p2 "build_sub" b);
1244    ignore (build_nsw_sub p1 p2 "build_nsw_sub" b);
1245    ignore (build_nuw_sub p1 p2 "build_nuw_sub" b);
1246    ignore (build_fsub f1 f2 "build_fsub" b);
1247    ignore (build_mul p1 p2 "build_mul" b);
1248    ignore (build_nsw_mul p1 p2 "build_nsw_mul" b);
1249    ignore (build_nuw_mul p1 p2 "build_nuw_mul" b);
1250    ignore (build_fmul f1 f2 "build_fmul" b);
1251    ignore (build_udiv p1 p2 "build_udiv" b);
1252    ignore (build_sdiv p1 p2 "build_sdiv" b);
1253    ignore (build_exact_sdiv p1 p2 "build_exact_sdiv" b);
1254    ignore (build_fdiv f1 f2 "build_fdiv" b);
1255    ignore (build_urem p1 p2 "build_urem" b);
1256    ignore (build_srem p1 p2 "build_srem" b);
1257    ignore (build_frem f1 f2 "build_frem" b);
1258    ignore (build_shl p1 p2 "build_shl" b);
1259    ignore (build_lshr p1 p2 "build_lshl" b);
1260    ignore (build_ashr p1 p2 "build_ashl" b);
1261    ignore (build_and p1 p2 "build_and" b);
1262    ignore (build_or p1 p2 "build_or" b);
1263    ignore (build_xor p1 p2 "build_xor" b);
1264    ignore (build_neg p1 "build_neg" b);
1265    ignore (build_nsw_neg p1 "build_nsw_neg" b);
1266    ignore (build_nuw_neg p1 "build_nuw_neg" b);
1267    ignore (build_fneg f1 "build_fneg" b);
1268    ignore (build_not p1 "build_not" b);
1269    ignore (build_unreachable b)
1270  end;
1271  
1272  group "memory"; begin
1273    let bb08 = append_block context "Bb08" fn in
1274    let b = builder_at_end context bb08 in
1275
1276    (* CHECK: %build_alloca = alloca i32
1277     * CHECK: %build_array_alloca = alloca i32, i32 %P2
1278     * CHECK: %build_load = load volatile i32* %build_array_alloca, align 4
1279     * CHECK: store volatile i32 %P2, i32* %build_alloca, align 4
1280     * CHECK: %build_gep = getelementptr i32* %build_array_alloca, i32 %P2
1281     * CHECK: %build_in_bounds_gep = getelementptr inbounds i32* %build_array_alloca, i32 %P2
1282     * CHECK: %build_struct_gep = getelementptr inbounds{{.*}}%build_alloca2, i32 0, i32 1
1283     * CHECK: %build_atomicrmw = atomicrmw xchg i8* %p, i8 42 seq_cst
1284     *)
1285    let alloca = build_alloca i32_type "build_alloca" b in
1286    let array_alloca = build_array_alloca i32_type p2 "build_array_alloca" b in
1287
1288    let load = build_load array_alloca "build_load" b in
1289    ignore(set_alignment 4 load);
1290    ignore(set_volatile true load);
1291    insist(true = is_volatile load);
1292    insist(4 = alignment load);
1293
1294    let store = build_store p2 alloca b in
1295    ignore(set_volatile true store);
1296    ignore(set_alignment 4 store);
1297    insist(true = is_volatile store);
1298    insist(4 = alignment store);
1299    ignore(build_gep array_alloca [| p2 |] "build_gep" b);
1300    ignore(build_in_bounds_gep array_alloca [| p2 |] "build_in_bounds_gep" b);
1301
1302    let sty = struct_type context [| i32_type; i8_type |] in
1303    let alloca2 = build_alloca sty "build_alloca2" b in
1304    ignore(build_struct_gep alloca2 1 "build_struct_gep" b);
1305
1306    let p = build_alloca i8_type "p" b in
1307    ignore(build_atomicrmw AtomicRMWBinOp.Xchg p (const_int i8_type 42)
1308              AtomicOrdering.SequentiallyConsistent false "build_atomicrmw"
1309              b);
1310
1311    ignore(build_unreachable b)
1312  end;
1313
1314  group "string"; begin
1315    let bb09 = append_block context "Bb09" fn in
1316    let b = builder_at_end context bb09 in
1317    let p = build_alloca (pointer_type i8_type) "p" b in
1318    (* build_global_string is emitted above.
1319     * CHECK: store{{.*}}build_global_string1{{.*}}p
1320     * *)
1321    ignore (build_global_string "stringval" "build_global_string" b);
1322    let g = build_global_stringptr "stringval" "build_global_string1" b in
1323    ignore (build_store g p b);
1324    ignore(build_unreachable b);
1325  end;
1326
1327  group "phi"; begin
1328    (* CHECK: PhiNode{{.*}}P1{{.*}}PhiBlock1{{.*}}P2{{.*}}PhiBlock2
1329     *)
1330    let b1 = append_block context "PhiBlock1" fn in
1331    let b2 = append_block context "PhiBlock2" fn in
1332    
1333    let jb = append_block context "PhiJoinBlock" fn in
1334    ignore (build_br jb (builder_at_end context b1));
1335    ignore (build_br jb (builder_at_end context b2));
1336    let at_jb = builder_at_end context jb in
1337    
1338    let phi = build_phi [(p1, b1)] "PhiNode" at_jb in
1339    insist ([(p1, b1)] = incoming phi);
1340    
1341    add_incoming (p2, b2) phi;
1342    insist ([(p1, b1); (p2, b2)] = incoming phi);
1343    
1344    ignore (build_unreachable at_jb);
1345  end
1346
1347(* End-of-file checks for things like metdata and attributes.
1348 * CHECK: attributes #0 = {{.*}}uwtable{{.*}}
1349 * CHECK: !llvm.module.flags = !{!0}
1350 * CHECK: !0 = metadata !{i32 1, metadata !"Debug Info Version", i32 1}
1351 * CHECK: !1 = metadata !{i32 1, metadata !"metadata test"}
1352 * CHECK: !2 = metadata !{i32 2, i32 3, metadata !3, metadata !3}
1353 *)
1354
1355(*===-- Pass Managers -----------------------------------------------------===*)
1356
1357let test_pass_manager () =
1358  let (++) x f = ignore (f x); x in
1359
1360  begin group "module pass manager";
1361    ignore (PassManager.create ()
1362             ++ PassManager.run_module m
1363             ++ PassManager.dispose)
1364  end;
1365  
1366  begin group "function pass manager";
1367    let fty = function_type void_type [| |] in
1368    let fn = define_function "FunctionPassManager" fty m in
1369    ignore (build_ret_void (builder_at_end context (entry_block fn)));
1370    
1371    ignore (PassManager.create_function m
1372             ++ PassManager.initialize
1373             ++ PassManager.run_function fn
1374             ++ PassManager.finalize
1375             ++ PassManager.dispose)
1376  end
1377
1378
1379(*===-- Memory Buffer -----------------------------------------------------===*)
1380
1381let test_memory_buffer () =
1382  group "memory buffer";
1383  let buf = MemoryBuffer.of_string "foobar" in
1384  insist ((MemoryBuffer.as_string buf) = "foobar")
1385
1386
1387(*===-- Writer ------------------------------------------------------------===*)
1388
1389let test_writer () =
1390  group "valid";
1391  insist (match Llvm_analysis.verify_module m with
1392          | None -> true
1393          | Some msg -> prerr_string msg; false);
1394
1395  group "writer";
1396  insist (write_bitcode_file m filename);
1397  
1398  dispose_module m
1399
1400
1401(*===-- Driver ------------------------------------------------------------===*)
1402
1403let _ =
1404  suite "conversion"       test_conversion;
1405  suite "target"           test_target;
1406  suite "constants"        test_constants;
1407  suite "global values"    test_global_values;
1408  suite "global variables" test_global_variables;
1409  suite "uses"             test_uses;
1410  suite "users"            test_users;
1411  suite "aliases"          test_aliases;
1412  suite "functions"        test_functions;
1413  suite "params"           test_params;
1414  suite "basic blocks"     test_basic_blocks;
1415  suite "instructions"     test_instructions;
1416  suite "builder"          test_builder;
1417  suite "pass manager"     test_pass_manager;
1418  suite "memory buffer"    test_memory_buffer;
1419  suite "writer"           test_writer; (* Keep this last; it disposes m. *)
1420  exit !exit_status
1421