1 /* $NetBSD: t_memchr.c,v 1.3 2012/04/06 07:53:10 jruoho Exp $ */
2 
3 /*
4  * Written by J.T. Conklin <jtc@acorntoolworks.com>
5  * Public domain.
6  */
7 
8 #include <atf-c.h>
9 #include <string.h>
10 #include <unistd.h>
11 #include <stdio.h>
12 #include <stdlib.h>
13 
14 ATF_TC(memchr_basic);
15 ATF_TC_HEAD(memchr_basic, tc)
16 {
17         atf_tc_set_md_var(tc, "descr", "Test memchr(3) results, #1");
18 }
19 
20 ATF_TC_BODY(memchr_basic, tc)
21 {
22 	/* try to trick the compiler */
23 	void * (*f)(const void *, int, size_t) = memchr;
24 
25 	unsigned int a, t;
26 	void *off, *off2;
27 	char buf[32];
28 
29 	struct tab {
30 		const char	*val;
31 		size_t  	 len;
32 		char		 match;
33 		ssize_t		 off;
34 	};
35 
36 	const struct tab tab[] = {
37 		{ "",			0, 0, 0 },
38 
39 		{ "/",			0, 0, 0 },
40 		{ "/",			1, 1, 0 },
41 		{ "/a",			2, 1, 0 },
42 		{ "/ab",		3, 1, 0 },
43 		{ "/abc",		4, 1, 0 },
44 		{ "/abcd",		5, 1, 0 },
45 		{ "/abcde",		6, 1, 0 },
46 		{ "/abcdef",		7, 1, 0 },
47 		{ "/abcdefg",		8, 1, 0 },
48 
49 		{ "a/",			1, 0, 0 },
50 		{ "a/",			2, 1, 1 },
51 		{ "a/b",		3, 1, 1 },
52 		{ "a/bc",		4, 1, 1 },
53 		{ "a/bcd",		5, 1, 1 },
54 		{ "a/bcde",		6, 1, 1 },
55 		{ "a/bcdef",		7, 1, 1 },
56 		{ "a/bcdefg",		8, 1, 1 },
57 
58 		{ "ab/",		2, 0, 0 },
59 		{ "ab/",		3, 1, 2 },
60 		{ "ab/c",		4, 1, 2 },
61 		{ "ab/cd",		5, 1, 2 },
62 		{ "ab/cde",		6, 1, 2 },
63 		{ "ab/cdef",		7, 1, 2 },
64 		{ "ab/cdefg",		8, 1, 2 },
65 
66 		{ "abc/",		3, 0, 0 },
67 		{ "abc/",		4, 1, 3 },
68 		{ "abc/d",		5, 1, 3 },
69 		{ "abc/de",		6, 1, 3 },
70 		{ "abc/def",		7, 1, 3 },
71 		{ "abc/defg",		8, 1, 3 },
72 
73 		{ "abcd/",		4, 0, 0 },
74 		{ "abcd/",		5, 1, 4 },
75 		{ "abcd/e",		6, 1, 4 },
76 		{ "abcd/ef",		7, 1, 4 },
77 		{ "abcd/efg",		8, 1, 4 },
78 
79 		{ "abcde/",		5, 0, 0 },
80 		{ "abcde/",		6, 1, 5 },
81 		{ "abcde/f",		7, 1, 5 },
82 		{ "abcde/fg",		8, 1, 5 },
83 
84 		{ "abcdef/",		6, 0, 0 },
85 		{ "abcdef/",		7, 1, 6 },
86 		{ "abcdef/g",		8, 1, 6 },
87 
88 		{ "abcdefg/",		7, 0, 0 },
89 		{ "abcdefg/",		8, 1, 7 },
90 
91 		{ "\xff\xff\xff\xff" "efg/",	8, 1, 7 },
92 		{ "a" "\xff\xff\xff\xff" "fg/",	8, 1, 7 },
93 		{ "ab" "\xff\xff\xff\xff" "g/",	8, 1, 7 },
94 		{ "abc" "\xff\xff\xff\xff" "/",	8, 1, 7 },
95 	};
96 
97 	for (a = 1; a < 1 + sizeof(long); ++a) {
98 		for (t = 0; t < (sizeof(tab) / sizeof(tab[0])); ++t) {
99 			buf[a-1] = '/';
100 			strcpy(&buf[a], tab[t].val);
101 
102 			off = f(&buf[a], '/', tab[t].len);
103 			if (tab[t].match == 0) {
104 				if (off != 0) {
105 					fprintf(stderr, "a = %d, t = %d\n",
106 					    a, t);
107 					atf_tc_fail("should not have found "
108 					    " char past len");
109 				}
110 			} else if (tab[t].match == 1) {
111 				if (tab[t].off != ((char*)off - &buf[a])) {
112 					fprintf(stderr, "a = %d, t = %d\n",
113 					    a, t);
114 					atf_tc_fail("char not found at "
115 					    "correct offset");
116 				}
117 	    		} else {
118 				fprintf(stderr, "a = %d, t = %d\n", a, t);
119 				atf_tc_fail("Corrupt test case data");
120 			}
121 
122 			/* check zero extension of char arg */
123 			off2 = f(&buf[a], 0xffffff00 | '/', tab[t].len);
124 			if (off2 != off)
125 				atf_tc_fail("zero extension of char arg "
126 				    "failed");
127 		}
128 	}
129 }
130 
131 ATF_TC(memchr_simple);
132 ATF_TC_HEAD(memchr_simple, tc)
133 {
134         atf_tc_set_md_var(tc, "descr", "Test memchr(3) results, #2");
135 }
136 
137 ATF_TC_BODY(memchr_simple, tc)
138 {
139 	char buf[] = "abcdefg";
140 	short i = 7;
141 
142 	ATF_CHECK(memchr(buf, 'a', 0) == NULL);
143 	ATF_CHECK(memchr(buf, 'g', 0) == NULL);
144 	ATF_CHECK(memchr(buf, 'x', 7) == NULL);
145 
146 	ATF_CHECK(memchr("\0", 'x', 0) == NULL);
147 	ATF_CHECK(memchr("\0", 'x', 1) == NULL);
148 
149 	while (i <= 14) {
150 
151 		ATF_CHECK(memchr(buf, 'a', i) == buf + 0);
152 		ATF_CHECK(memchr(buf, 'b', i) == buf + 1);
153 		ATF_CHECK(memchr(buf, 'c', i) == buf + 2);
154 		ATF_CHECK(memchr(buf, 'd', i) == buf + 3);
155 		ATF_CHECK(memchr(buf, 'e', i) == buf + 4);
156 		ATF_CHECK(memchr(buf, 'f', i) == buf + 5);
157 		ATF_CHECK(memchr(buf, 'g', i) == buf + 6);
158 
159 		i *= 2;
160 	}
161 }
162 
163 ATF_TC(memrchr_simple);
164 ATF_TC_HEAD(memrchr_simple, tc)
165 {
166         atf_tc_set_md_var(tc, "descr", "Test memrchr(3) results");
167 }
168 
169 ATF_TC_BODY(memrchr_simple, tc)
170 {
171 	char buf[] = "abcdabcd";
172 
173 	ATF_CHECK(memrchr(buf, 'a', 0) == NULL);
174 	ATF_CHECK(memrchr(buf, 'g', 0) == NULL);
175 	ATF_CHECK(memrchr(buf, 'x', 8) == NULL);
176 
177 	ATF_CHECK(memrchr("\0", 'x', 0) == NULL);
178 	ATF_CHECK(memrchr("\0", 'x', 1) == NULL);
179 
180 	ATF_CHECK(memrchr(buf, 'a', 8) == buf + 4);
181 	ATF_CHECK(memrchr(buf, 'b', 8) == buf + 5);
182 	ATF_CHECK(memrchr(buf, 'c', 8) == buf + 6);
183 	ATF_CHECK(memrchr(buf, 'd', 8) == buf + 7);
184 }
185 
186 ATF_TP_ADD_TCS(tp)
187 {
188 
189 	ATF_TP_ADD_TC(tp, memchr_basic);
190 	ATF_TP_ADD_TC(tp, memchr_simple);
191 	ATF_TP_ADD_TC(tp, memrchr_simple);
192 
193 	return atf_no_error();
194 }
195