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
tinythread(void * arg)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
rumptest_tsleep()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, ¬bigl[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