/*
 * PtokaX - hub server for Direct Connect peer to peer network.

 * Copyright (C) 2004-2022  Petr Kozelka, PPK at PtokaX dot org

 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3
 * as published by the Free Software Foundation.

 * This program 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 General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

//---------------------------------------------------------------------------
#include "stdinc.h"
//---------------------------------------------------------------------------
#include "ZlibUtility.h"
//---------------------------------------------------------------------------
#include "ServerManager.h"
#include "utility.h"
#include "GlobalDataQueue.h"

//---------------------------------------------------------------------------
#ifdef _WIN32
#pragma hdrstop
#endif
#include <zlib.h>
//---------------------------------------------------------------------------
static const uint32_t ZBUFFERLEN = PTOKAX_GLOBAL_BUFF_SIZE;
static const uint32_t ZMINLEN = 128;
#define Z_PTOKAX_COMPRESSION 8
//---------------------------------------------------------------------------
ZlibUtility * ZlibUtility::m_Ptr = nullptr;
//---------------------------------------------------------------------------

ZlibUtility::ZlibUtility() : m_pZbuffer(NULL), m_szZbufferSize(0)
{
	// allocate buffer for zlib
	m_pZbuffer = (char *)calloc(ZBUFFERLEN, 1);
	if (m_pZbuffer == NULL)
	{
		AppendDebugLogFormat("[MEM] Cannot allocate %u bytes for m_pZbuffer in ZlibUtility::ZlibUtility\n", ZBUFFERLEN);
		exit(EXIT_FAILURE);
	}
	memcpy(m_pZbuffer, "$ZOn|", 5);
	m_szZbufferSize = ZBUFFERLEN;
}
//---------------------------------------------------------------------------

ZlibUtility::~ZlibUtility()
{
	free(m_pZbuffer);
}
//---------------------------------------------------------------------------

char * ZlibUtility::CreateZPipe(const char *sInData, const size_t szInDataSize, uint32_t &ui32OutDataLen)
{

#ifdef USE_FLYLINKDC_EXT_JSON
#ifdef _WIN32
	printf("\r\n\r\n[1]CreateZPipe [size = %u], sInData = [%s]\r\n", szInDataSize, sInData);
#endif
#ifdef _DEBUG
	AppendDebugLog("\r\n[1] CreateZPipe", szInDataSize);
	AppendDebugLog(sInData, 0);
#endif
#endif

	// prepare Zbuffer
	if (m_szZbufferSize < szInDataSize + 128)
	{
		size_t szOldZbufferSize = m_szZbufferSize;

		m_szZbufferSize = Allign(szInDataSize + 128);

		char * pOldBuf = m_pZbuffer;
		m_pZbuffer = (char *)realloc(pOldBuf, m_szZbufferSize);
		if (m_pZbuffer == NULL)
		{
			m_pZbuffer = pOldBuf;
			m_szZbufferSize = szOldZbufferSize;
			ui32OutDataLen = 0;

			AppendDebugLogFormat("[MEM] Cannot reallocate %zu bytes for m_pZbuffer in ZlibUtility::CreateZPipe\n", m_szZbufferSize);

			return m_pZbuffer;
		}
	}

	z_stream stream;

	// init zlib struct
	memset(&stream, 0, sizeof(stream));

	stream.zalloc = Z_NULL;
	stream.zfree  = Z_NULL;
	stream.data_type = Z_TEXT;

	deflateInit(&stream, Z_PTOKAX_COMPRESSION);

	stream.next_in  = (Bytef*)sInData;
	stream.avail_in = (uInt)szInDataSize;

	stream.next_out = (Bytef*)m_pZbuffer + 5;
	stream.avail_out = (uInt)m_szZbufferSize-5;

	// compress
	if (deflate(&stream, Z_FINISH) != Z_STREAM_END)
	{
		deflateEnd(&stream);
		AppendDebugLog("%s - [ERR] deflate error\n");
		ui32OutDataLen = 0;
		return m_pZbuffer;
	}

	ui32OutDataLen = stream.total_out + 5;

	// cleanup zlib
	deflateEnd(&stream);

	if (ui32OutDataLen >= szInDataSize)
	{
		ui32OutDataLen = 0;
	}
    GlobalDataQueue::m_Ptr->PrometheusZlibBytes("ZPipeIn",szInDataSize);
    GlobalDataQueue::m_Ptr->PrometheusZlibBytes("ZPipeOut",ui32OutDataLen);
	return m_pZbuffer;
}
//---------------------------------------------------------------------------

char * ZlibUtility::CreateZPipe(const char *sInData, const size_t szInDataSize, char *sOutData, uint32_t &ui32OutDataLen, uint32_t &ui32OutDataSize)
{
	if (szInDataSize < ZMINLEN)
		return sOutData;
#ifdef USE_FLYLINKDC_EXT_JSON
#ifdef _DEBUG
	AppendDebugLog("\r\n[2] CreateZPipe", szInDataSize);
	AppendDebugLog(sInData, 0);
#endif
#ifdef _WIN32
	printf("\r\n\r\n[2]CreateZPipe [size = %u], sInData = [%s]\r\n", szInDataSize, sInData);
#endif
#endif

	// prepare Zbuffer
	if (m_szZbufferSize < szInDataSize + 128)
	{
		size_t szOldZbufferSize = m_szZbufferSize;

		m_szZbufferSize = Allign(szInDataSize + 128);

		char * pOldBuf = m_pZbuffer;
		m_pZbuffer = (char *)realloc(pOldBuf, m_szZbufferSize);
		if (m_pZbuffer == NULL)
		{
			m_pZbuffer = pOldBuf;
			m_szZbufferSize = szOldZbufferSize;
			ui32OutDataLen = 0;

			AppendDebugLogFormat("[MEM] Cannot reallocate %zu bytes for m_pZbuffer in ZlibUtility::CreateZPipe\n", m_szZbufferSize);

			return sOutData;
		}
	}

	z_stream stream;

	// init zlib struct
	memset(&stream, 0, sizeof(stream));

	stream.zalloc = Z_NULL;
	stream.zfree  = Z_NULL;
	stream.data_type = Z_TEXT;

	deflateInit(&stream, Z_PTOKAX_COMPRESSION);

	stream.next_in  = (Bytef*)sInData;
	stream.avail_in = (uInt)szInDataSize;

	stream.next_out = (Bytef*)m_pZbuffer + 5;
	stream.avail_out = (uInt)m_szZbufferSize-5;

	// compress
	if (deflate(&stream, Z_FINISH) != Z_STREAM_END)
	{
		deflateEnd(&stream);
		AppendDebugLog("%s - [ERR] deflate error\n");
		return sOutData;
	}

	ui32OutDataLen = stream.total_out + 5;

	// cleanup zlib
	deflateEnd(&stream);

	if (ui32OutDataLen >= szInDataSize)
	{
		ui32OutDataLen = 0;
		return sOutData;
	}

	// prepare out buffer
	if (ui32OutDataSize < ui32OutDataLen)
	{
		size_t uiOldOutDataSize = ui32OutDataSize;

		ui32OutDataSize = Allign(ui32OutDataLen) - 1;
		char * pOldBuf = sOutData;
		sOutData = (char *)realloc(pOldBuf, ui32OutDataSize + 1);
		if (sOutData == NULL)
		{
			sOutData = pOldBuf;
			ui32OutDataSize = uiOldOutDataSize;
			ui32OutDataLen = 0;

			AppendDebugLogFormat("[MEM] Cannot reallocate %u bytes for sOutData in ZlibUtility::CreateZPipe\n", ui32OutDataSize+1);

			return sOutData;
		}
	}

	memcpy(sOutData, m_pZbuffer, ui32OutDataLen);
    GlobalDataQueue::m_Ptr->PrometheusZlibBytes("ZPipe2In",szInDataSize);
    GlobalDataQueue::m_Ptr->PrometheusZlibBytes("ZPipe2Out",ui32OutDataLen);

	return sOutData;
}
//---------------------------------------------------------------------------

char * ZlibUtility::CreateZPipeAlign(const char *sInData, const size_t szInDataSize, char * sOutData, uint32_t &ui32OutDataLen, uint32_t &ui32OutDataSize)
{
	if (szInDataSize < ZMINLEN)
		return sOutData;
#ifdef USE_FLYLINKDC_EXT_JSON
#ifdef _DEBUG
	AppendDebugLog("\r\n[3] CreateZPipe", szInDataSize);
	AppendDebugLog(sInData, 0);
#endif
#ifdef _WIN32
	printf("\r\n\r\n[3]CreateZPipe [size = %u], sInData = [%s]\r\n", szInDataSize, sInData);
#endif
#endif

	// prepare Zbuffer
	if (m_szZbufferSize < szInDataSize + 128)
	{
		size_t szOldZbufferSize = m_szZbufferSize;

		m_szZbufferSize = Allign(szInDataSize + 128);

		char * pOldBuf = m_pZbuffer;
		m_pZbuffer = (char *)realloc(pOldBuf, m_szZbufferSize);
		if (m_pZbuffer == NULL)
		{
			m_pZbuffer = pOldBuf;
			m_szZbufferSize = szOldZbufferSize;
			ui32OutDataLen = 0;

			AppendDebugLogFormat("[MEM] Cannot reallocate %zu bytes for m_pZbuffer in ZlibUtility::CreateZPipe\n", m_szZbufferSize);

			return sOutData;
		}
	}

	z_stream stream;

	// init zlib struct
	memset(&stream, 0, sizeof(stream));

	stream.zalloc = Z_NULL;
	stream.zfree  = Z_NULL;
	stream.data_type = Z_TEXT;

	deflateInit(&stream, Z_PTOKAX_COMPRESSION);

	stream.next_in  = (Bytef*)sInData;
	stream.avail_in = (uInt)szInDataSize;

	stream.next_out = (Bytef*)m_pZbuffer + 5;
	stream.avail_out = (uInt)m_szZbufferSize-5;

	// compress
	if (deflate(&stream, Z_FINISH) != Z_STREAM_END)
	{
		deflateEnd(&stream);
		AppendDebugLog("%s - [ERR] deflate error\n");
		return sOutData;
	}

	ui32OutDataLen = stream.total_out + 5;

	// cleanup zlib
	deflateEnd(&stream);

	if (ui32OutDataLen >= szInDataSize)
	{
		ui32OutDataLen = 0;
		return sOutData;
	}

	// prepare out buffer
	if (ui32OutDataSize < ui32OutDataLen)
	{
		unsigned int uiOldOutDataSize = ui32OutDataSize;

		ui32OutDataSize = Allign(ui32OutDataLen + 1);

		char * pOldBuf = sOutData;
		sOutData = (char *)realloc(pOldBuf, ui32OutDataSize);
		if (sOutData == NULL)
		{
			sOutData = pOldBuf;
			ui32OutDataSize = uiOldOutDataSize;
			ui32OutDataLen = 0;

			AppendDebugLogFormat("[MEM] Cannot reallocate %u bytes for sOutData in ZlibUtility::CreateZPipe\n", ui32OutDataSize+1);

			return sOutData;
		}
	}

	memcpy(sOutData, m_pZbuffer, ui32OutDataLen);

    GlobalDataQueue::m_Ptr->PrometheusZlibBytes("ZPipeAlignIn",szInDataSize);
    GlobalDataQueue::m_Ptr->PrometheusZlibBytes("ZPipeAlignOut",ui32OutDataLen);

	return sOutData;
}
//---------------------------------------------------------------------------