mirror of
https://github.com/guanzhi/GmSSL.git
synced 2026-05-07 00:46:17 +08:00
add gf128_avx, add sm3 sm4 speed
This commit is contained in:
toorls
360
tools/sm4speed.c
Normal file
360
tools/sm4speed.c
Normal file
@@ -0,0 +1,360 @@
|
||||
/*
|
||||
* Copyright 2014-2022 The GmSSL Project. All Rights Reserved.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the License); you may
|
||||
* not use this file except in compliance with the License.
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <gmssl/hex.h>
|
||||
#include <gmssl/sm4.h>
|
||||
#include <gmssl/error.h>
|
||||
#include <gmssl/rand.h>
|
||||
|
||||
#ifdef WIN32
|
||||
#include <wincrypt.h>
|
||||
|
||||
static volatile int finish;
|
||||
|
||||
VOID CALLBACK TimerProc_sm4(HWND hwnd, UINT message, UINT iTimerID, DWORD dwTime)
|
||||
{
|
||||
finish = 0;
|
||||
}
|
||||
|
||||
int test_sm4()
|
||||
{
|
||||
uint8_t user_key[16] = {
|
||||
0x01,
|
||||
0x23,
|
||||
0x45,
|
||||
0x67,
|
||||
0x89,
|
||||
0xab,
|
||||
0xcd,
|
||||
0xef,
|
||||
0xfe,
|
||||
0xdc,
|
||||
0xba,
|
||||
0x98,
|
||||
0x76,
|
||||
0x54,
|
||||
0x32,
|
||||
0x10,
|
||||
};
|
||||
uint8_t iv[16] = {
|
||||
0x01,
|
||||
0x23,
|
||||
0x45,
|
||||
0x67,
|
||||
0x89,
|
||||
0xab,
|
||||
0xcd,
|
||||
0xef,
|
||||
0xfe,
|
||||
0xdc,
|
||||
0xba,
|
||||
0x98,
|
||||
0x76,
|
||||
0x54,
|
||||
0x32,
|
||||
0x10,
|
||||
};
|
||||
uint8_t ctr[16] = {0};
|
||||
uint8_t mac[16] = {0};
|
||||
uint8_t aad[16] = {
|
||||
0x01,
|
||||
0x23,
|
||||
0x45,
|
||||
0x67,
|
||||
0x89,
|
||||
0xab,
|
||||
0xcd,
|
||||
0xef,
|
||||
0xfe,
|
||||
0xdc,
|
||||
0xba,
|
||||
0x98,
|
||||
0x76,
|
||||
0x54,
|
||||
0x32,
|
||||
0x10,
|
||||
};
|
||||
uint8_t out[16384] = {0};
|
||||
|
||||
SM4_KEY key;
|
||||
int sizebox[] = {16, 64, 256, 1024, 8192, 16384};
|
||||
int countbox[18] = {0};
|
||||
uint8_t *testhex[];
|
||||
HCRYPTPROV hCryptProv;
|
||||
|
||||
testhex = (uint8_t **)malloc(sizeof(uint8_t *) * 6);
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
testhex[i] = (uint8_t *)malloc(sizebox[i]);
|
||||
CryptGenRandom(hCryptProv, sizebox[i], testhex[i]);
|
||||
}
|
||||
|
||||
int count;
|
||||
|
||||
sm4_set_encrypt_key(&key, user_key);
|
||||
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
finish = 1;
|
||||
count = 0;
|
||||
printf("Doing sm4-cbc for 3s on %d size blocks: ", sizebox[i]);
|
||||
UINT_PTR iTimerID = SetTimer(NULL, 0, 3000, TimerProc_sm4);
|
||||
while (finish)
|
||||
{
|
||||
sm4_cbc_encrypt(&key, iv, testhex[i], sizebox[i] / 16, out);
|
||||
count++;
|
||||
}
|
||||
KillTimer(NULL, iTimerID);
|
||||
countbox[i] = count;
|
||||
printf("%d sm4-cbc's in 3s\n", count);
|
||||
}
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
finish = 1;
|
||||
count = 0;
|
||||
printf("Doing sm4-ctr for 3s on %d size blocks: ", sizebox[i]);
|
||||
UINT_PTR iTimerID = SetTimer(NULL, 0, 3000, TimerProc);
|
||||
while (finish)
|
||||
{
|
||||
sm4_ctr_encrypt(&key, ctr, testhex[i], sizebox[i], out);
|
||||
count++;
|
||||
}
|
||||
KillTimer(NULL, iTimerID);
|
||||
countbox[i + 6] = count;
|
||||
printf("%d sm4-ctr's in 3s\n", count);
|
||||
}
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
finish = 1;
|
||||
count = 0;
|
||||
printf("Doing sm4-gcm for 3s on %d size blocks: ", sizebox[i]);
|
||||
UINT_PTR iTimerID = SetTimer(NULL, 0, 3000, TimerProc);
|
||||
while (finish)
|
||||
{
|
||||
sm4_gcm_encrypt(&key, iv, 16, aad, 16, testhex[i], sizebox[i], out, 16, mac);
|
||||
count++;
|
||||
}
|
||||
KillTimer(NULL, iTimerID);
|
||||
countbox[i + 12] = count;
|
||||
printf("%d sm4-gcm's in 3s\n", count);
|
||||
}
|
||||
printf("type\t\t16 bytes\t64 bytes\t256 bytes\t1024 bytes\t8192 bytes\t16384 bytes\n");
|
||||
printf("sm4-cbc\t");
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
printf("\t%.2fK", countbox[i] * sizebox[i] / 1024 / 3.00);
|
||||
}
|
||||
printf("\n");
|
||||
printf("sm4-ctr\t");
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
printf("\t%.2fK", countbox[i + 6] * sizebox[i] / 1024 / 3.00);
|
||||
}
|
||||
printf("\n");
|
||||
printf("sm4-gcm\t");
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
printf("\t%.2fK", countbox[i + 12] * sizebox[i] / 1024 / 3.00);
|
||||
}
|
||||
printf("\n");
|
||||
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
free(testhex[i]);
|
||||
}
|
||||
free(testhex);
|
||||
return 1;
|
||||
}
|
||||
|
||||
#else
|
||||
#include <signal.h>
|
||||
#include <sys/time.h>
|
||||
|
||||
static volatile int finish;
|
||||
|
||||
void sig_alm_handler_sm4(int sig_num)
|
||||
{
|
||||
if (sig_num = SIGALRM)
|
||||
finish = 0;
|
||||
}
|
||||
|
||||
int test_sm4()
|
||||
{
|
||||
uint8_t user_key[16] = {
|
||||
0x01,
|
||||
0x23,
|
||||
0x45,
|
||||
0x67,
|
||||
0x89,
|
||||
0xab,
|
||||
0xcd,
|
||||
0xef,
|
||||
0xfe,
|
||||
0xdc,
|
||||
0xba,
|
||||
0x98,
|
||||
0x76,
|
||||
0x54,
|
||||
0x32,
|
||||
0x10,
|
||||
};
|
||||
uint8_t iv[16] = {
|
||||
0x01,
|
||||
0x23,
|
||||
0x45,
|
||||
0x67,
|
||||
0x89,
|
||||
0xab,
|
||||
0xcd,
|
||||
0xef,
|
||||
0xfe,
|
||||
0xdc,
|
||||
0xba,
|
||||
0x98,
|
||||
0x76,
|
||||
0x54,
|
||||
0x32,
|
||||
0x10,
|
||||
};
|
||||
uint8_t ctr[16] = {0};
|
||||
uint8_t mac[16] = {0};
|
||||
uint8_t aad[16] = {
|
||||
0x01,
|
||||
0x23,
|
||||
0x45,
|
||||
0x67,
|
||||
0x89,
|
||||
0xab,
|
||||
0xcd,
|
||||
0xef,
|
||||
0xfe,
|
||||
0xdc,
|
||||
0xba,
|
||||
0x98,
|
||||
0x76,
|
||||
0x54,
|
||||
0x32,
|
||||
0x10,
|
||||
};
|
||||
uint8_t out[16384] = {0};
|
||||
int count;
|
||||
|
||||
SM4_KEY key;
|
||||
int sizebox[] = {16, 64, 256, 1024, 8192, 16384};
|
||||
int countbox[18] = {0};
|
||||
uint8_t **testhex;
|
||||
|
||||
FILE *fs_p = fopen("/dev/urandom", "r");
|
||||
if (NULL == fs_p)
|
||||
{
|
||||
printf("Can not open /dev/urandom\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
testhex = (uint8_t **)malloc(sizeof(uint8_t *) * 6);
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
testhex[i] = (uint8_t *)malloc(sizebox[i]);
|
||||
fread(testhex[i], sizebox[i], 1, fs_p);
|
||||
}
|
||||
|
||||
fclose(fs_p);
|
||||
|
||||
signal(SIGALRM, sig_alm_handler_sm4);
|
||||
|
||||
struct itimerval new_value, old_value;
|
||||
new_value.it_value.tv_sec = 3;
|
||||
new_value.it_value.tv_usec = 0;
|
||||
new_value.it_interval.tv_sec = 0;
|
||||
new_value.it_interval.tv_usec = 0;
|
||||
|
||||
sm4_set_encrypt_key(&key, user_key);
|
||||
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
finish = 1;
|
||||
count = 0;
|
||||
printf("Doing sm4-cbc for 3s on %d size blocks: ", sizebox[i]);
|
||||
setitimer(ITIMER_REAL, &new_value, &old_value);
|
||||
while (finish)
|
||||
{
|
||||
sm4_cbc_encrypt(&key, iv, testhex[i], sizebox[i] / 16, out);
|
||||
count++;
|
||||
}
|
||||
|
||||
countbox[i] = count;
|
||||
printf("%d sm4-cbc's in 3s\n", count);
|
||||
}
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
finish = 1;
|
||||
count = 0;
|
||||
printf("Doing sm4-ctr for 3s on %d size blocks: ", sizebox[i]);
|
||||
setitimer(ITIMER_REAL, &new_value, &old_value);
|
||||
while (finish)
|
||||
{
|
||||
sm4_ctr_encrypt(&key, ctr, testhex[i], sizebox[i], out);
|
||||
count++;
|
||||
}
|
||||
|
||||
countbox[i + 6] = count;
|
||||
printf("%d sm4-ctr's in 3s\n", count);
|
||||
}
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
finish = 1;
|
||||
count = 0;
|
||||
printf("Doing sm4-gcm for 3s on %d size blocks: ", sizebox[i]);
|
||||
setitimer(ITIMER_REAL, &new_value, &old_value);
|
||||
while (finish)
|
||||
{
|
||||
sm4_gcm_encrypt(&key, iv, 16, aad, 16, testhex[i], sizebox[i], out, 16, mac);
|
||||
count++;
|
||||
}
|
||||
countbox[i + 12] = count;
|
||||
printf("%d sm4-gcm's in 3s\n", count);
|
||||
}
|
||||
printf("type\t\t16 bytes\t64 bytes\t256 bytes\t1024 bytes\t8192 bytes\t16384 bytes\n");
|
||||
printf("sm4-cbc\t");
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
printf("\t%.2fK", countbox[i] * sizebox[i] / 1024 / 3.00);
|
||||
}
|
||||
printf("\n");
|
||||
printf("sm4-ctr\t");
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
printf("\t%.2fK", countbox[i + 6] * sizebox[i] / 1024 / 3.00);
|
||||
}
|
||||
printf("\n");
|
||||
printf("sm4-gcm\t");
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
printf("\t%.2fK", countbox[i + 12] * sizebox[i] / 1024 / 3.00);
|
||||
}
|
||||
printf("\n");
|
||||
|
||||
for (int i = 0; i < 6; i++)
|
||||
{
|
||||
free(testhex[i]);
|
||||
}
|
||||
free(testhex);
|
||||
return 1;
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
int sm4speed_main(void)
|
||||
{
|
||||
test_sm4();
|
||||
return 1;
|
||||
}
|
||||
Reference in New Issue
Block a user