OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright 2015 The Native Client Authors. All rights reserved. | |
3 * Use of this source code is governed by a BSD-style license that can be | |
4 * found in the LICENSE file. | |
5 */ | |
6 | |
7 #include <errno.h> | |
8 #include <pthread.h> | |
9 #include <stdio.h> | |
10 #include <time.h> | |
11 | |
12 #include "native_client/src/include/nacl_assert.h" | |
13 | |
14 pthread_rwlock_t g_rwlock; | |
15 volatile int g_thread_has_lock = 0; | |
16 volatile int g_thread_should_acquire_lock = 0; | |
17 volatile int g_thread_should_release_lock = 0; | |
18 | |
19 typedef enum { | |
20 READ_LOCK = 1, | |
21 WRITE_LOCK = 2, | |
22 } lock_type; | |
23 | |
24 void *locking_thread(void *unused) { | |
25 int rc; | |
26 for (;;) { | |
27 while (!g_thread_should_acquire_lock) { /* Spin. */ } | |
28 | |
29 ASSERT_EQ(g_thread_has_lock, 0); | |
30 if (g_thread_should_acquire_lock == WRITE_LOCK) | |
31 rc = pthread_rwlock_wrlock(&g_rwlock); | |
32 else | |
33 rc = pthread_rwlock_rdlock(&g_rwlock); | |
34 ASSERT_EQ(rc, 0); | |
35 __sync_fetch_and_add(&g_thread_has_lock, 1); | |
36 | |
37 while (!g_thread_should_release_lock) { /* Spin. */ } | |
38 | |
39 ASSERT_EQ(g_thread_has_lock, 1); | |
40 rc = pthread_rwlock_unlock(&g_rwlock); | |
41 ASSERT_EQ(rc, 0); | |
42 __sync_fetch_and_sub(&g_thread_has_lock, 1); | |
43 } | |
44 | |
45 return NULL; | |
46 } | |
47 | |
48 | |
49 void tell_thread_to_acquire_lock(lock_type lock_type) { | |
50 fprintf(stderr, "Thread acquiring lock: %s\n", | |
51 lock_type == WRITE_LOCK ? "WRITE" : "READ"); | |
52 | |
53 ASSERT_EQ(g_thread_has_lock, 0); | |
54 ASSERT_EQ(g_thread_should_acquire_lock, 0); | |
55 __sync_fetch_and_add(&g_thread_should_acquire_lock, lock_type); | |
56 | |
57 while (!g_thread_has_lock) { /* Spin. */ } | |
58 | |
59 __sync_fetch_and_sub(&g_thread_should_acquire_lock, lock_type); | |
60 ASSERT_EQ(g_thread_should_acquire_lock, 0); | |
61 | |
62 fprintf(stderr, "Thread acquired lock.\n"); | |
63 } | |
64 | |
65 void tell_thread_to_release_lock(void) { | |
66 fprintf(stderr, "Thread releasing lock.\n"); | |
67 | |
68 ASSERT_EQ(g_thread_has_lock, 1); | |
69 ASSERT_EQ(g_thread_should_release_lock, 0); | |
70 __sync_fetch_and_add(&g_thread_should_release_lock, 1); | |
71 | |
72 while (g_thread_has_lock) { /* Spin. */ } | |
73 | |
74 __sync_fetch_and_sub(&g_thread_should_release_lock, 1); | |
75 ASSERT_EQ(g_thread_should_release_lock, 0); | |
76 | |
77 fprintf(stderr, "Thread released lock.\n"); | |
78 } | |
79 | |
80 void test_reader_timedwait(void) { | |
81 fprintf(stderr, "test_reader_timedwait\n"); | |
82 tell_thread_to_acquire_lock(WRITE_LOCK); | |
83 | |
84 struct timespec t = { 0, 0 }; | |
85 int rc = pthread_rwlock_timedrdlock(&g_rwlock, &t); | |
86 ASSERT_EQ(rc, ETIMEDOUT); | |
87 | |
88 tell_thread_to_release_lock(); | |
89 } | |
90 | |
91 void test_writer_timedwait(void) { | |
92 fprintf(stderr, "test_writer_timedwait\n"); | |
93 tell_thread_to_acquire_lock(READ_LOCK); | |
94 | |
95 struct timespec t = { 0, 0 }; | |
96 int rc = pthread_rwlock_timedwrlock(&g_rwlock, &t); | |
97 ASSERT_EQ(rc, ETIMEDOUT); | |
98 | |
99 tell_thread_to_release_lock(); | |
100 } | |
101 | |
102 void test_multiple_writers(void) { | |
103 fprintf(stderr, "test_multiple_writers\n"); | |
104 tell_thread_to_acquire_lock(WRITE_LOCK); | |
105 | |
106 /* | |
107 * Attempt to acquire second write lock should fail. | |
108 */ | |
109 int rc = pthread_rwlock_trywrlock(&g_rwlock); | |
110 ASSERT_EQ(rc, EBUSY); | |
111 | |
112 tell_thread_to_release_lock(); | |
113 } | |
114 | |
115 void test_recursive_reader(void) { | |
116 /* | |
117 * Test that an rdlock can be recursively acquired even when there | |
118 * is a waiting writer. | |
119 */ | |
120 int rc = pthread_rwlock_rdlock(&g_rwlock); | |
121 ASSERT_EQ(rc, 0); | |
122 | |
123 /* | |
124 * Tell the locking thread to attempt to acquire the write lock. | |
125 * This should fail and block until all readers are unlocked. | |
126 */ | |
127 ASSERT_EQ(g_thread_has_lock, 0); | |
128 ASSERT_EQ(g_thread_should_acquire_lock, 0); | |
129 __sync_fetch_and_add(&g_thread_should_acquire_lock, WRITE_LOCK); | |
130 | |
131 /* | |
132 * Sleep for 10ms | |
133 */ | |
134 rc = usleep(10 * 1000); | |
135 ASSERT_EQ(rc, 0); | |
136 ASSERT_EQ(g_thread_has_lock, 0); | |
137 | |
138 /* | |
139 * Now make sure the waiting writer doesn't block the recursive acquisition | |
140 * of the rdlock (using both tryrdlock and rdlock). | |
141 */ | |
142 rc = pthread_rwlock_tryrdlock(&g_rwlock); | |
143 ASSERT_EQ(rc, 0); | |
144 rc = pthread_rwlock_unlock(&g_rwlock); | |
145 ASSERT_EQ(rc, 0); | |
146 ASSERT_EQ(g_thread_has_lock, 0); | |
147 rc = pthread_rwlock_rdlock(&g_rwlock); | |
148 ASSERT_EQ(rc, 0); | |
149 rc = pthread_rwlock_unlock(&g_rwlock); | |
150 ASSERT_EQ(rc, 0); | |
151 ASSERT_EQ(g_thread_has_lock, 0); | |
152 | |
153 /* | |
154 * Finally unlock the rdlock which should allow the secondary thread | |
155 * to acquire the wrlock | |
156 */ | |
157 rc = pthread_rwlock_unlock(&g_rwlock); | |
158 ASSERT_EQ(rc, 0); | |
159 while (!g_thread_has_lock) { /* Spin. */ } | |
160 __sync_fetch_and_sub(&g_thread_should_acquire_lock, WRITE_LOCK); | |
161 ASSERT_EQ(g_thread_should_acquire_lock, 0); | |
162 | |
163 tell_thread_to_release_lock(); | |
164 } | |
165 | |
166 void test_multiple_readers(void) { | |
167 fprintf(stderr, "test_multiple_readers\n"); | |
168 tell_thread_to_acquire_lock(READ_LOCK); | |
169 | |
170 /* | |
171 * Now attempt to acquire the lock on the main thread. | |
172 * Since they are both readers this should succeed. | |
173 * Try with tryrdlock, rdlock and timedrdlock. | |
174 */ | |
175 int rc = pthread_rwlock_tryrdlock(&g_rwlock); | |
176 ASSERT_EQ(rc, 0); | |
177 rc = pthread_rwlock_unlock(&g_rwlock); | |
178 ASSERT_EQ(rc, 0); | |
179 | |
180 struct timespec t = { 0, 0 }; | |
181 rc = pthread_rwlock_timedrdlock(&g_rwlock, &t); | |
182 ASSERT_EQ(rc, 0); | |
183 rc = pthread_rwlock_unlock(&g_rwlock); | |
184 ASSERT_EQ(rc, 0); | |
185 | |
186 rc = pthread_rwlock_rdlock(&g_rwlock); | |
187 ASSERT_EQ(rc, 0); | |
188 rc = pthread_rwlock_unlock(&g_rwlock); | |
189 ASSERT_EQ(rc, 0); | |
190 | |
191 tell_thread_to_release_lock(); | |
192 } | |
193 | |
194 void test_reader_plus_writer(void) { | |
195 fprintf(stderr, "test_reader_plus_writer\n"); | |
196 tell_thread_to_acquire_lock(READ_LOCK); | |
197 | |
198 /* | |
199 * Now attempt to acquire the write lock on the main thread. | |
200 * This should fail. | |
201 */ | |
202 int rc = pthread_rwlock_trywrlock(&g_rwlock); | |
203 ASSERT_EQ(rc, EBUSY); | |
204 | |
205 tell_thread_to_release_lock(); | |
206 } | |
207 | |
208 void test_writer_plus_reader(void) { | |
209 fprintf(stderr, "test_writer_plus_reader\n"); | |
210 | |
211 /* | |
212 * First get the write lock. | |
213 */ | |
214 int rc = pthread_rwlock_wrlock(&g_rwlock); | |
215 ASSERT_EQ(rc, 0); | |
216 | |
217 /* | |
218 * Attempt to acquire read lock should now fail | |
219 */ | |
220 rc = pthread_rwlock_tryrdlock(&g_rwlock); | |
221 ASSERT_EQ(rc, EBUSY); | |
222 | |
223 rc = pthread_rwlock_unlock(&g_rwlock); | |
224 ASSERT_EQ(rc, 0); | |
225 } | |
226 | |
227 void test_unlocked_with_zero_timestamp(void) { | |
228 fprintf(stderr, "test_unlocked_with_zero_timestamp\n"); | |
229 int rc; | |
230 struct timespec abstime = { 0, 0 }; | |
231 ASSERT_EQ(g_thread_has_lock, 0); | |
232 fprintf(stderr, "Trying to lock the unlocked rwlock with a valid " | |
233 "zero absolute timestamp. " | |
234 "Expected to succeed instantly since the lock is free.\n"); | |
235 rc = pthread_rwlock_timedrdlock(&g_rwlock, &abstime); | |
236 ASSERT_EQ(rc, 0); | |
237 rc = pthread_rwlock_unlock(&g_rwlock); | |
238 ASSERT_EQ(rc, 0); | |
239 } | |
240 | |
241 int main(int argc, char **argv) { | |
242 int rc; | |
243 fprintf(stderr, "Running...\n"); | |
244 | |
245 pthread_rwlockattr_t attrs; | |
246 rc = pthread_rwlockattr_init(&attrs); | |
247 ASSERT_EQ(rc, 0); | |
248 int shared = -1; | |
249 rc = pthread_rwlockattr_getpshared(&attrs, &shared); | |
250 ASSERT_EQ(rc, 0); | |
251 ASSERT_EQ(shared, PTHREAD_PROCESS_PRIVATE); | |
252 rc = pthread_rwlockattr_setpshared(&attrs, PTHREAD_PROCESS_SHARED); | |
253 ASSERT_EQ(rc, 0); | |
254 rc = pthread_rwlockattr_setpshared(&attrs, PTHREAD_PROCESS_PRIVATE); | |
255 ASSERT_EQ(rc, 0); | |
256 rc = pthread_rwlock_init(&g_rwlock, &attrs); | |
257 ASSERT_EQ(rc, 0); | |
258 rc = pthread_rwlockattr_destroy(&attrs); | |
259 ASSERT_EQ(rc, 0); | |
260 | |
261 pthread_t thread; | |
262 rc = pthread_create(&thread, NULL, locking_thread, NULL); | |
263 ASSERT_EQ(rc, 0); | |
264 fprintf(stderr, "Thread started.\n"); | |
265 | |
266 test_unlocked_with_zero_timestamp(); | |
267 test_multiple_readers(); | |
268 test_multiple_writers(); | |
269 test_reader_plus_writer(); | |
270 test_writer_plus_reader(); | |
271 test_reader_timedwait(); | |
272 test_writer_timedwait(); | |
273 test_recursive_reader(); | |
274 | |
275 rc = pthread_rwlock_destroy(&g_rwlock); | |
276 ASSERT_EQ(rc, 0); | |
277 fprintf(stderr, "Done.\n"); | |
278 return 0; | |
279 } | |
OLD | NEW |