Commit | Line | Data |
---|---|---|
65178db4 BS |
1 | /* |
2 | * Copyright (c) 2011 Bryan Schumaker <bjschuma@netapp.com> | |
3 | * | |
4 | * Uses debugfs to create fault injection points for client testing | |
5 | */ | |
6 | ||
7 | #include <linux/types.h> | |
8 | #include <linux/fs.h> | |
9 | #include <linux/debugfs.h> | |
10 | #include <linux/module.h> | |
6c1e82a4 | 11 | #include <linux/nsproxy.h> |
5976687a | 12 | #include <linux/sunrpc/addr.h> |
d7cc431e | 13 | #include <asm/uaccess.h> |
65178db4 BS |
14 | |
15 | #include "state.h" | |
6c1e82a4 | 16 | #include "netns.h" |
65178db4 BS |
17 | |
18 | struct nfsd_fault_inject_op { | |
19 | char *file; | |
8ce54e0d | 20 | u64 (*forget)(struct nfs4_client *, u64); |
184c1847 | 21 | u64 (*print)(struct nfs4_client *, u64); |
65178db4 BS |
22 | }; |
23 | ||
24 | static struct nfsd_fault_inject_op inject_ops[] = { | |
25 | { | |
26 | .file = "forget_clients", | |
8ce54e0d | 27 | .forget = nfsd_forget_client, |
184c1847 | 28 | .print = nfsd_print_client, |
65178db4 BS |
29 | }, |
30 | { | |
31 | .file = "forget_locks", | |
8ce54e0d | 32 | .forget = nfsd_forget_client_locks, |
184c1847 | 33 | .print = nfsd_print_client_locks, |
65178db4 BS |
34 | }, |
35 | { | |
36 | .file = "forget_openowners", | |
8ce54e0d | 37 | .forget = nfsd_forget_client_openowners, |
184c1847 | 38 | .print = nfsd_print_client_openowners, |
65178db4 BS |
39 | }, |
40 | { | |
41 | .file = "forget_delegations", | |
8ce54e0d | 42 | .forget = nfsd_forget_client_delegations, |
184c1847 | 43 | .print = nfsd_print_client_delegations, |
65178db4 BS |
44 | }, |
45 | { | |
46 | .file = "recall_delegations", | |
8ce54e0d | 47 | .forget = nfsd_recall_client_delegations, |
184c1847 | 48 | .print = nfsd_print_client_delegations, |
65178db4 BS |
49 | }, |
50 | }; | |
51 | ||
52 | static long int NUM_INJECT_OPS = sizeof(inject_ops) / sizeof(struct nfsd_fault_inject_op); | |
53 | static struct dentry *debug_dir; | |
54 | ||
d7cc431e | 55 | static void nfsd_inject_set(struct nfsd_fault_inject_op *op, u64 val) |
65178db4 | 56 | { |
8ce54e0d | 57 | u64 count = 0; |
65178db4 BS |
58 | |
59 | if (val == 0) | |
60 | printk(KERN_INFO "NFSD Fault Injection: %s (all)", op->file); | |
61 | else | |
62 | printk(KERN_INFO "NFSD Fault Injection: %s (n = %llu)", op->file, val); | |
63 | ||
04395839 | 64 | nfs4_lock_state(); |
8ce54e0d | 65 | count = nfsd_for_n_state(val, op->forget); |
04395839 | 66 | nfs4_unlock_state(); |
8ce54e0d | 67 | printk(KERN_INFO "NFSD: %s: found %llu", op->file, count); |
65178db4 BS |
68 | } |
69 | ||
6c1e82a4 BS |
70 | static void nfsd_inject_set_client(struct nfsd_fault_inject_op *op, |
71 | struct sockaddr_storage *addr, | |
72 | size_t addr_size) | |
73 | { | |
74 | char buf[INET6_ADDRSTRLEN]; | |
75 | struct nfs4_client *clp; | |
76 | u64 count; | |
77 | ||
78 | nfs4_lock_state(); | |
79 | clp = nfsd_find_client(addr, addr_size); | |
80 | if (clp) { | |
81 | count = op->forget(clp, 0); | |
0a5c33e2 | 82 | rpc_ntop((struct sockaddr *)&clp->cl_addr, buf, sizeof(buf)); |
6c1e82a4 BS |
83 | printk(KERN_INFO "NFSD [%s]: Client %s had %llu state object(s)\n", op->file, buf, count); |
84 | } | |
85 | nfs4_unlock_state(); | |
86 | } | |
87 | ||
d7cc431e | 88 | static void nfsd_inject_get(struct nfsd_fault_inject_op *op, u64 *val) |
65178db4 | 89 | { |
184c1847 BS |
90 | nfs4_lock_state(); |
91 | *val = nfsd_for_n_state(0, op->print); | |
92 | nfs4_unlock_state(); | |
65178db4 BS |
93 | } |
94 | ||
d7cc431e BS |
95 | static ssize_t fault_inject_read(struct file *file, char __user *buf, |
96 | size_t len, loff_t *ppos) | |
97 | { | |
98 | static u64 val; | |
99 | char read_buf[25]; | |
f3e41ec5 | 100 | size_t size; |
d7cc431e BS |
101 | loff_t pos = *ppos; |
102 | ||
103 | if (!pos) | |
496ad9aa | 104 | nfsd_inject_get(file_inode(file)->i_private, &val); |
d7cc431e BS |
105 | size = scnprintf(read_buf, sizeof(read_buf), "%llu\n", val); |
106 | ||
f3e41ec5 | 107 | return simple_read_from_buffer(buf, len, ppos, read_buf, size); |
d7cc431e BS |
108 | } |
109 | ||
110 | static ssize_t fault_inject_write(struct file *file, const char __user *buf, | |
111 | size_t len, loff_t *ppos) | |
112 | { | |
6c1e82a4 | 113 | char write_buf[INET6_ADDRSTRLEN]; |
18d9a2ca | 114 | size_t size = min(sizeof(write_buf) - 1, len); |
6c1e82a4 BS |
115 | struct net *net = current->nsproxy->net_ns; |
116 | struct sockaddr_storage sa; | |
d7cc431e BS |
117 | u64 val; |
118 | ||
119 | if (copy_from_user(write_buf, buf, size)) | |
120 | return -EFAULT; | |
6c1e82a4 BS |
121 | write_buf[size] = '\0'; |
122 | ||
123 | size = rpc_pton(net, write_buf, size, (struct sockaddr *)&sa, sizeof(sa)); | |
124 | if (size > 0) | |
496ad9aa | 125 | nfsd_inject_set_client(file_inode(file)->i_private, &sa, size); |
6c1e82a4 BS |
126 | else { |
127 | val = simple_strtoll(write_buf, NULL, 0); | |
496ad9aa | 128 | nfsd_inject_set(file_inode(file)->i_private, val); |
6c1e82a4 | 129 | } |
d7cc431e BS |
130 | return len; /* on success, claim we got the whole input */ |
131 | } | |
132 | ||
133 | static const struct file_operations fops_nfsd = { | |
134 | .owner = THIS_MODULE, | |
135 | .read = fault_inject_read, | |
136 | .write = fault_inject_write, | |
137 | }; | |
65178db4 BS |
138 | |
139 | void nfsd_fault_inject_cleanup(void) | |
140 | { | |
141 | debugfs_remove_recursive(debug_dir); | |
142 | } | |
143 | ||
144 | int nfsd_fault_inject_init(void) | |
145 | { | |
146 | unsigned int i; | |
147 | struct nfsd_fault_inject_op *op; | |
88187398 | 148 | umode_t mode = S_IFREG | S_IRUSR | S_IWUSR; |
65178db4 BS |
149 | |
150 | debug_dir = debugfs_create_dir("nfsd", NULL); | |
151 | if (!debug_dir) | |
152 | goto fail; | |
153 | ||
154 | for (i = 0; i < NUM_INJECT_OPS; i++) { | |
155 | op = &inject_ops[i]; | |
156 | if (!debugfs_create_file(op->file, mode, debug_dir, op, &fops_nfsd)) | |
157 | goto fail; | |
158 | } | |
159 | return 0; | |
160 | ||
161 | fail: | |
162 | nfsd_fault_inject_cleanup(); | |
163 | return -ENOMEM; | |
164 | } |