转载自:http://blog.csdn.net/ithzhang/article/details/8363951
Windows socket select模型开发。
套接字select模型是一种比较常用的IO模型。利用该模型可以使Windows socket应用程序可以同时管理多个套接字。
使用select模型,可以使当执行操作的套接字满足可读可写条件时,给应用程序发送通知。收到这个通知后,应用程序再去调用相应的Windows socket API去执行函数调用。
Select模型的核心是select函数。调用select函数检查当前各个套接字的状态。根据函数的返回值判断套接字的可读可写性。然后调用相应的Windows Sockets API完成数据的发送、接收等。
阻塞模式和非阻塞模式的优点和不足:
阻塞模式套接字执行IO操作时,如果执行操作的条件未满足,线程就会阻塞在调用的函数上。程序不得不处于等待状态,但是由于并不知道客户请求何时到来,因此函数在何时返回不得而知。
非阻塞模式套接字执行IO操作时,在任何时候函数都会立即返回。但程序员必须为此编写更多的代码。这增加了开发Windows socket应用程序的难度。另外由于不断的循环调用导致程序效率很低。
Select模型是Windows sockets中最常见的IO模型。它利用select函数实现IO 管理。通过对select函数的调用,应用程序可以判断套接字是否存在数据、能否向该套接字写入数据。
如:在调用recv函数之前,先调用select函数,如果系统没有可读数据那么select函数就会阻塞在这里。当系统存在可读或可写数据时,select函数返回,就可以调用recv函数接收数据了。
可以看出使用select模型,需要两次调用函数。第一次调用select函数第二次socket API。使用该模式的好处是:可以等待多个套接字。
select函数
int select (
Int nfds, //被忽略。传入0即可。
fd_set *readfds, //可读套接字集合。
fd_set *writefds, //可写套接字集合。
fd_set *exceptfds, //错误套接字集合。
const struct timeval*timeout); //select函数等待时间
该函数返回处于就绪态并且已经被包含在fd_set结构中的套接字总数。如果超时则返回0。
- 第一个参数nfds被忽略。
- 第二个参数readfds,可读性套接字集合指针。
- 第三个参数writefds,可写性套接字集合指针。
- 第四个参数exceptfds,检查错误套接字集合指针。
- 第五个参数timeout,等待时间。
fd_set结构是一个结构体:
typedef struct fd_set
{
u_int fd_count;
socket fd_array[FD_SETSIZE];
} fd_set;
fd_cout 表示该集合套接字数量。最大为64.
fd_array 套接字数组。
readfds 数组将包括满足以下条件的套接字:
- 有数据可读。此时在此套接字上调用recv,立即收到对方的数据。
- 连接已经关闭、重设或终止。
- 正在请求建立连接的套接字。此时调用accept函数会成功。
writefds 数组包含满足下列条件的套接字:
- 有数据可以发出。此时在此套接字上调用send,可以向对方发送数据。
- 调用connect函数,并连接成功的套接字。
exceptfds 数组将包括满足下列条件的套接字:
- 调用connection函数,但连接失败的套接字。
- 有带外(out of band)数据可读。
select函数的使用:
在调用select函数对套接字进行监视之前,必须将要监视的套接字分配给上述三个数组中的一个。然后调用select函数,再次判断需要监视的套接字是否还在原来的集合中。就可以知道该集合是否正在发生IO操作。
例如:应用程序想要判断某个套接字是否存在可读的数据,需要进行如下步骤:
- 将该套接字加入到readfds集合。
- 以readfds作为第二个参数调用select函数。
- 当select函数返回时,应用程序判断该套接字是否仍然存在于readfds集合。
- 如果该套接字存在与readfds集合,则表明该套接字可读。此时就可以调用recv函数接收数据。否则,该套接字不可读。
在调用select函数时,readfds、writefds和exceptfds三个参数至少有一个为非空。并且在该非空的参数中,必须至少包含一个套接字。否则select函数将没有任何套接字可以等待。
timeval结构体用于定义select的等待时间:
structure timeval
{
long tv_sec; //秒。
long tv_usec; //毫秒。
};
- 当 timeval 为空指针时,select 会一直等待,直到有符合条件的套接字时才返回。
- 当 tv_sec 和 tv_usec 之和为 0 时,无论是否有符合条件的套接字,select 都会立即返回。
- 当 tv_sec 和 tv_usec 之和为非 0 时,如果在等待的时间内有套接字满足条件,则该函数将返回符合条件的套接字。如果在等待的时间内没有套接字满足设置的条件,则 select 会在时间用完时返回,并且返回值为 0。
为了方便使用,windows sockets 提供了下列宏,用来对 fd_set 进行一系列操作。使用以下宏可以使编程工作简化:
FD_CLR(s, *set); // 从set集合中删除s套接字。
FD_ISSET(s, *set); // 检查s是否为set集合的成员。
FD_SET(s, *set); // 将套接字加入到set集合中。
FD_ZERO(*set); // 将set集合初始化为空集合。
在开发Windows sockets应用程序时,通过下面的步骤,可以完成对套接字的可读写判断:
- 使用FD_ZERO初始化套接字集合。如:FD_ZERO(&readfds);
- 使用FD_SET将某套接字放到readfds内。如:FD_SET(s, &readfds);
- 以 readfds 为第二个参数调用 select 函数。select 在返回时会返回所有 fd_set 集合中套接字的总个数,并对每个集合进行相应的更新。将满足条件的套接字放在相应的集合中。
- 使用FD_ISSET判断s是否还在某个集合中。如:FD_ISSET(s, &readfds);
- 调用相应的 Windows socket api 对某套接字进行操作。
select 返回后会修改每个 fd_set 结构。删除不存在的或没有完成 IO 操作的套接字。这也正是在第四步中可以使用 FD_ISSET 来判断一个套接字是否仍在集合中的原因。
看例子,该例演示了一个服务器程序使用 select 模型管理套接字:
SOCKET listenSocket;
SOCKET acceptSocket;
FD_SET socketSet;
FD_SET writeSet;
FD_SET readSet;
FD_ZERO(&socketSet);
FD_SET(listenSocket,&socketSet);
while(true)
{
FD_ZERO(&readSet);
FD_ZERO(&writeSet);
readSet = socketSet;
writeSet = socketSet;
// 同时检查套接字的可读可写性。
// 为等待时间传入 : NULL->永久等待; 0->立即返回。
// 不要误用!!!
int ret = select(0, &readSet, &writeSet, NULL, NULL);
if(ret == SOCKET_ERROR)
{
return false;
}
sockaddr_in addr;
int len = sizeof(addr);
// 是否存在客户端的连接请求。
if(FD_ISSET(listenSocket, &readSet)) // 在readset中会返回已经调用过listen的套接字。
{
acceptSocket = accept(listenSocket, (sockaddr*)&addr, &len);
if(acceptSocket == INVALID_SOCKET)
{
return false;
}
else
{
FD_SET(acceptSocket, &socketSet);
}
}
for(int i=0; i<socketSet.fd_count; i++)
{
if(FD_ISSET(socketSet.fd_array[i], &readSet))
{
// 调用recv,接收数据。
}
if(FD_ISSET(socketSet.fd_array[i]), &writeSet)
{
// 调用send,发送数据。
}
}
}
以下展示了一个客户端程序使用 select 模型的用法。注意与服务器用法相区别。主要区别就是不可能有请求进来,也就不需要使用 allsocketfds。仅仅对一个套接字进行判断:
CRemoteFileDownloadClientDlg* pdlg = (CRemoteFileDownloadClientDlg*)ppram;
FD_SET readfds;
FD_SET writefds;
while(pdlg->m_IsConnected)
{
FD_ZERO(&readfds);
FD_ZERO(&writefds);
FD_SET(pdlg->m_ServerSocket, &readfds);
FD_SET(pdlg->m_ServerSocket, &writefds);
int ret = select(0,&readfds,&writefds,NULL,NULL); //NULL为无限等待。0立即返回。
if(ret > 0)
{
if(FD_ISSET(pdlg->m_ServerSocket, &readfds)); //注意与服务器此处写法相区别。
{
pdlg->recvData();
}
if(FD_ISSET(pdlg->m_ServerSocket, &writefds))
{
//可写。
pdlg->sendData();
}
}
}
本文参考自《精通Windows sockets网络开发–基于Visual C++实现》孙海民著
2012.12.21 14:59山西大同