xref: /freebsd/lib/libc/locale/mbsrtowcs.c (revision 6155c34adf50d3d025ec0e862bb34746ea8c0a21)
1e92a3d83STim J. Robbins /*-
274f90defSTim J. Robbins  * Copyright (c) 2002-2004 Tim J. Robbins.
3e92a3d83STim J. Robbins  * All rights reserved.
4e92a3d83STim J. Robbins  *
5e92a3d83STim J. Robbins  * Redistribution and use in source and binary forms, with or without
6e92a3d83STim J. Robbins  * modification, are permitted provided that the following conditions
7e92a3d83STim J. Robbins  * are met:
8e92a3d83STim J. Robbins  * 1. Redistributions of source code must retain the above copyright
9e92a3d83STim J. Robbins  *    notice, this list of conditions and the following disclaimer.
10e92a3d83STim J. Robbins  * 2. Redistributions in binary form must reproduce the above copyright
11e92a3d83STim J. Robbins  *    notice, this list of conditions and the following disclaimer in the
12e92a3d83STim J. Robbins  *    documentation and/or other materials provided with the distribution.
13e92a3d83STim J. Robbins  *
14e92a3d83STim J. Robbins  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15e92a3d83STim J. Robbins  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16e92a3d83STim J. Robbins  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17e92a3d83STim J. Robbins  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18e92a3d83STim J. Robbins  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19e92a3d83STim J. Robbins  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20e92a3d83STim J. Robbins  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21e92a3d83STim J. Robbins  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22e92a3d83STim J. Robbins  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23e92a3d83STim J. Robbins  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24e92a3d83STim J. Robbins  * SUCH DAMAGE.
25e92a3d83STim J. Robbins  */
26e92a3d83STim J. Robbins 
27e92a3d83STim J. Robbins #include <sys/cdefs.h>
28e92a3d83STim J. Robbins __FBSDID("$FreeBSD$");
29e92a3d83STim J. Robbins 
30e92a3d83STim J. Robbins #include <errno.h>
31e92a3d83STim J. Robbins #include <limits.h>
32e92a3d83STim J. Robbins #include <stdlib.h>
33e92a3d83STim J. Robbins #include <wchar.h>
346155c34aSTim J. Robbins #include "mblocal.h"
35e92a3d83STim J. Robbins 
36e92a3d83STim J. Robbins size_t
37e92a3d83STim J. Robbins mbsrtowcs(wchar_t * __restrict dst, const char ** __restrict src, size_t len,
3874f90defSTim J. Robbins     mbstate_t * __restrict ps)
39e92a3d83STim J. Robbins {
4074f90defSTim J. Robbins 	static mbstate_t mbs;
41e92a3d83STim J. Robbins 	const char *s;
42e92a3d83STim J. Robbins 	size_t nchr;
43e92a3d83STim J. Robbins 	wchar_t wc;
44e92a3d83STim J. Robbins 	int nb;
45e92a3d83STim J. Robbins 
46e92a3d83STim J. Robbins 	s = *src;
47e92a3d83STim J. Robbins 	nchr = 0;
48e92a3d83STim J. Robbins 
4974f90defSTim J. Robbins 	if (ps == NULL)
5074f90defSTim J. Robbins 		ps = &mbs;
51e92a3d83STim J. Robbins 	if (dst == NULL) {
52e92a3d83STim J. Robbins 		for (;;) {
536155c34aSTim J. Robbins 			if ((nb = (int)__mbrtowc(&wc, s, MB_CUR_MAX, ps)) < 0)
54e92a3d83STim J. Robbins 				/* Invalid sequence - mbrtowc() sets errno. */
55e92a3d83STim J. Robbins 				return ((size_t)-1);
56e92a3d83STim J. Robbins 			else if (nb == 0)
57e92a3d83STim J. Robbins 				return (nchr);
58e92a3d83STim J. Robbins 			s += nb;
59e92a3d83STim J. Robbins 			nchr++;
60e92a3d83STim J. Robbins 		}
61e92a3d83STim J. Robbins 		/*NOTREACHED*/
62e92a3d83STim J. Robbins 	}
63e92a3d83STim J. Robbins 
64e92a3d83STim J. Robbins 	while (len-- > 0) {
656155c34aSTim J. Robbins 		if ((nb = (int)__mbrtowc(dst, s, MB_CUR_MAX, ps)) < 0) {
66e92a3d83STim J. Robbins 			*src = s;
67e92a3d83STim J. Robbins 			return ((size_t)-1);
68e92a3d83STim J. Robbins 		} else if (nb == 0) {
69e92a3d83STim J. Robbins 			*src = NULL;
70e92a3d83STim J. Robbins 			return (nchr);
71e92a3d83STim J. Robbins 		}
72e92a3d83STim J. Robbins 		s += nb;
73e92a3d83STim J. Robbins 		nchr++;
74e92a3d83STim J. Robbins 		dst++;
75e92a3d83STim J. Robbins 	}
76e92a3d83STim J. Robbins 	*src = s;
77e92a3d83STim J. Robbins 	return (nchr);
78e92a3d83STim J. Robbins }
79