summaryrefslogtreecommitdiff
path: root/lib/cpu-notifier-error-inject.c
blob: 0e2c9a1e958a490cdbbaae732f379c5ee3b26d87 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/cpu.h>

#include "notifier-error-inject.h"

static int priority;
module_param(priority, int, 0);
MODULE_PARM_DESC(priority, "specify cpu notifier priority");

#define UP_PREPARE 0
#define UP_PREPARE_FROZEN 0
#define DOWN_PREPARE 0
#define DOWN_PREPARE_FROZEN 0

static struct notifier_err_inject cpu_notifier_err_inject = {
	.actions = {
		{ NOTIFIER_ERR_INJECT_ACTION(UP_PREPARE) },
		{ NOTIFIER_ERR_INJECT_ACTION(UP_PREPARE_FROZEN) },
		{ NOTIFIER_ERR_INJECT_ACTION(DOWN_PREPARE) },
		{ NOTIFIER_ERR_INJECT_ACTION(DOWN_PREPARE_FROZEN) },
		{}
	}
};

static int notf_err_handle(struct notifier_err_inject_action *action)
{
	int ret;

	ret = action->error;
	if (ret)
		pr_info("Injecting error (%d) to %s\n", ret, action->name);
	return ret;
}

static int notf_err_inj_up_prepare(unsigned int cpu)
{
	if (!cpuhp_tasks_frozen)
		return notf_err_handle(&cpu_notifier_err_inject.actions[0]);
	else
		return notf_err_handle(&cpu_notifier_err_inject.actions[1]);
}

static int notf_err_inj_dead(unsigned int cpu)
{
	if (!cpuhp_tasks_frozen)
		return notf_err_handle(&cpu_notifier_err_inject.actions[2]);
	else
		return notf_err_handle(&cpu_notifier_err_inject.actions[3]);
}

static struct dentry *dir;

static int err_inject_init(void)
{
	int err;

	dir = notifier_err_inject_init("cpu", notifier_err_inject_dir,
					&cpu_notifier_err_inject, priority);
	if (IS_ERR(dir))
		return PTR_ERR(dir);

	err = cpuhp_setup_state_nocalls(CPUHP_NOTF_ERR_INJ_PREPARE,
					"cpu-err-notif:prepare",
					notf_err_inj_up_prepare,
					notf_err_inj_dead);
	if (err)
		debugfs_remove_recursive(dir);

	return err;
}

static void err_inject_exit(void)
{
	cpuhp_remove_state_nocalls(CPUHP_NOTF_ERR_INJ_PREPARE);
	debugfs_remove_recursive(dir);
}

module_init(err_inject_init);
module_exit(err_inject_exit);

MODULE_DESCRIPTION("CPU notifier error injection module");
MODULE_LICENSE("GPL");
MODULE_AUTHOR("Akinobu Mita <akinobu.mita@gmail.com>");