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

Linux/Unix编程

开发平台:

Unix_Linux

  1. /*
  2.  * Semaphore implementation Copyright (c) 2001 Matthew Wilcox, Hewlett-Packard
  3.  */
  4. #include <linux/sched.h>
  5. #include <linux/spinlock.h>
  6. /*
  7.  * Semaphores are complex as we wish to avoid using two variables.
  8.  * `count' has multiple roles, depending on its value.  If it is positive
  9.  * or zero, there are no waiters.  The functions here will never be
  10.  * called; see <asm/semaphore.h>
  11.  *
  12.  * When count is -1 it indicates there is at least one task waiting
  13.  * for the semaphore.
  14.  *
  15.  * When count is less than that, there are '- count - 1' wakeups
  16.  * pending.  ie if it has value -3, there are 2 wakeups pending.
  17.  *
  18.  * Note that these functions are only called when there is contention
  19.  * on the lock, and as such all this is the "non-critical" part of the
  20.  * whole semaphore business. The critical part is the inline stuff in
  21.  * <asm/semaphore.h> where we want to avoid any extra jumps and calls.
  22.  */
  23. void __up(struct semaphore *sem)
  24. {
  25. sem->count--;
  26. wake_up(&sem->wait);
  27. }
  28. #define wakers(count) (-1 - count)
  29. #define DOWN_HEAD
  30. int ret = 0;
  31. DECLARE_WAITQUEUE(wait, current);
  32. /* Note that someone is waiting */
  33. if (sem->count == 0)
  34. sem->count = -1;
  35. /* protected by the sentry still -- use unlocked version */
  36. wait.flags = WQ_FLAG_EXCLUSIVE;
  37. __add_wait_queue_tail(&sem->wait, &wait);
  38.  lost_race:
  39. spin_unlock_irq(&sem->sentry);
  40. #define DOWN_TAIL
  41. spin_lock_irq(&sem->sentry);
  42. if (wakers(sem->count) == 0 && ret == 0)
  43. goto lost_race; /* Someone stole our wakeup */
  44. __remove_wait_queue(&sem->wait, &wait);
  45. current->state = TASK_RUNNING;
  46. if (!waitqueue_active(&sem->wait) && (sem->count < 0))
  47. sem->count = wakers(sem->count);
  48. #define UPDATE_COUNT
  49. sem->count += (sem->count < 0) ? 1 : - 1;
  50. void __down(struct semaphore * sem)
  51. {
  52. DOWN_HEAD
  53. for(;;) {
  54. set_task_state(current, TASK_UNINTERRUPTIBLE);
  55. /* we can _read_ this without the sentry */
  56. if (sem->count != -1)
  57. break;
  58.   schedule();
  59.   }
  60. DOWN_TAIL
  61. UPDATE_COUNT
  62. }
  63. int __down_interruptible(struct semaphore * sem)
  64. {
  65. DOWN_HEAD
  66. for(;;) {
  67. set_task_state(current, TASK_INTERRUPTIBLE);
  68. /* we can _read_ this without the sentry */
  69. if (sem->count != -1)
  70. break;
  71. if (signal_pending(current)) {
  72. ret = -EINTR;
  73. break;
  74. }
  75. schedule();
  76. }
  77. DOWN_TAIL
  78. if (!ret) {
  79. UPDATE_COUNT
  80. }
  81. return ret;
  82. }