tests/codegen/c-variadic.rs

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