1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef _LIBLOCKDEP_MUTEX_H
3 #define _LIBLOCKDEP_MUTEX_H
4 
5 #include <pthread.h>
6 #include "common.h"
7 
8 struct liblockdep_pthread_mutex {
9 	pthread_mutex_t mutex;
10 	struct lockdep_map dep_map;
11 };
12 
13 typedef struct liblockdep_pthread_mutex liblockdep_pthread_mutex_t;
14 
15 #define LIBLOCKDEP_PTHREAD_MUTEX_INITIALIZER(mtx)			\
16 		(const struct liblockdep_pthread_mutex) {		\
17 	.mutex = PTHREAD_MUTEX_INITIALIZER,				\
18 	.dep_map = STATIC_LOCKDEP_MAP_INIT(#mtx, &((&(mtx))->dep_map)),	\
19 }
20 
__mutex_init(liblockdep_pthread_mutex_t * lock,const char * name,struct lock_class_key * key,const pthread_mutexattr_t * __mutexattr)21 static inline int __mutex_init(liblockdep_pthread_mutex_t *lock,
22 				const char *name,
23 				struct lock_class_key *key,
24 				const pthread_mutexattr_t *__mutexattr)
25 {
26 	lockdep_init_map(&lock->dep_map, name, key, 0);
27 	return pthread_mutex_init(&lock->mutex, __mutexattr);
28 }
29 
30 #define liblockdep_pthread_mutex_init(mutex, mutexattr)		\
31 ({								\
32 	static struct lock_class_key __key;			\
33 								\
34 	__mutex_init((mutex), #mutex, &__key, (mutexattr));	\
35 })
36 
liblockdep_pthread_mutex_lock(liblockdep_pthread_mutex_t * lock)37 static inline int liblockdep_pthread_mutex_lock(liblockdep_pthread_mutex_t *lock)
38 {
39 	lock_acquire(&lock->dep_map, 0, 0, 0, 1, NULL, (unsigned long)_RET_IP_);
40 	return pthread_mutex_lock(&lock->mutex);
41 }
42 
liblockdep_pthread_mutex_unlock(liblockdep_pthread_mutex_t * lock)43 static inline int liblockdep_pthread_mutex_unlock(liblockdep_pthread_mutex_t *lock)
44 {
45 	lock_release(&lock->dep_map, 0, (unsigned long)_RET_IP_);
46 	return pthread_mutex_unlock(&lock->mutex);
47 }
48 
liblockdep_pthread_mutex_trylock(liblockdep_pthread_mutex_t * lock)49 static inline int liblockdep_pthread_mutex_trylock(liblockdep_pthread_mutex_t *lock)
50 {
51 	lock_acquire(&lock->dep_map, 0, 1, 0, 1, NULL, (unsigned long)_RET_IP_);
52 	return pthread_mutex_trylock(&lock->mutex) == 0 ? 1 : 0;
53 }
54 
liblockdep_pthread_mutex_destroy(liblockdep_pthread_mutex_t * lock)55 static inline int liblockdep_pthread_mutex_destroy(liblockdep_pthread_mutex_t *lock)
56 {
57 	return pthread_mutex_destroy(&lock->mutex);
58 }
59 
60 #ifdef __USE_LIBLOCKDEP
61 
62 #define pthread_mutex_t         liblockdep_pthread_mutex_t
63 #define pthread_mutex_init      liblockdep_pthread_mutex_init
64 #define pthread_mutex_lock      liblockdep_pthread_mutex_lock
65 #define pthread_mutex_unlock    liblockdep_pthread_mutex_unlock
66 #define pthread_mutex_trylock   liblockdep_pthread_mutex_trylock
67 #define pthread_mutex_destroy   liblockdep_pthread_mutex_destroy
68 
69 #endif
70 
71 #endif
72