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
37static 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
43static char sccsid[] = "@(#)df.c 8.9 (Berkeley) 5/8/95";
44#endif /* not lint */
45#endif
46#include <sys/cdefs.h>
47__FBSDID("$FreeBSD: head/bin/df/df.c 139969 2005-01-10 08:39:26Z imp $");
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. */
69struct maxwidths {
70 int mntfrom;
71 int total;
72 int used;
73 int avail;
74 int iused;
75 int ifree;
76};
77
78static void addstat(struct statfs *, struct statfs *);
79static char *getmntpt(const char *);
80static int int64width(int64_t);
81static char *makenetvfslist(void);
82static void prthuman(const struct statfs *, int64_t);
83static void prthumanval(int64_t);
84static intmax_t fsbtoblk(int64_t, uint64_t, u_long);
85static void prtstat(struct statfs *, struct maxwidths *);
86static size_t regetmntinfo(struct statfs **, long, const char **);
87static void update_maxwidths(struct maxwidths *, const struct statfs *);
88static void usage(void);
89
90static __inline int
91imax(int a, int b)
92{
93 return (a > b ? a : b);
94}
95
96static int aflag = 0, cflag, hflag, iflag, nflag;
97static struct ufs_args mdev;
98
99int
100main(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
279static char *
280getmntpt(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 */
298static size_t
299regetmntinfo(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
330static void
331prthuman(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
339static void
340prthumanval(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 */
359static intmax_t
360fsbtoblk(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 */
372static void
373prtstat(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
432void
433addstat(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 */
449static void
450update_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. */
472static int
473int64width(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
491static void
492usage(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
500static char *
501makenetvfslist(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}