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
8struct liblockdep_pthread_mutex {
9	pthread_mutex_t mutex;
10	struct lock_class_key key;
11	struct lockdep_map dep_map;
12};
13
14typedef 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
22static 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
37static 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
43static 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
49static 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
55static 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