blob: c56afbc4fee9f769fedef5180a3a1133e2807584 [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
Eric Andersen3570a342000-09-25 21:45:58 +000025#include "busybox.h"
Erik Andersenfac10d72000-02-07 05:29:42 +000026#define BB_DECLARE_EXTERN
27#define bb_need_name_too_long
28#include "messages.c"
29
John Beppu0f5e1ab1999-12-09 18:23:54 +000030#include <sys/types.h>
31#include <fcntl.h>
32#include <dirent.h>
33#include <stdio.h>
Eric Andersened3ef502001-01-27 08:24:39 +000034#include <stdlib.h>
35#include <getopt.h>
John Beppu98355411999-12-10 07:40:08 +000036#include <errno.h>
John Beppu0f5e1ab1999-12-09 18:23:54 +000037
Richard June6d0921c2001-01-22 22:35:38 +000038#ifdef BB_FEATURE_HUMAN_READABLE
39unsigned long du_disp_hr = KILOBYTE;
40#endif
41
Erik Andersene49d5ec2000-02-08 19:58:47 +000042typedef void (Display) (long, char *);
John Beppu0f5e1ab1999-12-09 18:23:54 +000043
Erik Andersene49d5ec2000-02-08 19:58:47 +000044static int du_depth = 0;
Erik Andersen27fdd082000-02-19 18:16:49 +000045static int count_hardlinks = 0;
John Beppue1618e41999-12-15 18:52:17 +000046
Erik Andersene49d5ec2000-02-08 19:58:47 +000047static Display *print;
48
49static void print_normal(long size, char *filename)
John Beppu0f5e1ab1999-12-09 18:23:54 +000050{
Richard June6d0921c2001-01-22 22:35:38 +000051#ifdef BB_FEATURE_HUMAN_READABLE
52 printf("%s\t%s\n", format((size * KILOBYTE), du_disp_hr), filename);
53#else
Matt Kraai12f417e2001-01-18 02:57:08 +000054 printf("%ld\t%s\n", size, filename);
Richard June6d0921c2001-01-22 22:35:38 +000055#endif
John Beppu0f5e1ab1999-12-09 18:23:54 +000056}
57
Erik Andersene49d5ec2000-02-08 19:58:47 +000058static void print_summary(long size, char *filename)
John Beppue1618e41999-12-15 18:52:17 +000059{
Erik Andersene49d5ec2000-02-08 19:58:47 +000060 if (du_depth == 1) {
Richard June6d0921c2001-01-22 22:35:38 +000061printf("summary\n");
Erik Andersene49d5ec2000-02-08 19:58:47 +000062 print_normal(size, filename);
63 }
John Beppue1618e41999-12-15 18:52:17 +000064}
65
John Beppu0f5e1ab1999-12-09 18:23:54 +000066/* tiny recursive du */
Erik Andersene49d5ec2000-02-08 19:58:47 +000067static long du(char *filename)
John Beppu0f5e1ab1999-12-09 18:23:54 +000068{
Erik Andersene49d5ec2000-02-08 19:58:47 +000069 struct stat statbuf;
Erik Andersenfac10d72000-02-07 05:29:42 +000070 long sum;
Erik Andersen42387e42000-02-21 17:27:17 +000071 int len;
John Beppu14c82b61999-12-10 06:15:27 +000072
Erik Andersene49d5ec2000-02-08 19:58:47 +000073 if ((lstat(filename, &statbuf)) != 0) {
Eric Andersenb12e5062000-12-12 23:17:26 +000074 perror_msg_and_die("%s", filename);
Erik Andersene49d5ec2000-02-08 19:58:47 +000075 }
76
77 du_depth++;
John Beppu08c965a2000-02-13 04:10:57 +000078 sum = (statbuf.st_blocks >> 1);
Erik Andersene49d5ec2000-02-08 19:58:47 +000079
Erik Andersen27fdd082000-02-19 18:16:49 +000080 /* Don't add in stuff pointed to by symbolic links */
Erik Andersen9ffdaa62000-02-11 21:55:04 +000081 if (S_ISLNK(statbuf.st_mode)) {
Erik Andersen42387e42000-02-21 17:27:17 +000082 sum = 0L;
83 if (du_depth == 1)
84 print(sum, filename);
Erik Andersen9ffdaa62000-02-11 21:55:04 +000085 }
Erik Andersene49d5ec2000-02-08 19:58:47 +000086 if (S_ISDIR(statbuf.st_mode)) {
87 DIR *dir;
88 struct dirent *entry;
89
90 dir = opendir(filename);
91 if (!dir) {
Erik Andersen42387e42000-02-21 17:27:17 +000092 du_depth--;
Erik Andersene49d5ec2000-02-08 19:58:47 +000093 return 0;
94 }
Erik Andersen42387e42000-02-21 17:27:17 +000095
96 len = strlen(filename);
97 if (filename[len - 1] == '/')
98 filename[--len] = '\0';
99
Erik Andersene49d5ec2000-02-08 19:58:47 +0000100 while ((entry = readdir(dir))) {
Erik Andersen4f3f7572000-04-28 00:18:56 +0000101 char newfile[BUFSIZ + 1];
Erik Andersene49d5ec2000-02-08 19:58:47 +0000102 char *name = entry->d_name;
103
104 if ((strcmp(name, "..") == 0)
105 || (strcmp(name, ".") == 0)) {
106 continue;
107 }
108
Erik Andersen4f3f7572000-04-28 00:18:56 +0000109 if (len + strlen(name) + 1 > BUFSIZ) {
Mark Whitleyf57c9442000-12-07 19:56:48 +0000110 error_msg(name_too_long);
Erik Andersen42387e42000-02-21 17:27:17 +0000111 du_depth--;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000112 return 0;
113 }
114 sprintf(newfile, "%s/%s", filename, name);
115
116 sum += du(newfile);
117 }
118 closedir(dir);
119 print(sum, filename);
120 }
Erik Andersen27fdd082000-02-19 18:16:49 +0000121 else if (statbuf.st_nlink > 1 && !count_hardlinks) {
122 /* Add files with hard links only once */
Erik Andersen029011b2000-03-04 21:19:32 +0000123 if (is_in_ino_dev_hashtable(&statbuf, NULL)) {
Erik Andersen42387e42000-02-21 17:27:17 +0000124 sum = 0L;
125 if (du_depth == 1)
126 print(sum, filename);
127 }
128 else {
Erik Andersen029011b2000-03-04 21:19:32 +0000129 add_to_ino_dev_hashtable(&statbuf, NULL);
Erik Andersen42387e42000-02-21 17:27:17 +0000130 }
Erik Andersen27fdd082000-02-19 18:16:49 +0000131 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000132 du_depth--;
133 return sum;
John Beppu0f5e1ab1999-12-09 18:23:54 +0000134}
135
Erik Andersene49d5ec2000-02-08 19:58:47 +0000136int du_main(int argc, char **argv)
137{
Matt Kraai92ed8a32000-12-06 15:55:23 +0000138 int status = EXIT_SUCCESS;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000139 int i;
Pavel Roskin47d49262000-07-17 16:17:19 +0000140 int c;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000141
142 /* default behaviour */
143 print = print_normal;
144
145 /* parse argv[] */
Richard June6d0921c2001-01-22 22:35:38 +0000146 while ((c = getopt(argc, argv, "sl"
147#ifdef BB_FEATURE_HUMAN_READABLE
148"hm"
149#endif
150"k")) != EOF) {
Eric Andersen17ad45a2000-07-14 18:38:26 +0000151 switch (c) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000152 case 's':
Eric Andersen17ad45a2000-07-14 18:38:26 +0000153 print = print_summary;
154 break;
Erik Andersen27fdd082000-02-19 18:16:49 +0000155 case 'l':
Eric Andersen17ad45a2000-07-14 18:38:26 +0000156 count_hardlinks = 1;
157 break;
Richard June6d0921c2001-01-22 22:35:38 +0000158#ifdef BB_FEATURE_HUMAN_READABLE
159 case 'h': du_disp_hr = 0; break;
160 case 'm': du_disp_hr = MEGABYTE; break;
161 case 'k': du_disp_hr = KILOBYTE; break;
162#else
163 case 'k': break;
164#endif
Erik Andersene49d5ec2000-02-08 19:58:47 +0000165 default:
Eric Andersen17ad45a2000-07-14 18:38:26 +0000166 usage(du_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 Andersen8ec10a92001-01-27 09:33:39 +0000190/* $Id: du.c,v 1.36 2001/01/27 09:33:38 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*/