memory-notifier-error-inject.c 1.1 KB

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