1 /* $NetBSD: t_getenv_thread.c,v 1.2 2012/03/15 02:02:23 joerg Exp $ */
2
3 /*-
4 * Copyright (c) 2010 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Matthias Scheler.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31 #include <sys/cdefs.h>
32 __RCSID("$NetBSD: t_getenv_thread.c,v 1.2 2012/03/15 02:02:23 joerg Exp $");
33
34 #include <atf-c.h>
35 #include <errno.h>
36 #include <pthread.h>
37 #include <stdio.h>
38 #include <stdlib.h>
39 #include <string.h>
40 #include <time.h>
41
42 #define THREADED_NUM_THREADS 8
43 #define THREADED_NUM_VARS 16
44 #define THREADED_VAR_NAME "THREADED%zu"
45 #define THREADED_RUN_TIME 10
46
47 static void *thread_getenv_r(void *);
48 static void *thread_putenv(void *);
49 static void *thread_setenv(void *);
50 static void *thread_unsetenv(void *);
51
52 static void *
thread_getenv_r(void * arg)53 thread_getenv_r(void *arg)
54 {
55 time_t endtime;
56
57 endtime = *(time_t *)arg;
58 do {
59 size_t i;
60 char name[32], value[128];
61
62 i = lrand48() % THREADED_NUM_VARS;
63 (void)snprintf(name, sizeof(name), THREADED_VAR_NAME, i);
64
65 if (getenv_r(name, value, sizeof(value)) == -1) {
66 ATF_CHECK(errno == ENOENT);
67 }
68 } while (time(NULL) < endtime);
69
70 return NULL;
71 }
72
73
74 static void *
thread_putenv(void * arg)75 thread_putenv(void *arg)
76 {
77 time_t endtime;
78 size_t i;
79 static char vars[THREADED_NUM_VARS][128];
80
81 for (i = 0; i < THREADED_NUM_VARS; i++) {
82 (void)snprintf(vars[i], sizeof(vars[i]),
83 THREADED_VAR_NAME "=putenv %ld", i, lrand48());
84 }
85
86 endtime = *(time_t *)arg;
87 do {
88 char name[128];
89
90 i = lrand48() % THREADED_NUM_VARS;
91 (void)strlcpy(name, vars[i], sizeof(name));
92 *strchr(name, '=') = '\0';
93
94 ATF_CHECK(unsetenv(name) != -1);
95 ATF_CHECK(putenv(vars[i]) != -1);
96 } while (time(NULL) < endtime);
97
98 return NULL;
99 }
100
101 static void *
thread_setenv(void * arg)102 thread_setenv(void *arg)
103 {
104 time_t endtime;
105
106 endtime = *(time_t *)arg;
107 do {
108 size_t i;
109 char name[32], value[64];
110
111 i = lrand48() % THREADED_NUM_VARS;
112 (void)snprintf(name, sizeof(name), THREADED_VAR_NAME, i);
113 (void)snprintf(value, sizeof(value), "setenv %ld", lrand48());
114
115 ATF_CHECK(setenv(name, value, 1) != -1);
116 } while (time(NULL) < endtime);
117
118 return NULL;
119 }
120
121 static void *
thread_unsetenv(void * arg)122 thread_unsetenv(void *arg)
123 {
124 time_t endtime;
125
126 endtime = *(time_t *)arg;
127 do {
128 size_t i;
129 char name[32];
130
131 i = lrand48() % THREADED_NUM_VARS;
132 (void)snprintf(name, sizeof(name), THREADED_VAR_NAME, i);
133
134 ATF_CHECK(unsetenv(name) != -1);
135 } while (time(NULL) < endtime);
136
137 return NULL;
138 }
139
140 ATF_TC(getenv_r_thread);
ATF_TC_HEAD(getenv_r_thread,tc)141 ATF_TC_HEAD(getenv_r_thread, tc)
142 {
143
144 atf_tc_set_md_var(tc, "descr", "Test getenv_r(3) with threads");
145 atf_tc_set_md_var(tc, "timeout", "%d", THREADED_RUN_TIME + 5);
146 }
147
ATF_TC_BODY(getenv_r_thread,tc)148 ATF_TC_BODY(getenv_r_thread, tc)
149 {
150 pthread_t threads[THREADED_NUM_THREADS];
151 time_t endtime;
152 size_t i, j;
153
154 endtime = time(NULL) + THREADED_RUN_TIME;
155
156 for (i = j = 0; j < 2; j++) {
157
158 ATF_CHECK(pthread_create(&threads[i++], NULL, thread_getenv_r,
159 &endtime) == 0);
160 }
161
162 for (j = 0; j < i; j++)
163 ATF_CHECK(pthread_join(threads[j], NULL) == 0);
164 }
165
166 ATF_TC(putenv_thread);
ATF_TC_HEAD(putenv_thread,tc)167 ATF_TC_HEAD(putenv_thread, tc)
168 {
169 atf_tc_set_md_var(tc, "descr", "Test concurrent access by putenv(3)");
170 atf_tc_set_md_var(tc, "timeout", "%d", THREADED_RUN_TIME + 5);
171 }
172
ATF_TC_BODY(putenv_thread,tc)173 ATF_TC_BODY(putenv_thread, tc)
174 {
175 pthread_t threads[THREADED_NUM_THREADS];
176 time_t endtime;
177 size_t i, j;
178
179 endtime = time(NULL) + THREADED_RUN_TIME;
180
181 for (i = j = 0; j < 2; j++) {
182
183 ATF_CHECK(pthread_create(&threads[i++], NULL, thread_putenv,
184 &endtime) == 0);
185 }
186
187 for (j = 0; j < i; j++)
188 ATF_CHECK(pthread_join(threads[j], NULL) == 0);
189 }
190
191 ATF_TC(setenv_thread);
ATF_TC_HEAD(setenv_thread,tc)192 ATF_TC_HEAD(setenv_thread, tc)
193 {
194 atf_tc_set_md_var(tc, "descr", "Test concurrent access by setenv(3)");
195 atf_tc_set_md_var(tc, "timeout", "%d", THREADED_RUN_TIME + 5);
196 }
197
ATF_TC_BODY(setenv_thread,tc)198 ATF_TC_BODY(setenv_thread, tc)
199 {
200 pthread_t threads[THREADED_NUM_THREADS];
201 time_t endtime;
202 size_t i, j;
203
204 endtime = time(NULL) + THREADED_RUN_TIME;
205
206 for (i = j = 0; j < 2; j++) {
207
208 ATF_CHECK(pthread_create(&threads[i++], NULL, thread_setenv,
209 &endtime) == 0);
210 }
211
212 for (j = 0; j < i; j++)
213 ATF_CHECK(pthread_join(threads[j], NULL) == 0);
214 }
215
216 ATF_TC(unsetenv_thread);
ATF_TC_HEAD(unsetenv_thread,tc)217 ATF_TC_HEAD(unsetenv_thread, tc)
218 {
219 atf_tc_set_md_var(tc, "descr", "Test unsetenv(3) with threads");
220 atf_tc_set_md_var(tc, "timeout", "%d", THREADED_RUN_TIME + 5);
221 }
222
ATF_TC_BODY(unsetenv_thread,tc)223 ATF_TC_BODY(unsetenv_thread, tc)
224 {
225 pthread_t threads[THREADED_NUM_THREADS];
226 time_t endtime;
227 size_t i, j;
228
229 endtime = time(NULL) + THREADED_RUN_TIME;
230
231 for (i = j = 0; j < 2; j++) {
232
233 ATF_CHECK(pthread_create(&threads[i++], NULL, thread_unsetenv,
234 &endtime) == 0);
235 }
236
237 for (j = 0; j < i; j++)
238 ATF_CHECK(pthread_join(threads[j], NULL) == 0);
239 }
240
ATF_TP_ADD_TCS(tp)241 ATF_TP_ADD_TCS(tp)
242 {
243
244 ATF_TP_ADD_TC(tp, getenv_r_thread);
245 ATF_TP_ADD_TC(tp, putenv_thread);
246 ATF_TP_ADD_TC(tp, setenv_thread);
247 ATF_TP_ADD_TC(tp, unsetenv_thread);
248
249 return atf_no_error();
250 }
251