1 /* getndelim2 - Read a line from a stream, stopping at one of 2 delimiters,
2 with bounded memory allocation.
4 Copyright (C) 1993, 1996, 1997, 1998, 2000, 2003, 2004 Free Software
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2, or (at your option)
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software Foundation,
19 Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */
21 /* Originally written by Jan Brittenson, bson@gnu.ai.mit.edu. */
27 #include "getndelim2.h"
32 #include "unlocked-io.h"
36 # include <inttypes.h>
42 # define PTRDIFF_MAX ((ptrdiff_t) (SIZE_MAX / 2))
45 # define SIZE_MAX ((size_t) -1)
48 # define SSIZE_MAX ((ssize_t) (SIZE_MAX / 2))
51 /* The maximum value that getndelim2 can return without suffering from
52 overflow problems, either internally (because of pointer
53 subtraction overflow) or due to the API (because of ssize_t). */
54 #define GETNDELIM2_MAXIMUM (PTRDIFF_MAX < SSIZE_MAX ? PTRDIFF_MAX : SSIZE_MAX)
56 /* Try to add at least this many bytes when extending the buffer.
57 MIN_CHUNK must be no greater than GETNDELIM2_MAXIMUM. */
61 getndelim2 (char **lineptr, size_t *linesize, size_t offset, size_t nmax,
62 int delim1, int delim2, FILE *stream)
64 size_t nbytes_avail; /* Allocated but unused bytes in *LINEPTR. */
65 char *read_pos; /* Where we're reading into *LINEPTR. */
66 ssize_t bytes_stored = -1;
68 size_t size = *linesize;
72 size = nmax < MIN_CHUNK ? nmax : MIN_CHUNK;
81 nbytes_avail = size - offset;
82 read_pos = ptr + offset;
84 if (nbytes_avail == 0 && nmax <= size)
89 /* Here always ptr + size == read_pos + nbytes_avail. */
93 /* We always want at least one byte left in the buffer, since we
94 always (unless we get an error while reading the first byte)
95 NUL-terminate the line buffer. */
97 if (nbytes_avail < 2 && size < nmax)
99 size_t newsize = size < MIN_CHUNK ? size + MIN_CHUNK : 2 * size;
102 if (! (size < newsize && newsize <= nmax))
105 if (GETNDELIM2_MAXIMUM < newsize - offset)
107 size_t newsizemax = offset + GETNDELIM2_MAXIMUM + 1;
108 if (size == newsizemax)
110 newsize = newsizemax;
113 nbytes_avail = newsize - (read_pos - ptr);
114 newptr = realloc (ptr, newsize);
119 read_pos = size - nbytes_avail + ptr;
125 /* Return partial line, if any. */
132 if (nbytes_avail >= 2)
138 if (c == delim1 || c == delim2)
139 /* Return the line. */
143 /* Done - NUL terminate and return the number of bytes read.
144 At this point we know that nbytes_avail >= 1. */
147 bytes_stored = read_pos - (ptr + offset);