1*dd11376bSBart Van Assche // SPDX-License-Identifier: GPL-2.0-or-later
2*dd11376bSBart Van Assche 
3*dd11376bSBart Van Assche #include <linux/kconfig.h>
4*dd11376bSBart Van Assche #include <linux/types.h>
5*dd11376bSBart Van Assche #include <linux/fault-inject.h>
6*dd11376bSBart Van Assche #include <linux/module.h>
7*dd11376bSBart Van Assche #include "ufs-fault-injection.h"
8*dd11376bSBart Van Assche 
9*dd11376bSBart Van Assche static int ufs_fault_get(char *buffer, const struct kernel_param *kp);
10*dd11376bSBart Van Assche static int ufs_fault_set(const char *val, const struct kernel_param *kp);
11*dd11376bSBart Van Assche 
12*dd11376bSBart Van Assche static const struct kernel_param_ops ufs_fault_ops = {
13*dd11376bSBart Van Assche 	.get = ufs_fault_get,
14*dd11376bSBart Van Assche 	.set = ufs_fault_set,
15*dd11376bSBart Van Assche };
16*dd11376bSBart Van Assche 
17*dd11376bSBart Van Assche enum { FAULT_INJ_STR_SIZE = 80 };
18*dd11376bSBart Van Assche 
19*dd11376bSBart Van Assche /*
20*dd11376bSBart Van Assche  * For more details about fault injection, please refer to
21*dd11376bSBart Van Assche  * Documentation/fault-injection/fault-injection.rst.
22*dd11376bSBart Van Assche  */
23*dd11376bSBart Van Assche static char g_trigger_eh_str[FAULT_INJ_STR_SIZE];
24*dd11376bSBart Van Assche module_param_cb(trigger_eh, &ufs_fault_ops, g_trigger_eh_str, 0644);
25*dd11376bSBart Van Assche MODULE_PARM_DESC(trigger_eh,
26*dd11376bSBart Van Assche 	"Fault injection. trigger_eh=<interval>,<probability>,<space>,<times>");
27*dd11376bSBart Van Assche static DECLARE_FAULT_ATTR(ufs_trigger_eh_attr);
28*dd11376bSBart Van Assche 
29*dd11376bSBart Van Assche static char g_timeout_str[FAULT_INJ_STR_SIZE];
30*dd11376bSBart Van Assche module_param_cb(timeout, &ufs_fault_ops, g_timeout_str, 0644);
31*dd11376bSBart Van Assche MODULE_PARM_DESC(timeout,
32*dd11376bSBart Van Assche 	"Fault injection. timeout=<interval>,<probability>,<space>,<times>");
33*dd11376bSBart Van Assche static DECLARE_FAULT_ATTR(ufs_timeout_attr);
34*dd11376bSBart Van Assche 
35*dd11376bSBart Van Assche static int ufs_fault_get(char *buffer, const struct kernel_param *kp)
36*dd11376bSBart Van Assche {
37*dd11376bSBart Van Assche 	const char *fault_str = kp->arg;
38*dd11376bSBart Van Assche 
39*dd11376bSBart Van Assche 	return sysfs_emit(buffer, "%s\n", fault_str);
40*dd11376bSBart Van Assche }
41*dd11376bSBart Van Assche 
42*dd11376bSBart Van Assche static int ufs_fault_set(const char *val, const struct kernel_param *kp)
43*dd11376bSBart Van Assche {
44*dd11376bSBart Van Assche 	struct fault_attr *attr = NULL;
45*dd11376bSBart Van Assche 
46*dd11376bSBart Van Assche 	if (kp->arg == g_trigger_eh_str)
47*dd11376bSBart Van Assche 		attr = &ufs_trigger_eh_attr;
48*dd11376bSBart Van Assche 	else if (kp->arg == g_timeout_str)
49*dd11376bSBart Van Assche 		attr = &ufs_timeout_attr;
50*dd11376bSBart Van Assche 
51*dd11376bSBart Van Assche 	if (WARN_ON_ONCE(!attr))
52*dd11376bSBart Van Assche 		return -EINVAL;
53*dd11376bSBart Van Assche 
54*dd11376bSBart Van Assche 	if (!setup_fault_attr(attr, (char *)val))
55*dd11376bSBart Van Assche 		return -EINVAL;
56*dd11376bSBart Van Assche 
57*dd11376bSBart Van Assche 	strlcpy(kp->arg, val, FAULT_INJ_STR_SIZE);
58*dd11376bSBart Van Assche 
59*dd11376bSBart Van Assche 	return 0;
60*dd11376bSBart Van Assche }
61*dd11376bSBart Van Assche 
62*dd11376bSBart Van Assche bool ufs_trigger_eh(void)
63*dd11376bSBart Van Assche {
64*dd11376bSBart Van Assche 	return should_fail(&ufs_trigger_eh_attr, 1);
65*dd11376bSBart Van Assche }
66*dd11376bSBart Van Assche 
67*dd11376bSBart Van Assche bool ufs_fail_completion(void)
68*dd11376bSBart Van Assche {
69*dd11376bSBart Van Assche 	return should_fail(&ufs_timeout_attr, 1);
70*dd11376bSBart Van Assche }
71