blob: e6239bf0b0dfd6e73b871030e55dc2b772675fc9
1 | #include <linux/kernel.h> |
2 | #include <linux/module.h> |
3 | #include <linux/memory.h> |
4 | |
5 | #include "notifier-error-inject.h" |
6 | |
7 | static int priority; |
8 | module_param(priority, int, 0); |
9 | MODULE_PARM_DESC(priority, "specify memory notifier priority"); |
10 | |
11 | static struct notifier_err_inject memory_notifier_err_inject = { |
12 | .actions = { |
13 | { NOTIFIER_ERR_INJECT_ACTION(MEM_GOING_ONLINE) }, |
14 | { NOTIFIER_ERR_INJECT_ACTION(MEM_GOING_OFFLINE) }, |
15 | {} |
16 | } |
17 | }; |
18 | |
19 | static struct dentry *dir; |
20 | |
21 | static int err_inject_init(void) |
22 | { |
23 | int err; |
24 | |
25 | dir = notifier_err_inject_init("memory", notifier_err_inject_dir, |
26 | &memory_notifier_err_inject, priority); |
27 | if (IS_ERR(dir)) |
28 | return PTR_ERR(dir); |
29 | |
30 | err = register_memory_notifier(&memory_notifier_err_inject.nb); |
31 | if (err) |
32 | debugfs_remove_recursive(dir); |
33 | |
34 | return err; |
35 | } |
36 | |
37 | static void err_inject_exit(void) |
38 | { |
39 | unregister_memory_notifier(&memory_notifier_err_inject.nb); |
40 | debugfs_remove_recursive(dir); |
41 | } |
42 | |
43 | module_init(err_inject_init); |
44 | module_exit(err_inject_exit); |
45 | |
46 | MODULE_DESCRIPTION("memory notifier error injection module"); |
47 | MODULE_LICENSE("GPL"); |
48 | MODULE_AUTHOR("Akinobu Mita <akinobu.mita@gmail.com>"); |
49 |