semaphore.h 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. /*
  2. * This file is subject to the terms and conditions of the GNU General Public
  3. * License. See the file "COPYING" in the main directory of this archive
  4. * for more details.
  5. *
  6. * Copyright (C) 1996 Linus Torvalds
  7. * Copyright (C) 1998, 99, 2000, 01, 04 Ralf Baechle
  8. * Copyright (C) 1999, 2000, 01 Silicon Graphics, Inc.
  9. * Copyright (C) 2000, 01 MIPS Technologies, Inc.
  10. *
  11. * In all honesty, little of the old MIPS code left - the PPC64 variant was
  12. * just looking nice and portable so I ripped it. Credits to whoever wrote
  13. * it.
  14. */
  15. #ifndef __ASM_SEMAPHORE_H
  16. #define __ASM_SEMAPHORE_H
  17. /*
  18. * Remove spinlock-based RW semaphores; RW semaphore definitions are
  19. * now in rwsem.h and we use the generic lib/rwsem.c implementation.
  20. * Rework semaphores to use atomic_dec_if_positive.
  21. * -- Paul Mackerras (paulus@samba.org)
  22. */
  23. #ifdef __KERNEL__
  24. #include <asm/atomic.h>
  25. #include <asm/system.h>
  26. #include <linux/wait.h>
  27. #include <linux/rwsem.h>
  28. struct semaphore {
  29. /*
  30. * Note that any negative value of count is equivalent to 0,
  31. * but additionally indicates that some process(es) might be
  32. * sleeping on `wait'.
  33. */
  34. atomic_t count;
  35. wait_queue_head_t wait;
  36. };
  37. #define __SEMAPHORE_INITIALIZER(name, n) \
  38. { \
  39. .count = ATOMIC_INIT(n), \
  40. .wait = __WAIT_QUEUE_HEAD_INITIALIZER((name).wait) \
  41. }
  42. #define __DECLARE_SEMAPHORE_GENERIC(name, count) \
  43. struct semaphore name = __SEMAPHORE_INITIALIZER(name, count)
  44. #define DECLARE_MUTEX(name) __DECLARE_SEMAPHORE_GENERIC(name, 1)
  45. static inline void sema_init(struct semaphore *sem, int val)
  46. {
  47. atomic_set(&sem->count, val);
  48. init_waitqueue_head(&sem->wait);
  49. }
  50. static inline void init_MUTEX(struct semaphore *sem)
  51. {
  52. sema_init(sem, 1);
  53. }
  54. static inline void init_MUTEX_LOCKED(struct semaphore *sem)
  55. {
  56. sema_init(sem, 0);
  57. }
  58. extern void __down(struct semaphore * sem);
  59. extern int __down_interruptible(struct semaphore * sem);
  60. extern void __up(struct semaphore * sem);
  61. static inline void down(struct semaphore * sem)
  62. {
  63. might_sleep();
  64. /*
  65. * Try to get the semaphore, take the slow path if we fail.
  66. */
  67. if (unlikely(atomic_dec_return(&sem->count) < 0))
  68. __down(sem);
  69. }
  70. static inline int down_interruptible(struct semaphore * sem)
  71. {
  72. int ret = 0;
  73. might_sleep();
  74. if (unlikely(atomic_dec_return(&sem->count) < 0))
  75. ret = __down_interruptible(sem);
  76. return ret;
  77. }
  78. static inline int down_trylock(struct semaphore * sem)
  79. {
  80. return atomic_dec_if_positive(&sem->count) < 0;
  81. }
  82. static inline void up(struct semaphore * sem)
  83. {
  84. if (unlikely(atomic_inc_return(&sem->count) <= 0))
  85. __up(sem);
  86. }
  87. #endif /* __KERNEL__ */
  88. #endif /* __ASM_SEMAPHORE_H */