diff toys/df.c @ 7:fc9c0503d5e2

Implement df. Add -Wall to build and fix up warnings. Add copyright notices. Add error_msg() and itoa() to library. Remove argc from globals (since argv is null terminated), add optflags to globals.
author landley@driftwood
date Mon, 30 Oct 2006 01:38:00 -0500
parents 67b517913e56
children 8f8a8ac59c14
line wrap: on
line diff
--- a/toys/df.c	Thu Oct 26 12:04:37 2006 -0400
+++ b/toys/df.c	Mon Oct 30 01:38:00 2006 -0500
@@ -2,7 +2,7 @@
 /*
  * df.c - report free disk space.
  *
- * Implemented according to SUSv3:
+ * Implemented roughly according to SUSv3:
  * http://www.opengroup.org/onlinepubs/009695399/utilities/df.html
  * 
  * usage: df [-k] [-P|-t] [file...]
@@ -10,16 +10,110 @@
 
 #include "toys.h"
 
+static void show_mt(struct mtab_list *mt)
+{
+	int len;
+	long size, used, avail;
+	uint64_t block;
+
+	// Return if it wasn't found (should never happen, but with /etc/mtab...)
+	if (!mt) return;
+
+	// If we have -t, skip other filesystem types
+	if (toy.df.fstype) {
+		struct string_list *sl;
+
+		for (sl = toy.df.fstype; sl; sl = sl->next)
+			if (!strcmp(mt->type, sl->str)) break;
+		if (!sl) return;
+	}
+
+	// If we don't have -a, skip synthetic filesystems
+	if (!(toys.optflags & 1) && !mt->statvfs.f_blocks) return;
+
+	// Figure out how much total/used/free space this filesystem has,
+	// forcing 64-bit math because filesystems are big now.
+	block = mt->statvfs.f_bsize ? : 1;
+
+	size = (long)((block * mt->statvfs.f_blocks) / toy.df.units);
+	used = (long)((block * (mt->statvfs.f_blocks-mt->statvfs.f_bfree))
+			/ toy.df.units);
+	avail = (long)((block
+				* (getuid() ? mt->statvfs.f_bavail : mt->statvfs.f_bfree))
+			/ toy.df.units);
+
+	// Figure out appropriate spacing
+	len = 25 - strlen(mt->device);
+	if (len < 1) len = 1;
+	printf("%s% *ld % 10ld % 9ld % 3ld%% %s\n",mt->device, len,
+		size, used, avail, 100-(long)((100*(uint64_t)avail)/size), mt->dir);
+}
+
 int df_main(void)
 {
-	struct mtab_list *mt, *mtlist;
+	struct mtab_list *mt, *mt2, *mtlist;
+	char **argv;
+
+	// get_optflags("Pkt:a",&(toy.df.fstype));
+	argv = NULL;
 
-	//int units = 512;
+	// Handle -P and -k
+	toy.df.units = 1024;
+	if (CFG_DF_PEDANTIC && (toys.optflags & 8)) {
+		// Units are 512 bytes if you select "pedantic" without "kilobytes".
+		if ((toys.optflags&3) == 1) toy.df.units = 512;
+		printf("Filesystem %ld-blocks Used Available Capacity Mounted on\n",
+			toy.df.units);
+	} else puts("Filesystem\t1K-blocks\tUsed Available Use% Mounted on");
+
 	mtlist = getmountlist(1);
-	// Zap overmounts
-	for (mt = mtlist; mt; mt = mt->next) {
-		printf("type=%s dir=%s device=%s\n",mt->type,mt->dir,mt->device);
+
+	// If we have a list of filesystems on the command line, loop through them.
+	if (argv) {
+		char *next;
+
+		for(next = *argv; *next; next++) {
+			struct stat st;
+
+			// Stat it (complain if we can't).
+			if(!stat(next, &st)) {
+				perror_msg("`%s'", next);
+				toys.exitval = 1;
+				continue;
+			}
+
+			// Find and display this filesystem.  Use _last_ hit in case of
+			// -- bind mounts.
+			mt2 = NULL;
+			for (mt = mtlist; mt; mt = mt->next)
+				if (st.st_dev == mt->stat.st_dev) mt2 = mt;
+			show_mt(mt2);
+		}
+	} else {
+		// Get and loop through mount list.
+
+		for (mt = mtlist; mt; mt = mt->next) {
+			struct mtab_list *mt2, *mt3;
+
+			if (!mt->stat.st_dev) continue;
+
+			// Filter out overmounts.
+			mt3 = mt;
+			for (mt2 = mt->next; mt2; mt2 = mt2->next) {
+				if (mt->stat.st_dev == mt2->stat.st_dev) {
+					// For --bind mounts, take last match
+					if (!strcmp(mt->device, mt2->device)) mt3 = mt2;
+					// Filter out overmounts
+					mt2->stat.st_dev = 0;
+				}
+			}
+			show_mt(mt3);
+		}
 	}
 
+	if (CFG_TOYS_FREE) {
+		llist_free(mtlist, NULL);
+		free(argv);
+	}
 	return 0;
 }