src/test/test-hexdecoct.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2010 Lennart Poettering
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include "alloc-util.h"
  21 #include "hexdecoct.h"
  22 #include "macro.h"
  23 #include "string-util.h"
  24
  25 static void test_hexchar(void) {
  26         assert_se(hexchar(0xa) == 'a');
  27         assert_se(hexchar(0x0) == '0');
  28 }
  29
  30 static void test_unhexchar(void) {
  31         assert_se(unhexchar('a') == 0xA);
  32         assert_se(unhexchar('A') == 0xA);
  33         assert_se(unhexchar('0') == 0x0);
  34 }
  35
  36 static void test_base32hexchar(void) {
  37         assert_se(base32hexchar(0) == '0');
  38         assert_se(base32hexchar(9) == '9');
  39         assert_se(base32hexchar(10) == 'A');
  40         assert_se(base32hexchar(31) == 'V');
  41 }
  42
  43 static void test_unbase32hexchar(void) {
  44         assert_se(unbase32hexchar('0') == 0);
  45         assert_se(unbase32hexchar('9') == 9);
  46         assert_se(unbase32hexchar('A') == 10);
  47         assert_se(unbase32hexchar('V') == 31);
  48         assert_se(unbase32hexchar('=') == -EINVAL);
  49 }
  50
  51 static void test_base64char(void) {
  52         assert_se(base64char(0) == 'A');
  53         assert_se(base64char(26) == 'a');
  54         assert_se(base64char(63) == '/');
  55 }
  56
  57 static void test_unbase64char(void) {
  58         assert_se(unbase64char('A') == 0);
  59         assert_se(unbase64char('Z') == 25);
  60         assert_se(unbase64char('a') == 26);
  61         assert_se(unbase64char('z') == 51);
  62         assert_se(unbase64char('0') == 52);
  63         assert_se(unbase64char('9') == 61);
  64         assert_se(unbase64char('+') == 62);
  65         assert_se(unbase64char('/') == 63);
  66         assert_se(unbase64char('=') == -EINVAL);
  67 }
  68
  69 static void test_octchar(void) {
  70         assert_se(octchar(00) == '0');
  71         assert_se(octchar(07) == '7');
  72 }
  73
  74 static void test_unoctchar(void) {
  75         assert_se(unoctchar('0') == 00);
  76         assert_se(unoctchar('7') == 07);
  77 }
  78
  79 static void test_decchar(void) {
  80         assert_se(decchar(0) == '0');
  81         assert_se(decchar(9) == '9');
  82 }
  83
  84 static void test_undecchar(void) {
  85         assert_se(undecchar('0') == 0);
  86         assert_se(undecchar('9') == 9);
  87 }
  88
  89 static void test_unhexmem(void) {
  90         const char *hex = "efa214921";
  91         const char *hex_invalid = "efa214921o";
  92         _cleanup_free_ char *hex2 = NULL;
  93         _cleanup_free_ void *mem = NULL;
  94         size_t len;
  95
  96         assert_se(unhexmem(hex, strlen(hex), &mem, &len) == 0);
  97         assert_se(unhexmem(hex, strlen(hex) + 1, &mem, &len) == -EINVAL);
  98         assert_se(unhexmem(hex_invalid, strlen(hex_invalid), &mem, &len) == -EINVAL);
  99
 100         assert_se((hex2 = hexmem(mem, len)));
 101
 102         free(mem);
 103
 104         assert_se(memcmp(hex, hex2, strlen(hex)) == 0);
 105
 106         free(hex2);
 107
 108         assert_se(unhexmem(hex, strlen(hex) - 1, &mem, &len) == 0);
 109         assert_se((hex2 = hexmem(mem, len)));
 110         assert_se(memcmp(hex, hex2, strlen(hex) - 1) == 0);
 111 }
 112
 113 /* https://tools.ietf.org/html/rfc4648#section-10 */
 114 static void test_base32hexmem(void) {
 115         char *b32;
 116
 117         b32 = base32hexmem("", strlen(""), true);
 118         assert_se(b32);
 119         assert_se(streq(b32, ""));
 120         free(b32);
 121
 122         b32 = base32hexmem("f", strlen("f"), true);
 123         assert_se(b32);
 124         assert_se(streq(b32, "CO======"));
 125         free(b32);
 126
 127         b32 = base32hexmem("fo", strlen("fo"), true);
 128         assert_se(b32);
 129         assert_se(streq(b32, "CPNG===="));
 130         free(b32);
 131
 132         b32 = base32hexmem("foo", strlen("foo"), true);
 133         assert_se(b32);
 134         assert_se(streq(b32, "CPNMU==="));
 135         free(b32);
 136
 137         b32 = base32hexmem("foob", strlen("foob"), true);
 138         assert_se(b32);
 139         assert_se(streq(b32, "CPNMUOG="));
 140         free(b32);
 141
 142         b32 = base32hexmem("fooba", strlen("fooba"), true);
 143         assert_se(b32);
 144         assert_se(streq(b32, "CPNMUOJ1"));
 145         free(b32);
 146
 147         b32 = base32hexmem("foobar", strlen("foobar"), true);
 148         assert_se(b32);
 149         assert_se(streq(b32, "CPNMUOJ1E8======"));
 150         free(b32);
 151
 152         b32 = base32hexmem("", strlen(""), false);
 153         assert_se(b32);
 154         assert_se(streq(b32, ""));
 155         free(b32);
 156
 157         b32 = base32hexmem("f", strlen("f"), false);
 158         assert_se(b32);
 159         assert_se(streq(b32, "CO"));
 160         free(b32);
 161
 162         b32 = base32hexmem("fo", strlen("fo"), false);
 163         assert_se(b32);
 164         assert_se(streq(b32, "CPNG"));
 165         free(b32);
 166
 167         b32 = base32hexmem("foo", strlen("foo"), false);
 168         assert_se(b32);
 169         assert_se(streq(b32, "CPNMU"));
 170         free(b32);
 171
 172         b32 = base32hexmem("foob", strlen("foob"), false);
 173         assert_se(b32);
 174         assert_se(streq(b32, "CPNMUOG"));
 175         free(b32);
 176
 177         b32 = base32hexmem("fooba", strlen("fooba"), false);
 178         assert_se(b32);
 179         assert_se(streq(b32, "CPNMUOJ1"));
 180         free(b32);
 181
 182         b32 = base32hexmem("foobar", strlen("foobar"), false);
 183         assert_se(b32);
 184         assert_se(streq(b32, "CPNMUOJ1E8"));
 185         free(b32);
 186 }
 187
 188 static void test_unbase32hexmem(void) {
 189         void *mem;
 190         size_t len;
 191
 192         assert_se(unbase32hexmem("", strlen(""), true, &mem, &len) == 0);
 193         assert_se(streq(strndupa(mem, len), ""));
 194         free(mem);
 195
 196         assert_se(unbase32hexmem("CO======", strlen("CO======"), true, &mem, &len) == 0);
 197         assert_se(streq(strndupa(mem, len), "f"));
 198         free(mem);
 199
 200         assert_se(unbase32hexmem("CPNG====", strlen("CPNG===="), true, &mem, &len) == 0);
 201         assert_se(streq(strndupa(mem, len), "fo"));
 202         free(mem);
 203
 204         assert_se(unbase32hexmem("CPNMU===", strlen("CPNMU==="), true, &mem, &len) == 0);
 205         assert_se(streq(strndupa(mem, len), "foo"));
 206         free(mem);
 207
 208         assert_se(unbase32hexmem("CPNMUOG=", strlen("CPNMUOG="), true, &mem, &len) == 0);
 209         assert_se(streq(strndupa(mem, len), "foob"));
 210         free(mem);
 211
 212         assert_se(unbase32hexmem("CPNMUOJ1", strlen("CPNMUOJ1"), true, &mem, &len) == 0);
 213         assert_se(streq(strndupa(mem, len), "fooba"));
 214         free(mem);
 215
 216         assert_se(unbase32hexmem("CPNMUOJ1E8======", strlen("CPNMUOJ1E8======"), true, &mem, &len) == 0);
 217         assert_se(streq(strndupa(mem, len), "foobar"));
 218         free(mem);
 219
 220         assert_se(unbase32hexmem("A", strlen("A"), true, &mem, &len) == -EINVAL);
 221         assert_se(unbase32hexmem("A=======", strlen("A======="), true, &mem, &len) == -EINVAL);
 222         assert_se(unbase32hexmem("AAA=====", strlen("AAA====="), true, &mem, &len) == -EINVAL);
 223         assert_se(unbase32hexmem("AAAAAA==", strlen("AAAAAA=="), true, &mem, &len) == -EINVAL);
 224         assert_se(unbase32hexmem("AB======", strlen("AB======"), true, &mem, &len) == -EINVAL);
 225         assert_se(unbase32hexmem("AAAB====", strlen("AAAB===="), true, &mem, &len) == -EINVAL);
 226         assert_se(unbase32hexmem("AAAAB===", strlen("AAAAB==="), true, &mem, &len) == -EINVAL);
 227         assert_se(unbase32hexmem("AAAAAAB=", strlen("AAAAAAB="), true, &mem, &len) == -EINVAL);
 228
 229         assert_se(unbase32hexmem("XPNMUOJ1", strlen("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 230         assert_se(unbase32hexmem("CXNMUOJ1", strlen("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 231         assert_se(unbase32hexmem("CPXMUOJ1", strlen("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 232         assert_se(unbase32hexmem("CPNXUOJ1", strlen("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 233         assert_se(unbase32hexmem("CPNMXOJ1", strlen("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 234         assert_se(unbase32hexmem("CPNMUXJ1", strlen("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 235         assert_se(unbase32hexmem("CPNMUOX1", strlen("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 236         assert_se(unbase32hexmem("CPNMUOJX", strlen("CPNMUOJ1"), true, &mem, &len) == -EINVAL);
 237
 238         assert_se(unbase32hexmem("", strlen(""), false, &mem, &len) == 0);
 239         assert_se(streq(strndupa(mem, len), ""));
 240         free(mem);
 241
 242         assert_se(unbase32hexmem("CO", strlen("CO"), false, &mem, &len) == 0);
 243         assert_se(streq(strndupa(mem, len), "f"));
 244         free(mem);
 245
 246         assert_se(unbase32hexmem("CPNG", strlen("CPNG"), false, &mem, &len) == 0);
 247         assert_se(streq(strndupa(mem, len), "fo"));
 248         free(mem);
 249
 250         assert_se(unbase32hexmem("CPNMU", strlen("CPNMU"), false, &mem, &len) == 0);
 251         assert_se(streq(strndupa(mem, len), "foo"));
 252         free(mem);
 253
 254         assert_se(unbase32hexmem("CPNMUOG", strlen("CPNMUOG"), false, &mem, &len) == 0);
 255         assert_se(streq(strndupa(mem, len), "foob"));
 256         free(mem);
 257
 258         assert_se(unbase32hexmem("CPNMUOJ1", strlen("CPNMUOJ1"), false, &mem, &len) == 0);
 259         assert_se(streq(strndupa(mem, len), "fooba"));
 260         free(mem);
 261
 262         assert_se(unbase32hexmem("CPNMUOJ1E8", strlen("CPNMUOJ1E8"), false, &mem, &len) == 0);
 263         assert_se(streq(strndupa(mem, len), "foobar"));
 264         free(mem);
 265
 266         assert_se(unbase32hexmem("CPNMUOG=", strlen("CPNMUOG="), false, &mem, &len) == -EINVAL);
 267         assert_se(unbase32hexmem("CPNMUOJ1E8======", strlen("CPNMUOJ1E8======"), false, &mem, &len) == -EINVAL);
 268         assert_se(unbase32hexmem("A", strlen("A"), false, &mem, &len) == -EINVAL);
 269         assert_se(unbase32hexmem("A", strlen("A"), false, &mem, &len) == -EINVAL);
 270         assert_se(unbase32hexmem("AAA", strlen("AAA"), false, &mem, &len) == -EINVAL);
 271         assert_se(unbase32hexmem("AAAAAA", strlen("AAAAAA"), false, &mem, &len) == -EINVAL);
 272         assert_se(unbase32hexmem("AB", strlen("AB"), false, &mem, &len) == -EINVAL);
 273         assert_se(unbase32hexmem("AAAB", strlen("AAAB"), false, &mem, &len) == -EINVAL);
 274         assert_se(unbase32hexmem("AAAAB", strlen("AAAAB"), false, &mem, &len) == -EINVAL);
 275         assert_se(unbase32hexmem("AAAAAAB", strlen("AAAAAAB"), false, &mem, &len) == -EINVAL);
 276 }
 277
 278 /* https://tools.ietf.org/html/rfc4648#section-10 */
 279 static void test_base64mem(void) {
 280         char *b64;
 281
 282         assert_se(base64mem("", strlen(""), &b64) == 0);
 283         assert_se(streq(b64, ""));
 284         free(b64);
 285
 286         assert_se(base64mem("f", strlen("f"), &b64) == 4);
 287         assert_se(streq(b64, "Zg=="));
 288         free(b64);
 289
 290         assert_se(base64mem("fo", strlen("fo"), &b64) == 4);
 291         assert_se(streq(b64, "Zm8="));
 292         free(b64);
 293
 294         assert_se(base64mem("foo", strlen("foo"), &b64) == 4);
 295         assert_se(streq(b64, "Zm9v"));
 296         free(b64);
 297
 298         assert_se(base64mem("foob", strlen("foob"), &b64) == 8);
 299         assert_se(streq(b64, "Zm9vYg=="));
 300         free(b64);
 301
 302         assert_se(base64mem("fooba", strlen("fooba"), &b64) == 8);
 303         assert_se(streq(b64, "Zm9vYmE="));
 304         free(b64);
 305
 306         assert_se(base64mem("foobar", strlen("foobar"), &b64) == 8);
 307         assert_se(streq(b64, "Zm9vYmFy"));
 308         free(b64);
 309 }
 310
 311 static void test_unbase64mem(void) {
 312         void *mem;
 313         size_t len;
 314
 315         assert_se(unbase64mem("", strlen(""), &mem, &len) == 0);
 316         assert_se(streq(strndupa(mem, len), ""));
 317         free(mem);
 318
 319         assert_se(unbase64mem("Zg==", strlen("Zg=="), &mem, &len) == 0);
 320         assert_se(streq(strndupa(mem, len), "f"));
 321         free(mem);
 322
 323         assert_se(unbase64mem("Zm8=", strlen("Zm8="), &mem, &len) == 0);
 324         assert_se(streq(strndupa(mem, len), "fo"));
 325         free(mem);
 326
 327         assert_se(unbase64mem("Zm9v", strlen("Zm9v"), &mem, &len) == 0);
 328         assert_se(streq(strndupa(mem, len), "foo"));
 329         free(mem);
 330
 331         assert_se(unbase64mem("Zm9vYg==", strlen("Zm9vYg=="), &mem, &len) == 0);
 332         assert_se(streq(strndupa(mem, len), "foob"));
 333         free(mem);
 334
 335         assert_se(unbase64mem("Zm9vYmE=", strlen("Zm9vYmE="), &mem, &len) == 0);
 336         assert_se(streq(strndupa(mem, len), "fooba"));
 337         free(mem);
 338
 339         assert_se(unbase64mem("Zm9vYmFy", strlen("Zm9vYmFy"), &mem, &len) == 0);
 340         assert_se(streq(strndupa(mem, len), "foobar"));
 341         free(mem);
 342
 343         assert_se(unbase64mem("A", strlen("A"), &mem, &len) == -EINVAL);
 344         assert_se(unbase64mem("A====", strlen("A===="), &mem, &len) == -EINVAL);
 345         assert_se(unbase64mem("AAB==", strlen("AAB=="), &mem, &len) == -EINVAL);
 346         assert_se(unbase64mem("AAAB=", strlen("AAAB="), &mem, &len) == -EINVAL);
 347 }
 348
 349 static void test_hexdump(void) {
 350         uint8_t data[146];
 351         unsigned i;
 352
 353         hexdump(stdout, NULL, 0);
 354         hexdump(stdout, "", 0);
 355         hexdump(stdout, "", 1);
 356         hexdump(stdout, "x", 1);
 357         hexdump(stdout, "x", 2);
 358         hexdump(stdout, "foobar", 7);
 359         hexdump(stdout, "f\nobar", 7);
 360         hexdump(stdout, "xxxxxxxxxxxxxxxxxxxxyz", 23);
 361
 362         for (i = 0; i < ELEMENTSOF(data); i++)
 363                 data[i] = i*2;
 364
 365         hexdump(stdout, data, sizeof(data));
 366 }
 367
 368 int main(int argc, char *argv[]) {
 369         test_hexchar();
 370         test_unhexchar();
 371         test_base32hexchar();
 372         test_unbase32hexchar();
 373         test_base64char();
 374         test_unbase64char();
 375         test_octchar();
 376         test_unoctchar();
 377         test_decchar();
 378         test_undecchar();
 379         test_unhexmem();
 380         test_base32hexmem();
 381         test_unbase32hexmem();
 382         test_base64mem();
 383         test_unbase64mem();
 384         test_hexdump();
 385
 386         return 0;
 387 }