tools/lib/lockdep/include/liblockdep/mutex.h

   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
  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
  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
  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
  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
  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