From a258b4bbea7b955321ad124aa5f2708e71fe5fc7 Mon Sep 17 00:00:00 2001
From: ari melody <ari@arimelody.me>
Date: Mon, 12 Feb 2024 02:14:41 +0000
Subject: [PATCH] flush console to ensure progress output

---
 main.c | 71 +++++++++++++++++++++++++++++++---------------------------
 1 file changed, 38 insertions(+), 33 deletions(-)

diff --git a/main.c b/main.c
index 3353cae..960eb30 100644
--- a/main.c
+++ b/main.c
@@ -3,46 +3,51 @@
 
 int main() {
 
-    unsigned long long int size = 0;
-    unsigned int gib = 0;
-    
-    printf("please enter the number of gibibytes (GiB) you would like to allocate (1 GiB = 1073741824 bytes):\n> ");
-    unsigned int digit = 0;
-    while (1) {
+        unsigned long long int size = 0;
+        unsigned int gib = 0;
+
+        printf("please enter the number of gibibytes (GiB) you would like to allocate (1 GiB = 1073741824 bytes):\n> ");
+        unsigned int digit = 0;
+        while (1) {
+                unsigned char input = getchar();
+                if (input == '\n') break;
+                if (input < '0' || input > '9') continue;
+
+                if (digit > 0) gib = gib * 10;
+
+                gib = gib + (unsigned int) input - 48;
+                digit++;
+        }
+
+        size = (unsigned long long int) gib * 1024 * 1024 * 1024;
+
+        printf("you are about to allocate %d GiB (%llu bytes) of heap memory. are you sure? (y/n)\n> ", gib, size);
+
         unsigned char input = getchar();
-        if (input == '\n') break;
-        if (input < '0' || input > '9') continue;
+        if (input != 'y') return 0;
+        while (getchar() != '\n');
 
-        if (digit > 0) gib = gib * 10;
+        printf("please wait");
+        fflush(stdout);
 
-        gib = gib + (unsigned int) input - 48;
-        digit++;
-    }
+        unsigned long long int* wtf = (unsigned long long int*) malloc(size);
 
-    size = (unsigned long long int) gib * 1024 * 1024 * 1024;
+        for (unsigned long long int i = 0; i < size / sizeof(long long int); i++) {
+                wtf[i] = 0x80085;
+                if (i > 0 && i % (1024 * 1024 * 1024 / sizeof(long long int)) == 0) {
+                        printf(".");
+                        fflush(stdout);
+                }
+        }
+        printf("\n");
 
-    printf("you are about to allocate %d GiB (%llu bytes) of heap memory. are you sure? (y/n)\n> ", gib, size);
+        printf("%d GiB (%llu bytes) of heap memory has been allocated. you are insane.\n", gib, size);
+        printf("press ENTER to release this memory, or CTRL+C to exit the program.\n");
 
-    unsigned char input = getchar();
-    if (input != 'y') return 0;
-    while (getchar() != '\n');
+        getchar();
 
-    printf("please wait");
-    unsigned long long int* wtf = (unsigned long long int*) malloc(size);
+        free(wtf);
 
-    for (unsigned long long int i = 0; i < size / sizeof(long long int); i++) {
-        wtf[i] = 0x80085;
-        if (i > 0 && i % (1024 * 1024 * 1024 / sizeof(long long int)) == 0) printf(".");
-    }
-    printf("\n");
-
-    printf("%d GiB (%llu bytes) of heap memory has been allocated. you are insane.\n", gib, size);
-    printf("press ENTER to release this memory, or CTRL+C to exit the program.\n");
-
-    getchar();
-
-    free(wtf);
-
-    return 0;
+        return 0;
 
 }