c54d75505e
1、修复主控端CTalkDlg的内存泄漏问题，被控端即时消息对话框置于顶层。 2、SAFE_DELETE(ContextObject->olps)有崩溃概率。改为主控端退出时先令被控端退出，就没有内存泄漏。 3、开关音频时偶有内存泄漏，waveInCallBack线程不能正常退出。
737 lines
18 KiB
C++
737 lines
18 KiB
C++
#include "StdAfx.h"
|
||
#include "IOCPServer.h"
|
||
#include "2015Remote.h"
|
||
|
||
#include <iostream>
|
||
#include "zlib.h"
|
||
#include "zconf.h"
|
||
using namespace std;
|
||
|
||
CRITICAL_SECTION IOCPServer::m_cs = {0};
|
||
|
||
#define HUERISTIC_VALUE 2
|
||
|
||
#define SAFE_DELETE(p) if(p){ delete (p); (p) = NULL; }
|
||
|
||
IOCPServer::IOCPServer(void)
|
||
{
|
||
WSADATA wsaData;
|
||
if (WSAStartup(MAKEWORD(2,2), &wsaData)!=0)
|
||
{
|
||
return;
|
||
}
|
||
|
||
m_hCompletionPort = NULL;
|
||
m_sListenSocket = INVALID_SOCKET;
|
||
m_hListenEvent = WSA_INVALID_EVENT;
|
||
m_hListenThread = INVALID_HANDLE_VALUE;
|
||
|
||
m_ulMaxConnections = ((CMy2015RemoteApp*)AfxGetApp())->m_iniFile.GetInt("settings", "MaxConnection");
|
||
|
||
if (m_ulMaxConnections<=0)
|
||
{
|
||
m_ulMaxConnections = 10000;
|
||
}
|
||
|
||
InitializeCriticalSection(&m_cs);
|
||
|
||
m_ulWorkThreadCount = 0;
|
||
|
||
m_bTimeToKill = FALSE;
|
||
|
||
m_ulThreadPoolMin = 0;
|
||
m_ulThreadPoolMax = 0;
|
||
m_ulCPULowThreadsHold = 0;
|
||
m_ulCPUHighThreadsHold = 0;
|
||
m_ulCurrentThread = 0;
|
||
m_ulBusyThread = 0;
|
||
|
||
m_ulKeepLiveTime = 0;
|
||
|
||
m_hKillEvent = NULL;
|
||
|
||
memcpy(m_szPacketFlag,"Shine",FLAG_LENGTH);
|
||
|
||
m_NotifyProc = NULL;
|
||
m_OfflineProc = NULL;
|
||
}
|
||
|
||
|
||
IOCPServer::~IOCPServer(void)
|
||
{
|
||
m_bTimeToKill = TRUE;
|
||
|
||
Sleep(10);
|
||
SetEvent(m_hKillEvent);
|
||
|
||
Sleep(10);
|
||
|
||
if (m_hKillEvent!=NULL)
|
||
{
|
||
CloseHandle(m_hKillEvent);
|
||
}
|
||
|
||
if (m_sListenSocket!=INVALID_SOCKET)
|
||
{
|
||
closesocket(m_sListenSocket);
|
||
m_sListenSocket = INVALID_SOCKET;
|
||
}
|
||
|
||
if (m_hCompletionPort!=INVALID_HANDLE_VALUE)
|
||
{
|
||
CloseHandle(m_hCompletionPort);
|
||
m_hCompletionPort = INVALID_HANDLE_VALUE;
|
||
}
|
||
|
||
if (m_hListenEvent!=WSA_INVALID_EVENT)
|
||
{
|
||
CloseHandle(m_hListenEvent);
|
||
m_hListenEvent = WSA_INVALID_EVENT;
|
||
}
|
||
|
||
while (!m_ContextConnectionList.IsEmpty())
|
||
{
|
||
CONTEXT_OBJECT *ContextObject = m_ContextConnectionList.GetHead();
|
||
RemoveStaleContext(ContextObject);
|
||
}
|
||
|
||
while (!m_ContextFreePoolList.IsEmpty())
|
||
{
|
||
CONTEXT_OBJECT *ContextObject = m_ContextFreePoolList.RemoveHead();
|
||
// ÏÂÊöÓï¾äÓбÀÀ£¸ÅÂÊ£¬2019.1.14
|
||
//SAFE_DELETE(ContextObject->olps);
|
||
delete ContextObject;
|
||
}
|
||
|
||
while (m_ulWorkThreadCount)
|
||
Sleep(10);
|
||
|
||
DeleteCriticalSection(&m_cs);
|
||
m_ulWorkThreadCount = 0;
|
||
|
||
m_ulThreadPoolMin = 0;
|
||
m_ulThreadPoolMax = 0;
|
||
m_ulCPULowThreadsHold = 0;
|
||
m_ulCPUHighThreadsHold = 0;
|
||
m_ulCurrentThread = 0;
|
||
m_ulBusyThread = 0;
|
||
m_ulKeepLiveTime = 0;
|
||
|
||
WSACleanup();
|
||
}
|
||
|
||
BOOL IOCPServer::StartServer(pfnNotifyProc NotifyProc, pfnOfflineProc OffProc, USHORT uPort)
|
||
{
|
||
m_NotifyProc = NotifyProc;
|
||
m_OfflineProc = OffProc;
|
||
m_hKillEvent = CreateEvent(NULL,FALSE,FALSE,NULL);
|
||
|
||
if (m_hKillEvent==NULL)
|
||
{
|
||
return FALSE;
|
||
}
|
||
|
||
m_sListenSocket = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED); //´´½¨¼àÌýÌ×½Ó×Ö
|
||
|
||
if (m_sListenSocket == INVALID_SOCKET)
|
||
{
|
||
return FALSE;
|
||
}
|
||
|
||
m_hListenEvent = WSACreateEvent();
|
||
|
||
if (m_hListenEvent == WSA_INVALID_EVENT)
|
||
{
|
||
closesocket(m_sListenSocket);
|
||
|
||
m_sListenSocket = INVALID_SOCKET;
|
||
return FALSE;
|
||
}
|
||
|
||
int iRet = WSAEventSelect(m_sListenSocket, //½«¼àÌýÌ×½Ó×ÖÓëʼþ½øÐйØÁª²¢ÊÚÓèFD_ACCEPTµÄÊôÐÔ
|
||
m_hListenEvent,
|
||
FD_ACCEPT);
|
||
|
||
if (iRet == SOCKET_ERROR)
|
||
{
|
||
closesocket(m_sListenSocket);
|
||
|
||
m_sListenSocket = INVALID_SOCKET;
|
||
WSACloseEvent(m_hListenEvent);
|
||
|
||
m_hListenEvent = WSA_INVALID_EVENT;
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
SOCKADDR_IN ServerAddr;
|
||
ServerAddr.sin_port = htons(uPort);
|
||
ServerAddr.sin_family = AF_INET;
|
||
ServerAddr.sin_addr.s_addr = INADDR_ANY; //³õʼ»¯±¾µØÍø¿¨
|
||
|
||
//½«¼àÌýÌ×»ú×ÖºÍÍø¿¨½øÐÐbind
|
||
iRet = bind(m_sListenSocket,
|
||
(sockaddr*)&ServerAddr,
|
||
sizeof(ServerAddr));
|
||
|
||
if (iRet == SOCKET_ERROR)
|
||
{
|
||
int a = GetLastError();
|
||
closesocket(m_sListenSocket);
|
||
|
||
m_sListenSocket = INVALID_SOCKET;
|
||
WSACloseEvent(m_hListenEvent);
|
||
|
||
m_hListenEvent = WSA_INVALID_EVENT;
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
iRet = listen(m_sListenSocket, SOMAXCONN);
|
||
|
||
if (iRet == SOCKET_ERROR)
|
||
{
|
||
closesocket(m_sListenSocket);
|
||
|
||
m_sListenSocket = INVALID_SOCKET;
|
||
WSACloseEvent(m_hListenEvent);
|
||
|
||
m_hListenEvent = WSA_INVALID_EVENT;
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
m_hListenThread =
|
||
(HANDLE)CreateThread(NULL,
|
||
0,
|
||
(LPTHREAD_START_ROUTINE)ListenThreadProc,
|
||
(void*)this, //ÏòThread»Øµ÷º¯Êý´«Èëthis ·½±ãÎÒÃǵÄÏ̻߳ص÷·ÃÎÊÀàÖеijÉÔ±
|
||
0,
|
||
NULL);
|
||
if (m_hListenThread==INVALID_HANDLE_VALUE)
|
||
{
|
||
closesocket(m_sListenSocket);
|
||
|
||
m_sListenSocket = INVALID_SOCKET;
|
||
WSACloseEvent(m_hListenEvent);
|
||
|
||
m_hListenEvent = WSA_INVALID_EVENT;
|
||
return FALSE;
|
||
}
|
||
|
||
//Æô¶¯¹¤×÷Ïß³Ì 1 2
|
||
InitializeIOCP();
|
||
return TRUE;
|
||
}
|
||
|
||
|
||
//1´´½¨Íê³É¶Ë¿Ú
|
||
//2´´½¨¹¤×÷Ïß³Ì
|
||
BOOL IOCPServer::InitializeIOCP(VOID)
|
||
{
|
||
m_hCompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0 );
|
||
if ( m_hCompletionPort == NULL )
|
||
{
|
||
return FALSE;
|
||
}
|
||
|
||
if (m_hCompletionPort==INVALID_HANDLE_VALUE)
|
||
{
|
||
return FALSE;
|
||
}
|
||
|
||
SYSTEM_INFO SystemInfo;
|
||
GetSystemInfo(&SystemInfo); //»ñµÃPCÖÐÓм¸ºË
|
||
|
||
m_ulThreadPoolMin = 1;
|
||
m_ulThreadPoolMax = SystemInfo.dwNumberOfProcessors * 2;
|
||
m_ulCPULowThreadsHold = 10;
|
||
m_ulCPUHighThreadsHold = 75;
|
||
m_cpu.Init();
|
||
|
||
ULONG ulWorkThreadCount = m_ulThreadPoolMax;
|
||
|
||
HANDLE hWorkThread = NULL;
|
||
for (int i=0; i<ulWorkThreadCount; i++)
|
||
{
|
||
hWorkThread = (HANDLE)CreateThread(NULL, //´´½¨¹¤×÷Ïß³ÌÄ¿µÄÊÇ´¦ÀíͶµÝµ½Íê³É¶Ë¿ÚÖеÄÈÎÎñ
|
||
0,
|
||
(LPTHREAD_START_ROUTINE)WorkThreadProc,
|
||
(void*)this,
|
||
0,
|
||
NULL);
|
||
if (hWorkThread == NULL )
|
||
{
|
||
CloseHandle(m_hCompletionPort);
|
||
return FALSE;
|
||
}
|
||
|
||
m_ulWorkThreadCount++;
|
||
|
||
CloseHandle(hWorkThread);
|
||
}
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
|
||
DWORD IOCPServer::WorkThreadProc(LPVOID lParam)
|
||
{
|
||
OutputDebugStringA("======> IOCPServer WorkThreadProc begin \n");
|
||
|
||
IOCPServer* This = (IOCPServer*)(lParam);
|
||
|
||
HANDLE hCompletionPort = This->m_hCompletionPort;
|
||
DWORD dwTrans = 0;
|
||
|
||
PCONTEXT_OBJECT ContextObject = NULL;
|
||
LPOVERLAPPED Overlapped = NULL;
|
||
OVERLAPPEDPLUS* OverlappedPlus = NULL;
|
||
ULONG ulBusyThread = 0;
|
||
BOOL bError = FALSE;
|
||
|
||
InterlockedIncrement(&This->m_ulCurrentThread);
|
||
InterlockedIncrement(&This->m_ulBusyThread);
|
||
|
||
while (This->m_bTimeToKill==FALSE)
|
||
{
|
||
InterlockedDecrement(&This->m_ulBusyThread);
|
||
BOOL bOk = GetQueuedCompletionStatus(
|
||
hCompletionPort,
|
||
&dwTrans,
|
||
(LPDWORD)&ContextObject,
|
||
&Overlapped,INFINITE);
|
||
|
||
DWORD dwIOError = GetLastError();
|
||
|
||
OverlappedPlus = CONTAINING_RECORD(Overlapped, OVERLAPPEDPLUS, m_ol);
|
||
|
||
ulBusyThread = InterlockedIncrement(&This->m_ulBusyThread); //1 1
|
||
if ( !bOk && dwIOError != WAIT_TIMEOUT ) //µ±¶Ô·½µÄÌ×»úÖÆ·¢ÉúÁ˹رÕ
|
||
{
|
||
if (ContextObject && This->m_bTimeToKill == FALSE &&dwTrans==0)
|
||
{
|
||
ContextObject->olps = NULL;
|
||
This->RemoveStaleContext(ContextObject);
|
||
}
|
||
SAFE_DELETE(OverlappedPlus);
|
||
continue;
|
||
}
|
||
if (!bError)
|
||
{
|
||
//·ÖÅäÒ»¸öеÄÏ̵߳½Ï̵߳½Ï̳߳Ø
|
||
if (ulBusyThread == This->m_ulCurrentThread)
|
||
{
|
||
if (ulBusyThread < This->m_ulThreadPoolMax)
|
||
{
|
||
if (ContextObject != NULL)
|
||
{
|
||
HANDLE hThread = (HANDLE)CreateThread(NULL,
|
||
0,
|
||
(LPTHREAD_START_ROUTINE)WorkThreadProc,
|
||
(void*)This,
|
||
0,
|
||
NULL);
|
||
|
||
InterlockedIncrement(&This->m_ulWorkThreadCount);
|
||
|
||
CloseHandle(hThread);
|
||
}
|
||
}
|
||
}
|
||
|
||
if (!bOk && dwIOError == WAIT_TIMEOUT)
|
||
{
|
||
if (ContextObject == NULL)
|
||
{
|
||
if (This->m_ulCurrentThread > This->m_ulThreadPoolMin)
|
||
{
|
||
break;
|
||
}
|
||
|
||
bError = TRUE;
|
||
}
|
||
}
|
||
}
|
||
|
||
if (!bError)
|
||
{
|
||
if(bOk && OverlappedPlus!=NULL && ContextObject!=NULL)
|
||
{
|
||
try
|
||
{
|
||
This->HandleIO(OverlappedPlus->m_ioType, ContextObject, dwTrans);
|
||
|
||
ContextObject = NULL;
|
||
}
|
||
catch (...) {}
|
||
}
|
||
}
|
||
|
||
SAFE_DELETE(OverlappedPlus);
|
||
}
|
||
SAFE_DELETE(OverlappedPlus);
|
||
|
||
InterlockedDecrement(&This->m_ulWorkThreadCount);
|
||
InterlockedDecrement(&This->m_ulCurrentThread);
|
||
InterlockedDecrement(&This->m_ulBusyThread);
|
||
|
||
OutputDebugStringA("======> IOCPServer WorkThreadProc end \n");
|
||
|
||
return 0;
|
||
}
|
||
|
||
//ÔÚ¹¤×÷Ïß³ÌÖб»µ÷ÓÃ
|
||
BOOL IOCPServer::HandleIO(IOType PacketFlags,PCONTEXT_OBJECT ContextObject, DWORD dwTrans)
|
||
{
|
||
AUTO_TICK(5);
|
||
|
||
BOOL bRet = FALSE;
|
||
|
||
switch (PacketFlags)
|
||
{
|
||
case IOInitialize:
|
||
bRet = OnClientInitializing(ContextObject, dwTrans);
|
||
break;
|
||
case IORead:
|
||
bRet = OnClientReceiving(ContextObject,dwTrans);
|
||
break;
|
||
case IOWrite:
|
||
bRet = OnClientPostSending(ContextObject,dwTrans);
|
||
break;
|
||
case IOIdle:
|
||
OutputDebugStringA("=> HandleIO PacketFlags= IOIdle\n");
|
||
break;
|
||
default:
|
||
break;
|
||
}
|
||
|
||
return bRet;
|
||
}
|
||
|
||
|
||
BOOL IOCPServer::OnClientInitializing(PCONTEXT_OBJECT ContextObject, DWORD dwTrans)
|
||
{
|
||
return TRUE;
|
||
}
|
||
|
||
BOOL IOCPServer::OnClientReceiving(PCONTEXT_OBJECT ContextObject, DWORD dwTrans)
|
||
{
|
||
CLock cs(m_cs);
|
||
try
|
||
{
|
||
if (dwTrans == 0) //¶Ô·½¹Ø±ÕÁËÌ×½Ó×Ö
|
||
{
|
||
RemoveStaleContext(ContextObject);
|
||
return FALSE;
|
||
}
|
||
//½«½ÓÊÕµ½µÄÊý¾Ý¿½±´µ½ÎÒÃÇ×Ô¼ºµÄÄÚ´æÖÐwsabuff 8192
|
||
ContextObject->InCompressedBuffer.WriteBuffer((PBYTE)ContextObject->szBuffer,dwTrans);
|
||
//²é¿´Êý¾Ý°üÀïµÄÊý¾Ý
|
||
while (ContextObject->InCompressedBuffer.GetBufferLength() > HDR_LENGTH)
|
||
{
|
||
char szPacketFlag[FLAG_LENGTH + 3]= {0}; // 8×Ö½Ú¶ÔÆë
|
||
CopyMemory(szPacketFlag, ContextObject->InCompressedBuffer.GetBuffer(),FLAG_LENGTH);
|
||
if (memcmp(m_szPacketFlag, szPacketFlag, FLAG_LENGTH) != 0)
|
||
throw "Bad Buffer";
|
||
|
||
//Shine[50][kdjfkdjfkj]
|
||
ULONG ulPackTotalLength = 0;
|
||
CopyMemory(&ulPackTotalLength, ContextObject->InCompressedBuffer.GetBuffer(FLAG_LENGTH), sizeof(ULONG));
|
||
//È¡³öÊý¾Ý°üµÄ×ܳ¤
|
||
//50
|
||
if (ulPackTotalLength && (ContextObject->InCompressedBuffer.GetBufferLength()) >= ulPackTotalLength)
|
||
{
|
||
ULONG ulOriginalLength = 0;
|
||
ContextObject->InCompressedBuffer.ReadBuffer((PBYTE)szPacketFlag, FLAG_LENGTH);
|
||
ContextObject->InCompressedBuffer.ReadBuffer((PBYTE) &ulPackTotalLength, sizeof(ULONG));
|
||
ContextObject->InCompressedBuffer.ReadBuffer((PBYTE) &ulOriginalLength, sizeof(ULONG));
|
||
ULONG ulCompressedLength = ulPackTotalLength - HDR_LENGTH; //461 - 13 448
|
||
PBYTE CompressedBuffer = new BYTE[ulCompressedLength]; //ûÓнâѹ
|
||
PBYTE DeCompressedBuffer = new BYTE[ulOriginalLength]; //½âѹ¹ýµÄÄÚ´æ 436
|
||
//´ÓÊý¾Ý°üµ±Ç°½«Ô´Êý¾ÝûÓнâѹ¶ÁÈ¡µ½pData 448
|
||
ContextObject->InCompressedBuffer.ReadBuffer(CompressedBuffer, ulCompressedLength);
|
||
int iRet = uncompress(DeCompressedBuffer, &ulOriginalLength, CompressedBuffer, ulCompressedLength);
|
||
if (iRet == Z_OK)
|
||
{
|
||
ContextObject->InDeCompressedBuffer.ClearBuffer();
|
||
ContextObject->InCompressedBuffer.ClearBuffer();
|
||
ContextObject->InDeCompressedBuffer.WriteBuffer(DeCompressedBuffer, ulOriginalLength);
|
||
m_NotifyProc(ContextObject); //֪ͨ´°¿Ú
|
||
}else{
|
||
throw "Bad Buffer";
|
||
}
|
||
delete [] CompressedBuffer;
|
||
delete [] DeCompressedBuffer;
|
||
}else{
|
||
break;
|
||
}
|
||
}
|
||
PostRecv(ContextObject); //ͶµÝеĽÓÊÕÊý¾ÝµÄÇëÇó
|
||
}catch(...)
|
||
{
|
||
ContextObject->InCompressedBuffer.ClearBuffer();
|
||
ContextObject->InDeCompressedBuffer.ClearBuffer();
|
||
|
||
PostRecv(ContextObject);
|
||
}
|
||
return TRUE;
|
||
}
|
||
|
||
VOID IOCPServer::OnClientPreSending(CONTEXT_OBJECT* ContextObject, PBYTE szBuffer, ULONG ulOriginalLength)
|
||
{
|
||
assert (ContextObject);
|
||
|
||
try
|
||
{
|
||
if (ulOriginalLength > 0)
|
||
{
|
||
unsigned long ulCompressedLength = (double)ulOriginalLength * 1.001 + 12;
|
||
LPBYTE CompressedBuffer = new BYTE[ulCompressedLength];
|
||
int iRet = compress(CompressedBuffer, &ulCompressedLength, (LPBYTE)szBuffer, ulOriginalLength);
|
||
|
||
if (iRet != Z_OK)
|
||
{
|
||
delete [] CompressedBuffer;
|
||
return;
|
||
}
|
||
ULONG ulPackTotalLength = ulCompressedLength + HDR_LENGTH;
|
||
ContextObject->OutCompressedBuffer.WriteBuffer((LPBYTE)m_szPacketFlag,FLAG_LENGTH);
|
||
ContextObject->OutCompressedBuffer.WriteBuffer((PBYTE)&ulPackTotalLength, sizeof(ULONG));
|
||
ContextObject->OutCompressedBuffer.WriteBuffer((PBYTE) &ulOriginalLength, sizeof(ULONG));
|
||
ContextObject->OutCompressedBuffer.WriteBuffer(CompressedBuffer, ulCompressedLength);
|
||
delete [] CompressedBuffer;
|
||
}
|
||
|
||
OVERLAPPEDPLUS* OverlappedPlus = new OVERLAPPEDPLUS(IOWrite);
|
||
BOOL bOk = PostQueuedCompletionStatus(m_hCompletionPort, 0, (DWORD)ContextObject, &OverlappedPlus->m_ol);
|
||
if ( (!bOk && GetLastError() != ERROR_IO_PENDING) ) //Èç¹ûͶµÝʧ°Ü
|
||
{
|
||
RemoveStaleContext(ContextObject);
|
||
SAFE_DELETE(OverlappedPlus);
|
||
}
|
||
}catch(...){}
|
||
}
|
||
|
||
BOOL IOCPServer::OnClientPostSending(CONTEXT_OBJECT* ContextObject,ULONG ulCompletedLength)
|
||
{
|
||
try
|
||
{
|
||
DWORD ulFlags = MSG_PARTIAL;
|
||
|
||
ContextObject->OutCompressedBuffer.RemoveComletedBuffer(ulCompletedLength); //½«Íê³ÉµÄÊý¾Ý´ÓÊý¾Ý½á¹¹ÖÐÈ¥³ý
|
||
if (ContextObject->OutCompressedBuffer.GetBufferLength() == 0)
|
||
{
|
||
ContextObject->OutCompressedBuffer.ClearBuffer();
|
||
return true; //×ßµ½ÕâÀï˵Ã÷ÎÒÃǵÄÊý¾ÝÕæÕýÍêÈ«·¢ËÍ
|
||
}
|
||
else
|
||
{
|
||
OVERLAPPEDPLUS * OverlappedPlus = new OVERLAPPEDPLUS(IOWrite); //Êý¾ÝûÓÐÍê³É ÎÒÃǼÌÐøͶµÝ ·¢ËÍÇëÇó
|
||
|
||
ContextObject->wsaOutBuffer.buf = (char*)ContextObject->OutCompressedBuffer.GetBuffer();
|
||
ContextObject->wsaOutBuffer.len = ContextObject->OutCompressedBuffer.GetBufferLength();
|
||
int iOk = WSASend(ContextObject->sClientSocket, &ContextObject->wsaOutBuffer,1,
|
||
&ContextObject->wsaOutBuffer.len, ulFlags,&OverlappedPlus->m_ol, NULL);
|
||
if ( iOk == SOCKET_ERROR && WSAGetLastError() != WSA_IO_PENDING )
|
||
{
|
||
int a = GetLastError();
|
||
RemoveStaleContext(ContextObject);
|
||
SAFE_DELETE(OverlappedPlus);
|
||
}
|
||
}
|
||
}catch(...){}
|
||
|
||
return FALSE;
|
||
}
|
||
|
||
DWORD IOCPServer::ListenThreadProc(LPVOID lParam) //¼àÌýÏß³Ì
|
||
{
|
||
IOCPServer* This = (IOCPServer*)(lParam);
|
||
WSANETWORKEVENTS NetWorkEvents;
|
||
|
||
while(1)
|
||
{
|
||
if (WaitForSingleObject(This->m_hKillEvent, 100) == WAIT_OBJECT_0)
|
||
break;
|
||
|
||
DWORD dwRet;
|
||
dwRet = WSAWaitForMultipleEvents(1,&This->m_hListenEvent,FALSE,100,FALSE);
|
||
if (dwRet == WSA_WAIT_TIMEOUT)
|
||
continue;
|
||
|
||
int iRet = WSAEnumNetworkEvents(This->m_sListenSocket,
|
||
//Èç¹ûʼþÊÚÐÅ ÎÒÃǾͽ«¸Ãʼþת»»³ÉÒ»¸öÍøÂçʼþ ½øÐÐ ÅжÏ
|
||
This->m_hListenEvent,
|
||
&NetWorkEvents);
|
||
|
||
if (iRet == SOCKET_ERROR)
|
||
break;
|
||
|
||
if (NetWorkEvents.lNetworkEvents & FD_ACCEPT)
|
||
{
|
||
if (NetWorkEvents.iErrorCode[FD_ACCEPT_BIT] == 0)
|
||
{
|
||
This->OnAccept();
|
||
}else{
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
void IOCPServer::OnAccept()
|
||
{
|
||
SOCKADDR_IN ClientAddr = {0};
|
||
SOCKET sClientSocket = INVALID_SOCKET;
|
||
|
||
int iLen = sizeof(SOCKADDR_IN);
|
||
sClientSocket = accept(m_sListenSocket,
|
||
(sockaddr*)&ClientAddr,
|
||
&iLen); //ͨ¹ýÎÒÃǵļàÌýÌ×½Ó×ÖÀ´Éú³ÉÒ»¸öÓëÖ®ÐźÅͨÐŵÄÌ×½Ó×Ö
|
||
if (sClientSocket == SOCKET_ERROR)
|
||
{
|
||
return;
|
||
}
|
||
|
||
//ÎÒÃÇÔÚÕâÀïΪÿһ¸öµ½´ïµÄÐźÅά»¤ÁËÒ»¸öÓëÖ®¹ØÁªµÄÊý¾Ý½á¹¹ÕâÀï¼ò³ÆΪÓû§µÄÉÏϱ³¾°ÎÄ
|
||
PCONTEXT_OBJECT ContextObject = AllocateContext(); // Context
|
||
|
||
if (ContextObject == NULL)
|
||
{
|
||
closesocket(sClientSocket);
|
||
sClientSocket = INVALID_SOCKET;
|
||
return;
|
||
}
|
||
|
||
ContextObject->sClientSocket = sClientSocket;
|
||
|
||
ContextObject->wsaInBuf.buf = (char*)ContextObject->szBuffer;
|
||
ContextObject->wsaInBuf.len = sizeof(ContextObject->szBuffer);
|
||
|
||
HANDLE Handle = CreateIoCompletionPort((HANDLE)sClientSocket, m_hCompletionPort, (DWORD)ContextObject, 0);
|
||
|
||
if (Handle!=m_hCompletionPort)
|
||
{
|
||
delete ContextObject;
|
||
ContextObject = NULL;
|
||
|
||
if (sClientSocket!=INVALID_SOCKET)
|
||
{
|
||
closesocket(sClientSocket);
|
||
sClientSocket = INVALID_SOCKET;
|
||
}
|
||
|
||
return;
|
||
}
|
||
|
||
//ÉèÖÃÌ×½Ó×ÖµÄÑ¡Ï Set KeepAlive ¿ªÆô±£»î»úÖÆ SO_KEEPALIVE
|
||
//±£³ÖÁ¬½Ó¼ì²â¶Ô·½Ö÷»úÊÇ·ñ±ÀÀ£Èç¹û2СʱÄÚÔÚ´ËÌ×½Ó¿ÚµÄÈÎÒ»·½Ïò¶¼Ã»
|
||
//ÓÐÊý¾Ý½»»»£¬TCP¾Í×Ô¶¯¸ø¶Ô·½ ·¢Ò»¸ö±£³Ö´æ»î
|
||
m_ulKeepLiveTime = 3;
|
||
const BOOL bKeepAlive = TRUE;
|
||
setsockopt(ContextObject->sClientSocket,SOL_SOCKET,SO_KEEPALIVE,(char*)&bKeepAlive,sizeof(bKeepAlive));
|
||
|
||
//ÉèÖó¬Ê±ÏêϸÐÅÏ¢
|
||
tcp_keepalive KeepAlive;
|
||
KeepAlive.onoff = 1; // ÆôÓñ£»î
|
||
KeepAlive.keepalivetime = m_ulKeepLiveTime; //³¬¹ý3·ÖÖÓûÓÐÊý¾Ý£¬¾Í·¢ËÍ̽²â°ü
|
||
KeepAlive.keepaliveinterval = 1000 * 10; //ÖØÊÔ¼ä¸ôΪ10Ãë Resend if No-Reply
|
||
WSAIoctl(ContextObject->sClientSocket, SIO_KEEPALIVE_VALS,&KeepAlive,sizeof(KeepAlive),
|
||
NULL,0,(unsigned long *)&bKeepAlive,0,NULL);
|
||
|
||
//ÔÚ×ö·þÎñÆ÷ʱ£¬Èç¹û·¢Éú¿Í»§¶ËÍøÏß»ò¶ÏµçµÈ·ÇÕý³£¶Ï¿ªµÄÏÖÏó£¬Èç¹û·þÎñÆ÷ûÓÐÉèÖÃSO_KEEPALIVEÑ¡Ï
|
||
//Ôò»áÒ»Ö±²»¹Ø±ÕSOCKET¡£ÒòΪÉϵĵÄÉèÖÃÊÇĬÈÏÁ½¸öСʱʱ¼äÌ«³¤ÁËËùÒÔÎÒÃǾÍÐÞÕýÕâ¸öÖµ
|
||
CLock cs(m_cs);
|
||
m_ContextConnectionList.AddTail(ContextObject); //²åÈëµ½ÎÒÃǵÄÄÚ´æÁбíÖÐ
|
||
|
||
OVERLAPPEDPLUS *OverlappedPlus = new OVERLAPPEDPLUS(IOInitialize); //×¢ÒâÕâÀïµÄÖصþIOÇëÇóÊÇ Óû§ÇëÇóÉÏÏß
|
||
|
||
BOOL bOk = PostQueuedCompletionStatus(m_hCompletionPort, 0, (DWORD)ContextObject, &OverlappedPlus->m_ol); // ¹¤×÷Ïß³Ì
|
||
//ÒòΪÎÒÃǽÓÊܵ½ÁËÒ»¸öÓû§ÉÏÏßµÄÇëÇóÄÇôÎÒÃǾͽ«¸ÃÇëÇó·¢Ë͸øÎÒÃǵÄÍê³É¶Ë¿Ú ÈÃÎÒÃǵŤ×÷Ï̴߳¦ÀíËü
|
||
if ( (!bOk && GetLastError() != ERROR_IO_PENDING)) //Èç¹ûͶµÝʧ°Ü
|
||
{
|
||
RemoveStaleContext(ContextObject);
|
||
SAFE_DELETE(OverlappedPlus);
|
||
return;
|
||
}
|
||
|
||
PostRecv(ContextObject);
|
||
}
|
||
|
||
VOID IOCPServer::PostRecv(CONTEXT_OBJECT* ContextObject)
|
||
{
|
||
//ÏòÎÒÃǵĸÕÉÏÏßµÄÓû§µÄͶµÝÒ»¸ö½ÓÊÜÊý¾ÝµÄÇëÇó
|
||
// Èç¹ûÓû§µÄµÚÒ»¸öÊý¾Ý°üµ½´ïÒ²¾Í¾ÍÊDZ»¿Ø¶ËµÄµÇ½ÇëÇóµ½´ïÎÒÃǵŤ×÷Ï߳̾Í
|
||
// »áÏìÓ¦,²¢µ÷ÓÃProcessIOMessageº¯Êý
|
||
OVERLAPPEDPLUS * OverlappedPlus = new OVERLAPPEDPLUS(IORead);
|
||
ContextObject->olps = OverlappedPlus;
|
||
|
||
DWORD dwReturn;
|
||
ULONG ulFlags = MSG_PARTIAL;
|
||
int iOk = WSARecv(ContextObject->sClientSocket, &ContextObject->wsaInBuf,
|
||
1,&dwReturn, &ulFlags,&OverlappedPlus->m_ol, NULL);
|
||
|
||
if (iOk == SOCKET_ERROR && WSAGetLastError() != WSA_IO_PENDING)
|
||
{
|
||
int a = GetLastError();
|
||
RemoveStaleContext(ContextObject);
|
||
SAFE_DELETE(OverlappedPlus);
|
||
}
|
||
}
|
||
|
||
PCONTEXT_OBJECT IOCPServer::AllocateContext()
|
||
{
|
||
PCONTEXT_OBJECT ContextObject = NULL;
|
||
|
||
CLock cs(m_cs);
|
||
|
||
ContextObject = !m_ContextFreePoolList.IsEmpty() ? m_ContextFreePoolList.RemoveHead() : new CONTEXT_OBJECT;
|
||
|
||
if (ContextObject != NULL)
|
||
{
|
||
ContextObject->InitMember();
|
||
}
|
||
|
||
return ContextObject;
|
||
}
|
||
|
||
VOID IOCPServer::RemoveStaleContext(CONTEXT_OBJECT* ContextObject)
|
||
{
|
||
CLock cs(m_cs);
|
||
if (m_ContextConnectionList.Find(ContextObject)) //ÔÚÄÚ´æÖвéÕÒ¸ÃÓû§µÄÉÏÏÂÎÄÊý¾Ý½á¹¹
|
||
{
|
||
m_OfflineProc(ContextObject);
|
||
|
||
CancelIo((HANDLE)ContextObject->sClientSocket); //È¡ÏûÔÚµ±Ç°Ì×½Ó×ÖµÄÒì²½IO -->PostRecv
|
||
closesocket(ContextObject->sClientSocket); //¹Ø±ÕÌ×½Ó×Ö
|
||
ContextObject->sClientSocket = INVALID_SOCKET;
|
||
|
||
while (!HasOverlappedIoCompleted((LPOVERLAPPED)ContextObject))//Åжϻ¹ÓÐûÓÐÒì²½IOÇëÇóÔÚµ±Ç°Ì×½Ó×ÖÉÏ
|
||
{
|
||
Sleep(0);
|
||
}
|
||
|
||
MoveContextToFreePoolList(ContextObject); //½«¸ÃÄÚ´æ½á¹¹»ØÊÕÖÁÄÚ´æ³Ø
|
||
}
|
||
}
|
||
|
||
VOID IOCPServer::MoveContextToFreePoolList(CONTEXT_OBJECT* ContextObject)
|
||
{
|
||
CLock cs(m_cs);
|
||
|
||
POSITION Pos = m_ContextConnectionList.Find(ContextObject);
|
||
if (Pos)
|
||
{
|
||
ContextObject->InCompressedBuffer.ClearBuffer();
|
||
ContextObject->InDeCompressedBuffer.ClearBuffer();
|
||
ContextObject->OutCompressedBuffer.ClearBuffer();
|
||
|
||
memset(ContextObject->szBuffer,0,8192);
|
||
m_ContextFreePoolList.AddTail(ContextObject); //»ØÊÕÖÁÄÚ´æ³Ø
|
||
m_ContextConnectionList.RemoveAt(Pos); //´ÓÄÚ´æ½á¹¹ÖÐÒƳý
|
||
}
|
||
}
|