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