test/Bindings/Ocaml/vmcore.ml

   1 (* RUN: %ocamlopt -warn-error A llvm.cmxa llvm_analysis.cmxa llvm_bitwriter.cmxa %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 ocamlopt 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 suite_name = ref ""
  17 let group_name = ref ""
  18 let case_num = ref 0
  19 let print_checkpoints = false
  20 let context = global_context ()
  21 let i1_type = Llvm.i1_type context
  22 let i8_type = Llvm.i8_type context
  23 let i16_type = Llvm.i16_type context
  24 let i32_type = Llvm.i32_type context
  25 let i64_type = Llvm.i64_type context
  26 let void_type = Llvm.void_type context
  27 let float_type = Llvm.float_type context
  28 let double_type = Llvm.double_type context
  29 let fp128_type = Llvm.fp128_type context
  30
  31 let group name =
  32   group_name := !suite_name ^ "/" ^ name;
  33   case_num := 0;
  34   if print_checkpoints then
  35     prerr_endline ("  " ^ name ^ "...")
  36
  37 let insist cond =
  38   incr case_num;
  39   if not cond then
  40     exit_status := 10;
  41   match print_checkpoints, cond with
  42   | false, true -> ()
  43   | false, false ->
  44       prerr_endline ("FAILED: " ^ !suite_name ^ "/" ^ !group_name ^ " #" ^ (string_of_int !case_num))
  45   | true, true ->
  46       prerr_endline ("    " ^ (string_of_int !case_num))
  47   | true, false ->
  48       prerr_endline ("    " ^ (string_of_int !case_num) ^ " FAIL")
  49
  50 let suite name f =
  51   suite_name := name;
  52   if print_checkpoints then
  53     prerr_endline (name ^ ":");
  54   f ()
  55
  56
  57 (*===-- Fixture -----------------------------------------------------------===*)
  58
  59 let filename = Sys.argv.(1)
  60 let m = create_module context filename
  61 let mp = ModuleProvider.create m
  62
  63
  64 (*===-- Target ------------------------------------------------------------===*)
  65
  66 let test_target () =
  67   begin group "triple";
  68     (* RUN: grep "i686-apple-darwin8" < %t.ll
  69      *)
  70     let trip = "i686-apple-darwin8" in
  71     set_target_triple trip m;
  72     insist (trip = target_triple m)
  73   end;
  74
  75   begin group "layout";
  76     (* RUN: grep "bogus" < %t.ll
  77      *)
  78     let layout = "bogus" in
  79     set_data_layout layout m;
  80     insist (layout = data_layout m)
  81   end
  82
  83 (*===-- Types -------------------------------------------------------------===*)
  84
  85 let test_types () =
  86   (* RUN: grep {void_type.*void} < %t.ll
  87    *)
  88   group "void";
  89   insist (define_type_name "void_type" void_type m);
  90   insist (TypeKind.Void == classify_type void_type);
  91
  92   (* RUN: grep {i1_type.*i1} < %t.ll
  93    *)
  94   group "i1";
  95   insist (define_type_name "i1_type" i1_type m);
  96   insist (TypeKind.Integer == classify_type i1_type);
  97
  98   (* RUN: grep {i32_type.*i32} < %t.ll
  99    *)
 100   group "i32";
 101   insist (define_type_name "i32_type" i32_type m);
 102
 103   (* RUN: grep {i42_type.*i42} < %t.ll
 104    *)
 105   group "i42";
 106   let ty = integer_type context 42 in
 107   insist (define_type_name "i42_type" ty m);
 108
 109   (* RUN: grep {float_type.*float} < %t.ll
 110    *)
 111   group "float";
 112   insist (define_type_name "float_type" float_type m);
 113   insist (TypeKind.Float == classify_type float_type);
 114
 115   (* RUN: grep {double_type.*double} < %t.ll
 116    *)
 117   group "double";
 118   insist (define_type_name "double_type" double_type m);
 119   insist (TypeKind.Double == classify_type double_type);
 120
 121   (* RUN: grep {function_type.*i32.*i1, double} < %t.ll
 122    *)
 123   group "function";
 124   let ty = function_type i32_type [| i1_type; double_type |] in
 125   insist (define_type_name "function_type" ty m);
 126   insist (TypeKind.Function = classify_type ty);
 127   insist (not (is_var_arg ty));
 128   insist (i32_type == return_type ty);
 129   insist (double_type == (param_types ty).(1));
 130
 131   (* RUN: grep {var_arg_type.*\.\.\.} < %t.ll
 132    *)
 133   group "var arg function";
 134   let ty = var_arg_function_type void_type [| i32_type |] in
 135   insist (define_type_name "var_arg_type" ty m);
 136   insist (is_var_arg ty);
 137
 138   (* RUN: grep {array_type.*\\\[7 x i8\\\]} < %t.ll
 139    *)
 140   group "array";
 141   let ty = array_type i8_type 7 in
 142   insist (define_type_name "array_type" ty m);
 143   insist (7 = array_length ty);
 144   insist (i8_type == element_type ty);
 145   insist (TypeKind.Array == classify_type ty);
 146
 147   begin group "pointer";
 148     (* RUN: grep {pointer_type.*float\*} < %t.ll
 149      *)
 150     let ty = pointer_type float_type in
 151     insist (define_type_name "pointer_type" ty m);
 152     insist (float_type == element_type ty);
 153     insist (0 == address_space ty);
 154     insist (TypeKind.Pointer == classify_type ty)
 155   end;
 156
 157   begin group "qualified_pointer";
 158     (* RUN: grep {qualified_pointer_type.*i8.*3.*\*} < %t.ll
 159      *)
 160     let ty = qualified_pointer_type i8_type 3 in
 161     insist (define_type_name "qualified_pointer_type" ty m);
 162     insist (i8_type == element_type ty);
 163     insist (3 == address_space ty)
 164   end;
 165
 166   (* RUN: grep {vector_type.*\<4 x i16\>} < %t.ll
 167    *)
 168   group "vector";
 169   let ty = vector_type i16_type 4 in
 170   insist (define_type_name "vector_type" ty m);
 171   insist (i16_type == element_type ty);
 172   insist (4 = vector_size ty);
 173
 174   (* RUN: grep {opaque_type.*opaque} < %t.ll
 175    *)
 176   group "opaque";
 177   let ty = opaque_type context in
 178   insist (define_type_name "opaque_type" ty m);
 179   insist (ty == ty);
 180   insist (ty <> opaque_type context);
 181
 182   (* RUN: grep -v {delete_type} < %t.ll
 183    *)
 184   group "delete";
 185   let ty = opaque_type context in
 186   insist (define_type_name "delete_type" ty m);
 187   delete_type_name "delete_type" m;
 188
 189   (* RUN: grep -v {recursive_type.*recursive_type} < %t.ll
 190    *)
 191   group "recursive";
 192   let ty = opaque_type context in
 193   let th = handle_to_type ty in
 194   refine_type ty (pointer_type ty);
 195   let ty = type_of_handle th in
 196   insist (define_type_name "recursive_type" ty m);
 197   insist (ty == element_type ty)
 198
 199
 200 (*===-- Constants ---------------------------------------------------------===*)
 201
 202 let test_constants () =
 203   (* RUN: grep {const_int.*i32.*-1} < %t.ll
 204    *)
 205   group "int";
 206   let c = const_int i32_type (-1) in
 207   ignore (define_global "const_int" c m);
 208   insist (i32_type = type_of c);
 209   insist (is_constant c);
 210
 211   (* RUN: grep {const_sext_int.*i64.*-1} < %t.ll
 212    *)
 213   group "sext int";
 214   let c = const_int i64_type (-1) in
 215   ignore (define_global "const_sext_int" c m);
 216   insist (i64_type = type_of c);
 217
 218   (* RUN: grep {const_zext_int64.*i64.*4294967295} < %t.ll
 219    *)
 220   group "zext int64";
 221   let c = const_of_int64 i64_type (Int64.of_string "4294967295") false in
 222   ignore (define_global "const_zext_int64" c m);
 223   insist (i64_type = type_of c);
 224
 225   (* RUN: grep {const_int_string.*i32.*-1} < %t.ll
 226    *)
 227   group "int string";
 228   let c = const_int_of_string i32_type "-1" 10 in
 229   ignore (define_global "const_int_string" c m);
 230   insist (i32_type = type_of c);
 231
 232   (* RUN: grep {const_string.*"cruel\\\\00world"} < %t.ll
 233    *)
 234   group "string";
 235   let c = const_string context "cruel\000world" in
 236   ignore (define_global "const_string" c m);
 237   insist ((array_type i8_type 11) = type_of c);
 238
 239   (* RUN: grep {const_stringz.*"hi\\\\00again\\\\00"} < %t.ll
 240    *)
 241   group "stringz";
 242   let c = const_stringz context "hi\000again" in
 243   ignore (define_global "const_stringz" c m);
 244   insist ((array_type i8_type 9) = type_of c);
 245
 246   (* RUN: grep {const_single.*2.75} < %t.ll
 247    * RUN: grep {const_double.*3.1459} < %t.ll
 248    * RUN: grep {const_double_string.*1.25} < %t.ll
 249    *)
 250   begin group "real";
 251     let cs = const_float float_type 2.75 in
 252     ignore (define_global "const_single" cs m);
 253     insist (float_type = type_of cs);
 254
 255     let cd = const_float double_type 3.1459 in
 256     ignore (define_global "const_double" cd m);
 257     insist (double_type = type_of cd);
 258
 259     let cd = const_float_of_string double_type "1.25" in
 260     ignore (define_global "const_double_string" cd m);
 261     insist (double_type = type_of cd)
 262   end;
 263
 264   let one = const_int i16_type 1 in
 265   let two = const_int i16_type 2 in
 266   let three = const_int i32_type 3 in
 267   let four = const_int i32_type 4 in
 268
 269   (* RUN: grep {const_array.*\\\[i32 3, i32 4\\\]} < %t.ll
 270    *)
 271   group "array";
 272   let c = const_array i32_type [| three; four |] in
 273   ignore (define_global "const_array" c m);
 274   insist ((array_type i32_type 2) = (type_of c));
 275
 276   (* RUN: grep {const_vector.*<i16 1, i16 2.*>} < %t.ll
 277    *)
 278   group "vector";
 279   let c = const_vector [| one; two; one; two;
 280                           one; two; one; two |] in
 281   ignore (define_global "const_vector" c m);
 282   insist ((vector_type i16_type 8) = (type_of c));
 283
 284   (* RUN: grep {const_structure.*.i16 1, i16 2, i32 3, i32 4} < %t.ll
 285    *)
 286   group "structure";
 287   let c = const_struct context [| one; two; three; four |] in
 288   ignore (define_global "const_structure" c m);
 289   insist ((struct_type context [| i16_type; i16_type; i32_type; i32_type |])
 290         = (type_of c));
 291
 292   (* RUN: grep {const_null.*zeroinit} < %t.ll
 293    *)
 294   group "null";
 295   let c = const_null (packed_struct_type context [| i1_type; i8_type; i64_type;
 296                                                     double_type |]) in
 297   ignore (define_global "const_null" c m);
 298
 299   (* RUN: grep {const_all_ones.*-1} < %t.ll
 300    *)
 301   group "all ones";
 302   let c = const_all_ones i64_type in
 303   ignore (define_global "const_all_ones" c m);
 304
 305   (* RUN: grep {const_undef.*undef} < %t.ll
 306    *)
 307   group "undef";
 308   let c = undef i1_type in
 309   ignore (define_global "const_undef" c m);
 310   insist (i1_type = type_of c);
 311   insist (is_undef c);
 312
 313   group "constant arithmetic";
 314   (* RUN: grep {@const_neg = global i64 sub} < %t.ll
 315    * RUN: grep {@const_not = global i64 xor } < %t.ll
 316    * RUN: grep {@const_add = global i64 add } < %t.ll
 317    * RUN: grep {@const_sub = global i64 sub } < %t.ll
 318    * RUN: grep {@const_mul = global i64 mul } < %t.ll
 319    * RUN: grep {@const_udiv = global i64 udiv } < %t.ll
 320    * RUN: grep {@const_sdiv = global i64 sdiv } < %t.ll
 321    * RUN: grep {@const_fdiv = global double fdiv } < %t.ll
 322    * RUN: grep {@const_urem = global i64 urem } < %t.ll
 323    * RUN: grep {@const_srem = global i64 srem } < %t.ll
 324    * RUN: grep {@const_frem = global double frem } < %t.ll
 325    * RUN: grep {@const_and = global i64 and } < %t.ll
 326    * RUN: grep {@const_or = global i64 or } < %t.ll
 327    * RUN: grep {@const_xor = global i64 xor } < %t.ll
 328    * RUN: grep {@const_icmp = global i1 icmp sle } < %t.ll
 329    * RUN: grep {@const_fcmp = global i1 fcmp ole } < %t.ll
 330    *)
 331   let void_ptr = pointer_type i8_type in
 332   let five = const_int i64_type 5 in
 333   let ffive = const_uitofp five double_type in
 334   let foldbomb_gv = define_global "FoldBomb" (const_null i8_type) m in
 335   let foldbomb = const_ptrtoint foldbomb_gv i64_type in
 336   let ffoldbomb = const_uitofp foldbomb double_type in
 337   ignore (define_global "const_neg" (const_neg foldbomb) m);
 338   ignore (define_global "const_not" (const_not foldbomb) m);
 339   ignore (define_global "const_add" (const_add foldbomb five) m);
 340   ignore (define_global "const_sub" (const_sub foldbomb five) m);
 341   ignore (define_global "const_mul" (const_mul foldbomb five) m);
 342   ignore (define_global "const_udiv" (const_udiv foldbomb five) m);
 343   ignore (define_global "const_sdiv" (const_sdiv foldbomb five) m);
 344   ignore (define_global "const_fdiv" (const_fdiv ffoldbomb ffive) m);
 345   ignore (define_global "const_urem" (const_urem foldbomb five) m);
 346   ignore (define_global "const_srem" (const_srem foldbomb five) m);
 347   ignore (define_global "const_frem" (const_frem ffoldbomb ffive) m);
 348   ignore (define_global "const_and" (const_and foldbomb five) m);
 349   ignore (define_global "const_or" (const_or foldbomb five) m);
 350   ignore (define_global "const_xor" (const_xor foldbomb five) m);
 351   ignore (define_global "const_icmp" (const_icmp Icmp.Sle foldbomb five) m);
 352   ignore (define_global "const_fcmp" (const_fcmp Fcmp.Ole ffoldbomb ffive) m);
 353
 354   group "constant casts";
 355   (* RUN: grep {const_trunc.*trunc} < %t.ll
 356    * RUN: grep {const_sext.*sext} < %t.ll
 357    * RUN: grep {const_zext.*zext} < %t.ll
 358    * RUN: grep {const_fptrunc.*fptrunc} < %t.ll
 359    * RUN: grep {const_fpext.*fpext} < %t.ll
 360    * RUN: grep {const_uitofp.*uitofp} < %t.ll
 361    * RUN: grep {const_sitofp.*sitofp} < %t.ll
 362    * RUN: grep {const_fptoui.*fptoui} < %t.ll
 363    * RUN: grep {const_fptosi.*fptosi} < %t.ll
 364    * RUN: grep {const_ptrtoint.*ptrtoint} < %t.ll
 365    * RUN: grep {const_inttoptr.*inttoptr} < %t.ll
 366    * RUN: grep {const_bitcast.*bitcast} < %t.ll
 367    *)
 368   let i128_type = integer_type context 128 in
 369   ignore (define_global "const_trunc" (const_trunc (const_add foldbomb five)
 370                                                i8_type) m);
 371   ignore (define_global "const_sext" (const_sext foldbomb i128_type) m);
 372   ignore (define_global "const_zext" (const_zext foldbomb i128_type) m);
 373   ignore (define_global "const_fptrunc" (const_fptrunc ffoldbomb float_type) m);
 374   ignore (define_global "const_fpext" (const_fpext ffoldbomb fp128_type) m);
 375   ignore (define_global "const_uitofp" (const_uitofp foldbomb double_type) m);
 376   ignore (define_global "const_sitofp" (const_sitofp foldbomb double_type) m);
 377   ignore (define_global "const_fptoui" (const_fptoui ffoldbomb i32_type) m);
 378   ignore (define_global "const_fptosi" (const_fptosi ffoldbomb i32_type) m);
 379   ignore (define_global "const_ptrtoint" (const_ptrtoint
 380     (const_gep (const_null (pointer_type i8_type))
 381                [| const_int i32_type 1 |])
 382     i32_type) m);
 383   ignore (define_global "const_inttoptr" (const_inttoptr (const_add foldbomb five)
 384                                                   void_ptr) m);
 385   ignore (define_global "const_bitcast" (const_bitcast ffoldbomb i64_type) m);
 386
 387   group "misc constants";
 388   (* RUN: grep {const_size_of.*getelementptr.*null} < %t.ll
 389    * RUN: grep {const_gep.*getelementptr} < %t.ll
 390    * RUN: grep {const_select.*select} < %t.ll
 391    * RUN: grep {const_extractelement.*extractelement} < %t.ll
 392    * RUN: grep {const_insertelement.*insertelement} < %t.ll
 393    * RUN: grep {const_shufflevector.*shufflevector} < %t.ll
 394    *)
 395   ignore (define_global "const_size_of" (size_of (pointer_type i8_type)) m);
 396   ignore (define_global "const_gep" (const_gep foldbomb_gv [| five |]) m);
 397   ignore (define_global "const_select" (const_select
 398     (const_icmp Icmp.Sle foldbomb five)
 399     (const_int i8_type (-1))
 400     (const_int i8_type 0)) m);
 401   let zero = const_int i32_type 0 in
 402   let one  = const_int i32_type 1 in
 403   ignore (define_global "const_extractelement" (const_extractelement
 404     (const_vector [| zero; one; zero; one |])
 405     (const_trunc foldbomb i32_type)) m);
 406   ignore (define_global "const_insertelement" (const_insertelement
 407     (const_vector [| zero; one; zero; one |])
 408     zero (const_trunc foldbomb i32_type)) m);
 409   ignore (define_global "const_shufflevector" (const_shufflevector
 410     (const_vector [| zero; one |])
 411     (const_vector [| one; zero |])
 412     (const_bitcast foldbomb (vector_type i32_type 2))) m)
 413
 414
 415 (*===-- Global Values -----------------------------------------------------===*)
 416
 417 let test_global_values () =
 418   let (++) x f = f x; x in
 419   let zero32 = const_null i32_type in
 420
 421   (* RUN: grep {GVal01} < %t.ll
 422    *)
 423   group "naming";
 424   let g = define_global "TEMPORARY" zero32 m in
 425   insist ("TEMPORARY" = value_name g);
 426   set_value_name "GVal01" g;
 427   insist ("GVal01" = value_name g);
 428
 429   (* RUN: grep {GVal02.*linkonce} < %t.ll
 430    *)
 431   group "linkage";
 432   let g = define_global "GVal02" zero32 m ++
 433           set_linkage Linkage.Link_once in
 434   insist (Linkage.Link_once = linkage g);
 435
 436   (* RUN: grep {GVal03.*Hanalei} < %t.ll
 437    *)
 438   group "section";
 439   let g = define_global "GVal03" zero32 m ++
 440           set_section "Hanalei" in
 441   insist ("Hanalei" = section g);
 442
 443   (* RUN: grep {GVal04.*hidden} < %t.ll
 444    *)
 445   group "visibility";
 446   let g = define_global "GVal04" zero32 m ++
 447           set_visibility Visibility.Hidden in
 448   insist (Visibility.Hidden = visibility g);
 449
 450   (* RUN: grep {GVal05.*align 128} < %t.ll
 451    *)
 452   group "alignment";
 453   let g = define_global "GVal05" zero32 m ++
 454           set_alignment 128 in
 455   insist (128 = alignment g)
 456
 457
 458 (*===-- Global Variables --------------------------------------------------===*)
 459
 460 let test_global_variables () =
 461   let (++) x f = f x; x in
 462   let fourty_two32 = const_int i32_type 42 in
 463
 464   (* RUN: grep {GVar01.*external} < %t.ll
 465    *)
 466   group "declarations";
 467   insist (None == lookup_global "GVar01" m);
 468   let g = declare_global i32_type "GVar01" m in
 469   insist (is_declaration g);
 470   insist (pointer_type float_type ==
 471             type_of (declare_global float_type "GVar01" m));
 472   insist (g == declare_global i32_type "GVar01" m);
 473   insist (match lookup_global "GVar01" m with Some x -> x = g
 474                                             | None -> false);
 475
 476   (* RUN: grep {GVar02.*42} < %t.ll
 477    * RUN: grep {GVar03.*42} < %t.ll
 478    *)
 479   group "definitions";
 480   let g = define_global "GVar02" fourty_two32 m in
 481   let g2 = declare_global i32_type "GVar03" m ++
 482            set_initializer fourty_two32 in
 483   insist (not (is_declaration g));
 484   insist (not (is_declaration g2));
 485   insist ((global_initializer g) == (global_initializer g2));
 486
 487   (* RUN: grep {GVar04.*thread_local} < %t.ll
 488    *)
 489   group "threadlocal";
 490   let g = define_global "GVar04" fourty_two32 m ++
 491           set_thread_local true in
 492   insist (is_thread_local g);
 493
 494   (* RUN: grep -v {GVar05} < %t.ll
 495    *)
 496   group "delete";
 497   let g = define_global "GVar05" fourty_two32 m in
 498   delete_global g;
 499
 500   (* RUN: grep -v {ConstGlobalVar.*constant} < %t.ll
 501    *)
 502   group "constant";
 503   let g = define_global "ConstGlobalVar" fourty_two32 m in
 504   insist (not (is_global_constant g));
 505   set_global_constant true g;
 506   insist (is_global_constant g);
 507
 508   begin group "iteration";
 509     let m = create_module context "temp" in
 510
 511     insist (At_end m = global_begin m);
 512     insist (At_start m = global_end m);
 513
 514     let g1 = declare_global i32_type "One" m in
 515     let g2 = declare_global i32_type "Two" m in
 516
 517     insist (Before g1 = global_begin m);
 518     insist (Before g2 = global_succ g1);
 519     insist (At_end m = global_succ g2);
 520
 521     insist (After g2 = global_end m);
 522     insist (After g1 = global_pred g2);
 523     insist (At_start m = global_pred g1);
 524
 525     let lf s x = s ^ "->" ^ value_name x in
 526     insist ("->One->Two" = fold_left_globals lf "" m);
 527
 528     let rf x s = value_name x ^ "<-" ^ s in
 529     insist ("One<-Two<-" = fold_right_globals rf m "");
 530
 531     dispose_module m
 532   end
 533
 534
 535 (*===-- Functions ---------------------------------------------------------===*)
 536
 537 let test_functions () =
 538   let ty = function_type i32_type [| i32_type; i64_type |] in
 539   let ty2 = function_type i8_type [| i8_type; i64_type |] in
 540
 541   (* RUN: grep {declare i32 @Fn1\(i32, i64\)} < %t.ll
 542    *)
 543   begin group "declare";
 544     insist (None = lookup_function "Fn1" m);
 545     let fn = declare_function "Fn1" ty m in
 546     insist (pointer_type ty = type_of fn);
 547     insist (is_declaration fn);
 548     insist (0 = Array.length (basic_blocks fn));
 549     insist (pointer_type ty2 == type_of (declare_function "Fn1" ty2 m));
 550     insist (fn == declare_function "Fn1" ty m);
 551     insist (None <> lookup_function "Fn1" m);
 552     insist (match lookup_function "Fn1" m with Some x -> x = fn
 553                                              | None -> false);
 554     insist (m == global_parent fn)
 555   end;
 556
 557   (* RUN: grep -v {Fn2} < %t.ll
 558    *)
 559   group "delete";
 560   let fn = declare_function "Fn2" ty m in
 561   delete_function fn;
 562
 563   (* RUN: grep {define.*Fn3} < %t.ll
 564    *)
 565   group "define";
 566   let fn = define_function "Fn3" ty m in
 567   insist (not (is_declaration fn));
 568   insist (1 = Array.length (basic_blocks fn));
 569   ignore (build_unreachable (builder_at_end context (entry_block fn)));
 570
 571   (* RUN: grep {define.*Fn4.*Param1.*Param2} < %t.ll
 572    *)
 573   group "params";
 574   let fn = define_function "Fn4" ty m in
 575   let params = params fn in
 576   insist (2 = Array.length params);
 577   insist (params.(0) = param fn 0);
 578   insist (params.(1) = param fn 1);
 579   insist (i32_type = type_of params.(0));
 580   insist (i64_type = type_of params.(1));
 581   set_value_name "Param1" params.(0);
 582   set_value_name "Param2" params.(1);
 583   ignore (build_unreachable (builder_at_end context (entry_block fn)));
 584
 585   (* RUN: grep {fastcc.*Fn5} < %t.ll
 586    *)
 587   group "callconv";
 588   let fn = define_function "Fn5" ty m in
 589   insist (CallConv.c = function_call_conv fn);
 590   set_function_call_conv CallConv.fast fn;
 591   insist (CallConv.fast = function_call_conv fn);
 592   ignore (build_unreachable (builder_at_end context (entry_block fn)));
 593
 594   begin group "gc";
 595     (* RUN: grep {Fn6.*gc.*shadowstack} < %t.ll
 596      *)
 597     let fn = define_function "Fn6" ty m in
 598     insist (None = gc fn);
 599     set_gc (Some "ocaml") fn;
 600     insist (Some "ocaml" = gc fn);
 601     set_gc None fn;
 602     insist (None = gc fn);
 603     set_gc (Some "shadowstack") fn;
 604     ignore (build_unreachable (builder_at_end context (entry_block fn)));
 605   end;
 606
 607   begin group "iteration";
 608     let m = create_module context "temp" in
 609
 610     insist (At_end m = function_begin m);
 611     insist (At_start m = function_end m);
 612
 613     let f1 = define_function "One" ty m in
 614     let f2 = define_function "Two" ty m in
 615
 616     insist (Before f1 = function_begin m);
 617     insist (Before f2 = function_succ f1);
 618     insist (At_end m = function_succ f2);
 619
 620     insist (After f2 = function_end m);
 621     insist (After f1 = function_pred f2);
 622     insist (At_start m = function_pred f1);
 623
 624     let lf s x = s ^ "->" ^ value_name x in
 625     insist ("->One->Two" = fold_left_functions lf "" m);
 626
 627     let rf x s = value_name x ^ "<-" ^ s in
 628     insist ("One<-Two<-" = fold_right_functions rf m "");
 629
 630     dispose_module m
 631   end
 632
 633
 634 (*===-- Params ------------------------------------------------------------===*)
 635
 636 let test_params () =
 637   begin group "iteration";
 638     let m = create_module context "temp" in
 639
 640     let vf = define_function "void" (function_type void_type [| |]) m in
 641
 642     insist (At_end vf = param_begin vf);
 643     insist (At_start vf = param_end vf);
 644
 645     let ty = function_type void_type [| i32_type; i32_type |] in
 646     let f = define_function "f" ty m in
 647     let p1 = param f 0 in
 648     let p2 = param f 1 in
 649     set_value_name "One" p1;
 650     set_value_name "Two" p2;
 651     add_param_attr p1 Attribute.Sext;
 652     add_param_attr p2 Attribute.Noalias;
 653     remove_param_attr p2 Attribute.Noalias;
 654     add_function_attr f Attribute.Nounwind;
 655     add_function_attr f Attribute.Noreturn;
 656     remove_function_attr f Attribute.Noreturn;
 657
 658     insist (Before p1 = param_begin f);
 659     insist (Before p2 = param_succ p1);
 660     insist (At_end f = param_succ p2);
 661
 662     insist (After p2 = param_end f);
 663     insist (After p1 = param_pred p2);
 664     insist (At_start f = param_pred p1);
 665
 666     let lf s x = s ^ "->" ^ value_name x in
 667     insist ("->One->Two" = fold_left_params lf "" f);
 668
 669     let rf x s = value_name x ^ "<-" ^ s in
 670     insist ("One<-Two<-" = fold_right_params rf f "");
 671
 672     dispose_module m
 673   end
 674
 675
 676 (*===-- Basic Blocks ------------------------------------------------------===*)
 677
 678 let test_basic_blocks () =
 679   let ty = function_type void_type [| |] in
 680
 681   (* RUN: grep {Bb1} < %t.ll
 682    *)
 683   group "entry";
 684   let fn = declare_function "X" ty m in
 685   let bb = append_block context "Bb1" fn in
 686   insist (bb = entry_block fn);
 687   ignore (build_unreachable (builder_at_end context bb));
 688
 689   (* RUN: grep -v Bb2 < %t.ll
 690    *)
 691   group "delete";
 692   let fn = declare_function "X2" ty m in
 693   let bb = append_block context "Bb2" fn in
 694   delete_block bb;
 695
 696   group "insert";
 697   let fn = declare_function "X3" ty m in
 698   let bbb = append_block context "b" fn in
 699   let bba = insert_block context "a" bbb in
 700   insist ([| bba; bbb |] = basic_blocks fn);
 701   ignore (build_unreachable (builder_at_end context bba));
 702   ignore (build_unreachable (builder_at_end context bbb));
 703
 704   (* RUN: grep Bb3 < %t.ll
 705    *)
 706   group "name/value";
 707   let fn = define_function "X4" ty m in
 708   let bb = entry_block fn in
 709   ignore (build_unreachable (builder_at_end context bb));
 710   let bbv = value_of_block bb in
 711   set_value_name "Bb3" bbv;
 712   insist ("Bb3" = value_name bbv);
 713
 714   group "casts";
 715   let fn = define_function "X5" ty m in
 716   let bb = entry_block fn in
 717   ignore (build_unreachable (builder_at_end context bb));
 718   insist (bb = block_of_value (value_of_block bb));
 719   insist (value_is_block (value_of_block bb));
 720   insist (not (value_is_block (const_null i32_type)));
 721
 722   begin group "iteration";
 723     let m = create_module context "temp" in
 724     let f = declare_function "Temp" (function_type i32_type [| |]) m in
 725
 726     insist (At_end f = block_begin f);
 727     insist (At_start f = block_end f);
 728
 729     let b1 = append_block context "One" f in
 730     let b2 = append_block context "Two" f in
 731
 732     insist (Before b1 = block_begin f);
 733     insist (Before b2 = block_succ b1);
 734     insist (At_end f = block_succ b2);
 735
 736     insist (After b2 = block_end f);
 737     insist (After b1 = block_pred b2);
 738     insist (At_start f = block_pred b1);
 739
 740     let lf s x = s ^ "->" ^ value_name (value_of_block x) in
 741     insist ("->One->Two" = fold_left_blocks lf "" f);
 742
 743     let rf x s = value_name (value_of_block x) ^ "<-" ^ s in
 744     insist ("One<-Two<-" = fold_right_blocks rf f "");
 745
 746     dispose_module m
 747   end
 748
 749
 750 (*===-- Instructions ------------------------------------------------------===*)
 751
 752 let test_instructions () =
 753   begin group "iteration";
 754     let m = create_module context "temp" in
 755     let fty = function_type void_type [| i32_type; i32_type |] in
 756     let f = define_function "f" fty m in
 757     let bb = entry_block f in
 758     let b = builder_at context (At_end bb) in
 759
 760     insist (At_end bb = instr_begin bb);
 761     insist (At_start bb = instr_end bb);
 762
 763     let i1 = build_add (param f 0) (param f 1) "One" b in
 764     let i2 = build_sub (param f 0) (param f 1) "Two" b in
 765
 766     insist (Before i1 = instr_begin bb);
 767     insist (Before i2 = instr_succ i1);
 768     insist (At_end bb = instr_succ i2);
 769
 770     insist (After i2 = instr_end bb);
 771     insist (After i1 = instr_pred i2);
 772     insist (At_start bb = instr_pred i1);
 773
 774     let lf s x = s ^ "->" ^ value_name x in
 775     insist ("->One->Two" = fold_left_instrs lf "" bb);
 776
 777     let rf x s = value_name x ^ "<-" ^ s in
 778     insist ("One<-Two<-" = fold_right_instrs rf bb "");
 779
 780     dispose_module m
 781   end
 782
 783
 784 (*===-- Builder -----------------------------------------------------------===*)
 785
 786 let test_builder () =
 787   let (++) x f = f x; x in
 788
 789   begin group "parent";
 790     insist (try
 791               ignore (insertion_block (builder context));
 792               false
 793             with Not_found ->
 794               true);
 795
 796     let fty = function_type void_type [| i32_type |] in
 797     let fn = define_function "BuilderParent" fty m in
 798     let bb = entry_block fn in
 799     let b = builder_at_end context bb in
 800     let p = param fn 0 in
 801     let sum = build_add p p "sum" b in
 802     ignore (build_ret_void b);
 803
 804     insist (fn = block_parent bb);
 805     insist (fn = param_parent p);
 806     insist (bb = instr_parent sum);
 807     insist (bb = insertion_block b)
 808   end;
 809
 810   group "ret void";
 811   begin
 812     (* RUN: grep {ret void} < %t.ll
 813      *)
 814     let fty = function_type void_type [| |] in
 815     let fn = declare_function "X6" fty m in
 816     let b = builder_at_end context (append_block context "Bb01" fn) in
 817     ignore (build_ret_void b)
 818   end;
 819
 820   (* The rest of the tests will use one big function. *)
 821   let fty = function_type i32_type [| i32_type; i32_type |] in
 822   let fn = define_function "X7" fty m in
 823   let atentry = builder_at_end context (entry_block fn) in
 824   let p1 = param fn 0 ++ set_value_name "P1" in
 825   let p2 = param fn 1 ++ set_value_name "P2" in
 826   let f1 = build_uitofp p1 float_type "F1" atentry in
 827   let f2 = build_uitofp p2 float_type "F2" atentry in
 828
 829   let bb00 = append_block context "Bb00" fn in
 830   ignore (build_unreachable (builder_at_end context bb00));
 831
 832   group "ret"; begin
 833     (* RUN: grep {ret.*P1} < %t.ll
 834      *)
 835     let ret = build_ret p1 atentry in
 836     position_before ret atentry
 837   end;
 838
 839   group "br"; begin
 840     (* RUN: grep {br.*Bb02} < %t.ll
 841      *)
 842     let bb02 = append_block context "Bb02" fn in
 843     let b = builder_at_end context bb02 in
 844     ignore (build_br bb02 b)
 845   end;
 846
 847   group "cond_br"; begin
 848     (* RUN: grep {br.*build_br.*Bb03.*Bb00} < %t.ll
 849      *)
 850     let bb03 = append_block context "Bb03" fn in
 851     let b = builder_at_end context bb03 in
 852     let cond = build_trunc p1 i1_type "build_br" b in
 853     ignore (build_cond_br cond bb03 bb00 b)
 854   end;
 855
 856   group "switch"; begin
 857     (* RUN: grep {switch.*P1.*SwiBlock3} < %t.ll
 858      * RUN: grep {2,.*SwiBlock2} < %t.ll
 859      *)
 860     let bb1 = append_block context "SwiBlock1" fn in
 861     let bb2 = append_block context "SwiBlock2" fn in
 862     ignore (build_unreachable (builder_at_end context bb2));
 863     let bb3 = append_block context "SwiBlock3" fn in
 864     ignore (build_unreachable (builder_at_end context bb3));
 865     let si = build_switch p1 bb3 1 (builder_at_end context bb1) in
 866     ignore (add_case si (const_int i32_type 2) bb2)
 867   end;
 868
 869   group "invoke"; begin
 870     (* RUN: grep {build_invoke.*invoke.*P1.*P2} < %t.ll
 871      * RUN: grep {to.*Bb04.*unwind.*Bb00} < %t.ll
 872      *)
 873     let bb04 = append_block context "Bb04" fn in
 874     let b = builder_at_end context bb04 in
 875     ignore (build_invoke fn [| p1; p2 |] bb04 bb00 "build_invoke" b)
 876   end;
 877
 878   group "unwind"; begin
 879     (* RUN: grep {unwind} < %t.ll
 880      *)
 881     let bb05 = append_block context "Bb05" fn in
 882     let b = builder_at_end context bb05 in
 883     ignore (build_unwind b)
 884   end;
 885
 886   group "unreachable"; begin
 887     (* RUN: grep {unreachable} < %t.ll
 888      *)
 889     let bb06 = append_block context "Bb06" fn in
 890     let b = builder_at_end context bb06 in
 891     ignore (build_unreachable b)
 892   end;
 893
 894   group "arithmetic"; begin
 895     let bb07 = append_block context "Bb07" fn in
 896     let b = builder_at_end context bb07 in
 897
 898     (* RUN: grep {%build_add = add i32 %P1, %P2} < %t.ll
 899      * RUN: grep {%build_sub = sub i32 %P1, %P2} < %t.ll
 900      * RUN: grep {%build_mul = mul i32 %P1, %P2} < %t.ll
 901      * RUN: grep {%build_udiv = udiv i32 %P1, %P2} < %t.ll
 902      * RUN: grep {%build_sdiv = sdiv i32 %P1, %P2} < %t.ll
 903      * RUN: grep {%build_fdiv = fdiv float %F1, %F2} < %t.ll
 904      * RUN: grep {%build_urem = urem i32 %P1, %P2} < %t.ll
 905      * RUN: grep {%build_srem = srem i32 %P1, %P2} < %t.ll
 906      * RUN: grep {%build_frem = frem float %F1, %F2} < %t.ll
 907      * RUN: grep {%build_shl = shl i32 %P1, %P2} < %t.ll
 908      * RUN: grep {%build_lshl = lshr i32 %P1, %P2} < %t.ll
 909      * RUN: grep {%build_ashl = ashr i32 %P1, %P2} < %t.ll
 910      * RUN: grep {%build_and = and i32 %P1, %P2} < %t.ll
 911      * RUN: grep {%build_or = or i32 %P1, %P2} < %t.ll
 912      * RUN: grep {%build_xor = xor i32 %P1, %P2} < %t.ll
 913      * RUN: grep {%build_neg = sub i32 0, %P1} < %t.ll
 914      * RUN: grep {%build_not = xor i32 %P1, -1} < %t.ll
 915      *)
 916     ignore (build_add p1 p2 "build_add" b);
 917     ignore (build_sub p1 p2 "build_sub" b);
 918     ignore (build_mul p1 p2 "build_mul" b);
 919     ignore (build_udiv p1 p2 "build_udiv" b);
 920     ignore (build_sdiv p1 p2 "build_sdiv" b);
 921     ignore (build_fdiv f1 f2 "build_fdiv" b);
 922     ignore (build_urem p1 p2 "build_urem" b);
 923     ignore (build_srem p1 p2 "build_srem" b);
 924     ignore (build_frem f1 f2 "build_frem" b);
 925     ignore (build_shl p1 p2 "build_shl" b);
 926     ignore (build_lshr p1 p2 "build_lshl" b);
 927     ignore (build_ashr p1 p2 "build_ashl" b);
 928     ignore (build_and p1 p2 "build_and" b);
 929     ignore (build_or p1 p2 "build_or" b);
 930     ignore (build_xor p1 p2 "build_xor" b);
 931     ignore (build_neg p1 "build_neg" b);
 932     ignore (build_not p1 "build_not" b);
 933     ignore (build_unreachable b)
 934   end;
 935
 936   group "memory"; begin
 937     let bb08 = append_block context "Bb08" fn in
 938     let b = builder_at_end context bb08 in
 939
 940     (* RUN: grep {%build_alloca = alloca i32} < %t.ll
 941      * RUN: grep {%build_array_alloca = alloca i32, i32 %P2} < %t.ll
 942      * RUN: grep {%build_load = load i32\\* %build_array_alloca} < %t.ll
 943      * RUN: grep {store i32 %P2, i32\\* %build_alloca} < %t.ll
 944      * RUN: grep {%build_gep = getelementptr i32\\* %build_array_alloca, i32 %P2} < %t.ll
 945      *)
 946     let alloca = build_alloca i32_type "build_alloca" b in
 947     let array_alloca = build_array_alloca i32_type p2 "build_array_alloca" b in
 948     ignore(build_load array_alloca "build_load" b);
 949     ignore(build_store p2 alloca b);
 950     ignore(build_gep array_alloca [| p2 |] "build_gep" b);
 951     ignore(build_unreachable b)
 952   end;
 953
 954   group "casts"; begin
 955     let void_ptr = pointer_type i8_type in
 956
 957     (* RUN: grep {%build_trunc = trunc i32 %P1 to i8} < %t.ll
 958      * RUN: grep {%build_zext = zext i8 %build_trunc to i32} < %t.ll
 959      * RUN: grep {%build_sext = sext i32 %build_zext to i64} < %t.ll
 960      * RUN: grep {%build_uitofp = uitofp i64 %build_sext to float} < %t.ll
 961      * RUN: grep {%build_sitofp = sitofp i32 %build_zext to double} < %t.ll
 962      * RUN: grep {%build_fptoui = fptoui float %build_uitofp to i32} < %t.ll
 963      * RUN: grep {%build_fptosi = fptosi double %build_sitofp to i64} < %t.ll
 964      * RUN: grep {%build_fptrunc = fptrunc double %build_sitofp to float} < %t.ll
 965      * RUN: grep {%build_fpext = fpext float %build_fptrunc to double} < %t.ll
 966      * RUN: grep {%build_inttoptr = inttoptr i32 %P1 to i8\\*} < %t.ll
 967      * RUN: grep {%build_ptrtoint = ptrtoint i8\\* %build_inttoptr to i64} < %t.ll
 968      * RUN: grep {%build_bitcast = bitcast i64 %build_ptrtoint to double} < %t.ll
 969      *)
 970     let inst28 = build_trunc p1 i8_type "build_trunc" atentry in
 971     let inst29 = build_zext inst28 i32_type "build_zext" atentry in
 972     let inst30 = build_sext inst29 i64_type "build_sext" atentry in
 973     let inst31 = build_uitofp inst30 float_type "build_uitofp" atentry in
 974     let inst32 = build_sitofp inst29 double_type "build_sitofp" atentry in
 975     ignore(build_fptoui inst31 i32_type "build_fptoui" atentry);
 976     ignore(build_fptosi inst32 i64_type "build_fptosi" atentry);
 977     let inst35 = build_fptrunc inst32 float_type "build_fptrunc" atentry in
 978     ignore(build_fpext inst35 double_type "build_fpext" atentry);
 979     let inst37 = build_inttoptr p1 void_ptr "build_inttoptr" atentry in
 980     let inst38 = build_ptrtoint inst37 i64_type "build_ptrtoint" atentry in
 981     ignore(build_bitcast inst38 double_type "build_bitcast" atentry)
 982   end;
 983
 984   group "comparisons"; begin
 985     (* RUN: grep {%build_icmp_ne = icmp ne i32 %P1, %P2} < %t.ll
 986      * RUN: grep {%build_icmp_sle = icmp sle i32 %P2, %P1} < %t.ll
 987      * RUN: grep {%build_icmp_false = fcmp false float %F1, %F2} < %t.ll
 988      * RUN: grep {%build_icmp_true = fcmp true float %F2, %F1} < %t.ll
 989      *)
 990     ignore (build_icmp Icmp.Ne    p1 p2 "build_icmp_ne" atentry);
 991     ignore (build_icmp Icmp.Sle   p2 p1 "build_icmp_sle" atentry);
 992     ignore (build_fcmp Fcmp.False f1 f2 "build_icmp_false" atentry);
 993     ignore (build_fcmp Fcmp.True  f2 f1 "build_icmp_true" atentry)
 994   end;
 995
 996   group "miscellaneous"; begin
 997     (* RUN: grep {%build_call = tail call cc63 i32 @.*(i32 signext %P2, i32 %P1)} < %t.ll
 998      * RUN: grep {%build_select = select i1 %build_icmp, i32 %P1, i32 %P2} < %t.ll
 999      * RUN: grep {%build_va_arg = va_arg i8\\*\\* null, i32} < %t.ll
1000      * RUN: grep {%build_extractelement = extractelement <4 x i32> %Vec1, i32 %P2} < %t.ll
1001      * RUN: grep {%build_insertelement = insertelement <4 x i32> %Vec1, i32 %P1, i32 %P2} < %t.ll
1002      * RUN: grep {%build_shufflevector = shufflevector <4 x i32> %Vec1, <4 x i32> %Vec2, <4 x i32> <i32 1, i32 1, i32 0, i32 0>} < %t.ll
1003      *)
1004     let ci = build_call fn [| p2; p1 |] "build_call" atentry in
1005     insist (CallConv.c = instruction_call_conv ci);
1006     set_instruction_call_conv 63 ci;
1007     insist (63 = instruction_call_conv ci);
1008     insist (not (is_tail_call ci));
1009     set_tail_call true ci;
1010     insist (is_tail_call ci);
1011     add_instruction_param_attr ci 1 Attribute.Sext;
1012     add_instruction_param_attr ci 2 Attribute.Noalias;
1013     remove_instruction_param_attr ci 2 Attribute.Noalias;
1014
1015     let inst46 = build_icmp Icmp.Eq p1 p2 "build_icmp" atentry in
1016     ignore (build_select inst46 p1 p2 "build_select" atentry);
1017     ignore (build_va_arg
1018       (const_null (pointer_type (pointer_type i8_type)))
1019       i32_type "build_va_arg" atentry);
1020
1021     (* Set up some vector vregs. *)
1022     let one  = const_int i32_type 1 in
1023     let zero = const_int i32_type 0 in
1024     let t1 = const_vector [| one; zero; one; zero |] in
1025     let t2 = const_vector [| zero; one; zero; one |] in
1026     let t3 = const_vector [| one; one; zero; zero |] in
1027     let vec1 = build_insertelement t1 p1 p2 "Vec1" atentry in
1028     let vec2 = build_insertelement t2 p1 p2 "Vec2" atentry in
1029
1030     ignore (build_extractelement vec1 p2 "build_extractelement" atentry);
1031     ignore (build_insertelement vec1 p1 p2 "build_insertelement" atentry);
1032     ignore (build_shufflevector vec1 vec2 t3 "build_shufflevector" atentry);
1033   end;
1034
1035   group "phi"; begin
1036     (* RUN: grep {PhiNode.*P1.*PhiBlock1.*P2.*PhiBlock2} < %t.ll
1037      *)
1038     let b1 = append_block context "PhiBlock1" fn in
1039     let b2 = append_block context "PhiBlock2" fn in
1040
1041     let jb = append_block context "PhiJoinBlock" fn in
1042     ignore (build_br jb (builder_at_end context b1));
1043     ignore (build_br jb (builder_at_end context b2));
1044     let at_jb = builder_at_end context jb in
1045
1046     let phi = build_phi [(p1, b1)] "PhiNode" at_jb in
1047     insist ([(p1, b1)] = incoming phi);
1048
1049     add_incoming (p2, b2) phi;
1050     insist ([(p1, b1); (p2, b2)] = incoming phi);
1051
1052     ignore (build_unreachable at_jb);
1053   end
1054
1055
1056 (*===-- Module Provider ---------------------------------------------------===*)
1057
1058 let test_module_provider () =
1059   let m = create_module context "test" in
1060   let mp = ModuleProvider.create m in
1061   ModuleProvider.dispose mp
1062
1063
1064 (*===-- Pass Managers -----------------------------------------------------===*)
1065
1066 let test_pass_manager () =
1067   let (++) x f = ignore (f x); x in
1068
1069   begin group "module pass manager";
1070     ignore (PassManager.create ()
1071              ++ PassManager.run_module m
1072              ++ PassManager.dispose)
1073   end;
1074
1075   begin group "function pass manager";
1076     let fty = function_type void_type [| |] in
1077     let fn = define_function "FunctionPassManager" fty m in
1078     ignore (build_ret_void (builder_at_end context (entry_block fn)));
1079
1080     ignore (PassManager.create_function mp
1081              ++ PassManager.initialize
1082              ++ PassManager.run_function fn
1083              ++ PassManager.finalize
1084              ++ PassManager.dispose)
1085   end
1086
1087
1088 (*===-- Writer ------------------------------------------------------------===*)
1089
1090 let test_writer () =
1091   group "valid";
1092   insist (match Llvm_analysis.verify_module m with
1093           | None -> true
1094           | Some msg -> prerr_string msg; false);
1095
1096   group "writer";
1097   insist (write_bitcode_file m filename);
1098
1099   ModuleProvider.dispose mp
1100
1101
1102 (*===-- Driver ------------------------------------------------------------===*)
1103
1104 let _ =
1105   suite "target"           test_target;
1106   suite "types"            test_types;
1107   suite "constants"        test_constants;
1108   suite "global values"    test_global_values;
1109   suite "global variables" test_global_variables;
1110   suite "functions"        test_functions;
1111   suite "params"           test_params;
1112   suite "basic blocks"     test_basic_blocks;
1113   suite "instructions"     test_instructions;
1114   suite "builder"          test_builder;
1115   suite "module provider"  test_module_provider;
1116   suite "pass manager"     test_pass_manager;
1117   suite "writer"           test_writer; (* Keep this last; it disposes m. *)
1118   exit !exit_status