src/test/codegen/naked-functions.rs

   1 // compile-flags: -C no-prepopulate-passes -Zmir-opt-level=0
   2
   3 #![crate_type = "lib"]
   4 #![feature(naked_functions)]
   5
   6 // CHECK: Function Attrs: naked
   7 // CHECK-NEXT: define void @naked_empty()
   8 #[no_mangle]
   9 #[naked]
  10 pub fn naked_empty() {
  11     // CHECK-NEXT: {{.+}}:
  12     // CHECK-NEXT: ret void
  13 }
  14
  15 // CHECK: Function Attrs: naked
  16 #[no_mangle]
  17 #[naked]
  18 // CHECK-NEXT: define void @naked_with_args(i{{[0-9]+( %0)?}})
  19 pub fn naked_with_args(a: isize) {
  20     // CHECK-NEXT: {{.+}}:
  21     // CHECK-NEXT: %a = alloca i{{[0-9]+}}
  22     &a; // keep variable in an alloca
  23     // CHECK: ret void
  24 }
  25
  26 // CHECK: Function Attrs: naked
  27 // CHECK-NEXT: define i{{[0-9]+}} @naked_with_return()
  28 #[no_mangle]
  29 #[naked]
  30 pub fn naked_with_return() -> isize {
  31     // CHECK-NEXT: {{.+}}:
  32     // CHECK-NEXT: ret i{{[0-9]+}} 0
  33     0
  34 }
  35
  36 // CHECK: Function Attrs: naked
  37 // CHECK-NEXT: define i{{[0-9]+}} @naked_with_args_and_return(i{{[0-9]+( %0)?}})
  38 #[no_mangle]
  39 #[naked]
  40 pub fn naked_with_args_and_return(a: isize) -> isize {
  41     // CHECK-NEXT: {{.+}}:
  42     // CHECK-NEXT: %a = alloca i{{[0-9]+}}
  43     &a; // keep variable in an alloca
  44     // CHECK: ret i{{[0-9]+}} %{{[0-9]+}}
  45     a
  46 }
  47
  48 // CHECK: Function Attrs: naked
  49 // CHECK-NEXT: define void @naked_recursive()
  50 #[no_mangle]
  51 #[naked]
  52 pub fn naked_recursive() {
  53     // CHECK-NEXT: {{.+}}:
  54     // CHECK-NEXT: call void @naked_empty()
  55
  56     // FIXME(#39685) Avoid one block per call.
  57     // CHECK-NEXT: br label %bb1
  58     // CHECK: bb1:
  59
  60     naked_empty();
  61
  62     // CHECK-NEXT: %_4 = call i{{[0-9]+}} @naked_with_return()
  63
  64     // FIXME(#39685) Avoid one block per call.
  65     // CHECK-NEXT: br label %bb2
  66     // CHECK: bb2:
  67
  68     // CHECK-NEXT: %_3 = call i{{[0-9]+}} @naked_with_args_and_return(i{{[0-9]+}} %_4)
  69
  70     // FIXME(#39685) Avoid one block per call.
  71     // CHECK-NEXT: br label %bb3
  72     // CHECK: bb3:
  73
  74     // CHECK-NEXT: call void @naked_with_args(i{{[0-9]+}} %_3)
  75
  76     // FIXME(#39685) Avoid one block per call.
  77     // CHECK-NEXT: br label %bb4
  78     // CHECK: bb4:
  79
  80     naked_with_args(
  81         naked_with_args_and_return(
  82             naked_with_return()
  83         )
  84     );
  85     // CHECK-NEXT: ret void
  86 }