blob: b8e296ddd20f3bea2a1f761c2ebc8a38baf1ea76 [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 *
Erik Andersen61677fe2000-04-13 01:18:56 +00006 * Copyright (C) 1999,2000 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
25#include "internal.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>
John Beppu98355411999-12-10 07:40:08 +000034#include <errno.h>
John Beppu0f5e1ab1999-12-09 18:23:54 +000035
Erik Andersene49d5ec2000-02-08 19:58:47 +000036typedef void (Display) (long, char *);
John Beppu0f5e1ab1999-12-09 18:23:54 +000037
Erik Andersen27fdd082000-02-19 18:16:49 +000038static const char du_usage[] =
Erik Andersen59b9e872000-05-10 05:05:45 +000039 "du [OPTION]... [FILE]...\n"
40#ifndef BB_FEATURE_TRIVIAL_HELP
41 "\nSummarizes disk space used for each FILE and/or directory.\n"
Erik Andersen27fdd082000-02-19 18:16:49 +000042 "Disk space is printed in units of 1024 bytes.\n\n"
43 "Options:\n"
44 "\t-l\tcount sizes many times if hard linked\n"
Erik Andersen59b9e872000-05-10 05:05:45 +000045 "\t-s\tdisplay only a total for each argument\n"
46#endif
47 ;
John Beppue1618e41999-12-15 18:52:17 +000048
Erik Andersene49d5ec2000-02-08 19:58:47 +000049static int du_depth = 0;
Erik Andersen27fdd082000-02-19 18:16:49 +000050static int count_hardlinks = 0;
John Beppue1618e41999-12-15 18:52:17 +000051
Erik Andersene49d5ec2000-02-08 19:58:47 +000052static Display *print;
53
54static void print_normal(long size, char *filename)
John Beppu0f5e1ab1999-12-09 18:23:54 +000055{
Erik Andersen27fdd082000-02-19 18:16:49 +000056 fprintf(stdout, "%ld\t%s\n", size, filename);
John Beppu0f5e1ab1999-12-09 18:23:54 +000057}
58
Erik Andersene49d5ec2000-02-08 19:58:47 +000059static void print_summary(long size, char *filename)
John Beppue1618e41999-12-15 18:52:17 +000060{
Erik Andersene49d5ec2000-02-08 19:58:47 +000061 if (du_depth == 1) {
62 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) {
Erik Andersen42387e42000-02-21 17:27:17 +000074 printf("du: %s: %s\n", filename, strerror(errno));
Erik Andersene49d5ec2000-02-08 19:58:47 +000075 return 0;
76 }
77
78 du_depth++;
John Beppu08c965a2000-02-13 04:10:57 +000079 sum = (statbuf.st_blocks >> 1);
Erik Andersene49d5ec2000-02-08 19:58:47 +000080
Erik Andersen27fdd082000-02-19 18:16:49 +000081 /* Don't add in stuff pointed to by symbolic links */
Erik Andersen9ffdaa62000-02-11 21:55:04 +000082 if (S_ISLNK(statbuf.st_mode)) {
Erik Andersen42387e42000-02-21 17:27:17 +000083 sum = 0L;
84 if (du_depth == 1)
85 print(sum, filename);
Erik Andersen9ffdaa62000-02-11 21:55:04 +000086 }
Erik Andersene49d5ec2000-02-08 19:58:47 +000087 if (S_ISDIR(statbuf.st_mode)) {
88 DIR *dir;
89 struct dirent *entry;
90
91 dir = opendir(filename);
92 if (!dir) {
Erik Andersen42387e42000-02-21 17:27:17 +000093 du_depth--;
Erik Andersene49d5ec2000-02-08 19:58:47 +000094 return 0;
95 }
Erik Andersen42387e42000-02-21 17:27:17 +000096
97 len = strlen(filename);
98 if (filename[len - 1] == '/')
99 filename[--len] = '\0';
100
Erik Andersene49d5ec2000-02-08 19:58:47 +0000101 while ((entry = readdir(dir))) {
Erik Andersen4f3f7572000-04-28 00:18:56 +0000102 char newfile[BUFSIZ + 1];
Erik Andersene49d5ec2000-02-08 19:58:47 +0000103 char *name = entry->d_name;
104
105 if ((strcmp(name, "..") == 0)
106 || (strcmp(name, ".") == 0)) {
107 continue;
108 }
109
Erik Andersen4f3f7572000-04-28 00:18:56 +0000110 if (len + strlen(name) + 1 > BUFSIZ) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000111 fprintf(stderr, name_too_long, "du");
Erik Andersen42387e42000-02-21 17:27:17 +0000112 du_depth--;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000113 return 0;
114 }
115 sprintf(newfile, "%s/%s", filename, name);
116
117 sum += du(newfile);
118 }
119 closedir(dir);
120 print(sum, filename);
121 }
Erik Andersen27fdd082000-02-19 18:16:49 +0000122 else if (statbuf.st_nlink > 1 && !count_hardlinks) {
123 /* Add files with hard links only once */
Erik Andersen029011b2000-03-04 21:19:32 +0000124 if (is_in_ino_dev_hashtable(&statbuf, NULL)) {
Erik Andersen42387e42000-02-21 17:27:17 +0000125 sum = 0L;
126 if (du_depth == 1)
127 print(sum, filename);
128 }
129 else {
Erik Andersen029011b2000-03-04 21:19:32 +0000130 add_to_ino_dev_hashtable(&statbuf, NULL);
Erik Andersen42387e42000-02-21 17:27:17 +0000131 }
Erik Andersen27fdd082000-02-19 18:16:49 +0000132 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000133 du_depth--;
134 return sum;
John Beppu0f5e1ab1999-12-09 18:23:54 +0000135}
136
Erik Andersene49d5ec2000-02-08 19:58:47 +0000137int du_main(int argc, char **argv)
138{
139 int i;
140 char opt;
141
142 /* default behaviour */
143 print = print_normal;
144
145 /* parse argv[] */
146 for (i = 1; i < argc; i++) {
147 if (argv[i][0] == '-') {
148 opt = argv[i][1];
149 switch (opt) {
150 case 's':
151 print = print_summary;
152 break;
Erik Andersen27fdd082000-02-19 18:16:49 +0000153 case 'l':
154 count_hardlinks = 1;
155 break;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000156 case 'h':
Erik Andersen27fdd082000-02-19 18:16:49 +0000157 case '-':
Erik Andersene49d5ec2000-02-08 19:58:47 +0000158 usage(du_usage);
159 break;
160 default:
161 fprintf(stderr, "du: invalid option -- %c\n", opt);
162 usage(du_usage);
163 }
164 } else {
165 break;
166 }
167 }
168
169 /* go through remaining args (if any) */
170 if (i >= argc) {
171 du(".");
172 } else {
173 long sum;
174
175 for (; i < argc; i++) {
176 sum = du(argv[i]);
Erik Andersen42387e42000-02-21 17:27:17 +0000177 if (sum && isDirectory(argv[i], FALSE, NULL)) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000178 print_normal(sum, argv[i]);
179 }
Erik Andersen029011b2000-03-04 21:19:32 +0000180 reset_ino_dev_hashtable();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000181 }
182 }
183
Eric Andersenb6106152000-06-19 17:25:40 +0000184 return(0);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000185}
186
Eric Andersenb6106152000-06-19 17:25:40 +0000187/* $Id: du.c,v 1.20 2000/06/19 17:25:39 andersen Exp $ */
Erik Andersen029011b2000-03-04 21:19:32 +0000188/*
189Local Variables:
190c-file-style: "linux"
191c-basic-offset: 4
192tab-width: 4
193End:
194*/