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