Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 2 | /* |
| 3 | * Mini du implementation for busybox |
| 4 | * |
| 5 | * |
Eric Andersen | 8ec10a9 | 2001-01-27 09:33:39 +0000 | [diff] [blame] | 6 | * Copyright (C) 1999,2000,2001 by Lineo, inc. |
Eric Andersen | 70e2f0b | 1999-12-10 06:45:42 +0000 | [diff] [blame] | 7 | * Written by John Beppu <beppu@lineo.com> |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 8 | * |
| 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 Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 25 | #include <sys/types.h> |
| 26 | #include <fcntl.h> |
| 27 | #include <dirent.h> |
| 28 | #include <stdio.h> |
Eric Andersen | ed3ef50 | 2001-01-27 08:24:39 +0000 | [diff] [blame] | 29 | #include <stdlib.h> |
| 30 | #include <getopt.h> |
Eric Andersen | eba8ed7 | 2001-03-09 14:36:42 +0000 | [diff] [blame] | 31 | #include <string.h> |
John Beppu | 9835541 | 1999-12-10 07:40:08 +0000 | [diff] [blame] | 32 | #include <errno.h> |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 33 | #include "busybox.h" |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 34 | |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 35 | |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 36 | #ifdef BB_FEATURE_HUMAN_READABLE |
Eric Andersen | ec9fad9 | 2001-03-07 06:04:08 +0000 | [diff] [blame] | 37 | static unsigned long disp_hr = KILOBYTE; |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 38 | #endif |
| 39 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 40 | typedef void (Display) (long, char *); |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 41 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 42 | static int du_depth = 0; |
Erik Andersen | 27fdd08 | 2000-02-19 18:16:49 +0000 | [diff] [blame] | 43 | static int count_hardlinks = 0; |
John Beppu | e1618e4 | 1999-12-15 18:52:17 +0000 | [diff] [blame] | 44 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 45 | static Display *print; |
| 46 | |
| 47 | static void print_normal(long size, char *filename) |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 48 | { |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 49 | #ifdef BB_FEATURE_HUMAN_READABLE |
Eric Andersen | f429bac | 2001-06-13 08:02:45 +0000 | [diff] [blame] | 50 | printf("%s\t%s\n", make_human_readable_str(size<<10, 1, disp_hr), filename); |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 51 | #else |
Matt Kraai | 12f417e | 2001-01-18 02:57:08 +0000 | [diff] [blame] | 52 | printf("%ld\t%s\n", size, filename); |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 53 | #endif |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 54 | } |
| 55 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 56 | static void print_summary(long size, char *filename) |
John Beppu | e1618e4 | 1999-12-15 18:52:17 +0000 | [diff] [blame] | 57 | { |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 58 | if (du_depth == 1) { |
| 59 | print_normal(size, filename); |
| 60 | } |
John Beppu | e1618e4 | 1999-12-15 18:52:17 +0000 | [diff] [blame] | 61 | } |
| 62 | |
Eric Andersen | 8b113f9 | 2001-06-01 21:47:15 +0000 | [diff] [blame] | 63 | #define HASH_SIZE 311 /* Should be prime */ |
| 64 | #define hash_inode(i) ((i) % HASH_SIZE) |
| 65 | |
| 66 | typedef struct ino_dev_hash_bucket_struct { |
| 67 | struct ino_dev_hash_bucket_struct *next; |
| 68 | ino_t ino; |
| 69 | dev_t dev; |
| 70 | char name[1]; |
| 71 | } ino_dev_hashtable_bucket_t; |
| 72 | |
| 73 | static ino_dev_hashtable_bucket_t *ino_dev_hashtable[HASH_SIZE]; |
| 74 | |
| 75 | /* |
| 76 | * Return 1 if statbuf->st_ino && statbuf->st_dev are recorded in |
| 77 | * `ino_dev_hashtable', else return 0 |
| 78 | * |
| 79 | * If NAME is a non-NULL pointer to a character pointer, and there is |
| 80 | * a match, then set *NAME to the value of the name slot in that |
| 81 | * bucket. |
| 82 | */ |
| 83 | static int is_in_ino_dev_hashtable(const struct stat *statbuf, char **name) |
| 84 | { |
| 85 | ino_dev_hashtable_bucket_t *bucket; |
| 86 | |
| 87 | bucket = ino_dev_hashtable[hash_inode(statbuf->st_ino)]; |
| 88 | while (bucket != NULL) { |
| 89 | if ((bucket->ino == statbuf->st_ino) && |
| 90 | (bucket->dev == statbuf->st_dev)) |
| 91 | { |
| 92 | if (name) *name = bucket->name; |
| 93 | return 1; |
| 94 | } |
| 95 | bucket = bucket->next; |
| 96 | } |
| 97 | return 0; |
| 98 | } |
| 99 | |
| 100 | /* Add statbuf to statbuf hash table */ |
| 101 | static void add_to_ino_dev_hashtable(const struct stat *statbuf, const char *name) |
| 102 | { |
| 103 | int i; |
| 104 | size_t s; |
| 105 | ino_dev_hashtable_bucket_t *bucket; |
| 106 | |
| 107 | i = hash_inode(statbuf->st_ino); |
| 108 | s = name ? strlen(name) : 0; |
| 109 | bucket = xmalloc(sizeof(ino_dev_hashtable_bucket_t) + s); |
| 110 | bucket->ino = statbuf->st_ino; |
| 111 | bucket->dev = statbuf->st_dev; |
| 112 | if (name) |
| 113 | strcpy(bucket->name, name); |
| 114 | else |
| 115 | bucket->name[0] = '\0'; |
| 116 | bucket->next = ino_dev_hashtable[i]; |
| 117 | ino_dev_hashtable[i] = bucket; |
| 118 | } |
| 119 | |
| 120 | /* Clear statbuf hash table */ |
| 121 | static void reset_ino_dev_hashtable(void) |
| 122 | { |
| 123 | int i; |
| 124 | ino_dev_hashtable_bucket_t *bucket; |
| 125 | |
| 126 | for (i = 0; i < HASH_SIZE; i++) { |
| 127 | while (ino_dev_hashtable[i] != NULL) { |
| 128 | bucket = ino_dev_hashtable[i]->next; |
| 129 | free(ino_dev_hashtable[i]); |
| 130 | ino_dev_hashtable[i] = bucket; |
| 131 | } |
| 132 | } |
| 133 | } |
| 134 | |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 135 | /* tiny recursive du */ |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 136 | static long du(char *filename) |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 137 | { |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 138 | struct stat statbuf; |
Erik Andersen | fac10d7 | 2000-02-07 05:29:42 +0000 | [diff] [blame] | 139 | long sum; |
John Beppu | 14c82b6 | 1999-12-10 06:15:27 +0000 | [diff] [blame] | 140 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 141 | if ((lstat(filename, &statbuf)) != 0) { |
Eric Andersen | e5dfced | 2001-04-09 22:48:12 +0000 | [diff] [blame] | 142 | perror_msg("%s", filename); |
| 143 | return 0; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 144 | } |
| 145 | |
| 146 | du_depth++; |
John Beppu | 08c965a | 2000-02-13 04:10:57 +0000 | [diff] [blame] | 147 | sum = (statbuf.st_blocks >> 1); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 148 | |
Erik Andersen | 27fdd08 | 2000-02-19 18:16:49 +0000 | [diff] [blame] | 149 | /* Don't add in stuff pointed to by symbolic links */ |
Erik Andersen | 9ffdaa6 | 2000-02-11 21:55:04 +0000 | [diff] [blame] | 150 | if (S_ISLNK(statbuf.st_mode)) { |
Erik Andersen | 42387e4 | 2000-02-21 17:27:17 +0000 | [diff] [blame] | 151 | sum = 0L; |
Eric Andersen | 8fa1bf7 | 2001-06-30 17:54:20 +0000 | [diff] [blame] | 152 | if (du_depth == 1) { |
| 153 | } |
Erik Andersen | 9ffdaa6 | 2000-02-11 21:55:04 +0000 | [diff] [blame] | 154 | } |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 155 | if (S_ISDIR(statbuf.st_mode)) { |
| 156 | DIR *dir; |
| 157 | struct dirent *entry; |
Eric Andersen | 04b0354 | 2001-05-07 22:49:43 +0000 | [diff] [blame] | 158 | char *newfile; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 159 | |
| 160 | dir = opendir(filename); |
| 161 | if (!dir) { |
Erik Andersen | 42387e4 | 2000-02-21 17:27:17 +0000 | [diff] [blame] | 162 | du_depth--; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 163 | return 0; |
| 164 | } |
Erik Andersen | 42387e4 | 2000-02-21 17:27:17 +0000 | [diff] [blame] | 165 | |
Eric Andersen | 04b0354 | 2001-05-07 22:49:43 +0000 | [diff] [blame] | 166 | newfile = last_char_is(filename, '/'); |
| 167 | if (newfile) |
| 168 | *newfile = '\0'; |
Erik Andersen | 42387e4 | 2000-02-21 17:27:17 +0000 | [diff] [blame] | 169 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 170 | while ((entry = readdir(dir))) { |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 171 | char *name = entry->d_name; |
| 172 | |
| 173 | if ((strcmp(name, "..") == 0) |
| 174 | || (strcmp(name, ".") == 0)) { |
| 175 | continue; |
| 176 | } |
Eric Andersen | e5dfced | 2001-04-09 22:48:12 +0000 | [diff] [blame] | 177 | newfile = concat_path_file(filename, name); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 178 | sum += du(newfile); |
Eric Andersen | e5dfced | 2001-04-09 22:48:12 +0000 | [diff] [blame] | 179 | free(newfile); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 180 | } |
| 181 | closedir(dir); |
| 182 | print(sum, filename); |
| 183 | } |
Erik Andersen | 27fdd08 | 2000-02-19 18:16:49 +0000 | [diff] [blame] | 184 | else if (statbuf.st_nlink > 1 && !count_hardlinks) { |
| 185 | /* Add files with hard links only once */ |
Erik Andersen | 029011b | 2000-03-04 21:19:32 +0000 | [diff] [blame] | 186 | if (is_in_ino_dev_hashtable(&statbuf, NULL)) { |
Erik Andersen | 42387e4 | 2000-02-21 17:27:17 +0000 | [diff] [blame] | 187 | sum = 0L; |
| 188 | if (du_depth == 1) |
| 189 | print(sum, filename); |
| 190 | } |
| 191 | else { |
Erik Andersen | 029011b | 2000-03-04 21:19:32 +0000 | [diff] [blame] | 192 | add_to_ino_dev_hashtable(&statbuf, NULL); |
Erik Andersen | 42387e4 | 2000-02-21 17:27:17 +0000 | [diff] [blame] | 193 | } |
Erik Andersen | 27fdd08 | 2000-02-19 18:16:49 +0000 | [diff] [blame] | 194 | } |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 195 | du_depth--; |
| 196 | return sum; |
John Beppu | 0f5e1ab | 1999-12-09 18:23:54 +0000 | [diff] [blame] | 197 | } |
| 198 | |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 199 | int du_main(int argc, char **argv) |
| 200 | { |
Matt Kraai | 92ed8a3 | 2000-12-06 15:55:23 +0000 | [diff] [blame] | 201 | int status = EXIT_SUCCESS; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 202 | int i; |
Pavel Roskin | 47d4926 | 2000-07-17 16:17:19 +0000 | [diff] [blame] | 203 | int c; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 204 | |
| 205 | /* default behaviour */ |
| 206 | print = print_normal; |
| 207 | |
| 208 | /* parse argv[] */ |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 209 | while ((c = getopt(argc, argv, "sl" |
| 210 | #ifdef BB_FEATURE_HUMAN_READABLE |
| 211 | "hm" |
| 212 | #endif |
| 213 | "k")) != EOF) { |
Eric Andersen | 17ad45a | 2000-07-14 18:38:26 +0000 | [diff] [blame] | 214 | switch (c) { |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 215 | case 's': |
Eric Andersen | 17ad45a | 2000-07-14 18:38:26 +0000 | [diff] [blame] | 216 | print = print_summary; |
| 217 | break; |
Erik Andersen | 27fdd08 | 2000-02-19 18:16:49 +0000 | [diff] [blame] | 218 | case 'l': |
Eric Andersen | 17ad45a | 2000-07-14 18:38:26 +0000 | [diff] [blame] | 219 | count_hardlinks = 1; |
| 220 | break; |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 221 | #ifdef BB_FEATURE_HUMAN_READABLE |
Eric Andersen | ec9fad9 | 2001-03-07 06:04:08 +0000 | [diff] [blame] | 222 | case 'h': disp_hr = 0; break; |
| 223 | case 'm': disp_hr = MEGABYTE; break; |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 224 | #endif |
Eric Andersen | 8b728a2 | 2001-03-06 23:14:43 +0000 | [diff] [blame] | 225 | case 'k': break; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 226 | default: |
Eric Andersen | 67991cf | 2001-02-14 21:23:06 +0000 | [diff] [blame] | 227 | show_usage(); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 228 | } |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 229 | } |
| 230 | |
| 231 | /* go through remaining args (if any) */ |
Eric Andersen | 17ad45a | 2000-07-14 18:38:26 +0000 | [diff] [blame] | 232 | if (optind >= argc) { |
Matt Kraai | 92ed8a3 | 2000-12-06 15:55:23 +0000 | [diff] [blame] | 233 | if (du(".") == 0) |
| 234 | status = EXIT_FAILURE; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 235 | } else { |
| 236 | long sum; |
| 237 | |
Eric Andersen | 17ad45a | 2000-07-14 18:38:26 +0000 | [diff] [blame] | 238 | for (i=optind; i < argc; i++) { |
Eric Andersen | 8fa1bf7 | 2001-06-30 17:54:20 +0000 | [diff] [blame] | 239 | sum = du(argv[i]); |
Richard June | 6d0921c | 2001-01-22 22:35:38 +0000 | [diff] [blame] | 240 | if(is_directory(argv[i], FALSE, NULL)==FALSE) { |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 241 | print_normal(sum, argv[i]); |
| 242 | } |
Erik Andersen | 029011b | 2000-03-04 21:19:32 +0000 | [diff] [blame] | 243 | reset_ino_dev_hashtable(); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 244 | } |
| 245 | } |
| 246 | |
Matt Kraai | 92ed8a3 | 2000-12-06 15:55:23 +0000 | [diff] [blame] | 247 | return status; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 248 | } |
| 249 | |
Eric Andersen | 8fa1bf7 | 2001-06-30 17:54:20 +0000 | [diff] [blame] | 250 | /* $Id: du.c,v 1.50 2001/06/30 17:54:20 andersen Exp $ */ |
Erik Andersen | 029011b | 2000-03-04 21:19:32 +0000 | [diff] [blame] | 251 | /* |
| 252 | Local Variables: |
| 253 | c-file-style: "linux" |
| 254 | c-basic-offset: 4 |
| 255 | tab-width: 4 |
| 256 | End: |
| 257 | */ |