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     total;
  72         int     used;
  73         int     avail;
  74         int     iused;
  75         int     ifree;
  76 };
  77
  78 static void       addstat(struct statfs *, struct statfs *);
  79 static char      *getmntpt(const char *);
  80 static int        int64width(int64_t);
  81 static char      *makenetvfslist(void);
  82 static void       prthuman(const struct statfs *, int64_t);
  83 static void       prthumanval(int64_t);
  84 static intmax_t   fsbtoblk(int64_t, uint64_t, u_long);
  85 static void       prtstat(struct statfs *, struct maxwidths *);
  86 static size_t     regetmntinfo(struct statfs **, long, const char **);
  87 static void       update_maxwidths(struct maxwidths *, const struct statfs *);
  88 static void       usage(void);
  89
  90 static __inline int
  91 imax(int a, int b)
  92 {
  93         return (a > b ? a : b);
  94 }
  95
  96 static int      aflag = 0, cflag, hflag, iflag, nflag;
  97 static struct   ufs_args mdev;
  98
  99 int
 100 main(int argc, char *argv[])
 101 {
 102         struct stat stbuf;
 103         struct statfs statfsbuf, totalbuf;
 104         struct maxwidths maxwidths;
 105         struct statfs *mntbuf;
 106         const char *fstype;
 107         char *mntpath, *mntpt;
 108         const char **vfslist;
 109         size_t i, mntsize;
 110         int ch, rv;
 111
 112         fstype = "ufs";
 113
 114         memset(&totalbuf, 0, sizeof(totalbuf));
 115         totalbuf.f_bsize = DEV_BSIZE;
 116         strncpy(totalbuf.f_mntfromname, "total", MNAMELEN);
 117         vfslist = NULL;
 118         while ((ch = getopt(argc, argv, "abcgHhiklmnPt:")) != -1)
 119                 switch (ch) {
 120                 case 'a':
 121                         aflag = 1;
 122                         break;
 123                 case 'b':
 124                                 /* FALLTHROUGH */
 125                 case 'P':
 126                         putenv("BLOCKSIZE=512");
 127                         hflag = 0;
 128                         break;
 129                 case 'c':
 130                         cflag = 1;
 131                         break;
 132                 case 'g':
 133                         putenv("BLOCKSIZE=1g");
 134                         hflag = 0;
 135                         break;
 136                 case 'H':
 137                         hflag = UNITS_SI;
 138                         break;
 139                 case 'h':
 140                         hflag = UNITS_2;
 141                         break;
 142                 case 'i':
 143                         iflag = 1;
 144                         break;
 145                 case 'k':
 146                         putenv("BLOCKSIZE=1k");
 147                         hflag = 0;
 148                         break;
 149                 case 'l':
 150                         if (vfslist != NULL)
 151                                 errx(1, "-l and -t are mutually exclusive.");
 152                         vfslist = makevfslist(makenetvfslist());
 153                         break;
 154                 case 'm':
 155                         putenv("BLOCKSIZE=1m");
 156                         hflag = 0;
 157                         break;
 158                 case 'n':
 159                         nflag = 1;
 160                         break;
 161                 case 't':
 162                         if (vfslist != NULL)
 163                                 errx(1, "only one -t option may be specified");
 164                         fstype = optarg;
 165                         vfslist = makevfslist(optarg);
 166                         break;
 167                 case '?':
 168                 default:
 169                         usage();
 170                 }
 171         argc -= optind;
 172         argv += optind;
 173
 174         mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
 175         bzero(&maxwidths, sizeof(maxwidths));
 176         for (i = 0; i < mntsize; i++)
 177                 update_maxwidths(&maxwidths, &mntbuf[i]);
 178
 179         rv = 0;
 180         if (!*argv) {
 181                 mntsize = regetmntinfo(&mntbuf, mntsize, vfslist);
 182                 bzero(&maxwidths, sizeof(maxwidths));
 183                 for (i = 0; i < mntsize; i++) {
 184                         if (cflag)
 185                                 addstat(&totalbuf, &mntbuf[i]);
 186                         update_maxwidths(&maxwidths, &mntbuf[i]);
 187                 }
 188                 if (cflag)
 189                         update_maxwidths(&maxwidths, &totalbuf);
 190                 for (i = 0; i < mntsize; i++)
 191                         if (aflag || (mntbuf[i].f_flags & MNT_IGNORE) == 0)
 192                                 prtstat(&mntbuf[i], &maxwidths);
 193                 if (cflag)
 194                         prtstat(&totalbuf, &maxwidths);
 195                 exit(rv);
 196         }
 197
 198         for (; *argv; argv++) {
 199                 if (stat(*argv, &stbuf) < 0) {
 200                         if ((mntpt = getmntpt(*argv)) == 0) {
 201                                 warn("%s", *argv);
 202                                 rv = 1;
 203                                 continue;
 204                         }
 205                 } else if (S_ISCHR(stbuf.st_mode)) {
 206                         if ((mntpt = getmntpt(*argv)) == 0) {
 207                                 mdev.fspec = *argv;
 208                                 mntpath = strdup("/tmp/df.XXXXXX");
 209                                 if (mntpath == NULL) {
 210                                         warn("strdup failed");
 211                                         rv = 1;
 212                                         continue;
 213                                 }
 214                                 mntpt = mkdtemp(mntpath);
 215                                 if (mntpt == NULL) {
 216                                         warn("mkdtemp(\"%s\") failed", mntpath);
 217                                         rv = 1;
 218                                         free(mntpath);
 219                                         continue;
 220                                 }
 221                                 if (mount(fstype, mntpt, MNT_RDONLY,
 222                                     &mdev) != 0) {
 223                                         warn("%s", *argv);
 224                                         rv = 1;
 225                                         (void)rmdir(mntpt);
 226                                         free(mntpath);
 227                                         continue;
 228                                 } else if (statfs(mntpt, &statfsbuf) == 0) {
 229                                         statfsbuf.f_mntonname[0] = '\0';
 230                                         prtstat(&statfsbuf, &maxwidths);
 231                                         if (cflag)
 232                                                 addstat(&totalbuf, &statfsbuf);
 233                                 } else {
 234                                         warn("%s", *argv);
 235                                         rv = 1;
 236                                 }
 237                                 (void)unmount(mntpt, 0);
 238                                 (void)rmdir(mntpt);
 239                                 free(mntpath);
 240                                 continue;
 241                         }
 242                 } else
 243                         mntpt = *argv;
 244
 245                 /*
 246                  * Statfs does not take a `wait' flag, so we cannot
 247                  * implement nflag here.
 248                  */
 249                 if (statfs(mntpt, &statfsbuf) < 0) {
 250                         warn("%s", mntpt);
 251                         rv = 1;
 252                         continue;
 253                 }
 254
 255                 /*
 256                  * Check to make sure the arguments we've been given are
 257                  * satisfied.  Return an error if we have been asked to
 258                  * list a mount point that does not match the other args
 259                  * we've been given (-l, -t, etc.).
 260                  */
 261                 if (checkvfsname(statfsbuf.f_fstypename, vfslist)) {
 262                         rv = 1;
 263                         continue;
 264                 }
 265
 266                 if (argc == 1) {
 267                         bzero(&maxwidths, sizeof(maxwidths));
 268                         update_maxwidths(&maxwidths, &statfsbuf);
 269                 }
 270                 prtstat(&statfsbuf, &maxwidths);
 271                 if (cflag)
 272                         addstat(&totalbuf, &statfsbuf);
 273         }
 274         if (cflag)
 275                 prtstat(&totalbuf, &maxwidths);
 276         return (rv);
 277 }
 278
 279 static char *
 280 getmntpt(const char *name)
 281 {
 282         size_t mntsize, i;
 283         struct statfs *mntbuf;
 284
 285         mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
 286         for (i = 0; i < mntsize; i++) {
 287                 if (!strcmp(mntbuf[i].f_mntfromname, name))
 288                         return (mntbuf[i].f_mntonname);
 289         }
 290         return (0);
 291 }
 292
 293 /*
 294  * Make a pass over the file system info in ``mntbuf'' filtering out
 295  * file system types not in vfslist and possibly re-stating to get
 296  * current (not cached) info.  Returns the new count of valid statfs bufs.
 297  */
 298 static size_t
 299 regetmntinfo(struct statfs **mntbufp, long mntsize, const char **vfslist)
 300 {
 301         int error, i, j;
 302         struct statfs *mntbuf;
 303
 304         if (vfslist == NULL)
 305                 return (nflag ? mntsize : getmntinfo(mntbufp, MNT_WAIT));
 306
 307         mntbuf = *mntbufp;
 308         for (j = 0, i = 0; i < mntsize; i++) {
 309                 if (checkvfsname(mntbuf[i].f_fstypename, vfslist))
 310                         continue;
 311                 /*
 312                  * XXX statfs(2) can fail for various reasons. It may be
 313                  * possible that the user does not have access to the
 314                  * pathname, if this happens, we will fall back on
 315                  * "stale" filesystem statistics.
 316                  */
 317                 error = statfs(mntbuf[i].f_mntonname, &mntbuf[j]);
 318                 if (nflag || error < 0)
 319                         if (i != j) {
 320                                 if (error < 0)
 321                                         warnx("%s stats possibly stale",
 322                                             mntbuf[i].f_mntonname);
 323                                 mntbuf[j] = mntbuf[i];
 324                         }
 325                 j++;
 326         }
 327         return (j);
 328 }
 329
 330 static void
 331 prthuman(const struct statfs *sfsp, int64_t used)
 332 {
 333
 334         prthumanval(sfsp->f_blocks * sfsp->f_bsize);
 335         prthumanval(used * sfsp->f_bsize);
 336         prthumanval(sfsp->f_bavail * sfsp->f_bsize);
 337 }
 338
 339 static void
 340 prthumanval(int64_t bytes)
 341 {
 342         char buf[6];
 343         int flags;
 344
 345         flags = HN_B | HN_NOSPACE | HN_DECIMAL;
 346         if (hflag == UNITS_SI)
 347                 flags |= HN_DIVISOR_1000;
 348
 349         humanize_number(buf, sizeof(buf) - (bytes < 0 ? 0 : 1),
 350             bytes, "", HN_AUTOSCALE, flags);
 351
 352         (void)printf("  %6s", buf);
 353 }
 354
 355 /*
 356  * Convert statfs returned file system size into BLOCKSIZE units.
 357  * Attempts to avoid overflow for large file systems.
 358  */
 359 static intmax_t
 360 fsbtoblk(int64_t num, uint64_t fsbs, u_long bs)
 361 {
 362
 363         if (fsbs != 0 && fsbs < bs)
 364                 return (num / (intmax_t)(bs / fsbs));
 365         else
 366                 return (num * (intmax_t)(fsbs / bs));
 367 }
 368
 369 /*
 370  * Print out status about a file system.
 371  */
 372 static void
 373 prtstat(struct statfs *sfsp, struct maxwidths *mwp)
 374 {
 375         static u_long blocksize;
 376         static int headerlen, timesthrough = 0;
 377         static const char *header;
 378         int64_t used, availblks, inodes;
 379
 380         if (++timesthrough == 1) {
 381                 mwp->mntfrom = imax(mwp->mntfrom, (int)strlen("Filesystem"));
 382                 if (hflag) {
 383                         header = "   Size";
 384                         mwp->total = mwp->used = mwp->avail =
 385                             (int)strlen(header);
 386                 } else {
 387                         header = getbsize(&headerlen, &blocksize);
 388                         mwp->total = imax(mwp->total, headerlen);
 389                 }
 390                 mwp->used = imax(mwp->used, (int)strlen("Used"));
 391                 mwp->avail = imax(mwp->avail, (int)strlen("Avail"));
 392
 393                 (void)printf("%-*s %-*s %*s %*s Capacity",
 394                     mwp->mntfrom, "Filesystem", mwp->total, header,
 395                     mwp->used, "Used", mwp->avail, "Avail");
 396                 if (iflag) {
 397                         mwp->iused = imax(mwp->iused, (int)strlen("  iused"));
 398                         mwp->ifree = imax(mwp->ifree, (int)strlen("ifree"));
 399                         (void)printf(" %*s %*s %%iused",
 400                             mwp->iused - 2, "iused", mwp->ifree, "ifree");
 401                 }
 402                 (void)printf("  Mounted on\n");
 403         }
 404         (void)printf("%-*s", mwp->mntfrom, sfsp->f_mntfromname);
 405         used = sfsp->f_blocks - sfsp->f_bfree;
 406         availblks = sfsp->f_bavail + used;
 407         if (hflag) {
 408                 prthuman(sfsp, used);
 409         } else {
 410                 (void)printf(" %*jd %*jd %*jd",
 411                     mwp->total, fsbtoblk(sfsp->f_blocks,
 412                     sfsp->f_bsize, blocksize),
 413                     mwp->used, fsbtoblk(used, sfsp->f_bsize, blocksize),
 414                     mwp->avail, fsbtoblk(sfsp->f_bavail,
 415                     sfsp->f_bsize, blocksize));
 416         }
 417         (void)printf(" %5.0f%%",
 418             availblks == 0 ? 100.0 : (double)used / (double)availblks * 100.0);
 419         if (iflag) {
 420                 inodes = sfsp->f_files;
 421                 used = inodes - sfsp->f_ffree;
 422                 (void)printf(" %*jd %*jd %4.0f%% ", mwp->iused, (intmax_t)used,
 423                     mwp->ifree, (intmax_t)sfsp->f_ffree, inodes == 0 ? 100.0 :
 424                     (double)used / (double)inodes * 100.0);
 425         } else
 426                 (void)printf("  ");
 427         if (strncmp(sfsp->f_mntfromname, "total", MNAMELEN) != 0)
 428                 (void)printf("  %s", sfsp->f_mntonname);
 429         (void)printf("\n");
 430 }
 431
 432 void
 433 addstat(struct statfs *totalfsp, struct statfs *statfsp)
 434 {
 435         uint64_t bsize;
 436
 437         bsize = statfsp->f_bsize / totalfsp->f_bsize;
 438         totalfsp->f_blocks += statfsp->f_blocks * bsize;
 439         totalfsp->f_bfree += statfsp->f_bfree * bsize;
 440         totalfsp->f_bavail += statfsp->f_bavail * bsize;
 441         totalfsp->f_files += statfsp->f_files;
 442         totalfsp->f_ffree += statfsp->f_ffree;
 443 }
 444
 445 /*
 446  * Update the maximum field-width information in `mwp' based on
 447  * the file system specified by `sfsp'.
 448  */
 449 static void
 450 update_maxwidths(struct maxwidths *mwp, const struct statfs *sfsp)
 451 {
 452         static u_long blocksize = 0;
 453         int dummy;
 454
 455         if (blocksize == 0)
 456                 getbsize(&dummy, &blocksize);
 457
 458         mwp->mntfrom = imax(mwp->mntfrom, (int)strlen(sfsp->f_mntfromname));
 459         mwp->total = imax(mwp->total, int64width(
 460             fsbtoblk((int64_t)sfsp->f_blocks, sfsp->f_bsize, blocksize)));
 461         mwp->used = imax(mwp->used,
 462             int64width(fsbtoblk((int64_t)sfsp->f_blocks -
 463             (int64_t)sfsp->f_bfree, sfsp->f_bsize, blocksize)));
 464         mwp->avail = imax(mwp->avail, int64width(fsbtoblk(sfsp->f_bavail,
 465             sfsp->f_bsize, blocksize)));
 466         mwp->iused = imax(mwp->iused, int64width((int64_t)sfsp->f_files -
 467             sfsp->f_ffree));
 468         mwp->ifree = imax(mwp->ifree, int64width(sfsp->f_ffree));
 469 }
 470
 471 /* Return the width in characters of the specified value. */
 472 static int
 473 int64width(int64_t val)
 474 {
 475         int len;
 476
 477         len = 0;
 478         /* Negative or zero values require one extra digit. */
 479         if (val <= 0) {
 480                 val = -val;
 481                 len++;
 482         }
 483         while (val > 0) {
 484                 len++;
 485                 val /= 10;
 486         }
 487
 488         return (len);
 489 }
 490
 491 static void
 492 usage(void)
 493 {
 494
 495         (void)fprintf(stderr,
 496 "usage: df [-b | -g | -H | -h | -k | -m | -P] [-aciln] [-t type] [file | filesystem ...]\n");
 497         exit(EX_USAGE);
 498 }
 499
 500 static char *
 501 makenetvfslist(void)
 502 {
 503         char *str, *strptr, **listptr;
 504         struct xvfsconf *xvfsp, *keep_xvfsp;
 505         size_t buflen;
 506         int cnt, i, maxvfsconf;
 507
 508         if (sysctlbyname("vfs.conflist", NULL, &buflen, NULL, 0) < 0) {
 509                 warn("sysctl(vfs.conflist)");
 510                 return (NULL);
 511         }
 512         xvfsp = malloc(buflen);
 513         if (xvfsp == NULL) {
 514                 warnx("malloc failed");
 515                 return (NULL);
 516         }
 517         keep_xvfsp = xvfsp;
 518         if (sysctlbyname("vfs.conflist", xvfsp, &buflen, NULL, 0) < 0) {
 519                 warn("sysctl(vfs.conflist)");
 520                 free(keep_xvfsp);
 521                 return (NULL);
 522         }
 523         maxvfsconf = buflen / sizeof(struct xvfsconf);
 524
 525         if ((listptr = malloc(sizeof(char*) * maxvfsconf)) == NULL) {
 526                 warnx("malloc failed");
 527                 free(keep_xvfsp);
 528                 return (NULL);
 529         }
 530
 531         for (cnt = 0, i = 0; i < maxvfsconf; i++) {
 532                 if (xvfsp->vfc_flags & VFCF_NETWORK) {
 533                         listptr[cnt++] = strdup(xvfsp->vfc_name);
 534                         if (listptr[cnt-1] == NULL) {
 535                                 warnx("malloc failed");
 536                                 free(listptr);
 537                                 free(keep_xvfsp);
 538                                 return (NULL);
 539                         }
 540                 }
 541                 xvfsp++;
 542         }
 543
 544         if (cnt == 0 ||
 545             (str = malloc(sizeof(char) * (32 * cnt + cnt + 2))) == NULL) {
 546                 if (cnt > 0)
 547                         warnx("malloc failed");
 548                 free(listptr);
 549                 free(keep_xvfsp);
 550                 return (NULL);
 551         }
 552
 553         *str = 'n'; *(str + 1) = 'o';
 554         for (i = 0, strptr = str + 2; i < cnt; i++, strptr++) {
 555                 strncpy(strptr, listptr[i], 32);
 556                 strptr += strlen(listptr[i]);
 557                 *strptr = ',';
 558                 free(listptr[i]);
 559         }
 560         *(--strptr) = '\0';
 561
 562         free(keep_xvfsp);
 563         free(listptr);
 564         return (str);
 565 }