void CClientDlg::OnBnClickedStop() { SetAppState(ST_STOPING); for(size_t i = 0; i < m_vtClients.Size(); i++) { CClientSocket* pSocket = m_vtClients[i]; if(pSocket->Stop()) ::LogClientStop((DWORD)i + 1); else ASSERT(FALSE); } ::WaitWithMessageLoop(100L); CString strMsg; strMsg.Format( _T(" *** Summary: expect - %lld, send - %lld, recv - %lld"), m_llExpectReceived, m_llTotalSent, m_llTotalReceived); ::LogMsg(strMsg); if(m_llExpectReceived == m_llTotalSent && m_llTotalSent == m_llTotalReceived) strMsg.Format(_T(" *** Success: time consuming - %u millisecond !"), m_dwTimeconsuming); else strMsg.Format(_T(" *** Fail: manual terminated ? (or data lost)")); ::LogMsg(strMsg); SetAppState(ST_STOPED); }
UINT #else DWORD #endif WINAPI CClientSocket::WorkerThreadProc(LPVOID pv) { TRACE1("---------------> Worker Thread 0x%08X started <---------------\n", ::GetCurrentThreadId()); CClientSocket* pClient = (CClientSocket*)pv; HANDLE hEvents[] = {pClient->m_evSocket, pClient->m_evBuffer, pClient->m_evStop}; pClient->m_rcBuffer.Malloc(pClient->m_dwSocketBufferSize); while(pClient->HasStarted()) { DWORD retval = ::WSAWaitForMultipleEvents(3, hEvents, FALSE, WSA_INFINITE, FALSE); if(retval == WSA_WAIT_EVENT_0) { if(!pClient->ProcessNetworkEvent()) { if(pClient->HasStarted()) pClient->Stop(); break; } } else if(retval == WSA_WAIT_EVENT_0 + 1) { if(!pClient->SendData()) { if(pClient->HasStarted()) pClient->Stop(); break; } } else if(retval == WSA_WAIT_EVENT_0 + 2) break; else ASSERT(FALSE); } TRACE1("---------------> Worker Thread 0x%08X stoped <---------------\n", ::GetCurrentThreadId()); return 0; }