Home | History | Annotate | Line # | Download | only in kernspace 
      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