386BSD 0.1 development
[unix-history] / usr / src / bin / df / df.c
index 075beb1..1ed82c7 100644 (file)
-static char *sccsid = "@(#)df.c        4.1 (Berkeley) %G%";
-#include <stdio.h>
-#include <fstab.h>
-#include <sys/param.h>
-#include <sys/filsys.h>
-#include <sys/fblk.h>
-
-#define NFS    20      /* Max number of filesystems */
-
-
-struct {
-       char path[32];
-       char spec[32];
-} mtab[NFS];
-char root[32];
-
-char *mpath();
+/*
+ * 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.
+ */
 
 
-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    lflag;
-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)
 main(argc, argv)
-char **argv;
+       int argc;
+       char **argv;
 {
 {
-       int i;
-       FILE *f = fopen(FSTAB, "r");
-       char buf[128];
-       struct  fstab   fs;
-
-       while (argc >= 1 && argv[1][0]=='-') {
-               switch(argv[1][1]) {
-
-               case 'l':
-                       lflag++;
-                       break;
+       extern int errno, optind;
+       int err, ch, i;
+       long width, maxwidth, mntsize;
+       char *mntpt, *mktemp();
+       struct stat stbuf;
+       struct statfs statfsbuf, *mntbuf;
 
 
+       while ((ch = getopt(argc, argv, "ikn")) != EOF)
+               switch(ch) {
                case 'i':
                case 'i':
-                       iflag++;
+                       iflag = 1;
                        break;
                        break;
-
+               case 'k':
+                       kflag = 1;
+                       break;
+               case 'n':
+                       nflag = 1;
+                       break;
+               case '?':
                default:
                default:
-                       fprintf(stderr, "usage: df [ -il ] [ filsys... ]\n");
-                       exit(0);
+                       fprintf(stderr,
+                           "usage: df [-ikn] [file | file_system ...]\n");
+                       exit(1);
                }
                }
-               argc--, argv++;
-       }
+       argc -= optind;
+       argv += optind;
 
 
-       if ((i=open("/etc/mtab", 0)) >= 0) {
-               read(i, mtab, sizeof mtab);     /* Probably returns short */
-               close(i);
+       mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
+       maxwidth = 0;
+       for (i = 0; i < mntsize; i++) {
+               width = strlen(mntbuf[i].f_mntfromname);
+               if (width > maxwidth)
+                       maxwidth = width;
        }
        }
-       printf("Filesystem  Mounted on  blocks\t  used\t  free");
-       if (lflag)
-               printf("\thardway");
-       printf("\t%% used");
-       if (iflag)
-               printf("\tiused\tifree\t%%iused");
-       putchar('\n');
-       if(argc <= 1) {
-               if (f == NULL)
-                       perror(FSTAB), exit(1);
-               while (!feof(f)){
-                       fscanf(f, FSTABFMT, FSTABARG(&fs));
-                       if (root[0] == 0)
-                               strcpy(root, fs.fs_spec);
-                       dfree(fs.fs_spec);
-               }
+       if (!*argv) {
+               mntsize = getmntinfo(&mntbuf, (nflag ? MNT_NOWAIT : MNT_WAIT));
+               for (i = 0; i < mntsize; i++)
+                       prtstat(&mntbuf[i], maxwidth);
                exit(0);
        }
                exit(0);
        }
-
-       if (f){
-               fscanf(f, FSTABFMT, FSTABARG(&fs));
-               strcpy(root, fs.fs_spec);
-       }
-       for(i=1; i<argc; i++) {
-               dfree(argv[i]);
+       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;
+                       }
+               } 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;
+               }
+               if (argc == 1)
+                       maxwidth = strlen(statfsbuf.f_mntfromname) + 1;
+               prtstat(&statfsbuf, maxwidth);
        }
        }
+       return (0);
 }
 
 }
 
-dfree(file)
-char *file;
+char *
+getmntpt(name)
+       char *name;
 {
 {
-       daddr_t i;
-       long    blocks;
-       long    free;
-       long    used;
-       long    hardway;
-       char    *mp;
-
-       fi = open(file, 0);
-       if(fi < 0) {
-               fprintf(stderr,"cannot open %s\n", file);
-               return;
-       }
-       sync();
-       bread(1L, (char *)&sblock, sizeof(sblock));
-       printf("%-12.12s%s", file, mp = mpath(file));
-       if (strlen(mp) < 4)
-               putchar('\t');
-
-       blocks = (long) sblock.s_fsize - (long)sblock.s_isize;
-       free = sblock.s_tfree;
-       used = blocks - free;
-
-       printf("\t%6ld", blocks);
-       printf("\t%6ld", used);
-       printf("\t%6ld", free);
-       if (lflag) {
-               hardway = 0;
-               while(alloc())
-                       hardway++;
-               printf("\t%6ld", free=hardway);
-       }
-       printf("\t%5.0f%%", (double) used / (double)blocks * 100.0);
-       if (iflag) {
-               int inodes = (sblock.s_isize - 2) * INOPB;
-               used = inodes - sblock.s_tinode;
-               printf("\t%5ld\t%5ld\t%5.0f%%", used, sblock.s_tinode, (double)used/(double)inodes*100.0);
-       }
-       printf("\n");
-       close(fi);
-}
+       long mntsize, i;
+       struct statfs *mntbuf;
 
 
-daddr_t
-alloc()
-{
-       int i;
-       daddr_t b;
-       struct fblk buf;
-
-       i = --sblock.s_nfree;
-       if(i<0 || i>=NICFREE) {
-               printf("bad free count, b=%D\n", blkno);
-               return(0);
-       }
-       b = sblock.s_free[i];
-       if(b == 0)
-               return(0);
-       if(b<sblock.s_isize || b>=sblock.s_fsize) {
-               printf("bad free block (%D)\n", b);
-               return(0);
+       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);
        }
        }
-       if(sblock.s_nfree <= 0) {
-               bread(b, (char *)&buf, sizeof(buf));
-               blkno = b;
-               sblock.s_nfree = buf.df_nfree;
-               for(i=0; i<NICFREE; i++)
-                       sblock.s_free[i] = buf.df_free[i];
-       }
-       return(b);
+       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)
-char *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;
+       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;
 {
 {
-       if (strncmp(f1, "/dev/", 5) == 0)
-               f1 += 5;
-       if (strncmp(f2, "/dev/", 5) == 0)
-               f2 += 5;
-       if (strcmp(f1, f2) == 0)
-               return 1;
-       if (*f1 == 'r' && strcmp(f1+1, f2) == 0)
-               return 1;
-       if (*f2 == 'r' && strcmp(f1, f2+1) == 0)
-               return 1;
-       if (*f1 == 'r' && *f2 == 'r' && strcmp(f1+1, f2+1) == 0)
-               return 1;
-       return 0;
+       int n;
+       extern errno;
+
+       (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);
 }
 }