blob: 3e4821a390010fa885d1a2937963f43eac2a025d [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 *
5 *
Eric Andersen8ec10a92001-01-27 09:33:39 +00006 * Copyright (C) 1999,2000,2001 by Lineo, inc.
Eric Andersen70e2f0b1999-12-10 06:45:42 +00007 * Written by John Beppu <beppu@lineo.com>
John Beppu0f5e1ab1999-12-09 18:23:54 +00008 *
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 * General Public License for more details.
18 *
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 *
23 */
24
John Beppu0f5e1ab1999-12-09 18:23:54 +000025#include <sys/types.h>
26#include <fcntl.h>
27#include <dirent.h>
28#include <stdio.h>
Eric Andersened3ef502001-01-27 08:24:39 +000029#include <stdlib.h>
30#include <getopt.h>
Eric Anderseneba8ed72001-03-09 14:36:42 +000031#include <string.h>
John Beppu98355411999-12-10 07:40:08 +000032#include <errno.h>
Eric Andersencbe31da2001-02-20 06:14:08 +000033#include "busybox.h"
Eric Andersencbe31da2001-02-20 06:14:08 +000034
John Beppu0f5e1ab1999-12-09 18:23:54 +000035
Richard June6d0921c2001-01-22 22:35:38 +000036#ifdef BB_FEATURE_HUMAN_READABLE
Eric Andersenec9fad92001-03-07 06:04:08 +000037static unsigned long disp_hr = KILOBYTE;
Richard June6d0921c2001-01-22 22:35:38 +000038#endif
39
Erik Andersene49d5ec2000-02-08 19:58:47 +000040typedef void (Display) (long, char *);
John Beppu0f5e1ab1999-12-09 18:23:54 +000041
Erik Andersene49d5ec2000-02-08 19:58:47 +000042static int du_depth = 0;
Erik Andersen27fdd082000-02-19 18:16:49 +000043static int count_hardlinks = 0;
John Beppue1618e41999-12-15 18:52:17 +000044
Erik Andersene49d5ec2000-02-08 19:58:47 +000045static Display *print;
46
47static void print_normal(long size, char *filename)
John Beppu0f5e1ab1999-12-09 18:23:54 +000048{
Eric Andersenec9fad92001-03-07 06:04:08 +000049 unsigned long base;
Richard June6d0921c2001-01-22 22:35:38 +000050#ifdef BB_FEATURE_HUMAN_READABLE
Eric Andersenec9fad92001-03-07 06:04:08 +000051 switch (disp_hr) {
52 case MEGABYTE:
53 base = KILOBYTE;
54 break;
55 case KILOBYTE:
56 base = 1;
57 break;
58 default:
59 base = 0;
60 }
Mark Whitleyae5612c2001-03-07 17:42:07 +000061 printf("%s\t%s\n", make_human_readable_str(size, base), filename);
Richard June6d0921c2001-01-22 22:35:38 +000062#else
Matt Kraai12f417e2001-01-18 02:57:08 +000063 printf("%ld\t%s\n", size, filename);
Richard June6d0921c2001-01-22 22:35:38 +000064#endif
John Beppu0f5e1ab1999-12-09 18:23:54 +000065}
66
Erik Andersene49d5ec2000-02-08 19:58:47 +000067static void print_summary(long size, char *filename)
John Beppue1618e41999-12-15 18:52:17 +000068{
Erik Andersene49d5ec2000-02-08 19:58:47 +000069 if (du_depth == 1) {
70 print_normal(size, filename);
71 }
John Beppue1618e41999-12-15 18:52:17 +000072}
73
John Beppu0f5e1ab1999-12-09 18:23:54 +000074/* tiny recursive du */
Erik Andersene49d5ec2000-02-08 19:58:47 +000075static long du(char *filename)
John Beppu0f5e1ab1999-12-09 18:23:54 +000076{
Erik Andersene49d5ec2000-02-08 19:58:47 +000077 struct stat statbuf;
Erik Andersenfac10d72000-02-07 05:29:42 +000078 long sum;
John Beppu14c82b61999-12-10 06:15:27 +000079
Erik Andersene49d5ec2000-02-08 19:58:47 +000080 if ((lstat(filename, &statbuf)) != 0) {
Eric Andersene5dfced2001-04-09 22:48:12 +000081 perror_msg("%s", filename);
82 return 0;
Erik Andersene49d5ec2000-02-08 19:58:47 +000083 }
84
85 du_depth++;
John Beppu08c965a2000-02-13 04:10:57 +000086 sum = (statbuf.st_blocks >> 1);
Erik Andersene49d5ec2000-02-08 19:58:47 +000087
Erik Andersen27fdd082000-02-19 18:16:49 +000088 /* Don't add in stuff pointed to by symbolic links */
Erik Andersen9ffdaa62000-02-11 21:55:04 +000089 if (S_ISLNK(statbuf.st_mode)) {
Erik Andersen42387e42000-02-21 17:27:17 +000090 sum = 0L;
91 if (du_depth == 1)
92 print(sum, filename);
Erik Andersen9ffdaa62000-02-11 21:55:04 +000093 }
Erik Andersene49d5ec2000-02-08 19:58:47 +000094 if (S_ISDIR(statbuf.st_mode)) {
95 DIR *dir;
96 struct dirent *entry;
Eric Andersen04b03542001-05-07 22:49:43 +000097 char *newfile;
Erik Andersene49d5ec2000-02-08 19:58:47 +000098
99 dir = opendir(filename);
100 if (!dir) {
Erik Andersen42387e42000-02-21 17:27:17 +0000101 du_depth--;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000102 return 0;
103 }
Erik Andersen42387e42000-02-21 17:27:17 +0000104
Eric Andersen04b03542001-05-07 22:49:43 +0000105 newfile = last_char_is(filename, '/');
106 if (newfile)
107 *newfile = '\0';
Erik Andersen42387e42000-02-21 17:27:17 +0000108
Erik Andersene49d5ec2000-02-08 19:58:47 +0000109 while ((entry = readdir(dir))) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000110 char *name = entry->d_name;
111
112 if ((strcmp(name, "..") == 0)
113 || (strcmp(name, ".") == 0)) {
114 continue;
115 }
Eric Andersene5dfced2001-04-09 22:48:12 +0000116 newfile = concat_path_file(filename, name);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000117 sum += du(newfile);
Eric Andersene5dfced2001-04-09 22:48:12 +0000118 free(newfile);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000119 }
120 closedir(dir);
121 print(sum, filename);
122 }
Erik Andersen27fdd082000-02-19 18:16:49 +0000123 else if (statbuf.st_nlink > 1 && !count_hardlinks) {
124 /* Add files with hard links only once */
Erik Andersen029011b2000-03-04 21:19:32 +0000125 if (is_in_ino_dev_hashtable(&statbuf, NULL)) {
Erik Andersen42387e42000-02-21 17:27:17 +0000126 sum = 0L;
127 if (du_depth == 1)
128 print(sum, filename);
129 }
130 else {
Erik Andersen029011b2000-03-04 21:19:32 +0000131 add_to_ino_dev_hashtable(&statbuf, NULL);
Erik Andersen42387e42000-02-21 17:27:17 +0000132 }
Erik Andersen27fdd082000-02-19 18:16:49 +0000133 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000134 du_depth--;
135 return sum;
John Beppu0f5e1ab1999-12-09 18:23:54 +0000136}
137
Erik Andersene49d5ec2000-02-08 19:58:47 +0000138int du_main(int argc, char **argv)
139{
Matt Kraai92ed8a32000-12-06 15:55:23 +0000140 int status = EXIT_SUCCESS;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000141 int i;
Pavel Roskin47d49262000-07-17 16:17:19 +0000142 int c;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000143
144 /* default behaviour */
145 print = print_normal;
146
147 /* parse argv[] */
Richard June6d0921c2001-01-22 22:35:38 +0000148 while ((c = getopt(argc, argv, "sl"
149#ifdef BB_FEATURE_HUMAN_READABLE
150"hm"
151#endif
152"k")) != EOF) {
Eric Andersen17ad45a2000-07-14 18:38:26 +0000153 switch (c) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000154 case 's':
Eric Andersen17ad45a2000-07-14 18:38:26 +0000155 print = print_summary;
156 break;
Erik Andersen27fdd082000-02-19 18:16:49 +0000157 case 'l':
Eric Andersen17ad45a2000-07-14 18:38:26 +0000158 count_hardlinks = 1;
159 break;
Richard June6d0921c2001-01-22 22:35:38 +0000160#ifdef BB_FEATURE_HUMAN_READABLE
Eric Andersenec9fad92001-03-07 06:04:08 +0000161 case 'h': disp_hr = 0; break;
162 case 'm': disp_hr = MEGABYTE; break;
Richard June6d0921c2001-01-22 22:35:38 +0000163#endif
Eric Andersen8b728a22001-03-06 23:14:43 +0000164 case 'k': break;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000165 default:
Eric Andersen67991cf2001-02-14 21:23:06 +0000166 show_usage();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000167 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000168 }
169
170 /* go through remaining args (if any) */
Eric Andersen17ad45a2000-07-14 18:38:26 +0000171 if (optind >= argc) {
Matt Kraai92ed8a32000-12-06 15:55:23 +0000172 if (du(".") == 0)
173 status = EXIT_FAILURE;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000174 } else {
175 long sum;
176
Eric Andersen17ad45a2000-07-14 18:38:26 +0000177 for (i=optind; i < argc; i++) {
Matt Kraaie8849702000-12-06 15:56:31 +0000178 if ((sum = du(argv[i])) == 0)
Matt Kraai92ed8a32000-12-06 15:55:23 +0000179 status = EXIT_FAILURE;
Richard June6d0921c2001-01-22 22:35:38 +0000180 if(is_directory(argv[i], FALSE, NULL)==FALSE) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000181 print_normal(sum, argv[i]);
182 }
Erik Andersen029011b2000-03-04 21:19:32 +0000183 reset_ino_dev_hashtable();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000184 }
185 }
186
Matt Kraai92ed8a32000-12-06 15:55:23 +0000187 return status;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000188}
189
Eric Andersen04b03542001-05-07 22:49:43 +0000190/* $Id: du.c,v 1.47 2001/05/07 22:49:43 andersen Exp $ */
Erik Andersen029011b2000-03-04 21:19:32 +0000191/*
192Local Variables:
193c-file-style: "linux"
194c-basic-offset: 4
195tab-width: 4
196End:
197*/