coreutils/cksum.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * cksum - calculate the CRC32 checksum of a file
   4  *
   5  * Copyright (C) 2006 by Rob Sullivan, with ideas from code by Walter Harms
   6  *
   7  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
   8  */
   9 //config:config CKSUM
  10 //config:       bool "cksum (4.2 kb)"
  11 //config:       default y
  12 //config:       help
  13 //config:       cksum is used to calculate the CRC32 checksum of a file.
  14
  15 //applet:IF_CKSUM(APPLET_NOEXEC(cksum, cksum, BB_DIR_USR_BIN, BB_SUID_DROP, cksum))
  16 /* bb_common_bufsiz1 usage here is safe wrt NOEXEC: not expecting it to be zeroed. */
  17
  18 //kbuild:lib-$(CONFIG_CKSUM) += cksum.o
  19
  20 //usage:#define cksum_trivial_usage
  21 //usage:       "FILE..."
  22 //usage:#define cksum_full_usage "\n\n"
  23 //usage:       "Calculate the CRC32 checksums of FILEs"
  24
  25 #include "libbb.h"
  26 #include "common_bufsiz.h"
  27
  28 /* This is a NOEXEC applet. Be very careful! */
  29
  30 int cksum_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  31 int cksum_main(int argc UNUSED_PARAM, char **argv)
  32 {
  33         uint32_t *crc32_table = crc32_filltable(NULL, 1);
  34         uint32_t crc;
  35         off_t length, filesize;
  36         int bytes_read;
  37         int exit_code = EXIT_SUCCESS;
  38
  39 #if ENABLE_DESKTOP
  40         getopt32(argv, ""); /* coreutils 6.9 compat */
  41         argv += optind;
  42 #else
  43         argv++;
  44 #endif
  45
  46         setup_common_bufsiz();
  47         do {
  48                 int fd = open_or_warn_stdin(*argv ? *argv : bb_msg_standard_input);
  49
  50                 if (fd < 0) {
  51                         exit_code = EXIT_FAILURE;
  52                         continue;
  53                 }
  54                 crc = 0;
  55                 length = 0;
  56
  57 #define read_buf bb_common_bufsiz1
  58                 while ((bytes_read = safe_read(fd, read_buf, COMMON_BUFSIZE)) > 0) {
  59                         length += bytes_read;
  60                         crc = crc32_block_endian1(crc, read_buf, bytes_read, crc32_table);
  61                 }
  62                 close(fd);
  63
  64                 filesize = length;
  65
  66                 while (length) {
  67                         crc = (crc << 8) ^ crc32_table[(uint8_t)(crc >> 24) ^ (uint8_t)length];
  68                         /* must ensure that shift is unsigned! */
  69                         if (sizeof(length) <= sizeof(unsigned))
  70                                 length = (unsigned)length >> 8;
  71                         else if (sizeof(length) <= sizeof(unsigned long))
  72                                 length = (unsigned long)length >> 8;
  73                         else
  74                                 length = (unsigned long long)length >> 8;
  75                 }
  76                 crc = ~crc;
  77
  78                 printf((*argv ? "%"PRIu32" %"OFF_FMT"i %s\n" : "%"PRIu32" %"OFF_FMT"i\n"),
  79                                 crc, filesize, *argv);
  80         } while (*argv && *++argv);
  81
  82         fflush_stdout_and_exit(exit_code);
  83 }