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 },
};
|
