du.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini du implementation for busybox
   4  *
   5  *
   6  * Copyright (C) 1999,2000 by Lineo, inc.
   7  * Written by John Beppu <beppu@lineo.com>
   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
  25 #include "internal.h"
  26 #define BB_DECLARE_EXTERN
  27 #define bb_need_name_too_long
  28 #include "messages.c"
  29
  30 #include <sys/types.h>
  31 #include <fcntl.h>
  32 #include <dirent.h>
  33 #include <stdio.h>
  34 #include <errno.h>
  35
  36 typedef void (Display) (long, char *);
  37
  38 static const char du_usage[] =
  39         "du [OPTION]... [FILE]...\n"
  40 #ifndef BB_FEATURE_TRIVIAL_HELP
  41         "\nSummarizes disk space used for each FILE and/or directory.\n"
  42         "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"
  45         "\t-s\tdisplay only a total for each argument\n"
  46 #endif
  47         ;
  48
  49 static int du_depth = 0;
  50 static int count_hardlinks = 0;
  51
  52 static Display *print;
  53
  54 static void print_normal(long size, char *filename)
  55 {
  56         fprintf(stdout, "%ld\t%s\n", size, filename);
  57 }
  58
  59 static void print_summary(long size, char *filename)
  60 {
  61         if (du_depth == 1) {
  62                 print_normal(size, filename);
  63         }
  64 }
  65
  66 /* tiny recursive du */
  67 static long du(char *filename)
  68 {
  69         struct stat statbuf;
  70         long sum;
  71         int len;
  72
  73         if ((lstat(filename, &statbuf)) != 0) {
  74                 printf("du: %s: %s\n", filename, strerror(errno));
  75                 return 0;
  76         }
  77
  78         du_depth++;
  79         sum = (statbuf.st_blocks >> 1);
  80
  81         /* Don't add in stuff pointed to by symbolic links */
  82         if (S_ISLNK(statbuf.st_mode)) {
  83                 sum = 0L;
  84                 if (du_depth == 1)
  85                         print(sum, filename);
  86         }
  87         if (S_ISDIR(statbuf.st_mode)) {
  88                 DIR *dir;
  89                 struct dirent *entry;
  90
  91                 dir = opendir(filename);
  92                 if (!dir) {
  93                         du_depth--;
  94                         return 0;
  95                 }
  96
  97                 len = strlen(filename);
  98                 if (filename[len - 1] == '/')
  99                         filename[--len] = '\0';
 100
 101                 while ((entry = readdir(dir))) {
 102                         char newfile[BUFSIZ + 1];
 103                         char *name = entry->d_name;
 104
 105                         if ((strcmp(name, "..") == 0)
 106                                 || (strcmp(name, ".") == 0)) {
 107                                 continue;
 108                         }
 109
 110                         if (len + strlen(name) + 1 > BUFSIZ) {
 111                                 fprintf(stderr, name_too_long, "du");
 112                                 du_depth--;
 113                                 return 0;
 114                         }
 115                         sprintf(newfile, "%s/%s", filename, name);
 116
 117                         sum += du(newfile);
 118                 }
 119                 closedir(dir);
 120                 print(sum, filename);
 121         }
 122         else if (statbuf.st_nlink > 1 && !count_hardlinks) {
 123                 /* Add files with hard links only once */
 124                 if (is_in_ino_dev_hashtable(&statbuf, NULL)) {
 125                         sum = 0L;
 126                         if (du_depth == 1)
 127                                 print(sum, filename);
 128                 }
 129                 else {
 130                         add_to_ino_dev_hashtable(&statbuf, NULL);
 131                 }
 132         }
 133         du_depth--;
 134         return sum;
 135 }
 136
 137 int 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;
 153                         case 'l':
 154                                 count_hardlinks = 1;
 155                                 break;
 156                         case 'h':
 157                         case '-':
 158                                 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]);
 177                         if (sum && isDirectory(argv[i], FALSE, NULL)) {
 178                                 print_normal(sum, argv[i]);
 179                         }
 180                         reset_ino_dev_hashtable();
 181                 }
 182         }
 183
 184         return(0);
 185 }
 186
 187 /* $Id: du.c,v 1.20 2000/06/19 17:25:39 andersen Exp $ */
 188 /*
 189 Local Variables:
 190 c-file-style: "linux"
 191 c-basic-offset: 4
 192 tab-width: 4
 193 End:
 194 */