bin/df/df.c

   1 /*-
   2  * Copyright (c) 1980, 1990, 1993, 1994
   3  *      The Regents of the University of California.  All rights reserved.
   4  * (c) UNIX System Laboratories, Inc.
   5  * All or some portions of this file are derived from material licensed
   6  * to the University of California by American Telephone and Telegraph
   7  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
   8  * the permission of UNIX System Laboratories, Inc.
   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  * 4. Neither the name of the University nor the names of its contributors
  19  *    may be used to endorse or promote products derived from this software
  20  *    without specific prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  */
  34
  35 #if 0
  36 #ifndef lint
  37 static const char copyright[] =
  38 "@(#) Copyright (c) 1980, 1990, 1993, 1994\n\
  39         The Regents of the University of California.  All rights reserved.\n";
  40 #endif /* not lint */
  41
  42 #ifndef lint
  43 static char sccsid[] = "@(#)df.c        8.9 (Berkeley) 5/8/95";
  44 #endif /* not lint */
  45 #endif
  46 #include <sys/cdefs.h>
  47 __FBSDID("$FreeBSD$");
  48
  49 #include <sys/param.h>
  50 #include <sys/stat.h>
  51 #include <sys/mount.h>
  52 #include <sys/sysctl.h>
  53 #include <ufs/ufs/ufsmount.h>
  54 #include <err.h>
  55 #include <libutil.h>
  56 #include <stdint.h>
  57 #include <stdio.h>
  58 #include <stdlib.h>
  59 #include <string.h>
  60 #include <sysexits.h>
  61 #include <unistd.h>
  62
  63 #include "extern.h"
  64
  65 #define UNITS_SI        1
  66 #define UNITS_2         2
  67
  68 /* Maximum widths of various fields. */
  69 struct maxwidths {
  70         int     mntfrom;
  71         int     fstype;
  72         int     total;
  73         int     used;
  74         int     avail;
  75         int     iused;
  76         int     ifree;
  77 };
  78
  79 static void       addstat(struct statfs *, struct statfs *);
  80 static char      *getmntpt(const char *);
  81 static int        int64width(int64_t);
  82 static char      *makenetvfslist(void);
  83 static void       prthuman(const struct statfs *, int64_t);
  84 static void       prthumanval(int64_t);
  85 static intmax_t   fsbtoblk(int64_t, uint64_t, u_long);
  86 static void       prtstat(struct statfs *, struct maxwidths *);
  87 static size_t     regetmntinfo(struct statfs **, long, const char **);
  88 static void       update_maxwidths(struct maxwidths *, const struct statfs *);
  89 static void       usage(void);
  90
  91 static __inline int
  92 imax(int a, int b)
  93 {
  94         return (a > b ? a : b);
  95 }
  96
  97 static int      aflag = 0, cflag, hflag, iflag, kflag, lflag = 0, nflag, Tflag;
  98 static struct   ufs_args mdev;
  99
 100 int
 101 main(int argc, char *argv[])
 102 {
 103         struct stat stbuf;
 104         struct statfs statfsbuf, totalbuf;
 105         struct maxwidths maxwidths;
 106         struct statfs *mntbuf;
 107         const char *fstype;
 108         char *mntpath, *mntpt;
 109         const char **vfslist;
 110         int i, mntsize;
 111         int ch, rv;
 112
 113         fstype = "ufs";
 114
 115         memset(&totalbuf, 0, sizeof(totalbuf));
 116         totalbuf.f_bsize = DEV_BSIZE;
 117         strlcpy(totalbuf.f_mntfromname, "total", MNAMELEN);
 118         vfslist = NULL;
 119         while ((ch = getopt(argc, argv, "abcgHhiklmnPt:T")) != -1)
 120                 switch (ch) {
 121                 case 'a':
 122                         aflag = 1;
 123                         break;
 124                 case 'b':
 125                                 /* FALLTHROUGH */
 126                 case 'P':
 127                         /*
 128                          * POSIX specifically discusses the behavior of
 129                          * both -k and -P. It states that the blocksize should
 130                          * be set to 1024. Thus, if this occurs, simply break
 131                          * rather than clobbering the old blocksize.
 132                          */
 133                         if (kflag)
 134                                 break;
 135                         setenv("BLOCKSIZE", "512", 1);
 136                         hflag = 0;
 137                         break;
 138                 case 'c':
 139                         cflag = 1;
 140                         break;
 141                 case 'g':
 142                         setenv("BLOCKSIZE", "1g", 1);
 143                         hflag = 0;
 144                         break;
 145                 case 'H':
 146                         hflag = UNITS_SI;
 147                         break;
 148                 case 'h':
 149                         hflag = UNITS_2;
 150                         break;
 151                 case 'i':
 152                         iflag = 1;
 153                         break;
 154                 case 'k':
 155                         kflag++;
 156                         setenv("BLOCKSIZE", "1024", 1);
 157                         hflag = 0;
 158                         break;
 159                 case 'l':
 160                         if (vfslist != NULL)
 161                                 errx(1, "-l and -t are mutually exclusive.");
 162                         vfslist = makevfslist(makenetvfslist());
 163                         lflag = 1;
 164                         break;
 165                 case 'm':
 166                         setenv("BLOCKSIZE", "1m", 1);
 167                         hflag = 0;
 168                         break;
 169                 case 'n':
 170                         nflag = 1;
 171                         break;
 172                 case 't':
 173                         if (lflag)
 174                                 errx(1, "-l and -t are mutually exclusive.");
 175                         if (vfslist != NULL)
 176                                 errx(1, "only one -t option may be specified");
 177                         fstype = optarg;
 178                         vfslist = makevfslist(optarg);
 179                         break;
 180                 case 'T':
 181                         Tflag = 1;
 182                         break;
 183                 case '?':
 184                 default:
 185                         usage();
 186                 }
 187         argc -= optind;
 188         argv += optind;
 189
 190         rv = 0;
 191         if (!*argv) {
 192                 /* everything (modulo -t) */
 193                 mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
 194                 mntsize = regetmntinfo(&mntbuf, mntsize, vfslist);
 195         } else {
 196                 /* just the filesystems specified on the command line */
 197                 mntbuf = malloc(argc * sizeof(*mntbuf));
 198                 if (mntbuf == 0)
 199                         err(1, "malloc()");
 200                 mntsize = 0;
 201                 /* continued in for loop below */
 202         }
 203
 204         /* iterate through specified filesystems */
 205         for (; *argv; argv++) {
 206                 if (stat(*argv, &stbuf) < 0) {
 207                         if ((mntpt = getmntpt(*argv)) == 0) {
 208                                 warn("%s", *argv);
 209                                 rv = 1;
 210                                 continue;
 211                         }
 212                 } else if (S_ISCHR(stbuf.st_mode)) {
 213                         if ((mntpt = getmntpt(*argv)) == 0) {
 214                                 mdev.fspec = *argv;
 215                                 mntpath = strdup("/tmp/df.XXXXXX");
 216                                 if (mntpath == NULL) {
 217                                         warn("strdup failed");
 218                                         rv = 1;
 219                                         continue;
 220                                 }
 221                                 mntpt = mkdtemp(mntpath);
 222                                 if (mntpt == NULL) {
 223                                         warn("mkdtemp(\"%s\") failed", mntpath);
 224                                         rv = 1;
 225                                         free(mntpath);
 226                                         continue;
 227                                 }
 228                                 if (mount(fstype, mntpt, MNT_RDONLY,
 229                                     &mdev) != 0) {
 230                                         warn("%s", *argv);
 231                                         rv = 1;
 232                                         (void)rmdir(mntpt);
 233                                         free(mntpath);
 234                                         continue;
 235                                 } else if (statfs(mntpt, &statfsbuf) == 0) {
 236                                         statfsbuf.f_mntonname[0] = '\0';
 237                                         prtstat(&statfsbuf, &maxwidths);
 238                                         if (cflag)
 239                                                 addstat(&totalbuf, &statfsbuf);
 240                                 } else {
 241                                         warn("%s", *argv);
 242                                         rv = 1;
 243                                 }
 244                                 (void)unmount(mntpt, 0);
 245                                 (void)rmdir(mntpt);
 246                                 free(mntpath);
 247                                 continue;
 248                         }
 249                 } else
 250                         mntpt = *argv;
 251
 252                 /*
 253                  * Statfs does not take a `wait' flag, so we cannot
 254                  * implement nflag here.
 255                  */
 256                 if (statfs(mntpt, &statfsbuf) < 0) {
 257                         warn("%s", mntpt);
 258                         rv = 1;
 259                         continue;
 260                 }
 261
 262                 /*
 263                  * Check to make sure the arguments we've been given are
 264                  * satisfied.  Return an error if we have been asked to
 265                  * list a mount point that does not match the other args
 266                  * we've been given (-l, -t, etc.).
 267                  */
 268                 if (checkvfsname(statfsbuf.f_fstypename, vfslist)) {
 269                         rv = 1;
 270                         continue;
 271                 }
 272
 273                 /* the user asked for it, so ignore the ignore flag */
 274                 statfsbuf.f_flags &= ~MNT_IGNORE;
 275
 276                 /* add to list */
 277                 mntbuf[mntsize++] = statfsbuf;
 278         }
 279
 280         bzero(&maxwidths, sizeof(maxwidths));
 281         for (i = 0; i < mntsize; i++) {
 282                 if (aflag || (mntbuf[i].f_flags & MNT_IGNORE) == 0) {
 283                         update_maxwidths(&maxwidths, &mntbuf[i]);
 284                         if (cflag)
 285                                 addstat(&totalbuf, &mntbuf[i]);
 286                 }
 287         }
 288         for (i = 0; i < mntsize; i++)
 289                 if (aflag || (mntbuf[i].f_flags & MNT_IGNORE) == 0)
 290                         prtstat(&mntbuf[i], &maxwidths);
 291         if (cflag)
 292                 prtstat(&totalbuf, &maxwidths);
 293         return (rv);
 294 }
 295
 296 static char *
 297 getmntpt(const char *name)
 298 {
 299         size_t mntsize, i;
 300         struct statfs *mntbuf;
 301
 302         mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
 303         for (i = 0; i < mntsize; i++) {
 304                 if (!strcmp(mntbuf[i].f_mntfromname, name))
 305                         return (mntbuf[i].f_mntonname);
 306         }
 307         return (0);
 308 }
 309
 310 /*
 311  * Make a pass over the file system info in ``mntbuf'' filtering out
 312  * file system types not in vfslist and possibly re-stating to get
 313  * current (not cached) info.  Returns the new count of valid statfs bufs.
 314  */
 315 static size_t
 316 regetmntinfo(struct statfs **mntbufp, long mntsize, const char **vfslist)
 317 {
 318         int error, i, j;
 319         struct statfs *mntbuf;
 320
 321         if (vfslist == NULL)
 322                 return (nflag ? mntsize : getmntinfo(mntbufp, MNT_WAIT));
 323
 324         mntbuf = *mntbufp;
 325         for (j = 0, i = 0; i < mntsize; i++) {
 326                 if (checkvfsname(mntbuf[i].f_fstypename, vfslist))
 327                         continue;
 328                 /*
 329                  * XXX statfs(2) can fail for various reasons. It may be
 330                  * possible that the user does not have access to the
 331                  * pathname, if this happens, we will fall back on
 332                  * "stale" filesystem statistics.
 333                  */
 334                 error = statfs(mntbuf[i].f_mntonname, &mntbuf[j]);
 335                 if (nflag || error < 0)
 336                         if (i != j) {
 337                                 if (error < 0)
 338                                         warnx("%s stats possibly stale",
 339                                             mntbuf[i].f_mntonname);
 340                                 mntbuf[j] = mntbuf[i];
 341                         }
 342                 j++;
 343         }
 344         return (j);
 345 }
 346
 347 static void
 348 prthuman(const struct statfs *sfsp, int64_t used)
 349 {
 350
 351         prthumanval(sfsp->f_blocks * sfsp->f_bsize);
 352         prthumanval(used * sfsp->f_bsize);
 353         prthumanval(sfsp->f_bavail * sfsp->f_bsize);
 354 }
 355
 356 static void
 357 prthumanval(int64_t bytes)
 358 {
 359         char buf[6];
 360         int flags;
 361
 362         flags = HN_B | HN_NOSPACE | HN_DECIMAL;
 363         if (hflag == UNITS_SI)
 364                 flags |= HN_DIVISOR_1000;
 365
 366         humanize_number(buf, sizeof(buf) - (bytes < 0 ? 0 : 1),
 367             bytes, "", HN_AUTOSCALE, flags);
 368
 369         (void)printf("  %6s", buf);
 370 }
 371
 372 /*
 373  * Print an inode count in "human-readable" format.
 374  */
 375 static void
 376 prthumanvalinode(int64_t bytes)
 377 {
 378         char buf[6];
 379         int flags;
 380
 381         flags = HN_NOSPACE | HN_DECIMAL | HN_DIVISOR_1000;
 382
 383         humanize_number(buf, sizeof(buf) - (bytes < 0 ? 0 : 1),
 384             bytes, "", HN_AUTOSCALE, flags);
 385
 386         (void)printf(" %5s", buf);
 387 }
 388
 389 /*
 390  * Convert statfs returned file system size into BLOCKSIZE units.
 391  * Attempts to avoid overflow for large file systems.
 392  */
 393 static intmax_t
 394 fsbtoblk(int64_t num, uint64_t fsbs, u_long bs)
 395 {
 396
 397         if (fsbs != 0 && fsbs < bs)
 398                 return (num / (intmax_t)(bs / fsbs));
 399         else
 400                 return (num * (intmax_t)(fsbs / bs));
 401 }
 402
 403 /*
 404  * Print out status about a file system.
 405  */
 406 static void
 407 prtstat(struct statfs *sfsp, struct maxwidths *mwp)
 408 {
 409         static long blocksize;
 410         static int headerlen, timesthrough = 0;
 411         static const char *header;
 412         int64_t used, availblks, inodes;
 413
 414         if (++timesthrough == 1) {
 415                 mwp->mntfrom = imax(mwp->mntfrom, (int)strlen("Filesystem"));
 416                 mwp->fstype = imax(mwp->fstype, (int)strlen("Type"));
 417                 if (hflag) {
 418                         header = "   Size";
 419                         mwp->total = mwp->used = mwp->avail =
 420                             (int)strlen(header);
 421                 } else {
 422                         header = getbsize(&headerlen, &blocksize);
 423                         mwp->total = imax(mwp->total, headerlen);
 424                 }
 425                 mwp->used = imax(mwp->used, (int)strlen("Used"));
 426                 mwp->avail = imax(mwp->avail, (int)strlen("Avail"));
 427
 428                 (void)printf("%-*s", mwp->mntfrom, "Filesystem");
 429                 if (Tflag)
 430                         (void)printf("  %-*s", mwp->fstype, "Type");
 431                 (void)printf(" %-*s %*s %*s Capacity", mwp->total, header,
 432                     mwp->used, "Used", mwp->avail, "Avail");
 433                 if (iflag) {
 434                         mwp->iused = imax(hflag ? 0 : mwp->iused,
 435                             (int)strlen("  iused"));
 436                         mwp->ifree = imax(hflag ? 0 : mwp->ifree,
 437                             (int)strlen("ifree"));
 438                         (void)printf(" %*s %*s %%iused",
 439                             mwp->iused - 2, "iused", mwp->ifree, "ifree");
 440                 }
 441                 (void)printf("  Mounted on\n");
 442         }
 443         (void)printf("%-*s", mwp->mntfrom, sfsp->f_mntfromname);
 444         if (Tflag)
 445                 (void)printf("  %-*s", mwp->fstype, sfsp->f_fstypename);
 446         used = sfsp->f_blocks - sfsp->f_bfree;
 447         availblks = sfsp->f_bavail + used;
 448         if (hflag) {
 449                 prthuman(sfsp, used);
 450         } else {
 451                 (void)printf(" %*jd %*jd %*jd",
 452                     mwp->total, fsbtoblk(sfsp->f_blocks,
 453                     sfsp->f_bsize, blocksize),
 454                     mwp->used, fsbtoblk(used, sfsp->f_bsize, blocksize),
 455                     mwp->avail, fsbtoblk(sfsp->f_bavail,
 456                     sfsp->f_bsize, blocksize));
 457         }
 458         (void)printf(" %5.0f%%",
 459             availblks == 0 ? 100.0 : (double)used / (double)availblks * 100.0);
 460         if (iflag) {
 461                 inodes = sfsp->f_files;
 462                 used = inodes - sfsp->f_ffree;
 463                 if (hflag) {
 464                         (void)printf("  ");
 465                         prthumanvalinode(used);
 466                         prthumanvalinode(sfsp->f_ffree);
 467                 } else {
 468                         (void)printf(" %*jd %*jd", mwp->iused, (intmax_t)used,
 469                             mwp->ifree, (intmax_t)sfsp->f_ffree);
 470                 }
 471                 (void)printf(" %4.0f%% ", inodes == 0 ? 100.0 :
 472                     (double)used / (double)inodes * 100.0);
 473         } else
 474                 (void)printf("  ");
 475         if (strncmp(sfsp->f_mntfromname, "total", MNAMELEN) != 0)
 476                 (void)printf("  %s", sfsp->f_mntonname);
 477         (void)printf("\n");
 478 }
 479
 480 static void
 481 addstat(struct statfs *totalfsp, struct statfs *statfsp)
 482 {
 483         uint64_t bsize;
 484
 485         bsize = statfsp->f_bsize / totalfsp->f_bsize;
 486         totalfsp->f_blocks += statfsp->f_blocks * bsize;
 487         totalfsp->f_bfree += statfsp->f_bfree * bsize;
 488         totalfsp->f_bavail += statfsp->f_bavail * bsize;
 489         totalfsp->f_files += statfsp->f_files;
 490         totalfsp->f_ffree += statfsp->f_ffree;
 491 }
 492
 493 /*
 494  * Update the maximum field-width information in `mwp' based on
 495  * the file system specified by `sfsp'.
 496  */
 497 static void
 498 update_maxwidths(struct maxwidths *mwp, const struct statfs *sfsp)
 499 {
 500         static long blocksize = 0;
 501         int dummy;
 502
 503         if (blocksize == 0)
 504                 getbsize(&dummy, &blocksize);
 505
 506         mwp->mntfrom = imax(mwp->mntfrom, (int)strlen(sfsp->f_mntfromname));
 507         mwp->fstype = imax(mwp->fstype, (int)strlen(sfsp->f_fstypename));
 508         mwp->total = imax(mwp->total, int64width(
 509             fsbtoblk((int64_t)sfsp->f_blocks, sfsp->f_bsize, blocksize)));
 510         mwp->used = imax(mwp->used,
 511             int64width(fsbtoblk((int64_t)sfsp->f_blocks -
 512             (int64_t)sfsp->f_bfree, sfsp->f_bsize, blocksize)));
 513         mwp->avail = imax(mwp->avail, int64width(fsbtoblk(sfsp->f_bavail,
 514             sfsp->f_bsize, blocksize)));
 515         mwp->iused = imax(mwp->iused, int64width((int64_t)sfsp->f_files -
 516             sfsp->f_ffree));
 517         mwp->ifree = imax(mwp->ifree, int64width(sfsp->f_ffree));
 518 }
 519
 520 /* Return the width in characters of the specified value. */
 521 static int
 522 int64width(int64_t val)
 523 {
 524         int len;
 525
 526         len = 0;
 527         /* Negative or zero values require one extra digit. */
 528         if (val <= 0) {
 529                 val = -val;
 530                 len++;
 531         }
 532         while (val > 0) {
 533                 len++;
 534                 val /= 10;
 535         }
 536
 537         return (len);
 538 }
 539
 540 static void
 541 usage(void)
 542 {
 543
 544         (void)fprintf(stderr,
 545 "usage: df [-b | -g | -H | -h | -k | -m | -P] [-acilnT] [-t type] [file | filesystem ...]\n");
 546         exit(EX_USAGE);
 547 }
 548
 549 static char *
 550 makenetvfslist(void)
 551 {
 552         char *str, *strptr, **listptr;
 553         struct xvfsconf *xvfsp, *keep_xvfsp;
 554         size_t buflen;
 555         int cnt, i, maxvfsconf;
 556
 557         if (sysctlbyname("vfs.conflist", NULL, &buflen, NULL, 0) < 0) {
 558                 warn("sysctl(vfs.conflist)");
 559                 return (NULL);
 560         }
 561         xvfsp = malloc(buflen);
 562         if (xvfsp == NULL) {
 563                 warnx("malloc failed");
 564                 return (NULL);
 565         }
 566         keep_xvfsp = xvfsp;
 567         if (sysctlbyname("vfs.conflist", xvfsp, &buflen, NULL, 0) < 0) {
 568                 warn("sysctl(vfs.conflist)");
 569                 free(keep_xvfsp);
 570                 return (NULL);
 571         }
 572         maxvfsconf = buflen / sizeof(struct xvfsconf);
 573
 574         if ((listptr = malloc(sizeof(char*) * maxvfsconf)) == NULL) {
 575                 warnx("malloc failed");
 576                 free(keep_xvfsp);
 577                 return (NULL);
 578         }
 579
 580         for (cnt = 0, i = 0; i < maxvfsconf; i++) {
 581                 if (xvfsp->vfc_flags & VFCF_NETWORK) {
 582                         listptr[cnt++] = strdup(xvfsp->vfc_name);
 583                         if (listptr[cnt-1] == NULL) {
 584                                 warnx("malloc failed");
 585                                 free(listptr);
 586                                 free(keep_xvfsp);
 587                                 return (NULL);
 588                         }
 589                 }
 590                 xvfsp++;
 591         }
 592
 593         if (cnt == 0 ||
 594             (str = malloc(sizeof(char) * (32 * cnt + cnt + 2))) == NULL) {
 595                 if (cnt > 0)
 596                         warnx("malloc failed");
 597                 free(listptr);
 598                 free(keep_xvfsp);
 599                 return (NULL);
 600         }
 601
 602         *str = 'n'; *(str + 1) = 'o';
 603         for (i = 0, strptr = str + 2; i < cnt; i++, strptr++) {
 604                 strlcpy(strptr, listptr[i], 32);
 605                 strptr += strlen(listptr[i]);
 606                 *strptr = ',';
 607                 free(listptr[i]);
 608         }
 609         *(--strptr) = '\0';
 610
 611         free(keep_xvfsp);
 612         free(listptr);
 613         return (str);
 614 }