xref: /minix3/tests/rump/kernspace/tsleep.c (revision b5e2faaaaf60a8b9a02f8d72f64caa56a87eb312)
1 /*	$NetBSD: tsleep.c,v 1.3 2012/01/28 12:23:56 rmind Exp $	*/
2 
3 /*-
4  * Copyright (c) 2010 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
17  * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
18  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20  * IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27  * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29 
30 #include <sys/cdefs.h>
31 #if !defined(lint)
32 __RCSID("$NetBSD: tsleep.c,v 1.3 2012/01/28 12:23:56 rmind Exp $");
33 #endif /* !lint */
34 
35 #include <sys/param.h>
36 #include <sys/kernel.h>
37 #include <sys/kthread.h>
38 #include <sys/proc.h>
39 #include <sys/simplelock.h>
40 
41 #include "kernspace.h"
42 
43 #define NTHREADS 10
44 
45 /*
46  * mpsafe thread.  need dedicated interlock
47  */
48 static kmutex_t mymtx;
49 
50 static void
51 tinythread(void *arg)
52 {
53 	static int wakeups;
54 	int i, rv;
55 	bool relock = ((uintptr_t)arg % 2) == 0;
56 
57 	for (i = 0; i < 1000; i++) {
58 		mutex_enter(&mymtx);
59 		wakeup(tinythread);
60 		if (wakeups >= NTHREADS-1) {
61 			mutex_exit(&mymtx);
62 			break;
63 		}
64 		rv = mtsleep(tinythread, relock ? 0 : PNORELOCK,
65 		    "haa", 0, &mymtx);
66 		if (relock)
67 			mutex_exit(&mymtx);
68 		if (rv != 0)
69 			panic("mtsleep failed");
70 	}
71 
72 	mutex_enter(&mymtx);
73 	wakeups++;
74 	wakeup(tinythread);
75 
76 	rv = mtsleep(rumptest_tsleep, PNORELOCK, "kepuli", 1, &mymtx);
77 	if (rv != EWOULDBLOCK)
78 		panic("mtsleep unexpected return value %d", rv);
79 
80 	kthread_exit(0);
81 }
82 
83 void
84 rumptest_tsleep()
85 {
86 	struct lwp *notbigl[NTHREADS];
87 	int rv, i;
88 
89 	mutex_init(&mymtx, MUTEX_DEFAULT, IPL_NONE);
90 
91 	for (i = 0; i < NTHREADS; i++) {
92 		rv = kthread_create(PRI_NONE, KTHREAD_MUSTJOIN| KTHREAD_MPSAFE,
93 		    NULL, tinythread, (void *)(uintptr_t)i, &notbigl[i], "nb");
94 		if (rv)
95 			panic("thread create failed: %d", rv);
96 	}
97 
98 	for (i = 0; i < NTHREADS; i++) {
99 		kthread_join(notbigl[i]);
100 	}
101 }
102