Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/net
[pandora-kernel.git] / lib / locking-selftest.c
1 /*
2  * lib/locking-selftest.c
3  *
4  * Testsuite for various locking APIs: spinlocks, rwlocks,
5  * mutexes and rw-semaphores.
6  *
7  * It is checking both false positives and false negatives.
8  *
9  * Started by Ingo Molnar:
10  *
11  *  Copyright (C) 2006 Red Hat, Inc., Ingo Molnar <mingo@redhat.com>
12  */
13 #include <linux/rwsem.h>
14 #include <linux/mutex.h>
15 #include <linux/ww_mutex.h>
16 #include <linux/sched.h>
17 #include <linux/delay.h>
18 #include <linux/lockdep.h>
19 #include <linux/spinlock.h>
20 #include <linux/kallsyms.h>
21 #include <linux/interrupt.h>
22 #include <linux/debug_locks.h>
23 #include <linux/irqflags.h>
24
25 /*
26  * Change this to 1 if you want to see the failure printouts:
27  */
28 static unsigned int debug_locks_verbose;
29
30 static DEFINE_WW_CLASS(ww_lockdep);
31
32 static int __init setup_debug_locks_verbose(char *str)
33 {
34         get_option(&str, &debug_locks_verbose);
35
36         return 1;
37 }
38
39 __setup("debug_locks_verbose=", setup_debug_locks_verbose);
40
41 #define FAILURE         0
42 #define SUCCESS         1
43
44 #define LOCKTYPE_SPIN   0x1
45 #define LOCKTYPE_RWLOCK 0x2
46 #define LOCKTYPE_MUTEX  0x4
47 #define LOCKTYPE_RWSEM  0x8
48 #define LOCKTYPE_WW     0x10
49
50 static struct ww_acquire_ctx t, t2;
51 static struct ww_mutex o, o2, o3;
52
53 /*
54  * Normal standalone locks, for the circular and irq-context
55  * dependency tests:
56  */
57 static DEFINE_RAW_SPINLOCK(lock_A);
58 static DEFINE_RAW_SPINLOCK(lock_B);
59 static DEFINE_RAW_SPINLOCK(lock_C);
60 static DEFINE_RAW_SPINLOCK(lock_D);
61
62 static DEFINE_RWLOCK(rwlock_A);
63 static DEFINE_RWLOCK(rwlock_B);
64 static DEFINE_RWLOCK(rwlock_C);
65 static DEFINE_RWLOCK(rwlock_D);
66
67 static DEFINE_MUTEX(mutex_A);
68 static DEFINE_MUTEX(mutex_B);
69 static DEFINE_MUTEX(mutex_C);
70 static DEFINE_MUTEX(mutex_D);
71
72 static DECLARE_RWSEM(rwsem_A);
73 static DECLARE_RWSEM(rwsem_B);
74 static DECLARE_RWSEM(rwsem_C);
75 static DECLARE_RWSEM(rwsem_D);
76
77 /*
78  * Locks that we initialize dynamically as well so that
79  * e.g. X1 and X2 becomes two instances of the same class,
80  * but X* and Y* are different classes. We do this so that
81  * we do not trigger a real lockup:
82  */
83 static DEFINE_RAW_SPINLOCK(lock_X1);
84 static DEFINE_RAW_SPINLOCK(lock_X2);
85 static DEFINE_RAW_SPINLOCK(lock_Y1);
86 static DEFINE_RAW_SPINLOCK(lock_Y2);
87 static DEFINE_RAW_SPINLOCK(lock_Z1);
88 static DEFINE_RAW_SPINLOCK(lock_Z2);
89
90 static DEFINE_RWLOCK(rwlock_X1);
91 static DEFINE_RWLOCK(rwlock_X2);
92 static DEFINE_RWLOCK(rwlock_Y1);
93 static DEFINE_RWLOCK(rwlock_Y2);
94 static DEFINE_RWLOCK(rwlock_Z1);
95 static DEFINE_RWLOCK(rwlock_Z2);
96
97 static DEFINE_MUTEX(mutex_X1);
98 static DEFINE_MUTEX(mutex_X2);
99 static DEFINE_MUTEX(mutex_Y1);
100 static DEFINE_MUTEX(mutex_Y2);
101 static DEFINE_MUTEX(mutex_Z1);
102 static DEFINE_MUTEX(mutex_Z2);
103
104 static DECLARE_RWSEM(rwsem_X1);
105 static DECLARE_RWSEM(rwsem_X2);
106 static DECLARE_RWSEM(rwsem_Y1);
107 static DECLARE_RWSEM(rwsem_Y2);
108 static DECLARE_RWSEM(rwsem_Z1);
109 static DECLARE_RWSEM(rwsem_Z2);
110
111 /*
112  * non-inlined runtime initializers, to let separate locks share
113  * the same lock-class:
114  */
115 #define INIT_CLASS_FUNC(class)                          \
116 static noinline void                                    \
117 init_class_##class(raw_spinlock_t *lock, rwlock_t *rwlock, \
118         struct mutex *mutex, struct rw_semaphore *rwsem)\
119 {                                                       \
120         raw_spin_lock_init(lock);                       \
121         rwlock_init(rwlock);                            \
122         mutex_init(mutex);                              \
123         init_rwsem(rwsem);                              \
124 }
125
126 INIT_CLASS_FUNC(X)
127 INIT_CLASS_FUNC(Y)
128 INIT_CLASS_FUNC(Z)
129
130 static void init_shared_classes(void)
131 {
132         init_class_X(&lock_X1, &rwlock_X1, &mutex_X1, &rwsem_X1);
133         init_class_X(&lock_X2, &rwlock_X2, &mutex_X2, &rwsem_X2);
134
135         init_class_Y(&lock_Y1, &rwlock_Y1, &mutex_Y1, &rwsem_Y1);
136         init_class_Y(&lock_Y2, &rwlock_Y2, &mutex_Y2, &rwsem_Y2);
137
138         init_class_Z(&lock_Z1, &rwlock_Z1, &mutex_Z1, &rwsem_Z1);
139         init_class_Z(&lock_Z2, &rwlock_Z2, &mutex_Z2, &rwsem_Z2);
140 }
141
142 /*
143  * For spinlocks and rwlocks we also do hardirq-safe / softirq-safe tests.
144  * The following functions use a lock from a simulated hardirq/softirq
145  * context, causing the locks to be marked as hardirq-safe/softirq-safe:
146  */
147
148 #define HARDIRQ_DISABLE         local_irq_disable
149 #define HARDIRQ_ENABLE          local_irq_enable
150
151 #define HARDIRQ_ENTER()                         \
152         local_irq_disable();                    \
153         __irq_enter();                          \
154         WARN_ON(!in_irq());
155
156 #define HARDIRQ_EXIT()                          \
157         __irq_exit();                           \
158         local_irq_enable();
159
160 #define SOFTIRQ_DISABLE         local_bh_disable
161 #define SOFTIRQ_ENABLE          local_bh_enable
162
163 #define SOFTIRQ_ENTER()                         \
164                 local_bh_disable();             \
165                 local_irq_disable();            \
166                 lockdep_softirq_enter();        \
167                 WARN_ON(!in_softirq());
168
169 #define SOFTIRQ_EXIT()                          \
170                 lockdep_softirq_exit();         \
171                 local_irq_enable();             \
172                 local_bh_enable();
173
174 /*
175  * Shortcuts for lock/unlock API variants, to keep
176  * the testcases compact:
177  */
178 #define L(x)                    raw_spin_lock(&lock_##x)
179 #define U(x)                    raw_spin_unlock(&lock_##x)
180 #define LU(x)                   L(x); U(x)
181 #define SI(x)                   raw_spin_lock_init(&lock_##x)
182
183 #define WL(x)                   write_lock(&rwlock_##x)
184 #define WU(x)                   write_unlock(&rwlock_##x)
185 #define WLU(x)                  WL(x); WU(x)
186
187 #define RL(x)                   read_lock(&rwlock_##x)
188 #define RU(x)                   read_unlock(&rwlock_##x)
189 #define RLU(x)                  RL(x); RU(x)
190 #define RWI(x)                  rwlock_init(&rwlock_##x)
191
192 #define ML(x)                   mutex_lock(&mutex_##x)
193 #define MU(x)                   mutex_unlock(&mutex_##x)
194 #define MI(x)                   mutex_init(&mutex_##x)
195
196 #define WSL(x)                  down_write(&rwsem_##x)
197 #define WSU(x)                  up_write(&rwsem_##x)
198
199 #define RSL(x)                  down_read(&rwsem_##x)
200 #define RSU(x)                  up_read(&rwsem_##x)
201 #define RWSI(x)                 init_rwsem(&rwsem_##x)
202
203 #ifndef CONFIG_DEBUG_WW_MUTEX_SLOWPATH
204 #define WWAI(x)                 ww_acquire_init(x, &ww_lockdep)
205 #else
206 #define WWAI(x)                 do { ww_acquire_init(x, &ww_lockdep); (x)->deadlock_inject_countdown = ~0U; } while (0)
207 #endif
208 #define WWAD(x)                 ww_acquire_done(x)
209 #define WWAF(x)                 ww_acquire_fini(x)
210
211 #define WWL(x, c)               ww_mutex_lock(x, c)
212 #define WWT(x)                  ww_mutex_trylock(x)
213 #define WWL1(x)                 ww_mutex_lock(x, NULL)
214 #define WWU(x)                  ww_mutex_unlock(x)
215
216
217 #define LOCK_UNLOCK_2(x,y)      LOCK(x); LOCK(y); UNLOCK(y); UNLOCK(x)
218
219 /*
220  * Generate different permutations of the same testcase, using
221  * the same basic lock-dependency/state events:
222  */
223
224 #define GENERATE_TESTCASE(name)                 \
225                                                 \
226 static void name(void) { E(); }
227
228 #define GENERATE_PERMUTATIONS_2_EVENTS(name)    \
229                                                 \
230 static void name##_12(void) { E1(); E2(); }     \
231 static void name##_21(void) { E2(); E1(); }
232
233 #define GENERATE_PERMUTATIONS_3_EVENTS(name)            \
234                                                         \
235 static void name##_123(void) { E1(); E2(); E3(); }      \
236 static void name##_132(void) { E1(); E3(); E2(); }      \
237 static void name##_213(void) { E2(); E1(); E3(); }      \
238 static void name##_231(void) { E2(); E3(); E1(); }      \
239 static void name##_312(void) { E3(); E1(); E2(); }      \
240 static void name##_321(void) { E3(); E2(); E1(); }
241
242 /*
243  * AA deadlock:
244  */
245
246 #define E()                                     \
247                                                 \
248         LOCK(X1);                               \
249         LOCK(X2); /* this one should fail */
250
251 /*
252  * 6 testcases:
253  */
254 #include "locking-selftest-spin.h"
255 GENERATE_TESTCASE(AA_spin)
256 #include "locking-selftest-wlock.h"
257 GENERATE_TESTCASE(AA_wlock)
258 #include "locking-selftest-rlock.h"
259 GENERATE_TESTCASE(AA_rlock)
260 #include "locking-selftest-mutex.h"
261 GENERATE_TESTCASE(AA_mutex)
262 #include "locking-selftest-wsem.h"
263 GENERATE_TESTCASE(AA_wsem)
264 #include "locking-selftest-rsem.h"
265 GENERATE_TESTCASE(AA_rsem)
266
267 #undef E
268
269 /*
270  * Special-case for read-locking, they are
271  * allowed to recurse on the same lock class:
272  */
273 static void rlock_AA1(void)
274 {
275         RL(X1);
276         RL(X1); // this one should NOT fail
277 }
278
279 static void rlock_AA1B(void)
280 {
281         RL(X1);
282         RL(X2); // this one should NOT fail
283 }
284
285 static void rsem_AA1(void)
286 {
287         RSL(X1);
288         RSL(X1); // this one should fail
289 }
290
291 static void rsem_AA1B(void)
292 {
293         RSL(X1);
294         RSL(X2); // this one should fail
295 }
296 /*
297  * The mixing of read and write locks is not allowed:
298  */
299 static void rlock_AA2(void)
300 {
301         RL(X1);
302         WL(X2); // this one should fail
303 }
304
305 static void rsem_AA2(void)
306 {
307         RSL(X1);
308         WSL(X2); // this one should fail
309 }
310
311 static void rlock_AA3(void)
312 {
313         WL(X1);
314         RL(X2); // this one should fail
315 }
316
317 static void rsem_AA3(void)
318 {
319         WSL(X1);
320         RSL(X2); // this one should fail
321 }
322
323 /*
324  * ABBA deadlock:
325  */
326
327 #define E()                                     \
328                                                 \
329         LOCK_UNLOCK_2(A, B);                    \
330         LOCK_UNLOCK_2(B, A); /* fail */
331
332 /*
333  * 6 testcases:
334  */
335 #include "locking-selftest-spin.h"
336 GENERATE_TESTCASE(ABBA_spin)
337 #include "locking-selftest-wlock.h"
338 GENERATE_TESTCASE(ABBA_wlock)
339 #include "locking-selftest-rlock.h"
340 GENERATE_TESTCASE(ABBA_rlock)
341 #include "locking-selftest-mutex.h"
342 GENERATE_TESTCASE(ABBA_mutex)
343 #include "locking-selftest-wsem.h"
344 GENERATE_TESTCASE(ABBA_wsem)
345 #include "locking-selftest-rsem.h"
346 GENERATE_TESTCASE(ABBA_rsem)
347
348 #undef E
349
350 /*
351  * AB BC CA deadlock:
352  */
353
354 #define E()                                     \
355                                                 \
356         LOCK_UNLOCK_2(A, B);                    \
357         LOCK_UNLOCK_2(B, C);                    \
358         LOCK_UNLOCK_2(C, A); /* fail */
359
360 /*
361  * 6 testcases:
362  */
363 #include "locking-selftest-spin.h"
364 GENERATE_TESTCASE(ABBCCA_spin)
365 #include "locking-selftest-wlock.h"
366 GENERATE_TESTCASE(ABBCCA_wlock)
367 #include "locking-selftest-rlock.h"
368 GENERATE_TESTCASE(ABBCCA_rlock)
369 #include "locking-selftest-mutex.h"
370 GENERATE_TESTCASE(ABBCCA_mutex)
371 #include "locking-selftest-wsem.h"
372 GENERATE_TESTCASE(ABBCCA_wsem)
373 #include "locking-selftest-rsem.h"
374 GENERATE_TESTCASE(ABBCCA_rsem)
375
376 #undef E
377
378 /*
379  * AB CA BC deadlock:
380  */
381
382 #define E()                                     \
383                                                 \
384         LOCK_UNLOCK_2(A, B);                    \
385         LOCK_UNLOCK_2(C, A);                    \
386         LOCK_UNLOCK_2(B, C); /* fail */
387
388 /*
389  * 6 testcases:
390  */
391 #include "locking-selftest-spin.h"
392 GENERATE_TESTCASE(ABCABC_spin)
393 #include "locking-selftest-wlock.h"
394 GENERATE_TESTCASE(ABCABC_wlock)
395 #include "locking-selftest-rlock.h"
396 GENERATE_TESTCASE(ABCABC_rlock)
397 #include "locking-selftest-mutex.h"
398 GENERATE_TESTCASE(ABCABC_mutex)
399 #include "locking-selftest-wsem.h"
400 GENERATE_TESTCASE(ABCABC_wsem)
401 #include "locking-selftest-rsem.h"
402 GENERATE_TESTCASE(ABCABC_rsem)
403
404 #undef E
405
406 /*
407  * AB BC CD DA deadlock:
408  */
409
410 #define E()                                     \
411                                                 \
412         LOCK_UNLOCK_2(A, B);                    \
413         LOCK_UNLOCK_2(B, C);                    \
414         LOCK_UNLOCK_2(C, D);                    \
415         LOCK_UNLOCK_2(D, A); /* fail */
416
417 /*
418  * 6 testcases:
419  */
420 #include "locking-selftest-spin.h"
421 GENERATE_TESTCASE(ABBCCDDA_spin)
422 #include "locking-selftest-wlock.h"
423 GENERATE_TESTCASE(ABBCCDDA_wlock)
424 #include "locking-selftest-rlock.h"
425 GENERATE_TESTCASE(ABBCCDDA_rlock)
426 #include "locking-selftest-mutex.h"
427 GENERATE_TESTCASE(ABBCCDDA_mutex)
428 #include "locking-selftest-wsem.h"
429 GENERATE_TESTCASE(ABBCCDDA_wsem)
430 #include "locking-selftest-rsem.h"
431 GENERATE_TESTCASE(ABBCCDDA_rsem)
432
433 #undef E
434
435 /*
436  * AB CD BD DA deadlock:
437  */
438 #define E()                                     \
439                                                 \
440         LOCK_UNLOCK_2(A, B);                    \
441         LOCK_UNLOCK_2(C, D);                    \
442         LOCK_UNLOCK_2(B, D);                    \
443         LOCK_UNLOCK_2(D, A); /* fail */
444
445 /*
446  * 6 testcases:
447  */
448 #include "locking-selftest-spin.h"
449 GENERATE_TESTCASE(ABCDBDDA_spin)
450 #include "locking-selftest-wlock.h"
451 GENERATE_TESTCASE(ABCDBDDA_wlock)
452 #include "locking-selftest-rlock.h"
453 GENERATE_TESTCASE(ABCDBDDA_rlock)
454 #include "locking-selftest-mutex.h"
455 GENERATE_TESTCASE(ABCDBDDA_mutex)
456 #include "locking-selftest-wsem.h"
457 GENERATE_TESTCASE(ABCDBDDA_wsem)
458 #include "locking-selftest-rsem.h"
459 GENERATE_TESTCASE(ABCDBDDA_rsem)
460
461 #undef E
462
463 /*
464  * AB CD BC DA deadlock:
465  */
466 #define E()                                     \
467                                                 \
468         LOCK_UNLOCK_2(A, B);                    \
469         LOCK_UNLOCK_2(C, D);                    \
470         LOCK_UNLOCK_2(B, C);                    \
471         LOCK_UNLOCK_2(D, A); /* fail */
472
473 /*
474  * 6 testcases:
475  */
476 #include "locking-selftest-spin.h"
477 GENERATE_TESTCASE(ABCDBCDA_spin)
478 #include "locking-selftest-wlock.h"
479 GENERATE_TESTCASE(ABCDBCDA_wlock)
480 #include "locking-selftest-rlock.h"
481 GENERATE_TESTCASE(ABCDBCDA_rlock)
482 #include "locking-selftest-mutex.h"
483 GENERATE_TESTCASE(ABCDBCDA_mutex)
484 #include "locking-selftest-wsem.h"
485 GENERATE_TESTCASE(ABCDBCDA_wsem)
486 #include "locking-selftest-rsem.h"
487 GENERATE_TESTCASE(ABCDBCDA_rsem)
488
489 #undef E
490
491 /*
492  * Double unlock:
493  */
494 #define E()                                     \
495                                                 \
496         LOCK(A);                                \
497         UNLOCK(A);                              \
498         UNLOCK(A); /* fail */
499
500 /*
501  * 6 testcases:
502  */
503 #include "locking-selftest-spin.h"
504 GENERATE_TESTCASE(double_unlock_spin)
505 #include "locking-selftest-wlock.h"
506 GENERATE_TESTCASE(double_unlock_wlock)
507 #include "locking-selftest-rlock.h"
508 GENERATE_TESTCASE(double_unlock_rlock)
509 #include "locking-selftest-mutex.h"
510 GENERATE_TESTCASE(double_unlock_mutex)
511 #include "locking-selftest-wsem.h"
512 GENERATE_TESTCASE(double_unlock_wsem)
513 #include "locking-selftest-rsem.h"
514 GENERATE_TESTCASE(double_unlock_rsem)
515
516 #undef E
517
518 /*
519  * Bad unlock ordering:
520  */
521 #define E()                                     \
522                                                 \
523         LOCK(A);                                \
524         LOCK(B);                                \
525         UNLOCK(A); /* fail */                   \
526         UNLOCK(B);
527
528 /*
529  * 6 testcases:
530  */
531 #include "locking-selftest-spin.h"
532 GENERATE_TESTCASE(bad_unlock_order_spin)
533 #include "locking-selftest-wlock.h"
534 GENERATE_TESTCASE(bad_unlock_order_wlock)
535 #include "locking-selftest-rlock.h"
536 GENERATE_TESTCASE(bad_unlock_order_rlock)
537 #include "locking-selftest-mutex.h"
538 GENERATE_TESTCASE(bad_unlock_order_mutex)
539 #include "locking-selftest-wsem.h"
540 GENERATE_TESTCASE(bad_unlock_order_wsem)
541 #include "locking-selftest-rsem.h"
542 GENERATE_TESTCASE(bad_unlock_order_rsem)
543
544 #undef E
545
546 /*
547  * initializing a held lock:
548  */
549 #define E()                                     \
550                                                 \
551         LOCK(A);                                \
552         INIT(A); /* fail */
553
554 /*
555  * 6 testcases:
556  */
557 #include "locking-selftest-spin.h"
558 GENERATE_TESTCASE(init_held_spin)
559 #include "locking-selftest-wlock.h"
560 GENERATE_TESTCASE(init_held_wlock)
561 #include "locking-selftest-rlock.h"
562 GENERATE_TESTCASE(init_held_rlock)
563 #include "locking-selftest-mutex.h"
564 GENERATE_TESTCASE(init_held_mutex)
565 #include "locking-selftest-wsem.h"
566 GENERATE_TESTCASE(init_held_wsem)
567 #include "locking-selftest-rsem.h"
568 GENERATE_TESTCASE(init_held_rsem)
569
570 #undef E
571
572 /*
573  * locking an irq-safe lock with irqs enabled:
574  */
575 #define E1()                            \
576                                         \
577         IRQ_ENTER();                    \
578         LOCK(A);                        \
579         UNLOCK(A);                      \
580         IRQ_EXIT();
581
582 #define E2()                            \
583                                         \
584         LOCK(A);                        \
585         UNLOCK(A);
586
587 /*
588  * Generate 24 testcases:
589  */
590 #include "locking-selftest-spin-hardirq.h"
591 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_spin)
592
593 #include "locking-selftest-rlock-hardirq.h"
594 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_rlock)
595
596 #include "locking-selftest-wlock-hardirq.h"
597 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_wlock)
598
599 #include "locking-selftest-spin-softirq.h"
600 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_soft_spin)
601
602 #include "locking-selftest-rlock-softirq.h"
603 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_soft_rlock)
604
605 #include "locking-selftest-wlock-softirq.h"
606 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_soft_wlock)
607
608 #undef E1
609 #undef E2
610
611 /*
612  * Enabling hardirqs with a softirq-safe lock held:
613  */
614 #define E1()                            \
615                                         \
616         SOFTIRQ_ENTER();                \
617         LOCK(A);                        \
618         UNLOCK(A);                      \
619         SOFTIRQ_EXIT();
620
621 #define E2()                            \
622                                         \
623         HARDIRQ_DISABLE();              \
624         LOCK(A);                        \
625         HARDIRQ_ENABLE();               \
626         UNLOCK(A);
627
628 /*
629  * Generate 12 testcases:
630  */
631 #include "locking-selftest-spin.h"
632 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2A_spin)
633
634 #include "locking-selftest-wlock.h"
635 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2A_wlock)
636
637 #include "locking-selftest-rlock.h"
638 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2A_rlock)
639
640 #undef E1
641 #undef E2
642
643 /*
644  * Enabling irqs with an irq-safe lock held:
645  */
646 #define E1()                            \
647                                         \
648         IRQ_ENTER();                    \
649         LOCK(A);                        \
650         UNLOCK(A);                      \
651         IRQ_EXIT();
652
653 #define E2()                            \
654                                         \
655         IRQ_DISABLE();                  \
656         LOCK(A);                        \
657         IRQ_ENABLE();                   \
658         UNLOCK(A);
659
660 /*
661  * Generate 24 testcases:
662  */
663 #include "locking-selftest-spin-hardirq.h"
664 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_spin)
665
666 #include "locking-selftest-rlock-hardirq.h"
667 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_rlock)
668
669 #include "locking-selftest-wlock-hardirq.h"
670 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_wlock)
671
672 #include "locking-selftest-spin-softirq.h"
673 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_soft_spin)
674
675 #include "locking-selftest-rlock-softirq.h"
676 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_soft_rlock)
677
678 #include "locking-selftest-wlock-softirq.h"
679 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_soft_wlock)
680
681 #undef E1
682 #undef E2
683
684 /*
685  * Acquiring a irq-unsafe lock while holding an irq-safe-lock:
686  */
687 #define E1()                            \
688                                         \
689         LOCK(A);                        \
690         LOCK(B);                        \
691         UNLOCK(B);                      \
692         UNLOCK(A);                      \
693
694 #define E2()                            \
695                                         \
696         LOCK(B);                        \
697         UNLOCK(B);
698
699 #define E3()                            \
700                                         \
701         IRQ_ENTER();                    \
702         LOCK(A);                        \
703         UNLOCK(A);                      \
704         IRQ_EXIT();
705
706 /*
707  * Generate 36 testcases:
708  */
709 #include "locking-selftest-spin-hardirq.h"
710 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_spin)
711
712 #include "locking-selftest-rlock-hardirq.h"
713 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_rlock)
714
715 #include "locking-selftest-wlock-hardirq.h"
716 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_wlock)
717
718 #include "locking-selftest-spin-softirq.h"
719 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_soft_spin)
720
721 #include "locking-selftest-rlock-softirq.h"
722 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_soft_rlock)
723
724 #include "locking-selftest-wlock-softirq.h"
725 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_soft_wlock)
726
727 #undef E1
728 #undef E2
729 #undef E3
730
731 /*
732  * If a lock turns into softirq-safe, but earlier it took
733  * a softirq-unsafe lock:
734  */
735
736 #define E1()                            \
737         IRQ_DISABLE();                  \
738         LOCK(A);                        \
739         LOCK(B);                        \
740         UNLOCK(B);                      \
741         UNLOCK(A);                      \
742         IRQ_ENABLE();
743
744 #define E2()                            \
745         LOCK(B);                        \
746         UNLOCK(B);
747
748 #define E3()                            \
749         IRQ_ENTER();                    \
750         LOCK(A);                        \
751         UNLOCK(A);                      \
752         IRQ_EXIT();
753
754 /*
755  * Generate 36 testcases:
756  */
757 #include "locking-selftest-spin-hardirq.h"
758 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_spin)
759
760 #include "locking-selftest-rlock-hardirq.h"
761 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_rlock)
762
763 #include "locking-selftest-wlock-hardirq.h"
764 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_wlock)
765
766 #include "locking-selftest-spin-softirq.h"
767 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_soft_spin)
768
769 #include "locking-selftest-rlock-softirq.h"
770 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_soft_rlock)
771
772 #include "locking-selftest-wlock-softirq.h"
773 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_soft_wlock)
774
775 #undef E1
776 #undef E2
777 #undef E3
778
779 /*
780  * read-lock / write-lock irq inversion.
781  *
782  * Deadlock scenario:
783  *
784  * CPU#1 is at #1, i.e. it has write-locked A, but has not
785  * taken B yet.
786  *
787  * CPU#2 is at #2, i.e. it has locked B.
788  *
789  * Hardirq hits CPU#2 at point #2 and is trying to read-lock A.
790  *
791  * The deadlock occurs because CPU#1 will spin on B, and CPU#2
792  * will spin on A.
793  */
794
795 #define E1()                            \
796                                         \
797         IRQ_DISABLE();                  \
798         WL(A);                          \
799         LOCK(B);                        \
800         UNLOCK(B);                      \
801         WU(A);                          \
802         IRQ_ENABLE();
803
804 #define E2()                            \
805                                         \
806         LOCK(B);                        \
807         UNLOCK(B);
808
809 #define E3()                            \
810                                         \
811         IRQ_ENTER();                    \
812         RL(A);                          \
813         RU(A);                          \
814         IRQ_EXIT();
815
816 /*
817  * Generate 36 testcases:
818  */
819 #include "locking-selftest-spin-hardirq.h"
820 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_spin)
821
822 #include "locking-selftest-rlock-hardirq.h"
823 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_rlock)
824
825 #include "locking-selftest-wlock-hardirq.h"
826 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_wlock)
827
828 #include "locking-selftest-spin-softirq.h"
829 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_soft_spin)
830
831 #include "locking-selftest-rlock-softirq.h"
832 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_soft_rlock)
833
834 #include "locking-selftest-wlock-softirq.h"
835 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_soft_wlock)
836
837 #undef E1
838 #undef E2
839 #undef E3
840
841 /*
842  * read-lock / write-lock recursion that is actually safe.
843  */
844
845 #define E1()                            \
846                                         \
847         IRQ_DISABLE();                  \
848         WL(A);                          \
849         WU(A);                          \
850         IRQ_ENABLE();
851
852 #define E2()                            \
853                                         \
854         RL(A);                          \
855         RU(A);                          \
856
857 #define E3()                            \
858                                         \
859         IRQ_ENTER();                    \
860         RL(A);                          \
861         L(B);                           \
862         U(B);                           \
863         RU(A);                          \
864         IRQ_EXIT();
865
866 /*
867  * Generate 12 testcases:
868  */
869 #include "locking-selftest-hardirq.h"
870 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion_hard)
871
872 #include "locking-selftest-softirq.h"
873 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion_soft)
874
875 #undef E1
876 #undef E2
877 #undef E3
878
879 /*
880  * read-lock / write-lock recursion that is unsafe.
881  */
882
883 #define E1()                            \
884                                         \
885         IRQ_DISABLE();                  \
886         L(B);                           \
887         WL(A);                          \
888         WU(A);                          \
889         U(B);                           \
890         IRQ_ENABLE();
891
892 #define E2()                            \
893                                         \
894         RL(A);                          \
895         RU(A);                          \
896
897 #define E3()                            \
898                                         \
899         IRQ_ENTER();                    \
900         L(B);                           \
901         U(B);                           \
902         IRQ_EXIT();
903
904 /*
905  * Generate 12 testcases:
906  */
907 #include "locking-selftest-hardirq.h"
908 // GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion2_hard)
909
910 #include "locking-selftest-softirq.h"
911 // GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion2_soft)
912
913 #ifdef CONFIG_DEBUG_LOCK_ALLOC
914 # define I_SPINLOCK(x)  lockdep_reset_lock(&lock_##x.dep_map)
915 # define I_RWLOCK(x)    lockdep_reset_lock(&rwlock_##x.dep_map)
916 # define I_MUTEX(x)     lockdep_reset_lock(&mutex_##x.dep_map)
917 # define I_RWSEM(x)     lockdep_reset_lock(&rwsem_##x.dep_map)
918 # define I_WW(x)        lockdep_reset_lock(&x.dep_map)
919 #else
920 # define I_SPINLOCK(x)
921 # define I_RWLOCK(x)
922 # define I_MUTEX(x)
923 # define I_RWSEM(x)
924 # define I_WW(x)
925 #endif
926
927 #define I1(x)                                   \
928         do {                                    \
929                 I_SPINLOCK(x);                  \
930                 I_RWLOCK(x);                    \
931                 I_MUTEX(x);                     \
932                 I_RWSEM(x);                     \
933         } while (0)
934
935 #define I2(x)                                   \
936         do {                                    \
937                 raw_spin_lock_init(&lock_##x);  \
938                 rwlock_init(&rwlock_##x);       \
939                 mutex_init(&mutex_##x);         \
940                 init_rwsem(&rwsem_##x);         \
941         } while (0)
942
943 static void reset_locks(void)
944 {
945         local_irq_disable();
946         lockdep_free_key_range(&ww_lockdep.acquire_key, 1);
947         lockdep_free_key_range(&ww_lockdep.mutex_key, 1);
948
949         I1(A); I1(B); I1(C); I1(D);
950         I1(X1); I1(X2); I1(Y1); I1(Y2); I1(Z1); I1(Z2);
951         I_WW(t); I_WW(t2); I_WW(o.base); I_WW(o2.base); I_WW(o3.base);
952         lockdep_reset();
953         I2(A); I2(B); I2(C); I2(D);
954         init_shared_classes();
955
956         ww_mutex_init(&o, &ww_lockdep); ww_mutex_init(&o2, &ww_lockdep); ww_mutex_init(&o3, &ww_lockdep);
957         memset(&t, 0, sizeof(t)); memset(&t2, 0, sizeof(t2));
958         memset(&ww_lockdep.acquire_key, 0, sizeof(ww_lockdep.acquire_key));
959         memset(&ww_lockdep.mutex_key, 0, sizeof(ww_lockdep.mutex_key));
960         local_irq_enable();
961 }
962
963 #undef I
964
965 static int testcase_total;
966 static int testcase_successes;
967 static int expected_testcase_failures;
968 static int unexpected_testcase_failures;
969
970 static void dotest(void (*testcase_fn)(void), int expected, int lockclass_mask)
971 {
972         unsigned long saved_preempt_count = preempt_count();
973
974         WARN_ON(irqs_disabled());
975
976         testcase_fn();
977         /*
978          * Filter out expected failures:
979          */
980 #ifndef CONFIG_PROVE_LOCKING
981         if (expected == FAILURE && debug_locks) {
982                 expected_testcase_failures++;
983                 printk("failed|");
984         }
985         else
986 #endif
987         if (debug_locks != expected) {
988                 unexpected_testcase_failures++;
989                 printk("FAILED|");
990
991                 dump_stack();
992         } else {
993                 testcase_successes++;
994                 printk("  ok  |");
995         }
996         testcase_total++;
997
998         if (debug_locks_verbose)
999                 printk(" lockclass mask: %x, debug_locks: %d, expected: %d\n",
1000                         lockclass_mask, debug_locks, expected);
1001         /*
1002          * Some tests (e.g. double-unlock) might corrupt the preemption
1003          * count, so restore it:
1004          */
1005         preempt_count_set(saved_preempt_count);
1006 #ifdef CONFIG_TRACE_IRQFLAGS
1007         if (softirq_count())
1008                 current->softirqs_enabled = 0;
1009         else
1010                 current->softirqs_enabled = 1;
1011 #endif
1012
1013         reset_locks();
1014 }
1015
1016 static inline void print_testname(const char *testname)
1017 {
1018         printk("%33s:", testname);
1019 }
1020
1021 #define DO_TESTCASE_1(desc, name, nr)                           \
1022         print_testname(desc"/"#nr);                             \
1023         dotest(name##_##nr, SUCCESS, LOCKTYPE_RWLOCK);          \
1024         printk("\n");
1025
1026 #define DO_TESTCASE_1B(desc, name, nr)                          \
1027         print_testname(desc"/"#nr);                             \
1028         dotest(name##_##nr, FAILURE, LOCKTYPE_RWLOCK);          \
1029         printk("\n");
1030
1031 #define DO_TESTCASE_3(desc, name, nr)                           \
1032         print_testname(desc"/"#nr);                             \
1033         dotest(name##_spin_##nr, FAILURE, LOCKTYPE_SPIN);       \
1034         dotest(name##_wlock_##nr, FAILURE, LOCKTYPE_RWLOCK);    \
1035         dotest(name##_rlock_##nr, SUCCESS, LOCKTYPE_RWLOCK);    \
1036         printk("\n");
1037
1038 #define DO_TESTCASE_3RW(desc, name, nr)                         \
1039         print_testname(desc"/"#nr);                             \
1040         dotest(name##_spin_##nr, FAILURE, LOCKTYPE_SPIN|LOCKTYPE_RWLOCK);\
1041         dotest(name##_wlock_##nr, FAILURE, LOCKTYPE_RWLOCK);    \
1042         dotest(name##_rlock_##nr, SUCCESS, LOCKTYPE_RWLOCK);    \
1043         printk("\n");
1044
1045 #define DO_TESTCASE_6(desc, name)                               \
1046         print_testname(desc);                                   \
1047         dotest(name##_spin, FAILURE, LOCKTYPE_SPIN);            \
1048         dotest(name##_wlock, FAILURE, LOCKTYPE_RWLOCK);         \
1049         dotest(name##_rlock, FAILURE, LOCKTYPE_RWLOCK);         \
1050         dotest(name##_mutex, FAILURE, LOCKTYPE_MUTEX);          \
1051         dotest(name##_wsem, FAILURE, LOCKTYPE_RWSEM);           \
1052         dotest(name##_rsem, FAILURE, LOCKTYPE_RWSEM);           \
1053         printk("\n");
1054
1055 #define DO_TESTCASE_6_SUCCESS(desc, name)                       \
1056         print_testname(desc);                                   \
1057         dotest(name##_spin, SUCCESS, LOCKTYPE_SPIN);            \
1058         dotest(name##_wlock, SUCCESS, LOCKTYPE_RWLOCK);         \
1059         dotest(name##_rlock, SUCCESS, LOCKTYPE_RWLOCK);         \
1060         dotest(name##_mutex, SUCCESS, LOCKTYPE_MUTEX);          \
1061         dotest(name##_wsem, SUCCESS, LOCKTYPE_RWSEM);           \
1062         dotest(name##_rsem, SUCCESS, LOCKTYPE_RWSEM);           \
1063         printk("\n");
1064
1065 /*
1066  * 'read' variant: rlocks must not trigger.
1067  */
1068 #define DO_TESTCASE_6R(desc, name)                              \
1069         print_testname(desc);                                   \
1070         dotest(name##_spin, FAILURE, LOCKTYPE_SPIN);            \
1071         dotest(name##_wlock, FAILURE, LOCKTYPE_RWLOCK);         \
1072         dotest(name##_rlock, SUCCESS, LOCKTYPE_RWLOCK);         \
1073         dotest(name##_mutex, FAILURE, LOCKTYPE_MUTEX);          \
1074         dotest(name##_wsem, FAILURE, LOCKTYPE_RWSEM);           \
1075         dotest(name##_rsem, FAILURE, LOCKTYPE_RWSEM);           \
1076         printk("\n");
1077
1078 #define DO_TESTCASE_2I(desc, name, nr)                          \
1079         DO_TESTCASE_1("hard-"desc, name##_hard, nr);            \
1080         DO_TESTCASE_1("soft-"desc, name##_soft, nr);
1081
1082 #define DO_TESTCASE_2IB(desc, name, nr)                         \
1083         DO_TESTCASE_1B("hard-"desc, name##_hard, nr);           \
1084         DO_TESTCASE_1B("soft-"desc, name##_soft, nr);
1085
1086 #define DO_TESTCASE_6I(desc, name, nr)                          \
1087         DO_TESTCASE_3("hard-"desc, name##_hard, nr);            \
1088         DO_TESTCASE_3("soft-"desc, name##_soft, nr);
1089
1090 #define DO_TESTCASE_6IRW(desc, name, nr)                        \
1091         DO_TESTCASE_3RW("hard-"desc, name##_hard, nr);          \
1092         DO_TESTCASE_3RW("soft-"desc, name##_soft, nr);
1093
1094 #define DO_TESTCASE_2x3(desc, name)                             \
1095         DO_TESTCASE_3(desc, name, 12);                          \
1096         DO_TESTCASE_3(desc, name, 21);
1097
1098 #define DO_TESTCASE_2x6(desc, name)                             \
1099         DO_TESTCASE_6I(desc, name, 12);                         \
1100         DO_TESTCASE_6I(desc, name, 21);
1101
1102 #define DO_TESTCASE_6x2(desc, name)                             \
1103         DO_TESTCASE_2I(desc, name, 123);                        \
1104         DO_TESTCASE_2I(desc, name, 132);                        \
1105         DO_TESTCASE_2I(desc, name, 213);                        \
1106         DO_TESTCASE_2I(desc, name, 231);                        \
1107         DO_TESTCASE_2I(desc, name, 312);                        \
1108         DO_TESTCASE_2I(desc, name, 321);
1109
1110 #define DO_TESTCASE_6x2B(desc, name)                            \
1111         DO_TESTCASE_2IB(desc, name, 123);                       \
1112         DO_TESTCASE_2IB(desc, name, 132);                       \
1113         DO_TESTCASE_2IB(desc, name, 213);                       \
1114         DO_TESTCASE_2IB(desc, name, 231);                       \
1115         DO_TESTCASE_2IB(desc, name, 312);                       \
1116         DO_TESTCASE_2IB(desc, name, 321);
1117
1118 #define DO_TESTCASE_6x6(desc, name)                             \
1119         DO_TESTCASE_6I(desc, name, 123);                        \
1120         DO_TESTCASE_6I(desc, name, 132);                        \
1121         DO_TESTCASE_6I(desc, name, 213);                        \
1122         DO_TESTCASE_6I(desc, name, 231);                        \
1123         DO_TESTCASE_6I(desc, name, 312);                        \
1124         DO_TESTCASE_6I(desc, name, 321);
1125
1126 #define DO_TESTCASE_6x6RW(desc, name)                           \
1127         DO_TESTCASE_6IRW(desc, name, 123);                      \
1128         DO_TESTCASE_6IRW(desc, name, 132);                      \
1129         DO_TESTCASE_6IRW(desc, name, 213);                      \
1130         DO_TESTCASE_6IRW(desc, name, 231);                      \
1131         DO_TESTCASE_6IRW(desc, name, 312);                      \
1132         DO_TESTCASE_6IRW(desc, name, 321);
1133
1134 static void ww_test_fail_acquire(void)
1135 {
1136         int ret;
1137
1138         WWAI(&t);
1139         t.stamp++;
1140
1141         ret = WWL(&o, &t);
1142
1143         if (WARN_ON(!o.ctx) ||
1144             WARN_ON(ret))
1145                 return;
1146
1147         /* No lockdep test, pure API */
1148         ret = WWL(&o, &t);
1149         WARN_ON(ret != -EALREADY);
1150
1151         ret = WWT(&o);
1152         WARN_ON(ret);
1153
1154         t2 = t;
1155         t2.stamp++;
1156         ret = WWL(&o, &t2);
1157         WARN_ON(ret != -EDEADLK);
1158         WWU(&o);
1159
1160         if (WWT(&o))
1161                 WWU(&o);
1162 #ifdef CONFIG_DEBUG_LOCK_ALLOC
1163         else
1164                 DEBUG_LOCKS_WARN_ON(1);
1165 #endif
1166 }
1167
1168 static void ww_test_normal(void)
1169 {
1170         int ret;
1171
1172         WWAI(&t);
1173
1174         /*
1175          * None of the ww_mutex codepaths should be taken in the 'normal'
1176          * mutex calls. The easiest way to verify this is by using the
1177          * normal mutex calls, and making sure o.ctx is unmodified.
1178          */
1179
1180         /* mutex_lock (and indirectly, mutex_lock_nested) */
1181         o.ctx = (void *)~0UL;
1182         mutex_lock(&o.base);
1183         mutex_unlock(&o.base);
1184         WARN_ON(o.ctx != (void *)~0UL);
1185
1186         /* mutex_lock_interruptible (and *_nested) */
1187         o.ctx = (void *)~0UL;
1188         ret = mutex_lock_interruptible(&o.base);
1189         if (!ret)
1190                 mutex_unlock(&o.base);
1191         else
1192                 WARN_ON(1);
1193         WARN_ON(o.ctx != (void *)~0UL);
1194
1195         /* mutex_lock_killable (and *_nested) */
1196         o.ctx = (void *)~0UL;
1197         ret = mutex_lock_killable(&o.base);
1198         if (!ret)
1199                 mutex_unlock(&o.base);
1200         else
1201                 WARN_ON(1);
1202         WARN_ON(o.ctx != (void *)~0UL);
1203
1204         /* trylock, succeeding */
1205         o.ctx = (void *)~0UL;
1206         ret = mutex_trylock(&o.base);
1207         WARN_ON(!ret);
1208         if (ret)
1209                 mutex_unlock(&o.base);
1210         else
1211                 WARN_ON(1);
1212         WARN_ON(o.ctx != (void *)~0UL);
1213
1214         /* trylock, failing */
1215         o.ctx = (void *)~0UL;
1216         mutex_lock(&o.base);
1217         ret = mutex_trylock(&o.base);
1218         WARN_ON(ret);
1219         mutex_unlock(&o.base);
1220         WARN_ON(o.ctx != (void *)~0UL);
1221
1222         /* nest_lock */
1223         o.ctx = (void *)~0UL;
1224         mutex_lock_nest_lock(&o.base, &t);
1225         mutex_unlock(&o.base);
1226         WARN_ON(o.ctx != (void *)~0UL);
1227 }
1228
1229 static void ww_test_two_contexts(void)
1230 {
1231         WWAI(&t);
1232         WWAI(&t2);
1233 }
1234
1235 static void ww_test_diff_class(void)
1236 {
1237         WWAI(&t);
1238 #ifdef CONFIG_DEBUG_MUTEXES
1239         t.ww_class = NULL;
1240 #endif
1241         WWL(&o, &t);
1242 }
1243
1244 static void ww_test_context_done_twice(void)
1245 {
1246         WWAI(&t);
1247         WWAD(&t);
1248         WWAD(&t);
1249         WWAF(&t);
1250 }
1251
1252 static void ww_test_context_unlock_twice(void)
1253 {
1254         WWAI(&t);
1255         WWAD(&t);
1256         WWAF(&t);
1257         WWAF(&t);
1258 }
1259
1260 static void ww_test_context_fini_early(void)
1261 {
1262         WWAI(&t);
1263         WWL(&o, &t);
1264         WWAD(&t);
1265         WWAF(&t);
1266 }
1267
1268 static void ww_test_context_lock_after_done(void)
1269 {
1270         WWAI(&t);
1271         WWAD(&t);
1272         WWL(&o, &t);
1273 }
1274
1275 static void ww_test_object_unlock_twice(void)
1276 {
1277         WWL1(&o);
1278         WWU(&o);
1279         WWU(&o);
1280 }
1281
1282 static void ww_test_object_lock_unbalanced(void)
1283 {
1284         WWAI(&t);
1285         WWL(&o, &t);
1286         t.acquired = 0;
1287         WWU(&o);
1288         WWAF(&t);
1289 }
1290
1291 static void ww_test_object_lock_stale_context(void)
1292 {
1293         WWAI(&t);
1294         o.ctx = &t2;
1295         WWL(&o, &t);
1296 }
1297
1298 static void ww_test_edeadlk_normal(void)
1299 {
1300         int ret;
1301
1302         mutex_lock(&o2.base);
1303         o2.ctx = &t2;
1304         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1305
1306         WWAI(&t);
1307         t2 = t;
1308         t2.stamp--;
1309
1310         ret = WWL(&o, &t);
1311         WARN_ON(ret);
1312
1313         ret = WWL(&o2, &t);
1314         WARN_ON(ret != -EDEADLK);
1315
1316         o2.ctx = NULL;
1317         mutex_acquire(&o2.base.dep_map, 0, 1, _THIS_IP_);
1318         mutex_unlock(&o2.base);
1319         WWU(&o);
1320
1321         WWL(&o2, &t);
1322 }
1323
1324 static void ww_test_edeadlk_normal_slow(void)
1325 {
1326         int ret;
1327
1328         mutex_lock(&o2.base);
1329         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1330         o2.ctx = &t2;
1331
1332         WWAI(&t);
1333         t2 = t;
1334         t2.stamp--;
1335
1336         ret = WWL(&o, &t);
1337         WARN_ON(ret);
1338
1339         ret = WWL(&o2, &t);
1340         WARN_ON(ret != -EDEADLK);
1341
1342         o2.ctx = NULL;
1343         mutex_acquire(&o2.base.dep_map, 0, 1, _THIS_IP_);
1344         mutex_unlock(&o2.base);
1345         WWU(&o);
1346
1347         ww_mutex_lock_slow(&o2, &t);
1348 }
1349
1350 static void ww_test_edeadlk_no_unlock(void)
1351 {
1352         int ret;
1353
1354         mutex_lock(&o2.base);
1355         o2.ctx = &t2;
1356         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1357
1358         WWAI(&t);
1359         t2 = t;
1360         t2.stamp--;
1361
1362         ret = WWL(&o, &t);
1363         WARN_ON(ret);
1364
1365         ret = WWL(&o2, &t);
1366         WARN_ON(ret != -EDEADLK);
1367
1368         o2.ctx = NULL;
1369         mutex_acquire(&o2.base.dep_map, 0, 1, _THIS_IP_);
1370         mutex_unlock(&o2.base);
1371
1372         WWL(&o2, &t);
1373 }
1374
1375 static void ww_test_edeadlk_no_unlock_slow(void)
1376 {
1377         int ret;
1378
1379         mutex_lock(&o2.base);
1380         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1381         o2.ctx = &t2;
1382
1383         WWAI(&t);
1384         t2 = t;
1385         t2.stamp--;
1386
1387         ret = WWL(&o, &t);
1388         WARN_ON(ret);
1389
1390         ret = WWL(&o2, &t);
1391         WARN_ON(ret != -EDEADLK);
1392
1393         o2.ctx = NULL;
1394         mutex_acquire(&o2.base.dep_map, 0, 1, _THIS_IP_);
1395         mutex_unlock(&o2.base);
1396
1397         ww_mutex_lock_slow(&o2, &t);
1398 }
1399
1400 static void ww_test_edeadlk_acquire_more(void)
1401 {
1402         int ret;
1403
1404         mutex_lock(&o2.base);
1405         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1406         o2.ctx = &t2;
1407
1408         WWAI(&t);
1409         t2 = t;
1410         t2.stamp--;
1411
1412         ret = WWL(&o, &t);
1413         WARN_ON(ret);
1414
1415         ret = WWL(&o2, &t);
1416         WARN_ON(ret != -EDEADLK);
1417
1418         ret = WWL(&o3, &t);
1419 }
1420
1421 static void ww_test_edeadlk_acquire_more_slow(void)
1422 {
1423         int ret;
1424
1425         mutex_lock(&o2.base);
1426         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1427         o2.ctx = &t2;
1428
1429         WWAI(&t);
1430         t2 = t;
1431         t2.stamp--;
1432
1433         ret = WWL(&o, &t);
1434         WARN_ON(ret);
1435
1436         ret = WWL(&o2, &t);
1437         WARN_ON(ret != -EDEADLK);
1438
1439         ww_mutex_lock_slow(&o3, &t);
1440 }
1441
1442 static void ww_test_edeadlk_acquire_more_edeadlk(void)
1443 {
1444         int ret;
1445
1446         mutex_lock(&o2.base);
1447         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1448         o2.ctx = &t2;
1449
1450         mutex_lock(&o3.base);
1451         mutex_release(&o3.base.dep_map, 1, _THIS_IP_);
1452         o3.ctx = &t2;
1453
1454         WWAI(&t);
1455         t2 = t;
1456         t2.stamp--;
1457
1458         ret = WWL(&o, &t);
1459         WARN_ON(ret);
1460
1461         ret = WWL(&o2, &t);
1462         WARN_ON(ret != -EDEADLK);
1463
1464         ret = WWL(&o3, &t);
1465         WARN_ON(ret != -EDEADLK);
1466 }
1467
1468 static void ww_test_edeadlk_acquire_more_edeadlk_slow(void)
1469 {
1470         int ret;
1471
1472         mutex_lock(&o2.base);
1473         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1474         o2.ctx = &t2;
1475
1476         mutex_lock(&o3.base);
1477         mutex_release(&o3.base.dep_map, 1, _THIS_IP_);
1478         o3.ctx = &t2;
1479
1480         WWAI(&t);
1481         t2 = t;
1482         t2.stamp--;
1483
1484         ret = WWL(&o, &t);
1485         WARN_ON(ret);
1486
1487         ret = WWL(&o2, &t);
1488         WARN_ON(ret != -EDEADLK);
1489
1490         ww_mutex_lock_slow(&o3, &t);
1491 }
1492
1493 static void ww_test_edeadlk_acquire_wrong(void)
1494 {
1495         int ret;
1496
1497         mutex_lock(&o2.base);
1498         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1499         o2.ctx = &t2;
1500
1501         WWAI(&t);
1502         t2 = t;
1503         t2.stamp--;
1504
1505         ret = WWL(&o, &t);
1506         WARN_ON(ret);
1507
1508         ret = WWL(&o2, &t);
1509         WARN_ON(ret != -EDEADLK);
1510         if (!ret)
1511                 WWU(&o2);
1512
1513         WWU(&o);
1514
1515         ret = WWL(&o3, &t);
1516 }
1517
1518 static void ww_test_edeadlk_acquire_wrong_slow(void)
1519 {
1520         int ret;
1521
1522         mutex_lock(&o2.base);
1523         mutex_release(&o2.base.dep_map, 1, _THIS_IP_);
1524         o2.ctx = &t2;
1525
1526         WWAI(&t);
1527         t2 = t;
1528         t2.stamp--;
1529
1530         ret = WWL(&o, &t);
1531         WARN_ON(ret);
1532
1533         ret = WWL(&o2, &t);
1534         WARN_ON(ret != -EDEADLK);
1535         if (!ret)
1536                 WWU(&o2);
1537
1538         WWU(&o);
1539
1540         ww_mutex_lock_slow(&o3, &t);
1541 }
1542
1543 static void ww_test_spin_nest_unlocked(void)
1544 {
1545         raw_spin_lock_nest_lock(&lock_A, &o.base);
1546         U(A);
1547 }
1548
1549 static void ww_test_unneeded_slow(void)
1550 {
1551         WWAI(&t);
1552
1553         ww_mutex_lock_slow(&o, &t);
1554 }
1555
1556 static void ww_test_context_block(void)
1557 {
1558         int ret;
1559
1560         WWAI(&t);
1561
1562         ret = WWL(&o, &t);
1563         WARN_ON(ret);
1564         WWL1(&o2);
1565 }
1566
1567 static void ww_test_context_try(void)
1568 {
1569         int ret;
1570
1571         WWAI(&t);
1572
1573         ret = WWL(&o, &t);
1574         WARN_ON(ret);
1575
1576         ret = WWT(&o2);
1577         WARN_ON(!ret);
1578         WWU(&o2);
1579         WWU(&o);
1580 }
1581
1582 static void ww_test_context_context(void)
1583 {
1584         int ret;
1585
1586         WWAI(&t);
1587
1588         ret = WWL(&o, &t);
1589         WARN_ON(ret);
1590
1591         ret = WWL(&o2, &t);
1592         WARN_ON(ret);
1593
1594         WWU(&o2);
1595         WWU(&o);
1596 }
1597
1598 static void ww_test_try_block(void)
1599 {
1600         bool ret;
1601
1602         ret = WWT(&o);
1603         WARN_ON(!ret);
1604
1605         WWL1(&o2);
1606         WWU(&o2);
1607         WWU(&o);
1608 }
1609
1610 static void ww_test_try_try(void)
1611 {
1612         bool ret;
1613
1614         ret = WWT(&o);
1615         WARN_ON(!ret);
1616         ret = WWT(&o2);
1617         WARN_ON(!ret);
1618         WWU(&o2);
1619         WWU(&o);
1620 }
1621
1622 static void ww_test_try_context(void)
1623 {
1624         int ret;
1625
1626         ret = WWT(&o);
1627         WARN_ON(!ret);
1628
1629         WWAI(&t);
1630
1631         ret = WWL(&o2, &t);
1632         WARN_ON(ret);
1633 }
1634
1635 static void ww_test_block_block(void)
1636 {
1637         WWL1(&o);
1638         WWL1(&o2);
1639 }
1640
1641 static void ww_test_block_try(void)
1642 {
1643         bool ret;
1644
1645         WWL1(&o);
1646         ret = WWT(&o2);
1647         WARN_ON(!ret);
1648 }
1649
1650 static void ww_test_block_context(void)
1651 {
1652         int ret;
1653
1654         WWL1(&o);
1655         WWAI(&t);
1656
1657         ret = WWL(&o2, &t);
1658         WARN_ON(ret);
1659 }
1660
1661 static void ww_test_spin_block(void)
1662 {
1663         L(A);
1664         U(A);
1665
1666         WWL1(&o);
1667         L(A);
1668         U(A);
1669         WWU(&o);
1670
1671         L(A);
1672         WWL1(&o);
1673         WWU(&o);
1674         U(A);
1675 }
1676
1677 static void ww_test_spin_try(void)
1678 {
1679         bool ret;
1680
1681         L(A);
1682         U(A);
1683
1684         ret = WWT(&o);
1685         WARN_ON(!ret);
1686         L(A);
1687         U(A);
1688         WWU(&o);
1689
1690         L(A);
1691         ret = WWT(&o);
1692         WARN_ON(!ret);
1693         WWU(&o);
1694         U(A);
1695 }
1696
1697 static void ww_test_spin_context(void)
1698 {
1699         int ret;
1700
1701         L(A);
1702         U(A);
1703
1704         WWAI(&t);
1705
1706         ret = WWL(&o, &t);
1707         WARN_ON(ret);
1708         L(A);
1709         U(A);
1710         WWU(&o);
1711
1712         L(A);
1713         ret = WWL(&o, &t);
1714         WARN_ON(ret);
1715         WWU(&o);
1716         U(A);
1717 }
1718
1719 static void ww_tests(void)
1720 {
1721         printk("  --------------------------------------------------------------------------\n");
1722         printk("  | Wound/wait tests |\n");
1723         printk("  ---------------------\n");
1724
1725         print_testname("ww api failures");
1726         dotest(ww_test_fail_acquire, SUCCESS, LOCKTYPE_WW);
1727         dotest(ww_test_normal, SUCCESS, LOCKTYPE_WW);
1728         dotest(ww_test_unneeded_slow, FAILURE, LOCKTYPE_WW);
1729         printk("\n");
1730
1731         print_testname("ww contexts mixing");
1732         dotest(ww_test_two_contexts, FAILURE, LOCKTYPE_WW);
1733         dotest(ww_test_diff_class, FAILURE, LOCKTYPE_WW);
1734         printk("\n");
1735
1736         print_testname("finishing ww context");
1737         dotest(ww_test_context_done_twice, FAILURE, LOCKTYPE_WW);
1738         dotest(ww_test_context_unlock_twice, FAILURE, LOCKTYPE_WW);
1739         dotest(ww_test_context_fini_early, FAILURE, LOCKTYPE_WW);
1740         dotest(ww_test_context_lock_after_done, FAILURE, LOCKTYPE_WW);
1741         printk("\n");
1742
1743         print_testname("locking mismatches");
1744         dotest(ww_test_object_unlock_twice, FAILURE, LOCKTYPE_WW);
1745         dotest(ww_test_object_lock_unbalanced, FAILURE, LOCKTYPE_WW);
1746         dotest(ww_test_object_lock_stale_context, FAILURE, LOCKTYPE_WW);
1747         printk("\n");
1748
1749         print_testname("EDEADLK handling");
1750         dotest(ww_test_edeadlk_normal, SUCCESS, LOCKTYPE_WW);
1751         dotest(ww_test_edeadlk_normal_slow, SUCCESS, LOCKTYPE_WW);
1752         dotest(ww_test_edeadlk_no_unlock, FAILURE, LOCKTYPE_WW);
1753         dotest(ww_test_edeadlk_no_unlock_slow, FAILURE, LOCKTYPE_WW);
1754         dotest(ww_test_edeadlk_acquire_more, FAILURE, LOCKTYPE_WW);
1755         dotest(ww_test_edeadlk_acquire_more_slow, FAILURE, LOCKTYPE_WW);
1756         dotest(ww_test_edeadlk_acquire_more_edeadlk, FAILURE, LOCKTYPE_WW);
1757         dotest(ww_test_edeadlk_acquire_more_edeadlk_slow, FAILURE, LOCKTYPE_WW);
1758         dotest(ww_test_edeadlk_acquire_wrong, FAILURE, LOCKTYPE_WW);
1759         dotest(ww_test_edeadlk_acquire_wrong_slow, FAILURE, LOCKTYPE_WW);
1760         printk("\n");
1761
1762         print_testname("spinlock nest unlocked");
1763         dotest(ww_test_spin_nest_unlocked, FAILURE, LOCKTYPE_WW);
1764         printk("\n");
1765
1766         printk("  -----------------------------------------------------\n");
1767         printk("                                 |block | try  |context|\n");
1768         printk("  -----------------------------------------------------\n");
1769
1770         print_testname("context");
1771         dotest(ww_test_context_block, FAILURE, LOCKTYPE_WW);
1772         dotest(ww_test_context_try, SUCCESS, LOCKTYPE_WW);
1773         dotest(ww_test_context_context, SUCCESS, LOCKTYPE_WW);
1774         printk("\n");
1775
1776         print_testname("try");
1777         dotest(ww_test_try_block, FAILURE, LOCKTYPE_WW);
1778         dotest(ww_test_try_try, SUCCESS, LOCKTYPE_WW);
1779         dotest(ww_test_try_context, FAILURE, LOCKTYPE_WW);
1780         printk("\n");
1781
1782         print_testname("block");
1783         dotest(ww_test_block_block, FAILURE, LOCKTYPE_WW);
1784         dotest(ww_test_block_try, SUCCESS, LOCKTYPE_WW);
1785         dotest(ww_test_block_context, FAILURE, LOCKTYPE_WW);
1786         printk("\n");
1787
1788         print_testname("spinlock");
1789         dotest(ww_test_spin_block, FAILURE, LOCKTYPE_WW);
1790         dotest(ww_test_spin_try, SUCCESS, LOCKTYPE_WW);
1791         dotest(ww_test_spin_context, FAILURE, LOCKTYPE_WW);
1792         printk("\n");
1793 }
1794
1795 void locking_selftest(void)
1796 {
1797         /*
1798          * Got a locking failure before the selftest ran?
1799          */
1800         if (!debug_locks) {
1801                 printk("----------------------------------\n");
1802                 printk("| Locking API testsuite disabled |\n");
1803                 printk("----------------------------------\n");
1804                 return;
1805         }
1806
1807         /*
1808          * Run the testsuite:
1809          */
1810         printk("------------------------\n");
1811         printk("| Locking API testsuite:\n");
1812         printk("----------------------------------------------------------------------------\n");
1813         printk("                                 | spin |wlock |rlock |mutex | wsem | rsem |\n");
1814         printk("  --------------------------------------------------------------------------\n");
1815
1816         init_shared_classes();
1817         debug_locks_silent = !debug_locks_verbose;
1818
1819         DO_TESTCASE_6R("A-A deadlock", AA);
1820         DO_TESTCASE_6R("A-B-B-A deadlock", ABBA);
1821         DO_TESTCASE_6R("A-B-B-C-C-A deadlock", ABBCCA);
1822         DO_TESTCASE_6R("A-B-C-A-B-C deadlock", ABCABC);
1823         DO_TESTCASE_6R("A-B-B-C-C-D-D-A deadlock", ABBCCDDA);
1824         DO_TESTCASE_6R("A-B-C-D-B-D-D-A deadlock", ABCDBDDA);
1825         DO_TESTCASE_6R("A-B-C-D-B-C-D-A deadlock", ABCDBCDA);
1826         DO_TESTCASE_6("double unlock", double_unlock);
1827         DO_TESTCASE_6("initialize held", init_held);
1828         DO_TESTCASE_6_SUCCESS("bad unlock order", bad_unlock_order);
1829
1830         printk("  --------------------------------------------------------------------------\n");
1831         print_testname("recursive read-lock");
1832         printk("             |");
1833         dotest(rlock_AA1, SUCCESS, LOCKTYPE_RWLOCK);
1834         printk("             |");
1835         dotest(rsem_AA1, FAILURE, LOCKTYPE_RWSEM);
1836         printk("\n");
1837
1838         print_testname("recursive read-lock #2");
1839         printk("             |");
1840         dotest(rlock_AA1B, SUCCESS, LOCKTYPE_RWLOCK);
1841         printk("             |");
1842         dotest(rsem_AA1B, FAILURE, LOCKTYPE_RWSEM);
1843         printk("\n");
1844
1845         print_testname("mixed read-write-lock");
1846         printk("             |");
1847         dotest(rlock_AA2, FAILURE, LOCKTYPE_RWLOCK);
1848         printk("             |");
1849         dotest(rsem_AA2, FAILURE, LOCKTYPE_RWSEM);
1850         printk("\n");
1851
1852         print_testname("mixed write-read-lock");
1853         printk("             |");
1854         dotest(rlock_AA3, FAILURE, LOCKTYPE_RWLOCK);
1855         printk("             |");
1856         dotest(rsem_AA3, FAILURE, LOCKTYPE_RWSEM);
1857         printk("\n");
1858
1859         printk("  --------------------------------------------------------------------------\n");
1860
1861         /*
1862          * irq-context testcases:
1863          */
1864         DO_TESTCASE_2x6("irqs-on + irq-safe-A", irqsafe1);
1865         DO_TESTCASE_2x3("sirq-safe-A => hirqs-on", irqsafe2A);
1866         DO_TESTCASE_2x6("safe-A + irqs-on", irqsafe2B);
1867         DO_TESTCASE_6x6("safe-A + unsafe-B #1", irqsafe3);
1868         DO_TESTCASE_6x6("safe-A + unsafe-B #2", irqsafe4);
1869         DO_TESTCASE_6x6RW("irq lock-inversion", irq_inversion);
1870
1871         DO_TESTCASE_6x2("irq read-recursion", irq_read_recursion);
1872 //      DO_TESTCASE_6x2B("irq read-recursion #2", irq_read_recursion2);
1873
1874         ww_tests();
1875
1876         if (unexpected_testcase_failures) {
1877                 printk("-----------------------------------------------------------------\n");
1878                 debug_locks = 0;
1879                 printk("BUG: %3d unexpected failures (out of %3d) - debugging disabled! |\n",
1880                         unexpected_testcase_failures, testcase_total);
1881                 printk("-----------------------------------------------------------------\n");
1882         } else if (expected_testcase_failures && testcase_successes) {
1883                 printk("--------------------------------------------------------\n");
1884                 printk("%3d out of %3d testcases failed, as expected. |\n",
1885                         expected_testcase_failures, testcase_total);
1886                 printk("----------------------------------------------------\n");
1887                 debug_locks = 1;
1888         } else if (expected_testcase_failures && !testcase_successes) {
1889                 printk("--------------------------------------------------------\n");
1890                 printk("All %3d testcases failed, as expected. |\n",
1891                         expected_testcase_failures);
1892                 printk("----------------------------------------\n");
1893                 debug_locks = 1;
1894         } else {
1895                 printk("-------------------------------------------------------\n");
1896                 printk("Good, all %3d testcases passed! |\n",
1897                         testcase_successes);
1898                 printk("---------------------------------\n");
1899                 debug_locks = 1;
1900         }
1901         debug_locks_silent = 0;
1902 }