From 3b4fd2f1dfb8e93d292db615004eec38c2c086e4 Mon Sep 17 00:00:00 2001
From: Dan Sheppard <dan.sheppard.circle@gmail.com>
Date: Tue, 22 Apr 2025 16:10:48 +0100
Subject: [PATCH] Add sha2 tests to infra.

---
 src/sha2.c      | 103 ++++++++++++++++++++++++++++++++++++++++++++++++
 src/sha2.h      |   4 ++
 src/test.c      |   2 +
 src/test_sha2.c | 103 ------------------------------------------------
 4 files changed, 109 insertions(+), 103 deletions(-)
 delete mode 100644 src/test_sha2.c

diff --git a/src/sha2.c b/src/sha2.c
index e5264c8..ee4b0df 100644
--- a/src/sha2.c
+++ b/src/sha2.c
@@ -353,3 +353,106 @@ void sha2_init_hmac(struct sha2_ctx_t *ctx, int sha2_variety,
     sha2_more(ctx,i_pad,block_size);
     ctx->flags |= FLAG_HMAC;
 }
+
+#ifdef COQUET_TEST
+
+#include <stdlib.h>
+
+static void to_hex(char *hex, uint8_t *data, int len) {
+    int i;
+    char *h;
+
+    h = hex;
+    for(i=0;i<len;i++) {
+        snprintf(h,3,"%2.2x",data[i]);
+        h += 2;
+    }
+    *h = '\0';
+}
+
+static void compare(char * test_name, int variety, char *input, char *value, char *key) {
+    struct sha2_ctx_t ctx;
+    int len;
+    uint8_t out[SHA2_MAX_DIGEST_SIZE];
+    char hex[SHA2_MAX_DIGEST_SIZE*2+1];
+
+    if(key) {
+        sha2_init_hmac(&ctx,variety,(uint8_t *)key,strlen(key));
+    } else {
+        sha2_init(&ctx,variety);
+    }
+    sha2_more(&ctx,(uint8_t *)input,strlen(input));
+    len = sha2_finish(&ctx,out,SHA2_MAX_DIGEST_SIZE);
+    to_hex(hex,out,len);
+    if(strcmp(hex,value)) {
+        printf("test '%s' failed: expected='%s' got='%s'\n",test_name,value,hex);
+        exit(1);
+    }
+    printf("ok %s\n",test_name);
+}
+
+static void test_trunc() {
+    struct sha2_ctx_t ctx;
+    uint8_t out[SHA2_MAX_DIGEST_SIZE+1];
+    char hex[SHA2_MAX_DIGEST_SIZE*2+1];
+    int len,got;
+    char *expect_all = "cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e";
+    char expect[SHA2_MAX_DIGEST_SIZE*2+1];
+
+    for(len=0;len<SHA2_MAX_DIGEST_SIZE;len++) {
+        sha2_init(&ctx,SHA2_VARIETY_512);
+        out[len] = len&0xFF;
+        got = sha2_finish(&ctx,out,len);
+        to_hex(hex,out,got);
+        memcpy(expect,expect_all,len*2);
+        expect[len*2] = '\0';
+        if(strcmp(hex,expect)) {
+            printf("test trunc failed at len=%d: expected='%s' got='%s'\n",
+                    len,expect,hex);
+            exit(1);
+        }
+        if(out[len] != (len&0xFF)) {
+            printf("test trunc failed with overrun at len=%d\n",len);
+            exit(1);
+        }
+    }
+    printf("ok test-trunc\n");
+}
+
+#define MAX_LEN 1000
+static void test_lengths() { /* Only useful with valgrind */
+    struct sha2_ctx_t ctx;
+    uint8_t text[MAX_LEN],out[SHA2_MAX_DIGEST_SIZE];
+    int len;
+
+    memset(text,42,MAX_LEN);
+    for(len=0;len<1000;len++) {
+        sha2_init(&ctx,SHA2_VARIETY_512);
+        sha2_more(&ctx,text,len);
+        sha2_finish(&ctx,out,SHA2_MAX_DIGEST_SIZE);
+    }
+}
+
+int test_sha2() {
+    char *dog = "The quick brown fox jumps over the lazy dog";
+    char *k = "abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz";
+
+    printf("Run me under valgrind too!\n");
+    compare("empty-512",SHA2_VARIETY_512,"","cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e",NULL);
+    compare("empty-384",SHA2_VARIETY_384,"","38b060a751ac96384cd9327eb1b1e36a21fdb71114be07434c0cc7bf63f6e1da274edebfe76f65fbd51ad2f14898b95b",NULL);
+    compare("empty-256",SHA2_VARIETY_256,"","e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",NULL);
+    compare("empty-256",SHA2_VARIETY_224,"","d14a028c2a3a2bc9476102bb288234c415a2b01f828ea62ac5b3e42f",NULL);
+    compare("empty-512/256",SHA2_VARIETY_512_256,"","c672b8d1ef56ed28ab87c3622c5114069bdd3ad7b8f9737498d0c01ecef0967a",NULL);
+    compare("empty-512/224",SHA2_VARIETY_512_224,"","6ed0dd02806fa89e25de060c19d3ac86cabb87d6a0ddd05c333b84f4",NULL);
+    compare("dog-512",SHA2_VARIETY_512,dog,"07e547d9586f6a73f73fbac0435ed76951218fb7d0c8d788a309d785436bbb642e93a252a954f23912547d1e8a3b5ed6e1bfd7097821233fa0538f3db854fee6",NULL);
+    compare("dog-384",SHA2_VARIETY_384,dog,"ca737f1014a48f4c0b6dd43cb177b0afd9e5169367544c494011e3317dbf9a509cb1e5dc1e85a941bbee3d7f2afbc9b1",NULL);
+    compare("dog-256",SHA2_VARIETY_256,dog,"d7a8fbb307d7809469ca9abcb0082e4f8d5651e46d3cdb762d02d0bf37c9e592",NULL);
+    compare("dog-224",SHA2_VARIETY_224,dog,"730e109bd7a8a32b1cb9d9a09aa2325d2430587ddbc0c38bad911525",NULL);
+    compare("hmac-512",SHA2_VARIETY_512,dog,"2a31d580d74d604de3dce055477d0a5633411adeafa044e10a2c6cfee6e38df49ed336cb53e3e7fa6bbbf3f107a3067296560be3deb09afcaff9cb98d2169433",k);
+    compare("hmac-224",SHA2_VARIETY_224,dog,"610d38da56e06cf7d15bdf1ad83e250ae77ada28b5648036bba614ee",k);
+    test_lengths();
+    test_trunc();
+    return 0;
+}
+
+#endif
diff --git a/src/sha2.h b/src/sha2.h
index 5b7a4c3..7526ca4 100644
--- a/src/sha2.h
+++ b/src/sha2.h
@@ -74,4 +74,8 @@ void sha2_more(struct sha2_ctx_t *ctx, uint8_t *data, int len);
 /* Fill up the buffer provided with the digest, upto maxlen bytes */
 int sha2_finish(struct sha2_ctx_t *ctx, uint8_t *out, int maxlen);
 
