sbin/swapon/swapon.c

   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  * 4. 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/stat.h>
  45 #include <sys/param.h>
  46 #include <sys/sysctl.h>
  47 #include <vm/vm_param.h>
  48
  49 #include <err.h>
  50 #include <errno.h>
  51 #include <fstab.h>
  52 #include <stdio.h>
  53 #include <stdlib.h>
  54 #include <string.h>
  55 #include <unistd.h>
  56 #include <fcntl.h>
  57 #include <libutil.h>
  58
  59 static void usage(void);
  60 static int swap_on_off(char *name, int ignoreebusy);
  61 static void swaplist(int, int, int);
  62
  63 enum { SWAPON, SWAPOFF, SWAPCTL } orig_prog, which_prog = SWAPCTL;
  64
  65 int
  66 main(int argc, char **argv)
  67 {
  68         struct fstab *fsp;
  69         char *ptr;
  70         int stat;
  71         int ch, doall;
  72         int sflag = 0, lflag = 0, hflag = 0, qflag = 0;
  73
  74         if ((ptr = strrchr(argv[0], '/')) == NULL)
  75                 ptr = argv[0];
  76         if (strstr(ptr, "swapon"))
  77                 which_prog = SWAPON;
  78         else if (strstr(ptr, "swapoff"))
  79                 which_prog = SWAPOFF;
  80         orig_prog = which_prog;
  81
  82         doall = 0;
  83         while ((ch = getopt(argc, argv, "AadghklmqsU")) != -1) {
  84                 switch(ch) {
  85                 case 'A':
  86                         if (which_prog == SWAPCTL) {
  87                                 doall = 1;
  88                                 which_prog = SWAPON;
  89                         } else {
  90                                 usage();
  91                         }
  92                         break;
  93                 case 'a':
  94                         if (which_prog == SWAPON || which_prog == SWAPOFF)
  95                                 doall = 1;
  96                         else
  97                                 which_prog = SWAPON;
  98                         break;
  99                 case 'd':
 100                         if (which_prog == SWAPCTL)
 101                                 which_prog = SWAPOFF;
 102                         else
 103                                 usage();
 104                         break;
 105                 case 'g':
 106                         hflag = 'G';
 107                         break;
 108                 case 'h':
 109                         hflag = 'H';
 110                         break;
 111                 case 'k':
 112                         hflag = 'K';
 113                         break;
 114                 case 'l':
 115                         lflag = 1;
 116                         break;
 117                 case 'm':
 118                         hflag = 'M';
 119                         break;
 120                 case 'q':
 121                         if (which_prog == SWAPON || which_prog == SWAPOFF)
 122                                 qflag = 1;
 123                         break;
 124                 case 's':
 125                         sflag = 1;
 126                         break;
 127                 case 'U':
 128                         if (which_prog == SWAPCTL) {
 129                                 doall = 1;
 130                                 which_prog = SWAPOFF;
 131                         } else {
 132                                 usage();
 133                         }
 134                         break;
 135                 case '?':
 136                 default:
 137                         usage();
 138                 }
 139         }
 140         argv += optind;
 141
 142         stat = 0;
 143         if (which_prog == SWAPON || which_prog == SWAPOFF) {
 144                 if (doall) {
 145                         while ((fsp = getfsent()) != NULL) {
 146                                 if (strcmp(fsp->fs_type, FSTAB_SW))
 147                                         continue;
 148                                 if (strstr(fsp->fs_mntops, "noauto"))
 149                                         continue;
 150                                 if (swap_on_off(fsp->fs_spec, 1)) {
 151                                         stat = 1;
 152                                 } else {
 153                                         if (!qflag) {
 154                                                 printf("%s: %sing %s as swap device\n",
 155                                                     getprogname(),
 156                                                     which_prog == SWAPOFF ? "remov" : "add",
 157                                                     fsp->fs_spec);
 158                                         }
 159                                 }
 160                         }
 161                 }
 162                 else if (!*argv)
 163                         usage();
 164                 for (; *argv; ++argv) {
 165                         if (swap_on_off(*argv, 0)) {
 166                                 stat = 1;
 167                         } else if (orig_prog == SWAPCTL) {
 168                                 printf("%s: %sing %s as swap device\n",
 169                                     getprogname(), which_prog == SWAPOFF ? "remov" : "add",
 170                                     *argv);
 171                         }
 172                 }
 173         } else {
 174                 if (lflag || sflag)
 175                         swaplist(lflag, sflag, hflag);
 176                 else
 177                         usage();
 178         }
 179         exit(stat);
 180 }
 181
 182 static int
 183 swap_on_off(char *name, int doingall)
 184 {
 185         if ((which_prog == SWAPOFF ? swapoff(name) : swapon(name)) == -1) {
 186                 switch (errno) {
 187                 case EBUSY:
 188                         if (!doingall)
 189                                 warnx("%s: device already in use", name);
 190                         break;
 191                 case EINVAL:
 192                         if (which_prog == SWAPON)
 193                                 warnx("%s: NSWAPDEV limit reached", name);
 194                         else if (!doingall)
 195                                 warn("%s", name);
 196                         break;
 197                 default:
 198                         warn("%s", name);
 199                         break;
 200                 }
 201                 return(1);
 202         }
 203         return(0);
 204 }
 205
 206 static void
 207 usage(void)
 208 {
 209         fprintf(stderr, "usage: %s ", getprogname());
 210         switch(orig_prog) {
 211         case SWAPON:
 212         case SWAPOFF:
 213             fprintf(stderr, "-aq | file ...\n");
 214             break;
 215         case SWAPCTL:
 216             fprintf(stderr, "[-AghklmsU] [-a file ... | -d file ...]\n");
 217             break;
 218         }
 219         exit(1);
 220 }
 221
 222 static void
 223 sizetobuf(char *buf, size_t bufsize, int hflag, long long val, int hlen,
 224     long blocksize)
 225 {
 226
 227         if (hflag == 'H') {
 228                 char tmp[16];
 229
 230                 humanize_number(tmp, 5, (int64_t)val, "", HN_AUTOSCALE,
 231                     HN_B | HN_NOSPACE | HN_DECIMAL);
 232                 snprintf(buf, bufsize, "%*s", hlen, tmp);
 233         } else {
 234                 snprintf(buf, bufsize, "%*lld", hlen, val / blocksize);
 235         }
 236 }
 237
 238 static void
 239 swaplist(int lflag, int sflag, int hflag)
 240 {
 241         size_t mibsize, size;
 242         struct xswdev xsw;
 243         int hlen, mib[16], n, pagesize;
 244         long blocksize;
 245         long long total = 0;
 246         long long used = 0;
 247         long long tmp_total;
 248         long long tmp_used;
 249         char buf[32];
 250
 251         pagesize = getpagesize();
 252         switch(hflag) {
 253         case 'G':
 254             blocksize = 1024 * 1024 * 1024;
 255             strlcpy(buf, "1GB-blocks", sizeof(buf));
 256             hlen = 10;
 257             break;
 258         case 'H':
 259             blocksize = -1;
 260             strlcpy(buf, "Bytes", sizeof(buf));
 261             hlen = 10;
 262             break;
 263         case 'K':
 264             blocksize = 1024;
 265             strlcpy(buf, "1kB-blocks", sizeof(buf));
 266             hlen = 10;
 267             break;
 268         case 'M':
 269             blocksize = 1024 * 1024;
 270             strlcpy(buf, "1MB-blocks", sizeof(buf));
 271             hlen = 10;
 272             break;
 273         default:
 274             getbsize(&hlen, &blocksize);
 275             snprintf(buf, sizeof(buf), "%ld-blocks", blocksize);
 276             break;
 277         }
 278
 279         mibsize = sizeof mib / sizeof mib[0];
 280         if (sysctlnametomib("vm.swap_info", mib, &mibsize) == -1)
 281                 err(1, "sysctlnametomib()");
 282
 283         if (lflag) {
 284                 printf("%-13s %*s %*s\n",
 285                     "Device:",
 286                     hlen, buf,
 287                     hlen, "Used:");
 288         }
 289
 290         for (n = 0; ; ++n) {
 291                 mib[mibsize] = n;
 292                 size = sizeof xsw;
 293                 if (sysctl(mib, mibsize + 1, &xsw, &size, NULL, 0) == -1)
 294                         break;
 295                 if (xsw.xsw_version != XSWDEV_VERSION)
 296                         errx(1, "xswdev version mismatch");
 297
 298                 tmp_total = (long long)xsw.xsw_nblks * pagesize;
 299                 tmp_used  = (long long)xsw.xsw_used * pagesize;
 300                 total += tmp_total;
 301                 used  += tmp_used;
 302                 if (lflag) {
 303                         sizetobuf(buf, sizeof(buf), hflag, tmp_total, hlen,
 304                             blocksize);
 305                         printf("/dev/%-8s %s ", devname(xsw.xsw_dev, S_IFCHR),
 306                             buf);
 307                         sizetobuf(buf, sizeof(buf), hflag, tmp_used, hlen,
 308                             blocksize);
 309                         printf("%s\n", buf);
 310                 }
 311         }
 312         if (errno != ENOENT)
 313                 err(1, "sysctl()");
 314
 315         if (sflag) {
 316                 sizetobuf(buf, sizeof(buf), hflag, total, hlen, blocksize);
 317                 printf("Total:        %s ", buf);
 318                 sizetobuf(buf, sizeof(buf), hflag, used, hlen, blocksize);
 319                 printf("%s\n", buf);
 320         }
 321 }
 322