src/du.c

   1 /* du -- summarize disk usage
   2    Copyright (C) 1988-1991, 1995-2010 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation, either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  16
  17 /* Differences from the Unix du:
  18    * Doesn't simply ignore the names of regular files given as arguments
  19      when -a is given.
  20
  21    By tege@sics.se, Torbjorn Granlund,
  22    and djm@ai.mit.edu, David MacKenzie.
  23    Variable blocks added by lm@sgi.com and eggert@twinsun.com.
  24    Rewritten to use nftw, then to use fts by Jim Meyering.  */
  25
  26 #include <config.h>
  27 #include <getopt.h>
  28 #include <sys/types.h>
  29 #include <assert.h>
  30 #include "system.h"
  31 #include "argmatch.h"
  32 #include "argv-iter.h"
  33 #include "di-set.h"
  34 #include "error.h"
  35 #include "exclude.h"
  36 #include "fprintftime.h"
  37 #include "human.h"
  38 #include "quote.h"
  39 #include "quotearg.h"
  40 #include "same.h"
  41 #include "stat-time.h"
  42 #include "stdio--.h"
  43 #include "xfts.h"
  44 #include "xstrtol.h"
  45
  46 extern bool fts_debug;
  47
  48 /* The official name of this program (e.g., no `g' prefix).  */
  49 #define PROGRAM_NAME "du"
  50
  51 #define AUTHORS \
  52   proper_name_utf8 ("Torbjorn Granlund", "Torbj\303\266rn Granlund"), \
  53   proper_name ("David MacKenzie"), \
  54   proper_name ("Paul Eggert"), \
  55   proper_name ("Jim Meyering")
  56
  57 #if DU_DEBUG
  58 # define FTS_CROSS_CHECK(Fts) fts_cross_check (Fts)
  59 #else
  60 # define FTS_CROSS_CHECK(Fts)
  61 #endif
  62
  63 /* A set of dev/ino pairs.  */
  64 static struct di_set *di_set;
  65
  66 /* Define a class for collecting directory information. */
  67
  68 struct duinfo
  69 {
  70   /* Size of files in directory.  */
  71   uintmax_t size;
  72
  73   /* Latest time stamp found.  If tmax.tv_sec == TYPE_MINIMUM (time_t)
  74      && tmax.tv_nsec < 0, no time stamp has been found.  */
  75   struct timespec tmax;
  76 };
  77
  78 /* Initialize directory data.  */
  79 static inline void
  80 duinfo_init (struct duinfo *a)
  81 {
  82   a->size = 0;
  83   a->tmax.tv_sec = TYPE_MINIMUM (time_t);
  84   a->tmax.tv_nsec = -1;
  85 }
  86
  87 /* Set directory data.  */
  88 static inline void
  89 duinfo_set (struct duinfo *a, uintmax_t size, struct timespec tmax)
  90 {
  91   a->size = size;
  92   a->tmax = tmax;
  93 }
  94
  95 /* Accumulate directory data.  */
  96 static inline void
  97 duinfo_add (struct duinfo *a, struct duinfo const *b)
  98 {
  99   a->size += b->size;
 100   if (timespec_cmp (a->tmax, b->tmax) < 0)
 101     a->tmax = b->tmax;
 102 }
 103
 104 /* A structure for per-directory level information.  */
 105 struct dulevel
 106 {
 107   /* Entries in this directory.  */
 108   struct duinfo ent;
 109
 110   /* Total for subdirectories.  */
 111   struct duinfo subdir;
 112 };
 113
 114 /* If true, display counts for all files, not just directories.  */
 115 static bool opt_all = false;
 116
 117 /* If true, rather than using the disk usage of each file,
 118    use the apparent size (a la stat.st_size).  */
 119 static bool apparent_size = false;
 120
 121 /* If true, count each hard link of files with multiple links.  */
 122 static bool opt_count_all = false;
 123
 124 /* If true, hash all files to look for hard links.  */
 125 static bool hash_all;
 126
 127 /* If true, output the NUL byte instead of a newline at the end of each line. */
 128 static bool opt_nul_terminate_output = false;
 129
 130 /* If true, print a grand total at the end.  */
 131 static bool print_grand_total = false;
 132
 133 /* If nonzero, do not add sizes of subdirectories.  */
 134 static bool opt_separate_dirs = false;
 135
 136 /* Show the total for each directory (and file if --all) that is at
 137    most MAX_DEPTH levels down from the root of the hierarchy.  The root
 138    is at level 0, so `du --max-depth=0' is equivalent to `du -s'.  */
 139 static size_t max_depth = SIZE_MAX;
 140
 141 /* Human-readable options for output.  */
 142 static int human_output_opts;
 143
 144 /* If true, print most recently modified date, using the specified format.  */
 145 static bool opt_time = false;
 146
 147 /* Type of time to display. controlled by --time.  */
 148
 149 enum time_type
 150   {
 151     time_mtime,                 /* default */
 152     time_ctime,
 153     time_atime
 154   };
 155
 156 static enum time_type time_type = time_mtime;
 157
 158 /* User specified date / time style */
 159 static char const *time_style = NULL;
 160
 161 /* Format used to display date / time. Controlled by --time-style */
 162 static char const *time_format = NULL;
 163
 164 /* The units to use when printing sizes.  */
 165 static uintmax_t output_block_size;
 166
 167 /* File name patterns to exclude.  */
 168 static struct exclude *exclude;
 169
 170 /* Grand total size of all args, in bytes. Also latest modified date. */
 171 static struct duinfo tot_dui;
 172
 173 #define IS_DIR_TYPE(Type)       \
 174   ((Type) == FTS_DP             \
 175    || (Type) == FTS_DNR)
 176
 177 /* For long options that have no equivalent short option, use a
 178    non-character as a pseudo short option, starting with CHAR_MAX + 1.  */
 179 enum
 180 {
 181   APPARENT_SIZE_OPTION = CHAR_MAX + 1,
 182   EXCLUDE_OPTION,
 183   FILES0_FROM_OPTION,
 184   HUMAN_SI_OPTION,
 185   FTS_DEBUG,
 186   TIME_OPTION,
 187   TIME_STYLE_OPTION
 188 };
 189
 190 static struct option const long_options[] =
 191 {
 192   {"all", no_argument, NULL, 'a'},
 193   {"apparent-size", no_argument, NULL, APPARENT_SIZE_OPTION},
 194   {"block-size", required_argument, NULL, 'B'},
 195   {"bytes", no_argument, NULL, 'b'},
 196   {"count-links", no_argument, NULL, 'l'},
 197   /* {"-debug", no_argument, NULL, FTS_DEBUG}, */
 198   {"dereference", no_argument, NULL, 'L'},
 199   {"dereference-args", no_argument, NULL, 'D'},
 200   {"exclude", required_argument, NULL, EXCLUDE_OPTION},
 201   {"exclude-from", required_argument, NULL, 'X'},
 202   {"files0-from", required_argument, NULL, FILES0_FROM_OPTION},
 203   {"human-readable", no_argument, NULL, 'h'},
 204   {"si", no_argument, NULL, HUMAN_SI_OPTION},
 205   {"max-depth", required_argument, NULL, 'd'},
 206   {"null", no_argument, NULL, '0'},
 207   {"no-dereference", no_argument, NULL, 'P'},
 208   {"one-file-system", no_argument, NULL, 'x'},
 209   {"separate-dirs", no_argument, NULL, 'S'},
 210   {"summarize", no_argument, NULL, 's'},
 211   {"total", no_argument, NULL, 'c'},
 212   {"time", optional_argument, NULL, TIME_OPTION},
 213   {"time-style", required_argument, NULL, TIME_STYLE_OPTION},
 214   {GETOPT_HELP_OPTION_DECL},
 215   {GETOPT_VERSION_OPTION_DECL},
 216   {NULL, 0, NULL, 0}
 217 };
 218
 219 static char const *const time_args[] =
 220 {
 221   "atime", "access", "use", "ctime", "status", NULL
 222 };
 223 static enum time_type const time_types[] =
 224 {
 225   time_atime, time_atime, time_atime, time_ctime, time_ctime
 226 };
 227 ARGMATCH_VERIFY (time_args, time_types);
 228
 229 /* `full-iso' uses full ISO-style dates and times.  `long-iso' uses longer
 230    ISO-style time stamps, though shorter than `full-iso'.  `iso' uses shorter
 231    ISO-style time stamps.  */
 232 enum time_style
 233   {
 234     full_iso_time_style,       /* --time-style=full-iso */
 235     long_iso_time_style,       /* --time-style=long-iso */
 236     iso_time_style             /* --time-style=iso */
 237   };
 238
 239 static char const *const time_style_args[] =
 240 {
 241   "full-iso", "long-iso", "iso", NULL
 242 };
 243 static enum time_style const time_style_types[] =
 244 {
 245   full_iso_time_style, long_iso_time_style, iso_time_style
 246 };
 247 ARGMATCH_VERIFY (time_style_args, time_style_types);
 248
 249 void
 250 usage (int status)
 251 {
 252   if (status != EXIT_SUCCESS)
 253     fprintf (stderr, _("Try `%s --help' for more information.\n"),
 254              program_name);
 255   else
 256     {
 257       printf (_("\
 258 Usage: %s [OPTION]... [FILE]...\n\
 259   or:  %s [OPTION]... --files0-from=F\n\
 260 "), program_name, program_name);
 261       fputs (_("\
 262 Summarize disk usage of each FILE, recursively for directories.\n\
 263 \n\
 264 "), stdout);
 265       fputs (_("\
 266 Mandatory arguments to long options are mandatory for short options too.\n\
 267 "), stdout);
 268       fputs (_("\
 269   -a, --all             write counts for all files, not just directories\n\
 270       --apparent-size   print apparent sizes, rather than disk usage; although\n\
 271                           the apparent size is usually smaller, it may be\n\
 272                           larger due to holes in (`sparse') files, internal\n\
 273                           fragmentation, indirect blocks, and the like\n\
 274 "), stdout);
 275       fputs (_("\
 276   -B, --block-size=SIZE  scale sizes by SIZE before printing them.  E.g.,\n\
 277                            `-BM' prints sizes in units of 1,048,576 bytes.\n\
 278                            See SIZE format below.\n\
 279   -b, --bytes           equivalent to `--apparent-size --block-size=1'\n\
 280   -c, --total           produce a grand total\n\
 281   -D, --dereference-args  dereference only symlinks that are listed on the\n\
 282                           command line\n\
 283 "), stdout);
 284       fputs (_("\
 285       --files0-from=F   summarize disk usage of the NUL-terminated file\n\
 286                           names specified in file F;\n\
 287                           If F is - then read names from standard input\n\
 288   -H                    equivalent to --dereference-args (-D)\n\
 289   -h, --human-readable  print sizes in human readable format (e.g., 1K 234M 2G)\n\
 290       --si              like -h, but use powers of 1000 not 1024\n\
 291 "), stdout);
 292       fputs (_("\
 293   -k                    like --block-size=1K\n\
 294   -l, --count-links     count sizes many times if hard linked\n\
 295   -m                    like --block-size=1M\n\
 296 "), stdout);
 297       fputs (_("\
 298   -L, --dereference     dereference all symbolic links\n\
 299   -P, --no-dereference  don't follow any symbolic links (this is the default)\n\
 300   -0, --null            end each output line with 0 byte rather than newline\n\
 301   -S, --separate-dirs   do not include size of subdirectories\n\
 302   -s, --summarize       display only a total for each argument\n\
 303 "), stdout);
 304       fputs (_("\
 305   -x, --one-file-system    skip directories on different file systems\n\
 306   -X, --exclude-from=FILE  exclude files that match any pattern in FILE\n\
 307       --exclude=PATTERN    exclude files that match PATTERN\n\
 308   -d, --max-depth=N     print the total for a directory (or file, with --all)\n\
 309                           only if it is N or fewer levels below the command\n\
 310                           line argument;  --max-depth=0 is the same as\n\
 311                           --summarize\n\
 312 "), stdout);
 313       fputs (_("\
 314       --time            show time of the last modification of any file in the\n\
 315                           directory, or any of its subdirectories\n\
 316       --time=WORD       show time as WORD instead of modification time:\n\
 317                           atime, access, use, ctime or status\n\
 318       --time-style=STYLE  show times using style STYLE:\n\
 319                           full-iso, long-iso, iso, +FORMAT\n\
 320                           FORMAT is interpreted like `date'\n\
 321 "), stdout);
 322       fputs (HELP_OPTION_DESCRIPTION, stdout);
 323       fputs (VERSION_OPTION_DESCRIPTION, stdout);
 324       emit_blocksize_note ("DU");
 325       emit_size_note ();
 326       emit_ancillary_info ();
 327     }
 328   exit (status);
 329 }
 330
 331 /* Try to insert the INO/DEV pair into the global table, HTAB.
 332    Return true if the pair is successfully inserted,
 333    false if the pair is already in the table.  */
 334 static bool
 335 hash_ins (ino_t ino, dev_t dev)
 336 {
 337   int inserted = di_set_insert (di_set, dev, ino);
 338   if (inserted < 0)
 339     xalloc_die ();
 340   return inserted;
 341 }
 342
 343 /* FIXME: this code is nearly identical to code in date.c  */
 344 /* Display the date and time in WHEN according to the format specified
 345    in FORMAT.  */
 346
 347 static void
 348 show_date (const char *format, struct timespec when)
 349 {
 350   struct tm *tm = localtime (&when.tv_sec);
 351   if (! tm)
 352     {
 353       char buf[INT_BUFSIZE_BOUND (intmax_t)];
 354       error (0, 0, _("time %s is out of range"), timetostr (when.tv_sec, buf));
 355       fputs (buf, stdout);
 356       return;
 357     }
 358
 359   fprintftime (stdout, format, tm, 0, when.tv_nsec);
 360 }
 361
 362 /* Print N_BYTES.  Convert it to a readable value before printing.  */
 363
 364 static void
 365 print_only_size (uintmax_t n_bytes)
 366 {
 367   char buf[LONGEST_HUMAN_READABLE + 1];
 368   fputs (human_readable (n_bytes, buf, human_output_opts,
 369                          1, output_block_size), stdout);
 370 }
 371
 372 /* Print size (and optionally time) indicated by *PDUI, followed by STRING.  */
 373
 374 static void
 375 print_size (const struct duinfo *pdui, const char *string)
 376 {
 377   print_only_size (pdui->size);
 378   if (opt_time)
 379     {
 380       putchar ('\t');
 381       show_date (time_format, pdui->tmax);
 382     }
 383   printf ("\t%s%c", string, opt_nul_terminate_output ? '\0' : '\n');
 384   fflush (stdout);
 385 }
 386
 387 /* This function is called once for every file system object that fts
 388    encounters.  fts does a depth-first traversal.  This function knows
 389    that and accumulates per-directory totals based on changes in
 390    the depth of the current entry.  It returns true on success.  */
 391
 392 static bool
 393 process_file (FTS *fts, FTSENT *ent)
 394 {
 395   bool ok;
 396   struct duinfo dui;
 397   struct duinfo dui_to_print;
 398   size_t level;
 399   static size_t prev_level;
 400   static size_t n_alloc;
 401   /* First element of the structure contains:
 402      The sum of the st_size values of all entries in the single directory
 403      at the corresponding level.  Although this does include the st_size
 404      corresponding to each subdirectory, it does not include the size of
 405      any file in a subdirectory. Also corresponding last modified date.
 406      Second element of the structure contains:
 407      The sum of the sizes of all entries in the hierarchy at or below the
 408      directory at the specified level.  */
 409   static struct dulevel *dulvl;
 410   bool print = true;
 411
 412   const char *file = ent->fts_path;
 413   const struct stat *sb = ent->fts_statp;
 414   bool skip;
 415
 416   /* If necessary, set FTS_SKIP before returning.  */
 417   skip = excluded_file_name (exclude, file);
 418   if (skip)
 419     fts_set (fts, ent, FTS_SKIP);
 420
 421   switch (ent->fts_info)
 422     {
 423     case FTS_NS:
 424       error (0, ent->fts_errno, _("cannot access %s"), quote (file));
 425       return false;
 426
 427     case FTS_ERR:
 428       /* if (S_ISDIR (ent->fts_statp->st_mode) && FIXME */
 429       error (0, ent->fts_errno, "%s", quote (file));
 430       return false;
 431
 432     case FTS_DNR:
 433       /* Don't return just yet, since although the directory is not readable,
 434          we were able to stat it, so we do have a size.  */
 435       error (0, ent->fts_errno, _("cannot read directory %s"), quote (file));
 436       ok = false;
 437       break;
 438
 439     case FTS_DC:                /* directory that causes cycles */
 440       if (cycle_warning_required (fts, ent))
 441         {
 442           emit_cycle_warning (file);
 443           return false;
 444         }
 445       ok = true;
 446       break;
 447
 448     default:
 449       ok = true;
 450       break;
 451     }
 452
 453   /* If this is the first (pre-order) encounter with a directory,
 454      or if it's the second encounter for a skipped directory, then
 455      return right away.  */
 456   if (ent->fts_info == FTS_D || skip)
 457     return ok;
 458
 459   /* If the file is being excluded or if it has already been counted
 460      via a hard link, then don't let it contribute to the sums.  */
 461   if (skip
 462       || (!opt_count_all
 463           && (hash_all || (! S_ISDIR (sb->st_mode) && 1 < sb->st_nlink))
 464           && ! hash_ins (sb->st_ino, sb->st_dev)))
 465     {
 466       /* Note that we must not simply return here.
 467          We still have to update prev_level and maybe propagate
 468          some sums up the hierarchy.  */
 469       duinfo_init (&dui);
 470       print = false;
 471     }
 472   else
 473     {
 474       duinfo_set (&dui,
 475                   (apparent_size
 476                    ? sb->st_size
 477                    : (uintmax_t) ST_NBLOCKS (*sb) * ST_NBLOCKSIZE),
 478                   (time_type == time_mtime ? get_stat_mtime (sb)
 479                    : time_type == time_atime ? get_stat_atime (sb)
 480                    : get_stat_ctime (sb)));
 481     }
 482
 483   level = ent->fts_level;
 484   dui_to_print = dui;
 485
 486   if (n_alloc == 0)
 487     {
 488       n_alloc = level + 10;
 489       dulvl = xcalloc (n_alloc, sizeof *dulvl);
 490     }
 491   else
 492     {
 493       if (level == prev_level)
 494         {
 495           /* This is usually the most common case.  Do nothing.  */
 496         }
 497       else if (level > prev_level)
 498         {
 499           /* Descending the hierarchy.
 500              Clear the accumulators for *all* levels between prev_level
 501              and the current one.  The depth may change dramatically,
 502              e.g., from 1 to 10.  */
 503           size_t i;
 504
 505           if (n_alloc <= level)
 506             {
 507               dulvl = xnrealloc (dulvl, level, 2 * sizeof *dulvl);
 508               n_alloc = level * 2;
 509             }
 510
 511           for (i = prev_level + 1; i <= level; i++)
 512             {
 513               duinfo_init (&dulvl[i].ent);
 514               duinfo_init (&dulvl[i].subdir);
 515             }
 516         }
 517       else /* level < prev_level */
 518         {
 519           /* Ascending the hierarchy.
 520              Process a directory only after all entries in that
 521              directory have been processed.  When the depth decreases,
 522              propagate sums from the children (prev_level) to the parent.
 523              Here, the current level is always one smaller than the
 524              previous one.  */
 525           assert (level == prev_level - 1);
 526           duinfo_add (&dui_to_print, &dulvl[prev_level].ent);
 527           if (!opt_separate_dirs)
 528             duinfo_add (&dui_to_print, &dulvl[prev_level].subdir);
 529           duinfo_add (&dulvl[level].subdir, &dulvl[prev_level].ent);
 530           duinfo_add (&dulvl[level].subdir, &dulvl[prev_level].subdir);
 531         }
 532     }
 533
 534   prev_level = level;
 535
 536   /* Let the size of a directory entry contribute to the total for the
 537      containing directory, unless --separate-dirs (-S) is specified.  */
 538   if ( ! (opt_separate_dirs && IS_DIR_TYPE (ent->fts_info)))
 539     duinfo_add (&dulvl[level].ent, &dui);
 540
 541   /* Even if this directory is unreadable or we can't chdir into it,
 542      do let its size contribute to the total. */
 543   duinfo_add (&tot_dui, &dui);
 544
 545   /* If we're not counting an entry, e.g., because it's a hard link
 546      to a file we've already counted (and --count-links), then don't
 547      print a line for it.  */
 548   if (!print)
 549     return ok;
 550
 551   if ((IS_DIR_TYPE (ent->fts_info) && level <= max_depth)
 552       || ((opt_all && level <= max_depth) || level == 0))
 553     print_size (&dui_to_print, file);
 554
 555   return ok;
 556 }
 557
 558 /* Recursively print the sizes of the directories (and, if selected, files)
 559    named in FILES, the last entry of which is NULL.
 560    BIT_FLAGS controls how fts works.
 561    Return true if successful.  */
 562
 563 static bool
 564 du_files (char **files, int bit_flags)
 565 {
 566   bool ok = true;
 567
 568   if (*files)
 569     {
 570       FTS *fts = xfts_open (files, bit_flags, NULL);
 571
 572       while (1)
 573         {
 574           FTSENT *ent;
 575
 576           ent = fts_read (fts);
 577           if (ent == NULL)
 578             {
 579               if (errno != 0)
 580                 {
 581                   /* FIXME: try to give a better message  */
 582                   error (0, errno, _("fts_read failed"));
 583                   ok = false;
 584                 }
 585               break;
 586             }
 587           FTS_CROSS_CHECK (fts);
 588
 589           ok &= process_file (fts, ent);
 590         }
 591
 592       if (fts_close (fts) != 0)
 593         {
 594           error (0, errno, _("fts_close failed"));
 595           ok = false;
 596         }
 597     }
 598
 599   return ok;
 600 }
 601
 602 int
 603 main (int argc, char **argv)
 604 {
 605   char *cwd_only[2];
 606   bool max_depth_specified = false;
 607   bool ok = true;
 608   char *files_from = NULL;
 609
 610   /* Bit flags that control how fts works.  */
 611   int bit_flags = FTS_TIGHT_CYCLE_CHECK | FTS_DEFER_STAT;
 612
 613   /* Select one of the three FTS_ options that control if/when
 614      to follow a symlink.  */
 615   int symlink_deref_bits = FTS_PHYSICAL;
 616
 617   /* If true, display only a total for each argument. */
 618   bool opt_summarize_only = false;
 619
 620   cwd_only[0] = bad_cast (".");
 621   cwd_only[1] = NULL;
 622
 623   initialize_main (&argc, &argv);
 624   set_program_name (argv[0]);
 625   setlocale (LC_ALL, "");
 626   bindtextdomain (PACKAGE, LOCALEDIR);
 627   textdomain (PACKAGE);
 628
 629   atexit (close_stdout);
 630
 631   exclude = new_exclude ();
 632
 633   human_options (getenv ("DU_BLOCK_SIZE"),
 634                  &human_output_opts, &output_block_size);
 635
 636   while (true)
 637     {
 638       int oi = -1;
 639       int c = getopt_long (argc, argv, "0abd:chHklmsxB:DLPSX:",
 640                            long_options, &oi);
 641       if (c == -1)
 642         break;
 643
 644       switch (c)
 645         {
 646 #if DU_DEBUG
 647         case FTS_DEBUG:
 648           fts_debug = true;
 649           break;
 650 #endif
 651
 652         case '0':
 653           opt_nul_terminate_output = true;
 654           break;
 655
 656         case 'a':
 657           opt_all = true;
 658           break;
 659
 660         case APPARENT_SIZE_OPTION:
 661           apparent_size = true;
 662           break;
 663
 664         case 'b':
 665           apparent_size = true;
 666           human_output_opts = 0;
 667           output_block_size = 1;
 668           break;
 669
 670         case 'c':
 671           print_grand_total = true;
 672           break;
 673
 674         case 'h':
 675           human_output_opts = human_autoscale | human_SI | human_base_1024;
 676           output_block_size = 1;
 677           break;
 678
 679         case HUMAN_SI_OPTION:
 680           human_output_opts = human_autoscale | human_SI;
 681           output_block_size = 1;
 682           break;
 683
 684         case 'k':
 685           human_output_opts = 0;
 686           output_block_size = 1024;
 687           break;
 688
 689         case 'd':               /* --max-depth=N */
 690           {
 691             unsigned long int tmp_ulong;
 692             if (xstrtoul (optarg, NULL, 0, &tmp_ulong, NULL) == LONGINT_OK
 693                 && tmp_ulong <= SIZE_MAX)
 694               {
 695                 max_depth_specified = true;
 696                 max_depth = tmp_ulong;
 697               }
 698             else
 699               {
 700                 error (0, 0, _("invalid maximum depth %s"),
 701                        quote (optarg));
 702                 ok = false;
 703               }
 704           }
 705           break;
 706
 707         case 'm':
 708           human_output_opts = 0;
 709           output_block_size = 1024 * 1024;
 710           break;
 711
 712         case 'l':
 713           opt_count_all = true;
 714           break;
 715
 716         case 's':
 717           opt_summarize_only = true;
 718           break;
 719
 720         case 'x':
 721           bit_flags |= FTS_XDEV;
 722           break;
 723
 724         case 'B':
 725           {
 726             enum strtol_error e = human_options (optarg, &human_output_opts,
 727                                                  &output_block_size);
 728             if (e != LONGINT_OK)
 729               xstrtol_fatal (e, oi, c, long_options, optarg);
 730           }
 731           break;
 732
 733         case 'H':  /* NOTE: before 2008-12, -H was equivalent to --si.  */
 734         case 'D':
 735           symlink_deref_bits = FTS_COMFOLLOW | FTS_PHYSICAL;
 736           break;
 737
 738         case 'L': /* --dereference */
 739           symlink_deref_bits = FTS_LOGICAL;
 740           break;
 741
 742         case 'P': /* --no-dereference */
 743           symlink_deref_bits = FTS_PHYSICAL;
 744           break;
 745
 746         case 'S':
 747           opt_separate_dirs = true;
 748           break;
 749
 750         case 'X':
 751           if (add_exclude_file (add_exclude, exclude, optarg,
 752                                 EXCLUDE_WILDCARDS, '\n'))
 753             {
 754               error (0, errno, "%s", quotearg_colon (optarg));
 755               ok = false;
 756             }
 757           break;
 758
 759         case FILES0_FROM_OPTION:
 760           files_from = optarg;
 761           break;
 762
 763         case EXCLUDE_OPTION:
 764           add_exclude (exclude, optarg, EXCLUDE_WILDCARDS);
 765           break;
 766
 767         case TIME_OPTION:
 768           opt_time = true;
 769           time_type =
 770             (optarg
 771              ? XARGMATCH ("--time", optarg, time_args, time_types)
 772              : time_mtime);
 773           break;
 774
 775         case TIME_STYLE_OPTION:
 776           time_style = optarg;
 777           break;
 778
 779         case_GETOPT_HELP_CHAR;
 780
 781         case_GETOPT_VERSION_CHAR (PROGRAM_NAME, AUTHORS);
 782
 783         default:
 784           ok = false;
 785         }
 786     }
 787
 788   if (!ok)
 789     usage (EXIT_FAILURE);
 790
 791   if (opt_all && opt_summarize_only)
 792     {
 793       error (0, 0, _("cannot both summarize and show all entries"));
 794       usage (EXIT_FAILURE);
 795     }
 796
 797   if (opt_summarize_only && max_depth_specified && max_depth == 0)
 798     {
 799       error (0, 0,
 800              _("warning: summarizing is the same as using --max-depth=0"));
 801     }
 802
 803   if (opt_summarize_only && max_depth_specified && max_depth != 0)
 804     {
 805       unsigned long int d = max_depth;
 806       error (0, 0, _("warning: summarizing conflicts with --max-depth=%lu"), d);
 807       usage (EXIT_FAILURE);
 808     }
 809
 810   if (opt_summarize_only)
 811     max_depth = 0;
 812
 813   /* Process time style if printing last times.  */
 814   if (opt_time)
 815     {
 816       if (! time_style)
 817         {
 818           time_style = getenv ("TIME_STYLE");
 819
 820           /* Ignore TIMESTYLE="locale", for compatibility with ls.  */
 821           if (! time_style || STREQ (time_style, "locale"))
 822             time_style = "long-iso";
 823           else if (*time_style == '+')
 824             {
 825               /* Ignore anything after a newline, for compatibility
 826                  with ls.  */
 827               char *p = strchr (time_style, '\n');
 828               if (p)
 829                 *p = '\0';
 830             }
 831           else
 832             {
 833               /* Ignore "posix-" prefix, for compatibility with ls.  */
 834               static char const posix_prefix[] = "posix-";
 835               while (strncmp (time_style, posix_prefix, sizeof posix_prefix - 1)
 836                      == 0)
 837                 time_style += sizeof posix_prefix - 1;
 838             }
 839         }
 840
 841       if (*time_style == '+')
 842         time_format = time_style + 1;
 843       else
 844         {
 845           switch (XARGMATCH ("time style", time_style,
 846                              time_style_args, time_style_types))
 847             {
 848             case full_iso_time_style:
 849               time_format = "%Y-%m-%d %H:%M:%S.%N %z";
 850               break;
 851
 852             case long_iso_time_style:
 853               time_format = "%Y-%m-%d %H:%M";
 854               break;
 855
 856             case iso_time_style:
 857               time_format = "%Y-%m-%d";
 858               break;
 859             }
 860         }
 861     }
 862
 863   struct argv_iterator *ai;
 864   if (files_from)
 865     {
 866       /* When using --files0-from=F, you may not specify any files
 867          on the command-line.  */
 868       if (optind < argc)
 869         {
 870           error (0, 0, _("extra operand %s"), quote (argv[optind]));
 871           fprintf (stderr, "%s\n",
 872                    _("file operands cannot be combined with --files0-from"));
 873           usage (EXIT_FAILURE);
 874         }
 875
 876       if (! (STREQ (files_from, "-") || freopen (files_from, "r", stdin)))
 877         error (EXIT_FAILURE, errno, _("cannot open %s for reading"),
 878                quote (files_from));
 879
 880       ai = argv_iter_init_stream (stdin);
 881
 882       /* It's not easy here to count the arguments, so assume the
 883          worst.  */
 884       hash_all = true;
 885     }
 886   else
 887     {
 888       char **files = (optind < argc ? argv + optind : cwd_only);
 889       ai = argv_iter_init_argv (files);
 890
 891       /* Hash all dev,ino pairs if there are multiple arguments, or if
 892          following non-command-line symlinks, because in either case a
 893          file with just one hard link might be seen more than once.  */
 894       hash_all = (optind + 1 < argc || symlink_deref_bits == FTS_LOGICAL);
 895     }
 896
 897   if (!ai)
 898     xalloc_die ();
 899
 900   /* Initialize the set of dev,inode pairs.  */
 901   di_set = di_set_alloc ();
 902   if (!di_set)
 903     xalloc_die ();
 904
 905   bit_flags |= symlink_deref_bits;
 906   static char *temp_argv[] = { NULL, NULL };
 907
 908   while (true)
 909     {
 910       bool skip_file = false;
 911       enum argv_iter_err ai_err;
 912       char *file_name = argv_iter (ai, &ai_err);
 913       if (ai_err == AI_ERR_EOF)
 914         break;
 915       if (!file_name)
 916         {
 917           switch (ai_err)
 918             {
 919             case AI_ERR_READ:
 920               error (0, errno, _("%s: read error"), quote (files_from));
 921               continue;
 922
 923             case AI_ERR_MEM:
 924               xalloc_die ();
 925
 926             default:
 927               assert (!"unexpected error code from argv_iter");
 928             }
 929         }
 930       if (files_from && STREQ (files_from, "-") && STREQ (file_name, "-"))
 931         {
 932           /* Give a better diagnostic in an unusual case:
 933              printf - | du --files0-from=- */
 934           error (0, 0, _("when reading file names from stdin, "
 935                          "no file name of %s allowed"),
 936                  quote (file_name));
 937           skip_file = true;
 938         }
 939
 940       /* Report and skip any empty file names before invoking fts.
 941          This works around a glitch in fts, which fails immediately
 942          (without looking at the other file names) when given an empty
 943          file name.  */
 944       if (!file_name[0])
 945         {
 946           /* Diagnose a zero-length file name.  When it's one
 947              among many, knowing the record number may help.
 948              FIXME: currently print the record number only with
 949              --files0-from=FILE.  Maybe do it for argv, too?  */
 950           if (files_from == NULL)
 951             error (0, 0, "%s", _("invalid zero-length file name"));
 952           else
 953             {
 954               /* Using the standard `filename:line-number:' prefix here is
 955                  not totally appropriate, since NUL is the separator, not NL,
 956                  but it might be better than nothing.  */
 957               unsigned long int file_number = argv_iter_n_args (ai);
 958               error (0, 0, "%s:%lu: %s", quotearg_colon (files_from),
 959                      file_number, _("invalid zero-length file name"));
 960             }
 961           skip_file = true;
 962         }
 963
 964       if (skip_file)
 965         ok = false;
 966       else
 967         {
 968           temp_argv[0] = file_name;
 969           ok &= du_files (temp_argv, bit_flags);
 970         }
 971     }
 972
 973   argv_iter_free (ai);
 974   di_set_free (di_set);
 975
 976   if (files_from && (ferror (stdin) || fclose (stdin) != 0))
 977     error (EXIT_FAILURE, 0, _("error reading %s"), quote (files_from));
 978
 979   if (print_grand_total)
 980     print_size (&tot_dui, _("total"));
 981
 982   exit (ok ? EXIT_SUCCESS : EXIT_FAILURE);
 983 }