/**************************************************************************/
/*                                                                        */
/* Copyright (c) 2001, 2011 NoMachine (http://www.nomachine.com)          */
/* Copyright (c) 2008-2014 Oleksandr Shneyder <o.shneyder@phoca-gmbh.de>  */
/* Copyright (c) 2014-2016 Ulrich Sibiller <uli42@gmx.de>                 */
/* Copyright (c) 2014-2016 Mihai Moldovan <ionic@ionic.de>                */
/* Copyright (c) 2011-2016 Mike Gabriel <mike.gabriel@das-netzwerkteam.de>*/
/* Copyright (c) 2015-2016 Qindel Group (http://www.qindel.com)           */
/*                                                                        */
/* NXCOMP, NX protocol compression and NX extensions to this software     */
/* are copyright of the aforementioned persons and companies.             */
/*                                                                        */
/* Redistribution and use of the present software is allowed according    */
/* to terms specified in the file LICENSE.nxcomp which comes in the       */
/* source distribution.                                                   */
/*                                                                        */
/* All rights reserved.                                                   */
/*                                                                        */
/* NOTE: This software has received contributions from various other      */
/* contributors, only the core maintainers and supporters are listed as   */
/* copyright holders. Please contact us, if you feel you should be listed */
/* as copyright holder, as well.                                          */
/*                                                                        */
/**************************************************************************/

#include "Z.h"
#include "Misc.h"

int ZCompress(z_stream *stream, unsigned char *dest, unsigned int *destLen,
                  const unsigned char *source, unsigned int sourceLen)
{
  //
  // Deal with the possible overflow.
  //

  if (stream -> total_out & 0x80000000)
  {
    #ifdef TEST
    *logofs << "ZCompress: Reset stream counters with "
            << "total in " << stream -> total_in
            << " and total out " << stream -> total_out
            << ".\n" << logofs_flush;
    #endif

    stream -> total_in  = 0;
    stream -> total_out = 0;
  }

  unsigned int saveOut = stream -> total_out;

  stream -> next_in  = (Bytef *) source;
  stream -> avail_in = sourceLen;

  //
  // Check if the source is bigger than
  // 64K on 16-bit machine.
  //

  #ifdef MAXSEG_64K

  if ((uLong) stream -> avail_in != sourceLen) return Z_BUF_ERROR;

  #endif

  stream -> next_out  = dest;
  stream -> avail_out = *destLen;

  #ifdef MAXSEG_64K

  if ((uLong) stream -> avail_out != *destLen) return Z_BUF_ERROR;

  #endif

  int result = deflate(stream, Z_FINISH);

  if (result != Z_STREAM_END)
  {
    deflateReset(stream);

    return (result == Z_OK ? Z_BUF_ERROR : result);
  }

  *destLen = stream -> total_out - saveOut;

  result = deflateReset(stream);

  return result;
}

int ZDecompress(z_stream *stream, unsigned char *dest, unsigned int *destLen,
                    const unsigned char *source, unsigned int sourceLen)
{
  stream -> next_in  = (Bytef *) source;
  stream -> avail_in = sourceLen;

  //
  // Deal with the possible overflow.
  //

  if (stream -> total_out & 0x80000000)
  {
    #ifdef TEST
    *logofs << "ZDecompress: Reset stream counters with "
            << "total in " << stream -> total_in
            << " and total out " << stream -> total_out
            << ".\n" << logofs_flush;
    #endif

    stream -> total_in  = 0;
    stream -> total_out = 0;
  }

  unsigned int saveOut = stream -> total_out;

  if (stream -> avail_in != sourceLen)
  {
    return Z_BUF_ERROR;
  }

  stream -> next_out  = dest;
  stream -> avail_out = *destLen;

  if (stream -> avail_out != *destLen)
  {
    return Z_BUF_ERROR;
  }

  int result = inflate(stream, Z_FINISH);

  if (result != Z_STREAM_END)
  {
    inflateReset(stream);

    return (result == Z_OK ? Z_BUF_ERROR : result);
  }

  *destLen = stream -> total_out - saveOut;

  result = inflateReset(stream);

  return result;
}