select2

 DWORD WINAPI WorkerThread(LPVOID lpParam)
{
 int i;
 fd_set fdread;
 int ret;
 struct timeval tv = {1, 0};
 char szMessage[MSGSIZE];
 while (TRUE)
 {
  FD_ZERO(&fdread);
  for (i = 0; i < g_iTotalConn; i++)
  {
   FD_SET(g_CliSocketArr[i], &fdread);
  }
  // We only care read event
  ret = select(0, &fdread, NULL, NULL, &tv);
  if (ret == 0)
  {
   // Time expired
   continue;
  }
  for (i = 0; i < g_iTotalConn; i++)
  {
   if (FD_ISSET(g_CliSocketArr[i], &fdread))
   {
    // A read event happened on g_CliSocketArr
    ret = recv(g_CliSocketArr[i], szMessage, MSGSIZE, 0);
    if (ret == 0 || (ret == SOCKET_ERROR && WSAGetLastError() == WSAECONNRESET))
    {
     // Client socket closed
     printf("Client socket %d closed.\n", g_CliSocketArr[i]);
     closesocket(g_CliSocketArr[i]);
     if (i < g_iTotalConn-1)
     {
      g_CliSocketArr[i--] = g_CliSocketArr[--g_iTotalConn];
     }
    }
    else
    {
     // We reveived a message from client
     szMessage[ret] = '\0';
     send(g_CliSocketArr[i], szMessage, strlen(szMessage), 0);
    }
   }
  }
 }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章