2.7 SM3

sm3.h

  1. /*
  2. * sm3.h
  3. *
  4. * 为使此算法兼容32位、64位下Linux或Windows系统，
  5. * 选择 int 来表示 32 位整数。
  6. * 消息长度最大限定为 2**32 - 1（单位：比特），
  7. * 且为 8 的倍数（消息的最小单元为字节）。
  8. */
  9. #ifndef _SM3_H_
  10. #define _SM3_H_
  12. /*
  13. * SM3算法产生的哈希值大小（单位：字节）
  14. */
  15. #define SM3_HASH_SIZE 32 
  17. /*
  18. * SM3上下文
  19. */
  20. typedef struct SM3Context
  21. {
  22.     unsigned int intermediateHash[SM3_HASH_SIZE / 4];
  23.     unsigned char messageBlock[64];
  24. } SM3Context;
  26. /*
  27. * SM3计算函数
  28. */
  29. unsigned char *SM3Calc(const unsigned char *message,
  30.     unsigned int messageLen, unsigned char digest[SM3_HASH_SIZE]);
  32. #endif // _SM3_H_

sm3.c

  1. /*
  2. * sm3.c
  3. */
  4. #include "stdafx.h"
  5. #include <stdio.h>
  6. #include <memory.h>
  7. #include "sm3.h"
  9. /*
  10. * 判断运行环境是否为小端
  11. */
  12. static const int endianTest = 1;
  13. #define IsLittleEndian() (*(char *)&endianTest == 1)
  15. /*
  16. * 向左循环移位
  17. */
  18. #define LeftRotate(word, bits) ( (word) << (bits) | (word) >> (32 - (bits)) )
  20. /*
  21. * 反转四字节整型字节序
  22. */
  23. unsigned int *ReverseWord(unsigned int *word)
  24. {
  25.     unsigned char *byte, temp;
  27.     byte = (unsigned char *)word;
  28.     temp = byte[0];
  29.     byte[0] = byte[3];
  30.     byte[3] = temp;
  32.     temp = byte[1];
  33.     byte[1] = byte[2];
  34.     byte[2] = temp;
  35.     return word;
  36. }
  38. /*
  39. * T
  40. */
  41. unsigned int T(int i)
  42. {
  43.     if (i >= 0 && i <= 15)
  44.         return 0x79CC4519;
  45.     else if (i >= 16 && i <= 63)
  46.         return 0x7A879D8A;
  47.     else
  48.         return 0;
  49. }
  51. /*
  52. * FF
  53. */
  54. unsigned int FF(unsigned int X, unsigned int Y, unsigned int Z, int i)
  55. {
  56.     if (i >= 0 && i <= 15)
  57.         return X ^ Y ^ Z;
  58.     else if (i >= 16 && i <= 63)
  59.         return (X & Y) | (X & Z) | (Y & Z);
  60.     else
  61.         return 0;
  62. }
  64. /*
  65. * GG
  66. */
  67. unsigned int GG(unsigned int X, unsigned int Y, unsigned int Z, int i)
  68. {
  69.     if (i >= 0 && i <= 15)
  70.         return X ^ Y ^ Z;
  71.     else if (i >= 16 && i <= 63)
  72.         return (X & Y) | (~X & Z);
  73.     else
  74.         return 0;
  75. }
  77. /*
  78. * P0
  79. */
  80. unsigned int P0(unsigned int X)
  81. {
  82.     return X ^ LeftRotate(X, 9) ^ LeftRotate(X, 17);
  83. }
  85. /*
  86. * P1
  87. */
  88. unsigned int P1(unsigned int X)
  89. {
  90.     return X ^ LeftRotate(X, 15) ^ LeftRotate(X, 23);
  91. }
  93. /*
  94. * 初始化函数
  95. */
  96. void SM3Init(SM3Context *context)
  97. {
  98.     context->intermediateHash[0] = 0x7380166F;
  99.     context->intermediateHash[1] = 0x4914B2B9;
  100.     context->intermediateHash[2] = 0x172442D7;
  101.     context->intermediateHash[3] = 0xDA8A0600;
  102.     context->intermediateHash[4] = 0xA96F30BC;
  103.     context->intermediateHash[5] = 0x163138AA;
  104.     context->intermediateHash[6] = 0xE38DEE4D;
  105.     context->intermediateHash[7] = 0xB0FB0E4E;
  106. }
  108. /*
  109. * 处理消息块
  110. */
  111. void SM3ProcessMessageBlock(SM3Context *context)
  112. {
  113.     int i;
  114.     unsigned int W[68];
  115.     unsigned int W_[64];
  116.     unsigned int A, B, C, D, E, F, G, H, SS1, SS2, TT1, TT2;
  118.     /* 消息扩展 */
  119.     for (i = 0; i < 16; i++)
  120.     {
  121.         W[i] = *(unsigned int *)(context->messageBlock + i * 4);
  122.         if (IsLittleEndian())
  123.             ReverseWord(W + i);
  124.         //printf("%d: %x\n", i, W[i]);    
  125.     }
  126.     for (i = 16; i < 68; i++)
  127.     {
  128.         W[i] = P1(W[i - 16] ^ W[i - 9] ^ LeftRotate(W[i - 3], 15))
  129.             ^ LeftRotate(W[i - 13], 7)
  130.             ^ W[i - 6];
  131.         //printf("%d: %x\n", i, W[i]);    
  132.     }
  133.     for (i = 0; i < 64; i++)
  134.     {
  135.         W_[i] = W[i] ^ W[i + 4];
  136.         //printf("%d: %x\n", i, W_[i]);    
  137.     }
  139.     /* 消息压缩 */
  140.     A = context->intermediateHash[0];
  141.     B = context->intermediateHash[1];
  142.     C = context->intermediateHash[2];
  143.     D = context->intermediateHash[3];
  144.     E = context->intermediateHash[4];
  145.     F = context->intermediateHash[5];
  146.     G = context->intermediateHash[6];
  147.     H = context->intermediateHash[7];
  148.     for (i = 0; i < 64; i++)
  149.     {
  150.         SS1 = LeftRotate((LeftRotate(A, 12) + E + LeftRotate(T(i), i)), 7);
  151.         SS2 = SS1 ^ LeftRotate(A, 12);
  152.         TT1 = FF(A, B, C, i) + D + SS2 + W_[i];
  153.         TT2 = GG(E, F, G, i) + H + SS1 + W[i];
  154.         D = C;
  155.         C = LeftRotate(B, 9);
  156.         B = A;
  157.         A = TT1;
  158.         H = G;
  159.         G = LeftRotate(F, 19);
  160.         F = E;
  161.         E = P0(TT2);
  162.     }
  163.     context->intermediateHash[0] ^= A;
  164.     context->intermediateHash[1] ^= B;
  165.     context->intermediateHash[2] ^= C;
  166.     context->intermediateHash[3] ^= D;
  167.     context->intermediateHash[4] ^= E;
  168.     context->intermediateHash[5] ^= F;
  169.     context->intermediateHash[6] ^= G;
  170.     context->intermediateHash[7] ^= H;
  171. }
  173. /*
  174. * SM3算法主函数
  175. */
  176. unsigned char *SM3Calc(const unsigned char *message,
  177.     unsigned int messageLen, unsigned char digest[SM3_HASH_SIZE])
  178. {
  179.     SM3Context context;
  180.     unsigned int i, remainder, bitLen;
  182.     /* 初始化上下文 */
  183.     SM3Init(&context);
  185.     /* 对前面的消息分组进行处理 */
  186.     for (i = 0; i < messageLen / 64; i++)
  187.     {
  188.         memcpy(context.messageBlock, message + i * 64, 64);
  189.         SM3ProcessMessageBlock(&context);
  190.     }
  192.     /* 填充消息分组，并处理 */
  193.     bitLen = messageLen * 8;
  194.     if (IsLittleEndian())
  195.         ReverseWord(&bitLen);
  196.     remainder = messageLen % 64;
  197.     memcpy(context.messageBlock, message + i * 64, remainder);
  198.     context.messageBlock[remainder] = 0x80;
  199.     if (remainder <= 55)
  200.     {
  201.         /* 长度按照大端法占8个字节，该程序只考虑长度在 2**32 - 1（单位：比特）以内的情况，
  202.         * 故将高 4 个字节赋为 0 。*/
  203.         memset(context.messageBlock + remainder + 1, 0, 64 - remainder - 1 - 8 + 4);
  204.         memcpy(context.messageBlock + 64 - 4, &bitLen, 4);
  205.         SM3ProcessMessageBlock(&context);
  206.     }
  207.     else
  208.     {
  209.         memset(context.messageBlock + remainder + 1, 0, 64 - remainder - 1);
  210.         SM3ProcessMessageBlock(&context);
  211.         /* 长度按照大端法占8个字节，该程序只考虑长度在 2**32 - 1（单位：比特）以内的情况，
  212.         * 故将高 4 个字节赋为 0 。*/
  213.         memset(context.messageBlock, 0, 64 - 4);
  214.         memcpy(context.messageBlock + 64 - 4, &bitLen, 4);
  215.         SM3ProcessMessageBlock(&context);
  216.     }
  218.     /* 返回结果 */
  219.     if (IsLittleEndian())
  220.     for (i = 0; i < 8; i++)
  221.         ReverseWord(context.intermediateHash + i);
  222.     memcpy(digest, context.intermediateHash, SM3_HASH_SIZE);
  224.     return digest;
  225. }

