Fix egrep
[oweals/busybox.git] / df.c
diff --git a/df.c b/df.c
index 94b6b823180b861da819f01e90082949d351d8c6..21a1dbb319fc3a04d619606437d3d2458e412bec 100644 (file)
--- a/df.c
+++ b/df.c
@@ -1,7 +1,8 @@
+/* vi: set sw=4 ts=4: */
 /*
  * Mini df implementation for busybox
  *
- * Copyright (C) 1999 by Lineo, inc.
+ * Copyright (C) 1999,2000,2001 by Lineo, inc.
  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
  * based on original code by (I think) Bruce Perens <bruce@pixar.com>.
  *
  *
  */
 
-#include "internal.h"
 #include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
 #include <mntent.h>
-#include <sys/stat.h>
 #include <sys/vfs.h>
-#include <fstab.h>
+#include <getopt.h>
+#include "busybox.h"
 
-static const char df_usage[] = "df [filesystem ...]\n"
-    "\n" "\tPrint the filesystem space used and space available.\n";
+extern const char mtab_file[]; /* Defined in utility.c */
+#ifdef BB_FEATURE_HUMAN_READABLE
+static unsigned long df_disp_hr = KILOBYTE; 
+#endif
 
-extern const char mtab_file[]; /* Defined in utility.c */
-
-static int df(char *device, const char *mountPoint)
+static int do_df(char *device, const char *mount_point)
 {
-    struct statfs s;
-    long blocks_used;
-    long blocks_percent_used;
-
-    if (statfs(mountPoint, &s) != 0) {
-       perror(mountPoint);
-       return 1;
-    }
-
-    if (s.f_blocks > 0) {
-       blocks_used = s.f_blocks - s.f_bfree;
-       blocks_percent_used = (long)
-           (blocks_used * 100.0 / (blocks_used + s.f_bavail) + 0.5);
-       if (strcmp(device, "/dev/root") == 0)
-           device = (getfsfile("/"))->fs_spec;
-
-       printf("%-20s %9ld %9ld %9ld %3ld%% %s\n",
-              device,
-              (long) (s.f_blocks * (s.f_bsize / 1024.0)),
-              (long) ((s.f_blocks - s.f_bfree) * (s.f_bsize / 1024.0)),
-              (long) (s.f_bavail * (s.f_bsize / 1024.0)),
-              blocks_percent_used, mountPoint);
-
-    }
-
-    return 0;
-}
-
-/*
- * Given a block device, find the mount table entry if that block device
- * is mounted.
- *
- * Given any other file (or directory), find the mount table entry for its
- * filesystem.
- */
-extern struct mntent *findMountPoint(const char *name, const char *table)
-{
-    struct stat s;
-    dev_t mountDevice;
-    FILE *mountTable;
-    struct mntent *mountEntry;
-
-    if (stat(name, &s) != 0)
-       return 0;
-
-    if ((s.st_mode & S_IFMT) == S_IFBLK)
-       mountDevice = s.st_rdev;
-    else
-       mountDevice = s.st_dev;
-
+       struct statfs s;
+       long blocks_used;
+       long blocks_percent_used;
+#ifdef BB_FEATURE_HUMAN_READABLE
+       long base;
+#endif 
+
+       if (statfs(mount_point, &s) != 0) {
+               perror_msg("%s", mount_point);
+               return FALSE;
+       }
 
-    if ((mountTable = setmntent(table, "r")) == 0)
-       return 0;
+       if (s.f_blocks > 0) {
+               blocks_used = s.f_blocks - s.f_bfree;
+               if(blocks_used == 0)
+                       blocks_percent_used = 0;
+               else {
+                       blocks_percent_used = (long)
+                         (blocks_used * 100.0 / (blocks_used + s.f_bavail) + 0.5);
+               }
+               if (strcmp(device, "/dev/root") == 0) {
+                       /* Adjusts device to be the real root device,
+                        * or leaves device alone if it can't find it */
+                       find_real_root_device_name( device);
+               }
+#ifdef BB_FEATURE_HUMAN_READABLE
+               switch (df_disp_hr) {
+                       case MEGABYTE:
+                               base = KILOBYTE;
+                               break;
+                       case KILOBYTE:
+                               base = 1;
+                               break;
+                       default:
+                               base = 0;
+               }
+               printf("%-20s %9s ", device,
+                          make_human_readable_str((unsigned long)(s.f_blocks * 
+                                          (s.f_bsize/(double)KILOBYTE)), base));
+               printf("%9s ",
+                          make_human_readable_str((unsigned long)(
+                                          (s.f_blocks - s.f_bfree) * 
+                                          (s.f_bsize/(double)KILOBYTE)), base));
+               printf("%9s %3ld%% %s\n",
+                          make_human_readable_str((unsigned long)(s.f_bavail * 
+                                          (s.f_bsize/(double)KILOBYTE)), base),
+                          blocks_percent_used, mount_point);
+#else
+               printf("%-20s %9ld %9ld %9ld %3ld%% %s\n",
+                               device,
+                               (long) (s.f_blocks * (s.f_bsize / (double)KILOBYTE)),
+                               (long) ((s.f_blocks - s.f_bfree)*(s.f_bsize/(double)KILOBYTE)),
+                               (long) (s.f_bavail * (s.f_bsize / (double)KILOBYTE)),
+                               blocks_percent_used, mount_point);
+#endif
+       }
 
-    while ((mountEntry = getmntent(mountTable)) != 0) {
-       if (strcmp(name, mountEntry->mnt_dir) == 0
-           || strcmp(name, mountEntry->mnt_fsname) == 0)       /* String match. */
-           break;
-       if (stat(mountEntry->mnt_fsname, &s) == 0 && s.st_rdev == mountDevice)  /* Match the device. */
-           break;
-       if (stat(mountEntry->mnt_dir, &s) == 0 && s.st_dev == mountDevice)      /* Match the directory's mount point. */
-           break;
-    }
-    endmntent(mountTable);
-    return mountEntry;
+       return TRUE;
 }
 
