memory-notifier-error-inject.c (1157B)
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>");