home · contact · privacy
Simplified textfile_sizes() and replaced all get_linemax() calls with it.
[plomrogue] / src / readwrite.c
index 6d24992520b6687a99e91007231d1ec10c455651..06a63448cc9d157974e07140a5d8ad2f2710215a 100644 (file)
 /* readwrite.c */
 
 #include "readwrite.h"
-#include <stdio.h> /* for FILE typedef*/
-#include <stdint.h> /* for uint16_t, uint32_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() */
+#include "main.h"   /* for world global */
 
 
 
-/* Read/write "x" from/to "file" as bigendian representation of "size" bits. */
-static uint8_t read_uintX_bigendian(FILE * file, uint32_t * x, uint8_t size);
-static uint8_t write_uintX_bigendian(FILE * file, uint32_t x, uint8_t size);
+extern FILE * try_fopen(char * path, char * mode, char * f)
+{
+    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;
+}
 
 
 
-static uint8_t read_uintX_bigendian(FILE * file, uint32_t * x, uint8_t size)
+extern void try_fclose(FILE * file, char * f)
 {
-    if (0 != size % 8)
-    {
-        return 1;
-    }
-    int16_t bitshift = size - 8;
+    exit_trouble(fclose(file), f, "fclose()");
+}
 
-    * x = 0;
-    int test;
-    for (; bitshift >= 0; bitshift = bitshift - 8)
-    {
-        test = fgetc(file);
-        if (EOF == test)
-        {
-            return 1;
-        }
-        * x = * x + ((uint32_t) test << bitshift);
-    }
-    return 0;
+
+
+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()");
 }
 
 
 
-static uint8_t write_uintX_bigendian(FILE * file, uint32_t x, uint8_t size)
+extern void try_fputc(uint8_t c, FILE * file, char * f)
 {
-    if (0 != size % 8)
-    {
-        return 1;
-    }
-    int16_t bitshift = size - 8;
+    exit_trouble(EOF == fputc(c, file), f, "fputc()");
+}
 
-    for (; bitshift >= 0; bitshift = bitshift - 8)
-    {
-        if (EOF == fputc((x >> bitshift) & 0xFF, file))
-        {
-            return 1;
-        }
-    }
-    return 0;
+
+
+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 uint8_t read_uint8(FILE * file, uint8_t * x)
+extern int try_fgetc(FILE * file, char * f)
 {
-    uint32_t y = * x;
-    uint8_t fail = read_uintX_bigendian(file, &y, 8);
-    * x = (uint8_t) y;
-    return fail;
+    int test = fgetc(file);
+    exit_trouble(EOF == test && ferror(file), f, "fgetc()");
+    return test;
 }
 
 
 
-extern uint8_t read_uint16_bigendian(FILE * file, uint16_t * x)
+extern uint8_t try_fgetc_noeof(FILE * file, char * f)
 {
-    uint32_t y = * x;
-    uint8_t fail = read_uintX_bigendian(file, &y, 16);
-    * x = (uint16_t) y;
-    return fail;
+    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 uint8_t read_uint32_bigendian(FILE * file, uint32_t * x)
+extern void try_fclose_unlink_rename(FILE * file, char * p1, char * p2,
+                                     char * f)
 {
-    return read_uintX_bigendian(file, x, 32);
+    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 uint8_t write_uint8(uint8_t x, FILE * file)
+extern uint16_t textfile_sizes(FILE * file, uint16_t * n_lines_p)
 {
-    return write_uintX_bigendian(file, x, 8);
+    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 uint8_t write_uint16_bigendian(uint16_t x, FILE * file)
+extern uint32_t read_uint32_bigendian(FILE * file)
 {
-    return write_uintX_bigendian(file, x, 16);
+    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 uint8_t write_uint32_bigendian(uint32_t x, FILE * file)
+extern void write_uint32_bigendian(uint32_t x, FILE * file)
 {
-    return write_uintX_bigendian(file, x, 32);
+    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);
 }