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  }