天天看点

Windows下select模型

阻塞模式和非阻塞模式的优点和不足:

     阻塞模式套接字执行IO操作时,如果执行操作的条件未满足,线程就会阻塞在调用的函数上。程序不得不处于等待状态,但是由于并不知道客户请求何时到来,因此函数在何时返回不得而知。

    非阻塞模式套接字执行IO操作时,在任何时候函数都会立即返回。但程序员必须为此编写更多的代码。这增加了开发Windows socket应用程序的难度。另外由于不断的循环调用导致程序效率很低。

     select模型是一种比较常用的IO模型。利用该模型可以使Windows socket应用程序可以同时管理多个套接字。 使用select模型,可以使当执行操作的套接字满足可读可写条件时,给应用程序发送通知。收到这个通知后,应用程序再去调用相应的Windows socket API去执行函数调用。

     Select模型的核心是select函数。调用select函数检查当前各个套接字的状态。根据函数的返回值判断套接字的可读可写性。然后调用相应的Windows Sockets API完成数据的发送、接收等。

     Select模型是Windows sockets中最常见的IO模型。它利用select函数实现IO 管理。通过对select函数的调用,应用程序可以判断套接字是否存在数据、能否向该套接字写入数据。   如:在调用recv函数之前,先调用select函数,如果系统没有可读数据那么select函数就会阻塞在这里。当系统存在可读或可写数据时,select函数返回,就可以调用recv函数接收数据了。 

select函数

[cpp]  view plain  copy

  1. int select (  
  2.    Int nfds,//被忽略。传入0即可。  
  3.    fd_set *readfds,//可读套接字集合。  
  4.    fd_set *writefds,//可写套接字集合。  
  5.    fd_set *exceptfds,//错误套接字集合。  
  6.    const struct timeval*timeout);//select函数等待时间。  

   该函数返回处于就绪态并且已经被包含在fd_set结构中的套接字总数。如果超时则返回0。

    第一个参数nfds被忽略。

    第二个参数readfds,可读性套接字集合指针。

    第三个参数writefds,可写性套接字集合指针。

    第四个参数exceptfds,检查错误套接字集合指针。

    第五个参数timeout,等待时间。

fd_set结构是一个结构体。

[cpp]  view plain  copy

  1. typedef struct fd_set  
  2. {  
  3.      u_int fd_count;  
  4.      socket fd_array[FD_SETSIZE];  
  5. }fd_set; 

fd_cout表示该集合套接字数量。最大为64.

fd_array套接字数组。

select函数中需要三个fd_set结构:

    一:准备接收数据的套接字集合,即可读性集合。

    二:准备发送数据的套接字集合,即可写性集合。

     在select函数返回时,会在fd_set结构中,填入相应的套接字。

readfds数组将包括满足以下条件的套接字:

     1:有数据可读。此时在此套接字上调用recv,立即收到对方的数据。

     2:连接已经关闭、重设或终止。

     3:正在请求建立连接的套接字。此时调用accept函数会成功。

writefds数组包含满足下列条件的套接字:

    1:有数据可以发出。此时在此套接字上调用send,可以向对方发送数据。

    2:调用connect函数,并连接成功的套接字。

exceptfds数组将包括满足下列条件的套接字:

    1:调用connection函数,但连接失败的套接字。

    2:有带外(out of band)数据可读。

select函数的使用:

    在调用select函数对套接字进行监视之前,必须将要监视的套接字分配给上述三个数组中的一个。然后调用select函数,再次判断需要监视的套接字是否还在原来的集合中。就可以知道该集合是否正在发生IO操作。

    例如:应用程序想要判断某个套接字是否存在可读的数据,需要进行如下步骤:

    1:将该套接字加入到readfds集合。

    2:以readfds作为第二个参数调用select函数。

    3:当select函数返回时,应用程序判断该套接字是否仍然存在于readfds集合。

    4:如果该套接字存在与readfds集合,则表明该套接字可读。此时就可以调用recv函数接收数据。否则,该套接字不可读。

     在调用select函数时,readfds、writefds和exceptfds三个参数至少有一个为非空。并且在该非空的参数中,必须至少包含一个套接字。否则select函数将没有任何套接字可以等待。

timeval结构体用于定义select的等待时间。

