X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=du.c;h=b8e296ddd20f3bea2a1f761c2ebc8a38baf1ea76;hb=e714bce003a1b0e2c49e58fe14af86abc8b31f23;hp=ed90bc445dc45b1fa7bea6c45791033a267d5898;hpb=0aa611816772021e706f0996aa2eed56d26ad696;p=oweals%2Fbusybox.git diff --git a/du.c b/du.c index ed90bc445..b8e296ddd 100644 --- a/du.c +++ b/du.c @@ -1,8 +1,9 @@ +/* vi: set sw=4 ts=4: */ /* * Mini du implementation for busybox * * - * Copyright (C) 1999 by Lineo, inc. + * Copyright (C) 1999,2000 by Lineo, inc. * Written by John Beppu * * This program is free software; you can redistribute it and/or modify @@ -22,125 +23,172 @@ */ #include "internal.h" +#define BB_DECLARE_EXTERN +#define bb_need_name_too_long +#include "messages.c" + #include #include #include #include #include -#if 0 -#include -#include -#endif -typedef void (Display)(size_t, char *); +typedef void (Display) (long, 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; + "du [OPTION]... [FILE]...\n" +#ifndef BB_FEATURE_TRIVIAL_HELP + "\nSummarizes disk space used for each FILE and/or directory.\n" + "Disk space is printed in units of 1024 bytes.\n\n" + "Options:\n" + "\t-l\tcount sizes many times if hard linked\n" + "\t-s\tdisplay only a total for each argument\n" +#endif + ; -static Display *print; +static int du_depth = 0; +static int count_hardlinks = 0; -static void -print_null(size_t size, char *filename) { } +static Display *print; -static void -print_normal(size_t size, char *filename) +static void print_normal(long size, char *filename) { - fprintf(stdout, "%-7d %s\n", (size >> 1), filename); + fprintf(stdout, "%ld\t%s\n", size, filename); } -static void -print_summary(size_t size, char *filename) +static void print_summary(long size, char *filename) { - if (du_depth == 1) { - print_normal(size, filename); - } + if (du_depth == 1) { + print_normal(size, filename); + } } - /* tiny recursive du */ -static size_t -du(char *filename) +static long du(char *filename) { - struct stat statbuf; - size_t sum; - - if ((lstat(filename, &statbuf)) != 0) { - fprintf(stdout, "du: %s: %s\n", filename, strerror(errno)); - return 0; - } - - du_depth++; - sum = statbuf.st_blocks; + struct stat statbuf; + long sum; + int len; - if (S_ISDIR(statbuf.st_mode)) { - DIR *dir; - struct dirent *entry; - - dir = opendir(filename); - if (!dir) { return 0; } - while ((entry = readdir(dir))) { - char newfile[512]; - char *name = entry->d_name; + if ((lstat(filename, &statbuf)) != 0) { + printf("du: %s: %s\n", filename, strerror(errno)); + return 0; + } - if ( (strcmp(name, "..") == 0) - || (strcmp(name, ".") == 0)) - { continue; } + du_depth++; + sum = (statbuf.st_blocks >> 1); - sprintf(newfile, "%s/%s", filename, name); - sum += du(newfile); + /* Don't add in stuff pointed to by symbolic links */ + if (S_ISLNK(statbuf.st_mode)) { + sum = 0L; + if (du_depth == 1) + print(sum, filename); + } + if (S_ISDIR(statbuf.st_mode)) { + DIR *dir; + struct dirent *entry; + + dir = opendir(filename); + if (!dir) { + du_depth--; + return 0; + } + + len = strlen(filename); + if (filename[len - 1] == '/') + filename[--len] = '\0'; + + while ((entry = readdir(dir))) { + char newfile[BUFSIZ + 1]; + char *name = entry->d_name; + + if ((strcmp(name, "..") == 0) + || (strcmp(name, ".") == 0)) { + continue; + } + + if (len + strlen(name) + 1 > BUFSIZ) { + fprintf(stderr, name_too_long, "du"); + du_depth--; + return 0; + } + sprintf(newfile, "%s/%s", filename, name); + + sum += du(newfile); + } + closedir(dir); + print(sum, filename); + } + else if (statbuf.st_nlink > 1 && !count_hardlinks) { + /* Add files with hard links only once */ + if (is_in_ino_dev_hashtable(&statbuf, NULL)) { + sum = 0L; + if (du_depth == 1) + print(sum, filename); + } + else { + add_to_ino_dev_hashtable(&statbuf, NULL); + } } - closedir(dir); - print(sum, filename); - } - du_depth--; - return sum; + du_depth--; + return sum; } -int -du_main(int argc, char **argv) +int du_main(int argc, char **argv) { - 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); - break; - default: - fprintf(stderr, "du: invalid option -- %c\n", opt); - usage(du_usage); - } - } else { - break; + 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 'l': + count_hardlinks = 1; + break; + case 'h': + case '-': + 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 { - int sum; - for ( ; i < argc; i++) { - sum = du(argv[i]); - if ((sum) && (isDirectory(argv[i]))) { print_normal(sum, argv[i]); } + + /* go through remaining args (if any) */ + if (i >= argc) { + du("."); + } else { + long sum; + + for (; i < argc; i++) { + sum = du(argv[i]); + if (sum && isDirectory(argv[i], FALSE, NULL)) { + print_normal(sum, argv[i]); + } + reset_ino_dev_hashtable(); + } } - } - exit(0); + return(0); } -/* $Id: du.c,v 1.7 1999/12/16 21:16:47 beppu Exp $ */ +/* $Id: du.c,v 1.20 2000/06/19 17:25:39 andersen Exp $ */ +/* +Local Variables: +c-file-style: "linux" +c-basic-offset: 4 +tab-width: 4 +End: +*/