further cleanup

Signed-off-by: Leah Rowe <leah@libreboot.org>

1 files changed, 20 insertions, 7 deletions

diff --git a/util/libreboot-utils/lottery.c b/util/libreboot-utils/lottery.c
index 4c3b0f70..3b3cffb4 100644
--- a/util/libreboot-utils/lottery.c
+++ b/util/libreboot-utils/lottery.c
@@ -7,26 +7,39 @@
 #endif
 
 #include <stdio.h>
+#include <string.h>
+
 #include "include/common.h"
 
+#define MAX_ALLOC (1 << 17)
+
+static int lottery(char **s);
+
 int
 main(int argc, char *argv[])
 {
-	int lucky;
+	char *s = "";
+	int unlucky;
+
 #if defined(__OpenBSD__) && defined(OpenBSD)
 #if (OpenBSD) >= 509
 	if (pledge("stdio", NULL) == -1)
 		err_no_cleanup(0, errno, "openbsd won it");
 #endif
 #endif
-	setvbuf(stdout, NULL, _IONBF, 0);
-
-	lucky = win_lottery();
+	printf("%s\n", (unlucky = lottery(&s)) ? "You lose!" : "You win!");
+	return unlucky;
+}
 
-	printf("%s\n", lucky ? "You won!" : "You lose! Sorry!");
-	return lucky ? EXIT_SUCCESS : EXIT_FAILURE;
+static int
+lottery(char **s) /* are u lucky? */
+{
+	size_t size[2] = { rsize(MAX_ALLOC), rsize(MAX_ALLOC) };
+	return !(size[0] && size[0] == size[1] && size[0] <= MAX_ALLOC << 1 &&
+	    s != NULL) || memcmp(*s = mkrbuf(size[0] << 1), *s + size[0],
+	    size[0]);
 }/*
 
   ( >:3 )    
-   /| |\
+   /| |\      it could be you!
     / \                                                                      */