test/Bindings/Ocaml/vmcore.ml

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