linux中实现多路复用服务器

linux中实现多路复用服务器

使用一个进程(且只有主线程)同时监控若干个文件描述符的读写情况,这种读写模式称为多路复用

多用于TCP的服务端,用于监控客户端的连接和数据的发送

优点:

不需要频繁地创建、销毁进程,从而节约了内存资源、时间资源,也避免了进程之间的竞争、等待

缺点:

要求单个客户端的任务不能太过于耗时,否则其他客户端就会感知到卡顿

c 复制代码
select:
    fd_set 是文件描述符的集合,使用以下函数操作:
    void FD_CLR(int fd, fd_set *set);
    功能:从集合set中删除fd文件描述符

    int  FD_ISSET(int fd, fd_set *set);
    功能:判断集合set中是否存在fd文件描述符,可以用于监视多个文件描述符以检测哪些文件描述符准备好进行读、写或异常处理。
    
    void FD_SET(int fd, fd_set *set);
    功能:向集合set中添加fd文件描述符

    void FD_ZERO(fd_set *set);
    功能:清空集合set

    int select(int nfds, fd_set *readfds, fd_set *writefds,fd_set *exceptfds, struct timeval *timeout);
    功能:同时监控多个文件描述的读、写、异常操作
    nfds:被监控的文件描述符中的最大值+1
    readfds:监控读操作的文件描述符集合
    writefds:监控写操作的文件描述符集合
    exceptfds:监控异常操作的文件描述符集合
    timeout:设置超时时间
        NULL    一直阻塞,直到某个文件描述符发生了变化
        0秒0微秒 非阻塞
        大于0秒 等待超时时间,超时返回0
        struct timeval {
               long    tv_sec;  //  秒
               long    tv_usec; //  微秒
           };
    返回值:监控到发生相关操作的文件描述符的个数,超时返回0,错误返回-1

    注意:
        readfds、writefds、exceptfds 这三个集合参数既是输入也是输出,调用select时这三个集合需要存储被监控的文件描述符,当由于有文件描述符发生了相应的操作而导致函数返回时,这三个集合中存储了这些文件描述符并返回给调用者

    select设计不合理的地方:
        1、每次调用select都需要向它重新传递被监控的文件描述符集合
        2、调用结束后如果想知道具体是哪个文件描述符发生了相关操作,必须对所有被监控的文件描述符进行一遍测试
    select的优点:
        它是最早的多路复用函数,几乎所有的操作系统都支持,兼容性很高

pselect:
    int pselect(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, const struct timespec *timeout,const sigset_t *sigmask);
    功能:大致与select
    区别:
        1、超时时间的结构类型不同
        struct timespec {
            long    tv_sec;     //秒
            long    tv_nsec;    //纳秒
        };
        2、pselect监控时可以通过sigmask参数设置要屏蔽的信号,可以保障pselect监控时不受这些信号干扰
    共同点:本质上与select差别不大、select的缺点pelect也是一样的,只是个别功能有所增强而已

服务器具体代码实现

c 复制代码
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#include <errno.h>
#include <signal.h>
#include <ctype.h>
#include <sys/stat.h>
#include <fcntl.h>
int main(int argc,const char* argv[])
{
	//创建套接字
    int svr_fd = socket(AF_INET, SOCK_STREAM, 0);
	if(svr_fd < 0)
	{
        perror("socket");
        return -1;
    }
    //准备通信地址
    struct sockaddr_in addr={};
    addr.sin_family = AF_INET;
    addr.sin_port = htons(8888);//设置端口号
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");//设置ip地址
    socklen_t addrlen = sizeof(addr);
    //绑定地址
    if(bind(svr_fd, (struct sockaddr*)&addr, addrlen) < 0)
    {
        perror("bind");
        return -1;
    }
    //监听
    if(listen(svr_fd, 5) < 0)
    {
        perror("listen");
        return -1;
    }
    //定义一个文件描述符集合,并清空
    fd_set rfds;
    FD_ZERO(&rfds);
    //将套接字加入到集合中
    FD_SET(svr_fd, &rfds);

    //定义超时时间
    struct timeval timeout={};
    timeout.tv_sec = 5;
    timeout.tv_usec = 0;

    //记录最大的文件描述符
    int max_fd = svr_fd;//当前socket描述符就这一个,所以也是最大值
    char buf[1024];
    size_t buf_size = sizeof(buf);
    while(1)
    {
        //若有多个新的连接,集合会发生变化,而把之前的没发生变化的文件描述符给覆盖了,所以需要复制一份
        fd_set ready_copy=rfds;
        int ret = select(max_fd+1, &ready_copy, NULL, NULL, &timeout);//select函数返回值表示有多少文件描述符发生变化
        if(ret > 0)
        {
          //一开始,先测网络等待的socket描述符
            if(FD_ISSET(svr_fd, &ready_copy))//如果有新的连接
            {
                //调用accept()连接客户端
                int clt_fd = accept(svr_fd,(struct sockaddr*)&addr, &addrlen);
                if(clt_fd < 0)
                {
                    perror("accept");
                    continue;
                }
                else
                {
                    //把客户端的描述符加入到集合中
                    FD_SET(clt_fd, &rfds);
                    if(clt_fd > max_fd)
                    {
                        max_fd = clt_fd;
                    }
                }
            }
            else
            {
                //测试其他socket描述符是否发生变化
                for(int i=3; i<=max_fd; i++)
                {
                    if(FD_ISSET(i, &ready_copy) && i!= svr_fd)
                    {
                        int ret = recv(i, buf, buf_size, 0);
                        if(ret < 0)
                        {
                            FD_CLR(i, &rfds);
                            printf("客户端%d断开连接\n", i);
                            continue;
                        }
                        printf("recv:%s bits:%d\n", buf, ret);
                        strcat(buf,":return");
                        ret=send(i,buf,strlen(buf)+1,0);
                        if(ret <= 0 || 0==strcmp(buf,"quit"))
                        {
                            FD_CLR(i, &rfds);
                            printf("客户端%d断开连接\n", i);
                            continue;
                        } 
                    }
                }
            }
        }
    }
}
相关推荐
DO_Community3 分钟前
DigitalOcean Droplet 云服务器:新增自动扩展池功能
服务器
安科瑞刘鸿鹏7 分钟前
老旧小区用电安全保护装置#限流式防火保护器参数介绍#
运维·服务器·物联网·能源
Rain_Rong26 分钟前
linux检测硬盘
linux·运维·服务器
过过过呀Glik30 分钟前
在 Ubuntu 上安装 Muduo 网络库的详细指南
linux·c++·ubuntu·boost·muduo
我曾经是个程序员1 小时前
鸿蒙学习记录之http网络请求
服务器·学习·http
0zxm1 小时前
06 - Django 视图view
网络·后端·python·django
真真-真真1 小时前
WebXR
linux·运维·服务器
轩辰~2 小时前
网络协议入门
linux·服务器·开发语言·网络·arm开发·c++·网络协议
燕雀安知鸿鹄之志哉.2 小时前
攻防世界 web ics-06
网络·经验分享·安全·web安全·网络安全
wanhengidc2 小时前
短视频运营行业该如何选择服务器?
运维·服务器