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 } 215