tftp: optional tftp-hpa compat

[oweals/busybox.git] / coreutils / cksum.c

diff --git a/coreutils/cksum.c b/coreutils/cksum.c
index 70febdf3ad2250d8665f3f14e8a47a5e2981cab3..633322bc7eb015bc52e7d44b1021b4cf4365c127 100644 (file)
--- a/coreutils/cksum.c
+++ b/coreutils/cksum.c
@@ -4,50 +4,81 @@
  *
  * Copyright (C) 2006 by Rob Sullivan, with ideas from code by Walter Harms
  *
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. */
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
+ */
+//config:config CKSUM
+//config:      bool "cksum (4.1 kb)"
+//config:      default y
+//config:      help
+//config:      cksum is used to calculate the CRC32 checksum of a file.
 
-#include "busybox.h"
+//applet:IF_CKSUM(APPLET_NOEXEC(cksum, cksum, BB_DIR_USR_BIN, BB_SUID_DROP, cksum))
+/* bb_common_bufsiz1 usage here is safe wrt NOEXEC: not expecting it to be zeroed. */
 
-int cksum_main(int argc, char **argv);
-int cksum_main(int argc, char **argv)
-{
-       uint32_t *crc32_table = crc32_filltable(NULL, 1);
+//kbuild:lib-$(CONFIG_CKSUM) += cksum.o
 
-       FILE *fp;
-       uint32_t crc;
-       long length, filesize;
-       int bytes_read;
-       char *cp;
+//usage:#define cksum_trivial_usage
+//usage:       "FILE..."
+//usage:#define cksum_full_usage "\n\n"
+//usage:       "Calculate the CRC32 checksums of FILEs"
 
-       int inp_stdin = (argc == optind) ? 1 : 0;
+#include "libbb.h"
+#include "common_bufsiz.h"
 
-       do {
-               fp = fopen_or_warn_stdin((inp_stdin) ? bb_msg_standard_input : *++argv);
+/* This is a NOEXEC applet. Be very careful! */
 
-               crc = 0;
-               length = 0;
+int cksum_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int cksum_main(int argc UNUSED_PARAM, char **argv)
+{
+       uint32_t *crc32_table = crc32_filltable(NULL, 1);
+       int exit_code = EXIT_SUCCESS;
 
-               while ((bytes_read = fread(bb_common_bufsiz1, 1, BUFSIZ, fp)) > 0) {
-                       cp = bb_common_bufsiz1;
-                       length += bytes_read;
-                       while (bytes_read--)
-                               crc = (crc << 8) ^ crc32_table[((crc >> 24) ^ (*cp++)) & 0xffL];
-               }
+#if ENABLE_DESKTOP
+       getopt32(argv, ""); /* coreutils 6.9 compat */
+       argv += optind;
+#else
+       argv++;
+#endif
 
-               filesize = length;
+       setup_common_bufsiz();
+       do {
+               uint32_t crc;
+               off_t filesize;
+               int fd = open_or_warn_stdin(*argv ? *argv : bb_msg_standard_input);
 
-               for (; length; length >>= 8)
-                       crc = (crc << 8) ^ crc32_table[((crc >> 24) ^ length) & 0xffL];
-               crc ^= 0xffffffffL;
+               if (fd < 0) {
+                       exit_code = EXIT_FAILURE;
+                       continue;
+               }
 
-               if (inp_stdin) {
-                       printf("%" PRIu32 " %li\n", crc, filesize);
-                       break;
+               crc = 0;
+               filesize = 0;
+#define read_buf bb_common_bufsiz1
+               for (;;) {
+                       uoff_t t;
+                       int bytes_read = safe_read(fd, read_buf, COMMON_BUFSIZE);
+                       if (bytes_read > 0) {
+                               filesize += bytes_read;
+                       } else {
+                               /* Checksum filesize bytes, LSB first, and exit */
+                               close(fd);
+                               fd = -1; /* break flag */
+                               t = filesize;
+                               bytes_read = 0;
+                               while (t != 0) {
+                                       read_buf[bytes_read++] = (uint8_t)t;
+                                       t >>= 8;
+                               }
+                       }
+                       crc = crc32_block_endian1(crc, read_buf, bytes_read, crc32_table);
+                       if (fd < 0)
+                               break;
                }
 
-               printf("%" PRIu32 " %li %s\n", crc, filesize, *argv);
-               fclose(fp);
-       } while (*(argv + 1));
+               crc = ~crc;
+               printf((*argv ? "%u %"OFF_FMT"u %s\n" : "%u %"OFF_FMT"u\n"),
+                               (unsigned)crc, filesize, *argv);
+       } while (*argv && *++argv);
 
-       fflush_stdout_and_exit(EXIT_SUCCESS);
+       fflush_stdout_and_exit(exit_code);
 }