采用比zlib更快的压缩库zstd

远程桌面帧率提高到12；补充上传zlib.lib。

client/IOCPClient.cpp

@@ -5,8 +5,18 @@
 #include "stdafx.h"
 #include "IOCPClient.h"
 #include <IOSTREAM>
-#include "zconf.h"
+#if USING_ZLIB
 #include "zlib.h"
+#define Z_FAILED(p) (Z_OK != (p))
+#define Z_SUCCESS(p) (!Z_FAILED(p))
+#else
+#include "zstd/zstd.h"
+#pragma comment(lib, "zstd/zstd.lib")
+#define Z_FAILED(p) ZSTD_isError(p)
+#define Z_SUCCESS(p) (!Z_FAILED(p))
+#define compress(dest, destLen, source, sourceLen) ZSTD_compress(dest, *(destLen), source, sourceLen, ZSTD_CLEVEL_DEFAULT)
+#define uncompress(dest, destLen, source, sourceLen) ZSTD_decompress(dest, *(destLen), source, sourceLen)
+#endif
 #include <assert.h>
 #include "Manager.h"
 
@@ -32,6 +42,7 @@ IOCPClient::IOCPClient(bool exit_while_disconnect)
 	m_hWorkThread   = NULL;
 	m_bWorkThread = S_STOP;
 
+	memset(m_szPacketFlag, 0, sizeof(m_szPacketFlag));
 	memcpy(m_szPacketFlag,"Shine",FLAG_LENGTH);
 
 	m_bIsRunning = TRUE;
