完成端口(IOCP)的实现(5)

整理的一个简单的完成端口类

on);
DeleteCriticalSection(&m_conSection);
DeleteCriticalSection(&m_ioMenListSection);
DeleteCriticalSection(&m_hdMemListSection);
}

//启动完成端口服务
bool NetCompletionPort::StartServer(int port)
{
WSADATA m_wsaData;
m_port = port;

if(WSAStartup(MAKEWORD(2,2), &m_wsaData))
return FALSE;

if(InitWinSock() && InitListenSocket() && InitListenEvent() && InitWorkThread())
{
m_isCreate = TRUE;
m_pThreadAccept->Start(); //启动accept线程
std::vector<WorkThread*>::iterator pos = m_pWorkThreadList->begin();
while (pos != m_pWorkThreadList->end())
{
(*pos)->Start(); //启动工作者线程
pos++;
}
}
else
{
StopServer();
return FALSE;
}

return TRUE;
}

//初始化完成端口，创建工作者线程
bool NetCompletionPort::InitWinSock()
{
m_comPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE,NULL,0,0);

if(!m_comPort)
return FALSE;

SYSTEM_INFO sysInfo;
GetSystemInfo(&sysInfo);

for(int i=0;i<(sysInfo.dwNumberOfProcessors*2);i++)
{
WorkThread* pThread = new WorkThread(&WorkThreadFun,this);
if(pThread->Init())
m_pWorkThreadList->push_back(pThread);
else
{
delete pThread;
return FALSE;
}
}
return TRUE;
}

//创建监听端口，并关联完成端口，启动监听
bool NetCompletionPort::InitListenSocket()
{
m_listenSocket = WSASocket(PF_INET,SOCK_STREAM,IPPROTO_TCP,NULL,0,WSA_FLAG_OVERLAPPED); //创建监听服务端口

if(INVALID_SOCKET == m_listenSocket)
{
m_listenSocket = NULL;
return FALSE;
}

if(!CreateIoCompletionPort((HANDLE)m_listenSocket,m_comPort,0,0)) //关联完成端口
return FALSE;
SOCKADDR_IN SockAddr;
SockAddr.sin_family = AF_INET;
SockAddr.sin_port = htons(m_port);
SockAddr.sin_addr.s_addr = INADDR_ANY;

if(SOCKET_ERROR == bind(m_listenSocket,(PSOCKADDR)&SockAddr,sizeof(SockAddr))) //绑定
return FALSE;

if(SOCKET_ERROR == listen(m_listenSocket,10)) //监听
return FALSE;

DWORD dwTemp=0;

if(SOCKET_ERROR == WSAIoctl(m_listenSocket,SIO_GET_EXTENSION_FUNCTION_POINTER, //获取扩展函数调用
&g_GUIDAccep
tEx,sizeof(g_GUIDAcceptEx),&m_pAcceptEx,sizeof(m_pAcceptEx),
&dwTemp,NULL,NULL))
{
m_pAcceptEx = NULL;
return FALSE;
}