15 #include "arch/arch.h"
21 void __fio_mutex_remove(struct fio_mutex *mutex)
23 assert(mutex->magic == FIO_MUTEX_MAGIC);
24 pthread_cond_destroy(&mutex->cond);
27 * Ensure any subsequent attempt to grab this mutex will fail
28 * with an assert, instead of just silently hanging.
30 memset(mutex, 0, sizeof(*mutex));
33 void fio_mutex_remove(struct fio_mutex *mutex)
35 __fio_mutex_remove(mutex);
36 munmap((void *) mutex, sizeof(*mutex));
39 int cond_init_pshared(pthread_cond_t *cond)
41 pthread_condattr_t cattr;
44 ret = pthread_condattr_init(&cattr);
46 log_err("pthread_condattr_init: %s\n", strerror(ret));
51 ret = pthread_condattr_setpshared(&cattr, PTHREAD_PROCESS_SHARED);
53 log_err("pthread_condattr_setpshared: %s\n", strerror(ret));
57 ret = pthread_cond_init(cond, &cattr);
59 log_err("pthread_cond_init: %s\n", strerror(ret));
66 int mutex_init_pshared(pthread_mutex_t *mutex)
68 pthread_mutexattr_t mattr;
71 ret = pthread_mutexattr_init(&mattr);
73 log_err("pthread_mutexattr_init: %s\n", strerror(ret));
78 * Not all platforms support process shared mutexes (FreeBSD)
81 ret = pthread_mutexattr_setpshared(&mattr, PTHREAD_PROCESS_SHARED);
83 log_err("pthread_mutexattr_setpshared: %s\n", strerror(ret));
87 ret = pthread_mutex_init(mutex, &mattr);
89 log_err("pthread_mutex_init: %s\n", strerror(ret));
96 int mutex_cond_init_pshared(pthread_mutex_t *mutex, pthread_cond_t *cond)
100 ret = mutex_init_pshared(mutex);
104 ret = cond_init_pshared(cond);
111 int __fio_mutex_init(struct fio_mutex *mutex, int value)
115 mutex->value = value;
116 mutex->magic = FIO_MUTEX_MAGIC;
118 ret = mutex_cond_init_pshared(&mutex->lock, &mutex->cond);
125 struct fio_mutex *fio_mutex_init(int value)
127 struct fio_mutex *mutex = NULL;
129 mutex = (void *) mmap(NULL, sizeof(struct fio_mutex),
130 PROT_READ | PROT_WRITE,
131 OS_MAP_ANON | MAP_SHARED, -1, 0);
132 if (mutex == MAP_FAILED) {
133 perror("mmap mutex");
137 if (!__fio_mutex_init(mutex, value))
140 fio_mutex_remove(mutex);
144 static bool mutex_timed_out(struct timeval *t, unsigned int msecs)
148 gettimeofday(&now, NULL);
149 return mtime_since(t, &now) >= msecs;
152 int fio_mutex_down_timeout(struct fio_mutex *mutex, unsigned int msecs)
158 assert(mutex->magic == FIO_MUTEX_MAGIC);
160 gettimeofday(&tv_s, NULL);
161 t.tv_sec = tv_s.tv_sec;
162 t.tv_nsec = tv_s.tv_usec * 1000;
164 t.tv_sec += msecs / 1000;
165 t.tv_nsec += ((msecs * 1000000ULL) % 1000000000);
166 if (t.tv_nsec >= 1000000000) {
167 t.tv_nsec -= 1000000000;
171 pthread_mutex_lock(&mutex->lock);
174 while (!mutex->value && !ret) {
176 * Some platforms (FreeBSD 9?) seems to return timed out
177 * way too early, double check.
179 ret = pthread_cond_timedwait(&mutex->cond, &mutex->lock, &t);
180 if (ret == ETIMEDOUT && !mutex_timed_out(&tv_s, msecs))
187 pthread_mutex_unlock(&mutex->lock);
191 pthread_mutex_unlock(&mutex->lock);
195 bool fio_mutex_down_trylock(struct fio_mutex *mutex)
199 assert(mutex->magic == FIO_MUTEX_MAGIC);
201 pthread_mutex_lock(&mutex->lock);
206 pthread_mutex_unlock(&mutex->lock);
211 void fio_mutex_down(struct fio_mutex *mutex)
213 assert(mutex->magic == FIO_MUTEX_MAGIC);
215 pthread_mutex_lock(&mutex->lock);
217 while (!mutex->value) {
219 pthread_cond_wait(&mutex->cond, &mutex->lock);
224 pthread_mutex_unlock(&mutex->lock);
227 void fio_mutex_up(struct fio_mutex *mutex)
231 assert(mutex->magic == FIO_MUTEX_MAGIC);
233 pthread_mutex_lock(&mutex->lock);
235 if (!mutex->value && mutex->waiters)
238 pthread_mutex_unlock(&mutex->lock);
241 pthread_cond_signal(&mutex->cond);
244 void fio_rwlock_write(struct fio_rwlock *lock)
246 assert(lock->magic == FIO_RWLOCK_MAGIC);
247 pthread_rwlock_wrlock(&lock->lock);
250 void fio_rwlock_read(struct fio_rwlock *lock)
252 assert(lock->magic == FIO_RWLOCK_MAGIC);
253 pthread_rwlock_rdlock(&lock->lock);
256 void fio_rwlock_unlock(struct fio_rwlock *lock)
258 assert(lock->magic == FIO_RWLOCK_MAGIC);
259 pthread_rwlock_unlock(&lock->lock);
262 void fio_rwlock_remove(struct fio_rwlock *lock)
264 assert(lock->magic == FIO_RWLOCK_MAGIC);
265 munmap((void *) lock, sizeof(*lock));
268 struct fio_rwlock *fio_rwlock_init(void)
270 struct fio_rwlock *lock;
271 pthread_rwlockattr_t attr;
274 lock = (void *) mmap(NULL, sizeof(struct fio_rwlock),
275 PROT_READ | PROT_WRITE,
276 OS_MAP_ANON | MAP_SHARED, -1, 0);
277 if (lock == MAP_FAILED) {
278 perror("mmap rwlock");
283 lock->magic = FIO_RWLOCK_MAGIC;
285 ret = pthread_rwlockattr_init(&attr);
287 log_err("pthread_rwlockattr_init: %s\n", strerror(ret));
290 #ifdef CONFIG_PSHARED
291 ret = pthread_rwlockattr_setpshared(&attr, PTHREAD_PROCESS_SHARED);
293 log_err("pthread_rwlockattr_setpshared: %s\n", strerror(ret));
297 ret = pthread_rwlock_init(&lock->lock, &attr);
299 ret = pthread_rwlock_init(&lock->lock, NULL);
303 log_err("pthread_rwlock_init: %s\n", strerror(ret));
307 pthread_rwlockattr_destroy(&attr);
311 pthread_rwlockattr_destroy(&attr);
314 fio_rwlock_remove(lock);