static char* test_encode_and_decode_base64()
{
load_histograms();
uint8_t* buffer = NULL;
uint8_t* decoded = NULL;
char* encoded = NULL;
size_t len = 0;
int rc = 0;
rc = hdr_encode_compressed(cor_histogram, &buffer, &len);
mu_assert("Did not encode", validate_return_code(rc));
size_t encoded_len = hdr_base64_encoded_len(len);
size_t decoded_len = hdr_base64_decoded_len(encoded_len);
encoded = calloc(encoded_len + 1, sizeof(char));
decoded = calloc(decoded_len, sizeof(uint8_t));
hdr_base64_encode(buffer, len, encoded, encoded_len);
hdr_base64_decode(encoded, encoded_len, decoded, decoded_len);
mu_assert("Should be same", memcmp(buffer, decoded, len) == 0);
return 0;
}
int hdr_log_encode(struct hdr_histogram* histogram, char** encoded_histogram)
{
char *encoded_histogram_tmp = NULL;
uint8_t* compressed_histogram = NULL;
size_t compressed_len = 0;
int rc = 0;
int result = 0;
size_t encoded_len;
rc = hdr_encode_compressed(histogram, &compressed_histogram, &compressed_len);
if (rc != 0)
{
FAIL_AND_CLEANUP(cleanup, result, rc);
}
encoded_len = hdr_base64_encoded_len(compressed_len);
encoded_histogram_tmp = calloc(encoded_len + 1, sizeof(char));
rc = hdr_base64_encode(
compressed_histogram, compressed_len, encoded_histogram_tmp, encoded_len);
if (rc != 0)
{
FAIL_AND_CLEANUP(cleanup, result, rc);
}
*encoded_histogram = encoded_histogram_tmp;
cleanup:
free(compressed_histogram);
return result;
}
static char* base64_encode_fails_with_invalid_lengths()
{
mu_assert(
"Output length not 4/3 of input length",
hdr_base64_encode(NULL, 9, NULL, 11));
return 0;
}
static bool assert_base64_encode(const char* input, const char* expected)
{
size_t input_len = strlen(input);
int output_len = (int) (ceil(input_len / 3.0) * 4.0);
char* output = calloc(sizeof(char), output_len);
int r = hdr_base64_encode((uint8_t*)input, input_len, output, output_len);
bool result = r == 0 && compare_string(expected, output, output_len);
free(output);
return result;
}
int hdr_log_write(
struct hdr_log_writer* writer,
FILE* file,
const struct timespec* start_timestamp,
const struct timespec* end_timestamp,
struct hdr_histogram* histogram)
{
uint8_t* compressed_histogram = NULL;
size_t compressed_len = 0;
char* encoded_histogram = NULL;
int rc = 0;
int result = 0;
size_t encoded_len;
(void)writer;
rc = hdr_encode_compressed(histogram, &compressed_histogram, &compressed_len);
if (rc != 0)
{
FAIL_AND_CLEANUP(cleanup, result, rc);
}
encoded_len = hdr_base64_encoded_len(compressed_len);
encoded_histogram = calloc(encoded_len + 1, sizeof(char));
rc = hdr_base64_encode(
compressed_histogram, compressed_len, encoded_histogram, encoded_len);
if (rc != 0)
{
FAIL_AND_CLEANUP(cleanup, result, rc);
}
if (fprintf(
file, "%d.%d,%d.%d,%"PRIu64".0,%s\n",
(int) start_timestamp->tv_sec, (int) (start_timestamp->tv_nsec / 1000000),
(int) end_timestamp->tv_sec, (int) (end_timestamp->tv_nsec / 1000000),
hdr_max(histogram),
encoded_histogram) < 0)
{
result = EIO;
}
cleanup:
free(compressed_histogram);
free(encoded_histogram);
return result;
}