[cpp]  view plain  copy

  1. structure timeval  
  2. {  
  3.     long tv_sec;//秒。  
  4.     long tv_usec;//毫秒。  
  5. }; 

    当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应用程序时,通过下面的步骤,可以完成对套接字的可读写判断:

    1:使用FD_ZERO初始化套接字集合。如FD_ZERO(&readfds);

    2:使用FD_SET将某套接字放到readfds内。如:     FD_SET(s,&readfds);

    3:以readfds为第二个参数调用select函数。select在返回时会返回所有fd_set集合中套接字的总个数,并对每个集合进行相应的更新。将满足条件的套接字放在相应的集合中。

    4:使用FD_ISSET判断s是否还在某个集合中。如:  FD_ISSET(s,&readfds);

     5:调用相应的Windows socket api 对某套接字进行操作。

     select返回后会修改每个fd_set结构。删除不存在的或没有完成IO操作的套接字。这也正是在第四步中可以使用FD_ISSET来判断一个套接字是否仍在集合中的原因。

看例子,该例演示了一个服务器程序使用select模型管理套接字。 

#include <WinSock2.h>  
#include <Windows.h>  
#include <MSWSock.h>  
#include <stdio.h>  
#include <map>  
using namespace std;  


#pragma comment(lib,"Ws2_32.lib")  
#pragma comment(lib,"Mswsock.lib")  

 
int main()  
{  
    WSAData wsaData;  
    if(0 != WSAStartup(MAKEWORD(2,2),&wsaData))  
    {  
        printf("初始化失败!%d\n",WSAGetLastError());  
        Sleep(5000);  
        return -1;  
    }  
	
    USHORT nport = 6000;  
    SOCKET sListen = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);  
	
	u_long ul = 1;  
    ioctlsocket(sListen,FIONBIO,&ul);  
	
    sockaddr_in sin;  
    sin.sin_family = AF_INET;  
    sin.sin_port = htons(nport);  
    sin.sin_addr.S_un.S_addr = ADDR_ANY;  
	
	
    if(SOCKET_ERROR == bind(sListen,(sockaddr*)&sin,sizeof(sin)))  
    {  
        printf("bind failed!%d\n",WSAGetLastError());  
        Sleep(5000);  
        return -1;  
    }  
	
	
    listen(sListen,5);  
	
	
    //1)初始化一个套接字集合fdSocket,并将监听套接字放入  
    fd_set socketSet;  
    FD_ZERO(&socketSet);  
    FD_SET(sListen,&socketSet);  
	
    TIMEVAL time={1,0};  
    char buf[4096];  
	
	
    fd_set    readSet;  
    FD_ZERO(&readSet);  
	
    fd_set    writeSet;  
    FD_ZERO(&readSet);  
	 
	
    while(true)  
    {  
        //2)将fdSocket的一个拷贝fdRead传给select函数  
		readSet = socketSet;   
		writeSet = socketSet;   
		
		//同时检查套接字的可读可写性。
        int   nRetAll = select(0,&readSet,&writeSet,NULL,NULL/*&time*/);//若不设置超时则select为阻塞  
        if(nRetAll >0 )   //-1
		{  
			//是否存在客户端的连接请求。  
			if(FD_ISSET(sListen ,&readSet))//在readset中会返回已经调用过listen的套接字。  
			{  
				
				if(socketSet.fd_count < FD_SETSIZE)  
				{  
					sockaddr_in addrRemote;  
					int nAddrLen = sizeof(addrRemote);  
					SOCKET sClient = accept(sListen,(sockaddr*)&addrRemote,&nAddrLen); 
					if(sClient!=INVALID_SOCKET)  
					{ 
						FD_SET(sClient,&socketSet);  
						printf("\n接收到连接:(%s)",inet_ntoa(addrRemote.sin_addr));  
					}
				}  
				else  
				{  
					printf("连接数量已达上限!\n");  
					continue;  
				}   
			}  
			
			
			for(int i=0;i<socketSet.fd_count;i++)  
			{  
				if(FD_ISSET(socketSet.fd_array[i], &readSet)  )  
				{  
					//调用recv,接收数据。 
					int nRecv = recv(socketSet.fd_array[i],buf,4096,0);  
					if(nRecv > 0)  
					{  
						buf[nRecv] = 0;  
						printf("\nrecv  %d :  %s", socketSet.fd_array[i],buf);  
					}  
				}  

				if(FD_ISSET(socketSet.fd_array[i] , &writeSet)  )
				{   

				      //调用send,发送数据。
					    char buf[]="hello!"; 
                        int nRet = send(socketSet.fd_array[i],buf, strlen(buf)+1 ,0);   
                        if(nRet <= 0)  
                        {  
                            if(GetLastError() == WSAEWOULDBLOCK)  
                            {  
								//do nothing  
                            }  
                            else  
                            {  
                                closesocket(socketSet.fd_array[i]);   
                                FD_CLR(socketSet.fd_array[i],&socketSet);  
                            }   
						}
						else 
						{
						  	printf("\nsend  hello!");
						}
				}  
			}  
			
		}  
        else if(nRetAll == 0)  
        {  
			printf("time out!\n");  
        }  
        else  
        {  
            printf("select error!%d\n",WSAGetLastError());  
            Sleep(5000);  
            break;  
        }  
		
		 Sleep(1000);

    }  
    closesocket(sListen);  
    WSACleanup();  
}  
           

