NetBSD/regress/sys/kern/mutex2/test_mutex2.c

151 lines
3.9 KiB
C

/* $NetBSD: test_mutex2.c,v 1.3 2007/02/05 22:48:02 ad Exp $ */
/*-
* Copyright (c) 2007 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Andrew Doran.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the NetBSD
* Foundation, Inc. and its contributors.
* 4. Neither the name of The NetBSD Foundation nor the names of its
* contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
* ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: test_mutex2.c,v 1.3 2007/02/05 22:48:02 ad Exp $");
#include <sys/param.h>
#include <sys/ioctl.h>
#include <sys/systm.h>
#include <sys/proc.h>
#include <sys/kthread.h>
#include <sys/kernel.h>
#define NTHREADS 32
#define SECONDS 60
int testcall(struct lwp *, void *, register_t *);
void thread1(void *);
void thread_exit(int);
void thread_enter(int *);
struct proc *test_threads[NTHREADS];
kmutex_t test_mutex;
kcondvar_t test_cv;
int test_count;
int test_exit;
static int primes[NTHREADS] = {
131, 137, 139, 149,
151, 157, 163, 167,
173, 179, 181, 191,
193, 197, 199, 211,
223, 227, 229, 233,
239, 241, 251, 257,
263, 269, 271, 277,
281, 283, 293, 307,
};
void
thread_enter(int *nlocks)
{
struct lwp *l = curlwp;
KERNEL_UNLOCK_ALL(l, nlocks);
lwp_lock(l);
l->l_usrpri = MAXPRI;
lwp_changepri(l, MAXPRI);
lwp_unlock(l);
}
void
thread_exit(int nlocks)
{
mutex_enter(&test_mutex);
if (--test_count == 0)
cv_signal(&test_cv);
mutex_exit(&test_mutex);
KERNEL_LOCK(nlocks, curlwp);
kthread_exit(0);
}
void
thread1(void *cookie)
{
int count, nlocks;
thread_enter(&nlocks);
for (count = 0; !test_exit; count++) {
mutex_enter(&test_mutex);
if ((count % *(int *)cookie) == 0)
yield();
mutex_exit(&test_mutex);
if ((count % curproc->p_pid) == 0)
yield();
}
thread_exit(nlocks);
}
int
testcall(struct lwp *l, void *uap, register_t *retval)
{
int i;
mutex_init(&test_mutex, MUTEX_DEFAULT, IPL_NONE);
cv_init(&test_cv, "testcv");
printf("test: creating threads\n");
test_count = NTHREADS;
test_exit = 0;
for (i = 0; i < test_count; i++)
kthread_create1(thread1, &primes[i], &test_threads[i],
"thread%d", i);
printf("test: sleeping\n");
mutex_enter(&test_mutex);
while (test_count != 0) {
(void)cv_timedwait(&test_cv, &test_mutex, hz * SECONDS);
test_exit = 1;
}
mutex_exit(&test_mutex);
printf("test: finished\n");
cv_destroy(&test_cv);
mutex_destroy(&test_mutex);
return 0;
}