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