util/libreboot-utils: finish implementing hell

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

1 files changed, 63 insertions, 20 deletions

diff --git a/util/libreboot-utils/lottery.c b/util/libreboot-utils/lottery.c
index 76548191..c66eb2b7 100644
--- a/util/libreboot-utils/lottery.c
+++ b/util/libreboot-utils/lottery.c
@@ -1,28 +1,71 @@
 /* SPDX-License-Identifier: MIT                                        ( >:3 )
- * Copyright (c) 2026 Leah Rowe <leah@libreboot.org>                    /| |\*/
-#include <stdio.h> /*                                                    / \ */
+ * Copyright (c) 2026 Leah Rowe <leah@libreboot.org>                    /| |\
+                                                                         / \ */
+#include <stdio.h>
 #include <stdint.h>
 #include <stdlib.h> /*                           (^.>)  - are u lucky?       */
 #include <string.h> /*                           \| /=                       */
 #include "include/common.h" /*                    l \                        */
+
 #define len_hell(x, y) ((((x) - (y)) < BUFSIZ) ? ((x) - (y)) : BUFSIZ)
-static int rigged(char **s) {
-	size_t size[3] = { rsize(SIZE_MAX), rsize(SIZE_MAX), 0 };
-	if (!(size[0] && size[0]==size[1] && size[0]<=SIZE_MAX && s != NULL) ||
-	    ((size[0] & 1) && (*mkrstr(1) != *mkrstr(1))))
-		return 1;
-	if (!(size[0] &= ~(size_t)1)) /*         (^:3) - it could be you!    */
-		return 0; /*                     /| |l                       */
-	if ((*s = malloc(BUFSIZ)) == NULL) /*     / \                        */
-		for (size[1] = 0; size[1] < size[0]; size[1] += BUFSIZ) {
-			rset(*s, size[2] = len_hell(size[0], size[1]));
-			if (!memcmp(*s, *s + (size[2] >> 1), size[2] >> 1))
-				return 1;
-		} else return 1;
-	return 0; /* <---- winner! */	
-} /*                                             (^.^) - come again soon!    */
-int main(int argc, char **argv) { /*             /| |l                       */
-	xpledgex("stdio", NULL); /*               / \                        */
-	printf("%s", (argc = rigged(argv))? "You lose!\n" : "");
+
+static int rigged(void)
+{
+	size_t size[5] = { 0, 0, 0, SIZE_MAX, SIZE_MAX};
+	char *b1 = NULL, *b2 = NULL;
+	char *s = NULL;
+	size_t e = 0;
+
+	for (e = 0; e < SIZE_MAX; e++) {
+	size[3] = SIZE_MAX;
+	size[4] = SIZE_MAX;
+again:
+	if (!size[3]) {
+		if (!size[4])
+			if (e == SIZE_MAX - 1)
+				return 0; /* <--- WINNER!!! */
+			else
+				continue;
+		else --size[4];
+	} else --size[3];
+
+	free_and_set_null(&b1);
+	free_and_set_null(&b2);
+
+	size[0] = rsize(SIZE_MAX); /*          \( ^o^)/ - then come play!    */
+	size[1] = rsize(SIZE_MAX); /*            |  |                        */
+/*                                               /  \                        */
+	if (!(size[0] && (size[0] == size[1]) && (size[0] <= SIZE_MAX)) ||
+	    ((size[0] & 1) && (*(b1 = mkrstr(1)) != *(b2 = mkrstr(1)))))
+		goto out; /*                   \(^-^)/  - it could be you!   */
+	if (!(size[0] &= ~(size_t)1)) /*        \ /                          */
+		goto again; /*                  / \                          */
+
+	if (s == NULL) {
+		if ((s = malloc(BUFSIZ)) == NULL)
+			goto out;
+	}
+
+	for (size[1] = 0; size[1] < size[0]; size[1] += BUFSIZ) {
+		rset(s, size[2] = len_hell(size[0], size[1]));
+		if (!memcmp(s, s + (size[2] >> 1), size[2] >> 1))
+			goto out;
+	} 
+
+	goto again;
+	}
+out:
+	free_and_set_null(&b1);
+	free_and_set_null(&b2);
+	free_and_set_null(&s);
+
+	return 1;
+}
+/*                                              \(^o^)/ - come again soon!   */
+int main(int argc, char **argv) /*               \ /                         */
+{ /*                                             / \                         */
+	xpledgex("stdio", NULL); /*                                          */
+	printf("%s", (argc = rigged()) ? "You lose!\n" : "");
+
 	return argc;
 }