Linux网络编程IO多路复用(一)

Linux网络编程IO多路复用(一)

一、介绍

epoll是一种I/O事件通知机制,是linux 内核实现IO多路复用的一个实现,linux内部是通过红黑树和链表队列实现的。 IO多路复用是指,在一个操作里同时监听多个输入输出源,在其中一个或多个输入输出源可用的时候返回,然后对其的进行读写操作。它所支持的文件描述符上限是系统可以最大打开的文件的数目;eg:1GB机器上,这个上限10万个左右。每个fd上面有callback(回调函数)函数,只有活跃的fd才有主动调用callback,不需要轮询,效率高。 epoll处理高并发,百万级,不关心底层怎样实现,只需要会调用就可以。

二、epoll接口调用的步骤

1.创建表(创建一颗红黑树)
2.将需要监听的文件描述符添加到树上
3.循环调用epoll_wait检测链表中是否有事件

函数接口

epoll_create( )
epoll_ctl( )
epoll_wait( )

创建控制句柄

int epoll_create(int size); 
功能:创建红黑树根节点
参数:size:不作为实际意义值 >0 即可
返回值:成功时返回epoll文件描述符,失败时返回-1

epoll控制函数

int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);
功能:控制epoll属性
    epfd:epoll_create函数的返回句柄。
    op:表示动作类型。有三个宏 来表示:
            EPOLL_CTL_ADD:注册新的fd到epfd中
            EPOLL_CTL_MOD:修改已注册fd的监听事件
            EPOLL_CTL_DEL:从epfd中删除一个fd
    fd:需要监听的fd。
            event:告诉内核需要监听什么事件
            EPOLLIN:表示对应文件描述符可读
            EPOLLOUT:可写
            EPOLLPRI:有紧急数据可读;
            EPOLLERR:错误;
            EPOLLHUP:被挂断;
            EPOLLET:触发方式,边缘触发;(默认使用边缘触发)
            ET模式:表示状态的变化;
返回值:成功时返回0,失败时返回-1

typedef union epoll_data {
    void *ptr;            //(无效)
    int fd;
    uint32_t u32;
    uint64_t u64;
} epoll_data_t;
struct epoll_event
{
    uint32_t events;       /* Epoll事件 */
        epoll_data_t data; /* 用户数据变量 */
};

等待事件到来

int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);
功能:等待事件的产生,类似于select的用法
     epfd:句柄;
     events:用来保存从内核得到事件的集合;
     maxevents:表示每次能处理事件最大个数;
     timeout:超时时间,毫秒,0立即返回,-1阻塞
成功时返回发生事件的文件描述个数,失败时返回-1

typedef union epoll_data {
    void *ptr;
    int fd;
    uint32_t u32;
    uint64_t u64;
} epoll_data_t;

struct epoll_event
{
    uint32_t events;   /* Epoll events */
    epoll_data_t data; /* User data variable */
};

二、I/O多路复用案例

举个栗子

Server.c

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <string.h>
#include <poll.h>
 #include <sys/epoll.h>

