1/* Copyright (C) 2003-2022 Free Software Foundation, Inc.
2 This file is part of the GNU C Library.
3
4 The GNU C Library is free software; you can redistribute it and/or
5 modify it under the terms of the GNU Lesser General Public
6 License as published by the Free Software Foundation; either
7 version 2.1 of the License, or (at your option) any later version.
8
9 The GNU C Library is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 Lesser General Public License for more details.
13
14 You should have received a copy of the GNU Lesser General Public
15 License along with the GNU C Library; if not, see
16 <https://www.gnu.org/licenses/>. */
17
18#include <errno.h>
19#include <error.h>
20#include <limits.h>
21#include <pthread.h>
22#include <stdio.h>
23#include <stdlib.h>
24#include <ucontext.h>
25#include <support/support.h>
26
27#define N 4
28#if __WORDSIZE == 64
29#define GUARD_PATTERN 0xdeadbeafdeadbeaf
30#else
31#define GUARD_PATTERN 0xdeadbeaf
32#endif
33
34typedef struct {
35 ucontext_t uctx;
36 unsigned long guard[3];
37 } tst_context_t;
38
39static char *stacks[N];
40static size_t stack_size;
41static tst_context_t ctx[N][2];
42static volatile int failures;
43
44
45static void
46fct (long int n)
47{
48 char on_stack[1];
49
50 /* Just to use the thread local descriptor. */
51 printf (format: "%ld: in %s now, on_stack = %p\n", n, __FUNCTION__, on_stack);
52 errno = 0;
53
54 if (ctx[n][1].uctx.uc_link != &ctx[n][0].uctx)
55 {
56 printf (format: "context[%ld][1] uc_link damaged, = %p\n", n,
57 ctx[n][1].uctx.uc_link);
58 exit (1);
59 }
60
61 if ((ctx[n][0].guard[0] != GUARD_PATTERN)
62 || (ctx[n][0].guard[1] != GUARD_PATTERN)
63 || (ctx[n][0].guard[2] != GUARD_PATTERN))
64 {
65 printf (format: "%ld: %s context[0] overflow detected!\n", n, __FUNCTION__);
66 ++failures;
67 }
68
69 if ((ctx[n][1].guard[0] != GUARD_PATTERN)
70 || (ctx[n][1].guard[1] != GUARD_PATTERN)
71 || (ctx[n][1].guard[2] != GUARD_PATTERN))
72 {
73 printf (format: "%ld: %s context[1] overflow detected!\n", n, __FUNCTION__);
74 ++failures;
75 }
76
77 if (n < 0 || n >= N)
78 {
79 printf (format: "%ld out of range\n", n);
80 exit (1);
81 }
82
83 if (on_stack < stacks[n] || on_stack >= stacks[n] + stack_size)
84 {
85 printf (format: "%ld: on_stack not on appropriate stack\n", n);
86 exit (1);
87 }
88}
89
90
91static void *
92tf (void *arg)
93{
94 int n = (int) (long int) arg;
95
96 ctx[n][0].guard[0] = GUARD_PATTERN;
97 ctx[n][0].guard[1] = GUARD_PATTERN;
98 ctx[n][0].guard[2] = GUARD_PATTERN;
99
100 ctx[n][1].guard[0] = GUARD_PATTERN;
101 ctx[n][1].guard[1] = GUARD_PATTERN;
102 ctx[n][1].guard[2] = GUARD_PATTERN;
103
104 if (getcontext (ucp: &ctx[n][1].uctx) != 0)
105 {
106 printf (format: "%d: cannot get context: %m\n", n);
107 exit (1);
108 }
109
110 printf (format: "%d: %s: before makecontext\n", n, __FUNCTION__);
111
112 ctx[n][1].uctx.uc_stack.ss_sp = stacks[n];
113 ctx[n][1].uctx.uc_stack.ss_size = stack_size;
114 ctx[n][1].uctx.uc_link = &ctx[n][0].uctx;
115 makecontext (ucp: &ctx[n][1].uctx, func: (void (*) (void)) fct, argc: 1, (long int) n);
116
117 printf (format: "%d: %s: before swapcontext\n", n, __FUNCTION__);
118
119 if (swapcontext (oucp: &ctx[n][0].uctx, ucp: &ctx[n][1].uctx) != 0)
120 {
121 ++failures;
122 printf (format: "%d: %s: swapcontext failed\n", n, __FUNCTION__);
123 }
124 else
125 printf (format: "%d: back in %s\n", n, __FUNCTION__);
126
127 return NULL;
128}
129
130
131static volatile int global;
132
133
134static int
135do_test (void)
136{
137 int n;
138 pthread_t th[N];
139 ucontext_t mctx;
140
141 stack_size = 2 * PTHREAD_STACK_MIN;
142 for (int i = 0; i < N; i++)
143 stacks[i] = xmalloc (n: stack_size);
144
145 puts (s: "making contexts");
146 if (getcontext (ucp: &mctx) != 0)
147 {
148 if (errno == ENOSYS)
149 {
150 puts (s: "context handling not supported");
151 exit (0);
152 }
153
154 printf (format: "%s: getcontext: %m\n", __FUNCTION__);
155 exit (1);
156 }
157
158 /* Play some tricks with this context. */
159 if (++global == 1)
160 if (setcontext (&mctx) != 0)
161 {
162 puts (s: "setcontext failed");
163 exit (1);
164 }
165 if (global != 2)
166 {
167 puts (s: "global not incremented twice");
168 exit (1);
169 }
170 puts (s: "global OK");
171
172 pthread_attr_t at;
173
174 if (pthread_attr_init (attr: &at) != 0)
175 {
176 puts (s: "attr_init failed");
177 return 1;
178 }
179
180 if (pthread_attr_setstacksize (attr: &at, stacksize: 1 * 1024 * 1024) != 0)
181 {
182 puts (s: "attr_setstacksize failed");
183 return 1;
184 }
185
186 for (n = 0; n < N; ++n)
187 if (pthread_create (newthread: &th[n], attr: &at, start_routine: tf, arg: (void *) (long int) n) != 0)
188 {
189 puts (s: "create failed");
190 exit (1);
191 }
192
193 if (pthread_attr_destroy (attr: &at) != 0)
194 {
195 puts (s: "attr_destroy failed");
196 return 1;
197 }
198
199 for (n = 0; n < N; ++n)
200 if (pthread_join (th: th[n], NULL) != 0)
201 {
202 puts (s: "join failed");
203 exit (1);
204 }
205
206 for (int i = 0; i < N; i++)
207 free (ptr: stacks[i]);
208
209 return failures;
210}
211
212#define TEST_FUNCTION do_test ()
213#include "../test-skeleton.c"
214

source code of glibc/nptl/tst-context1.c