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