X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;ds=sidebyside;f=coreutils%2Fdu.c;h=b8bbe3d9ef10d0550d31fb349da01065e7718d15;hb=561f9c8585e7cb0882472297f4fd2ed763dc085a;hp=19748719a0dd285441703109c2cd8875f75d6988;hpb=6ca409e0e4c198fe3081346eebbae3f068fe605a;p=oweals%2Fbusybox.git diff --git a/coreutils/du.c b/coreutils/du.c index 19748719a..b8bbe3d9e 100644 --- a/coreutils/du.c +++ b/coreutils/du.c @@ -6,7 +6,7 @@ * Copyright (C) 1999,2000,2001 by John Beppu * Copyright (C) 2002 Edward Betts * - * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. + * Licensed under GPLv2 or later, see file LICENSE in this source tree. */ /* BB_AUDIT SUSv3 compliant (unless default blocksize set to 1k) */ @@ -23,62 +23,107 @@ * 4) Fixed busybox bug #1284 involving long overflow with human_readable. */ +//usage:#define du_trivial_usage +//usage: "[-aHLdclsx" IF_FEATURE_HUMAN_READABLE("hm") "k] [FILE]..." +//usage:#define du_full_usage "\n\n" +//usage: "Summarize disk space used for each FILE and/or directory.\n" +//usage: "Disk space is printed in units of " +//usage: IF_FEATURE_DU_DEFAULT_BLOCKSIZE_1K("1024") +//usage: IF_NOT_FEATURE_DU_DEFAULT_BLOCKSIZE_1K("512") +//usage: " bytes.\n" +//usage: "\n -a Show file sizes too" +//usage: "\n -L Follow all symlinks" +//usage: "\n -H Follow symlinks on command line" +//usage: "\n -d N Limit output to directories (and files with -a) of depth < N" +//usage: "\n -c Show grand total" +//usage: "\n -l Count sizes many times if hard linked" +//usage: "\n -s Display only a total for each argument" +//usage: "\n -x Skip directories on different filesystems" +//usage: IF_FEATURE_HUMAN_READABLE( +//usage: "\n -h Sizes in human readable format (e.g., 1K 243M 2G )" +//usage: "\n -m Sizes in megabytes" +//usage: ) +//usage: "\n -k Sizes in kilobytes" +//usage: IF_FEATURE_DU_DEFAULT_BLOCKSIZE_1K(" (default)") +//usage: +//usage:#define du_example_usage +//usage: "$ du\n" +//usage: "16 ./CVS\n" +//usage: "12 ./kernel-patches/CVS\n" +//usage: "80 ./kernel-patches\n" +//usage: "12 ./tests/CVS\n" +//usage: "36 ./tests\n" +//usage: "12 ./scripts/CVS\n" +//usage: "16 ./scripts\n" +//usage: "12 ./docs/CVS\n" +//usage: "104 ./docs\n" +//usage: "2417 .\n" + #include "libbb.h" +enum { + OPT_a_files_too = (1 << 0), + OPT_H_follow_links = (1 << 1), + OPT_k_kbytes = (1 << 2), + OPT_L_follow_links = (1 << 3), + OPT_s_total_norecurse = (1 << 4), + OPT_x_one_FS = (1 << 5), + OPT_d_maxdepth = (1 << 6), + OPT_l_hardlinks = (1 << 7), + OPT_c_total = (1 << 8), + OPT_h_for_humans = (1 << 9), + OPT_m_mbytes = (1 << 10), +}; + +struct globals { #if ENABLE_FEATURE_HUMAN_READABLE -# if ENABLE_FEATURE_DU_DEFAULT_BLOCKSIZE_1K -static unsigned long disp_hr = 1024; -# else -static unsigned long disp_hr = 512; -# endif -#elif ENABLE_FEATURE_DU_DEFAULT_BLOCKSIZE_1K -static unsigned disp_k = 1; + unsigned long disp_hr; #else -static unsigned disp_k; /* bss inits to 0 */ + unsigned disp_k; #endif + int max_print_depth; + bool status; + int slink_depth; + int du_depth; + dev_t dir_dev; +} FIX_ALIASING; +#define G (*(struct globals*)&bb_common_bufsiz1) -static int max_print_depth = INT_MAX; -static nlink_t count_hardlinks = 1; - -static int status; -static int print_files; -static int slink_depth; -static int du_depth; -static int one_file_system; -static dev_t dir_dev; - -static void print(long size, const char *const filename) +static void print(unsigned long size, const char *filename) { /* TODO - May not want to defer error checking here. */ #if ENABLE_FEATURE_HUMAN_READABLE - printf("%s\t%s\n", make_human_readable_str(size, 512, disp_hr), + printf("%s\t%s\n", + /* size x 512 / G.disp_hr, show one fractional, + * use suffixes if G.disp_hr == 0 */ + make_human_readable_str(size, 512, G.disp_hr), filename); #else - if (disp_k) { + if (G.disp_k) { size++; size >>= 1; } - printf("%ld\t%s\n", size, filename); + printf("%lu\t%s\n", size, filename); #endif } /* tiny recursive du */ -static long du(const char *const filename) +static unsigned long du(const char *filename) { struct stat statbuf; - long sum; + unsigned long sum; if (lstat(filename, &statbuf) != 0) { - bb_perror_msg("%s", filename); - status = EXIT_FAILURE; + bb_simple_perror_msg(filename); + G.status = EXIT_FAILURE; return 0; } - if (one_file_system) { - if (du_depth == 0) { - dir_dev = statbuf.st_dev; - } else if (dir_dev != statbuf.st_dev) { + if (option_mask32 & OPT_x_one_FS) { + if (G.du_depth == 0) { + G.dir_dev = statbuf.st_dev; + } else if (G.dir_dev != statbuf.st_dev) { return 0; } } @@ -86,20 +131,23 @@ static long du(const char *const filename) sum = statbuf.st_blocks; if (S_ISLNK(statbuf.st_mode)) { - if (slink_depth > du_depth) { /* -H or -L */ + if (G.slink_depth > G.du_depth) { /* -H or -L */ if (stat(filename, &statbuf) != 0) { - bb_perror_msg("%s", filename); - status = EXIT_FAILURE; + bb_simple_perror_msg(filename); + G.status = EXIT_FAILURE; return 0; } sum = statbuf.st_blocks; - if (slink_depth == 1) { - slink_depth = INT_MAX; /* Convert -H to -L. */ + if (G.slink_depth == 1) { + /* Convert -H to -L */ + G.slink_depth = INT_MAX; } } } - if (statbuf.st_nlink > count_hardlinks) { + if (!(option_mask32 & OPT_l_hardlinks) + && statbuf.st_nlink > 1 + ) { /* Add files/directories with links only once */ if (is_in_ino_dev_hashtable(&statbuf)) { return 0; @@ -114,53 +162,47 @@ static long du(const char *const filename) dir = warn_opendir(filename); if (!dir) { - status = EXIT_FAILURE; + G.status = EXIT_FAILURE; return sum; } - newfile = last_char_is(filename, '/'); - if (newfile) - *newfile = '\0'; - while ((entry = readdir(dir))) { - char *name = entry->d_name; - - newfile = concat_subpath_file(filename, name); + newfile = concat_subpath_file(filename, entry->d_name); if (newfile == NULL) continue; - ++du_depth; + ++G.du_depth; sum += du(newfile); - --du_depth; + --G.du_depth; free(newfile); } closedir(dir); - } else if (du_depth > print_files) { - return sum; + } else { + if (!(option_mask32 & OPT_a_files_too) && G.du_depth != 0) + return sum; } - if (du_depth <= max_print_depth) { + if (G.du_depth <= G.max_print_depth) { print(sum, filename); } return sum; } -int du_main(int argc, char **argv); -int du_main(int argc, char **argv) +int du_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; +int du_main(int argc UNUSED_PARAM, char **argv) { - long total; + unsigned long total; int slink_depth_save; - int print_final_total; - char *smax_print_depth; unsigned opt; -#if ENABLE_FEATURE_DU_DEFAULT_BLOCKSIZE_1K - if (getenv("POSIXLY_CORRECT")) { /* TODO - a new libbb function? */ #if ENABLE_FEATURE_HUMAN_READABLE - disp_hr = 512; + IF_FEATURE_DU_DEFAULT_BLOCKSIZE_1K(G.disp_hr = 1024;) + IF_NOT_FEATURE_DU_DEFAULT_BLOCKSIZE_1K(G.disp_hr = 512;) + if (getenv("POSIXLY_CORRECT")) /* TODO - a new libbb function? */ + G.disp_hr = 512; #else - disp_k = 0; -#endif - } + IF_FEATURE_DU_DEFAULT_BLOCKSIZE_1K(G.disp_k = 1;) + /* IF_NOT_FEATURE_DU_DEFAULT_BLOCKSIZE_1K(G.disp_k = 0;) - G is pre-zeroed */ #endif + G.max_print_depth = INT_MAX; /* Note: SUSv3 specifies that -a and -s options cannot be used together * in strictly conforming applications. However, it also says that some @@ -169,77 +211,57 @@ int du_main(int argc, char **argv) * ignore -a. This is consistent with -s being equivalent to -d 0. */ #if ENABLE_FEATURE_HUMAN_READABLE - opt_complementary = "h-km:k-hm:m-hk:H-L:L-H:s-d:d-s"; - opt = getopt32(argc, argv, "aHkLsx" "d:" "lc" "hm", &smax_print_depth); - if (opt & (1 << 9)) { - /* -h opt */ - disp_hr = 0; + opt_complementary = "h-km:k-hm:m-hk:H-L:L-H:s-d:d-s:d+"; + opt = getopt32(argv, "aHkLsx" "d:" "lc" "hm", &G.max_print_depth); + argv += optind; + if (opt & OPT_h_for_humans) { + G.disp_hr = 0; } - if (opt & (1 << 10)) { - /* -m opt */ - disp_hr = 1024*1024; + if (opt & OPT_m_mbytes) { + G.disp_hr = 1024*1024; } - if (opt & (1 << 2)) { - /* -k opt */ - disp_hr = 1024; + if (opt & OPT_k_kbytes) { + G.disp_hr = 1024; } #else - opt_complementary = "H-L:L-H:s-d:d-s"; - opt = getopt32(argc, argv, "aHkLsx" "d:" "lc", &smax_print_depth); + opt_complementary = "H-L:L-H:s-d:d-s:d+"; + opt = getopt32(argv, "aHkLsx" "d:" "lc", &G.max_print_depth); + argv += optind; #if !ENABLE_FEATURE_DU_DEFAULT_BLOCKSIZE_1K - if (opt & (1 << 2)) { - /* -k opt */ - disp_k = 1; + if (opt & OPT_k_kbytes) { + G.disp_k = 1; } #endif #endif - if (opt & (1 << 0)) { - /* -a opt */ - print_files = INT_MAX; - } - if (opt & (1 << 1)) { - /* -H opt */ - slink_depth = 1; - } - if (opt & (1 << 3)) { - /* -L opt */ - slink_depth = INT_MAX; + if (opt & OPT_H_follow_links) { + G.slink_depth = 1; } - if (opt & (1 << 4)) { - /* -s opt */ - max_print_depth = 0; + if (opt & OPT_L_follow_links) { + G.slink_depth = INT_MAX; } - one_file_system = opt & (1 << 5); /* -x opt */ - if (opt & (1 << 6)) { - /* -d opt */ - max_print_depth = xatoi_u(smax_print_depth); + if (opt & OPT_s_total_norecurse) { + G.max_print_depth = 0; } - if (opt & (1 << 7)) { - /* -l opt */ - count_hardlinks = MAXINT(nlink_t); - } - print_final_total = opt & (1 << 8); /* -c opt */ /* go through remaining args (if any) */ - argv += optind; - if (optind >= argc) { + if (!*argv) { *--argv = (char*)"."; - if (slink_depth == 1) { - slink_depth = 0; + if (G.slink_depth == 1) { + G.slink_depth = 0; } } - slink_depth_save = slink_depth; + slink_depth_save = G.slink_depth; total = 0; do { total += du(*argv); - slink_depth = slink_depth_save; - } while (*++argv); - if (ENABLE_FEATURE_CLEAN_UP) + /* otherwise du /dir /dir won't show /dir twice: */ reset_ino_dev_hashtable(); - if (print_final_total) { + G.slink_depth = slink_depth_save; + } while (*++argv); + + if (opt & OPT_c_total) print(total, "total"); - } - fflush_stdout_and_exit(status); + fflush_stdout_and_exit(G.status); }