-
-
 extern int df_main(int argc, char **argv)
 {
-    printf("%-20s %-14s %s %s %s %s\n", "Filesystem",
-          "1k-blocks", "Used", "Available", "Use%", "Mounted on");
-
-    if (argc > 1) {
-       struct mntent *mountEntry;
-       int status;
-
-       while (argc > 1) {
-           if ((mountEntry = findMountPoint(argv[1], mtab_file)) ==
-               0) {
-               fprintf(stderr, "%s: can't find mount point.\n", argv[1]);
-               return 1;
-           }
-           status = df(mountEntry->mnt_fsname, mountEntry->mnt_dir);
-           if (status != 0)
-               return status;
-           argc--;
-           argv++;
-       }
-       return 0;
-    } else {
-       FILE *mountTable;
-       struct mntent *mountEntry;
-
-       mountTable = setmntent(mtab_file, "r");
-       if (mountTable == 0) {
-           perror(mtab_file);
-           exit(FALSE);
+       int status = EXIT_SUCCESS;
+       int opt = 0;
+       int i = 0;
+       char disp_units_hdr[80] = "1k-blocks"; /* default display is kilobytes */
+
+       while ((opt = getopt(argc, argv, "k"
+#ifdef BB_FEATURE_HUMAN_READABLE
+       "hm"
+#endif
+)) > 0)
+       {
+               switch (opt) {
+#ifdef BB_FEATURE_HUMAN_READABLE
+                       case 'h':
+                               df_disp_hr = 0;
+                               strcpy(disp_units_hdr, "     Size");
+                               break;
+                       case 'm':
+                               df_disp_hr = MEGABYTE;
+                               strcpy(disp_units_hdr, "1M-blocks");
+                               break;
+#endif
+                       case 'k':
+                               /* default display is kilobytes */
+                               break;
+                       default:
+                                         show_usage();
+               }
        }
 
-       while ((mountEntry = getmntent(mountTable))) {
-           int status = df(mountEntry->mnt_fsname, mountEntry->mnt_dir);
-           if (status)
-               return status;
+       printf("%-20s %-14s %s %s %s %s\n", "Filesystem", disp_units_hdr,
+              "Used", "Available", "Use%", "Mounted on");
+
+       if(optind < argc) {
+               struct mntent *mount_entry;
+               for(i = optind; i < argc; i++)
+               {
+                       if ((mount_entry = find_mount_point(argv[i], mtab_file)) == 0) {
+                               error_msg("%s: can't find mount point.", argv[i]);
+                               status = EXIT_FAILURE;
+                       } else if (!do_df(mount_entry->mnt_fsname, mount_entry->mnt_dir))
+                               status = EXIT_FAILURE;
+               }
+       } else {
+               FILE *mount_table;
+               struct mntent *mount_entry;
+
+               mount_table = setmntent(mtab_file, "r");
+               if (mount_table == 0) {
+                       perror_msg("%s", mtab_file);
+                       return EXIT_FAILURE;
+               }
+
+               while ((mount_entry = getmntent(mount_table))) {
+                       if (!do_df(mount_entry->mnt_fsname, mount_entry->mnt_dir))
+                               status = EXIT_FAILURE;
+               }
+               endmntent(mount_table);
        }
-       endmntent(mountTable);
-    }
 
-    return 0;
+       return status;
 }
+
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/