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