2 * Copyright (c) 2013 Hudson River Trading LLC
3 * Written by: John H. Baldwin <jhb@FreeBSD.org>
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * $FreeBSD: head/lib/libc/stdio/open_wmemstream.c 281887 2015-04-23 14:22:20Z jhb $
30 #include "namespace.h"
38 #include "un-namespace.h"
40 /* XXX: There is no FPOS_MAX. This assumes fpos_t is an off_t. */
41 #define FPOS_MAX OFF_MAX
52 wmemstream_grow(struct wmemstream *ms, fpos_t newoff)
57 if (newoff < 0 || newoff >= SSIZE_MAX / sizeof(wchar_t))
58 newsize = SSIZE_MAX / sizeof(wchar_t) - 1;
61 if (newsize > ms->len) {
62 buf = realloc(*ms->bufp, (newsize + 1) * sizeof(wchar_t));
65 fprintf(stderr, "WMS: %p growing from %zd to %zd\n",
66 ms, ms->len, newsize);
68 wmemset(buf + ms->len + 1, 0, newsize - ms->len);
79 wmemstream_update(struct wmemstream *ms)
82 assert(ms->len >= 0 && ms->offset >= 0);
83 *ms->sizep = ms->len < ms->offset ? ms->len : ms->offset;
87 * Based on a starting multibyte state and an input buffer, determine
88 * how many wchar_t's would be output. This doesn't use mbsnrtowcs()
89 * so that it can handle embedded null characters.
92 wbuflen(const mbstate_t *state, const char *buf, int len)
95 size_t charlen, count;
100 charlen = mbrlen(buf, len, &lenstate);
101 if (charlen == (size_t)-1)
103 if (charlen == (size_t)-2)
106 /* XXX: Not sure how else to handle this. */
116 wmemstream_write(void *cookie, const char *buf, int len)
118 struct wmemstream *ms;
119 ssize_t consumed, wlen;
123 wlen = wbuflen(&ms->mbstate, buf, len);
128 if (!wmemstream_grow(ms, ms->offset + wlen))
132 * This copies characters one at a time rather than using
133 * mbsnrtowcs() so it can properly handle embedded null
137 while (len > 0 && ms->offset < ms->len) {
138 charlen = mbrtowc(*ms->bufp + ms->offset, buf, len,
140 if (charlen == (size_t)-1) {
145 /* Treat it as a successful short write. */
149 /* XXX: Not sure how else to handle this. */
151 if (charlen == (size_t)-2) {
161 wmemstream_update(ms);
163 fprintf(stderr, "WMS: write(%p, %d) = %zd\n", ms, len, consumed);
169 wmemstream_seek(void *cookie, fpos_t pos, int whence)
171 struct wmemstream *ms;
178 /* _fseeko() checks for negative offsets. */
183 /* This is only called by _ftello(). */
188 if (pos + ms->len < 0) {
191 "WMS: bad SEEK_END: pos %jd, len %zd\n",
192 (intmax_t)pos, ms->len);
198 if (FPOS_MAX - ms->len < pos) {
201 "WMS: bad SEEK_END: pos %jd, len %zd\n",
202 (intmax_t)pos, ms->len);
208 ms->offset = ms->len + pos;
211 /* Reset the multibyte state if a seek changes the position. */
212 if (ms->offset != old)
213 memset(&ms->mbstate, 0, sizeof(ms->mbstate));
214 wmemstream_update(ms);
216 fprintf(stderr, "WMS: seek(%p, %jd, %d) %jd -> %jd\n", ms,
217 (intmax_t)pos, whence, (intmax_t)old, (intmax_t)ms->offset);
223 wmemstream_close(void *cookie)
231 open_wmemstream(wchar_t **bufp, size_t *sizep)
233 struct wmemstream *ms;
237 if (bufp == NULL || sizep == NULL) {
241 *bufp = calloc(1, sizeof(wchar_t));
244 ms = malloc(sizeof(*ms));
256 memset(&ms->mbstate, 0, sizeof(mbstate_t));
257 wmemstream_update(ms);
258 fp = funopen(ms, NULL, wmemstream_write, wmemstream_seek,