diff options
author | Alberto Gonzalez Iniesta <agi@inittab.org> | 2016-12-27 18:25:47 +0100 |
---|---|---|
committer | Alberto Gonzalez Iniesta <agi@inittab.org> | 2016-12-27 18:25:47 +0100 |
commit | 79f3537f69e125f19f59c36aa090120a63186a54 (patch) | |
tree | 2089a3b7dac990841dbc2e4d9b2f535b82dbb0af /src/openvpn/comp-lz4.c | |
parent | f2137fedb30cb87448eb03b2f288920df6187571 (diff) | |
parent | 3a2bbdb05ca6a6996e424c9fb225cb0d53804125 (diff) |
Merge tag 'upstream/2.4.0'
Upstream version 2.4.0
Diffstat (limited to 'src/openvpn/comp-lz4.c')
-rw-r--r-- | src/openvpn/comp-lz4.c | 348 |
1 files changed, 181 insertions, 167 deletions
diff --git a/src/openvpn/comp-lz4.c b/src/openvpn/comp-lz4.c index 395f3d2..fa65f87 100644 --- a/src/openvpn/comp-lz4.c +++ b/src/openvpn/comp-lz4.c @@ -5,8 +5,8 @@ * packet encryption, packet authentication, and * packet compression. * - * Copyright (C) 2002-2012 OpenVPN Technologies, Inc. <sales@openvpn.net> - * Copyright (C) 2013 Gert Doering <gert@greenie.muc.de> + * Copyright (C) 2002-2017 OpenVPN Technologies, Inc. <sales@openvpn.net> + * Copyright (C) 2013-2017 Gert Doering <gert@greenie.muc.de> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License version 2 @@ -45,263 +45,277 @@ #include "memdbg.h" static void -lz4_compress_init (struct compress_context *compctx) +lz4_compress_init(struct compress_context *compctx) { - msg (D_INIT_MEDIUM, "LZ4 compression initializing"); - ASSERT(compctx->flags & COMP_F_SWAP); + msg(D_INIT_MEDIUM, "LZ4 compression initializing"); + ASSERT(compctx->flags & COMP_F_SWAP); } static void -lz4v2_compress_init (struct compress_context *compctx) +lz4v2_compress_init(struct compress_context *compctx) { - msg (D_INIT_MEDIUM, "LZ4v2 compression initializing"); + msg(D_INIT_MEDIUM, "LZ4v2 compression initializing"); } static void -lz4_compress_uninit (struct compress_context *compctx) +lz4_compress_uninit(struct compress_context *compctx) { } static bool -do_lz4_compress (struct buffer *buf, - struct buffer *work, - struct compress_context *compctx, - const struct frame* frame) +do_lz4_compress(struct buffer *buf, + struct buffer *work, + struct compress_context *compctx, + const struct frame *frame) { - /* - * In order to attempt compression, length must be at least COMPRESS_THRESHOLD. - */ - if (buf->len >= COMPRESS_THRESHOLD) + /* + * In order to attempt compression, length must be at least COMPRESS_THRESHOLD. + */ + if (buf->len >= COMPRESS_THRESHOLD) { - const size_t ps = PAYLOAD_SIZE (frame); - int zlen_max = ps + COMP_EXTRA_BUFFER (ps); - int zlen; + const size_t ps = PAYLOAD_SIZE(frame); + int zlen_max = ps + COMP_EXTRA_BUFFER(ps); + int zlen; - ASSERT (buf_init (work, FRAME_HEADROOM (frame))); - ASSERT (buf_safe (work, zlen_max)); + ASSERT(buf_init(work, FRAME_HEADROOM(frame))); + ASSERT(buf_safe(work, zlen_max)); - if (buf->len > ps) - { - dmsg (D_COMP_ERRORS, "LZ4 compression buffer overflow"); - buf->len = 0; - return false; - } + if (buf->len > ps) + { + dmsg(D_COMP_ERRORS, "LZ4 compression buffer overflow"); + buf->len = 0; + return false; + } - zlen = LZ4_compress_limitedOutput((const char *)BPTR(buf), (char *)BPTR(work), BLEN(buf), zlen_max ); + zlen = LZ4_compress_limitedOutput((const char *)BPTR(buf), (char *)BPTR(work), BLEN(buf), zlen_max ); - if (zlen <= 0) - { - dmsg (D_COMP_ERRORS, "LZ4 compression error"); - buf->len = 0; - return false; - } + if (zlen <= 0) + { + dmsg(D_COMP_ERRORS, "LZ4 compression error"); + buf->len = 0; + return false; + } - ASSERT (buf_safe (work, zlen)); - work->len = zlen; + ASSERT(buf_safe(work, zlen)); + work->len = zlen; - dmsg (D_COMP, "LZ4 compress %d -> %d", buf->len, work->len); - compctx->pre_compress += buf->len; - compctx->post_compress += work->len; - return true; + dmsg(D_COMP, "LZ4 compress %d -> %d", buf->len, work->len); + compctx->pre_compress += buf->len; + compctx->post_compress += work->len; + return true; } - return false; + return false; } static void -lz4_compress (struct buffer *buf, struct buffer work, - struct compress_context *compctx, - const struct frame* frame) +lz4_compress(struct buffer *buf, struct buffer work, + struct compress_context *compctx, + const struct frame *frame) { - bool compressed; - if (buf->len <= 0) - return; - - compressed = do_lz4_compress(buf, &work, compctx, frame); + bool compressed; + if (buf->len <= 0) + { + return; + } - /* On error do_lz4_compress sets buf len to zero, just return */ - if (buf->len == 0) - return; + compressed = do_lz4_compress(buf, &work, compctx, frame); - /* did compression save us anything? */ - { - uint8_t comp_head_byte = NO_COMPRESS_BYTE_SWAP; - if (compressed && work.len < buf->len) - { - *buf = work; - comp_head_byte = LZ4_COMPRESS_BYTE; - } + /* On error do_lz4_compress sets buf len to zero, just return */ + if (buf->len == 0) + { + return; + } + /* did compression save us anything? */ { - uint8_t *head = BPTR (buf); - uint8_t *tail = BEND (buf); - ASSERT (buf_safe (buf, 1)); - ++buf->len; - - /* move head byte of payload to tail */ - *tail = *head; - *head = comp_head_byte; + uint8_t comp_head_byte = NO_COMPRESS_BYTE_SWAP; + if (compressed && work.len < buf->len) + { + *buf = work; + comp_head_byte = LZ4_COMPRESS_BYTE; + } + + { + uint8_t *head = BPTR(buf); + uint8_t *tail = BEND(buf); + ASSERT(buf_safe(buf, 1)); + ++buf->len; + + /* move head byte of payload to tail */ + *tail = *head; + *head = comp_head_byte; + } } - } } static void -lz4v2_compress (struct buffer *buf, struct buffer work, - struct compress_context *compctx, - const struct frame* frame) +lz4v2_compress(struct buffer *buf, struct buffer work, + struct compress_context *compctx, + const struct frame *frame) { - bool compressed; - if (buf->len <= 0) - return; + bool compressed; + if (buf->len <= 0) + { + return; + } - compressed = do_lz4_compress(buf, &work, compctx, frame); + compressed = do_lz4_compress(buf, &work, compctx, frame); - /* On Error just return */ - if (buf->len == 0) - return; + /* On Error just return */ + if (buf->len == 0) + { + return; + } - /* did compression save us anything? Include 2 byte compression header - in calculation */ - if (compressed && work.len + 2 < buf->len) + /* did compression save us anything? Include 2 byte compression header + * in calculation */ + if (compressed && work.len + 2 < buf->len) { - ASSERT(buf_prepend(&work, 2)); - uint8_t *head = BPTR (&work); - head[0] = COMP_ALGV2_INDICATOR_BYTE; - head[1] = COMP_ALGV2_LZ4_BYTE; - *buf = work; + ASSERT(buf_prepend(&work, 2)); + uint8_t *head = BPTR(&work); + head[0] = COMP_ALGV2_INDICATOR_BYTE; + head[1] = COMP_ALGV2_LZ4_BYTE; + *buf = work; } - else + else { - compv2_escape_data_ifneeded(buf); + compv2_escape_data_ifneeded(buf); } } void do_lz4_decompress(size_t zlen_max, - struct buffer *work, - struct buffer *buf, - struct compress_context *compctx) + struct buffer *work, + struct buffer *buf, + struct compress_context *compctx) { - int uncomp_len; - ASSERT (buf_safe (work, zlen_max)); - uncomp_len = LZ4_decompress_safe((const char *)BPTR(buf), (char *)BPTR(work), (size_t)BLEN(buf), zlen_max); - if (uncomp_len <= 0) + int uncomp_len; + ASSERT(buf_safe(work, zlen_max)); + uncomp_len = LZ4_decompress_safe((const char *)BPTR(buf), (char *)BPTR(work), (size_t)BLEN(buf), zlen_max); + if (uncomp_len <= 0) { - dmsg (D_COMP_ERRORS, "LZ4 decompression error: %d", uncomp_len); - buf->len = 0; - return; + dmsg(D_COMP_ERRORS, "LZ4 decompression error: %d", uncomp_len); + buf->len = 0; + return; } - ASSERT (buf_safe (work, uncomp_len)); - work->len = uncomp_len; + ASSERT(buf_safe(work, uncomp_len)); + work->len = uncomp_len; - dmsg (D_COMP, "LZ4 decompress %d -> %d", buf->len, work->len); - compctx->pre_decompress += buf->len; - compctx->post_decompress += work->len; + dmsg(D_COMP, "LZ4 decompress %d -> %d", buf->len, work->len); + compctx->pre_decompress += buf->len; + compctx->post_decompress += work->len; - *buf = *work; + *buf = *work; } static void -lz4_decompress (struct buffer *buf, struct buffer work, - struct compress_context *compctx, - const struct frame* frame) +lz4_decompress(struct buffer *buf, struct buffer work, + struct compress_context *compctx, + const struct frame *frame) { - size_t zlen_max = EXPANDED_SIZE (frame); - uint8_t c; /* flag indicating whether or not our peer compressed */ + size_t zlen_max = EXPANDED_SIZE(frame); + uint8_t c; /* flag indicating whether or not our peer compressed */ - if (buf->len <= 0) - return; + if (buf->len <= 0) + { + return; + } - ASSERT (buf_init (&work, FRAME_HEADROOM (frame))); + ASSERT(buf_init(&work, FRAME_HEADROOM(frame))); - /* do unframing/swap (assumes buf->len > 0) */ - { - uint8_t *head = BPTR (buf); - c = *head; - --buf->len; - *head = *BEND (buf); - } + /* do unframing/swap (assumes buf->len > 0) */ + { + uint8_t *head = BPTR(buf); + c = *head; + --buf->len; + *head = *BEND(buf); + } - if (c == LZ4_COMPRESS_BYTE) /* packet was compressed */ + if (c == LZ4_COMPRESS_BYTE) /* packet was compressed */ { - do_lz4_decompress(zlen_max, &work, buf, compctx); + do_lz4_decompress(zlen_max, &work, buf, compctx); } - else if (c == NO_COMPRESS_BYTE_SWAP) /* packet was not compressed */ + else if (c == NO_COMPRESS_BYTE_SWAP) /* packet was not compressed */ { - ; } - else + else { - dmsg (D_COMP_ERRORS, "Bad LZ4 decompression header byte: %d", c); - buf->len = 0; + dmsg(D_COMP_ERRORS, "Bad LZ4 decompression header byte: %d", c); + buf->len = 0; } } static void -lz4v2_decompress (struct buffer *buf, struct buffer work, - struct compress_context *compctx, - const struct frame* frame) +lz4v2_decompress(struct buffer *buf, struct buffer work, + struct compress_context *compctx, + const struct frame *frame) { - size_t zlen_max = EXPANDED_SIZE (frame); - uint8_t c; /* flag indicating whether or not our peer compressed */ + size_t zlen_max = EXPANDED_SIZE(frame); + uint8_t c; /* flag indicating whether or not our peer compressed */ - if (buf->len <= 0) - return; + if (buf->len <= 0) + { + return; + } - ASSERT (buf_init (&work, FRAME_HEADROOM (frame))); + ASSERT(buf_init(&work, FRAME_HEADROOM(frame))); - /* do unframing/swap (assumes buf->len > 0) */ - uint8_t *head = BPTR (buf); - c = *head; + /* do unframing/swap (assumes buf->len > 0) */ + uint8_t *head = BPTR(buf); + c = *head; - /* Not compressed */ - if (c != COMP_ALGV2_INDICATOR_BYTE) { - return; - } + /* Not compressed */ + if (c != COMP_ALGV2_INDICATOR_BYTE) + { + return; + } - /* Packet to short to make sense */ - if (buf->len <= 1) + /* Packet to short to make sense */ + if (buf->len <= 1) { - buf->len=0; - return; + buf->len = 0; + return; } - c = head[1]; - if (c == COMP_ALGV2_LZ4_BYTE) /* packet was compressed */ + c = head[1]; + if (c == COMP_ALGV2_LZ4_BYTE) /* packet was compressed */ { - buf_advance(buf,2); - do_lz4_decompress(zlen_max, &work, buf, compctx); + buf_advance(buf,2); + do_lz4_decompress(zlen_max, &work, buf, compctx); } - else if (c == COMP_ALGV2_UNCOMPRESSED_BYTE) + else if (c == COMP_ALGV2_UNCOMPRESSED_BYTE) { - buf_advance(buf,2); + buf_advance(buf,2); } - else + else { - dmsg (D_COMP_ERRORS, "Bad LZ4v2 decompression header byte: %d", c); - buf->len = 0; + dmsg(D_COMP_ERRORS, "Bad LZ4v2 decompression header byte: %d", c); + buf->len = 0; } } const struct compress_alg lz4_alg = { - "lz4", - lz4_compress_init, - lz4_compress_uninit, - lz4_compress, - lz4_decompress + "lz4", + lz4_compress_init, + lz4_compress_uninit, + lz4_compress, + lz4_decompress }; const struct compress_alg lz4v2_alg = { - "lz4v2", - lz4v2_compress_init, - lz4_compress_uninit, - lz4v2_compress, - lz4v2_decompress + "lz4v2", + lz4v2_compress_init, + lz4_compress_uninit, + lz4v2_compress, + lz4v2_decompress }; -#else -static void dummy(void) {} +#else /* if defined(ENABLE_LZ4) */ +static void +dummy(void) { +} #endif /* ENABLE_LZ4 */ |