43 lines
1.2 KiB
C++
43 lines
1.2 KiB
C++
|
// 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;
|
||
|
|
}
|