Allocate just enough to read the input file.

icculus · Feb 19, 2009 · e4e7b5e · e4e7b5e
1 parent 9c33281
commit e4e7b5e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/utils/preprocess.c b/utils/preprocess.c
@@ -210,8 +210,13 @@ int main(int argc, char **argv)
         printf(" ... fopen('%s') failed.\n", infile);
     else
     {
-        char *buf = (char *) malloc(1000000);
-        int rc = fread(buf, 1, 1000000, io);
+        fseek(io, 0, SEEK_END);
+        long fsize = ftell(io);
+        fseek(io, 0, SEEK_SET);
+        if (fsize == -1)
+            fsize = 1000000;
+        char *buf = (char *) malloc(fsize);
+        const int rc = fread(buf, 1, fsize, io);
         fclose(io);
         if (rc == EOF)
             printf(" ... fread('%s') failed.\n", infile);