diff options
Diffstat (limited to 'streaming/compression_brotli.c')
-rw-r--r-- | streaming/compression_brotli.c | 142 |
1 files changed, 0 insertions, 142 deletions
diff --git a/streaming/compression_brotli.c b/streaming/compression_brotli.c deleted file mode 100644 index cf52f3bca..000000000 --- a/streaming/compression_brotli.c +++ /dev/null @@ -1,142 +0,0 @@ -// SPDX-License-Identifier: GPL-3.0-or-later - -#include "compression_brotli.h" - -#ifdef ENABLE_BROTLI -#include <brotli/encode.h> -#include <brotli/decode.h> - -void rrdpush_compressor_init_brotli(struct compressor_state *state) { - if (!state->initialized) { - state->initialized = true; - state->stream = BrotliEncoderCreateInstance(NULL, NULL, NULL); - - if (state->level < BROTLI_MIN_QUALITY) { - state->level = BROTLI_MIN_QUALITY; - } else if (state->level > BROTLI_MAX_QUALITY) { - state->level = BROTLI_MAX_QUALITY; - } - - BrotliEncoderSetParameter(state->stream, BROTLI_PARAM_QUALITY, state->level); - } -} - -void rrdpush_compressor_destroy_brotli(struct compressor_state *state) { - if (state->stream) { - BrotliEncoderDestroyInstance(state->stream); - state->stream = NULL; - } -} - -size_t rrdpush_compress_brotli(struct compressor_state *state, const char *data, size_t size, const char **out) { - if (unlikely(!state || !size || !out)) - return 0; - - simple_ring_buffer_make_room(&state->output, MAX(BrotliEncoderMaxCompressedSize(size), COMPRESSION_MAX_CHUNK)); - - size_t available_out = state->output.size; - - size_t available_in = size; - const uint8_t *next_in = (const uint8_t *)data; - uint8_t *next_out = (uint8_t *)state->output.data; - - if (!BrotliEncoderCompressStream(state->stream, BROTLI_OPERATION_FLUSH, &available_in, &next_in, &available_out, &next_out, NULL)) { - netdata_log_error("STREAM: Brotli compression failed."); - return 0; - } - - if(available_in != 0) { - netdata_log_error("STREAM: BrotliEncoderCompressStream() did not use all the input buffer, %zu bytes out of %zu remain", - available_in, size); - return 0; - } - - size_t compressed_size = state->output.size - available_out; - if(available_out == 0) { - netdata_log_error("STREAM: BrotliEncoderCompressStream() needs a bigger output buffer than the one we provided " - "(output buffer %zu bytes, compressed payload %zu bytes)", - state->output.size, size); - return 0; - } - - if(compressed_size == 0) { - netdata_log_error("STREAM: BrotliEncoderCompressStream() did not produce any output from the input provided " - "(input buffer %zu bytes)", - size); - return 0; - } - - state->sender_locked.total_compressions++; - state->sender_locked.total_uncompressed += size - available_in; - state->sender_locked.total_compressed += compressed_size; - - *out = state->output.data; - return compressed_size; -} - -void rrdpush_decompressor_init_brotli(struct decompressor_state *state) { - if (!state->initialized) { - state->initialized = true; - state->stream = BrotliDecoderCreateInstance(NULL, NULL, NULL); - - simple_ring_buffer_make_room(&state->output, COMPRESSION_MAX_CHUNK); - } -} - -void rrdpush_decompressor_destroy_brotli(struct decompressor_state *state) { - if (state->stream) { - BrotliDecoderDestroyInstance(state->stream); - state->stream = NULL; - } -} - -size_t rrdpush_decompress_brotli(struct decompressor_state *state, const char *compressed_data, size_t compressed_size) { - if (unlikely(!state || !compressed_data || !compressed_size)) - return 0; - - // The state.output ring buffer is always EMPTY at this point, - // meaning that (state->output.read_pos == state->output.write_pos) - // However, THEY ARE NOT ZERO. - - size_t available_out = state->output.size; - size_t available_in = compressed_size; - const uint8_t *next_in = (const uint8_t *)compressed_data; - uint8_t *next_out = (uint8_t *)state->output.data; - - if (BrotliDecoderDecompressStream(state->stream, &available_in, &next_in, &available_out, &next_out, NULL) == BROTLI_DECODER_RESULT_ERROR) { - netdata_log_error("STREAM: Brotli decompression failed."); - return 0; - } - - if(available_in != 0) { - netdata_log_error("STREAM: BrotliDecoderDecompressStream() did not use all the input buffer, %zu bytes out of %zu remain", - available_in, compressed_size); - return 0; - } - - size_t decompressed_size = state->output.size - available_out; - if(available_out == 0) { - netdata_log_error("STREAM: BrotliDecoderDecompressStream() needs a bigger output buffer than the one we provided " - "(output buffer %zu bytes, compressed payload %zu bytes)", - state->output.size, compressed_size); - return 0; - } - - if(decompressed_size == 0) { - netdata_log_error("STREAM: BrotliDecoderDecompressStream() did not produce any output from the input provided " - "(input buffer %zu bytes)", - compressed_size); - return 0; - } - - state->output.read_pos = 0; - state->output.write_pos = decompressed_size; - - state->total_compressed += compressed_size - available_in; - state->total_uncompressed += decompressed_size; - state->total_compressions++; - - return decompressed_size; -} - -#endif // ENABLE_BROTLI |