1 /*	$NetBSD: h_tools.c,v 1.4 2011/06/11 18:03:17 christos Exp $	*/
2 
3 /*
4  * Copyright (c) 2005, 2006 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26  * POSSIBILITY OF SUCH DAMAGE.
27  */
28 
29 /*
30  * Helper tools for several tests.  These are kept in a single file due
31  * to the limitations of bsd.prog.mk to build a single program in a
32  * given directory.
33  */
34 
35 #include <sys/param.h>
36 #include <sys/types.h>
37 #include <sys/event.h>
38 #include <sys/mount.h>
39 #include <sys/statvfs.h>
40 #include <sys/socket.h>
41 #include <sys/time.h>
42 #include <sys/un.h>
43 
44 #include <assert.h>
45 #include <err.h>
46 #include <errno.h>
47 #include <fcntl.h>
48 #include <stdio.h>
49 #include <stdlib.h>
50 #include <string.h>
51 #include <unistd.h>
52 
53 /* --------------------------------------------------------------------- */
54 
55 static int getfh_main(int, char **);
56 static int kqueue_main(int, char **);
57 static int rename_main(int, char **);
58 static int sockets_main(int, char **);
59 static int statvfs_main(int, char **);
60 
61 /* --------------------------------------------------------------------- */
62 
63 int
64 getfh_main(int argc, char **argv)
65 {
66 	int error;
67 	void *fh;
68 	size_t fh_size;
69 
70 	if (argc < 2)
71 		return EXIT_FAILURE;
72 
73 	fh_size = 0;
74 	fh = NULL;
75 	for (;;) {
76 		if (fh_size) {
77 			fh = malloc(fh_size);
78 			if (fh == NULL) {
79 				fprintf(stderr, "out of memory");
80 				return EXIT_FAILURE;
81 			}
82 		}
83 		/*
84 		 * The kernel provides the necessary size in fh_size -
85 		 * but it may change if someone moves things around,
86 		 * so retry untill we have enough memory.
87 		 */
88 		error = getfh(argv[1], fh, &fh_size);
89 		if (error == 0) {
90 			break;
91 		} else {
92 			if (fh != NULL)
93 				free(fh);
94 			if (errno != E2BIG) {
95 				warn("getfh");
96 				return EXIT_FAILURE;
97 			}
98 		}
99 	}
100 
101 	error = write(STDOUT_FILENO, fh, fh_size);
102 	if (error == -1) {
103 		warn("write");
104 		return EXIT_FAILURE;
105 	}
106 	free(fh);
107 
108 	return 0;
109 }
110 
111 /* --------------------------------------------------------------------- */
112 
113 int
114 kqueue_main(int argc, char **argv)
115 {
116 	char *line;
117 	int i, kq;
118 	size_t len;
119 	struct kevent *changes, event;
120 
121 	if (argc < 2)
122 		return EXIT_FAILURE;
123 
124 	argc--;
125 	argv++;
126 
127 	changes = malloc(sizeof(struct kevent) * argc);
128 	if (changes == NULL)
129 		errx(EXIT_FAILURE, "not enough memory");
130 
131 	for (i = 0; i < argc; i++) {
132 		int fd;
133 
134 		fd = open(argv[i], O_RDONLY);
135 		if (fd == -1)
136 			err(EXIT_FAILURE, "cannot open %s", argv[i]);
137 
138 		EV_SET(&changes[i], fd, EVFILT_VNODE,
139 		    EV_ADD | EV_ENABLE | EV_ONESHOT,
140 		    NOTE_ATTRIB | NOTE_DELETE | NOTE_EXTEND | NOTE_LINK |
141 		    NOTE_RENAME | NOTE_REVOKE | NOTE_WRITE,
142 		    0, 0);
143 	}
144 
145 	kq = kqueue();
146 	if (kq == -1)
147 		err(EXIT_FAILURE, "kqueue");
148 
149 	while ((line = fgetln(stdin, &len)) != NULL) {
150 		int ec, nev;
151 		struct timespec to;
152 
153 		to.tv_sec = 0;
154 		to.tv_nsec = 100000;
155 
156 		(void)kevent(kq, changes, argc, &event, 1, &to);
157 
158 		assert(len > 0);
159 		assert(line[len - 1] == '\n');
160 		line[len - 1] = '\0';
161 		ec = system(line);
162 		if (ec != EXIT_SUCCESS)
163 			errx(ec, "%s returned %d", line, ec);
164 
165 		do {
166 			nev = kevent(kq, changes, argc, &event, 1, &to);
167 			if (nev == -1)
168 				err(EXIT_FAILURE, "kevent");
169 			else if (nev > 0) {
170 				for (i = 0; i < argc; i++)
171 					if (event.ident == changes[i].ident)
172 						break;
173 
174 				if (event.fflags & NOTE_ATTRIB)
175 					printf("%s - NOTE_ATTRIB\n", argv[i]);
176 				if (event.fflags & NOTE_DELETE)
177 					printf("%s - NOTE_DELETE\n", argv[i]);
178 				if (event.fflags & NOTE_EXTEND)
179 					printf("%s - NOTE_EXTEND\n", argv[i]);
180 				if (event.fflags & NOTE_LINK)
181 					printf("%s - NOTE_LINK\n", argv[i]);
182 				if (event.fflags & NOTE_RENAME)
183 					printf("%s - NOTE_RENAME\n", argv[i]);
184 				if (event.fflags & NOTE_REVOKE)
185 					printf("%s - NOTE_REVOKE\n", argv[i]);
186 				if (event.fflags & NOTE_WRITE)
187 					printf("%s - NOTE_WRITE\n", argv[i]);
188 			}
189 		} while (nev > 0);
190 	}
191 
192 	for (i = 0; i < argc; i++)
193 		close(changes[i].ident);
194 	free(changes);
195 
196 	return EXIT_SUCCESS;
197 }
198 
199 /* --------------------------------------------------------------------- */
200 
201 int
202 rename_main(int argc, char **argv)
203 {
204 
205 	if (argc < 3)
206 		return EXIT_FAILURE;
207 
208 	if (rename(argv[1], argv[2]) == -1) {
209 		warn("rename");
210 		return EXIT_FAILURE;
211 	}
212 
213 	return EXIT_SUCCESS;
214 }
215 
216 /* --------------------------------------------------------------------- */
217 
218 int
219 sockets_main(int argc, char **argv)
220 {
221 	int error, fd;
222 	struct sockaddr_un addr;
223 
224 	if (argc < 2)
225 		return EXIT_FAILURE;
226 
227 	fd = socket(PF_LOCAL, SOCK_STREAM, 0);
228 	if (fd == -1) {
229 		warn("socket");
230 		return EXIT_FAILURE;
231 	}
232 
233 	(void)strlcpy(addr.sun_path, argv[1], sizeof(addr.sun_path));
234 	addr.sun_family = PF_UNIX;
235 
236 	error = bind(fd, (struct sockaddr *)&addr, sizeof(addr));
237 	if (error == -1) {
238 		warn("connect");
239 		return EXIT_FAILURE;
240 	}
241 
242 	close(fd);
243 
244 	return EXIT_SUCCESS;
245 }
246 
247 /* --------------------------------------------------------------------- */
248 
249 int
250 statvfs_main(int argc, char **argv)
251 {
252 	int error;
253 	struct statvfs buf;
254 
255 	if (argc < 2)
256 		return EXIT_FAILURE;
257 
258 	error = statvfs(argv[1], &buf);
259 	if (error != 0) {
260 		warn("statvfs");
261 		return EXIT_FAILURE;
262 	}
263 
264 	(void)printf("f_bsize=%lu\n", buf.f_bsize);
265 	(void)printf("f_blocks=%" PRId64 "\n", buf.f_blocks);
266 	(void)printf("f_bfree=%" PRId64 "\n", buf.f_bfree);
267 	(void)printf("f_files=%" PRId64 "\n", buf.f_files);
268 
269 	return EXIT_SUCCESS;
270 }
271 
272 /* --------------------------------------------------------------------- */
273 
274 int
275 main(int argc, char **argv)
276 {
277 	int error;
278 
279 	if (argc < 2)
280 		return EXIT_FAILURE;
281 
282 	argc -= 1;
283 	argv += 1;
284 
285 	if (strcmp(argv[0], "getfh") == 0)
286 		error = getfh_main(argc, argv);
287 	else if (strcmp(argv[0], "kqueue") == 0)
288 		error = kqueue_main(argc, argv);
289 	else if (strcmp(argv[0], "rename") == 0)
290 		error = rename_main(argc, argv);
291 	else if (strcmp(argv[0], "sockets") == 0)
292 		error = sockets_main(argc, argv);
293 	else if (strcmp(argv[0], "statvfs") == 0)
294 		error = statvfs_main(argc, argv);
295 	else
296 		error = EXIT_FAILURE;
297 
298 	return error;
299 }
300