Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : :
3 : : #include <errno.h>
4 : :
5 : : #include "alloc-util.h"
6 : : #include "hexdecoct.h"
7 : : #include "macro.h"
8 : : #include "string-util.h"
9 : :
10 : 4 : static void test_hexchar(void) {
11 [ - + ]: 4 : assert_se(hexchar(0xa) == 'a');
12 [ - + ]: 4 : assert_se(hexchar(0x0) == '0');
13 : 4 : }
14 : :
15 : 4 : static void test_unhexchar(void) {
16 [ - + ]: 4 : assert_se(unhexchar('a') == 0xA);
17 [ - + ]: 4 : assert_se(unhexchar('A') == 0xA);
18 [ - + ]: 4 : assert_se(unhexchar('0') == 0x0);
19 : 4 : }
20 : :
21 : 4 : static void test_base32hexchar(void) {
22 [ - + ]: 4 : assert_se(base32hexchar(0) == '0');
23 [ - + ]: 4 : assert_se(base32hexchar(9) == '9');
24 [ - + ]: 4 : assert_se(base32hexchar(10) == 'A');
25 [ - + ]: 4 : assert_se(base32hexchar(31) == 'V');
26 : 4 : }
27 : :
28 : 4 : static void test_unbase32hexchar(void) {
29 [ - + ]: 4 : assert_se(unbase32hexchar('0') == 0);
30 [ - + ]: 4 : assert_se(unbase32hexchar('9') == 9);
31 [ - + ]: 4 : assert_se(unbase32hexchar('A') == 10);
32 [ - + ]: 4 : assert_se(unbase32hexchar('V') == 31);
33 [ - + ]: 4 : assert_se(unbase32hexchar('=') == -EINVAL);
34 : 4 : }
35 : :
36 : 4 : static void test_base64char(void) {
37 [ - + ]: 4 : assert_se(base64char(0) == 'A');
38 [ - + ]: 4 : assert_se(base64char(26) == 'a');
39 [ - + ]: 4 : assert_se(base64char(63) == '/');
40 : 4 : }
41 : :
42 : 4 : static void test_unbase64char(void) {
43 [ - + ]: 4 : assert_se(unbase64char('A') == 0);
44 [ - + ]: 4 : assert_se(unbase64char('Z') == 25);
45 [ - + ]: 4 : assert_se(unbase64char('a') == 26);
46 [ - + ]: 4 : assert_se(unbase64char('z') == 51);
47 [ - + ]: 4 : assert_se(unbase64char('0') == 52);
48 [ - + ]: 4 : assert_se(unbase64char('9') == 61);
49 [ - + ]: 4 : assert_se(unbase64char('+') == 62);
50 [ - + ]: 4 : assert_se(unbase64char('/') == 63);
51 [ - + ]: 4 : assert_se(unbase64char('=') == -EINVAL);
52 : 4 : }
53 : :
54 : 4 : static void test_octchar(void) {
55 [ - + ]: 4 : assert_se(octchar(00) == '0');
56 [ - + ]: 4 : assert_se(octchar(07) == '7');
57 : 4 : }
58 : :
59 : 4 : static void test_unoctchar(void) {
60 [ - + ]: 4 : assert_se(unoctchar('0') == 00);
61 [ - + ]: 4 : assert_se(unoctchar('7') == 07);
62 : 4 : }
63 : :
64 : 4 : static void test_decchar(void) {
65 [ - + ]: 4 : assert_se(decchar(0) == '0');
66 [ - + ]: 4 : assert_se(decchar(9) == '9');
67 : 4 : }
68 : :
69 : 4 : static void test_undecchar(void) {
70 [ - + ]: 4 : assert_se(undecchar('0') == 0);
71 [ - + ]: 4 : assert_se(undecchar('9') == 9);
72 : 4 : }
73 : :
74 : 44 : static void test_unhexmem_one(const char *s, size_t l, int retval) {
75 : 44 : _cleanup_free_ char *hex = NULL;
76 : 44 : _cleanup_free_ void *mem = NULL;
77 : : size_t len;
78 : :
79 [ - + ]: 44 : assert_se(unhexmem(s, l, &mem, &len) == retval);
80 [ + + ]: 44 : if (retval == 0) {
81 : : char *answer;
82 : :
83 [ + + ]: 32 : if (l == (size_t) -1)
84 : 16 : l = strlen(s);
85 : :
86 [ - + ]: 32 : assert_se(hex = hexmem(mem, len));
87 : 32 : answer = strndupa(strempty(s), l);
88 [ - + ]: 32 : assert_se(streq(delete_chars(answer, WHITESPACE), hex));
89 : : }
90 : 44 : }
91 : :
92 : 4 : static void test_unhexmem(void) {
93 : 4 : const char *hex = "efa2149213";
94 : 4 : const char *hex_space = " e f a\n 2\r 14\n\r\t9\t2 \n1\r3 \r\r\t";
95 : 4 : const char *hex_invalid = "efa214921o";
96 : :
97 : 4 : test_unhexmem_one(NULL, 0, 0);
98 : 4 : test_unhexmem_one("", 0, 0);
99 : 4 : test_unhexmem_one("", (size_t) -1, 0);
100 : 4 : test_unhexmem_one(" \n \t\r \t\t \n\n\n", (size_t) -1, 0);
101 : 4 : test_unhexmem_one(hex_invalid, strlen(hex_invalid), -EINVAL);
102 : 4 : test_unhexmem_one(hex_invalid, (size_t) - 1, -EINVAL);
103 : 4 : test_unhexmem_one(hex, strlen(hex) - 1, -EPIPE);
104 : 4 : test_unhexmem_one(hex, strlen(hex), 0);
105 : 4 : test_unhexmem_one(hex, (size_t) -1, 0);
106 : 4 : test_unhexmem_one(hex_space, strlen(hex_space), 0);
107 : 4 : test_unhexmem_one(hex_space, (size_t) -1, 0);
108 : 4 : }
109 : :
110 : : /* https://tools.ietf.org/html/rfc4648#section-10 */
111 : 4 : static void test_base32hexmem(void) {
112 : : char *b32;
113 : :
114 : 4 : b32 = base32hexmem("", STRLEN(""), true);
115 [ - + ]: 4 : assert_se(b32);
116 [ - + ]: 4 : assert_se(streq(b32, ""));
117 : 4 : free(b32);
118 : :
119 : 4 : b32 = base32hexmem("f", STRLEN("f"), true);
120 [ - + ]: 4 : assert_se(b32);
121 [ - + ]: 4 : assert_se(streq(b32, "CO======"));
122 : 4 : free(b32);
123 : :
124 : 4 : b32 = base32hexmem("fo", STRLEN("fo"), true);
125 [ - + ]: 4 : assert_se(b32);
126 [ - + ]: 4 : assert_se(streq(b32, "CPNG===="));
127 : 4 : free(b32);
128 : :
129 : 4 : b32 = base32hexmem("foo", STRLEN("foo"), true);
130 [ - + ]: 4 : assert_se(b32);
131 [ - + ]: 4 : assert_se(streq(b32, "CPNMU==="));
132 : 4 : free(b32);
133 : :
134 : 4 : b32 = base32hexmem("foob", STRLEN("foob"), true);
135 [ - + ]: 4 : assert_se(b32);
136 [ - + ]: 4 : assert_se(streq(b32, "CPNMUOG="));
137 : 4 : free(b32);
138 : :
139 : 4 : b32 = base32hexmem("fooba", STRLEN("fooba"), true);
140 [ - + ]: 4 : assert_se(b32);
141 [ - + ]: 4 : assert_se(streq(b32, "CPNMUOJ1"));
142 : 4 : free(b32);
143 : :
144 : 4 : b32 = base32hexmem("foobar", STRLEN("foobar"), true);
145 [ - + ]: 4 : assert_se(b32);
146 [ - + ]: 4 : assert_se(streq(b32, "CPNMUOJ1E8======"));
147 : 4 : free(b32);
148 : :
149 : 4 : b32 = base32hexmem("", STRLEN(""), false);
150 [ - + ]: 4 : assert_se(b32);
151 [ - + ]: 4 : assert_se(streq(b32, ""));
152 : 4 : free(b32);
153 : :
154 : 4 : b32 = base32hexmem("f", STRLEN("f"), false);
155 [ - + ]: 4 : assert_se(b32);
156 [ - + ]: 4 : assert_se(streq(b32, "CO"));
157 : 4 : free(b32);
158 : :
159 : 4 : b32 = base32hexmem("fo", STRLEN("fo"), false);
160 [ - + ]: 4 : assert_se(b32);
161 [ - + ]: 4 : assert_se(streq(b32, "CPNG"));
162 : 4 : free(b32);
163 : :
164 : 4 : b32 = base32hexmem("foo", STRLEN("foo"), false);
165 [ - + ]: 4 : assert_se(b32);
166 [ - + ]: 4 : assert_se(streq(b32, "CPNMU"));
167 : 4 : free(b32);
168 : :
169 : 4 : b32 = base32hexmem("foob", STRLEN("foob"), false);
170 [ - + ]: 4 : assert_se(b32);
171 [ - + ]: 4 : assert_se(streq(b32, "CPNMUOG"));
172 : 4 : free(b32);
173 : :
174 : 4 : b32 = base32hexmem("fooba", STRLEN("fooba"), false);
175 [ - + ]: 4 : assert_se(b32);
176 [ - + ]: 4 : assert_se(streq(b32, "CPNMUOJ1"));
177 : 4 : free(b32);
178 : :
179 : 4 : b32 = base32hexmem("foobar", STRLEN("foobar"), false);
180 [ - + ]: 4 : assert_se(b32);
181 [ - + ]: 4 : assert_se(streq(b32, "CPNMUOJ1E8"));
182 : 4 : free(b32);
183 : 4 : }
184 : :
185 : 156 : static void test_unbase32hexmem_one(const char *hex, bool padding, int retval, const char *ans) {
186 : 156 : _cleanup_free_ void *mem = NULL;
187 : : size_t len;
188 : :
189 [ - + ]: 156 : assert_se(unbase32hexmem(hex, (size_t) -1, padding, &mem, &len) == retval);
190 [ + + ]: 156 : if (retval == 0) {
191 : : char *str;
192 : :
193 : 56 : str = strndupa(mem, len);
194 [ - + ]: 56 : assert_se(streq(str, ans));
195 : : }
196 : 156 : }
197 : :
198 : 4 : static void test_unbase32hexmem(void) {
199 : 4 : test_unbase32hexmem_one("", true, 0, "");
200 : :
201 : 4 : test_unbase32hexmem_one("CO======", true, 0, "f");
202 : 4 : test_unbase32hexmem_one("CPNG====", true, 0, "fo");
203 : 4 : test_unbase32hexmem_one("CPNMU===", true, 0, "foo");
204 : 4 : test_unbase32hexmem_one("CPNMUOG=", true, 0, "foob");
205 : 4 : test_unbase32hexmem_one("CPNMUOJ1", true, 0, "fooba");
206 : 4 : test_unbase32hexmem_one("CPNMUOJ1E8======", true, 0, "foobar");
207 : :
208 : 4 : test_unbase32hexmem_one("A", true, -EINVAL, NULL);
209 : 4 : test_unbase32hexmem_one("A=======", true, -EINVAL, NULL);
210 : 4 : test_unbase32hexmem_one("AAA=====", true, -EINVAL, NULL);
211 : 4 : test_unbase32hexmem_one("AAAAAA==", true, -EINVAL, NULL);
212 : 4 : test_unbase32hexmem_one("AB======", true, -EINVAL, NULL);
213 : 4 : test_unbase32hexmem_one("AAAB====", true, -EINVAL, NULL);
214 : 4 : test_unbase32hexmem_one("AAAAB===", true, -EINVAL, NULL);
215 : 4 : test_unbase32hexmem_one("AAAAAAB=", true, -EINVAL, NULL);
216 : :
217 : 4 : test_unbase32hexmem_one("XPNMUOJ1", true, -EINVAL, NULL);
218 : 4 : test_unbase32hexmem_one("CXNMUOJ1", true, -EINVAL, NULL);
219 : 4 : test_unbase32hexmem_one("CPXMUOJ1", true, -EINVAL, NULL);
220 : 4 : test_unbase32hexmem_one("CPNXUOJ1", true, -EINVAL, NULL);
221 : 4 : test_unbase32hexmem_one("CPNMXOJ1", true, -EINVAL, NULL);
222 : 4 : test_unbase32hexmem_one("CPNMUXJ1", true, -EINVAL, NULL);
223 : 4 : test_unbase32hexmem_one("CPNMUOX1", true, -EINVAL, NULL);
224 : 4 : test_unbase32hexmem_one("CPNMUOJX", true, -EINVAL, NULL);
225 : :
226 : 4 : test_unbase32hexmem_one("", false, 0, "");
227 : 4 : test_unbase32hexmem_one("CO", false, 0, "f");
228 : 4 : test_unbase32hexmem_one("CPNG", false, 0, "fo");
229 : 4 : test_unbase32hexmem_one("CPNMU", false, 0, "foo");
230 : 4 : test_unbase32hexmem_one("CPNMUOG", false, 0, "foob");
231 : 4 : test_unbase32hexmem_one("CPNMUOJ1", false, 0, "fooba");
232 : 4 : test_unbase32hexmem_one("CPNMUOJ1E8", false, 0, "foobar");
233 : 4 : test_unbase32hexmem_one("CPNMUOG=", false, -EINVAL, NULL);
234 : 4 : test_unbase32hexmem_one("CPNMUOJ1E8======", false, -EINVAL, NULL);
235 : :
236 : 4 : test_unbase32hexmem_one("A", false, -EINVAL, NULL);
237 : 4 : test_unbase32hexmem_one("AAA", false, -EINVAL, NULL);
238 : 4 : test_unbase32hexmem_one("AAAAAA", false, -EINVAL, NULL);
239 : 4 : test_unbase32hexmem_one("AB", false, -EINVAL, NULL);
240 : 4 : test_unbase32hexmem_one("AAAB", false, -EINVAL, NULL);
241 : 4 : test_unbase32hexmem_one("AAAAB", false, -EINVAL, NULL);
242 : 4 : test_unbase32hexmem_one("AAAAAAB", false, -EINVAL, NULL);
243 : 4 : }
244 : :
245 : : /* https://tools.ietf.org/html/rfc4648#section-10 */
246 : 4 : static void test_base64mem(void) {
247 : : char *b64;
248 : :
249 [ - + ]: 4 : assert_se(base64mem("", STRLEN(""), &b64) == 0);
250 [ - + ]: 4 : assert_se(streq(b64, ""));
251 : 4 : free(b64);
252 : :
253 [ - + ]: 4 : assert_se(base64mem("f", STRLEN("f"), &b64) == 4);
254 [ - + ]: 4 : assert_se(streq(b64, "Zg=="));
255 : 4 : free(b64);
256 : :
257 [ - + ]: 4 : assert_se(base64mem("fo", STRLEN("fo"), &b64) == 4);
258 [ - + ]: 4 : assert_se(streq(b64, "Zm8="));
259 : 4 : free(b64);
260 : :
261 [ - + ]: 4 : assert_se(base64mem("foo", STRLEN("foo"), &b64) == 4);
262 [ - + ]: 4 : assert_se(streq(b64, "Zm9v"));
263 : 4 : free(b64);
264 : :
265 [ - + ]: 4 : assert_se(base64mem("foob", STRLEN("foob"), &b64) == 8);
266 [ - + ]: 4 : assert_se(streq(b64, "Zm9vYg=="));
267 : 4 : free(b64);
268 : :
269 [ - + ]: 4 : assert_se(base64mem("fooba", STRLEN("fooba"), &b64) == 8);
270 [ - + ]: 4 : assert_se(streq(b64, "Zm9vYmE="));
271 : 4 : free(b64);
272 : :
273 [ - + ]: 4 : assert_se(base64mem("foobar", STRLEN("foobar"), &b64) == 8);
274 [ - + ]: 4 : assert_se(streq(b64, "Zm9vYmFy"));
275 : 4 : free(b64);
276 : 4 : }
277 : :
278 : 80 : static void test_unbase64mem_one(const char *input, const char *output, int ret) {
279 : 80 : _cleanup_free_ void *buffer = NULL;
280 : 80 : size_t size = 0;
281 : :
282 [ - + ]: 80 : assert_se(unbase64mem(input, (size_t) -1, &buffer, &size) == ret);
283 : :
284 [ + + ]: 80 : if (ret >= 0) {
285 [ - + ]: 60 : assert_se(size == strlen(output));
286 [ - + ]: 60 : assert_se(memcmp(buffer, output, size) == 0);
287 [ - + ]: 60 : assert_se(((char*) buffer)[size] == 0);
288 : : }
289 : 80 : }
290 : :
291 : 4 : static void test_unbase64mem(void) {
292 : :
293 : 4 : test_unbase64mem_one("", "", 0);
294 : 4 : test_unbase64mem_one("Zg==", "f", 0);
295 : 4 : test_unbase64mem_one("Zm8=", "fo", 0);
296 : 4 : test_unbase64mem_one("Zm9v", "foo", 0);
297 : 4 : test_unbase64mem_one("Zm9vYg==", "foob", 0);
298 : 4 : test_unbase64mem_one("Zm9vYmE=", "fooba", 0);
299 : 4 : test_unbase64mem_one("Zm9vYmFy", "foobar", 0);
300 : :
301 : 4 : test_unbase64mem_one(" ", "", 0);
302 : 4 : test_unbase64mem_one(" \n\r ", "", 0);
303 : 4 : test_unbase64mem_one(" Zg\n== ", "f", 0);
304 : 4 : test_unbase64mem_one(" Zm 8=\r", "fo", 0);
305 : 4 : test_unbase64mem_one(" Zm9\n\r\r\nv ", "foo", 0);
306 : 4 : test_unbase64mem_one(" Z m9vYg==\n\r", "foob", 0);
307 : 4 : test_unbase64mem_one(" Zm 9vYmE= ", "fooba", 0);
308 : 4 : test_unbase64mem_one(" Z m9v YmFy ", "foobar", 0);
309 : :
310 : 4 : test_unbase64mem_one("A", NULL, -EPIPE);
311 : 4 : test_unbase64mem_one("A====", NULL, -EINVAL);
312 : 4 : test_unbase64mem_one("AAB==", NULL, -EINVAL);
313 : 4 : test_unbase64mem_one(" A A A B = ", NULL, -EINVAL);
314 : 4 : test_unbase64mem_one(" Z m 8 = q u u x ", NULL, -ENAMETOOLONG);
315 : 4 : }
316 : :
317 : 4 : static void test_hexdump(void) {
318 : : uint8_t data[146];
319 : : unsigned i;
320 : :
321 : 4 : hexdump(stdout, NULL, 0);
322 : 4 : hexdump(stdout, "", 0);
323 : 4 : hexdump(stdout, "", 1);
324 : 4 : hexdump(stdout, "x", 1);
325 : 4 : hexdump(stdout, "x", 2);
326 : 4 : hexdump(stdout, "foobar", 7);
327 : 4 : hexdump(stdout, "f\nobar", 7);
328 : 4 : hexdump(stdout, "xxxxxxxxxxxxxxxxxxxxyz", 23);
329 : :
330 [ + + ]: 588 : for (i = 0; i < ELEMENTSOF(data); i++)
331 : 584 : data[i] = i*2;
332 : :
333 : 4 : hexdump(stdout, data, sizeof(data));
334 : 4 : }
335 : :
336 : 4 : int main(int argc, char *argv[]) {
337 : 4 : test_hexchar();
338 : 4 : test_unhexchar();
339 : 4 : test_base32hexchar();
340 : 4 : test_unbase32hexchar();
341 : 4 : test_base64char();
342 : 4 : test_unbase64char();
343 : 4 : test_octchar();
344 : 4 : test_unoctchar();
345 : 4 : test_decchar();
346 : 4 : test_undecchar();
347 : 4 : test_unhexmem();
348 : 4 : test_base32hexmem();
349 : 4 : test_unbase32hexmem();
350 : 4 : test_base64mem();
351 : 4 : test_unbase64mem();
352 : 4 : test_hexdump();
353 : :
354 : 4 : return 0;
355 : : }
|
