rwsem.h 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173
  1. /*
  2. * include/asm-sh/rwsem.h: R/W semaphores for SH using the stuff
  3. * in lib/rwsem.c.
  4. */
  5. #ifndef _ASM_SH_RWSEM_H
  6. #define _ASM_SH_RWSEM_H
  7. #ifndef _LINUX_RWSEM_H
  8. #error "please don't include asm/rwsem.h directly, use linux/rwsem.h instead"
  9. #endif
  10. #ifdef __KERNEL__
  11. #define RWSEM_UNLOCKED_VALUE 0x00000000
  12. #define RWSEM_ACTIVE_BIAS 0x00000001
  13. #define RWSEM_ACTIVE_MASK 0x0000ffff
  14. #define RWSEM_WAITING_BIAS (-0x00010000)
  15. #define RWSEM_ACTIVE_READ_BIAS RWSEM_ACTIVE_BIAS
  16. #define RWSEM_ACTIVE_WRITE_BIAS (RWSEM_WAITING_BIAS + RWSEM_ACTIVE_BIAS)
  17. #ifdef CONFIG_DEBUG_LOCK_ALLOC
  18. # define __RWSEM_DEP_MAP_INIT(lockname) , .dep_map = { .name = #lockname }
  19. #else
  20. # define __RWSEM_DEP_MAP_INIT(lockname)
  21. #endif
  22. #define __RWSEM_INITIALIZER(name) \
  23. { RWSEM_UNLOCKED_VALUE, __SPIN_LOCK_UNLOCKED((name).wait_lock), \
  24. LIST_HEAD_INIT((name).wait_list) \
  25. __RWSEM_DEP_MAP_INIT(name) }
  26. #define DECLARE_RWSEM(name) \
  27. struct rw_semaphore name = __RWSEM_INITIALIZER(name)
  28. extern struct rw_semaphore *rwsem_down_read_failed(struct rw_semaphore *sem);
  29. extern struct rw_semaphore *rwsem_down_write_failed(struct rw_semaphore *sem);
  30. extern struct rw_semaphore *rwsem_wake(struct rw_semaphore *sem);
  31. extern struct rw_semaphore *rwsem_downgrade_wake(struct rw_semaphore *sem);
  32. extern void __init_rwsem(struct rw_semaphore *sem, const char *name,
  33. struct lock_class_key *key);
  34. #define init_rwsem(sem) \
  35. do { \
  36. static struct lock_class_key __key; \
  37. \
  38. __init_rwsem((sem), #sem, &__key); \
  39. } while (0)
  40. static inline void init_rwsem(struct rw_semaphore *sem)
  41. {
  42. sem->count = RWSEM_UNLOCKED_VALUE;
  43. spin_lock_init(&sem->wait_lock);
  44. INIT_LIST_HEAD(&sem->wait_list);
  45. }
  46. /*
  47. * lock for reading
  48. */
  49. static inline void __down_read(struct rw_semaphore *sem)
  50. {
  51. if (atomic_inc_return((atomic_t *)(&sem->count)) > 0)
  52. smp_wmb();
  53. else
  54. rwsem_down_read_failed(sem);
  55. }
  56. static inline int __down_read_trylock(struct rw_semaphore *sem)
  57. {
  58. int tmp;
  59. while ((tmp = sem->count) >= 0) {
  60. if (tmp == cmpxchg(&sem->count, tmp,
  61. tmp + RWSEM_ACTIVE_READ_BIAS)) {
  62. smp_wmb();
  63. return 1;
  64. }
  65. }
  66. return 0;
  67. }
  68. /*
  69. * lock for writing
  70. */
  71. static inline void __down_write(struct rw_semaphore *sem)
  72. {
  73. int tmp;
  74. tmp = atomic_add_return(RWSEM_ACTIVE_WRITE_BIAS,
  75. (atomic_t *)(&sem->count));
  76. if (tmp == RWSEM_ACTIVE_WRITE_BIAS)
  77. smp_wmb();
  78. else
  79. rwsem_down_write_failed(sem);
  80. }
  81. static inline int __down_write_trylock(struct rw_semaphore *sem)
  82. {
  83. int tmp;
  84. tmp = cmpxchg(&sem->count, RWSEM_UNLOCKED_VALUE,
  85. RWSEM_ACTIVE_WRITE_BIAS);
  86. smp_wmb();
  87. return tmp == RWSEM_UNLOCKED_VALUE;
  88. }
  89. /*
  90. * unlock after reading
  91. */
  92. static inline void __up_read(struct rw_semaphore *sem)
  93. {
  94. int tmp;
  95. smp_wmb();
  96. tmp = atomic_dec_return((atomic_t *)(&sem->count));
  97. if (tmp < -1 && (tmp & RWSEM_ACTIVE_MASK) == 0)
  98. rwsem_wake(sem);
  99. }
  100. /*
  101. * unlock after writing
  102. */
  103. static inline void __up_write(struct rw_semaphore *sem)
  104. {
  105. smp_wmb();
  106. if (atomic_sub_return(RWSEM_ACTIVE_WRITE_BIAS,
  107. (atomic_t *)(&sem->count)) < 0)
  108. rwsem_wake(sem);
  109. }
  110. /*
  111. * implement atomic add functionality
  112. */
  113. static inline void rwsem_atomic_add(int delta, struct rw_semaphore *sem)
  114. {
  115. atomic_add(delta, (atomic_t *)(&sem->count));
  116. }
  117. /*
  118. * downgrade write lock to read lock
  119. */
  120. static inline void __downgrade_write(struct rw_semaphore *sem)
  121. {
  122. int tmp;
  123. smp_wmb();
  124. tmp = atomic_add_return(-RWSEM_WAITING_BIAS, (atomic_t *)(&sem->count));
  125. if (tmp < 0)
  126. rwsem_downgrade_wake(sem);
  127. }
  128. static inline void __down_write_nested(struct rw_semaphore *sem, int subclass)
  129. {
  130. __down_write(sem);
  131. }
  132. /*
  133. * implement exchange and add functionality
  134. */
  135. static inline int rwsem_atomic_update(int delta, struct rw_semaphore *sem)
  136. {
  137. smp_mb();
  138. return atomic_add_return(delta, (atomic_t *)(&sem->count));
  139. }
  140. static inline int rwsem_is_locked(struct rw_semaphore *sem)
  141. {
  142. return (sem->count != 0);
  143. }
  144. #endif /* __KERNEL__ */
  145. #endif /* _ASM_SH_RWSEM_H */