Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	lib/libc/stdio/open_wmemstream.c	Lines:	0	68	0.0 %
Date:	2017-11-13	Branches:	0	33	0.0 %


/*	$OpenBSD: open_wmemstream.c,v 1.8 2015/09/12 16:23:14 guenther Exp $	*/

/*
 * Copyright (c) 2011 Martin Pieuchot <mpi@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <wchar.h>
#include "local.h"

#define	MINIMUM(a, b)	(((a) < (b)) ? (a) : (b))

struct state {
	wchar_t		 *string;	/* actual stream */
	wchar_t		**pbuf;		/* point to the stream */
	size_t		 *psize;	/* point to min(pos, len) */
	size_t		  pos;		/* current position */
	size_t		  size;		/* number of allocated wchar_t */
	size_t		  len;		/* length of the data */
	mbstate_t	  mbs;		/* conversion state of the stream */
};

static int
wmemstream_write(void *v, const char *b, int l)
{
	struct state	*st = v;
	wchar_t		*p;
	size_t		 nmc, len, end;

	end = (st->pos + l);

	if (end >= st->size) {
		/* 1.6 is (very) close to the golden ratio. */
		size_t	sz = st->size * 8 / 5;

		if (sz < end + 1)
			sz = end + 1;
		p = reallocarray(st->string, sz, sizeof(wchar_t));
		if (!p)
			return (-1);
		bzero(p + st->size, (sz - st->size) * sizeof(wchar_t));
		*st->pbuf = st->string = p;
		st->size = sz;
	}

	nmc = (st->size - st->pos) * sizeof(wchar_t);
	len = mbsnrtowcs(st->string + st->pos, &b, nmc, l, &st->mbs);
	if (len == (size_t)-1)
		return (-1);
	st->pos += len;

	if (st->pos > st->len) {
		st->len = st->pos;
		st->string[st->len] = L'\0';
	}

	*st->psize = st->pos;

	return (len);
}

static fpos_t
wmemstream_seek(void *v, fpos_t off, int whence)
{
	struct state	*st = v;
	ssize_t		 base = 0;

	switch (whence) {
	case SEEK_SET:
		break;
	case SEEK_CUR:
		base = st->pos;
		break;
	case SEEK_END:
		base = st->len;
		break;
	}

	if (off > (SIZE_MAX / sizeof(wchar_t)) - base || off < -base) {
		errno = EOVERFLOW;
		return (-1);
	}

	/*
	 * XXX Clearing mbs here invalidates shift state for state-
	 * dependent encodings, but they are not (yet) supported.
	 */
	bzero(&st->mbs, sizeof(st->mbs));

	st->pos = base + off;
	*st->psize = MINIMUM(st->pos, st->len);

	return (st->pos);
}

static int
wmemstream_close(void *v)
{
	struct state	*st = v;

	free(st);

	return (0);
}

