IO复用之select

目录

一.select方法介绍

2.1 select 系统调用的原型

2.2 集合的数据结构

2.2.1 fd_set 结构如下:

2.2.2 关于集合fd_set的解析

2.3 select第一个参数

2.4 select方法之超时时间timeout

2.5 select方法的用法简述及返回值

2.6 如何检测集合中有哪些描述符有事件就绪

三.select应用

3.1 select小实例

 3.2 结合tcp编程


一.select方法介绍

select 系统调用的用途是:在一段指定时间内,监听用户感兴趣的文件描述符的可读、可写和异常等事件。

2.1 select 系统调用的原型

select 成功时返回就绪(可读、可写和异常)文件描述符的总数。如果在超时时间内没有任何文件描述符就绪,select将返回 0。select 失败是返回-1.如果在 select 等待期间,程序接收到信号,则select 立即返回-1,并设置errno 为EINTR。

#include <sys/select.h>
int select(int maxfd, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout);
  • maxfd 参数指定的被监听的文件描述符的总数。它通常被设置为 select 监听的所有文件描述符中的最大值+1。
  • readfds、 writefds 和 exceptfds 参数分别指向可读、可写和异常等事件对应的文件描述符集合。应用程序调用 select 函数时,通过这 3 个参数传入自己感兴趣的文件描述符。
  • select 返回时,内核将修改它们来通知应用程序哪些文件描述符已经就绪

2.2 集合的数据结构

select如何通知应用程序有哪些描述符数据就绪了,用户又是如何将这些描述符添加到select这个集合中?具体看下面关于集合fd_set的解析;

2.2.1 fd_set 结构如下:

#define __FD_SETSIZE 1024
typedef long int fd_mask;
#define __NFDBITS(8*(int)sizeof(__fd_mask))

typedef struct
{
    #ifdef __USE_XOPEN
    __fd_mask fds_bits[__FD_SETSIZE/__NFDBITS];//这是一个数组,依次是数组类型 数组名[数组个数]
    # define __FDS_BITS(set)((set)->fds_bits)
    #else
    __fd_mask __fds_bits[__FD_SETSIZE / __NFDBITS];
    # define __FDS_BITS(set)((set)->__fds_bits)
    #endif
}fd_set;

2.2.2 关于集合fd_set的解析

select如何通知应用程序有哪些描述符数据就绪了,用户又是如何将这些描述符添加到select这个集合中?我们首先来看关于集合fd_set的解析 :

也就是说这个结构体fd_set实际定义了1024个位。

通过下列宏可以访问 fd_set 结构中的位:

FD_ZERO(fd_set *fdset);// 清除 fdset 的所有位,就是将所有的位都置为0;

FD_SET(int fd,fd_set *fdset);// 设置 fdset 的位 fd,就是将某个描述符对应的位设置为1,就是将某个描述符添加到这个集合中;

FD_CLR(int fd,fd_set *fdset);// 清除 fdset 的位 fd,就是将某个位清零;

int FD_ISSET(int fd,fd_set *fdset);// 测试 fdset 的位 fd 是否被设置,其实就是测试某个描述符对应的位是不是1,如果被设置为1返回值为真,否则返回值为假; 

2.3 select第一个参数

select第一个参数通常被设置为描述符的最大值+1

也就是说书上maxfd是描述符的总数目,其实应该理解为描述符的最大值+1;也就是需要关注的位的总数目,比如上面的例子是8;

2.4 select方法之超时时间timeout

timeout 参数用来设置 select 函数的超时时间。它是一个 timeval 结构类型的指针,采用指针参数是因为内核将修改它以告诉应用程序 select 等待了多久。

timeval结构的定义如下:

struct timeval
{
    long tv_sec;//秒数
    long tv_usec;// 微秒数
};

也就是我们能够精确到微秒,实际上它还要看系统能不能达到;

1秒=1000毫秒,1毫秒=1000微秒

如果给 timeout 的两个成员都是 0,则 select 将立即返回。如果 timeout 传递NULL,则 select 将一直阻塞,直到某个文件描述符就绪 

2.5 select方法的用法简述及返回值

我们先将描述符添加到集合中,将集合传参给select,select返回以后我们就要关注它的返回值;

返回值如果等于0,就说明超时了:

如果返回值为n,n>0,那么就说明这个集合中有n(n>0)个描述符有事件就绪;

