386BSD 0.1 development

[unix-history] / usr / src / bin / df / df.c

diff --git a/usr/src/bin/df/df.c b/usr/src/bin/df/df.c
index 759b645..1ed82c7 100644 (file)
--- a/usr/src/bin/df/df.c
+++ b/usr/src/bin/df/df.c
@@ -1,183 +1,298 @@
-static char *sccsid = "@(#)df.c        4.10 (Berkeley) %G%";
-
-#include <stdio.h>
-#include <fstab.h>
-#include <sys/param.h>
-#include <sys/filsys.h>
-#include <sys/fblk.h>
-#include <sys/stat.h>
-

 /*
 /*

- * df
+ * Copyright (c) 1980, 1990 The Regents of the University of California.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. All advertising materials mentioning features or use of this software
+ *    must display the following acknowledgement:
+ *     This product includes software developed by the University of
+ *     California, Berkeley and its contributors.
+ * 4. Neither the name of the University nor the names of its contributors
+ *    may be used to endorse or promote products derived from this software
+ *    without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.

  */
  */

-#define NFS    20      /* Max number of filesystems */

 
 

-struct {
-       char path[32];
-       char spec[32];
-} mtab[NFS];
-char   root[32];
-char   *mpath();
-daddr_t        blkno   = 1;
+#ifndef lint
+char copyright[] =
+"@(#) Copyright (c) 1980, 1990 The Regents of the University of California.\n\
+ All rights reserved.\n";
+#endif /* not lint */

 
 

-int    iflag;
+#ifndef lint
+static char sccsid[] = "@(#)df.c       5.24 (Berkeley) 3/6/91";
+#endif /* not lint */

 
 

-struct filsys sblock;
+/*
+ * df
+ */
+#include <sys/param.h>
+#include <sys/stat.h>
+#include <sys/mount.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>

 
 

-int    fi;
-daddr_t        alloc();
+char   *getmntpt();
+void   ufs_df(), prtstat();
+int    iflag, kflag, nflag;
+struct ufs_args mdev;

 
 

