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
30static void *alloc_wrapper(void *opaque, uInt items, uInt size)
31{
32    return av_malloc_array(items, size);
33}
34
35static void free_wrapper(void *opaque, void *ptr)
36{
37    av_free(ptr);
38}
39
40#if CONFIG_INFLATE_WRAPPER
41int 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
64void 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
74int 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
95void ff_deflate_end(FFZStream *z)
96{
97    if (z->inited) {
98        z->inited = 0;
99        deflateEnd(&z->zstream);
100    }
101}
102#endif
103