rcutorture: Abstract torture_create_kthread()

Creation of kthreads is not RCU-specific, so this commit abstracts
out torture_create_kthread(), saving a few tens of lines of code in
the process.

This change requires modifying VERBOSE_TOROUT_ERRSTRING() to take a
non-const string, so that _torture_create_kthread() can avoid an
open-coded substitute.

Signed-off-by: Paul E. McKenney <paulmck@linux.vnet.ibm.com>
Reviewed-by: Josh Triplett <josh@joshtriplett.org>
This commit is contained in:
Paul E. McKenney 2014-02-03 11:52:27 -08:00
parent bc8f83e2c0
commit 47cf29b9e7
3 changed files with 60 additions and 126 deletions

View File

@ -47,7 +47,7 @@
#define VERBOSE_TOROUT_STRING(s) \ #define VERBOSE_TOROUT_STRING(s) \
do { if (verbose) pr_alert("%s" TORTURE_FLAG " %s\n", torture_type, s); } while (0) do { if (verbose) pr_alert("%s" TORTURE_FLAG " %s\n", torture_type, s); } while (0)
#define VERBOSE_TOROUT_ERRSTRING(s) \ #define VERBOSE_TOROUT_ERRSTRING(s) \
do { if (verbose) pr_alert("%s" TORTURE_FLAG "!!! " s "\n", torture_type); } while (0) do { if (verbose) pr_alert("%s" TORTURE_FLAG "!!! %s\n", torture_type, s); } while (0)
/* Definitions for a non-string torture-test module parameter. */ /* Definitions for a non-string torture-test module parameter. */
#define torture_parm(type, name, init, msg) \ #define torture_parm(type, name, init, msg) \
@ -89,5 +89,11 @@ bool torture_cleanup(void);
bool torture_must_stop(void); bool torture_must_stop(void);
bool torture_must_stop_irq(void); bool torture_must_stop_irq(void);
void torture_kthread_stopping(char *title); void torture_kthread_stopping(char *title);
int _torture_create_kthread(int (*fn)(void *arg), void *arg, char *s, char *m,
char *f, struct task_struct **tp);
#define torture_create_kthread(n, arg, tp) \
_torture_create_kthread(n, (arg), #n, "Creating " #n " task", \
"Failed to create " #n, &(tp))
#endif /* __LINUX_TORTURE_H */ #endif /* __LINUX_TORTURE_H */

View File

@ -1105,19 +1105,9 @@ static int rcu_torture_stall(void *args)
/* Spawn CPU-stall kthread, if stall_cpu specified. */ /* Spawn CPU-stall kthread, if stall_cpu specified. */
static int __init rcu_torture_stall_init(void) static int __init rcu_torture_stall_init(void)
{ {
int ret;
if (stall_cpu <= 0) if (stall_cpu <= 0)
return 0; return 0;
VERBOSE_TOROUT_STRING("Creating rcu_torture_stall task"); return torture_create_kthread(rcu_torture_stall, NULL, stall_task);
stall_task = kthread_run(rcu_torture_stall, NULL, "rcu_torture_stall");
if (IS_ERR(stall_task)) {
ret = PTR_ERR(stall_task);
stall_task = NULL;
return ret;
}
torture_shuffle_task_register(stall_task);
return 0;
} }
/* Clean up after the CPU-stall kthread, if one was spawned. */ /* Clean up after the CPU-stall kthread, if one was spawned. */
@ -1226,29 +1216,13 @@ static int rcu_torture_barrier_init(void)
return -ENOMEM; return -ENOMEM;
for (i = 0; i < n_barrier_cbs; i++) { for (i = 0; i < n_barrier_cbs; i++) {
init_waitqueue_head(&barrier_cbs_wq[i]); init_waitqueue_head(&barrier_cbs_wq[i]);
VERBOSE_TOROUT_STRING("Creating rcu_torture_barrier_cbs task"); ret = torture_create_kthread(rcu_torture_barrier_cbs,
barrier_cbs_tasks[i] = kthread_run(rcu_torture_barrier_cbs,
(void *)(long)i, (void *)(long)i,
"rcu_torture_barrier_cbs"); barrier_cbs_tasks[i]);
if (IS_ERR(barrier_cbs_tasks[i])) { if (ret)
ret = PTR_ERR(barrier_cbs_tasks[i]);
VERBOSE_TOROUT_ERRSTRING("Failed to create rcu_torture_barrier_cbs");
barrier_cbs_tasks[i] = NULL;
return ret; return ret;
} }
torture_shuffle_task_register(barrier_cbs_tasks[i]); return torture_create_kthread(rcu_torture_barrier, NULL, barrier_task);
}
VERBOSE_TOROUT_STRING("Creating rcu_torture_barrier task");
barrier_task = kthread_run(rcu_torture_barrier, NULL,
"rcu_torture_barrier");
if (IS_ERR(barrier_task)) {
ret = PTR_ERR(barrier_task);
VERBOSE_TOROUT_ERRSTRING("Failed to create rcu_torture_barrier");
barrier_task = NULL;
return ret;
}
torture_shuffle_task_register(barrier_task);
return 0;
} }
/* Clean up after RCU barrier testing. */ /* Clean up after RCU barrier testing. */
@ -1516,17 +1490,10 @@ rcu_torture_init(void)
/* Start up the kthreads. */ /* Start up the kthreads. */
VERBOSE_TOROUT_STRING("Creating rcu_torture_writer task"); firsterr = torture_create_kthread(rcu_torture_writer, NULL,
writer_task = kthread_create(rcu_torture_writer, NULL, writer_task);
"rcu_torture_writer"); if (firsterr)
if (IS_ERR(writer_task)) {
firsterr = PTR_ERR(writer_task);
VERBOSE_TOROUT_ERRSTRING("Failed to create writer");
writer_task = NULL;
goto unwind; goto unwind;
}
torture_shuffle_task_register(writer_task);
wake_up_process(writer_task);
fakewriter_tasks = kzalloc(nfakewriters * sizeof(fakewriter_tasks[0]), fakewriter_tasks = kzalloc(nfakewriters * sizeof(fakewriter_tasks[0]),
GFP_KERNEL); GFP_KERNEL);
if (fakewriter_tasks == NULL) { if (fakewriter_tasks == NULL) {
@ -1535,17 +1502,11 @@ rcu_torture_init(void)
goto unwind; goto unwind;
} }
for (i = 0; i < nfakewriters; i++) { for (i = 0; i < nfakewriters; i++) {
VERBOSE_TOROUT_STRING("Creating rcu_torture_fakewriter task"); firsterr = torture_create_kthread(rcu_torture_fakewriter,
fakewriter_tasks[i] = kthread_run(rcu_torture_fakewriter, NULL, NULL, fakewriter_tasks[i]);
"rcu_torture_fakewriter"); if (firsterr)
if (IS_ERR(fakewriter_tasks[i])) {
firsterr = PTR_ERR(fakewriter_tasks[i]);
VERBOSE_TOROUT_ERRSTRING("Failed to create fakewriter");
fakewriter_tasks[i] = NULL;
goto unwind; goto unwind;
} }
torture_shuffle_task_register(fakewriter_tasks[i]);
}
reader_tasks = kzalloc(nrealreaders * sizeof(reader_tasks[0]), reader_tasks = kzalloc(nrealreaders * sizeof(reader_tasks[0]),
GFP_KERNEL); GFP_KERNEL);
if (reader_tasks == NULL) { if (reader_tasks == NULL) {
@ -1554,29 +1515,17 @@ rcu_torture_init(void)
goto unwind; goto unwind;
} }
for (i = 0; i < nrealreaders; i++) { for (i = 0; i < nrealreaders; i++) {
VERBOSE_TOROUT_STRING("Creating rcu_torture_reader task"); firsterr = torture_create_kthread(rcu_torture_reader, NULL,
reader_tasks[i] = kthread_run(rcu_torture_reader, NULL, reader_tasks[i]);
"rcu_torture_reader"); if (firsterr)
if (IS_ERR(reader_tasks[i])) {
firsterr = PTR_ERR(reader_tasks[i]);
VERBOSE_TOROUT_ERRSTRING("Failed to create reader");
reader_tasks[i] = NULL;
goto unwind; goto unwind;
} }
torture_shuffle_task_register(reader_tasks[i]);
}
if (stat_interval > 0) { if (stat_interval > 0) {
VERBOSE_TOROUT_STRING("Creating rcu_torture_stats task"); firsterr = torture_create_kthread(rcu_torture_stats, NULL,
stats_task = kthread_run(rcu_torture_stats, NULL, stats_task);
"rcu_torture_stats"); if (firsterr)
if (IS_ERR(stats_task)) {
firsterr = PTR_ERR(stats_task);
VERBOSE_TOROUT_ERRSTRING("Failed to create stats");
stats_task = NULL;
goto unwind; goto unwind;
} }
torture_shuffle_task_register(stats_task);
}
if (test_no_idle_hz) { if (test_no_idle_hz) {
firsterr = torture_shuffle_init(shuffle_interval * HZ); firsterr = torture_shuffle_init(shuffle_interval * HZ);
if (firsterr) if (firsterr)
@ -1593,17 +1542,10 @@ rcu_torture_init(void)
fqs_duration = 0; fqs_duration = 0;
if (fqs_duration) { if (fqs_duration) {
/* Create the fqs thread */ /* Create the fqs thread */
VERBOSE_TOROUT_STRING("Creating rcu_torture_fqs task"); torture_create_kthread(rcu_torture_fqs, NULL, fqs_task);
fqs_task = kthread_run(rcu_torture_fqs, NULL, if (firsterr)
"rcu_torture_fqs");
if (IS_ERR(fqs_task)) {
firsterr = PTR_ERR(fqs_task);
VERBOSE_TOROUT_ERRSTRING("Failed to create fqs");
fqs_task = NULL;
goto unwind; goto unwind;
} }
torture_shuffle_task_register(fqs_task);
}
if (test_boost_interval < 1) if (test_boost_interval < 1)
test_boost_interval = 1; test_boost_interval = 1;
if (test_boost_duration < 2) if (test_boost_duration < 2)

View File

@ -180,23 +180,16 @@ torture_onoff(void *arg)
*/ */
int torture_onoff_init(long ooholdoff, long oointerval) int torture_onoff_init(long ooholdoff, long oointerval)
{ {
#ifdef CONFIG_HOTPLUG_CPU int ret = 0;
int ret;
#ifdef CONFIG_HOTPLUG_CPU
onoff_holdoff = ooholdoff; onoff_holdoff = ooholdoff;
onoff_interval = oointerval; onoff_interval = oointerval;
if (onoff_interval <= 0) if (onoff_interval <= 0)
return 0; return 0;
VERBOSE_TOROUT_STRING("Creating torture_onoff task"); ret = torture_create_kthread(torture_onoff, NULL, onoff_task);
onoff_task = kthread_run(torture_onoff, NULL, "torture_onoff");
if (IS_ERR(onoff_task)) {
ret = PTR_ERR(onoff_task);
onoff_task = NULL;
return ret;
}
torture_shuffle_task_register(onoff_task);
#endif /* #ifdef CONFIG_HOTPLUG_CPU */ #endif /* #ifdef CONFIG_HOTPLUG_CPU */
return 0; return ret;
} }
EXPORT_SYMBOL_GPL(torture_onoff_init); EXPORT_SYMBOL_GPL(torture_onoff_init);
@ -379,8 +372,6 @@ static int torture_shuffle(void *arg)
*/ */
int torture_shuffle_init(long shuffint) int torture_shuffle_init(long shuffint)
{ {
int ret;
shuffle_interval = shuffint; shuffle_interval = shuffint;
shuffle_idle_cpu = -1; shuffle_idle_cpu = -1;
@ -391,17 +382,7 @@ int torture_shuffle_init(long shuffint)
} }
/* Create the shuffler thread */ /* Create the shuffler thread */
VERBOSE_TOROUT_STRING("Creating torture_shuffle task"); return torture_create_kthread(torture_shuffle, NULL, shuffler_task);
shuffler_task = kthread_run(torture_shuffle, NULL, "torture_shuffle");
if (IS_ERR(shuffler_task)) {
ret = PTR_ERR(shuffler_task);
free_cpumask_var(shuffle_tmp_mask);
VERBOSE_TOROUT_ERRSTRING("Failed to create shuffler");
shuffler_task = NULL;
return ret;
}
torture_shuffle_task_register(shuffler_task);
return 0;
} }
EXPORT_SYMBOL_GPL(torture_shuffle_init); EXPORT_SYMBOL_GPL(torture_shuffle_init);
@ -483,26 +464,17 @@ static int torture_shutdown(void *arg)
*/ */
int torture_shutdown_init(int ssecs, void (*cleanup)(void)) int torture_shutdown_init(int ssecs, void (*cleanup)(void))
{ {
int ret; int ret = 0;
shutdown_secs = ssecs; shutdown_secs = ssecs;
torture_shutdown_hook = cleanup; torture_shutdown_hook = cleanup;
if (shutdown_secs > 0) { if (shutdown_secs > 0) {
VERBOSE_TOROUT_STRING("Creating torture_shutdown task");
shutdown_time = jiffies + shutdown_secs * HZ; shutdown_time = jiffies + shutdown_secs * HZ;
shutdown_task = kthread_create(torture_shutdown, NULL, ret = torture_create_kthread(torture_shutdown, NULL,
"torture_shutdown"); shutdown_task);
if (IS_ERR(shutdown_task)) { }
ret = PTR_ERR(shutdown_task);
VERBOSE_TOROUT_ERRSTRING("Failed to create shutdown");
shutdown_task = NULL;
return ret; return ret;
} }
torture_shuffle_task_register(shutdown_task);
wake_up_process(shutdown_task);
}
return 0;
}
EXPORT_SYMBOL_GPL(torture_shutdown_init); EXPORT_SYMBOL_GPL(torture_shutdown_init);
/* /*
@ -595,17 +567,9 @@ int torture_stutter_init(int s)
int ret; int ret;
stutter = s; stutter = s;
VERBOSE_TOROUT_STRING("Creating torture_stutter task"); ret = torture_create_kthread(torture_stutter, NULL, stutter_task);
stutter_task = kthread_run(torture_stutter, NULL, "torture_stutter");
if (IS_ERR(stutter_task)) {
ret = PTR_ERR(stutter_task);
VERBOSE_TOROUT_ERRSTRING("Failed to create stutter");
stutter_task = NULL;
return ret; return ret;
} }
torture_shuffle_task_register(stutter_task);
return 0;
}
EXPORT_SYMBOL_GPL(torture_stutter_init); EXPORT_SYMBOL_GPL(torture_stutter_init);
/* /*
@ -714,3 +678,25 @@ void torture_kthread_stopping(char *title)
} }
} }
EXPORT_SYMBOL_GPL(torture_kthread_stopping); EXPORT_SYMBOL_GPL(torture_kthread_stopping);
/*
* Create a generic torture kthread that is immediately runnable. If you
* need the kthread to be stopped so that you can do something to it before
* it starts, you will need to open-code your own.
*/
int _torture_create_kthread(int (*fn)(void *arg), void *arg, char *s, char *m,
char *f, struct task_struct **tp)
{
int ret = 0;
VERBOSE_TOROUT_STRING(m);
*tp = kthread_run(fn, arg, s);
if (IS_ERR(*tp)) {
ret = PTR_ERR(*tp);
VERBOSE_TOROUT_ERRSTRING(f);
*tp = NULL;
}
torture_shuffle_task_register(*tp);
return ret;
}
EXPORT_SYMBOL_GPL(_torture_create_kthread);