int main(int argc, char const *argv[])
{
    if (argc != 2)
    {
        printf("please input %s <port>\n", argv[0]);
        return -1;
    }
    //1.创建套接字 socket
    int sockfd, acceptfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err.");
        return -1;
    }
    printf("socket ok %d\n", sockfd);

    //填充ipv4的通信结构体
    struct sockaddr_in serveraddr, caddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[1]));
    //serveraddr.sin_addr.s_addr = inet_addr("0.0.0.0"); //0.0.0.0 自动获取主机ip
    // serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
    serveraddr.sin_addr.s_addr = INADDR_ANY;

    socklen_t len = sizeof(caddr);

    //&serveraddr -->struct sockaddr_in *

    //2.绑定套接字  bind  (绑定自己的ip和端口,便于别人找到自己)
    if (bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
    {
        perror("bind err.");
        return -1;
    }
    printf("bind ok.\n");

    //3.listen  监听 将主动套接子变为被动等待
    if (listen(sockfd, 5) < 0)
    {
        perror("listen err.");
        return -1;
    }
    printf("listen ok.\n");

    //功能:epoll 服务器响应多个客户端请求且进行通信
    struct epoll_event event;       //暂时保存添加到树上的事件
    struct epoll_event revents[20]; //保存从链表中获取的产生的事件
    //1.创建表 -创建一颗树
    int epfd = epoll_create(1);

    //2.添加关心的文件描述符到树上  epoll_ctl
    event.events = EPOLLIN | EPOLLET;
    event.data.fd = 0;
    epoll_ctl(epfd, EPOLL_CTL_ADD, 0, &event);

    event.events = EPOLLIN | EPOLLET;
    event.data.fd = sockfd;
    epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &event);

    int n = 2, ret;
    char buf[128];
    int recvbyte;
    //3.循环调用epoll_wait检测链表中是否有事件
    while (1)
    {
        //返回值是实际从链表中拿出来事件的个数
        ret = epoll_wait(epfd, revents, 20, -1);
        if (ret < 0)
        {
            perror("poll err.");
            return -1;
        }
        //处理事件
        for (int i = 0; i < ret; i++)
        {
            if (revents[i].data.fd == 0)
            {
                fgets(buf, sizeof(buf), stdin);
                printf("key:%s\n", buf);
            }
            else if (revents[i].data.fd == sockfd)
            {
                //4.阻塞等待客户端链接,链接成功返回一个通信文件描述符 accept
                acceptfd = accept(sockfd, (struct sockaddr *)&caddr, &len);

                if (acceptfd < 0)
                {
                    perror("accept err.");
                    return -1;
                }
                printf("accept ok.\n");

                //输出查看链接的客户端的ip和端口
                printf("ip:%s ,port:%d\n",
                       inet_ntoa(caddr.sin_addr), ntohs(caddr.sin_port));

                //需要将产生通信文件描述符添加到树
                event.events = EPOLLIN | EPOLLET;
                event.data.fd = acceptfd;
                epoll_ctl(epfd, EPOLL_CTL_ADD, acceptfd, &event);
            }
            else
            {
                //接收消息
                recvbyte = recv(revents[i].data.fd, buf, sizeof(buf), 0);
                if (recvbyte < 0)
                {
                    perror("recv err.");
                    // return -1;
                }
                else if (recvbyte == 0)
                {
                    printf("%d client exit.\n", revents[i].data.fd);
                    close(revents[i].data.fd);
                    epoll_ctl(epfd, EPOLL_CTL_DEL, revents[i].data.fd,NULL);
                    break;
                }
                else
                {
                    printf("%d buf:%s\n",revents[i].data.fd, buf);
                }
            }
        }
    }
    close(sockfd);
    return 0;
}

Client.c

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <string.h>

int main(int argc, char const *argv[])
{
    if (argc != 3)
    {
        printf("please input %s <ip> <port>\n", argv[0]);
        return -1;
    }
    //1.创建套接字 socket
    int sockfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err.");
        return -1;
    }
    printf("socket ok %d\n", sockfd);

    //填充ipv4的通信结构体  服务器端ip和端口
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);

    //&serveraddr -->struct sockaddr_in *

    //2.请求链接  connect
    if (connect(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
    {
        perror("connect err.");
        return -1;
    }
    printf("connect ok.\n");

    //3.循环收发消息
    char buf[128];
    int recvbyte;
    pid_t pid = fork();
    if (pid < 0)
    {
        perror("fork err.");
        return -1;
    }
    else if (pid == 0)
    {
        while (1)
        {

            fgets(buf, sizeof(buf), stdin);
            if (buf[strlen(buf) - 1] == '\n')
                buf[strlen(buf) - 1] = '\0';
            //发送消息
            send(sockfd, buf, sizeof(buf), 0);
        }
    }
    else
    {
        while (1)
        {
            //接收消息
            recvbyte = recv(sockfd, buf, sizeof(buf), 0);
            if (recvbyte < 0)
            {
                perror("recv err.");
                // return -1;
            }

            printf("buf:%s\n", buf);
        }
    }
    close(sockfd);
    return 0;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值