1// Test ASan detection of stack-overflow condition.
2
3// RUN: %clangxx_asan -O0 %s -DSMALL_FRAME -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
4// RUN: %clangxx_asan -O3 %s -DSMALL_FRAME -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
5// RUN: %clangxx_asan -O0 %s -DSAVE_ALL_THE_REGISTERS -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
6// RUN: %clangxx_asan -O3 %s -DSAVE_ALL_THE_REGISTERS -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
7// RUN: %clangxx_asan -O0 %s -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
8// RUN: %clangxx_asan -O3 %s -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
9
10// RUN: %clangxx_asan -O0 %s -DTHREAD -DSMALL_FRAME -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
11// RUN: %clangxx_asan -O3 %s -DTHREAD -DSMALL_FRAME -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
12// RUN: %clangxx_asan -O0 %s -DTHREAD -DSAVE_ALL_THE_REGISTERS -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
13// RUN: %clangxx_asan -O3 %s -DTHREAD -DSAVE_ALL_THE_REGISTERS -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
14// RUN: %clangxx_asan -O0 %s -DTHREAD -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
15// RUN: %clangxx_asan -O3 %s -DTHREAD -pthread -o %t && %env_asan_opts=use_sigaltstack=1 not %run %t 2>&1 | FileCheck %s
16// RUN: not %run %t 2>&1 | FileCheck %s
17// REQUIRES: stable-runtime
18
19// Issue #109771
20// XFAIL: target={{sparc.*-.*-linux.*}}
21
22// UNSUPPORTED: ios
23
24#include <assert.h>
25#include <stdlib.h>
26#include <pthread.h>
27#include <unistd.h>
28#include <sys/time.h>
29#include <sys/resource.h>
30#include <sanitizer/asan_interface.h>
31
32const int BS = 1024;
33volatile char x;
34volatile int y = 1;
35volatile int z0, z1, z2, z3, z4, z5, z6, z7, z8, z9, z10, z11, z12, z13;
36
37void recursive_func(uintptr_t parent_frame_address) {
38#if defined(SMALL_FRAME)
39 char *buf = 0;
40#elif defined(SAVE_ALL_THE_REGISTERS)
41 char *buf = 0;
42 int t0, t1, t2, t3, t4, t5, t6, t7, t8, t9, t10, t11, t12, t13;
43 t0 = z0;
44 t1 = z1;
45 t2 = z2;
46 t3 = z3;
47 t4 = z4;
48 t5 = z5;
49 t6 = z6;
50 t7 = z7;
51 t8 = z8;
52 t9 = z9;
53 t10 = z10;
54 t11 = z11;
55 t12 = z12;
56 t13 = z13;
57
58 z0 = t0;
59 z1 = t1;
60 z2 = t2;
61 z3 = t3;
62 z4 = t4;
63 z5 = t5;
64 z6 = t6;
65 z7 = t7;
66 z8 = t8;
67 z9 = t9;
68 z10 = t10;
69 z11 = t11;
70 z12 = t12;
71 z13 = t13;
72#else
73 char buf[BS];
74 // Check that the stack grows in the right direction, unless we use fake stack.
75 assert(parent_frame_address > (uintptr_t)__builtin_frame_address(0));
76 buf[rand() % BS] = 1;
77 buf[rand() % BS] = 2;
78 x = buf[rand() % BS];
79#endif
80 if (y)
81 recursive_func(parent_frame_address: (uintptr_t)__builtin_frame_address(0));
82 x = 1; // prevent tail call optimization
83 // CHECK: {{stack-overflow on address 0x.* \(pc 0x.* bp 0x.* sp 0x.* T.*\)}}
84 // If stack overflow happens during function prologue, stack trace may be
85 // corrupted. Unwind tables are not always 100% exact there.
86 // For this reason, we don't do any further checks.
87}
88
89void *ThreadFn(void* unused) {
90 recursive_func(parent_frame_address: (uintptr_t)__builtin_frame_address(0));
91 return 0;
92}
93
94void LimitStackAndReexec(int argc, char **argv) {
95 struct rlimit rlim;
96 int res = getrlimit(RLIMIT_STACK, rlimits: &rlim);
97 assert(res == 0);
98 if (rlim.rlim_cur == RLIM_INFINITY) {
99 rlim.rlim_cur = 256 * 1024;
100 res = setrlimit(RLIMIT_STACK, rlimits: &rlim);
101 assert(res == 0);
102
103 execv(path: argv[0], argv: argv);
104 assert(0 && "unreachable");
105 }
106}
107
108int main(int argc, char **argv) {
109 LimitStackAndReexec(argc, argv);
110#ifdef THREAD
111 pthread_t t;
112 pthread_create(&t, 0, ThreadFn, 0);
113 pthread_join(t, 0);
114#else
115 recursive_func(parent_frame_address: (uintptr_t)__builtin_frame_address(0));
116#endif
117 return 0;
118}
119

source code of compiler-rt/test/asan/TestCases/Posix/stack-overflow.cpp