further cleanup

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

1 files changed, 21 insertions, 14 deletions

diff --git a/util/libreboot-utils/lottery.c b/util/libreboot-utils/lottery.c
index 4c3b0f70..dcede791 100644
--- a/util/libreboot-utils/lottery.c
+++ b/util/libreboot-utils/lottery.c
@@ -2,31 +2,38 @@
  * Copyright (c) 2026 Leah Rowe <leah@libreboot.org>
  */
 
-#ifdef __OpenBSD__
-#include <sys/param.h> /* pledge(2) */
-#endif
-
 #include <stdio.h>
+#include <string.h>
+
 #include "include/common.h"
+#define MAX_ALLOC (1 << 17)
+
+static int rigged(char **s);
 
 int
 main(int argc, char *argv[])
 {
-	int lucky;
-#if defined(__OpenBSD__) && defined(OpenBSD)
-#if (OpenBSD) >= 509
+	char *s = "You lose!";
+	int lucky = rigged(&s);
+
+#ifdef __OpenBSD__
 	if (pledge("stdio", NULL) == -1)
-		err_no_cleanup(0, errno, "openbsd won it");
+		err_no_cleanup(0, errno, "openbsd wins");
 #endif
-#endif
-	setvbuf(stdout, NULL, _IONBF, 0);
+	printf("%s\n", s);
+	return lucky;
+}
 
-	lucky = win_lottery();
+static int
+rigged(char **s) /* are u lucky? */
+{
+	size_t size[2] = { rsize(MAX_ALLOC), rsize(MAX_ALLOC) };
 
-	printf("%s\n", lucky ? "You won!" : "You lose! Sorry!");
-	return lucky ? EXIT_SUCCESS : EXIT_FAILURE;
+	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!
     / \                                                                      */