1/* $FreeBSD$ */
2/* $NetBSD: citrus_memstream.h,v 1.3 2005/05/14 17:55:42 tshiozak Exp $ */
3
4/*-
5 * Copyright (c)2003 Citrus Project,
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 *
29 */
30
31#ifndef _CITRUS_MEMSTREAM_H_
32#define _CITRUS_MEMSTREAM_H_
33
34struct _citrus_memory_stream {
35	struct _citrus_region	ms_region;
36	size_t			ms_pos;
37};
38
39__BEGIN_DECLS
40const char	*_citrus_memory_stream_getln(
41		    struct _citrus_memory_stream * __restrict,
42		    size_t * __restrict);
43const char	*_citrus_memory_stream_matchline(
44		    struct _citrus_memory_stream * __restrict,
45		    const char * __restrict, size_t * __restrict, int);
46void		*_citrus_memory_stream_chr(struct _citrus_memory_stream *,
47		    struct _citrus_region *, char);
48void		_citrus_memory_stream_skip_ws(struct _citrus_memory_stream *);
49__END_DECLS
50
51static __inline int
52_citrus_memory_stream_iseof(struct _citrus_memory_stream *ms)
53{
54
55	return (ms->ms_pos >= _citrus_region_size(&ms->ms_region));
56}
57
58static __inline void
59_citrus_memory_stream_bind(struct _citrus_memory_stream * __restrict ms,
60    const struct _citrus_region * __restrict r)
61{
62
63	ms->ms_region = *r;
64	ms->ms_pos = 0;
65}
66
67static __inline void
68_citrus_memory_stream_bind_ptr(struct _citrus_memory_stream * __restrict ms,
69    void *ptr, size_t sz)
70{
71	struct _citrus_region r;
72
73	_citrus_region_init(&r, ptr, sz);
74	_citrus_memory_stream_bind(ms, &r);
75}
76
77static __inline void
78_citrus_memory_stream_rewind(struct _citrus_memory_stream *ms)
79{
80
81	ms->ms_pos = 0;
82}
83
84static __inline size_t
85_citrus_memory_stream_tell(struct _citrus_memory_stream *ms)
86{
87
88	return (ms->ms_pos);
89}
90
91static __inline size_t
92_citrus_memory_stream_remainder(struct _citrus_memory_stream *ms)
93{
94	size_t sz;
95
96	sz = _citrus_region_size(&ms->ms_region);
97	if (ms->ms_pos>sz)
98		return (0);
99	return (sz-ms->ms_pos);
100}
101
102static __inline int
103_citrus_memory_stream_seek(struct _citrus_memory_stream *ms, size_t pos, int w)
104{
105	size_t sz;
106
107	sz = _citrus_region_size(&ms->ms_region);
108
109	switch (w) {
110	case SEEK_SET:
111		if (pos >= sz)
112			return (-1);
113		ms->ms_pos = pos;
114		break;
115	case SEEK_CUR:
116		pos += (ssize_t)ms->ms_pos;
117		if (pos >= sz)
118			return (-1);
119		ms->ms_pos = pos;
120		break;
121	case SEEK_END:
122		if (sz < pos)
123			return (-1);
124		ms->ms_pos = sz - pos;
125		break;
126	}
127	return (0);
128}
129
130static __inline int
131_citrus_memory_stream_getc(struct _citrus_memory_stream *ms)
132{
133
134	if (_citrus_memory_stream_iseof(ms))
135		return (EOF);
136	return (_citrus_region_peek8(&ms->ms_region, ms->ms_pos++));
137}
138
139static __inline void
140_citrus_memory_stream_ungetc(struct _citrus_memory_stream *ms, int ch)
141{
142
143	if (ch != EOF && ms->ms_pos > 0)
144		ms->ms_pos--;
145}
146
147static __inline int
148_citrus_memory_stream_peek(struct _citrus_memory_stream *ms)
149{
150
151	if (_citrus_memory_stream_iseof(ms))
152		return (EOF);
153	return (_citrus_region_peek8(&ms->ms_region, ms->ms_pos));
154}
155
156static __inline void *
157_citrus_memory_stream_getregion(struct _citrus_memory_stream *ms,
158    struct _citrus_region *r, size_t sz)
159{
160	void *ret;
161
162	if (ms->ms_pos + sz > _citrus_region_size(&ms->ms_region))
163		return (NULL);
164
165	ret = _citrus_region_offset(&ms->ms_region, ms->ms_pos);
166	ms->ms_pos += sz;
167	if (r)
168		_citrus_region_init(r, ret, sz);
169
170	return (ret);
171}
172
173static __inline int
174_citrus_memory_stream_get8(struct _citrus_memory_stream *ms, uint8_t *rval)
175{
176
177	if (ms->ms_pos + 1 > _citrus_region_size(&ms->ms_region))
178		return (-1);
179
180	*rval = _citrus_region_peek8(&ms->ms_region, ms->ms_pos);
181	ms->ms_pos += 2;
182
183	return (0);
184}
185
186static __inline int
187_citrus_memory_stream_get16(struct _citrus_memory_stream *ms, uint16_t *rval)
188{
189
190	if (ms->ms_pos + 2 > _citrus_region_size(&ms->ms_region))
191		return (-1);
192
193	*rval = _citrus_region_peek16(&ms->ms_region, ms->ms_pos);
194	ms->ms_pos += 2;
195
196	return (0);
197}
198
199static __inline int
200_citrus_memory_stream_get32(struct _citrus_memory_stream *ms, uint32_t *rval)
201{
202
203	if (ms->ms_pos + 4 > _citrus_region_size(&ms->ms_region))
204		return (-1);
205
206	*rval = _citrus_region_peek32(&ms->ms_region, ms->ms_pos);
207	ms->ms_pos += 4;
208
209	return (0);
210}
211
212static __inline int
213_citrus_memory_stream_getln_region(struct _citrus_memory_stream *ms,
214    struct _citrus_region *r)
215{
216	const char *ptr;
217	size_t sz;
218
219	ptr = _citrus_memory_stream_getln(ms, &sz);
220	if (ptr)
221		_citrus_region_init(r, __DECONST(void *, ptr), sz);
222
223	return (ptr == NULL);
224}
225
226#endif
227