WebSVN – HelenOS – Diff – /trunk/kernel/test/synch/rwlock4.c


/*
 * Copyright (c) 2001-2004 Jakub Jermar
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - 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.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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 <test.h>
#include <arch.h>
#include <atomic.h>
#include <print.h>
#include <proc/thread.h>
#include <arch/types.h>
#include <arch/context.h>
#include <context.h>
 
#include <synch/waitq.h>
#include <synch/rwlock.h>
#include <synch/synch.h>
#include <synch/spinlock.h>
 
#define READERS     50
#define WRITERS     50
 
static atomic_t thread_count;
static rwlock_t rwlock;
static atomic_t threads_fault;
static bool sh_quiet;
 
SPINLOCK_INITIALIZE(rw_lock);
 
static waitq_t can_start;
 
static uint32_t seed = 0xdeadbeef;
 
static uint32_t random(uint32_t max)
{
    uint32_t rc;
 
    spinlock_lock(&rw_lock);   
    rc = seed % max;
    seed = (((seed<<2) ^ (seed>>2)) * 487) + rc;
    spinlock_unlock(&rw_lock);
    return rc;
}
 
static void writer(void *arg)
{
    int rc, to;
    thread_detach(THREAD);
    waitq_sleep(&can_start);
 
    to = random(40000);
   
    if (!sh_quiet)
        printf("cpu%u, tid %" PRIu64 " w+ (%d)\n", CPU->id, THREAD->tid, to);
   
    rc = rwlock_write_lock_timeout(&rwlock, to);
    if (SYNCH_FAILED(rc)) {
        if (!sh_quiet)
            printf("cpu%u, tid %" PRIu64 " w!\n", CPU->id, THREAD->tid);
        atomic_dec(&thread_count);
        return;
    }
   
    if (!sh_quiet)
        printf("cpu%u, tid %" PRIu64 " w=\n", CPU->id, THREAD->tid);
 
    if (rwlock.readers_in) {
        if (!sh_quiet)
            printf("Oops.");
        atomic_inc(&threads_fault);
        atomic_dec(&thread_count);
        return;
    }
    thread_usleep(random(1000000));
    if (rwlock.readers_in) {
        if (!sh_quiet)
            printf("Oops.");   
        atomic_inc(&threads_fault);
        atomic_dec(&thread_count);
        return;
    }
 
    rwlock_write_unlock(&rwlock);
   
    if (!sh_quiet)
        printf("cpu%u, tid %" PRIu64 " w-\n", CPU->id, THREAD->tid);
    atomic_dec(&thread_count);
}
 
static void reader(void *arg)
{
    int rc, to;
    thread_detach(THREAD);
    waitq_sleep(&can_start);
   
    to = random(2000);
   
    if (!sh_quiet)
        printf("cpu%u, tid %" PRIu64 " r+ (%d)\n", CPU->id, THREAD->tid, to);
   
    rc = rwlock_read_lock_timeout(&rwlock, to);
    if (SYNCH_FAILED(rc)) {
        if (!sh_quiet)
            printf("cpu%u, tid %" PRIu64 " r!\n", CPU->id, THREAD->tid);
        atomic_dec(&thread_count);
        return;
    }
   
    if (!sh_quiet)
        printf("cpu%u, tid %" PRIu64 " r=\n", CPU->id, THREAD->tid);
   
    thread_usleep(30000);
    rwlock_read_unlock(&rwlock);
   
    if (!sh_quiet)
        printf("cpu%u, tid %" PRIu64 " r-\n", CPU->id, THREAD->tid);
    atomic_dec(&thread_count);
}
 
char * test_rwlock4(bool quiet)
{
    context_t ctx;
    uint32_t i;
    sh_quiet = quiet;
   
    waitq_initialize(&can_start);
    rwlock_initialize(&rwlock);
    atomic_set(&threads_fault, 0);
   
    uint32_t rd = random(7) + 1;
    uint32_t wr = random(5) + 1;
   
    atomic_set(&thread_count, rd + wr);
   
    thread_t *thrd;
   
    context_save(&ctx);
    if (!quiet) {
        printf("sp=%#x, readers_in=%" PRIc "\n", ctx.sp, rwlock.readers_in);
        printf("Creating %" PRIu32 " readers\n", rd);
    }
   
    for (i = 0; i < rd; i++) {
        thrd = thread_create(reader, NULL, TASK, 0, "reader", false);
        if (thrd)
            thread_ready(thrd);
        else if (!quiet)
            printf("Could not create reader %" PRIu32 "\n", i);
    }
 
    if (!quiet)
        printf("Creating %" PRIu32 " writers\n", wr);
   
    for (i = 0; i < wr; i++) {
        thrd = thread_create(writer, NULL, TASK, 0, "writer", false);
        if (thrd)
            thread_ready(thrd);
        else if (!quiet)
            printf("Could not create writer %" PRIu32 "\n", i);
    }
   
    thread_usleep(20000);
    waitq_wakeup(&can_start, WAKEUP_ALL);
   
    while (atomic_get(&thread_count) > 0) {
        if (!quiet)
            printf("Threads left: %ld\n", atomic_get(&thread_count));
        thread_sleep(1);
    }
   
    if (atomic_get(&threads_fault) == 0)
        return NULL;
   
    return "Test failed";
}
 

Rev 2216	Rev 3069
1	/*	1	/*
2	* Copyright (c) 2001-2004 Jakub Jermar	2	* Copyright (c) 2001-2004 Jakub Jermar
3	* All rights reserved.	3	* All rights reserved.
4	*	4	*
5	* Redistribution and use in source and binary forms, with or without	5	* Redistribution and use in source and binary forms, with or without
6	* modification, are permitted provided that the following conditions	6	* modification, are permitted provided that the following conditions
7	* are met:	7	* are met:
8	*	8	*
9	* - Redistributions of source code must retain the above copyright	9	* - Redistributions of source code must retain the above copyright
10	* notice, this list of conditions and the following disclaimer.	10	* notice, this list of conditions and the following disclaimer.
11	* - Redistributions in binary form must reproduce the above copyright	11	* - Redistributions in binary form must reproduce the above copyright
12	* notice, this list of conditions and the following disclaimer in the	12	* notice, this list of conditions and the following disclaimer in the
13	* documentation and/or other materials provided with the distribution.	13	* documentation and/or other materials provided with the distribution.
14	* - The name of the author may not be used to endorse or promote products	14	* - The name of the author may not be used to endorse or promote products
15	* derived from this software without specific prior written permission.	15	* derived from this software without specific prior written permission.
16	*	16	*
17	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR	17	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES	18	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.	19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,	20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT	21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,	22	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY	23	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT	24	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF	25	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.	26	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27	*/	27	*/
28		28
29	#include <test.h>	29	#include <test.h>
30	#include <arch.h>	30	#include <arch.h>
31	#include <atomic.h>	31	#include <atomic.h>
32	#include <print.h>	32	#include <print.h>
33	#include <proc/thread.h>	33	#include <proc/thread.h>
34	#include <arch/types.h>	34	#include <arch/types.h>
35	#include <arch/context.h>	35	#include <arch/context.h>
36	#include <context.h>	36	#include <context.h>
37		37
38	#include <synch/waitq.h>	38	#include <synch/waitq.h>
39	#include <synch/rwlock.h>	39	#include <synch/rwlock.h>
40	#include <synch/synch.h>	40	#include <synch/synch.h>
41	#include <synch/spinlock.h>	41	#include <synch/spinlock.h>
42		42
43	#define READERS 50	43	#define READERS 50
44	#define WRITERS 50	44	#define WRITERS 50
45		45
46	static atomic_t thread_count;	46	static atomic_t thread_count;
47	static rwlock_t rwlock;	47	static rwlock_t rwlock;
48	static atomic_t threads_fault;	48	static atomic_t threads_fault;
49	static bool sh_quiet;	49	static bool sh_quiet;
50		50
51	SPINLOCK_INITIALIZE(rw_lock);	51	SPINLOCK_INITIALIZE(rw_lock);
52		52
53	static waitq_t can_start;	53	static waitq_t can_start;
54		54
55	static uint32_t seed = 0xdeadbeef;	55	static uint32_t seed = 0xdeadbeef;
56		56
57	static uint32_t random(uint32_t max)	57	static uint32_t random(uint32_t max)
58	{	58	{
59	uint32_t rc;	59	uint32_t rc;
60		60
61	spinlock_lock(&rw_lock);	61	spinlock_lock(&rw_lock);
62	rc = seed % max;	62	rc = seed % max;
63	seed = (((seed<<2) ^ (seed>>2)) * 487) + rc;	63	seed = (((seed<<2) ^ (seed>>2)) * 487) + rc;
64	spinlock_unlock(&rw_lock);	64	spinlock_unlock(&rw_lock);
65	return rc;	65	return rc;
66	}	66	}
67		67
68	static void writer(void *arg)	68	static void writer(void *arg)
69	{	69	{
70	int rc, to;	70	int rc, to;
71	thread_detach(THREAD);	71	thread_detach(THREAD);
72	waitq_sleep(&can_start);	72	waitq_sleep(&can_start);
73		73
74	to = random(40000);	74	to = random(40000);
75		75
76	if (!sh_quiet)	76	if (!sh_quiet)
77	printf("cpu%d, tid %llu w+ (%d)\n", CPU->id, THREAD->tid, to);	77	printf("cpu%u, tid %" PRIu64 " w+ (%d)\n", CPU->id, THREAD->tid, to);
78		78
79	rc = rwlock_write_lock_timeout(&rwlock, to);	79	rc = rwlock_write_lock_timeout(&rwlock, to);
80	if (SYNCH_FAILED(rc)) {	80	if (SYNCH_FAILED(rc)) {
81	if (!sh_quiet)	81	if (!sh_quiet)
82	printf("cpu%d, tid %llu w!\n", CPU->id, THREAD->tid);	82	printf("cpu%u, tid %" PRIu64 " w!\n", CPU->id, THREAD->tid);
83	atomic_dec(&thread_count);	83	atomic_dec(&thread_count);
84	return;	84	return;
85	}	85	}
86		86
87	if (!sh_quiet)	87	if (!sh_quiet)
88	printf("cpu%d, tid %llu w=\n", CPU->id, THREAD->tid);	88	printf("cpu%u, tid %" PRIu64 " w=\n", CPU->id, THREAD->tid);
89		89
90	if (rwlock.readers_in) {	90	if (rwlock.readers_in) {
91	if (!sh_quiet)	91	if (!sh_quiet)
92	printf("Oops.");	92	printf("Oops.");
93	atomic_inc(&threads_fault);	93	atomic_inc(&threads_fault);
94	atomic_dec(&thread_count);	94	atomic_dec(&thread_count);
95	return;	95	return;
96	}	96	}
97	thread_usleep(random(1000000));	97	thread_usleep(random(1000000));
98	if (rwlock.readers_in) {	98	if (rwlock.readers_in) {
99	if (!sh_quiet)	99	if (!sh_quiet)
100	printf("Oops.");	100	printf("Oops.");
101	atomic_inc(&threads_fault);	101	atomic_inc(&threads_fault);
102	atomic_dec(&thread_count);	102	atomic_dec(&thread_count);
103	return;	103	return;
104	}	104	}
105		105
106	rwlock_write_unlock(&rwlock);	106	rwlock_write_unlock(&rwlock);
107		107
108	if (!sh_quiet)	108	if (!sh_quiet)
109	printf("cpu%d, tid %llu w-\n", CPU->id, THREAD->tid);	109	printf("cpu%u, tid %" PRIu64 " w-\n", CPU->id, THREAD->tid);
110	atomic_dec(&thread_count);	110	atomic_dec(&thread_count);
111	}	111	}
112		112
113	static void reader(void *arg)	113	static void reader(void *arg)
114	{	114	{
115	int rc, to;	115	int rc, to;
116	thread_detach(THREAD);	116	thread_detach(THREAD);
117	waitq_sleep(&can_start);	117	waitq_sleep(&can_start);
118		118
119	to = random(2000);	119	to = random(2000);
120		120
121	if (!sh_quiet)	121	if (!sh_quiet)
122	printf("cpu%d, tid %llu r+ (%d)\n", CPU->id, THREAD->tid, to);	122	printf("cpu%u, tid %" PRIu64 " r+ (%d)\n", CPU->id, THREAD->tid, to);
123		123
124	rc = rwlock_read_lock_timeout(&rwlock, to);	124	rc = rwlock_read_lock_timeout(&rwlock, to);
125	if (SYNCH_FAILED(rc)) {	125	if (SYNCH_FAILED(rc)) {
126	if (!sh_quiet)	126	if (!sh_quiet)
127	printf("cpu%d, tid %llu r!\n", CPU->id, THREAD->tid);	127	printf("cpu%u, tid %" PRIu64 " r!\n", CPU->id, THREAD->tid);
128	atomic_dec(&thread_count);	128	atomic_dec(&thread_count);
129	return;	129	return;
130	}	130	}
131		131
132	if (!sh_quiet)	132	if (!sh_quiet)
133	printf("cpu%d, tid %llu r=\n", CPU->id, THREAD->tid);	133	printf("cpu%u, tid %" PRIu64 " r=\n", CPU->id, THREAD->tid);
134		134
135	thread_usleep(30000);	135	thread_usleep(30000);
136	rwlock_read_unlock(&rwlock);	136	rwlock_read_unlock(&rwlock);
137		137
138	if (!sh_quiet)	138	if (!sh_quiet)
139	printf("cpu%d, tid %llu r-\n", CPU->id, THREAD->tid);	139	printf("cpu%u, tid %" PRIu64 " r-\n", CPU->id, THREAD->tid);
140	atomic_dec(&thread_count);	140	atomic_dec(&thread_count);
141	}	141	}
142		142
143	char * test_rwlock4(bool quiet)	143	char * test_rwlock4(bool quiet)
144	{	144	{
145	context_t ctx;	145	context_t ctx;
146	uint32_t i;	146	uint32_t i;
147	sh_quiet = quiet;	147	sh_quiet = quiet;
148		148
149	waitq_initialize(&can_start);	149	waitq_initialize(&can_start);
150	rwlock_initialize(&rwlock);	150	rwlock_initialize(&rwlock);
151	atomic_set(&threads_fault, 0);	151	atomic_set(&threads_fault, 0);
152		152
153	uint32_t rd = random(7) + 1;	153	uint32_t rd = random(7) + 1;
154	uint32_t wr = random(5) + 1;	154	uint32_t wr = random(5) + 1;
155		155
156	atomic_set(&thread_count, rd + wr);	156	atomic_set(&thread_count, rd + wr);
157		157
158	thread_t *thrd;	158	thread_t *thrd;
159		159
160	context_save(&ctx);	160	context_save(&ctx);
161	if (!quiet) {	161	if (!quiet) {
162	printf("sp=%#x, readers_in=%d\n", ctx.sp, rwlock.readers_in);	162	printf("sp=%#x, readers_in=%" PRIc "\n", ctx.sp, rwlock.readers_in);
163	printf("Creating %d readers\n", rd);	163	printf("Creating %" PRIu32 " readers\n", rd);
164	}	164	}
165		165
166	for (i = 0; i < rd; i++) {	166	for (i = 0; i < rd; i++) {
167	thrd = thread_create(reader, NULL, TASK, 0, "reader", false);	167	thrd = thread_create(reader, NULL, TASK, 0, "reader", false);
168	if (thrd)	168	if (thrd)
169	thread_ready(thrd);	169	thread_ready(thrd);
170	else if (!quiet)	170	else if (!quiet)
171	printf("Could not create reader %d\n", i);	171	printf("Could not create reader %" PRIu32 "\n", i);
172	}	172	}
173		173
174	if (!quiet)	174	if (!quiet)
175	printf("Creating %d writers\n", wr);	175	printf("Creating %" PRIu32 " writers\n", wr);
176		176
177	for (i = 0; i < wr; i++) {	177	for (i = 0; i < wr; i++) {
178	thrd = thread_create(writer, NULL, TASK, 0, "writer", false);	178	thrd = thread_create(writer, NULL, TASK, 0, "writer", false);
179	if (thrd)	179	if (thrd)
180	thread_ready(thrd);	180	thread_ready(thrd);
181	else if (!quiet)	181	else if (!quiet)
182	printf("Could not create writer %d\n", i);	182	printf("Could not create writer %" PRIu32 "\n", i);
183	}	183	}
184		184
185	thread_usleep(20000);	185	thread_usleep(20000);
186	waitq_wakeup(&can_start, WAKEUP_ALL);	186	waitq_wakeup(&can_start, WAKEUP_ALL);
187		187
188	while (atomic_get(&thread_count) > 0) {	188	while (atomic_get(&thread_count) > 0) {
189	if (!quiet)	189	if (!quiet)
190	printf("Threads left: %d\n", atomic_get(&thread_count));	190	printf("Threads left: %ld\n", atomic_get(&thread_count));
191	thread_sleep(1);	191	thread_sleep(1);
192	}	192	}
193		193
194	if (atomic_get(&threads_fault) == 0)	194	if (atomic_get(&threads_fault) == 0)
195	return NULL;	195	return NULL;
196		196
197	return "Test failed";	197	return "Test failed";
198	}	198	}
199		199

Subversion Repositories HelenOS

(root)/trunk/kernel/test/synch/rwlock4.c @ 4055 – Rev 2216 → 3069