1 files changed, 257 insertions, 0 deletions

diff --git a/sys/dev/random/unit_test.c b/sys/dev/random/unit_test.c
new file mode 100644
index 000000000000..32b3363e247b
--- /dev/null
+++ b/sys/dev/random/unit_test.c
@@ -0,0 +1,257 @@
+/*-
+ * Copyright (c) 2000-2013 Mark R V Murray
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer
+ *    in this position and unchanged.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * $FreeBSD$
+ */
+
+/*
+ Build this by going:
+
+cc -g -O0 -pthread -DRANDOM_<alg> -DRANDOM_DEBUG -I../.. -lstdthreads -Wall \
+	unit_test.c \
+	yarrow.c \
+	fortuna.c \
+	hash.c \
+	../../crypto/rijndael/rijndael-api-fst.c \
+	../../crypto/rijndael/rijndael-alg-fst.c \
+	../../crypto/sha2/sha2.c \
+	-o unit_test
+./unit_test
+
+Where <alg> is YARROW or FORTUNA.
+*/
+
+#include <sys/types.h>
+#include <inttypes.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <threads.h>
+#include <unistd.h>
+
+#include "unit_test.h"
+
+#ifdef RANDOM_YARROW
+#include "dev/random/yarrow.h"
+#endif
+#ifdef RANDOM_FORTUNA
+#include "dev/random/fortuna.h"
+#endif
+
+#define NUM_THREADS	  3
+
+static volatile int stopseeding = 0;
+
+void
+random_adaptor_unblock(void)
+{
+
+#if 0
+	if (mtx_trylock(&random_reseed_mtx) == thrd_busy)
+		printf("Mutex held. Good.\n");
+	else {
+		printf("Mutex not held. PANIC!!\n");
+		thrd_exit(0);
+	}
+#endif
+	printf("random: unblocking device.\n");
+}
+
+static int
+RunHarvester(void *arg __unused)
+{
+	int i, r;
+	struct harvest_event e;
+
+	for (i = 0; ; i++) {
+		if (stopseeding)
+			break;
+		if (i % 1000 == 0)
+			printf("Harvest: %d\n", i);
+		r = random()%10;
+		e.he_somecounter = i;
+		*((uint64_t *)e.he_entropy) = random();
+		e.he_size = 8;
+		e.he_bits = random()%4;
+		e.he_destination = i;
+		e.he_source = (i + 3)%7;
+		e.he_next = NULL;
+#ifdef RANDOM_YARROW
+		random_yarrow_process_event(&e);
+#endif
+#ifdef RANDOM_FORTUNA
+		random_fortuna_process_event(&e);
+#endif
+		usleep(r);
+	}
+
+	printf("Thread #0 ends\n");
+
+	thrd_exit(0);
+
+	return (0);
+}
+
+static int
+WriteCSPRNG(void *threadid)
+{
+	uint8_t *buf;
+	int i;
+
+	printf("Thread #1 starts\n");
+
+	for (i = 0; ; i++) {
+		if (stopseeding)
+			break;
+		buf = malloc(4096);
+		if (i % 1000 == 0)
+			printf("Thread write 1 - %d\n", i);
+		if (buf != NULL) {
+#ifdef RANDOM_YARROW
+			random_yarrow_write(buf, i);
+#endif
+#ifdef RANDOM_FORTUNA
+			random_fortuna_write(buf, i);
+#endif
+			free(buf);
+		}
+		usleep(1000000);
+	}
+
+	printf("Thread #1 ends\n");
+
+	thrd_exit(0);
+
+	return (0);
+}
+
+static int
+ReadCSPRNG(void *threadid)
+{
+	size_t tid;
+	uint8_t *buf;
+	int i;
+
+	tid = (size_t)threadid;
+	printf("Thread #%zd starts\n", tid);
+
+#ifdef RANDOM_YARROW
+	while (!random_yarrow_seeded())
+#endif
+#ifdef RANDOM_FORTUNA
+	while (!random_fortuna_seeded())
+#endif
+	{
+#ifdef RANDOM_YARROW
+		random_yarrow_read(NULL, 0);
+		random_yarrow_read(NULL, 1);
+#endif
+#ifdef RANDOM_FORTUNA
+		random_fortuna_read(NULL, 0);
+		random_fortuna_read(NULL, 1);
+#endif
+		usleep(100);
+	}
+
+	for (i = 0; i < 100000; i++) {
+		buf = malloc(i);
+		if (i % 1000 == 0)
+			printf("Thread read %zd - %d %p\n", tid, i, buf);
+		if (buf != NULL) {
+#ifdef RANDOM_YARROW
+			random_yarrow_read(NULL, 0);
+			random_yarrow_read(buf, i);
+			random_yarrow_read(NULL, 1);
+#endif
+#ifdef RANDOM_FORTUNA
+			random_fortuna_read(NULL, 0);
+			random_fortuna_read(buf, i);
+			random_fortuna_read(NULL, 1);
+#endif
+#if 0
+			{
+			int j;
+
+			for (j = 0; j < i; j++) {
+				printf(" %02X", buf[j]);
+				if (j % 32 == 31 || j == i - 1)
+					printf("\n");
+			}
+			}
+#endif
+			free(buf);
+		}
+		usleep(100);
+	}
+
+	printf("Thread #%zd ends\n", tid);
+
+	thrd_exit(0);
+
+	return (0);
+}
+
+int
+main(int argc, char *argv[])
+{
+	thrd_t threads[NUM_THREADS];
+	int rc;
+	long t;
+
+#ifdef RANDOM_YARROW
+	random_yarrow_init_alg();
+#endif
+#ifdef RANDOM_FORTUNA
+	random_fortuna_init_alg();
+#endif
+
+	for (t = 0; t < NUM_THREADS; t++) {
+		printf("In main: creating thread %ld\n", t);
+		rc = thrd_create(&threads[t], (t == 0 ? RunHarvester : (t == 1 ? WriteCSPRNG : ReadCSPRNG)), t);
+		if (rc != thrd_success) {
+			printf("ERROR; return code from thrd_create() is %d\n", rc);
+			exit(-1);
+		}
+	}
+
+	for (t = 2; t < NUM_THREADS; t++)
+		thrd_join(threads[t], &rc);
+
+	stopseeding = 1;
+
+	thrd_join(threads[1], &rc);
+	thrd_join(threads[0], &rc);
+
+#ifdef RANDOM_YARROW
+	random_yarrow_deinit_alg();
+#endif
+#ifdef RANDOM_FORTUNA
+	random_fortuna_deinit_alg();
+#endif
+
+	/* Last thing that main() should do */
+	thrd_exit(0);
+
+	return (0);
+}