Socket编程之Select

2 篇文章 0 订阅
1 篇文章 0 订阅

 Select的函数格式(Unix系统下的伯克利socket编程,和windows下的略有区别,体现两个方面:一是select函数的第一个参数,在windows下可以忽略,但在linux下必须设为最大文件描述符加1;二是结构fd_set在两个系统里定义不一样):

int select(int maxfdp,fd_set *readfds,fd_set *writefds,fd_set *errorfds,struct timeval *timeout);
/*参数列表
int maxfdp是一个整数值,是指集合中所有文件描述符的范围,即所有文件描述符的最大值加1,不能错!在Windows中这个参数的值无所谓,可以设置不正确。 
  
fd_set *readfds是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的读变化的,即我们关心是否可以从这些文件中读取数据了,如果这个集合中有一个文件可读,select就会返回一个大于0的值,表示有文件可读,如果没有可读的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的读变化。 
  
fd_set *writefds是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的写变化的,即我们关心是否可以向这些文件中写入数据了,如果这个集合中有一个文件可写,select就会返回一个大于0的值,表示有文件可写,如果没有可写的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的写变化。 
  
fd_set *errorfds同上面两个参数的意图,用来监视文件错误异常。 
  
struct timeval* timeout是select的超时时间,这个参数至关重要,它可以使select处于三种状态:
第一,若将NULL以形参传入,即不传入时间结构,就是将select置于阻塞状态,一定等到监视文件描述符集合中某个文件描述符发生变化为止;
第二,若将时间值设为0秒0毫秒,就变成一个纯粹的非阻塞函数,不管文件描述符是否有变化,都立刻返回继续执行,文件无变化返回0,有变化返回一个正值;
第三,timeout的值大于0,这就是等待的超时时间,即 select在timeout时间内阻塞,超时时间之内有事件到来就返回了,否则在超时后不管怎样一定返回,返回值同上述。
struct timeval
{
time_t tv_sec; //second 秒
time_t tv_usec; //microsecond 微妙
};
*/
/*
返回值: 
负值:select错误
正值:某些文件可读写或出错
0:等待超时,没有可读写或错误的文件
*/
struct fd_set可以理解为一个集合,这个集合中存放的是文件描述符(file descriptor),即文件句柄,这可以是我们所说的普通意义的文件,当然Unix下任何设备、管道、FIFO等都是文件形式,全部包括在内,所以毫无疑问一个socket就是一个文件,socket句柄就是一个文件描述符。fd_set集合可以通过一些宏由人为来操作,比如清空集合 FD_ZERO(fd_set *),将一个给定的文件描述符加入集合之中FD_SET(int ,fd_set *),将一个给定的文件描述符从集合中删除FD_CLR(int ,fd_set*),检查集合中指定的文件描述符是否可以读写FD_ISSET(int ,fd_set* )。
在winsock.h头文件中共定义了四个宏来操作描述字集。FD_SETSIZE变量用于确定一个集合中最多有多少描述字(FD_SETSIZE缺省值为64,可在包含winsock.h前用#define FD_SETSIZE来改变该值)。对于内部表示,fd_set被表示成一个套接口的队列,最后一个有效元素的后续元素为INVAL_SOCKET。宏为:
   FD_CLR(s,*set):从集合set中删除描述字s。
   FD_ISSET(s,*set):若s为集合中一员,非零;否则为零。
   FD_SET(s,*set):向集合添加描述字s。
   FD_ZERO(*set):将set初始化为空集NULL。
在有了select后可以写出像样的网络程序来!举个简单的例子,就是从网络上接受数据写入一个文件中。
server.cpp 

  
  
复制代码
#include <winsock2.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <string>
using namespace std;
#define port    2020 //本地端口
#define msgsize 1024
#pragma comment(lib,"ws2_32.lib")


int g_iTotalClient=0;//当前连接的client总数
SOCKET g_CliSocketArray[64];//存放client的socket的数组
DWORD WINAPI WorkerThread(LPVOID lpParameter);
int main()
{
    WSADATA wsaData;
    SOCKET sListen,sClient;//本机监听套接字,客户端套接字
    SOCKADDR_IN client;//客户端地址
    int iaddrSize=sizeof(SOCKADDR_IN);
    DWORD dwThreadId;
    WSAStartup(0x0202,&wsaData);
    string local_ip; 
    char szhostname[100];  
    gethostname(szhostname,sizeof(szhostname));
    struct hostent *hn=gethostbyname(szhostname);
    struct in_addr in;
    memcpy(&in,hn->h_addr_list[0],sizeof(struct in_addr));
    local_ip=inet_ntoa(in);
    cout<<"本机IP为:"<<local_ip<<endl;
  //  int nClient;
//    cout<<"请输入本地ip:";
//    cin>>local_ip;
/*  cout<<"请输入Client数目:";
    cin>>nClient;
    string *pclient_ip=new string[nClient];
    cout<<"请输入client ip:";
    for(int i=0;i<nClient;i++)
    {
         cin>>*pclient_ip++;
    }
*/
    // 创建一个套接字sListen与本机地址local_addr和端口port绑定
    sListen=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);//socket()创建一个套接口描述字
    LPHOSTENT lpHost;					//返回对应于给定服务器的包含服务器名字和IP的hostent结构指针
	struct   sockaddr_in   local_addr;	//将IP地址格式转化为sockaddr_in格式
	memset(&local_addr,0,sizeof(struct   sockaddr));
	local_addr.sin_family=AF_INET;
	local_addr.sin_addr.s_addr=inet_addr(local_ip.c_str());
	local_addr.sin_port=htons(port);
	bind(sListen,(struct sockaddr*)&local_addr,sizeof(SOCKADDR_IN));
    //listen
    /***1,内核用了两个队列,可以针对并发请求的时候,及时拒绝掉一部分处理不过来的请求,防止盲等待。
        2,listen的第二个参数。跟系统的链接数量没有任何关系。相当于设置一个瞬间能够处理的阈值。
        3,一般情况下都会去开启 syncookie。所有其实现在已经可以不太关系listen的第二个值了***/
    listen(sListen,3);//listen()监听申请的连接,仅适用于sock_stream类型的
    //Create WorkerThread
    CreateThread(NULL,0,WorkerThread,NULL,0,&dwThreadId);
    while(TRUE)
    {
        sClient=accept(sListen,(struct sockaddr *)&client,&iaddrSize);//阻塞等待客户端请求连接
        //将客户端套接字添加进数组g_CliSocketArray
        g_CliSocketArray[g_iTotalClient++]=sClient;
    }
    system("Pause");
    return 0;
}
/*
工作者线程里面是一个死循环,一次循环完成的动作是: 
     1.将当前所有的客户端套接字加入到读集fdread中; 
     2.调用select函数; 
     3.查看某个套接字是否仍然处于读集中,如果是,则接收数据。如果接收的数据长度为0,
     或者发生WSAECONNRESET错误,则表示客户端套接字主动关闭,这时需要将服务器中对应的套接字所绑定的资源释放掉,
     然后调整我们的套接字数组(将数组中最后一个套接字挪到当前的位置上)。
*/
DWORD WINAPI WorkerThread(LPVOID lpParam)
{
    int i;
    fd_set fdread;
    int ret;
    struct timeval tv={1,0};//这个参数使Select处于三种状态 使select在timeout时间内阻塞,超时返回
    char szMessage[msgsize];
    while(TRUE)
    {
        FD_ZERO(&fdread);
        for(i=0;i<g_iTotalClient;i++)
        {
            FD_SET(g_CliSocketArray[i],&fdread);//向集合添加套接字
        }
        ret=select(0,&fdread,NULL,NULL,&tv);
        if(ret==0)
            continue;
        for(i=0;i<g_iTotalClient;i++)
        {
            if(FD_ISSET(g_CliSocketArray[i],&fdread))
            {
                ret=recv(g_CliSocketArray[i],szMessage,msgsize,0);
                if(ret==0 || (ret==SOCKET_ERROR && WSAGetLastError()==WSAECONNRESET))
                {
                    //客户端套接字关闭 释放资源
                    printf("client socket %d closed.\n",g_CliSocketArray[i]);
                    closesocket(g_CliSocketArray[i]);
                    WSACleanup();
                    if(i<g_iTotalClient-1)
                        g_CliSocketArray[i--]=g_CliSocketArray[--g_iTotalClient];
                }
                else
                {
                    //从客户端接收数据并将数据返回
                    //szMessage[ret]='\0';
                    printf("接收到客户端 %d 数据:%s \n", i ,szMessage);
                    send(g_CliSocketArray[i],szMessage,strlen(szMessage),0);
                }
            }
        }
    }
    return 0;
}

