1 /* $NetBSD: t_vis.c,v 1.8 2015/05/23 14:02:11 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
39 static 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
59 ATF_TC(strvis_basic);
ATF_TC_HEAD(strvis_basic,tc)60 ATF_TC_HEAD(strvis_basic, tc)
61 {
62
63 atf_tc_set_md_var(tc, "descr", "Test strvis(3)");
64 }
65
ATF_TC_BODY(strvis_basic,tc)66 ATF_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
93 ATF_TC(strvis_null);
ATF_TC_HEAD(strvis_null,tc)94 ATF_TC_HEAD(strvis_null, tc)
95 {
96 atf_tc_set_md_var(tc, "descr", "Test strvis(3) NULL");
97 }
98
ATF_TC_BODY(strvis_null,tc)99 ATF_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
106 ATF_TC(strvis_empty);
ATF_TC_HEAD(strvis_empty,tc)107 ATF_TC_HEAD(strvis_empty, tc)
108 {
109 atf_tc_set_md_var(tc, "descr", "Test strvis(3) empty");
110 }
111
ATF_TC_BODY(strvis_empty,tc)112 ATF_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
119 ATF_TC(strunvis_hex);
ATF_TC_HEAD(strunvis_hex,tc)120 ATF_TC_HEAD(strunvis_hex, tc)
121 {
122 atf_tc_set_md_var(tc, "descr", "Test strunvis(3) \\xXX");
123 }
124
ATF_TC_BODY(strunvis_hex,tc)125 ATF_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 ATF_TC(strvis_locale);
ATF_TC_HEAD(strvis_locale,tc)148 ATF_TC_HEAD(strvis_locale, tc)
149 {
150 atf_tc_set_md_var(tc, "descr", "Test strvis(3) with locale");
151 }
152
ATF_TC_BODY(strvis_locale,tc)153 ATF_TC_BODY(strvis_locale, tc)
154 {
155 char s[256], cd[sizeof(s) * 4 + 1], jd[sizeof(cd)], *ol;
156 int jr, cr;
157
158 for (size_t i = 0; i < sizeof(s) - 1; i++)
159 s[i] = i + 1;
160 s[sizeof(s) - 1] = '\0';
161
162 ol = setlocale(LC_CTYPE, "ja_JP.UTF-8");
163 ATF_REQUIRE(ol != NULL);
164 jr = strvisx(jd, s, sizeof(s), VIS_WHITE | VIS_NOLOCALE);
165 ATF_REQUIRE(jr != -1);
166 ol = strdup(ol);
167 ATF_REQUIRE(ol != NULL);
168 ATF_REQUIRE(setlocale(LC_CTYPE, "C") != NULL);
169 cr = strvisx(cd, s, sizeof(s), VIS_WHITE);
170 ATF_REQUIRE(jr == cr);
171 ATF_REQUIRE(memcmp(jd, cd, jr) == 0);
172 setlocale(LC_CTYPE, ol);
173 free(ol);
174 }
175
ATF_TP_ADD_TCS(tp)176 ATF_TP_ADD_TCS(tp)
177 {
178
179 ATF_TP_ADD_TC(tp, strvis_basic);
180 ATF_TP_ADD_TC(tp, strvis_null);
181 ATF_TP_ADD_TC(tp, strvis_empty);
182 ATF_TP_ADD_TC(tp, strunvis_hex);
183 ATF_TP_ADD_TC(tp, strvis_locale);
184
185 return atf_no_error();
186 }
187