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(void) {
 194         void *mem;
 195         size_t len;
 196
 197         assert_se(unbase32hexmem("", STRLEN(""), true, &mem, &len) == 0);
 198         assert_se(streq(strndupa(mem, len), ""));
 199         free(mem);
 200
 201         assert_se(unbase32hexmem("CO======", STRLEN("CO======"), true, &mem, &len) == 0);
 202         assert_se(streq(strndupa(mem, len), "f"));
 203         free(mem);
 204
 205         assert_se(unbase32hexmem("CPNG====", STRLEN("CPNG===="), true, &mem, &len) == 0);
 206         assert_se(streq(strndupa(mem, len), "fo"));
 207         free(mem);
 208
 209         assert_se(unbase32hexmem("CPNMU===", STRLEN("CPNMU==="), true, &mem, &len) == 0);
 210         assert_se(streq(strndupa(mem, len), "foo"));
 211         free(mem);
 212
 213         assert_se(unbase32hexmem("CPNMUOG=", STRLEN("CPNMUOG="), true, &mem, &len) == 0);
 214         assert_se(streq(strndupa(mem, len), "foob"));
 215         free(mem);
 216
 217         assert_se(unbase32hexmem("CPNMUOJ1", STRLEN("CPNMUOJ1"), true, &mem, &len) == 0);
 218         assert_se(streq(strndupa(mem, len), "fooba"));
 219         free(mem);
 220
 221         assert_se(unbase32hexmem("CPNMUOJ1E8======", STRLEN("CPNMUOJ1E8======"), true, &mem, &len) == 0);
 222         assert_se(streq(strndupa(mem, len), "foobar"));
 223         free(mem);
 224
 225         assert_se(unbase32hexmem("A", STRLEN("A"), true, &mem, &len) == -EINVAL);
 226         assert_se(unbase32hexmem("A=======", STRLEN("A======="), true, &mem, &len) == -EINVAL);
 227         assert_se(unbase32hexmem("AAA=====", STRLEN("AAA====="), true, &mem, &len) == -EINVAL);
 228         assert_se(unbase32hexmem("AAAAAA==", STRLEN("AAAAAA=="), true, &mem, &len) == -EINVAL);
 229         assert_se(unbase32hexmem("AB======", STRLEN("AB======"), true, &mem, &len) == -EINVAL);
 230         assert_se(unbase32hexmem("AAAB====", STRLEN("AAAB===="), true, &mem, &len) == -EINVAL);
 231         assert_se(unbase32hexmem("AAAAB===", STRLEN("AAAAB==="), true, &mem, &len) == -EINVAL);
 232         assert_se(unbase32hexmem("AAAAAAB=", STRLEN("AAAAAAB="), true, &mem, &len) == -EINVAL);
 233
 234         assert_se(unbase32hexmem("XPNMUOJ1", STRLEN("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 235         assert_se(unbase32hexmem("CXNMUOJ1", STRLEN("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 236         assert_se(unbase32hexmem("CPXMUOJ1", STRLEN("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 237         assert_se(unbase32hexmem("CPNXUOJ1", STRLEN("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 238         assert_se(unbase32hexmem("CPNMXOJ1", STRLEN("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 239         assert_se(unbase32hexmem("CPNMUXJ1", STRLEN("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 240         assert_se(unbase32hexmem("CPNMUOX1", STRLEN("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 241         assert_se(unbase32hexmem("CPNMUOJX", STRLEN("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 242
 243         assert_se(unbase32hexmem("", STRLEN(""), false, &mem, &len) == 0);
 244         assert_se(streq(strndupa(mem, len), ""));
 245         free(mem);
 246
 247         assert_se(unbase32hexmem("CO", STRLEN("CO"), false, &mem, &len) == 0);
 248         assert_se(streq(strndupa(mem, len), "f"));
 249         free(mem);
 250
 251         assert_se(unbase32hexmem("CPNG", STRLEN("CPNG"), false, &mem, &len) == 0);
 252         assert_se(streq(strndupa(mem, len), "fo"));
 253         free(mem);
 254
 255         assert_se(unbase32hexmem("CPNMU", STRLEN("CPNMU"), false, &mem, &len) == 0);
 256         assert_se(streq(strndupa(mem, len), "foo"));
 257         free(mem);
 258
 259         assert_se(unbase32hexmem("CPNMUOG", STRLEN("CPNMUOG"), false, &mem, &len) == 0);
 260         assert_se(streq(strndupa(mem, len), "foob"));
 261         free(mem);
 262
 263         assert_se(unbase32hexmem("CPNMUOJ1", STRLEN("CPNMUOJ1"), false, &mem, &len) == 0);
 264         assert_se(streq(strndupa(mem, len), "fooba"));
 265         free(mem);
 266
 267         assert_se(unbase32hexmem("CPNMUOJ1E8", STRLEN("CPNMUOJ1E8"), false, &mem, &len) == 0);
 268         assert_se(streq(strndupa(mem, len), "foobar"));
 269         free(mem);
 270
 271         assert_se(unbase32hexmem("CPNMUOG=", STRLEN("CPNMUOG="), false, &mem, &len) == -EINVAL);
 272         assert_se(unbase32hexmem("CPNMUOJ1E8======", STRLEN("CPNMUOJ1E8======"), false, &mem, &len) == -EINVAL);
 273         assert_se(unbase32hexmem("A", STRLEN("A"), false, &mem, &len) == -EINVAL);
 274         assert_se(unbase32hexmem("A", STRLEN("A"), false, &mem, &len) == -EINVAL);
 275         assert_se(unbase32hexmem("AAA", STRLEN("AAA"), false, &mem, &len) == -EINVAL);
 276         assert_se(unbase32hexmem("AAAAAA", STRLEN("AAAAAA"), false, &mem, &len) == -EINVAL);
 277         assert_se(unbase32hexmem("AB", STRLEN("AB"), false, &mem, &len) == -EINVAL);
 278         assert_se(unbase32hexmem("AAAB", STRLEN("AAAB"), false, &mem, &len) == -EINVAL);
 279         assert_se(unbase32hexmem("AAAAB", STRLEN("AAAAB"), false, &mem, &len) == -EINVAL);
 280         assert_se(unbase32hexmem("AAAAAAB", STRLEN("AAAAAAB"), false, &mem, &len) == -EINVAL);
 281 }
 282
 283 /* https://tools.ietf.org/html/rfc4648#section-10 */
 284 static void test_base64mem(void) {
 285         char *b64;
 286
 287         assert_se(base64mem("", STRLEN(""), &b64) == 0);
 288         assert_se(streq(b64, ""));
 289         free(b64);
 290
 291         assert_se(base64mem("f", STRLEN("f"), &b64) == 4);
 292         assert_se(streq(b64, "Zg=="));
 293         free(b64);
 294
 295         assert_se(base64mem("fo", STRLEN("fo"), &b64) == 4);
 296         assert_se(streq(b64, "Zm8="));
 297         free(b64);
 298
 299         assert_se(base64mem("foo", STRLEN("foo"), &b64) == 4);
 300         assert_se(streq(b64, "Zm9v"));
 301         free(b64);
 302
 303         assert_se(base64mem("foob", STRLEN("foob"), &b64) == 8);
 304         assert_se(streq(b64, "Zm9vYg=="));
 305         free(b64);
 306
 307         assert_se(base64mem("fooba", STRLEN("fooba"), &b64) == 8);
 308         assert_se(streq(b64, "Zm9vYmE="));
 309         free(b64);
 310
 311         assert_se(base64mem("foobar", STRLEN("foobar"), &b64) == 8);
 312         assert_se(streq(b64, "Zm9vYmFy"));
 313         free(b64);
 314 }
 315
 316 static void test_unbase64mem_one(const char *input, const char *output, int ret) {
 317         _cleanup_free_ void *buffer = NULL;
 318         size_t size = 0;
 319
 320         assert_se(unbase64mem(input, (size_t) -1, &buffer, &size) == ret);
 321
 322         if (ret >= 0) {
 323                 assert_se(size == strlen(output));
 324                 assert_se(memcmp(buffer, output, size) == 0);
 325                 assert_se(((char*) buffer)[size] == 0);
 326         }
 327 }
 328
 329 static void test_unbase64mem(void) {
 330
 331         test_unbase64mem_one("", "", 0);
 332         test_unbase64mem_one("Zg==", "f", 0);
 333         test_unbase64mem_one("Zm8=", "fo", 0);
 334         test_unbase64mem_one("Zm9v", "foo", 0);
 335         test_unbase64mem_one("Zm9vYg==", "foob", 0);
 336         test_unbase64mem_one("Zm9vYmE=", "fooba", 0);
 337         test_unbase64mem_one("Zm9vYmFy", "foobar", 0);
 338
 339         test_unbase64mem_one(" ", "", 0);
 340         test_unbase64mem_one(" \n\r ", "", 0);
 341         test_unbase64mem_one("    Zg\n==       ", "f", 0);
 342         test_unbase64mem_one(" Zm 8=\r", "fo", 0);
 343         test_unbase64mem_one("  Zm9\n\r\r\nv   ", "foo", 0);
 344         test_unbase64mem_one(" Z m9vYg==\n\r", "foob", 0);
 345         test_unbase64mem_one(" Zm 9vYmE=   ", "fooba", 0);
 346         test_unbase64mem_one("   Z m9v    YmFy   ", "foobar", 0);
 347
 348         test_unbase64mem_one("A", NULL, -EPIPE);
 349         test_unbase64mem_one("A====", NULL, -EINVAL);
 350         test_unbase64mem_one("AAB==", NULL, -EINVAL);
 351         test_unbase64mem_one(" A A A B = ", NULL, -EINVAL);
 352         test_unbase64mem_one(" Z m 8 = q u u x ", NULL, -ENAMETOOLONG);
 353 }
 354
 355 static void test_hexdump(void) {
 356         uint8_t data[146];
 357         unsigned i;
 358
 359         hexdump(stdout, NULL, 0);
 360         hexdump(stdout, "", 0);
 361         hexdump(stdout, "", 1);
 362         hexdump(stdout, "x", 1);
 363         hexdump(stdout, "x", 2);
 364         hexdump(stdout, "foobar", 7);
 365         hexdump(stdout, "f\nobar", 7);
 366         hexdump(stdout, "xxxxxxxxxxxxxxxxxxxxyz", 23);
 367
 368         for (i = 0; i < ELEMENTSOF(data); i++)
 369                 data[i] = i*2;
 370
 371         hexdump(stdout, data, sizeof(data));
 372 }
 373
 374 int main(int argc, char *argv[]) {
 375         test_hexchar();
 376         test_unhexchar();
 377         test_base32hexchar();
 378         test_unbase32hexchar();
 379         test_base64char();
 380         test_unbase64char();
 381         test_octchar();
 382         test_unoctchar();
 383         test_decchar();
 384         test_undecchar();
 385         test_unhexmem();
 386         test_base32hexmem();
 387         test_unbase32hexmem();
 388         test_base64mem();
 389         test_unbase64mem();
 390         test_hexdump();
 391
 392         return 0;
 393 }