1/*	$NetBSD: t_vis.c,v 1.9 2017/01/10 15:16:57 christos Exp $	*/
2
3/*-
4 * Copyright (c) 2002 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code was contributed to The NetBSD Foundation by Christos Zoulas.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 *    notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 *    notice, this list of conditions and the following disclaimer in the
16 *    documentation and/or other materials provided with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
19 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
20 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
22 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28 * POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#include <atf-c.h>
32
33#include <string.h>
34#include <stdlib.h>
35#include <locale.h>
36#include <err.h>
37#include <vis.h>
38
39static int styles[] = {
40	VIS_OCTAL,
41	VIS_CSTYLE,
42	VIS_SP,
43	VIS_TAB,
44	VIS_NL,
45	VIS_WHITE,
46	VIS_SAFE,
47#if 0	/* Not reversible */
48	VIS_NOSLASH,
49#endif
50	VIS_HTTP1808,
51	VIS_MIMESTYLE,
52#if 0	/* Not supported by vis(3) */
53	VIS_HTTP1866,
54#endif
55};
56
57#define SIZE	256
58
59ATF_TC(strvis_basic);
60ATF_TC_HEAD(strvis_basic, tc)
61{
62
63	atf_tc_set_md_var(tc, "descr", "Test strvis(3)");
64}
65
66ATF_TC_BODY(strvis_basic, tc)
67{
68	char *srcbuf, *dstbuf, *visbuf;
69	unsigned int i, j;
70
71	ATF_REQUIRE((dstbuf = malloc(SIZE)) != NULL);
72	ATF_REQUIRE((srcbuf = malloc(SIZE)) != NULL);
73	ATF_REQUIRE((visbuf = malloc(SIZE * 4 + 1)) != NULL);
74
75	for (i = 0; i < SIZE; i++)
76		srcbuf[i] = (char)i;
77
78	for (i = 0; i < __arraycount(styles); i++) {
79		ATF_REQUIRE(strsvisx(visbuf, srcbuf, SIZE, styles[i], "") > 0);
80		memset(dstbuf, 0, SIZE);
81		ATF_REQUIRE(strunvisx(dstbuf, visbuf,
82		    styles[i] & (VIS_HTTP1808|VIS_MIMESTYLE)) > 0);
83		for (j = 0; j < SIZE; j++)
84			if (dstbuf[j] != (char)j)
85				atf_tc_fail_nonfatal("Failed for style %x, "
86				    "char %d [%d]", styles[i], j, dstbuf[j]);
87	}
88	free(dstbuf);
89	free(srcbuf);
90	free(visbuf);
91}
92
93ATF_TC(strvis_null);
94ATF_TC_HEAD(strvis_null, tc)
95{
96	atf_tc_set_md_var(tc, "descr", "Test strvis(3) NULL");
97}
98
99ATF_TC_BODY(strvis_null, tc)
100{
101	char dst[] = "fail";
102	strvis(dst, NULL, VIS_SAFE);
103	ATF_REQUIRE(dst[0] == '\0' && dst[1] == 'a');
104}
105
106ATF_TC(strvis_empty);
107ATF_TC_HEAD(strvis_empty, tc)
108{
109	atf_tc_set_md_var(tc, "descr", "Test strvis(3) empty");
110}
111
112ATF_TC_BODY(strvis_empty, tc)
113{
114	char dst[] = "fail";
115	strvis(dst, "", VIS_SAFE);
116	ATF_REQUIRE(dst[0] == '\0' && dst[1] == 'a');
117}
118
119ATF_TC(strunvis_hex);
120ATF_TC_HEAD(strunvis_hex, tc)
121{
122	atf_tc_set_md_var(tc, "descr", "Test strunvis(3) \\xXX");
123}
124
125ATF_TC_BODY(strunvis_hex, tc)
126{
127	static const struct {
128		const char *e;
129		const char *d;
130		int error;
131	} ed[] = {
132		{ "\\xff", "\xff", 1 },
133		{ "\\x1", "\x1", 1 },
134		{ "\\x1\\x02", "\x1\x2", 2 },
135		{ "\\x1x", "\x1x", 2 },
136		{ "\\xx", "", -1 },
137	};
138	char uv[10];
139
140	for (size_t i = 0; i < __arraycount(ed); i++) {
141		ATF_REQUIRE(strunvis(uv, ed[i].e) == ed[i].error);
142		if (ed[i].error > 0)
143			ATF_REQUIRE(memcmp(ed[i].d, uv, ed[i].error) == 0);
144	}
145}
146
147#ifdef VIS_NOLOCALE
148ATF_TC(strvis_locale);
149ATF_TC_HEAD(strvis_locale, tc)
150{
151	atf_tc_set_md_var(tc, "descr", "Test strvis(3) with locale");
152}
153
154ATF_TC_BODY(strvis_locale, tc)
155{
156	char s[256], cd[sizeof(s) * 4 + 1], jd[sizeof(cd)], *ol;
157	int jr, cr;
158
159	for (size_t i = 0; i < sizeof(s) - 1; i++)
160		s[i] = i + 1;
161	s[sizeof(s) - 1] = '\0';
162
163	ol = setlocale(LC_CTYPE, "ja_JP.UTF-8");
164	ATF_REQUIRE(ol != NULL);
165	jr = strvisx(jd, s, sizeof(s), VIS_WHITE | VIS_NOLOCALE);
166	ATF_REQUIRE(jr != -1);
167	ol = strdup(ol);
168	ATF_REQUIRE(ol != NULL);
169	ATF_REQUIRE(setlocale(LC_CTYPE, "C") != NULL);
170	cr = strvisx(cd, s, sizeof(s), VIS_WHITE);
171	ATF_REQUIRE(jr == cr);
172	ATF_REQUIRE(memcmp(jd, cd, jr) == 0);
173	setlocale(LC_CTYPE, ol);
174	free(ol);
175}
176#endif /* VIS_NOLOCALE */
177
178ATF_TP_ADD_TCS(tp)
179{
180
181	ATF_TP_ADD_TC(tp, strvis_basic);
182	ATF_TP_ADD_TC(tp, strvis_null);
183	ATF_TP_ADD_TC(tp, strvis_empty);
184	ATF_TP_ADD_TC(tp, strunvis_hex);
185#ifdef VIS_NOLOCALE
186	ATF_TP_ADD_TC(tp, strvis_locale);
187#endif /* VIS_NOLOCALE */
188
189	return atf_no_error();
190}
191