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
28 #include <sys/cdefs.h>
29 __FBSDID("$FreeBSD$");
31 #include "namespace.h"
42 #include "un-namespace.h"
44 /* XXX: There is no FPOS_MAX. This assumes fpos_t is an off_t. */
45 #define FPOS_MAX OFF_MAX
56 wmemstream_grow(struct wmemstream *ms, fpos_t newoff)
61 if (newoff < 0 || newoff >= SSIZE_MAX / sizeof(wchar_t))
62 newsize = SSIZE_MAX / sizeof(wchar_t) - 1;
65 if (newsize > ms->len) {
66 buf = realloc(*ms->bufp, (newsize + 1) * sizeof(wchar_t));
69 fprintf(stderr, "WMS: %p growing from %zd to %zd\n",
70 ms, ms->len, newsize);
72 wmemset(buf + ms->len + 1, 0, newsize - ms->len);
83 wmemstream_update(struct wmemstream *ms)
86 assert(ms->len >= 0 && ms->offset >= 0);
87 *ms->sizep = ms->len < ms->offset ? ms->len : ms->offset;
91 * Based on a starting multibyte state and an input buffer, determine
92 * how many wchar_t's would be output. This doesn't use mbsnrtowcs()
93 * so that it can handle embedded null characters.
96 wbuflen(const mbstate_t *state, const char *buf, int len)
99 size_t charlen, count;
104 charlen = mbrlen(buf, len, &lenstate);
105 if (charlen == (size_t)-1)
107 if (charlen == (size_t)-2)
110 /* XXX: Not sure how else to handle this. */
120 wmemstream_write(void *cookie, const char *buf, int len)
122 struct wmemstream *ms;
123 ssize_t consumed, wlen;
127 wlen = wbuflen(&ms->mbstate, buf, len);
132 if (!wmemstream_grow(ms, ms->offset + wlen))
136 * This copies characters one at a time rather than using
137 * mbsnrtowcs() so it can properly handle embedded null
141 while (len > 0 && ms->offset < ms->len) {
142 charlen = mbrtowc(*ms->bufp + ms->offset, buf, len,
144 if (charlen == (size_t)-1) {
149 /* Treat it as a successful short write. */
153 /* XXX: Not sure how else to handle this. */
155 if (charlen == (size_t)-2) {
165 wmemstream_update(ms);
167 fprintf(stderr, "WMS: write(%p, %d) = %zd\n", ms, len, consumed);
173 wmemstream_seek(void *cookie, fpos_t pos, int whence)
175 struct wmemstream *ms;
182 /* _fseeko() checks for negative offsets. */
187 /* This is only called by _ftello(). */
192 if (pos + ms->len < 0) {
195 "WMS: bad SEEK_END: pos %jd, len %zd\n",
196 (intmax_t)pos, ms->len);
202 if (FPOS_MAX - ms->len < pos) {
205 "WMS: bad SEEK_END: pos %jd, len %zd\n",
206 (intmax_t)pos, ms->len);
212 ms->offset = ms->len + pos;
215 /* Reset the multibyte state if a seek changes the position. */
216 if (ms->offset != old)
217 memset(&ms->mbstate, 0, sizeof(ms->mbstate));
218 wmemstream_update(ms);
220 fprintf(stderr, "WMS: seek(%p, %jd, %d) %jd -> %jd\n", ms,
221 (intmax_t)pos, whence, (intmax_t)old, (intmax_t)ms->offset);
227 wmemstream_close(void *cookie)
235 open_wmemstream(wchar_t **bufp, size_t *sizep)
237 struct wmemstream *ms;
241 if (bufp == NULL || sizep == NULL) {
245 *bufp = calloc(1, sizeof(wchar_t));
248 ms = malloc(sizeof(*ms));
260 memset(&ms->mbstate, 0, sizeof(mbstate_t));
261 wmemstream_update(ms);
262 fp = funopen(ms, NULL, wmemstream_write, wmemstream_seek,