多路IO转接之select(Linux网络编程)


多路IO转接服务器也叫做多任务IO服务器。该类服务器实现的主旨思想是,不再由应用程序自己监视客户端连接,取而代之由内核替应用程序监视文件。

select 函数原型分析

1.select能监听的文件描述符个数受限于FD_SETSIZE,一般为1024,单纯改变进程打开的文件描述符个数并不能改变select监听文件个数

2.解决1024以下客户端时使用select是很合适的,但如果链接客户端过多,select采用的是轮询模型,会大大降低服务器响应效率,不应在select上投入更多精力

#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
int select(int nfds, fd_set *readfds, fd_set *writefds,
			fd_set *exceptfds, struct timeval *timeout);

	nfds: 		指定被监听的文件描述符的总数。它通常被设置为select监听的所有文件描述符中的最大值加1,
	因为文件描述符是从0开始计数的。
	readfds:	监控有读数据到达文件描述符集合,传入传出参数
	writefds:	监控写数据到达文件描述符集合,传入传出参数
	exceptfds:	监控异常发生达文件描述符集合,如带外数据到达异常,传入传出参数
	timeout:	定时阻塞监控时间,3种情况
				1. NULL,永远等下去,阻塞监听
				2. >0,等待固定时间
				3. 0,非阻塞监听,轮询
	struct timeval {
		long tv_sec; /* seconds */
		long tv_usec; /* microseconds */
	};
	
	返回值:
			>0: 所有监听集合(3)中,满足对应事件的总数
			0: 没有满足监听条件的文件描述符
			-1:errno

select相关函数参数分析

void FD_CLR(int fd, fd_set *set)		把某一个fd清除出去
int FD_ISSET(int fd, fd_set *set)		判定某个fd是否在位图中
void FD_SET(int fd, fd_set *set)		把某一个fd添加到位图
void FD_ZERO(fd_set *set)				位图所有二进制位置零

select多路IO转接:

	原理:  借助内核, select 来监听, 客户端连接、数据通信事件。

	void FD_ZERO(fd_set *set);	--- 清空一个文件描述符集合。

		fd_set rset;

		FD_ZERO(&rset);

	void FD_SET(int fd, fd_set *set);	--- 将待监听的文件描述符,添加到监听集合中

		FD_SET(3, &rset);	FD_SET(5, &rset);	FD_SET(6, &rset);


	void FD_CLR(int fd, fd_set *set);	--- 将一个文件描述符从监听集合中 移除。

		FD_CLR(4&rset);

	int  FD_ISSET(int fd, fd_set *set);	--- 判断一个文件描述符是否在监听集合中。

		返回值: 在:1;不在:0;

		FD_ISSET(4&rset);

select实现多路IO转接设计思路

