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 lockdep_map dep_map;
11};
12
13typedef 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
21static 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
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, 0, (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	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