2 * Copyright (c) 2009 David Schultz <das@FreeBSD.org>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
39 /* The assertions depend on this string. */
40 char apothegm[] = "All work and no play\0 makes Jack a dull boy.\n";
43 * This is a neurotic reader function designed to give getdelim() a
44 * hard time. It reads through the string `apothegm' and returns a
45 * random number of bytes up to the requested length.
48 _reader(void *cookie, char *buf, int len)
50 size_t *offp = cookie;
53 r = random() % CHUNK_MAX + 1;
56 if (len > sizeof(apothegm) - *offp)
57 len = sizeof(apothegm) - *offp;
58 memcpy(buf, apothegm + *offp, len);
68 offp = malloc(sizeof(*offp)); /* XXX leak */
70 return (fropen(offp, _reader));
74 main(int argc, char *argv[])
86 * Test multiple times with different buffer sizes
87 * and different _reader() return values.
90 for (i = 0; i < 8; i++) {
94 /* First line: the full apothegm */
95 assert(getline(&line, &linecap, fp) == sizeof(apothegm) - 1);
96 assert(memcmp(line, apothegm, sizeof(apothegm)) == 0);
97 assert(linecap >= sizeof(apothegm));
98 /* Second line: the NUL terminator following the newline */
99 assert(getline(&line, &linecap, fp) == 1);
100 assert(line[0] == '\0' && line[1] == '\0');
101 /* Third line: EOF */
103 assert(getline(&line, &linecap, fp) == -1);
104 assert(line[0] == '\0');
111 printf("ok 1 - getline basic\n");
113 /* Make sure read errors are handled properly. */
116 assert(getline(&line, &linecap, stdout) == -1);
117 assert(errno == EBADF);
119 assert(getdelim(&line, &linecap, 'X', stdout) == -1);
120 assert(errno == EBADF);
121 assert(ferror(stdout));
122 printf("ok 2 - stream error\n");
124 /* Make sure NULL linep or linecapp pointers are handled. */
126 assert(getline(NULL, &linecap, fp) == -1);
127 assert(errno == EINVAL);
128 assert(getline(&line, NULL, fp) == -1);
129 assert(errno == EINVAL);
132 printf("ok 3 - invalid params\n");
134 /* Make sure getline() allocates memory as needed if fp is at EOF. */
137 while (!feof(fp)) /* advance to EOF; can't fseek this stream */
142 assert(getline(&line, &linecap, fp) == -1);
143 assert(line[0] == '\0');
149 printf("ok 4 - eof\n");
151 /* Make sure a NUL delimiter works. */
153 n = strlen(apothegm);
154 assert(getdelim(&line, &linecap, '\0', fp) == n + 1);
155 assert(strcmp(line, apothegm) == 0);
156 assert(line[n + 1] == '\0');
157 assert(linecap > n + 1);
158 n = strlen(apothegm + n + 1);
159 assert(getdelim(&line, &linecap, '\0', fp) == n + 1);
160 assert(line[n + 1] == '\0');
161 assert(linecap > n + 1);
165 printf("ok 5 - nul\n");