Linux C/C++ I/O多路复用(epoll)

---

一、参数介绍

#include<sys/epoll.h>

//创建一个新的epoll实例。在内核中创建了一块数据，这个数据中有两个比较重要的数据，一个是需要检测的文件描述符的信息（红黑树），还有一个是就绪列表，存放检测到数据发送改变的文件描述符信息（双向链表）。
int epoll_create(int size);
	-参数：
        size：现在不起作用了，必须大于0
    -返回值
        -1：失败
        >0:文件描述符，

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;		/*用户数据*/
}
常见的	epoll检测事件：
    -EPOLLIN
    -EPOLLOUT
    -EPOLLERR

//对epoll实例进行管理：添加文件描述符信息，删除信息，修改信息
int epoll_ctl(int epfd,int op,int fd,struct epoll_event* event);
	-参数：
        epfd：epoll实例对应的文件描述符
        op：要进行什么操作
        	EPOLL_CTR_ADD：添加
        	EPOLL_CTR_MOD：修改
        	EPOLL_CTR_DEL：删除
        fd：要检测的文件描述符
        event：检测文件描述符什么事情
        
//检测函数        
int epoll_wait(int epfd,struct epoll_event* events,int maxevents，int timeout);
	-参数：
        epfd：epoll实例对应的文件描述符
        events：传出参数，保存了发送了变化的文件描述符的信息
        maxevents：第二个参数结构体数组的大小
        timeout：阻塞时间
        	0：不阻塞
        	-1：阻塞，直到检测到fd数据发生变化
        	>0:阻塞的时(毫秒)
     -返回值：
           -成功，返回发送变化的文件描述符的个数 > 0
           -失败 -1

二、代码实现

1.服务器

代码如下（示例）：

#include <stdio.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/epoll.h>

int main()
{
    //创建socket
    int lfd = socket(PF_INET, SOCK_STREAM, 0);
    struct sockaddr_in saddr;
    saddr.sin_port = htons(9999);
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = INADDR_ANY;

    //绑定
    bind(lfd, (struct sockaddr *)&saddr, sizeof(saddr));

    //监听
    listen(lfd, 5);

    //调用epoll_create()创建一个epoll实例
    int epfd = epoll_create(128);

    //将监听的文件描述符相关的检测信息添加到epoll实例中
    struct epoll_event epev;
    epev.events = EPOLLIN;
    epev.data.fd = lfd;
    epoll_ctl(epfd, EPOLL_CTL_ADD, lfd, &epev);

    struct epoll_event epevs[1024];

    while (1)
    {
        int ret = epoll_wait(epfd, epevs, 1024, -1);
        if (ret == -1)
        {
            perror("epoll_wait");
            exit(-1);
        }

        printf("ret= %d\n", ret);

        for (int i = 0; i < ret; i++)
        {
            int curfd = epevs[i].data.fd;

            if (curfd == lfd)
            { //监听的文件描述符有数据到达，有客户端连接
                //表示有新的客户端连接进来了
                struct sockaddr_in caddr;
                int len = sizeof(caddr);
                int cfd = accept(lfd, (struct sockaddr *)&caddr, &len);
                epev.events = EPOLLIN | EPOLLOUT;
                epev.data.fd = cfd;
                epoll_ctl(epfd, EPOLL_CTL_ADD, cfd, &epev);
            }
            else
            {   
                if(epevs[i].events & EPOLLOUT)
                {
                    continue;
                }
                //有数据到达，需要读数据
                char buf[1024] = {0};
                int len = read(curfd, buf, sizeof(buf));
                if (len == -1)
                {
                    perror("read");
                }
                else if (len == 0)
                {
                    printf("client closed...\n");
                    close(curfd);
                    epoll_ctl(epfd, EPOLL_CTL_DEL, curfd, NULL);
                }
                else if (len > 0)
                {
                    printf("read buf=%s\n", buf);
                    write(curfd, buf, strlen(buf) + 1);
                }
            }
        }
    }

    close(lfd);
    close(epfd);
    return 0;
}

2.客户端

代码如下（示例）：

#include<stdio.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<string.h>
#include<stdlib.h>

int main()
{
    //1.创建套接字
    int fd = socket(AF_INET, SOCK_STREAM, 0);

    if(fd==-1)
    {
        perror("socket");
        exit(-1);
    }

    //2.连接服务器
    struct sockaddr_in saddr;
    inet_pton(AF_INET, "182.61.11.197", &saddr.sin_addr.s_addr);
    saddr.sin_family = AF_INET;
    saddr.sin_port = htons(9999);

    //连接服务器
    int ret = connect(fd, (struct sockaddr *)&saddr, sizeof(saddr));

    if(ret==-1)
    {
        perror("connet");
        exit(-1);
    }

    //3.通信
    char recvbuf[1024] = {0};
    int i = 0;
    int num = 0;
    while (1)
    {
        char sendbuf[1024] = {0};

        sprintf(sendbuf, "send data %d", num++);

        write(fd, sendbuf, strlen(sendbuf) + 1);

        //接收
        int len = read(fd, sendbuf, sizeof(sendbuf));
        if(len==-1)
        {
            perror("read");
            return -1;
        }else if(len>0)
        {
            printf("read buf = %s\n", sendbuf);
        }else{
            printf("服务器已经断开连接...\n");
            break;
        }
        usleep(1000);
    }

    //关闭连接
    close(fd);

    return 0;
}

---