failslab.c 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. // SPDX-License-Identifier: GPL-2.0
  2. #include <linux/fault-inject.h>
  3. #include <linux/slab.h>
  4. #include <linux/mm.h>
  5. #include "slab.h"
  6. static struct {
  7. struct fault_attr attr;
  8. bool ignore_gfp_reclaim;
  9. bool cache_filter;
  10. } failslab = {
  11. .attr = FAULT_ATTR_INITIALIZER,
  12. .ignore_gfp_reclaim = true,
  13. .cache_filter = false,
  14. };
  15. bool __should_failslab(struct kmem_cache *s, gfp_t gfpflags)
  16. {
  17. /* No fault-injection for bootstrap cache */
  18. if (unlikely(s == kmem_cache))
  19. return false;
  20. if (gfpflags & __GFP_NOFAIL)
  21. return false;
  22. if (failslab.ignore_gfp_reclaim &&
  23. (gfpflags & __GFP_DIRECT_RECLAIM))
  24. return false;
  25. if (failslab.cache_filter && !(s->flags & SLAB_FAILSLAB))
  26. return false;
  27. return should_fail(&failslab.attr, s->object_size);
  28. }
  29. static int __init setup_failslab(char *str)
  30. {
  31. return setup_fault_attr(&failslab.attr, str);
  32. }
  33. __setup("failslab=", setup_failslab);
  34. #ifdef CONFIG_FAULT_INJECTION_DEBUG_FS
  35. static int __init failslab_debugfs_init(void)
  36. {
  37. struct dentry *dir;
  38. umode_t mode = S_IFREG | 0600;
  39. dir = fault_create_debugfs_attr("failslab", NULL, &failslab.attr);
  40. if (IS_ERR(dir))
  41. return PTR_ERR(dir);
  42. debugfs_create_bool("ignore-gfp-wait", mode, dir,
  43. &failslab.ignore_gfp_reclaim);
  44. debugfs_create_bool("cache-filter", mode, dir,
  45. &failslab.cache_filter);
  46. return 0;
  47. }
  48. late_initcall(failslab_debugfs_init);
  49. #endif /* CONFIG_FAULT_INJECTION_DEBUG_FS */