xref: /freebsd/lib/libc/locale/wcsrtombs.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 <string.h>
34e92a3d83STim J. Robbins #include <wchar.h>
35e92a3d83STim J. Robbins 
36e92a3d83STim J. Robbins size_t
37e92a3d83STim J. Robbins wcsrtombs(char *__restrict dst, const wchar_t **__restrict src, size_t len,
38e92a3d83STim J. Robbins     mbstate_t *__restrict ps __unused)
39e92a3d83STim J. Robbins {
40e92a3d83STim J. Robbins 	char buf[MB_LEN_MAX];
41e92a3d83STim J. Robbins 	const wchar_t *s;
42e92a3d83STim J. Robbins 	size_t nbytes;
43e92a3d83STim J. Robbins 	int nb;
44e92a3d83STim J. Robbins 
45e92a3d83STim J. Robbins 	s = *src;
46e92a3d83STim J. Robbins 	nbytes = 0;
47e92a3d83STim J. Robbins 
48e92a3d83STim J. Robbins 	if (dst == NULL) {
49e92a3d83STim J. Robbins 		for (;;) {
50e92a3d83STim J. Robbins 			if ((nb = (int)wcrtomb(buf, *s, NULL)) < 0)
51e92a3d83STim J. Robbins 				/* Invalid character - wcrtomb() sets errno. */
52e92a3d83STim J. Robbins 				return ((size_t)-1);
53e92a3d83STim J. Robbins 			else if (*s == L'\0')
54e92a3d83STim J. Robbins 				return (nbytes + nb - 1);
55e92a3d83STim J. Robbins 			s++;
56e92a3d83STim J. Robbins 			nbytes += nb;
57e92a3d83STim J. Robbins 		}
58e92a3d83STim J. Robbins 		/*NOTREACHED*/
59e92a3d83STim J. Robbins 	}
60e92a3d83STim J. Robbins 
61e92a3d83STim J. Robbins 	while (len > 0) {
62e92a3d83STim J. Robbins 		if (len > (size_t)MB_CUR_MAX) {
63e92a3d83STim J. Robbins 			/* Enough space to translate in-place. */
64e92a3d83STim J. Robbins 			if ((nb = (int)wcrtomb(dst, *s, NULL)) < 0) {
65e92a3d83STim J. Robbins 				*src = s;
66e92a3d83STim J. Robbins 				return ((size_t)-1);
67e92a3d83STim J. Robbins 			}
68e92a3d83STim J. Robbins 		} else {
69e92a3d83STim J. Robbins 			/* May not be enough space; use temp. buffer. */
70e92a3d83STim J. Robbins 			if ((nb = (int)wcrtomb(buf, *s, NULL)) < 0) {
71e92a3d83STim J. Robbins 				*src = s;
72e92a3d83STim J. Robbins 				return ((size_t)-1);
73e92a3d83STim J. Robbins 			}
74e92a3d83STim J. Robbins 			if (nb > (int)len)
75e92a3d83STim J. Robbins 				/* MB sequence for character won't fit. */
76e92a3d83STim J. Robbins 				break;
77e92a3d83STim J. Robbins 			memcpy(dst, buf, nb);
78e92a3d83STim J. Robbins 		}
79e92a3d83STim J. Robbins 		if (*s == L'\0') {
80e92a3d83STim J. Robbins 			*src = NULL;
81e92a3d83STim J. Robbins 			return (nbytes + nb - 1);
82e92a3d83STim J. Robbins 		}
83e92a3d83STim J. Robbins 		s++;
84e92a3d83STim J. Robbins 		dst += nb;
85e92a3d83STim J. Robbins 		len -= nb;
86e92a3d83STim J. Robbins 		nbytes += nb;
87e92a3d83STim J. Robbins 	}
88e92a3d83STim J. Robbins 	*src = s;
89e92a3d83STim J. Robbins 	return (nbytes);
90e92a3d83STim J. Robbins }
91