md5.c
上传用户:wxp200602
上传日期:2007-10-30
资源大小:4028k
文件大小:16k
源码类别:

SNMP编程

开发平台:

Unix_Linux

  1. /*
  2.  * ** **************************************************************************
  3.  * ** md5.c -- Implementation of MD5 Message Digest Algorithm                 **
  4.  * ** Updated: 2/16/90 by Ronald L. Rivest                                    **
  5.  * ** (C) 1990 RSA Data Security, Inc.                                        **
  6.  * ** **************************************************************************
  7.  */
  8. /*
  9.  * ** To use MD5:
  10.  * **   -- Include md5.h in your program
  11.  * **   -- Declare an MDstruct MD to hold the state of the digest computation.
  12.  * **   -- Initialize MD using MDbegin(&MD)
  13.  * **   -- For each full block (64 bytes) X you wish to process, call
  14.  * **          MDupdate(&MD,X,512)
  15.  * **      (512 is the number of bits in a full block.)
  16.  * **   -- For the last block (less than 64 bytes) you wish to process,
  17.  * **          MDupdate(&MD,X,n)
  18.  * **      where n is the number of bits in the partial block. A partial
  19.  * **      block terminates the computation, so every MD computation should
  20.  * **      terminate by processing a partial block, even if it has n = 0.
  21.  * **   -- The message digest is available in MD.buffer[0] ... MD.buffer[3].
  22.  * **      (Least-significant byte of each word should be output first.)
  23.  * **   -- You can print out the digest using MDprint(&MD)
  24.  */
  25. /*
  26.  * Implementation notes:
  27.  * ** This implementation assumes that ints are 32-bit quantities.
  28.  * ** If the machine stores the least-significant byte of an int in the
  29.  * ** least-addressed byte (eg., VAX and 8086), then LOWBYTEFIRST should be
  30.  * ** set to TRUE.  Otherwise (eg., SUNS), LOWBYTEFIRST should be set to
  31.  * ** FALSE.  Note that on machines with LOWBYTEFIRST FALSE the routine
  32.  * ** MDupdate modifies has a side-effect on its input array (the order of bytes
  33.  * ** in each word are reversed).  If this is undesired a call to MDreverse(X) can
  34.  * ** reverse the bytes of X back into order after each call to MDupdate.
  35.  */
  36. /*
  37.  * code uses WORDS_BIGENDIAN defined by configure now  -- WH 9/27/95 
  38.  */
  39. /*
  40.  * Compile-time includes 
  41.  */
  42. #include <net-snmp/net-snmp-config.h>
  43. #ifndef DISABLE_MD5
  44. #include <stdio.h>
  45. #include <sys/types.h>
  46. #if HAVE_STRING_H
  47. #include <string.h>
  48. #else
  49. #include <strings.h>
  50. #endif
  51. #if HAVE_WINSOCK_H
  52. #include <winsock.h>
  53. #endif
  54. #if HAVE_STDLIB_H
  55. #include <stdlib.h>
  56. #endif
  57. #include <net-snmp/utilities.h>
  58. #include <net-snmp/library/md5.h>
  59. /*
  60.  * Compile-time declarations of MD5 ``magic constants''.
  61.  */
  62. #define I0  0x67452301          /* Initial values for MD buffer */
  63. #define I1  0xefcdab89
  64. #define I2  0x98badcfe
  65. #define I3  0x10325476
  66. #define fs1  7                  /* round 1 shift amounts */
  67. #define fs2 12
  68. #define fs3 17
  69. #define fs4 22
  70. #define gs1  5                  /* round 2 shift amounts */
  71. #define gs2  9
  72. #define gs3 14
  73. #define gs4 20
  74. #define hs1  4                  /* round 3 shift amounts */
  75. #define hs2 11
  76. #define hs3 16
  77. #define hs4 23
  78. #define is1  6                  /* round 4 shift amounts */
  79. #define is2 10
  80. #define is3 15
  81. #define is4 21
  82. /*
  83.  * Compile-time macro declarations for MD5.
  84.  * ** Note: The ``rot'' operator uses the variable ``tmp''.
  85.  * ** It assumes tmp is declared as unsigned int, so that the >>
  86.  * ** operator will shift in zeros rather than extending the sign bit.
  87.  */
  88. #define f(X,Y,Z)             ((X&Y) | ((~X)&Z))
  89. #define g(X,Y,Z)             ((X&Z) | (Y&(~Z)))
  90. #define h(X,Y,Z)             (X^Y^Z)
  91. #define i_(X,Y,Z)            (Y ^ ((X) | (~Z)))
  92. #define rot(X,S)             (tmp=X,(tmp<<S) | (tmp>>(32-S)))
  93. #define ff(A,B,C,D,i,s,lp)   A = rot((A + f(B,C,D) + X[i] + lp),s) + B
  94. #define gg(A,B,C,D,i,s,lp)   A = rot((A + g(B,C,D) + X[i] + lp),s) + B
  95. #define hh(A,B,C,D,i,s,lp)   A = rot((A + h(B,C,D) + X[i] + lp),s) + B
  96. #define ii(A,B,C,D,i,s,lp)   A = rot((A + i_(B,C,D) + X[i] + lp),s) + B
  97. #ifdef STDC_HEADERS
  98. #define Uns(num) num##U
  99. #else
  100. #define Uns(num) num
  101. #endif                          /* STDC_HEADERS */
  102. void            MDreverse(unsigned int *);
  103. static void     MDblock(MDptr, unsigned int *);
  104. #ifdef SNMP_TESTING_CODE
  105. /*
  106.  * MDprint(MDp)
  107.  * ** Print message digest buffer MDp as 32 hexadecimal digits.
  108.  * ** Order is from low-order byte of buffer[0] to high-order byte of buffer[3].
  109.  * ** Each byte is printed with high-order hexadecimal digit first.
  110.  * ** This is a user-callable routine.
  111.  */
  112. void
  113. MDprint(MDptr MDp)
  114. {
  115.     int             i, j;
  116.     for (i = 0; i < 4; i++)
  117.         for (j = 0; j < 32; j = j + 8)
  118.             printf("%02x", (MDp->buffer[i] >> j) & 0xFF);
  119.     printf("n");
  120.     fflush(stdout);
  121. }
  122. #endif                          /* SNMP_TESTING_CODE */
  123. /*
  124.  * MDbegin(MDp)
  125.  * ** Initialize message digest buffer MDp. 
  126.  * ** This is a user-callable routine.
  127.  */
  128. void
  129. MDbegin(MDptr MDp)
  130. {
  131.     int             i;
  132.     MDp->buffer[0] = I0;
  133.     MDp->buffer[1] = I1;
  134.     MDp->buffer[2] = I2;
  135.     MDp->buffer[3] = I3;
  136.     for (i = 0; i < 8; i++)
  137.         MDp->count[i] = 0;
  138.     MDp->done = 0;
  139. }
  140. /*
  141.  * MDreverse(X)
  142.  * ** Reverse the byte-ordering of every int in X.
  143.  * ** Assumes X is an array of 16 ints.
  144.  * ** The macro revx reverses the byte-ordering of the next word of X.
  145.  */
  146. #define revx { t = (*X << 16) | (*X >> 16); 
  147.        *X++ = ((t & 0xFF00FF00) >> 8) | ((t & 0x00FF00FF) << 8); }
  148. void
  149. MDreverse(unsigned int *X)
  150. {
  151.     register unsigned int t;
  152.     revx;
  153.     revx;
  154.     revx;
  155.     revx;
  156.     revx;
  157.     revx;
  158.     revx;
  159.     revx;
  160.     revx;
  161.     revx;
  162.     revx;
  163.     revx;
  164.     revx;
  165.     revx;
  166.     revx;
  167.     revx;
  168. }
  169. /*
  170.  * MDblock(MDp,X)
  171.  * ** Update message digest buffer MDp->buffer using 16-word data block X.
  172.  * ** Assumes all 16 words of X are full of data.
  173.  * ** Does not update MDp->count.
  174.  * ** This routine is not user-callable. 
  175.  */
  176. static void
  177. MDblock(MDptr MDp, unsigned int *X)
  178. {
  179.     register unsigned int tmp, A, B, C, D;      /* hpux sysv sun */
  180. #ifdef WORDS_BIGENDIAN
  181.     MDreverse(X);
  182. #endif
  183.     A = MDp->buffer[0];
  184.     B = MDp->buffer[1];
  185.     C = MDp->buffer[2];
  186.     D = MDp->buffer[3];
  187.     /*
  188.      * Update the message digest buffer 
  189.      */
  190.     ff(A, B, C, D, 0, fs1, Uns(3614090360));    /* Round 1 */
  191.     ff(D, A, B, C, 1, fs2, Uns(3905402710));
  192.     ff(C, D, A, B, 2, fs3, Uns(606105819));
  193.     ff(B, C, D, A, 3, fs4, Uns(3250441966));
  194.     ff(A, B, C, D, 4, fs1, Uns(4118548399));
  195.     ff(D, A, B, C, 5, fs2, Uns(1200080426));
  196.     ff(C, D, A, B, 6, fs3, Uns(2821735955));
  197.     ff(B, C, D, A, 7, fs4, Uns(4249261313));
  198.     ff(A, B, C, D, 8, fs1, Uns(1770035416));
  199.     ff(D, A, B, C, 9, fs2, Uns(2336552879));
  200.     ff(C, D, A, B, 10, fs3, Uns(4294925233));
  201.     ff(B, C, D, A, 11, fs4, Uns(2304563134));
  202.     ff(A, B, C, D, 12, fs1, Uns(1804603682));
  203.     ff(D, A, B, C, 13, fs2, Uns(4254626195));
  204.     ff(C, D, A, B, 14, fs3, Uns(2792965006));
  205.     ff(B, C, D, A, 15, fs4, Uns(1236535329));
  206.     gg(A, B, C, D, 1, gs1, Uns(4129170786));    /* Round 2 */
  207.     gg(D, A, B, C, 6, gs2, Uns(3225465664));
  208.     gg(C, D, A, B, 11, gs3, Uns(643717713));
  209.     gg(B, C, D, A, 0, gs4, Uns(3921069994));
  210.     gg(A, B, C, D, 5, gs1, Uns(3593408605));
  211.     gg(D, A, B, C, 10, gs2, Uns(38016083));
  212.     gg(C, D, A, B, 15, gs3, Uns(3634488961));
  213.     gg(B, C, D, A, 4, gs4, Uns(3889429448));
  214.     gg(A, B, C, D, 9, gs1, Uns(568446438));
  215.     gg(D, A, B, C, 14, gs2, Uns(3275163606));
  216.     gg(C, D, A, B, 3, gs3, Uns(4107603335));
  217.     gg(B, C, D, A, 8, gs4, Uns(1163531501));
  218.     gg(A, B, C, D, 13, gs1, Uns(2850285829));
  219.     gg(D, A, B, C, 2, gs2, Uns(4243563512));
  220.     gg(C, D, A, B, 7, gs3, Uns(1735328473));
  221.     gg(B, C, D, A, 12, gs4, Uns(2368359562));
  222.     hh(A, B, C, D, 5, hs1, Uns(4294588738));    /* Round 3 */
  223.     hh(D, A, B, C, 8, hs2, Uns(2272392833));
  224.     hh(C, D, A, B, 11, hs3, Uns(1839030562));
  225.     hh(B, C, D, A, 14, hs4, Uns(4259657740));
  226.     hh(A, B, C, D, 1, hs1, Uns(2763975236));
  227.     hh(D, A, B, C, 4, hs2, Uns(1272893353));
  228.     hh(C, D, A, B, 7, hs3, Uns(4139469664));
  229.     hh(B, C, D, A, 10, hs4, Uns(3200236656));
  230.     hh(A, B, C, D, 13, hs1, Uns(681279174));
  231.     hh(D, A, B, C, 0, hs2, Uns(3936430074));
  232.     hh(C, D, A, B, 3, hs3, Uns(3572445317));
  233.     hh(B, C, D, A, 6, hs4, Uns(76029189));
  234.     hh(A, B, C, D, 9, hs1, Uns(3654602809));
  235.     hh(D, A, B, C, 12, hs2, Uns(3873151461));
  236.     hh(C, D, A, B, 15, hs3, Uns(530742520));
  237.     hh(B, C, D, A, 2, hs4, Uns(3299628645));
  238.     ii(A, B, C, D, 0, is1, Uns(4096336452));    /* Round 4 */
  239.     ii(D, A, B, C, 7, is2, Uns(1126891415));
  240.     ii(C, D, A, B, 14, is3, Uns(2878612391));
  241.     ii(B, C, D, A, 5, is4, Uns(4237533241));
  242.     ii(A, B, C, D, 12, is1, Uns(1700485571));
  243.     ii(D, A, B, C, 3, is2, Uns(2399980690));
  244.     ii(C, D, A, B, 10, is3, Uns(4293915773));
  245.     ii(B, C, D, A, 1, is4, Uns(2240044497));
  246.     ii(A, B, C, D, 8, is1, Uns(1873313359));
  247.     ii(D, A, B, C, 15, is2, Uns(4264355552));
  248.     ii(C, D, A, B, 6, is3, Uns(2734768916));
  249.     ii(B, C, D, A, 13, is4, Uns(1309151649));
  250.     ii(A, B, C, D, 4, is1, Uns(4149444226));
  251.     ii(D, A, B, C, 11, is2, Uns(3174756917));
  252.     ii(C, D, A, B, 2, is3, Uns(718787259));
  253.     ii(B, C, D, A, 9, is4, Uns(3951481745));
  254.     MDp->buffer[0] += A;
  255.     MDp->buffer[1] += B;
  256.     MDp->buffer[2] += C;
  257.     MDp->buffer[3] += D;
  258. #ifdef WORDS_BIGENDIAN
  259.     MDreverse(X);
  260. #endif
  261. }
  262. /*
  263.  * MDupdate(MDp,X,count)
  264.  * ** Input: MDp -- an MDptr
  265.  * **        X -- a pointer to an array of unsigned characters.
  266.  * **        count -- the number of bits of X to use.
  267.  * **                 (if not a multiple of 8, uses high bits of last byte.)
  268.  * ** Update MDp using the number of bits of X given by count.
  269.  * ** This is the basic input routine for an MD5 user.
  270.  * ** The routine completes the MD computation when count < 512, so
  271.  * ** every MD computation should end with one call to MDupdate with a
  272.  * ** count less than 512.  A call with count 0 will be ignored if the
  273.  * ** MD has already been terminated (done != 0), so an extra call with count
  274.  * ** 0 can be given as a ``courtesy close'' to force termination if desired.
  275.  * ** Returns : 0 if processing succeeds or was already done;
  276.  * **          -1 if processing was already done
  277.  * **          -2 if count was too large
  278.  */
  279. int
  280. MDupdate(MDptr MDp, unsigned char *X, unsigned int count)
  281. {
  282.     unsigned int    i, tmp, bit, byte, mask;
  283.     unsigned char   XX[64];
  284.     unsigned char  *p;
  285.     /*
  286.      * return with no error if this is a courtesy close with count
  287.      * ** zero and MDp->done is true.
  288.      */
  289.     if (count == 0 && MDp->done)
  290.         return 0;
  291.     /*
  292.      * check to see if MD is already done and report error 
  293.      */
  294.     if (MDp->done) {
  295.         return -1;
  296.     }
  297.     /*
  298.      * if (MDp->done) { fprintf(stderr,"nError: MDupdate MD already done."); return; }
  299.      */
  300.     /*
  301.      * Add count to MDp->count 
  302.      */
  303.     tmp = count;
  304.     p = MDp->count;
  305.     while (tmp) {
  306.         tmp += *p;
  307.         *p++ = tmp;
  308.         tmp = tmp >> 8;
  309.     }
  310.     /*
  311.      * Process data 
  312.      */
  313.     if (count == 512) {         /* Full block of data to handle */
  314.         MDblock(MDp, (unsigned int *) X);
  315.     } else if (count > 512)     /* Check for count too large */
  316.         return -2;
  317.     /*
  318.      * { fprintf(stderr,"nError: MDupdate called with illegal count value %d.",count);
  319.      * return;
  320.      * }
  321.      */
  322.     else {                      /* partial block -- must be last block so finish up */
  323.         /*
  324.          * Find out how many bytes and residual bits there are 
  325.          */
  326.         int             copycount;
  327.         byte = count >> 3;
  328.         bit = count & 7;
  329.         copycount = byte;
  330.         if (bit)
  331.             copycount++;
  332.         /*
  333.          * Copy X into XX since we need to modify it 
  334.          */
  335.         memset(XX, 0, sizeof(XX));
  336.         memcpy(XX, X, copycount);
  337.         /*
  338.          * Add padding '1' bit and low-order zeros in last byte 
  339.          */
  340.         mask = ((unsigned long) 1) << (7 - bit);
  341.         XX[byte] = (XX[byte] | mask) & ~(mask - 1);
  342.         /*
  343.          * If room for bit count, finish up with this block 
  344.          */
  345.         if (byte <= 55) {
  346.             for (i = 0; i < 8; i++)
  347.                 XX[56 + i] = MDp->count[i];
  348.             MDblock(MDp, (unsigned int *) XX);
  349.         } else {                /* need to do two blocks to finish up */
  350.             MDblock(MDp, (unsigned int *) XX);
  351.             for (i = 0; i < 56; i++)
  352.                 XX[i] = 0;
  353.             for (i = 0; i < 8; i++)
  354.                 XX[56 + i] = MDp->count[i];
  355.             MDblock(MDp, (unsigned int *) XX);
  356.         }
  357.         /*
  358.          * Set flag saying we're done with MD computation 
  359.          */
  360.         MDp->done = 1;
  361.     }
  362.     return 0;
  363. }
  364. /*
  365.  * MDchecksum(data, len, MD5): do a checksum on an arbirtrary amount of data 
  366.  */
  367. int
  368. MDchecksum(u_char * data, size_t len, u_char * mac, size_t maclen)
  369. {
  370.     MDstruct        md;
  371.     MDstruct       *MD = &md;
  372.     int             rc = 0;
  373.     MDbegin(MD);
  374.     while (len >= 64) {
  375.         rc = MDupdate(MD, data, 64 * 8);
  376.         if (rc)
  377.             goto check_end;
  378.         data += 64;
  379.         len -= 64;
  380.     }
  381.     rc = MDupdate(MD, data, len * 8);
  382.     if (rc)
  383.         goto check_end;
  384.     /*
  385.      * copy the checksum to the outgoing data (all of it that is requested). 
  386.      */
  387.     MDget(MD, mac, maclen);
  388.   check_end:
  389.     memset(&md, 0, sizeof(md));
  390.     return rc;
  391. }
  392. /*
  393.  * MDsign(data, len, MD5): do a checksum on an arbirtrary amount
  394.  * of data, and prepended with a secret in the standard fashion 
  395.  */
  396. int
  397. MDsign(u_char * data, size_t len, u_char * mac, size_t maclen,
  398.        u_char * secret, size_t secretlen)
  399. {
  400. #define HASHKEYLEN 64
  401.     MDstruct        MD;
  402.     u_char          K1[HASHKEYLEN];
  403.     u_char          K2[HASHKEYLEN];
  404.     u_char          extendedAuthKey[HASHKEYLEN];
  405.     u_char          buf[HASHKEYLEN];
  406.     size_t          i;
  407.     u_char         *cp, *newdata = 0;
  408.     int             rc = 0;
  409.     /*
  410.      * memset(K1,0,HASHKEYLEN);
  411.      * memset(K2,0,HASHKEYLEN);
  412.      * memset(buf,0,HASHKEYLEN);
  413.      * memset(extendedAuthKey,0,HASHKEYLEN);
  414.      */
  415.     if (secretlen != 16 || secret == NULL || mac == NULL || data == NULL ||
  416.         len <= 0 || maclen <= 0) {
  417.         /*
  418.          * DEBUGMSGTL(("md5","MD5 signing not properly initialized")); 
  419.          */
  420.         return -1;
  421.     }
  422.     memset(extendedAuthKey, 0, HASHKEYLEN);
  423.     memcpy(extendedAuthKey, secret, secretlen);
  424.     for (i = 0; i < HASHKEYLEN; i++) {
  425.         K1[i] = extendedAuthKey[i] ^ 0x36;
  426.         K2[i] = extendedAuthKey[i] ^ 0x5c;
  427.     }
  428.     MDbegin(&MD);
  429.     rc = MDupdate(&MD, K1, HASHKEYLEN * 8);
  430.     if (rc)
  431.         goto update_end;
  432.     i = len;
  433.     if (((unsigned int) data) % sizeof(long) != 0) {
  434.         /*
  435.          * this relies on the ability to use integer math and thus we
  436.          * must rely on data that aligns on 32-bit-word-boundries 
  437.          */
  438.         memdup(&newdata, data, len);
  439.         cp = newdata;
  440.     } else {
  441.         cp = data;
  442.     }
  443.     while (i >= 64) {
  444.         rc = MDupdate(&MD, cp, 64 * 8);
  445.         if (rc)
  446.             goto update_end;
  447.         cp += 64;
  448.         i -= 64;
  449.     }
  450.     rc = MDupdate(&MD, cp, i * 8);
  451.     if (rc)
  452.         goto update_end;
  453.     memset(buf, 0, HASHKEYLEN);
  454.     MDget(&MD, buf, HASHKEYLEN);
  455.     MDbegin(&MD);
  456.     rc = MDupdate(&MD, K2, HASHKEYLEN * 8);
  457.     if (rc)
  458.         goto update_end;
  459.     rc = MDupdate(&MD, buf, 16 * 8);
  460.     if (rc)
  461.         goto update_end;
  462.     /*
  463.      * copy the sign checksum to the outgoing pointer 
  464.      */
  465.     MDget(&MD, mac, maclen);
  466.   update_end:
  467.     memset(buf, 0, HASHKEYLEN);
  468.     memset(K1, 0, HASHKEYLEN);
  469.     memset(K2, 0, HASHKEYLEN);
  470.     memset(extendedAuthKey, 0, HASHKEYLEN);
  471.     memset(&MD, 0, sizeof(MD));
  472.     if (newdata)
  473.         free(newdata);
  474.     return rc;
  475. }
  476. void
  477. MDget(MDstruct * MD, u_char * buf, size_t buflen)
  478. {
  479.     int             i, j;
  480.     /*
  481.      * copy the checksum to the outgoing data (all of it that is requested). 
  482.      */
  483.     for (i = 0; i < 4 && i * 4 < (int) buflen; i++)
  484.         for (j = 0; j < 4 && i * 4 + j < (int) buflen; j++)
  485.             buf[i * 4 + j] = (MD->buffer[i] >> j * 8) & 0xff;
  486. }
  487. /*
  488.  * ** End of md5.c
  489.  * ****************************(cut)****************************************
  490.  */
  491. #endif /* DISABLE_MD5 */