xref: /freebsd/lib/libc/locale/mbsrtowcs.c (revision e92a3d83fcba95f35d5f162df92f2d5b8351c2e3)
1e92a3d83STim J. Robbins /*-
2e92a3d83STim J. Robbins  * Copyright (c) 2002 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>
34e92a3d83STim J. Robbins 
35e92a3d83STim J. Robbins size_t
36e92a3d83STim J. Robbins mbsrtowcs(wchar_t *__restrict dst, const char **__restrict src, size_t len,
37e92a3d83STim J. Robbins     mbstate_t *__restrict ps __unused)
38e92a3d83STim J. Robbins {
39e92a3d83STim J. Robbins 	const char *s;
40e92a3d83STim J. Robbins 	size_t nchr;
41e92a3d83STim J. Robbins 	wchar_t wc;
42e92a3d83STim J. Robbins 	int nb;
43e92a3d83STim J. Robbins 
44e92a3d83STim J. Robbins 	s = *src;
45e92a3d83STim J. Robbins 	nchr = 0;
46e92a3d83STim J. Robbins 
47e92a3d83STim J. Robbins 	if (dst == NULL) {
48e92a3d83STim J. Robbins 		for (;;) {
49e92a3d83STim J. Robbins 			if ((nb = (int)mbrtowc(&wc, s, MB_CUR_MAX, NULL)) < 0)
50e92a3d83STim J. Robbins 				/* Invalid sequence - mbrtowc() sets errno. */
51e92a3d83STim J. Robbins 				return ((size_t)-1);
52e92a3d83STim J. Robbins 			else if (nb == 0)
53e92a3d83STim J. Robbins 				return (nchr);
54e92a3d83STim J. Robbins 			s += nb;
55e92a3d83STim J. Robbins 			nchr++;
56e92a3d83STim J. Robbins 		}
57e92a3d83STim J. Robbins 		/*NOTREACHED*/
58e92a3d83STim J. Robbins 	}
59e92a3d83STim J. Robbins 
60e92a3d83STim J. Robbins 	while (len-- > 0) {
61e92a3d83STim J. Robbins 		if ((nb = (int)mbrtowc(dst, s, MB_CUR_MAX, NULL)) < 0) {
62e92a3d83STim J. Robbins 			*src = s;
63e92a3d83STim J. Robbins 			return ((size_t)-1);
64e92a3d83STim J. Robbins 		} else if (nb == 0) {
65e92a3d83STim J. Robbins 			*src = NULL;
66e92a3d83STim J. Robbins 			return (nchr);
67e92a3d83STim J. Robbins 		}
68e92a3d83STim J. Robbins 		s += nb;
69e92a3d83STim J. Robbins 		nchr++;
70e92a3d83STim J. Robbins 		dst++;
71e92a3d83STim J. Robbins 	}
72e92a3d83STim J. Robbins 	*src = s;
73e92a3d83STim J. Robbins 	return (nchr);
74e92a3d83STim J. Robbins }
75