src/test/codegen/c-variadic.rs

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