Add LFIDEBUG check

zyedidia · Nov 28, 2024 · cba9d08 · cba9d08
1 parent 7d83297
commit cba9d08
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lfi-leg/main.c b/lfi-leg/main.c
@@ -257,6 +257,18 @@ main(int argc, char** argv)
     input = argopen(args.input, "r", stdin);
     output = argopen(args.output, "w", stdout);
 
+    char* lfidebug = getenv("LFIDEBUG");
+    if (lfidebug) {
+        // write input to a debug location
+        FILE* debug = argopen("debug-in.s", "w", NULL);
+        char buf[BUFSIZ];
+        size_t n;
+        while ((n = fread(buf, 1, BUFSIZ, input)) != 0)
+            fwrite(buf, 1, n, debug);
+        fseek(input, 0, SEEK_SET);
+        fclose(debug);
+    }
+
     if (args.arch == NULL) {
         args.arch = getarch();
     }