1/* $NetBSD: t_wctomb.c,v 1.5 2017/07/12 17:32:51 perseant 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.5 2017/07/12 17:32:51 perseant 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
74static struct test {
75	const char *locale;
76	const char *data;
77	size_t wclen;
78	size_t mblen[16];
79	size_t stateful;
80} tests[] = {
81{
82	"ja_JP.ISO2022-JP",
83	"\x1b$B"	/* JIS X 0208-1983 */
84	"\x46\x7c\x4b\x5c\x38\x6c" /* "nihongo" */
85	"\x1b(B"	/* ISO 646 */
86	"ABC"
87	"\x1b(I"	/* JIS X 0201 katakana */
88	"\xb1\xb2\xb3"	/* "aiu" */
89	"\x1b(B",	/* ISO 646 */
90	3 + 3 + 3,
91	{ 3+2, 2, 2, 3+1, 1, 1, 3+1, 1, 1, 3+1 },
92	1,
93}, {
94	"C",
95	"ABC",
96	3,
97	{ 1, 1, 1, 1 },
98	0,
99}, { NULL, NULL, 0, { }, 0 }
100};
101
102static void
103h_wctomb(const struct test *t, char tc)
104{
105	wchar_t wcs[16 + 2];
106	char buf[128];
107	char cs[MB_LEN_MAX];
108	const char *pcs;
109	char *str;
110	mbstate_t st;
111	mbstate_t *stp = NULL;
112	size_t sz, ret, i;
113
114	ATF_REQUIRE_STREQ(setlocale(LC_ALL, "C"), "C");
115	(void)printf("Trying locale: %s\n", t->locale);
116	ATF_REQUIRE(setlocale(LC_CTYPE, t->locale) != NULL);
117
118	if (tc == TC_WCRTOMB_ST) {
119		(void)memset(&st, 0, sizeof(st));
120		stp = &st;
121	} else {
122		(void)printf("Checking correct reporting of statefulness\n");
123		ret = wctomb(NULL, 0);
124		ATF_REQUIRE_EQ(t->stateful, !!ret);
125	}
126
127	(void)strvis(buf, t->data, VIS_WHITE | VIS_OCTAL);
128	(void)printf("Checking sequence: \"%s\"\n", buf);
129
130	ATF_REQUIRE((str = setlocale(LC_ALL, NULL)) != NULL);
131	(void)printf("Using locale: %s\n", str);
132
133	wcs[t->wclen] = L'X'; /* poison */
134	pcs = t->data;
135	sz = mbsrtowcs(wcs, &pcs, t->wclen + 2, NULL);
136	ATF_REQUIRE_EQ_MSG(sz, t->wclen, "mbsrtowcs() returned: "
137		"%zu, expected: %zu", sz, t->wclen);
138	ATF_REQUIRE_EQ(wcs[t->wclen], 0);
139
140	for (i = 0; i < t->wclen + 1; i++) {
141		if (tc == TC_WCTOMB)
142			ret = wctomb(cs, wcs[i]);
143		else
144			ret = wcrtomb(cs, wcs[i], stp);
145
146		if (ret == t->mblen[i])
147			continue;
148
149		(void)printf("At position %zd:\n", i);
150		(void)printf("  expected: %zd\n", t->mblen[i]);
151		(void)printf("  got     : %zd\n", ret);
152		(void)strvis(buf, cs, VIS_WHITE | VIS_OCTAL);
153		(void)printf("  sequence: \"%s\"\n", buf);
154		atf_tc_fail("Test failed");
155		/* NOTREACHED */
156	}
157
158	(void)printf("Ok.\n");
159}
160
161ATF_TC(wctomb);
162ATF_TC_HEAD(wctomb, tc)
163{
164	atf_tc_set_md_var(tc, "descr", "Checks wctomb(3)");
165}
166ATF_TC_BODY(wctomb, tc)
167{
168	struct test *t;
169
170	(void)printf("Checking wctomb()\n");
171
172	for (t = &tests[0]; t->data != NULL; ++t)
173		h_wctomb(t, TC_WCTOMB);
174}
175
176ATF_TC(wcrtomb_state);
177ATF_TC_HEAD(wcrtomb_state, tc)
178{
179	atf_tc_set_md_var(tc, "descr",
180		"Checks wcrtomb(3) (using state object)");
181}
182ATF_TC_BODY(wcrtomb_state, tc)
183{
184	struct test *t;
185
186	(void)printf("Checking wcrtomb() (with state object)\n");
187
188	for (t = &tests[0]; t->data != NULL; ++t)
189		h_wctomb(t, TC_WCRTOMB_ST);
190}
191
192ATF_TC(wcrtomb);
193ATF_TC_HEAD(wcrtomb, tc)
194{
195	atf_tc_set_md_var(tc, "descr",
196		"Checks wcrtomb(3) (using internal state)");
197}
198ATF_TC_BODY(wcrtomb, tc)
199{
200	struct test *t;
201
202	(void)printf("Checking wcrtomb() (using internal state)\n");
203
204	for (t = &tests[0]; t->data != NULL; ++t)
205		h_wctomb(t, TC_WCRTOMB);
206}
207
208ATF_TP_ADD_TCS(tp)
209{
210	ATF_TP_ADD_TC(tp, wctomb);
211	ATF_TP_ADD_TC(tp, wcrtomb);
212	ATF_TP_ADD_TC(tp, wcrtomb_state);
213
214	return atf_no_error();
215}
216