1/*-
2 * Copyright (c) 2002-2004 Tim J. Robbins.
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD: src/lib/libc/locale/mbsnrtowcs.c,v 1.1 2004/07/21 10:54:57 tjr Exp $");
29
30#include "xlocale_private.h"
31
32#include <errno.h>
33#include <limits.h>
34#include <stdlib.h>
35#include <wchar.h>
36#include "mblocal.h"
37
38size_t
39mbsnrtowcs_l(wchar_t * __restrict dst, const char ** __restrict src,
40    size_t nms, size_t len, mbstate_t * __restrict ps, locale_t loc)
41{
42	NORMALIZE_LOCALE(loc);
43	if (ps == NULL)
44		ps = &loc->__mbs_mbsnrtowcs;
45	return (loc->__lc_ctype->__mbsnrtowcs(dst, src, nms, len, ps, loc));
46}
47
48size_t
49mbsnrtowcs(wchar_t * __restrict dst, const char ** __restrict src,
50    size_t nms, size_t len, mbstate_t * __restrict ps)
51{
52	return mbsnrtowcs_l(dst, src, nms, len, ps, __current_locale());
53}
54
55__private_extern__ size_t
56__mbsnrtowcs_std(wchar_t * __restrict dst, const char ** __restrict src,
57    size_t nms, size_t len, mbstate_t * __restrict ps, locale_t loc)
58{
59	const char *s;
60	size_t nchr;
61	wchar_t wc;
62	size_t nb;
63	size_t (*__mbrtowc)(wchar_t * __restrict, const char * __restrict,
64	    size_t, mbstate_t * __restrict, locale_t)
65	    = loc->__lc_ctype->__mbrtowc;
66
67	s = *src;
68	nchr = 0;
69
70	if (dst == NULL) {
71		for (;;) {
72			if ((nb = __mbrtowc(&wc, s, nms, ps, loc)) == (size_t)-1)
73				/* Invalid sequence - mbrtowc() sets errno. */
74				return ((size_t)-1);
75			else if (nb == 0 || nb == (size_t)-2)
76				return (nchr);
77			s += nb;
78			nms -= nb;
79			nchr++;
80		}
81		/*NOTREACHED*/
82	}
83
84	while (len-- > 0) {
85		if ((nb = __mbrtowc(dst, s, nms, ps, loc)) == (size_t)-1) {
86			*src = s;
87			return ((size_t)-1);
88		} else if (nb == (size_t)-2) {
89			*src = s + nms;
90			return (nchr);
91		} else if (nb == 0) {
92			*src = NULL;
93			return (nchr);
94		}
95		s += nb;
96		nms -= nb;
97		nchr++;
98		dst++;
99	}
100	*src = s;
101	return (nchr);
102}
103