1 /* $NetBSD: t_wctomb.c,v 1.3 2013/03/25 15:31:03 gson Exp $ */
2 
3 /*-
4  * Copyright (c) 2011 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26  * POSSIBILITY OF SUCH DAMAGE.
27  */
28 
29 /*-
30  * Copyright (c)2004 Citrus Project,
31  * All rights reserved.
32  *
33  * Redistribution and use in source and binary forms, with or without
34  * modification, are permitted provided that the following conditions
35  * are met:
36  * 1. Redistributions of source code must retain the above copyright
37  *    notice, this list of conditions and the following disclaimer.
38  * 2. Redistributions in binary form must reproduce the above copyright
39  *    notice, this list of conditions and the following disclaimer in the
40  *    documentation and/or other materials provided with the distribution.
41  *
42  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
43  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
44  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
45  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
46  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
47  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
48  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
49  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
50  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
51  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
52  * SUCH DAMAGE.
53  */
54 
55 #include <sys/cdefs.h>
56 __COPYRIGHT("@(#) Copyright (c) 2011\
57  The NetBSD Foundation, inc. All rights reserved.");
58 __RCSID("$NetBSD: t_wctomb.c,v 1.3 2013/03/25 15:31:03 gson Exp $");
59 
60 #include <stdio.h>
61 #include <stdlib.h>
62 #include <locale.h>
63 #include <vis.h>
64 #include <wchar.h>
65 #include <string.h>
66 #include <limits.h>
67 
68 #include <atf-c.h>
69 
70 #define TC_WCTOMB	0
71 #define TC_WCRTOMB	1
72 #define TC_WCRTOMB_ST	2
73 
74 static struct test {
75 	const char *locale;
76 	const char *data;
77 	size_t wclen;
78 	size_t mblen[16];
79 } tests[] = {
80 {
81 	"ja_JP.ISO2022-JP",
82 	"\x1b$B"	/* JIS X 0208-1983 */
83 	"\x46\x7c\x4b\x5c\x38\x6c" /* "nihongo" */
84 	"\x1b(B"	/* ISO 646 */
85 	"ABC"
86 	"\x1b(I"	/* JIS X 0201 katakana */
87 	"\xb1\xb2\xb3"	/* "aiu" */
88 	"\x1b(B",	/* ISO 646 */
89 	3 + 3 + 3,
90 	{ 3+2, 2, 2, 3+1, 1, 1, 3+1, 1, 1, 3+1 }
91 }, {
92 	"C",
93 	"ABC",
94 	3,
95 	{ 1, 1, 1, 1 }
96 }, { NULL, NULL, 0, { } }
97 };
98 
99 static void
100 h_wctomb(const struct test *t, char tc)
101 {
102 	wchar_t wcs[16 + 2];
103 	char buf[128];
104 	char cs[MB_LEN_MAX];
105 	const char *pcs;
106 	char *str;
107 	mbstate_t st;
108 	mbstate_t *stp = NULL;
109 	size_t sz, ret, i;
110 
111 	ATF_REQUIRE_STREQ(setlocale(LC_ALL, "C"), "C");
112 #ifdef __NetBSD__
113 	ATF_REQUIRE(setlocale(LC_CTYPE, t->locale) != NULL);
114 #else
115 	if (setlocale(LC_CTYPE, t->locale) == NULL) {
116 		fprintf(stderr, "Locale %s not found.\n", t->locale);
117 		return;
118 	}
119 #endif
120 
121 	(void)strvis(buf, t->data, VIS_WHITE | VIS_OCTAL);
122 	(void)printf("Checking sequence: \"%s\"\n", buf);
123 
124 	ATF_REQUIRE((str = setlocale(LC_ALL, NULL)) != NULL);
125 	(void)printf("Using locale: %s\n", str);
126 
127 	if (tc == TC_WCRTOMB_ST) {
128 		(void)memset(&st, 0, sizeof(st));
129 		stp = &st;
130 	}
131 
132 	wcs[t->wclen] = L'X'; /* poison */
133 	pcs = t->data;
134 	sz = mbsrtowcs(wcs, &pcs, t->wclen + 2, NULL);
135 	ATF_REQUIRE_EQ_MSG(sz, t->wclen, "mbsrtowcs() returned: "
136 		"%zu, expected: %zu", sz, t->wclen);
137 	ATF_REQUIRE_EQ(wcs[t->wclen], 0);
138 
139 	for (i = 0; i < t->wclen + 1; i++) {
140 		if (tc == TC_WCTOMB)
141 			ret = wctomb(cs, wcs[i]);
142 		else
143 			ret = wcrtomb(cs, wcs[i], stp);
144 
145 		if (ret == t->mblen[i])
146 			continue;
147 
148 		(void)printf("At position %zd:\n", i);
149 		(void)printf("  expected: %zd\n", t->mblen[i]);
150 		(void)printf("  got     : %zd\n", ret);
151 		atf_tc_fail("Test failed");
152 		/* NOTREACHED */
153 	}
154 
155 	(void)printf("Ok.\n");
156 }
157 
158 ATF_TC(wctomb);
159 ATF_TC_HEAD(wctomb, tc)
160 {
161 	atf_tc_set_md_var(tc, "descr", "Checks wctomb(3)");
162 }
163 ATF_TC_BODY(wctomb, tc)
164 {
165 	struct test *t;
166 
167 	(void)printf("Checking wctomb()\n");
168 
169 	for (t = &tests[0]; t->data != NULL; ++t)
170 		h_wctomb(t, TC_WCTOMB);
171 }
172 
173 ATF_TC(wcrtomb_state);
174 ATF_TC_HEAD(wcrtomb_state, tc)
175 {
176 	atf_tc_set_md_var(tc, "descr",
177 		"Checks wcrtomb(3) (using state object)");
178 }
179 ATF_TC_BODY(wcrtomb_state, tc)
180 {
181 	struct test *t;
182 
183 	(void)printf("Checking wcrtomb() (with state object)\n");
184 
185 	for (t = &tests[0]; t->data != NULL; ++t)
186 		h_wctomb(t, TC_WCRTOMB_ST);
187 }
188 
189 ATF_TC(wcrtomb);
190 ATF_TC_HEAD(wcrtomb, tc)
191 {
192 	atf_tc_set_md_var(tc, "descr",
193 		"Checks wcrtomb(3) (using internal state)");
194 }
195 ATF_TC_BODY(wcrtomb, tc)
196 {
197 	struct test *t;
198 
199 	(void)printf("Checking wcrtomb() (using internal state)\n");
200 
201 	for (t = &tests[0]; t->data != NULL; ++t)
202 		h_wctomb(t, TC_WCRTOMB);
203 }
204 
205 ATF_TP_ADD_TCS(tp)
206 {
207 	ATF_TP_ADD_TC(tp, wctomb);
208 	ATF_TP_ADD_TC(tp, wcrtomb);
209 	ATF_TP_ADD_TC(tp, wcrtomb_state);
210 
211 	return atf_no_error();
212 }
213