1*7c3d14c8STreehugger Robot // Test that ASan doesn't hang on stack overflow in recovery mode.
2*7c3d14c8STreehugger Robot //
3*7c3d14c8STreehugger Robot // RUN: %clang_asan -O0 -fsanitize-recover=address %s -o %t
4*7c3d14c8STreehugger Robot // RUN: %env_asan_opts=halt_on_error=false not %run %t 2>&1 | FileCheck %s
5*7c3d14c8STreehugger Robot
6*7c3d14c8STreehugger Robot #include <assert.h>
7*7c3d14c8STreehugger Robot #include <unistd.h>
8*7c3d14c8STreehugger Robot #include <sys/mman.h>
9*7c3d14c8STreehugger Robot #include <sys/resource.h>
10*7c3d14c8STreehugger Robot
recurse(volatile int n,volatile int * p)11*7c3d14c8STreehugger Robot static volatile int *recurse(volatile int n, volatile int *p) {
12*7c3d14c8STreehugger Robot // CHECK: {{stack-overflow on address 0x.* \(pc 0x.* bp 0x.* sp 0x.* T.*\)}}
13*7c3d14c8STreehugger Robot if (n >= 0) *recurse(n + 1, p) += n;
14*7c3d14c8STreehugger Robot return p;
15*7c3d14c8STreehugger Robot }
16*7c3d14c8STreehugger Robot
17*7c3d14c8STreehugger Robot
LimitStackAndReexec(int argc,char ** argv)18*7c3d14c8STreehugger Robot void LimitStackAndReexec(int argc, char **argv) {
19*7c3d14c8STreehugger Robot struct rlimit rlim;
20*7c3d14c8STreehugger Robot int res = getrlimit(RLIMIT_STACK, &rlim);
21*7c3d14c8STreehugger Robot assert(res == 0);
22*7c3d14c8STreehugger Robot if (rlim.rlim_cur == RLIM_INFINITY) {
23*7c3d14c8STreehugger Robot rlim.rlim_cur = 256 * 1024;
24*7c3d14c8STreehugger Robot res = setrlimit(RLIMIT_STACK, &rlim);
25*7c3d14c8STreehugger Robot assert(res == 0);
26*7c3d14c8STreehugger Robot
27*7c3d14c8STreehugger Robot execv(argv[0], argv);
28*7c3d14c8STreehugger Robot assert(0 && "unreachable");
29*7c3d14c8STreehugger Robot }
30*7c3d14c8STreehugger Robot }
31*7c3d14c8STreehugger Robot
main(int argc,char ** argv)32*7c3d14c8STreehugger Robot int main(int argc, char **argv) {
33*7c3d14c8STreehugger Robot LimitStackAndReexec(argc, argv);
34*7c3d14c8STreehugger Robot volatile int res;
35*7c3d14c8STreehugger Robot return *recurse(argc + 1, &res);
36*7c3d14c8STreehugger Robot }
37