如果返回值为-1,说明出错了;

2.6 如何检测集合中有哪些描述符有事件就绪

三.select应用

3.1 select小实例

select检查键盘是否有数据

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>

#define STDIN 0
int main()
{
	int fd=STDIN;
	fd_set fdset;
	while(1)
	{
		FD_ZERO(&fdset);
		FD_SET(fd,&fdset);

		struct timeval tv={5,0};
		int n=select(fd+1,&fdset,NULL,NULL,&tv);
		if(n==-1)
		{
			printf("select error!\n");
			continue;
		}
		else if(n==0)
		{
			printf("time out!\n");
			continue;
		}
		else
		{
			if(FD_ISSET(fd,&fdset))
			{
				char buff[128]={0};
				read(fd,buff,127);
				printf("read:%s\n",buff);
			}
		}
	}
}

 3.2 实现select的tcp服务器端

 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
 #include <string.h>
 #include <assert.h>
 #include <sys/socket.h>
 #include <arpa/inet.h>
 #include <netinet/in.h>
 #include <fcntl.h>
 #include <sys/select.h>
 #include <sys/time.h>
  
 #define MAXFD 100

int create_socket()
 {
     int sockfd=socket(AF_INET,SOCK_STREAM,0);
     if(sockfd==-1)
     {
         return -1;
     }
 
     struct sockaddr_in saddr;
     memset(&saddr,0,sizeof(saddr));
     saddr.sin_family=AF_INET;
     saddr.sin_port=htons(6000);
     saddr.sin_addr.s_addr=inet_addr("127.0.0.1");
 
     int res=bind(sockfd,(struct sockaddr *)&saddr,sizeof(saddr));
     if(res==-1)
     {
         return -1;
     }
 
     res=listen(sockfd,5);
     if(res==-1)
     {
         return -1;
     }
 
     return sockfd;
 }

 void fds_init(int fds[])
 {
	 for(int i=0;i<MAXFD;i++)
	 {
		 fds[i]=-1;
	 }
 }

 void fds_add(int fds[],int fd)
 {
	 for(int i=0;i<MAXFD;i++)
	 {
		 if(fds[i]==-1)
		 {
			 fds[i]=fd;
			 break;
		 }
	 }
 }

 void fds_del(int fds[],int fd)
 {
	 for(int i=0;i<MAXFD;i++)
	 {
		 if(fds[i]==fd)
		 {
			 fds[i]=-1;
			 break;
		 }
	 }
 }

 int main()
 {
	 int sockfd=create_socket();
	 assert(sockfd!=-1);

	 int fds[MAXFD];
	 fds_init(fds);

	 fds_add(fds,sockfd);

	 fd_set fdset;
	 while(1)
	 {
		 FD_ZERO(&fdset);
		 int maxfd=-1;

		 for(int i=0;i<MAXFD;i++)
		 {
			 if(fds[i]!=-1)
			 {
				 FD_SET(fds[i],&fdset);
				 if(maxfd<fds[i])
				 {
					 maxfd=fds[i];
				 }
			 }
		 }

		 struct timeval tv={5,0};

		 int n=select(maxfd+1,&fdset,NULL,NULL,&tv);
		 if(n==-1)
		 {
			 printf("select error!\n");
			 continue;
		 }
		 else if(n==0)
		 {
			 printf("time out!\n");
			 continue;
		 }
		 else
		 {
			 for(int i=0;i<MAXFD;i++)
			 {
				 if(fds[i]==-1)
				 {
					continue;
				 }
				 if(FD_ISSET(fds[i],&fdset))
				 {
					 if(fds[i]==sockfd)
					 {
						 //accept;
						 struct sockaddr_in caddr;
						 int len=sizeof(caddr);
						 int c=accept(sockfd,(struct sockaddr*)&caddr,&len);
						 if(c==-1)
						 {
							 continue;
						 }
						 printf("accept c=%d\n",c);

						 fds_add(fds,c);
					 }
					 else
					 {
						 //recv
						 char buff[128]={0};
						 int res=recv(fds[i],buff,127,0);
						 if(res<=0)
						 {
							 close(fds[i]);
							 fds_del(fds,fds[i]);
							 printf("one client over!\n");
						 }
						 else
						 {
							 printf("buff(c=%d)=%s\n",fds[i],buff);
							 send(fds[i],"ok",2,0);
						 }
					 }
				 }
			 }
		 }
	 }
	 exit(0);
 }

