X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=src%2Freadwrite.c;h=e6660ae4238ef9ed16a0ad0dfc2634c9e865b5ce;hb=60a77cb1894c678fff62840dc6b24931d839f820;hp=449e96b4c3d5d5051e08d9d45a13864870ca9513;hpb=27cee0624f3dc430199c99f000179a3385e4d7c5;p=plomrogue

diff --git a/src/readwrite.c b/src/readwrite.c
index 449e96b..e6660ae 100644
--- a/src/readwrite.c
+++ b/src/readwrite.c
@@ -1,37 +1,170 @@
-/*  readwrite.c */
+/* readwrite.c */
 
 #include "readwrite.h"
-#include <stdio.h>
-#include <stdint.h>
+#include <stdlib.h> /* for size_t */
+#include <stdio.h>  /* for FILE typedef, fopen(), fgetc(), fputc(), fseek(),
+                     * sprintf(), fwrite(), ferror()
+                     */
+#include <stdint.h> /* for uint8_t, uint16_t, uint32_t */
+#include <string.h> /* for strlen() */
+#include <unistd.h> /* for unlink() */
+#include "rexit.h"  /* for exit_err(), exit_trouble() */
 
-extern uint16_t read_uint16_bigendian( FILE * file )
+
+
+extern FILE * try_fopen(char * path, char * mode, char * f)
 {
-    uint16_t x;
-    x =     (uint16_t) fgetc(file) << 8;
-    x = x + (uint16_t) fgetc(file);
-    return x;
+    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);
+    FILE * file_p = fopen(path, mode);
+    exit_err(NULL == file_p, msg);
+    return file_p;
 }
 
-extern uint32_t read_uint32_bigendian( FILE * file )
+
+
+extern void try_fclose(FILE * file, char * f)
 {
-    uint32_t x;
-    x =       (uint32_t) fgetc(file) << 24;
-    x = x + ( (uint32_t) fgetc(file) << 16 );
-    x = x + ( (uint32_t) fgetc(file) <<  8 );
-    x = x +   (uint32_t) fgetc(file);
-    return x;
+    exit_trouble(fclose(file), f, "fclose()");
+}
+
+
+
+extern void try_fwrite(void * ptr, size_t size, size_t nmemb, FILE * stream,
+                       char * f)
+{
+    exit_trouble(0 == fwrite(ptr, size, nmemb, stream), f, "fwrite()");
+}
+
+
+
+extern void try_fputc(uint8_t c, FILE * file, char * f)
+{
+    exit_trouble(EOF == fputc(c, file), f, "fputc()");
+}
+
+
+
+extern char * try_fgets(char * line, int linemax, FILE * file, char * f)
+{
+    char * test = fgets(line, linemax, file);
+    exit_trouble(NULL == test && ferror(file), f, "fgets()");
+    return test;
+}
+
+
+
+extern int try_fgetc(FILE * file, char * f)
+{
+    int test = fgetc(file);
+    exit_trouble(EOF == test && ferror(file), f, "fgetc()");
+    return test;
+}
+
+
+
+extern uint8_t try_fgetc_noeof(FILE * file, char * f)
+{
+    char * f_name = "try_fgetc_noeof()";
+    int test = try_fgetc(file, f_name);
+    exit_trouble(EOF == test, f, "fgetc() unexpectedly hitting EOF");
+    return (uint8_t) test;
+}
+
+
+
+extern void try_fclose_unlink_rename(FILE * file, char * p1, char * p2,
+                                     char * f)
+{
+    try_fclose(file, f);
+    char * msg1 = "Trouble in ";
+    char * msg4 = "'.";
+    if (!access(p2, F_OK))
+    {
+        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);
+        exit_err(unlink(p2), 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);
+    exit_err(rename(p1, p2), msg);
+}
+
+
+
+extern uint16_t textfile_sizes(FILE * file, uint16_t * n_lines_p)
+{
+    char * f_name = "textfile_sizes()";
+    int c = 0;
+    uint16_t c_count = 0;
+    uint16_t n_lines = 0;
+    uint16_t linemax = 0;
+    while (1)
+    {
+        c = try_fgetc(file, f_name);
+        if (EOF == c)
+        {
+            break;
+        }
+        c_count++;
+        if ('\n' == c)
+        {
+            if (c_count > linemax)
+            {
+                linemax = c_count;
+            }
+            c_count = 0;
+            if (n_lines_p)
+            {
+                n_lines++;
+            }
+        }
+    }
+    if (0 == linemax && 0 < c_count) /* Handle files that consist of only one */
+    {                                /* line / lack newline chars.            */
+        linemax = c_count;
+    }
+    exit_trouble(-1 == fseek(file, 0, SEEK_SET), f_name, "fseek()");
+    if (n_lines_p)
+    {
+        * n_lines_p = n_lines;
+    }
+    return linemax;
 }
 
-extern void write_uint16_bigendian( uint16_t x, FILE * file )
+
+
+extern uint32_t read_uint32_bigendian(FILE * file)
 {
-    fputc( x >> 8,   file );
-    fputc( x & 0xFF, file );
+    char * f_name = "read_uint32_bigendian()";
+    uint32_t x;
+    x =       (uint32_t) try_fgetc_noeof(file, f_name) << 24;
+    x = x + ( (uint32_t) try_fgetc_noeof(file, f_name) << 16 );
+    x = x + ( (uint32_t) try_fgetc_noeof(file, f_name) <<  8 );
+    x = x +   (uint32_t) try_fgetc_noeof(file, f_name);
+    return x;
 }
 
-extern void write_uint32_bigendian( uint32_t x, FILE * file )
+
+
+extern void write_uint32_bigendian(uint32_t x, FILE * file)
 {
-    fputc(   x >> 24,          file);
-    fputc( ( x >> 16 ) & 0xFF, file);
-    fputc( ( x >>  8 ) & 0xFF, file);
-    fputc(   x         & 0xFF, file);
+    char * f_name = "write_uint32_bigendian()";
+    try_fputc(   x >> 24,          file, f_name);
+    try_fputc( ( x >> 16 ) & 0xFF, file, f_name);
+    try_fputc( ( x >>  8 ) & 0xFF, file, f_name);
+    try_fputc(   x         & 0xFF, file, f_name);
 }