WinSocket 非阻塞之select用法

winSocket下有4个非阻塞的方式;

select模型是winSocket下最常见的i/o模型,通过select可以判断是否存在一个或多个有效的socket连接状态,判断套接字是否可以读取或者写入。他既能防止socket处于阻塞状态下时候,最后一次io操作以后进入阻塞状态,也可以套接字处于非阻塞状态下出现的wsaewouldblock错误,

select  函数原型:

int select (
  int nfds, //用来与Berkely socket兼容, 通常写0   fd_set* readfds, //检查可读性的socket集合   fd_set* writefds, //检查可写性的socket集合   fd_set* exceptfds,//检查异常的socket集合   const struct timeval* timeout // 设置超时时间 
);
返回值:
    readfds, writefds, exceptfds中被检查复合条件的socket总数。
 
注意:
   由于select每次返回时,readfds, writefds, exceptfds都会被改变,所以,下次调用select前,要重新初始化fds。
对于三种检查各自如下:
readfds:
> 若当前socket处在listen状态,则有个连接到来,调用accept将成功建立连接
>若连接已经建立,则意味着有新数据到来,此时可以调用recv接收数据
>对方连接终止,调用recv,根据返回值判断是否是正常终止: 0 对方正常终止, ErrorCode 对方重置连接或强制断开连接
writefds:
>当前socket调用了connect, 则成功连接
>当前socket调用了send,则能成功发送出去
exceptfds:
>当前socket调用connect,则连接失败
>OOB 数据可读(暂时没研究,OOB是什么个东东)
以下附上简单的测试代码:
 
设置socket为非阻塞模式的时候要先设置
 
int status = ioctlsocket(socket s,FIOBIO,&cmd);
如果status 不等于socket_error表示非阻塞模式设置成功
 
server端代码:
 

#include "SocketInit.h"
#include "iostream"
using namespace std;

CInitSock sockInit;

int main()
{
 SOCKET s = SocketFactory::getTCPSocket();
 
 hostent *thisHost = gethostbyname("");
 char *localIP = inet_ntoa (*(struct in_addr *)*thisHost->h_addr_list);
 const int localIPInt = inet_addr(localIP);
 const unsigned short linkPort = ntohs(4001);//本地监听

 sockaddr_in servAddr;
 servAddr.sin_family = AF_INET;
 servAddr.sin_port = linkPort;
 servAddr.sin_addr.S_un.S_addr = localIPInt;

 ::bind(s, (sockaddr *)&servAddr, sizeof(servAddr));
 ::listen(s, 1);

 sockaddr_in clientAddr;
 int lenC = sizeof(clientAddr);

 cout<<"TCP server: "<<endl;
 SOCKET clientSocket = ::accept(s, (sockaddr *)&clientAddr, &lenC);
 cout<<"Recv a connection..."<<endl;

 timeval waitTime;
 waitTime.tv_sec=2;
 waitTime.tv_usec=0;
 while(true)
 {
  fd_set fdSocket;
  FD_ZERO(&fdSocket);
  FD_SET(clientSocket, &fdSocket);

  int nRet = ::select(0, &fdSocket, 0, 0, &waitTime);
  if(nRet<0)
  {
   cout<<"Faild Select()"<<endl;
   exit(-1);
  }
  else if(nRet ==0)
  {
   cout<<"time out()"<<endl;
   continue;
  }
  else
  {
   if(FD_ISSET(clientSocket, &fdSocket))
   {
    char rBuff[100];
    int nRec=::recv(clientSocket, rBuff, 100, 0);
    cout<<"Recv from client data: "<<nRec<<endl;
    if(nRec < 0)
    {
     cout<<"Not normal close: "<<::GetLastError()<<endl;
     break;
    }
    else if(nRec == 0)
    {
     cout<<"Normal close"<<endl;
     break;
    }
    else
    {
     rBuff[nRec]=0;
     cout<<"Data : "<<rBuff<<endl;
    }

   }
  }
 }

 ::closesocket(clientSocket);
 ::closesocket(s);

 return 0;
}

客户端代码;

#include "SocketInit.h"
#include "iostream"
using namespace std;

CInitSock sockInit;
int main()
{
 
 SOCKET s = SocketFactory::getTCPSocket();

 hostent *thisHost = gethostbyname("");
 char *localIP = inet_ntoa (*(struct in_addr *)*thisHost->h_addr_list);
 const int localIPInt = inet_addr(localIP);
 const unsigned short linkPort = ntohs(4001);//本地监听

 sockaddr_in servAddr;
 servAddr.sin_family = AF_INET;
 servAddr.sin_port = linkPort;
 servAddr.sin_addr.S_un.S_addr = localIPInt;

 cout<<"Client : "<<endl;
 if(::connect(s, (sockaddr *)&servAddr, sizeof(servAddr)) == -1)
 {
  printf("Faild connect(): %d\n", ::GetLastError());
  return 0;
 }

 char sendStr[100];
 while(true)
 {
  cout<<"Input send data('q' to normal quit): ";
  cin.getline(sendStr, 100);
  if(sendStr[0]=='q') break;
   ::send(s, sendStr, strlen(sendStr), 0);
 }

 ::closesocket(s);

 return 0;
}

本文使用Blog_Backup未注册版本导出,请到soft.pt42.com注册。

原文地址:https://www.cnblogs.com/zjypp/p/2319407.html