1/* $NetBSD: t_once.c,v 1.2 2017/08/25 22:59:47 ginsbach Exp $ */
2
3/*
4 * Copyright (c) 2008 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29#include <sys/cdefs.h>
30__COPYRIGHT("@(#) Copyright (c) 2008\
31 The NetBSD Foundation, inc. All rights reserved.");
32__RCSID("$NetBSD: t_once.c,v 1.2 2017/08/25 22:59:47 ginsbach Exp $");
33
34#include <sys/time.h>
35#include <pthread.h>
36#include <signal.h>
37#include <stdio.h>
38#include <stdlib.h>
39
40#include <atf-c.h>
41
42#include "h_common.h"
43
44static pthread_once_t once = PTHREAD_ONCE_INIT;
45static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
46static int x;
47
48#define NTHREADS 25
49
50static void
51ofunc(void)
52{
53	printf("Variable x has value %d\n", x);
54	x++;
55}
56
57ATF_TC(once1);
58ATF_TC_HEAD(once1, tc)
59{
60	atf_tc_set_md_var(tc, "descr", "Checks pthread_once()");
61}
62ATF_TC_BODY(once1, tc)
63{
64
65	printf("1: Test 1 of pthread_once()\n");
66
67	PTHREAD_REQUIRE(pthread_once(&once, ofunc));
68	PTHREAD_REQUIRE(pthread_once(&once, ofunc));
69
70	printf("1: X has value %d\n",x );
71	ATF_REQUIRE_EQ(x, 1);
72}
73
74static void
75once2_ofunc(void)
76{
77	x++;
78	printf("ofunc: Variable x has value %d\n", x);
79	x++;
80}
81
82static void *
83once2_threadfunc(void *arg)
84{
85	int num;
86
87	PTHREAD_REQUIRE(pthread_once(&once, once2_ofunc));
88
89	num = *(int *)arg;
90	printf("Thread %d sees x with value %d\n", num, x);
91	ATF_REQUIRE_EQ(x, 2);
92
93	return NULL;
94}
95
96ATF_TC(once2);
97ATF_TC_HEAD(once2, tc)
98{
99	atf_tc_set_md_var(tc, "descr", "Checks pthread_once()");
100}
101ATF_TC_BODY(once2, tc)
102{
103	pthread_t  threads[NTHREADS];
104	int id[NTHREADS];
105	int i;
106
107	printf("1: Test 2 of pthread_once()\n");
108
109	for (i=0; i < NTHREADS; i++) {
110		id[i] = i;
111		PTHREAD_REQUIRE(pthread_create(&threads[i], NULL, once2_threadfunc, &id[i]));
112	}
113
114	for (i=0; i < NTHREADS; i++)
115		PTHREAD_REQUIRE(pthread_join(threads[i], NULL));
116
117	printf("1: X has value %d\n",x );
118	ATF_REQUIRE_EQ(x, 2);
119}
120
121static void
122once3_cleanup(void *m)
123{
124	pthread_mutex_t *mu = m;
125
126	PTHREAD_REQUIRE(pthread_mutex_unlock(mu));
127}
128
129static void
130once3_ofunc(void)
131{
132	pthread_testcancel();
133}
134
135static void *
136once3_threadfunc(void *arg)
137{
138	PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
139	pthread_cleanup_push(once3_cleanup, &mutex);
140	PTHREAD_REQUIRE(pthread_once(&once, once3_ofunc));
141	pthread_cleanup_pop(1);
142
143	return NULL;
144}
145
146static void
147handler(int sig, siginfo_t *info, void *ctx)
148{
149	atf_tc_fail("Signal handler was called; "
150		"main thread deadlocked in pthread_once()");
151}
152
153ATF_TC(once3);
154ATF_TC_HEAD(once3, tc)
155{
156	atf_tc_set_md_var(tc, "descr", "Checks pthread_once()");
157}
158ATF_TC_BODY(once3, tc)
159{
160	pthread_t thread;
161	struct sigaction act;
162	struct itimerval it;
163	printf("Test 3 of pthread_once() (test versus cancellation)\n");
164
165	act.sa_sigaction = handler;
166	sigemptyset(&act.sa_mask);
167	act.sa_flags = SA_SIGINFO;
168	sigaction(SIGALRM, &act, NULL);
169
170	timerclear(&it.it_value);
171	it.it_value.tv_usec = 500000;
172	timerclear(&it.it_interval);
173	setitimer(ITIMER_REAL, &it, NULL);
174
175	PTHREAD_REQUIRE(pthread_mutex_lock(&mutex));
176	PTHREAD_REQUIRE(pthread_create(&thread, NULL, once3_threadfunc, NULL));
177	PTHREAD_REQUIRE(pthread_cancel(thread));
178	PTHREAD_REQUIRE(pthread_mutex_unlock(&mutex));
179	PTHREAD_REQUIRE(pthread_join(thread, NULL));
180
181	PTHREAD_REQUIRE(pthread_once(&once, ofunc));
182
183	/* Cancel timer */
184	timerclear(&it.it_value);
185	setitimer(ITIMER_REAL, &it, NULL);
186
187	printf("Test succeeded\n");
188}
189
190ATF_TP_ADD_TCS(tp)
191{
192	ATF_TP_ADD_TC(tp, once1);
193	ATF_TP_ADD_TC(tp, once2);
194	ATF_TP_ADD_TC(tp, once3);
195
196	return atf_no_error();
197}
198