rwsem.h 3.9 KB

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