1 files changed, 134 insertions, 0 deletions
diff --git a/nxcomp/Z.cpp b/nxcomp/Z.cpp
new file mode 100644
index 000000000..e6c93cd10
--- /dev/null
+++ b/nxcomp/Z.cpp
@@ -0,0 +1,134 @@
+/**************************************************************************/
+/*                                                                        */
+/* Copyright (c) 2001, 2010 NoMachine, http://www.nomachine.com/.         */
+/*                                                                        */
+/* NXCOMP, NX protocol compression and NX extensions to this software     */
+/* are copyright of NoMachine. Redistribution and use of the present      */
+/* software is allowed according to terms specified in the file LICENSE   */
+/* which comes in the source distribution.                                */
+/*                                                                        */
+/* Check http://www.nomachine.com/licensing.html for applicability.       */
+/*                                                                        */
+/* NX and NoMachine are trademarks of Medialogic S.p.A.                   */
+/*                                                                        */
+/* All rights reserved.                                                   */
+/*                                                                        */
+/**************************************************************************/
+
+#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;
+}