+int

 main(argc, argv)
        int argc;
        char **argv;
 {
 main(argc, argv)
        int argc;
        char **argv;
 {

-       int i;
-
-       while (argc >= 1 && argv[1][0]=='-') {
-       switch (argv[1][1]) {
+       extern int errno, optind;
+       int err, ch, i;
+       long width, maxwidth, mntsize;
+       char *mntpt, *mktemp();
+       struct stat stbuf;
+       struct statfs statfsbuf, *mntbuf;

 
 

-       case 'i':
-               iflag++;
-               break;
+       while ((ch = getopt(argc, argv, "ikn")) != EOF)
+               switch(ch) {
+               case 'i':
+                       iflag = 1;
+                       break;
+               case 'k':
+                       kflag = 1;
+                       break;
+               case 'n':
+                       nflag = 1;
+                       break;
+               case '?':
+               default:
+                       fprintf(stderr,
+                           "usage: df [-ikn] [file | file_system ...]\n");
+                       exit(1);
+               }
+       argc -= optind;
+       argv += optind;

 
 

-       default:
-               fprintf(stderr, "usage: df [ -i ] [ filsys... ]\n");
-               exit(0);
+       mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
+       maxwidth = 0;
+       for (i = 0; i < mntsize; i++) {
+               width = strlen(mntbuf[i].f_mntfromname);
+               if (width > maxwidth)
+                       maxwidth = width;

        }
        }

-       argc--, argv++;
-       }
-       i = open("/etc/mtab", 0);
-       if (i >= 0) {
-               read(i, mtab, sizeof mtab);     /* Probably returns short */
-               close(i);
+       if (!*argv) {
+               mntsize = getmntinfo(&mntbuf, (nflag ? MNT_NOWAIT : MNT_WAIT));
+               for (i = 0; i < mntsize; i++)
+                       prtstat(&mntbuf[i], maxwidth);
+               exit(0);

        }
        }

-       printf("Filesystem  Mounted on      kbytes    used    free");
-       printf("   %% used");
-       if (iflag)
-               printf("   iused   ifree  %%iused");
-       putchar('\n');
-       if (argc <= 1) {
-               struct fstab *fsp;
-
-               if (setfsent() == 0)
-                       perror(FSTAB), exit(1);
-               while (fsp = getfsent()) {
-                       if (strcmp(fsp->fs_type, FSTAB_RW) &&
-                           strcmp(fsp->fs_type, FSTAB_RO))
+       for (; *argv; argv++) {
+               if (stat(*argv, &stbuf) < 0) {
+                       err = errno;
+                       if ((mntpt = getmntpt(*argv)) == 0) {
+                               fprintf(stderr, "df: %s: %s\n", *argv,
+                                   strerror(err));

                                continue;
                                continue;

-                       if (root[0] == 0)
-                               strcpy(root, fsp->fs_spec);
-                       dfree(fsp->fs_spec, 1);
+                       }
+               } else if ((stbuf.st_mode & S_IFMT) == S_IFCHR) {
+                       ufs_df(*argv, maxwidth);
+                       continue;
+               } else if ((stbuf.st_mode & S_IFMT) == S_IFBLK) {
+                       if ((mntpt = getmntpt(*argv)) == 0) {
+                               mntpt = mktemp(strdup("/tmp/df.XXXXXX"));
+                               mdev.fspec = *argv;
+                               if (mkdir(mntpt, DEFFILEMODE) != 0) {
+                                       fprintf(stderr, "df: %s: %s\n",
+                                           mntpt, strerror(errno));
+                                       continue;
+                               }
+                               if (mount(MOUNT_UFS, mntpt, MNT_RDONLY,
+                                   &mdev) != 0) {
+                                       ufs_df(*argv, maxwidth);
+                                       (void)rmdir(mntpt);
+                                       continue;
+                               } else if (statfs(mntpt, &statfsbuf)) {
+                                       statfsbuf.f_mntonname[0] = '\0';
+                                       prtstat(&statfsbuf, maxwidth);
+                               } else
+                                       fprintf(stderr, "df: %s: %s\n",
+                                           *argv, strerror(errno));
+                               (void)unmount(mntpt, MNT_NOFORCE);
+                               (void)rmdir(mntpt);
+                               continue;
+                       }
+               } else
+                       mntpt = *argv;
+               /*
+                * Statfs does not take a `wait' flag, so we cannot
+                * implement nflag here.
+                */
+               if (statfs(mntpt, &statfsbuf) < 0) {
+                       fprintf(stderr,
+                           "df: %s: %s\n", mntpt, strerror(errno));
+                       continue;

                }
                }

-               endfsent();
-               exit(0);
+               if (argc == 1)
+                       maxwidth = strlen(statfsbuf.f_mntfromname) + 1;
+               prtstat(&statfsbuf, maxwidth);

        }
        }

-       for (i=1; i<argc; i++)
-               dfree(argv[i], 0);
+       return (0);

 }
 
 }
 

