init: v1.0.0

2026-05-27 23:03:00 +08:00
commit 8d97f750eb
466 changed files with 80067 additions and 0 deletions
@@ -0,0 +1,128 @@
+// 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;
+}
+
+