/*
   Copyright (C) 2010 Red Hat, Inc.

   This library 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.

   This library 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 this library; if not, see <http://www.gnu.org/licenses/>.
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include "common.h"
#include "zlib_decoder.h"
#include "debug.h"
#include "utils.h"

static void op_decode(SpiceZlibDecoder *decoder,
                      uint8_t *data,
                      int data_size,
                      uint8_t *dest,
                      int dest_size)
{
    ZlibDecoder* _decoder = static_cast<ZlibDecoder*>(decoder);
    _decoder->decode(data, data_size, dest, dest_size);
}

ZlibDecoder::ZlibDecoder()
{
    int z_ret;

    _z_strm.zalloc = Z_NULL;
    _z_strm.zfree = Z_NULL;
    _z_strm.opaque = Z_NULL;
    _z_strm.next_in = Z_NULL;
    _z_strm.avail_in = 0;
    z_ret = inflateInit(&_z_strm);
    if (z_ret != Z_OK) {
        THROW("zlib decoder init failed, error %d", z_ret);
    }

    static SpiceZlibDecoderOps decoder_ops = {
        op_decode,
    };

    ops = &decoder_ops;
}

ZlibDecoder::~ZlibDecoder()
{
    inflateEnd(&_z_strm);
}


void ZlibDecoder::decode(uint8_t *data, int data_size, uint8_t *dest, int dest_size)
{
    int z_ret;

    inflateReset(&_z_strm);
    _z_strm.next_in = data;
    _z_strm.avail_in = data_size;
    _z_strm.next_out = dest;
    _z_strm.avail_out = dest_size;

    z_ret = inflate(&_z_strm, Z_FINISH);

    if (z_ret != Z_STREAM_END) {
        THROW("zlib inflate failed, error %d", z_ret);
    }
}