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 */
38static unsigned int debug_locks_verbose;
39unsigned int force_read_lock_recursive;
40
41static DEFINE_WD_CLASS(ww_lockdep);
42
43static 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
64static struct ww_acquire_ctx t, t2;
65static struct ww_mutex o, o2, o3;
66
67/*
68 * Normal standalone locks, for the circular and irq-context
69 * dependency tests:
70 */
71static DEFINE_SPINLOCK(lock_A);
72static DEFINE_SPINLOCK(lock_B);
73static DEFINE_SPINLOCK(lock_C);
74static DEFINE_SPINLOCK(lock_D);
75
76static DEFINE_RAW_SPINLOCK(raw_lock_A);
77static DEFINE_RAW_SPINLOCK(raw_lock_B);
78
79static DEFINE_RWLOCK(rwlock_A);
80static DEFINE_RWLOCK(rwlock_B);
81static DEFINE_RWLOCK(rwlock_C);
82static DEFINE_RWLOCK(rwlock_D);
83
84static DEFINE_MUTEX(mutex_A);
85static DEFINE_MUTEX(mutex_B);
86static DEFINE_MUTEX(mutex_C);
87static DEFINE_MUTEX(mutex_D);
88
89static DECLARE_RWSEM(rwsem_A);
90static DECLARE_RWSEM(rwsem_B);
91static DECLARE_RWSEM(rwsem_C);
92static DECLARE_RWSEM(rwsem_D);
93
94#ifdef CONFIG_RT_MUTEXES
95
96static DEFINE_RT_MUTEX(rtmutex_A);
97static DEFINE_RT_MUTEX(rtmutex_B);
98static DEFINE_RT_MUTEX(rtmutex_C);
99static 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 */
109static DEFINE_SPINLOCK(lock_X1);
110static DEFINE_SPINLOCK(lock_X2);
111static DEFINE_SPINLOCK(lock_Y1);
112static DEFINE_SPINLOCK(lock_Y2);
113static DEFINE_SPINLOCK(lock_Z1);
114static DEFINE_SPINLOCK(lock_Z2);
115
116static DEFINE_RWLOCK(rwlock_X1);
117static DEFINE_RWLOCK(rwlock_X2);
118static DEFINE_RWLOCK(rwlock_Y1);
119static DEFINE_RWLOCK(rwlock_Y2);
120static DEFINE_RWLOCK(rwlock_Z1);
121static DEFINE_RWLOCK(rwlock_Z2);
122
123static DEFINE_MUTEX(mutex_X1);
124static DEFINE_MUTEX(mutex_X2);
125static DEFINE_MUTEX(mutex_Y1);
126static DEFINE_MUTEX(mutex_Y2);
127static DEFINE_MUTEX(mutex_Z1);
128static DEFINE_MUTEX(mutex_Z2);
129
130static DECLARE_RWSEM(rwsem_X1);
131static DECLARE_RWSEM(rwsem_X2);
132static DECLARE_RWSEM(rwsem_Y1);
133static DECLARE_RWSEM(rwsem_Y2);
134static DECLARE_RWSEM(rwsem_Z1);
135static DECLARE_RWSEM(rwsem_Z2);
136
137#ifdef CONFIG_RT_MUTEXES
138
139static DEFINE_RT_MUTEX(rtmutex_X1);
140static DEFINE_RT_MUTEX(rtmutex_X2);
141static DEFINE_RT_MUTEX(rtmutex_Y1);
142static DEFINE_RT_MUTEX(rtmutex_Y2);
143static DEFINE_RT_MUTEX(rtmutex_Z1);
144static DEFINE_RT_MUTEX(rtmutex_Z2);
145
146#endif
147
148static 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) 				\
155static noinline void					\
156init_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
165INIT_CLASS_FUNC(X)
166INIT_CLASS_FUNC(Y)
167INIT_CLASS_FUNC(Z)
168
169static 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						\
281static void name(void) { E(); }
282
283#define GENERATE_PERMUTATIONS_2_EVENTS(name)	\
284						\
285static void name##_12(void) { E1(); E2(); }	\
286static void name##_21(void) { E2(); E1(); }
287
288#define GENERATE_PERMUTATIONS_3_EVENTS(name)		\
289							\
290static void name##_123(void) { E1(); E2(); E3(); }	\
291static void name##_132(void) { E1(); E3(); E2(); }	\
292static void name##_213(void) { E2(); E1(); E3(); }	\
293static void name##_231(void) { E2(); E3(); E1(); }	\
294static void name##_312(void) { E3(); E1(); E2(); }	\
295static 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"
310GENERATE_TESTCASE(AA_spin)
311#include "locking-selftest-wlock.h"
312GENERATE_TESTCASE(AA_wlock)
313#include "locking-selftest-rlock.h"
314GENERATE_TESTCASE(AA_rlock)
315#include "locking-selftest-mutex.h"
316GENERATE_TESTCASE(AA_mutex)
317#include "locking-selftest-wsem.h"
318GENERATE_TESTCASE(AA_wsem)
319#include "locking-selftest-rsem.h"
320GENERATE_TESTCASE(AA_rsem)
321
322#ifdef CONFIG_RT_MUTEXES
323#include "locking-selftest-rtmutex.h"
324GENERATE_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 */
333static void rlock_AA1(void)
334{
335	RL(X1);
336	RL(X1); // this one should NOT fail
337}
338
339static void rlock_AA1B(void)
340{
341	RL(X1);
342	RL(X2); // this one should NOT fail
343}
344
345static void rsem_AA1(void)
346{
347	RSL(X1);
348	RSL(X1); // this one should fail
349}
350
351static 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 */
359static void rlock_AA2(void)
360{
361	RL(X1);
362	WL(X2); // this one should fail
363}
364
365static void rsem_AA2(void)
366{
367	RSL(X1);
368	WSL(X2); // this one should fail
369}
370
371static void rlock_AA3(void)
372{
373	WL(X1);
374	RL(X2); // this one should fail
375}
376
377static 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 */
389static 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
402static 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 */
440static 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 */
464static 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
477static 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 */
497static 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
510static 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"
536GENERATE_TESTCASE(ABBA_spin)
537#include "locking-selftest-wlock.h"
538GENERATE_TESTCASE(ABBA_wlock)
539#include "locking-selftest-rlock.h"
540GENERATE_TESTCASE(ABBA_rlock)
541#include "locking-selftest-mutex.h"
542GENERATE_TESTCASE(ABBA_mutex)
543#include "locking-selftest-wsem.h"
544GENERATE_TESTCASE(ABBA_wsem)
545#include "locking-selftest-rsem.h"
546GENERATE_TESTCASE(ABBA_rsem)
547
548#ifdef CONFIG_RT_MUTEXES
549#include "locking-selftest-rtmutex.h"
550GENERATE_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"
569GENERATE_TESTCASE(ABBCCA_spin)
570#include "locking-selftest-wlock.h"
571GENERATE_TESTCASE(ABBCCA_wlock)
572#include "locking-selftest-rlock.h"
573GENERATE_TESTCASE(ABBCCA_rlock)
574#include "locking-selftest-mutex.h"
575GENERATE_TESTCASE(ABBCCA_mutex)
576#include "locking-selftest-wsem.h"
577GENERATE_TESTCASE(ABBCCA_wsem)
578#include "locking-selftest-rsem.h"
579GENERATE_TESTCASE(ABBCCA_rsem)
580
581#ifdef CONFIG_RT_MUTEXES
582#include "locking-selftest-rtmutex.h"
583GENERATE_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"
602GENERATE_TESTCASE(ABCABC_spin)
603#include "locking-selftest-wlock.h"
604GENERATE_TESTCASE(ABCABC_wlock)
605#include "locking-selftest-rlock.h"
606GENERATE_TESTCASE(ABCABC_rlock)
607#include "locking-selftest-mutex.h"
608GENERATE_TESTCASE(ABCABC_mutex)
609#include "locking-selftest-wsem.h"
610GENERATE_TESTCASE(ABCABC_wsem)
611#include "locking-selftest-rsem.h"
612GENERATE_TESTCASE(ABCABC_rsem)
613
614#ifdef CONFIG_RT_MUTEXES
615#include "locking-selftest-rtmutex.h"
616GENERATE_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"
636GENERATE_TESTCASE(ABBCCDDA_spin)
637#include "locking-selftest-wlock.h"
638GENERATE_TESTCASE(ABBCCDDA_wlock)
639#include "locking-selftest-rlock.h"
640GENERATE_TESTCASE(ABBCCDDA_rlock)
641#include "locking-selftest-mutex.h"
642GENERATE_TESTCASE(ABBCCDDA_mutex)
643#include "locking-selftest-wsem.h"
644GENERATE_TESTCASE(ABBCCDDA_wsem)
645#include "locking-selftest-rsem.h"
646GENERATE_TESTCASE(ABBCCDDA_rsem)
647
648#ifdef CONFIG_RT_MUTEXES
649#include "locking-selftest-rtmutex.h"
650GENERATE_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"
669GENERATE_TESTCASE(ABCDBDDA_spin)
670#include "locking-selftest-wlock.h"
671GENERATE_TESTCASE(ABCDBDDA_wlock)
672#include "locking-selftest-rlock.h"
673GENERATE_TESTCASE(ABCDBDDA_rlock)
674#include "locking-selftest-mutex.h"
675GENERATE_TESTCASE(ABCDBDDA_mutex)
676#include "locking-selftest-wsem.h"
677GENERATE_TESTCASE(ABCDBDDA_wsem)
678#include "locking-selftest-rsem.h"
679GENERATE_TESTCASE(ABCDBDDA_rsem)
680
681#ifdef CONFIG_RT_MUTEXES
682#include "locking-selftest-rtmutex.h"
683GENERATE_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"
702GENERATE_TESTCASE(ABCDBCDA_spin)
703#include "locking-selftest-wlock.h"
704GENERATE_TESTCASE(ABCDBCDA_wlock)
705#include "locking-selftest-rlock.h"
706GENERATE_TESTCASE(ABCDBCDA_rlock)
707#include "locking-selftest-mutex.h"
708GENERATE_TESTCASE(ABCDBCDA_mutex)
709#include "locking-selftest-wsem.h"
710GENERATE_TESTCASE(ABCDBCDA_wsem)
711#include "locking-selftest-rsem.h"
712GENERATE_TESTCASE(ABCDBCDA_rsem)
713
714#ifdef CONFIG_RT_MUTEXES
715#include "locking-selftest-rtmutex.h"
716GENERATE_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"
740GENERATE_TESTCASE(double_unlock_spin)
741#include "locking-selftest-wlock.h"
742GENERATE_TESTCASE(double_unlock_wlock)
743#include "locking-selftest-rlock.h"
744GENERATE_TESTCASE(double_unlock_rlock)
745#include "locking-selftest-mutex.h"
746GENERATE_TESTCASE(double_unlock_mutex)
747#include "locking-selftest-wsem.h"
748GENERATE_TESTCASE(double_unlock_wsem)
749#include "locking-selftest-rsem.h"
750GENERATE_TESTCASE(double_unlock_rsem)
751
752#ifdef CONFIG_RT_MUTEXES
753#include "locking-selftest-rtmutex.h"
754GENERATE_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"
771GENERATE_TESTCASE(init_held_spin)
772#include "locking-selftest-wlock.h"
773GENERATE_TESTCASE(init_held_wlock)
774#include "locking-selftest-rlock.h"
775GENERATE_TESTCASE(init_held_rlock)
776#include "locking-selftest-mutex.h"
777GENERATE_TESTCASE(init_held_mutex)
778#include "locking-selftest-wsem.h"
779GENERATE_TESTCASE(init_held_wsem)
780#include "locking-selftest-rsem.h"
781GENERATE_TESTCASE(init_held_rsem)
782
783#ifdef CONFIG_RT_MUTEXES
784#include "locking-selftest-rtmutex.h"
785GENERATE_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"
809GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_spin)
810
811#include "locking-selftest-rlock-hardirq.h"
812GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_rlock)
813
814#include "locking-selftest-wlock-hardirq.h"
815GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_hard_wlock)
816
817#ifndef CONFIG_PREEMPT_RT
818#include "locking-selftest-spin-softirq.h"
819GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_soft_spin)
820
821#include "locking-selftest-rlock-softirq.h"
822GENERATE_PERMUTATIONS_2_EVENTS(irqsafe1_soft_rlock)
823
824#include "locking-selftest-wlock-softirq.h"
825GENERATE_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"
853GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2A_spin)
854
855#include "locking-selftest-wlock.h"
856GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2A_wlock)
857
858#include "locking-selftest-rlock.h"
859GENERATE_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"
887GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_spin)
888
889#include "locking-selftest-rlock-hardirq.h"
890GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_rlock)
891
892#include "locking-selftest-wlock-hardirq.h"
893GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_hard_wlock)
894
895#ifndef CONFIG_PREEMPT_RT
896#include "locking-selftest-spin-softirq.h"
897GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_soft_spin)
898
899#include "locking-selftest-rlock-softirq.h"
900GENERATE_PERMUTATIONS_2_EVENTS(irqsafe2B_soft_rlock)
901
902#include "locking-selftest-wlock-softirq.h"
903GENERATE_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"
935GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_spin)
936
937#include "locking-selftest-rlock-hardirq.h"
938GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_rlock)
939
940#include "locking-selftest-wlock-hardirq.h"
941GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_hard_wlock)
942
943#ifndef CONFIG_PREEMPT_RT
944#include "locking-selftest-spin-softirq.h"
945GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_soft_spin)
946
947#include "locking-selftest-rlock-softirq.h"
948GENERATE_PERMUTATIONS_3_EVENTS(irqsafe3_soft_rlock)
949
950#include "locking-selftest-wlock-softirq.h"
951GENERATE_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"
985GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_spin)
986
987#include "locking-selftest-rlock-hardirq.h"
988GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_rlock)
989
990#include "locking-selftest-wlock-hardirq.h"
991GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_hard_wlock)
992
993#ifndef CONFIG_PREEMPT_RT
994#include "locking-selftest-spin-softirq.h"
995GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_soft_spin)
996
997#include "locking-selftest-rlock-softirq.h"
998GENERATE_PERMUTATIONS_3_EVENTS(irqsafe4_soft_rlock)
999
1000#include "locking-selftest-wlock-softirq.h"
1001GENERATE_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"
1049GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_spin)
1050
1051#include "locking-selftest-rlock-hardirq.h"
1052GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_rlock)
1053
1054#include "locking-selftest-wlock-hardirq.h"
1055GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_hard_wlock)
1056
1057#ifndef CONFIG_PREEMPT_RT
1058#include "locking-selftest-spin-softirq.h"
1059GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_soft_spin)
1060
1061#include "locking-selftest-rlock-softirq.h"
1062GENERATE_PERMUTATIONS_3_EVENTS(irq_inversion_soft_rlock)
1063
1064#include "locking-selftest-wlock-softirq.h"
1065GENERATE_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"
1098GENERATE_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"
1130GENERATE_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"
1162GENERATE_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"
1194GENERATE_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"
1229GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion_hard_rlock)
1230
1231#include "locking-selftest-wlock.h"
1232GENERATE_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"
1237GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion_soft_rlock)
1238
1239#include "locking-selftest-wlock.h"
1240GENERATE_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"
1277GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion2_hard_rlock)
1278
1279#include "locking-selftest-wlock.h"
1280GENERATE_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"
1285GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion2_soft_rlock)
1286
1287#include "locking-selftest-wlock.h"
1288GENERATE_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"
1333GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion3_hard_rlock)
1334
1335#include "locking-selftest-wlock.h"
1336GENERATE_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"
1341GENERATE_PERMUTATIONS_3_EVENTS(irq_read_recursion3_soft_rlock)
1342
1343#include "locking-selftest-wlock.h"
1344GENERATE_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
1396static 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
1425static int testcase_total;
1426static int testcase_successes;
1427static int expected_testcase_failures;
1428static int unexpected_testcase_failures;
1429
1430static 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
1500static 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
1669static 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
1719static 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
1780static void ww_test_two_contexts(void)
1781{
1782	WWAI(&t);
1783	WWAI(&t2);
1784}
1785
1786static 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
1795static void ww_test_context_done_twice(void)
1796{
1797	WWAI(&t);
1798	WWAD(&t);
1799	WWAD(&t);
1800	WWAF(&t);
1801}
1802
1803static void ww_test_context_unlock_twice(void)
1804{
1805	WWAI(&t);
1806	WWAD(&t);
1807	WWAF(&t);
1808	WWAF(&t);
1809}
1810
1811static void ww_test_context_fini_early(void)
1812{
1813	WWAI(&t);
1814	WWL(&o, &t);
1815	WWAD(&t);
1816	WWAF(&t);
1817}
1818
1819static void ww_test_context_lock_after_done(void)
1820{
1821	WWAI(&t);
1822	WWAD(&t);
1823	WWL(&o, &t);
1824}
1825
1826static void ww_test_object_unlock_twice(void)
1827{
1828	WWL1(&o);
1829	WWU(&o);
1830	WWU(&o);
1831}
1832
1833static 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
1842static void ww_test_object_lock_stale_context(void)
1843{
1844	WWAI(&t);
1845	o.ctx = &t2;
1846	WWL(&o, &t);
1847}
1848
1849static 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
1875static 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
1901static 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
1926static 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
1951static 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
1972static 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
1993static 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
2019static 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
2044static 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
2069static 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
2094static 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 */
2101static 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
2113static void ww_test_unneeded_slow(void)
2114{
2115	WWAI(&t);
2116
2117	ww_mutex_lock_slow(&o, &t);
2118}
2119
2120static 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
2131static 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
2146static 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
2162static 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
2174static 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
2186static 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
2199static void ww_test_block_block(void)
2200{
2201	WWL1(&o);
2202	WWL1(&o2);
2203}
2204
2205static void ww_test_block_try(void)
2206{
2207	bool ret;
2208
2209	WWL1(&o);
2210	ret = WWT(&o2);
2211	WARN_ON(!ret);
2212}
2213
2214static 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
2225static 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
2241static 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
2261static 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
2283static 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 */
2374static 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 */
2401static 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 */
2428static 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
2447static 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
2465static 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
2472static 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
2479static 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
2490static 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/* Defines guard classes to create contexts */
2510DEFINE_LOCK_GUARD_0(HARDIRQ, HARDIRQ_ENTER(), HARDIRQ_EXIT())
2511DEFINE_LOCK_GUARD_0(NOTTHREADED_HARDIRQ,
2512	do {
2513		local_irq_disable();
2514		__irq_enter();
2515		WARN_ON(!in_irq());
2516	} while(0), HARDIRQ_EXIT())
2517DEFINE_LOCK_GUARD_0(SOFTIRQ, SOFTIRQ_ENTER(), SOFTIRQ_EXIT())
2518
2519/* Define RCU guards, should go away when RCU has its own guard definitions */
2520DEFINE_LOCK_GUARD_0(RCU, rcu_read_lock(), rcu_read_unlock())
2521DEFINE_LOCK_GUARD_0(RCU_BH, rcu_read_lock_bh(), rcu_read_unlock_bh())
2522DEFINE_LOCK_GUARD_0(RCU_SCHED, rcu_read_lock_sched(), rcu_read_unlock_sched())
2523
2524
2525#define GENERATE_2_CONTEXT_TESTCASE(outer, outer_lock, inner, inner_lock)	\
2526										\
2527static void __maybe_unused inner##_in_##outer(void)				\
2528{										\
2529	/* Relies the reversed clean-up ordering: inner first */		\
2530	guard(outer)(outer_lock);						\
2531	guard(inner)(inner_lock);						\
2532}
2533
2534/*
2535 * wait contexts (considering PREEMPT_RT)
2536 *
2537 * o: inner is allowed in outer
2538 * x: inner is disallowed in outer
2539 *
2540 *       \  inner |  RCU  | RAW_SPIN | SPIN | MUTEX
2541 * outer  \       |       |          |      |
2542 * ---------------+-------+----------+------+-------
2543 * HARDIRQ        |   o   |    o     |  o   |  x
2544 * ---------------+-------+----------+------+-------
2545 * NOTTHREADED_IRQ|   o   |    o     |  x   |  x
2546 * ---------------+-------+----------+------+-------
2547 * SOFTIRQ        |   o   |    o     |  o   |  x
2548 * ---------------+-------+----------+------+-------
2549 * RCU            |   o   |    o     |  o   |  x
2550 * ---------------+-------+----------+------+-------
2551 * RCU_BH         |   o   |    o     |  o   |  x
2552 * ---------------+-------+----------+------+-------
2553 * RCU_SCHED      |   o   |    o     |  x   |  x
2554 * ---------------+-------+----------+------+-------
2555 * RAW_SPIN       |   o   |    o     |  x   |  x
2556 * ---------------+-------+----------+------+-------
2557 * SPIN           |   o   |    o     |  o   |  x
2558 * ---------------+-------+----------+------+-------
2559 * MUTEX          |   o   |    o     |  o   |  o
2560 * ---------------+-------+----------+------+-------
2561 */
2562
2563#define GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(inner, inner_lock)		\
2564GENERATE_2_CONTEXT_TESTCASE(HARDIRQ, , inner, inner_lock)			\
2565GENERATE_2_CONTEXT_TESTCASE(NOTTHREADED_HARDIRQ, , inner, inner_lock)		\
2566GENERATE_2_CONTEXT_TESTCASE(SOFTIRQ, , inner, inner_lock)			\
2567GENERATE_2_CONTEXT_TESTCASE(RCU, , inner, inner_lock)				\
2568GENERATE_2_CONTEXT_TESTCASE(RCU_BH, , inner, inner_lock)			\
2569GENERATE_2_CONTEXT_TESTCASE(RCU_SCHED, , inner, inner_lock)			\
2570GENERATE_2_CONTEXT_TESTCASE(raw_spinlock, &raw_lock_A, inner, inner_lock)	\
2571GENERATE_2_CONTEXT_TESTCASE(spinlock, &lock_A, inner, inner_lock)		\
2572GENERATE_2_CONTEXT_TESTCASE(mutex, &mutex_A, inner, inner_lock)
2573
2574GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(RCU, )
2575GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(raw_spinlock, &raw_lock_B)
2576GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(spinlock, &lock_B)
2577GENERATE_2_CONTEXT_TESTCASE_FOR_ALL_OUTER(mutex, &mutex_B)
2578
2579/* the outer context allows all kinds of preemption */
2580#define DO_CONTEXT_TESTCASE_OUTER_PREEMPTIBLE(outer)			\
2581	dotest(RCU_in_##outer, SUCCESS, LOCKTYPE_RWLOCK);		\
2582	dotest(raw_spinlock_in_##outer, SUCCESS, LOCKTYPE_SPIN);	\
2583	dotest(spinlock_in_##outer, SUCCESS, LOCKTYPE_SPIN);		\
2584	dotest(mutex_in_##outer, SUCCESS, LOCKTYPE_MUTEX);		\
2585
2586/*
2587 * the outer context only allows the preemption introduced by spinlock_t (which
2588 * is a sleepable lock for PREEMPT_RT)
2589 */
2590#define DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(outer)		\
2591	dotest(RCU_in_##outer, SUCCESS, LOCKTYPE_RWLOCK);		\
2592	dotest(raw_spinlock_in_##outer, SUCCESS, LOCKTYPE_SPIN);	\
2593	dotest(spinlock_in_##outer, SUCCESS, LOCKTYPE_SPIN);		\
2594	dotest(mutex_in_##outer, FAILURE, LOCKTYPE_MUTEX);		\
2595
2596/* the outer doesn't allows any kind of preemption */
2597#define DO_CONTEXT_TESTCASE_OUTER_NOT_PREEMPTIBLE(outer)			\
2598	dotest(RCU_in_##outer, SUCCESS, LOCKTYPE_RWLOCK);		\
2599	dotest(raw_spinlock_in_##outer, SUCCESS, LOCKTYPE_SPIN);	\
2600	dotest(spinlock_in_##outer, FAILURE, LOCKTYPE_SPIN);		\
2601	dotest(mutex_in_##outer, FAILURE, LOCKTYPE_MUTEX);		\
2602
2603static void wait_context_tests(void)
2604{
2605	printk("  --------------------------------------------------------------------------\n");
2606	printk("  | wait context tests |\n");
2607	printk("  --------------------------------------------------------------------------\n");
2608	printk("                                 | rcu  | raw  | spin |mutex |\n");
2609	printk("  --------------------------------------------------------------------------\n");
2610	print_testname("in hardirq context");
2611	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(HARDIRQ);
2612	pr_cont("\n");
2613
2614	print_testname("in hardirq context (not threaded)");
2615	DO_CONTEXT_TESTCASE_OUTER_NOT_PREEMPTIBLE(NOTTHREADED_HARDIRQ);
2616	pr_cont("\n");
2617
2618	print_testname("in softirq context");
2619	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(SOFTIRQ);
2620	pr_cont("\n");
2621
2622	print_testname("in RCU context");
2623	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(RCU);
2624	pr_cont("\n");
2625
2626	print_testname("in RCU-bh context");
2627	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(RCU_BH);
2628	pr_cont("\n");
2629
2630	print_testname("in RCU-sched context");
2631	DO_CONTEXT_TESTCASE_OUTER_NOT_PREEMPTIBLE(RCU_SCHED);
2632	pr_cont("\n");
2633
2634	print_testname("in RAW_SPINLOCK context");
2635	DO_CONTEXT_TESTCASE_OUTER_NOT_PREEMPTIBLE(raw_spinlock);
2636	pr_cont("\n");
2637
2638	print_testname("in SPINLOCK context");
2639	DO_CONTEXT_TESTCASE_OUTER_LIMITED_PREEMPTIBLE(spinlock);
2640	pr_cont("\n");
2641
2642	print_testname("in MUTEX context");
2643	DO_CONTEXT_TESTCASE_OUTER_PREEMPTIBLE(mutex);
2644	pr_cont("\n");
2645}
2646
2647static void local_lock_2(void)
2648{
2649	local_lock(&local_A);	/* IRQ-ON */
2650	local_unlock(&local_A);
2651
2652	HARDIRQ_ENTER();
2653	spin_lock(&lock_A);		/* IN-IRQ */
2654	spin_unlock(&lock_A);
2655	HARDIRQ_EXIT()
2656
2657	HARDIRQ_DISABLE();
2658	spin_lock(&lock_A);
2659	local_lock(&local_A);	/* IN-IRQ <-> IRQ-ON cycle, false */
2660	local_unlock(&local_A);
2661	spin_unlock(&lock_A);
2662	HARDIRQ_ENABLE();
2663}
2664
2665static void local_lock_3A(void)
2666{
2667	local_lock(&local_A);	/* IRQ-ON */
2668	spin_lock(&lock_B);		/* IRQ-ON */
2669	spin_unlock(&lock_B);
2670	local_unlock(&local_A);
2671
2672	HARDIRQ_ENTER();
2673	spin_lock(&lock_A);		/* IN-IRQ */
2674	spin_unlock(&lock_A);
2675	HARDIRQ_EXIT()
2676
2677	HARDIRQ_DISABLE();
2678	spin_lock(&lock_A);
2679	local_lock(&local_A);	/* IN-IRQ <-> IRQ-ON cycle only if we count local_lock(), false */
2680	local_unlock(&local_A);
2681	spin_unlock(&lock_A);
2682	HARDIRQ_ENABLE();
2683}
2684
2685static void local_lock_3B(void)
2686{
2687	local_lock(&local_A);	/* IRQ-ON */
2688	spin_lock(&lock_B);		/* IRQ-ON */
2689	spin_unlock(&lock_B);
2690	local_unlock(&local_A);
2691
2692	HARDIRQ_ENTER();
2693	spin_lock(&lock_A);		/* IN-IRQ */
2694	spin_unlock(&lock_A);
2695	HARDIRQ_EXIT()
2696
2697	HARDIRQ_DISABLE();
2698	spin_lock(&lock_A);
2699	local_lock(&local_A);	/* IN-IRQ <-> IRQ-ON cycle only if we count local_lock(), false */
2700	local_unlock(&local_A);
2701	spin_unlock(&lock_A);
2702	HARDIRQ_ENABLE();
2703
2704	HARDIRQ_DISABLE();
2705	spin_lock(&lock_A);
2706	spin_lock(&lock_B);		/* IN-IRQ <-> IRQ-ON cycle, true */
2707	spin_unlock(&lock_B);
2708	spin_unlock(&lock_A);
2709	HARDIRQ_DISABLE();
2710
2711}
2712
2713static void local_lock_tests(void)
2714{
2715	printk("  --------------------------------------------------------------------------\n");
2716	printk("  | local_lock tests |\n");
2717	printk("  ---------------------\n");
2718
2719	print_testname("local_lock inversion  2");
2720	dotest(local_lock_2, SUCCESS, LOCKTYPE_LL);
2721	pr_cont("\n");
2722
2723	print_testname("local_lock inversion 3A");
2724	dotest(local_lock_3A, SUCCESS, LOCKTYPE_LL);
2725	pr_cont("\n");
2726
2727	print_testname("local_lock inversion 3B");
2728	dotest(local_lock_3B, FAILURE, LOCKTYPE_LL);
2729	pr_cont("\n");
2730}
2731
2732static void hardirq_deadlock_softirq_not_deadlock(void)
2733{
2734	/* mutex_A is hardirq-unsafe and softirq-unsafe */
2735	/* mutex_A -> lock_C */
2736	mutex_lock(&mutex_A);
2737	HARDIRQ_DISABLE();
2738	spin_lock(&lock_C);
2739	spin_unlock(&lock_C);
2740	HARDIRQ_ENABLE();
2741	mutex_unlock(&mutex_A);
2742
2743	/* lock_A is hardirq-safe */
2744	HARDIRQ_ENTER();
2745	spin_lock(&lock_A);
2746	spin_unlock(&lock_A);
2747	HARDIRQ_EXIT();
2748
2749	/* lock_A -> lock_B */
2750	HARDIRQ_DISABLE();
2751	spin_lock(&lock_A);
2752	spin_lock(&lock_B);
2753	spin_unlock(&lock_B);
2754	spin_unlock(&lock_A);
2755	HARDIRQ_ENABLE();
2756
2757	/* lock_B -> lock_C */
2758	HARDIRQ_DISABLE();
2759	spin_lock(&lock_B);
2760	spin_lock(&lock_C);
2761	spin_unlock(&lock_C);
2762	spin_unlock(&lock_B);
2763	HARDIRQ_ENABLE();
2764
2765	/* lock_D is softirq-safe */
2766	SOFTIRQ_ENTER();
2767	spin_lock(&lock_D);
2768	spin_unlock(&lock_D);
2769	SOFTIRQ_EXIT();
2770
2771	/* And lock_D is hardirq-unsafe */
2772	SOFTIRQ_DISABLE();
2773	spin_lock(&lock_D);
2774	spin_unlock(&lock_D);
2775	SOFTIRQ_ENABLE();
2776
2777	/*
2778	 * mutex_A -> lock_C -> lock_D is softirq-unsafe -> softirq-safe, not
2779	 * deadlock.
2780	 *
2781	 * lock_A -> lock_B -> lock_C -> lock_D is hardirq-safe ->
2782	 * hardirq-unsafe, deadlock.
2783	 */
2784	HARDIRQ_DISABLE();
2785	spin_lock(&lock_C);
2786	spin_lock(&lock_D);
2787	spin_unlock(&lock_D);
2788	spin_unlock(&lock_C);
2789	HARDIRQ_ENABLE();
2790}
2791
2792void locking_selftest(void)
2793{
2794	/*
2795	 * Got a locking failure before the selftest ran?
2796	 */
2797	if (!debug_locks) {
2798		printk("----------------------------------\n");
2799		printk("| Locking API testsuite disabled |\n");
2800		printk("----------------------------------\n");
2801		return;
2802	}
2803
2804	/*
2805	 * treats read_lock() as recursive read locks for testing purpose
2806	 */
2807	force_read_lock_recursive = 1;
2808
2809	/*
2810	 * Run the testsuite:
2811	 */
2812	printk("------------------------\n");
2813	printk("| Locking API testsuite:\n");
2814	printk("----------------------------------------------------------------------------\n");
2815	printk("                                 | spin |wlock |rlock |mutex | wsem | rsem |rtmutex\n");
2816	printk("  --------------------------------------------------------------------------\n");
2817
2818	init_shared_classes();
2819	lockdep_set_selftest_task(current);
2820
2821	DO_TESTCASE_6R("A-A deadlock", AA);
2822	DO_TESTCASE_6R("A-B-B-A deadlock", ABBA);
2823	DO_TESTCASE_6R("A-B-B-C-C-A deadlock", ABBCCA);
2824	DO_TESTCASE_6R("A-B-C-A-B-C deadlock", ABCABC);
2825	DO_TESTCASE_6R("A-B-B-C-C-D-D-A deadlock", ABBCCDDA);
2826	DO_TESTCASE_6R("A-B-C-D-B-D-D-A deadlock", ABCDBDDA);
2827	DO_TESTCASE_6R("A-B-C-D-B-C-D-A deadlock", ABCDBCDA);
2828	DO_TESTCASE_6("double unlock", double_unlock);
2829	DO_TESTCASE_6("initialize held", init_held);
2830
2831	printk("  --------------------------------------------------------------------------\n");
2832	print_testname("recursive read-lock");
2833	pr_cont("             |");
2834	dotest(rlock_AA1, SUCCESS, LOCKTYPE_RWLOCK);
2835	pr_cont("             |");
2836	dotest(rsem_AA1, FAILURE, LOCKTYPE_RWSEM);
2837	pr_cont("\n");
2838
2839	print_testname("recursive read-lock #2");
2840	pr_cont("             |");
2841	dotest(rlock_AA1B, SUCCESS, LOCKTYPE_RWLOCK);
2842	pr_cont("             |");
2843	dotest(rsem_AA1B, FAILURE, LOCKTYPE_RWSEM);
2844	pr_cont("\n");
2845
2846	print_testname("mixed read-write-lock");
2847	pr_cont("             |");
2848	dotest(rlock_AA2, FAILURE, LOCKTYPE_RWLOCK);
2849	pr_cont("             |");
2850	dotest(rsem_AA2, FAILURE, LOCKTYPE_RWSEM);
2851	pr_cont("\n");
2852
2853	print_testname("mixed write-read-lock");
2854	pr_cont("             |");
2855	dotest(rlock_AA3, FAILURE, LOCKTYPE_RWLOCK);
2856	pr_cont("             |");
2857	dotest(rsem_AA3, FAILURE, LOCKTYPE_RWSEM);
2858	pr_cont("\n");
2859
2860	print_testname("mixed read-lock/lock-write ABBA");
2861	pr_cont("             |");
2862	dotest(rlock_ABBA1, FAILURE, LOCKTYPE_RWLOCK);
2863	pr_cont("             |");
2864	dotest(rwsem_ABBA1, FAILURE, LOCKTYPE_RWSEM);
2865
2866	print_testname("mixed read-lock/lock-read ABBA");
2867	pr_cont("             |");
2868	dotest(rlock_ABBA2, SUCCESS, LOCKTYPE_RWLOCK);
2869	pr_cont("             |");
2870	dotest(rwsem_ABBA2, FAILURE, LOCKTYPE_RWSEM);
2871
2872	print_testname("mixed write-lock/lock-write ABBA");
2873	pr_cont("             |");
2874	dotest(rlock_ABBA3, FAILURE, LOCKTYPE_RWLOCK);
2875	pr_cont("             |");
2876	dotest(rwsem_ABBA3, FAILURE, LOCKTYPE_RWSEM);
2877
2878	print_testname("chain cached mixed R-L/L-W ABBA");
2879	pr_cont("             |");
2880	dotest(rlock_chaincache_ABBA1, FAILURE, LOCKTYPE_RWLOCK);
2881
2882	DO_TESTCASE_6x1RRB("rlock W1R2/W2R3/W3R1", W1R2_W2R3_W3R1);
2883	DO_TESTCASE_6x1RRB("rlock W1W2/R2R3/W3R1", W1W2_R2R3_W3R1);
2884	DO_TESTCASE_6x1RR("rlock W1W2/R2R3/R3W1", W1W2_R2R3_R3W1);
2885	DO_TESTCASE_6x1RR("rlock W1R2/R2R3/W3W1", W1R2_R2R3_W3W1);
2886
2887	printk("  --------------------------------------------------------------------------\n");
2888	/*
2889	 * irq-context testcases:
2890	 */
2891	DO_TESTCASE_2x6("irqs-on + irq-safe-A", irqsafe1);
2892	NON_RT(DO_TESTCASE_2x3("sirq-safe-A => hirqs-on", irqsafe2A));
2893	DO_TESTCASE_2x6("safe-A + irqs-on", irqsafe2B);
2894	DO_TESTCASE_6x6("safe-A + unsafe-B #1", irqsafe3);
2895	DO_TESTCASE_6x6("safe-A + unsafe-B #2", irqsafe4);
2896	DO_TESTCASE_6x6RW("irq lock-inversion", irq_inversion);
2897
2898	DO_TESTCASE_6x2x2RW("irq read-recursion", irq_read_recursion);
2899	DO_TESTCASE_6x2x2RW("irq read-recursion #2", irq_read_recursion2);
2900	DO_TESTCASE_6x2x2RW("irq read-recursion #3", irq_read_recursion3);
2901
2902	ww_tests();
2903
2904	force_read_lock_recursive = 0;
2905	/*
2906	 * queued_read_lock() specific test cases can be put here
2907	 */
2908	if (IS_ENABLED(CONFIG_QUEUED_RWLOCKS))
2909		queued_read_lock_tests();
2910
2911	fs_reclaim_tests();
2912
2913	/* Wait context test cases that are specific for RAW_LOCK_NESTING */
2914	if (IS_ENABLED(CONFIG_PROVE_RAW_LOCK_NESTING))
2915		wait_context_tests();
2916
2917	local_lock_tests();
2918
2919	print_testname("hardirq_unsafe_softirq_safe");
2920	dotest(hardirq_deadlock_softirq_not_deadlock, FAILURE, LOCKTYPE_SPECIAL);
2921	pr_cont("\n");
2922
2923	if (unexpected_testcase_failures) {
2924		printk("-----------------------------------------------------------------\n");
2925		debug_locks = 0;
2926		printk("BUG: %3d unexpected failures (out of %3d) - debugging disabled! |\n",
2927			unexpected_testcase_failures, testcase_total);
2928		printk("-----------------------------------------------------------------\n");
2929	} else if (expected_testcase_failures && testcase_successes) {
2930		printk("--------------------------------------------------------\n");
2931		printk("%3d out of %3d testcases failed, as expected. |\n",
2932			expected_testcase_failures, testcase_total);
2933		printk("----------------------------------------------------\n");
2934		debug_locks = 1;
2935	} else if (expected_testcase_failures && !testcase_successes) {
2936		printk("--------------------------------------------------------\n");
2937		printk("All %3d testcases failed, as expected. |\n",
2938			expected_testcase_failures);
2939		printk("----------------------------------------\n");
2940		debug_locks = 1;
2941	} else {
2942		printk("-------------------------------------------------------\n");
2943		printk("Good, all %3d testcases passed! |\n",
2944			testcase_successes);
2945		printk("---------------------------------\n");
2946		debug_locks = 1;
2947	}
2948	lockdep_set_selftest_task(NULL);
2949	debug_locks_silent = 0;
2950}
2951