Start function is used to enable service
1 state variable initialization
2 Create a listening socket
3 loaded using the extended API functions
4 port object is created
5 association between the listening socket and completion port object
6 is registered FD_ACCEPT time listening socket
Create a listener thread after delivery 7 AcceptEx can be notified when the IO is not enough
BOOL CIOCOPServer::Start(int nPort,int nMaxConnnections,int nMaxFreeBuffers,int nMaxFreeContexts,int nInitialReads)
{
//检查服务是否启动
if(m_bServerStarted)
return FALSE;
//保存参数
m_nPort = nPort;
m_nMaxConnnections = nMaxConnnections;
m_nMaxFreeBuffers = nMaxFreeBuffers;
m_nMaxFreeContexts = nMaxFreeContexts;
m_nInitialReads = nInitialReads;
//初始化变量
m_bServerStarted = TRUE;
m_bShutDown = FALSE;
//创建监听套接字,绑定到本地端口, 进入监听模式
m_sListen = ::WSASocket(AF_INET,SOCK_STREAM,0,NULL,0,WSA_FLAG_OVERLAPPED);
SOCKADDR_IN si;
si.sin_family = AF_INET;
si.sin_port = nPort;
si.sin_addr.S_un.S_addr = INADDR_ANY;
if(::bind(m_sListen,(sockaddr*)&si,sizeof(si))==SOCKET_ERROR)
{
m_bServerStarted = FALSE;
return FALSE;
}
::listen(m_sListen,200);
//创建完成端口
m_hConnection = ::CreateIoCompletionPort(INVALID_HANDLE_VALUE,0,0,0);
//加载扩展函数AcceptEx
GUID GuidAcceptEx = WSAID_ACCEPTEX;
DWORD dwBytes;
WSAIotcl(
m_sListen,
SIO_GET_EXTENSION_FUNCTION_POINTER,
&GuidAcceptEx,
sizeof(GuidAcceptEx),
&m_lpfnAcceptEx,
sizeof(m_lpfnAcceptEx),
&dwBytes,
NULL,
NULL
);
//加载GetAcceptExSockaddrs
GUID GuidAcceptExSockaddrs = WSAID_GETACCEPTEXSOCKADDRS;
::WSAIoctl(m_sListen,
SIO_GET_EXTENSION_FUNCTION_POINTER,
&GuiGetAcceptExSockaddrs,
sizeof(GuidGetAcceptExSockaddrs),
&m_lpfnAcceptExSockaddrs,
sizeof(m_lofnAcceptExSockaddrs),
&dwBytes,
NULL,
NULL);
//将监听套接字关联到完成端口
::CreateIoCompletionPort((HANDLE)m_sListen,m_hConnection,(DWORD)0,0);
//注册FD_ACCEPT事件
WSAEventSelect(m_sListen,m_hAcceptEvent,FD_ACCEPT);
//创建监听线程
m_hListenThread = ::CreateThread(NULL,0,_ListenThreadProc,this,0,NULL);
return TRUE;
}
Listening thread _ListenThreadProc primary responsibility: listening socket delivery AcceptEx IO request.
m_hAcceptEvent : winsock when a new connection request is received, but AcceptEx IO, when receiving the connection request, which the object is triggered.
m_hRepostEvent : interact with the IO.
_ListenThreadProc delivery Accept request in the following three cases:
An initialization procedure, a few first delivery Accept request number specified by the user
2 Threading the IO receives a customer, the m_hRepostEvent time trusted, _ListenThreadProc thread to be notified before posting a request Accept.
3 during running, if the delivery request Accept enough, the user is not able to handle the connection request immediately, this time to a plurality of delivery Accept request.
DWORD WINAPI CIOCPServer::_ListenThreadProc(LPVOID lpParam)
{
CIOCPServer *pThis = (CIOCPServer*)lpParam;
//在监听套接字上投递几个AcceptIO
CIOCPBuffer *pBuffer;
for(int i=0;i<pThis->m_nInitialAccepts;i++)
{
pBuffer = pThis->AllocateBuffer(BUFFER_SIZE);
if(pBuffer==NULL)
return -1;
pTHis->InsertPendingAccept(pBuffer);
pThis->PostAccept(pBuffer);
}
//构建事件对象数组
HANDLE hWaitEvents[2+MAX_THREAD];
int nEventCount = 0;
hWaitEvents[nEventCount++]=pThis->m_hAcceptEvent;
hWaitEvents[nEventCount++]=pThis->m_hRepostEvent;
//创建指定数量的工作线程在完成端口上处理IO
for(i=0;i<MAX_THREAD;i++)
{
hWaitEvents[nEventCount++]=::CreateThread(NULL,0,_WorkerThreadProc,pThis,0,NULL);
}
//下面进入无限循环,处理时间对象数组中的事件
while(TRUE)
{
int nIndex = ::WSAWaitForMultipleEvents(nEventCount,hWaitEvents,FALSE,60*1000,FALSE);
//检查是否要停止服务
if(pThis->m_bShutDown || nIndex==WSA_WAIT_FAILED)
{
//关闭所有连接
pThis->CloseAllConnections();
::Sleep(0);
//关闭监听套接字
::closesocket(pThis->m_sListen);
pThis->m_sListen=INVALID_SOCKET;
::Sleep(0);
//通知所有IO处理线程退出
for(int i=2;i<MAX_THREAD+2;i++)
{
::PostQueuedCompletionStatus(pThis->m_hCompletion,-1,0,NULL);
}
//等待IO处理线程退出
::WaitForMultipleObjects(MAX_THREAD,&hWaitEvents[2],TRUE,5*1000);
for(i=2;i<MAX_THREAD+2;i++)
{
::CloseHandle(hWaitEvents[i]);
}
::CloseHandle(pThis->m_hCompletion);
pThis->FreeBuffers();
pThis->FreeContexts();
::ExitThread(0);
}
//定时检查所有未返回的AcceptEx IO的连接建立多长时间
if(nIndex == WSA_WAIT_TIMEOUT)
{
pBuffer = pThis->m_pPendingAccepts;
while(pBuffer!=NULL)
{
int nSeconds;
int nLen = sizeof(nSeconds);
//取得连接建立时间
::getsockopt(pBuffer->sClient,SOL_SOCKET,SO_CONNECT_TIME,(char*)&nSeconds,&nLen);
//如果超过两分钟,就丢弃
if(nSeconds!=-1 && nSeconds>=2*60)
{
closesocket(pBuffer->sClient);
pBuffer->sClient = INVALIDE_SOCKET;
}
pBuffer = pBuffer->pNext;
}
}
else
{
nIndex = nIndex-WAIT_OBJECT_0;
WSANETWORKEVENTS ne;
int nLimit=0;
if(nIndex==0)//m_hAcceptEvent时间对象受信,说明投递的Accept请求不够,需要增加
{
::WSAEnumNetworkEvents(pThis->m_sListen,hWaitEvents[nIndex],&ne);
if(ne.lNetworkEvents & FD_ACCEPT)
{
nLimit = 50;
}
}
else if(nIndex==1)//m_hRepostEvent事件对象受信,说明处理IO的线程接受到新的客户
{
nLimit = InterlockedExchange(&pThis->m_nRepostCount,0);
}
else if(nIndex>1)//IO服务线程退出,说明有错误发生,关闭服务器
{
pThis->m_bShutDown = TRUE;
continue;
}
//投递nLimit个AcceptEx IO 请求
int i=0;
while(i++ < nLimit && pThis->m_nPendingAcceptCount < pThis->m_nMaxAccepts)
{
pBuffer = pThis->AllocateBuffer(BUFFER_SIZE);
if(pBuffer!=NULL)
{
pThis->InsertPendingAccept(pBuffer);
pThis->PostAccept(pBuffer);
}
}
}
}
return 0;
}
3 Stop the service function ShutDown
void CIOCPServer::ShutDown()
{
if(!m_bServerStarted)
return;
//通知监听线程,马上停止服务
m_bShutDown = TRUE;
::SetEvent(m_hAcceptEvent);
//等待监听线程退出
::WaitForSingleObject(m_hListenThread,INFINITE);
::CloseHandle(m_hListenThread);
m_hListenThread = NUll;
m_bServerStarted = FALSE;
}
Reproduced in: https: //my.oschina.net/u/204616/blog/545431