X-Git-Url: https://plomlompom.com/repos/processes?a=blobdiff_plain;f=src%2Fcommon%2Freadwrite.c;h=c27e86143bc7feb5ff0471df654f70ff960d4a33;hb=778534bf6946fe0fef17e353c55678d248d8d09d;hp=c2c970d292cb9579d5e6fddc64eb733ac3524246;hpb=ef60270da2acf29953b411632e26503d7787d9d8;p=plomrogue

diff --git a/src/common/readwrite.c b/src/common/readwrite.c
index c2c970d..c27e861 100644
--- a/src/common/readwrite.c
+++ b/src/common/readwrite.c
@@ -4,26 +4,31 @@
 #include <stddef.h> /* size_t */
 #include <stdint.h> /* uint8_t, uint16_t, uint32_t, UINT32_MAX */
 #include <stdio.h> /* FILE, fseek(), sprintf(), fgets(), fgetc(), ferror(),
-                    * fputc(), fwrite(), fclose(), fopen()
+                    * fputc(), fwrite(), fclose(), fopen(), clearerr()
                     */
+#include <stdlib.h> /* free() */
 #include <string.h> /* strlen() */
-#include <unistd.h> /* for access(), unlink() */
+#include <unistd.h> /* access(), unlink() */
 #include "rexit.h" /* exit_err(), exit_trouble() */
+#include "try_malloc.h" /* try_malloc() */
 
 
 
 extern FILE * try_fopen(char * path, char * mode, char * f)
 {
+    char * f_name = "try_fopen()";
     char * msg1 = "Trouble in ";
     char * msg2 = " with fopen() (mode '";
     char * msg3 = "') on path '";
     char * msg4 = "'.";
     uint16_t size = strlen(msg1) + strlen(msg2) + strlen(msg3) + strlen(msg4)
                     + strlen(f) + strlen(path) + strlen(mode) + 1;
-    char msg[size];
-    sprintf(msg, "%s%s%s%s%s%s%s", msg1, f, msg2, mode, msg3, path, msg4);
+    char * msg = try_malloc(size, f_name);
+    int test = sprintf(msg, "%s%s%s%s%s%s%s", msg1,f,msg2,mode,msg3,path,msg4);
+    exit_trouble(test < 0, f_name, "sprintf()");
     FILE * file_p = fopen(path, mode);
     exit_err(NULL == file_p, msg);
+    free(msg);
     return file_p;
 }
 
@@ -53,6 +58,7 @@ extern void try_fputc(uint8_t c, FILE * file, char * f)
 
 extern int try_fgetc(FILE * file, char * f)
 {
+    clearerr(file); /* OSX' (BSD?) fgetc() needs this to undo previous EOFs. */
     int test = fgetc(file);
     exit_trouble(EOF == test && ferror(file), f, "fgetc()");
     return test;
@@ -72,6 +78,7 @@ extern char * try_fgets(char * line, int linemax, FILE * file, char * f)
 extern void try_fclose_unlink_rename(FILE * file, char * p1, char * p2,
                                      char * f)
 {
+    char * f_name = "try_fclose_unlink_rename()";
     try_fclose(file, f);
     char * msg1 = "Trouble in ";
     char * msg4 = "'.";
@@ -80,17 +87,21 @@ extern void try_fclose_unlink_rename(FILE * file, char * p1, char * p2,
         char * msg2 = " with unlink() on path '";
         uint16_t size = strlen(msg1) + strlen(msg2) + strlen(msg4)
                         + strlen(f) + strlen(p2) + 1;
-        char msg[size];
-        sprintf(msg, "%s%s%s%s%s", msg1, f, msg2, p2, msg4);
+        char * msg = try_malloc(size, f_name);
+        int test = sprintf(msg, "%s%s%s%s%s", msg1, f, msg2, p2, msg4);
+        exit_trouble(test < 0, f_name, "sprintf()");
         exit_err(unlink(p2), msg);
+        free(msg);
     }
     char * msg2 = " with rename() from '";
     char * msg3 = "' to '";
     uint16_t size = strlen(msg1) + strlen(f) + strlen(msg2) + strlen(p1)
                     + strlen(msg3) + strlen(p2) + strlen(msg4) + 1;
-    char msg[size];
-    sprintf(msg, "%s%s%s%s%s%s%s", msg1, f, msg2, p1, msg3, p2, msg4);
+    char * msg = try_malloc(size, f_name);
+    int test = sprintf(msg, "%s%s%s%s%s%s%s", msg1,f,msg2,p1,msg3,p2,msg4);
+    exit_trouble(test < 0, f_name, "sprintf()");
     exit_err(rename(p1, p2), msg);
+    free(msg);
 }