1 /*
2 * Wrappers for zlib
3 * Copyright (C) 2022 Andreas Rheinhardt
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include <zlib.h>
23
24 #include "config.h"
25 #include "libavutil/error.h"
26 #include "libavutil/log.h"
27 #include "libavutil/mem.h"
28 #include "zlib_wrapper.h"
29
alloc_wrapper(void * opaque,uInt items,uInt size)30 static void *alloc_wrapper(void *opaque, uInt items, uInt size)
31 {
32 return av_malloc_array(items, size);
33 }
34
free_wrapper(void * opaque,void * ptr)35 static void free_wrapper(void *opaque, void *ptr)
36 {
37 av_free(ptr);
38 }
39
40 #if CONFIG_INFLATE_WRAPPER
ff_inflate_init(FFZStream * z,void * logctx)41 int ff_inflate_init(FFZStream *z, void *logctx)
42 {
43 z_stream *const zstream = &z->zstream;
44 int zret;
45
46 z->inited = 0;
47 zstream->next_in = Z_NULL;
48 zstream->avail_in = 0;
49 zstream->zalloc = alloc_wrapper;
50 zstream->zfree = free_wrapper;
51 zstream->opaque = Z_NULL;
52
53 zret = inflateInit(zstream);
54 if (zret == Z_OK) {
55 z->inited = 1;
56 } else {
57 av_log(logctx, AV_LOG_ERROR, "inflateInit error %d, message: %s\n",
58 zret, zstream->msg ? zstream->msg : "");
59 return AVERROR_EXTERNAL;
60 }
61 return 0;
62 }
63
ff_inflate_end(FFZStream * z)64 void ff_inflate_end(FFZStream *z)
65 {
66 if (z->inited) {
67 z->inited = 0;
68 inflateEnd(&z->zstream);
69 }
70 }
71 #endif
72
73 #if CONFIG_DEFLATE_WRAPPER
ff_deflate_init(FFZStream * z,int level,void * logctx)74 int ff_deflate_init(FFZStream *z, int level, void *logctx)
75 {
76 z_stream *const zstream = &z->zstream;
77 int zret;
78
79 z->inited = 0;
80 zstream->zalloc = alloc_wrapper;
81 zstream->zfree = free_wrapper;
82 zstream->opaque = Z_NULL;
83
84 zret = deflateInit(zstream, level);
85 if (zret == Z_OK) {
86 z->inited = 1;
87 } else {
88 av_log(logctx, AV_LOG_ERROR, "deflateInit error %d, message: %s\n",
89 zret, zstream->msg ? zstream->msg : "");
90 return AVERROR_EXTERNAL;
91 }
92 return 0;
93 }
94
ff_deflate_end(FFZStream * z)95 void ff_deflate_end(FFZStream *z)
96 {
97 if (z->inited) {
98 z->inited = 0;
99 deflateEnd(&z->zstream);
100 }
101 }
102 #endif
103