1/*- 2 * Copyright (c) 2009 David Schultz <das@FreeBSD.org> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 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. 13 * 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 24 * SUCH DAMAGE. 25 */ 26 27#include <sys/cdefs.h> 28__FBSDID("$FreeBSD$"); 29 30#define _WITH_GETLINE 31#include <assert.h> 32#include <errno.h> 33#include <stdio.h> 34#include <stdlib.h> 35#include <string.h> 36 37#define CHUNK_MAX 10 38 39/* The assertions depend on this string. */ 40char apothegm[] = "All work and no play\0 makes Jack a dull boy.\n"; 41 42/* 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. 46 */ 47static int 48_reader(void *cookie, char *buf, int len) 49{ 50 size_t *offp = cookie; 51 size_t r; 52 53 r = random() % CHUNK_MAX + 1; 54 if (len > r) 55 len = r; 56 if (len > sizeof(apothegm) - *offp) 57 len = sizeof(apothegm) - *offp; 58 memcpy(buf, apothegm + *offp, len); 59 *offp += len; 60 return (len); 61} 62 63static FILE * 64mkfilebuf(void) 65{ 66 size_t *offp; 67 68 offp = malloc(sizeof(*offp)); /* XXX leak */ 69 *offp = 0; 70 return (fropen(offp, _reader)); 71} 72 73int 74main(int argc, char *argv[]) 75{ 76 FILE *fp; 77 char *line; 78 size_t linecap; 79 int i, n; 80 81 srandom(0); 82 83 printf("1..6\n"); 84 85 /* 86 * Test multiple times with different buffer sizes 87 * and different _reader() return values. 88 */ 89 errno = 0; 90 for (i = 0; i < 8; i++) { 91 fp = mkfilebuf(); 92 linecap = i; 93 line = malloc(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 */ 102 line[0] = 'X'; 103 assert(getline(&line, &linecap, fp) == -1); 104 assert(line[0] == '\0'); 105 free(line); 106 line = NULL; 107 assert(feof(fp)); 108 assert(!ferror(fp)); 109 fclose(fp); 110 } 111 assert(errno == 0); 112 printf("ok 1 - getline basic\n"); 113 114 /* Make sure read errors are handled properly. */ 115 linecap = 0; 116 errno = 0; 117 assert(getline(&line, &linecap, stdout) == -1); 118 assert(errno == EBADF); 119 errno = 0; 120 assert(getdelim(&line, &linecap, 'X', stdout) == -1); 121 assert(errno == EBADF); 122 assert(ferror(stdout)); 123 printf("ok 2 - stream error\n"); 124 125 /* Make sure NULL linep or linecapp pointers are handled. */ 126 fp = mkfilebuf(); 127 assert(getline(NULL, &linecap, fp) == -1); 128 assert(errno == EINVAL); 129 assert(getline(&line, NULL, fp) == -1); 130 assert(errno == EINVAL); 131 assert(ferror(fp)); 132 fclose(fp); 133 printf("ok 3 - invalid params\n"); 134 135 /* Make sure getline() allocates memory as needed if fp is at EOF. */ 136 errno = 0; 137 fp = mkfilebuf(); 138 while (!feof(fp)) /* advance to EOF; can't fseek this stream */ 139 getc(fp); 140 free(line); 141 line = NULL; 142 linecap = 0; 143 assert(getline(&line, &linecap, fp) == -1); 144 assert(line[0] == '\0'); 145 assert(linecap > 0); 146 assert(errno == 0); 147 assert(feof(fp)); 148 assert(!ferror(fp)); 149 fclose(fp); 150 printf("ok 4 - eof\n"); 151 152 /* Make sure a NUL delimiter works. */ 153 fp = mkfilebuf(); 154 n = strlen(apothegm); 155 assert(getdelim(&line, &linecap, '\0', fp) == n + 1); 156 assert(strcmp(line, apothegm) == 0); 157 assert(line[n + 1] == '\0'); 158 assert(linecap > n + 1); 159 n = strlen(apothegm + n + 1); 160 assert(getdelim(&line, &linecap, '\0', fp) == n + 1); 161 assert(line[n + 1] == '\0'); 162 assert(linecap > n + 1); 163 assert(errno == 0); 164 assert(!ferror(fp)); 165 fclose(fp); 166 printf("ok 5 - nul\n"); 167 168 /* Make sure NULL *linep and zero *linecapp are handled. */ 169 fp = mkfilebuf(); 170 free(line); 171 line = NULL; 172 linecap = 42; 173 assert(getline(&line, &linecap, fp) == sizeof(apothegm) - 1); 174 assert(memcmp(line, apothegm, sizeof(apothegm)) == 0); 175 fp = mkfilebuf(); 176 free(line); 177 line = malloc(100); 178 linecap = 0; 179 assert(getline(&line, &linecap, fp) == sizeof(apothegm) - 1); 180 assert(memcmp(line, apothegm, sizeof(apothegm)) == 0); 181 free(line); 182 assert(!ferror(fp)); 183 fclose(fp); 184 printf("ok 6 - empty/NULL initial buffer\n"); 185 186 exit(0); 187} 188