@@ -187,7 +198,7 @@ VOID IOCPClient::OnServerReceiving(char* szBuffer, ULONG ulLength)
 		//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƿ<EFBFBD><C7B7><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ͷ<EFBFBD><CDB7>С <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ǾͲ<C7BE><CDB2><EFBFBD><EFBFBD><EFBFBD>ȷ<EFBFBD><C8B7><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
 		while (m_CompressedBuffer.GetBufferLength() > HDR_LENGTH)
 		{
-			char szPacketFlag[FLAG_LENGTH] = {0};
+			char szPacketFlag[FLAG_LENGTH + 3] = {0};
 			CopyMemory(szPacketFlag, m_CompressedBuffer.GetBuffer(),FLAG_LENGTH);
 			//<2F>ж<EFBFBD><D0B6><EFBFBD><EFBFBD><EFBFBD>ͷ
 			if (memcmp(m_szPacketFlag, szPacketFlag, FLAG_LENGTH) != 0)
@@ -220,7 +231,7 @@ VOID IOCPClient::OnServerReceiving(char* szBuffer, ULONG ulLength)
 				int	iRet = uncompress(DeCompressedBuffer, 
 					&ulOriginalLength, CompressedBuffer, ulCompressedLength);
 
-				if (iRet == Z_OK)//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ѹ<EFBFBD>ɹ<EFBFBD>
+				if (Z_SUCCESS(iRet))//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ѹ<EFBFBD>ɹ<EFBFBD>
 				{
 					CBuffer m_DeCompressedBuffer;
 					m_DeCompressedBuffer.WriteBuffer(DeCompressedBuffer,
@@ -231,8 +242,10 @@ VOID IOCPClient::OnServerReceiving(char* szBuffer, ULONG ulLength)
 					m_Manager->OnReceive((PBYTE)m_DeCompressedBuffer.GetBuffer(0),
 						m_DeCompressedBuffer.GetBufferLength());
 				}
-				else
+				else{
+					printf("[ERROR] uncompress failed \n");
 					throw "Bad Buffer";
+				}
 
 				delete [] CompressedBuffer;
 				delete [] DeCompressedBuffer;
@@ -246,6 +259,7 @@ VOID IOCPClient::OnServerReceiving(char* szBuffer, ULONG ulLength)
 
 int IOCPClient::OnServerSending(const char* szBuffer, ULONG ulOriginalLength)  //Hello
 {
+	AUTO_TICK(10);
 	assert (ulOriginalLength > 0);
 	{
 		//<2F><><EFBFBD><EFBFBD>1.001<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ҳ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ѹ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>ռ<EFBFBD>õ<EFBFBD><EFBFBD>ڴ<EFBFBD><EFBFBD>ռ<EFBFBD><EFBFBD><EFBFBD>ԭ<EFBFBD><EFBFBD>һ<EFBFBD><EFBFBD> +12
@@ -253,20 +267,28 @@ int IOCPClient::OnServerSending(const char* szBuffer, ULONG ulOriginalLength)  /
 		//<2F><><EFBFBD><EFBFBD>ѹ<EFBFBD><D1B9> ѹ<><D1B9><EFBFBD>㷨 ΢<><CEA2><EFBFBD>ṩ
 		//nSize   = 436
 		//destLen = 448
+#if USING_ZLIB
 		unsigned long	ulCompressedLength = (double)ulOriginalLength * 1.001  + 12;
+#else
+		unsigned long	ulCompressedLength = ZSTD_compressBound(ulOriginalLength);
+#endif
 		LPBYTE			CompressedBuffer = new BYTE[ulCompressedLength];
 
 		int	iRet = compress(CompressedBuffer, &ulCompressedLength, (PBYTE)szBuffer, ulOriginalLength);
-		if (iRet != Z_OK)
+		if (Z_FAILED(iRet))
 		{
+			printf("[ERROR] compress failed \n");
 			delete [] CompressedBuffer;
 			return FALSE;
 		}
+#if !USING_ZLIB
+		ulCompressedLength = iRet;
+#endif
 
 		ULONG ulPackTotalLength = ulCompressedLength + HDR_LENGTH;
 		CBuffer m_WriteBuffer;
 
-		m_WriteBuffer.WriteBuffer((PBYTE)m_szPacketFlag, sizeof(m_szPacketFlag));
+		m_WriteBuffer.WriteBuffer((PBYTE)m_szPacketFlag, FLAG_LENGTH);
 
 		m_WriteBuffer.WriteBuffer((PBYTE) &ulPackTotalLength,sizeof(ULONG));
 		//  5      4

client/IOCPClient.h

@@ -49,7 +49,7 @@ public:
 	BOOL m_bIsRunning;
 	BOOL m_bConnected;
 
-	char    m_szPacketFlag[FLAG_LENGTH];
+	char    m_szPacketFlag[FLAG_LENGTH + 3];
 
 	VOID setManagerCallBack(CManager* Manager);
 

client/ScreenManager.cpp

@@ -43,7 +43,7 @@ DWORD WINAPI CScreenManager::WorkThreadProc(LPVOID lParam)
 
 	clock_t last = clock();
 	This->SendFirstScreen();
-	const int fps = 10;// ֡<><D6A1>
+	const int fps = 12;// ֡<><D6A1>
 	const int sleep = 1000 / fps;// <20><><EFBFBD><EFBFBD>ʱ<EFBFBD>䣨ms<6D><73>
 	while (This->m_bIsWorking)
 	{
@@ -202,7 +202,6 @@ VOID CScreenManager::SendFirstScreen()
 
 const char* CScreenManager::GetNextScreen(ULONG &ulNextSendLength)
 {
-	AUTO_TICK(5);
 	LPVOID	NextScreenData = m_ScreenSpyObject->GetNextScreenData(&ulNextSendLength);
 
 	if (ulNextSendLength == 0 || NextScreenData == NULL)

client/ScreenSpy.cpp

@@ -208,8 +208,8 @@ VOID CScreenSpy::WriteRectBuffer(LPBYTE	szBuffer,ULONG ulLength)
 
 VOID CScreenSpy::ScanScreen(HDC hdcDest, HDC hdcSour, ULONG ulWidth, ULONG ulHeight)
 {
-	AUTO_TICK(1);
-#ifdef COPY_ALL
+	AUTO_TICK(70);
+#if COPY_ALL
 	BitBlt(hdcDest, 0, 0, ulWidth, ulHeight, hdcSour, 0, 0, m_dwBitBltRop);
 #else
 	const ULONG	ulJumpLine = 50;
@@ -233,7 +233,7 @@ VOID CScreenSpy::ScanScreen(HDC hdcDest, HDC hdcSour, ULONG ulWidth, ULONG ulHei
 ULONG CScreenSpy::CompareBitmap(LPBYTE CompareSourData, LPBYTE CompareDestData, 
 								LPBYTE szBuffer, DWORD ulCompareLength)
 {
-	AUTO_TICK(1);
+	AUTO_TICK(20);
 	// Windows<77>涨һ<E6B6A8><D2BB>ɨ<EFBFBD><C9A8><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ռ<EFBFBD><D5BC><EFBFBD>ֽ<EFBFBD><D6BD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>4<EFBFBD>ı<EFBFBD><C4B1><EFBFBD>, <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>DWORD<52>Ƚ<EFBFBD>
 	LPDWORD	p1 = (LPDWORD)CompareDestData, p2 = (LPDWORD)CompareSourData;
 	// ƫ<>Ƶ<EFBFBD>ƫ<EFBFBD>ƣ<EFBFBD><C6A3><EFBFBD>ͬ<EFBFBD><CDAC><EFBFBD>ȵ<EFBFBD>ƫ<EFBFBD><C6AB>

client/StdAfx.h

@@ -6,6 +6,8 @@
 #if !defined(AFX_STDAFX_H__46CA6496_AAD6_4658_B6E9_D7AEB26CDCD5__INCLUDED_)
 #define AFX_STDAFX_H__46CA6496_AAD6_4658_B6E9_D7AEB26CDCD5__INCLUDED_
 
+#define USING_ZLIB 0
+
 #if _MSC_VER > 1000
 #pragma once
 #endif // _MSC_VER > 1000
@@ -61,7 +63,7 @@ public:
 
 #ifdef _DEBUG
 // <20><><EFBFBD>ܼ<EFBFBD><DCBC>㵱ǰ<E3B5B1><C7B0><EFBFBD><EFBFBD><EFBFBD>ĺ<EFBFBD>ʱ<EFBFBD><CAB1><EFBFBD><EFBFBD>ʱ<EFBFBD><CAB1><EFBFBD><EFBFBD>ӡ
-#define AUTO_TICK(thresh) auto_tick(__FUNCTION__, thresh)
+#define AUTO_TICK(thresh) auto_tick TICK(__FUNCTION__, thresh)
 #else
 #define AUTO_TICK(thresh) 
 #endif