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

Linux/Unix编程

开发平台:

Unix_Linux

  1. /*
  2.  * SMP- and interrupt-safe semaphores..
  3.  *
  4.  * This file is subject to the terms and conditions of the GNU General Public
  5.  * License.  See the file "COPYING" in the main directory of this archive
  6.  * for more details.
  7.  *
  8.  * (C) Copyright 1996  Linus Torvalds
  9.  * (C) Copyright 1998, 99, 2000, 01  Ralf Baechle
  10.  * (C) Copyright 1999, 2000  Silicon Graphics, Inc.
  11.  * Copyright (C) 2000, 01 MIPS Technologies, Inc.  All rights reserved.
  12.  */
  13. #ifndef _ASM_SEMAPHORE_H
  14. #define _ASM_SEMAPHORE_H
  15. #include <asm/system.h>
  16. #include <asm/atomic.h>
  17. #include <linux/spinlock.h>
  18. #include <linux/wait.h>
  19. #include <linux/config.h>
  20. #include <linux/rwsem.h>
  21. struct semaphore {
  22. #ifdef __MIPSEB__
  23. atomic_t count;
  24. atomic_t waking;
  25. #else
  26. atomic_t waking;
  27. atomic_t count;
  28. #endif
  29. wait_queue_head_t wait;
  30. #if WAITQUEUE_DEBUG
  31. long __magic;
  32. #endif
  33. } __attribute__((aligned(8)));
  34. #if WAITQUEUE_DEBUG
  35. # define __SEM_DEBUG_INIT(name) 
  36. , (long)&(name).__magic
  37. #else
  38. # define __SEM_DEBUG_INIT(name)
  39. #endif
  40. #ifdef __MIPSEB__
  41. #define __SEMAPHORE_INITIALIZER(name,count) 
  42. { ATOMIC_INIT(count), ATOMIC_INIT(0), __WAIT_QUEUE_HEAD_INITIALIZER((name).wait) 
  43. __SEM_DEBUG_INIT(name) }
  44. #else
  45. #define __SEMAPHORE_INITIALIZER(name,count) 
  46. { ATOMIC_INIT(0), ATOMIC_INIT(count), __WAIT_QUEUE_HEAD_INITIALIZER((name).wait) 
  47. __SEM_DEBUG_INIT(name) }
  48. #endif
  49. #define __MUTEX_INITIALIZER(name) 
  50. __SEMAPHORE_INITIALIZER(name,1)
  51. #define __DECLARE_SEMAPHORE_GENERIC(name,count) 
  52. struct semaphore name = __SEMAPHORE_INITIALIZER(name,count)
  53. #define DECLARE_MUTEX(name) __DECLARE_SEMAPHORE_GENERIC(name,1)
  54. #define DECLARE_MUTEX_LOCKED(name) __DECLARE_SEMAPHORE_GENERIC(name,0)
  55. static inline void sema_init (struct semaphore *sem, int val)
  56. {
  57. atomic_set(&sem->count, val);
  58. atomic_set(&sem->waking, 0);
  59. init_waitqueue_head(&sem->wait);
  60. #if WAITQUEUE_DEBUG
  61. sem->__magic = (long)&sem->__magic;
  62. #endif
  63. }
  64. static inline void init_MUTEX (struct semaphore *sem)
  65. {
  66. sema_init(sem, 1);
  67. }
  68. static inline void init_MUTEX_LOCKED (struct semaphore *sem)
  69. {
  70. sema_init(sem, 0);
  71. }
  72. asmlinkage void __down(struct semaphore * sem);
  73. asmlinkage int  __down_interruptible(struct semaphore * sem);
  74. asmlinkage int __down_trylock(struct semaphore * sem);
  75. asmlinkage void __up(struct semaphore * sem);
  76. static inline void down(struct semaphore * sem)
  77. {
  78. #if WAITQUEUE_DEBUG
  79. CHECK_MAGIC(sem->__magic);
  80. #endif
  81. if (atomic_dec_return(&sem->count) < 0)
  82. __down(sem);
  83. }
  84. /*
  85.  * Interruptible try to acquire a semaphore.  If we obtained
  86.  * it, return zero.  If we were interrupted, returns -EINTR
  87.  */
  88. static inline int down_interruptible(struct semaphore * sem)
  89. {
  90. int ret = 0;
  91. #if WAITQUEUE_DEBUG
  92. CHECK_MAGIC(sem->__magic);
  93. #endif
  94. if (atomic_dec_return(&sem->count) < 0)
  95. ret = __down_interruptible(sem);
  96. return ret;
  97. }
  98. #ifndef CONFIG_CPU_HAS_LLDSCD
  99. /*
  100.  * Non-blockingly attempt to down() a semaphore.
  101.  * Returns zero if we acquired it
  102.  */
  103. static inline int down_trylock(struct semaphore * sem)
  104. {
  105. int ret = 0;
  106. if (atomic_dec_return(&sem->count) < 0)
  107. ret = __down_trylock(sem);
  108. return ret;
  109. }
  110. #else
  111. /*
  112.  * down_trylock returns 0 on success, 1 if we failed to get the lock.
  113.  *
  114.  * We must manipulate count and waking simultaneously and atomically.
  115.  * Here, we do this by using lld/scd on the pair of 32-bit words.  This
  116.  * won't work on MIPS32 platforms, however, and must be rewritten.
  117.  *
  118.  * Pseudocode:
  119.  *
  120.  *   Decrement(sem->count)
  121.  *   If(sem->count >=0) {
  122.  * Return(SUCCESS) // resource is free
  123.  *   } else {
  124.  * If(sem->waking <= 0) { // if no wakeup pending
  125.  *    Increment(sem->count) // undo decrement
  126.  *    Return(FAILURE)
  127.  *      } else {
  128.  *    Decrement(sem->waking) // otherwise "steal" wakeup
  129.  *    Return(SUCCESS)
  130.  * }
  131.  *   }
  132.  */
  133. static inline int down_trylock(struct semaphore * sem)
  134. {
  135. long ret, tmp, tmp2, sub;
  136. #if WAITQUEUE_DEBUG
  137. CHECK_MAGIC(sem->__magic);
  138. #endif
  139. __asm__ __volatile__(
  140. ".settmips3ttt# down_trylockn"
  141. "0:tlldt%1, %4nt"
  142. "dlit%3, 0x0000000100000000nt"
  143. "dsubut%1, %3nt"
  144. "lit%0, 0nt"
  145. "bgezt%1, 2fnt"
  146. "sllt%2, %1, 0nt"
  147. "blezt%2, 1fnt"
  148. "daddiut%1, %1, -1nt"
  149. "bt2fnt"
  150. "1:tdaddut%1, %1, %3n"
  151. "lit%0, 1n"
  152. "2:tscdt%1, %4nt"
  153. "beqzt%1, 0bnt"
  154. ".set mips0"
  155. : "=&r"(ret), "=&r"(tmp), "=&r"(tmp2), "=&r"(sub)
  156. : "m"(*sem)
  157. : "memory");
  158. return ret;
  159. }
  160. #endif /* CONFIG_CPU_HAS_LLDSCD */
  161. /*
  162.  * Note! This is subtle. We jump to wake people up only if
  163.  * the semaphore was negative (== somebody was waiting on it).
  164.  */
  165. static inline void up(struct semaphore * sem)
  166. {
  167. #if WAITQUEUE_DEBUG
  168. CHECK_MAGIC(sem->__magic);
  169. #endif
  170. if (atomic_inc_return(&sem->count) <= 0)
  171. __up(sem);
  172. }
  173. static inline int sem_getcount(struct semaphore *sem)
  174. {
  175. return atomic_read(&sem->count);
  176. }
  177. #endif /* _ASM_SEMAPHORE_H */