int main()
{
unsigned char out[CHUNK];
z_stream strm;
strm_init(&strm);
strm.next_in = (unsigned char *) message;
strm.avail_in = strlen(message);
do
{
int have;
strm.avail_out = CHUNK;
strm.next_out = out;
CALL_ZLIB(deflate(&strm, Z_FINISH) );
have = CHUNK - strm.avail_out;
fwrite(out, sizeof(char), have, stdout);
}
while ( strm.avail_out == 0 );
deflateEnd(&strm);
return (0);
}
static int hdr_decode_compressed_v0(
_compression_flyweight* compression_flyweight,
size_t length,
struct hdr_histogram** histogram)
{
struct hdr_histogram* h = NULL;
int result = 0;
uint8_t* counts_array = NULL;
_encoding_flyweight_v0 encoding_flyweight;
z_stream strm;
strm_init(&strm);
if (inflateInit(&strm) != Z_OK)
{
FAIL_AND_CLEANUP(cleanup, result, HDR_INFLATE_FAIL);
}
int32_t compressed_length = be32toh(compression_flyweight->length);
if (compressed_length < 0 || length - sizeof(_compression_flyweight) < (size_t)compressed_length)
{
FAIL_AND_CLEANUP(cleanup, result, EINVAL);
}
strm.next_in = compression_flyweight->data;
strm.avail_in = (uInt) compressed_length;
strm.next_out = (uint8_t *) &encoding_flyweight;
strm.avail_out = sizeof(_encoding_flyweight_v0);
if (inflate(&strm, Z_SYNC_FLUSH) != Z_OK)
{
FAIL_AND_CLEANUP(cleanup, result, HDR_INFLATE_FAIL);
}
int32_t encoding_cookie = get_cookie_base(be32toh(encoding_flyweight.cookie));
if (V0_ENCODING_COOKIE != encoding_cookie)
{
FAIL_AND_CLEANUP(cleanup, result, HDR_ENCODING_COOKIE_MISMATCH);
}
int32_t word_size = word_size_from_cookie(be32toh(encoding_flyweight.cookie));
int64_t lowest_trackable_value = be64toh(encoding_flyweight.lowest_trackable_value);
int64_t highest_trackable_value = be64toh(encoding_flyweight.highest_trackable_value);
int32_t significant_figures = be32toh(encoding_flyweight.significant_figures);
if (hdr_init(
lowest_trackable_value,
highest_trackable_value,
significant_figures,
&h) != 0)
{
FAIL_AND_CLEANUP(cleanup, result, ENOMEM);
}
int32_t counts_array_len = h->counts_len * word_size;
if ((counts_array = calloc(1, (size_t) counts_array_len)) == NULL)
{
FAIL_AND_CLEANUP(cleanup, result, ENOMEM);
}
strm.next_out = counts_array;
strm.avail_out = (uInt) counts_array_len;
if (inflate(&strm, Z_FINISH) != Z_STREAM_END)
{
FAIL_AND_CLEANUP(cleanup, result, HDR_INFLATE_FAIL);
}
_apply_to_counts(h, word_size, counts_array, h->counts_len);
hdr_reset_internal_counters(h);
h->normalizing_index_offset = 0;
h->conversion_ratio = 1.0;
cleanup:
(void)inflateEnd(&strm);
free(counts_array);
if (result != 0)
{
free(h);
}
else if (NULL == *histogram)
{
*histogram = h;
}
else
{
hdr_add(*histogram, h);
free(h);
}
return result;
}
static int hdr_decode_compressed_v2(
_compression_flyweight* compression_flyweight,
size_t length,
struct hdr_histogram** histogram)
{
struct hdr_histogram* h = NULL;
int result = 0;
uint8_t* counts_array = NULL;
_encoding_flyweight_v1 encoding_flyweight;
z_stream strm;
strm_init(&strm);
if (inflateInit(&strm) != Z_OK)
{
FAIL_AND_CLEANUP(cleanup, result, HDR_INFLATE_FAIL);
}
int32_t compressed_length = be32toh(compression_flyweight->length);
if (compressed_length < 0 || length - sizeof(_compression_flyweight) < (size_t)compressed_length)
{
FAIL_AND_CLEANUP(cleanup, result, EINVAL);
}
strm.next_in = compression_flyweight->data;
strm.avail_in = (uInt) compressed_length;
strm.next_out = (uint8_t *) &encoding_flyweight;
strm.avail_out = sizeof(_encoding_flyweight_v1);
if (inflate(&strm, Z_SYNC_FLUSH) != Z_OK)
{
FAIL_AND_CLEANUP(cleanup, result, HDR_INFLATE_FAIL);
}
int32_t encoding_cookie = get_cookie_base(be32toh(encoding_flyweight.cookie));
if (V2_ENCODING_COOKIE != encoding_cookie)
{
FAIL_AND_CLEANUP(cleanup, result, HDR_ENCODING_COOKIE_MISMATCH);
}
int32_t counts_limit = be32toh(encoding_flyweight.payload_len);
int64_t lowest_trackable_value = be64toh(encoding_flyweight.lowest_trackable_value);
int64_t highest_trackable_value = be64toh(encoding_flyweight.highest_trackable_value);
int32_t significant_figures = be32toh(encoding_flyweight.significant_figures);
if (hdr_init(
lowest_trackable_value,
highest_trackable_value,
significant_figures,
&h) != 0)
{
FAIL_AND_CLEANUP(cleanup, result, ENOMEM);
}
// Make sure there at least 9 bytes to read
// if there is a corrupt value at the end
// of the array we won't read corrupt data or crash.
if ((counts_array = calloc(1, (size_t) counts_limit + 9)) == NULL)
{
FAIL_AND_CLEANUP(cleanup, result, ENOMEM);
}
strm.next_out = counts_array;
strm.avail_out = (uInt) counts_limit;
if (inflate(&strm, Z_FINISH) != Z_STREAM_END)
{
FAIL_AND_CLEANUP(cleanup, result, HDR_INFLATE_FAIL);
}
int r = _apply_to_counts_zz(h, counts_array, counts_limit);
if (0 != r)
{
FAIL_AND_CLEANUP(cleanup, result, r);
}
h->normalizing_index_offset = be32toh(encoding_flyweight.normalizing_index_offset);
h->conversion_ratio = int64_bits_to_double(be64toh(encoding_flyweight.conversion_ratio_bits));
hdr_reset_internal_counters(h);
cleanup:
(void)inflateEnd(&strm);
free(counts_array);
if (result != 0)
{
free(h);
}
else if (NULL == *histogram)
{
*histogram = h;
}
else
{
hdr_add(*histogram, h);
free(h);
}
return result;
}
/*
* ======== api_init ========
* Purpose:
* Module initialization used by Bridge API.
*/
bool api_init(void)
{
bool ret = true;
bool fdrv, fdev, fcod, fchnl, fmsg, fio;
bool fmgr, fproc, fnode, fdisp, fstrm, frmm;
if (api_c_refs == 0) {
/* initialize driver and other modules */
fdrv = drv_init();
fmgr = mgr_init();
fproc = proc_init();
fnode = node_init();
fdisp = disp_init();
fstrm = strm_init();
frmm = rmm_init();
fchnl = chnl_init();
fmsg = msg_mod_init();
fio = io_init();
fdev = dev_init();
fcod = cod_init();
ret = fdrv && fdev && fchnl && fcod && fmsg && fio;
ret = ret && fmgr && fproc && frmm;
if (!ret) {
if (fdrv)
drv_exit();
if (fmgr)
mgr_exit();
if (fstrm)
strm_exit();
if (fproc)
proc_exit();
if (fnode)
node_exit();
if (fdisp)
disp_exit();
if (fchnl)
chnl_exit();
if (fmsg)
msg_exit();
if (fio)
io_exit();
if (fdev)
dev_exit();
if (fcod)
cod_exit();
if (frmm)
rmm_exit();
}
}
if (ret)
api_c_refs++;
return ret;
}