服务器的几个主要动作如下: 
     1.创建监听套接字,绑定,监听; 
     2.创建工作者线程; 
     3.创建一个套接字数组,用来存放当前所有活动的客户端套接字,每accept一个连接就更新一次数组; 
     4.接受客户端的连接。 

     这里有一点需要注意的,就是我没有重新定义FD_SETSIZE宏,所以服务器最多支持的并发连接数为64。而且,这里决不能无条件的accept,服务器应该根据当前的连接数来决定是否接受来自某个客户端的连接。一种比较好的实现方案就是采用WSAAccept函数,而且让WSAAccept回调自己实现的Condition Function。
  如下所示: 

复制代码
int CALLBACK ConditionFunc(LPWSABUF lpCallerId,LPWSABUF lpCallerData, LPQOS lpSQOS,LPQOS lpGQOS,LPWSABUF lpCalleeId, LPWSABUF lpCalleeData,GROUP FAR * 
g,DWORD dwCallbackData) 
{ 
  if (当前连接数 < FD_SETSIZE) 
    return CF_ACCEPT; 
  else   
    return CF_REJECT; 
} 
复制代码

     工作者线程里面是一个死循环 
     除了需要有条件接受客户端的连接外,还需要在连接数为0的情形下做特殊处理,因为如果读集中没有任何套接字,select函数会立刻返回,这将导致工作者线程成为一个毫无停顿的死循环,CPU的占用率马上达到100%。

