1 /* $NetBSD: cksum.c,v 1.48 2015/06/16 22:54:10 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 1991, 1993 5 * The Regents of the University of California. All rights reserved. 6 * 7 * This code is derived from software contributed to Berkeley by 8 * James W. Williams of NASA Goddard Space Flight Center. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 3. Neither the name of the University nor the names of its contributors 19 * may be used to endorse or promote products derived from this software 20 * without specific prior written permission. 21 * 22 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 25 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 32 * SUCH DAMAGE. 33 */ 34 35 /*- 36 * Copyright (c) 1997 Jason R. Thorpe. All rights reserved. 37 * 38 * This code is derived from software contributed to Berkeley by 39 * James W. Williams of NASA Goddard Space Flight Center. 40 * 41 * Redistribution and use in source and binary forms, with or without 42 * modification, are permitted provided that the following conditions 43 * are met: 44 * 1. Redistributions of source code must retain the above copyright 45 * notice, this list of conditions and the following disclaimer. 46 * 2. Redistributions in binary form must reproduce the above copyright 47 * notice, this list of conditions and the following disclaimer in the 48 * documentation and/or other materials provided with the distribution. 49 * 3. All advertising materials mentioning features or use of this software 50 * must display the following acknowledgement: 51 * This product includes software developed by the University of 52 * California, Berkeley and its contributors. 53 * 4. Neither the name of the University nor the names of its contributors 54 * may be used to endorse or promote products derived from this software 55 * without specific prior written permission. 56 * 57 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 58 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 59 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 60 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 61 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 62 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 63 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 64 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 65 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 66 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 67 * SUCH DAMAGE. 68 */ 69 70 #if HAVE_NBTOOL_CONFIG_H 71 #include "nbtool_config.h" 72 #endif 73 74 #include <sys/cdefs.h> 75 #if defined(__COPYRIGHT) && !defined(lint) 76 __COPYRIGHT("@(#) Copyright (c) 1991, 1993\ 77 The Regents of the University of California. All rights reserved."); 78 #endif /* not lint */ 79 80 #if defined(__RCSID) && !defined(lint) 81 #if 0 82 static char sccsid[] = "@(#)cksum.c 8.2 (Berkeley) 4/28/95"; 83 #endif 84 __RCSID("$NetBSD: cksum.c,v 1.48 2015/06/16 22:54:10 christos Exp $"); 85 #endif /* not lint */ 86 87 #include <sys/types.h> 88 89 #include <ctype.h> 90 #include <err.h> 91 #include <errno.h> 92 #include <fcntl.h> 93 #include <locale.h> 94 #include <md2.h> 95 #include <md4.h> 96 #include <md5.h> 97 #include <rmd160.h> 98 #include <sha1.h> 99 #include <sha2.h> 100 #include <stdio.h> 101 #include <stdlib.h> 102 #include <string.h> 103 #include <unistd.h> 104 105 #include "extern.h" 106 107 #define PRINT_NORMAL 0x01 108 #define PRINT_QUIET 0x02 109 110 typedef char *(*_filefunc)(const char *, char *); 111 112 const struct hash { 113 const char *progname; 114 const char *hashname; 115 void (*stringfunc)(const char *); 116 void (*timetrialfunc)(void); 117 void (*testsuitefunc)(void); 118 void (*filterfunc)(int); 119 char *(*filefunc)(const char *, char *); 120 } hashes[] = { 121 { "md2", "MD2", 122 MD2String, MD2TimeTrial, MD2TestSuite, 123 MD2Filter, MD2File }, 124 { "md4", "MD4", 125 MD4String, MD4TimeTrial, MD4TestSuite, 126 MD4Filter, MD4File }, 127 { "md5", "MD5", 128 MD5String, MD5TimeTrial, MD5TestSuite, 129 MD5Filter, MD5File }, 130 { "rmd160", "RMD160", 131 RMD160String, RMD160TimeTrial, RMD160TestSuite, 132 RMD160Filter, (_filefunc) RMD160File }, 133 { "sha1", "SHA1", 134 SHA1String, SHA1TimeTrial, SHA1TestSuite, 135 SHA1Filter, (_filefunc) SHA1File }, 136 { "sha256", "SHA256", 137 SHA256_String, SHA256_TimeTrial, SHA256_TestSuite, 138 SHA256_Filter, (_filefunc) SHA256_File }, 139 { "sha384", "SHA384", 140 SHA384_String, SHA384_TimeTrial, SHA384_TestSuite, 141 SHA384_Filter, (_filefunc) SHA384_File }, 142 { "sha512", "SHA512", 143 SHA512_String, SHA512_TimeTrial, SHA512_TestSuite, 144 SHA512_Filter, (_filefunc) SHA512_File }, 145 { .progname = NULL, }, 146 }; 147 148 static int hash_digest_file(char *, const struct hash *, int); 149 __dead static void requirehash(const char *); 150 __dead static void usage(void); 151 152 int 153 main(int argc, char **argv) 154 { 155 int ch, fd, rval, pflag, nohashstdin; 156 u_int32_t val; 157 off_t len; 158 char *fn; 159 const char *progname; 160 int (*cfncn) (int, u_int32_t *, off_t *); 161 void (*pfncn) (char *, u_int32_t, off_t); 162 const struct hash *hash; 163 int i, check_warn, do_check; 164 int print_flags; 165 166 cfncn = NULL; 167 pfncn = NULL; 168 pflag = nohashstdin = 0; 169 check_warn = 0; 170 do_check = 0; 171 print_flags = 0; 172 173 setlocale(LC_ALL, ""); 174 175 progname = getprogname(); 176 177 for (hash = hashes; hash->hashname != NULL; hash++) 178 if (strcmp(progname, hash->progname) == 0) 179 break; 180 181 if (hash->hashname == NULL) { 182 hash = NULL; 183 184 if (!strcmp(progname, "sum")) { 185 cfncn = csum1; 186 pfncn = psum1; 187 } else { 188 cfncn = crc; 189 pfncn = pcrc; 190 } 191 } 192 193 while ((ch = getopt(argc, argv, "a:cno:pqs:twx")) != -1) 194 switch(ch) { 195 case 'a': 196 if (hash) { 197 warnx("illegal use of -a option"); 198 usage(); 199 } 200 i = 0; 201 while (hashes[i].hashname != NULL) { 202 if (!strcasecmp(hashes[i].hashname, optarg)) { 203 hash = &hashes[i]; 204 break; 205 } 206 i++; 207 } 208 if (hash == NULL) { 209 if (!strcasecmp(optarg, "old1")) { 210 cfncn = csum1; 211 pfncn = psum1; 212 } else if (!strcasecmp(optarg, "old2")) { 213 cfncn = csum2; 214 pfncn = psum2; 215 } else if (!strcasecmp(optarg, "crc")) { 216 cfncn = crc; 217 pfncn = pcrc; 218 } else { 219 warnx("illegal argument to -a option"); 220 usage(); 221 } 222 } 223 break; 224 case 'c': 225 do_check = 1; 226 break; 227 case 'n': 228 print_flags |= PRINT_NORMAL; 229 break; 230 case 'o': 231 if (hash) { 232 warnx("%s mutually exclusive with sum", 233 hash->hashname); 234 usage(); 235 } 236 if (!strcmp(optarg, "1")) { 237 cfncn = csum1; 238 pfncn = psum1; 239 } else if (!strcmp(optarg, "2")) { 240 cfncn = csum2; 241 pfncn = psum2; 242 } else { 243 warnx("illegal argument to -o option"); 244 usage(); 245 } 246 break; 247 case 'p': 248 if (hash == NULL) 249 requirehash("-p"); 250 pflag = 1; 251 break; 252 case 'q': 253 print_flags |= PRINT_QUIET; 254 break; 255 case 's': 256 if (hash == NULL) 257 requirehash("-s"); 258 nohashstdin = 1; 259 hash->stringfunc(optarg); 260 break; 261 case 't': 262 if (hash == NULL) 263 requirehash("-t"); 264 nohashstdin = 1; 265 hash->timetrialfunc(); 266 break; 267 case 'w': 268 check_warn = 1; 269 break; 270 case 'x': 271 if (hash == NULL) 272 requirehash("-x"); 273 nohashstdin = 1; 274 hash->testsuitefunc(); 275 break; 276 case '?': 277 default: 278 usage(); 279 } 280 argc -= optind; 281 argv += optind; 282 283 if (do_check) { 284 /* 285 * Verify checksums 286 */ 287 FILE *f; 288 char buf[BUFSIZ]; 289 char *s, *p_filename, *p_cksum; 290 int l_filename, l_cksum; 291 char filename[BUFSIZ]; 292 char cksum[BUFSIZ]; 293 int ok,cnt,badcnt; 294 295 rval = 0; 296 cnt = badcnt = 0; 297 298 if (argc == 0) { 299 f = fdopen(STDIN_FILENO, "r"); 300 } else { 301 f = fopen(argv[0], "r"); 302 } 303 if (f == NULL) 304 err(1, "Cannot read %s", 305 argc>0?argv[0]:"stdin"); 306 307 while(fgets(buf, sizeof(buf), f) != NULL) { 308 s = strrchr(buf, '\n'); 309 if (s) 310 *s = '\0'; 311 312 p_cksum = p_filename = NULL; 313 314 p_filename = strchr(buf, '('); 315 if (p_filename) { 316 /* 317 * Assume 'normal' output if there's a '(' 318 */ 319 p_filename += 1; 320 print_flags &= ~(PRINT_NORMAL); 321 322 p_cksum = strrchr(p_filename, ')'); 323 if (p_cksum == NULL) { 324 if (check_warn) 325 warnx("bogus format: %s. " 326 "Skipping...", 327 buf); 328 rval = 1; 329 continue; 330 } 331 p_cksum += 4; 332 333 l_cksum = strlen(p_cksum); 334 l_filename = p_cksum - p_filename - 4; 335 336 /* Sanity check, and find proper hash if 337 * it's not the same as the current program 338 */ 339 if (hash == NULL || 340 strncmp(buf, hash->hashname, 341 strlen(hash->hashname)) != 0) { 342 /* 343 * Search proper hash 344 */ 345 const struct hash *nhash; 346 347 for (nhash = hashes ; 348 nhash->hashname != NULL; 349 nhash++) 350 if (strncmp(buf, 351 nhash->hashname, 352 strlen(nhash->hashname)) == 0) 353 break; 354 355 356 if (nhash->hashname == NULL) { 357 if (check_warn) 358 warnx("unknown hash: %s", 359 buf); 360 rval = 1; 361 continue; 362 } else { 363 hash = nhash; 364 } 365 } 366 367 } else { 368 if (hash) { 369 int nspaces; 370 371 /* 372 * 'normal' output, no (ck)sum 373 */ 374 print_flags |= PRINT_NORMAL; 375 nspaces = 1; 376 377 p_cksum = buf; 378 p_filename = strchr(buf, ' '); 379 if (p_filename == NULL) { 380 if (check_warn) 381 warnx("no filename in %s? " 382 "Skipping...", buf); 383 rval = 1; 384 continue; 385 } 386 while (isspace((int)*++p_filename)) 387 nspaces++; 388 l_filename = strlen(p_filename); 389 l_cksum = p_filename - buf - nspaces; 390 } else { 391 /* 392 * sum/cksum output format 393 */ 394 p_cksum = buf; 395 s=strchr(p_cksum, ' '); 396 if (s == NULL) { 397 if (check_warn) 398 warnx("bogus format: %s." 399 " Skipping...", 400 buf); 401 rval = 1; 402 continue; 403 } 404 l_cksum = s - p_cksum; 405 406 p_filename = strrchr(buf, ' '); 407 if (p_filename == NULL) { 408 if (check_warn) 409 warnx("no filename in %s?" 410 " Skipping...", 411 buf); 412 rval = 1; 413 continue; 414 } 415 p_filename++; 416 l_filename = strlen(p_filename); 417 } 418 } 419 420 strlcpy(filename, p_filename, l_filename+1); 421 strlcpy(cksum, p_cksum, l_cksum+1); 422 423 if (hash) { 424 if (access(filename, R_OK) == 0 425 && strcmp(cksum, hash->filefunc(filename, NULL)) == 0) 426 ok = 1; 427 else 428 ok = 0; 429 } else { 430 if ((fd = open(filename, O_RDONLY, 0)) < 0) { 431 if (check_warn) 432 warn("%s", filename); 433 rval = 1; 434 ok = 0; 435 } else { 436 if (cfncn(fd, &val, &len)) 437 ok = 0; 438 else { 439 u_int32_t should_val; 440 441 should_val = 442 strtoul(cksum, NULL, 10); 443 if (val == should_val) 444 ok = 1; 445 else 446 ok = 0; 447 } 448 close(fd); 449 } 450 } 451 452 if (! ok) { 453 if (hash) 454 printf("(%s) ", hash->hashname); 455 printf("%s: FAILED\n", filename); 456 badcnt++; 457 } 458 cnt++; 459 460 } 461 fclose(f); 462 463 if (badcnt > 0) 464 rval = 1; 465 466 } else { 467 /* 468 * Calculate checksums 469 */ 470 471 fd = STDIN_FILENO; 472 fn = NULL; 473 rval = 0; 474 do { 475 if (*argv) { 476 fn = *argv++; 477 if (hash != NULL) { 478 if (hash_digest_file(fn, hash, print_flags)) { 479 warn("%s", fn); 480 rval = 1; 481 } 482 continue; 483 } 484 if ((fd = open(fn, O_RDONLY, 0)) < 0) { 485 warn("%s", fn); 486 rval = 1; 487 continue; 488 } 489 } else if (hash && !nohashstdin) { 490 hash->filterfunc(pflag); 491 } 492 493 if (hash == NULL) { 494 if (cfncn(fd, &val, &len)) { 495 warn("%s", fn ? fn : "stdin"); 496 rval = 1; 497 } else 498 pfncn(fn, val, len); 499 (void)close(fd); 500 } 501 } while (*argv); 502 } 503 exit(rval); 504 } 505 506 static int 507 hash_digest_file(char *fn, const struct hash *hash, int flags) 508 { 509 char *cp; 510 511 cp = hash->filefunc(fn, NULL); 512 if (cp == NULL) 513 return 1; 514 515 if (flags & PRINT_QUIET) 516 printf("%s\n", cp); 517 else if (flags & PRINT_NORMAL) 518 printf("%s %s\n", cp, fn); 519 else 520 printf("%s (%s) = %s\n", hash->hashname, fn, cp); 521 522 free(cp); 523 524 return 0; 525 } 526 527 static void 528 requirehash(const char *flg) 529 { 530 warnx("%s flag requires `-a algorithm'", flg); 531 usage(); 532 } 533 534 static void 535 usage(void) 536 { 537 const char fileargs[] = "[file ... | -c [-w] [sumfile]]"; 538 const char sumargs[] = "[-n] [-a algorithm [-ptx] [-s string]] [-o 1|2]"; 539 const char hashargs[] = "[-nptx] [-s string]"; 540 541 (void)fprintf(stderr, "usage: cksum %s\n %s\n", 542 sumargs, fileargs); 543 (void)fprintf(stderr, " sum %s\n %s\n", 544 sumargs, fileargs); 545 (void)fprintf(stderr, " md2 %s %s\n", hashargs, fileargs); 546 (void)fprintf(stderr, " md4 %s %s\n", hashargs, fileargs); 547 (void)fprintf(stderr, " md5 %s %s\n", hashargs, fileargs); 548 (void)fprintf(stderr, " rmd160 %s %s\n", hashargs, fileargs); 549 (void)fprintf(stderr, " sha1 %s %s\n", hashargs, fileargs); 550 exit(1); 551 } 552