1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
extern crate stacker;
use std::sync::mpsc;
use std::thread;
#[inline(never)]
fn __stacker_black_box(_: *const u8) {}
#[test]
fn deep() {
fn foo(n: usize, s: &mut [u8]) {
__stacker_black_box(s.as_ptr());
if n > 0 {
stacker::maybe_grow(64 * 1024, 1024 * 1024, || {
let mut s = [0u8; 1024];
foo(n - 1, &mut s);
__stacker_black_box(s.as_ptr());
})
} else {
println!("bottom");
}
}
let limit = if cfg!(target_arch = "wasm32") {
2000
} else {
256 * 1024
};
foo(limit, &mut []);
}
#[test]
#[cfg_attr(target_arch = "wasm32", ignore)]
fn panic() {
fn foo(n: usize, s: &mut [u8]) {
__stacker_black_box(s.as_ptr());
if n > 0 {
stacker::maybe_grow(64 * 1024, 1024 * 1024, || {
let mut s = [0u8; 1024];
foo(n - 1, &mut s);
__stacker_black_box(s.as_ptr());
})
} else {
panic!("bottom");
}
}
let (tx, rx) = mpsc::channel::<()>();
thread::spawn(move || {
foo(64 * 1024, &mut []);
drop(tx);
})
.join()
.unwrap_err();
assert!(rx.recv().is_err());
}
fn recursive<F: FnOnce()>(n: usize, f: F) -> usize {
if n > 0 {
stacker::grow(64 * 1024, || recursive(n - 1, f) + 1)
} else {
f();
0
}
}
#[test]
#[cfg_attr(target_arch = "wasm32", ignore)]
fn catch_panic() {
let panic_result = std::panic::catch_unwind(|| {
recursive(100, || panic!());
});
assert!(panic_result.is_err());
}
#[test]
#[cfg_attr(target_arch = "wasm32", ignore)]
fn catch_panic_inside() {
stacker::grow(64 * 1024, || {
let panic_result = std::panic::catch_unwind(|| {
recursive(100, || panic!());
});
assert!(panic_result.is_err());
});
}
#[test]
#[cfg_attr(target_arch = "wasm32", ignore)]
fn catch_panic_leaf() {
stacker::grow(64 * 1024, || {
let panic_result = std::panic::catch_unwind(|| panic!());
assert!(panic_result.is_err());
});
}
|
