ARCEmu项目wow私服服务器网络模块分析（二）

 

/* * Multiplatform Async Network Library * Copyright (c) 2007 Burlex * * SocketMgr - iocp-based SocketMgr for windows. * */#include "Network.h"#ifdef CONFIG_USE_IOCP#include "../CrashHandler.h"initialiseSingleton(SocketMgr);SocketMgr::SocketMgr(){WSADATA wsaData;WSAStartup(MAKEWORD(2, 0), &wsaData);m_completionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, (ULONG_PTR)0, 0);}SocketMgr::~SocketMgr(){}void SocketMgr::SpawnWorkerThreads(){SYSTEM_INFO si;GetSystemInfo(&si);threadcount = si.dwNumberOfProcessors;sLog.outDetail("IOCP: Spawning %u worker threads.", threadcount);for(long x = 0; x < threadcount; ++x)ThreadPool.ExecuteTask(new SocketWorkerThread());}bool SocketWorkerThread::run(){THREAD_TRY_EXECUTIONHANDLE cp = sSocketMgr.GetCompletionPort();DWORD len;Socket* s;OverlappedStruct* ov;LPOVERLAPPED ol_ptr;while(true){#ifndef _WIN64if(!GetQueuedCompletionStatus(cp, &len, (LPDWORD)&s, &ol_ptr, 10000))#elseif(!GetQueuedCompletionStatus(cp, &len, (PULONG_PTR)&s, &ol_ptr, 10000))#endifcontinue;ov = CONTAINING_RECORD(ol_ptr, OverlappedStruct, m_overlap);if(ov->m_event == SOCKET_IO_THREAD_SHUTDOWN){delete ov;return true;}if(ov->m_event < NUM_SOCKET_IO_EVENTS)ophandlers[ov->m_event](s, len);}THREAD_HANDLE_CRASHreturn true;}void HandleReadComplete(Socket* s, uint32 len){//s->m_readEvent= NULL;if(!s->IsDeleted()){s->m_readEvent.Unmark();if(len){s->readBuffer.IncrementWritten(len);s->OnRead();s->SetupReadEvent();}elses->Delete();  // Queue deletion.}}void HandleWriteComplete(Socket* s, uint32 len){if(!s->IsDeleted()){s->m_writeEvent.Unmark();s->BurstBegin();// Locks->writeBuffer.Remove(len);if(s->writeBuffer.GetContiguiousBytes() > 0)s->WriteCallback();elses->DecSendLock();s->BurstEnd();  // Unlock}}void HandleShutdown(Socket* s, uint32 len){}void SocketMgr::CloseAll(){list<Socket*> tokill;socketLock.Acquire();for(set<Socket*>::iterator itr = _sockets.begin(); itr != _sockets.end(); ++itr)tokill.push_back(*itr);socketLock.Release();for(list<Socket*>::iterator itr = tokill.begin(); itr != tokill.end(); ++itr)(*itr)->Disconnect();size_t size;do{socketLock.Acquire();size = _sockets.size();socketLock.Release();}while(size);}void SocketMgr::ShutdownThreads(){for(int i = 0; i < threadcount; ++i){OverlappedStruct* ov = new OverlappedStruct(SOCKET_IO_THREAD_SHUTDOWN);PostQueuedCompletionStatus(m_completionPort, 0, (ULONG_PTR)0, &ov->m_overlap);}}void SocketMgr::ShowStatus(){sLog.outString("sockets count = %u", socket_count.GetVal());}#endif