Rev 3069 | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed
| Rev 3069 | Rev 4227 | ||
|---|---|---|---|
| Line 32... | Line 32... | ||
| 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 | } |