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注册。