-dfree(file, infsent)
-       char *file;
-       int infsent;
+char *
+getmntpt(name)
+       char *name;

 {
 {

-       daddr_t i;
-       long blocks, free, used, hardway;
-       char *mp;
-       struct stat stbuf;
-       struct fstab *fsp;
-
-       if (stat(file, &stbuf) == 0 &&
-           (stbuf.st_mode&S_IFMT) != S_IFCHR &&
-           (stbuf.st_mode&S_IFMT) != S_IFBLK) {
-               if (infsent) {
-                       fprintf(stderr, "%s: screwy /etc/fstab entry\n", file);
-                       return;
-               }
-               setfsent();
-               while (fsp = getfsent()) {
-                       struct stat stb;
-
-                       if (stat(fsp->fs_spec, &stb) == 0 &&
-                           stb.st_rdev == stbuf.st_dev) {
-                               file = fsp->fs_spec;
-                               endfsent();
-                               goto found;
-                       }
-               }
-               endfsent();
-               fprintf(stderr, "%s: mounted on unknown device\n", file);
-               return;
-       }
-found:
-       fi = open(file, 0);
-       if (fi < 0) {
-               perror(file);
-               return;
-       }
-       bread((long) 1, (char *)&sblock, sizeof (sblock));
-       printf("%-12.12s%-14.14s", file, mp = mpath(file));
-       blocks = (long) sblock.s_fsize - (long)sblock.s_isize;
-       free = sblock.s_tfree;
-       used = blocks - free;
-       printf("%8d%8d%8d", blocks, used, free);
-       printf("%8.0f%%", 
-           blocks == 0 ? 0.0 : (double) used / (double)blocks * 100.0);
-       if (iflag) {
-               int inodes = (sblock.s_isize - 2) * INOPB;
-               used = inodes - sblock.s_tinode;
-               printf("%8ld%8ld%8.0f%%", used, sblock.s_tinode, 
-                   inodes == 0 ? 0.0 : (double)used/(double)inodes*100.0);
+       long mntsize, i;
+       struct statfs *mntbuf;
+
+       mntsize = getmntinfo(&mntbuf, (nflag ? MNT_NOWAIT : MNT_WAIT));
+       for (i = 0; i < mntsize; i++) {
+               if (!strcmp(mntbuf[i].f_mntfromname, name))
+                       return (mntbuf[i].f_mntonname);

        }
        }

-       printf("\n");
-       close(fi);
+       return (0);

 }
 
 }
 

