mddriver.c
上传用户:zfm_0418
上传日期:2013-03-09
资源大小:7k
文件大小:5k
源码类别:

加密解密

开发平台:

C/C++

  1. /* MDDRIVER.C - test driver for MD2, MD4 and MD5
  2.  */
  3. /* Copyright (C) 1990-2, RSA Data Security, Inc. Created 1990. All
  4. rights reserved.
  5. RSA Data Security, Inc. makes no representations concerning either
  6. the merchantability of this software or the suitability of this
  7. software for any particular purpose. It is provided "as is"
  8. without express or implied warranty of any kind.
  9. These notices must be retained in any copies of any part of this
  10. documentation and/or software.
  11.  */
  12. /* The following makes MD default to MD5 if it has not already been
  13.   defined with C compiler flags.
  14.  */
  15. #ifndef MD
  16. #define MD 5
  17. #endif
  18. #include <stdio.h>
  19. #include <time.h>
  20. #include <string.h>
  21. #include "global.h"
  22. #if MD == 2
  23. #include "md2.h"
  24. #endif
  25. #if MD == 4
  26. #include "md4.h"
  27. #endif
  28. #if MD == 5
  29. #include "md5.h"
  30. #endif
  31. /* Length of test block, number of test blocks.
  32.  */
  33. #define TEST_BLOCK_LEN 1000
  34. #define TEST_BLOCK_COUNT 1000
  35. static void MDString PROTO_LIST ((char *));
  36. static void MDTimeTrial PROTO_LIST ((void));
  37. static void MDTestSuite PROTO_LIST ((void));
  38. static void MDFile PROTO_LIST ((char *));
  39. static void MDFilter PROTO_LIST ((void));
  40. static void MDPrint PROTO_LIST ((unsigned char [16]));
  41. #if MD == 2
  42. #define MD5_CTX MD2_CTX
  43. #define MDInit MD2Init
  44. #define MDUpdate MD2Update
  45. #define MDFinal MD2Final
  46. #endif
  47. #if MD == 4
  48. #define MD5_CTX MD4_CTX
  49. #define MDInit MD4Init
  50. #define MDUpdate MD4Update
  51. #define MDFinal MD4Final
  52. #endif
  53. #if MD == 5
  54. #define MD5_CTX MD5_CTX
  55. #define MDInit MD5Init
  56. #define MDUpdate MD5Update
  57. #define MDFinal MD5Final
  58. #endif
  59. /* Main driver.
  60. Arguments (may be any combination):
  61.   -sstring - digests string
  62.   -t       - runs time trial
  63.   -x       - runs test script
  64.   filename - digests file
  65.   (none)   - digests standard input
  66.  */
  67. int main (argc, argv)
  68. int argc;
  69. char *argv[];
  70. {
  71.   int i;
  72.   if (argc > 1)
  73.  for (i = 1; i < argc; i++)
  74.    if (argv[i][0] == '-' && argv[i][1] == 's')
  75.      MDString (argv[i] + 2);
  76.    else if (strcmp (argv[i], "-t") == 0)
  77.      MDTimeTrial ();
  78.    else if (strcmp (argv[i], "-x") == 0)
  79.      MDTestSuite ();
  80.    else
  81.      MDFile (argv[i]);
  82.   else
  83.  MDFilter ();
  84.   return (0);
  85. }
  86. /* Digests a string and prints the result.
  87.  */
  88. static void MDString (string)
  89. char *string;
  90. {
  91.   MD5_CTX context;
  92.   unsigned char digest[16];
  93.   unsigned int len = strlen (string);
  94.   MDInit (&context);
  95.   MDUpdate (&context, string, len);
  96.   MDFinal (digest, &context);
  97.   printf ("MD%d ("%s") = ", MD, string);
  98.   MDPrint (digest);
  99.   printf ("n");
  100. }
  101. /* Measures the time to digest TEST_BLOCK_COUNT TEST_BLOCK_LEN-byte
  102.   blocks.
  103.  */
  104. static void MDTimeTrial ()
  105. {
  106.   MD5_CTX context;
  107.   time_t endTime, startTime;
  108.   unsigned char block[TEST_BLOCK_LEN], digest[16];
  109.   unsigned int i;
  110.   printf
  111.  ("MD%d time trial. Digesting %d %d-byte blocks ...", MD,
  112.   TEST_BLOCK_LEN, TEST_BLOCK_COUNT);
  113.   /* Initialize block */
  114.   for (i = 0; i < TEST_BLOCK_LEN; i++)
  115.  block[i] = (unsigned char)(i & 0xff);
  116.   /* Start timer */
  117.   time (&startTime);
  118.   /* Digest blocks */
  119.   MDInit (&context);
  120.   for (i = 0; i < TEST_BLOCK_COUNT; i++)
  121.  MDUpdate (&context, block, TEST_BLOCK_LEN);
  122.   MDFinal (digest, &context);
  123.   /* Stop timer */
  124.   time (&endTime);
  125.   printf (" donen");
  126.   printf ("Digest = ");
  127.   MDPrint (digest);
  128.   printf ("nTime = %ld secondsn", (long)(endTime-startTime));
  129.   printf
  130.  ("Speed = %ld bytes/secondn",
  131.   (long)TEST_BLOCK_LEN * (long)TEST_BLOCK_COUNT/(endTime-startTime));
  132. }
  133. /* Digests a reference suite of strings and prints the results.
  134.  */
  135. static void MDTestSuite ()
  136. {
  137.   printf ("MD%d test suite:n", MD);
  138.   MDString ("");
  139.   MDString ("a");
  140.   MDString ("abc");
  141.   MDString ("message digest");
  142.   MDString ("abcdefghijklmnopqrstuvwxyz");
  143.   MDString
  144.  ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789");
  145.   MDString
  146.  ("1234567890123456789012345678901234567890
  147. 1234567890123456789012345678901234567890");
  148. }
  149. /* Digests a file and prints the result.
  150.  */
  151. static void MDFile (filename)
  152. char *filename;
  153. {
  154.   FILE *file;
  155.   MD5_CTX context;
  156.   int len;
  157.   unsigned char buffer[1024], digest[16];
  158.   if ((file = fopen (filename, "rb")) == NULL)
  159.  printf ("%s can't be openedn", filename);
  160.   else {
  161.  MDInit (&context);
  162.  while (len = fread (buffer, 1, 1024, file))
  163.    MDUpdate (&context, buffer, len);
  164.  MDFinal (digest, &context);
  165.  fclose (file);
  166.  printf ("MD%d (%s) = ", MD, filename);
  167.  MDPrint (digest);
  168.  printf ("n");
  169.   }
  170. }
  171. /* Digests the standard input and prints the result.
  172.  */
  173. static void MDFilter ()
  174. {
  175.   MD5_CTX context;
  176.   int len;
  177.   unsigned char buffer[16], digest[16];
  178.   MDInit (&context);
  179.   while (len = fread (buffer, 1, 16, stdin))
  180.  MDUpdate (&context, buffer, len);
  181.   MDFinal (digest, &context);
  182.   MDPrint (digest);
  183.   printf ("n");
  184. }
  185. /* Prints a message digest in hexadecimal.
  186.  */
  187. static void MDPrint (digest)
  188. unsigned char digest[16];
  189. {
  190.   unsigned int i;
  191.   for (i = 0; i < 16; i++)
  192.  printf ("%02x", digest[i]);
  193. }