xref: /titanic_51/usr/src/lib/libc/port/locale/wcsnrtombs.c (revision eb8f03cde11ee9fc86ee179ef61fc5861756b1c6)
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 /*
28  * Copyright 2010 Nexenta Systems, Inc.  All rights reserved.
29  * Use is subject to license terms.
30  */
31 
32 #include "lint.h"
33 #include <limits.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <wchar.h>
37 #include "mblocal.h"
38 
39 size_t
40 wcsnrtombs(char *_RESTRICT_KYWD dst, const wchar_t **_RESTRICT_KYWD src,
41     size_t nwc, size_t len, mbstate_t *_RESTRICT_KYWD ps)
42 {
43 	static mbstate_t mbs;
44 
45 	if (ps == NULL)
46 		ps = &mbs;
47 	return (__wcsnrtombs(dst, src, nwc, len, ps));
48 }
49 
50 size_t
51 __wcsnrtombs_std(char *_RESTRICT_KYWD dst, const wchar_t **_RESTRICT_KYWD src,
52     size_t nwc, size_t len, mbstate_t *_RESTRICT_KYWD ps)
53 {
54 	mbstate_t mbsbak;
55 	char buf[MB_LEN_MAX];
56 	const wchar_t *s;
57 	size_t nbytes;
58 	size_t nb;
59 
60 	s = *src;
61 	nbytes = 0;
62 
63 	if (dst == NULL) {
64 		while (nwc-- > 0) {
65 			if ((nb = __wcrtomb(buf, *s, ps)) == (size_t)-1)
66 				/* Invalid character - wcrtomb() sets errno. */
67 				return ((size_t)-1);
68 			else if (*s == L'\0')
69 				return (nbytes + nb - 1);
70 			s++;
71 			nbytes += nb;
72 		}
73 		return (nbytes);
74 	}
75 
76 	while (len > 0 && nwc-- > 0) {
77 		if (len > (size_t)MB_CUR_MAX) {
78 			/* Enough space to translate in-place. */
79 			if ((nb = __wcrtomb(dst, *s, ps)) == (size_t)-1) {
80 				*src = s;
81 				return ((size_t)-1);
82 			}
83 		} else {
84 			/*
85 			 * May not be enough space; use temp. buffer.
86 			 *
87 			 * We need to save a copy of the conversion state
88 			 * here so we can restore it if the multibyte
89 			 * character is too long for the buffer.
90 			 */
91 			mbsbak = *ps;
92 			if ((nb = __wcrtomb(buf, *s, ps)) == (size_t)-1) {
93 				*src = s;
94 				return ((size_t)-1);
95 			}
96 			if (nb > (int)len) {
97 				/* MB sequence for character won't fit. */
98 				*ps = mbsbak;
99 				break;
100 			}
101 			(void) memcpy(dst, buf, nb);
102 		}
103 		if (*s == L'\0') {
104 			*src = NULL;
105 			return (nbytes + nb - 1);
106 		}
107 		s++;
108 		dst += nb;
109 		len -= nb;
110 		nbytes += nb;
111 	}
112 	*src = s;
113 	return (nbytes);
114 }
115