Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 2 | /* |
| 3 | * Mini du implementation for busybox |
| 4 | * |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 5 | * Copyright (C) 1999,2000,2001 by Lineo, inc. and John Beppu |
| 6 | * Copyright (C) 1999,2000,2001 by John Beppu <beppu@codepoet.org> |
Eric Andersen | 265d229 | 2002-04-06 23:16:44 +0000 | [diff] [blame] | 7 | * Copyright (C) 2002 Edward Betts <edward@debian.org> |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 8 | * |
Bernhard Reutner-Fischer | cb44816 | 2006-04-12 07:35:12 +0000 | [diff] [blame] | 9 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 10 | */ |
| 11 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 12 | /* BB_AUDIT SUSv3 compliant (unless default blocksize set to 1k) */ |
| 13 | /* http://www.opengroup.org/onlinepubs/007904975/utilities/du.html */ |
| 14 | |
| 15 | /* Mar 16, 2003 Manuel Novoa III (mjn3@codepoet.org) |
| 16 | * |
| 17 | * Mostly rewritten for SUSv3 compliance and to fix bugs/defects. |
| 18 | * 1) Added support for SUSv3 -a, -H, -L, gnu -c, and (busybox) -d options. |
| 19 | * The -d option allows setting of max depth (similar to gnu --max-depth). |
| 20 | * 2) Fixed incorrect size calculations for links and directories, especially |
| 21 | * when errors occurred. Calculates sizes should now match gnu du output. |
| 22 | * 3) Added error checking of output. |
| 23 | * 4) Fixed busybox bug #1284 involving long overflow with human_readable. |
| 24 | */ |
| 25 | |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 26 | #include "busybox.h" |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 27 | |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 28 | #ifdef CONFIG_FEATURE_HUMAN_READABLE |
Rob Landley | 5fe8d5b | 2006-04-25 22:40:53 +0000 | [diff] [blame] | 29 | # ifdef CONFIG_FEATURE_DU_DEFAULT_BLOCKSIZE_1K |
Denis Vlasenko | dca0b70 | 2006-10-27 09:05:02 +0000 | [diff] [blame] | 30 | static unsigned long disp_hr = 1024; |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 31 | # else |
| 32 | static unsigned long disp_hr = 512; |
| 33 | # endif |
Rob Landley | 5fe8d5b | 2006-04-25 22:40:53 +0000 | [diff] [blame] | 34 | #elif defined CONFIG_FEATURE_DU_DEFAULT_BLOCKSIZE_1K |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 35 | static unsigned int disp_k = 1; |
| 36 | #else |
| 37 | static unsigned int disp_k; /* bss inits to 0 */ |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 38 | #endif |
| 39 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 40 | static int max_print_depth = INT_MAX; |
"Vladimir N. Oleynik" | 57545c8 | 2006-01-31 12:06:57 +0000 | [diff] [blame] | 41 | static nlink_t count_hardlinks = 1; |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 42 | |
| 43 | static int status |
| 44 | #if EXIT_SUCCESS == 0 |
| 45 | = EXIT_SUCCESS |
| 46 | #endif |
| 47 | ; |
| 48 | static int print_files; |
| 49 | static int slink_depth; |
| 50 | static int du_depth; |
| 51 | static int one_file_system; |
Glenn L McGrath | 14dad70 | 2002-08-23 03:25:22 +0000 | [diff] [blame] | 52 | static dev_t dir_dev; |
John Beppu | e1618e4 | 1999-12-15 18:52:17 +0000 | [diff] [blame] | 53 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 54 | |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 55 | static void print(long size, const char * const filename) |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 56 | { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 57 | /* TODO - May not want to defer error checking here. */ |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 58 | #ifdef CONFIG_FEATURE_HUMAN_READABLE |
Denis Vlasenko | f0ed376 | 2006-10-26 23:21:47 +0000 | [diff] [blame] | 59 | printf("%s\t%s\n", make_human_readable_str(size, 512, disp_hr), |
Glenn L McGrath | 14dad70 | 2002-08-23 03:25:22 +0000 | [diff] [blame] | 60 | filename); |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 61 | #else |
Glenn L McGrath | c66ebe4 | 2004-03-10 09:58:51 +0000 | [diff] [blame] | 62 | if (disp_k) { |
| 63 | size++; |
| 64 | size >>= 1; |
| 65 | } |
Denis Vlasenko | f0ed376 | 2006-10-26 23:21:47 +0000 | [diff] [blame] | 66 | printf("%ld\t%s\n", size, filename); |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 67 | #endif |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 68 | } |
| 69 | |
| 70 | /* tiny recursive du */ |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 71 | static long du(const char * const filename) |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 72 | { |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 73 | struct stat statbuf; |
Erik Andersen | fac10d7 | 2000-02-07 05:29:42 +0000 | [diff] [blame] | 74 | long sum; |
John Beppu | 14c82b6 | 1999-12-10 06:15:27 +0000 | [diff] [blame] | 75 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 76 | if ((lstat(filename, &statbuf)) != 0) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 77 | bb_perror_msg("%s", filename); |
| 78 | status = EXIT_FAILURE; |
Eric Andersen | e5dfced | 2001-04-09 22:48:12 +0000 | [diff] [blame] | 79 | return 0; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 80 | } |
| 81 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 82 | if (one_file_system) { |
| 83 | if (du_depth == 0) { |
| 84 | dir_dev = statbuf.st_dev; |
| 85 | } else if (dir_dev != statbuf.st_dev) { |
| 86 | return 0; |
Eric Andersen | 8fa1bf7 | 2001-06-30 17:54:20 +0000 | [diff] [blame] | 87 | } |
Erik Andersen | 9ffdaa6 | 2000-02-11 21:55:04 +0000 | [diff] [blame] | 88 | } |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 89 | |
| 90 | sum = statbuf.st_blocks; |
| 91 | |
| 92 | if (S_ISLNK(statbuf.st_mode)) { |
| 93 | if (slink_depth > du_depth) { /* -H or -L */ |
| 94 | if ((stat(filename, &statbuf)) != 0) { |
| 95 | bb_perror_msg("%s", filename); |
| 96 | status = EXIT_FAILURE; |
| 97 | return 0; |
| 98 | } |
| 99 | sum = statbuf.st_blocks; |
| 100 | if (slink_depth == 1) { |
| 101 | slink_depth = INT_MAX; /* Convert -H to -L. */ |
| 102 | } |
| 103 | } |
| 104 | } |
| 105 | |
| 106 | if (statbuf.st_nlink > count_hardlinks) { |
| 107 | /* Add files/directories with links only once */ |
| 108 | if (is_in_ino_dev_hashtable(&statbuf, NULL)) { |
| 109 | return 0; |
| 110 | } |
| 111 | add_to_ino_dev_hashtable(&statbuf, NULL); |
| 112 | } |
| 113 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 114 | if (S_ISDIR(statbuf.st_mode)) { |
| 115 | DIR *dir; |
| 116 | struct dirent *entry; |
Eric Andersen | 04b0354 | 2001-05-07 22:49:43 +0000 | [diff] [blame] | 117 | char *newfile; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 118 | |
Rob Landley | 86b4d64 | 2006-08-03 17:58:17 +0000 | [diff] [blame] | 119 | dir = warn_opendir(filename); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 120 | if (!dir) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 121 | status = EXIT_FAILURE; |
| 122 | return sum; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 123 | } |
Erik Andersen | 42387e4 | 2000-02-21 17:27:17 +0000 | [diff] [blame] | 124 | |
Eric Andersen | 04b0354 | 2001-05-07 22:49:43 +0000 | [diff] [blame] | 125 | newfile = last_char_is(filename, '/'); |
| 126 | if (newfile) |
| 127 | *newfile = '\0'; |
Erik Andersen | 42387e4 | 2000-02-21 17:27:17 +0000 | [diff] [blame] | 128 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 129 | while ((entry = readdir(dir))) { |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 130 | char *name = entry->d_name; |
| 131 | |
Glenn L McGrath | 393183d | 2003-05-26 14:07:50 +0000 | [diff] [blame] | 132 | newfile = concat_subpath_file(filename, name); |
| 133 | if(newfile == NULL) |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 134 | continue; |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 135 | ++du_depth; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 136 | sum += du(newfile); |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 137 | --du_depth; |
Eric Andersen | e5dfced | 2001-04-09 22:48:12 +0000 | [diff] [blame] | 138 | free(newfile); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 139 | } |
| 140 | closedir(dir); |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 141 | } else if (du_depth > print_files) { |
| 142 | return sum; |
Erik Andersen | 27fdd08 | 2000-02-19 18:16:49 +0000 | [diff] [blame] | 143 | } |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 144 | if (du_depth <= max_print_depth) { |
| 145 | print(sum, filename); |
| 146 | } |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 147 | return sum; |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 148 | } |
| 149 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 150 | int du_main(int argc, char **argv) |
| 151 | { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 152 | long total; |
| 153 | int slink_depth_save; |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 154 | int print_final_total; |
| 155 | char *smax_print_depth; |
Denis Vlasenko | 67b23e6 | 2006-10-03 21:00:06 +0000 | [diff] [blame] | 156 | unsigned opt; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 157 | |
Rob Landley | 5fe8d5b | 2006-04-25 22:40:53 +0000 | [diff] [blame] | 158 | #ifdef CONFIG_FEATURE_DU_DEFUALT_BLOCKSIZE_1K |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 159 | if (getenv("POSIXLY_CORRECT")) { /* TODO - a new libbb function? */ |
| 160 | #ifdef CONFIG_FEATURE_HUMAN_READABLE |
| 161 | disp_hr = 512; |
| 162 | #else |
| 163 | disp_k = 0; |
| 164 | #endif |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 165 | } |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 166 | #endif |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 167 | |
Denis Vlasenko | e1a0d48 | 2006-10-20 13:28:22 +0000 | [diff] [blame] | 168 | /* Note: SUSv3 specifies that -a and -s options cannot be used together |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 169 | * in strictly conforming applications. However, it also says that some |
| 170 | * du implementations may produce output when -a and -s are used together. |
| 171 | * gnu du exits with an error code in this case. We choose to simply |
| 172 | * ignore -a. This is consistent with -s being equivalent to -d 0. |
| 173 | */ |
Eric Andersen | bdfd0d7 | 2001-10-24 05:00:29 +0000 | [diff] [blame] | 174 | #ifdef CONFIG_FEATURE_HUMAN_READABLE |
Denis Vlasenko | 67b23e6 | 2006-10-03 21:00:06 +0000 | [diff] [blame] | 175 | opt_complementary = "h-km:k-hm:m-hk:H-L:L-H:s-d:d-s"; |
| 176 | opt = getopt32(argc, argv, "aHkLsx" "d:" "lc" "hm", &smax_print_depth); |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 177 | if((opt & (1 << 9))) { |
| 178 | /* -h opt */ |
| 179 | disp_hr = 0; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 180 | } |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 181 | if((opt & (1 << 10))) { |
| 182 | /* -m opt */ |
Denis Vlasenko | dca0b70 | 2006-10-27 09:05:02 +0000 | [diff] [blame] | 183 | disp_hr = 1024*1024; |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 184 | } |
| 185 | if((opt & (1 << 2))) { |
| 186 | /* -k opt */ |
Denis Vlasenko | dca0b70 | 2006-10-27 09:05:02 +0000 | [diff] [blame] | 187 | disp_hr = 1024; |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 188 | } |
| 189 | #else |
Denis Vlasenko | 67b23e6 | 2006-10-03 21:00:06 +0000 | [diff] [blame] | 190 | opt_complementary = "H-L:L-H:s-d:d-s"; |
| 191 | opt = getopt32(argc, argv, "aHkLsx" "d:" "lc", &smax_print_depth); |
Rob Landley | 5fe8d5b | 2006-04-25 22:40:53 +0000 | [diff] [blame] | 192 | #if !defined CONFIG_FEATURE_DU_DEFAULT_BLOCKSIZE_1K |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 193 | if((opt & (1 << 2))) { |
| 194 | /* -k opt */ |
| 195 | disp_k = 1; |
| 196 | } |
| 197 | #endif |
| 198 | #endif |
| 199 | if((opt & (1 << 0))) { |
| 200 | /* -a opt */ |
| 201 | print_files = INT_MAX; |
| 202 | } |
| 203 | if((opt & (1 << 1))) { |
| 204 | /* -H opt */ |
| 205 | slink_depth = 1; |
| 206 | } |
| 207 | if((opt & (1 << 3))) { |
| 208 | /* -L opt */ |
| 209 | slink_depth = INT_MAX; |
| 210 | } |
| 211 | if((opt & (1 << 4))) { |
| 212 | /* -s opt */ |
| 213 | max_print_depth = 0; |
| 214 | } |
| 215 | one_file_system = opt & (1 << 5); /* -x opt */ |
| 216 | if((opt & (1 << 6))) { |
| 217 | /* -d opt */ |
Denis Vlasenko | 1385899 | 2006-10-08 12:49:22 +0000 | [diff] [blame] | 218 | max_print_depth = xatoi_u(smax_print_depth); |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 219 | } |
| 220 | if((opt & (1 << 7))) { |
| 221 | /* -l opt */ |
Glenn L McGrath | 44c0e17 | 2004-03-10 09:10:53 +0000 | [diff] [blame] | 222 | count_hardlinks = INT_MAX; |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 223 | } |
| 224 | print_final_total = opt & (1 << 8); /* -c opt */ |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 225 | |
| 226 | /* go through remaining args (if any) */ |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 227 | argv += optind; |
Eric Andersen | 17ad45a | 2000-07-14 18:38:26 +0000 | [diff] [blame] | 228 | if (optind >= argc) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 229 | *--argv = "."; |
| 230 | if (slink_depth == 1) { |
| 231 | slink_depth = 0; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 232 | } |
| 233 | } |
| 234 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 235 | slink_depth_save = slink_depth; |
| 236 | total = 0; |
| 237 | do { |
| 238 | total += du(*argv); |
| 239 | slink_depth = slink_depth_save; |
| 240 | } while (*++argv); |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 241 | #ifdef CONFIG_FEATURE_CLEAN_UP |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 242 | reset_ino_dev_hashtable(); |
Eric Andersen | 8876fb2 | 2003-06-20 09:01:58 +0000 | [diff] [blame] | 243 | #endif |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 244 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 245 | if (print_final_total) { |
| 246 | print(total, "total"); |
| 247 | } |
| 248 | |
Denis Vlasenko | f0ed376 | 2006-10-26 23:21:47 +0000 | [diff] [blame] | 249 | fflush_stdout_and_exit(status); |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 250 | } |