1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef _LIBLOCKDEP_RWLOCK_H
3 #define _LIBLOCKDEP_RWLOCK_H
4
5 #include <pthread.h>
6 #include "common.h"
7
8 struct liblockdep_pthread_rwlock {
9 pthread_rwlock_t rwlock;
10 struct lockdep_map dep_map;
11 };
12
13 typedef struct liblockdep_pthread_rwlock liblockdep_pthread_rwlock_t;
14
15 #define LIBLOCKDEP_PTHREAD_RWLOCK_INITIALIZER(rwl) \
16 (struct liblockdep_pthread_rwlock) { \
17 .rwlock = PTHREAD_RWLOCK_INITIALIZER, \
18 .dep_map = STATIC_LOCKDEP_MAP_INIT(#rwl, &((&(rwl))->dep_map)), \
19 }
20
__rwlock_init(liblockdep_pthread_rwlock_t * lock,const char * name,struct lock_class_key * key,const pthread_rwlockattr_t * attr)21 static inline int __rwlock_init(liblockdep_pthread_rwlock_t *lock,
22 const char *name,
23 struct lock_class_key *key,
24 const pthread_rwlockattr_t *attr)
25 {
26 lockdep_init_map(&lock->dep_map, name, key, 0);
27
28 return pthread_rwlock_init(&lock->rwlock, attr);
29 }
30
31 #define liblockdep_pthread_rwlock_init(lock, attr) \
32 ({ \
33 static struct lock_class_key __key; \
34 \
35 __rwlock_init((lock), #lock, &__key, (attr)); \
36 })
37
liblockdep_pthread_rwlock_rdlock(liblockdep_pthread_rwlock_t * lock)38 static inline int liblockdep_pthread_rwlock_rdlock(liblockdep_pthread_rwlock_t *lock)
39 {
40 lock_acquire(&lock->dep_map, 0, 0, 2, 1, NULL, (unsigned long)_RET_IP_);
41 return pthread_rwlock_rdlock(&lock->rwlock);
42
43 }
44
liblockdep_pthread_rwlock_unlock(liblockdep_pthread_rwlock_t * lock)45 static inline int liblockdep_pthread_rwlock_unlock(liblockdep_pthread_rwlock_t *lock)
46 {
47 lock_release(&lock->dep_map, (unsigned long)_RET_IP_);
48 return pthread_rwlock_unlock(&lock->rwlock);
49 }
50
liblockdep_pthread_rwlock_wrlock(liblockdep_pthread_rwlock_t * lock)51 static inline int liblockdep_pthread_rwlock_wrlock(liblockdep_pthread_rwlock_t *lock)
52 {
53 lock_acquire(&lock->dep_map, 0, 0, 0, 1, NULL, (unsigned long)_RET_IP_);
54 return pthread_rwlock_wrlock(&lock->rwlock);
55 }
56
liblockdep_pthread_rwlock_tryrdlock(liblockdep_pthread_rwlock_t * lock)57 static inline int liblockdep_pthread_rwlock_tryrdlock(liblockdep_pthread_rwlock_t *lock)
58 {
59 lock_acquire(&lock->dep_map, 0, 1, 2, 1, NULL, (unsigned long)_RET_IP_);
60 return pthread_rwlock_tryrdlock(&lock->rwlock) == 0 ? 1 : 0;
61 }
62
liblockdep_pthread_rwlock_trywrlock(liblockdep_pthread_rwlock_t * lock)63 static inline int liblockdep_pthread_rwlock_trywrlock(liblockdep_pthread_rwlock_t *lock)
64 {
65 lock_acquire(&lock->dep_map, 0, 1, 0, 1, NULL, (unsigned long)_RET_IP_);
66 return pthread_rwlock_trywrlock(&lock->rwlock) == 0 ? 1 : 0;
67 }
68
liblockdep_rwlock_destroy(liblockdep_pthread_rwlock_t * lock)69 static inline int liblockdep_rwlock_destroy(liblockdep_pthread_rwlock_t *lock)
70 {
71 return pthread_rwlock_destroy(&lock->rwlock);
72 }
73
74 #ifdef __USE_LIBLOCKDEP
75
76 #define pthread_rwlock_t liblockdep_pthread_rwlock_t
77 #define pthread_rwlock_init liblockdep_pthread_rwlock_init
78 #define pthread_rwlock_rdlock liblockdep_pthread_rwlock_rdlock
79 #define pthread_rwlock_unlock liblockdep_pthread_rwlock_unlock
80 #define pthread_rwlock_wrlock liblockdep_pthread_rwlock_wrlock
81 #define pthread_rwlock_tryrdlock liblockdep_pthread_rwlock_tryrdlock
82 #define pthread_rwlock_trywrlock liblockdep_pthread_rwlock_trywrlock
83 #define pthread_rwlock_destroy liblockdep_rwlock_destroy
84
85 #endif
86
87 #endif
88