about summary refs log tree commit diff
path: root/tests/ui/runtime/out-of-stack.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tests/ui/runtime/out-of-stack.rs')
-rw-r--r--tests/ui/runtime/out-of-stack.rs88
1 files changed, 88 insertions, 0 deletions
diff --git a/tests/ui/runtime/out-of-stack.rs b/tests/ui/runtime/out-of-stack.rs
new file mode 100644
index 00000000000..6873abc49b2
--- /dev/null
+++ b/tests/ui/runtime/out-of-stack.rs
@@ -0,0 +1,88 @@
+// run-pass
+
+#![allow(unused_must_use)]
+#![allow(unconditional_recursion)]
+// ignore-android: FIXME (#20004)
+// ignore-emscripten no processes
+// ignore-sgx no processes
+// ignore-fuchsia must translate zircon signal to SIGABRT, FIXME (#58590)
+
+#![feature(core_intrinsics)]
+#![feature(rustc_private)]
+
+#[cfg(unix)]
+extern crate libc;
+
+use std::env;
+use std::process::Command;
+use std::thread;
+
+// Inlining to avoid llvm turning the recursive functions into tail calls,
+// which doesn't consume stack.
+#[inline(always)]
+pub fn black_box<T>(dummy: T) { std::intrinsics::black_box(dummy); }
+
+fn silent_recurse() {
+    let buf = [0u8; 1000];
+    black_box(buf);
+    silent_recurse();
+}
+
+fn loud_recurse() {
+    println!("hello!");
+    loud_recurse();
+    black_box(()); // don't optimize this into a tail call. please.
+}
+
+#[cfg(unix)]
+fn check_status(status: std::process::ExitStatus)
+{
+    use std::os::unix::process::ExitStatusExt;
+
+    assert!(!status.success());
+    assert_eq!(status.signal(), Some(libc::SIGABRT));
+}
+
+#[cfg(not(unix))]
+fn check_status(status: std::process::ExitStatus)
+{
+    assert!(!status.success());
+}
+
+
+fn main() {
+    let args: Vec<String> = env::args().collect();
+    if args.len() > 1 && args[1] == "silent" {
+        silent_recurse();
+    } else if args.len() > 1 && args[1] == "loud" {
+        loud_recurse();
+    } else if args.len() > 1 && args[1] == "silent-thread" {
+        thread::spawn(silent_recurse).join();
+    } else if args.len() > 1 && args[1] == "loud-thread" {
+        thread::spawn(loud_recurse).join();
+    } else {
+        let mut modes = vec![
+            "silent-thread",
+            "loud-thread",
+        ];
+
+        // On linux it looks like the main thread can sometimes grow its stack
+        // basically without bounds, so we only test the child thread cases
+        // there.
+        if !cfg!(target_os = "linux") {
+            modes.push("silent");
+            modes.push("loud");
+        }
+        for mode in modes {
+            println!("testing: {}", mode);
+
+            let silent = Command::new(&args[0]).arg(mode).output().unwrap();
+
+            check_status(silent.status);
+
+            let error = String::from_utf8_lossy(&silent.stderr);
+            assert!(error.contains("has overflowed its stack"),
+                    "missing overflow message: {}", error);
+        }
+    }
+}