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
40#include "kernspace.h"
41
42#define NTHREADS 10
43
44/*
45 * mpsafe thread.  need dedicated interlock
46 */
47static kmutex_t mymtx;
48
49static void
50tinythread(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
82void
83rumptest_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