-bread(bno, buf, cnt)
-       daddr_t bno;
-       char *buf;
+/*
+ * Print out status about a filesystem.
+ */
+void
+prtstat(sfsp, maxwidth)
+       register struct statfs *sfsp;
+       long maxwidth;

 {
 {

-       int n;
-       extern errno;
+       long used, availblks, inodes;
+       static int timesthrough;

 
 

-       lseek(fi, bno<<BSHIFT, 0);
-       if ((n=read(fi, buf, cnt)) != cnt) {
-               printf("\nread error bno = %ld\n", bno);
-               printf("count = %d; errno = %d\n", n, errno);
-               exit(0);
+       if (maxwidth < 11)
+               maxwidth = 11;
+       if (++timesthrough == 1) {
+               printf("%-*.*s%s    used   avail capacity",
+                   maxwidth, maxwidth, "Filesystem",
+                   kflag ? "  kbytes" : "512-blks");
+               if (iflag)
+                       printf(" iused   ifree  %%iused");
+               printf("  Mounted on\n");

        }
        }

+       printf("%-*.*s", maxwidth, maxwidth, sfsp->f_mntfromname);
+       used = sfsp->f_blocks - sfsp->f_bfree;
+       availblks = sfsp->f_bavail + used;
+       printf("%8ld%8ld%8ld",
+           sfsp->f_blocks * sfsp->f_fsize / (kflag ? 1024 : 512),
+           used * sfsp->f_fsize / (kflag ? 1024 : 512),
+           sfsp->f_bavail * sfsp->f_fsize / (kflag ? 1024 : 512));
+       printf("%6.0f%%",
+           availblks == 0 ? 100.0 : (double)used / (double)availblks * 100.0);
+       if (iflag) {
+               inodes = sfsp->f_files;
+               used = inodes - sfsp->f_ffree;
+               printf("%8ld%8ld%6.0f%% ", used, sfsp->f_ffree,
+                  inodes == 0 ? 100.0 : (double)used / (double)inodes * 100.0);
+       } else 
+               printf("  ");
+       printf("  %s\n", sfsp->f_mntonname);

 }
 
 /*
 }
 
 /*

- * Given a name like /dev/rrp0h, returns the mounted path, like /usr.
+ * This code constitutes the old df code for extracting
+ * information from filesystem superblocks.

  */
  */

-char *mpath(file)
+#include <ufs/fs.h>
+#include <errno.h>
+#include <fstab.h>
+
+union {
+       struct fs iu_fs;
+       char dummy[SBSIZE];
+} sb;
+#define sblock sb.iu_fs
+
+int    fi;
+int    bread();
+
+void
+ufs_df(file, maxwidth)

        char *file;
        char *file;

+       long maxwidth;

 {
 {

-       register int i;
-
-       if (eq(file, root))
-               return "/";
-       for (i=0; i<NFS; i++)
-               if (eq(file, mtab[i].spec))
-                       return (mtab[i].path);
-       return "";
+       extern int errno;
+       struct statfs statfsbuf;
+       register struct statfs *sfsp;
+       char *mntpt;
+       static int synced;
+
+       if (synced++ == 0)
+               sync();
+
+       if ((fi = open(file, O_RDONLY)) < 0) {
+               fprintf(stderr, "df: %s: %s\n", file, strerror(errno));
+               return;
+       }
+       if (bread((long)SBOFF, (char *)&sblock, SBSIZE) == 0) {
+               (void) close(fi);
+               return;
+       }
+       sfsp = &statfsbuf;
+       sfsp->f_type = MOUNT_UFS;
+       sfsp->f_flags = 0;
+       sfsp->f_fsize = sblock.fs_fsize;
+       sfsp->f_bsize = sblock.fs_bsize;
+       sfsp->f_blocks = sblock.fs_dsize;
+       sfsp->f_bfree = sblock.fs_cstotal.cs_nbfree * sblock.fs_frag +
+               sblock.fs_cstotal.cs_nffree;
+       sfsp->f_bavail = (sblock.fs_dsize * (100 - sblock.fs_minfree) / 100) -
+               (sblock.fs_dsize - sfsp->f_bfree);
+       if (sfsp->f_bavail < 0)
+               sfsp->f_bavail = 0;
+       sfsp->f_files =  sblock.fs_ncg * sblock.fs_ipg;
+       sfsp->f_ffree = sblock.fs_cstotal.cs_nifree;
+       sfsp->f_fsid.val[0] = 0;
+       sfsp->f_fsid.val[1] = 0;
+       if ((mntpt = getmntpt(file)) == 0)
+               mntpt = "";
+       bcopy((caddr_t)mntpt, (caddr_t)&sfsp->f_mntonname[0], MNAMELEN);
+       bcopy((caddr_t)file, (caddr_t)&sfsp->f_mntfromname[0], MNAMELEN);
+       prtstat(sfsp, maxwidth);
+       (void) close(fi);

 }
 
 }
 

-eq(f1, f2)
-       char *f1, *f2;
+long lseek();
+
+int
+bread(off, buf, cnt)
+       long off;
+       char *buf;
+       int cnt;

 {
 {

+       int n;
+       extern errno;

 
 

-       if (strncmp(f1, "/dev/", 5) == 0)
-               f1 += 5;
-       if (strncmp(f2, "/dev/", 5) == 0)
-               f2 += 5;
-       if (!strcmp(f1, f2))
-               return (1);
-       if (*f1 == 'r' && !strcmp(f1+1, f2))
-               return (1);
-       if (*f2 == 'r' && !strcmp(f1, f2+1))
-               return (1);
-       if (*f1 == 'r' && *f2 == 'r' && strcmp(f1+1, f2+1) == 0)
-               return (1);
-       return (0);
+       (void) lseek(fi, off, SEEK_SET);
+       if ((n=read(fi, buf, cnt)) != cnt) {
+               /* probably a dismounted disk if errno == EIO */
+               if (errno != EIO) {
+                       printf("\nread error off = %ld\n", off);
+                       printf("count = %d; errno = %d\n", n, errno);
+               }
+               return (0);
+       }
+       return (1);

 }
 }