rwsem.h
上传用户:jlfgdled
上传日期:2013-04-10
资源大小:33168k
文件大小:3k
源码类别:

Linux/Unix编程

开发平台:

Unix_Linux

  1. /*
  2.  * BK Id: %F% %I% %G% %U% %#%
  3.  */
  4. /*
  5.  * include/asm-ppc/rwsem.h: R/W semaphores for PPC using the stuff
  6.  * in lib/rwsem.c.  Adapted largely from include/asm-i386/rwsem.h
  7.  * by Paul Mackerras <paulus@samba.org>.
  8.  */
  9. #ifndef _PPC_RWSEM_H
  10. #define _PPC_RWSEM_H
  11. #ifdef __KERNEL__
  12. #include <linux/list.h>
  13. #include <linux/spinlock.h>
  14. #include <asm/atomic.h>
  15. #include <asm/system.h>
  16. /*
  17.  * the semaphore definition
  18.  */
  19. struct rw_semaphore {
  20. /* XXX this should be able to be an atomic_t  -- paulus */
  21. signed long count;
  22. #define RWSEM_UNLOCKED_VALUE 0x00000000
  23. #define RWSEM_ACTIVE_BIAS 0x00000001
  24. #define RWSEM_ACTIVE_MASK 0x0000ffff
  25. #define RWSEM_WAITING_BIAS (-0x00010000)
  26. #define RWSEM_ACTIVE_READ_BIAS RWSEM_ACTIVE_BIAS
  27. #define RWSEM_ACTIVE_WRITE_BIAS (RWSEM_WAITING_BIAS + RWSEM_ACTIVE_BIAS)
  28. spinlock_t wait_lock;
  29. struct list_head wait_list;
  30. #if RWSEM_DEBUG
  31. int debug;
  32. #endif
  33. };
  34. /*
  35.  * initialisation
  36.  */
  37. #if RWSEM_DEBUG
  38. #define __RWSEM_DEBUG_INIT      , 0
  39. #else
  40. #define __RWSEM_DEBUG_INIT /* */
  41. #endif
  42. #define __RWSEM_INITIALIZER(name) 
  43. { RWSEM_UNLOCKED_VALUE, SPIN_LOCK_UNLOCKED, 
  44.   LIST_HEAD_INIT((name).wait_list) 
  45.   __RWSEM_DEBUG_INIT }
  46. #define DECLARE_RWSEM(name)
  47. struct rw_semaphore name = __RWSEM_INITIALIZER(name)
  48. extern struct rw_semaphore *rwsem_down_read_failed(struct rw_semaphore *sem);
  49. extern struct rw_semaphore *rwsem_down_write_failed(struct rw_semaphore *sem);
  50. extern struct rw_semaphore *rwsem_wake(struct rw_semaphore *sem);
  51. static inline void init_rwsem(struct rw_semaphore *sem)
  52. {
  53. sem->count = RWSEM_UNLOCKED_VALUE;
  54. spin_lock_init(&sem->wait_lock);
  55. INIT_LIST_HEAD(&sem->wait_list);
  56. #if RWSEM_DEBUG
  57. sem->debug = 0;
  58. #endif
  59. }
  60. /*
  61.  * lock for reading
  62.  */
  63. static inline void __down_read(struct rw_semaphore *sem)
  64. {
  65. if (atomic_inc_return((atomic_t *)(&sem->count)) >= 0)
  66. smp_wmb();
  67. else
  68. rwsem_down_read_failed(sem);
  69. }
  70. static inline int __down_read_trylock(struct rw_semaphore *sem)
  71. {
  72. int tmp;
  73. while ((tmp = sem->count) >= 0) {
  74. if (tmp == cmpxchg(&sem->count, tmp,
  75.    tmp + RWSEM_ACTIVE_READ_BIAS)) {
  76. smp_wmb();
  77. return 1;
  78. }
  79. }
  80. return 0;
  81. }
  82. /*
  83.  * lock for writing
  84.  */
  85. static inline void __down_write(struct rw_semaphore *sem)
  86. {
  87. int tmp;
  88. tmp = atomic_add_return(RWSEM_ACTIVE_WRITE_BIAS,
  89. (atomic_t *)(&sem->count));
  90. if (tmp == RWSEM_ACTIVE_WRITE_BIAS)
  91. smp_wmb();
  92. else
  93. rwsem_down_write_failed(sem);
  94. }
  95. static inline int __down_write_trylock(struct rw_semaphore *sem)
  96. {
  97. int tmp;
  98. tmp = cmpxchg(&sem->count, RWSEM_UNLOCKED_VALUE,
  99.       RWSEM_ACTIVE_WRITE_BIAS);
  100. smp_wmb();
  101. return tmp == RWSEM_UNLOCKED_VALUE;
  102. }
  103. /*
  104.  * unlock after reading
  105.  */
  106. static inline void __up_read(struct rw_semaphore *sem)
  107. {
  108. int tmp;
  109. smp_wmb();
  110. tmp = atomic_dec_return((atomic_t *)(&sem->count));
  111. if (tmp < -1 && (tmp & RWSEM_ACTIVE_MASK) == 0)
  112. rwsem_wake(sem);
  113. }
  114. /*
  115.  * unlock after writing
  116.  */
  117. static inline void __up_write(struct rw_semaphore *sem)
  118. {
  119. smp_wmb();
  120. if (atomic_sub_return(RWSEM_ACTIVE_WRITE_BIAS,
  121.       (atomic_t *)(&sem->count)) < 0)
  122. rwsem_wake(sem);
  123. }
  124. /*
  125.  * implement atomic add functionality
  126.  */
  127. static inline void rwsem_atomic_add(int delta, struct rw_semaphore *sem)
  128. {
  129. atomic_add(delta, (atomic_t *)(&sem->count));
  130. }
  131. /*
  132.  * implement exchange and add functionality
  133.  */
  134. static inline int rwsem_atomic_update(int delta, struct rw_semaphore *sem)
  135. {
  136. smp_mb();
  137. return atomic_add_return(delta, (atomic_t *)(&sem->count));
  138. }
  139. #endif /* __KERNEL__ */
  140. #endif /* _PPC_RWSEM_XADD_H */