vnconfig.c revision 1.42 1 /* $NetBSD: vnconfig.c,v 1.42 2014/05/23 20:50:16 dholland Exp $ */
2
3 /*-
4 * Copyright (c) 1997 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Jason R. Thorpe.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 /*
33 * Copyright (c) 1993 University of Utah.
34 * Copyright (c) 1990, 1993
35 * The Regents of the University of California. All rights reserved.
36 *
37 * This code is derived from software contributed to Berkeley by
38 * the Systems Programming Group of the University of Utah Computer
39 * Science Department.
40 *
41 * Redistribution and use in source and binary forms, with or without
42 * modification, are permitted provided that the following conditions
43 * are met:
44 * 1. Redistributions of source code must retain the above copyright
45 * notice, this list of conditions and the following disclaimer.
46 * 2. Redistributions in binary form must reproduce the above copyright
47 * notice, this list of conditions and the following disclaimer in the
48 * documentation and/or other materials provided with the distribution.
49 * 3. Neither the name of the University nor the names of its contributors
50 * may be used to endorse or promote products derived from this software
51 * without specific prior written permission.
52 *
53 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
54 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
55 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
56 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
57 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
58 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
59 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
60 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
61 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
62 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
63 * SUCH DAMAGE.
64 *
65 * from: Utah $Hdr: vnconfig.c 1.1 93/12/15$
66 *
67 * @(#)vnconfig.c 8.1 (Berkeley) 12/15/93
68 */
69
70 #include <sys/param.h>
71 #include <sys/ioctl.h>
72 #include <sys/mount.h>
73 #include <sys/buf.h>
74 #include <sys/disklabel.h>
75 #include <sys/disk.h>
76 #include <sys/bitops.h>
77
78 #include <dev/vndvar.h>
79
80 #include <disktab.h>
81 #include <err.h>
82 #include <dirent.h>
83 #include <errno.h>
84 #include <fcntl.h>
85 #include <stddef.h>
86 #include <stdio.h>
87 #include <stdlib.h>
88 #include <string.h>
89 #include <unistd.h>
90 #include <util.h>
91 #include <paths.h>
92
93 #define VND_CONFIG 1
94 #define VND_UNCONFIG 2
95 #define VND_GET 3
96
97 static int verbose = 0;
98 static int readonly = 0;
99 static int force = 0;
100 static int compressed = 0;
101 static char *tabname;
102
103 static void show(int, int);
104 static int config(char *, char *, char *, int);
105 static int getgeom(struct vndgeom *, char *);
106 __dead static void usage(void);
107
108 int
109 main(int argc, char *argv[])
110 {
111 int ch, rv, action = VND_CONFIG;
112
113 while ((ch = getopt(argc, argv, "Fcf:lrt:uvz")) != -1) {
114 switch (ch) {
115 case 'F':
116 force = 1;
117 break;
118 case 'c':
119 action = VND_CONFIG;
120 break;
121 case 'f':
122 if (setdisktab(optarg) == -1)
123 usage();
124 break;
125 case 'l':
126 action = VND_GET;
127 break;
128 case 'r':
129 readonly = 1;
130 break;
131 case 't':
132 tabname = optarg;
133 break;
134 case 'u':
135 action = VND_UNCONFIG;
136 break;
137 case 'v':
138 verbose = 1;
139 break;
140 case 'z':
141 compressed = 1;
142 readonly = 1;
143 break;
144 default:
145 case '?':
146 usage();
147 /* NOTREACHED */
148 }
149 }
150 argc -= optind;
151 argv += optind;
152
153 if (action == VND_CONFIG) {
154 if ((argc < 2 || argc > 3) ||
155 (argc == 3 && tabname != NULL))
156 usage();
157 rv = config(argv[0], argv[1], (argc == 3) ? argv[2] : NULL,
158 action);
159 } else if (action == VND_UNCONFIG) {
160 if (argc != 1 || tabname != NULL)
161 usage();
162 rv = config(argv[0], NULL, NULL, action);
163 } else { /* VND_GET */
164 int n, v;
165 const char *vn;
166 char path[64];
167
168 if (argc != 0 && argc != 1)
169 usage();
170
171 vn = argc ? argv[0] : "vnd0";
172
173 v = opendisk(vn, O_RDONLY, path, sizeof(path), 0);
174 if (v == -1)
175 err(1, "open: %s", vn);
176
177 if (argc)
178 show(v, -1);
179 else {
180 DIR *dirp;
181 struct dirent *dp;
182 __BITMAP_TYPE(, uint32_t, 65536) bm;
183
184 __BITMAP_ZERO(&bm);
185
186 if ((dirp = opendir(_PATH_DEV)) == NULL)
187 err(1, "opendir: %s", _PATH_DEV);
188
189 while ((dp = readdir(dirp)) != NULL) {
190 if (strncmp(dp->d_name, "rvnd", 4) != 0)
191 continue;
192 n = atoi(dp->d_name + 4);
193 if (__BITMAP_ISSET(n, &bm))
194 continue;
195 __BITMAP_SET(n, &bm);
196 show(v, n);
197 }
198
199 closedir(dirp);
200 }
201 close(v);
202 rv = 0;
203 }
204 return rv;
205 }
206
207 static void
208 show(int v, int n)
209 {
210 struct vnd_user vnu;
211 char *dev;
212 struct statvfs *mnt;
213 int i, nmount;
214
215 vnu.vnu_unit = n;
216 if (ioctl(v, VNDIOCGET, &vnu) == -1)
217 err(1, "VNDIOCGET");
218
219 if (vnu.vnu_ino == 0) {
220 printf("vnd%d: not in use\n", vnu.vnu_unit);
221 return;
222 }
223
224 printf("vnd%d: ", vnu.vnu_unit);
225
226 dev = devname(vnu.vnu_dev, S_IFBLK);
227 if (dev != NULL)
228 nmount = getmntinfo(&mnt, MNT_NOWAIT);
229 else {
230 mnt = NULL;
231 nmount = 0;
232 }
233
234 if (mnt != NULL) {
235 for (i = 0; i < nmount; i++) {
236 if (strncmp(mnt[i].f_mntfromname, "/dev/", 5) == 0 &&
237 strcmp(mnt[i].f_mntfromname + 5, dev) == 0)
238 break;
239 }
240 if (i < nmount)
241 printf("%s (%s) ", mnt[i].f_mntonname,
242 mnt[i].f_mntfromname);
243 else
244 printf("%s ", dev);
245 }
246 else if (dev != NULL)
247 printf("%s ", dev);
248 else
249 printf("dev %llu,%llu ",
250 (unsigned long long)major(vnu.vnu_dev),
251 (unsigned long long)minor(vnu.vnu_dev));
252
253 printf("inode %llu\n", (unsigned long long)vnu.vnu_ino);
254 }
255
256 static int
257 config(char *dev, char *file, char *geom, int action)
258 {
259 struct vnd_ioctl vndio;
260 struct disklabel *lp;
261 char rdev[MAXPATHLEN + 1];
262 int fd, rv;
263
264 fd = opendisk(dev, O_RDWR, rdev, sizeof(rdev), 0);
265 if (fd < 0) {
266 warn("%s: opendisk", rdev);
267 return (1);
268 }
269
270 memset(&vndio, 0, sizeof(vndio));
271 #ifdef __GNUC__
272 rv = 0; /* XXX */
273 #endif
274
275 vndio.vnd_file = file;
276 if (geom != NULL) {
277 rv = getgeom(&vndio.vnd_geom, geom);
278 if (rv != 0)
279 errx(1, "invalid geometry: %s", geom);
280 vndio.vnd_flags = VNDIOF_HASGEOM;
281 } else if (tabname != NULL) {
282 lp = getdiskbyname(tabname);
283 if (lp == NULL)
284 errx(1, "unknown disk type: %s", tabname);
285 vndio.vnd_geom.vng_secsize = lp->d_secsize;
286 vndio.vnd_geom.vng_nsectors = lp->d_nsectors;
287 vndio.vnd_geom.vng_ntracks = lp->d_ntracks;
288 vndio.vnd_geom.vng_ncylinders = lp->d_ncylinders;
289 vndio.vnd_flags = VNDIOF_HASGEOM;
290 }
291
292 if (readonly)
293 vndio.vnd_flags |= VNDIOF_READONLY;
294
295 if (compressed)
296 vndio.vnd_flags |= VNF_COMP;
297
298 /*
299 * Clear (un-configure) the device
300 */
301 if (action == VND_UNCONFIG) {
302 if (force)
303 vndio.vnd_flags |= VNDIOF_FORCE;
304 rv = ioctl(fd, VNDIOCCLR, &vndio);
305 #ifdef VNDIOOCCLR
306 if (rv && errno == ENOTTY)
307 rv = ioctl(fd, VNDIOOCCLR, &vndio);
308 #endif
309 if (rv)
310 warn("%s: VNDIOCCLR", rdev);
311 else if (verbose)
312 printf("%s: cleared\n", rdev);
313 }
314 /*
315 * Configure the device
316 */
317 if (action == VND_CONFIG) {
318 int ffd;
319
320 ffd = open(file, readonly ? O_RDONLY : O_RDWR);
321 if (ffd < 0) {
322 warn("%s", file);
323 rv = -1;
324 } else {
325 (void) close(ffd);
326
327 rv = ioctl(fd, VNDIOCSET, &vndio);
328 #ifdef VNDIOOCSET
329 if (rv && errno == ENOTTY) {
330 rv = ioctl(fd, VNDIOOCSET, &vndio);
331 vndio.vnd_size = vndio.vnd_osize;
332 }
333 #endif
334 if (rv)
335 warn("%s: VNDIOCSET", rdev);
336 else if (verbose) {
337 printf("%s: %" PRIu64 " bytes on %s", rdev,
338 vndio.vnd_size, file);
339 if (vndio.vnd_flags & VNDIOF_HASGEOM)
340 printf(" using geometry %d/%d/%d/%d",
341 vndio.vnd_geom.vng_secsize,
342 vndio.vnd_geom.vng_nsectors,
343 vndio.vnd_geom.vng_ntracks,
344 vndio.vnd_geom.vng_ncylinders);
345 printf("\n");
346 }
347 }
348 }
349
350 (void) close(fd);
351 fflush(stdout);
352 return (rv < 0);
353 }
354
355 static int
356 getgeom(struct vndgeom *vng, char *cp)
357 {
358 char *secsize, *nsectors, *ntracks, *ncylinders;
359
360 #define GETARG(arg) \
361 do { \
362 if (cp == NULL || *cp == '\0') \
363 return (1); \
364 arg = strsep(&cp, "/"); \
365 if (arg == NULL) \
366 return (1); \
367 } while (0)
368
369 GETARG(secsize);
370 GETARG(nsectors);
371 GETARG(ntracks);
372 GETARG(ncylinders);
373
374 #undef GETARG
375
376 /* Too many? */
377 if (cp != NULL)
378 return (1);
379
380 #define CVTARG(str, num) \
381 do { \
382 num = strtol(str, &cp, 10); \
383 if (*cp != '\0') \
384 return (1); \
385 } while (0)
386
387 CVTARG(secsize, vng->vng_secsize);
388 CVTARG(nsectors, vng->vng_nsectors);
389 CVTARG(ntracks, vng->vng_ntracks);
390 CVTARG(ncylinders, vng->vng_ncylinders);
391
392 #undef CVTARG
393
394 return (0);
395 }
396
397 static void
398 usage(void)
399 {
400
401 (void)fprintf(stderr, "%s%s",
402 "usage: vnconfig [-crvz] [-f disktab] [-t typename] vnode_disk"
403 " regular-file [geomspec]\n",
404 " vnconfig -u [-Fv] vnode_disk\n"
405 " vnconfig -l [vnode_disk]\n");
406 exit(1);
407 }
408