Implemented du -s
diff --git a/du.c b/du.c
index 14e0bdf..e5788c6 100644
--- a/du.c
+++ b/du.c
@@ -32,17 +32,35 @@
#include <sys/stat.h>
#endif
-static const char du_usage[] =
-"Usage: du [OPTION]... [FILE]...\n";
-
typedef void (Display)(size_t, char *);
+static const char du_usage[] =
+"Usage: du [OPTION]... [FILE]...\n\n"
+"\t-s\tdisplay only a total for each argument\n"
+;
+
+static int du_depth = 0;
+
+static Display *print;
+
static void
-print(size_t size, char *filename)
+print_null(size_t size, char *filename) { }
+
+static void
+print_normal(size_t size, char *filename)
{
fprintf(stdout, "%-7d %s\n", (size >> 1), filename);
}
+static void
+print_summary(size_t size, char *filename)
+{
+ if (du_depth == 1) {
+ print_normal(size, filename);
+ }
+}
+
+
/* tiny recursive du */
static size_t
du(char *filename)
@@ -54,6 +72,8 @@
fprintf(stdout, "du: %s: %s\n", filename, strerror(errno));
return 0;
}
+
+ du_depth++;
sum = statbuf.st_blocks;
if (S_ISDIR(statbuf.st_mode)) {
@@ -76,6 +96,7 @@
closedir(dir);
print(sum, filename);
}
+ du_depth--;
return sum;
}
@@ -85,12 +106,16 @@
int i;
char opt;
+ /* default behaviour */
+ print = print_normal;
+
/* parse argv[] */
for (i = 1; i < argc; i++) {
if (argv[i][0] == '-') {
opt = argv[i][1];
switch (opt) {
case 's':
+ print = print_summary;
break;
case 'h':
usage(du_usage);
@@ -110,12 +135,12 @@
} else {
int sum;
for ( ; i < argc; i++) {
- sum = du(argv[i]) >> 1;
- if (sum) printf("%-7d %s\n", sum, argv[i]);
+ sum = du(argv[i]);
+ if (sum) { print_normal(sum, argv[i]); }
}
}
exit(0);
}
-/* $Id: du.c,v 1.5 1999/12/10 15:23:47 beppu Exp $ */
+/* $Id: du.c,v 1.6 1999/12/15 18:52:17 beppu Exp $ */