sm3Test.cpp

  1. // sm3Test.cpp : 定义控制台应用程序的入口点。
  2. //
  4. #include "stdafx.h"
  5. #include <string.h>
  6. #include <stdio.h>
  7. #include "sm3.h"
  8. #include <iostream>
  10. //#pragma comment(lib,"sm3dll2")
  11. //extern "C" void SM3Call(const unsigned char *message,unsigned int messageLen, unsigned char digest[SM3_HASH_SIZE]);
  13. int main( int argc, char *argv[] )
  14. {
  15.     unsigned char input[256] = "12345678";
  16.     int ilen = 3;
  17.     unsigned char output[32];
  18.     int i;
  19.     // ctx;
  21.     printf("Message:\n");
  22.     printf("%s\n",input);
  24.     SM3Calc(input, 8, output);
  25.     printf("Hash:\n   ");
  26.     for(i=0; i<32; i++)
  27.     {
  28.         printf("%02x",output[i]);
  29.         if (((i+1) % 4 ) == 0) printf(" ");
  30.     } 
  32.     printf("\n");
  35.     unsigned char input2[256] = "abcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcdabcd";
  36.     int ilen2 = 64;
  37.     unsigned char output2[32];
  38.     int i2;
  39.     // ctx;
  41.     printf("Message:\n");
  42.     printf("%s\n", input2);
  44.     SM3Calc(input2, ilen2, output2);
  45.     printf("Hash:\n   ");
  46.     for (i2 = 0; i2<32; i2++)
  47.     {
  48.         printf("%02x", output2[i2]);
  49.         if (((i2 + 1) % 4) == 0) printf(" ");
  50.     }
  51.     printf("\n");
  52.     system("pause");
  53. }