1/*	$NetBSD: h_hash.c,v 1.1 2011/01/02 22:03:25 pgoyette Exp $	*/
2
3/*-
4 * Copyright (c) 2000 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29/*
30 * Combined MD5/SHA1 time and regression test.
31 */
32
33#include <stdio.h>
34#include <stdlib.h>
35#include <unistd.h>
36#include <string.h>
37#include <md5.h>
38#ifdef __NetBSD__
39#include <sha1.h>
40#endif
41
42#ifdef __FreeBSD__
43#include <sha.h>
44#endif
45
46int mflag, rflag, sflag, tflag;
47
48static void
49usage(void)
50{
51	(void)fprintf(stderr,
52	    "Usage:\t%s -r[ms] < test-file\n"
53	    "\t%s -t[ms]\n",
54	    getprogname(), getprogname());
55	exit(1);
56	/* NOTREACHED */
57}
58
59static void
60hexdump (unsigned char *buf, int len)
61{
62	int i;
63	for (i=0; i<len; i++) {
64		printf("%02x", buf[i]);
65	}
66	printf("\n");
67}
68
69
70static void
71timetest(void)
72{
73	printf("sorry, not yet\n");
74}
75
76#define CHOMP(buf, len, last) 				\
77	if ((len > 0) &&				\
78	    (buf[len-1] == '\n')) {			\
79		buf[len-1] = '\0';			\
80		len--;					\
81		last = 1;				\
82	}
83
84static void
85regress(void)
86{
87	unsigned char buf[1024];
88	unsigned char out[20];
89	int len, outlen, last;
90
91	while (fgets((char *)buf, sizeof(buf), stdin) != NULL) {
92		last = 0;
93
94		len = strlen((char *)buf);
95		CHOMP(buf, len, last);
96		if (mflag) {
97			MD5_CTX ctx;
98
99			MD5Init(&ctx);
100			MD5Update(&ctx, buf, len);
101			while (!last &&
102			    fgets((char *)buf, sizeof(buf), stdin) != NULL) {
103				len = strlen((char *)buf);
104				CHOMP(buf, len, last);
105				MD5Update(&ctx, buf, len);
106			}
107			MD5Final(out, &ctx);
108			outlen = 16;
109		} else {
110#ifdef __FreeBSD__
111			SHA_CTX ctx;
112
113			SHA1_Init(&ctx);
114			SHA1_Update(&ctx, buf, len);
115#else
116			SHA1_CTX ctx;
117
118			SHA1Init(&ctx);
119			SHA1Update(&ctx, buf, len);
120#endif
121			while (!last &&
122			    fgets((char *)buf, sizeof(buf), stdin) != NULL) {
123				len = strlen((char *)buf);
124				CHOMP(buf, len, last);
125#ifdef __FreeBSD__
126				SHA1_Update(&ctx, buf, len);
127#else
128				SHA1Update(&ctx, buf, len);
129#endif
130			}
131#ifdef __FreeBSD__
132			SHA1_Final(out, &ctx);
133#else
134			SHA1Final(out, &ctx);
135#endif
136			outlen = 20;
137		}
138		hexdump(out, outlen);
139	}
140}
141
142int
143main(int argc, char **argv)
144{
145	int ch;
146
147	while ((ch = getopt(argc, argv, "mrst")) != -1)
148		switch (ch) {
149		case 'm':
150			mflag = 1;
151			break;
152		case 'r':
153			rflag = 1;
154			break;
155		case 's':
156			sflag = 1;
157			break;
158		case 't':
159			tflag = 1;
160			break;
161		case '?':
162		default:
163			usage();
164		}
165	argc -= optind;
166	argv += optind;
167	if (argc > 0)
168		usage();
169
170	if (!(mflag || sflag))
171		mflag = 1;
172
173	if ((mflag ^ sflag) != 1)
174		usage();
175
176	if ((tflag ^ rflag) != 1)
177		usage();
178
179	if (tflag)
180		timetest();
181
182	if (rflag)
183		regress();
184
185	exit(0);
186
187}
188