wcsnrtombs_test.c revision 132499
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 * Test program for wcsnrtombs().
29 *
30 * The function is tested with both the "C" ("POSIX") LC_CTYPE setting and
31 * "ja_JP.eucJP". Other encodings are not tested.
32 */
33
34#include <sys/cdefs.h>
35__FBSDID("$FreeBSD: head/tools/regression/lib/libc/locale/test-wcsnrtombs.c 132499 2004-07-21 13:47:48Z tjr $");
36
37#include <assert.h>
38#include <errno.h>
39#include <limits.h>
40#include <locale.h>
41#include <stdio.h>
42#include <stdlib.h>
43#include <string.h>
44#include <wchar.h>
45
46int
47main(int argc, char *argv[])
48{
49	wchar_t srcbuf[128];
50	char dstbuf[128];
51	wchar_t *src;
52	mbstate_t s;
53
54	/*
55	 * C/POSIX locale.
56	 */
57
58	/* Simple null terminated string. */
59	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
60	wcscpy(srcbuf, L"hello");
61	memset(dstbuf, 0xcc, sizeof(dstbuf));
62	src = srcbuf;
63	memset(&s, 0, sizeof(s));
64	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 6, sizeof(dstbuf),
65	    &s) == 5);
66	assert(strcmp(dstbuf, "hello") == 0);
67	assert((unsigned char)dstbuf[6] == 0xcc);
68	assert(src == NULL);
69
70	/* Simple null terminated string, stopping early. */
71	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
72	wcscpy(srcbuf, L"hello");
73	memset(dstbuf, 0xcc, sizeof(dstbuf));
74	src = srcbuf;
75	memset(&s, 0, sizeof(s));
76	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 4, sizeof(dstbuf),
77	    &s) == 4);
78	assert(memcmp(dstbuf, "hell", 4) == 0);
79	assert((unsigned char)dstbuf[5] == 0xcc);
80	assert(src == srcbuf + 4);
81
82	/* Not enough space in destination buffer. */
83	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
84	wcscpy(srcbuf, L"hello");
85	memset(dstbuf, 0xcc, sizeof(dstbuf));
86	src = srcbuf;
87	memset(&s, 0, sizeof(s));
88	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 6, 4,
89	    &s) == 4);
90	assert(memcmp(dstbuf, "hell", 4) == 0);
91	assert((unsigned char)dstbuf[5] == 0xcc);
92	assert(src == srcbuf + 4);
93
94	/* Null terminated string, internal dest. buffer */
95	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
96	wcscpy(srcbuf, L"hello");
97	src = srcbuf;
98	memset(&s, 0, sizeof(s));
99	assert(wcsnrtombs(NULL, (const wchar_t **)&src, 6, sizeof(dstbuf),
100	    &s) == 5);
101
102	/* Null terminated string, internal dest. buffer, stopping early. */
103	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
104	wcscpy(srcbuf, L"hello");
105	src = srcbuf;
106	memset(&s, 0, sizeof(s));
107	assert(wcsnrtombs(NULL, (const wchar_t **)&src, 4, sizeof(dstbuf),
108	    &s) == 4);
109
110	/* Null terminated string, internal state. */
111	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
112	wcscpy(srcbuf, L"hello");
113	memset(dstbuf, 0xcc, sizeof(dstbuf));
114	src = srcbuf;
115	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 6, sizeof(dstbuf),
116	    NULL) == 5);
117	assert(strcmp(dstbuf, "hello") == 0);
118	assert((unsigned char)dstbuf[6] == 0xcc);
119	assert(src == NULL);
120
121	/* Null terminated string, internal state, internal dest. buffer. */
122	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
123	wcscpy(srcbuf, L"hello");
124	src = srcbuf;
125	assert(wcsnrtombs(NULL, (const wchar_t **)&src, 6, 0, NULL) == 5);
126
127	/* Empty source buffer. */
128	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
129	srcbuf[0] = L'\0';
130	memset(dstbuf, 0xcc, sizeof(dstbuf));
131	src = srcbuf;
132	memset(&s, 0, sizeof(s));
133	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 1, sizeof(dstbuf),
134	    &s) == 0);
135	assert(dstbuf[0] == L'\0');
136
137	/* Zero length destination buffer. */
138	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
139	wcscpy(srcbuf, L"hello");
140	memset(dstbuf, 0xcc, sizeof(dstbuf));
141	src = srcbuf;
142	memset(&s, 0, sizeof(s));
143	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 6, 0, &s) == 0);
144	assert((unsigned char)dstbuf[0] == 0xcc);
145
146	/* Zero length source buffer. */
147	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
148	memset(dstbuf, 0xcc, sizeof(dstbuf));
149	src = srcbuf;
150	memset(&s, 0, sizeof(s));
151	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 0, sizeof(dstbuf),
152	    &s) == 0);
153	assert((unsigned char)dstbuf[0] == 0xcc);
154	assert(src == srcbuf);
155
156	/*
157	 * Japanese (EUC) locale.
158	 */
159
160	assert(strcmp(setlocale(LC_CTYPE, "ja_JP.eucJP"), "ja_JP.eucJP") == 0);
161	assert(MB_CUR_MAX > 1);
162
163	wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf));
164	srcbuf[0] = 0xA3C1;
165	srcbuf[1] = 0x0020;
166	srcbuf[2] = 0x0042;
167	srcbuf[3] = 0x0020;
168	srcbuf[4] = 0xA3C3;
169	srcbuf[5] = 0x0000;
170	memset(dstbuf, 0xcc, sizeof(dstbuf));
171	src = srcbuf;
172	memset(&s, 0, sizeof(s));
173	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 6, sizeof(dstbuf),
174	    &s) == 7);
175	assert(strcmp(dstbuf, "\xA3\xC1 B \xA3\xC3") == 0);
176	assert((unsigned char)dstbuf[8] == 0xcc);
177	assert(src == NULL);
178
179	/* Stopping early. */
180	memset(dstbuf, 0xcc, sizeof(dstbuf));
181	src = srcbuf;
182	memset(&s, 0, sizeof(s));
183	assert(wcsnrtombs(dstbuf, (const wchar_t **)&src, 6, 6,
184	    &s) == 5);
185	assert(memcmp(dstbuf, "\xA3\xC1 B ", 5) == 0);
186	assert((unsigned char)dstbuf[5] == 0xcc);
187	assert(src == srcbuf + 4);
188
189	printf("PASS wcsnrtombs()\n");
190
191	return (0);
192}
193