封装函数版(跟上面的代码作用是一样的) 

//select_ser.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <assert.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/select.h>

#define MAX_FD  128
#define DATALEN 1024

//初始化服务器端的sockfd套接字
int InitSocket()
{
    int sockfd=socket(AF_INET,SOCK_STREAM,0);
    if(sockfd==-1)
    {
        return -1;
    }
    
    struct sockaddr_in saddr;
    memset(&saddr,0,sizeof(saddr));
    saddr.sin_family=AF_INET;
    saddr.sin_port=htons(6000);
    saddr.sin_addr.s_addr=inet_addr("127.0.0.1");
    
    int res=bind(sockfd,(struct sockaddr *)&saddr,sizeof(saddr));
    if(res==-1)
    {
        return -1;
    }
    
    res=listen(sockfd,5);
    if(res==-1)
    {
        return -1;
    }
    
    return sockfd;  
}


//初始化记录服务器套接字的数组
void InitFds(int fds[],int n)
{
    int i=0;
    for(;i<n;++i)
    {
        fds[i]=-1;
    }
}


//将套接字描述符添加到数组中
void AddFdToFds(int fds[],int fd,int n)
{
    int i=0;
    for(;i<n;++i)
    {
        if(fds[i]==-1)
        {
            fds[i]=fd;
            break;
        }
    }
}


//删除数组中的套接字描述符
void DelFdFromFds(int fds[],int fd,int n)
{
    int i=0;
    for(;i<n;++i)
    {
        if(fds[i]==fd)
        {
            fds[i]=-1;
            break;
        }
    }
}



//将数组中的套接字描述符设置到fd_set变量中,并返回当前最大的文件描述符值
int SetFdToFdset(fd_set *fdset,int fds[],int n)
{
    FD_ZERO(fdset);
    int i=0,maxfd=fds[0];
    for(;i<n;++i)
    {
        if(fds[i]!=-1)
        {
            FD_SET(fds[i],fdset);
            if(fds[i]>maxfd)
            {
                maxfd=fds[i];
            }
        }
    }
    
    return maxfd;
}

void GetClientLink(int sockfd,int fds[],int n)
{
    struct sockaddr_in caddr;
    memset(&caddr,0,sizeof(caddr));
    socklen_t len=sizeof(caddr);
    int c=accept(sockfd,(struct sockaddr *)&caddr,&len);
    if(c<0)
    {
        return ;
    }
    printf("A client connection was successful\n");
    AddFdToFds(fds,c,n);//新的链接套接字c添加进去集合中
}

//处理客户端数据
void DealClientData(int fds[],int n,int clifd)
{
    char data[DATALEN]={0};
    int num=recv(clifd,data,DATALEN-1,0);
    if(num<=0)
    {
        DelFdFromFds(fds,clifd,n);
        close(clifd);
        printf("A client disconnected\n");
    }
    else
    {
        printf("%d:%s\n",clifd,data);
        send(clifd,"Ok",2,0);
    }
}


//处理select返回的就绪事件
void DealReadyEvent(int fds[],int n,fd_set*fdset,int sockfd)
{
    int i=0;
    for(;i<n;++i)
    {
        if(fds[i]!=-1&& FD_ISSET(fds[i],fdset))
        {
            if(fds[i]==sockfd)
            {
                GetClientLink(sockfd,fds,n);
            }
            else
            {
               DealClientData(fds,n,fds[i]);
            }
        }
    }
}




int main()
{
    int sockfd=InitSocket();
    assert(sockfd!=-1);
    
     fd_set readfds;
     int fds[MAX_FD];
     InitFds(fds,MAX_FD);
     AddFdToFds(fds,sockfd,MAX_FD);
        
     while(1)
     {
         int maxfd=SetFdToFdset(&readfds,fds,MAX_FD);
         struct timeval timeout;
         timeout.tv_sec=5;//秒数
         timeout.tv_usec=0;//微秒
         int n=select(maxfd+1,&readfds,NULL,NULL,&timeout);
         if(n<0)
         {
             printf("select error\n");
             break;
         }
         else if(n==0)
         {
             printf("time  out\n");
             continue;
         }
         
         DealReadyEvent(fds,MAX_FD,&readfds,sockfd);
     }
    exit(0);
    
}

  • 19
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值