home · contact · privacy
Make textfile_width() check for too large lines.
[plomrogue] / src / common / readwrite.c
1 /* src/common/readwrite.c */
2
3 #include "readwrite.h"
4 #include <stddef.h> /* size_t */
5 #include <stdint.h> /* uint8_t, uint16_t, uint32_t, UINT32_MAX */
6 #include <stdio.h> /* FILE, fseek(), sprintf(), fgets(), fgetc(), ferror(),
7                     * fputc(), fwrite(), fclose(), fopen()
8                     */
9 #include <string.h> /* strlen() */
10 #include <unistd.h> /* for access(), unlink() */
11 #include "rexit.h" /* exit_err(), exit_trouble() */
12
13
14
15 extern FILE * try_fopen(char * path, char * mode, char * f)
16 {
17     char * msg1 = "Trouble in ";
18     char * msg2 = " with fopen() (mode '";
19     char * msg3 = "') on path '";
20     char * msg4 = "'.";
21     uint16_t size = strlen(msg1) + strlen(msg2) + strlen(msg3) + strlen(msg4)
22                     + strlen(f) + strlen(path) + strlen(mode) + 1;
23     char msg[size];
24     sprintf(msg, "%s%s%s%s%s%s%s", msg1, f, msg2, mode, msg3, path, msg4);
25     FILE * file_p = fopen(path, mode);
26     exit_err(NULL == file_p, msg);
27     return file_p;
28 }
29
30
31
32 extern void try_fclose(FILE * file, char * f)
33 {
34     exit_trouble(fclose(file), f, "fclose()");
35 }
36
37
38
39 extern void try_fwrite(void * ptr, size_t size, size_t nmemb, FILE * stream,
40                        char * f)
41 {
42     exit_trouble(0 == fwrite(ptr, size, nmemb, stream), f, "fwrite()");
43 }
44
45
46
47 extern void try_fputc(uint8_t c, FILE * file, char * f)
48 {
49     exit_trouble(EOF == fputc(c, file), f, "fputc()");
50 }
51
52
53
54 extern int try_fgetc(FILE * file, char * f)
55 {
56     int test = fgetc(file);
57     exit_trouble(EOF == test && ferror(file), f, "fgetc()");
58     return test;
59 }
60
61
62
63 extern char * try_fgets(char * line, int linemax, FILE * file, char * f)
64 {
65     char * test = fgets(line, linemax, file);
66     exit_trouble(NULL == test && ferror(file), f, "fgets()");
67     return test;
68 }
69
70
71
72 extern void try_fclose_unlink_rename(FILE * file, char * p1, char * p2,
73                                      char * f)
74 {
75     try_fclose(file, f);
76     char * msg1 = "Trouble in ";
77     char * msg4 = "'.";
78     if (!access(p2, F_OK))
79     {
80         char * msg2 = " with unlink() on path '";
81         uint16_t size = strlen(msg1) + strlen(msg2) + strlen(msg4)
82                         + strlen(f) + strlen(p2) + 1;
83         char msg[size];
84         sprintf(msg, "%s%s%s%s%s", msg1, f, msg2, p2, msg4);
85         exit_err(unlink(p2), msg);
86     }
87     char * msg2 = " with rename() from '";
88     char * msg3 = "' to '";
89     uint16_t size = strlen(msg1) + strlen(f) + strlen(msg2) + strlen(p1)
90                     + strlen(msg3) + strlen(p2) + strlen(msg4) + 1;
91     char msg[size];
92     sprintf(msg, "%s%s%s%s%s%s%s", msg1, f, msg2, p1, msg3, p2, msg4);
93     exit_err(rename(p1, p2), msg);
94 }
95
96
97
98 extern uint32_t textfile_width(FILE * file)
99 {
100     char * f_name = "textfile_width()";
101     int c = 0;
102     uint32_t c_count = 0;
103     uint32_t linemax = 0;
104     while (1)
105     {
106         c = try_fgetc(file, f_name);
107         if (EOF == c)
108         {
109             break;
110         }
111         exit_trouble(UINT32_MAX == c_count, f_name, "too large text file line");
112         c_count++;
113         if ('\n' == c)
114         {
115             if (c_count > linemax)
116             {
117                 linemax = c_count;
118             }
119             c_count = 0;
120         }
121     }
122   if (0 == linemax && 0 < c_count) /* Handle files that consist of only one */
123   {                                /* line / lack newline chars.            */
124       linemax = c_count;
125    }
126    exit_trouble(-1 == fseek(file, 0, SEEK_SET), f_name, "fseek()");
127    return linemax;
128 }