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 lock_class_key key;
11 	struct lockdep_map dep_map;
12 };
13 
14 typedef struct liblockdep_pthread_mutex liblockdep_pthread_mutex_t;
15 
16 #define LIBLOCKDEP_PTHREAD_MUTEX_INITIALIZER(mtx)			\
17 		(const struct liblockdep_pthread_mutex) {		\
18 	.mutex = PTHREAD_MUTEX_INITIALIZER,				\
19 	.dep_map = STATIC_LOCKDEP_MAP_INIT(#mtx, &((&(mtx))->dep_map)),	\
20 }
21 
__mutex_init(liblockdep_pthread_mutex_t * lock,const char * name,struct lock_class_key * key,const pthread_mutexattr_t * __mutexattr)22 static inline int __mutex_init(liblockdep_pthread_mutex_t *lock,
23 				const char *name,
24 				struct lock_class_key *key,
25 				const pthread_mutexattr_t *__mutexattr)
26 {
27 	lockdep_init_map(&lock->dep_map, name, key, 0);
28 	return pthread_mutex_init(&lock->mutex, __mutexattr);
29 }
30 
31 #define liblockdep_pthread_mutex_init(mutex, mutexattr)			\
32 ({									\
33 	lockdep_register_key(&(mutex)->key);				\
34 	__mutex_init((mutex), #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, (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 	lockdep_reset_lock(&lock->dep_map);
58 	lockdep_unregister_key(&lock->key);
59 	return pthread_mutex_destroy(&lock->mutex);
60 }
61 
62 #ifdef __USE_LIBLOCKDEP
63 
64 #define pthread_mutex_t         liblockdep_pthread_mutex_t
65 #define pthread_mutex_init      liblockdep_pthread_mutex_init
66 #define pthread_mutex_lock      liblockdep_pthread_mutex_lock
67 #define pthread_mutex_unlock    liblockdep_pthread_mutex_unlock
68 #define pthread_mutex_trylock   liblockdep_pthread_mutex_trylock
69 #define pthread_mutex_destroy   liblockdep_pthread_mutex_destroy
70 
71 #endif
72 
73 #endif
74