xref: /openbmc/linux/lib/locking-selftest.c (revision 26b32974)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * lib/locking-selftest.c
4  *
5  * Testsuite for various locking APIs: spinlocks, rwlocks,
6  * mutexes and rw-semaphores.
7  *
8  * It is checking both false positives and false negatives.
9  *
10  * Started by Ingo Molnar:
11  *
12  *  Copyright (C) 2006 Red Hat, Inc., Ingo Molnar <mingo@redhat.com>
13  */
14 #include <linux/rwsem.h>
15 #include <linux/mutex.h>
16 #include <linux/ww_mutex.h>
17 #include <linux/sched.h>
18 #include <linux/sched/mm.h>
19 #include <linux/delay.h>
20 #include <linux/lockdep.h>
21 #include <linux/spinlock.h>
22 #include <linux/kallsyms.h>
23 #include <linux/interrupt.h>
24 #include <linux/debug_locks.h>
25 #include <linux/irqflags.h>
26 #include <linux/rtmutex.h>
27 #include <linux/local_lock.h>
28 
29 #ifdef CONFIG_PREEMPT_RT
30 # define NON_RT(...)
31 #else
32 # define NON_RT(...)	__VA_ARGS__
33 #endif
34 
35 /*
36  * Change this to 1 if you want to see the failure printouts:
37  */
38 static unsigned int debug_locks_verbose;
39 unsigned int force_read_lock_recursive;
40 
41 static DEFINE_WD_CLASS(ww_lockdep);
42 
43 static int __init setup_debug_locks_verbose(char *str)
44 {
45 	get_option(&str, &debug_locks_verbose);
46 
47 	return 1;
48 }
49 
50 __setup("debug_locks_verbose=", setup_debug_locks_verbose);
51 
52 #define FAILURE		0
53 #define SUCCESS		1
54 
55 #define LOCKTYPE_SPIN	0x1
56 #define LOCKTYPE_RWLOCK	0x2
57 #define LOCKTYPE_MUTEX	0x4
58 #define LOCKTYPE_RWSEM	0x8
59 #define LOCKTYPE_WW	0x10
60 #define LOCKTYPE_RTMUTEX 0x20
61 #define LOCKTYPE_LL	0x40
62 #define LOCKTYPE_SPECIAL 0x80
63 
64 static struct ww_acquire_ctx t, t2;
65 static struct ww_mutex o, o2, o3;
66 
67 /*
68  * Normal standalone locks, for the circular and irq-context
69  * dependency tests:
70  */
71 static DEFINE_SPINLOCK(lock_A);
72 static DEFINE_SPINLOCK(lock_B);
73 static DEFINE_SPINLOCK(lock_C);
74 static DEFINE_SPINLOCK(lock_D);
75 
76 static DEFINE_RAW_SPINLOCK(raw_lock_A);
77 static DEFINE_RAW_SPINLOCK(raw_lock_B);
78 
79 static DEFINE_RWLOCK(rwlock_A);
80 static DEFINE_RWLOCK(rwlock_B);
81 static DEFINE_RWLOCK(rwlock_C);
82 static DEFINE_RWLOCK(rwlock_D);
83 
84 static DEFINE_MUTEX(mutex_A);
85 static DEFINE_MUTEX(mutex_B);
86 static DEFINE_MUTEX(mutex_C);
87 static DEFINE_MUTEX(mutex_D);
88 
89 static DECLARE_RWSEM(rwsem_A);
90 static DECLARE_RWSEM(rwsem_B);
91 static DECLARE_RWSEM(rwsem_C);
92 static DECLARE_RWSEM(rwsem_D);
93 
94 #ifdef CONFIG_RT_MUTEXES
95 
96 static DEFINE_RT_MUTEX(rtmutex_A);
97 static DEFINE_RT_MUTEX(rtmutex_B);
98 static DEFINE_RT_MUTEX(rtmutex_C);
99 static DEFINE_RT_MUTEX(rtmutex_D);
100 
101 #endif
102 
103 /*
104  * Locks that we initialize dynamically as well so that
105  * e.g. X1 and X2 becomes two instances of the same class,
106  * but X* and Y* are different classes. We do this so that
107  * we do not trigger a real lockup:
108  */
109 static DEFINE_SPINLOCK(lock_X1);
110 static DEFINE_SPINLOCK(lock_X2);
111 static DEFINE_SPINLOCK(lock_Y1);
112 static DEFINE_SPINLOCK(lock_Y2);
113 static DEFINE_SPINLOCK(lock_Z1);
114 static DEFINE_SPINLOCK(lock_Z2);
115 
116 static DEFINE_RWLOCK(rwlock_X1);
117 static DEFINE_RWLOCK(rwlock_X2);
118 static DEFINE_RWLOCK(rwlock_Y1);
119 static DEFINE_RWLOCK(rwlock_Y2);
120 static DEFINE_RWLOCK(rwlock_Z1);
121 static DEFINE_RWLOCK(rwlock_Z2);
122 
123 static DEFINE_MUTEX(mutex_X1);
124 static DEFINE_MUTEX(mutex_X2);
125 static DEFINE_MUTEX(mutex_Y1);
126 static DEFINE_MUTEX(mutex_Y2);
127 static DEFINE_MUTEX(mutex_Z1);
128 static DEFINE_MUTEX(mutex_Z2);
129 
130 static DECLARE_RWSEM(rwsem_X1);
131 static DECLARE_RWSEM(rwsem_X2);
132 static DECLARE_RWSEM(rwsem_Y1);
133 static DECLARE_RWSEM(rwsem_Y2);
134 static DECLARE_RWSEM(rwsem_Z1);
135 static DECLARE_RWSEM(rwsem_Z2);
136 
137 #ifdef CONFIG_RT_MUTEXES
138 
139 static DEFINE_RT_MUTEX(rtmutex_X1);
140 static DEFINE_RT_MUTEX(rtmutex_X2);
141 static DEFINE_RT_MUTEX(rtmutex_Y1);
142 static DEFINE_RT_MUTEX(rtmutex_Y2);
143 static DEFINE_RT_MUTEX(rtmutex_Z1);
144 static DEFINE_RT_MUTEX(rtmutex_Z2);
145 
146 #endif
147 
148 static DEFINE_PER_CPU(local_lock_t, local_A);
149 
150 /*
151  * non-inlined runtime initializers, to let separate locks share
152  * the same lock-class:
153  */
154 #define INIT_CLASS_FUNC(class) 				\
155 static noinline void					\
156 init_class_##class(spinlock_t *lock, rwlock_t *rwlock, \
157 	struct mutex *mutex, struct rw_semaphore *rwsem)\
158 {							\
159 	spin_lock_init(lock);			\
160 	rwlock_init(rwlock);				\
161 	mutex_init(mutex);				\
162 	init_rwsem(rwsem);				\
163 }
164 
165 INIT_CLASS_FUNC(X)
166 INIT_CLASS_FUNC(Y)
167 INIT_CLASS_FUNC(Z)
168 
169 static void init_shared_classes(void)
170 {
171 #ifdef CONFIG_RT_MUTEXES
172 	static struct lock_class_key rt_X, rt_Y, rt_Z;
173 
174 	__rt_mutex_init(&rtmutex_X1, __func__, &rt_X);
175 	__rt_mutex_init(&rtmutex_X2, __func__, &rt_X);
176 	__rt_mutex_init(&rtmutex_Y1, __func__, &rt_Y);
177 	__rt_mutex_init(&rtmutex_Y2, __func__, &rt_Y);
178 	__rt_mutex_init(&rtmutex_Z1, __func__, &rt_Z);
179 	__rt_mutex_init(&rtmutex_Z2, __func__, &rt_Z);
180 #endif
181 
182 	init_class_X(&lock_X1, &rwlock_X1, &mutex_X1, &rwsem_X1);
183 	init_class_X(&lock_X2, &rwlock_X2, &mutex_X2, &rwsem_X2);
184 
185 	init_class_Y(&lock_Y1, &rwlock_Y1, &mutex_Y1, &rwsem_Y1);
186 	init_class_Y(&lock_Y2, &rwlock_Y2, &mutex_Y2, &rwsem_Y2);
187 
188 	init_class_Z(&lock_Z1, &rwlock_Z1, &mutex_Z1, &rwsem_Z1);
189 	init_class_Z(&lock_Z2, &rwlock_Z2, &mutex_Z2, &rwsem_Z2);
190 }
191 
192 /*
193  * For spinlocks and rwlocks we also do hardirq-safe / softirq-safe tests.
194  * The following functions use a lock from a simulated hardirq/softirq
195  * context, causing the locks to be marked as hardirq-safe/softirq-safe:
196  */
197 
198 #define HARDIRQ_DISABLE		local_irq_disable
199 #define HARDIRQ_ENABLE		local_irq_enable
200 
201 #define HARDIRQ_ENTER()				\
202 	local_irq_disable();			\
203 	__irq_enter();				\
204 	lockdep_hardirq_threaded();		\
205 	WARN_ON(!in_irq());
206 
207 #define HARDIRQ_EXIT()				\
208 	__irq_exit();				\
209 	local_irq_enable();
210 
211 #define SOFTIRQ_DISABLE		local_bh_disable
212 #define SOFTIRQ_ENABLE		local_bh_enable
213 
214 #define SOFTIRQ_ENTER()				\
215 		local_bh_disable();		\
216 		local_irq_disable();		\
217 		lockdep_softirq_enter();	\
218 		WARN_ON(!in_softirq());
219 
220 #define SOFTIRQ_EXIT()				\
221 		lockdep_softirq_exit();		\
222 		local_irq_enable();		\
223 		local_bh_enable();
224 
225 /*
226  * Shortcuts for lock/unlock API variants, to keep
227  * the testcases compact:
228  */
229 #define L(x)			spin_lock(&lock_##x)
230 #define U(x)			spin_unlock(&lock_##x)
231 #define LU(x)			L(x); U(x)
232 #define SI(x)			spin_lock_init(&lock_##x)
233 
234 #define WL(x)			write_lock(&rwlock_##x)
235 #define WU(x)			write_unlock(&rwlock_##x)
236 #define WLU(x)			WL(x); WU(x)
237 
238 #define RL(x)			read_lock(&rwlock_##x)
239 #define RU(x)			read_unlock(&rwlock_##x)
240 #define RLU(x)			RL(x); RU(x)
241 #define RWI(x)			rwlock_init(&rwlock_##x)
242 
243 #define ML(x)			mutex_lock(&mutex_##x)
244 #define MU(x)			mutex_unlock(&mutex_##x)
245 #define MI(x)			mutex_init(&mutex_##x)
246 
247 #define RTL(x)			rt_mutex_lock(&rtmutex_##x)
248 #define RTU(x)			rt_mutex_unlock(&rtmutex_##x)
249 #define RTI(x)			rt_mutex_init(&rtmutex_##x)
250 
251 #define WSL(x)			down_write(&rwsem_##x)
252 #define WSU(x)			up_write(&rwsem_##x)
253 
254 #define RSL(x)			down_read(&rwsem_##x)
255 #define RSU(x)			up_read(&rwsem_##x)
256 #define RWSI(x)			init_rwsem(&rwsem_##x)
257 
258 #ifndef CONFIG_DEBUG_WW_MUTEX_SLOWPATH
259 #define WWAI(x)			ww_acquire_init(x, &ww_lockdep)
260 #else
261 #define WWAI(x)			do { ww_acquire_init(x, &ww_lockdep); (x)->deadlock_inject_countdown = ~0U; } while (0)
262 #endif
263 #define WWAD(x)			ww_acquire_done(x)
264 #define WWAF(x)			ww_acquire_fini(x)
265 
266 #define WWL(x, c)		ww_mutex_lock(x, c)
267 #define WWT(x)			ww_mutex_trylock(x, NULL)
268 #define WWL1(x)			ww_mutex_lock(x, NULL)
269 #define WWU(x)			ww_mutex_unlock(x)
270 
271 
272 #define LOCK_UNLOCK_2(x,y)	LOCK(x); LOCK(y); UNLOCK(y); UNLOCK(x)
273 
274 /*
275  * Generate different permutations of the same testcase, using
276  * the same basic lock-dependency/state events:
277  */
278 
279 #define GENERATE_TESTCASE(name)			\
280 						\
281 static void name(void) { E(); }
282 
283 #define GENERATE_PERMUTATIONS_2_EVENTS(name)	\
284 						\
285 static void name##_12(void) { E1(); E2(); }	\
286 static void name##_21(void) { E2(); E1(); }
287 
288 #define GENERATE_PERMUTATIONS_3_EVENTS(name)		\
289 							\
290 static void name##_123(void) { E1(); E2(); E3(); }	\
291 static void name##_132(void) { E1(); E3(); E2(); }	\
292 static void name##_213(void) { E2(); E1(); E3(); }	\
293 static void name##_231(void) { E2(); E3(); E1(); }	\
294 static void name##_312(void) { E3(); E1(); E2(); }	\
295 static void name##_321(void) { E3(); E2(); E1(); }
296 
297 /*
298  * AA deadlock:
299  */
300 
301 #define E()					\
302 						\
303 	LOCK(X1);				\
304 	LOCK(X2); /* this one should fail */
305 
306 /*
307  * 6 testcases:
308  */
309 #include "locking-selftest-spin.h"
310 GENERATE_TESTCASE(AA_spin)
311 #include "locking-selftest-wlock.h"
312 GENERATE_TESTCASE(AA_wlock)
313 #include "locking-selftest-rlock.h"
314 GENERATE_TESTCASE(AA_rlock)
315 #include "locking-selftest-mutex.h"
316 GENERATE_TESTCASE(AA_mutex)
317 #include "locking-selftest-wsem.h"
318 GENERATE_TESTCASE(AA_wsem)
319 #include "locking-selftest-rsem.h"
320 GENERATE_TESTCASE(AA_rsem)
321 
322 #ifdef CONFIG_RT_MUTEXES
323 #include "locking-selftest-rtmutex.h"
324 GENERATE_TESTCASE(AA_rtmutex);
325 #endif
326 
327 #undef E
328 
329 /*
330  * Special-case for read-locking, they are
331  * allowed to recurse on the same lock class:
332  */
333 static void rlock_AA1(void)
334 {
335 	RL(X1);
336 	RL(X1); // this one should NOT fail
337 }
338 
339 static void rlock_AA1B(void)
340 {
341 	RL(X1);
342 	RL(X2); // this one should NOT fail
343 }
344 
345 static void rsem_AA1(void)
346 {
347 	RSL(X1);
348 	RSL(X1); // this one should fail
349 }
350 
351 static void rsem_AA1B(void)
352 {
353 	RSL(X1);
354 	RSL(X2); // this one should fail
355 }
356 /*
357  * The mixing of read and write locks is not allowed:
358  */
359 static void rlock_AA2(void)
360 {
361 	RL(X1);
362 	WL(X2); // this one should fail
363 }
364 
365 static void rsem_AA2(void)
366 {
367 	RSL(X1);
368 	WSL(X2); // this one should fail
369 }
370 
371 static void rlock_AA3(void)
372 {
373 	WL(X1);
374 	RL(X2); // this one should fail
375 }
376 
377 static void rsem_AA3(void)
378 {
379 	WSL(X1);
380 	RSL(X2); // this one should fail
381 }
382 
383 /*
384  * read_lock(A)
385  * spin_lock(B)
386  *		spin_lock(B)
387  *		write_lock(A)
388  */
389 static void rlock_ABBA1(void)
390 {
391 	RL(X1);
392 	L(Y1);
393 	U(Y1);
394 	RU(X1);
395 
396 	L(Y1);
397 	WL(X1);
398 	WU(X1);
399 	U(Y1); // should fail
400 }
401 
402 static void rwsem_ABBA1(void)
403 {
404 	RSL(X1);
405 	ML(Y1);
406 	MU(Y1);
407 	RSU(X1);
408 
409 	ML(Y1);
410 	WSL(X1);
411 	WSU(X1);
412 	MU(Y1); // should fail
413 }
414 
415 /*
416  * read_lock(A)
417  * spin_lock(B)
418  *		spin_lock(B)
419  *		write_lock(A)
420  *
421  * This test case is aimed at poking whether the chain cache prevents us from
422  * detecting a read-lock/lock-write deadlock: if the chain cache doesn't differ
423  * read/write locks, the following case may happen
424  *
425  * 	{ read_lock(A)->lock(B) dependency exists }
426  *
427  * 	P0:
428  * 	lock(B);
429  * 	read_lock(A);
430  *
431  *	{ Not a deadlock, B -> A is added in the chain cache }
432  *
433  *	P1:
434  *	lock(B);
435  *	write_lock(A);
436  *
437  *	{ B->A found in chain cache, not reported as a deadlock }
438  *
439  */
440 static void rlock_chaincache_ABBA1(void)
441 {
442 	RL(X1);
443 	L(Y1);
444 	U(Y1);
445 	RU(X1);
446 
447 	L(Y1);
448 	RL(X1);
449 	RU(X1);
450 	U(Y1);
451 
452 	L(Y1);
453 	WL(X1);
454 	WU(X1);
455 	U(Y1); // should fail
456 }
457 
458 /*
459  * read_lock(A)
460  * spin_lock(B)
461  *		spin_lock(B)
462  *		read_lock(A)
463  */
464 static void rlock_ABBA2(void)
465 {
466 	RL(X1);
467 	L(Y1);
468 	U(Y1);
469 	RU(X1);
470 
471 	L(Y1);
472 	RL(X1);
473 	RU(X1);
474 	U(Y1); // should NOT fail
475 }
476 
477 static void rwsem_ABBA2(void)
478 {
479 	RSL(X1);
480 	ML(Y1);
481 	MU(Y1);
482 	RSU(X1);
483 
484 	ML(Y1);
485 	RSL(X1);
486 	RSU(X1);
487 	MU(Y1); // should fail
488 }
489 
490 
491 /*
492  * write_lock(A)
493  * spin_lock(B)
494  *		spin_lock(B)
495  *		write_lock(A)
496  */
497 static void rlock_ABBA3(void)
498 {
499 	WL(X1);
500 	L(Y1);
501 	U(Y1);
502 	WU(X1);
503 
504 	L(Y1);
505 	WL(X1);
506 	WU(X1);
507 	U(Y1); // should fail
508 }
509 
510 static void rwsem_ABBA3(void)
511 {
512 	WSL(X1);
513 	ML(Y1);
514 	MU(Y1);
515 	WSU(X1);
516 
517 	ML(Y1);
518 	WSL(X1);
519 	WSU(X1);
520 	MU(Y1); // should fail
521 }
522 
523 /*
524  * ABBA deadlock:
525  */
526 
527 #define E()					\
528 						\
529 	LOCK_UNLOCK_2(A, B);			\
530 	LOCK_UNLOCK_2(B, A); /* fail */
531 
532 /*
533  * 6 testcases:
534  */
535 #include "locking-selftest-spin.h"
536 GENERATE_TESTCASE(ABBA_spin)
537 #include "locking-selftest-wlock.h"
538 GENERATE_TESTCASE(ABBA_wlock)
539 #include "locking-selftest-rlock.h"
540 GENERATE_TESTCASE(ABBA_rlock)
541 #include "locking-selftest-mutex.h"
542 GENERATE_TESTCASE(ABBA_mutex)
543 #include "locking-selftest-wsem.h"
544 GENERATE_TESTCASE(ABBA_wsem)
545 #include "locking-selftest-rsem.h"
546 GENERATE_TESTCASE(ABBA_rsem)
547 
548 #ifdef CONFIG_RT_MUTEXES
549 #include "locking-selftest-rtmutex.h"
550 GENERATE_TESTCASE(ABBA_rtmutex);
551 #endif
552 
553 #undef E
554 
555 /*
556  * AB BC CA deadlock:
557  */
558 
559 #define E()					\
560 						\
561 	LOCK_UNLOCK_2(A, B);			\
562 	LOCK_UNLOCK_2(B, C);			\
563 	LOCK_UNLOCK_2(C, A); /* fail */
564 
565 /*
566  * 6 testcases:
567  */
568 #include "locking-selftest-spin.h"
569 GENERATE_TESTCASE(ABBCCA_spin)
570 #include "locking-selftest-wlock.h"
571 GENERATE_TESTCASE(ABBCCA_wlock)
572 #include "locking-selftest-rlock.h"
573 GENERATE_TESTCASE(ABBCCA_rlock)
574 #include "locking-selftest-mutex.h"
575 GENERATE_TESTCASE(ABBCCA_mutex)
576 #include "locking-selftest-wsem.h"
577 GENERATE_TESTCASE(ABBCCA_wsem)
578 #include "locking-selftest-rsem.h"
579 GENERATE_TESTCASE(ABBCCA_rsem)
580 
581 #ifdef CONFIG_RT_MUTEXES
582 #include "locking-selftest-rtmutex.h"
583 GENERATE_TESTCASE(ABBCCA_rtmutex);
584 #endif
585 
586 #undef E
587 
588 /*
589  * AB CA BC deadlock:
590  */
591 
592 #define E()					\
593 						\
594 	LOCK_UNLOCK_2(A, B);			\
595 	LOCK_UNLOCK_2(C, A);			\
596 	LOCK_UNLOCK_2(B, C); /* fail */
597 
598 /*
599  * 6 testcases:
600  */
601 #include "locking-selftest-spin.h"
602 GENERATE_TESTCASE(ABCABC_spin)
603 #include "locking-selftest-wlock.h"
604 GENERATE_TESTCASE(ABCABC_wlock)
605 #include "locking-selftest-rlock.h"
606 GENERATE_TESTCASE(ABCABC_rlock)
607 #include "locking-selftest-mutex.h"
608 GENERATE_TESTCASE(ABCABC_mutex)
609 #include "locking-selftest-wsem.h"
610 GENERATE_TESTCASE(ABCABC_wsem)
611 #include "locking-selftest-rsem.h"
612 GENERATE_TESTCASE(ABCABC_rsem)
613 
614 #ifdef CONFIG_RT_MUTEXES
615 #include "locking-selftest-rtmutex.h"
616 GENERATE_TESTCASE(ABCABC_rtmutex);
617 #endif
618 
619 #undef E
620 
621 /*
622  * AB BC CD DA deadlock:
623  */
624 
625 #define E()					\
626 						\
627 	LOCK_UNLOCK_2(A, B);			\
628 	LOCK_UNLOCK_2(B, C);			\
629 	LOCK_UNLOCK_2(C, D);			\
630 	LOCK_UNLOCK_2(D, A); /* fail */
631 
632 /*
633  * 6 testcases:
634  */
635 #include "locking-selftest-spin.h"
636 GENERATE_TESTCASE(ABBCCDDA_spin)
637 #include "locking-selftest-wlock.h"
638 GENERATE_TESTCASE(ABBCCDDA_wlock)
639 #include "locking-selftest-rlock.h"
640 GENERATE_TESTCASE(ABBCCDDA_rlock)
641 #include "locking-selftest-mutex.h"
642 GENERATE_TESTCASE(ABBCCDDA_mutex)
643 #include "locking-selftest-wsem.h"
644 GENERATE_TESTCASE(ABBCCDDA_wsem)
645 #include "locking-selftest-rsem.h"
646 GENERATE_TESTCASE(ABBCCDDA_rsem)
647 
648 #ifdef CONFIG_RT_MUTEXES
649 #include "locking-selftest-rtmutex.h"
650 GENERATE_TESTCASE(ABBCCDDA_rtmutex);
651 #endif
652 
653 #undef E
654 
655 /*
656  * AB CD BD DA deadlock:
657  */
658 #define E()					\
659 						\
660 	LOCK_UNLOCK_2(A, B);			\
661 	LOCK_UNLOCK_2(C, D);			\
662 	LOCK_UNLOCK_2(B, D);			\
663 	LOCK_UNLOCK_2(D, A); /* fail */
664 
665 /*
666  * 6 testcases:
667  */
668 #include "locking-selftest-spin.h"
669 GENERATE_TESTCASE(ABCDBDDA_spin)
670 #include "locking-selftest-wlock.h"
671 GENERATE_TESTCASE(ABCDBDDA_wlock)
672 #include "locking-selftest-rlock.h"
673 GENERATE_TESTCASE(ABCDBDDA_rlock)
674 #include "locking-selftest-mutex.h"
675 GENERATE_TESTCASE(ABCDBDDA_mutex)
676 #include "locking-selftest-wsem.h"
677 GENERATE_TESTCASE(ABCDBDDA_wsem)
678 #include "locking-selftest-rsem.h"
679 GENERATE_TESTCASE(ABCDBDDA_rsem)
680 
681 #ifdef CONFIG_RT_MUTEXES
682 #include "locking-selftest-rtmutex.h"
683 GENERATE_TESTCASE(ABCDBDDA_rtmutex);
684 #endif
685 
686 #undef E
687 
688 /*
689  * AB CD BC DA deadlock:
690  */
691 #define E()					\
692 						\
693 	LOCK_UNLOCK_2(A, B);			\
694 	LOCK_UNLOCK_2(C, D);			\
695 	LOCK_UNLOCK_2(B, C);			\
696 	LOCK_UNLOCK_2(D, A); /* fail */
697 
698 /*
699  * 6 testcases:
700  */
701 #include "locking-selftest-spin.h"
702 GENERATE_TESTCASE(ABCDBCDA_spin)
703 #include "locking-selftest-wlock.h"
704 GENERATE_TESTCASE(ABCDBCDA_wlock)
705 #include "locking-selftest-rlock.h"
706 GENERATE_TESTCASE(ABCDBCDA_rlock)
707 #include "locking-selftest-mutex.h"
708 GENERATE_TESTCASE(ABCDBCDA_mutex)
709 #include "locking-selftest-wsem.h"
710 GENERATE_TESTCASE(ABCDBCDA_wsem)
711 #include "locking-selftest-rsem.h"
712 GENERATE_TESTCASE(ABCDBCDA_rsem)
713 
714 #ifdef CONFIG_RT_MUTEXES
715 #include "locking-selftest-rtmutex.h"
716 GENERATE_TESTCASE(ABCDBCDA_rtmutex);
717 #endif
718 
719 #undef E
720 
721 #ifdef CONFIG_PREEMPT_RT
722 # define RT_PREPARE_DBL_UNLOCK()	{ migrate_disable(); rcu_read_lock(); }
723 #else
724 # define RT_PREPARE_DBL_UNLOCK()
725 #endif
726 /*
727  * Double unlock:
728  */
729 #define E()					\
730 						\
731 	LOCK(A);				\
732 	RT_PREPARE_DBL_UNLOCK();		\
733 	UNLOCK(A);				\
734 	UNLOCK(A); /* fail */
735 
736 /*
737  * 6 testcases:
738  */
739 #include "locking-selftest-spin.h"
740 GENERATE_TESTCASE(double_unlock_spin)
741 #include "locking-selftest-wlock.h"
742 GENERATE_TESTCASE(double_unlock_wlock)
743 #include "locking-selftest-rlock.h"
744 GENERATE_TESTCASE(double_unlock_rlock)
745 #include "locking-selftest-mutex.h"
746 GENERATE_TESTCASE(double_unlock_mutex)
747 #include "locking-selftest-wsem.h"
748 GENERATE_TESTCASE(double_unlock_wsem)
749 #include "locking-selftest-rsem.h"
750 GENERATE_TESTCASE(double_unlock_rsem)
751 
752 #ifdef CONFIG_RT_MUTEXES
753 #include "locking-selftest-rtmutex.h"
754 GENERATE_TESTCASE(double_unlock_rtmutex);
755 #endif
756 
757 #undef E
758 
759 /*
760  * initializing a held lock:
761  */
762 #define E()					\
763 						\
764 	LOCK(A);				\
765 	INIT(A); /* fail */
766 
767 /*
768  * 6 testcases:
769  */
770 #include "locking-selftest-spin.h"
771 GENERATE_TESTCASE(init_held_spin)
772 #include "locking-selftest-wlock.h"
773 GENERATE_TESTCASE(init_held_wlock)
774 #include "locking-selftest-rlock.h"
775 GENERATE_TESTCASE(init_held_rlock)
776 #include "locking-selftest-mutex.h"
777 GENERATE_TESTCASE(init_held_mutex)
778 #include "locking-selftest-wsem.h"
779 GENERATE_TESTCASE(init_held_wsem)
780 #include "locking-selftest-rsem.h"
781 GENERATE_TESTCASE(init_held_rsem)
782 
783 #ifdef CONFIG_RT_MUTEXES
784 #include "locking-selftest-rtmutex.h"
785 GENERATE_TESTCASE(init_held_rtmutex);
786 #endif
787 
788 #undef E
789 
790 /*
791  * locking an irq-safe lock with irqs enabled:
792  */
793 #define E1()				\
794 					\
795 	IRQ_ENTER();			\
796 	LOCK(A);			\
797 	UNLOCK(A);			\
798 	IRQ_EXIT();
799 
800 #define E2()				\
801 					\
802 	LOCK(A);			\
803 	UNLOCK(A);
804 
805 /*
806  * Generate 24 testcases:
807  */
808 #include "locking-selftest-spin-hardirq.h"
809 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_spin)
810 
811 #include "locking-selftest-rlock-hardirq.h"
812 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_rlock)
813 
814 #include "locking-selftest-wlock-hardirq.h"
815 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_wlock)
816 
817 #ifndef CONFIG_PREEMPT_RT
818 #include "locking-selftest-spin-softirq.h"
819 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_soft_spin)
820 
821 #include "locking-selftest-rlock-softirq.h"
822 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_soft_rlock)
823 
824 #include "locking-selftest-wlock-softirq.h"
825 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_soft_wlock)
826 #endif
827 
828 #undef E1
829 #undef E2
830 
831 #ifndef CONFIG_PREEMPT_RT
832 /*
833  * Enabling hardirqs with a softirq-safe lock held:
834  */
835 #define E1()				\
836 					\
837 	SOFTIRQ_ENTER();		\
838 	LOCK(A);			\
839 	UNLOCK(A);			\
840 	SOFTIRQ_EXIT();
841 
842 #define E2()				\
843 					\
844 	HARDIRQ_DISABLE();		\
845 	LOCK(A);			\
846 	HARDIRQ_ENABLE();		\
847 	UNLOCK(A);
848 
849 /*
850  * Generate 12 testcases:
851  */
852 #include "locking-selftest-spin.h"
853 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2A_spin)
854 
855 #include "locking-selftest-wlock.h"
856 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2A_wlock)
857 
858 #include "locking-selftest-rlock.h"
859 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2A_rlock)
860 
861 #undef E1
862 #undef E2
863 
864 #endif
865 
866 /*
867  * Enabling irqs with an irq-safe lock held:
868  */
869 #define E1()				\
870 					\
871 	IRQ_ENTER();			\
872 	LOCK(A);			\
873 	UNLOCK(A);			\
874 	IRQ_EXIT();
875 
876 #define E2()				\
877 					\
878 	IRQ_DISABLE();			\
879 	LOCK(A);			\
880 	IRQ_ENABLE();			\
881 	UNLOCK(A);
882 
883 /*
884  * Generate 24 testcases:
885  */
886 #include "locking-selftest-spin-hardirq.h"
887 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_spin)
888 
889 #include "locking-selftest-rlock-hardirq.h"
890 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_rlock)
891 
892 #include "locking-selftest-wlock-hardirq.h"
893 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_wlock)
894 
895 #ifndef CONFIG_PREEMPT_RT
896 #include "locking-selftest-spin-softirq.h"
897 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_soft_spin)
898 
899 #include "locking-selftest-rlock-softirq.h"
900 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_soft_rlock)
901 
902 #include "locking-selftest-wlock-softirq.h"
903 GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_soft_wlock)
904 #endif
905 
906 #undef E1
907 #undef E2
908 
909 /*
910  * Acquiring a irq-unsafe lock while holding an irq-safe-lock:
911  */
912 #define E1()				\
913 					\
914 	LOCK(A);			\
915 	LOCK(B);			\
916 	UNLOCK(B);			\
917 	UNLOCK(A);			\
918 
919 #define E2()				\
920 					\
921 	LOCK(B);			\
922 	UNLOCK(B);
923 
924 #define E3()				\
925 					\
926 	IRQ_ENTER();			\
927 	LOCK(A);			\
928 	UNLOCK(A);			\
929 	IRQ_EXIT();
930 
931 /*
932  * Generate 36 testcases:
933  */
934 #include "locking-selftest-spin-hardirq.h"
935 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_spin)
936 
937 #include "locking-selftest-rlock-hardirq.h"
938 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_rlock)
939 
940 #include "locking-selftest-wlock-hardirq.h"
941 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_wlock)
942 
943 #ifndef CONFIG_PREEMPT_RT
944 #include "locking-selftest-spin-softirq.h"
945 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_soft_spin)
946 
947 #include "locking-selftest-rlock-softirq.h"
948 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_soft_rlock)
949 
950 #include "locking-selftest-wlock-softirq.h"
951 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_soft_wlock)
952 #endif
953 
954 #undef E1
955 #undef E2
956 #undef E3
957 
958 /*
959  * If a lock turns into softirq-safe, but earlier it took
960  * a softirq-unsafe lock:
961  */
962 
963 #define E1()				\
964 	IRQ_DISABLE();			\
965 	LOCK(A);			\
966 	LOCK(B);			\
967 	UNLOCK(B);			\
968 	UNLOCK(A);			\
969 	IRQ_ENABLE();
970 
971 #define E2()				\
972 	LOCK(B);			\
973 	UNLOCK(B);
974 
975 #define E3()				\
976 	IRQ_ENTER();			\
977 	LOCK(A);			\
978 	UNLOCK(A);			\
979 	IRQ_EXIT();
980 
981 /*
982  * Generate 36 testcases:
983  */
984 #include "locking-selftest-spin-hardirq.h"
985 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_spin)
986 
987 #include "locking-selftest-rlock-hardirq.h"
988 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_rlock)
989 
990 #include "locking-selftest-wlock-hardirq.h"
991 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_wlock)
992 
993 #ifndef CONFIG_PREEMPT_RT
994 #include "locking-selftest-spin-softirq.h"
995 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_soft_spin)
996 
997 #include "locking-selftest-rlock-softirq.h"
998 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_soft_rlock)
999 
1000 #include "locking-selftest-wlock-softirq.h"
1001 GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_soft_wlock)
1002 #endif
1003 
1004 #undef E1
1005 #undef E2
1006 #undef E3
1007 
1008 /*
1009  * read-lock / write-lock irq inversion.
1010  *
1011  * Deadlock scenario:
1012  *
1013  * CPU#1 is at #1, i.e. it has write-locked A, but has not
1014  * taken B yet.
1015  *
1016  * CPU#2 is at #2, i.e. it has locked B.
1017  *
1018  * Hardirq hits CPU#2 at point #2 and is trying to read-lock A.
1019  *
1020  * The deadlock occurs because CPU#1 will spin on B, and CPU#2
1021  * will spin on A.
1022  */
1023 
1024 #define E1()				\
1025 					\
1026 	IRQ_DISABLE();			\
1027 	WL(A);				\
1028 	LOCK(B);			\
1029 	UNLOCK(B);			\
1030 	WU(A);				\
1031 	IRQ_ENABLE();
1032 
1033 #define E2()				\
1034 					\
1035 	LOCK(B);			\
1036 	UNLOCK(B);
1037 
1038 #define E3()				\
1039 					\
1040 	IRQ_ENTER();			\
1041 	RL(A);				\
1042 	RU(A);				\
1043 	IRQ_EXIT();
1044 
1045 /*
1046  * Generate 36 testcases:
1047  */
1048 #include "locking-selftest-spin-hardirq.h"
1049 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_spin)
1050 
1051 #include "locking-selftest-rlock-hardirq.h"
1052 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_rlock)
1053 
1054 #include "locking-selftest-wlock-hardirq.h"
1055 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_wlock)
1056 
1057 #ifndef CONFIG_PREEMPT_RT
1058 #include "locking-selftest-spin-softirq.h"
1059 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_soft_spin)
1060 
1061 #include "locking-selftest-rlock-softirq.h"
1062 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_soft_rlock)
1063 
1064 #include "locking-selftest-wlock-softirq.h"
1065 GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_soft_wlock)
1066 #endif
1067 
1068 #undef E1
1069 #undef E2
1070 #undef E3
1071 
1072 /*
1073  * write-read / write-read / write-read deadlock even if read is recursive
1074  */
1075 
1076 #define E1()				\
1077 					\
1078 	WL(X1);				\
1079 	RL(Y1);				\
1080 	RU(Y1);				\
1081 	WU(X1);
1082 
1083 #define E2()				\
1084 					\
1085 	WL(Y1);				\
1086 	RL(Z1);				\
1087 	RU(Z1);				\
1088 	WU(Y1);
1089 
1090 #define E3()				\
1091 					\
1092 	WL(Z1);				\
1093 	RL(X1);				\
1094 	RU(X1);				\
1095 	WU(Z1);
1096 
1097 #include "locking-selftest-rlock.h"
1098 GENERATE_PERMUTATIONS_3_EVENTS(W1R2_W2R3_W3R1)
1099 
1100 #undef E1
1101 #undef E2
1102 #undef E3
1103 
1104 /*
1105  * write-write / read-read / write-read deadlock even if read is recursive
1106  */
1107 
1108 #define E1()				\
1109 					\
1110 	WL(X1);				\
1111 	WL(Y1);				\
1112 	WU(Y1);				\
1113 	WU(X1);
1114 
1115 #define E2()				\
1116 					\
1117 	RL(Y1);				\
1118 	RL(Z1);				\
1119 	RU(Z1);				\
1120 	RU(Y1);
1121 
1122 #define E3()				\
1123 					\
1124 	WL(Z1);				\
1125 	RL(X1);				\
1126 	RU(X1);				\
1127 	WU(Z1);
1128 
1129 #include "locking-selftest-rlock.h"
1130 GENERATE_PERMUTATIONS_3_EVENTS(W1W2_R2R3_W3R1)
1131 
1132 #undef E1
1133 #undef E2
1134 #undef E3
1135 
1136 /*
1137  * write-write / read-read / read-write is not deadlock when read is recursive
1138  */
1139 
1140 #define E1()				\
1141 					\
1142 	WL(X1);				\
1143 	WL(Y1);				\
1144 	WU(Y1);				\
1145 	WU(X1);
1146 
1147 #define E2()				\
1148 					\
1149 	RL(Y1);				\
1150 	RL(Z1);				\
1151 	RU(Z1);				\
1152 	RU(Y1);
1153 
1154 #define E3()				\
1155 					\
1156 	RL(Z1);				\
1157 	WL(X1);				\
1158 	WU(X1);				\
1159 	RU(Z1);
1160 
1161 #include "locking-selftest-rlock.h"
1162 GENERATE_PERMUTATIONS_3_EVENTS(W1R2_R2R3_W3W1)
1163 
1164 #undef E1
1165 #undef E2
1166 #undef E3
1167 
1168 /*
1169  * write-read / read-read / write-write is not deadlock when read is recursive
1170  */
1171 
1172 #define E1()				\
1173 					\
1174 	WL(X1);				\
1175 	RL(Y1);				\
1176 	RU(Y1);				\
1177 	WU(X1);
1178 
1179 #define E2()				\
1180 					\
1181 	RL(Y1);				\
1182 	RL(Z1);				\
1183 	RU(Z1);				\
1184 	RU(Y1);
1185 
1186 #define E3()				\
1187 					\
1188 	WL(Z1);				\
1189 	WL(X1);				\
1190 	WU(X1);				\
1191 	WU(Z1);
1192 
1193 #include "locking-selftest-rlock.h"
1194 GENERATE_PERMUTATIONS_3_EVENTS(W1W2_R2R3_R3W1)
1195 
1196 #undef E1
1197 #undef E2
1198 #undef E3
1199 /*
1200  * read-lock / write-lock recursion that is actually safe.
1201  */
1202 
1203 #define E1()				\
1204 					\
1205 	IRQ_DISABLE();			\
1206 	WL(A);				\
1207 	WU(A);				\
1208 	IRQ_ENABLE();
1209 
1210 #define E2()				\
1211 					\
1212 	RL(A);				\
1213 	RU(A);				\
1214 
1215 #define E3()				\
1216 					\
1217 	IRQ_ENTER();			\
1218 	LOCK(A);			\
1219 	L(B);				\
1220 	U(B);				\
1221 	UNLOCK(A);			\
1222 	IRQ_EXIT();
1223 
1224 /*
1225  * Generate 24 testcases:
1226  */
1227 #include "locking-selftest-hardirq.h"
1228 #include "locking-selftest-rlock.h"
1229 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion_hard_rlock)
1230 
1231 #include "locking-selftest-wlock.h"
1232 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion_hard_wlock)
1233 
1234 #ifndef CONFIG_PREEMPT_RT
1235 #include "locking-selftest-softirq.h"
1236 #include "locking-selftest-rlock.h"
1237 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion_soft_rlock)
1238 
1239 #include "locking-selftest-wlock.h"
1240 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion_soft_wlock)
1241 #endif
1242 
1243 #undef E1
1244 #undef E2
1245 #undef E3
1246 
1247 /*
1248  * read-lock / write-lock recursion that is unsafe.
1249  */
1250 
1251 #define E1()				\
1252 					\
1253 	IRQ_DISABLE();			\
1254 	L(B);				\
1255 	LOCK(A);			\
1256 	UNLOCK(A);			\
1257 	U(B);				\
1258 	IRQ_ENABLE();
1259 
1260 #define E2()				\
1261 					\
1262 	RL(A);				\
1263 	RU(A);				\
1264 
1265 #define E3()				\
1266 					\
1267 	IRQ_ENTER();			\
1268 	L(B);				\
1269 	U(B);				\
1270 	IRQ_EXIT();
1271 
1272 /*
1273  * Generate 24 testcases:
1274  */
1275 #include "locking-selftest-hardirq.h"
1276 #include "locking-selftest-rlock.h"
1277 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion2_hard_rlock)
1278 
1279 #include "locking-selftest-wlock.h"
1280 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion2_hard_wlock)
1281 
1282 #ifndef CONFIG_PREEMPT_RT
1283 #include "locking-selftest-softirq.h"
1284 #include "locking-selftest-rlock.h"
1285 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion2_soft_rlock)
1286 
1287 #include "locking-selftest-wlock.h"
1288 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion2_soft_wlock)
1289 #endif
1290 
1291 #undef E1
1292 #undef E2
1293 #undef E3
1294 /*
1295  * read-lock / write-lock recursion that is unsafe.
1296  *
1297  * A is a ENABLED_*_READ lock
1298  * B is a USED_IN_*_READ lock
1299  *
1300  * read_lock(A);
1301  *			write_lock(B);
1302  * <interrupt>
1303  * read_lock(B);
1304  * 			write_lock(A); // if this one is read_lock(), no deadlock
1305  */
1306 
1307 #define E1()				\
1308 					\
1309 	IRQ_DISABLE();			\
1310 	WL(B);				\
1311 	LOCK(A);			\
1312 	UNLOCK(A);			\
1313 	WU(B);				\
1314 	IRQ_ENABLE();
1315 
1316 #define E2()				\
1317 					\
1318 	RL(A);				\
1319 	RU(A);				\
1320 
1321 #define E3()				\
1322 					\
1323 	IRQ_ENTER();			\
1324 	RL(B);				\
1325 	RU(B);				\
1326 	IRQ_EXIT();
1327 
1328 /*
1329  * Generate 24 testcases:
1330  */
1331 #include "locking-selftest-hardirq.h"
1332 #include "locking-selftest-rlock.h"
1333 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion3_hard_rlock)
1334 
1335 #include "locking-selftest-wlock.h"
1336 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion3_hard_wlock)
1337 
1338 #ifndef CONFIG_PREEMPT_RT
1339 #include "locking-selftest-softirq.h"
1340 #include "locking-selftest-rlock.h"
1341 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion3_soft_rlock)
1342 
1343 #include "locking-selftest-wlock.h"
1344 GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion3_soft_wlock)
1345 #endif
1346 
1347 #ifdef CONFIG_DEBUG_LOCK_ALLOC
1348 # define I_SPINLOCK(x)	lockdep_reset_lock(&lock_##x.dep_map)
1349 # define I_RAW_SPINLOCK(x)	lockdep_reset_lock(&raw_lock_##x.dep_map)
1350 # define I_RWLOCK(x)	lockdep_reset_lock(&rwlock_##x.dep_map)
1351 # define I_MUTEX(x)	lockdep_reset_lock(&mutex_##x.dep_map)
1352 # define I_RWSEM(x)	lockdep_reset_lock(&rwsem_##x.dep_map)
1353 # define I_WW(x)	lockdep_reset_lock(&x.dep_map)
1354 # define I_LOCAL_LOCK(x) lockdep_reset_lock(this_cpu_ptr(&local_##x.dep_map))
1355 #ifdef CONFIG_RT_MUTEXES
1356 # define I_RTMUTEX(x)	lockdep_reset_lock(&rtmutex_##x.dep_map)
1357 #endif
1358 #else
1359 # define I_SPINLOCK(x)
1360 # define I_RAW_SPINLOCK(x)
1361 # define I_RWLOCK(x)
1362 # define I_MUTEX(x)
1363 # define I_RWSEM(x)
1364 # define I_WW(x)
1365 # define I_LOCAL_LOCK(x)
1366 #endif
1367 
1368 #ifndef I_RTMUTEX
1369 # define I_RTMUTEX(x)
1370 #endif
1371 
1372 #ifdef CONFIG_RT_MUTEXES
1373 #define I2_RTMUTEX(x)	rt_mutex_init(&rtmutex_##x)
1374 #else
1375 #define I2_RTMUTEX(x)
1376 #endif
1377 
1378 #define I1(x)					\
1379 	do {					\
1380 		I_SPINLOCK(x);			\
1381 		I_RWLOCK(x);			\
1382 		I_MUTEX(x);			\
1383 		I_RWSEM(x);			\
1384 		I_RTMUTEX(x);			\
1385 	} while (0)
1386 
1387 #define I2(x)					\
1388 	do {					\
1389 		spin_lock_init(&lock_##x);	\
1390 		rwlock_init(&rwlock_##x);	\
1391 		mutex_init(&mutex_##x);		\
1392 		init_rwsem(&rwsem_##x);		\
1393 		I2_RTMUTEX(x);			\
1394 	} while (0)
1395 
1396 static void reset_locks(void)
1397 {
1398 	local_irq_disable();
1399 	lockdep_free_key_range(&ww_lockdep.acquire_key, 1);
1400 	lockdep_free_key_range(&ww_lockdep.mutex_key, 1);
1401 
1402 	I1(A); I1(B); I1(C); I1(D);
1403 	I1(X1); I1(X2); I1(Y1); I1(Y2); I1(Z1); I1(Z2);
1404 	I_WW(t); I_WW(t2); I_WW(o.base); I_WW(o2.base); I_WW(o3.base);
1405 	I_RAW_SPINLOCK(A); I_RAW_SPINLOCK(B);
1406 	I_LOCAL_LOCK(A);
1407 
1408 	lockdep_reset();
1409 
1410 	I2(A); I2(B); I2(C); I2(D);
1411 	init_shared_classes();
1412 	raw_spin_lock_init(&raw_lock_A);
1413 	raw_spin_lock_init(&raw_lock_B);
1414 	local_lock_init(this_cpu_ptr(&local_A));
1415 
1416 	ww_mutex_init(&o, &ww_lockdep); ww_mutex_init(&o2, &ww_lockdep); ww_mutex_init(&o3, &ww_lockdep);
1417 	memset(&t, 0, sizeof(t)); memset(&t2, 0, sizeof(t2));
1418 	memset(&ww_lockdep.acquire_key, 0, sizeof(ww_lockdep.acquire_key));
1419 	memset(&ww_lockdep.mutex_key, 0, sizeof(ww_lockdep.mutex_key));
1420 	local_irq_enable();
1421 }
1422 
1423 #undef I
1424 
1425 static int testcase_total;
1426 static int testcase_successes;
1427 static int expected_testcase_failures;
1428 static int unexpected_testcase_failures;
1429 
1430 static void dotest(void (*testcase_fn)(void), int expected, int lockclass_mask)
1431 {
1432 	int saved_preempt_count = preempt_count();
1433 #ifdef CONFIG_PREEMPT_RT
1434 #ifdef CONFIG_SMP
1435 	int saved_mgd_count = current->migration_disabled;
1436 #endif
1437 	int saved_rcu_count = current->rcu_read_lock_nesting;
1438 #endif
1439 
1440 	WARN_ON(irqs_disabled());
1441 
1442 	debug_locks_silent = !(debug_locks_verbose & lockclass_mask);
1443 
1444 	testcase_fn();
1445 	/*
1446 	 * Filter out expected failures:
1447 	 */
1448 #ifndef CONFIG_PROVE_LOCKING
1449 	if (expected == FAILURE && debug_locks) {
1450 		expected_testcase_failures++;
1451 		pr_cont("failed|");
1452 	}
1453 	else
1454 #endif
1455 	if (debug_locks != expected) {
1456 		unexpected_testcase_failures++;
1457 		pr_cont("FAILED|");
1458 	} else {
1459 		testcase_successes++;
1460 		pr_cont("  ok  |");
1461 	}
1462 	testcase_total++;
1463 
1464 	if (debug_locks_verbose & lockclass_mask)
1465 		pr_cont(" lockclass mask: %x, debug_locks: %d, expected: %d\n",
1466 			lockclass_mask, debug_locks, expected);
1467 	/*
1468 	 * Some tests (e.g. double-unlock) might corrupt the preemption
1469 	 * count, so restore it:
1470 	 */
1471 	preempt_count_set(saved_preempt_count);
1472 
1473 #ifdef CONFIG_PREEMPT_RT
1474 #ifdef CONFIG_SMP
1475 	while (current->migration_disabled > saved_mgd_count)
1476 		migrate_enable();
1477 #endif
1478 
1479 	while (current->rcu_read_lock_nesting > saved_rcu_count)
1480 		rcu_read_unlock();
1481 	WARN_ON_ONCE(current->rcu_read_lock_nesting < saved_rcu_count);
1482 #endif
1483 
1484 #ifdef CONFIG_TRACE_IRQFLAGS
1485 	if (softirq_count())
1486 		current->softirqs_enabled = 0;
1487 	else
1488 		current->softirqs_enabled = 1;
1489 #endif
1490 
1491 	reset_locks();
1492 }
1493 
1494 #ifdef CONFIG_RT_MUTEXES
1495 #define dotest_rt(fn, e, m)	dotest((fn), (e), (m))
1496 #else
1497 #define dotest_rt(fn, e, m)
1498 #endif
1499 
1500 static inline void print_testname(const char *testname)
1501 {
1502 	printk("%33s:", testname);
1503 }
1504 
1505 #define DO_TESTCASE_1(desc, name, nr)				\
1506 	print_testname(desc"/"#nr);				\
1507 	dotest(name##_##nr, SUCCESS, LOCKTYPE_RWLOCK);		\
1508 	pr_cont("\n");
1509 
1510 #define DO_TESTCASE_1B(desc, name, nr)				\
1511 	print_testname(desc"/"#nr);				\
1512 	dotest(name##_##nr, FAILURE, LOCKTYPE_RWLOCK);		\
1513 	pr_cont("\n");
1514 
1515 #define DO_TESTCASE_1RR(desc, name, nr)				\
1516 	print_testname(desc"/"#nr);				\
1517 	pr_cont("             |");				\
1518 	dotest(name##_##nr, SUCCESS, LOCKTYPE_RWLOCK);		\
1519 	pr_cont("\n");
1520 
1521 #define DO_TESTCASE_1RRB(desc, name, nr)			\
1522 	print_testname(desc"/"#nr);				\
1523 	pr_cont("             |");				\
1524 	dotest(name##_##nr, FAILURE, LOCKTYPE_RWLOCK);		\
1525 	pr_cont("\n");
1526 
1527 
1528 #define DO_TESTCASE_3(desc, name, nr)				\
1529 	print_testname(desc"/"#nr);				\
1530 	dotest(name##_spin_##nr, FAILURE, LOCKTYPE_SPIN);	\
1531 	dotest(name##_wlock_##nr, FAILURE, LOCKTYPE_RWLOCK);	\
1532 	dotest(name##_rlock_##nr, SUCCESS, LOCKTYPE_RWLOCK);	\
1533 	pr_cont("\n");
1534 
1535 #define DO_TESTCASE_3RW(desc, name, nr)				\
1536 	print_testname(desc"/"#nr);				\
1537 	dotest(name##_spin_##nr, FAILURE, LOCKTYPE_SPIN|LOCKTYPE_RWLOCK);\
1538 	dotest(name##_wlock_##nr, FAILURE, LOCKTYPE_RWLOCK);	\
1539 	dotest(name##_rlock_##nr, SUCCESS, LOCKTYPE_RWLOCK);	\
1540 	pr_cont("\n");
1541 
1542 #define DO_TESTCASE_2RW(desc, name, nr)				\
1543 	print_testname(desc"/"#nr);				\
1544 	pr_cont("      |");					\
1545 	dotest(name##_wlock_##nr, FAILURE, LOCKTYPE_RWLOCK);	\
1546 	dotest(name##_rlock_##nr, SUCCESS, LOCKTYPE_RWLOCK);	\
1547 	pr_cont("\n");
1548 
1549 #define DO_TESTCASE_2x2RW(desc, name, nr)			\
1550 	DO_TESTCASE_2RW("hard-"desc, name##_hard, nr)		\
1551 	NON_RT(DO_TESTCASE_2RW("soft-"desc, name##_soft, nr))	\
1552 
1553 #define DO_TESTCASE_6x2x2RW(desc, name)				\
1554 	DO_TESTCASE_2x2RW(desc, name, 123);			\
1555 	DO_TESTCASE_2x2RW(desc, name, 132);			\
1556 	DO_TESTCASE_2x2RW(desc, name, 213);			\
1557 	DO_TESTCASE_2x2RW(desc, name, 231);			\
1558 	DO_TESTCASE_2x2RW(desc, name, 312);			\
1559 	DO_TESTCASE_2x2RW(desc, name, 321);
1560 
1561 #define DO_TESTCASE_6(desc, name)				\
1562 	print_testname(desc);					\
1563 	dotest(name##_spin, FAILURE, LOCKTYPE_SPIN);		\
1564 	dotest(name##_wlock, FAILURE, LOCKTYPE_RWLOCK);		\
1565 	dotest(name##_rlock, FAILURE, LOCKTYPE_RWLOCK);		\
1566 	dotest(name##_mutex, FAILURE, LOCKTYPE_MUTEX);		\
1567 	dotest(name##_wsem, FAILURE, LOCKTYPE_RWSEM);		\
1568 	dotest(name##_rsem, FAILURE, LOCKTYPE_RWSEM);		\
1569 	dotest_rt(name##_rtmutex, FAILURE, LOCKTYPE_RTMUTEX);	\
1570 	pr_cont("\n");
1571 
1572 #define DO_TESTCASE_6_SUCCESS(desc, name)			\
1573 	print_testname(desc);					\
1574 	dotest(name##_spin, SUCCESS, LOCKTYPE_SPIN);		\
1575 	dotest(name##_wlock, SUCCESS, LOCKTYPE_RWLOCK);		\
1576 	dotest(name##_rlock, SUCCESS, LOCKTYPE_RWLOCK);		\
1577 	dotest(name##_mutex, SUCCESS, LOCKTYPE_MUTEX);		\
1578 	dotest(name##_wsem, SUCCESS, LOCKTYPE_RWSEM);		\
1579 	dotest(name##_rsem, SUCCESS, LOCKTYPE_RWSEM);		\
1580 	dotest_rt(name##_rtmutex, SUCCESS, LOCKTYPE_RTMUTEX);	\
1581 	pr_cont("\n");
1582 
1583 /*
1584  * 'read' variant: rlocks must not trigger.
1585  */
1586 #define DO_TESTCASE_6R(desc, name)				\
1587 	print_testname(desc);					\
1588 	dotest(name##_spin, FAILURE, LOCKTYPE_SPIN);		\
1589 	dotest(name##_wlock, FAILURE, LOCKTYPE_RWLOCK);		\
1590 	dotest(name##_rlock, SUCCESS, LOCKTYPE_RWLOCK);		\
1591 	dotest(name##_mutex, FAILURE, LOCKTYPE_MUTEX);		\
1592 	dotest(name##_wsem, FAILURE, LOCKTYPE_RWSEM);		\
1593 	dotest(name##_rsem, FAILURE, LOCKTYPE_RWSEM);		\
1594 	dotest_rt(name##_rtmutex, FAILURE, LOCKTYPE_RTMUTEX);	\
1595 	pr_cont("\n");
1596 
1597 #define DO_TESTCASE_2I(desc, name, nr)				\
1598 	DO_TESTCASE_1("hard-"desc, name##_hard, nr);		\
1599 	NON_RT(DO_TESTCASE_1("soft-"desc, name##_soft, nr));
1600 
1601 #define DO_TESTCASE_2IB(desc, name, nr)				\
1602 	DO_TESTCASE_1B("hard-"desc, name##_hard, nr);		\
1603 	NON_RT(DO_TESTCASE_1B("soft-"desc, name##_soft, nr));
1604 
1605 #define DO_TESTCASE_6I(desc, name, nr)				\
1606 	DO_TESTCASE_3("hard-"desc, name##_hard, nr);		\
1607 	NON_RT(DO_TESTCASE_3("soft-"desc, name##_soft, nr));
1608 
1609 #define DO_TESTCASE_6IRW(desc, name, nr)			\
1610 	DO_TESTCASE_3RW("hard-"desc, name##_hard, nr);		\
1611 	NON_RT(DO_TESTCASE_3RW("soft-"desc, name##_soft, nr));
1612 
1613 #define DO_TESTCASE_2x3(desc, name)				\
1614 	DO_TESTCASE_3(desc, name, 12);				\
1615 	DO_TESTCASE_3(desc, name, 21);
1616 
1617 #define DO_TESTCASE_2x6(desc, name)				\
1618 	DO_TESTCASE_6I(desc, name, 12);				\
1619 	DO_TESTCASE_6I(desc, name, 21);
1620 
1621 #define DO_TESTCASE_6x2(desc, name)				\
1622 	DO_TESTCASE_2I(desc, name, 123);			\
1623 	DO_TESTCASE_2I(desc, name, 132);			\
1624 	DO_TESTCASE_2I(desc, name, 213);			\
1625 	DO_TESTCASE_2I(desc, name, 231);			\
1626 	DO_TESTCASE_2I(desc, name, 312);			\
1627 	DO_TESTCASE_2I(desc, name, 321);
1628 
1629 #define DO_TESTCASE_6x2B(desc, name)				\
1630 	DO_TESTCASE_2IB(desc, name, 123);			\
1631 	DO_TESTCASE_2IB(desc, name, 132);			\
1632 	DO_TESTCASE_2IB(desc, name, 213);			\
1633 	DO_TESTCASE_2IB(desc, name, 231);			\
1634 	DO_TESTCASE_2IB(desc, name, 312);			\
1635 	DO_TESTCASE_2IB(desc, name, 321);
1636 
1637 #define DO_TESTCASE_6x1RR(desc, name)				\
1638 	DO_TESTCASE_1RR(desc, name, 123);			\
1639 	DO_TESTCASE_1RR(desc, name, 132);			\
1640 	DO_TESTCASE_1RR(desc, name, 213);			\
1641 	DO_TESTCASE_1RR(desc, name, 231);			\
1642 	DO_TESTCASE_1RR(desc, name, 312);			\
1643 	DO_TESTCASE_1RR(desc, name, 321);
1644 
1645 #define DO_TESTCASE_6x1RRB(desc, name)				\
1646 	DO_TESTCASE_1RRB(desc, name, 123);			\
1647 	DO_TESTCASE_1RRB(desc, name, 132);			\
1648 	DO_TESTCASE_1RRB(desc, name, 213);			\
1649 	DO_TESTCASE_1RRB(desc, name, 231);			\
1650 	DO_TESTCASE_1RRB(desc, name, 312);			\
1651 	DO_TESTCASE_1RRB(desc, name, 321);
1652 
1653 #define DO_TESTCASE_6x6(desc, name)				\
1654 	DO_TESTCASE_6I(desc, name, 123);			\
1655 	DO_TESTCASE_6I(desc, name, 132);			\
1656 	DO_TESTCASE_6I(desc, name, 213);			\
1657 	DO_TESTCASE_6I(desc, name, 231);			\
1658 	DO_TESTCASE_6I(desc, name, 312);			\
1659 	DO_TESTCASE_6I(desc, name, 321);
1660 
1661 #define DO_TESTCASE_6x6RW(desc, name)				\
1662 	DO_TESTCASE_6IRW(desc, name, 123);			\
1663 	DO_TESTCASE_6IRW(desc, name, 132);			\
1664 	DO_TESTCASE_6IRW(desc, name, 213);			\
1665 	DO_TESTCASE_6IRW(desc, name, 231);			\
1666 	DO_TESTCASE_6IRW(desc, name, 312);			\
1667 	DO_TESTCASE_6IRW(desc, name, 321);
1668 
1669 static void ww_test_fail_acquire(void)
1670 {
1671 	int ret;
1672 
1673 	WWAI(&t);
1674 	t.stamp++;
1675 
1676 	ret = WWL(&o, &t);
1677 
1678 	if (WARN_ON(!o.ctx) ||
1679 	    WARN_ON(ret))
1680 		return;
1681 
1682 	/* No lockdep test, pure API */
1683 	ret = WWL(&o, &t);
1684 	WARN_ON(ret != -EALREADY);
1685 
1686 	ret = WWT(&o);
1687 	WARN_ON(ret);
1688 
1689 	t2 = t;
1690 	t2.stamp++;
1691 	ret = WWL(&o, &t2);
1692 	WARN_ON(ret != -EDEADLK);
1693 	WWU(&o);
1694 
1695 	if (WWT(&o))
1696 		WWU(&o);
1697 #ifdef CONFIG_DEBUG_LOCK_ALLOC
1698 	else
1699 		DEBUG_LOCKS_WARN_ON(1);
1700 #endif
1701 }
1702 
1703 #ifdef CONFIG_PREEMPT_RT
1704 #define ww_mutex_base_lock(b)			rt_mutex_lock(b)
1705 #define ww_mutex_base_trylock(b)		rt_mutex_trylock(b)
1706 #define ww_mutex_base_lock_nest_lock(b, b2)	rt_mutex_lock_nest_lock(b, b2)
1707 #define ww_mutex_base_lock_interruptible(b)	rt_mutex_lock_interruptible(b)
1708 #define ww_mutex_base_lock_killable(b)		rt_mutex_lock_killable(b)
1709 #define ww_mutex_base_unlock(b)			rt_mutex_unlock(b)
1710 #else
1711 #define ww_mutex_base_lock(b)			mutex_lock(b)
1712 #define ww_mutex_base_trylock(b)		mutex_trylock(b)
1713 #define ww_mutex_base_lock_nest_lock(b, b2)	mutex_lock_nest_lock(b, b2)
1714 #define ww_mutex_base_lock_interruptible(b)	mutex_lock_interruptible(b)
1715 #define ww_mutex_base_lock_killable(b)		mutex_lock_killable(b)
1716 #define ww_mutex_base_unlock(b)			mutex_unlock(b)
1717 #endif
1718 
1719 static void ww_test_normal(void)
1720 {
1721 	int ret;
1722 
1723 	WWAI(&t);
1724 
1725 	/*
1726 	 * None of the ww_mutex codepaths should be taken in the 'normal'
1727 	 * mutex calls. The easiest way to verify this is by using the
1728 	 * normal mutex calls, and making sure o.ctx is unmodified.
1729 	 */
1730 
1731 	/* mutex_lock (and indirectly, mutex_lock_nested) */
1732 	o.ctx = (void *)~0UL;
1733 	ww_mutex_base_lock(&o.base);
1734 	ww_mutex_base_unlock(&o.base);
1735 	WARN_ON(o.ctx != (void *)~0UL);
1736 
1737 	/* mutex_lock_interruptible (and *_nested) */
1738 	o.ctx = (void *)~0UL;
1739 	ret = ww_mutex_base_lock_interruptible(&o.base);
1740 	if (!ret)
1741 		ww_mutex_base_unlock(&o.base);
1742 	else
1743 		WARN_ON(1);
1744 	WARN_ON(o.ctx != (void *)~0UL);
1745 
1746 	/* mutex_lock_killable (and *_nested) */
1747 	o.ctx = (void *)~0UL;
1748 	ret = ww_mutex_base_lock_killable(&o.base);
1749 	if (!ret)
1750 		ww_mutex_base_unlock(&o.base);
1751 	else
1752 		WARN_ON(1);
1753 	WARN_ON(o.ctx != (void *)~0UL);
1754 
1755 	/* trylock, succeeding */
1756 	o.ctx = (void *)~0UL;
1757 	ret = ww_mutex_base_trylock(&o.base);
1758 	WARN_ON(!ret);
1759 	if (ret)
1760 		ww_mutex_base_unlock(&o.base);
1761 	else
1762 		WARN_ON(1);
1763 	WARN_ON(o.ctx != (void *)~0UL);
1764 
1765 	/* trylock, failing */
1766 	o.ctx = (void *)~0UL;
1767 	ww_mutex_base_lock(&o.base);
1768 	ret = ww_mutex_base_trylock(&o.base);
1769 	WARN_ON(ret);
1770 	ww_mutex_base_unlock(&o.base);
1771 	WARN_ON(o.ctx != (void *)~0UL);
1772 
1773 	/* nest_lock */
1774 	o.ctx = (void *)~0UL;
1775 	ww_mutex_base_lock_nest_lock(&o.base, &t);
1776 	ww_mutex_base_unlock(&o.base);
1777 	WARN_ON(o.ctx != (void *)~0UL);
1778 }
1779 
1780 static void ww_test_two_contexts(void)
1781 {
1782 	WWAI(&t);
1783 	WWAI(&t2);
1784 }
1785 
1786 static void ww_test_diff_class(void)
1787 {
1788 	WWAI(&t);
1789 #ifdef DEBUG_WW_MUTEXES
1790 	t.ww_class = NULL;
1791 #endif
1792 	WWL(&o, &t);
1793 }
1794 
1795 static void ww_test_context_done_twice(void)
1796 {
1797 	WWAI(&t);
1798 	WWAD(&t);
1799 	WWAD(&t);
1800 	WWAF(&t);
1801 }
1802 
1803 static void ww_test_context_unlock_twice(void)
1804 {
1805 	WWAI(&t);
1806 	WWAD(&t);
1807 	WWAF(&t);
1808 	WWAF(&t);
1809 }
1810 
1811 static void ww_test_context_fini_early(void)
1812 {
1813 	WWAI(&t);
1814 	WWL(&o, &t);
1815 	WWAD(&t);
1816 	WWAF(&t);
1817 }
1818 
1819 static void ww_test_context_lock_after_done(void)
1820 {
1821 	WWAI(&t);
1822 	WWAD(&t);
1823 	WWL(&o, &t);
1824 }
1825 
1826 static void ww_test_object_unlock_twice(void)
1827 {
1828 	WWL1(&o);
1829 	WWU(&o);
1830 	WWU(&o);
1831 }
1832 
1833 static void ww_test_object_lock_unbalanced(void)
1834 {
1835 	WWAI(&t);
1836 	WWL(&o, &t);
1837 	t.acquired = 0;
1838 	WWU(&o);
1839 	WWAF(&t);
1840 }
1841 
1842 static void ww_test_object_lock_stale_context(void)
1843 {
1844 	WWAI(&t);
1845 	o.ctx = &t2;
1846 	WWL(&o, &t);
1847 }
1848 
1849 static void ww_test_edeadlk_normal(void)
1850 {
1851 	int ret;
1852 
1853 	ww_mutex_base_lock(&o2.base);
1854 	o2.ctx = &t2;
1855 	mutex_release(&o2.base.dep_map, _THIS_IP_);
1856 
1857 	WWAI(&t);
1858 	t2 = t;
1859 	t2.stamp--;
1860 
1861 	ret = WWL(&o, &t);
1862 	WARN_ON(ret);
1863 
1864 	ret = WWL(&o2, &t);
1865 	WARN_ON(ret != -EDEADLK);
1866 
1867 	o2.ctx = NULL;
1868 	mutex_acquire(&o2.base.dep_map, 0, 1, _THIS_IP_);
1869 	ww_mutex_base_unlock(&o2.base);
1870 	WWU(&o);
1871 
1872 	WWL(&o2, &t);
1873 }
1874 
1875 static void ww_test_edeadlk_normal_slow(void)
1876 {
1877 	int ret;
1878 
1879 	ww_mutex_base_lock(&o2.base);
1880 	mutex_release(&o2.base.dep_map, _THIS_IP_);
1881 	o2.ctx = &t2;
1882 
1883 	WWAI(&t);
1884 	t2 = t;
1885 	t2.stamp--;
1886 
1887 	ret = WWL(&o, &t);
1888 	WARN_ON(ret);
1889 
1890 	ret = WWL(&o2, &t);
1891 	WARN_ON(ret != -EDEADLK);
1892 
1893 	o2.ctx = NULL;
1894 	mutex_acquire(&o2.base.dep_map, 0, 1, _THIS_IP_);
1895 	ww_mutex_base_unlock(&o2.base);
1896 	WWU(&o);
1897 
1898 	ww_mutex_lock_slow(&o2, &t);
1899 }
1900 
1901 static void ww_test_edeadlk_no_unlock(void)
1902 {
1903 	int ret;
1904 
1905 	ww_mutex_base_lock(&o2.base);
1906 	o2.ctx = &t2;
1907 	mutex_release(&o2.base.dep_map, _THIS_IP_);
1908 
1909 	WWAI(&t);
1910 	t2 = t;
1911 	t2.stamp--;
1912 
1913 	ret = WWL(&o, &t);
1914 	WARN_ON(ret);
1915 
1916 	ret = WWL(&o2, &t);
1917 	WARN_ON(ret != -EDEADLK);
1918 
1919 	o2.ctx = NULL;
1920 	mutex_acquire(&o2.base.dep_map, 0, 1, _THIS_IP_);
1921 	ww_mutex_base_unlock(&o2.base);
1922 
1923 	WWL(&o2, &t);
1924 }
1925 
1926 static void ww_test_edeadlk_no_unlock_slow(void)
1927 {
1928 	int ret;
1929 
1930 	ww_mutex_base_lock(&o2.base);
1931 	mutex_release(&o2.base.dep_map, _THIS_IP_);
1932 	o2.ctx = &t2;
1933 
1934 	WWAI(&t);
1935 	t2 = t;
1936 	t2.stamp--;
1937 
1938 	ret = WWL(&o, &t);
1939 	WARN_ON(ret);
1940 
1941 	ret = WWL(&o2, &t);
1942 	WARN_ON(ret != -EDEADLK);
1943 
1944 	o2.ctx = NULL;
1945 	mutex_acquire(&o2.base.dep_map, 0, 1, _THIS_IP_);
1946 	ww_mutex_base_unlock(&o2.base);
1947 
1948 	ww_mutex_lock_slow(&o2, &t);
1949 }
1950 
1951 static void ww_test_edeadlk_acquire_more(void)
1952 {
1953 	int ret;
1954 
1955 	ww_mutex_base_lock(&o2.base);
1956 	mutex_release(&o2.base.dep_map, _THIS_IP_);
1957 	o2.ctx = &t2;
1958 
1959 	WWAI(&t);
1960 	t2 = t;
1961 	t2.stamp--;
1962 
1963 	ret = WWL(&o, &t);
1964 	WARN_ON(ret);
1965 
1966 	ret = WWL(&o2, &t);
1967 	WARN_ON(ret != -EDEADLK);
1968 
1969 	ret = WWL(&o3, &t);
1970 }
1971 
1972 static void ww_test_edeadlk_acquire_more_slow(void)
1973 {
1974 	int ret;
1975 
1976 	ww_mutex_base_lock(&o2.base);
1977 	mutex_release(&o2.base.dep_map, _THIS_IP_);
1978 	o2.ctx = &t2;
1979 
1980 	WWAI(&t);
1981 	t2 = t;
1982 	t2.stamp--;
1983 
1984 	ret = WWL(&o, &t);
1985 	WARN_ON(ret);
1986 
1987 	ret = WWL(&o2, &t);
1988 	WARN_ON(ret != -EDEADLK);
1989 
1990 	ww_mutex_lock_slow(&o3, &t);
1991 }
1992 
1993 static void ww_test_edeadlk_acquire_more_edeadlk(void)
1994 {
1995 	int ret;
1996 
1997 	ww_mutex_base_lock(&o2.base);
1998 	mutex_release(&o2.base.dep_map, _THIS_IP_);
1999 	o2.ctx = &t2;
2000 
2001 	ww_mutex_base_lock(&o3.base);
2002 	mutex_release(&o3.base.dep_map, _THIS_IP_);
2003 	o3.ctx = &t2;
2004 
2005 	WWAI(&t);
2006 	t2 = t;
2007 	t2.stamp--;
2008 
2009 	ret = WWL(&o, &t);
2010 	WARN_ON(ret);
2011 
2012 	ret = WWL(&o2, &t);
2013 	WARN_ON(ret != -EDEADLK);
2014 
2015 	ret = WWL(&o3, &t);
2016 	WARN_ON(ret != -EDEADLK);
2017 }
2018 
2019 static void ww_test_edeadlk_acquire_more_edeadlk_slow(void)
2020 {
2021 	int ret;
2022 
2023 	ww_mutex_base_lock(&o2.base);
2024 	mutex_release(&o2.base.dep_map, _THIS_IP_);
2025 	o2.ctx = &t2;
2026 
2027 	ww_mutex_base_lock(&o3.base);
2028 	mutex_release(&o3.base.dep_map, _THIS_IP_);
2029 	o3.ctx = &t2;
2030 
2031 	WWAI(&t);
2032 	t2 = t;
2033 	t2.stamp--;
2034 
2035 	ret = WWL(&o, &t);
2036 	WARN_ON(ret);
2037 
2038 	ret = WWL(&o2, &t);
2039 	WARN_ON(ret != -EDEADLK);
2040 
2041 	ww_mutex_lock_slow(&o3, &t);
2042 }
2043 
2044 static void ww_test_edeadlk_acquire_wrong(void)
2045 {
2046 	int ret;
2047 
2048 	ww_mutex_base_lock(&o2.base);
2049 	mutex_release(&o2.base.dep_map, _THIS_IP_);
2050 	o2.ctx = &t2;
2051 
2052 	WWAI(&t);
2053 	t2 = t;
2054 	t2.stamp--;
2055 
2056 	ret = WWL(&o, &t);
2057 	WARN_ON(ret);
2058 
2059 	ret = WWL(&o2, &t);
2060 	WARN_ON(ret != -EDEADLK);
2061 	if (!ret)
2062 		WWU(&o2);
2063 
2064 	WWU(&o);
2065 
2066 	ret = WWL(&o3, &t);
2067 }
2068 
2069 static void ww_test_edeadlk_acquire_wrong_slow(void)
2070 {
2071 	int ret;
2072 
2073 	ww_mutex_base_lock(&o2.base);
2074 	mutex_release(&o2.base.dep_map, _THIS_IP_);
2075 	o2.ctx = &t2;
2076 
2077 	WWAI(&t);
2078 	t2 = t;
2079 	t2.stamp--;
2080 
2081 	ret = WWL(&o, &t);
2082 	WARN_ON(ret);
2083 
2084 	ret = WWL(&o2, &t);
2085 	WARN_ON(ret != -EDEADLK);
2086 	if (!ret)
2087 		WWU(&o2);
2088 
2089 	WWU(&o);
2090 
2091 	ww_mutex_lock_slow(&o3, &t);
2092 }
2093 
2094 static void ww_test_spin_nest_unlocked(void)
2095 {
2096 	spin_lock_nest_lock(&lock_A, &o.base);
2097 	U(A);
2098 }
2099 
2100 /* This is not a deadlock, because we have X1 to serialize Y1 and Y2 */
2101 static void ww_test_spin_nest_lock(void)
2102 {
2103 	spin_lock(&lock_X1);
2104 	spin_lock_nest_lock(&lock_Y1, &lock_X1);
2105 	spin_lock(&lock_A);
2106 	spin_lock_nest_lock(&lock_Y2, &lock_X1);
2107 	spin_unlock(&lock_A);
2108 	spin_unlock(&lock_Y2);
2109 	spin_unlock(&lock_Y1);
2110 	spin_unlock(&lock_X1);
2111 }
2112 
2113 static void ww_test_unneeded_slow(void)
2114 {
2115 	WWAI(&t);
2116 
2117 	ww_mutex_lock_slow(&o, &t);
2118 }
2119 
2120 static void ww_test_context_block(void)
2121 {
2122 	int ret;
2123 
2124 	WWAI(&t);
2125 
2126 	ret = WWL(&o, &t);
2127 	WARN_ON(ret);
2128 	WWL1(&o2);
2129 }
2130 
2131 static void ww_test_context_try(void)
2132 {
2133 	int ret;
2134 
2135 	WWAI(&t);
2136 
2137 	ret = WWL(&o, &t);
2138 	WARN_ON(ret);
2139 
2140 	ret = WWT(&o2);
2141 	WARN_ON(!ret);
2142 	WWU(&o2);
2143 	WWU(&o);
2144 }
2145 
2146 static void ww_test_context_context(void)
2147 {
2148 	int ret;
2149 
2150 	WWAI(&t);
2151 
2152 	ret = WWL(&o, &t);
2153 	WARN_ON(ret);
2154 
2155 	ret = WWL(&o2, &t);
2156 	WARN_ON(ret);
2157 
2158 	WWU(&o2);
2159 	WWU(&o);
2160 }
2161 
2162 static void ww_test_try_block(void)
2163 {
2164 	bool ret;
2165 
2166 	ret = WWT(&o);
2167 	WARN_ON(!ret);
2168 
2169 	WWL1(&o2);
2170 	WWU(&o2);
2171 	WWU(&o);
2172 }
2173 
2174 static void ww_test_try_try(void)
2175 {
2176 	bool ret;
2177 
2178 	ret = WWT(&o);
2179 	WARN_ON(!ret);
2180 	ret = WWT(&o2);
2181 	WARN_ON(!ret);
2182 	WWU(&o2);
2183 	WWU(&o);
2184 }
2185 
2186 static void ww_test_try_context(void)
2187 {
2188 	int ret;
2189 
2190 	ret = WWT(&o);
2191 	WARN_ON(!ret);
2192 
2193 	WWAI(&t);
2194 
2195 	ret = WWL(&o2, &t);
2196 	WARN_ON(ret);
2197 }
2198 
2199 static void ww_test_block_block(void)
2200 {
2201 	WWL1(&o);
2202 	WWL1(&o2);
2203 }
2204 
2205 static void ww_test_block_try(void)
2206 {
2207 	bool ret;
2208 
2209 	WWL1(&o);
2210 	ret = WWT(&o2);
2211 	WARN_ON(!ret);
2212 }
2213 
2214 static void ww_test_block_context(void)
2215 {
2216 	int ret;
2217 
2218 	WWL1(&o);
2219 	WWAI(&t);
2220 
2221 	ret = WWL(&o2, &t);
2222 	WARN_ON(ret);
2223 }
2224 
2225 static void ww_test_spin_block(void)
2226 {
2227 	L(A);
2228 	U(A);
2229 
2230 	WWL1(&o);
2231 	L(A);
2232 	U(A);
2233 	WWU(&o);
2234 
2235 	L(A);
2236 	WWL1(&o);
2237 	WWU(&o);
2238 	U(A);
2239 }
2240 
2241 static void ww_test_spin_try(void)
2242 {
2243 	bool ret;
2244 
2245 	L(A);
2246 	U(A);
2247 
2248 	ret = WWT(&o);
2249 	WARN_ON(!ret);
2250 	L(A);
2251 	U(A);
2252 	WWU(&o);
2253 
2254 	L(A);
2255 	ret = WWT(&o);
2256 	WARN_ON(!ret);
2257 	WWU(&o);
2258 	U(A);
2259 }
2260 
2261 static void ww_test_spin_context(void)
2262 {
2263 	int ret;
2264 
2265 	L(A);
2266 	U(A);
2267 
2268 	WWAI(&t);
2269 
2270 	ret = WWL(&o, &t);
2271 	WARN_ON(ret);
2272 	L(A);
2273 	U(A);
2274 	WWU(&o);
2275 
2276 	L(A);
2277 	ret = WWL(&o, &t);
2278 	WARN_ON(ret);
2279 	WWU(&o);
2280 	U(A);
2281 }
2282 
2283 static void ww_tests(void)
2284 {
2285 	printk("  --------------------------------------------------------------------------\n");
2286 	printk("  | Wound/wait tests |\n");
2287 	printk("  ---------------------\n");
2288 
2289 	print_testname("ww api failures");
2290 	dotest(ww_test_fail_acquire, SUCCESS, LOCKTYPE_WW);
2291 	dotest(ww_test_normal, SUCCESS, LOCKTYPE_WW);
2292 	dotest(ww_test_unneeded_slow, FAILURE, LOCKTYPE_WW);
2293 	pr_cont("\n");
2294 
2295 	print_testname("ww contexts mixing");
2296 	dotest(ww_test_two_contexts, FAILURE, LOCKTYPE_WW);
2297 	dotest(ww_test_diff_class, FAILURE, LOCKTYPE_WW);
2298 	pr_cont("\n");
2299 
2300 	print_testname("finishing ww context");
2301 	dotest(ww_test_context_done_twice, FAILURE, LOCKTYPE_WW);
2302 	dotest(ww_test_context_unlock_twice, FAILURE, LOCKTYPE_WW);
2303 	dotest(ww_test_context_fini_early, FAILURE, LOCKTYPE_WW);
2304 	dotest(ww_test_context_lock_after_done, FAILURE, LOCKTYPE_WW);
2305 	pr_cont("\n");
2306 
2307 	print_testname("locking mismatches");
2308 	dotest(ww_test_object_unlock_twice, FAILURE, LOCKTYPE_WW);
2309 	dotest(ww_test_object_lock_unbalanced, FAILURE, LOCKTYPE_WW);
2310 	dotest(ww_test_object_lock_stale_context, FAILURE, LOCKTYPE_WW);
2311 	pr_cont("\n");
2312 
2313 	print_testname("EDEADLK handling");
2314 	dotest(ww_test_edeadlk_normal, SUCCESS, LOCKTYPE_WW);
2315 	dotest(ww_test_edeadlk_normal_slow, SUCCESS, LOCKTYPE_WW);
2316 	dotest(ww_test_edeadlk_no_unlock, FAILURE, LOCKTYPE_WW);
2317 	dotest(ww_test_edeadlk_no_unlock_slow, FAILURE, LOCKTYPE_WW);
2318 	dotest(ww_test_edeadlk_acquire_more, FAILURE, LOCKTYPE_WW);
2319 	dotest(ww_test_edeadlk_acquire_more_slow, FAILURE, LOCKTYPE_WW);
2320 	dotest(ww_test_edeadlk_acquire_more_edeadlk, FAILURE, LOCKTYPE_WW);
2321 	dotest(ww_test_edeadlk_acquire_more_edeadlk_slow, FAILURE, LOCKTYPE_WW);
2322 	dotest(ww_test_edeadlk_acquire_wrong, FAILURE, LOCKTYPE_WW);
2323 	dotest(ww_test_edeadlk_acquire_wrong_slow, FAILURE, LOCKTYPE_WW);
2324 	pr_cont("\n");
2325 
2326 	print_testname("spinlock nest unlocked");
2327 	dotest(ww_test_spin_nest_unlocked, FAILURE, LOCKTYPE_WW);
2328 	pr_cont("\n");
2329 
2330 	print_testname("spinlock nest test");
2331 	dotest(ww_test_spin_nest_lock, SUCCESS, LOCKTYPE_WW);
2332 	pr_cont("\n");
2333 
2334 	printk("  -----------------------------------------------------\n");
2335 	printk("                                 |block | try  |context|\n");
2336 	printk("  -----------------------------------------------------\n");
2337 
2338 	print_testname("context");
2339 	dotest(ww_test_context_block, FAILURE, LOCKTYPE_WW);
2340 	dotest(ww_test_context_try, SUCCESS, LOCKTYPE_WW);
2341 	dotest(ww_test_context_context, SUCCESS, LOCKTYPE_WW);
2342 	pr_cont("\n");
2343 
2344 	print_testname("try");
2345 	dotest(ww_test_try_block, FAILURE, LOCKTYPE_WW);
2346 	dotest(ww_test_try_try, SUCCESS, LOCKTYPE_WW);
2347 	dotest(ww_test_try_context, FAILURE, LOCKTYPE_WW);
2348 	pr_cont("\n");
2349 
2350 	print_testname("block");
2351 	dotest(ww_test_block_block, FAILURE, LOCKTYPE_WW);
2352 	dotest(ww_test_block_try, SUCCESS, LOCKTYPE_WW);
2353 	dotest(ww_test_block_context, FAILURE, LOCKTYPE_WW);
2354 	pr_cont("\n");
2355 
2356 	print_testname("spinlock");
2357 	dotest(ww_test_spin_block, FAILURE, LOCKTYPE_WW);
2358 	dotest(ww_test_spin_try, SUCCESS, LOCKTYPE_WW);
2359 	dotest(ww_test_spin_context, FAILURE, LOCKTYPE_WW);
2360 	pr_cont("\n");
2361 }
2362 
2363 
2364 /*
2365  * <in hardirq handler>
2366  * read_lock(&A);
2367  *			<hardirq disable>
2368  *			spin_lock(&B);
2369  * spin_lock(&B);
2370  *			read_lock(&A);
2371  *
2372  * is a deadlock.
2373  */
2374 static void queued_read_lock_hardirq_RE_Er(void)
2375 {
2376 	HARDIRQ_ENTER();
2377 	read_lock(&rwlock_A);
2378 	LOCK(B);
2379 	UNLOCK(B);
2380 	read_unlock(&rwlock_A);
2381 	HARDIRQ_EXIT();
2382 
2383 	HARDIRQ_DISABLE();
2384 	LOCK(B);
2385 	read_lock(&rwlock_A);
2386 	read_unlock(&rwlock_A);
2387 	UNLOCK(B);
2388 	HARDIRQ_ENABLE();
2389 }
2390 
2391 /*
2392  * <in hardirq handler>
2393  * spin_lock(&B);
2394  *			<hardirq disable>
2395  *			read_lock(&A);
2396  * read_lock(&A);
2397  *			spin_lock(&B);
2398  *
2399  * is not a deadlock.
2400  */
2401 static void queued_read_lock_hardirq_ER_rE(void)
2402 {
2403 	HARDIRQ_ENTER();
2404 	LOCK(B);
2405 	read_lock(&rwlock_A);
2406 	read_unlock(&rwlock_A);
2407 	UNLOCK(B);
2408 	HARDIRQ_EXIT();
2409 
2410 	HARDIRQ_DISABLE();
2411 	read_lock(&rwlock_A);
2412 	LOCK(B);
2413 	UNLOCK(B);
2414 	read_unlock(&rwlock_A);
2415 	HARDIRQ_ENABLE();
2416 }
2417 
2418 /*
2419  * <hardirq disable>
2420  * spin_lock(&B);
2421  *			read_lock(&A);
2422  *			<in hardirq handler>
2423  *			spin_lock(&B);
2424  * read_lock(&A);
2425  *
2426  * is a deadlock. Because the two read_lock()s are both non-recursive readers.
2427  */
2428 static void queued_read_lock_hardirq_inversion(void)
2429 {
2430 
2431 	HARDIRQ_ENTER();
2432 	LOCK(B);
2433 	UNLOCK(B);
2434 	HARDIRQ_EXIT();
2435 
2436 	HARDIRQ_DISABLE();
2437 	LOCK(B);
2438 	read_lock(&rwlock_A);
2439 	read_unlock(&rwlock_A);
2440 	UNLOCK(B);
2441 	HARDIRQ_ENABLE();
2442 
2443 	read_lock(&rwlock_A);
2444 	read_unlock(&rwlock_A);
2445 }
2446 
2447 static void queued_read_lock_tests(void)
2448 {
2449 	printk("  --------------------------------------------------------------------------\n");
2450 	printk("  | queued read lock tests |\n");
2451 	printk("  ---------------------------\n");
2452 	print_testname("hardirq read-lock/lock-read");
2453 	dotest(queued_read_lock_hardirq_RE_Er, FAILURE, LOCKTYPE_RWLOCK);
2454 	pr_cont("\n");
2455 
2456 	print_testname("hardirq lock-read/read-lock");
2457 	dotest(queued_read_lock_hardirq_ER_rE, SUCCESS, LOCKTYPE_RWLOCK);
2458 	pr_cont("\n");
2459 
2460 	print_testname("hardirq inversion");
2461 	dotest(queued_read_lock_hardirq_inversion, FAILURE, LOCKTYPE_RWLOCK);
2462 	pr_cont("\n");
2463 }
2464 
2465 static void fs_reclaim_correct_nesting(void)
2466 {
2467 	fs_reclaim_acquire(GFP_KERNEL);
2468 	might_alloc(GFP_NOFS);
2469 	fs_reclaim_release(GFP_KERNEL);
2470 }
2471 
2472 static void fs_reclaim_wrong_nesting(void)
2473 {
2474 	fs_reclaim_acquire(GFP_KERNEL);
2475 	might_alloc(GFP_KERNEL);
2476 	fs_reclaim_release(GFP_KERNEL);
2477 }
2478 
2479 static void fs_reclaim_protected_nesting(void)
2480 {
2481 	unsigned int flags;
2482 
2483 	fs_reclaim_acquire(GFP_KERNEL);
2484 	flags = memalloc_nofs_save();
2485 	might_alloc(GFP_KERNEL);
2486 	memalloc_nofs_restore(flags);
2487 	fs_reclaim_release(GFP_KERNEL);
2488 }
2489 
2490 static void fs_reclaim_tests(void)
2491 {
2492 	printk("  --------------------\n");
2493 	printk("  | fs_reclaim tests |\n");
2494 	printk("  --------------------\n");
2495 
2496 	print_testname("correct nesting");
2497 	dotest(fs_reclaim_correct_nesting, SUCCESS, 0);
2498 	pr_cont("\n");
2499 
2500 	print_testname("wrong nesting");
2501 	dotest(fs_reclaim_wrong_nesting, FAILURE, 0);
2502 	pr_cont("\n");
2503 
2504 	print_testname("protected nesting");
2505 	dotest(fs_reclaim_protected_nesting, SUCCESS, 0);
2506 	pr_cont("\n");
2507 }
2508 
2509 #define __guard(cleanup) __maybe_unused __attribute__((__cleanup__(cleanup)))
2510 
2511 static void hardirq_exit(int *_)
2512 {
2513 	HARDIRQ_EXIT();
2514 }
2515 
2516 #define HARDIRQ_CONTEXT(name, ...)					\
2517 	int hardirq_guard_##name __guard(hardirq_exit);			\
2518 	HARDIRQ_ENTER();
2519 
2520 #define NOTTHREADED_HARDIRQ_CONTEXT(name, ...)				\
2521 	int notthreaded_hardirq_guard_##name __guard(hardirq_exit);	\
2522 	local_irq_disable();						\
2523 	__irq_enter();							\
2524 	WARN_ON(!in_irq());
2525 
2526 static void softirq_exit(int *_)
2527 {
2528 	SOFTIRQ_EXIT();
2529 }
2530 
2531 #define SOFTIRQ_CONTEXT(name, ...)				\
2532 	int softirq_guard_##name __guard(softirq_exit);		\
2533 	SOFTIRQ_ENTER();
2534 
2535 static void rcu_exit(int *_)
2536 {
2537 	rcu_read_unlock();
2538 }
2539 
2540 #define RCU_CONTEXT(name, ...)					\
2541 	int rcu_guard_##name __guard(rcu_exit);			\
2542 	rcu_read_lock();
2543 
2544 static void rcu_bh_exit(int *_)
2545 {
2546 	rcu_read_unlock_bh();
2547 }
2548 
2549 #define RCU_BH_CONTEXT(name, ...)				\
2550 	int rcu_bh_guard_##name __guard(rcu_bh_exit);		\
2551 	rcu_read_lock_bh();
2552 
2553 static void rcu_sched_exit(int *_)
2554 {
2555 	rcu_read_unlock_sched();
2556 }
2557 
2558 #define RCU_SCHED_CONTEXT(name, ...)				\
2559 	int rcu_sched_guard_##name __guard(rcu_sched_exit);	\
2560 	rcu_read_lock_sched();
2561 
2562 static void raw_spinlock_exit(raw_spinlock_t **lock)
2563 {
2564 	raw_spin_unlock(*lock);
2565 }
2566 
2567 #define RAW_SPINLOCK_CONTEXT(name, lock)						\
2568 	raw_spinlock_t *raw_spinlock_guard_##name __guard(raw_spinlock_exit) = &(lock);	\
2569 	raw_spin_lock(&(lock));
2570 
2571 static void spinlock_exit(spinlock_t **lock)
2572 {
2573 	spin_unlock(*lock);
2574 }
2575 
2576 #define SPINLOCK_CONTEXT(name, lock)						\
2577 	spinlock_t *spinlock_guard_##name __guard(spinlock_exit) = &(lock);	\
2578 	spin_lock(&(lock));
2579 
2580 static void mutex_exit(struct mutex **lock)
2581 {
2582 	mutex_unlock(*lock);
2583 }
2584 
2585 #define MUTEX_CONTEXT(name, lock)					\
2586 	struct mutex *mutex_guard_##name __guard(mutex_exit) = &(lock);	\
2587 	mutex_lock(&(lock));
2588 
2589 #define GENERATE_2_CONTEXT_TESTCASE(outer, outer_lock, inner, inner_lock)	\
2590 										\
2591 static void __maybe_unused inner##_in_##outer(void)				\
2592 {										\
2593 	outer##_CONTEXT(_, outer_lock);						\
2594 	{									\
2595 		inner##_CONTEXT(_, inner_lock);					\
2596 	}									\
2597 }
2598 
2599 /*
2600  * wait contexts (considering PREEMPT_RT)
2601  *
2602  * o: inner is allowed in outer
2603  * x: inner is disallowed in outer
2604  *
2605  *       \  inner |  RCU  | RAW_SPIN | SPIN | MUTEX
2606  * outer  \       |       |          |      |
2607  * ---------------+-------+----------+------+-------
2608  * HARDIRQ        |   o   |    o     |  o   |  x
2609  * ---------------+-------+----------+------+-------
2610  * NOTTHREADED_IRQ|   o   |    o     |  x   |  x
2611  * ---------------+-------+----------+------+-------
2612  * SOFTIRQ        |   o   |    o     |  o   |  x
2613  * ---------------+-------+----------+------+-------
2614  * RCU            |   o   |    o     |  o   |  x
2615  * ---------------+-------+----------+------+-------
2616  * RCU_BH         |   o   |    o     |  o   |  x
2617  * ---------------+-------+----------+------+-------
2618  * RCU_SCHED      |   o   |    o     |  x   |  x
2619  * ---------------+-------+----------+------+-------
2620  * RAW_SPIN       |   o   |    o     |  x   |  x
2621  * ---------------+-------+----------+------+-------
2622  * SPIN           |   o   |    o     |  o   |  x
2623  * ---------------+-------+----------+------+-------
2624  * MUTEX          |   o   |    o     |  o   |  o
2625  * ---------------+-------+----------+------+-------
2626  */
2627 
2628 #define GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(inner, inner_lock)		\
2629 GENERATE_2_CONTEXT_TESTCASE(HARDIRQ, , inner, inner_lock)			\
2630 GENERATE_2_CONTEXT_TESTCASE(NOTTHREADED_HARDIRQ, , inner, inner_lock)		\
2631 GENERATE_2_CONTEXT_TESTCASE(SOFTIRQ, , inner, inner_lock)			\
2632 GENERATE_2_CONTEXT_TESTCASE(RCU, , inner, inner_lock)				\
2633 GENERATE_2_CONTEXT_TESTCASE(RCU_BH, , inner, inner_lock)			\
2634 GENERATE_2_CONTEXT_TESTCASE(RCU_SCHED, , inner, inner_lock)			\
2635 GENERATE_2_CONTEXT_TESTCASE(RAW_SPINLOCK, raw_lock_A, inner, inner_lock)	\
2636 GENERATE_2_CONTEXT_TESTCASE(SPINLOCK, lock_A, inner, inner_lock)		\
2637 GENERATE_2_CONTEXT_TESTCASE(MUTEX, mutex_A, inner, inner_lock)
2638 
2639 GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(RCU, )
2640 GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(RAW_SPINLOCK, raw_lock_B)
2641 GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(SPINLOCK, lock_B)
2642 GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(MUTEX, mutex_B)
2643 
2644 /* the outer context allows all kinds of preemption */
2645 #define DO_CONTEXT_TESTCASE_OUTER_PREEMPTIBLE(outer)			\
2646 	dotest(RCU_in_##outer, SUCCESS, LOCKTYPE_RWLOCK);		\
2647 	dotest(RAW_SPINLOCK_in_##outer, SUCCESS, LOCKTYPE_SPIN);	\
2648 	dotest(SPINLOCK_in_##outer, SUCCESS, LOCKTYPE_SPIN);		\
2649 	dotest(MUTEX_in_##outer, SUCCESS, LOCKTYPE_MUTEX);		\
2650 
2651 /*
2652  * the outer context only allows the preemption introduced by spinlock_t (which
2653  * is a sleepable lock for PREEMPT_RT)
2654  */
2655 #define DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(outer)		\
2656 	dotest(RCU_in_##outer, SUCCESS, LOCKTYPE_RWLOCK);		\
2657 	dotest(RAW_SPINLOCK_in_##outer, SUCCESS, LOCKTYPE_SPIN);	\
2658 	dotest(SPINLOCK_in_##outer, SUCCESS, LOCKTYPE_SPIN);		\
2659 	dotest(MUTEX_in_##outer, FAILURE, LOCKTYPE_MUTEX);		\
2660 
2661 /* the outer doesn't allows any kind of preemption */
2662 #define DO_CONTEXT_TESTCASE_OUTER_NOT_PREEMPTIBLE(outer)			\
2663 	dotest(RCU_in_##outer, SUCCESS, LOCKTYPE_RWLOCK);		\
2664 	dotest(RAW_SPINLOCK_in_##outer, SUCCESS, LOCKTYPE_SPIN);	\
2665 	dotest(SPINLOCK_in_##outer, FAILURE, LOCKTYPE_SPIN);		\
2666 	dotest(MUTEX_in_##outer, FAILURE, LOCKTYPE_MUTEX);		\
2667 
2668 static void wait_context_tests(void)
2669 {
2670 	printk("  --------------------------------------------------------------------------\n");
2671 	printk("  | wait context tests |\n");
2672 	printk("  --------------------------------------------------------------------------\n");
2673 	printk("                                 | rcu  | raw  | spin |mutex |\n");
2674 	printk("  --------------------------------------------------------------------------\n");
2675 	print_testname("in hardirq context");
2676 	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(HARDIRQ);
2677 	pr_cont("\n");
2678 
2679 	print_testname("in hardirq context (not threaded)");
2680 	DO_CONTEXT_TESTCASE_OUTER_NOT_PREEMPTIBLE(NOTTHREADED_HARDIRQ);
2681 	pr_cont("\n");
2682 
2683 	print_testname("in softirq context");
2684 	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(SOFTIRQ);
2685 	pr_cont("\n");
2686 
2687 	print_testname("in RCU context");
2688 	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(RCU);
2689 	pr_cont("\n");
2690 
2691 	print_testname("in RCU-bh context");
2692 	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(RCU_BH);
2693 	pr_cont("\n");
2694 
2695 	print_testname("in RCU-sched context");
2696 	DO_CONTEXT_TESTCASE_OUTER_NOT_PREEMPTIBLE(RCU_SCHED);
2697 	pr_cont("\n");
2698 
2699 	print_testname("in RAW_SPINLOCK context");
2700 	DO_CONTEXT_TESTCASE_OUTER_NOT_PREEMPTIBLE(RAW_SPINLOCK);
2701 	pr_cont("\n");
2702 
2703 	print_testname("in SPINLOCK context");
2704 	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(SPINLOCK);
2705 	pr_cont("\n");
2706 
2707 	print_testname("in MUTEX context");
2708 	DO_CONTEXT_TESTCASE_OUTER_PREEMPTIBLE(MUTEX);
2709 	pr_cont("\n");
2710 }
2711 
2712 static void local_lock_2(void)
2713 {
2714 	local_lock(&local_A);	/* IRQ-ON */
2715 	local_unlock(&local_A);
2716 
2717 	HARDIRQ_ENTER();
2718 	spin_lock(&lock_A);		/* IN-IRQ */
2719 	spin_unlock(&lock_A);
2720 	HARDIRQ_EXIT()
2721 
2722 	HARDIRQ_DISABLE();
2723 	spin_lock(&lock_A);
2724 	local_lock(&local_A);	/* IN-IRQ <-> IRQ-ON cycle, false */
2725 	local_unlock(&local_A);
2726 	spin_unlock(&lock_A);
2727 	HARDIRQ_ENABLE();
2728 }
2729 
2730 static void local_lock_3A(void)
2731 {
2732 	local_lock(&local_A);	/* IRQ-ON */
2733 	spin_lock(&lock_B);		/* IRQ-ON */
2734 	spin_unlock(&lock_B);
2735 	local_unlock(&local_A);
2736 
2737 	HARDIRQ_ENTER();
2738 	spin_lock(&lock_A);		/* IN-IRQ */
2739 	spin_unlock(&lock_A);
2740 	HARDIRQ_EXIT()
2741 
2742 	HARDIRQ_DISABLE();
2743 	spin_lock(&lock_A);
2744 	local_lock(&local_A);	/* IN-IRQ <-> IRQ-ON cycle only if we count local_lock(), false */
2745 	local_unlock(&local_A);
2746 	spin_unlock(&lock_A);
2747 	HARDIRQ_ENABLE();
2748 }
2749 
2750 static void local_lock_3B(void)
2751 {
2752 	local_lock(&local_A);	/* IRQ-ON */
2753 	spin_lock(&lock_B);		/* IRQ-ON */
2754 	spin_unlock(&lock_B);
2755 	local_unlock(&local_A);
2756 
2757 	HARDIRQ_ENTER();
2758 	spin_lock(&lock_A);		/* IN-IRQ */
2759 	spin_unlock(&lock_A);
2760 	HARDIRQ_EXIT()
2761 
2762 	HARDIRQ_DISABLE();
2763 	spin_lock(&lock_A);
2764 	local_lock(&local_A);	/* IN-IRQ <-> IRQ-ON cycle only if we count local_lock(), false */
2765 	local_unlock(&local_A);
2766 	spin_unlock(&lock_A);
2767 	HARDIRQ_ENABLE();
2768 
2769 	HARDIRQ_DISABLE();
2770 	spin_lock(&lock_A);
2771 	spin_lock(&lock_B);		/* IN-IRQ <-> IRQ-ON cycle, true */
2772 	spin_unlock(&lock_B);
2773 	spin_unlock(&lock_A);
2774 	HARDIRQ_DISABLE();
2775 
2776 }
2777 
2778 static void local_lock_tests(void)
2779 {
2780 	printk("  --------------------------------------------------------------------------\n");
2781 	printk("  | local_lock tests |\n");
2782 	printk("  ---------------------\n");
2783 
2784 	print_testname("local_lock inversion  2");
2785 	dotest(local_lock_2, SUCCESS, LOCKTYPE_LL);
2786 	pr_cont("\n");
2787 
2788 	print_testname("local_lock inversion 3A");
2789 	dotest(local_lock_3A, SUCCESS, LOCKTYPE_LL);
2790 	pr_cont("\n");
2791 
2792 	print_testname("local_lock inversion 3B");
2793 	dotest(local_lock_3B, FAILURE, LOCKTYPE_LL);
2794 	pr_cont("\n");
2795 }
2796 
2797 static void hardirq_deadlock_softirq_not_deadlock(void)
2798 {
2799 	/* mutex_A is hardirq-unsafe and softirq-unsafe */
2800 	/* mutex_A -> lock_C */
2801 	mutex_lock(&mutex_A);
2802 	HARDIRQ_DISABLE();
2803 	spin_lock(&lock_C);
2804 	spin_unlock(&lock_C);
2805 	HARDIRQ_ENABLE();
2806 	mutex_unlock(&mutex_A);
2807 
2808 	/* lock_A is hardirq-safe */
2809 	HARDIRQ_ENTER();
2810 	spin_lock(&lock_A);
2811 	spin_unlock(&lock_A);
2812 	HARDIRQ_EXIT();
2813 
2814 	/* lock_A -> lock_B */
2815 	HARDIRQ_DISABLE();
2816 	spin_lock(&lock_A);
2817 	spin_lock(&lock_B);
2818 	spin_unlock(&lock_B);
2819 	spin_unlock(&lock_A);
2820 	HARDIRQ_ENABLE();
2821 
2822 	/* lock_B -> lock_C */
2823 	HARDIRQ_DISABLE();
2824 	spin_lock(&lock_B);
2825 	spin_lock(&lock_C);
2826 	spin_unlock(&lock_C);
2827 	spin_unlock(&lock_B);
2828 	HARDIRQ_ENABLE();
2829 
2830 	/* lock_D is softirq-safe */
2831 	SOFTIRQ_ENTER();
2832 	spin_lock(&lock_D);
2833 	spin_unlock(&lock_D);
2834 	SOFTIRQ_EXIT();
2835 
2836 	/* And lock_D is hardirq-unsafe */
2837 	SOFTIRQ_DISABLE();
2838 	spin_lock(&lock_D);
2839 	spin_unlock(&lock_D);
2840 	SOFTIRQ_ENABLE();
2841 
2842 	/*
2843 	 * mutex_A -> lock_C -> lock_D is softirq-unsafe -> softirq-safe, not
2844 	 * deadlock.
2845 	 *
2846 	 * lock_A -> lock_B -> lock_C -> lock_D is hardirq-safe ->
2847 	 * hardirq-unsafe, deadlock.
2848 	 */
2849 	HARDIRQ_DISABLE();
2850 	spin_lock(&lock_C);
2851 	spin_lock(&lock_D);
2852 	spin_unlock(&lock_D);
2853 	spin_unlock(&lock_C);
2854 	HARDIRQ_ENABLE();
2855 }
2856 
2857 void locking_selftest(void)
2858 {
2859 	/*
2860 	 * Got a locking failure before the selftest ran?
2861 	 */
2862 	if (!debug_locks) {
2863 		printk("----------------------------------\n");
2864 		printk("| Locking API testsuite disabled |\n");
2865 		printk("----------------------------------\n");
2866 		return;
2867 	}
2868 
2869 	/*
2870 	 * treats read_lock() as recursive read locks for testing purpose
2871 	 */
2872 	force_read_lock_recursive = 1;
2873 
2874 	/*
2875 	 * Run the testsuite:
2876 	 */
2877 	printk("------------------------\n");
2878 	printk("| Locking API testsuite:\n");
2879 	printk("----------------------------------------------------------------------------\n");
2880 	printk("                                 | spin |wlock |rlock |mutex | wsem | rsem |rtmutex\n");
2881 	printk("  --------------------------------------------------------------------------\n");
2882 
2883 	init_shared_classes();
2884 	lockdep_set_selftest_task(current);
2885 
2886 	DO_TESTCASE_6R("A-A deadlock", AA);
2887 	DO_TESTCASE_6R("A-B-B-A deadlock", ABBA);
2888 	DO_TESTCASE_6R("A-B-B-C-C-A deadlock", ABBCCA);
2889 	DO_TESTCASE_6R("A-B-C-A-B-C deadlock", ABCABC);
2890 	DO_TESTCASE_6R("A-B-B-C-C-D-D-A deadlock", ABBCCDDA);
2891 	DO_TESTCASE_6R("A-B-C-D-B-D-D-A deadlock", ABCDBDDA);
2892 	DO_TESTCASE_6R("A-B-C-D-B-C-D-A deadlock", ABCDBCDA);
2893 	DO_TESTCASE_6("double unlock", double_unlock);
2894 	DO_TESTCASE_6("initialize held", init_held);
2895 
2896 	printk("  --------------------------------------------------------------------------\n");
2897 	print_testname("recursive read-lock");
2898 	pr_cont("             |");
2899 	dotest(rlock_AA1, SUCCESS, LOCKTYPE_RWLOCK);
2900 	pr_cont("             |");
2901 	dotest(rsem_AA1, FAILURE, LOCKTYPE_RWSEM);
2902 	pr_cont("\n");
2903 
2904 	print_testname("recursive read-lock #2");
2905 	pr_cont("             |");
2906 	dotest(rlock_AA1B, SUCCESS, LOCKTYPE_RWLOCK);
2907 	pr_cont("             |");
2908 	dotest(rsem_AA1B, FAILURE, LOCKTYPE_RWSEM);
2909 	pr_cont("\n");
2910 
2911 	print_testname("mixed read-write-lock");
2912 	pr_cont("             |");
2913 	dotest(rlock_AA2, FAILURE, LOCKTYPE_RWLOCK);
2914 	pr_cont("             |");
2915 	dotest(rsem_AA2, FAILURE, LOCKTYPE_RWSEM);
2916 	pr_cont("\n");
2917 
2918 	print_testname("mixed write-read-lock");
2919 	pr_cont("             |");
2920 	dotest(rlock_AA3, FAILURE, LOCKTYPE_RWLOCK);
2921 	pr_cont("             |");
2922 	dotest(rsem_AA3, FAILURE, LOCKTYPE_RWSEM);
2923 	pr_cont("\n");
2924 
2925 	print_testname("mixed read-lock/lock-write ABBA");
2926 	pr_cont("             |");
2927 	dotest(rlock_ABBA1, FAILURE, LOCKTYPE_RWLOCK);
2928 	pr_cont("             |");
2929 	dotest(rwsem_ABBA1, FAILURE, LOCKTYPE_RWSEM);
2930 
2931 	print_testname("mixed read-lock/lock-read ABBA");
2932 	pr_cont("             |");
2933 	dotest(rlock_ABBA2, SUCCESS, LOCKTYPE_RWLOCK);
2934 	pr_cont("             |");
2935 	dotest(rwsem_ABBA2, FAILURE, LOCKTYPE_RWSEM);
2936 
2937 	print_testname("mixed write-lock/lock-write ABBA");
2938 	pr_cont("             |");
2939 	dotest(rlock_ABBA3, FAILURE, LOCKTYPE_RWLOCK);
2940 	pr_cont("             |");
2941 	dotest(rwsem_ABBA3, FAILURE, LOCKTYPE_RWSEM);
2942 
2943 	print_testname("chain cached mixed R-L/L-W ABBA");
2944 	pr_cont("             |");
2945 	dotest(rlock_chaincache_ABBA1, FAILURE, LOCKTYPE_RWLOCK);
2946 
2947 	DO_TESTCASE_6x1RRB("rlock W1R2/W2R3/W3R1", W1R2_W2R3_W3R1);
2948 	DO_TESTCASE_6x1RRB("rlock W1W2/R2R3/W3R1", W1W2_R2R3_W3R1);
2949 	DO_TESTCASE_6x1RR("rlock W1W2/R2R3/R3W1", W1W2_R2R3_R3W1);
2950 	DO_TESTCASE_6x1RR("rlock W1R2/R2R3/W3W1", W1R2_R2R3_W3W1);
2951 
2952 	printk("  --------------------------------------------------------------------------\n");
2953 	/*
2954 	 * irq-context testcases:
2955 	 */
2956 	DO_TESTCASE_2x6("irqs-on + irq-safe-A", irqsafe1);
2957 	NON_RT(DO_TESTCASE_2x3("sirq-safe-A => hirqs-on", irqsafe2A));
2958 	DO_TESTCASE_2x6("safe-A + irqs-on", irqsafe2B);
2959 	DO_TESTCASE_6x6("safe-A + unsafe-B #1", irqsafe3);
2960 	DO_TESTCASE_6x6("safe-A + unsafe-B #2", irqsafe4);
2961 	DO_TESTCASE_6x6RW("irq lock-inversion", irq_inversion);
2962 
2963 	DO_TESTCASE_6x2x2RW("irq read-recursion", irq_read_recursion);
2964 	DO_TESTCASE_6x2x2RW("irq read-recursion #2", irq_read_recursion2);
2965 	DO_TESTCASE_6x2x2RW("irq read-recursion #3", irq_read_recursion3);
2966 
2967 	ww_tests();
2968 
2969 	force_read_lock_recursive = 0;
2970 	/*
2971 	 * queued_read_lock() specific test cases can be put here
2972 	 */
2973 	if (IS_ENABLED(CONFIG_QUEUED_RWLOCKS))
2974 		queued_read_lock_tests();
2975 
2976 	fs_reclaim_tests();
2977 
2978 	/* Wait context test cases that are specific for RAW_LOCK_NESTING */
2979 	if (IS_ENABLED(CONFIG_PROVE_RAW_LOCK_NESTING))
2980 		wait_context_tests();
2981 
2982 	local_lock_tests();
2983 
2984 	print_testname("hardirq_unsafe_softirq_safe");
2985 	dotest(hardirq_deadlock_softirq_not_deadlock, FAILURE, LOCKTYPE_SPECIAL);
2986 	pr_cont("\n");
2987 
2988 	if (unexpected_testcase_failures) {
2989 		printk("-----------------------------------------------------------------\n");
2990 		debug_locks = 0;
2991 		printk("BUG: %3d unexpected failures (out of %3d) - debugging disabled! |\n",
2992 			unexpected_testcase_failures, testcase_total);
2993 		printk("-----------------------------------------------------------------\n");
2994 	} else if (expected_testcase_failures && testcase_successes) {
2995 		printk("--------------------------------------------------------\n");
2996 		printk("%3d out of %3d testcases failed, as expected. |\n",
2997 			expected_testcase_failures, testcase_total);
2998 		printk("----------------------------------------------------\n");
2999 		debug_locks = 1;
3000 	} else if (expected_testcase_failures && !testcase_successes) {
3001 		printk("--------------------------------------------------------\n");
3002 		printk("All %3d testcases failed, as expected. |\n",
3003 			expected_testcase_failures);
3004 		printk("----------------------------------------\n");
3005 		debug_locks = 1;
3006 	} else {
3007 		printk("-------------------------------------------------------\n");
3008 		printk("Good, all %3d testcases passed! |\n",
3009 			testcase_successes);
3010 		printk("---------------------------------\n");
3011 		debug_locks = 1;
3012 	}
3013 	lockdep_set_selftest_task(NULL);
3014 	debug_locks_silent = 0;
3015 }
3016