summaryrefslogtreecommitdiff
path: root/tests/lib/b64_decode-test.c
blob: e6bbca50eef50be149ed5bfec0126412b9b8489a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
/* SPDX-License-Identifier: GPL-2.0-only */

#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <b64_decode.h>
#include <tests/test.h>

struct messages_t {
	const char *enc;
	const char *dec;
} messages[] = {
	{"QQ==", "A"},
	{"Q\r\nUI=", "AB"},
	{"QUJD", "ABC"},
	{"\nQUJDRA==", "ABCD"},
	{"SGVsbG8\r=", "Hello"},
	{"SGVsbG8h", "Hello!"}
};

const char *invalid[] = {
	"QQ=-=",
	"SGVsbG-8="
};

static void test_b64_decode(void **state)
{
	uint8_t *decoded;
	size_t res;

	for (int i = 0; i < ARRAY_SIZE(messages); i++) {
		decoded = malloc(strlen(messages[i].enc) * sizeof(char));

		res = b64_decode((uint8_t *)messages[i].enc, strlen(messages[i].enc), decoded);

		assert_int_equal(res, (strlen(messages[i].dec)));
		assert_string_equal(decoded, messages[i].dec);

		free(decoded);
	}

	for (int i = 0; i < ARRAY_SIZE(invalid); i++) {
		decoded = malloc(strlen(invalid[i]) * sizeof(char));

		res = b64_decode((uint8_t *)invalid[i], strlen(invalid[i]), decoded);

		assert_int_equal(res, 0);

		free(decoded);
	}
}

int main(void)
{
	const struct CMUnitTest tests[] = {
		cmocka_unit_test(test_b64_decode),
	};

	return cmocka_run_group_tests(tests, NULL, NULL);
}