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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
| //
| // Copyright 2019 Staysail Systems, Inc. <info@staysail.tech>
| // Copyright 2018 Capitar IT Group BV <info@capitar.com>
| //
| // This software is supplied under the terms of the MIT License, a
| // copy of which should be located in the distribution where this
| // file was obtained (LICENSE.txt). A copy of the license may also be
| // found online at https://opensource.org/licenses/MIT.
| //
|
| #include <string.h>
|
| #include <nng/nng.h>
|
| #include "base64.h"
|
| #include <acutest.h>
|
| typedef struct {
| char *decoded;
| char *encoded;
| } test_case;
|
| static test_case cases[] = {
| { "", "" },
| { "f", "Zg==" },
| { "fo", "Zm8=" },
| { "foo", "Zm9v" },
| { "foob", "Zm9vYg==" },
| { "fooba", "Zm9vYmE=" },
| { "foobar", "Zm9vYmFy" },
| { NULL, NULL },
| };
|
| void
| test_encode(void)
| {
| int i;
| void *dec;
|
| for (i = 0; (dec = cases[i].decoded) != NULL; i++) {
| char buf[1024];
| char name[8];
| int rv;
|
| (void) snprintf(name, sizeof(name), "%d", i);
| TEST_CASE(name);
| rv = nni_base64_encode(dec, strlen(dec), buf, 1024);
| TEST_CHECK(rv >= 0);
| TEST_CHECK(rv == (int) strlen(cases[i].encoded));
| buf[rv] = 0;
| TEST_CHECK(strcmp(buf, cases[i].encoded) == 0);
| }
| }
|
| void
| test_decode(void)
| {
| int i;
| void *enc;
|
| for (i = 0; (enc = cases[i].encoded) != NULL; i++) {
| char buf[1024];
| char name[8];
| size_t sz;
|
| (void) snprintf(name, sizeof(name), "%d", i);
| TEST_CASE(name);
|
| sz = nni_base64_decode(enc, strlen(enc), (void *) buf, 1024);
| TEST_CHECK(sz != (size_t) -1);
| TEST_CHECK(sz == strlen(cases[i].decoded));
| buf[sz] = 0;
| TEST_CHECK(strcmp(buf, cases[i].decoded) == 0);
| }
| }
|
| void
| test_overflow(void)
| {
| char tmp[1024];
| for (int i = 1; cases[i].decoded != NULL; i++) {
| void *enc = cases[i].encoded;
| void *dec = cases[i].decoded;
| void *buf = tmp;
|
| char name[8];
| (void) snprintf(name, sizeof(name), "%d", i);
| TEST_CASE(name);
|
| TEST_CHECK(nni_base64_encode(dec, strlen(dec), buf,
| strlen(enc) - 1) == (size_t) -1);
| TEST_CHECK(nni_base64_encode(dec, strlen(dec), buf, 0) ==
| (size_t) -1);
|
| TEST_CHECK(nni_base64_decode(enc, strlen(enc), buf,
| strlen(dec) - 1) == (size_t) -1);
| TEST_CHECK(nni_base64_encode(enc, strlen(enc), buf, 0) ==
| (size_t) -1);
| }
| }
|
| TEST_LIST = {
| { "encode", test_encode },
| { "decode", test_decode },
| { "overflow", test_overflow },
| { NULL, NULL },
| };
|
|
