1 /* $NetBSD: cksum.c,v 1.41 2007/01/17 00:21:43 hubertf 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\n\ 77 The Regents of the University of California. All rights reserved.\n"); 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.41 2007/01/17 00:21:43 hubertf 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 <md5.h> 95 #include <md4.h> 96 #include <md2.h> 97 #include <sha1.h> 98 #include <sha2.h> 99 #include <rmd160.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 HASH_MD2 0 108 #define HASH_MD4 1 109 #define HASH_MD5 2 110 #define HASH_SHA1 3 111 #define HASH_RMD160 4 112 113 typedef char *(*_filefunc)(const char *, char *); 114 115 struct hash { 116 const char *progname; 117 const char *hashname; 118 void (*stringfunc)(const char *); 119 void (*timetrialfunc)(void); 120 void (*testsuitefunc)(void); 121 void (*filterfunc)(int); 122 char *(*filefunc)(const char *, char *); 123 } hashes[] = { 124 { "md2", "MD2", 125 MD2String, MD2TimeTrial, MD2TestSuite, 126 MD2Filter, MD2File }, 127 { "md4", "MD4", 128 MD4String, MD4TimeTrial, MD4TestSuite, 129 MD4Filter, MD4File }, 130 { "md5", "MD5", 131 MD5String, MD5TimeTrial, MD5TestSuite, 132 MD5Filter, MD5File }, 133 { "sha1", "SHA1", 134 SHA1String, SHA1TimeTrial, SHA1TestSuite, 135 SHA1Filter, (_filefunc) SHA1File }, 136 { "rmd160", "RMD160", 137 RMD160String, RMD160TimeTrial, RMD160TestSuite, 138 RMD160Filter, (_filefunc) RMD160File }, 139 { "sha256", "SHA256", 140 SHA256_String, SHA256_TimeTrial, SHA256_TestSuite, 141 SHA256_Filter, (_filefunc) SHA256_File }, 142 { "sha384", "SHA384", 143 SHA384_String, SHA384_TimeTrial, SHA384_TestSuite, 144 SHA384_Filter, (_filefunc) SHA384_File }, 145 { "sha512", "SHA512", 146 SHA512_String, SHA512_TimeTrial, SHA512_TestSuite, 147 SHA512_Filter, (_filefunc) SHA512_File }, 148 { .progname = NULL, }, 149 }; 150 151 int hash_digest_file(char *, struct hash *, int); 152 void requirehash(const char *); 153 void usage(void); 154 155 int 156 main(int argc, char **argv) 157 { 158 int ch, fd, rval, dosum, pflag, nohashstdin; 159 u_int32_t val; 160 off_t len; 161 char *fn; 162 const char *progname; 163 int (*cfncn) (int, u_int32_t *, off_t *); 164 void (*pfncn) (char *, u_int32_t, off_t); 165 struct hash *hash; 166 int normal, i, check_warn, do_check; 167 168 cfncn = NULL; 169 pfncn = NULL; 170 dosum = pflag = nohashstdin = 0; 171 normal = 0; 172 check_warn = 0; 173 do_check = 0; 174 175 setlocale(LC_ALL, ""); 176 177 progname = getprogname(); 178 179 for (hash = hashes; hash->hashname != NULL; hash++) 180 if (strcmp(progname, hash->progname) == 0) 181 break; 182 183 if (hash->hashname == NULL) { 184 hash = NULL; 185 186 if (!strcmp(progname, "sum")) { 187 dosum = 1; 188 cfncn = csum1; 189 pfncn = psum1; 190 } else { 191 cfncn = crc; 192 pfncn = pcrc; 193 } 194 } 195 196 while ((ch = getopt(argc, argv, "a:cno:ps:twx")) != -1) 197 switch(ch) { 198 case 'a': 199 if (hash != NULL || dosum) { 200 warnx("illegal use of -a option\n"); 201 usage(); 202 } 203 i = 0; 204 while (hashes[i].hashname != NULL) { 205 if (!strcasecmp(hashes[i].hashname, optarg)) { 206 hash = &hashes[i]; 207 break; 208 } 209 i++; 210 } 211 if (hash == NULL) { 212 if (!strcasecmp(optarg, "old1")) { 213 cfncn = csum1; 214 pfncn = psum1; 215 } else if (!strcasecmp(optarg, "old2")) { 216 cfncn = csum2; 217 pfncn = psum2; 218 } else if (!strcasecmp(optarg, "crc")) { 219 cfncn = crc; 220 pfncn = pcrc; 221 } else { 222 warnx("illegal argument to -a option"); 223 usage(); 224 } 225 } 226 break; 227 case 'c': 228 do_check = 1; 229 break; 230 case 'n': 231 normal = 1; 232 break; 233 case 'o': 234 if (hash) { 235 warnx("%s mutually exclusive with sum", 236 hash->hashname); 237 usage(); 238 } 239 if (!strcmp(optarg, "1")) { 240 cfncn = csum1; 241 pfncn = psum1; 242 } else if (!strcmp(optarg, "2")) { 243 cfncn = csum2; 244 pfncn = psum2; 245 } else { 246 warnx("illegal argument to -o option"); 247 usage(); 248 } 249 break; 250 case 'p': 251 if (hash == NULL) 252 requirehash("-p"); 253 pflag = 1; 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 normal = 0; 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 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 normal = 1; 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, normal)) { 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 int 507 hash_digest_file(char *fn, struct hash *hash, int normal) 508 { 509 char *cp; 510 511 cp = hash->filefunc(fn, NULL); 512 if (cp == NULL) 513 return 1; 514 515 if (normal) 516 printf("%s %s\n", cp, fn); 517 else 518 printf("%s (%s) = %s\n", hash->hashname, fn, cp); 519 520 free(cp); 521 522 return 0; 523 } 524 525 void 526 requirehash(const char *flg) 527 { 528 warnx("%s flag requires `md2', `md4', `md5', `sha1', or `rmd160'", 529 flg); 530 usage(); 531 } 532 533 void 534 usage(void) 535 { 536 537 (void)fprintf(stderr, "usage: cksum [-nw] [-a algorithm | -c file ]\n\t\t| [-o 1 | 2]] [file ...]\n"); 538 (void)fprintf(stderr, " sum [-c] [file ...]\n"); 539 (void)fprintf(stderr, 540 " md2 [-n] [-p | -t | -x | -s string] [file ...]\n"); 541 (void)fprintf(stderr, 542 " md4 [-n] [-p | -t | -x | -s string] [file ...]\n"); 543 (void)fprintf(stderr, 544 " md5 [-n] [-p | -t | -x | -s string] [file ...]\n"); 545 (void)fprintf(stderr, 546 " sha1 [-n] [-p | -t | -x | -s string] [file ...]\n"); 547 (void)fprintf(stderr, 548 " rmd160 [-n] [-p | -t | -x | -s string] [file ...]\n"); 549 exit(1); 550 } 551