yaole
129 lines
3.0 KiB
C
129 lines
3.0 KiB
C
// testmain.c
|
|
// 2018-04-20 Markku-Juhani O. Saarinen <mjos@iki.fi>
|
|
|
|
#include <stdio.h>
|
|
#include <time.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
// reference implementation
|
|
#include "sm4_ref.h"
|
|
|
|
// AES-NI / SSE3 implementation, encrypt 4 blocks at once
|
|
void sm4_encrypt4(const uint32_t rk[32], void *src, const void *dst);
|
|
|
|
// high-precision time
|
|
|
|
static double clk_now()
|
|
{
|
|
struct timespec ts;
|
|
|
|
// You may onsider CLOCK_MONOTONIC and CLOCK_MONOTONIC_RAW here too
|
|
if (clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &ts) != 0) {
|
|
perror("clock_gettime()");
|
|
exit(-1);
|
|
}
|
|
return ((double) ts.tv_sec) + 1E-9 * ((double) ts.tv_nsec);
|
|
}
|
|
|
|
// test speed and validity
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int i, e;
|
|
double clk1, clk2, n;
|
|
|
|
// test vectors from the standard
|
|
|
|
const uint8_t key[16] = {
|
|
0x01, 0x23, 0x45, 0x67, 0x89, 0xAB, 0xCD, 0xEF,
|
|
0xFE, 0xDC, 0xBA, 0x98, 0x76, 0x54, 0x32, 0x10
|
|
};
|
|
|
|
const uint8_t ref[16] = {
|
|
0x68, 0x1E, 0xDF, 0x34, 0xD2, 0x06, 0x96, 0x5E,
|
|
0x86, 0xB3, 0xE9, 0x4F, 0x53, 0x6E, 0x42, 0x46
|
|
};
|
|
|
|
uint8_t buf1[64], buf2[64];
|
|
uint32_t rk[32];
|
|
|
|
memset(buf1, 0x55, sizeof(buf1));
|
|
memset(buf2, 0xAA, sizeof(buf2));
|
|
|
|
// Test reference implementation with a test vector.
|
|
|
|
sm4_key_schedule(key, rk);
|
|
sm4_encrypt(rk, key, buf1);
|
|
if (memcmp(buf1, ref, 16) != 0) {
|
|
fprintf(stderr, "sm4_encrypt() test failed.\n");
|
|
return -1;
|
|
}
|
|
sm4_decrypt(rk, buf1, buf2);
|
|
if (memcmp(buf2, key, 16) != 0) {
|
|
fprintf(stderr, "sm4_decrypt() test failed.\n");
|
|
return -1;
|
|
}
|
|
|
|
// Test the SM4NI four-block version against reference implementation.
|
|
|
|
for (i = 0; i < 64; i++)
|
|
buf1[i] = i;
|
|
|
|
// encrypt 4 blocks at once to buf2
|
|
sm4_encrypt4(rk, buf1, buf2);
|
|
|
|
// individual blocks in place (with ref algoritm)
|
|
sm4_encrypt(rk, buf1, buf1);
|
|
sm4_encrypt(rk, buf1 + 16, buf1 + 16);
|
|
sm4_encrypt(rk, buf1 + 32, buf1 + 32);
|
|
sm4_encrypt(rk, buf1 + 48, buf1 + 48);
|
|
|
|
if (memcmp(buf1, buf2, 64) != 0) {
|
|
fprintf(stderr, "sm4_encrypt4() test failed.\n");
|
|
return -1;
|
|
}
|
|
|
|
// bench reference implementation
|
|
|
|
n = 0;
|
|
e = 16;
|
|
clk1 = clk_now();
|
|
do {
|
|
|
|
for (i = 0; i < e; i++) {
|
|
sm4_encrypt(rk, buf1, buf1);
|
|
sm4_encrypt(rk, buf1 + 16, buf1 + 16);
|
|
sm4_encrypt(rk, buf1 + 32, buf1 + 32);
|
|
sm4_encrypt(rk, buf1 + 48, buf1 + 48);
|
|
}
|
|
n += 64 * e;
|
|
e <<= 1;
|
|
clk2 = clk_now() - clk1;
|
|
} while (clk2 < 2.0);
|
|
|
|
printf("SM4 reference %10.3f MB/s\n", (n / 1E6) / ((double) clk2));
|
|
|
|
// bench reference implementation
|
|
|
|
n = 0;
|
|
e = 16;
|
|
clk1 = clk_now();
|
|
do {
|
|
|
|
for (i = 0; i < e; i++) {
|
|
sm4_encrypt4(rk, buf1, buf1);
|
|
}
|
|
n += 64 * e;
|
|
e <<= 1;
|
|
clk2 = clk_now() - clk1;
|
|
} while (clk2 < 2.0);
|
|
|
|
printf("Vector SM4NI %10.3f MB/s\n", (n / 1E6) / ((double) clk2));
|
|
|
|
|
|
return 0;
|
|
}
|
|
|
|
|