From 67d0733f517cc202e54702b3066968e7ddca5fb3 Mon Sep 17 00:00:00 2001
From: Ken MacKay <kmackay@gmail.com>
Date: Mon, 15 Jun 2015 21:41:21 -0700
Subject: [PATCH] Added test for deterministic ECDSA (#37)

---
 test/test_ecdsa_deterministic.c | 104 ++++++++++++++++++++++++++++++++
 uECC.h                          |   2 +-
 2 files changed, 105 insertions(+), 1 deletion(-)
 create mode 100644 test/test_ecdsa_deterministic.c

diff --git a/test/test_ecdsa_deterministic.c b/test/test_ecdsa_deterministic.c
new file mode 100644
index 0000000..17a041c
--- /dev/null
+++ b/test/test_ecdsa_deterministic.c
@@ -0,0 +1,104 @@
+/* Copyright 2014, Kenneth MacKay. Licensed under the BSD 2-clause license. */
+
+#include "uECC.h"
+
+#include <stdio.h>
+#include <string.h>
+
+#if LPC11XX
+
+#include "/Projects/lpc11xx/peripherals/uart.h"
+#include "/Projects/lpc11xx/peripherals/time.h"
+
+static uint64_t g_rand = 88172645463325252ull;
+int fake_rng(uint8_t *dest, unsigned size) {
+    while (size) {
+        g_rand ^= (g_rand << 13);
+        g_rand ^= (g_rand >> 7);
+        g_rand ^= (g_rand << 17);
+
+        unsigned amount = (size > 8 ? 8 : size);
+        memcpy(dest, &g_rand, amount);
+        size -= amount;
+    }
+    return 1;
+}
+
+#endif
+
+#define SHA256_BLOCK_LENGTH  64
+#define SHA256_DIGEST_LENGTH 32
+
+typedef struct SHA256_CTX {
+	uint32_t	state[8];
+	uint64_t	bitcount;
+	uint8_t	buffer[SHA256_BLOCK_LENGTH];
+} SHA256_CTX;
+
+extern void SHA256_Init(SHA256_CTX *ctx);
+extern void SHA256_Update(SHA256_CTX *ctx, const uint8_t *message, size_t message_size);
+extern void SHA256_Final(uint8_t digest[SHA256_DIGEST_LENGTH], SHA256_CTX *ctx);
+
+typedef struct SHA256_HashContext {
+    uECC_HashContext uECC;
+    SHA256_CTX ctx;
+} SHA256_HashContext;
+
+static void SHA256_init(uECC_HashContext *base) {
+    SHA256_HashContext *context = (SHA256_HashContext *)base;
+    SHA256_Init(&context->ctx);
+}
+
+static void SHA256_update(uECC_HashContext *base,
+                          const uint8_t *message,
+                          unsigned message_size) {
+    SHA256_HashContext *context = (SHA256_HashContext *)base;
+    SHA256_Update(&context->ctx, message, message_size);
+}
+
+static void SHA256_finish(uECC_HashContext *base, uint8_t *hash_result) {
+    SHA256_HashContext *context = (SHA256_HashContext *)base;
+    SHA256_Final(hash_result, &context->ctx);
+}
+
+int main() {
+#if LPC11XX
+    uartInit(BAUD_115200);
+	initTime();
+
+    uECC_set_rng(&fake_rng);
+#endif
+    
+    uint8_t public[uECC_BYTES * 2];
+    uint8_t private[uECC_BYTES];
+    uint8_t hash[uECC_BYTES];
+    uint8_t sig[uECC_BYTES * 2];
+    SHA256_HashContext ctx = {{&SHA256_init, &SHA256_update, &SHA256_finish}};
+    
+    int i;
+    printf("Testing 256 signatures\n");
+    for (i = 0; i < 256; ++i) {
+        printf(".");
+    #if !LPC11XX
+        fflush(stdout);
+    #endif
+        
+        if (!uECC_make_key(public, private)) {
+            printf("uECC_make_key() failed\n");
+            continue;
+        }
+        memcpy(hash, public, uECC_BYTES);
+        
+        if (!uECC_sign_deterministic(private, hash, &ctx.uECC, sig)) {
+            printf("uECC_sign() failed\n");
+            continue;
+        }
+        
+        if (!uECC_verify(public, hash, sig)) {
+            printf("uECC_verify() failed\n");
+        }
+    }
+    printf("\n");
+    
+    return 0;
+}
diff --git a/uECC.h b/uECC.h
index 4d947e9..22a81cd 100644
--- a/uECC.h
+++ b/uECC.h
@@ -183,7 +183,7 @@ void SHA256_finish(uECC_HashContext *base, uint8_t *hash_result) {
 ... when signing ...
 {
     SHA256_HashContext ctx = {{&SHA256_init, &SHA256_update, &SHA256_finish}};
-    uECC_sign_deterministic(key, message_hash, &ctx, signature);
+    uECC_sign_deterministic(key, message_hash, &ctx.uECC, signature);
 }
 */
 typedef struct uECC_HashContext {