use pthread_cleanup to acquire and release lock

In chunk_prepare(), if multiple threads wait for acquiring f->lock,
and then pthread_cancel() is called, the waiting threads are never
canceled because pthread_mutex_lock() is not a cancellation point.
So, use pthread_cleanup_push/pop to release the lock.

1 files changed, 13 insertions, 0 deletions

diff --git a/src/atomic.h b/src/atomic.h
index e0e735d..cdbd21e 100644
--- a/src/atomic.h
+++ b/src/atomic.h
@@ -55,4 +55,17 @@ static inline void lock_release(lock *l)
 	}
 }
 
+static inline void lock_release_via_cleanup(void *l)
+{
+	lock_release(l);
+}
+
+#define LOCK_ACQUIRE_THREAD(l)				\
+		lock_acquire(l);			\
+		pthread_cleanup_push(lock_release_via_cleanup, l)
+
+
+#define LOCK_RELEASE_THREAD(l) \
+	pthread_cleanup_pop(1)
+
 #endif /* _ATOMIC_H_ */