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;
                        } 
                    }
                }
            }
        }
    }
}
相关推荐
薛定谔的悦3 小时前
储能 EMS 的功率策略:从一块电表到一次逆流的 200 毫秒
大数据·linux·能源·储能·bms
新时代牛马6 小时前
Linux 内核入门地图:架构、源码目录与五大子系统
linux·运维·架构
Julien20047 小时前
控制 SELinux 文件上下文
linux·运维·服务器
sulikey7 小时前
Linux Core Dump 完全指南:从原理到实战的崩溃调试手册。什么是core dump?程序报错core dumped怎么办?
linux·服务器·操作系统·调试·coredump
Lifangyun_WD7 小时前
第一次租 GPU 服务器怎么开始?镜像、SSH、数据存储和停机前要做什么
服务器·云计算·gpu算力·gpu算力租赁
DevLoom_8 小时前
SD卡插电脑提示:驱动器中的磁盘未被格式化?恢复数据正确处理步骤
运维·服务器·电脑
STQY燊桐启元(深圳)电子科技8 小时前
医疗精密电源散热方案:相变陶瓷片保障医疗设备长效稳定低干扰
大数据·网络·人工智能
MayZork10 小时前
Windows 与 Linux 下 vcpkg 的安装部署指南
linux·运维·windows
yyuuuzz10 小时前
记一次 VPS 性能排查:共享 CPU 突发额度导致的限流
运维·服务器·人工智能
冰雪青松10 小时前
Ubuntu 运维命令大全:从入门到精通的实战参考手册
运维·服务器·ubuntu