add tests for probestack linking

drivers/char/rust_example.rs

@@ -2,6 +2,7 @@
 
 #![no_std]
 #![feature(allocator_api, global_asm)]
+#![feature(test)]
 
 use alloc::boxed::Box;
 use core::pin::Pin;
@@ -52,6 +53,13 @@ impl KernelModule for RustExample {
  println!(" my_bool: {}", my_bool.read());
  println!(" my_i32: {}", my_i32.read());
 
+ // Including this large variable on the stack will trigger
+ // stack probing on the supported archs.
+ // This will verify that stack probing does not lead to
+ // any errors if we need to link `__rust_probestack`.
+ let x: [u64; 1028] = core::hint::black_box([5; 1028]);
+ println!("Large array has length: {}", x.len());
+
  Ok(RustExample {
  message: "on the heap!".to_owned(),
  _dev: miscdev::Registration::new_pinned::<RustFile>(cstr!("rust_miscdev"), None)?,

drivers/char/rust_example_2.rs

@@ -2,6 +2,7 @@
 
 #![no_std]
 #![feature(allocator_api, global_asm)]
+#![feature(test)]
 
 use kernel::prelude::*;
 
@@ -36,6 +37,14 @@ impl KernelModule for RustExample2 {
  println!("[2] Parameters:");
  println!("[2] my_bool: {}", my_bool.read());
  println!("[2] my_i32: {}", my_i32.read());
+
+ // Including this large variable on the stack will trigger
+ // stack probing on the supported archs.
+ // This will verify that stack probing does not lead to
+ // any errors if we need to link `__rust_probestack`.
+ let x: [u64; 1028] = core::hint::black_box([5; 1028]);
+ println!("Large array has length: {}", x.len());
+
  Ok(RustExample2 {
  message: "on the heap!".to_owned(),
  })

drivers/char/rust_example_3.rs

@@ -2,6 +2,7 @@
 
 #![no_std]
 #![feature(allocator_api, global_asm)]
+#![feature(test)]
 
 use kernel::prelude::*;
 
@@ -36,6 +37,14 @@ impl KernelModule for RustExample3 {
  println!("[3] Parameters:");
  println!("[3] my_bool: {}", my_bool.read());
  println!("[3] my_i32: {}", my_i32.read());
+
+ // Including this large variable on the stack will trigger
+ // stack probing on the supported archs.
+ // This will verify that stack probing does not lead to
+ // any errors if we need to link `__rust_probestack`.
+ let x: [u64; 1028] = core::hint::black_box([5; 1028]);
+ println!("Large array has length: {}", x.len());
+
  Ok(RustExample3 {
  message: "on the heap!".to_owned(),
  })

drivers/char/rust_example_4.rs

@@ -2,6 +2,7 @@
 
 #![no_std]
 #![feature(allocator_api, global_asm)]
+#![feature(test)]
 
 use kernel::prelude::*;
 
@@ -36,6 +37,14 @@ impl KernelModule for RustExample4 {
  println!("[4] Parameters:");
  println!("[4] my_bool: {}", my_bool.read());
  println!("[4] my_i32: {}", my_i32.read());
+
+ // Including this large variable on the stack will trigger
+ // stack probing on the supported archs.
+ // This will verify that stack probing does not lead to
+ // any errors if we need to link `__rust_probestack`.
+ let x: [u64; 1028] = core::hint::black_box([5; 1028]);
+ println!("Large array has length: {}", x.len());
+
  Ok(RustExample4 {
  message: "on the heap!".to_owned(),
  })