1ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown/*
2ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   This test makes sure the thread exit notification signals don't
3ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   interfere with the stack growth signals.
4ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
5ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   Thread death notifications are sent as RT signals, which are
6ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   queued.  In general, these notifications are ignored, since they're
7ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   only used by the main thread if it has exited and is still waiting
8ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   for the rest to exit.
9ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
10ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   The system has a finite limit to the number of RT signals which can
11ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   be queued (typically 1024), and beyond that it stops queueing
12ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   siginfo.  We rely on getting SIGSEGVs with siginfo information to
13ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   grow the stack.  If we don't get the siginfo, then it just looks
14ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   like the program crashed.
15ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
16ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   The extra complication in this test is making sure that the
17ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   unwanted signals are discarded while the main thread is blocked in
18ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   a syscall.  So, to check this, main creates a new process, which
19ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   attempts to grow the stack once all the threads have been created
20ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   and exited.  main() itself is blocked waiting for the child
21ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   process.
22ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
23ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   Oh, and this test also makes sure that thread resources are cleaned
24ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown   up properly.
25ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown */
26ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <pthread.h>
27ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <stdio.h>
28ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <string.h>
29ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <stdlib.h>
30ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <unistd.h>
31ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <signal.h>
32ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#include <sys/wait.h>
33ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
34ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic int grower;
35ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
36ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void handler(int sig)
37ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{
38ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown}
39ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
40ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void *thr(void *v)
41ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{
42ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	return 0;
43ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown}
44ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
45ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown#define FRAME 4096
46ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
47ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void grow(int depth)
48ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{
49ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	volatile char frame[FRAME];
50ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
51ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	memset((char *)frame, 0xff, sizeof(frame));
52ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
53ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	if (depth > 1)
54ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		grow(depth-1);
55ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown}
56ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
57ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownstatic void *maker(void *v)
58ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{
59ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	int i;
60ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
61ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sleep(1);
62ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
63ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	/* Create lots of threads */
64ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	printf("creating threads...\n");
65ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	for(i = 0; i < 1300; i++) {
66ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		pthread_t t;
67ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		int ret;
68ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
69ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		if (i % 100 == 0)
70ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown			printf("%d...\n", i);
71ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
72ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		ret = pthread_create(&t, NULL, thr, NULL);
73ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		if (ret) {
74ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown			printf("pthread_create failed: %s\n", strerror(ret));
75ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown			exit(1);
76ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		}
77ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
78ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		ret = pthread_join(t, NULL);
79ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		if (ret) {
80ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown			printf("pthread_join failed: %s\n", strerror(ret));
81ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown			exit(1);
82ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		}
83ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	}
84ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
85ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	kill(grower, SIGUSR1);
86ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
87ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	return NULL;
88ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown}
89ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
90ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brownint main()
91ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown{
92ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	pthread_t pth;
93ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sigset_t mask;
94ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	int status;
95ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	struct sigaction sa;
96ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
97ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sigemptyset(&mask);
98ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sigaddset(&mask, SIGCHLD);
99ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sigprocmask(SIG_BLOCK, &mask, NULL);
100ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
101ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sa.sa_handler = handler;
102ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sa.sa_flags = 0;
103ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sigfillset(&sa.sa_mask);
104ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	sigaction(SIGUSR1, &sa, NULL);
105ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
106ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	grower = fork();
107ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
108ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	if (grower == -1) {
109ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		perror("fork");
110ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		exit(1);
111ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	}
112ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
113ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	if (grower == 0) {
114ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		pause();	/* child - wait for SIGUSR1 */
115ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		grow(10);
116ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		printf("stack grew OK\n");
117ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		exit(0);
118ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	}
119ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
120ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	pthread_create(&pth, NULL, maker, NULL);
121ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
122ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	/* wait for child */
123ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	if (waitpid(grower, &status, 0) != grower)
124ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		printf("FAILED\n");
125ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	else if (WIFEXITED(status) && WEXITSTATUS(status) == 0)
126ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		printf("PASS: child OK\n");
127ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	else
128ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown		printf("FAILED: exit status=%d\n", status);
129ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
130ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	pthread_join(pth, NULL);
131ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown
132ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown	return 0;
133ed07e00d438c74b7a23c01bfffde77e3968305e4Jeff Brown}
134