/*

 * Wrappers for zlib

 * Copyright (C) 2022 Andreas Rheinhardt

 *

 * This file is part of FFmpeg.

 *

 * FFmpeg is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public

 * License as published by the Free Software Foundation; either

 * version 2.1 of the License, or (at your option) any later version.

 *

 * FFmpeg is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with FFmpeg; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

 */

#include <zlib.h>

#include "config.h"

#include "libavutil/error.h"

#include "libavutil/log.h"

#include "libavutil/mem.h"

#include "zlib_wrapper.h"

static void *alloc_wrapper(void *opaque, uInt items, uInt size)

{

    return av_malloc_array(items, size);

}

static void free_wrapper(void *opaque, void *ptr)

{

    av_free(ptr);

}

#if CONFIG_INFLATE_WRAPPER

int ff_inflate_init(FFZStream *z, void *logctx)

{

    z_stream *const zstream = &z->zstream;

    int zret;

    z->inited = 0;

    zstream->next_in  = Z_NULL;

    zstream->avail_in = 0;

    zstream->zalloc   = alloc_wrapper;

    zstream->zfree    = free_wrapper;

    zstream->opaque   = Z_NULL;

    zret = inflateInit(zstream);

    if (zret == Z_OK) {

        z->inited = 1;

    } else {

        av_log(logctx, AV_LOG_ERROR, "inflateInit error %d, message: %s\n",

               zret, zstream->msg ? zstream->msg : "");

        return AVERROR_EXTERNAL;

    }

    return 0;

}

void ff_inflate_end(FFZStream *z)

{

    if (z->inited) {

        z->inited = 0;

        inflateEnd(&z->zstream);

    }

}

#endif

#if CONFIG_DEFLATE_WRAPPER

int ff_deflate_init(FFZStream *z, int level, void *logctx)

{

    z_stream *const zstream = &z->zstream;

    int zret;

    z->inited = 0;

    zstream->zalloc = alloc_wrapper;

    zstream->zfree  = free_wrapper;

    zstream->opaque = Z_NULL;

    zret = deflateInit(zstream, level);

    if (zret == Z_OK) {

        z->inited = 1;

    } else {

        av_log(logctx, AV_LOG_ERROR, "deflateInit error %d, message: %s\n",

               zret, zstream->msg ? zstream->msg : "");

        return AVERROR_EXTERNAL;

    }

    return 0;

}

void ff_deflate_end(FFZStream *z)

{

    if (z->inited) {

        z->inited = 0;

        deflateEnd(&z->zstream);

    }

}

#endif