int maxfd = 0;

	lfd = socket() ;			创建套接字

	maxfd = lfd;

	bind();					绑定地址结构

	listen();				设置监听上限

	fd_set rset, allset;			创建r监听集合

	FD_ZERO(&allset);				将r监听集合清空

	FD_SET(lfd, &allset);			将 lfd 添加至读集合中。

	while1{

		rset = allset;			保存监听集合
	
		ret  = select(lfd+1&rset, NULLNULLNULL);		监听文件描述符集合对应事件。

		if(ret > 0{							有监听的描述符满足对应事件
		
			if (FD_ISSET(lfd, &rset)) {				// 1 在。 0不在。

				cfd = accept();				建立连接,返回用于通信的文件描述符

				maxfd = cfd;

				FD_SET(cfd, &allset);				添加到监听通信描述符集合中。
			}

			for (i = lfd+1; i <= 最大文件描述符; i++{

				FD_ISSET(i, &rset)				有read、write事件

				read()

				小 --write();
			}	
		}
	}

select实现多路IO转接

select_server.c

#include <stdio.h>  
#include <stdlib.h>  
#include <unistd.h>  
#include <string.h>  
#include <arpa/inet.h>  
#include <ctype.h>  
  
#include "wrap.h"  
  
#define SERV_PORT 6666  
  
int main(int argc, char *argv[])  
{  
    int i, j, n, nready;  
  
    int maxfd = 0;  
  
    int listenfd, connfd;  
  
    char buf[BUFSIZ];         /* #define INET_ADDRSTRLEN 16 */  
  
    struct sockaddr_in clie_addr, serv_addr;  
    socklen_t clie_addr_len;  
  
    listenfd = Socket(AF_INET, SOCK_STREAM, 0);    
    int opt = 1;  
    setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));  
    bzero(&serv_addr, sizeof(serv_addr));  
    serv_addr.sin_family= AF_INET;  
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);  
    serv_addr.sin_port= htons(SERV_PORT);  
    Bind(listenfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));  
    Listen(listenfd, 128);  
      
  
    fd_set rset, allset;                            /* rset 读事件文件描述符集合 allset用来暂存 */  
  
    maxfd = listenfd;  
  
    FD_ZERO(&allset);  
    FD_SET(listenfd, &allset);                                  /* 构造select监控文件描述符集 */  
  
    while (1) {     
        rset = allset;                                          /* 每次循环时都从新设置select监控信集 */  
        nready = select(maxfd+1, &rset, NULL, NULL, NULL);  
        if (nready < 0)  
            perr_exit("select error");  
  
        if (FD_ISSET(listenfd, &rset)) {                        /* 说明有新的客户端链接请求 */  
  
            clie_addr_len = sizeof(clie_addr);  
            connfd = Accept(listenfd, (struct sockaddr *)&clie_addr, &clie_addr_len);       /* Accept不会阻塞 */  
  
            FD_SET(connfd, &allset);                            /* 向监控文件描述符集合allset添加新的文描述符connfd */ 
 
           if (maxfd < connfd)  
                maxfd = connfd;  
  
            if (0 == --nready)                                  /* 只有listenfd有事件, 后续的 for 不需执行 */  
                continue;  
        }   
  
        for (i = listenfd+1; i <= maxfd; i++) {                 /* 检测哪个clients 有数据就绪 */  
  
            if (FD_ISSET(i, &rset)) {  
  
                if ((n = Read(i, buf, sizeof(buf))) == 0) {    /* 当client关闭链接时,服务器端也关闭对应接 */  
                    Close(i);  
                    FD_CLR(i, &allset);                        /* 解除select对此文件描述符的监控 */  
  
                } else if (n > 0) {  
  
                    for (j = 0; j < n; j++)  
                        buf[j] = toupper(buf[j]);  
                    Write(i, buf, n);  
                }  
            }  
        }  
    }  
  
    Close(listenfd);  
  
    return 0;  
}  

select优缺点

缺点: 监听上限受文件描述符限制。 最大 1024.

检测满足条件的fd, 自己添加业务逻辑提高效率。 提高了编码难度。

优点: 跨平台。win、linux、macOS、Unix、类Unix、mips

添加自定义数组提高效率

这里就是改进之前代码的问题,之前的代码,如果最大fd是1023,每次确定有事件发生的fd时,就要扫描3-1023的所有文件描述符,这看起来很蠢。于是定义一个数组,把要监听的文件描述符存下来,每次扫描这个数组就行了。看起来科学得多。

代码

#include <stdio.h>  
#include <stdlib.h>  
#include <unistd.h>  
#include <string.h>  
#include <arpa/inet.h>  
#include <ctype.h>  
  
#include "wrap.h"  
  
#define SERV_PORT 6666  
  
