semaphore.c
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:4k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

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