semaphore.c 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133
  1. /*
  2. * Generic semaphore code. Buyer beware. Do your own
  3. * specific changes in <asm/semaphore-helper.h>
  4. */
  5. #include <linux/config.h>
  6. #include <linux/sched.h>
  7. #include <linux/init.h>
  8. #include <asm/semaphore-helper.h>
  9. #ifndef CONFIG_RMW_INSNS
  10. spinlock_t semaphore_wake_lock;
  11. #endif
  12. /*
  13. * Semaphores are implemented using a two-way counter:
  14. * The "count" variable is decremented for each process
  15. * that tries to sleep, while the "waking" variable is
  16. * incremented when the "up()" code goes to wake up waiting
  17. * processes.
  18. *
  19. * Notably, the inline "up()" and "down()" functions can
  20. * efficiently test if they need to do any extra work (up
  21. * needs to do something only if count was negative before
  22. * the increment operation.
  23. *
  24. * waking_non_zero() (from asm/semaphore.h) must execute
  25. * atomically.
  26. *
  27. * When __up() is called, the count was negative before
  28. * incrementing it, and we need to wake up somebody.
  29. *
  30. * This routine adds one to the count of processes that need to
  31. * wake up and exit. ALL waiting processes actually wake up but
  32. * only the one that gets to the "waking" field first will gate
  33. * through and acquire the semaphore. The others will go back
  34. * to sleep.
  35. *
  36. * Note that these functions are only called when there is
  37. * contention on the lock, and as such all this is the
  38. * "non-critical" part of the whole semaphore business. The
  39. * critical part is the inline stuff in <asm/semaphore.h>
  40. * where we want to avoid any extra jumps and calls.
  41. */
  42. void __up(struct semaphore *sem)
  43. {
  44. wake_one_more(sem);
  45. wake_up(&sem->wait);
  46. }
  47. /*
  48. * Perform the "down" function. Return zero for semaphore acquired,
  49. * return negative for signalled out of the function.
  50. *
  51. * If called from __down, the return is ignored and the wait loop is
  52. * not interruptible. This means that a task waiting on a semaphore
  53. * using "down()" cannot be killed until someone does an "up()" on
  54. * the semaphore.
  55. *
  56. * If called from __down_interruptible, the return value gets checked
  57. * upon return. If the return value is negative then the task continues
  58. * with the negative value in the return register (it can be tested by
  59. * the caller).
  60. *
  61. * Either form may be used in conjunction with "up()".
  62. *
  63. */
  64. #define DOWN_HEAD(task_state) \
  65. \
  66. \
  67. current->state = (task_state); \
  68. add_wait_queue(&sem->wait, &wait); \
  69. \
  70. /* \
  71. * Ok, we're set up. sem->count is known to be less than zero \
  72. * so we must wait. \
  73. * \
  74. * We can let go the lock for purposes of waiting. \
  75. * We re-acquire it after awaking so as to protect \
  76. * all semaphore operations. \
  77. * \
  78. * If "up()" is called before we call waking_non_zero() then \
  79. * we will catch it right away. If it is called later then \
  80. * we will have to go through a wakeup cycle to catch it. \
  81. * \
  82. * Multiple waiters contend for the semaphore lock to see \
  83. * who gets to gate through and who has to wait some more. \
  84. */ \
  85. for (;;) {
  86. #define DOWN_TAIL(task_state) \
  87. current->state = (task_state); \
  88. } \
  89. current->state = TASK_RUNNING; \
  90. remove_wait_queue(&sem->wait, &wait);
  91. void __sched __down(struct semaphore * sem)
  92. {
  93. DECLARE_WAITQUEUE(wait, current);
  94. DOWN_HEAD(TASK_UNINTERRUPTIBLE)
  95. if (waking_non_zero(sem))
  96. break;
  97. schedule();
  98. DOWN_TAIL(TASK_UNINTERRUPTIBLE)
  99. }
  100. int __sched __down_interruptible(struct semaphore * sem)
  101. {
  102. DECLARE_WAITQUEUE(wait, current);
  103. int ret = 0;
  104. DOWN_HEAD(TASK_INTERRUPTIBLE)
  105. ret = waking_non_zero_interruptible(sem, current);
  106. if (ret)
  107. {
  108. if (ret == 1)
  109. /* ret != 0 only if we get interrupted -arca */
  110. ret = 0;
  111. break;
  112. }
  113. schedule();
  114. DOWN_TAIL(TASK_INTERRUPTIBLE)
  115. return ret;
  116. }
  117. int __down_trylock(struct semaphore * sem)
  118. {
  119. return waking_non_zero_trylock(sem);
  120. }