客户端的代码如下,可以同时打开多个客户端调试服务器是否非阻塞接收数据

#include <winsock2.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <string>
using namespace std;
#define port    2020 //服务器端口
#define msgsize 1024
#pragma comment(lib,"ws2_32.lib")

int main()
{
    string server_ip;   
    cout<<"请输入服务器ip:";
    cin>>server_ip;
    /*SOCKET INIT*/
    WSADATA wsaData; 
    char szMessage[msgsize];
    WSAStartup(0x0202,&wsaData);
    SOCKET localsocket=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);//socket()创建一个套接口描述字
    LPHOSTENT lpHost;					//返回对应于给定服务器的包含服务器名字和IP的hostent结构指针
	struct   sockaddr_in   server_addr;	//将IP地址格式转化为sockaddr_in格式
	memset(&server_addr,0,sizeof(struct   sockaddr));
	server_addr.sin_family=AF_INET;
	server_addr.sin_addr.s_addr=inet_addr(server_ip.c_str());
	server_addr.sin_port=htons(port);
    /*将localsocket连接至server_addr指定的服务器IP和Port
      localsocket为一个没有bind的套接口,connect()会自动设置一个关联值且设置为bind
    */
	if(connect(localsocket,(struct sockaddr*)&server_addr,sizeof(SOCKADDR_IN))!=0)
    {
        printf("connnect failed\n");
        exit(0);
    }
    while(TRUE)
    {
       string strdata="123das";
//       cout<<"请输入您想要发的数据:";
//       cin>>strdata;
       send(localsocket,strdata.c_str(),strlen(strdata.c_str()),0);
       int ret=recv(localsocket,szMessage,msgsize,0);
       if(ret==0 || (ret==SOCKET_ERROR && WSAGetLastError()==WSAECONNRESET))
        {
            //socket closed
            printf("server socket closed.\n");
            closesocket(localsocket);
            WSACleanup();
           
        }
        else
        {
            //received a message 
            //szMessage[ret]='\0';
            printf("接收到服务器返回的数据:%s \n",szMessage);
        }
    }
    system("Pause");
    return 0;
}

     关系到套接字列表的操作都需要使用循环,在轮询的时候,需要遍历一次,再新的一轮开始时,将列表加入队列又需要遍历一次.也就是说,Select在工作一次时,需要至少遍历2次列表,这是它效率较低的原因之一. 
     在大规模的网络连接方面,还是推荐使用IOCP或EPOLL模型.但是Select模型可以使用在诸如对战类游戏上,比如类似星际这种,因为它小巧易于实现,且对战类游戏的网络连接量并不大. 对于Select模型想要突破Windows 64个限制的话,可以采取分段轮询,一次轮询64个.例如套接字列表为128个,在第一次轮询时,将前64个放入队列中用Select进行状态查询, 待本次操作全部结束后.将后64个再加入轮询队列中进行轮询处理.这样处理需要在非阻塞式下工作.以此类推,Select也能支持无限多个。


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值