| /* |
| zip_source_deflate.c -- deflate (de)compressoin routines |
| Copyright (C) 2009-2014 Dieter Baron and Thomas Klausner |
| |
| This file is part of libzip, a library to manipulate ZIP archives. |
| The authors can be contacted at <libzip@nih.at> |
| |
| Redistribution and use in source and binary forms, with or without |
| modification, are permitted provided that the following conditions |
| are met: |
| 1. Redistributions of source code must retain the above copyright |
| notice, this list of conditions and the following disclaimer. |
| 2. Redistributions in binary form must reproduce the above copyright |
| notice, this list of conditions and the following disclaimer in |
| the documentation and/or other materials provided with the |
| distribution. |
| 3. The names of the authors may not be used to endorse or promote |
| products derived from this software without specific prior |
| written permission. |
| |
| THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS |
| OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY |
| DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE |
| GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER |
| IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR |
| OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN |
| IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #include <stdlib.h> |
| #include <string.h> |
| #include <limits.h> |
| |
| #include "zipint.h" |
| |
| struct deflate { |
| zip_error_t error; |
| |
| bool eof; |
| bool can_store; |
| bool is_stored; |
| int mem_level; |
| zip_uint64_t size; |
| zip_uint8_t buffer[BUFSIZE]; |
| z_stream zstr; |
| }; |
| |
| static zip_int64_t compress_read(zip_source_t *, struct deflate *, void *, zip_uint64_t); |
| static zip_int64_t decompress_read(zip_source_t *, struct deflate *, void *, zip_uint64_t); |
| static zip_int64_t deflate_compress(zip_source_t *, void *, void *, zip_uint64_t, zip_source_cmd_t); |
| static zip_int64_t deflate_decompress(zip_source_t *, void *, void *, zip_uint64_t, zip_source_cmd_t); |
| static void deflate_free(struct deflate *); |
| |
| |
| zip_source_t * |
| zip_source_deflate(zip_t *za, zip_source_t *src, zip_int32_t cm, int flags) |
| { |
| struct deflate *ctx; |
| zip_source_t *s2; |
| |
| if (src == NULL || (cm != ZIP_CM_DEFLATE && !ZIP_CM_IS_DEFAULT(cm))) { |
| zip_error_set(&za->error, ZIP_ER_INVAL, 0); |
| return NULL; |
| } |
| |
| if ((ctx=(struct deflate *)malloc(sizeof(*ctx))) == NULL) { |
| zip_error_set(&za->error, ZIP_ER_MEMORY, 0); |
| return NULL; |
| } |
| |
| zip_error_init(&ctx->error); |
| ctx->eof = false; |
| ctx->is_stored = false; |
| ctx->can_store = ZIP_CM_IS_DEFAULT(cm); |
| if (flags & ZIP_CODEC_ENCODE) { |
| ctx->mem_level = MAX_MEM_LEVEL; |
| } |
| |
| if ((s2=zip_source_layered(za, src, |
| ((flags & ZIP_CODEC_ENCODE) |
| ? deflate_compress : deflate_decompress), |
| ctx)) == NULL) { |
| deflate_free(ctx); |
| return NULL; |
| } |
| |
| return s2; |
| } |
| |
| |
| static zip_int64_t |
| compress_read(zip_source_t *src, struct deflate *ctx, void *data, zip_uint64_t len) |
| { |
| int end, ret; |
| zip_int64_t n; |
| zip_uint64_t out_offset; |
| uInt out_len; |
| |
| if (zip_error_code_zip(&ctx->error) != ZIP_ER_OK) |
| return -1; |
| |
| if (len == 0 || ctx->is_stored) { |
| return 0; |
| } |
| |
| out_offset = 0; |
| out_len = (uInt)ZIP_MIN(UINT_MAX, len); |
| ctx->zstr.next_out = (Bytef *)data; |
| ctx->zstr.avail_out = out_len; |
| |
| end = 0; |
| while (!end) { |
| ret = deflate(&ctx->zstr, ctx->eof ? Z_FINISH : 0); |
| |
| switch (ret) { |
| case Z_STREAM_END: |
| if (ctx->can_store && ctx->zstr.total_in <= ctx->zstr.total_out) { |
| ctx->is_stored = true; |
| ctx->size = ctx->zstr.total_in; |
| memcpy(data, ctx->buffer, ctx->size); |
| return (zip_int64_t)ctx->size; |
| } |
| /* fallthrough */ |
| case Z_OK: |
| /* all ok */ |
| |
| if (ctx->zstr.avail_out == 0) { |
| out_offset += out_len; |
| if (out_offset < len) { |
| out_len = (uInt)ZIP_MIN(UINT_MAX, len-out_offset); |
| ctx->zstr.next_out = (Bytef *)data+out_offset; |
| ctx->zstr.avail_out = out_len; |
| } |
| else { |
| ctx->can_store = false; |
| end = 1; |
| } |
| } |
| else if (ctx->eof && ctx->zstr.avail_in == 0) |
| end = 1; |
| break; |
| |
| case Z_BUF_ERROR: |
| if (ctx->zstr.avail_in == 0) { |
| if (ctx->eof) { |
| end = 1; |
| break; |
| } |
| |
| if ((n=zip_source_read(src, ctx->buffer, sizeof(ctx->buffer))) < 0) { |
| _zip_error_set_from_source(&ctx->error, src); |
| end = 1; |
| break; |
| } |
| else if (n == 0) { |
| ctx->eof = true; |
| /* TODO: check against stat of src? */ |
| ctx->size = ctx->zstr.total_in; |
| } |
| else { |
| if (ctx->zstr.total_in > 0) { |
| /* we overwrote a previously filled ctx->buffer */ |
| ctx->can_store = false; |
| } |
| ctx->zstr.next_in = (Bytef *)ctx->buffer; |
| ctx->zstr.avail_in = (uInt)n; |
| } |
| continue; |
| } |
| /* fallthrough */ |
| case Z_NEED_DICT: |
| case Z_DATA_ERROR: |
| case Z_STREAM_ERROR: |
| case Z_MEM_ERROR: |
| zip_error_set(&ctx->error, ZIP_ER_ZLIB, ret); |
| |
| end = 1; |
| break; |
| } |
| } |
| |
| if (ctx->zstr.avail_out < len) { |
| ctx->can_store = false; |
| return (zip_int64_t)(len - ctx->zstr.avail_out); |
| } |
| |
| return (zip_error_code_zip(&ctx->error) == ZIP_ER_OK) ? 0 : -1; |
| } |
| |
| |
| static zip_int64_t |
| decompress_read(zip_source_t *src, struct deflate *ctx, void *data, zip_uint64_t len) |
| { |
| int end, ret; |
| zip_int64_t n; |
| zip_uint64_t out_offset; |
| uInt out_len; |
| |
| if (zip_error_code_zip(&ctx->error) != ZIP_ER_OK) |
| return -1; |
| |
| if (len == 0) |
| return 0; |
| |
| out_offset = 0; |
| out_len = (uInt)ZIP_MIN(UINT_MAX, len); |
| ctx->zstr.next_out = (Bytef *)data; |
| ctx->zstr.avail_out = out_len; |
| |
| end = 0; |
| while (!end) { |
| ret = inflate(&ctx->zstr, Z_SYNC_FLUSH); |
| |
| switch (ret) { |
| case Z_OK: |
| if (ctx->zstr.avail_out == 0) { |
| out_offset += out_len; |
| if (out_offset < len) { |
| out_len = (uInt)ZIP_MIN(UINT_MAX, len-out_offset); |
| ctx->zstr.next_out = (Bytef *)data+out_offset; |
| ctx->zstr.avail_out = out_len; |
| } |
| else { |
| end = 1; |
| } |
| } |
| break; |
| |
| case Z_STREAM_END: |
| ctx->eof = 1; |
| end = 1; |
| break; |
| |
| case Z_BUF_ERROR: |
| if (ctx->zstr.avail_in == 0) { |
| if (ctx->eof) { |
| end = 1; |
| break; |
| } |
| |
| if ((n=zip_source_read(src, ctx->buffer, sizeof(ctx->buffer))) < 0) { |
| _zip_error_set_from_source(&ctx->error, src); |
| end = 1; |
| break; |
| } |
| else if (n == 0) { |
| ctx->eof = 1; |
| } |
| else { |
| ctx->zstr.next_in = (Bytef *)ctx->buffer; |
| ctx->zstr.avail_in = (uInt)n; |
| } |
| continue; |
| } |
| /* fallthrough */ |
| case Z_NEED_DICT: |
| case Z_DATA_ERROR: |
| case Z_STREAM_ERROR: |
| case Z_MEM_ERROR: |
| zip_error_set(&ctx->error, ZIP_ER_ZLIB, ret); |
| end = 1; |
| break; |
| } |
| } |
| |
| if (ctx->zstr.avail_out < len) |
| return (zip_int64_t)(len - ctx->zstr.avail_out); |
| |
| return (zip_error_code_zip(&ctx->error) == ZIP_ER_OK) ? 0 : -1; |
| } |
| |
| |
| static zip_int64_t |
| deflate_compress(zip_source_t *src, void *ud, void *data, zip_uint64_t len, zip_source_cmd_t cmd) |
| { |
| struct deflate *ctx; |
| int ret; |
| |
| ctx = (struct deflate *)ud; |
| |
| switch (cmd) { |
| case ZIP_SOURCE_OPEN: |
| ctx->zstr.zalloc = Z_NULL; |
| ctx->zstr.zfree = Z_NULL; |
| ctx->zstr.opaque = NULL; |
| ctx->zstr.avail_in = 0; |
| ctx->zstr.next_in = NULL; |
| ctx->zstr.avail_out = 0; |
| ctx->zstr.next_out = NULL; |
| |
| /* negative value to tell zlib not to write a header */ |
| if ((ret=deflateInit2(&ctx->zstr, Z_BEST_COMPRESSION, Z_DEFLATED, -MAX_WBITS, ctx->mem_level, Z_DEFAULT_STRATEGY)) != Z_OK) { |
| zip_error_set(&ctx->error, ZIP_ER_ZLIB, ret); |
| return -1; |
| } |
| |
| return 0; |
| |
| case ZIP_SOURCE_READ: |
| return compress_read(src, ctx, data, len); |
| |
| case ZIP_SOURCE_CLOSE: |
| deflateEnd(&ctx->zstr); |
| return 0; |
| |
| case ZIP_SOURCE_STAT: |
| { |
| zip_stat_t *st; |
| |
| st = (zip_stat_t *)data; |
| |
| st->comp_method = ctx->is_stored ? ZIP_CM_STORE : ZIP_CM_DEFLATE; |
| st->valid |= ZIP_STAT_COMP_METHOD; |
| if (ctx->eof) { |
| st->comp_size = ctx->size; |
| st->valid |= ZIP_STAT_COMP_SIZE; |
| } |
| else |
| st->valid &= ~ZIP_STAT_COMP_SIZE; |
| } |
| return 0; |
| |
| case ZIP_SOURCE_ERROR: |
| return zip_error_to_data(&ctx->error, data, len); |
| |
| case ZIP_SOURCE_FREE: |
| deflate_free(ctx); |
| return 0; |
| |
| case ZIP_SOURCE_SUPPORTS: |
| return ZIP_SOURCE_SUPPORTS_READABLE; |
| |
| default: |
| zip_error_set(&ctx->error, ZIP_ER_INTERNAL, 0); |
| return -1; |
| } |
| } |
| |
| |
| static zip_int64_t |
| deflate_decompress(zip_source_t *src, void *ud, void *data, |
| zip_uint64_t len, zip_source_cmd_t cmd) |
| { |
| struct deflate *ctx; |
| zip_int64_t n; |
| int ret; |
| |
| ctx = (struct deflate *)ud; |
| |
| switch (cmd) { |
| case ZIP_SOURCE_OPEN: |
| if ((n=zip_source_read(src, ctx->buffer, sizeof(ctx->buffer))) < 0) { |
| _zip_error_set_from_source(&ctx->error, src); |
| return -1; |
| } |
| |
| ctx->zstr.zalloc = Z_NULL; |
| ctx->zstr.zfree = Z_NULL; |
| ctx->zstr.opaque = NULL; |
| ctx->zstr.next_in = (Bytef *)ctx->buffer; |
| ctx->zstr.avail_in = (uInt)n; |
| |
| /* negative value to tell zlib that there is no header */ |
| if ((ret=inflateInit2(&ctx->zstr, -MAX_WBITS)) != Z_OK) { |
| zip_error_set(&ctx->error, ZIP_ER_ZLIB, ret); |
| return -1; |
| } |
| return 0; |
| |
| case ZIP_SOURCE_READ: |
| return decompress_read(src, ctx, data, len); |
| |
| case ZIP_SOURCE_CLOSE: |
| inflateEnd(&ctx->zstr); |
| return 0; |
| |
| case ZIP_SOURCE_STAT: |
| { |
| zip_stat_t *st; |
| |
| st = (zip_stat_t *)data; |
| |
| st->comp_method = ZIP_CM_STORE; |
| if (st->comp_size > 0 && st->size > 0) |
| st->comp_size = st->size; |
| |
| return 0; |
| } |
| |
| case ZIP_SOURCE_ERROR: |
| return zip_error_to_data(&ctx->error, data, len); |
| |
| case ZIP_SOURCE_FREE: |
| free(ctx); |
| return 0; |
| |
| case ZIP_SOURCE_SUPPORTS: |
| return zip_source_make_command_bitmap(ZIP_SOURCE_OPEN, ZIP_SOURCE_READ, ZIP_SOURCE_CLOSE, ZIP_SOURCE_STAT, ZIP_SOURCE_ERROR, ZIP_SOURCE_FREE, -1); |
| |
| default: |
| zip_error_set(&ctx->error, ZIP_ER_OPNOTSUPP, 0); |
| return -1; |
| } |
| } |
| |
| |
| static void |
| deflate_free(struct deflate *ctx) |
| { |
| free(ctx); |
| } |