summaryrefslogtreecommitdiffstats
path: root/tools/testing/selftests/powerpc/mm/stack_expansion_signal.c
blob: c8b32a29e27476cc17dc5741530ba5fbf77b3dda (plain)
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
// SPDX-License-Identifier: GPL-2.0
/*
 * Test that signal delivery is able to expand the stack segment without
 * triggering a SEGV.
 *
 * Based on test code by Tom Lane.
 */

#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/types.h>
#include <unistd.h>

#include "../pmu/lib.h"
#include "utils.h"

#define _KB (1024)
#define _MB (1024 * 1024)

static char *stack_base_ptr;
static char *stack_top_ptr;

static volatile sig_atomic_t sig_occurred = 0;

static void sigusr1_handler(int signal_arg)
{
	sig_occurred = 1;
}

static int consume_stack(unsigned int stack_size, union pipe write_pipe)
{
	char stack_cur;

	if ((stack_base_ptr - &stack_cur) < stack_size)
		return consume_stack(stack_size, write_pipe);
	else {
		stack_top_ptr = &stack_cur;

		FAIL_IF(notify_parent(write_pipe));

		while (!sig_occurred)
			barrier();
	}

	return 0;
}

static int child(unsigned int stack_size, union pipe write_pipe)
{
	struct sigaction act;
	char stack_base;

	act.sa_handler = sigusr1_handler;
	sigemptyset(&act.sa_mask);
	act.sa_flags = 0;
	if (sigaction(SIGUSR1, &act, NULL) < 0)
		err(1, "sigaction");

	stack_base_ptr = (char *) (((size_t) &stack_base + 65535) & ~65535UL);

	FAIL_IF(consume_stack(stack_size, write_pipe));

	printf("size 0x%06x: OK, stack base %p top %p (%zx used)\n",
		stack_size, stack_base_ptr, stack_top_ptr,
		stack_base_ptr - stack_top_ptr);

	return 0;
}

static int test_one_size(unsigned int stack_size)
{
	union pipe read_pipe, write_pipe;
	pid_t pid;

	FAIL_IF(pipe(read_pipe.fds) == -1);
	FAIL_IF(pipe(write_pipe.fds) == -1);

	pid = fork();
	if (pid == 0) {
		close(read_pipe.read_fd);
		close(write_pipe.write_fd);
		exit(child(stack_size, read_pipe));
	}

	close(read_pipe.write_fd);
	close(write_pipe.read_fd);
	FAIL_IF(sync_with_child(read_pipe, write_pipe));

	kill(pid, SIGUSR1);

	FAIL_IF(wait_for_child(pid));

	close(read_pipe.read_fd);
	close(write_pipe.write_fd);

	return 0;
}

int test(void)
{
	unsigned int i, size;

	// Test with used stack from 1MB - 64K to 1MB + 64K
	// Increment by 64 to get more coverage of odd sizes
	for (i = 0; i < (128 * _KB); i += 64) {
		size = i + (1 * _MB) - (64 * _KB);
		FAIL_IF(test_one_size(size));
	}

	return 0;
}

int main(void)
{
	return test_harness(test, "stack_expansion_signal");
}