int main(int argc, char *argv[])  
{  
    int i, j, n, maxi;  
  
    int nready, client[FD_SETSIZE];                 /* 自定义数组client, 防止遍历1024个文件描符  FD_SETSIZE默认为1024 */  
    int maxfd, listenfd, connfd, sockfd;  
    char buf[BUFSIZ], str[INET_ADDRSTRLEN];         /* #define INET_ADDRSTRLEN 16 */  
  
    struct sockaddr_in clie_addr, serv_addr;  
    socklen_t clie_addr_len;  
    fd_set rset, allset;                            /* rset 读事件文件描述符集合 allset用来暂存 */  
  
    listenfd = Socket(AF_INET, SOCK_STREAM, 0);  
  
    int opt = 1;  
    setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));  
  
    bzero(&serv_addr, sizeof(serv_addr));  
    serv_addr.sin_family= AF_INET;  
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);  
    serv_addr.sin_port= htons(SERV_PORT);  
  
    Bind(listenfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));  
    Listen(listenfd, 128);  
  
    maxfd = listenfd;                                           /* 起初 listenfd 即为最大文件描述符 */  
  
    maxi = -1;                                                  /* 将来用作client[]的下标, 初始值指向0个元素之前下标位置 */  
    for (i = 0; i < FD_SETSIZE; i++)  
        client[i] = -1;                                         /* 用-1初始化client[] */  
  
    FD_ZERO(&allset);  
    FD_SET(listenfd, &allset);                                  /* 构造select监控文件描述符集 */  
  
    while (1) {     
        rset = allset;                                          /* 每次循环时都重新设置select监控信集 */  
  
        nready = select(maxfd+1, &rset, NULL, NULL, NULL);  //2  1--lfd  1--connfd  
        if (nready < 0)  
            perr_exit("select error");  
  
        if (FD_ISSET(listenfd, &rset)) {                        /* 说明有新的客户端链接请求 */  
  
            clie_addr_len = sizeof(clie_addr);  
            connfd = Accept(listenfd, (struct sockaddr *)&clie_addr, &clie_addr_len);       /* Accept 不会阻塞 */  
            printf("received from %s at PORT %d\n",  
                    inet_ntop(AF_INET, &clie_addr.sin_addr, str, sizeof(str)),  
                    ntohs(clie_addr.sin_port));  
  
            for (i = 0; i < FD_SETSIZE; i++)  
                if (client[i] < 0) {                            /* 找client[]中没有使用的位置 */  
                    client[i] = connfd;                         /* 保存accept返回的文件描述符到client[]里 */  
                    break;  
                }  
  
            if (i == FD_SETSIZE) {                              /* 达到select能监控的文件个数上限 1024 */  
                fputs("too many clients\n", stderr);  
                exit(1);  
            }  
  
            FD_SET(connfd, &allset);                            /* 向监控文件描述符集合allset添加新的文件描述符connfd */  
  
            if (connfd > maxfd)  
                maxfd = connfd;                                 /* select第一个参数需要 */  
  
            if (i > maxi)  
                maxi = i;                                       /* 保证maxi存的总是client[]最后一个元素下标 */  
  
            if (--nready == 0)  
                continue;  
        }   
  
        for (i = 0; i <= maxi; i++) {                               /* 检测哪个clients 有数据就绪 */  
  
            if ((sockfd = client[i]) < 0)  
                continue;  
            if (FD_ISSET(sockfd, &rset)) {  
  
                if ((n = Read(sockfd, buf, sizeof(buf))) == 0) {    /* 当client关闭链接时,服务器端也关闭对应链接 */  
                    Close(sockfd);  
                    FD_CLR(sockfd, &allset);                        /* 解除select对此文件描述符的监控 */  
                    client[i] = -1;  
                } else if (n > 0) {  
                    for (j = 0; j < n; j++)  
                        buf[j] = toupper(buf[j]);  
                    Write(sockfd, buf, n);  
                    Write(STDOUT_FILENO, buf, n);  
                }  
                if (--nready == 0)  
                    break;                                          /* 跳出for, 但还在while中 */  
            }  
        }  
    }  
    Close(listenfd);  
    return 0;  
}  
  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值