blob: fb524b009eefeca3ffd2a2a929421a7eeb997660 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * Copyright (2004) Linus Torvalds
3 *
4 * Author: Zwane Mwaikambo <zwane@fsmlabs.com>
5 *
Ingo Molnarfb1c8f92005-09-10 00:25:56 -07006 * Copyright (2004, 2005) Ingo Molnar
7 *
8 * This file contains the spinlock/rwlock implementations for the
9 * SMP and the DEBUG_SPINLOCK cases. (UP-nondebug inlines them)
Linus Torvalds1da177e2005-04-16 15:20:36 -070010 */
11
Linus Torvalds1da177e2005-04-16 15:20:36 -070012#include <linux/linkage.h>
13#include <linux/preempt.h>
14#include <linux/spinlock.h>
15#include <linux/interrupt.h>
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070016#include <linux/debug_locks.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070017#include <linux/module.h>
18
19/*
20 * Generic declaration of the raw read_trylock() function,
21 * architectures are supposed to optimize this:
22 */
Ingo Molnarfb1c8f92005-09-10 00:25:56 -070023int __lockfunc generic__raw_read_trylock(raw_rwlock_t *lock)
Linus Torvalds1da177e2005-04-16 15:20:36 -070024{
Ingo Molnarfb1c8f92005-09-10 00:25:56 -070025 __raw_read_lock(lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -070026 return 1;
27}
Ingo Molnarfb1c8f92005-09-10 00:25:56 -070028EXPORT_SYMBOL(generic__raw_read_trylock);
Linus Torvalds1da177e2005-04-16 15:20:36 -070029
30int __lockfunc _spin_trylock(spinlock_t *lock)
31{
32 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070033 if (_raw_spin_trylock(lock)) {
34 spin_acquire(&lock->dep_map, 0, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -070035 return 1;
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070036 }
Linus Torvalds1da177e2005-04-16 15:20:36 -070037
38 preempt_enable();
39 return 0;
40}
41EXPORT_SYMBOL(_spin_trylock);
42
43int __lockfunc _read_trylock(rwlock_t *lock)
44{
45 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070046 if (_raw_read_trylock(lock)) {
47 rwlock_acquire_read(&lock->dep_map, 0, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -070048 return 1;
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070049 }
Linus Torvalds1da177e2005-04-16 15:20:36 -070050
51 preempt_enable();
52 return 0;
53}
54EXPORT_SYMBOL(_read_trylock);
55
56int __lockfunc _write_trylock(rwlock_t *lock)
57{
58 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070059 if (_raw_write_trylock(lock)) {
60 rwlock_acquire(&lock->dep_map, 0, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -070061 return 1;
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070062 }
Linus Torvalds1da177e2005-04-16 15:20:36 -070063
64 preempt_enable();
65 return 0;
66}
67EXPORT_SYMBOL(_write_trylock);
68
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070069/*
70 * If lockdep is enabled then we use the non-preemption spin-ops
71 * even on CONFIG_PREEMPT, because lockdep assumes that interrupts are
72 * not re-enabled during lock-acquire (which the preempt-spin-ops do):
73 */
74#if !defined(CONFIG_PREEMPT) || !defined(CONFIG_SMP) || \
Jarek Poplawskifc47e7b2006-09-06 00:03:39 -070075 defined(CONFIG_DEBUG_LOCK_ALLOC)
Linus Torvalds1da177e2005-04-16 15:20:36 -070076
77void __lockfunc _read_lock(rwlock_t *lock)
78{
79 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070080 rwlock_acquire_read(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -070081 _raw_read_lock(lock);
82}
83EXPORT_SYMBOL(_read_lock);
84
85unsigned long __lockfunc _spin_lock_irqsave(spinlock_t *lock)
86{
87 unsigned long flags;
88
89 local_irq_save(flags);
90 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -070091 spin_acquire(&lock->dep_map, 0, 0, _RET_IP_);
92 /*
93 * On lockdep we dont want the hand-coded irq-enable of
94 * _raw_spin_lock_flags() code, because lockdep assumes
95 * that interrupts are not re-enabled during lock-acquire:
96 */
97#ifdef CONFIG_PROVE_LOCKING
98 _raw_spin_lock(lock);
99#else
Ingo Molnarfb1c8f92005-09-10 00:25:56 -0700100 _raw_spin_lock_flags(lock, &flags);
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700101#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -0700102 return flags;
103}
104EXPORT_SYMBOL(_spin_lock_irqsave);
105
106void __lockfunc _spin_lock_irq(spinlock_t *lock)
107{
108 local_irq_disable();
109 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700110 spin_acquire(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700111 _raw_spin_lock(lock);
112}
113EXPORT_SYMBOL(_spin_lock_irq);
114
115void __lockfunc _spin_lock_bh(spinlock_t *lock)
116{
117 local_bh_disable();
118 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700119 spin_acquire(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700120 _raw_spin_lock(lock);
121}
122EXPORT_SYMBOL(_spin_lock_bh);
123
124unsigned long __lockfunc _read_lock_irqsave(rwlock_t *lock)
125{
126 unsigned long flags;
127
128 local_irq_save(flags);
129 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700130 rwlock_acquire_read(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700131 _raw_read_lock(lock);
132 return flags;
133}
134EXPORT_SYMBOL(_read_lock_irqsave);
135
136void __lockfunc _read_lock_irq(rwlock_t *lock)
137{
138 local_irq_disable();
139 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700140 rwlock_acquire_read(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700141 _raw_read_lock(lock);
142}
143EXPORT_SYMBOL(_read_lock_irq);
144
145void __lockfunc _read_lock_bh(rwlock_t *lock)
146{
147 local_bh_disable();
148 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700149 rwlock_acquire_read(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700150 _raw_read_lock(lock);
151}
152EXPORT_SYMBOL(_read_lock_bh);
153
154unsigned long __lockfunc _write_lock_irqsave(rwlock_t *lock)
155{
156 unsigned long flags;
157
158 local_irq_save(flags);
159 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700160 rwlock_acquire(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700161 _raw_write_lock(lock);
162 return flags;
163}
164EXPORT_SYMBOL(_write_lock_irqsave);
165
166void __lockfunc _write_lock_irq(rwlock_t *lock)
167{
168 local_irq_disable();
169 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700170 rwlock_acquire(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700171 _raw_write_lock(lock);
172}
173EXPORT_SYMBOL(_write_lock_irq);
174
175void __lockfunc _write_lock_bh(rwlock_t *lock)
176{
177 local_bh_disable();
178 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700179 rwlock_acquire(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700180 _raw_write_lock(lock);
181}
182EXPORT_SYMBOL(_write_lock_bh);
183
184void __lockfunc _spin_lock(spinlock_t *lock)
185{
186 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700187 spin_acquire(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700188 _raw_spin_lock(lock);
189}
190
191EXPORT_SYMBOL(_spin_lock);
192
193void __lockfunc _write_lock(rwlock_t *lock)
194{
195 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700196 rwlock_acquire(&lock->dep_map, 0, 0, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700197 _raw_write_lock(lock);
198}
199
200EXPORT_SYMBOL(_write_lock);
201
202#else /* CONFIG_PREEMPT: */
203
204/*
205 * This could be a long-held lock. We both prepare to spin for a long
206 * time (making _this_ CPU preemptable if possible), and we also signal
207 * towards that other CPU that it should break the lock ASAP.
208 *
209 * (We do this in a function because inlining it would be excessive.)
210 */
211
212#define BUILD_LOCK_OPS(op, locktype) \
213void __lockfunc _##op##_lock(locktype##_t *lock) \
214{ \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700215 for (;;) { \
Oleg Nesterovee25e962006-03-23 03:00:58 -0800216 preempt_disable(); \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700217 if (likely(_raw_##op##_trylock(lock))) \
218 break; \
219 preempt_enable(); \
Oleg Nesterovee25e962006-03-23 03:00:58 -0800220 \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700221 if (!(lock)->break_lock) \
222 (lock)->break_lock = 1; \
223 while (!op##_can_lock(lock) && (lock)->break_lock) \
224 cpu_relax(); \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700225 } \
226 (lock)->break_lock = 0; \
227} \
228 \
229EXPORT_SYMBOL(_##op##_lock); \
230 \
231unsigned long __lockfunc _##op##_lock_irqsave(locktype##_t *lock) \
232{ \
233 unsigned long flags; \
234 \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700235 for (;;) { \
Oleg Nesterovee25e962006-03-23 03:00:58 -0800236 preempt_disable(); \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700237 local_irq_save(flags); \
238 if (likely(_raw_##op##_trylock(lock))) \
239 break; \
240 local_irq_restore(flags); \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700241 preempt_enable(); \
Oleg Nesterovee25e962006-03-23 03:00:58 -0800242 \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700243 if (!(lock)->break_lock) \
244 (lock)->break_lock = 1; \
245 while (!op##_can_lock(lock) && (lock)->break_lock) \
246 cpu_relax(); \
Linus Torvalds1da177e2005-04-16 15:20:36 -0700247 } \
248 (lock)->break_lock = 0; \
249 return flags; \
250} \
251 \
252EXPORT_SYMBOL(_##op##_lock_irqsave); \
253 \
254void __lockfunc _##op##_lock_irq(locktype##_t *lock) \
255{ \
256 _##op##_lock_irqsave(lock); \
257} \
258 \
259EXPORT_SYMBOL(_##op##_lock_irq); \
260 \
261void __lockfunc _##op##_lock_bh(locktype##_t *lock) \
262{ \
263 unsigned long flags; \
264 \
265 /* */ \
266 /* Careful: we must exclude softirqs too, hence the */ \
267 /* irq-disabling. We use the generic preemption-aware */ \
268 /* function: */ \
269 /**/ \
270 flags = _##op##_lock_irqsave(lock); \
271 local_bh_disable(); \
272 local_irq_restore(flags); \
273} \
274 \
275EXPORT_SYMBOL(_##op##_lock_bh)
276
277/*
278 * Build preemption-friendly versions of the following
279 * lock-spinning functions:
280 *
281 * _[spin|read|write]_lock()
282 * _[spin|read|write]_lock_irq()
283 * _[spin|read|write]_lock_irqsave()
284 * _[spin|read|write]_lock_bh()
285 */
286BUILD_LOCK_OPS(spin, spinlock);
287BUILD_LOCK_OPS(read, rwlock);
288BUILD_LOCK_OPS(write, rwlock);
289
290#endif /* CONFIG_PREEMPT */
291
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700292#ifdef CONFIG_DEBUG_LOCK_ALLOC
293
294void __lockfunc _spin_lock_nested(spinlock_t *lock, int subclass)
295{
296 preempt_disable();
297 spin_acquire(&lock->dep_map, subclass, 0, _RET_IP_);
298 _raw_spin_lock(lock);
299}
300
301EXPORT_SYMBOL(_spin_lock_nested);
302
303#endif
304
Linus Torvalds1da177e2005-04-16 15:20:36 -0700305void __lockfunc _spin_unlock(spinlock_t *lock)
306{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700307 spin_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700308 _raw_spin_unlock(lock);
309 preempt_enable();
310}
311EXPORT_SYMBOL(_spin_unlock);
312
313void __lockfunc _write_unlock(rwlock_t *lock)
314{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700315 rwlock_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700316 _raw_write_unlock(lock);
317 preempt_enable();
318}
319EXPORT_SYMBOL(_write_unlock);
320
321void __lockfunc _read_unlock(rwlock_t *lock)
322{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700323 rwlock_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700324 _raw_read_unlock(lock);
325 preempt_enable();
326}
327EXPORT_SYMBOL(_read_unlock);
328
329void __lockfunc _spin_unlock_irqrestore(spinlock_t *lock, unsigned long flags)
330{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700331 spin_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700332 _raw_spin_unlock(lock);
333 local_irq_restore(flags);
334 preempt_enable();
335}
336EXPORT_SYMBOL(_spin_unlock_irqrestore);
337
338void __lockfunc _spin_unlock_irq(spinlock_t *lock)
339{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700340 spin_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700341 _raw_spin_unlock(lock);
342 local_irq_enable();
343 preempt_enable();
344}
345EXPORT_SYMBOL(_spin_unlock_irq);
346
347void __lockfunc _spin_unlock_bh(spinlock_t *lock)
348{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700349 spin_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700350 _raw_spin_unlock(lock);
Samuel Thibault10f02d1c2005-05-21 17:50:15 +0200351 preempt_enable_no_resched();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700352 local_bh_enable_ip((unsigned long)__builtin_return_address(0));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700353}
354EXPORT_SYMBOL(_spin_unlock_bh);
355
356void __lockfunc _read_unlock_irqrestore(rwlock_t *lock, unsigned long flags)
357{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700358 rwlock_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700359 _raw_read_unlock(lock);
360 local_irq_restore(flags);
361 preempt_enable();
362}
363EXPORT_SYMBOL(_read_unlock_irqrestore);
364
365void __lockfunc _read_unlock_irq(rwlock_t *lock)
366{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700367 rwlock_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700368 _raw_read_unlock(lock);
369 local_irq_enable();
370 preempt_enable();
371}
372EXPORT_SYMBOL(_read_unlock_irq);
373
374void __lockfunc _read_unlock_bh(rwlock_t *lock)
375{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700376 rwlock_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700377 _raw_read_unlock(lock);
Samuel Thibault10f02d1c2005-05-21 17:50:15 +0200378 preempt_enable_no_resched();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700379 local_bh_enable_ip((unsigned long)__builtin_return_address(0));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700380}
381EXPORT_SYMBOL(_read_unlock_bh);
382
383void __lockfunc _write_unlock_irqrestore(rwlock_t *lock, unsigned long flags)
384{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700385 rwlock_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700386 _raw_write_unlock(lock);
387 local_irq_restore(flags);
388 preempt_enable();
389}
390EXPORT_SYMBOL(_write_unlock_irqrestore);
391
392void __lockfunc _write_unlock_irq(rwlock_t *lock)
393{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700394 rwlock_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700395 _raw_write_unlock(lock);
396 local_irq_enable();
397 preempt_enable();
398}
399EXPORT_SYMBOL(_write_unlock_irq);
400
401void __lockfunc _write_unlock_bh(rwlock_t *lock)
402{
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700403 rwlock_release(&lock->dep_map, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700404 _raw_write_unlock(lock);
Samuel Thibault10f02d1c2005-05-21 17:50:15 +0200405 preempt_enable_no_resched();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700406 local_bh_enable_ip((unsigned long)__builtin_return_address(0));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700407}
408EXPORT_SYMBOL(_write_unlock_bh);
409
410int __lockfunc _spin_trylock_bh(spinlock_t *lock)
411{
412 local_bh_disable();
413 preempt_disable();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700414 if (_raw_spin_trylock(lock)) {
415 spin_acquire(&lock->dep_map, 0, 1, _RET_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700416 return 1;
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700417 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700418
Samuel Thibault10f02d1c2005-05-21 17:50:15 +0200419 preempt_enable_no_resched();
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700420 local_bh_enable_ip((unsigned long)__builtin_return_address(0));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700421 return 0;
422}
423EXPORT_SYMBOL(_spin_trylock_bh);
424
425int in_lock_functions(unsigned long addr)
426{
427 /* Linker adds these: start and end of __lockfunc functions */
428 extern char __lock_text_start[], __lock_text_end[];
429
430 return addr >= (unsigned long)__lock_text_start
431 && addr < (unsigned long)__lock_text_end;
432}
433EXPORT_SYMBOL(in_lock_functions);