Fleshed out du_main().

I'm not sure which options to support.
This commit is contained in:
John Beppu 1999-12-10 06:15:27 +00:00
parent 2ad2afd05c
commit 14c82b64c9
2 changed files with 76 additions and 12 deletions

View File

@ -26,11 +26,13 @@
#include <fcntl.h> #include <fcntl.h>
#include <dirent.h> #include <dirent.h>
#include <stdio.h> #include <stdio.h>
/* #if 0
#include <unistd.h> #include <unistd.h>
#include <sys/stat.h> #include <sys/stat.h>
*/ #endif
static const char du_usage[] =
"Usage: du [OPTION]... [FILE]...\n";
typedef void (Display)(size_t, char *); typedef void (Display)(size_t, char *);
@ -42,7 +44,7 @@ print(size_t size, char *filename)
/* tiny recursive du */ /* tiny recursive du */
static size_t static size_t
size(char *filename) du(char *filename)
{ {
struct stat statbuf; struct stat statbuf;
size_t sum; size_t sum;
@ -65,7 +67,7 @@ size(char *filename)
{ continue; } { continue; }
sprintf(newfile, "%s/%s", filename, name); sprintf(newfile, "%s/%s", filename, name);
sum += size(newfile); sum += du(newfile);
} }
closedir(dir); closedir(dir);
print(sum, filename); print(sum, filename);
@ -76,8 +78,38 @@ size(char *filename)
int int
du_main(int argc, char **argv) du_main(int argc, char **argv)
{ {
/* I'll fill main() in shortly */ int i;
size("."); char opt;
/* parse argv[] */
for (i = 1; i < argc; i++) {
if (argv[i][0] == '-') {
opt = argv[i][1];
switch (opt) {
case 's':
break;
case 'h':
usage(du_usage);
break;
default:
fprintf(stderr, "du: invalid option -- %c\n", opt);
usage(du_usage);
}
} else {
break;
}
}
/* go through remaining args (if any) */
if (i >= argc) {
du(".");
} else {
for ( ; i < argc; i++) {
printf("%-7d %s\n", du(argv[i]) >> 1, argv[i]);
}
}
exit(0); exit(0);
} }
/* $Id: du.c,v 1.2 1999/12/10 06:15:27 beppu Exp $ */

44
du.c
View File

@ -26,11 +26,13 @@
#include <fcntl.h> #include <fcntl.h>
#include <dirent.h> #include <dirent.h>
#include <stdio.h> #include <stdio.h>
/* #if 0
#include <unistd.h> #include <unistd.h>
#include <sys/stat.h> #include <sys/stat.h>
*/ #endif
static const char du_usage[] =
"Usage: du [OPTION]... [FILE]...\n";
typedef void (Display)(size_t, char *); typedef void (Display)(size_t, char *);
@ -42,7 +44,7 @@ print(size_t size, char *filename)
/* tiny recursive du */ /* tiny recursive du */
static size_t static size_t
size(char *filename) du(char *filename)
{ {
struct stat statbuf; struct stat statbuf;
size_t sum; size_t sum;
@ -65,7 +67,7 @@ size(char *filename)
{ continue; } { continue; }
sprintf(newfile, "%s/%s", filename, name); sprintf(newfile, "%s/%s", filename, name);
sum += size(newfile); sum += du(newfile);
} }
closedir(dir); closedir(dir);
print(sum, filename); print(sum, filename);
@ -76,8 +78,38 @@ size(char *filename)
int int
du_main(int argc, char **argv) du_main(int argc, char **argv)
{ {
/* I'll fill main() in shortly */ int i;
size("."); char opt;
/* parse argv[] */
for (i = 1; i < argc; i++) {
if (argv[i][0] == '-') {
opt = argv[i][1];
switch (opt) {
case 's':
break;
case 'h':
usage(du_usage);
break;
default:
fprintf(stderr, "du: invalid option -- %c\n", opt);
usage(du_usage);
}
} else {
break;
}
}
/* go through remaining args (if any) */
if (i >= argc) {
du(".");
} else {
for ( ; i < argc; i++) {
printf("%-7d %s\n", du(argv[i]) >> 1, argv[i]);
}
}
exit(0); exit(0);
} }
/* $Id: du.c,v 1.2 1999/12/10 06:15:27 beppu Exp $ */