blob: a547b1e14c7bff35c1397d02970758ec25e50df7 [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
John Beppu0f5e1ab1999-12-09 18:23:54 +00002/*
3 * Mini du implementation for busybox
4 *
Eric Andersenbdfd0d72001-10-24 05:00:29 +00005 * Copyright (C) 1999,2000,2001 by Lineo, inc. and John Beppu
6 * Copyright (C) 1999,2000,2001 by John Beppu <beppu@codepoet.org>
Eric Andersen265d2292002-04-06 23:16:44 +00007 * Copyright (C) 2002 Edward Betts <edward@debian.org>
John Beppu0f5e1ab1999-12-09 18:23:54 +00008 *
Bernhard Reutner-Fischercb448162006-04-12 07:35:12 +00009 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
John Beppu0f5e1ab1999-12-09 18:23:54 +000010 */
11
Manuel Novoa III cad53642003-03-19 09:13:01 +000012/* 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 Andersencbe31da2001-02-20 06:14:08 +000026#include "busybox.h"
Eric Andersencbe31da2001-02-20 06:14:08 +000027
Eric Andersenbdfd0d72001-10-24 05:00:29 +000028#ifdef CONFIG_FEATURE_HUMAN_READABLE
Rob Landley5fe8d5b2006-04-25 22:40:53 +000029# ifdef CONFIG_FEATURE_DU_DEFAULT_BLOCKSIZE_1K
Denis Vlasenkodca0b702006-10-27 09:05:02 +000030static unsigned long disp_hr = 1024;
Manuel Novoa III cad53642003-03-19 09:13:01 +000031# else
32static unsigned long disp_hr = 512;
33# endif
Rob Landley5fe8d5b2006-04-25 22:40:53 +000034#elif defined CONFIG_FEATURE_DU_DEFAULT_BLOCKSIZE_1K
Manuel Novoa III cad53642003-03-19 09:13:01 +000035static unsigned int disp_k = 1;
36#else
37static unsigned int disp_k; /* bss inits to 0 */
Richard June6d0921c2001-01-22 22:35:38 +000038#endif
39
Manuel Novoa III cad53642003-03-19 09:13:01 +000040static int max_print_depth = INT_MAX;
"Vladimir N. Oleynik"57545c82006-01-31 12:06:57 +000041static nlink_t count_hardlinks = 1;
Manuel Novoa III cad53642003-03-19 09:13:01 +000042
43static int status
44#if EXIT_SUCCESS == 0
45 = EXIT_SUCCESS
46#endif
47 ;
48static int print_files;
49static int slink_depth;
50static int du_depth;
51static int one_file_system;
Glenn L McGrath14dad702002-08-23 03:25:22 +000052static dev_t dir_dev;
John Beppue1618e41999-12-15 18:52:17 +000053
Erik Andersene49d5ec2000-02-08 19:58:47 +000054
Bernhard Reutner-Fischer73561cc2006-08-28 23:31:54 +000055static void print(long size, const char * const filename)
John Beppu0f5e1ab1999-12-09 18:23:54 +000056{
Manuel Novoa III cad53642003-03-19 09:13:01 +000057 /* TODO - May not want to defer error checking here. */
Eric Andersenbdfd0d72001-10-24 05:00:29 +000058#ifdef CONFIG_FEATURE_HUMAN_READABLE
Denis Vlasenkof0ed3762006-10-26 23:21:47 +000059 printf("%s\t%s\n", make_human_readable_str(size, 512, disp_hr),
Glenn L McGrath14dad702002-08-23 03:25:22 +000060 filename);
Richard June6d0921c2001-01-22 22:35:38 +000061#else
Glenn L McGrathc66ebe42004-03-10 09:58:51 +000062 if (disp_k) {
63 size++;
64 size >>= 1;
65 }
Denis Vlasenkof0ed3762006-10-26 23:21:47 +000066 printf("%ld\t%s\n", size, filename);
Richard June6d0921c2001-01-22 22:35:38 +000067#endif
John Beppu0f5e1ab1999-12-09 18:23:54 +000068}
69
70/* tiny recursive du */
Bernhard Reutner-Fischer73561cc2006-08-28 23:31:54 +000071static long du(const char * const filename)
John Beppu0f5e1ab1999-12-09 18:23:54 +000072{
Erik Andersene49d5ec2000-02-08 19:58:47 +000073 struct stat statbuf;
Erik Andersenfac10d72000-02-07 05:29:42 +000074 long sum;
John Beppu14c82b61999-12-10 06:15:27 +000075
Erik Andersene49d5ec2000-02-08 19:58:47 +000076 if ((lstat(filename, &statbuf)) != 0) {
Manuel Novoa III cad53642003-03-19 09:13:01 +000077 bb_perror_msg("%s", filename);
78 status = EXIT_FAILURE;
Eric Andersene5dfced2001-04-09 22:48:12 +000079 return 0;
Erik Andersene49d5ec2000-02-08 19:58:47 +000080 }
81
Manuel Novoa III cad53642003-03-19 09:13:01 +000082 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 Andersen8fa1bf72001-06-30 17:54:20 +000087 }
Erik Andersen9ffdaa62000-02-11 21:55:04 +000088 }
Manuel Novoa III cad53642003-03-19 09:13:01 +000089
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 Andersene49d5ec2000-02-08 19:58:47 +0000114 if (S_ISDIR(statbuf.st_mode)) {
115 DIR *dir;
116 struct dirent *entry;
Eric Andersen04b03542001-05-07 22:49:43 +0000117 char *newfile;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000118
Rob Landley86b4d642006-08-03 17:58:17 +0000119 dir = warn_opendir(filename);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000120 if (!dir) {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000121 status = EXIT_FAILURE;
122 return sum;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000123 }
Erik Andersen42387e42000-02-21 17:27:17 +0000124
Eric Andersen04b03542001-05-07 22:49:43 +0000125 newfile = last_char_is(filename, '/');
126 if (newfile)
127 *newfile = '\0';
Erik Andersen42387e42000-02-21 17:27:17 +0000128
Erik Andersene49d5ec2000-02-08 19:58:47 +0000129 while ((entry = readdir(dir))) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000130 char *name = entry->d_name;
131
Glenn L McGrath393183d2003-05-26 14:07:50 +0000132 newfile = concat_subpath_file(filename, name);
133 if(newfile == NULL)
Erik Andersene49d5ec2000-02-08 19:58:47 +0000134 continue;
Manuel Novoa III cad53642003-03-19 09:13:01 +0000135 ++du_depth;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000136 sum += du(newfile);
Manuel Novoa III cad53642003-03-19 09:13:01 +0000137 --du_depth;
Eric Andersene5dfced2001-04-09 22:48:12 +0000138 free(newfile);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000139 }
140 closedir(dir);
Manuel Novoa III cad53642003-03-19 09:13:01 +0000141 } else if (du_depth > print_files) {
142 return sum;
Erik Andersen27fdd082000-02-19 18:16:49 +0000143 }
Manuel Novoa III cad53642003-03-19 09:13:01 +0000144 if (du_depth <= max_print_depth) {
145 print(sum, filename);
146 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000147 return sum;
John Beppu0f5e1ab1999-12-09 18:23:54 +0000148}
149
Erik Andersene49d5ec2000-02-08 19:58:47 +0000150int du_main(int argc, char **argv)
151{
Manuel Novoa III cad53642003-03-19 09:13:01 +0000152 long total;
153 int slink_depth_save;
Eric Andersen8876fb22003-06-20 09:01:58 +0000154 int print_final_total;
155 char *smax_print_depth;
Denis Vlasenko67b23e62006-10-03 21:00:06 +0000156 unsigned opt;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000157
Rob Landley5fe8d5b2006-04-25 22:40:53 +0000158#ifdef CONFIG_FEATURE_DU_DEFUALT_BLOCKSIZE_1K
Manuel Novoa III cad53642003-03-19 09:13:01 +0000159 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 Andersenc7bda1c2004-03-15 08:29:22 +0000165 }
Manuel Novoa III cad53642003-03-19 09:13:01 +0000166#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000167
Denis Vlasenkoe1a0d482006-10-20 13:28:22 +0000168 /* Note: SUSv3 specifies that -a and -s options cannot be used together
Manuel Novoa III cad53642003-03-19 09:13:01 +0000169 * 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 Andersenbdfd0d72001-10-24 05:00:29 +0000174#ifdef CONFIG_FEATURE_HUMAN_READABLE
Denis Vlasenko67b23e62006-10-03 21:00:06 +0000175 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 Andersen8876fb22003-06-20 09:01:58 +0000177 if((opt & (1 << 9))) {
178 /* -h opt */
179 disp_hr = 0;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000180 }
Eric Andersen8876fb22003-06-20 09:01:58 +0000181 if((opt & (1 << 10))) {
182 /* -m opt */
Denis Vlasenkodca0b702006-10-27 09:05:02 +0000183 disp_hr = 1024*1024;
Eric Andersen8876fb22003-06-20 09:01:58 +0000184 }
185 if((opt & (1 << 2))) {
186 /* -k opt */
Denis Vlasenkodca0b702006-10-27 09:05:02 +0000187 disp_hr = 1024;
Eric Andersen8876fb22003-06-20 09:01:58 +0000188 }
189#else
Denis Vlasenko67b23e62006-10-03 21:00:06 +0000190 opt_complementary = "H-L:L-H:s-d:d-s";
191 opt = getopt32(argc, argv, "aHkLsx" "d:" "lc", &smax_print_depth);
Rob Landley5fe8d5b2006-04-25 22:40:53 +0000192#if !defined CONFIG_FEATURE_DU_DEFAULT_BLOCKSIZE_1K
Eric Andersen8876fb22003-06-20 09:01:58 +0000193 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 Vlasenko13858992006-10-08 12:49:22 +0000218 max_print_depth = xatoi_u(smax_print_depth);
Eric Andersen8876fb22003-06-20 09:01:58 +0000219 }
220 if((opt & (1 << 7))) {
221 /* -l opt */
Glenn L McGrath44c0e172004-03-10 09:10:53 +0000222 count_hardlinks = INT_MAX;
Eric Andersen8876fb22003-06-20 09:01:58 +0000223 }
224 print_final_total = opt & (1 << 8); /* -c opt */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000225
226 /* go through remaining args (if any) */
Manuel Novoa III cad53642003-03-19 09:13:01 +0000227 argv += optind;
Eric Andersen17ad45a2000-07-14 18:38:26 +0000228 if (optind >= argc) {
Manuel Novoa III cad53642003-03-19 09:13:01 +0000229 *--argv = ".";
230 if (slink_depth == 1) {
231 slink_depth = 0;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000232 }
233 }
234
Manuel Novoa III cad53642003-03-19 09:13:01 +0000235 slink_depth_save = slink_depth;
236 total = 0;
237 do {
238 total += du(*argv);
239 slink_depth = slink_depth_save;
240 } while (*++argv);
Eric Andersen8876fb22003-06-20 09:01:58 +0000241#ifdef CONFIG_FEATURE_CLEAN_UP
Manuel Novoa III cad53642003-03-19 09:13:01 +0000242 reset_ino_dev_hashtable();
Eric Andersen8876fb22003-06-20 09:01:58 +0000243#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000244
Manuel Novoa III cad53642003-03-19 09:13:01 +0000245 if (print_final_total) {
246 print(total, "total");
247 }
248
Denis Vlasenkof0ed3762006-10-26 23:21:47 +0000249 fflush_stdout_and_exit(status);
Manuel Novoa III cad53642003-03-19 09:13:01 +0000250}