blob: fd19855e1bc9b37882833b15d08853fc6f422ae6 [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
Eric Andersen8b113f92001-06-01 21:47:15 +000074#define HASH_SIZE 311 /* Should be prime */
75#define hash_inode(i) ((i) % HASH_SIZE)
76
77typedef struct ino_dev_hash_bucket_struct {
78 struct ino_dev_hash_bucket_struct *next;
79 ino_t ino;
80 dev_t dev;
81 char name[1];
82} ino_dev_hashtable_bucket_t;
83
84static ino_dev_hashtable_bucket_t *ino_dev_hashtable[HASH_SIZE];
85
86/*
87 * Return 1 if statbuf->st_ino && statbuf->st_dev are recorded in
88 * `ino_dev_hashtable', else return 0
89 *
90 * If NAME is a non-NULL pointer to a character pointer, and there is
91 * a match, then set *NAME to the value of the name slot in that
92 * bucket.
93 */
94static int is_in_ino_dev_hashtable(const struct stat *statbuf, char **name)
95{
96 ino_dev_hashtable_bucket_t *bucket;
97
98 bucket = ino_dev_hashtable[hash_inode(statbuf->st_ino)];
99 while (bucket != NULL) {
100 if ((bucket->ino == statbuf->st_ino) &&
101 (bucket->dev == statbuf->st_dev))
102 {
103 if (name) *name = bucket->name;
104 return 1;
105 }
106 bucket = bucket->next;
107 }
108 return 0;
109}
110
111/* Add statbuf to statbuf hash table */
112static void add_to_ino_dev_hashtable(const struct stat *statbuf, const char *name)
113{
114 int i;
115 size_t s;
116 ino_dev_hashtable_bucket_t *bucket;
117
118 i = hash_inode(statbuf->st_ino);
119 s = name ? strlen(name) : 0;
120 bucket = xmalloc(sizeof(ino_dev_hashtable_bucket_t) + s);
121 bucket->ino = statbuf->st_ino;
122 bucket->dev = statbuf->st_dev;
123 if (name)
124 strcpy(bucket->name, name);
125 else
126 bucket->name[0] = '\0';
127 bucket->next = ino_dev_hashtable[i];
128 ino_dev_hashtable[i] = bucket;
129}
130
131/* Clear statbuf hash table */
132static void reset_ino_dev_hashtable(void)
133{
134 int i;
135 ino_dev_hashtable_bucket_t *bucket;
136
137 for (i = 0; i < HASH_SIZE; i++) {
138 while (ino_dev_hashtable[i] != NULL) {
139 bucket = ino_dev_hashtable[i]->next;
140 free(ino_dev_hashtable[i]);
141 ino_dev_hashtable[i] = bucket;
142 }
143 }
144}
145
John Beppu0f5e1ab1999-12-09 18:23:54 +0000146/* tiny recursive du */
Erik Andersene49d5ec2000-02-08 19:58:47 +0000147static long du(char *filename)
John Beppu0f5e1ab1999-12-09 18:23:54 +0000148{
Erik Andersene49d5ec2000-02-08 19:58:47 +0000149 struct stat statbuf;
Erik Andersenfac10d72000-02-07 05:29:42 +0000150 long sum;
John Beppu14c82b61999-12-10 06:15:27 +0000151
Erik Andersene49d5ec2000-02-08 19:58:47 +0000152 if ((lstat(filename, &statbuf)) != 0) {
Eric Andersene5dfced2001-04-09 22:48:12 +0000153 perror_msg("%s", filename);
154 return 0;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000155 }
156
157 du_depth++;
John Beppu08c965a2000-02-13 04:10:57 +0000158 sum = (statbuf.st_blocks >> 1);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000159
Erik Andersen27fdd082000-02-19 18:16:49 +0000160 /* Don't add in stuff pointed to by symbolic links */
Erik Andersen9ffdaa62000-02-11 21:55:04 +0000161 if (S_ISLNK(statbuf.st_mode)) {
Erik Andersen42387e42000-02-21 17:27:17 +0000162 sum = 0L;
163 if (du_depth == 1)
164 print(sum, filename);
Erik Andersen9ffdaa62000-02-11 21:55:04 +0000165 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000166 if (S_ISDIR(statbuf.st_mode)) {
167 DIR *dir;
168 struct dirent *entry;
Eric Andersen04b03542001-05-07 22:49:43 +0000169 char *newfile;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000170
171 dir = opendir(filename);
172 if (!dir) {
Erik Andersen42387e42000-02-21 17:27:17 +0000173 du_depth--;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000174 return 0;
175 }
Erik Andersen42387e42000-02-21 17:27:17 +0000176
Eric Andersen04b03542001-05-07 22:49:43 +0000177 newfile = last_char_is(filename, '/');
178 if (newfile)
179 *newfile = '\0';
Erik Andersen42387e42000-02-21 17:27:17 +0000180
Erik Andersene49d5ec2000-02-08 19:58:47 +0000181 while ((entry = readdir(dir))) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000182 char *name = entry->d_name;
183
184 if ((strcmp(name, "..") == 0)
185 || (strcmp(name, ".") == 0)) {
186 continue;
187 }
Eric Andersene5dfced2001-04-09 22:48:12 +0000188 newfile = concat_path_file(filename, name);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000189 sum += du(newfile);
Eric Andersene5dfced2001-04-09 22:48:12 +0000190 free(newfile);
Erik Andersene49d5ec2000-02-08 19:58:47 +0000191 }
192 closedir(dir);
193 print(sum, filename);
194 }
Erik Andersen27fdd082000-02-19 18:16:49 +0000195 else if (statbuf.st_nlink > 1 && !count_hardlinks) {
196 /* Add files with hard links only once */
Erik Andersen029011b2000-03-04 21:19:32 +0000197 if (is_in_ino_dev_hashtable(&statbuf, NULL)) {
Erik Andersen42387e42000-02-21 17:27:17 +0000198 sum = 0L;
199 if (du_depth == 1)
200 print(sum, filename);
201 }
202 else {
Erik Andersen029011b2000-03-04 21:19:32 +0000203 add_to_ino_dev_hashtable(&statbuf, NULL);
Erik Andersen42387e42000-02-21 17:27:17 +0000204 }
Erik Andersen27fdd082000-02-19 18:16:49 +0000205 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000206 du_depth--;
207 return sum;
John Beppu0f5e1ab1999-12-09 18:23:54 +0000208}
209
Erik Andersene49d5ec2000-02-08 19:58:47 +0000210int du_main(int argc, char **argv)
211{
Matt Kraai92ed8a32000-12-06 15:55:23 +0000212 int status = EXIT_SUCCESS;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000213 int i;
Pavel Roskin47d49262000-07-17 16:17:19 +0000214 int c;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000215
216 /* default behaviour */
217 print = print_normal;
218
219 /* parse argv[] */
Richard June6d0921c2001-01-22 22:35:38 +0000220 while ((c = getopt(argc, argv, "sl"
221#ifdef BB_FEATURE_HUMAN_READABLE
222"hm"
223#endif
224"k")) != EOF) {
Eric Andersen17ad45a2000-07-14 18:38:26 +0000225 switch (c) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000226 case 's':
Eric Andersen17ad45a2000-07-14 18:38:26 +0000227 print = print_summary;
228 break;
Erik Andersen27fdd082000-02-19 18:16:49 +0000229 case 'l':
Eric Andersen17ad45a2000-07-14 18:38:26 +0000230 count_hardlinks = 1;
231 break;
Richard June6d0921c2001-01-22 22:35:38 +0000232#ifdef BB_FEATURE_HUMAN_READABLE
Eric Andersenec9fad92001-03-07 06:04:08 +0000233 case 'h': disp_hr = 0; break;
234 case 'm': disp_hr = MEGABYTE; break;
Richard June6d0921c2001-01-22 22:35:38 +0000235#endif
Eric Andersen8b728a22001-03-06 23:14:43 +0000236 case 'k': break;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000237 default:
Eric Andersen67991cf2001-02-14 21:23:06 +0000238 show_usage();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000239 }
Erik Andersene49d5ec2000-02-08 19:58:47 +0000240 }
241
242 /* go through remaining args (if any) */
Eric Andersen17ad45a2000-07-14 18:38:26 +0000243 if (optind >= argc) {
Matt Kraai92ed8a32000-12-06 15:55:23 +0000244 if (du(".") == 0)
245 status = EXIT_FAILURE;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000246 } else {
247 long sum;
248
Eric Andersen17ad45a2000-07-14 18:38:26 +0000249 for (i=optind; i < argc; i++) {
Matt Kraaie8849702000-12-06 15:56:31 +0000250 if ((sum = du(argv[i])) == 0)
Matt Kraai92ed8a32000-12-06 15:55:23 +0000251 status = EXIT_FAILURE;
Richard June6d0921c2001-01-22 22:35:38 +0000252 if(is_directory(argv[i], FALSE, NULL)==FALSE) {
Erik Andersene49d5ec2000-02-08 19:58:47 +0000253 print_normal(sum, argv[i]);
254 }
Erik Andersen029011b2000-03-04 21:19:32 +0000255 reset_ino_dev_hashtable();
Erik Andersene49d5ec2000-02-08 19:58:47 +0000256 }
257 }
258
Matt Kraai92ed8a32000-12-06 15:55:23 +0000259 return status;
Erik Andersene49d5ec2000-02-08 19:58:47 +0000260}
261
Eric Andersen8b113f92001-06-01 21:47:15 +0000262/* $Id: du.c,v 1.48 2001/06/01 21:47:15 andersen Exp $ */
Erik Andersen029011b2000-03-04 21:19:32 +0000263/*
264Local Variables:
265c-file-style: "linux"
266c-basic-offset: 4
267tab-width: 4
268End:
269*/