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
|
// RUN: %clangxx %s -o %t && %run %t %p
#include <assert.h>
#include <resolv.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
int main(int iArgc, char *szArgv[]) {
// Check NTOP writing
const char *src = "base64 test data";
size_t src_len = strlen(src);
size_t dst_len = ((src_len + 2) / 3) * 4 + 1;
char dst[dst_len];
int res = b64_ntop(reinterpret_cast<const unsigned char *>(src), src_len, dst,
dst_len);
assert(res >= 0);
assert(strcmp(dst, "YmFzZTY0IHRlc3QgZGF0YQ==") == 0);
// Check PTON writing
unsigned char target[dst_len];
res = b64_pton(dst, target, dst_len);
assert(res >= 0);
assert(strncmp(reinterpret_cast<const char *>(target), src, res) == 0);
// Check NTOP writing for zero length src
src = "";
src_len = strlen(src);
assert(((src_len + 2) / 3) * 4 + 1 < dst_len);
res = b64_ntop(reinterpret_cast<const unsigned char *>(src), src_len, dst,
dst_len);
assert(res >= 0);
assert(strcmp(dst, "") == 0);
// Check PTON writing for zero length src
dst[0] = '\0';
res = b64_pton(dst, target, dst_len);
assert(res >= 0);
assert(strncmp(reinterpret_cast<const char *>(target), src, res) == 0);
return 0;
}
|
