xref: /freebsd/sbin/swapon/swapon.c (revision 193d9e768ba63fcfb187cfd17f461f7d41345048)
1 /*-
2  * Copyright (c) 1980, 1993
3  *	The Regents of the University of California.  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  * 3. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #if 0
31 #ifndef lint
32 static const char copyright[] =
33 "@(#) Copyright (c) 1980, 1993\n\
34 	The Regents of the University of California.  All rights reserved.\n";
35 #endif /* not lint */
36 
37 #ifndef lint
38 static char sccsid[] = "@(#)swapon.c	8.1 (Berkeley) 6/5/93";
39 #endif /* not lint */
40 #endif
41 #include <sys/cdefs.h>
42 __FBSDID("$FreeBSD$");
43 
44 #include <sys/param.h>
45 #include <sys/types.h>
46 #include <sys/mdioctl.h>
47 #include <sys/stat.h>
48 #include <sys/sysctl.h>
49 #include <sys/wait.h>
50 #include <vm/vm_param.h>
51 
52 #include <err.h>
53 #include <errno.h>
54 #include <fcntl.h>
55 #include <fnmatch.h>
56 #include <fstab.h>
57 #include <libgen.h>
58 #include <libutil.h>
59 #include <limits.h>
60 #include <paths.h>
61 #include <stdarg.h>
62 #include <stdio.h>
63 #include <stdlib.h>
64 #include <string.h>
65 #include <unistd.h>
66 
67 static void usage(void);
68 static const char *swap_on_off(const char *, int, char *);
69 static const char *swap_on_off_gbde(const char *, int);
70 static const char *swap_on_off_geli(const char *, char *, int);
71 static const char *swap_on_off_md(const char *, char *, int);
72 static const char *swap_on_off_sfile(const char *, int);
73 static void swaplist(int, int, int);
74 static int run_cmd(int *, const char *, ...) __printflike(2, 3);
75 
76 static enum { SWAPON, SWAPOFF, SWAPCTL } orig_prog, which_prog = SWAPCTL;
77 
78 static int qflag;
79 
80 int
81 main(int argc, char **argv)
82 {
83 	struct fstab *fsp;
84 	const char *swfile;
85 	char *ptr;
86 	int ret, ch, doall;
87 	int sflag, lflag, late, hflag;
88 	const char *etc_fstab;
89 
90 	sflag = lflag = late = hflag = 0;
91 	if ((ptr = strrchr(argv[0], '/')) == NULL)
92 		ptr = argv[0];
93 	if (strstr(ptr, "swapon") != NULL)
94 		which_prog = SWAPON;
95 	else if (strstr(ptr, "swapoff") != NULL)
96 		which_prog = SWAPOFF;
97 	orig_prog = which_prog;
98 
99 	doall = 0;
100 	etc_fstab = NULL;
101 	while ((ch = getopt(argc, argv, "AadghklLmqsUF:")) != -1) {
102 		switch(ch) {
103 		case 'A':
104 			if (which_prog == SWAPCTL) {
105 				doall = 1;
106 				which_prog = SWAPON;
107 			} else
108 				usage();
109 			break;
110 		case 'a':
111 			if (which_prog == SWAPON || which_prog == SWAPOFF)
112 				doall = 1;
113 			else
114 				which_prog = SWAPON;
115 			break;
116 		case 'd':
117 			if (which_prog == SWAPCTL)
118 				which_prog = SWAPOFF;
119 			else
120 				usage();
121 			break;
122 		case 'g':
123 			hflag = 'G';
124 			break;
125 		case 'h':
126 			hflag = 'H';
127 			break;
128 		case 'k':
129 			hflag = 'K';
130 			break;
131 		case 'l':
132 			lflag = 1;
133 			break;
134 		case 'L':
135 			late = 1;
136 			break;
137 		case 'm':
138 			hflag = 'M';
139 			break;
140 		case 'q':
141 			if (which_prog == SWAPON || which_prog == SWAPOFF)
142 				qflag = 1;
143 			break;
144 		case 's':
145 			sflag = 1;
146 			break;
147 		case 'U':
148 			if (which_prog == SWAPCTL) {
149 				doall = 1;
150 				which_prog = SWAPOFF;
151 			} else
152 				usage();
153 			break;
154 		case 'F':
155 			etc_fstab = optarg;
156 			break;
157 		case '?':
158 		default:
159 			usage();
160 		}
161 	}
162 	argv += optind;
163 
164 	ret = 0;
165 	swfile = NULL;
166 	if (etc_fstab != NULL)
167 		setfstab(etc_fstab);
168 	if (which_prog == SWAPON || which_prog == SWAPOFF) {
169 		if (doall) {
170 			while ((fsp = getfsent()) != NULL) {
171 				if (strcmp(fsp->fs_type, FSTAB_SW) != 0)
172 					continue;
173 				if (strstr(fsp->fs_mntops, "noauto") != NULL)
174 					continue;
175 				if (which_prog != SWAPOFF &&
176 				    strstr(fsp->fs_mntops, "late") &&
177 				    late == 0)
178 					continue;
179 				if (which_prog == SWAPOFF &&
180 				    strstr(fsp->fs_mntops, "late") == NULL &&
181 				    late != 0)
182 					continue;
183 				swfile = swap_on_off(fsp->fs_spec, 1,
184 				    fsp->fs_mntops);
185 				if (swfile == NULL) {
186 					ret = 1;
187 					continue;
188 				}
189 				if (qflag == 0) {
190 					printf("%s: %sing %s as swap device\n",
191 					    getprogname(),
192 					    (which_prog == SWAPOFF) ?
193 					    "remov" : "add", swfile);
194 				}
195 			}
196 		} else if (*argv == NULL)
197 			usage();
198 		for (; *argv; ++argv) {
199 			swfile = swap_on_off(*argv, 0, NULL);
200 			if (swfile == NULL) {
201 				ret = 1;
202 				continue;
203 			}
204 			if (orig_prog == SWAPCTL) {
205 				printf("%s: %sing %s as swap device\n",
206 				    getprogname(),
207 				    (which_prog == SWAPOFF) ? "remov" : "add",
208 				    swfile);
209 			}
210 		}
211 	} else {
212 		if (lflag || sflag)
213 			swaplist(lflag, sflag, hflag);
214 		else
215 			usage();
216 	}
217 	exit(ret);
218 }
219 
220 static const char *
221 swap_on_off(const char *name, int doingall, char *mntops)
222 {
223 	char *base, *basebuf;
224 
225 	/* Swap on vnode-backed md(4) device. */
226 	if (mntops != NULL &&
227 	    (fnmatch(_PATH_DEV MD_NAME "[0-9]*", name, 0) == 0 ||
228 	     fnmatch(MD_NAME "[0-9]*", name, 0) == 0 ||
229 	     strncmp(_PATH_DEV MD_NAME, name,
230 		sizeof(_PATH_DEV) + sizeof(MD_NAME)) == 0 ||
231 	     strncmp(MD_NAME, name, sizeof(MD_NAME)) == 0))
232 		return (swap_on_off_md(name, mntops, doingall));
233 
234 	basebuf = strdup(name);
235 	base = basename(basebuf);
236 
237 	/* Swap on encrypted device by GEOM_BDE. */
238 	if (fnmatch("*.bde", base, 0) == 0) {
239 		free(basebuf);
240 		return (swap_on_off_gbde(name, doingall));
241 	}
242 
243 	/* Swap on encrypted device by GEOM_ELI. */
244 	if (fnmatch("*.eli", base, 0) == 0) {
245 		free(basebuf);
246 		return (swap_on_off_geli(name, mntops, doingall));
247 	}
248 
249 	/* Swap on special file. */
250 	free(basebuf);
251 	return (swap_on_off_sfile(name, doingall));
252 }
253 
254 /* Strip off .bde or .eli suffix from swap device name */
255 static char *
256 swap_basename(const char *name)
257 {
258 	char *dname, *p;
259 
260 	dname = strdup(name);
261 	p = strrchr(dname, '.');
262 	/* assert(p != NULL); */
263 	*p = '\0';
264 
265 	return (dname);
266 }
267 
268 static const char *
269 swap_on_off_gbde(const char *name, int doingall)
270 {
271 	const char *ret;
272 	char pass[64 * 2 + 1];
273 	unsigned char bpass[64];
274 	char *dname;
275 	int i, error;
276 
277 	dname = swap_basename(name);
278 	if (dname == NULL)
279 		return (NULL);
280 
281 	if (which_prog == SWAPON) {
282 		arc4random_buf(bpass, sizeof(bpass));
283 		for (i = 0; i < (int)sizeof(bpass); i++)
284 			sprintf(&pass[2 * i], "%02x", bpass[i]);
285 		pass[sizeof(pass) - 1] = '\0';
286 
287 		error = run_cmd(NULL, "%s init %s -P %s", _PATH_GBDE,
288 		    dname, pass);
289 		if (error) {
290 			/* bde device found.  Ignore it. */
291 			free(dname);
292 			if (qflag == 0)
293 				warnx("%s: Device already in use", name);
294 			return (NULL);
295 		}
296 		error = run_cmd(NULL, "%s attach %s -p %s", _PATH_GBDE,
297 		    dname, pass);
298 		free(dname);
299 		if (error) {
300 			warnx("gbde (attach) error: %s", name);
301 			return (NULL);
302 		}
303 	}
304 
305 	ret = swap_on_off_sfile(name, doingall);
306 
307 	if (which_prog == SWAPOFF) {
308 		error = run_cmd(NULL, "%s detach %s", _PATH_GBDE, dname);
309 		free(dname);
310 		if (error) {
311 			/* bde device not found.  Ignore it. */
312 			if (qflag == 0)
313 				warnx("%s: Device not found", name);
314 			return (NULL);
315 		}
316 	}
317 
318 	return (ret);
319 }
320 
321 /* Build geli(8) arguments from mntops */
322 static char *
323 swap_on_geli_args(const char *mntops)
324 {
325 	const char *aalgo, *ealgo, *keylen_str, *sectorsize_str;
326 	const char *aflag, *eflag, *lflag, *Tflag, *sflag;
327 	char *p, *args, *token, *string, *ops;
328 	int pagesize;
329 	size_t pagesize_len;
330 	u_long ul;
331 
332 	/* Use built-in defaults for geli(8). */
333 	aalgo = ealgo = keylen_str = "";
334 	aflag = eflag = lflag = Tflag = "";
335 
336 	/* We will always specify sectorsize. */
337 	sflag = " -s ";
338 	sectorsize_str = NULL;
339 
340 	if (mntops != NULL) {
341 		string = ops = strdup(mntops);
342 
343 		while ((token = strsep(&string, ",")) != NULL) {
344 			if ((p = strstr(token, "aalgo=")) == token) {
345 				aalgo = p + sizeof("aalgo=") - 1;
346 				aflag = " -a ";
347 			} else if ((p = strstr(token, "ealgo=")) == token) {
348 				ealgo = p + sizeof("ealgo=") - 1;
349 				eflag = " -e ";
350 			} else if ((p = strstr(token, "keylen=")) == token) {
351 				keylen_str = p + sizeof("keylen=") - 1;
352 				errno = 0;
353 				ul = strtoul(keylen_str, &p, 10);
354 				if (errno == 0) {
355 					if (*p != '\0' || ul > INT_MAX)
356 						errno = EINVAL;
357 				}
358 				if (errno) {
359 					warn("Invalid keylen: %s", keylen_str);
360 					free(ops);
361 					return (NULL);
362 				}
363 				lflag = " -l ";
364 			} else if ((p = strstr(token, "sectorsize=")) == token) {
365 				sectorsize_str = p + sizeof("sectorsize=") - 1;
366 				errno = 0;
367 				ul = strtoul(sectorsize_str, &p, 10);
368 				if (errno == 0) {
369 					if (*p != '\0' || ul > INT_MAX)
370 						errno = EINVAL;
371 				}
372 				if (errno) {
373 					warn("Invalid sectorsize: %s",
374 					    sectorsize_str);
375 					free(ops);
376 					return (NULL);
377 				}
378 			} else if ((p = strstr(token, "notrim")) == token) {
379 				Tflag = " -T ";
380 			} else if (strcmp(token, "sw") != 0) {
381 				warnx("Invalid option: %s", token);
382 				free(ops);
383 				return (NULL);
384 			}
385 		}
386 	} else
387 		ops = NULL;
388 
389 	/*
390 	 * If we do not have a sector size at this point, fill in
391 	 * pagesize as sector size.
392 	 */
393 	if (sectorsize_str == NULL) {
394 		/* Use pagesize as default sectorsize. */
395 		pagesize = getpagesize();
396 		pagesize_len = snprintf(NULL, 0, "%d", pagesize) + 1;
397 		p = alloca(pagesize_len);
398 		snprintf(p, pagesize_len, "%d", pagesize);
399 		sectorsize_str = p;
400 	}
401 
402 	(void)asprintf(&args, "%s%s%s%s%s%s%s%s%s -d",
403 	    aflag, aalgo, eflag, ealgo, lflag, keylen_str, Tflag,
404 	    sflag, sectorsize_str);
405 
406 	free(ops);
407 	return (args);
408 }
409 
410 static const char *
411 swap_on_off_geli(const char *name, char *mntops, int doingall)
412 {
413 	struct stat sb;
414 	char *dname, *args;
415 	int error;
416 
417 	error = stat(name, &sb);
418 
419 	if (which_prog == SWAPON) do {
420 		/* Skip if the .eli device already exists. */
421 		if (error == 0)
422 			break;
423 
424 		args = swap_on_geli_args(mntops);
425 		if (args == NULL)
426 			return (NULL);
427 
428 		dname = swap_basename(name);
429 		if (dname == NULL) {
430 			free(args);
431 			return (NULL);
432 		}
433 
434 		error = run_cmd(NULL, "%s onetime%s %s", _PATH_GELI, args,
435 		    dname);
436 
437 		free(dname);
438 		free(args);
439 
440 		if (error) {
441 			/* error occurred during creation. */
442 			if (qflag == 0)
443 				warnx("%s: Invalid parameters", name);
444 			return (NULL);
445 		}
446 	} while (0);
447 
448 	return (swap_on_off_sfile(name, doingall));
449 }
450 
451 static const char *
452 swap_on_off_md(const char *name, char *mntops, int doingall)
453 {
454 	FILE *sfd;
455 	int fd, mdunit, error;
456 	const char *ret;
457 	static char mdpath[PATH_MAX], linebuf[PATH_MAX];
458 	char *p, *vnodefile;
459 	size_t linelen;
460 	u_long ul;
461 
462 	fd = -1;
463 	sfd = NULL;
464 	if (strlen(name) == (sizeof(MD_NAME) - 1))
465 		mdunit = -1;
466 	else {
467 		errno = 0;
468 		ul = strtoul(name + 2, &p, 10);
469 		if (errno == 0) {
470 			if (*p != '\0' || ul > INT_MAX)
471 				errno = EINVAL;
472 		}
473 		if (errno) {
474 			warn("Bad device unit: %s", name);
475 			return (NULL);
476 		}
477 		mdunit = (int)ul;
478 	}
479 
480 	vnodefile = NULL;
481 	if ((p = strstr(mntops, "file=")) != NULL) {
482 		vnodefile = strdup(p + sizeof("file=") - 1);
483 		p = strchr(vnodefile, ',');
484 		if (p != NULL)
485 			*p = '\0';
486 	}
487 	if (vnodefile == NULL) {
488 		warnx("file option not found for %s", name);
489 		return (NULL);
490 	}
491 
492 	if (which_prog == SWAPON) {
493 		if (mdunit == -1) {
494 			error = run_cmd(&fd, "%s -l -n -f %s",
495 			    _PATH_MDCONFIG, vnodefile);
496 			if (error == 0) {
497 				/* md device found.  Ignore it. */
498 				close(fd);
499 				if (!qflag)
500 					warnx("%s: Device already in use",
501 					    vnodefile);
502 				free(vnodefile);
503 				return (NULL);
504 			}
505 			error = run_cmd(&fd, "%s -a -t vnode -n -f %s",
506 			    _PATH_MDCONFIG, vnodefile);
507 			if (error) {
508 				warnx("mdconfig (attach) error: file=%s",
509 				    vnodefile);
510 				free(vnodefile);
511 				return (NULL);
512 			}
513 			sfd = fdopen(fd, "r");
514 			if (sfd == NULL) {
515 				warn("mdconfig (attach) fdopen error");
516 				ret = NULL;
517 				goto err;
518 			}
519 			p = fgetln(sfd, &linelen);
520 			if (p == NULL &&
521 			    (linelen < 2 || linelen > sizeof(linebuf))) {
522 				warn("mdconfig (attach) unexpected output");
523 				ret = NULL;
524 				goto err;
525 			}
526 			strncpy(linebuf, p, linelen);
527 			linebuf[linelen - 1] = '\0';
528 			errno = 0;
529 			ul = strtoul(linebuf, &p, 10);
530 			if (errno == 0) {
531 				if (*p != '\0' || ul > INT_MAX)
532 					errno = EINVAL;
533 			}
534 			if (errno) {
535 				warn("mdconfig (attach) unexpected output: %s",
536 				    linebuf);
537 				ret = NULL;
538 				goto err;
539 			}
540 			mdunit = (int)ul;
541 		} else {
542 			error = run_cmd(&fd, "%s -l -n -f %s -u %d",
543 			    _PATH_MDCONFIG, vnodefile, mdunit);
544 			if (error == 0) {
545 				/* md device found.  Ignore it. */
546 				close(fd);
547 				if (qflag == 0)
548 					warnx("md%d on %s: Device already "
549 					    "in use", mdunit, vnodefile);
550 				free(vnodefile);
551 				return (NULL);
552 			}
553 			error = run_cmd(NULL, "%s -a -t vnode -u %d -f %s",
554 			    _PATH_MDCONFIG, mdunit, vnodefile);
555 			if (error) {
556 				warnx("mdconfig (attach) error: "
557 				    "md%d on file=%s", mdunit, vnodefile);
558 				free(vnodefile);
559 				return (NULL);
560 			}
561 		}
562 	} else /* SWAPOFF */ {
563 		if (mdunit == -1) {
564 			error = run_cmd(&fd, "%s -l -n -f %s",
565 			    _PATH_MDCONFIG, vnodefile);
566 			if (error) {
567 				/* md device not found.  Ignore it. */
568 				close(fd);
569 				if (!qflag)
570 					warnx("md on %s: Device not found",
571 					    vnodefile);
572 				free(vnodefile);
573 				return (NULL);
574 			}
575 			sfd = fdopen(fd, "r");
576 			if (sfd == NULL) {
577 				warn("mdconfig (list) fdopen error");
578 				ret = NULL;
579 				goto err;
580 			}
581 			p = fgetln(sfd, &linelen);
582 			if (p == NULL &&
583 			    (linelen < 2 || linelen > sizeof(linebuf) - 1)) {
584 				warn("mdconfig (list) unexpected output");
585 				ret = NULL;
586 				goto err;
587 			}
588 			strncpy(linebuf, p, linelen);
589 			linebuf[linelen - 1] = '\0';
590 			p = strchr(linebuf, ' ');
591 			if (p != NULL)
592 				*p = '\0';
593 			errno = 0;
594 			ul = strtoul(linebuf, &p, 10);
595 			if (errno == 0) {
596 				if (*p != '\0' || ul > INT_MAX)
597 					errno = EINVAL;
598 			}
599 			if (errno) {
600 				warn("mdconfig (list) unexpected output: %s",
601 				    linebuf);
602 				ret = NULL;
603 				goto err;
604 			}
605 			mdunit = (int)ul;
606 		} else {
607 			error = run_cmd(&fd, "%s -l -n -f %s -u %d",
608 			    _PATH_MDCONFIG, vnodefile, mdunit);
609 			if (error) {
610 				/* md device not found.  Ignore it. */
611 				close(fd);
612 				if (!qflag)
613 					warnx("md%d on %s: Device not found",
614 					    mdunit, vnodefile);
615 				free(vnodefile);
616 				return (NULL);
617 			}
618 		}
619 	}
620 	snprintf(mdpath, sizeof(mdpath), "%s%s%d", _PATH_DEV,
621 	    MD_NAME, mdunit);
622 	mdpath[sizeof(mdpath) - 1] = '\0';
623 	ret = swap_on_off_sfile(mdpath, doingall);
624 
625 	if (which_prog == SWAPOFF) {
626 		if (ret != NULL) {
627 			error = run_cmd(NULL, "%s -d -u %d",
628 			    _PATH_MDCONFIG, mdunit);
629 			if (error)
630 				warn("mdconfig (detach) detach failed: %s%s%d",
631 				    _PATH_DEV, MD_NAME, mdunit);
632 		}
633 	}
634 err:
635 	if (sfd != NULL)
636 		fclose(sfd);
637 	if (fd != -1)
638 		close(fd);
639 	free(vnodefile);
640 	return (ret);
641 }
642 
643 static int
644 run_cmd(int *ofd, const char *cmdline, ...)
645 {
646 	va_list ap;
647 	char **argv, **argvp, *cmd, *p;
648 	int argc, pid, status, rv;
649 	int pfd[2], nfd, dup2dn;
650 
651 	va_start(ap, cmdline);
652 	rv = vasprintf(&cmd, cmdline, ap);
653 	if (rv == -1) {
654 		warn("%s", __func__);
655 		va_end(ap);
656 		return (rv);
657 	}
658 	va_end(ap);
659 
660 	for (argc = 1, p = cmd; (p = strchr(p, ' ')) != NULL; p++)
661 		argc++;
662 	argv = (char **)malloc(sizeof(*argv) * (argc + 1));
663 	for (p = cmd, argvp = argv; (*argvp = strsep(&p, " ")) != NULL;)
664 		if (**argvp != '\0' && (++argvp > &argv[argc])) {
665 			*argvp = NULL;
666 			break;
667 		}
668 	/* The argv array ends up NULL-terminated here. */
669 #if 0
670 	{
671 		int i;
672 
673 		fprintf(stderr, "DEBUG: running:");
674 		/* Should be equivalent to 'cmd' (before strsep, of course). */
675 		for (i = 0; argv[i] != NULL; i++)
676 			fprintf(stderr, " %s", argv[i]);
677 		fprintf(stderr, "\n");
678 	}
679 #endif
680 	dup2dn = 1;
681 	if (ofd != NULL) {
682 		if (pipe(&pfd[0]) == -1) {
683 			warn("%s: pipe", __func__);
684 			return (-1);
685 		}
686 		*ofd = pfd[0];
687 		dup2dn = 0;
688 	}
689 	pid = fork();
690 	switch (pid) {
691 	case 0:
692 		/* Child process. */
693 		if (ofd != NULL)
694 			if (dup2(pfd[1], STDOUT_FILENO) < 0)
695 				err(1, "dup2 in %s", __func__);
696 		nfd = open(_PATH_DEVNULL, O_RDWR);
697 		if (nfd == -1)
698 			err(1, "%s: open %s", __func__, _PATH_DEVNULL);
699 		if (dup2(nfd, STDIN_FILENO) < 0)
700 			err(1, "%s: dup2", __func__);
701 		if (dup2dn && dup2(nfd, STDOUT_FILENO) < 0)
702 			err(1, "%s: dup2", __func__);
703 		if (dup2(nfd, STDERR_FILENO) < 0)
704 			err(1, "%s: dup2", __func__);
705 		execv(argv[0], argv);
706 		warn("exec: %s", argv[0]);
707 		_exit(-1);
708 	case -1:
709 		err(1, "%s: fork", __func__);
710 	}
711 	free(cmd);
712 	free(argv);
713 	while (waitpid(pid, &status, 0) != pid)
714 		;
715 	return (WEXITSTATUS(status));
716 }
717 
718 static const char *
719 swap_on_off_sfile(const char *name, int doingall)
720 {
721 	int error;
722 
723 	if (which_prog == SWAPON)
724 		error = swapon(name);
725 	else /* SWAPOFF */
726 		error = swapoff(name);
727 
728 	if (error == -1) {
729 		switch (errno) {
730 		case EBUSY:
731 			if (doingall == 0)
732 				warnx("%s: Device already in use", name);
733 			break;
734 		case EINVAL:
735 			if (which_prog == SWAPON)
736 				warnx("%s: NSWAPDEV limit reached", name);
737 			else if (doingall == 0)
738 				warn("%s", name);
739 			break;
740 		default:
741 			warn("%s", name);
742 			break;
743 		}
744 		return (NULL);
745 	}
746 	return (name);
747 }
748 
749 static void
750 usage(void)
751 {
752 
753 	fprintf(stderr, "usage: %s ", getprogname());
754 	switch(orig_prog) {
755 	case SWAPON:
756 	case SWAPOFF:
757 	    fprintf(stderr, "[-F fstab] -aLq | file ...\n");
758 	    break;
759 	case SWAPCTL:
760 	    fprintf(stderr, "[-AghklmsU] [-a file ... | -d file ...]\n");
761 	    break;
762 	}
763 	exit(1);
764 }
765 
766 static void
767 sizetobuf(char *buf, size_t bufsize, int hflag, long long val, int hlen,
768     long blocksize)
769 {
770 	char tmp[16];
771 
772 	if (hflag == 'H') {
773 		humanize_number(tmp, 5, (int64_t)val, "", HN_AUTOSCALE,
774 		    HN_B | HN_NOSPACE | HN_DECIMAL);
775 		snprintf(buf, bufsize, "%*s", hlen, tmp);
776 	} else
777 		snprintf(buf, bufsize, "%*lld", hlen, val / blocksize);
778 }
779 
780 static void
781 swaplist(int lflag, int sflag, int hflag)
782 {
783 	size_t mibsize, size;
784 	struct xswdev xsw;
785 	int hlen, mib[16], n, pagesize;
786 	long blocksize;
787 	long long total = 0;
788 	long long used = 0;
789 	long long tmp_total;
790 	long long tmp_used;
791 	char buf[32];
792 
793 	pagesize = getpagesize();
794 	switch(hflag) {
795 	case 'G':
796 		blocksize = 1024 * 1024 * 1024;
797 		strlcpy(buf, "1GB-blocks", sizeof(buf));
798 		hlen = 10;
799 		break;
800 	case 'H':
801 		blocksize = -1;
802 		strlcpy(buf, "Bytes", sizeof(buf));
803 		hlen = 10;
804 		break;
805 	case 'K':
806 		blocksize = 1024;
807 		strlcpy(buf, "1kB-blocks", sizeof(buf));
808 		hlen = 10;
809 		break;
810 	case 'M':
811 		blocksize = 1024 * 1024;
812 		strlcpy(buf, "1MB-blocks", sizeof(buf));
813 		hlen = 10;
814 		break;
815 	default:
816 		getbsize(&hlen, &blocksize);
817 		snprintf(buf, sizeof(buf), "%ld-blocks", blocksize);
818 		break;
819 	}
820 
821 	mibsize = nitems(mib);
822 	if (sysctlnametomib("vm.swap_info", mib, &mibsize) == -1)
823 		err(1, "sysctlnametomib()");
824 
825 	if (lflag) {
826 		printf("%-13s %*s %*s\n",
827 		    "Device:",
828 		    hlen, buf,
829 		    hlen, "Used:");
830 	}
831 
832 	for (n = 0; ; ++n) {
833 		mib[mibsize] = n;
834 		size = sizeof xsw;
835 		if (sysctl(mib, mibsize + 1, &xsw, &size, NULL, 0) == -1)
836 			break;
837 		if (xsw.xsw_version != XSWDEV_VERSION)
838 			errx(1, "xswdev version mismatch");
839 
840 		tmp_total = (long long)xsw.xsw_nblks * pagesize;
841 		tmp_used  = (long long)xsw.xsw_used * pagesize;
842 		total += tmp_total;
843 		used  += tmp_used;
844 		if (lflag) {
845 			sizetobuf(buf, sizeof(buf), hflag, tmp_total, hlen,
846 			    blocksize);
847 			printf("/dev/%-8s %s ", devname(xsw.xsw_dev, S_IFCHR),
848 			    buf);
849 			sizetobuf(buf, sizeof(buf), hflag, tmp_used, hlen,
850 			    blocksize);
851 			printf("%s\n", buf);
852 		}
853 	}
854 	if (errno != ENOENT)
855 		err(1, "sysctl()");
856 
857 	if (sflag) {
858 		sizetobuf(buf, sizeof(buf), hflag, total, hlen, blocksize);
859 		printf("Total:        %s ", buf);
860 		sizetobuf(buf, sizeof(buf), hflag, used, hlen, blocksize);
861 		printf("%s\n", buf);
862 	}
863 }
864 
865