xref: /freebsd/contrib/netbsd-tests/rump/kernspace/tsleep.c (revision e6bfd18d21b225af6a0ed67ceeaf1293b7b9eba5)
1 /*	$NetBSD: tsleep.c,v 1.4 2014/03/21 22:18:57 dholland 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.4 2014/03/21 22:18:57 dholland 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 
40 #include "kernspace.h"
41 
42 #define NTHREADS 10
43 
44 /*
45  * mpsafe thread.  need dedicated interlock
46  */
47 static kmutex_t mymtx;
48 
49 static void
50 tinythread(void *arg)
51 {
52 	static int wakeups;
53 	int i, rv;
54 	bool relock = ((uintptr_t)arg % 2) == 0;
55 
56 	for (i = 0; i < 1000; i++) {
57 		mutex_enter(&mymtx);
58 		wakeup(tinythread);
59 		if (wakeups >= NTHREADS-1) {
60 			mutex_exit(&mymtx);
61 			break;
62 		}
63 		rv = mtsleep(tinythread, relock ? 0 : PNORELOCK,
64 		    "haa", 0, &mymtx);
65 		if (relock)
66 			mutex_exit(&mymtx);
67 		if (rv != 0)
68 			panic("mtsleep failed");
69 	}
70 
71 	mutex_enter(&mymtx);
72 	wakeups++;
73 	wakeup(tinythread);
74 
75 	rv = mtsleep(rumptest_tsleep, PNORELOCK, "kepuli", 1, &mymtx);
76 	if (rv != EWOULDBLOCK)
77 		panic("mtsleep unexpected return value %d", rv);
78 
79 	kthread_exit(0);
80 }
81 
82 void
83 rumptest_tsleep()
84 {
85 	struct lwp *notbigl[NTHREADS];
86 	int rv, i;
87 
88 	mutex_init(&mymtx, MUTEX_DEFAULT, IPL_NONE);
89 
90 	for (i = 0; i < NTHREADS; i++) {
91 		rv = kthread_create(PRI_NONE, KTHREAD_MUSTJOIN| KTHREAD_MPSAFE,
92 		    NULL, tinythread, (void *)(uintptr_t)i, &notbigl[i], "nb");
93 		if (rv)
94 			panic("thread create failed: %d", rv);
95 	}
96 
97 	for (i = 0; i < NTHREADS; i++) {
98 		kthread_join(notbigl[i]);
99 	}
100 }
101