From d1ba9bae0358d87b8dc73d89b7df956722c26997 Mon Sep 17 00:00:00 2001 From: Leah Rowe Date: Thu, 26 Mar 2026 11:41:37 +0000 Subject: further cleanup Signed-off-by: Leah Rowe --- util/libreboot-utils/lottery.c | 35 +++++++++++++++++++++-------------- 1 file changed, 21 insertions(+), 14 deletions(-) (limited to 'util/libreboot-utils/lottery.c') 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 */ -#ifdef __OpenBSD__ -#include /* pledge(2) */ -#endif - #include +#include + #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! / \ */ -- cgit v1.2.1