seven-zip 1.4.2 → 1.7.0.rc1

data/ext/p7zip/CPP/7zip/Compress/Lzham/lzhamcomp/lzham_lzcomp.cpp

@@ -0,0 +1,750 @@
+// File: lzham_lzcomp.cpp
+// See Copyright Notice and license at the end of include/lzham.h
+#include "lzham_core.h"
+#include "lzham.h"
+#include "lzham_comp.h"
+#include "lzham_lzcomp_internal.h"
+
+using namespace lzham;
+
+namespace lzham
+{
+   struct lzham_compress_state
+   {
+      lzham_compress_state(lzham_malloc_context malloc_context) : 
+         m_tp(malloc_context),
+         m_malloc_context(malloc_context),
+         m_compressor(malloc_context)
+      {
+      }
+
+      // task_pool requires 8 or 16 alignment
+      task_pool m_tp;
+      
+      lzham_malloc_context m_malloc_context;
+
+      lzcompressor m_compressor;
+
+      uint m_dict_size_log2;
+
+      const uint8 *m_pIn_buf;
+      size_t *m_pIn_buf_size;
+      uint8 *m_pOut_buf;
+      size_t *m_pOut_buf_size;
+
+      size_t m_comp_data_ofs;
+
+      bool m_finished_compression;
+
+      lzham_compress_params m_params;
+
+      lzham_compress_status_t m_status;
+   };
+   
+   static lzham_compress_status_t create_internal_init_params(lzcompressor::init_params &internal_params, const lzham_compress_params *pParams)
+   {
+      if ((pParams->m_dict_size_log2 < CLZBase::cMinDictSizeLog2) || (pParams->m_dict_size_log2 > CLZBase::cMaxDictSizeLog2))
+      {
+         LZHAM_LOG_ERROR(6000);
+         return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+      }
+      
+      if (pParams->m_extreme_parsing_max_best_arrivals > cMaxParseNodeStates)
+      {
+         LZHAM_LOG_ERROR(6001);
+         return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+      }
+      
+      if (pParams->m_extreme_parsing_max_best_arrivals <= 1)
+         internal_params.m_extreme_parsing_max_best_arrivals = cDefaultMaxParseNodeStates;
+      else
+         internal_params.m_extreme_parsing_max_best_arrivals = pParams->m_extreme_parsing_max_best_arrivals;			     
+
+      if (pParams->m_fast_bytes > 0)
+         internal_params.m_fast_bytes_override = math::clamp<uint>(pParams->m_fast_bytes, LZHAM_MIN_FAST_BYTES, LZHAM_MAX_FAST_BYTES);
+
+      internal_params.m_dict_size_log2 = pParams->m_dict_size_log2;
+      
+      if (pParams->m_max_helper_threads < 0)
+         internal_params.m_max_helper_threads = lzham_get_max_helper_threads();
+      else
+         internal_params.m_max_helper_threads = pParams->m_max_helper_threads;
+      internal_params.m_max_helper_threads = LZHAM_MIN(LZHAM_MAX_HELPER_THREADS, internal_params.m_max_helper_threads);
+
+      internal_params.m_lzham_compress_flags = pParams->m_compress_flags;
+
+      if (pParams->m_num_seed_bytes)
+      {
+         if ((!pParams->m_pSeed_bytes) || (pParams->m_num_seed_bytes > (1U << pParams->m_dict_size_log2)))
+         {
+            LZHAM_LOG_ERROR(6002);
+            return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+         }
+
+         internal_params.m_num_seed_bytes = pParams->m_num_seed_bytes;
+         internal_params.m_pSeed_bytes = pParams->m_pSeed_bytes;
+      }
+
+      switch (pParams->m_level)
+      {
+         case LZHAM_COMP_LEVEL_FASTEST:   internal_params.m_compression_level = cCompressionLevelFastest; break;
+         case LZHAM_COMP_LEVEL_FASTER:    internal_params.m_compression_level = cCompressionLevelFaster; break;
+         case LZHAM_COMP_LEVEL_DEFAULT:   internal_params.m_compression_level = cCompressionLevelDefault; break;
+         case LZHAM_COMP_LEVEL_BETTER:    internal_params.m_compression_level = cCompressionLevelBetter; break;
+         case LZHAM_COMP_LEVEL_UBER:      internal_params.m_compression_level = cCompressionLevelUber; break;
+         default:
+            LZHAM_LOG_ERROR(6003);
+            return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+      };
+
+		if (pParams->m_table_max_update_interval || pParams->m_table_update_interval_slow_rate)
+		{
+			internal_params.m_table_max_update_interval = pParams->m_table_max_update_interval;
+			internal_params.m_table_update_interval_slow_rate = pParams->m_table_update_interval_slow_rate;
+		}
+		else 
+		{
+			uint rate = pParams->m_table_update_rate;
+			if (!rate)
+				rate = LZHAM_DEFAULT_TABLE_UPDATE_RATE;
+			rate = math::clamp<uint>(rate, 1, LZHAM_FASTEST_TABLE_UPDATE_RATE) - 1;
+			internal_params.m_table_max_update_interval = g_table_update_settings[rate].m_max_update_interval;
+			internal_params.m_table_update_interval_slow_rate = g_table_update_settings[rate].m_slow_rate;
+		}
+
+      return LZHAM_COMP_STATUS_SUCCESS;
+   }
+
+   lzham_compress_state_ptr LZHAM_CDECL lzham_lib_compress_init(const lzham_compress_params *pParams)
+   {
+      if ((!pParams) || (pParams->m_struct_size != sizeof(lzham_compress_params)))
+      {
+         LZHAM_LOG_ERROR(6004);
+         return NULL;
+      }
+
+      if ((pParams->m_dict_size_log2 < CLZBase::cMinDictSizeLog2) || (pParams->m_dict_size_log2 > CLZBase::cMaxDictSizeLog2))
+      {
+         LZHAM_LOG_ERROR(6005);
+         return NULL;
+      }
+
+      lzcompressor::init_params internal_params;
+      lzham_compress_status_t status = create_internal_init_params(internal_params, pParams);
+      if (status != LZHAM_COMP_STATUS_SUCCESS)
+      {
+         LZHAM_LOG_ERROR(6006);
+         return NULL;
+      }
+
+      lzham_malloc_context malloc_context = lzham_create_malloc_context(0);
+
+      lzham_compress_state *pState = lzham_new<lzham_compress_state>(malloc_context, malloc_context);
+      if (!pState)
+      {
+         lzham_destroy_malloc_context(malloc_context);
+         LZHAM_LOG_ERROR(6007);
+         return NULL;
+      }
+
+      pState->m_params = *pParams;
+
+      pState->m_pIn_buf = NULL;
+      pState->m_pIn_buf_size = NULL;
+      pState->m_pOut_buf = NULL;
+      pState->m_pOut_buf_size = NULL;
+      pState->m_status = LZHAM_COMP_STATUS_NOT_FINISHED;
+      pState->m_comp_data_ofs = 0;
+      pState->m_finished_compression = false;
+
+      if (internal_params.m_max_helper_threads)
+      {
+         if (!pState->m_tp.init(internal_params.m_max_helper_threads))
+         {
+            lzham_delete(malloc_context, pState);
+            lzham_destroy_malloc_context(malloc_context);
+            LZHAM_LOG_ERROR(6008);
+            return NULL;
+         }
+         if (pState->m_tp.get_num_threads() >= internal_params.m_max_helper_threads)
+         {
+            internal_params.m_pTask_pool = &pState->m_tp;
+         }
+         else
+         {
+            internal_params.m_max_helper_threads = 0;
+         }
+      }
+
+      if (!pState->m_compressor.init(internal_params))
+      {
+         lzham_delete(malloc_context, pState);
+         lzham_destroy_malloc_context(malloc_context);
+         LZHAM_LOG_ERROR(6009);
+         return NULL;
+      }
+
+      return pState;
+   }
+
+   lzham_compress_state_ptr LZHAM_CDECL lzham_lib_compress_reinit(lzham_compress_state_ptr p)
+   {
+      lzham_compress_state *pState = static_cast<lzham_compress_state*>(p);
+      if (pState)
+      {
+         if (!pState->m_compressor.reset())
+         {
+            LZHAM_LOG_ERROR(6010);
+            return NULL;
+         }
+
+         pState->m_pIn_buf = NULL;
+         pState->m_pIn_buf_size = NULL;
+         pState->m_pOut_buf = NULL;
+         pState->m_pOut_buf_size = NULL;
+         pState->m_status = LZHAM_COMP_STATUS_NOT_FINISHED;
+         pState->m_comp_data_ofs = 0;
+         pState->m_finished_compression = false;
+      }
+
+      return pState;
+   }
+
+   lzham_uint32 LZHAM_CDECL lzham_lib_compress_deinit(lzham_compress_state_ptr p)
+   {
+      lzham_compress_state *pState = static_cast<lzham_compress_state *>(p);
+      if (!pState)
+      {
+         LZHAM_LOG_ERROR(6011);
+         return 0;
+      }
+
+      uint32 adler32 = pState->m_compressor.get_src_adler32();
+
+      lzham_malloc_context malloc_context = pState->m_malloc_context;
+
+      lzham_delete(malloc_context, pState);
+      lzham_destroy_malloc_context(malloc_context);
+      
+      return adler32;
+   }
+
+   lzham_compress_status_t LZHAM_CDECL lzham_lib_compress(
+      lzham_compress_state_ptr p,
+      const lzham_uint8 *pIn_buf, size_t *pIn_buf_size,
+      lzham_uint8 *pOut_buf, size_t *pOut_buf_size,
+      lzham_bool no_more_input_bytes_flag)
+   {
+      return lzham_lib_compress2(p, pIn_buf, pIn_buf_size, pOut_buf, pOut_buf_size, no_more_input_bytes_flag ? LZHAM_FINISH : LZHAM_NO_FLUSH);
+   }
+
+   lzham_compress_status_t LZHAM_CDECL lzham_lib_compress2(
+      lzham_compress_state_ptr p,
+      const lzham_uint8 *pIn_buf, size_t *pIn_buf_size,
+      lzham_uint8 *pOut_buf, size_t *pOut_buf_size,
+      lzham_flush_t flush_type)
+   {
+      lzham_compress_state *pState = static_cast<lzham_compress_state*>(p);
+
+      if ((!pState) || (!pState->m_params.m_dict_size_log2) || (pState->m_status >= LZHAM_COMP_STATUS_FIRST_SUCCESS_OR_FAILURE_CODE) || (!pIn_buf_size) || (!pOut_buf_size))
+      {
+         LZHAM_LOG_ERROR(6012);
+         return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+      }
+
+      if ((*pIn_buf_size) && (!pIn_buf))
+      {
+         LZHAM_LOG_ERROR(6013);
+         return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+      }
+
+      if ((!*pOut_buf_size) || (!pOut_buf))
+      {
+         LZHAM_LOG_ERROR(6014);
+         return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+      }
+
+      byte_vec &comp_data = pState->m_compressor.get_compressed_data();
+      size_t num_bytes_written_to_out_buf = 0;
+      if (pState->m_comp_data_ofs < comp_data.size())
+      {
+         size_t n = LZHAM_MIN(comp_data.size() - pState->m_comp_data_ofs, *pOut_buf_size);
+
+         memcpy(pOut_buf, comp_data.get_ptr() + pState->m_comp_data_ofs, n);
+
+         pState->m_comp_data_ofs += n;
+
+         const bool has_no_more_output = (pState->m_comp_data_ofs >= comp_data.size());
+         if (has_no_more_output)
+         {
+            pOut_buf += n;
+            *pOut_buf_size -= n;
+            num_bytes_written_to_out_buf += n;
+         }
+         else
+         {
+            *pIn_buf_size = 0;
+            *pOut_buf_size = n;
+            pState->m_status = LZHAM_COMP_STATUS_HAS_MORE_OUTPUT;
+            return pState->m_status;
+         }
+      }
+
+      comp_data.try_resize(0);
+      pState->m_comp_data_ofs = 0;
+
+      if (pState->m_finished_compression)
+      {
+         if ((*pIn_buf_size) || (flush_type != LZHAM_FINISH))
+         {
+            pState->m_status = LZHAM_COMP_STATUS_INVALID_PARAMETER;
+            LZHAM_LOG_ERROR(6015);
+            return pState->m_status;
+         }
+
+         *pIn_buf_size = 0;
+         *pOut_buf_size = num_bytes_written_to_out_buf;
+
+         pState->m_status = LZHAM_COMP_STATUS_SUCCESS;
+         return pState->m_status;
+      }
+
+      const size_t cMaxBytesToPutPerIteration = 4*1024*1024;
+      size_t bytes_to_put = LZHAM_MIN(cMaxBytesToPutPerIteration, *pIn_buf_size);
+      const bool consumed_entire_input_buf = (bytes_to_put == *pIn_buf_size);
+
+      if (bytes_to_put)
+      {
+         if (!pState->m_compressor.put_bytes(pIn_buf, (uint)bytes_to_put))
+         {
+            *pIn_buf_size = 0;
+            *pOut_buf_size = num_bytes_written_to_out_buf;
+            pState->m_status = LZHAM_COMP_STATUS_FAILED;
+            LZHAM_LOG_ERROR(6016);
+            return pState->m_status;
+         }
+      }
+
+      if ((consumed_entire_input_buf) && (flush_type != LZHAM_NO_FLUSH))
+      {
+         if ((flush_type == LZHAM_SYNC_FLUSH) || (flush_type == LZHAM_FULL_FLUSH) || (flush_type == LZHAM_TABLE_FLUSH))
+         {
+            if (!pState->m_compressor.flush(flush_type))
+            {
+               *pIn_buf_size = 0;
+               *pOut_buf_size = num_bytes_written_to_out_buf;
+               pState->m_status = LZHAM_COMP_STATUS_FAILED;
+               LZHAM_LOG_ERROR(6017);
+               return pState->m_status;
+            }
+         }
+         else if (!pState->m_finished_compression)
+         {
+            if (!pState->m_compressor.put_bytes(NULL, 0))
+            {
+               *pIn_buf_size = 0;
+               *pOut_buf_size = num_bytes_written_to_out_buf;
+               pState->m_status = LZHAM_COMP_STATUS_FAILED;
+               LZHAM_LOG_ERROR(6018);
+               return pState->m_status;
+            }
+            pState->m_finished_compression = true;
+         }
+      }
+
+      size_t num_comp_bytes_to_output = LZHAM_MIN(comp_data.size() - pState->m_comp_data_ofs, *pOut_buf_size);
+      if (num_comp_bytes_to_output)
+      {
+         memcpy(pOut_buf, comp_data.get_ptr() + pState->m_comp_data_ofs, num_comp_bytes_to_output);
+
+         pState->m_comp_data_ofs += num_comp_bytes_to_output;
+      }
+
+      *pIn_buf_size = bytes_to_put;
+      *pOut_buf_size = num_bytes_written_to_out_buf + num_comp_bytes_to_output;
+
+      const bool has_no_more_output = (pState->m_comp_data_ofs >= comp_data.size());
+      if ((has_no_more_output) && (flush_type == LZHAM_FINISH) && (pState->m_finished_compression))
+         pState->m_status = LZHAM_COMP_STATUS_SUCCESS;
+      else if ((has_no_more_output) && (consumed_entire_input_buf) && (flush_type == LZHAM_NO_FLUSH))
+         pState->m_status = LZHAM_COMP_STATUS_NEEDS_MORE_INPUT;
+      else
+         pState->m_status = has_no_more_output ? LZHAM_COMP_STATUS_NOT_FINISHED : LZHAM_COMP_STATUS_HAS_MORE_OUTPUT;
+
+      return pState->m_status;
+   }
+
+   lzham_compress_status_t LZHAM_CDECL lzham_lib_compress_memory(const lzham_compress_params *pParams, lzham_uint8* pDst_buf, size_t *pDst_len, const lzham_uint8* pSrc_buf, size_t src_len, lzham_uint32 *pAdler32)
+   {
+      if ((!pParams) || (!pDst_len))
+      {
+         LZHAM_LOG_ERROR(6019);
+         return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+      }
+
+      if (src_len)
+      {
+         if (!pSrc_buf)
+         {
+            LZHAM_LOG_ERROR(6020);
+            return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+         }
+      }
+
+      if (sizeof(size_t) > sizeof(uint32))
+      {
+         if (src_len > cUINT32_MAX)
+         {
+            LZHAM_LOG_ERROR(6021);
+            return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+         }
+      }
+
+      lzcompressor::init_params internal_params;
+      lzham_compress_status_t status = create_internal_init_params(internal_params, pParams);
+      if (status != LZHAM_COMP_STATUS_SUCCESS)
+      {
+         LZHAM_LOG_ERROR(6022);
+         return status;
+      }
+
+      lzham_malloc_context malloc_context = lzham_create_malloc_context(0);
+
+      task_pool *pTP = NULL;
+      if (internal_params.m_max_helper_threads)
+      {
+         pTP = lzham_new<task_pool>(malloc_context, malloc_context);
+         if (!pTP->init(internal_params.m_max_helper_threads))
+         {
+            lzham_destroy_malloc_context(malloc_context);
+            LZHAM_LOG_ERROR(6023);
+            return LZHAM_COMP_STATUS_FAILED_INITIALIZING;
+         }
+
+         internal_params.m_pTask_pool = pTP;
+      }
+
+      lzcompressor *pCompressor = lzham_new<lzcompressor>(malloc_context, malloc_context);
+      if (!pCompressor)
+      {
+         lzham_delete(malloc_context, pTP);
+         lzham_destroy_malloc_context(malloc_context);
+         LZHAM_LOG_ERROR(6024);
+         return LZHAM_COMP_STATUS_FAILED_INITIALIZING;
+      }
+
+      if (!pCompressor->init(internal_params))
+      {
+         lzham_delete(malloc_context, pTP);
+         lzham_delete(malloc_context, pCompressor);
+         lzham_destroy_malloc_context(malloc_context);
+         LZHAM_LOG_ERROR(6025);
+         return LZHAM_COMP_STATUS_INVALID_PARAMETER;
+      }
+
+      if (src_len)
+      {
+         if (!pCompressor->put_bytes(pSrc_buf, static_cast<uint32>(src_len)))
+         {
+            *pDst_len = 0;
+            lzham_delete(malloc_context, pTP);
+            lzham_delete(malloc_context, pCompressor);
+            lzham_destroy_malloc_context(malloc_context);
+            LZHAM_LOG_ERROR(6026);
+            return LZHAM_COMP_STATUS_FAILED;
+         }
+      }
+
+      if (!pCompressor->put_bytes(NULL, 0))
+      {
+         *pDst_len = 0;
+         lzham_delete(malloc_context, pTP);
+         lzham_delete(malloc_context, pCompressor);
+         lzham_destroy_malloc_context(malloc_context);
+         LZHAM_LOG_ERROR(6027);
+         return LZHAM_COMP_STATUS_FAILED;
+      }
+
+      const byte_vec &comp_data = pCompressor->get_compressed_data();
+
+      size_t dst_buf_size = *pDst_len;
+      *pDst_len = comp_data.size();
+
+      if (pAdler32)
+         *pAdler32 = pCompressor->get_src_adler32();
+
+      if (comp_data.size() > dst_buf_size)
+      {
+         lzham_delete(malloc_context, pTP);
+         lzham_delete(malloc_context, pCompressor);
+         lzham_destroy_malloc_context(malloc_context);
+         LZHAM_LOG_ERROR(6028);
+         return LZHAM_COMP_STATUS_OUTPUT_BUF_TOO_SMALL;
+      }
+
+      memcpy(pDst_buf, comp_data.get_ptr(), comp_data.size());
+
+      lzham_delete(malloc_context, pTP);
+      lzham_delete(malloc_context, pCompressor);
+      lzham_destroy_malloc_context(malloc_context);
+      return LZHAM_COMP_STATUS_SUCCESS;
+   }
+
+   // ----------------- zlib-style API's
+
+   int lzham_lib_z_deflateInit(lzham_z_streamp pStream, int level)
+   {
+      return lzham_lib_z_deflateInit2(pStream, level, LZHAM_Z_LZHAM, LZHAM_Z_DEFAULT_WINDOW_BITS, 9, LZHAM_Z_DEFAULT_STRATEGY);
+   }
+
+   int lzham_lib_z_deflateInit2(lzham_z_streamp pStream, int level, int method, int window_bits, int mem_level, int strategy)
+   {
+      LZHAM_NOTE_UNUSED(strategy);
+
+      if (!pStream)
+      {
+         LZHAM_LOG_ERROR(6029);
+         return LZHAM_Z_STREAM_ERROR;
+      }
+      if ((mem_level < 1) || (mem_level > 9))
+      {
+         LZHAM_LOG_ERROR(6030);
+         return LZHAM_Z_PARAM_ERROR;
+      }
+      if ((method != LZHAM_Z_DEFLATED) && (method != LZHAM_Z_LZHAM))
+      {
+         LZHAM_LOG_ERROR(6031);
+         return LZHAM_Z_PARAM_ERROR;
+      }
+
+      if (level == LZHAM_Z_DEFAULT_COMPRESSION)
+         level = 9;
+
+      if (method == LZHAM_Z_DEFLATED)
+      {
+         // Force Deflate to LZHAM with default window_bits.
+         method = LZHAM_Z_LZHAM;
+         window_bits = LZHAM_Z_DEFAULT_WINDOW_BITS;
+      }
+
+#ifdef LZHAM_Z_API_FORCE_WINDOW_BITS
+      window_bits = LZHAM_Z_API_FORCE_WINDOW_BITS;
+#endif
+
+      int max_window_bits = LZHAM_64BIT_POINTERS ? LZHAM_MAX_DICT_SIZE_LOG2_X64 : LZHAM_MAX_DICT_SIZE_LOG2_X86;
+      if ((labs(window_bits) < LZHAM_MIN_DICT_SIZE_LOG2) || (labs(window_bits) > max_window_bits))
+      {
+         LZHAM_LOG_ERROR(6032);
+         return LZHAM_Z_PARAM_ERROR;
+      }
+
+      lzham_compress_params comp_params;
+
+      utils::zero_object(comp_params);
+      comp_params.m_struct_size = sizeof(lzham_compress_params);
+
+      comp_params.m_level = LZHAM_COMP_LEVEL_UBER;
+      if (level <= 1)
+         comp_params.m_level = LZHAM_COMP_LEVEL_FASTEST;
+      else if (level <= 3)
+         comp_params.m_level = LZHAM_COMP_LEVEL_FASTER;
+      else if (level <= 5)
+         comp_params.m_level = LZHAM_COMP_LEVEL_DEFAULT;
+      else if (level <= 7)
+         comp_params.m_level = LZHAM_COMP_LEVEL_BETTER;
+
+      if (level == 10)
+         comp_params.m_compress_flags |= LZHAM_COMP_FLAG_EXTREME_PARSING;
+
+      // Use all CPU's. TODO: This is not always the best idea depending on the dictionary size and the # of bytes to compress.
+      comp_params.m_max_helper_threads = -1;
+
+      comp_params.m_dict_size_log2 = static_cast<lzham_uint32>(labs(window_bits));
+
+      if (window_bits > 0)
+         comp_params.m_compress_flags |= LZHAM_COMP_FLAG_WRITE_ZLIB_STREAM;
+
+      pStream->data_type = 0;
+      pStream->adler = LZHAM_Z_ADLER32_INIT;
+      pStream->msg = NULL;
+      pStream->reserved = 0;
+      pStream->total_in = 0;
+      pStream->total_out = 0;
+
+      lzham_compress_state_ptr pComp = lzham_lib_compress_init(&comp_params);
+      if (!pComp)
+      {
+         LZHAM_LOG_ERROR(6033);
+         return LZHAM_Z_PARAM_ERROR;
+      }
+
+      pStream->state = (struct lzham_z_internal_state *)pComp;
+
+      return LZHAM_Z_OK;
+   }
+
+   int lzham_lib_z_deflateReset(lzham_z_streamp pStream)
+   {
+      if (!pStream)
+      {
+         LZHAM_LOG_ERROR(6034);
+         return LZHAM_Z_STREAM_ERROR;
+      }
+
+      lzham_compress_state_ptr pComp = (lzham_compress_state_ptr)pStream->state;
+      if (!pComp)
+      {
+         LZHAM_LOG_ERROR(6035);
+         return LZHAM_Z_STREAM_ERROR;
+      }
+
+      pComp = lzham_lib_compress_reinit(pComp);
+      if (!pComp)
+      {
+         LZHAM_LOG_ERROR(6036);
+         return LZHAM_Z_STREAM_ERROR;
+      }
+
+      pStream->state = (struct lzham_z_internal_state *)pComp;
+
+      return LZHAM_Z_OK;
+   }
+
+   int lzham_lib_z_deflate(lzham_z_streamp pStream, int flush)
+   {
+      if ((!pStream) || (!pStream->state) || (flush < 0) || (flush > LZHAM_Z_FINISH) || (!pStream->next_out))
+      {
+         LZHAM_LOG_ERROR(6037);
+         return LZHAM_Z_STREAM_ERROR;
+      }
+
+      if (!pStream->avail_out)
+      {
+         LZHAM_LOG_ERROR(6038);
+         return LZHAM_Z_BUF_ERROR;
+      }
+
+      if (flush == LZHAM_Z_PARTIAL_FLUSH)
+         flush = LZHAM_Z_SYNC_FLUSH;
+
+      int lzham_status = LZHAM_Z_OK;
+      lzham_z_ulong orig_total_in = pStream->total_in, orig_total_out = pStream->total_out;
+      for ( ; ; )
+      {
+         size_t in_bytes = pStream->avail_in, out_bytes = pStream->avail_out;
+
+         lzham_compress_state_ptr pComp = (lzham_compress_state_ptr)pStream->state;
+         lzham_compress_state *pState = static_cast<lzham_compress_state*>(pComp);
+
+         lzham_compress_status_t status = lzham_lib_compress2(
+            pComp,
+            pStream->next_in, &in_bytes,
+            pStream->next_out, &out_bytes,
+            (lzham_flush_t)flush);
+
+         pStream->next_in += (uint)in_bytes;
+         pStream->avail_in -= (uint)in_bytes;
+         pStream->total_in += (uint)in_bytes;
+
+         pStream->next_out += (uint)out_bytes;
+         pStream->avail_out -= (uint)out_bytes;
+         pStream->total_out += (uint)out_bytes;
+
+         pStream->adler = pState->m_compressor.get_src_adler32();
+
+         if (status >= LZHAM_COMP_STATUS_FIRST_FAILURE_CODE)
+         {
+            lzham_status = LZHAM_Z_STREAM_ERROR;
+            LZHAM_LOG_ERROR(6039);
+            break;
+         }
+         else if (status == LZHAM_COMP_STATUS_SUCCESS)
+         {
+            lzham_status = LZHAM_Z_STREAM_END;
+            break;
+         }
+         else if (!pStream->avail_out)
+            break;
+         else if ((!pStream->avail_in) && (flush != LZHAM_Z_FINISH))
+         {
+            if ((flush) || (pStream->total_in != orig_total_in) || (pStream->total_out != orig_total_out))
+               break;
+            LZHAM_LOG_ERROR(6040);
+            return LZHAM_Z_BUF_ERROR; // Can't make forward progress without some input.
+         }
+      }
+      return lzham_status;
+   }
+
+   int lzham_lib_z_deflateEnd(lzham_z_streamp pStream)
+   {
+      if (!pStream)
+      {
+         LZHAM_LOG_ERROR(6041);
+         return LZHAM_Z_STREAM_ERROR;
+      }
+
+      lzham_compress_state_ptr pComp = (lzham_compress_state_ptr)pStream->state;
+      if (pComp)
+      {
+         pStream->adler = lzham_lib_compress_deinit(pComp);
+
+         pStream->state = NULL;
+      }
+
+      return LZHAM_Z_OK;
+   }
+
+   lzham_z_ulong lzham_lib_z_deflateBound(lzham_z_streamp pStream, lzham_z_ulong source_len)
+   {
+      LZHAM_NOTE_UNUSED(pStream);
+      return 64 + source_len + ((source_len + 4095) / 4096) * 4;
+   }
+
+   int lzham_lib_z_compress2(unsigned char *pDest, lzham_z_ulong *pDest_len, const unsigned char *pSource, lzham_z_ulong source_len, int level)
+   {
+      int status;
+      lzham_z_stream stream;
+      memset(&stream, 0, sizeof(stream));
+
+      // In case lzham_z_ulong is 64-bits (argh I hate longs).
+      if ((source_len | *pDest_len) > 0xFFFFFFFFU)
+      {
+         LZHAM_LOG_ERROR(6042);
+         return LZHAM_Z_PARAM_ERROR;
+      }
+
+      stream.next_in = pSource;
+      stream.avail_in = (uint)source_len;
+      stream.next_out = pDest;
+      stream.avail_out = (uint)*pDest_len;
+
+      status = lzham_lib_z_deflateInit(&stream, level);
+      if (status != LZHAM_Z_OK)
+      {
+         LZHAM_LOG_ERROR(6043);
+         return status;
+      }
+
+      status = lzham_lib_z_deflate(&stream, LZHAM_Z_FINISH);
+      if (status != LZHAM_Z_STREAM_END)
+      {
+         lzham_lib_z_deflateEnd(&stream);
+         return (status == LZHAM_Z_OK) ? LZHAM_Z_BUF_ERROR : status;
+      }
+
+      *pDest_len = stream.total_out;
+      return lzham_lib_z_deflateEnd(&stream);
+   }
+
+   int lzham_lib_z_compress(unsigned char *pDest, lzham_z_ulong *pDest_len, const unsigned char *pSource, lzham_z_ulong source_len)
+   {
+      return lzham_lib_z_compress2(pDest, pDest_len, pSource, source_len, (int)LZHAM_Z_DEFAULT_COMPRESSION);
+   }
+
+   lzham_z_ulong lzham_lib_z_compressBound(lzham_z_ulong source_len)
+   {
+      return lzham_lib_z_deflateBound(NULL, source_len);
+   }
+
+} // namespace lzham