1 /* Test of getndelim2() function.
2 Copyright (C) 2008-2018 Free Software Foundation, Inc.
3
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 3, or (at your option)
7 any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, see <https://www.gnu.org/licenses/>. */
16
17 /* Written by Eric Blake <ebb9@byu.net>, 2008. */
18
19 #include <config.h>
20
21 #include "getndelim2.h"
22
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26
27 #include "macros.h"
28
29 int
main(void)30 main (void)
31 {
32 FILE *f;
33 char *line = NULL;
34 size_t len = 0;
35 ssize_t result;
36
37 /* Create test file. */
38 f = fopen ("test-getndelim2.txt", "wb+");
39 if (!f || fwrite ("a\nbc\nd\0f", 1, 8, f) != 8)
40 {
41 fputs ("Failed to create sample file.\n", stderr);
42 remove ("test-getndelim2.txt");
43 return 1;
44 }
45 rewind (f);
46
47 /* Unlimited lines. */
48
49 /* Test initial allocation, which must include trailing NUL. */
50 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', '\n', f);
51 ASSERT (result == 2);
52 ASSERT (strcmp (line, "a\n") == 0);
53 ASSERT (2 < len);
54
55 /* Test growth of buffer, must not leak. */
56 free (line);
57 line = malloc (1);
58 len = 0;
59 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, EOF, '\n', f);
60 ASSERT (result == 3);
61 ASSERT (strcmp (line, "bc\n") == 0);
62 ASSERT (3 < len);
63
64 /* Test embedded NULs and EOF behavior. */
65 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', EOF, f);
66 ASSERT (result == 3);
67 ASSERT (memcmp (line, "d\0f", 4) == 0);
68 ASSERT (3 < len);
69
70 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', EOF, f);
71 ASSERT (result == -1);
72
73 /* Using offset and nmax. */
74
75 /* Limit growth to four bytes, including NUL, but parse to next 'd'. */
76 free (line);
77 rewind (f);
78 line = malloc (8);
79 memset (line, 'e', 8);
80 len = 8;
81 result = getndelim2 (&line, &len, 6, 10, 'd', 'd', f);
82 ASSERT (result == 3);
83 ASSERT (10 == len);
84 ASSERT (strcmp (line, "eeeeeea\nb") == 0);
85
86 /* No change if offset larger than limit. */
87 result = getndelim2 (&line, &len, len, 1, EOF, EOF, f);
88 ASSERT (result == -1);
89 ASSERT (10 == len);
90 ASSERT (strcmp (line, "eeeeeea\nb") == 0);
91
92 /* Consume to end of file, including embedded NUL. */
93 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, EOF, EOF, f);
94 ASSERT (result == 2);
95 ASSERT (10 == len);
96 ASSERT (memcmp (line, "\0f\0eeea\nb", 10) == 0);
97
98 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', '\r', f);
99 ASSERT (result == -1);
100
101 /* Larger file size. */
102 rewind (f);
103 {
104 int i;
105 for (i = 0; i < 16; i++)
106 fprintf (f, "%500x%c", i + 0u, i % 2 ? '\n' : '\r');
107 }
108 rewind (f);
109 {
110 char buffer[502];
111
112 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', '\r', f);
113 ASSERT (result == 501);
114 ASSERT (501 < len);
115 memset (buffer, ' ', 499);
116 buffer[499] = '0';
117 buffer[500] = '\r';
118 buffer[501] = '\0';
119 ASSERT (strcmp (buffer, line) == 0);
120
121 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, '\n', '\r', f);
122 ASSERT (result == 501);
123 ASSERT (501 < len);
124 buffer[499] = '1';
125 buffer[500] = '\n';
126 ASSERT (strcmp (buffer, line) == 0);
127
128 result = getndelim2 (&line, &len, 0, GETNLINE_NO_LIMIT, 'g', 'f', f);
129 ASSERT (result == 501 * 14 - 1);
130 ASSERT (501 * 14 <= len);
131 buffer[499] = 'f';
132 buffer[500] = '\0';
133 ASSERT (strcmp (buffer, line + 501 * 13) == 0);
134
135 result = getndelim2 (&line, &len, 501 * 14 - 1, GETNLINE_NO_LIMIT,
136 EOF, EOF, f);
137 ASSERT (result == 1);
138 buffer[500] = '\n';
139 ASSERT (strcmp (buffer, line + 501 * 13) == 0);
140
141 result = getndelim2 (&line, &len, 501 * 14 - 1, GETNLINE_NO_LIMIT,
142 EOF, EOF, f);
143 buffer[500] = '\0';
144 ASSERT (strcmp (buffer, line + 501 * 13) == 0);
145 ASSERT (result == -1);
146 }
147
148 fclose (f);
149 remove ("test-getndelim2.txt");
150 return 0;
151 }
152