select模型
具体编程流程:
1、 初始化套接字集合fdSocket,向这个集合添加监听套接字。
2、 将fdSocket集合拷贝到一个每一次临时要读取的fdRead集合,然后将fdRead传递给select函数。当有事件发生 时,select函数移除集合中没有未决I/O操作的套接字句柄,然后返回。
3、 比较原来fdSocket集合与select处理过的fdRead集合,确定哪些套接字有未决I/O,并进一步处理这些I/O。
4、 回到第2步继续进行处理
#define _WIN32_WINNT 0x0400
#include<windows.h>
#include<cstdio>
#include"InitSocket.h"
CInitSock initSock ; //进入main函数前已经进行了初始化
int main(void)
{
USHORT nPort = 4567 ; //此服务器监听的端口号
//创建监听套接字
SOCKET sListen = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP) ;
sockaddr_in sin ;
sin.sin_family = AF_INET ;
sin.sin_port = htons(nPort) ;
sin.sin_addr.s_addr = INADDR_ANY ;
//绑定套接字到本地机器
if(bind(sListen,(sockaddr* )&sin,sizeof(sin)) == SOCKET_ERROR)
{
printf("Failed bind() \n") ;
return 0 ;
}
//进入监听模式
listen(sListen,5) ;
//select模型处理过程
//初始化一个套接字集合fdSocket,添加监听套接字句柄到这个集合
fd_set fdSocket ; //所有可用套接字集合
FD_ZERO(&fdSocket) ;
FD_SET(sListen,&fdSocket) ;
while(TRUE)
{
//2将fdSocket集合的一个拷贝fdRead传递给select函数
//当有事件发生时,select函数移除fdRead集合中没有未决I/O操作的套接字句柄,然后返回。
timeval reportTime ;
reportTime.tv_sec = 5 ;
reportTime.tv_usec = 5000 ;
fd_set fdRead = fdSocket ;
int nRet = select(0,&fdRead,NULL,NULL,&reportTime) ;
if(nRet > 0)
{
//确定都有哪些套接字有未决I/O,并进一步处理这些I/O
//通过将原来fdSocket集合与select处理过的fdRead集合比较
for(int i = 0 ; i < (int)fdSocket.fd_count ; ++i)
{
if(FD_ISSET(fdSocket.fd_array[i],&fdRead))
{
if(fdSocket.fd_array[i] == sListen) //监听套接字接收到新连接
{
if(fdSocket.fd_count < FD_SETSIZE)
{
sockaddr_in addrRemote ;
int nAddrLen = sizeof(addrRemote);
SOCKET sNew = accept(sListen,(SOCKADDR *)&addrRemote,&nAddrLen) ;
FD_SET(sNew,&fdSocket) ;
printf("接收到连接(%s)\n",inet_ntoa(addrRemote.sin_addr)) ;
}
else
{
printf("Too much connectinos!\n") ;
continue ;
}
}
else //读事件,结束事件也是
{
char szText[256] ;
// int nRecv = recv(fdSocket.fd_array[i],szText,strlen(szText),0);
int nRecv = recv(fdSocket.fd_array[i],szText,sizeof(szText),0);
if(nRecv > 0) //可读
{
szText[nRecv] = '\0' ;
printf("接收到数据:%s\n",szText) ;
}
else //连接关闭、重启或者中断
{
printf("关闭一个连接\n") ;
closesocket(fdSocket.fd_array[i]) ;
FD_CLR(fdSocket.fd_array[i],&fdSocket) ;
}
}
}
}
}
else if(0 == nRet) //测试所用,检测服务器是否能够读出客户端退出的消息
{
printf("Time Out: %d\n",fdSocket.fd_count) ;
continue ;
}
else
{
printf("Failed select() \n") ;
break ;
}
}
return 0 ;
}
总结:
使用select模型的好处是可以在单个线程内同时处理多个套接字连接,这避免了阻塞模型下的线程膨胀问题。但是,添加到fd_set的结构的套接字数量是有限制的,默认情况下,最大值FD_SETSIZE,它在winsock2.h文件中定义为64。为了增加套接字的数量,应用程序可以将FD_SETSIZE定义为更大值(这个定义必须在winsock2.h之前出现)。不过,自定义的值也不能超过Winsock下层提供者的限制(通常是1024)。(或者采用多个工作线程的方法,select另外64个套接字)
另外如果FD_SETSIZE的数值过大的话,服务器性能也会受到影响。例如,如果有1000个套接字,那么在调用select之前,必须设置这1000个套接字,select返回之后,又必须检查这1000个套接字。