WebSVN – HelenOS – Diff – /branches/network/kernel/test/synch/rwlock3.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 <synch/rwlock.h>
 
#define THREADS  4
 
static atomic_t thread_count;
static rwlock_t rwlock;
 
 
static void reader(void *arg)
{
    thread_detach(THREAD);
   
 
    TPRINTF("cpu%u, tid %" PRIu64 ": trying to lock rwlock for reading....\n", CPU->id, THREAD->tid);
   
    rwlock_read_lock(&rwlock);
    rwlock_read_unlock(&rwlock);
   
 
    TPRINTF("cpu%u, tid %" PRIu64 ": success\n", CPU->id, THREAD->tid);
    TPRINTF("cpu%u, tid %" PRIu64 ": trying to lock rwlock for writing....\n", CPU->id, THREAD->tid);
   
 
    rwlock_write_lock(&rwlock);
    rwlock_write_unlock(&rwlock);
   
 
    TPRINTF("cpu%u, tid %" PRIu64 ": success\n", CPU->id, THREAD->tid);
   
    atomic_dec(&thread_count);
}
 
char *test_rwlock3(void)
{
    int i;
    thread_t *thrd;
 
   
    atomic_set(&thread_count, THREADS);
   
    rwlock_initialize(&rwlock);
    rwlock_write_lock(&rwlock);
   
    for (i = 0; i < THREADS; i++) {
        thrd = thread_create(reader, NULL, TASK, 0, "reader", false);
        if (thrd)
            thread_ready(thrd);
        else
            TPRINTF("Could not create reader %d\n", i);
    }
   
    thread_sleep(1);
    rwlock_write_unlock(&rwlock);
   
    while (atomic_get(&thread_count) > 0) {
 
        TPRINTF("Threads left: %ld\n", atomic_get(&thread_count));
        thread_sleep(1);
    }
   
    return NULL;
}
 

Rev 3386	Rev 4263
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		34
35	#include <synch/rwlock.h>	35	#include <synch/rwlock.h>
36		36
37	#define THREADS 4	37	#define THREADS 4
38		38
39	static atomic_t thread_count;	39	static atomic_t thread_count;
40	static rwlock_t rwlock;	40	static rwlock_t rwlock;
41	static bool sh_quiet;	-
42		41
43	static void reader(void *arg)	42	static void reader(void *arg)
44	{	43	{
45	thread_detach(THREAD);	44	thread_detach(THREAD);
46		45
47	if (!sh_quiet)	-
48	printf("cpu%u, tid %" PRIu64 ": trying to lock rwlock for reading....\n", CPU->id, THREAD->tid);	46	TPRINTF("cpu%u, tid %" PRIu64 ": trying to lock rwlock for reading....\n", CPU->id, THREAD->tid);
49		47
50	rwlock_read_lock(&rwlock);	48	rwlock_read_lock(&rwlock);
51	rwlock_read_unlock(&rwlock);	49	rwlock_read_unlock(&rwlock);
52		50
53	if (!sh_quiet) {	-
54	printf("cpu%u, tid %" PRIu64 ": success\n", CPU->id, THREAD->tid);	51	TPRINTF("cpu%u, tid %" PRIu64 ": success\n", CPU->id, THREAD->tid);
55	printf("cpu%u, tid %" PRIu64 ": trying to lock rwlock for writing....\n", CPU->id, THREAD->tid);	52	TPRINTF("cpu%u, tid %" PRIu64 ": trying to lock rwlock for writing....\n", CPU->id, THREAD->tid);
56	}	53
57		-
58	rwlock_write_lock(&rwlock);	54	rwlock_write_lock(&rwlock);
59	rwlock_write_unlock(&rwlock);	55	rwlock_write_unlock(&rwlock);
60		56
61	if (!sh_quiet)	-
62	printf("cpu%u, tid %" PRIu64 ": success\n", CPU->id, THREAD->tid);	57	TPRINTF("cpu%u, tid %" PRIu64 ": success\n", CPU->id, THREAD->tid);
63		58
64	atomic_dec(&thread_count);	59	atomic_dec(&thread_count);
65	}	60	}
66		61
67	char * test_rwlock3(bool quiet)	62	char *test_rwlock3(void)
68	{	63	{
69	int i;	64	int i;
70	thread_t *thrd;	65	thread_t *thrd;
71	sh_quiet = quiet;	-
72		66
73	atomic_set(&thread_count, THREADS);	67	atomic_set(&thread_count, THREADS);
74		68
75	rwlock_initialize(&rwlock);	69	rwlock_initialize(&rwlock);
76	rwlock_write_lock(&rwlock);	70	rwlock_write_lock(&rwlock);
77		71
78	for (i = 0; i < THREADS; i++) {	72	for (i = 0; i < THREADS; i++) {
79	thrd = thread_create(reader, NULL, TASK, 0, "reader", false);	73	thrd = thread_create(reader, NULL, TASK, 0, "reader", false);
80	if (thrd)	74	if (thrd)
81	thread_ready(thrd);	75	thread_ready(thrd);
82	else if (!quiet)	76	else
83	printf("Could not create reader %d\n", i);	77	TPRINTF("Could not create reader %d\n", i);
84	}	78	}
85		79
86	thread_sleep(1);	80	thread_sleep(1);
87	rwlock_write_unlock(&rwlock);	81	rwlock_write_unlock(&rwlock);
88		82
89	while (atomic_get(&thread_count) > 0) {	83	while (atomic_get(&thread_count) > 0) {
90	if (!quiet)	-
91	printf("Threads left: %ld\n", atomic_get(&thread_count));	84	TPRINTF("Threads left: %ld\n", atomic_get(&thread_count));
92	thread_sleep(1);	85	thread_sleep(1);
93	}	86	}
94		87
95	return NULL;	88	return NULL;
96	}	89	}
97		90

Subversion Repositories HelenOS

(root)/branches/network/kernel/test/synch/rwlock3.c – Rev 3386 → 4263