1 /* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
2 *
3 * Permission is hereby granted, free of charge, to any person obtaining a copy
4 * of this software and associated documentation files (the "Software"), to
5 * deal in the Software without restriction, including without limitation the
6 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
7 * sell copies of the Software, and to permit persons to whom the Software is
8 * furnished to do so, subject to the following conditions:
9 *
10 * The above copyright notice and this permission notice shall be included in
11 * all copies or substantial portions of the Software.
12 *
13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
18 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
19 * IN THE SOFTWARE.
20 */
21
22 #include "uv.h"
23 #include "task.h"
24
25 #include <stdio.h>
26 #include <stdlib.h>
27
28 static uv_cond_t condvar;
29 static uv_mutex_t mutex;
30 static uv_rwlock_t rwlock;
31 static int step;
32
33 /* The mutex and rwlock tests are really poor.
34 * They're very basic sanity checks and nothing more.
35 * Apologies if that rhymes.
36 */
37
TEST_IMPL(thread_mutex)38 TEST_IMPL(thread_mutex) {
39 uv_mutex_t mutex;
40 int r;
41
42 r = uv_mutex_init(&mutex);
43 ASSERT(r == 0);
44
45 uv_mutex_lock(&mutex);
46 uv_mutex_unlock(&mutex);
47 uv_mutex_destroy(&mutex);
48
49 return 0;
50 }
51
52
TEST_IMPL(thread_mutex_recursive)53 TEST_IMPL(thread_mutex_recursive) {
54 uv_mutex_t mutex;
55 int r;
56
57 r = uv_mutex_init_recursive(&mutex);
58 ASSERT(r == 0);
59
60 uv_mutex_lock(&mutex);
61 uv_mutex_lock(&mutex);
62 ASSERT(0 == uv_mutex_trylock(&mutex));
63
64 uv_mutex_unlock(&mutex);
65 uv_mutex_unlock(&mutex);
66 uv_mutex_unlock(&mutex);
67 uv_mutex_destroy(&mutex);
68
69 return 0;
70 }
71
72
TEST_IMPL(thread_rwlock)73 TEST_IMPL(thread_rwlock) {
74 uv_rwlock_t rwlock;
75 int r;
76
77 r = uv_rwlock_init(&rwlock);
78 ASSERT(r == 0);
79
80 uv_rwlock_rdlock(&rwlock);
81 uv_rwlock_rdunlock(&rwlock);
82 uv_rwlock_wrlock(&rwlock);
83 uv_rwlock_wrunlock(&rwlock);
84 uv_rwlock_destroy(&rwlock);
85
86 return 0;
87 }
88
89
90 /* Call when holding |mutex|. */
synchronize_nowait(void)91 static void synchronize_nowait(void) {
92 step += 1;
93 uv_cond_signal(&condvar);
94 }
95
96
97 /* Call when holding |mutex|. */
synchronize(void)98 static void synchronize(void) {
99 int current;
100
101 synchronize_nowait();
102 /* Wait for the other thread. Guard against spurious wakeups. */
103 for (current = step; current == step; uv_cond_wait(&condvar, &mutex));
104 ASSERT(step == current + 1);
105 }
106
107
thread_rwlock_trylock_peer(void * unused)108 static void thread_rwlock_trylock_peer(void* unused) {
109 (void) &unused;
110
111 uv_mutex_lock(&mutex);
112
113 /* Write lock held by other thread. */
114 ASSERT(UV_EBUSY == uv_rwlock_tryrdlock(&rwlock));
115 ASSERT(UV_EBUSY == uv_rwlock_trywrlock(&rwlock));
116 synchronize();
117
118 /* Read lock held by other thread. */
119 ASSERT(0 == uv_rwlock_tryrdlock(&rwlock));
120 uv_rwlock_rdunlock(&rwlock);
121 ASSERT(UV_EBUSY == uv_rwlock_trywrlock(&rwlock));
122 synchronize();
123
124 /* Acquire write lock. */
125 ASSERT(0 == uv_rwlock_trywrlock(&rwlock));
126 synchronize();
127
128 /* Release write lock and acquire read lock. */
129 uv_rwlock_wrunlock(&rwlock);
130 ASSERT(0 == uv_rwlock_tryrdlock(&rwlock));
131 synchronize();
132
133 uv_rwlock_rdunlock(&rwlock);
134 synchronize_nowait(); /* Signal main thread we're going away. */
135 uv_mutex_unlock(&mutex);
136 }
137
138
TEST_IMPL(thread_rwlock_trylock)139 TEST_IMPL(thread_rwlock_trylock) {
140 uv_thread_t thread;
141
142 ASSERT(0 == uv_cond_init(&condvar));
143 ASSERT(0 == uv_mutex_init(&mutex));
144 ASSERT(0 == uv_rwlock_init(&rwlock));
145
146 uv_mutex_lock(&mutex);
147 ASSERT(0 == uv_thread_create(&thread, thread_rwlock_trylock_peer, NULL));
148
149 /* Hold write lock. */
150 ASSERT(0 == uv_rwlock_trywrlock(&rwlock));
151 synchronize(); /* Releases the mutex to the other thread. */
152
153 /* Release write lock and acquire read lock. Pthreads doesn't support
154 * the notion of upgrading or downgrading rwlocks, so neither do we.
155 */
156 uv_rwlock_wrunlock(&rwlock);
157 ASSERT(0 == uv_rwlock_tryrdlock(&rwlock));
158 synchronize();
159
160 /* Release read lock. */
161 uv_rwlock_rdunlock(&rwlock);
162 synchronize();
163
164 /* Write lock held by other thread. */
165 ASSERT(UV_EBUSY == uv_rwlock_tryrdlock(&rwlock));
166 ASSERT(UV_EBUSY == uv_rwlock_trywrlock(&rwlock));
167 synchronize();
168
169 /* Read lock held by other thread. */
170 ASSERT(0 == uv_rwlock_tryrdlock(&rwlock));
171 uv_rwlock_rdunlock(&rwlock);
172 ASSERT(UV_EBUSY == uv_rwlock_trywrlock(&rwlock));
173 synchronize();
174
175 ASSERT(0 == uv_thread_join(&thread));
176 uv_rwlock_destroy(&rwlock);
177 uv_mutex_unlock(&mutex);
178 uv_mutex_destroy(&mutex);
179 uv_cond_destroy(&condvar);
180
181 return 0;
182 }
183