1 /*-
2  * Copyright (c) 2009 Rick Macklem, University of Guelph
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  */
27 
28 #include <sys/cdefs.h>
29 __FBSDID("$FreeBSD$");
30 
31 #include <sys/param.h>
32 #include <sys/linker.h>
33 #include <sys/module.h>
34 #include <sys/socket.h>
35 
36 #include <arpa/inet.h>
37 
38 #include <netinet/in.h>
39 
40 #include <nfs/nfssvc.h>
41 
42 #include <fs/nfs/rpcv2.h>
43 #include <fs/nfs/nfsproto.h>
44 #include <fs/nfs/nfskpiport.h>
45 #include <fs/nfs/nfs.h>
46 
47 #include <ctype.h>
48 #include <err.h>
49 #include <errno.h>
50 #include <stdio.h>
51 #include <stdlib.h>
52 #include <string.h>
53 #include <unistd.h>
54 
55 #define	DUMPSIZE	10000
56 
57 static void dump_lockstate(char *);
58 static void dump_openstate(void);
59 static void usage(void);
60 static char *open_flags(uint32_t);
61 static char *deleg_flags(uint32_t);
62 static char *lock_flags(uint32_t);
63 static char *client_flags(uint32_t);
64 
65 static struct nfsd_dumpclients dp[DUMPSIZE];
66 static struct nfsd_dumplocks lp[DUMPSIZE];
67 static char flag_string[20];
68 
69 int
70 main(int argc, char **argv)
71 {
72 	int ch, openstate;
73 	char *lockfile;
74 
75 	if (modfind("nfsd") < 0)
76 		errx(1, "nfsd not loaded - self terminating");
77 	openstate = 0;
78 	lockfile = NULL;
79 	while ((ch = getopt(argc, argv, "ol:")) != -1)
80 		switch (ch) {
81 		case 'o':
82 			openstate = 1;
83 			break;
84 		case 'l':
85 			lockfile = optarg;
86 			break;
87 		default:
88 			usage();
89 		}
90 	argc -= optind;
91 	argv += optind;
92 
93 	if (openstate == 0 && lockfile == NULL)
94 		openstate = 1;
95 	else if (openstate != 0 && lockfile != NULL)
96 		errx(1, "-o and -l cannot both be specified");
97 
98 	/*
99 	 * For -o, dump all open/lock state.
100 	 * For -l, dump lock state for that file.
101 	 */
102 	if (openstate != 0)
103 		dump_openstate();
104 	else
105 		dump_lockstate(lockfile);
106 	exit(0);
107 }
108 
109 static void
110 usage(void)
111 {
112 
113 	errx(1, "usage: nfsdumpstate [-o] [-l]");
114 }
115 
116 /*
117  * Dump all open/lock state.
118  */
119 static void
120 dump_openstate(void)
121 {
122 	struct nfsd_dumplist dumplist;
123 	int cnt, i;
124 	char nbuf[INET6_ADDRSTRLEN];
125 
126 	dumplist.ndl_size = DUMPSIZE;
127 	dumplist.ndl_list = (void *)dp;
128 	if (nfssvc(NFSSVC_DUMPCLIENTS, &dumplist) < 0)
129 		errx(1, "Can't perform dump clients syscall");
130 
131 	printf("%-13s %9.9s %9.9s %9.9s %9.9s %9.9s %9.9s %-45s %s\n",
132 	    "Flags", "OpenOwner", "Open", "LockOwner",
133 	    "Lock", "Deleg", "OldDeleg", "Clientaddr", "ClientID");
134 	/*
135 	 * Loop through results, printing them out.
136 	 */
137 	cnt = 0;
138 	while (dp[cnt].ndcl_clid.nclid_idlen > 0 && cnt < DUMPSIZE) {
139 		printf("%-13s ", client_flags(dp[cnt].ndcl_flags));
140 		printf("%9d %9d %9d %9d %9d %9d ",
141 		    dp[cnt].ndcl_nopenowners,
142 		    dp[cnt].ndcl_nopens,
143 		    dp[cnt].ndcl_nlockowners,
144 		    dp[cnt].ndcl_nlocks,
145 		    dp[cnt].ndcl_ndelegs,
146 		    dp[cnt].ndcl_nolddelegs);
147 		switch (dp[cnt].ndcl_addrfam) {
148 #ifdef INET
149 		case AF_INET:
150 			printf("%-45s ",
151 			    inet_ntoa(dp[cnt].ndcl_cbaddr.sin_addr));
152 			break;
153 #endif
154 #ifdef INET6
155 		case AF_INET6:
156 			if (inet_ntop(AF_INET6, &dp[cnt].ndcl_cbaddr.sin6_addr,
157 			    nbuf, sizeof(nbuf)) != NULL)
158 				printf("%-45s ", nbuf);
159 			else
160 				printf("%-45s ", " ");
161 			break;
162 #endif
163 		}
164 		for (i = 0; i < dp[cnt].ndcl_clid.nclid_idlen; i++)
165 			printf("%02x", dp[cnt].ndcl_clid.nclid_id[i]);
166 		printf("\n");
167 		cnt++;
168 	}
169 }
170 
171 /*
172  * Dump the lock state for a file.
173  */
174 static void
175 dump_lockstate(char *fname)
176 {
177 	struct nfsd_dumplocklist dumplocklist;
178 	int cnt, i;
179 
180 	dumplocklist.ndllck_size = DUMPSIZE;
181 	dumplocklist.ndllck_list = (void *)lp;
182 	dumplocklist.ndllck_fname = fname;
183 	if (nfssvc(NFSSVC_DUMPLOCKS, &dumplocklist) < 0)
184 		errx(1, "Can't dump locks for %s\n", fname);
185 
186 	printf("%-11s %-36s %-15s %s\n",
187 	    "Open/Lock",
188 	    "          Stateid or Lock Range",
189 	    "Clientaddr",
190 	    "Owner and ClientID");
191 	/*
192 	 * Loop through results, printing them out.
193 	 */
194 	cnt = 0;
195 	while (lp[cnt].ndlck_clid.nclid_idlen > 0 && cnt < DUMPSIZE) {
196 		if (lp[cnt].ndlck_flags & NFSLCK_OPEN)
197 			printf("%-11s %9d %08x %08x %08x ",
198 			    open_flags(lp[cnt].ndlck_flags),
199 			    lp[cnt].ndlck_stateid.seqid,
200 			    lp[cnt].ndlck_stateid.other[0],
201 			    lp[cnt].ndlck_stateid.other[1],
202 			    lp[cnt].ndlck_stateid.other[2]);
203 		else if (lp[cnt].ndlck_flags & (NFSLCK_DELEGREAD |
204 		    NFSLCK_DELEGWRITE))
205 			printf("%-11s %9d %08x %08x %08x ",
206 			    deleg_flags(lp[cnt].ndlck_flags),
207 			    lp[cnt].ndlck_stateid.seqid,
208 			    lp[cnt].ndlck_stateid.other[0],
209 			    lp[cnt].ndlck_stateid.other[1],
210 			    lp[cnt].ndlck_stateid.other[2]);
211 		else
212 			printf("%-11s  %17jd %17jd ",
213 			    lock_flags(lp[cnt].ndlck_flags),
214 			    lp[cnt].ndlck_first,
215 			    lp[cnt].ndlck_end);
216 		if (lp[cnt].ndlck_addrfam == AF_INET)
217 			printf("%-15s ",
218 			    inet_ntoa(lp[cnt].ndlck_cbaddr.sin_addr));
219 		else
220 			printf("%-15s ", "  ");
221 		for (i = 0; i < lp[cnt].ndlck_owner.nclid_idlen; i++)
222 			printf("%02x", lp[cnt].ndlck_owner.nclid_id[i]);
223 		printf(" ");
224 		for (i = 0; i < lp[cnt].ndlck_clid.nclid_idlen; i++)
225 			printf("%02x", lp[cnt].ndlck_clid.nclid_id[i]);
226 		printf("\n");
227 		cnt++;
228 	}
229 }
230 
231 /*
232  * Parse the Open/Lock flag bits and create a string to be printed.
233  */
234 static char *
235 open_flags(uint32_t flags)
236 {
237 	int i, j;
238 
239 	strlcpy(flag_string, "Open ", sizeof (flag_string));
240 	i = 5;
241 	if (flags & NFSLCK_READACCESS)
242 		flag_string[i++] = 'R';
243 	if (flags & NFSLCK_WRITEACCESS)
244 		flag_string[i++] = 'W';
245 	flag_string[i++] = ' ';
246 	flag_string[i++] = 'D';
247 	flag_string[i] = 'N';
248 	j = i;
249 	if (flags & NFSLCK_READDENY)
250 		flag_string[i++] = 'R';
251 	if (flags & NFSLCK_WRITEDENY)
252 		flag_string[i++] = 'W';
253 	if (i == j)
254 		i++;
255 	flag_string[i] = '\0';
256 	return (flag_string);
257 }
258 
259 static char *
260 deleg_flags(uint32_t flags)
261 {
262 
263 	if (flags & NFSLCK_DELEGREAD)
264 		strlcpy(flag_string, "Deleg R", sizeof (flag_string));
265 	else
266 		strlcpy(flag_string, "Deleg W", sizeof (flag_string));
267 	return (flag_string);
268 }
269 
270 static char *
271 lock_flags(uint32_t flags)
272 {
273 
274 	if (flags & NFSLCK_READ)
275 		strlcpy(flag_string, "Lock R", sizeof (flag_string));
276 	else
277 		strlcpy(flag_string, "Lock W", sizeof (flag_string));
278 	return (flag_string);
279 }
280 
281 static char *
282 client_flags(uint32_t flags)
283 {
284 
285 	flag_string[0] = '\0';
286 	if (flags & LCL_NEEDSCONFIRM)
287 		strlcat(flag_string, "NC ", sizeof (flag_string));
288 	if (flags & LCL_CALLBACKSON)
289 		strlcat(flag_string, "CB ", sizeof (flag_string));
290 	if (flags & LCL_GSS)
291 		strlcat(flag_string, "GSS ", sizeof (flag_string));
292 	if (flags & LCL_ADMINREVOKED)
293 		strlcat(flag_string, "REV", sizeof (flag_string));
294 	return (flag_string);
295 }
296