客户端

#include<stdlib.h>
#include<WINSOCK2.H>
 #include <windows.h> 
#include <process.h>  

#include<iostream>
#include<string>
using namespace std;

#define BUF_SIZE 64
#pragma comment(lib,"WS2_32.lib")


void recv(PVOID pt)  
{  
	    SOCKET  sHost=  *((SOCKET *)pt);

      	while(true)
		{
		    char buf[BUF_SIZE];//清空接收数据的缓冲区
			memset(buf,0 , BUF_SIZE);
			int retVal=recv(sHost,buf,sizeof(buf),0);
			if(SOCKET_ERROR==retVal)
			{
				int  err=WSAGetLastError();
				//无法立即完成非阻塞Socket上的操作
				if(err==WSAEWOULDBLOCK)
				{
					Sleep(1000);
					printf("\nwaiting  reply!");
					continue;
				}
				else if(err==WSAETIMEDOUT||err==WSAENETDOWN|| err==WSAECONNRESET)//已建立连接
				{
					printf("\nrecv failed!");
					closesocket(sHost);
					WSACleanup();
					return  ;
				}

			}

				Sleep(100);

	        	printf("\n%s", buf);
			 //break;
		} 
}  


int main()
{
	WSADATA wsd;
	SOCKET sHost;
	SOCKADDR_IN servAddr;//服务器地址
	int retVal;//调用Socket函数的返回值
	char buf[BUF_SIZE];
	//初始化Socket环境
	if(WSAStartup(MAKEWORD(2,2),&wsd)!=0)
	{
		printf("WSAStartup failed!\n");
		return -1;
	}
	sHost=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
	//设置服务器Socket地址
	servAddr.sin_family=AF_INET;
	servAddr.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
	//在实际应用中,建议将服务器的IP地址和端口号保存在配置文件中
	servAddr.sin_port=htons(6000);
	//计算地址的长度
	int sServerAddlen=sizeof(servAddr);
	
	
	
 	//调用ioctlsocket()将其设置为非阻塞模式
   int iMode=1;
	retVal=ioctlsocket(sHost,FIONBIO,(u_long FAR*)&iMode); 
	
	
	if(retVal==SOCKET_ERROR)
	{
		printf("ioctlsocket failed!");
		WSACleanup();
		return -1;
	}
	
	
	//循环等待
	while(true)
	{
		//连接到服务器
		retVal=connect(sHost,(LPSOCKADDR)&servAddr,sizeof(servAddr));
		if(SOCKET_ERROR==retVal)
		{
			int err=WSAGetLastError();
			//无法立即完成非阻塞Socket上的操作
			if(err==WSAEWOULDBLOCK||err==WSAEINVAL)
			{
				Sleep(1);
				printf("check  connect!\n");
				continue;
			}
			else if(err==WSAEISCONN)//已建立连接
			{
				break;
			}
			else
			{
				printf("connection failed!\n");
				closesocket(sHost);
				WSACleanup();
				return -1;
			}
		}
	}
	
	
    unsigned long     threadId=_beginthread(recv,0,&sHost);//启动一个线程接收数据的线程   


	
	while(true)
	{
		//向服务器发送字符串,并显示反馈信息
		printf("input a string to send:\n");
		std::string str;
		//接收输入的数据
		std::cin>>str;
		//将用户输入的数据复制到buf中
		ZeroMemory(buf,BUF_SIZE);
		strcpy(buf,str.c_str());
		if(strcmp(buf,"quit")==0)
		{
			printf("quit!\n");
			break;
		}
		
		while(true)
		{
			retVal=send(sHost,buf,strlen(buf),0);
			if(SOCKET_ERROR==retVal)
			{
				int err=WSAGetLastError();
				if(err==WSAEWOULDBLOCK)
				{
					//无法立即完成非阻塞Socket上的操作
					Sleep(5);
					continue;
				}
				
				else
				{
					printf("send failed!\n");
					closesocket(sHost);
					WSACleanup();
					return -1;
				}
			}
			break;
		}
		  
		
	}
	
	return 0;
}
           

继续阅读