FILE *
open_wmemstream(wchar_t **pbuf, size_t *psize)
{
	struct state	*st;
	FILE		*fp;

	if (pbuf == NULL || psize == NULL) {
		errno = EINVAL;
		return (NULL);
	}

	if ((st = malloc(sizeof(*st))) == NULL)
		return (NULL);

	if ((fp = __sfp()) == NULL) {
		free(st);
		return (NULL);
	}

	st->size = BUFSIZ * sizeof(wchar_t);
	if ((st->string = calloc(1, st->size)) == NULL) {
		free(st);
		fp->_flags = 0;
		return (NULL);
	}

	*st->string = L'\0';
	st->pos = 0;
	st->len = 0;
	st->pbuf = pbuf;
	st->psize = psize;
	bzero(&st->mbs, sizeof(st->mbs));

	*pbuf = st->string;
	*psize = st->len;

	fp->_flags = __SWR;
	fp->_file = -1;
	fp->_cookie = st;
	fp->_read = NULL;
	fp->_write = wmemstream_write;
	fp->_seek = wmemstream_seek;
	fp->_close = wmemstream_close;
	_SET_ORIENTATION(fp, 1);

	return (fp);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: open_wmemstream.c,v 1.8 2015/09/12 16:23:14 guenther Exp $ */
2
3			/*
4			* Copyright (c) 2011 Martin Pieuchot <mpi@openbsd.org>
5			*
6			* Permission to use, copy, modify, and distribute this software for any
7			* purpose with or without fee is hereby granted, provided that the above
8			* copyright notice and this permission notice appear in all copies.
9			*
10			* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11			* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12			* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13			* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14			* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15			* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16			* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17			*/
18
19			#include <errno.h>
20			#include <fcntl.h>
21			#include <stdio.h>
22			#include <stdint.h>
23			#include <stdlib.h>
24			#include <string.h>
25			#include <wchar.h>
26			#include "local.h"
27
28			#define MINIMUM(a, b) (((a) < (b)) ? (a) : (b))
29
30			struct state {
31			wchar_t string; / actual stream */
32			wchar_t *pbuf; / point to the stream */
33			size_t psize; / point to min(pos, len) */
34			size_t pos; /* current position */
35			size_t size; /* number of allocated wchar_t */
36			size_t len; /* length of the data */
37			mbstate_t mbs; /* conversion state of the stream */
38			};
39
40			static int
41			wmemstream_write(void v, const char b, int l)
42			{
43			struct state *st = v;
44			wchar_t *p;
45			size_t nmc, len, end;
46
47			end = (st->pos + l);
48
49			if (end >= st->size) {
50			/* 1.6 is (very) close to the golden ratio. */
51			size_t sz = st->size * 8 / 5;
52
53			if (sz < end + 1)
54			sz = end + 1;
55			p = reallocarray(st->string, sz, sizeof(wchar_t));
56			if (!p)
57			return (-1);
58			bzero(p + st->size, (sz - st->size) * sizeof(wchar_t));
59			*st->pbuf = st->string = p;
60			st->size = sz;
61			}
62
63			nmc = (st->size - st->pos) * sizeof(wchar_t);
64			len = mbsnrtowcs(st->string + st->pos, &b, nmc, l, &st->mbs);
65			if (len == (size_t)-1)
66			return (-1);
67			st->pos += len;
68
69			if (st->pos > st->len) {
70			st->len = st->pos;
71			st->string[st->len] = L'\0';
72			}
73
74			*st->psize = st->pos;
75
76			return (len);
77			}
78
79			static fpos_t
80			wmemstream_seek(void *v, fpos_t off, int whence)
81			{
82			struct state *st = v;
83			ssize_t base = 0;
84
85			switch (whence) {
86			case SEEK_SET:
87			break;
88			case SEEK_CUR:
89			base = st->pos;
90			break;
91			case SEEK_END:
92			base = st->len;
93			break;
94			}
95
96			if (off > (SIZE_MAX / sizeof(wchar_t)) - base \|\| off < -base) {
97			errno = EOVERFLOW;
98			return (-1);
99			}
100
101			/*
102			* XXX Clearing mbs here invalidates shift state for state-
103			* dependent encodings, but they are not (yet) supported.
104			*/
105			bzero(&st->mbs, sizeof(st->mbs));
106
107			st->pos = base + off;
108			*st->psize = MINIMUM(st->pos, st->len);
109
110			return (st->pos);
111			}
112
113			static int
114			wmemstream_close(void *v)
115			{
116			struct state *st = v;
117
118			free(st);
119
120			return (0);
121			}
122
123			FILE *
124			open_wmemstream(wchar_t *pbuf, size_t psize)
125			{
126			struct state *st;
127			FILE *fp;
128
129			if (pbuf == NULL \|\| psize == NULL) {
130			errno = EINVAL;
131			return (NULL);
132			}
133
134			if ((st = malloc(sizeof(*st))) == NULL)
135			return (NULL);
136
137			if ((fp = __sfp()) == NULL) {
138			free(st);
139			return (NULL);
140			}
141
142			st->size = BUFSIZ * sizeof(wchar_t);
143			if ((st->string = calloc(1, st->size)) == NULL) {
144			free(st);
145			fp->_flags = 0;
146			return (NULL);
147			}
148
149			*st->string = L'\0';
150			st->pos = 0;
151			st->len = 0;
152			st->pbuf = pbuf;
153			st->psize = psize;
154			bzero(&st->mbs, sizeof(st->mbs));
155
156			*pbuf = st->string;
157			*psize = st->len;
158
159			fp->_flags = __SWR;
160			fp->_file = -1;
161			fp->_cookie = st;
162			fp->_read = NULL;
163			fp->_write = wmemstream_write;
164			fp->_seek = wmemstream_seek;
165			fp->_close = wmemstream_close;
166			_SET_ORIENTATION(fp, 1);
167
168			return (fp);
169			}