50 lines · c
1// SPDX-License-Identifier: GPL-2.0-only2#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