1
2
3
4
5
6
7
8
9
10
11
12#include <linux/sched.h>
13
14#include <asm/semaphore.h>
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47void __up(struct semaphore *sem)
48{
49 wake_up(&sem->wait);
50}
51
52static spinlock_t semaphore_lock = SPIN_LOCK_UNLOCKED;
53
54void __down(struct semaphore * sem)
55{
56 struct task_struct *tsk = current;
57 DECLARE_WAITQUEUE(wait, tsk);
58 tsk->state = TASK_UNINTERRUPTIBLE;
59 add_wait_queue_exclusive(&sem->wait, &wait);
60
61 spin_lock_irq(&semaphore_lock);
62 sem->sleepers++;
63 for (;;) {
64 int sleepers = sem->sleepers;
65
66
67
68
69
70 if (!atomic_add_negative(sleepers - 1, &sem->count)) {
71 sem->sleepers = 0;
72 break;
73 }
74 sem->sleepers = 1;
75 spin_unlock_irq(&semaphore_lock);
76
77 schedule();
78 tsk->state = TASK_UNINTERRUPTIBLE;
79 spin_lock_irq(&semaphore_lock);
80 }
81 spin_unlock_irq(&semaphore_lock);
82 remove_wait_queue(&sem->wait, &wait);
83 tsk->state = TASK_RUNNING;
84 wake_up(&sem->wait);
85}
86
87int __down_interruptible(struct semaphore * sem)
88{
89 int retval = 0;
90 struct task_struct *tsk = current;
91 DECLARE_WAITQUEUE(wait, tsk);
92 tsk->state = TASK_INTERRUPTIBLE;
93 add_wait_queue_exclusive(&sem->wait, &wait);
94
95 spin_lock_irq(&semaphore_lock);
96 sem->sleepers ++;
97 for (;;) {
98 int sleepers = sem->sleepers;
99
100
101
102
103
104
105
106
107 if (signal_pending(current)) {
108 retval = -EINTR;
109 sem->sleepers = 0;
110 atomic_add(sleepers, &sem->count);
111 break;
112 }
113
114
115
116
117
118
119
120 if (!atomic_add_negative(sleepers - 1, &sem->count)) {
121 sem->sleepers = 0;
122 break;
123 }
124 sem->sleepers = 1;
125 spin_unlock_irq(&semaphore_lock);
126
127 schedule();
128 tsk->state = TASK_INTERRUPTIBLE;
129 spin_lock_irq(&semaphore_lock);
130 }
131 spin_unlock_irq(&semaphore_lock);
132 tsk->state = TASK_RUNNING;
133 remove_wait_queue(&sem->wait, &wait);
134 wake_up(&sem->wait);
135 return retval;
136}
137
138
139
140
141
142int __down_trylock(struct semaphore * sem)
143{
144 unsigned long flags;
145 int sleepers;
146
147 spin_lock_irqsave(&semaphore_lock, flags);
148 sleepers = sem->sleepers + 1;
149 sem->sleepers = 0;
150
151
152
153
154
155 if (!atomic_add_negative(sleepers, &sem->count))
156 wake_up(&sem->wait);
157
158 spin_unlock_irqrestore(&semaphore_lock, flags);
159 return 1;
160}
161