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