tests/lib/b64_decode-test.c

   1 /* SPDX-License-Identifier: GPL-2.0-only */
   2
   3 #include <stddef.h>
   4 #include <stdint.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include <b64_decode.h>
   8 #include <tests/test.h>
   9
  10 struct messages_t {
  11         const char *enc;
  12         const char *dec;
  13 } messages[] = {
  14         {"QQ==", "A"},
  15         {"Q\r\nUI=", "AB"},
  16         {"QUJD", "ABC"},
  17         {"\nQUJDRA==", "ABCD"},
  18         {"SGVsbG8\r=", "Hello"},
  19         {"SGVsbG8h", "Hello!"}
  20 };
  21
  22 const char *invalid[] = {
  23         "QQ=-=",
  24         "SGVsbG-8="
  25 };
  26
  27 static void test_b64_decode(void **state)
  28 {
  29         uint8_t *decoded;
  30         size_t res;
  31
  32         for (int i = 0; i < ARRAY_SIZE(messages); i++) {
  33                 decoded = malloc(strlen(messages[i].enc) * sizeof(char));
  34
  35                 res = b64_decode((uint8_t *)messages[i].enc, strlen(messages[i].enc), decoded);
  36
  37                 assert_int_equal(res, (strlen(messages[i].dec)));
  38
  39                 decoded[res] = 0x00;
  40
  41                 assert_string_equal((const char *)decoded, messages[i].dec);
  42
  43                 free(decoded);
  44         }
  45
  46         for (int i = 0; i < ARRAY_SIZE(invalid); i++) {
  47                 decoded = malloc(strlen(invalid[i]) * sizeof(char));
  48
  49                 res = b64_decode((uint8_t *)invalid[i], strlen(invalid[i]), decoded);
  50
  51                 assert_int_equal(res, 0);
  52
  53                 free(decoded);
  54         }
  55 }
  56
  57 int main(void)
  58 {
  59         const struct CMUnitTest tests[] = {
  60                 cmocka_unit_test(test_b64_decode),
  61         };
  62
  63         return cb_run_group_tests(tests, NULL, NULL);
  64 }