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