lost the SCCS id's...

[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 80436ca..3227804 100644 (file)
--- a/usr/src/bin/df/df.c
+++ b/usr/src/bin/df/df.c
@@ -1,192 +1,272 @@
+/*
+ * Copyright (c) 1980, 1990 The Regents of the University of California.
+ * All rights reserved.
+ *
+ * %sccs.include.redist.c%
+ */
+

 #ifndef lint
 #ifndef lint

-static char *sccsid = "@(#)df.c        4.12 %G%";
-#endif
+char copyright[] =
+"@(#) Copyright (c) 1980, 1990 The Regents of the University of California.\n\
+ All rights reserved.\n";
+#endif /* not lint */

 
 

-#include <stdio.h>
-#include <fstab.h>
-#include <sys/param.h>
-#include <sys/fs.h>
-#include <sys/stat.h>
+#ifndef lint
+static char sccsid[] = "@(#)df.c       5.24 (Berkeley) %G%";
+#endif /* not lint */

 
 /*
  * df
  */
 
 /*
  * df
  */

-#define NFS    20      /* Max number of filesystems */
-
-struct {
-       char path[32];
-       char spec[32];
-} mtab[NFS];
-char   root[32];
-char   *mpath();
-
-int    iflag;
-
-union {
-       struct fs iu_fs;
-       char dummy[SBSIZE];
-} sb;
-#define sblock sb.iu_fs
+#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   *strcpy();
+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) {
-               (void) read(i, (char *)mtab, sizeof mtab);
-               (void) 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    kbytes    used   avail capacity");
-       if (iflag)
-               printf(" iused   ifree  %%iused");
-       printf("  Mounted on\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)
-                               (void) 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;

 {
 {

-       long totalblks, availblks, avail, free, used;
-       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;
+       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);

        }
        }

-found:
-       fi = open(file, 0);
-       if (fi < 0) {
-               perror(file);
-               return;
+       return (0);
+}
+
+/*
+ * Print out status about a filesystem.
+ */
+void
+prtstat(sfsp, maxwidth)
+       register struct statfs *sfsp;
+       long maxwidth;
+{
+       long used, availblks, inodes;
+       static int timesthrough;
+
+       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");

        }
        }

-       bread(SBLOCK, (char *)&sblock, SBSIZE);
-       printf("%-12.12s", file);
-       totalblks = sblock.fs_dsize;
-       free = sblock.fs_cstotal.cs_nbfree * sblock.fs_frag +
-           sblock.fs_cstotal.cs_nffree;
-       used = totalblks - free;
-       availblks = totalblks * (100 - sblock.fs_minfree) / 100;
-       avail = availblks > used ? availblks - used : 0;
-       printf("%8d%8d%8d", totalblks * sblock.fs_fsize / 1024,
-           used * sblock.fs_fsize / 1024, avail * sblock.fs_fsize / 1024);
+       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%%",
        printf("%6.0f%%",

-           availblks == 0 ? 0.0 : (double) used / (double) availblks * 100.0);
+           availblks == 0 ? 100.0 : (double)used / (double)availblks * 100.0);

        if (iflag) {
        if (iflag) {

-               int inodes = sblock.fs_ncg * sblock.fs_ipg;
-               used = inodes - sblock.fs_cstotal.cs_nifree;
-               printf("%8ld%8ld%6.0f%% ", used, sblock.fs_cstotal.cs_nifree,
-                   inodes == 0 ? 0.0 : (double)used / (double)inodes * 100.0);
+               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("  ");
        } else 
                printf("  ");

-       printf("  %s\n", mpath(file));
+       printf("  %s\n", sfsp->f_mntonname);
+}
+
+/*
+ * This code constitutes the old df code for extracting
+ * information from filesystem superblocks.
+ */
+#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;
+       long maxwidth;
+{
+       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);
 }
 
 long lseek();
 
        (void) close(fi);
 }
 
 long lseek();
 

-bread(bno, buf, cnt)
-       daddr_t bno;
+int
+bread(off, buf, cnt)
+       long off;

        char *buf;
        char *buf;

+       int cnt;

 {
        int n;
        extern errno;
 
 {
        int n;
        extern errno;
 

-       (void) lseek(fi, (long)(bno * DEV_BSIZE), 0);
+       (void) lseek(fi, off, SEEK_SET);

        if ((n=read(fi, buf, cnt)) != cnt) {
        if ((n=read(fi, buf, cnt)) != cnt) {

-               printf("\nread error bno = %ld\n", bno);
-               printf("count = %d; errno = %d\n", n, errno);
-               exit(0);
+               /* 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);

        }
        }

-}
-
-/*
- * Given a name like /dev/rrp0h, returns the mounted path, like /usr.
- */
-char *mpath(file)
-       char *file;
-{
-       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 "";
-}
-
-eq(f1, f2)
-       char *f1, *f2;
-{
-
-       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);
+       return (1);

 }
 }