]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - sbin/md5/md5.c
ldns: Upgrade to 1.8.3.
[FreeBSD/FreeBSD.git] / sbin / md5 / md5.c
1 /*
2  * Derived from:
3  *
4  * MDDRIVER.C - test driver for MD2, MD4 and MD5
5  */
6
7 /*
8  *  Copyright (C) 1990-2, RSA Data Security, Inc. Created 1990. All
9  *  rights reserved.
10  *
11  *  RSA Data Security, Inc. makes no representations concerning either
12  *  the merchantability of this software or the suitability of this
13  *  software for any particular purpose. It is provided "as is"
14  *  without express or implied warranty of any kind.
15  *
16  *  These notices must be retained in any copies of any part of this
17  *  documentation and/or software.
18  */
19
20 #include <sys/cdefs.h>
21 __FBSDID("$FreeBSD$");
22
23 #include <sys/param.h>
24 #include <sys/resource.h>
25 #include <sys/stat.h>
26 #include <sys/time.h>
27
28 #include <err.h>
29 #include <errno.h>
30 #include <fcntl.h>
31 #include <getopt.h>
32 #include <md5.h>
33 #include <osreldate.h>
34 #include <ripemd.h>
35 #include <sha.h>
36 #include <sha224.h>
37 #include <sha256.h>
38 #include <sha384.h>
39 #include <sha512.h>
40 #include <sha512t.h>
41 #include <skein.h>
42 #include <stdbool.h>
43 #include <stdio.h>
44 #include <stdlib.h>
45 #include <string.h>
46 #include <time.h>
47 #include <unistd.h>
48
49 #ifdef HAVE_CAPSICUM
50 #include <sys/capsicum.h>
51 #include <capsicum_helpers.h>
52 #endif
53
54 /*
55  * Length of test block, number of test blocks.
56  */
57 #define TEST_BLOCK_LEN 10000
58 #define TEST_BLOCK_COUNT 100000
59 #define MDTESTCOUNT 8
60
61 static char *progname;
62
63 static bool cflag;
64 static bool pflag;
65 static bool qflag;
66 static bool sflag;
67 static bool wflag;
68 static bool strict;
69 static bool skip;
70 static bool ignoreMissing;
71 static char* checkAgainst;
72 static int checksFailed;
73 static bool failed;
74 static int endl = '\n';
75
76 typedef void (DIGEST_Init)(void *);
77 typedef void (DIGEST_Update)(void *, const unsigned char *, size_t);
78 typedef char *(DIGEST_End)(void *, char *);
79
80 extern const char *MD5TestOutput[MDTESTCOUNT];
81 extern const char *SHA1_TestOutput[MDTESTCOUNT];
82 extern const char *SHA224_TestOutput[MDTESTCOUNT];
83 extern const char *SHA256_TestOutput[MDTESTCOUNT];
84 extern const char *SHA384_TestOutput[MDTESTCOUNT];
85 extern const char *SHA512_TestOutput[MDTESTCOUNT];
86 extern const char *SHA512t224_TestOutput[MDTESTCOUNT];
87 extern const char *SHA512t256_TestOutput[MDTESTCOUNT];
88 extern const char *RIPEMD160_TestOutput[MDTESTCOUNT];
89 extern const char *SKEIN256_TestOutput[MDTESTCOUNT];
90 extern const char *SKEIN512_TestOutput[MDTESTCOUNT];
91 extern const char *SKEIN1024_TestOutput[MDTESTCOUNT];
92
93 typedef struct Algorithm_t {
94         const char *progname;
95         const char *perlname;
96         const char *name;
97         const char *(*TestOutput)[MDTESTCOUNT];
98         DIGEST_Init *Init;
99         DIGEST_Update *Update;
100         DIGEST_End *End;
101         char *(*Data)(const void *, unsigned int, char *);
102 } Algorithm_t;
103
104 static void MD5_Update(MD5_CTX *, const unsigned char *, size_t);
105 static char *MDInput(const Algorithm_t *, FILE *, char *, bool);
106 static void MDOutput(const Algorithm_t *, char *, const char *);
107 static void MDTimeTrial(const Algorithm_t *);
108 static void MDTestSuite(const Algorithm_t *);
109 static void usage(const Algorithm_t *);
110 static void version(void);
111
112 typedef union {
113         MD5_CTX md5;
114         SHA1_CTX sha1;
115         SHA224_CTX sha224;
116         SHA256_CTX sha256;
117         SHA384_CTX sha384;
118         SHA512_CTX sha512;
119         RIPEMD160_CTX ripemd160;
120         SKEIN256_CTX skein256;
121         SKEIN512_CTX skein512;
122         SKEIN1024_CTX skein1024;
123 } DIGEST_CTX;
124
125 /* max(MD5_DIGEST_LENGTH, SHA_DIGEST_LENGTH,
126         SHA256_DIGEST_LENGTH, SHA512_DIGEST_LENGTH,
127         RIPEMD160_DIGEST_LENGTH, SKEIN1024_DIGEST_LENGTH)*2+1 */
128 #define HEX_DIGEST_LENGTH 257
129
130 /* algorithm function table */
131
132 static const struct Algorithm_t Algorithm[] = {
133         { "md5", NULL, "MD5",
134                 &MD5TestOutput, (DIGEST_Init*)&MD5Init,
135                 (DIGEST_Update*)&MD5_Update, (DIGEST_End*)&MD5End,
136                 &MD5Data },
137         { "sha1", "1", "SHA1",
138                 &SHA1_TestOutput, (DIGEST_Init*)&SHA1_Init,
139                 (DIGEST_Update*)&SHA1_Update, (DIGEST_End*)&SHA1_End,
140                 &SHA1_Data },
141         { "sha224", "224", "SHA224",
142                 &SHA224_TestOutput, (DIGEST_Init*)&SHA224_Init,
143                 (DIGEST_Update*)&SHA224_Update, (DIGEST_End*)&SHA224_End,
144                 &SHA224_Data },
145         { "sha256", "256", "SHA256",
146                 &SHA256_TestOutput, (DIGEST_Init*)&SHA256_Init,
147                 (DIGEST_Update*)&SHA256_Update, (DIGEST_End*)&SHA256_End,
148                 &SHA256_Data },
149         { "sha384", "384", "SHA384",
150                 &SHA384_TestOutput, (DIGEST_Init*)&SHA384_Init,
151                 (DIGEST_Update*)&SHA384_Update, (DIGEST_End*)&SHA384_End,
152                 &SHA384_Data },
153         { "sha512", "512", "SHA512",
154                 &SHA512_TestOutput, (DIGEST_Init*)&SHA512_Init,
155                 (DIGEST_Update*)&SHA512_Update, (DIGEST_End*)&SHA512_End,
156                 &SHA512_Data },
157         { "sha512t224", "512224", "SHA512t224",
158                 &SHA512t224_TestOutput, (DIGEST_Init*)&SHA512_224_Init,
159                 (DIGEST_Update*)&SHA512_224_Update, (DIGEST_End*)&SHA512_224_End,
160                 &SHA512_224_Data },
161         { "sha512t256", "512256", "SHA512t256",
162                 &SHA512t256_TestOutput, (DIGEST_Init*)&SHA512_256_Init,
163                 (DIGEST_Update*)&SHA512_256_Update, (DIGEST_End*)&SHA512_256_End,
164                 &SHA512_256_Data },
165         { "rmd160", NULL, "RMD160",
166                 &RIPEMD160_TestOutput,
167                 (DIGEST_Init*)&RIPEMD160_Init, (DIGEST_Update*)&RIPEMD160_Update,
168                 (DIGEST_End*)&RIPEMD160_End, &RIPEMD160_Data },
169         { "skein256", NULL, "Skein256",
170                 &SKEIN256_TestOutput,
171                 (DIGEST_Init*)&SKEIN256_Init, (DIGEST_Update*)&SKEIN256_Update,
172                 (DIGEST_End*)&SKEIN256_End, &SKEIN256_Data },
173         { "skein512", NULL, "Skein512",
174                 &SKEIN512_TestOutput,
175                 (DIGEST_Init*)&SKEIN512_Init, (DIGEST_Update*)&SKEIN512_Update,
176                 (DIGEST_End*)&SKEIN512_End, &SKEIN512_Data },
177         { "skein1024", NULL, "Skein1024",
178                 &SKEIN1024_TestOutput,
179                 (DIGEST_Init*)&SKEIN1024_Init, (DIGEST_Update*)&SKEIN1024_Update,
180                 (DIGEST_End*)&SKEIN1024_End, &SKEIN1024_Data },
181         { }
182 };
183
184 static int digest = -1;
185 static unsigned int malformed;
186
187 static enum mode {
188         mode_bsd,
189         mode_gnu,
190         mode_perl,
191 } mode = mode_bsd;
192
193 static enum input_mode {
194         input_binary     = '*',
195         input_text       = ' ',
196         input_universal  = 'U',
197         input_bits       = '^',
198 } input_mode = input_binary;
199
200 static enum output_mode {
201         output_bare,
202         output_tagged,
203         output_reverse,
204         output_gnu,
205 } output_mode = output_tagged;
206
207 enum optval {
208         opt_end = -1,
209         /* ensure we don't collide with shortopts */
210         opt_dummy = CHAR_MAX,
211         /* BSD options */
212         opt_check,
213         opt_passthrough,
214         opt_quiet,
215         opt_reverse,
216         opt_string,
217         opt_time_trial,
218         opt_self_test,
219         /* GNU options */
220         opt_binary,
221         opt_help,
222         opt_ignore_missing,
223         opt_status,
224         opt_strict,
225         opt_tag,
226         opt_text,
227         opt_warn,
228         opt_version,
229         opt_zero,
230         /* Perl options */
231         opt_algorithm,
232         opt_bits,
233         opt_universal,
234 };
235
236 static const struct option bsd_longopts[] = {
237         { "check",              required_argument,      0, opt_check },
238         { "passthrough",        no_argument,            0, opt_passthrough },
239         { "quiet",              no_argument,            0, opt_quiet },
240         { "reverse",            no_argument,            0, opt_reverse },
241         { "string",             required_argument,      0, opt_string },
242         { "time-trial",         no_argument,            0, opt_time_trial },
243         { "self-test",          no_argument,            0, opt_self_test },
244         { }
245 };
246 static const char *bsd_shortopts = "bc:pqrs:tx";
247
248 static const struct option gnu_longopts[] = {
249         { "binary",             no_argument,            0, opt_binary },
250         { "check",              no_argument,            0, opt_check },
251         { "help",               no_argument,            0, opt_help },
252         { "ignore-missing",     no_argument,            0, opt_ignore_missing },
253         { "quiet",              no_argument,            0, opt_quiet },
254         { "status",             no_argument,            0, opt_status },
255         { "strict",             no_argument,            0, opt_strict },
256         { "tag",                no_argument,            0, opt_tag },
257         { "text",               no_argument,            0, opt_text },
258         { "version",            no_argument,            0, opt_version },
259         { "warn",               no_argument,            0, opt_warn },
260         { "zero",               no_argument,            0, opt_zero },
261         { }
262 };
263 static const char *gnu_shortopts = "bctwz";
264
265 static const struct option perl_longopts[] = {
266         { "algorithm",          required_argument,      0, opt_algorithm },
267         { "check",              required_argument,      0, opt_check },
268         { "help",               no_argument,            0, opt_help },
269         { "ignore-missing",     no_argument,            0, opt_ignore_missing },
270         { "quiet",              no_argument,            0, opt_quiet },
271         { "status",             no_argument,            0, opt_status },
272         { "strict",             no_argument,            0, opt_strict },
273         { "tag",                no_argument,            0, opt_tag },
274         { "text",               no_argument,            0, opt_text },
275         { "UNIVERSAL",          no_argument,            0, opt_universal },
276         { "version",            no_argument,            0, opt_version },
277         { "warn",               no_argument,            0, opt_warn },
278         { "01",                 no_argument,            0, opt_bits },
279         { }
280 };
281 static const char *perl_shortopts = "0a:bchqstUvw";
282
283 static void
284 MD5_Update(MD5_CTX *c, const unsigned char *data, size_t len)
285 {
286         MD5Update(c, data, len);
287 }
288
289 struct chksumrec {
290         char    *filename;
291         char    *chksum;
292         struct  chksumrec       *next;
293 };
294
295 static struct chksumrec *head = NULL;
296 static struct chksumrec **next = &head;
297 static unsigned int numrecs;
298
299 #define PADDING 7       /* extra padding for "SHA512t256 (...) = ...\n" style */
300 #define CHKFILELINELEN  (HEX_DIGEST_LENGTH + MAXPATHLEN + PADDING)
301
302 static void
303 gnu_check(const char *checksumsfile)
304 {
305         FILE    *inp;
306         char    *linebuf = NULL;
307         size_t  linecap;
308         ssize_t linelen;
309         int     lineno;
310         char    *filename;
311         char    *hashstr;
312         struct chksumrec        *rec;
313         const char      *digestname;
314         size_t  digestnamelen;
315         size_t  hashstrlen;
316
317         if (strcmp(checksumsfile, "-") == 0)
318                 inp = stdin;
319         else if ((inp = fopen(checksumsfile, "r")) == NULL)
320                 err(1, "%s", checksumsfile);
321         digestname = Algorithm[digest].name;
322         digestnamelen = strlen(digestname);
323         hashstrlen = strlen(*(Algorithm[digest].TestOutput[0]));
324         lineno = 0;
325         linecap = CHKFILELINELEN;
326         while ((linelen = getline(&linebuf, &linecap, inp)) > 0) {
327                 lineno++;
328                 while (linelen > 0 && linebuf[linelen - 1] == '\n')
329                         linelen--;
330                 linebuf[linelen] = '\0';
331                 filename = linebuf + digestnamelen + 2;
332                 hashstr = linebuf + linelen - hashstrlen;
333                 /*
334                  * supported formats:
335                  * BSD: <DigestName> (<Filename>): <Digest>
336                  * GNU: <Digest> [ *U^]<Filename>
337                  */
338                 if ((size_t)linelen >= digestnamelen + hashstrlen + 6 &&
339                     strncmp(linebuf, digestname, digestnamelen) == 0 &&
340                     strncmp(filename - 2, " (", 2) == 0 &&
341                     strncmp(hashstr - 4, ") = ", 4) == 0 &&
342                     strspn(hashstr, "0123456789ABCDEFabcdef") == hashstrlen) {
343                         *(hashstr - 4) = '\0';
344                 } else if ((size_t)linelen >= hashstrlen + 3 &&
345                     strspn(linebuf, "0123456789ABCDEFabcdef") == hashstrlen &&
346                     linebuf[hashstrlen] == ' ') {
347                         linebuf[hashstrlen] = '\0';
348                         hashstr = linebuf;
349                         filename = linebuf + hashstrlen + 1;
350                 } else {
351                         if (wflag) {
352                                 warnx("%s: %d: improperly formatted "
353                                     "%s checksum line",
354                                     checksumsfile, lineno,
355                                     mode == mode_perl ? "SHA" : digestname);
356                         }
357                         malformed++;
358                         continue;
359                 }
360                 rec = malloc(sizeof(*rec));
361                 if (rec == NULL)
362                         errx(1, "malloc failed");
363                 rec->chksum = strdup(hashstr);
364                 rec->filename = strdup(filename);
365                 if (rec->chksum == NULL || rec->filename == NULL)
366                         errx(1, "malloc failed");
367                 rec->next = NULL;
368                 *next = rec;
369                 next = &rec->next;
370                 numrecs++;
371         }
372         if (inp != stdin)
373                 fclose(inp);
374 }
375
376 /* Main driver.
377
378 Arguments (may be any combination):
379   -sstring - digests string
380   -t       - runs time trial
381   -x       - runs test script
382   filename - digests file
383   (none)   - digests standard input
384  */
385 int
386 main(int argc, char *argv[])
387 {
388 #ifdef HAVE_CAPSICUM
389         cap_rights_t    rights;
390 #endif
391         const struct option *longopts;
392         const char *shortopts;
393         FILE   *f;
394         int     i, opt;
395         char   *p, *string = NULL;
396         char    buf[HEX_DIGEST_LENGTH];
397         size_t  len;
398         struct chksumrec        *rec;
399
400         if ((progname = strrchr(argv[0], '/')) == NULL)
401                 progname = argv[0];
402         else
403                 progname++;
404
405         /*
406          * GNU coreutils has a number of programs named *sum. These produce
407          * similar results to the BSD version, but in a different format,
408          * similar to BSD's -r flag. We install links to this program with
409          * ending 'sum' to provide this compatibility. Check here to see if the
410          * name of the program ends in 'sum', set the flag and drop the 'sum' so
411          * the digest lookup works. Also, make -t a nop when running in this mode
412          * since that means 'text file' there (though it's a nop in coreutils
413          * on unix-like systems). The -c flag conflicts, so it's just disabled
414          * in this mode (though in the future it might be implemented).
415          *
416          * We also strive to be compatible with the shasum script which is
417          * included in Perl.  It is roughly equivalent to the GNU offering
418          * but uses a command-line argument to select the algorithm, and
419          * supports only SHA-1 and SHA-2.
420          */
421         len = strlen(progname);
422         if (strcmp(progname, "shasum") == 0) {
423                 mode = mode_perl;
424                 input_mode = input_text;
425                 output_mode = output_gnu;
426                 digest = 1;
427                 longopts = perl_longopts;
428                 shortopts = perl_shortopts;
429         } else if (len > 3 && strcmp(progname + len - 3, "sum") == 0) {
430                 len -= 3;
431                 mode = mode_gnu;
432                 input_mode = input_text;
433                 /*
434                  * The historical behavior in GNU emulation mode is
435                  * output_reverse, however this not true to the original
436                  * and the flag that was used to force the correct output
437                  * was -b, which means something else (input_binary) in
438                  * GNU land.  Switch to the correct behavior.
439                  */
440                 output_mode = output_gnu;
441                 longopts = gnu_longopts;
442                 shortopts = gnu_shortopts;
443         } else {
444                 mode = mode_bsd;
445                 input_mode = input_binary;
446                 output_mode = output_tagged;
447                 longopts = bsd_longopts;
448                 shortopts = bsd_shortopts;
449         }
450
451         if (digest < 0) {
452                 for (digest = 0; Algorithm[digest].progname != NULL; digest++)
453                         if (strncasecmp(Algorithm[digest].progname, progname, len) == 0)
454                                 break;
455
456                 if (Algorithm[digest].progname == NULL)
457                         digest = 0;
458         }
459
460         failed = false;
461         checkAgainst = NULL;
462         checksFailed = 0;
463         skip = false;
464         while ((opt = getopt_long(argc, argv, shortopts, longopts, NULL)) != opt_end)
465                 switch (opt) {
466                 case opt_bits:
467                 case '0':
468                         input_mode = input_bits;
469                         break;
470                 case opt_algorithm:
471                 case 'a':
472                         for (i = 0; Algorithm[i].progname != NULL; i++) {
473                                 if (Algorithm[i].perlname != NULL &&
474                                     strcasecmp(Algorithm[i].perlname, optarg) == 0) {
475                                         digest = i;
476                                         break;
477                                 }
478                         }
479                         if (Algorithm[i].progname == NULL)
480                                 usage(&Algorithm[digest]);
481                         break;
482                 case opt_binary:
483                 case 'b':
484                         /* in BSD mode, -b is now a no-op */
485                         if (mode != mode_bsd)
486                                 input_mode = input_binary;
487                         break;
488                 case opt_check:
489                 case 'c':
490                         cflag = true;
491                         if (mode == mode_bsd)
492                                 checkAgainst = optarg;
493                         break;
494                 case opt_passthrough:
495                 case 'p':
496                         pflag = true;
497                         break;
498                 case opt_quiet:
499                 case 'q':
500                         output_mode = output_bare;
501                         qflag = true;
502                         break;
503                 case opt_reverse:
504                 case 'r':
505                         if (!qflag)
506                                 output_mode = output_reverse;
507                         break;
508                 case opt_status:
509                         sflag = true;
510                         break;
511                 case opt_strict:
512                         strict = 1;
513                         break;
514                 case 's':
515                         if (mode == mode_perl) {
516                                 sflag = true;
517                                 break;
518                         }
519                         /* fall through */
520                 case opt_string:
521                         output_mode = output_bare;
522                         string = optarg;
523                         break;
524                 case opt_tag:
525                         output_mode = output_tagged;
526                         break;
527                 case opt_time_trial:
528                 case opt_text:
529                 case 't':
530                         if (mode == mode_bsd) {
531                                 MDTimeTrial(&Algorithm[digest]);
532                                 skip = true;
533                         } else {
534                                 input_mode = input_text;
535                         }
536                         break;
537                 case opt_universal:
538                 case 'U':
539                         input_mode = input_universal;
540                         break;
541                 case opt_version:
542                         version();
543                         break;
544                 case opt_warn:
545                 case 'w':
546                         wflag = true;
547                         break;
548                 case opt_self_test:
549                 case 'x':
550                         MDTestSuite(&Algorithm[digest]);
551                         skip = true;
552                         break;
553                 case opt_zero:
554                 case 'z':
555                         endl = '\0';
556                         break;
557                 case opt_ignore_missing:
558                         ignoreMissing = true;
559                         break;
560                 default:
561                         usage(&Algorithm[digest]);
562                 }
563         argc -= optind;
564         argv += optind;
565
566 #ifdef HAVE_CAPSICUM
567         if (caph_limit_stdout() < 0 || caph_limit_stderr() < 0)
568                 err(1, "unable to limit rights for stdio");
569 #endif
570
571         if (cflag && mode != mode_bsd) {
572                 /*
573                  * Read digest files into a linked list, then replace argv
574                  * with an array of the filenames from that list.
575                  */
576                 if (argc < 1)
577                         usage(&Algorithm[digest]);
578                 while (argc--)
579                         gnu_check(*argv++);
580                 argc = 0;
581                 argv = calloc(sizeof(char *), numrecs + 1);
582                 for (rec = head; rec != NULL; rec = rec->next) {
583                         argv[argc] = rec->filename;
584                         argc++;
585                 }
586                 argv[argc] = NULL;
587                 rec = head;
588         }
589
590         if (*argv) {
591                 do {
592                         struct stat st;
593                         const char *filename = *argv;
594                         const char *filemode = "rb";
595
596                         if (*filename == '*' ||
597                             *filename == ' ' ||
598                             *filename == 'U' ||
599                             *filename == '^') {
600                                 if (lstat(filename, &st) != 0) {
601                                         input_mode = (int)*filename;
602                                         filename++;
603                                 }
604                         }
605                         if (input_mode == input_text)
606                                 filemode = "r";
607                         if ((f = fopen(filename, filemode)) == NULL) {
608                                 if (errno != ENOENT || !(cflag && ignoreMissing)) {
609                                         warn("%s", filename);
610                                         failed = true;
611                                 }
612                                 if (cflag && mode != mode_bsd)
613                                         rec = rec->next;
614                                 continue;
615                         }
616                         /*
617                          * XXX Enter capability mode on the last argv file.
618                          * When a casper file service or other approach is
619                          * available, switch to that and enter capability mode
620                          * earlier.
621                          */
622                         if (*(argv + 1) == NULL) {
623 #ifdef HAVE_CAPSICUM
624                                 cap_rights_init(&rights, CAP_READ);
625                                 if (caph_rights_limit(fileno(f), &rights) < 0 ||
626                                     caph_enter() < 0)
627                                         err(1, "capsicum");
628 #endif
629                         }
630                         if (cflag && mode != mode_bsd) {
631                                 checkAgainst = rec->chksum;
632                                 rec = rec->next;
633                         }
634                         p = MDInput(&Algorithm[digest], f, buf, false);
635                         (void)fclose(f);
636                         MDOutput(&Algorithm[digest], p, filename);
637                 } while (*++argv);
638         } else if (!cflag && string == NULL && !skip) {
639 #ifdef HAVE_CAPSICUM
640                 if (caph_limit_stdin() < 0 || caph_enter() < 0)
641                         err(1, "capsicum");
642 #endif
643                 if (mode == mode_bsd)
644                         output_mode = output_bare;
645                 p = MDInput(&Algorithm[digest], stdin, buf, pflag);
646                 MDOutput(&Algorithm[digest], p, "-");
647         } else if (string != NULL) {
648                 len = strlen(string);
649                 p = Algorithm[digest].Data(string, len, buf);
650                 MDOutput(&Algorithm[digest], p, string);
651         }
652         if (cflag && mode != mode_bsd) {
653                 if (!sflag && malformed > 1)
654                         warnx("WARNING: %d lines are improperly formatted", malformed);
655                 else if (!sflag && malformed > 0)
656                         warnx("WARNING: %d line is improperly formatted", malformed);
657                 if (!sflag && checksFailed > 1)
658                         warnx("WARNING: %d computed checksums did NOT match", checksFailed);
659                 else if (!sflag && checksFailed > 0)
660                         warnx("WARNING: %d computed checksum did NOT match", checksFailed);
661                 if (checksFailed != 0 || (strict && malformed > 0))
662                         return (1);
663         }
664         if (failed)
665                 return (1);
666         if (checksFailed > 0)
667                 return (2);
668
669         return (0);
670 }
671
672 /*
673  * Common input handling
674  */
675 static char *
676 MDInput(const Algorithm_t *alg, FILE *f, char *buf, bool tee)
677 {
678         char block[4096];
679         DIGEST_CTX context;
680         char *end, *p, *q;
681         size_t len;
682         int bits;
683         uint8_t byte;
684         bool cr = false;
685
686         alg->Init(&context);
687         while ((len = fread(block, 1, sizeof(block), f)) > 0) {
688                 switch (input_mode) {
689                 case input_binary:
690                 case input_text:
691                         if (tee && fwrite(block, 1, len, stdout) != len)
692                                 err(1, "stdout");
693                         alg->Update(&context, block, len);
694                         break;
695                 case input_universal:
696                         end = block + len;
697                         for (p = q = block; p < end; p = q) {
698                                 if (cr) {
699                                         if (*p == '\n')
700                                                 p++;
701                                         if (tee && putchar('\n') == EOF)
702                                                 err(1, "stdout");
703                                         alg->Update(&context, "\n", 1);
704                                         cr = false;
705                                 }
706                                 for (q = p; q < end && *q != '\r'; q++)
707                                         /* nothing */;
708                                 if (q > p) {
709                                         if (tee &&
710                                             fwrite(p, 1, q - p, stdout) !=
711                                             (size_t)(q - p))
712                                                 err(1, "stdout");
713                                         alg->Update(&context, p, q - p);
714                                 }
715                                 if (q < end && *q == '\r') {
716                                         cr = true;
717                                         q++;
718                                 }
719                         }
720                         break;
721                 case input_bits:
722                         end = block + len;
723                         bits = byte = 0;
724                         for (p = block; p < end; p++) {
725                                 if (*p == '0' || *p == '1') {
726                                         byte <<= 1;
727                                         byte |= *p - '0';
728                                         if (++bits == 8) {
729                                                 if (tee && putchar(byte) == EOF)
730                                                         err(1, "stdout");
731                                                 alg->Update(&context, &byte, 1);
732                                                 bits = byte = 0;
733                                         }
734                                 }
735                         }
736                         break;
737                 }
738         }
739         if (ferror(f)) {
740                 alg->End(&context, buf);
741                 return (NULL);
742         }
743         if (cr) {
744                 if (tee && putchar('\n') == EOF)
745                         err(1, "stdout");
746                 alg->Update(&context, "\n", 1);
747         }
748         if (input_mode == input_bits && bits != 0)
749                 errx(1, "input length was not a multiple of 8");
750         return (alg->End(&context, buf));
751 }
752
753 /*
754  * Common output handling
755  */
756 static void
757 MDOutput(const Algorithm_t *alg, char *p, const char *name)
758 {
759         bool checkfailed = false;
760
761         if (p == NULL) {
762                 warn("%s", name);
763                 failed = true;
764         } else if (cflag && mode != mode_bsd) {
765                 checkfailed = strcasecmp(checkAgainst, p) != 0;
766                 if (!sflag && (!qflag || checkfailed))
767                         printf("%s: %s%c", name, checkfailed ? "FAILED" : "OK",
768                             endl);
769         } else {
770                 switch (output_mode) {
771                 case output_bare:
772                         printf("%s", p);
773                         break;
774                 case output_gnu:
775                         printf("%s %c%s", p, input_mode, name);
776                         break;
777                 case output_reverse:
778                         printf("%s %s", p, name);
779                         break;
780                 case output_tagged:
781                         if (mode == mode_perl &&
782                             strncmp(alg->name, "SHA512t", 7) == 0) {
783                                 printf("%.6s/%s", alg->name, alg->name + 7);
784                         } else {
785                                 printf("%s", alg->name);
786                         }
787                         printf(" (%s) = %s", name, p);
788                         break;
789                 }
790                 if (checkAgainst) {
791                         checkfailed = strcasecmp(checkAgainst, p) != 0;
792                         if (!qflag && checkfailed)
793                                 printf(" [ Failed ]");
794                 }
795                 printf("%c", endl);
796         }
797         if (checkfailed)
798                 checksFailed++;
799 }
800
801 /*
802  * Measures the time to digest TEST_BLOCK_COUNT TEST_BLOCK_LEN-byte blocks.
803  */
804 static void
805 MDTimeTrial(const Algorithm_t *alg)
806 {
807         DIGEST_CTX context;
808         struct rusage before, after;
809         struct timeval total;
810         float seconds;
811         unsigned char block[TEST_BLOCK_LEN];
812         unsigned int i;
813         char *p, buf[HEX_DIGEST_LENGTH];
814
815         printf("%s time trial. Digesting %d %d-byte blocks ...",
816             alg->name, TEST_BLOCK_COUNT, TEST_BLOCK_LEN);
817         fflush(stdout);
818
819         /* Initialize block */
820         for (i = 0; i < TEST_BLOCK_LEN; i++)
821                 block[i] = (unsigned char) (i & 0xff);
822
823         /* Start timer */
824         getrusage(RUSAGE_SELF, &before);
825
826         /* Digest blocks */
827         alg->Init(&context);
828         for (i = 0; i < TEST_BLOCK_COUNT; i++)
829                 alg->Update(&context, block, TEST_BLOCK_LEN);
830         p = alg->End(&context, buf);
831
832         /* Stop timer */
833         getrusage(RUSAGE_SELF, &after);
834         timersub(&after.ru_utime, &before.ru_utime, &total);
835         seconds = total.tv_sec + (float) total.tv_usec / 1000000;
836
837         printf(" done\n");
838         printf("Digest = %s", p);
839         printf("\nTime = %f seconds\n", seconds);
840         printf("Speed = %f MiB/second\n", (float) TEST_BLOCK_LEN *
841                 (float) TEST_BLOCK_COUNT / seconds / (1 << 20));
842 }
843 /*
844  * Digests a reference suite of strings and prints the results.
845  */
846
847 static const char *MDTestInput[MDTESTCOUNT] = {
848         "",
849         "a",
850         "abc",
851         "message digest",
852         "abcdefghijklmnopqrstuvwxyz",
853         "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789",
854         "12345678901234567890123456789012345678901234567890123456789012345678901234567890",
855         "MD5 has not yet (2001-09-03) been broken, but sufficient attacks have been made \
856 that its security is in some doubt"
857 };
858
859 const char *MD5TestOutput[MDTESTCOUNT] = {
860         "d41d8cd98f00b204e9800998ecf8427e",
861         "0cc175b9c0f1b6a831c399e269772661",
862         "900150983cd24fb0d6963f7d28e17f72",
863         "f96b697d7cb7938d525a2f31aaf161d0",
864         "c3fcd3d76192e4007dfb496cca67e13b",
865         "d174ab98d277d9f5a5611c2c9f419d9f",
866         "57edf4a22be3c955ac49da2e2107b67a",
867         "b50663f41d44d92171cb9976bc118538"
868 };
869
870 const char *SHA1_TestOutput[MDTESTCOUNT] = {
871         "da39a3ee5e6b4b0d3255bfef95601890afd80709",
872         "86f7e437faa5a7fce15d1ddcb9eaeaea377667b8",
873         "a9993e364706816aba3e25717850c26c9cd0d89d",
874         "c12252ceda8be8994d5fa0290a47231c1d16aae3",
875         "32d10c7b8cf96570ca04ce37f2a19d84240d3a89",
876         "761c457bf73b14d27e9e9265c46f4b4dda11f940",
877         "50abf5706a150990a08b2c5ea40fa0e585554732",
878         "18eca4333979c4181199b7b4fab8786d16cf2846"
879 };
880
881 const char *SHA224_TestOutput[MDTESTCOUNT] = {
882         "d14a028c2a3a2bc9476102bb288234c415a2b01f828ea62ac5b3e42f",
883         "abd37534c7d9a2efb9465de931cd7055ffdb8879563ae98078d6d6d5",
884         "23097d223405d8228642a477bda255b32aadbce4bda0b3f7e36c9da7",
885         "2cb21c83ae2f004de7e81c3c7019cbcb65b71ab656b22d6d0c39b8eb",
886         "45a5f72c39c5cff2522eb3429799e49e5f44b356ef926bcf390dccc2",
887         "bff72b4fcb7d75e5632900ac5f90d219e05e97a7bde72e740db393d9",
888         "b50aecbe4e9bb0b57bc5f3ae760a8e01db24f203fb3cdcd13148046e",
889         "5ae55f3779c8a1204210d7ed7689f661fbe140f96f272ab79e19d470"
890 };
891
892 const char *SHA256_TestOutput[MDTESTCOUNT] = {
893         "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
894         "ca978112ca1bbdcafac231b39a23dc4da786eff8147c4e72b9807785afee48bb",
895         "ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad",
896         "f7846f55cf23e14eebeab5b4e1550cad5b509e3348fbc4efa3a1413d393cb650",
897         "71c480df93d6ae2f1efad1447c66c9525e316218cf51fc8d9ed832f2daf18b73",
898         "db4bfcbd4da0cd85a60c3c37d3fbd8805c77f15fc6b1fdfe614ee0a7c8fdb4c0",
899         "f371bc4a311f2b009eef952dd83ca80e2b60026c8e935592d0f9c308453c813e",
900         "e6eae09f10ad4122a0e2a4075761d185a272ebd9f5aa489e998ff2f09cbfdd9f"
901 };
902
903 const char *SHA384_TestOutput[MDTESTCOUNT] = {
904         "38b060a751ac96384cd9327eb1b1e36a21fdb71114be07434c0cc7bf63f6e1da274edebfe76f65fbd51ad2f14898b95b",
905         "54a59b9f22b0b80880d8427e548b7c23abd873486e1f035dce9cd697e85175033caa88e6d57bc35efae0b5afd3145f31",
906         "cb00753f45a35e8bb5a03d699ac65007272c32ab0eded1631a8b605a43ff5bed8086072ba1e7cc2358baeca134c825a7",
907         "473ed35167ec1f5d8e550368a3db39be54639f828868e9454c239fc8b52e3c61dbd0d8b4de1390c256dcbb5d5fd99cd5",
908         "feb67349df3db6f5924815d6c3dc133f091809213731fe5c7b5f4999e463479ff2877f5f2936fa63bb43784b12f3ebb4",
909         "1761336e3f7cbfe51deb137f026f89e01a448e3b1fafa64039c1464ee8732f11a5341a6f41e0c202294736ed64db1a84",
910         "b12932b0627d1c060942f5447764155655bd4da0c9afa6dd9b9ef53129af1b8fb0195996d2de9ca0df9d821ffee67026",
911         "99428d401bf4abcd4ee0695248c9858b7503853acfae21a9cffa7855f46d1395ef38596fcd06d5a8c32d41a839cc5dfb"
912 };
913
914 const char *SHA512_TestOutput[MDTESTCOUNT] = {
915         "cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e",
916         "1f40fc92da241694750979ee6cf582f2d5d7d28e18335de05abc54d0560e0f5302860c652bf08d560252aa5e74210546f369fbbbce8c12cfc7957b2652fe9a75",
917         "ddaf35a193617abacc417349ae20413112e6fa4e89a97ea20a9eeee64b55d39a2192992a274fc1a836ba3c23a3feebbd454d4423643ce80e2a9ac94fa54ca49f",
918         "107dbf389d9e9f71a3a95f6c055b9251bc5268c2be16d6c13492ea45b0199f3309e16455ab1e96118e8a905d5597b72038ddb372a89826046de66687bb420e7c",
919         "4dbff86cc2ca1bae1e16468a05cb9881c97f1753bce3619034898faa1aabe429955a1bf8ec483d7421fe3c1646613a59ed5441fb0f321389f77f48a879c7b1f1",
920         "1e07be23c26a86ea37ea810c8ec7809352515a970e9253c26f536cfc7a9996c45c8370583e0a78fa4a90041d71a4ceab7423f19c71b9d5a3e01249f0bebd5894",
921         "72ec1ef1124a45b047e8b7c75a932195135bb61de24ec0d1914042246e0aec3a2354e093d76f3048b456764346900cb130d2a4fd5dd16abb5e30bcb850dee843",
922         "e8a835195e039708b13d9131e025f4441dbdc521ce625f245a436dcd762f54bf5cb298d96235e6c6a304e087ec8189b9512cbdf6427737ea82793460c367b9c3"
923 };
924
925 const char *SHA512t224_TestOutput[MDTESTCOUNT] = {
926         "6ed0dd02806fa89e25de060c19d3ac86cabb87d6a0ddd05c333b84f4",
927         "d5cdb9ccc769a5121d4175f2bfdd13d6310e0d3d361ea75d82108327",
928         "4634270f707b6a54daae7530460842e20e37ed265ceee9a43e8924aa",
929         "ad1a4db188fe57064f4f24609d2a83cd0afb9b398eb2fcaeaae2c564",
930         "ff83148aa07ec30655c1b40aff86141c0215fe2a54f767d3f38743d8",
931         "a8b4b9174b99ffc67d6f49be9981587b96441051e16e6dd036b140d3",
932         "ae988faaa47e401a45f704d1272d99702458fea2ddc6582827556dd2",
933         "b3c3b945249b0c8c94aba76ea887bcaad5401665a1fbeb384af4d06b"
934 };
935
936 const char *SHA512t256_TestOutput[MDTESTCOUNT] = {
937         "c672b8d1ef56ed28ab87c3622c5114069bdd3ad7b8f9737498d0c01ecef0967a",
938         "455e518824bc0601f9fb858ff5c37d417d67c2f8e0df2babe4808858aea830f8",
939         "53048e2681941ef99b2e29b76b4c7dabe4c2d0c634fc6d46e0e2f13107e7af23",
940         "0cf471fd17ed69d990daf3433c89b16d63dec1bb9cb42a6094604ee5d7b4e9fb",
941         "fc3189443f9c268f626aea08a756abe7b726b05f701cb08222312ccfd6710a26",
942         "cdf1cc0effe26ecc0c13758f7b4a48e000615df241284185c39eb05d355bb9c8",
943         "2c9fdbc0c90bdd87612ee8455474f9044850241dc105b1e8b94b8ddf5fac9148",
944         "dd095fc859b336c30a52548b3dc59fcc0d1be8616ebcf3368fad23107db2d736"
945 };
946
947 const char *RIPEMD160_TestOutput[MDTESTCOUNT] = {
948         "9c1185a5c5e9fc54612808977ee8f548b2258d31",
949         "0bdc9d2d256b3ee9daae347be6f4dc835a467ffe",
950         "8eb208f7e05d987a9b044a8e98c6b087f15a0bfc",
951         "5d0689ef49d2fae572b881b123a85ffa21595f36",
952         "f71c27109c692c1b56bbdceb5b9d2865b3708dbc",
953         "b0e20b6e3116640286ed3a87a5713079b21f5189",
954         "9b752e45573d4b39f4dbd3323cab82bf63326bfb",
955         "5feb69c6bf7c29d95715ad55f57d8ac5b2b7dd32"
956 };
957
958 const char *SKEIN256_TestOutput[MDTESTCOUNT] = {
959         "c8877087da56e072870daa843f176e9453115929094c3a40c463a196c29bf7ba",
960         "7fba44ff1a31d71a0c1f82e6e82fb5e9ac6c92a39c9185b9951fed82d82fe635",
961         "258bdec343b9fde1639221a5ae0144a96e552e5288753c5fec76c05fc2fc1870",
962         "4d2ce0062b5eb3a4db95bc1117dd8aa014f6cd50fdc8e64f31f7d41f9231e488",
963         "46d8440685461b00e3ddb891b2ecc6855287d2bd8834a95fb1c1708b00ea5e82",
964         "7c5eb606389556b33d34eb2536459528dc0af97adbcd0ce273aeb650f598d4b2",
965         "4def7a7e5464a140ae9c3a80279fbebce4bd00f9faad819ab7e001512f67a10d",
966         "d9c017dbe355f318d036469eb9b5fbe129fc2b5786a9dc6746a516eab6fe0126"
967 };
968
969 const char *SKEIN512_TestOutput[MDTESTCOUNT] = {
970         "bc5b4c50925519c290cc634277ae3d6257212395cba733bbad37a4af0fa06af41fca7903d06564fea7a2d3730dbdb80c1f85562dfcc070334ea4d1d9e72cba7a",
971         "b1cd8d33f61b3737adfd59bb13ad82f4a9548e92f22956a8976cca3fdb7fee4fe91698146c4197cec85d38b83c5d93bdba92c01fd9a53870d0c7f967bc62bdce",
972         "8f5dd9ec798152668e35129496b029a960c9a9b88662f7f9482f110b31f9f93893ecfb25c009baad9e46737197d5630379816a886aa05526d3a70df272d96e75",
973         "15b73c158ffb875fed4d72801ded0794c720b121c0c78edf45f900937e6933d9e21a3a984206933d504b5dbb2368000411477ee1b204c986068df77886542fcc",
974         "23793ad900ef12f9165c8080da6fdfd2c8354a2929b8aadf83aa82a3c6470342f57cf8c035ec0d97429b626c4d94f28632c8f5134fd367dca5cf293d2ec13f8c",
975         "0c6bed927e022f5ddcf81877d42e5f75798a9f8fd3ede3d83baac0a2f364b082e036c11af35fe478745459dd8f5c0b73efe3c56ba5bb2009208d5a29cc6e469c",
976         "2ca9fcffb3456f297d1b5f407014ecb856f0baac8eb540f534b1f187196f21e88f31103128c2f03fcc9857d7a58eb66f9525e2302d88833ee069295537a434ce",
977         "1131f2aaa0e97126c9314f9f968cc827259bbfabced2943bb8c9274448998fb3b78738b4580dd500c76105fd3c03e465e1414f2c29664286b1f79d3e51128125"
978 };
979
980 const char *SKEIN1024_TestOutput[MDTESTCOUNT] = {
981         "0fff9563bb3279289227ac77d319b6fff8d7e9f09da1247b72a0a265cd6d2a62645ad547ed8193db48cff847c06494a03f55666d3b47eb4c20456c9373c86297d630d5578ebd34cb40991578f9f52b18003efa35d3da6553ff35db91b81ab890bec1b189b7f52cb2a783ebb7d823d725b0b4a71f6824e88f68f982eefc6d19c6",
982         "6ab4c4ba9814a3d976ec8bffa7fcc638ceba0544a97b3c98411323ffd2dc936315d13dc93c13c4e88cda6f5bac6f2558b2d8694d3b6143e40d644ae43ca940685cb37f809d3d0550c56cba8036dee729a4f8fb960732e59e64d57f7f7710f8670963cdcdc95b41daab4855fcf8b6762a64b173ee61343a2c7689af1d293eba97",
983         "35a599a0f91abcdb4cb73c19b8cb8d947742d82c309137a7caed29e8e0a2ca7a9ff9a90c34c1908cc7e7fd99bb15032fb86e76df21b72628399b5f7c3cc209d7bb31c99cd4e19465622a049afbb87c03b5ce3888d17e6e667279ec0aa9b3e2712624c01b5f5bbe1a564220bdcf6990af0c2539019f313fdd7406cca3892a1f1f",
984         "ea891f5268acd0fac97467fc1aa89d1ce8681a9992a42540e53babee861483110c2d16f49e73bac27653ff173003e40cfb08516cd34262e6af95a5d8645c9c1abb3e813604d508b8511b30f9a5c1b352aa0791c7d2f27b2706dccea54bc7de6555b5202351751c3299f97c09cf89c40f67187e2521c0fad82b30edbb224f0458",
985         "f23d95c2a25fbcd0e797cd058fec39d3c52d2b5afd7a9af1df934e63257d1d3dcf3246e7329c0f1104c1e51e3d22e300507b0c3b9f985bb1f645ef49835080536becf83788e17fed09c9982ba65c3cb7ffe6a5f745b911c506962adf226e435c42f6f6bc08d288f9c810e807e3216ef444f3db22744441deefa4900982a1371f",
986         "cf3889e8a8d11bfd3938055d7d061437962bc5eac8ae83b1b71c94be201b8cf657fdbfc38674997a008c0c903f56a23feb3ae30e012377f1cfa080a9ca7fe8b96138662653fb3335c7d06595bf8baf65e215307532094cfdfa056bd8052ab792a3944a2adaa47b30335b8badb8fe9eb94fe329cdca04e58bbc530f0af709f469",
987         "cf21a613620e6c119eca31fdfaad449a8e02f95ca256c21d2a105f8e4157048f9fe1e897893ea18b64e0e37cb07d5ac947f27ba544caf7cbc1ad094e675aed77a366270f7eb7f46543bccfa61c526fd628408058ed00ed566ac35a9761d002e629c4fb0d430b2f4ad016fcc49c44d2981c4002da0eecc42144160e2eaea4855a",
988         "e6799b78db54085a2be7ff4c8007f147fa88d326abab30be0560b953396d8802feee9a15419b48a467574e9283be15685ca8a079ee52b27166b64dd70b124b1d4e4f6aca37224c3f2685e67e67baef9f94b905698adc794a09672aba977a61b20966912acdb08c21a2c37001785355dc884751a21f848ab36e590331ff938138"
989 };
990
991 static void
992 MDTestSuite(const Algorithm_t *alg)
993 {
994         int i;
995         char buffer[HEX_DIGEST_LENGTH];
996
997         printf("%s test suite:\n", alg->name);
998         for (i = 0; i < MDTESTCOUNT; i++) {
999                 (*alg->Data)(MDTestInput[i], strlen(MDTestInput[i]), buffer);
1000                 printf("%s (\"%s\") = %s", alg->name, MDTestInput[i], buffer);
1001                 if (strcmp(buffer, (*alg->TestOutput)[i]) == 0) {
1002                         printf(" - verified correct\n");
1003                 } else {
1004                         printf(" - INCORRECT RESULT!\n");
1005                         failed = true;
1006                 }
1007         }
1008 }
1009
1010 static void
1011 usage(const Algorithm_t *alg)
1012 {
1013
1014         switch (mode) {
1015         case mode_gnu:
1016                 fprintf(stderr, "usage: %ssum [-bctwz] [files ...]\n", alg->progname);
1017                 break;
1018         case mode_perl:
1019                 fprintf(stderr, "usage: shasum [-0bchqstUvw] [-a alg] [files ...]\n");
1020                 break;
1021         default:
1022                 fprintf(stderr, "usage: %s [-pqrtx] [-c string] [-s string] [files ...]\n",
1023                     alg->progname);
1024         }
1025         exit(1);
1026 }
1027
1028 static void
1029 version(void)
1030 {
1031         if (mode == mode_gnu)
1032                 printf("%s (FreeBSD) ", progname);
1033         printf("%d.%d\n",
1034             __FreeBSD_version / 100000,
1035             (__FreeBSD_version / 1000) % 100);
1036         exit(0);
1037 }