@pinkparrot/qsafe-mayo-wasm 0.0.3

package/mayo-c/test/test_mayo.c

@@ -0,0 +1,139 @@
+// SPDX-License-Identifier: Apache-2.0
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <ctype.h>
+#include <randombytes.h>
+#include <mayo.h>
+#include <stdalign.h>
+
+#ifdef ENABLE_CT_TESTING
+#include <valgrind/memcheck.h>
+#endif
+
+#ifdef ENABLE_CT_TESTING
+static void print_hex(const unsigned char *hex, int len) {
+    unsigned char *copy  = calloc(len, 1);
+    memcpy(copy, hex, len); // make a copy that we can tell valgrind is okay to leak
+    VALGRIND_MAKE_MEM_DEFINED(copy, len);
+
+    for (int i = 0; i < len;  ++i) {
+        printf("%02x", copy[i]);
+    }
+    printf("\n");
+    free(copy);
+}
+#else
+static void print_hex(const unsigned char *hex, int len) {
+    for (int i = 0; i < len;  ++i) {
+        printf("%02x", hex[i]);
+    }
+    printf("\n");
+}
+#endif
+
+
+static int test_mayo(const mayo_params_t *p) {
+    unsigned char _pk[CPK_BYTES_MAX + 1] = {0};  
+    unsigned char _sk[CSK_BYTES_MAX + 1] = {0};
+    unsigned char _sig[SIG_BYTES_MAX + 32 + 1] = {0};
+    unsigned char _msg[32+1] = { 0 };
+
+    // Enforce unaligned memory addresses
+    unsigned char *pk  = (unsigned char *) ((uintptr_t)_pk | (uintptr_t)1);
+    unsigned char *sk  = (unsigned char *) ((uintptr_t)_sk | (uintptr_t)1);
+    unsigned char *sig = (unsigned char *) ((uintptr_t)_sig | (uintptr_t)1);
+    unsigned char *msg = (unsigned char *) ((uintptr_t)_msg | (uintptr_t)1);
+
+    for (int i = 0; i < 32; i++) {
+        msg[i] = i;
+    }
+
+    unsigned char seed[48] = { 0 };
+    size_t msglen = 32;
+
+    randombytes_init(seed, NULL, 256);
+
+    printf("Testing Keygen, Sign, Open: %s\n", PARAM_name(p));
+
+    int res = mayo_keypair(p, pk, sk);
+    if (res != MAYO_OK) {
+        res = -1;
+        printf("keygen failed!\n");
+        goto err;
+    }
+
+#ifdef ENABLE_CT_TESTING
+    VALGRIND_MAKE_MEM_DEFINED(pk, PARAM_cpk_bytes(p));
+#endif
+
+    size_t smlen = PARAM_sig_bytes(p) + 32;
+
+    res = mayo_sign(p, sig, &smlen, msg, 32, sk);
+    if (res != MAYO_OK) {
+        res = -1;
+        printf("sign failed!\n");
+        goto err;
+    }
+
+    printf("pk: ");
+    print_hex(pk, PARAM_cpk_bytes(p));
+    printf("sk: ");
+    print_hex(sk, PARAM_csk_bytes(p));
+    printf("sm: ");
+    print_hex(sig, smlen);
+
+#ifdef ENABLE_CT_TESTING
+    VALGRIND_MAKE_MEM_DEFINED(sig, smlen);
+#endif
+
+    res = mayo_open(p, msg, &msglen, sig, smlen, pk);
+    if (res != MAYO_OK) {
+        res = -1;
+        printf("verify failed!\n");
+        goto err;
+    }
+
+    printf("verify success!\n");
+
+    sig[0] = ~sig[0];
+    res = mayo_open(p, msg, &msglen, sig, smlen, pk);
+    if (res != MAYO_ERR) {
+        res = -1;
+        printf("wrong signature still verified!\n");
+        goto err;
+    } else {
+        res = MAYO_OK;
+    }
+
+err:
+    return res;
+}
+
+int main(int argc, char *argv[]) {
+    int rc = 0;
+
+#ifdef ENABLE_PARAMS_DYNAMIC
+    if (!strcmp(argv[1], "MAYO-1")) {
+        rc = test_mayo(&MAYO_1);
+    } else if (!strcmp(argv[1], "MAYO-2")) {
+        rc = test_mayo(&MAYO_2);
+    } else if (!strcmp(argv[1], "MAYO-3")) {
+        rc = test_mayo(&MAYO_3);
+    } else if (!strcmp(argv[1], "MAYO-5")) {
+        rc = test_mayo(&MAYO_5);
+    } else {
+        printf("unknown parameter set\n");
+        return MAYO_ERR;
+    }
+#else
+    rc = test_mayo(NULL);
+#endif
+
+    if (rc != MAYO_OK) {
+        printf("test failed for %s\n", argv[1]);
+    }
+    return rc;
+}
+

package/mayo-c/test/test_sample_solution.c

@@ -0,0 +1,75 @@
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * Test case for sample_solution failing in crypto_sign.
+ */
+#include <api.h>
+#include <mem.h>
+#include <string.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <randombytes.h>
+
+static int test_sample_solution(void) {
+
+#if CRYPTO_BYTES == 454 // MAYO-1
+    unsigned char entropy_input[48] = { 0xbf, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
+#elif CRYPTO_BYTES == 186 // MAYO-2
+    unsigned char entropy_input[48] = { 0x33, 0x34, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
+#elif CRYPTO_BYTES == 681 // MAYO-3
+    unsigned char entropy_input[48] = { 0xdb, 0x15, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
+#elif CRYPTO_BYTES == 964 // MAYO-5
+    unsigned char entropy_input[48] = { 0x87, 0x06, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
+#else
+    #error "variant not supported"
+#endif
+
+    size_t msglen = 32;
+    size_t smlen = CRYPTO_BYTES + msglen;
+
+    unsigned char *pk  = calloc(CRYPTO_PUBLICKEYBYTES, 1);
+    unsigned char *sk  = calloc(CRYPTO_SECRETKEYBYTES, 1);
+
+    unsigned char *sig = calloc(smlen, 1);
+
+    unsigned char msg[32];
+    unsigned char msgOpen[32] = { 0 };
+
+    int res = 0;
+    randombytes_init(entropy_input, NULL, 256);
+
+    for (int i = 0; i < 32; i++) {
+        msg[i] = i;
+    }
+
+    res = crypto_sign_keypair(pk, sk);
+    if (res) {
+        printf("crypto_sign_keypair failed\n");
+        goto err;
+    }
+
+    res = crypto_sign(sig, &smlen, msg, msglen, sk);
+
+    if (res) {
+        printf("crypto_sign failed\n");
+        goto err;
+    }
+
+    res = crypto_sign_open(msgOpen, &msglen, sig, smlen, pk);
+    if (res || memcmp(msg, msgOpen, msglen)) {
+        printf("crypto_sign_open failed\n");
+        res = -1;
+        goto err;
+    }
+
+err:
+    free(pk);
+    mayo_secure_free(sk, CRYPTO_SECRETKEYBYTES);
+    free(sig);
+    return res;
+}
+
+int main(int argc, char *argv[]) {
+    return test_sample_solution();
+}
+