diff options
Diffstat (limited to 'src/zlib.c')
| -rw-r--r-- | src/zlib.c | 28 |
1 files changed, 25 insertions, 3 deletions
@@ -1,5 +1,5 @@ /* - * Copyright (c) 2020, yzrh <yzrh@noema.org> + * Copyright (c) 2020-2021, yzrh <yzrh@noema.org> * * SPDX-License-Identifier: Apache-2.0 */ @@ -20,12 +20,34 @@ strinflate(char **dst, int dst_size, unsigned long size = dst_size; - uncompress((Bytef *) *dst, &size, (const Bytef *) src, src_size); + if (uncompress((Bytef *) *dst, + &size, (const Bytef *) src, src_size) != Z_OK) { + free(*dst); + return 1; + } + + return 0; +} - if (size != dst_size) { +int +strdeflate(char **dst, int *dst_size, + const char * restrict src, int src_size) +{ + *dst_size = compressBound(src_size); + *dst = malloc(*dst_size); + + if (*dst == NULL) + return 1; + + unsigned long size = *dst_size; + + if (compress((Bytef *) *dst, &size, + (const Bytef *) src, src_size) != Z_OK) { free(*dst); return 1; } + *dst_size = size; + return 0; } |