Home | History | Annotate | Line # | Download | only in kernspace
tsleep.c revision 1.2.2.1 
      1 /*	$NetBSD: tsleep.c,v 1.2.2.1 2012/04/17 00:09:15 yamt 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.2.2.1 2012/04/17 00:09:15 yamt 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