src/test/codegen/c-variadic.rs

   1 // ignore-wasm32-bare compiled with panic=abort by default
   2 // compile-flags: -C no-prepopulate-passes
   3 // ignore-tidy-linelength
   4
   5 #![crate_type = "lib"]
   6 #![feature(c_variadic)]
   7 #![feature(unwind_attributes)]
   8 #![no_std]
   9 use core::ffi::VaList;
  10
  11 extern "C" {
  12     fn foreign_c_variadic_0(_: i32, ...);
  13     fn foreign_c_variadic_1(_: VaList, ...);
  14 }
  15
  16 #[unwind(aborts)] // FIXME(#58794)
  17 pub unsafe extern "C" fn use_foreign_c_variadic_0() {
  18     // Ensure that we correctly call foreign C-variadic functions.
  19     // CHECK: call void (i32, ...) @foreign_c_variadic_0([[PARAM:i32( signext)?]] 0)
  20     foreign_c_variadic_0(0);
  21     // CHECK: call void (i32, ...) @foreign_c_variadic_0([[PARAM]] 0, [[PARAM]] 42)
  22     foreign_c_variadic_0(0, 42i32);
  23     // CHECK: call void (i32, ...) @foreign_c_variadic_0([[PARAM]] 0, [[PARAM]] 42, [[PARAM]] 1024)
  24     foreign_c_variadic_0(0, 42i32, 1024i32);
  25     // CHECK: call void (i32, ...) @foreign_c_variadic_0([[PARAM]] 0, [[PARAM]] 42, [[PARAM]] 1024, [[PARAM]] 0)
  26     foreign_c_variadic_0(0, 42i32, 1024i32, 0i32);
  27 }
  28
  29 // Ensure that we do not remove the `va_list` passed to the foreign function when
  30 // removing the "spoofed" `VaListImpl` that is used by Rust defined C-variadics.
  31 #[unwind(aborts)] // FIXME(#58794)
  32 pub unsafe extern "C" fn use_foreign_c_variadic_1_0(ap: VaList) {
  33     // CHECK: call void ({{.*}}*, ...) @foreign_c_variadic_1({{.*}} %ap)
  34     foreign_c_variadic_1(ap);
  35 }
  36
  37 #[unwind(aborts)] // FIXME(#58794)
  38 pub unsafe extern "C" fn use_foreign_c_variadic_1_1(ap: VaList) {
  39     // CHECK: call void ({{.*}}*, ...) @foreign_c_variadic_1({{.*}} %ap, [[PARAM]] 42)
  40     foreign_c_variadic_1(ap, 42i32);
  41 }
  42 #[unwind(aborts)] // FIXME(#58794)
  43 pub unsafe extern "C" fn use_foreign_c_variadic_1_2(ap: VaList) {
  44     // CHECK: call void ({{.*}}*, ...) @foreign_c_variadic_1({{.*}} %ap, [[PARAM]] 2, [[PARAM]] 42)
  45     foreign_c_variadic_1(ap, 2i32, 42i32);
  46 }
  47
  48 #[unwind(aborts)] // FIXME(#58794)
  49 pub unsafe extern "C" fn use_foreign_c_variadic_1_3(ap: VaList) {
  50     // CHECK: call void ({{.*}}*, ...) @foreign_c_variadic_1({{.*}} %ap, [[PARAM]] 2, [[PARAM]] 42, [[PARAM]] 0)
  51     foreign_c_variadic_1(ap, 2i32, 42i32, 0i32);
  52 }
  53
  54 // Ensure that `va_start` and `va_end` are properly injected.
  55 #[no_mangle]
  56 pub unsafe extern "C" fn c_variadic(n: i32, mut ap: ...) -> i32 {
  57     // CHECK: call void @llvm.va_start
  58     let mut sum = 0;
  59     for _ in 0..n {
  60         sum += ap.arg::<i32>();
  61     }
  62     sum
  63     // CHECK: call void @llvm.va_end
  64 }
  65
  66 // Ensure that we generate the correct `call` signature when calling a Rust
  67 // defined C-variadic.
  68 pub unsafe fn test_c_variadic_call() {
  69     // CHECK: call [[RET:(signext )?i32]] (i32, ...) @c_variadic([[PARAM]] 0)
  70     c_variadic(0);
  71     // CHECK: call [[RET]] (i32, ...) @c_variadic([[PARAM]] 0, [[PARAM]] 42)
  72     c_variadic(0, 42i32);
  73     // CHECK: call [[RET]] (i32, ...) @c_variadic([[PARAM]] 0, [[PARAM]] 42, [[PARAM]] 1024)
  74     c_variadic(0, 42i32, 1024i32);
  75     // CHECK: call [[RET]] (i32, ...) @c_variadic([[PARAM]] 0, [[PARAM]] 42, [[PARAM]] 1024, [[PARAM]] 0)
  76     c_variadic(0, 42i32, 1024i32, 0i32);
  77 }