xref: /openbmc/linux/lib/locking-selftest.c (revision b34e08d5)
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 }
1903