+#ifdef COQUET_TEST
+int test_sha2();
+#endif
+
 #endif
diff --git a/src/test.c b/src/test.c
index bbc8db7..e73f892 100644
--- a/src/test.c
+++ b/src/test.c
@@ -2,6 +2,7 @@
 #include <stdlib.h>
 #include <errno.h>
 #include <string.h>
+#include "sha2.h"
 #include "superblock.h"
 
 void test_unlink(char *path) {
@@ -33,6 +34,7 @@ void test_bail(coquet_t * cq, int error_code) {
 
 int main() {
     printf("testing\n");
+    test_sha2();
     test_superblock();
     return 0;
 }
diff --git a/src/test_sha2.c b/src/test_sha2.c
deleted file mode 100644
index 7c7167d..0000000
--- a/src/test_sha2.c
+++ /dev/null
@@ -1,103 +0,0 @@
-#include <stdio.h>
-#include <stdlib.h>
-#include <stdint.h>
-#include <string.h>
-#include "sha2.h"
-
-static void to_hex(char *hex, uint8_t *data, int len) {
-    int i;
-    char *h;
-
-    h = hex;
-    for(i=0;i<len;i++) {
-        snprintf(h,3,"%2.2x",data[i]);
-        h += 2;
-    }
-    *h = '\0';
-}
-
-static void compare(char * test_name, int variety, char *input, char *value, char *key) {
-    struct sha2_ctx_t ctx;
-    int len;
-    uint8_t out[SHA2_MAX_DIGEST_SIZE];
-    char hex[SHA2_MAX_DIGEST_SIZE*2+1];
-
-    if(key) {
-        sha2_init_hmac(&ctx,variety,(uint8_t *)key,strlen(key));
-    } else {
-        sha2_init(&ctx,variety);
-    }
-    sha2_more(&ctx,(uint8_t *)input,strlen(input));
-    len = sha2_finish(&ctx,out,SHA2_MAX_DIGEST_SIZE);
-    to_hex(hex,out,len);
-    if(strcmp(hex,value)) {
-        printf("test '%s' failed: expected='%s' got='%s'\n",test_name,value,hex);
-        exit(1);
-    }
-    printf("ok %s\n",test_name);
-}
-
-static void test_trunc() {
-    struct sha2_ctx_t ctx;
-    uint8_t out[SHA2_MAX_DIGEST_SIZE+1];
-    char hex[SHA2_MAX_DIGEST_SIZE*2+1];
-    int len,got;
-    char *expect_all = "cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e";
-    char expect[SHA2_MAX_DIGEST_SIZE*2+1];
-
-    for(len=0;len<SHA2_MAX_DIGEST_SIZE;len++) {
-        sha2_init(&ctx,SHA2_VARIETY_512);
-        out[len] = len&0xFF;
-        got = sha2_finish(&ctx,out,len);
-        to_hex(hex,out,got);
-        memcpy(expect,expect_all,len*2);
-        expect[len*2] = '\0';
-        if(strcmp(hex,expect)) {
-            printf("test trunc failed at len=%d: expected='%s' got='%s'\n",
-                    len,expect,hex);
-            exit(1);
-        }
-        if(out[len] != (len&0xFF)) {
-            printf("test trunc failed with overrun at len=%d\n",len);
-            exit(1);
-        }
-    }
-    printf("ok test-trunc\n");
-}
-
-#define MAX_LEN 1000
-static void test_lengths() { /* Only useful with valgrind */
-    struct sha2_ctx_t ctx;
-    uint8_t text[MAX_LEN],out[SHA2_MAX_DIGEST_SIZE];
-    int len;
-
-    memset(text,42,MAX_LEN);
-    for(len=0;len<1000;len++) {
-        sha2_init(&ctx,SHA2_VARIETY_512);
-        sha2_more(&ctx,text,len);
-        sha2_finish(&ctx,out,SHA2_MAX_DIGEST_SIZE);
-    }
-}
-
-
-int main() {
-    char *dog = "The quick brown fox jumps over the lazy dog";
-    char *k = "abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz";
-
-    printf("Run me under valgrind too!\n");
-    compare("empty-512",SHA2_VARIETY_512,"","cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e",NULL);
-    compare("empty-384",SHA2_VARIETY_384,"","38b060a751ac96384cd9327eb1b1e36a21fdb71114be07434c0cc7bf63f6e1da274edebfe76f65fbd51ad2f14898b95b",NULL);
-    compare("empty-256",SHA2_VARIETY_256,"","e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",NULL);
-    compare("empty-256",SHA2_VARIETY_224,"","d14a028c2a3a2bc9476102bb288234c415a2b01f828ea62ac5b3e42f",NULL);
-    compare("empty-512/256",SHA2_VARIETY_512_256,"","c672b8d1ef56ed28ab87c3622c5114069bdd3ad7b8f9737498d0c01ecef0967a",NULL);
-    compare("empty-512/224",SHA2_VARIETY_512_224,"","6ed0dd02806fa89e25de060c19d3ac86cabb87d6a0ddd05c333b84f4",NULL);
-    compare("dog-512",SHA2_VARIETY_512,dog,"07e547d9586f6a73f73fbac0435ed76951218fb7d0c8d788a309d785436bbb642e93a252a954f23912547d1e8a3b5ed6e1bfd7097821233fa0538f3db854fee6",NULL);
-    compare("dog-384",SHA2_VARIETY_384,dog,"ca737f1014a48f4c0b6dd43cb177b0afd9e5169367544c494011e3317dbf9a509cb1e5dc1e85a941bbee3d7f2afbc9b1",NULL);
-    compare("dog-256",SHA2_VARIETY_256,dog,"d7a8fbb307d7809469ca9abcb0082e4f8d5651e46d3cdb762d02d0bf37c9e592",NULL);
-    compare("dog-224",SHA2_VARIETY_224,dog,"730e109bd7a8a32b1cb9d9a09aa2325d2430587ddbc0c38bad911525",NULL);
-    compare("hmac-512",SHA2_VARIETY_512,dog,"2a31d580d74d604de3dce055477d0a5633411adeafa044e10a2c6cfee6e38df49ed336cb53e3e7fa6bbbf3f107a3067296560be3deb09afcaff9cb98d2169433",k);
-    compare("hmac-224",SHA2_VARIETY_224,dog,"610d38da56e06cf7d15bdf1ad83e250ae77ada28b5648036bba614ee",k);
-    test_lengths();
-    test_trunc();
-    return 0;
-}
-- 
2.30.2