WIN网络编程-select(I/O模型)

//////////////////////////////////////////////////////
// select.cpp文件

//select的优点是程序能够在单个线程内同时处理多个套接字连接，但是增加

//到fd_set结构的套接字是有限制的。winsock2.h定义为64，在包含winsock2.h

//之前重新定义它是可以的，但是最大不能超过1024，并且此值太大，影响服

//务器性能，因为select返回之前会检查这些集合中的套接字，并移除没有未

//决I/O操作的套接字。

//s为套接字

//FD_ZERO(*set)初始化set为空集合

//FD_CLR(s,*set)从set移除s

//FD_ISSET(s,*set)检查s是不是set的成员,如果是返回true

//FD_SET(s,*set)增加套接字到集合

#include "../common/initsock.h"
#include <stdio.h>

CInitSock theSock;//初始化Winsock库
int main()
{
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_un.S_addr = INADDR_ANY;
// 绑定套节字到本地机器
if(::bind(sListen,(sockaddr*)&sin, sizeof(sin)) ==SOCKET_ERROR)
{
printf(" Failed bind()\n");
return -1;
}
// 进入监听模式
::listen(sListen, 5);

// select模型处理过程
// 1）初始化一个套节字集合fdSocket，添加监听套节字句柄到这个集合
fd_setfdSocket;// 所有可用套节字集合
FD_ZERO(&fdSocket);
FD_SET(sListen, &fdSocket);
while(TRUE)
{
//2）将fdSocket集合的一个拷贝fdRead传递给select函数，
//当有事件发生时，select函数移除fdRead集合中没有未决I/O操作的套节字句柄，然后返回。
fd_set fdRead = fdSocket;
int nRet = ::select(0,&fdRead, NULL, NULL, NULL);
if(nRet >0)
{
//3）通过将原来fdSocket集合与select处理过的fdRead集合比较，
//确定都有哪些套节字有未决I/O，并进一步处理这些I/O。
for(int i=0;i<(int)fdSocket.fd_count; i++)
{
if(FD_ISSET(fdSocket.fd_array[i],&fdRead))
{
if(fdSocket.fd_array[i]== sListen)//（1）监听套节字接收到新连接
{
if(fdSocket.fd_count< FD_SETSIZE)
{
sockaddr_inaddrRemote;
intnAddrLen = sizeof(addrRemote);
SOCKETsNew = ::accept(sListen, (SOCKADDR*)&addrRemote,&nAddrLen);
FD_SET(sNew,&fdSocket);
printf("接收到连接（%s）\n",::inet_ntoa(addrRemote.sin_addr));
}
else
{
printf("Too much connections! \n");
continue;
}
}
else
{
charszText[256];
intnRecv = ::recv(fdSocket.fd_array[i], szText, strlen(szText),0);
if(nRecv>0)//（2）可读
{
szText[nRecv]= '\0';
printf("接收到数据：%s\n", szText);
}
else//（3）连接关闭、重启或者中断
{
::closesocket(fdSocket.fd_array[i]);
FD_CLR(fdSocket.fd_array[i],&fdSocket);
}
}
}
}
}
else
{
printf("Failed select() \n");
break;
}
}
return 0;