GC !__DragonFly__ section.
[dragonfly.git] / sbin / md5 / md5.c
1 /*
2  * Derived from:
3  *
4  * MDDRIVER.C - test driver for MD2, MD4 and MD5
5  *
6  * $FreeBSD: src/sbin/md5/md5.c,v 1.20.2.5 2001/12/26 09:44:56 phk Exp $
7  * $DragonFly: src/sbin/md5/md5.c,v 1.2 2003/06/17 04:27:33 dillon Exp $
8  */
9
10 /*
11  *  Copyright (C) 1990-2, RSA Data Security, Inc. Created 1990. All
12  *  rights reserved.
13  *
14  *  RSA Data Security, Inc. makes no representations concerning either
15  *  the merchantability of this software or the suitability of this
16  *  software for any particular purpose. It is provided "as is"
17  *  without express or implied warranty of any kind.
18  *
19  *  These notices must be retained in any copies of any part of this
20  *  documentation and/or software.
21  */
22
23 #include <sys/types.h>
24 #include <err.h>
25 #include <md5.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <time.h>
30 #include <unistd.h>
31
32 /*
33  * Length of test block, number of test blocks.
34  */
35 #define TEST_BLOCK_LEN 10000
36 #define TEST_BLOCK_COUNT 100000
37
38 int qflag;
39 int rflag;
40 int sflag;
41
42 static void MDString(const char *);
43 static void MDTimeTrial(void);
44 static void MDTestSuite(void);
45 static void MDFilter(int);
46 static void usage(void);
47
48 /* Main driver.
49
50 Arguments (may be any combination):
51   -sstring - digests string
52   -t       - runs time trial
53   -x       - runs test script
54   filename - digests file
55   (none)   - digests standard input
56  */
57 int
58 main(int argc, char *argv[])
59 {
60         int     ch;
61         char   *p;
62         char    buf[33];
63
64         while ((ch = getopt(argc, argv, "pqrs:tx")) != -1)
65                 switch (ch) {
66                 case 'p':
67                         MDFilter(1);
68                         break;
69                 case 'q':
70                         qflag = 1;
71                         break;
72                 case 'r':
73                         rflag = 1;
74                         break;
75                 case 's':
76                         sflag = 1;
77                         MDString(optarg);
78                         break;
79                 case 't':
80                         MDTimeTrial();
81                         break;
82                 case 'x':
83                         MDTestSuite();
84                         break;
85                 default:
86                         usage();
87                 }
88         argc -= optind;
89         argv += optind;
90
91         if (*argv) {
92                 do {
93                         p = MD5File(*argv, buf);
94                         if (!p)
95                                 warn("%s", *argv);
96                         else
97                                 if (qflag)
98                                         printf("%s\n", p);
99                                 else if (rflag)
100                                         printf("%s %s\n", p, *argv);
101                                 else
102                                         printf("MD5 (%s) = %s\n", *argv, p);
103                 } while (*++argv);
104         } else if (!sflag && (optind == 1 || qflag || rflag))
105                 MDFilter(0);
106
107         return (0);
108 }
109 /*
110  * Digests a string and prints the result.
111  */
112 static void
113 MDString(const char *string)
114 {
115         size_t len = strlen(string);
116         char buf[33];
117
118         if (qflag)
119                 printf("%s\n", MD5Data(string, len, buf));
120         else if (rflag)
121                 printf("%s \"%s\"\n", MD5Data(string, len, buf), string);
122         else
123                 printf("MD5 (\"%s\") = %s\n", string, MD5Data(string, len, buf));
124 }
125 /*
126  * Measures the time to digest TEST_BLOCK_COUNT TEST_BLOCK_LEN-byte blocks.
127  */
128 static void
129 MDTimeTrial(void)
130 {
131         MD5_CTX context;
132         time_t  endTime, startTime;
133         unsigned char block[TEST_BLOCK_LEN];
134         unsigned int i;
135         char   *p, buf[33];
136
137         printf
138             ("MD5 time trial. Digesting %d %d-byte blocks ...",
139             TEST_BLOCK_COUNT, TEST_BLOCK_LEN);
140         fflush(stdout);
141
142         /* Initialize block */
143         for (i = 0; i < TEST_BLOCK_LEN; i++)
144                 block[i] = (unsigned char) (i & 0xff);
145
146         /* Start timer */
147         time(&startTime);
148
149         /* Digest blocks */
150         MD5Init(&context);
151         for (i = 0; i < TEST_BLOCK_COUNT; i++)
152                 MD5Update(&context, block, TEST_BLOCK_LEN);
153         p = MD5End(&context,buf);
154
155         /* Stop timer */
156         time(&endTime);
157
158         printf(" done\n");
159         printf("Digest = %s", p);
160         printf("\nTime = %ld seconds\n", (long) (endTime - startTime));
161         /* Be careful that endTime-startTime is not zero. (Bug fix from Ric
162          * Anderson, ric@Artisoft.COM.) */
163         printf
164             ("Speed = %ld bytes/second\n",
165             (long) TEST_BLOCK_LEN * (long) TEST_BLOCK_COUNT / ((endTime - startTime) != 0 ? (long)(endTime - startTime) : 1));
166 }
167 /*
168  * Digests a reference suite of strings and prints the results.
169  */
170 static void
171 MDTestSuite(void)
172 {
173
174         printf("MD5 test suite:\n");
175
176         MDString("");
177         MDString("a");
178         MDString("abc");
179         MDString("message digest");
180         MDString("abcdefghijklmnopqrstuvwxyz");
181         MDString
182             ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789");
183         MDString
184             ("1234567890123456789012345678901234567890\
185 1234567890123456789012345678901234567890");
186 }
187
188 /*
189  * Digests the standard input and prints the result.
190  */
191 static void
192 MDFilter(int tee)
193 {
194         MD5_CTX context;
195         unsigned int len;
196         unsigned char buffer[BUFSIZ];
197         char buf[33];
198
199         MD5Init(&context);
200         while ((len = fread(buffer, 1, BUFSIZ, stdin))) {
201                 if (tee && len != fwrite(buffer, 1, len, stdout))
202                         err(1, "stdout");
203                 MD5Update(&context, buffer, len);
204         }
205         printf("%s\n", MD5End(&context,buf));
206 }
207
208 static void
209 usage(void)
210 {
211
212         fprintf(stderr, "usage: md5 [-pqrtx] [-s string] [files ...]\n");
213         exit(1);
214 }