coreutils/sum.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * sum -- checksum and count the blocks in a file
   4  *     Like BSD sum or SysV sum -r, except like SysV sum if -s option is given.
   5  *
   6  * Copyright (C) 86, 89, 91, 1995-2002, 2004 Free Software Foundation, Inc.
   7  * Copyright (C) 2005 by Erik Andersen <andersen@codepoet.org>
   8  * Copyright (C) 2005 by Mike Frysinger <vapier@gentoo.org>
   9  *
  10  * Written by Kayvan Aghaiepour and David MacKenzie
  11  * Taken from coreutils and turned into a busybox applet by Mike Frysinger
  12  *
  13  * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
  14  */
  15
  16 #include "busybox.h"
  17
  18 /* 1 if any of the files read were the standard input */
  19 static int have_read_stdin;
  20
  21 /* make a little more readable and avoid using strcmp for just 2 bytes */
  22 #define IS_STDIN(s) (s[0] == '-' && s[1] == '\0')
  23
  24 /* Calculate and print the rotated checksum and the size in 1K blocks
  25    of file FILE, or of the standard input if FILE is "-".
  26    If PRINT_NAME is >1, print FILE next to the checksum and size.
  27    The checksum varies depending on sizeof (int).
  28    Return 1 if successful.  */
  29 static int bsd_sum_file(const char *file, int print_name)
  30 {
  31         FILE *fp;
  32         int checksum = 0;          /* The checksum mod 2^16. */
  33         uintmax_t total_bytes = 0; /* The number of bytes. */
  34         int ch;                    /* Each character read. */
  35         int ret = 0;
  36
  37         if (IS_STDIN(file)) {
  38                 fp = stdin;
  39                 have_read_stdin++;
  40         } else {
  41                 fp = fopen_or_warn(file, "r");
  42                 if (fp == NULL)
  43                         goto out;
  44         }
  45
  46         while ((ch = getc(fp)) != EOF) {
  47                 ++total_bytes;
  48                 checksum = (checksum >> 1) + ((checksum & 1) << 15);
  49                 checksum += ch;
  50                 checksum &= 0xffff;             /* Keep it within bounds. */
  51         }
  52
  53         if (ferror(fp)) {
  54                 bb_perror_msg(file);
  55                 fclose_if_not_stdin(fp);
  56                 goto out;
  57         }
  58
  59         if (fclose_if_not_stdin(fp) == EOF) {
  60                 bb_perror_msg(file);
  61                 goto out;
  62         }
  63         ret++;
  64         printf("%05d %5ju ", checksum, (total_bytes+1023)/1024);
  65         if (print_name > 1)
  66                 puts(file);
  67         else
  68                 puts("");
  69 out:
  70         return ret;
  71 }
  72
  73 /* Calculate and print the checksum and the size in 512-byte blocks
  74    of file FILE, or of the standard input if FILE is "-".
  75    If PRINT_NAME is >0, print FILE next to the checksum and size.
  76    Return 1 if successful.  */
  77 #define MY_BUF_SIZE 8192
  78 static int sysv_sum_file(const char *file, int print_name)
  79 {
  80         RESERVE_CONFIG_BUFFER(buf, MY_BUF_SIZE);
  81         int fd;
  82         uintmax_t total_bytes = 0;
  83
  84         /* The sum of all the input bytes, modulo (UINT_MAX + 1).  */
  85         unsigned int s = 0;
  86
  87         if (IS_STDIN(file)) {
  88                 fd = 0;
  89                 have_read_stdin = 1;
  90         } else {
  91                 fd = open(file, O_RDONLY);
  92                 if (fd == -1)
  93                         goto release_and_ret;
  94         }
  95
  96         while (1) {
  97                 size_t bytes_read = safe_read(fd, buf, MY_BUF_SIZE);
  98
  99                 if (bytes_read == 0)
 100                         break;
 101
 102                 if (bytes_read == -1) {
 103 release_and_ret:
 104                         bb_perror_msg(file);
 105                         RELEASE_CONFIG_BUFFER(buf);
 106                         if (!IS_STDIN(file))
 107                                 close(fd);
 108                         return 0;
 109                 }
 110
 111                 total_bytes += bytes_read;
 112                 while (bytes_read--)
 113                         s += buf[bytes_read];
 114         }
 115
 116         if (!IS_STDIN(file) && close(fd) == -1)
 117                 goto release_and_ret;
 118         else
 119                 RELEASE_CONFIG_BUFFER(buf);
 120
 121         {
 122                 int r = (s & 0xffff) + ((s & 0xffffffff) >> 16);
 123                 s = (r & 0xffff) + (r >> 16);
 124
 125                 printf("%d %ju ", s, (total_bytes+511)/512);
 126         }
 127         puts(print_name ? file : "");
 128
 129         return 1;
 130 }
 131
 132 int sum_main(int argc, char **argv)
 133 {
 134         int flags;
 135         int ok;
 136         int (*sum_func)(const char *, int) = bsd_sum_file;
 137
 138         /* give the bsd func priority over sysv func */
 139         flags = getopt32(argc, argv, "sr");
 140         if (flags & 1)
 141                 sum_func = sysv_sum_file;
 142         if (flags & 2)
 143                 sum_func = bsd_sum_file;
 144
 145         have_read_stdin = 0;
 146         if ((argc - optind) == 0)
 147                 ok = sum_func("-", 0);
 148         else
 149                 for (ok = 1; optind < argc; optind++)
 150                         ok &= sum_func(argv[optind], 1);
 151
 152         if (have_read_stdin && fclose(stdin) == EOF)
 153                 bb_perror_msg_and_die("-");
 154
 155         exit(ok ? EXIT_SUCCESS : EXIT_FAILURE);
 156 }