Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 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 |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 6 | * |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 7 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. */ |
| 8 | |
Denis Vlasenko | b6adbf1 | 2007-05-26 19:00:18 +0000 | [diff] [blame] | 9 | #include "libbb.h" |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 10 | |
Denis Vlasenko | 9b49a5e | 2007-10-11 10:05:36 +0000 | [diff] [blame] | 11 | int cksum_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 12 | int cksum_main(int argc UNUSED_PARAM, char **argv) |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 13 | { |
Denis Vlasenko | c6758a0 | 2007-04-10 21:40:19 +0000 | [diff] [blame] | 14 | uint32_t *crc32_table = crc32_filltable(NULL, 1); |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 15 | uint32_t crc; |
Denis Vlasenko | 8f0e342 | 2008-08-27 21:31:23 +0000 | [diff] [blame] | 16 | off_t length, filesize; |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 17 | int bytes_read; |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 18 | uint8_t *cp; |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 19 | |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 20 | #if ENABLE_DESKTOP |
| 21 | getopt32(argv, ""); /* coreutils 6.9 compat */ |
| 22 | argv += optind; |
| 23 | #else |
| 24 | argv++; |
| 25 | #endif |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 26 | |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 27 | do { |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 28 | int fd = open_or_warn_stdin(*argv ? *argv : bb_msg_standard_input); |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 29 | |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 30 | if (fd < 0) |
| 31 | continue; |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 32 | crc = 0; |
| 33 | length = 0; |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 34 | |
Denis Vlasenko | 74324c8 | 2007-06-04 10:16:52 +0000 | [diff] [blame] | 35 | #define read_buf bb_common_bufsiz1 |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 36 | while ((bytes_read = safe_read(fd, read_buf, sizeof(read_buf))) > 0) { |
Denis Vlasenko | 5599502 | 2008-05-18 22:28:26 +0000 | [diff] [blame] | 37 | cp = (uint8_t *) read_buf; |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 38 | length += bytes_read; |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 39 | do { |
| 40 | crc = (crc << 8) ^ crc32_table[(crc >> 24) ^ *cp++]; |
| 41 | } while (--bytes_read); |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 42 | } |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 43 | close(fd); |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 44 | |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 45 | filesize = length; |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 46 | |
Denis Vlasenko | 8f0e342 | 2008-08-27 21:31:23 +0000 | [diff] [blame] | 47 | while (length) { |
| 48 | crc = (crc << 8) ^ crc32_table[(uint8_t)(crc >> 24) ^ (uint8_t)length]; |
| 49 | /* must ensure that shift is unsigned! */ |
| 50 | if (sizeof(length) <= sizeof(unsigned)) |
| 51 | length = (unsigned)length >> 8; |
| 52 | else if (sizeof(length) <= sizeof(unsigned long)) |
| 53 | length = (unsigned long)length >> 8; |
| 54 | else |
| 55 | length = (unsigned long long)length >> 8; |
| 56 | } |
| 57 | crc = ~crc; |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 58 | |
Denis Vlasenko | 8f0e342 | 2008-08-27 21:31:23 +0000 | [diff] [blame] | 59 | printf((*argv ? "%"PRIu32" %"OFF_FMT"i %s\n" : "%"PRIu32" %"OFF_FMT"i\n"), |
Denis Vlasenko | 62a90cd | 2008-03-17 09:07:36 +0000 | [diff] [blame] | 60 | crc, filesize, *argv); |
| 61 | } while (*argv && *++argv); |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 62 | |
Denis Vlasenko | f0ed376 | 2006-10-26 23:21:47 +0000 | [diff] [blame] | 63 | fflush_stdout_and_exit(EXIT_SUCCESS); |
Rob Landley | 856489b | 2006-04-18 20:57:28 +0000 | [diff] [blame] | 64 | } |