ufs/core/ufs-fault-injection.c

0001 // SPDX-License-Identifier: GPL-2.0-or-later
0002
0003 #include <linux/kconfig.h>
0004 #include <linux/types.h>
0005 #include <linux/fault-inject.h>
0006 #include <linux/module.h>
0007 #include "ufs-fault-injection.h"
0008
0009 static int ufs_fault_get(char *buffer, const struct kernel_param *kp);
0010 static int ufs_fault_set(const char *val, const struct kernel_param *kp);
0011
0012 static const struct kernel_param_ops ufs_fault_ops = {
0013     .get = ufs_fault_get,
0014     .set = ufs_fault_set,
0015 };
0016
0017 enum { FAULT_INJ_STR_SIZE = 80 };
0018
0019 /*
0020  * For more details about fault injection, please refer to
0021  * Documentation/fault-injection/fault-injection.rst.
0022  */
0023 static char g_trigger_eh_str[FAULT_INJ_STR_SIZE];
0024 module_param_cb(trigger_eh, &ufs_fault_ops, g_trigger_eh_str, 0644);
0025 MODULE_PARM_DESC(trigger_eh,
0026     "Fault injection. trigger_eh=<interval>,<probability>,<space>,<times>");
0027 static DECLARE_FAULT_ATTR(ufs_trigger_eh_attr);
0028
0029 static char g_timeout_str[FAULT_INJ_STR_SIZE];
0030 module_param_cb(timeout, &ufs_fault_ops, g_timeout_str, 0644);
0031 MODULE_PARM_DESC(timeout,
0032     "Fault injection. timeout=<interval>,<probability>,<space>,<times>");
0033 static DECLARE_FAULT_ATTR(ufs_timeout_attr);
0034
0035 static int ufs_fault_get(char *buffer, const struct kernel_param *kp)
0036 {
0037     const char *fault_str = kp->arg;
0038
0039     return sysfs_emit(buffer, "%s\n", fault_str);
0040 }
0041
0042 static int ufs_fault_set(const char *val, const struct kernel_param *kp)
0043 {
0044     struct fault_attr *attr = NULL;
0045
0046     if (kp->arg == g_trigger_eh_str)
0047         attr = &ufs_trigger_eh_attr;
0048     else if (kp->arg == g_timeout_str)
0049         attr = &ufs_timeout_attr;
0050
0051     if (WARN_ON_ONCE(!attr))
0052         return -EINVAL;
0053
0054     if (!setup_fault_attr(attr, (char *)val))
0055         return -EINVAL;
0056
0057     strlcpy(kp->arg, val, FAULT_INJ_STR_SIZE);
0058
0059     return 0;
0060 }
0061
0062 bool ufs_trigger_eh(void)
0063 {
0064     return should_fail(&ufs_trigger_eh_attr, 1);
0065 }
0066
0067 bool ufs_fail_completion(void)
0068 {
0069     return should_fail(&ufs_timeout_attr, 1);
0070 }