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