sha1.c
上传用户:tany51
上传日期:2013-06-12
资源大小:1397k
文件大小:13k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. #include <stdio.h>
  2. #include <string.h>
  3. typedef unsigned char *POINTER;
  4. typedef bn_int UINT4;
  5. typedef bn_char BYTE;
  6. void endianTest(int *endian_ness)
  7. {
  8. if((*(unsigned short *) ("#S") >> 8) == '#')
  9. {
  10. /* printf("Big endian = no changen"); */
  11. *endian_ness = !(0);
  12. }
  13. else
  14. {
  15. /* printf("Little endian = swapn"); */
  16. *endian_ness = 0;
  17. }
  18. }
  19. static void SHAtoByte(BYTE *output, UINT4 *input, unsigned int len);
  20. /* The SHS block size and message digest sizes, in bytes */
  21. #define SHS_DATASIZE    64
  22. #define SHS_DIGESTSIZE  20
  23. /* The SHS f()-functions.  The f1 and f3 functions can be optimized to
  24.    save one boolean operation each - thanks to Rich Schroeppel,
  25.    rcs@cs.arizona.edu for discovering this */
  26. /*#define f1(x,y,z) ( ( x & y ) | ( ~x & z ) )          // Rounds  0-19 */
  27. #define f1(x,y,z)   ( z ^ ( x & ( y ^ z ) ) )           /* Rounds  0-19 */
  28. #define f2(x,y,z)   ( x ^ y ^ z )                       /* Rounds 20-39 */
  29. /*#define f3(x,y,z) ( ( x & y ) | ( x & z ) | ( y & z ) )   // Rounds 40-59 */
  30. #define f3(x,y,z)   ( ( x & y ) | ( z & ( x | y ) ) )   /* Rounds 40-59 */
  31. #define f4(x,y,z)   ( x ^ y ^ z )                       /* Rounds 60-79 */
  32. /* The SHS Mysterious Constants */
  33. #define K1  0x5A827999L                                 /* Rounds  0-19 */
  34. #define K2  0x6ED9EBA1L                                 /* Rounds 20-39 */
  35. #define K3  0x8F1BBCDCL                                 /* Rounds 40-59 */
  36. #define K4  0xCA62C1D6L                                 /* Rounds 60-79 */
  37. /* SHS initial values */
  38. #define h0init  0x67452301L
  39. #define h1init  0xEFCDAB89L
  40. #define h2init  0x98BADCFEL
  41. #define h3init  0x10325476L
  42. #define h4init  0xC3D2E1F0L
  43. /* Note that it may be necessary to add parentheses to these macros if they
  44.    are to be called with expressions as arguments */
  45. /* 32-bit rotate left - kludged with shifts */
  46. #define ROTL(n,X)  ( ( ( X ) << n ) | ( ( X ) >> ( 32 - n ) ) )
  47. /* The initial expanding function.  The hash function is defined over an
  48.    80-UINT2 expanded input array W, where the first 16 are copies of the input
  49.    data, and the remaining 64 are defined by
  50.         W[ i ] = W[ i - 16 ] ^ W[ i - 14 ] ^ W[ i - 8 ] ^ W[ i - 3 ]
  51.    This implementation generates these values on the fly in a circular
  52.    buffer - thanks to Colin Plumb, colin@nyx10.cs.du.edu for this
  53.    optimization.
  54.    The updated SHS changes the expanding function by adding a rotate of 1
  55.    bit.  Thanks to Jim Gillogly, jim@rand.org, and an anonymous contributor
  56.    for this information */
  57. #define expand(W,i) ( W[ i & 15 ] = ROTL( 1, ( W[ i & 15 ] ^ W[ (i - 14) & 15 ] ^ 
  58.                                                  W[ (i - 8) & 15 ] ^ W[ (i - 3) & 15 ] ) ) )
  59. /* The prototype SHS sub-round.  The fundamental sub-round is:
  60.         a' = e + ROTL( 5, a ) + f( b, c, d ) + k + data;
  61.         b' = a;
  62.         c' = ROTL( 30, b );
  63.         d' = c;
  64.         e' = d;
  65.    but this is implemented by unrolling the loop 5 times and renaming the
  66.    variables ( e, a, b, c, d ) = ( a', b', c', d', e' ) each iteration.
  67.    This code is then replicated 20 times for each of the 4 functions, using
  68.    the next 20 values from the W[] array each time */
  69. #define subRound(a, b, c, d, e, f, k, data) 
  70.     ( e += ROTL( 5, a ) + f( b, c, d ) + k + data, b = ROTL( 30, b ) )
  71. /* Initialize the SHS values */
  72. void SHAInit(SHA_CTX *shsInfo)
  73. {
  74.     endianTest(&shsInfo->Endianness);
  75.     /* Set the h-vars to their initial values */
  76.     shsInfo->digest[ 0 ] = h0init;
  77.     shsInfo->digest[ 1 ] = h1init;
  78.     shsInfo->digest[ 2 ] = h2init;
  79.     shsInfo->digest[ 3 ] = h3init;
  80.     shsInfo->digest[ 4 ] = h4init;
  81.     /* Initialise bit count */
  82.     shsInfo->countLo = shsInfo->countHi = 0;
  83. }
  84. /* Perform the SHS transformation.  Note that this code, like MD5, seems to
  85.    break some optimizing compilers due to the complexity of the expressions
  86.    and the size of the basic block.  It may be necessary to split it into
  87.    sections, e.g. based on the four subrounds
  88.    Note that this corrupts the shsInfo->data area */
  89. static void SHSTransform( digest, data )
  90.      UINT4 *digest, *data ;
  91.     {
  92.     UINT4 A, B, C, D, E;     /* Local vars */
  93.     UINT4 eData[ 16 ];       /* Expanded data */
  94.     /* Set up first buffer and local data buffer */
  95.     A = digest[ 0 ];
  96.     B = digest[ 1 ];
  97.     C = digest[ 2 ];
  98.     D = digest[ 3 ];
  99.     E = digest[ 4 ];
  100.     memcpy( (POINTER)eData, (POINTER)data, SHS_DATASIZE );
  101.     /* Heavy mangling, in 4 sub-rounds of 20 interations each. */
  102.     subRound( A, B, C, D, E, f1, K1, eData[  0 ] );
  103.     subRound( E, A, B, C, D, f1, K1, eData[  1 ] );
  104.     subRound( D, E, A, B, C, f1, K1, eData[  2 ] );
  105.     subRound( C, D, E, A, B, f1, K1, eData[  3 ] );
  106.     subRound( B, C, D, E, A, f1, K1, eData[  4 ] );
  107.     subRound( A, B, C, D, E, f1, K1, eData[  5 ] );
  108.     subRound( E, A, B, C, D, f1, K1, eData[  6 ] );
  109.     subRound( D, E, A, B, C, f1, K1, eData[  7 ] );
  110.     subRound( C, D, E, A, B, f1, K1, eData[  8 ] );
  111.     subRound( B, C, D, E, A, f1, K1, eData[  9 ] );
  112.     subRound( A, B, C, D, E, f1, K1, eData[ 10 ] );
  113.     subRound( E, A, B, C, D, f1, K1, eData[ 11 ] );
  114.     subRound( D, E, A, B, C, f1, K1, eData[ 12 ] );
  115.     subRound( C, D, E, A, B, f1, K1, eData[ 13 ] );
  116.     subRound( B, C, D, E, A, f1, K1, eData[ 14 ] );
  117.     subRound( A, B, C, D, E, f1, K1, eData[ 15 ] );
  118.     subRound( E, A, B, C, D, f1, K1, expand( eData, 16 ) );
  119.     subRound( D, E, A, B, C, f1, K1, expand( eData, 17 ) );
  120.     subRound( C, D, E, A, B, f1, K1, expand( eData, 18 ) );
  121.     subRound( B, C, D, E, A, f1, K1, expand( eData, 19 ) );
  122.     subRound( A, B, C, D, E, f2, K2, expand( eData, 20 ) );
  123.     subRound( E, A, B, C, D, f2, K2, expand( eData, 21 ) );
  124.     subRound( D, E, A, B, C, f2, K2, expand( eData, 22 ) );
  125.     subRound( C, D, E, A, B, f2, K2, expand( eData, 23 ) );
  126.     subRound( B, C, D, E, A, f2, K2, expand( eData, 24 ) );
  127.     subRound( A, B, C, D, E, f2, K2, expand( eData, 25 ) );
  128.     subRound( E, A, B, C, D, f2, K2, expand( eData, 26 ) );
  129.     subRound( D, E, A, B, C, f2, K2, expand( eData, 27 ) );
  130.     subRound( C, D, E, A, B, f2, K2, expand( eData, 28 ) );
  131.     subRound( B, C, D, E, A, f2, K2, expand( eData, 29 ) );
  132.     subRound( A, B, C, D, E, f2, K2, expand( eData, 30 ) );
  133.     subRound( E, A, B, C, D, f2, K2, expand( eData, 31 ) );
  134.     subRound( D, E, A, B, C, f2, K2, expand( eData, 32 ) );
  135.     subRound( C, D, E, A, B, f2, K2, expand( eData, 33 ) );
  136.     subRound( B, C, D, E, A, f2, K2, expand( eData, 34 ) );
  137.     subRound( A, B, C, D, E, f2, K2, expand( eData, 35 ) );
  138.     subRound( E, A, B, C, D, f2, K2, expand( eData, 36 ) );
  139.     subRound( D, E, A, B, C, f2, K2, expand( eData, 37 ) );
  140.     subRound( C, D, E, A, B, f2, K2, expand( eData, 38 ) );
  141.     subRound( B, C, D, E, A, f2, K2, expand( eData, 39 ) );
  142.     subRound( A, B, C, D, E, f3, K3, expand( eData, 40 ) );
  143.     subRound( E, A, B, C, D, f3, K3, expand( eData, 41 ) );
  144.     subRound( D, E, A, B, C, f3, K3, expand( eData, 42 ) );
  145.     subRound( C, D, E, A, B, f3, K3, expand( eData, 43 ) );
  146.     subRound( B, C, D, E, A, f3, K3, expand( eData, 44 ) );
  147.     subRound( A, B, C, D, E, f3, K3, expand( eData, 45 ) );
  148.     subRound( E, A, B, C, D, f3, K3, expand( eData, 46 ) );
  149.     subRound( D, E, A, B, C, f3, K3, expand( eData, 47 ) );
  150.     subRound( C, D, E, A, B, f3, K3, expand( eData, 48 ) );
  151.     subRound( B, C, D, E, A, f3, K3, expand( eData, 49 ) );
  152.     subRound( A, B, C, D, E, f3, K3, expand( eData, 50 ) );
  153.     subRound( E, A, B, C, D, f3, K3, expand( eData, 51 ) );
  154.     subRound( D, E, A, B, C, f3, K3, expand( eData, 52 ) );
  155.     subRound( C, D, E, A, B, f3, K3, expand( eData, 53 ) );
  156.     subRound( B, C, D, E, A, f3, K3, expand( eData, 54 ) );
  157.     subRound( A, B, C, D, E, f3, K3, expand( eData, 55 ) );
  158.     subRound( E, A, B, C, D, f3, K3, expand( eData, 56 ) );
  159.     subRound( D, E, A, B, C, f3, K3, expand( eData, 57 ) );
  160.     subRound( C, D, E, A, B, f3, K3, expand( eData, 58 ) );
  161.     subRound( B, C, D, E, A, f3, K3, expand( eData, 59 ) );
  162.     subRound( A, B, C, D, E, f4, K4, expand( eData, 60 ) );
  163.     subRound( E, A, B, C, D, f4, K4, expand( eData, 61 ) );
  164.     subRound( D, E, A, B, C, f4, K4, expand( eData, 62 ) );
  165.     subRound( C, D, E, A, B, f4, K4, expand( eData, 63 ) );
  166.     subRound( B, C, D, E, A, f4, K4, expand( eData, 64 ) );
  167.     subRound( A, B, C, D, E, f4, K4, expand( eData, 65 ) );
  168.     subRound( E, A, B, C, D, f4, K4, expand( eData, 66 ) );
  169.     subRound( D, E, A, B, C, f4, K4, expand( eData, 67 ) );
  170.     subRound( C, D, E, A, B, f4, K4, expand( eData, 68 ) );
  171.     subRound( B, C, D, E, A, f4, K4, expand( eData, 69 ) );
  172.     subRound( A, B, C, D, E, f4, K4, expand( eData, 70 ) );
  173.     subRound( E, A, B, C, D, f4, K4, expand( eData, 71 ) );
  174.     subRound( D, E, A, B, C, f4, K4, expand( eData, 72 ) );
  175.     subRound( C, D, E, A, B, f4, K4, expand( eData, 73 ) );
  176.     subRound( B, C, D, E, A, f4, K4, expand( eData, 74 ) );
  177.     subRound( A, B, C, D, E, f4, K4, expand( eData, 75 ) );
  178.     subRound( E, A, B, C, D, f4, K4, expand( eData, 76 ) );
  179.     subRound( D, E, A, B, C, f4, K4, expand( eData, 77 ) );
  180.     subRound( C, D, E, A, B, f4, K4, expand( eData, 78 ) );
  181.     subRound( B, C, D, E, A, f4, K4, expand( eData, 79 ) );
  182.     /* Build message digest */
  183.     digest[ 0 ] += A;
  184.     digest[ 1 ] += B;
  185.     digest[ 2 ] += C;
  186.     digest[ 3 ] += D;
  187.     digest[ 4 ] += E;
  188.     }
  189. /* When run on a little-endian CPU we need to perform byte reversal on an
  190.    array of long words. */
  191. static void longReverse(UINT4 *buffer, int byteCount, int Endianness )
  192. {
  193.     UINT4 value;
  194.     if (Endianness==TRUE) return;
  195.     byteCount /= sizeof( UINT4 );
  196.     while( byteCount-- )
  197.         {
  198.         value = *buffer;
  199.         value = ( ( value & 0xFF00FF00L ) >> 8  ) | 
  200.                 ( ( value & 0x00FF00FFL ) << 8 );
  201.         *buffer++ = ( value << 16 ) | ( value >> 16 );
  202.         }
  203. }
  204. /* Update SHS for a block of data */
  205. void SHAUpdate(SHA_CTX *shsInfo, BYTE *buffer, int count)
  206. {
  207.     UINT4 tmp;
  208.     int dataCount;
  209.     /* Update bitcount */
  210.     tmp = shsInfo->countLo;
  211.     if ( ( shsInfo->countLo = tmp + ( ( UINT4 ) count << 3 ) ) < tmp )
  212.         shsInfo->countHi++;             /* Carry from low to high */
  213.     shsInfo->countHi += count >> 29;
  214.     /* Get count of bytes already in data */
  215.     dataCount = ( int ) ( tmp >> 3 ) & 0x3F;
  216.     /* Handle any leading odd-sized chunks */
  217.     if( dataCount )
  218.         {
  219.         BYTE *p = ( BYTE * ) shsInfo->data + dataCount;
  220.         dataCount = SHS_DATASIZE - dataCount;
  221.         if( count < dataCount )
  222.             {
  223.             memcpy( p, buffer, count );
  224.             return;
  225.             }
  226.         memcpy( p, buffer, dataCount );
  227.         longReverse( shsInfo->data, SHS_DATASIZE, shsInfo->Endianness);
  228.         SHSTransform( shsInfo->digest, shsInfo->data );
  229.         buffer += dataCount;
  230.         count -= dataCount;
  231.         }
  232.     /* Process data in SHS_DATASIZE chunks */
  233.     while( count >= SHS_DATASIZE )
  234.         {
  235.         memcpy( (POINTER)shsInfo->data, (POINTER)buffer, SHS_DATASIZE );
  236.         longReverse( shsInfo->data, SHS_DATASIZE, shsInfo->Endianness );
  237.         SHSTransform( shsInfo->digest, shsInfo->data );
  238.         buffer += SHS_DATASIZE;
  239.         count -= SHS_DATASIZE;
  240.         }
  241.     /* Handle any remaining bytes of data. */
  242.     memcpy( (POINTER)shsInfo->data, (POINTER)buffer, count );
  243.     }
  244. /* Final wrapup - pad to SHS_DATASIZE-byte boundary with the bit pattern
  245.    1 0* (64-bit count of bits processed, MSB-first) */
  246. void SHAFinal(BYTE *output, SHA_CTX *shsInfo)
  247. {
  248.     int count;
  249.     BYTE *dataPtr;
  250.     /* Compute number of bytes mod 64 */
  251.     count = ( int ) shsInfo->countLo;
  252.     count = ( count >> 3 ) & 0x3F;
  253.     /* Set the first char of padding to 0x80.  This is safe since there is
  254.        always at least one byte free */
  255.     dataPtr = ( BYTE * ) shsInfo->data + count;
  256.     *dataPtr++ = 0x80;
  257.     /* Bytes of padding needed to make 64 bytes */
  258.     count = SHS_DATASIZE - 1 - count;
  259.     /* Pad out to 56 mod 64 */
  260.     if( count < 8 )
  261.         {
  262.         /* Two lots of padding:  Pad the first block to 64 bytes */
  263.         memset( dataPtr, 0, count );
  264.         longReverse( shsInfo->data, SHS_DATASIZE, shsInfo->Endianness );
  265.         SHSTransform( shsInfo->digest, shsInfo->data );
  266.         /* Now fill the next block with 56 bytes */
  267.         memset( (POINTER)shsInfo->data, 0, SHS_DATASIZE - 8 );
  268.         }
  269.     else
  270.         /* Pad block to 56 bytes */
  271.         memset( dataPtr, 0, count - 8 );
  272.     /* Append length in bits and transform */
  273.     shsInfo->data[ 14 ] = shsInfo->countHi;
  274.     shsInfo->data[ 15 ] = shsInfo->countLo;
  275.     longReverse( shsInfo->data, SHS_DATASIZE - 8, shsInfo->Endianness );
  276.     SHSTransform( shsInfo->digest, shsInfo->data );
  277. /* Output to an array of bytes */
  278. SHAtoByte(output, shsInfo->digest, SHS_DIGESTSIZE);
  279. /* Zeroise sensitive stuff */
  280. memset((POINTER)shsInfo, 0, sizeof(shsInfo));
  281. }
  282. static void SHAtoByte(BYTE *output, UINT4 *input, unsigned int len)
  283. { /* Output SHA digest in byte array */
  284. unsigned int i, j;
  285. for(i = 0, j = 0; j < len; i++, j += 4) 
  286. {
  287.         output[j+3] = (BYTE)( input[i]        & 0xff);
  288.         output[j+2] = (BYTE)((input[i] >> 8 ) & 0xff);
  289.         output[j+1] = (BYTE)((input[i] >> 16) & 0xff);
  290.         output[j  ] = (BYTE)((input[i] >> 24) & 0xff);
  291. }
  292. }