blob: aa032936cae1bf44552f8d2eea42d7afc7153b5c [file] [log] [blame]
extern crate rayon;
use rayon::*;
use std::process::{self, Command};
use std::env;
#[cfg(target_os = "linux")]
use std::os::unix::process::ExitStatusExt;
fn force_stack_overflow(depth: u32) {
let buffer = [0u8; 1024*1024];
if depth > 0 {
force_stack_overflow(depth - 1);
}
}
fn main() {
if env::args().len() == 1 {
// first check that the recursivecall actually causes a stack overflow, and does not get optimized away
{
let status = Command::new(env::current_exe().unwrap())
.arg("8")
.status()
.unwrap();
#[cfg(windows)]
assert_eq!(status.code(), Some(0xc00000fd /*STATUS_STACK_OVERFLOW*/));
#[cfg(unix)]
assert_eq!(status.code(), None);
#[cfg(target_os = "linux")]
assert!(status.signal() == Some(11 /*SIGABRT*/) ||
status.signal() == Some(6 /*SIGSEGV*/));
}
// now run with a larger stack and verify correct operation
{
let status = Command::new(env::current_exe().unwrap())
.arg("48")
.status()
.unwrap();
assert_eq!(status.code(), Some(0));
#[cfg(target_os = "linux")]
assert_eq!(status.signal(), None);
}
} else {
let stack_size_in_mb: usize = env::args().nth(1).unwrap().parse().unwrap();
let pool = ThreadPoolBuilder::new().stack_size(stack_size_in_mb * 1024 * 1024).build().unwrap();
let index = pool.install(|| {
force_stack_overflow(32);
});
}
}