Linux Audio

Check our new training course

Loading...
v4.6
 
  1/*
  2 * Module-based torture test facility for locking
  3 *
  4 * This program is free software; you can redistribute it and/or modify
  5 * it under the terms of the GNU General Public License as published by
  6 * the Free Software Foundation; either version 2 of the License, or
  7 * (at your option) any later version.
  8 *
  9 * This program is distributed in the hope that it will be useful,
 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 12 * GNU General Public License for more details.
 13 *
 14 * You should have received a copy of the GNU General Public License
 15 * along with this program; if not, you can access it online at
 16 * http://www.gnu.org/licenses/gpl-2.0.html.
 17 *
 18 * Copyright (C) IBM Corporation, 2014
 19 *
 20 * Authors: Paul E. McKenney <paulmck@us.ibm.com>
 21 *          Davidlohr Bueso <dave@stgolabs.net>
 22 *	Based on kernel/rcu/torture.c.
 23 */
 
 
 
 24#include <linux/kernel.h>
 25#include <linux/module.h>
 26#include <linux/kthread.h>
 27#include <linux/sched/rt.h>
 28#include <linux/spinlock.h>
 29#include <linux/rwlock.h>
 30#include <linux/mutex.h>
 31#include <linux/rwsem.h>
 32#include <linux/smp.h>
 33#include <linux/interrupt.h>
 34#include <linux/sched.h>
 
 
 35#include <linux/atomic.h>
 36#include <linux/moduleparam.h>
 37#include <linux/delay.h>
 38#include <linux/slab.h>
 39#include <linux/percpu-rwsem.h>
 40#include <linux/torture.h>
 
 41
 42MODULE_LICENSE("GPL");
 43MODULE_AUTHOR("Paul E. McKenney <paulmck@us.ibm.com>");
 44
 45torture_param(int, nwriters_stress, -1,
 46	     "Number of write-locking stress-test threads");
 47torture_param(int, nreaders_stress, -1,
 48	     "Number of read-locking stress-test threads");
 49torture_param(int, onoff_holdoff, 0, "Time after boot before CPU hotplugs (s)");
 50torture_param(int, onoff_interval, 0,
 51	     "Time between CPU hotplugs (s), 0=disable");
 52torture_param(int, shuffle_interval, 3,
 53	     "Number of jiffies between shuffles, 0=disable");
 54torture_param(int, shutdown_secs, 0, "Shutdown time (j), <= zero to disable.");
 55torture_param(int, stat_interval, 60,
 56	     "Number of seconds between stats printk()s");
 57torture_param(int, stutter, 5, "Number of jiffies to run/halt test, 0=disable");
 58torture_param(bool, verbose, true,
 59	     "Enable verbose debugging printk()s");
 60
 61static char *torture_type = "spin_lock";
 62module_param(torture_type, charp, 0444);
 63MODULE_PARM_DESC(torture_type,
 64		 "Type of lock to torture (spin_lock, spin_lock_irq, mutex_lock, ...)");
 65
 66static struct task_struct *stats_task;
 67static struct task_struct **writer_tasks;
 68static struct task_struct **reader_tasks;
 69
 70static bool lock_is_write_held;
 71static bool lock_is_read_held;
 
 72
 73struct lock_stress_stats {
 74	long n_lock_fail;
 75	long n_lock_acquired;
 76};
 77
 78#if defined(MODULE)
 79#define LOCKTORTURE_RUNNABLE_INIT 1
 80#else
 81#define LOCKTORTURE_RUNNABLE_INIT 0
 82#endif
 83int torture_runnable = LOCKTORTURE_RUNNABLE_INIT;
 84module_param(torture_runnable, int, 0444);
 85MODULE_PARM_DESC(torture_runnable, "Start locktorture at module init");
 86
 87/* Forward reference. */
 88static void lock_torture_cleanup(void);
 89
 90/*
 91 * Operations vector for selecting different types of tests.
 92 */
 93struct lock_torture_ops {
 94	void (*init)(void);
 95	int (*writelock)(void);
 
 96	void (*write_delay)(struct torture_random_state *trsp);
 97	void (*task_boost)(struct torture_random_state *trsp);
 98	void (*writeunlock)(void);
 99	int (*readlock)(void);
100	void (*read_delay)(struct torture_random_state *trsp);
101	void (*readunlock)(void);
102
103	unsigned long flags; /* for irq spinlocks */
104	const char *name;
105};
106
107struct lock_torture_cxt {
108	int nrealwriters_stress;
109	int nrealreaders_stress;
110	bool debug_lock;
 
111	atomic_t n_lock_torture_errors;
112	struct lock_torture_ops *cur_ops;
113	struct lock_stress_stats *lwsa; /* writer statistics */
114	struct lock_stress_stats *lrsa; /* reader statistics */
115};
116static struct lock_torture_cxt cxt = { 0, 0, false,
117				       ATOMIC_INIT(0),
118				       NULL, NULL};
119/*
120 * Definitions for lock torture testing.
121 */
122
123static int torture_lock_busted_write_lock(void)
124{
125	return 0;  /* BUGGY, do not use in real life!!! */
126}
127
128static void torture_lock_busted_write_delay(struct torture_random_state *trsp)
129{
130	const unsigned long longdelay_ms = 100;
131
132	/* We want a long delay occasionally to force massive contention.  */
133	if (!(torture_random(trsp) %
134	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
135		mdelay(longdelay_ms);
136#ifdef CONFIG_PREEMPT
137	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
138		preempt_schedule();  /* Allow test to be preempted. */
139#endif
140}
141
142static void torture_lock_busted_write_unlock(void)
143{
144	  /* BUGGY, do not use in real life!!! */
145}
146
147static void torture_boost_dummy(struct torture_random_state *trsp)
148{
149	/* Only rtmutexes care about priority */
150}
151
152static struct lock_torture_ops lock_busted_ops = {
153	.writelock	= torture_lock_busted_write_lock,
154	.write_delay	= torture_lock_busted_write_delay,
155	.task_boost     = torture_boost_dummy,
156	.writeunlock	= torture_lock_busted_write_unlock,
157	.readlock       = NULL,
158	.read_delay     = NULL,
159	.readunlock     = NULL,
160	.name		= "lock_busted"
161};
162
163static DEFINE_SPINLOCK(torture_spinlock);
164
165static int torture_spin_lock_write_lock(void) __acquires(torture_spinlock)
 
166{
167	spin_lock(&torture_spinlock);
168	return 0;
169}
170
171static void torture_spin_lock_write_delay(struct torture_random_state *trsp)
172{
173	const unsigned long shortdelay_us = 2;
174	const unsigned long longdelay_ms = 100;
175
176	/* We want a short delay mostly to emulate likely code, and
177	 * we want a long delay occasionally to force massive contention.
178	 */
179	if (!(torture_random(trsp) %
180	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
181		mdelay(longdelay_ms);
182	if (!(torture_random(trsp) %
183	      (cxt.nrealwriters_stress * 2 * shortdelay_us)))
184		udelay(shortdelay_us);
185#ifdef CONFIG_PREEMPT
186	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
187		preempt_schedule();  /* Allow test to be preempted. */
188#endif
189}
190
191static void torture_spin_lock_write_unlock(void) __releases(torture_spinlock)
 
192{
193	spin_unlock(&torture_spinlock);
194}
195
196static struct lock_torture_ops spin_lock_ops = {
197	.writelock	= torture_spin_lock_write_lock,
198	.write_delay	= torture_spin_lock_write_delay,
199	.task_boost     = torture_boost_dummy,
200	.writeunlock	= torture_spin_lock_write_unlock,
201	.readlock       = NULL,
202	.read_delay     = NULL,
203	.readunlock     = NULL,
204	.name		= "spin_lock"
205};
206
207static int torture_spin_lock_write_lock_irq(void)
208__acquires(torture_spinlock)
209{
210	unsigned long flags;
211
212	spin_lock_irqsave(&torture_spinlock, flags);
213	cxt.cur_ops->flags = flags;
214	return 0;
215}
216
217static void torture_lock_spin_write_unlock_irq(void)
218__releases(torture_spinlock)
219{
220	spin_unlock_irqrestore(&torture_spinlock, cxt.cur_ops->flags);
221}
222
223static struct lock_torture_ops spin_lock_irq_ops = {
224	.writelock	= torture_spin_lock_write_lock_irq,
225	.write_delay	= torture_spin_lock_write_delay,
226	.task_boost     = torture_boost_dummy,
227	.writeunlock	= torture_lock_spin_write_unlock_irq,
228	.readlock       = NULL,
229	.read_delay     = NULL,
230	.readunlock     = NULL,
231	.name		= "spin_lock_irq"
232};
233
234static DEFINE_RWLOCK(torture_rwlock);
235
236static int torture_rwlock_write_lock(void) __acquires(torture_rwlock)
 
237{
238	write_lock(&torture_rwlock);
239	return 0;
240}
241
242static void torture_rwlock_write_delay(struct torture_random_state *trsp)
243{
244	const unsigned long shortdelay_us = 2;
245	const unsigned long longdelay_ms = 100;
246
247	/* We want a short delay mostly to emulate likely code, and
248	 * we want a long delay occasionally to force massive contention.
249	 */
250	if (!(torture_random(trsp) %
251	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
252		mdelay(longdelay_ms);
253	else
254		udelay(shortdelay_us);
255}
256
257static void torture_rwlock_write_unlock(void) __releases(torture_rwlock)
 
258{
259	write_unlock(&torture_rwlock);
260}
261
262static int torture_rwlock_read_lock(void) __acquires(torture_rwlock)
 
263{
264	read_lock(&torture_rwlock);
265	return 0;
266}
267
268static void torture_rwlock_read_delay(struct torture_random_state *trsp)
269{
270	const unsigned long shortdelay_us = 10;
271	const unsigned long longdelay_ms = 100;
272
273	/* We want a short delay mostly to emulate likely code, and
274	 * we want a long delay occasionally to force massive contention.
275	 */
276	if (!(torture_random(trsp) %
277	      (cxt.nrealreaders_stress * 2000 * longdelay_ms)))
278		mdelay(longdelay_ms);
279	else
280		udelay(shortdelay_us);
281}
282
283static void torture_rwlock_read_unlock(void) __releases(torture_rwlock)
 
284{
285	read_unlock(&torture_rwlock);
286}
287
288static struct lock_torture_ops rw_lock_ops = {
289	.writelock	= torture_rwlock_write_lock,
290	.write_delay	= torture_rwlock_write_delay,
291	.task_boost     = torture_boost_dummy,
292	.writeunlock	= torture_rwlock_write_unlock,
293	.readlock       = torture_rwlock_read_lock,
294	.read_delay     = torture_rwlock_read_delay,
295	.readunlock     = torture_rwlock_read_unlock,
296	.name		= "rw_lock"
297};
298
299static int torture_rwlock_write_lock_irq(void) __acquires(torture_rwlock)
 
300{
301	unsigned long flags;
302
303	write_lock_irqsave(&torture_rwlock, flags);
304	cxt.cur_ops->flags = flags;
305	return 0;
306}
307
308static void torture_rwlock_write_unlock_irq(void)
309__releases(torture_rwlock)
310{
311	write_unlock_irqrestore(&torture_rwlock, cxt.cur_ops->flags);
312}
313
314static int torture_rwlock_read_lock_irq(void) __acquires(torture_rwlock)
 
315{
316	unsigned long flags;
317
318	read_lock_irqsave(&torture_rwlock, flags);
319	cxt.cur_ops->flags = flags;
320	return 0;
321}
322
323static void torture_rwlock_read_unlock_irq(void)
324__releases(torture_rwlock)
325{
326	read_unlock_irqrestore(&torture_rwlock, cxt.cur_ops->flags);
327}
328
329static struct lock_torture_ops rw_lock_irq_ops = {
330	.writelock	= torture_rwlock_write_lock_irq,
331	.write_delay	= torture_rwlock_write_delay,
332	.task_boost     = torture_boost_dummy,
333	.writeunlock	= torture_rwlock_write_unlock_irq,
334	.readlock       = torture_rwlock_read_lock_irq,
335	.read_delay     = torture_rwlock_read_delay,
336	.readunlock     = torture_rwlock_read_unlock_irq,
337	.name		= "rw_lock_irq"
338};
339
340static DEFINE_MUTEX(torture_mutex);
341
342static int torture_mutex_lock(void) __acquires(torture_mutex)
 
343{
344	mutex_lock(&torture_mutex);
345	return 0;
346}
347
348static void torture_mutex_delay(struct torture_random_state *trsp)
349{
350	const unsigned long longdelay_ms = 100;
351
352	/* We want a long delay occasionally to force massive contention.  */
353	if (!(torture_random(trsp) %
354	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
355		mdelay(longdelay_ms * 5);
356	else
357		mdelay(longdelay_ms / 5);
358#ifdef CONFIG_PREEMPT
359	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
360		preempt_schedule();  /* Allow test to be preempted. */
361#endif
362}
363
364static void torture_mutex_unlock(void) __releases(torture_mutex)
 
365{
366	mutex_unlock(&torture_mutex);
367}
368
369static struct lock_torture_ops mutex_lock_ops = {
370	.writelock	= torture_mutex_lock,
371	.write_delay	= torture_mutex_delay,
372	.task_boost     = torture_boost_dummy,
373	.writeunlock	= torture_mutex_unlock,
374	.readlock       = NULL,
375	.read_delay     = NULL,
376	.readunlock     = NULL,
377	.name		= "mutex_lock"
378};
379
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
380#ifdef CONFIG_RT_MUTEXES
381static DEFINE_RT_MUTEX(torture_rtmutex);
382
383static int torture_rtmutex_lock(void) __acquires(torture_rtmutex)
 
384{
385	rt_mutex_lock(&torture_rtmutex);
386	return 0;
387}
388
389static void torture_rtmutex_boost(struct torture_random_state *trsp)
390{
391	int policy;
392	struct sched_param param;
393	const unsigned int factor = 50000; /* yes, quite arbitrary */
394
395	if (!rt_task(current)) {
396		/*
397		 * (1) Boost priority once every ~50k operations. When the
398		 * task tries to take the lock, the rtmutex it will account
399		 * for the new priority, and do any corresponding pi-dance.
400		 */
401		if (!(torture_random(trsp) %
402		      (cxt.nrealwriters_stress * factor))) {
403			policy = SCHED_FIFO;
404			param.sched_priority = MAX_RT_PRIO - 1;
405		} else /* common case, do nothing */
406			return;
407	} else {
408		/*
409		 * The task will remain boosted for another ~500k operations,
410		 * then restored back to its original prio, and so forth.
411		 *
412		 * When @trsp is nil, we want to force-reset the task for
413		 * stopping the kthread.
414		 */
415		if (!trsp || !(torture_random(trsp) %
416			       (cxt.nrealwriters_stress * factor * 2))) {
417			policy = SCHED_NORMAL;
418			param.sched_priority = 0;
419		} else /* common case, do nothing */
420			return;
421	}
422
423	sched_setscheduler_nocheck(current, policy, &param);
424}
425
426static void torture_rtmutex_delay(struct torture_random_state *trsp)
427{
428	const unsigned long shortdelay_us = 2;
429	const unsigned long longdelay_ms = 100;
430
431	/*
432	 * We want a short delay mostly to emulate likely code, and
433	 * we want a long delay occasionally to force massive contention.
434	 */
435	if (!(torture_random(trsp) %
436	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
437		mdelay(longdelay_ms);
438	if (!(torture_random(trsp) %
439	      (cxt.nrealwriters_stress * 2 * shortdelay_us)))
440		udelay(shortdelay_us);
441#ifdef CONFIG_PREEMPT
442	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
443		preempt_schedule();  /* Allow test to be preempted. */
444#endif
445}
446
447static void torture_rtmutex_unlock(void) __releases(torture_rtmutex)
 
448{
449	rt_mutex_unlock(&torture_rtmutex);
450}
451
452static struct lock_torture_ops rtmutex_lock_ops = {
453	.writelock	= torture_rtmutex_lock,
454	.write_delay	= torture_rtmutex_delay,
455	.task_boost     = torture_rtmutex_boost,
456	.writeunlock	= torture_rtmutex_unlock,
457	.readlock       = NULL,
458	.read_delay     = NULL,
459	.readunlock     = NULL,
460	.name		= "rtmutex_lock"
461};
462#endif
463
464static DECLARE_RWSEM(torture_rwsem);
465static int torture_rwsem_down_write(void) __acquires(torture_rwsem)
 
466{
467	down_write(&torture_rwsem);
468	return 0;
469}
470
471static void torture_rwsem_write_delay(struct torture_random_state *trsp)
472{
473	const unsigned long longdelay_ms = 100;
474
475	/* We want a long delay occasionally to force massive contention.  */
476	if (!(torture_random(trsp) %
477	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
478		mdelay(longdelay_ms * 10);
479	else
480		mdelay(longdelay_ms / 10);
481#ifdef CONFIG_PREEMPT
482	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
483		preempt_schedule();  /* Allow test to be preempted. */
484#endif
485}
486
487static void torture_rwsem_up_write(void) __releases(torture_rwsem)
 
488{
489	up_write(&torture_rwsem);
490}
491
492static int torture_rwsem_down_read(void) __acquires(torture_rwsem)
 
493{
494	down_read(&torture_rwsem);
495	return 0;
496}
497
498static void torture_rwsem_read_delay(struct torture_random_state *trsp)
499{
500	const unsigned long longdelay_ms = 100;
501
502	/* We want a long delay occasionally to force massive contention.  */
503	if (!(torture_random(trsp) %
504	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
505		mdelay(longdelay_ms * 2);
506	else
507		mdelay(longdelay_ms / 2);
508#ifdef CONFIG_PREEMPT
509	if (!(torture_random(trsp) % (cxt.nrealreaders_stress * 20000)))
510		preempt_schedule();  /* Allow test to be preempted. */
511#endif
512}
513
514static void torture_rwsem_up_read(void) __releases(torture_rwsem)
 
515{
516	up_read(&torture_rwsem);
517}
518
519static struct lock_torture_ops rwsem_lock_ops = {
520	.writelock	= torture_rwsem_down_write,
521	.write_delay	= torture_rwsem_write_delay,
522	.task_boost     = torture_boost_dummy,
523	.writeunlock	= torture_rwsem_up_write,
524	.readlock       = torture_rwsem_down_read,
525	.read_delay     = torture_rwsem_read_delay,
526	.readunlock     = torture_rwsem_up_read,
527	.name		= "rwsem_lock"
528};
529
530#include <linux/percpu-rwsem.h>
531static struct percpu_rw_semaphore pcpu_rwsem;
532
533void torture_percpu_rwsem_init(void)
534{
535	BUG_ON(percpu_init_rwsem(&pcpu_rwsem));
536}
537
538static int torture_percpu_rwsem_down_write(void) __acquires(pcpu_rwsem)
 
 
 
 
 
 
539{
540	percpu_down_write(&pcpu_rwsem);
541	return 0;
542}
543
544static void torture_percpu_rwsem_up_write(void) __releases(pcpu_rwsem)
 
545{
546	percpu_up_write(&pcpu_rwsem);
547}
548
549static int torture_percpu_rwsem_down_read(void) __acquires(pcpu_rwsem)
 
550{
551	percpu_down_read(&pcpu_rwsem);
552	return 0;
553}
554
555static void torture_percpu_rwsem_up_read(void) __releases(pcpu_rwsem)
 
556{
557	percpu_up_read(&pcpu_rwsem);
558}
559
560static struct lock_torture_ops percpu_rwsem_lock_ops = {
561	.init		= torture_percpu_rwsem_init,
 
562	.writelock	= torture_percpu_rwsem_down_write,
563	.write_delay	= torture_rwsem_write_delay,
564	.task_boost     = torture_boost_dummy,
565	.writeunlock	= torture_percpu_rwsem_up_write,
566	.readlock       = torture_percpu_rwsem_down_read,
567	.read_delay     = torture_rwsem_read_delay,
568	.readunlock     = torture_percpu_rwsem_up_read,
569	.name		= "percpu_rwsem_lock"
570};
571
572/*
573 * Lock torture writer kthread.  Repeatedly acquires and releases
574 * the lock, checking for duplicate acquisitions.
575 */
576static int lock_torture_writer(void *arg)
577{
578	struct lock_stress_stats *lwsp = arg;
579	static DEFINE_TORTURE_RANDOM(rand);
 
580
581	VERBOSE_TOROUT_STRING("lock_torture_writer task started");
582	set_user_nice(current, MAX_NICE);
583
584	do {
585		if ((torture_random(&rand) & 0xfffff) == 0)
586			schedule_timeout_uninterruptible(1);
587
588		cxt.cur_ops->task_boost(&rand);
589		cxt.cur_ops->writelock();
590		if (WARN_ON_ONCE(lock_is_write_held))
591			lwsp->n_lock_fail++;
592		lock_is_write_held = 1;
593		if (WARN_ON_ONCE(lock_is_read_held))
594			lwsp->n_lock_fail++; /* rare, but... */
595
596		lwsp->n_lock_acquired++;
597		cxt.cur_ops->write_delay(&rand);
598		lock_is_write_held = 0;
599		cxt.cur_ops->writeunlock();
 
600
601		stutter_wait("lock_torture_writer");
602	} while (!torture_must_stop());
603
604	cxt.cur_ops->task_boost(NULL); /* reset prio */
605	torture_kthread_stopping("lock_torture_writer");
606	return 0;
607}
608
609/*
610 * Lock torture reader kthread.  Repeatedly acquires and releases
611 * the reader lock.
612 */
613static int lock_torture_reader(void *arg)
614{
615	struct lock_stress_stats *lrsp = arg;
616	static DEFINE_TORTURE_RANDOM(rand);
 
617
618	VERBOSE_TOROUT_STRING("lock_torture_reader task started");
619	set_user_nice(current, MAX_NICE);
620
621	do {
622		if ((torture_random(&rand) & 0xfffff) == 0)
623			schedule_timeout_uninterruptible(1);
624
625		cxt.cur_ops->readlock();
626		lock_is_read_held = 1;
627		if (WARN_ON_ONCE(lock_is_write_held))
628			lrsp->n_lock_fail++; /* rare, but... */
629
630		lrsp->n_lock_acquired++;
631		cxt.cur_ops->read_delay(&rand);
632		lock_is_read_held = 0;
633		cxt.cur_ops->readunlock();
634
635		stutter_wait("lock_torture_reader");
636	} while (!torture_must_stop());
637	torture_kthread_stopping("lock_torture_reader");
638	return 0;
639}
640
641/*
642 * Create an lock-torture-statistics message in the specified buffer.
643 */
644static void __torture_print_stats(char *page,
645				  struct lock_stress_stats *statp, bool write)
646{
647	bool fail = 0;
 
648	int i, n_stress;
649	long max = 0;
650	long min = statp[0].n_lock_acquired;
651	long long sum = 0;
652
653	n_stress = write ? cxt.nrealwriters_stress : cxt.nrealreaders_stress;
654	for (i = 0; i < n_stress; i++) {
655		if (statp[i].n_lock_fail)
656			fail = true;
657		sum += statp[i].n_lock_acquired;
658		if (max < statp[i].n_lock_fail)
659			max = statp[i].n_lock_fail;
660		if (min > statp[i].n_lock_fail)
661			min = statp[i].n_lock_fail;
 
662	}
663	page += sprintf(page,
664			"%s:  Total: %lld  Max/Min: %ld/%ld %s  Fail: %d %s\n",
665			write ? "Writes" : "Reads ",
666			sum, max, min, max / 2 > min ? "???" : "",
 
667			fail, fail ? "!!!" : "");
668	if (fail)
669		atomic_inc(&cxt.n_lock_torture_errors);
670}
671
672/*
673 * Print torture statistics.  Caller must ensure that there is only one
674 * call to this function at a given time!!!  This is normally accomplished
675 * by relying on the module system to only have one copy of the module
676 * loaded, and then by giving the lock_torture_stats kthread full control
677 * (or the init/cleanup functions when lock_torture_stats thread is not
678 * running).
679 */
680static void lock_torture_stats_print(void)
681{
682	int size = cxt.nrealwriters_stress * 200 + 8192;
683	char *buf;
684
685	if (cxt.cur_ops->readlock)
686		size += cxt.nrealreaders_stress * 200 + 8192;
687
688	buf = kmalloc(size, GFP_KERNEL);
689	if (!buf) {
690		pr_err("lock_torture_stats_print: Out of memory, need: %d",
691		       size);
692		return;
693	}
694
695	__torture_print_stats(buf, cxt.lwsa, true);
696	pr_alert("%s", buf);
697	kfree(buf);
698
699	if (cxt.cur_ops->readlock) {
700		buf = kmalloc(size, GFP_KERNEL);
701		if (!buf) {
702			pr_err("lock_torture_stats_print: Out of memory, need: %d",
703			       size);
704			return;
705		}
706
707		__torture_print_stats(buf, cxt.lrsa, false);
708		pr_alert("%s", buf);
709		kfree(buf);
710	}
711}
712
713/*
714 * Periodically prints torture statistics, if periodic statistics printing
715 * was specified via the stat_interval module parameter.
716 *
717 * No need to worry about fullstop here, since this one doesn't reference
718 * volatile state or register callbacks.
719 */
720static int lock_torture_stats(void *arg)
721{
722	VERBOSE_TOROUT_STRING("lock_torture_stats task started");
723	do {
724		schedule_timeout_interruptible(stat_interval * HZ);
725		lock_torture_stats_print();
726		torture_shutdown_absorb("lock_torture_stats");
727	} while (!torture_must_stop());
728	torture_kthread_stopping("lock_torture_stats");
729	return 0;
730}
731
732static inline void
733lock_torture_print_module_parms(struct lock_torture_ops *cur_ops,
734				const char *tag)
735{
736	pr_alert("%s" TORTURE_FLAG
737		 "--- %s%s: nwriters_stress=%d nreaders_stress=%d stat_interval=%d verbose=%d shuffle_interval=%d stutter=%d shutdown_secs=%d onoff_interval=%d onoff_holdoff=%d\n",
738		 torture_type, tag, cxt.debug_lock ? " [debug]": "",
739		 cxt.nrealwriters_stress, cxt.nrealreaders_stress, stat_interval,
740		 verbose, shuffle_interval, stutter, shutdown_secs,
741		 onoff_interval, onoff_holdoff);
742}
743
744static void lock_torture_cleanup(void)
745{
746	int i;
747
748	if (torture_cleanup_begin())
749		return;
750
 
 
 
 
 
 
 
 
 
 
751	if (writer_tasks) {
752		for (i = 0; i < cxt.nrealwriters_stress; i++)
753			torture_stop_kthread(lock_torture_writer,
754					     writer_tasks[i]);
755		kfree(writer_tasks);
756		writer_tasks = NULL;
757	}
758
759	if (reader_tasks) {
760		for (i = 0; i < cxt.nrealreaders_stress; i++)
761			torture_stop_kthread(lock_torture_reader,
762					     reader_tasks[i]);
763		kfree(reader_tasks);
764		reader_tasks = NULL;
765	}
766
767	torture_stop_kthread(lock_torture_stats, stats_task);
768	lock_torture_stats_print();  /* -After- the stats thread is stopped! */
769
770	if (atomic_read(&cxt.n_lock_torture_errors))
771		lock_torture_print_module_parms(cxt.cur_ops,
772						"End of test: FAILURE");
773	else if (torture_onoff_failures())
774		lock_torture_print_module_parms(cxt.cur_ops,
775						"End of test: LOCK_HOTPLUG");
776	else
777		lock_torture_print_module_parms(cxt.cur_ops,
778						"End of test: SUCCESS");
 
 
 
 
 
 
 
 
 
 
 
 
779	torture_cleanup_end();
780}
781
782static int __init lock_torture_init(void)
783{
784	int i, j;
785	int firsterr = 0;
786	static struct lock_torture_ops *torture_ops[] = {
787		&lock_busted_ops,
788		&spin_lock_ops, &spin_lock_irq_ops,
789		&rw_lock_ops, &rw_lock_irq_ops,
790		&mutex_lock_ops,
 
791#ifdef CONFIG_RT_MUTEXES
792		&rtmutex_lock_ops,
793#endif
794		&rwsem_lock_ops,
795		&percpu_rwsem_lock_ops,
796	};
797
798	if (!torture_init_begin(torture_type, verbose, &torture_runnable))
799		return -EBUSY;
800
801	/* Process args and tell the world that the torturer is on the job. */
802	for (i = 0; i < ARRAY_SIZE(torture_ops); i++) {
803		cxt.cur_ops = torture_ops[i];
804		if (strcmp(torture_type, cxt.cur_ops->name) == 0)
805			break;
806	}
807	if (i == ARRAY_SIZE(torture_ops)) {
808		pr_alert("lock-torture: invalid torture type: \"%s\"\n",
809			 torture_type);
810		pr_alert("lock-torture types:");
811		for (i = 0; i < ARRAY_SIZE(torture_ops); i++)
812			pr_alert(" %s", torture_ops[i]->name);
813		pr_alert("\n");
814		firsterr = -EINVAL;
815		goto unwind;
816	}
817	if (cxt.cur_ops->init)
818		cxt.cur_ops->init();
 
 
 
 
 
819
820	if (nwriters_stress >= 0)
821		cxt.nrealwriters_stress = nwriters_stress;
822	else
823		cxt.nrealwriters_stress = 2 * num_online_cpus();
824
 
 
 
 
 
825#ifdef CONFIG_DEBUG_MUTEXES
826	if (strncmp(torture_type, "mutex", 5) == 0)
827		cxt.debug_lock = true;
828#endif
829#ifdef CONFIG_DEBUG_RT_MUTEXES
830	if (strncmp(torture_type, "rtmutex", 7) == 0)
831		cxt.debug_lock = true;
832#endif
833#ifdef CONFIG_DEBUG_SPINLOCK
834	if ((strncmp(torture_type, "spin", 4) == 0) ||
835	    (strncmp(torture_type, "rw_lock", 7) == 0))
836		cxt.debug_lock = true;
837#endif
838
839	/* Initialize the statistics so that each run gets its own numbers. */
 
 
 
 
 
 
 
 
 
 
840
841	lock_is_write_held = 0;
842	cxt.lwsa = kmalloc(sizeof(*cxt.lwsa) * cxt.nrealwriters_stress, GFP_KERNEL);
843	if (cxt.lwsa == NULL) {
844		VERBOSE_TOROUT_STRING("cxt.lwsa: Out of memory");
845		firsterr = -ENOMEM;
846		goto unwind;
847	}
848	for (i = 0; i < cxt.nrealwriters_stress; i++) {
849		cxt.lwsa[i].n_lock_fail = 0;
850		cxt.lwsa[i].n_lock_acquired = 0;
851	}
852
853	if (cxt.cur_ops->readlock) {
854		if (nreaders_stress >= 0)
855			cxt.nrealreaders_stress = nreaders_stress;
856		else {
857			/*
858			 * By default distribute evenly the number of
859			 * readers and writers. We still run the same number
860			 * of threads as the writer-only locks default.
861			 */
862			if (nwriters_stress < 0) /* user doesn't care */
863				cxt.nrealwriters_stress = num_online_cpus();
864			cxt.nrealreaders_stress = cxt.nrealwriters_stress;
865		}
866
867		lock_is_read_held = 0;
868		cxt.lrsa = kmalloc(sizeof(*cxt.lrsa) * cxt.nrealreaders_stress, GFP_KERNEL);
869		if (cxt.lrsa == NULL) {
870			VERBOSE_TOROUT_STRING("cxt.lrsa: Out of memory");
871			firsterr = -ENOMEM;
872			kfree(cxt.lwsa);
873			goto unwind;
874		}
875
876		for (i = 0; i < cxt.nrealreaders_stress; i++) {
877			cxt.lrsa[i].n_lock_fail = 0;
878			cxt.lrsa[i].n_lock_acquired = 0;
 
 
 
 
879		}
880	}
 
881	lock_torture_print_module_parms(cxt.cur_ops, "Start of test");
882
883	/* Prepare torture context. */
884	if (onoff_interval > 0) {
885		firsterr = torture_onoff_init(onoff_holdoff * HZ,
886					      onoff_interval * HZ);
887		if (firsterr)
888			goto unwind;
889	}
890	if (shuffle_interval > 0) {
891		firsterr = torture_shuffle_init(shuffle_interval);
892		if (firsterr)
893			goto unwind;
894	}
895	if (shutdown_secs > 0) {
896		firsterr = torture_shutdown_init(shutdown_secs,
897						 lock_torture_cleanup);
898		if (firsterr)
899			goto unwind;
900	}
901	if (stutter > 0) {
902		firsterr = torture_stutter_init(stutter);
903		if (firsterr)
904			goto unwind;
905	}
906
907	writer_tasks = kzalloc(cxt.nrealwriters_stress * sizeof(writer_tasks[0]),
908			       GFP_KERNEL);
909	if (writer_tasks == NULL) {
910		VERBOSE_TOROUT_ERRSTRING("writer_tasks: Out of memory");
911		firsterr = -ENOMEM;
912		goto unwind;
 
 
 
913	}
914
915	if (cxt.cur_ops->readlock) {
916		reader_tasks = kzalloc(cxt.nrealreaders_stress * sizeof(reader_tasks[0]),
 
917				       GFP_KERNEL);
918		if (reader_tasks == NULL) {
919			VERBOSE_TOROUT_ERRSTRING("reader_tasks: Out of memory");
 
 
920			firsterr = -ENOMEM;
921			goto unwind;
922		}
923	}
924
925	/*
926	 * Create the kthreads and start torturing (oh, those poor little locks).
927	 *
928	 * TODO: Note that we interleave writers with readers, giving writers a
929	 * slight advantage, by creating its kthread first. This can be modified
930	 * for very specific needs, or even let the user choose the policy, if
931	 * ever wanted.
932	 */
933	for (i = 0, j = 0; i < cxt.nrealwriters_stress ||
934		    j < cxt.nrealreaders_stress; i++, j++) {
935		if (i >= cxt.nrealwriters_stress)
936			goto create_reader;
937
938		/* Create writer. */
939		firsterr = torture_create_kthread(lock_torture_writer, &cxt.lwsa[i],
940						  writer_tasks[i]);
941		if (firsterr)
942			goto unwind;
943
944	create_reader:
945		if (cxt.cur_ops->readlock == NULL || (j >= cxt.nrealreaders_stress))
946			continue;
947		/* Create reader. */
948		firsterr = torture_create_kthread(lock_torture_reader, &cxt.lrsa[j],
949						  reader_tasks[j]);
950		if (firsterr)
951			goto unwind;
952	}
953	if (stat_interval > 0) {
954		firsterr = torture_create_kthread(lock_torture_stats, NULL,
955						  stats_task);
956		if (firsterr)
957			goto unwind;
958	}
959	torture_init_end();
960	return 0;
961
962unwind:
963	torture_init_end();
964	lock_torture_cleanup();
 
 
 
 
965	return firsterr;
966}
967
968module_init(lock_torture_init);
969module_exit(lock_torture_cleanup);
v6.2
   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * Module-based torture test facility for locking
   4 *
 
 
 
 
 
 
 
 
 
 
 
 
 
 
   5 * Copyright (C) IBM Corporation, 2014
   6 *
   7 * Authors: Paul E. McKenney <paulmck@linux.ibm.com>
   8 *          Davidlohr Bueso <dave@stgolabs.net>
   9 *	Based on kernel/rcu/torture.c.
  10 */
  11
  12#define pr_fmt(fmt) fmt
  13
  14#include <linux/kernel.h>
  15#include <linux/module.h>
  16#include <linux/kthread.h>
  17#include <linux/sched/rt.h>
  18#include <linux/spinlock.h>
 
  19#include <linux/mutex.h>
  20#include <linux/rwsem.h>
  21#include <linux/smp.h>
  22#include <linux/interrupt.h>
  23#include <linux/sched.h>
  24#include <uapi/linux/sched/types.h>
  25#include <linux/rtmutex.h>
  26#include <linux/atomic.h>
  27#include <linux/moduleparam.h>
  28#include <linux/delay.h>
  29#include <linux/slab.h>
 
  30#include <linux/torture.h>
  31#include <linux/reboot.h>
  32
  33MODULE_LICENSE("GPL");
  34MODULE_AUTHOR("Paul E. McKenney <paulmck@linux.ibm.com>");
  35
  36torture_param(int, nwriters_stress, -1,
  37	     "Number of write-locking stress-test threads");
  38torture_param(int, nreaders_stress, -1,
  39	     "Number of read-locking stress-test threads");
  40torture_param(int, onoff_holdoff, 0, "Time after boot before CPU hotplugs (s)");
  41torture_param(int, onoff_interval, 0,
  42	     "Time between CPU hotplugs (s), 0=disable");
  43torture_param(int, shuffle_interval, 3,
  44	     "Number of jiffies between shuffles, 0=disable");
  45torture_param(int, shutdown_secs, 0, "Shutdown time (j), <= zero to disable.");
  46torture_param(int, stat_interval, 60,
  47	     "Number of seconds between stats printk()s");
  48torture_param(int, stutter, 5, "Number of jiffies to run/halt test, 0=disable");
  49torture_param(int, verbose, 1,
  50	     "Enable verbose debugging printk()s");
  51
  52static char *torture_type = "spin_lock";
  53module_param(torture_type, charp, 0444);
  54MODULE_PARM_DESC(torture_type,
  55		 "Type of lock to torture (spin_lock, spin_lock_irq, mutex_lock, ...)");
  56
  57static struct task_struct *stats_task;
  58static struct task_struct **writer_tasks;
  59static struct task_struct **reader_tasks;
  60
  61static bool lock_is_write_held;
  62static atomic_t lock_is_read_held;
  63static unsigned long last_lock_release;
  64
  65struct lock_stress_stats {
  66	long n_lock_fail;
  67	long n_lock_acquired;
  68};
  69
 
 
 
 
 
 
 
 
 
  70/* Forward reference. */
  71static void lock_torture_cleanup(void);
  72
  73/*
  74 * Operations vector for selecting different types of tests.
  75 */
  76struct lock_torture_ops {
  77	void (*init)(void);
  78	void (*exit)(void);
  79	int (*writelock)(int tid);
  80	void (*write_delay)(struct torture_random_state *trsp);
  81	void (*task_boost)(struct torture_random_state *trsp);
  82	void (*writeunlock)(int tid);
  83	int (*readlock)(int tid);
  84	void (*read_delay)(struct torture_random_state *trsp);
  85	void (*readunlock)(int tid);
  86
  87	unsigned long flags; /* for irq spinlocks */
  88	const char *name;
  89};
  90
  91struct lock_torture_cxt {
  92	int nrealwriters_stress;
  93	int nrealreaders_stress;
  94	bool debug_lock;
  95	bool init_called;
  96	atomic_t n_lock_torture_errors;
  97	struct lock_torture_ops *cur_ops;
  98	struct lock_stress_stats *lwsa; /* writer statistics */
  99	struct lock_stress_stats *lrsa; /* reader statistics */
 100};
 101static struct lock_torture_cxt cxt = { 0, 0, false, false,
 102				       ATOMIC_INIT(0),
 103				       NULL, NULL};
 104/*
 105 * Definitions for lock torture testing.
 106 */
 107
 108static int torture_lock_busted_write_lock(int tid __maybe_unused)
 109{
 110	return 0;  /* BUGGY, do not use in real life!!! */
 111}
 112
 113static void torture_lock_busted_write_delay(struct torture_random_state *trsp)
 114{
 115	const unsigned long longdelay_ms = 100;
 116
 117	/* We want a long delay occasionally to force massive contention.  */
 118	if (!(torture_random(trsp) %
 119	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
 120		mdelay(longdelay_ms);
 
 121	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
 122		torture_preempt_schedule();  /* Allow test to be preempted. */
 
 123}
 124
 125static void torture_lock_busted_write_unlock(int tid __maybe_unused)
 126{
 127	  /* BUGGY, do not use in real life!!! */
 128}
 129
 130static void torture_boost_dummy(struct torture_random_state *trsp)
 131{
 132	/* Only rtmutexes care about priority */
 133}
 134
 135static struct lock_torture_ops lock_busted_ops = {
 136	.writelock	= torture_lock_busted_write_lock,
 137	.write_delay	= torture_lock_busted_write_delay,
 138	.task_boost     = torture_boost_dummy,
 139	.writeunlock	= torture_lock_busted_write_unlock,
 140	.readlock       = NULL,
 141	.read_delay     = NULL,
 142	.readunlock     = NULL,
 143	.name		= "lock_busted"
 144};
 145
 146static DEFINE_SPINLOCK(torture_spinlock);
 147
 148static int torture_spin_lock_write_lock(int tid __maybe_unused)
 149__acquires(torture_spinlock)
 150{
 151	spin_lock(&torture_spinlock);
 152	return 0;
 153}
 154
 155static void torture_spin_lock_write_delay(struct torture_random_state *trsp)
 156{
 157	const unsigned long shortdelay_us = 2;
 158	const unsigned long longdelay_ms = 100;
 159
 160	/* We want a short delay mostly to emulate likely code, and
 161	 * we want a long delay occasionally to force massive contention.
 162	 */
 163	if (!(torture_random(trsp) %
 164	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
 165		mdelay(longdelay_ms);
 166	if (!(torture_random(trsp) %
 167	      (cxt.nrealwriters_stress * 2 * shortdelay_us)))
 168		udelay(shortdelay_us);
 
 169	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
 170		torture_preempt_schedule();  /* Allow test to be preempted. */
 
 171}
 172
 173static void torture_spin_lock_write_unlock(int tid __maybe_unused)
 174__releases(torture_spinlock)
 175{
 176	spin_unlock(&torture_spinlock);
 177}
 178
 179static struct lock_torture_ops spin_lock_ops = {
 180	.writelock	= torture_spin_lock_write_lock,
 181	.write_delay	= torture_spin_lock_write_delay,
 182	.task_boost     = torture_boost_dummy,
 183	.writeunlock	= torture_spin_lock_write_unlock,
 184	.readlock       = NULL,
 185	.read_delay     = NULL,
 186	.readunlock     = NULL,
 187	.name		= "spin_lock"
 188};
 189
 190static int torture_spin_lock_write_lock_irq(int tid __maybe_unused)
 191__acquires(torture_spinlock)
 192{
 193	unsigned long flags;
 194
 195	spin_lock_irqsave(&torture_spinlock, flags);
 196	cxt.cur_ops->flags = flags;
 197	return 0;
 198}
 199
 200static void torture_lock_spin_write_unlock_irq(int tid __maybe_unused)
 201__releases(torture_spinlock)
 202{
 203	spin_unlock_irqrestore(&torture_spinlock, cxt.cur_ops->flags);
 204}
 205
 206static struct lock_torture_ops spin_lock_irq_ops = {
 207	.writelock	= torture_spin_lock_write_lock_irq,
 208	.write_delay	= torture_spin_lock_write_delay,
 209	.task_boost     = torture_boost_dummy,
 210	.writeunlock	= torture_lock_spin_write_unlock_irq,
 211	.readlock       = NULL,
 212	.read_delay     = NULL,
 213	.readunlock     = NULL,
 214	.name		= "spin_lock_irq"
 215};
 216
 217static DEFINE_RWLOCK(torture_rwlock);
 218
 219static int torture_rwlock_write_lock(int tid __maybe_unused)
 220__acquires(torture_rwlock)
 221{
 222	write_lock(&torture_rwlock);
 223	return 0;
 224}
 225
 226static void torture_rwlock_write_delay(struct torture_random_state *trsp)
 227{
 228	const unsigned long shortdelay_us = 2;
 229	const unsigned long longdelay_ms = 100;
 230
 231	/* We want a short delay mostly to emulate likely code, and
 232	 * we want a long delay occasionally to force massive contention.
 233	 */
 234	if (!(torture_random(trsp) %
 235	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
 236		mdelay(longdelay_ms);
 237	else
 238		udelay(shortdelay_us);
 239}
 240
 241static void torture_rwlock_write_unlock(int tid __maybe_unused)
 242__releases(torture_rwlock)
 243{
 244	write_unlock(&torture_rwlock);
 245}
 246
 247static int torture_rwlock_read_lock(int tid __maybe_unused)
 248__acquires(torture_rwlock)
 249{
 250	read_lock(&torture_rwlock);
 251	return 0;
 252}
 253
 254static void torture_rwlock_read_delay(struct torture_random_state *trsp)
 255{
 256	const unsigned long shortdelay_us = 10;
 257	const unsigned long longdelay_ms = 100;
 258
 259	/* We want a short delay mostly to emulate likely code, and
 260	 * we want a long delay occasionally to force massive contention.
 261	 */
 262	if (!(torture_random(trsp) %
 263	      (cxt.nrealreaders_stress * 2000 * longdelay_ms)))
 264		mdelay(longdelay_ms);
 265	else
 266		udelay(shortdelay_us);
 267}
 268
 269static void torture_rwlock_read_unlock(int tid __maybe_unused)
 270__releases(torture_rwlock)
 271{
 272	read_unlock(&torture_rwlock);
 273}
 274
 275static struct lock_torture_ops rw_lock_ops = {
 276	.writelock	= torture_rwlock_write_lock,
 277	.write_delay	= torture_rwlock_write_delay,
 278	.task_boost     = torture_boost_dummy,
 279	.writeunlock	= torture_rwlock_write_unlock,
 280	.readlock       = torture_rwlock_read_lock,
 281	.read_delay     = torture_rwlock_read_delay,
 282	.readunlock     = torture_rwlock_read_unlock,
 283	.name		= "rw_lock"
 284};
 285
 286static int torture_rwlock_write_lock_irq(int tid __maybe_unused)
 287__acquires(torture_rwlock)
 288{
 289	unsigned long flags;
 290
 291	write_lock_irqsave(&torture_rwlock, flags);
 292	cxt.cur_ops->flags = flags;
 293	return 0;
 294}
 295
 296static void torture_rwlock_write_unlock_irq(int tid __maybe_unused)
 297__releases(torture_rwlock)
 298{
 299	write_unlock_irqrestore(&torture_rwlock, cxt.cur_ops->flags);
 300}
 301
 302static int torture_rwlock_read_lock_irq(int tid __maybe_unused)
 303__acquires(torture_rwlock)
 304{
 305	unsigned long flags;
 306
 307	read_lock_irqsave(&torture_rwlock, flags);
 308	cxt.cur_ops->flags = flags;
 309	return 0;
 310}
 311
 312static void torture_rwlock_read_unlock_irq(int tid __maybe_unused)
 313__releases(torture_rwlock)
 314{
 315	read_unlock_irqrestore(&torture_rwlock, cxt.cur_ops->flags);
 316}
 317
 318static struct lock_torture_ops rw_lock_irq_ops = {
 319	.writelock	= torture_rwlock_write_lock_irq,
 320	.write_delay	= torture_rwlock_write_delay,
 321	.task_boost     = torture_boost_dummy,
 322	.writeunlock	= torture_rwlock_write_unlock_irq,
 323	.readlock       = torture_rwlock_read_lock_irq,
 324	.read_delay     = torture_rwlock_read_delay,
 325	.readunlock     = torture_rwlock_read_unlock_irq,
 326	.name		= "rw_lock_irq"
 327};
 328
 329static DEFINE_MUTEX(torture_mutex);
 330
 331static int torture_mutex_lock(int tid __maybe_unused)
 332__acquires(torture_mutex)
 333{
 334	mutex_lock(&torture_mutex);
 335	return 0;
 336}
 337
 338static void torture_mutex_delay(struct torture_random_state *trsp)
 339{
 340	const unsigned long longdelay_ms = 100;
 341
 342	/* We want a long delay occasionally to force massive contention.  */
 343	if (!(torture_random(trsp) %
 344	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
 345		mdelay(longdelay_ms * 5);
 346	else
 347		mdelay(longdelay_ms / 5);
 
 348	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
 349		torture_preempt_schedule();  /* Allow test to be preempted. */
 
 350}
 351
 352static void torture_mutex_unlock(int tid __maybe_unused)
 353__releases(torture_mutex)
 354{
 355	mutex_unlock(&torture_mutex);
 356}
 357
 358static struct lock_torture_ops mutex_lock_ops = {
 359	.writelock	= torture_mutex_lock,
 360	.write_delay	= torture_mutex_delay,
 361	.task_boost     = torture_boost_dummy,
 362	.writeunlock	= torture_mutex_unlock,
 363	.readlock       = NULL,
 364	.read_delay     = NULL,
 365	.readunlock     = NULL,
 366	.name		= "mutex_lock"
 367};
 368
 369#include <linux/ww_mutex.h>
 370/*
 371 * The torture ww_mutexes should belong to the same lock class as
 372 * torture_ww_class to avoid lockdep problem. The ww_mutex_init()
 373 * function is called for initialization to ensure that.
 374 */
 375static DEFINE_WD_CLASS(torture_ww_class);
 376static struct ww_mutex torture_ww_mutex_0, torture_ww_mutex_1, torture_ww_mutex_2;
 377static struct ww_acquire_ctx *ww_acquire_ctxs;
 378
 379static void torture_ww_mutex_init(void)
 380{
 381	ww_mutex_init(&torture_ww_mutex_0, &torture_ww_class);
 382	ww_mutex_init(&torture_ww_mutex_1, &torture_ww_class);
 383	ww_mutex_init(&torture_ww_mutex_2, &torture_ww_class);
 384
 385	ww_acquire_ctxs = kmalloc_array(cxt.nrealwriters_stress,
 386					sizeof(*ww_acquire_ctxs),
 387					GFP_KERNEL);
 388	if (!ww_acquire_ctxs)
 389		VERBOSE_TOROUT_STRING("ww_acquire_ctx: Out of memory");
 390}
 391
 392static void torture_ww_mutex_exit(void)
 393{
 394	kfree(ww_acquire_ctxs);
 395}
 396
 397static int torture_ww_mutex_lock(int tid)
 398__acquires(torture_ww_mutex_0)
 399__acquires(torture_ww_mutex_1)
 400__acquires(torture_ww_mutex_2)
 401{
 402	LIST_HEAD(list);
 403	struct reorder_lock {
 404		struct list_head link;
 405		struct ww_mutex *lock;
 406	} locks[3], *ll, *ln;
 407	struct ww_acquire_ctx *ctx = &ww_acquire_ctxs[tid];
 408
 409	locks[0].lock = &torture_ww_mutex_0;
 410	list_add(&locks[0].link, &list);
 411
 412	locks[1].lock = &torture_ww_mutex_1;
 413	list_add(&locks[1].link, &list);
 414
 415	locks[2].lock = &torture_ww_mutex_2;
 416	list_add(&locks[2].link, &list);
 417
 418	ww_acquire_init(ctx, &torture_ww_class);
 419
 420	list_for_each_entry(ll, &list, link) {
 421		int err;
 422
 423		err = ww_mutex_lock(ll->lock, ctx);
 424		if (!err)
 425			continue;
 426
 427		ln = ll;
 428		list_for_each_entry_continue_reverse(ln, &list, link)
 429			ww_mutex_unlock(ln->lock);
 430
 431		if (err != -EDEADLK)
 432			return err;
 433
 434		ww_mutex_lock_slow(ll->lock, ctx);
 435		list_move(&ll->link, &list);
 436	}
 437
 438	return 0;
 439}
 440
 441static void torture_ww_mutex_unlock(int tid)
 442__releases(torture_ww_mutex_0)
 443__releases(torture_ww_mutex_1)
 444__releases(torture_ww_mutex_2)
 445{
 446	struct ww_acquire_ctx *ctx = &ww_acquire_ctxs[tid];
 447
 448	ww_mutex_unlock(&torture_ww_mutex_0);
 449	ww_mutex_unlock(&torture_ww_mutex_1);
 450	ww_mutex_unlock(&torture_ww_mutex_2);
 451	ww_acquire_fini(ctx);
 452}
 453
 454static struct lock_torture_ops ww_mutex_lock_ops = {
 455	.init		= torture_ww_mutex_init,
 456	.exit		= torture_ww_mutex_exit,
 457	.writelock	= torture_ww_mutex_lock,
 458	.write_delay	= torture_mutex_delay,
 459	.task_boost     = torture_boost_dummy,
 460	.writeunlock	= torture_ww_mutex_unlock,
 461	.readlock       = NULL,
 462	.read_delay     = NULL,
 463	.readunlock     = NULL,
 464	.name		= "ww_mutex_lock"
 465};
 466
 467#ifdef CONFIG_RT_MUTEXES
 468static DEFINE_RT_MUTEX(torture_rtmutex);
 469
 470static int torture_rtmutex_lock(int tid __maybe_unused)
 471__acquires(torture_rtmutex)
 472{
 473	rt_mutex_lock(&torture_rtmutex);
 474	return 0;
 475}
 476
 477static void torture_rtmutex_boost(struct torture_random_state *trsp)
 478{
 
 
 479	const unsigned int factor = 50000; /* yes, quite arbitrary */
 480
 481	if (!rt_task(current)) {
 482		/*
 483		 * Boost priority once every ~50k operations. When the
 484		 * task tries to take the lock, the rtmutex it will account
 485		 * for the new priority, and do any corresponding pi-dance.
 486		 */
 487		if (trsp && !(torture_random(trsp) %
 488			      (cxt.nrealwriters_stress * factor))) {
 489			sched_set_fifo(current);
 
 490		} else /* common case, do nothing */
 491			return;
 492	} else {
 493		/*
 494		 * The task will remain boosted for another ~500k operations,
 495		 * then restored back to its original prio, and so forth.
 496		 *
 497		 * When @trsp is nil, we want to force-reset the task for
 498		 * stopping the kthread.
 499		 */
 500		if (!trsp || !(torture_random(trsp) %
 501			       (cxt.nrealwriters_stress * factor * 2))) {
 502			sched_set_normal(current, 0);
 
 503		} else /* common case, do nothing */
 504			return;
 505	}
 
 
 506}
 507
 508static void torture_rtmutex_delay(struct torture_random_state *trsp)
 509{
 510	const unsigned long shortdelay_us = 2;
 511	const unsigned long longdelay_ms = 100;
 512
 513	/*
 514	 * We want a short delay mostly to emulate likely code, and
 515	 * we want a long delay occasionally to force massive contention.
 516	 */
 517	if (!(torture_random(trsp) %
 518	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
 519		mdelay(longdelay_ms);
 520	if (!(torture_random(trsp) %
 521	      (cxt.nrealwriters_stress * 2 * shortdelay_us)))
 522		udelay(shortdelay_us);
 
 523	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
 524		torture_preempt_schedule();  /* Allow test to be preempted. */
 
 525}
 526
 527static void torture_rtmutex_unlock(int tid __maybe_unused)
 528__releases(torture_rtmutex)
 529{
 530	rt_mutex_unlock(&torture_rtmutex);
 531}
 532
 533static struct lock_torture_ops rtmutex_lock_ops = {
 534	.writelock	= torture_rtmutex_lock,
 535	.write_delay	= torture_rtmutex_delay,
 536	.task_boost     = torture_rtmutex_boost,
 537	.writeunlock	= torture_rtmutex_unlock,
 538	.readlock       = NULL,
 539	.read_delay     = NULL,
 540	.readunlock     = NULL,
 541	.name		= "rtmutex_lock"
 542};
 543#endif
 544
 545static DECLARE_RWSEM(torture_rwsem);
 546static int torture_rwsem_down_write(int tid __maybe_unused)
 547__acquires(torture_rwsem)
 548{
 549	down_write(&torture_rwsem);
 550	return 0;
 551}
 552
 553static void torture_rwsem_write_delay(struct torture_random_state *trsp)
 554{
 555	const unsigned long longdelay_ms = 100;
 556
 557	/* We want a long delay occasionally to force massive contention.  */
 558	if (!(torture_random(trsp) %
 559	      (cxt.nrealwriters_stress * 2000 * longdelay_ms)))
 560		mdelay(longdelay_ms * 10);
 561	else
 562		mdelay(longdelay_ms / 10);
 
 563	if (!(torture_random(trsp) % (cxt.nrealwriters_stress * 20000)))
 564		torture_preempt_schedule();  /* Allow test to be preempted. */
 
 565}
 566
 567static void torture_rwsem_up_write(int tid __maybe_unused)
 568__releases(torture_rwsem)
 569{
 570	up_write(&torture_rwsem);
 571}
 572
 573static int torture_rwsem_down_read(int tid __maybe_unused)
 574__acquires(torture_rwsem)
 575{
 576	down_read(&torture_rwsem);
 577	return 0;
 578}
 579
 580static void torture_rwsem_read_delay(struct torture_random_state *trsp)
 581{
 582	const unsigned long longdelay_ms = 100;
 583
 584	/* We want a long delay occasionally to force massive contention.  */
 585	if (!(torture_random(trsp) %
 586	      (cxt.nrealreaders_stress * 2000 * longdelay_ms)))
 587		mdelay(longdelay_ms * 2);
 588	else
 589		mdelay(longdelay_ms / 2);
 
 590	if (!(torture_random(trsp) % (cxt.nrealreaders_stress * 20000)))
 591		torture_preempt_schedule();  /* Allow test to be preempted. */
 
 592}
 593
 594static void torture_rwsem_up_read(int tid __maybe_unused)
 595__releases(torture_rwsem)
 596{
 597	up_read(&torture_rwsem);
 598}
 599
 600static struct lock_torture_ops rwsem_lock_ops = {
 601	.writelock	= torture_rwsem_down_write,
 602	.write_delay	= torture_rwsem_write_delay,
 603	.task_boost     = torture_boost_dummy,
 604	.writeunlock	= torture_rwsem_up_write,
 605	.readlock       = torture_rwsem_down_read,
 606	.read_delay     = torture_rwsem_read_delay,
 607	.readunlock     = torture_rwsem_up_read,
 608	.name		= "rwsem_lock"
 609};
 610
 611#include <linux/percpu-rwsem.h>
 612static struct percpu_rw_semaphore pcpu_rwsem;
 613
 614static void torture_percpu_rwsem_init(void)
 615{
 616	BUG_ON(percpu_init_rwsem(&pcpu_rwsem));
 617}
 618
 619static void torture_percpu_rwsem_exit(void)
 620{
 621	percpu_free_rwsem(&pcpu_rwsem);
 622}
 623
 624static int torture_percpu_rwsem_down_write(int tid __maybe_unused)
 625__acquires(pcpu_rwsem)
 626{
 627	percpu_down_write(&pcpu_rwsem);
 628	return 0;
 629}
 630
 631static void torture_percpu_rwsem_up_write(int tid __maybe_unused)
 632__releases(pcpu_rwsem)
 633{
 634	percpu_up_write(&pcpu_rwsem);
 635}
 636
 637static int torture_percpu_rwsem_down_read(int tid __maybe_unused)
 638__acquires(pcpu_rwsem)
 639{
 640	percpu_down_read(&pcpu_rwsem);
 641	return 0;
 642}
 643
 644static void torture_percpu_rwsem_up_read(int tid __maybe_unused)
 645__releases(pcpu_rwsem)
 646{
 647	percpu_up_read(&pcpu_rwsem);
 648}
 649
 650static struct lock_torture_ops percpu_rwsem_lock_ops = {
 651	.init		= torture_percpu_rwsem_init,
 652	.exit		= torture_percpu_rwsem_exit,
 653	.writelock	= torture_percpu_rwsem_down_write,
 654	.write_delay	= torture_rwsem_write_delay,
 655	.task_boost     = torture_boost_dummy,
 656	.writeunlock	= torture_percpu_rwsem_up_write,
 657	.readlock       = torture_percpu_rwsem_down_read,
 658	.read_delay     = torture_rwsem_read_delay,
 659	.readunlock     = torture_percpu_rwsem_up_read,
 660	.name		= "percpu_rwsem_lock"
 661};
 662
 663/*
 664 * Lock torture writer kthread.  Repeatedly acquires and releases
 665 * the lock, checking for duplicate acquisitions.
 666 */
 667static int lock_torture_writer(void *arg)
 668{
 669	struct lock_stress_stats *lwsp = arg;
 670	int tid = lwsp - cxt.lwsa;
 671	DEFINE_TORTURE_RANDOM(rand);
 672
 673	VERBOSE_TOROUT_STRING("lock_torture_writer task started");
 674	set_user_nice(current, MAX_NICE);
 675
 676	do {
 677		if ((torture_random(&rand) & 0xfffff) == 0)
 678			schedule_timeout_uninterruptible(1);
 679
 680		cxt.cur_ops->task_boost(&rand);
 681		cxt.cur_ops->writelock(tid);
 682		if (WARN_ON_ONCE(lock_is_write_held))
 683			lwsp->n_lock_fail++;
 684		lock_is_write_held = true;
 685		if (WARN_ON_ONCE(atomic_read(&lock_is_read_held)))
 686			lwsp->n_lock_fail++; /* rare, but... */
 687
 688		lwsp->n_lock_acquired++;
 689		cxt.cur_ops->write_delay(&rand);
 690		lock_is_write_held = false;
 691		WRITE_ONCE(last_lock_release, jiffies);
 692		cxt.cur_ops->writeunlock(tid);
 693
 694		stutter_wait("lock_torture_writer");
 695	} while (!torture_must_stop());
 696
 697	cxt.cur_ops->task_boost(NULL); /* reset prio */
 698	torture_kthread_stopping("lock_torture_writer");
 699	return 0;
 700}
 701
 702/*
 703 * Lock torture reader kthread.  Repeatedly acquires and releases
 704 * the reader lock.
 705 */
 706static int lock_torture_reader(void *arg)
 707{
 708	struct lock_stress_stats *lrsp = arg;
 709	int tid = lrsp - cxt.lrsa;
 710	DEFINE_TORTURE_RANDOM(rand);
 711
 712	VERBOSE_TOROUT_STRING("lock_torture_reader task started");
 713	set_user_nice(current, MAX_NICE);
 714
 715	do {
 716		if ((torture_random(&rand) & 0xfffff) == 0)
 717			schedule_timeout_uninterruptible(1);
 718
 719		cxt.cur_ops->readlock(tid);
 720		atomic_inc(&lock_is_read_held);
 721		if (WARN_ON_ONCE(lock_is_write_held))
 722			lrsp->n_lock_fail++; /* rare, but... */
 723
 724		lrsp->n_lock_acquired++;
 725		cxt.cur_ops->read_delay(&rand);
 726		atomic_dec(&lock_is_read_held);
 727		cxt.cur_ops->readunlock(tid);
 728
 729		stutter_wait("lock_torture_reader");
 730	} while (!torture_must_stop());
 731	torture_kthread_stopping("lock_torture_reader");
 732	return 0;
 733}
 734
 735/*
 736 * Create an lock-torture-statistics message in the specified buffer.
 737 */
 738static void __torture_print_stats(char *page,
 739				  struct lock_stress_stats *statp, bool write)
 740{
 741	long cur;
 742	bool fail = false;
 743	int i, n_stress;
 744	long max = 0, min = statp ? data_race(statp[0].n_lock_acquired) : 0;
 
 745	long long sum = 0;
 746
 747	n_stress = write ? cxt.nrealwriters_stress : cxt.nrealreaders_stress;
 748	for (i = 0; i < n_stress; i++) {
 749		if (data_race(statp[i].n_lock_fail))
 750			fail = true;
 751		cur = data_race(statp[i].n_lock_acquired);
 752		sum += cur;
 753		if (max < cur)
 754			max = cur;
 755		if (min > cur)
 756			min = cur;
 757	}
 758	page += sprintf(page,
 759			"%s:  Total: %lld  Max/Min: %ld/%ld %s  Fail: %d %s\n",
 760			write ? "Writes" : "Reads ",
 761			sum, max, min,
 762			!onoff_interval && max / 2 > min ? "???" : "",
 763			fail, fail ? "!!!" : "");
 764	if (fail)
 765		atomic_inc(&cxt.n_lock_torture_errors);
 766}
 767
 768/*
 769 * Print torture statistics.  Caller must ensure that there is only one
 770 * call to this function at a given time!!!  This is normally accomplished
 771 * by relying on the module system to only have one copy of the module
 772 * loaded, and then by giving the lock_torture_stats kthread full control
 773 * (or the init/cleanup functions when lock_torture_stats thread is not
 774 * running).
 775 */
 776static void lock_torture_stats_print(void)
 777{
 778	int size = cxt.nrealwriters_stress * 200 + 8192;
 779	char *buf;
 780
 781	if (cxt.cur_ops->readlock)
 782		size += cxt.nrealreaders_stress * 200 + 8192;
 783
 784	buf = kmalloc(size, GFP_KERNEL);
 785	if (!buf) {
 786		pr_err("lock_torture_stats_print: Out of memory, need: %d",
 787		       size);
 788		return;
 789	}
 790
 791	__torture_print_stats(buf, cxt.lwsa, true);
 792	pr_alert("%s", buf);
 793	kfree(buf);
 794
 795	if (cxt.cur_ops->readlock) {
 796		buf = kmalloc(size, GFP_KERNEL);
 797		if (!buf) {
 798			pr_err("lock_torture_stats_print: Out of memory, need: %d",
 799			       size);
 800			return;
 801		}
 802
 803		__torture_print_stats(buf, cxt.lrsa, false);
 804		pr_alert("%s", buf);
 805		kfree(buf);
 806	}
 807}
 808
 809/*
 810 * Periodically prints torture statistics, if periodic statistics printing
 811 * was specified via the stat_interval module parameter.
 812 *
 813 * No need to worry about fullstop here, since this one doesn't reference
 814 * volatile state or register callbacks.
 815 */
 816static int lock_torture_stats(void *arg)
 817{
 818	VERBOSE_TOROUT_STRING("lock_torture_stats task started");
 819	do {
 820		schedule_timeout_interruptible(stat_interval * HZ);
 821		lock_torture_stats_print();
 822		torture_shutdown_absorb("lock_torture_stats");
 823	} while (!torture_must_stop());
 824	torture_kthread_stopping("lock_torture_stats");
 825	return 0;
 826}
 827
 828static inline void
 829lock_torture_print_module_parms(struct lock_torture_ops *cur_ops,
 830				const char *tag)
 831{
 832	pr_alert("%s" TORTURE_FLAG
 833		 "--- %s%s: nwriters_stress=%d nreaders_stress=%d stat_interval=%d verbose=%d shuffle_interval=%d stutter=%d shutdown_secs=%d onoff_interval=%d onoff_holdoff=%d\n",
 834		 torture_type, tag, cxt.debug_lock ? " [debug]": "",
 835		 cxt.nrealwriters_stress, cxt.nrealreaders_stress, stat_interval,
 836		 verbose, shuffle_interval, stutter, shutdown_secs,
 837		 onoff_interval, onoff_holdoff);
 838}
 839
 840static void lock_torture_cleanup(void)
 841{
 842	int i;
 843
 844	if (torture_cleanup_begin())
 845		return;
 846
 847	/*
 848	 * Indicates early cleanup, meaning that the test has not run,
 849	 * such as when passing bogus args when loading the module.
 850	 * However cxt->cur_ops.init() may have been invoked, so beside
 851	 * perform the underlying torture-specific cleanups, cur_ops.exit()
 852	 * will be invoked if needed.
 853	 */
 854	if (!cxt.lwsa && !cxt.lrsa)
 855		goto end;
 856
 857	if (writer_tasks) {
 858		for (i = 0; i < cxt.nrealwriters_stress; i++)
 859			torture_stop_kthread(lock_torture_writer,
 860					     writer_tasks[i]);
 861		kfree(writer_tasks);
 862		writer_tasks = NULL;
 863	}
 864
 865	if (reader_tasks) {
 866		for (i = 0; i < cxt.nrealreaders_stress; i++)
 867			torture_stop_kthread(lock_torture_reader,
 868					     reader_tasks[i]);
 869		kfree(reader_tasks);
 870		reader_tasks = NULL;
 871	}
 872
 873	torture_stop_kthread(lock_torture_stats, stats_task);
 874	lock_torture_stats_print();  /* -After- the stats thread is stopped! */
 875
 876	if (atomic_read(&cxt.n_lock_torture_errors))
 877		lock_torture_print_module_parms(cxt.cur_ops,
 878						"End of test: FAILURE");
 879	else if (torture_onoff_failures())
 880		lock_torture_print_module_parms(cxt.cur_ops,
 881						"End of test: LOCK_HOTPLUG");
 882	else
 883		lock_torture_print_module_parms(cxt.cur_ops,
 884						"End of test: SUCCESS");
 885
 886	kfree(cxt.lwsa);
 887	cxt.lwsa = NULL;
 888	kfree(cxt.lrsa);
 889	cxt.lrsa = NULL;
 890
 891end:
 892	if (cxt.init_called) {
 893		if (cxt.cur_ops->exit)
 894			cxt.cur_ops->exit();
 895		cxt.init_called = false;
 896	}
 897	torture_cleanup_end();
 898}
 899
 900static int __init lock_torture_init(void)
 901{
 902	int i, j;
 903	int firsterr = 0;
 904	static struct lock_torture_ops *torture_ops[] = {
 905		&lock_busted_ops,
 906		&spin_lock_ops, &spin_lock_irq_ops,
 907		&rw_lock_ops, &rw_lock_irq_ops,
 908		&mutex_lock_ops,
 909		&ww_mutex_lock_ops,
 910#ifdef CONFIG_RT_MUTEXES
 911		&rtmutex_lock_ops,
 912#endif
 913		&rwsem_lock_ops,
 914		&percpu_rwsem_lock_ops,
 915	};
 916
 917	if (!torture_init_begin(torture_type, verbose))
 918		return -EBUSY;
 919
 920	/* Process args and tell the world that the torturer is on the job. */
 921	for (i = 0; i < ARRAY_SIZE(torture_ops); i++) {
 922		cxt.cur_ops = torture_ops[i];
 923		if (strcmp(torture_type, cxt.cur_ops->name) == 0)
 924			break;
 925	}
 926	if (i == ARRAY_SIZE(torture_ops)) {
 927		pr_alert("lock-torture: invalid torture type: \"%s\"\n",
 928			 torture_type);
 929		pr_alert("lock-torture types:");
 930		for (i = 0; i < ARRAY_SIZE(torture_ops); i++)
 931			pr_alert(" %s", torture_ops[i]->name);
 932		pr_alert("\n");
 933		firsterr = -EINVAL;
 934		goto unwind;
 935	}
 936
 937	if (nwriters_stress == 0 &&
 938	    (!cxt.cur_ops->readlock || nreaders_stress == 0)) {
 939		pr_alert("lock-torture: must run at least one locking thread\n");
 940		firsterr = -EINVAL;
 941		goto unwind;
 942	}
 943
 944	if (nwriters_stress >= 0)
 945		cxt.nrealwriters_stress = nwriters_stress;
 946	else
 947		cxt.nrealwriters_stress = 2 * num_online_cpus();
 948
 949	if (cxt.cur_ops->init) {
 950		cxt.cur_ops->init();
 951		cxt.init_called = true;
 952	}
 953
 954#ifdef CONFIG_DEBUG_MUTEXES
 955	if (str_has_prefix(torture_type, "mutex"))
 956		cxt.debug_lock = true;
 957#endif
 958#ifdef CONFIG_DEBUG_RT_MUTEXES
 959	if (str_has_prefix(torture_type, "rtmutex"))
 960		cxt.debug_lock = true;
 961#endif
 962#ifdef CONFIG_DEBUG_SPINLOCK
 963	if ((str_has_prefix(torture_type, "spin")) ||
 964	    (str_has_prefix(torture_type, "rw_lock")))
 965		cxt.debug_lock = true;
 966#endif
 967
 968	/* Initialize the statistics so that each run gets its own numbers. */
 969	if (nwriters_stress) {
 970		lock_is_write_held = false;
 971		cxt.lwsa = kmalloc_array(cxt.nrealwriters_stress,
 972					 sizeof(*cxt.lwsa),
 973					 GFP_KERNEL);
 974		if (cxt.lwsa == NULL) {
 975			VERBOSE_TOROUT_STRING("cxt.lwsa: Out of memory");
 976			firsterr = -ENOMEM;
 977			goto unwind;
 978		}
 979
 980		for (i = 0; i < cxt.nrealwriters_stress; i++) {
 981			cxt.lwsa[i].n_lock_fail = 0;
 982			cxt.lwsa[i].n_lock_acquired = 0;
 983		}
 
 
 
 
 
 
 984	}
 985
 986	if (cxt.cur_ops->readlock) {
 987		if (nreaders_stress >= 0)
 988			cxt.nrealreaders_stress = nreaders_stress;
 989		else {
 990			/*
 991			 * By default distribute evenly the number of
 992			 * readers and writers. We still run the same number
 993			 * of threads as the writer-only locks default.
 994			 */
 995			if (nwriters_stress < 0) /* user doesn't care */
 996				cxt.nrealwriters_stress = num_online_cpus();
 997			cxt.nrealreaders_stress = cxt.nrealwriters_stress;
 998		}
 999
1000		if (nreaders_stress) {
1001			cxt.lrsa = kmalloc_array(cxt.nrealreaders_stress,
1002						 sizeof(*cxt.lrsa),
1003						 GFP_KERNEL);
1004			if (cxt.lrsa == NULL) {
1005				VERBOSE_TOROUT_STRING("cxt.lrsa: Out of memory");
1006				firsterr = -ENOMEM;
1007				kfree(cxt.lwsa);
1008				cxt.lwsa = NULL;
1009				goto unwind;
1010			}
1011
1012			for (i = 0; i < cxt.nrealreaders_stress; i++) {
1013				cxt.lrsa[i].n_lock_fail = 0;
1014				cxt.lrsa[i].n_lock_acquired = 0;
1015			}
1016		}
1017	}
1018
1019	lock_torture_print_module_parms(cxt.cur_ops, "Start of test");
1020
1021	/* Prepare torture context. */
1022	if (onoff_interval > 0) {
1023		firsterr = torture_onoff_init(onoff_holdoff * HZ,
1024					      onoff_interval * HZ, NULL);
1025		if (torture_init_error(firsterr))
1026			goto unwind;
1027	}
1028	if (shuffle_interval > 0) {
1029		firsterr = torture_shuffle_init(shuffle_interval);
1030		if (torture_init_error(firsterr))
1031			goto unwind;
1032	}
1033	if (shutdown_secs > 0) {
1034		firsterr = torture_shutdown_init(shutdown_secs,
1035						 lock_torture_cleanup);
1036		if (torture_init_error(firsterr))
1037			goto unwind;
1038	}
1039	if (stutter > 0) {
1040		firsterr = torture_stutter_init(stutter, stutter);
1041		if (torture_init_error(firsterr))
1042			goto unwind;
1043	}
1044
1045	if (nwriters_stress) {
1046		writer_tasks = kcalloc(cxt.nrealwriters_stress,
1047				       sizeof(writer_tasks[0]),
1048				       GFP_KERNEL);
1049		if (writer_tasks == NULL) {
1050			TOROUT_ERRSTRING("writer_tasks: Out of memory");
1051			firsterr = -ENOMEM;
1052			goto unwind;
1053		}
1054	}
1055
1056	if (cxt.cur_ops->readlock) {
1057		reader_tasks = kcalloc(cxt.nrealreaders_stress,
1058				       sizeof(reader_tasks[0]),
1059				       GFP_KERNEL);
1060		if (reader_tasks == NULL) {
1061			TOROUT_ERRSTRING("reader_tasks: Out of memory");
1062			kfree(writer_tasks);
1063			writer_tasks = NULL;
1064			firsterr = -ENOMEM;
1065			goto unwind;
1066		}
1067	}
1068
1069	/*
1070	 * Create the kthreads and start torturing (oh, those poor little locks).
1071	 *
1072	 * TODO: Note that we interleave writers with readers, giving writers a
1073	 * slight advantage, by creating its kthread first. This can be modified
1074	 * for very specific needs, or even let the user choose the policy, if
1075	 * ever wanted.
1076	 */
1077	for (i = 0, j = 0; i < cxt.nrealwriters_stress ||
1078		    j < cxt.nrealreaders_stress; i++, j++) {
1079		if (i >= cxt.nrealwriters_stress)
1080			goto create_reader;
1081
1082		/* Create writer. */
1083		firsterr = torture_create_kthread(lock_torture_writer, &cxt.lwsa[i],
1084						  writer_tasks[i]);
1085		if (torture_init_error(firsterr))
1086			goto unwind;
1087
1088	create_reader:
1089		if (cxt.cur_ops->readlock == NULL || (j >= cxt.nrealreaders_stress))
1090			continue;
1091		/* Create reader. */
1092		firsterr = torture_create_kthread(lock_torture_reader, &cxt.lrsa[j],
1093						  reader_tasks[j]);
1094		if (torture_init_error(firsterr))
1095			goto unwind;
1096	}
1097	if (stat_interval > 0) {
1098		firsterr = torture_create_kthread(lock_torture_stats, NULL,
1099						  stats_task);
1100		if (torture_init_error(firsterr))
1101			goto unwind;
1102	}
1103	torture_init_end();
1104	return 0;
1105
1106unwind:
1107	torture_init_end();
1108	lock_torture_cleanup();
1109	if (shutdown_secs) {
1110		WARN_ON(!IS_MODULE(CONFIG_LOCK_TORTURE_TEST));
1111		kernel_power_off();
1112	}
1113	return firsterr;
1114}
1115
1116module_init(lock_torture_init);
1117module_exit(lock_torture_cleanup);