linux epoll 例子

epoll 一般使用

服务端代码

//server.cpp
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <errno.h>
#include <netinet/in.h>
#include <sys/epoll.h>
#include <fcntl.h>

int main(int argc, char *argv[])
{
        int serverfd = socket(AF_INET, SOCK_STREAM, 0);
        if (serverfd == -1)
        {
                printf("create socket error: %s\n", strerror(errno));
                return -1;
        }

        struct sockaddr_in     servaddr;
        memset(&servaddr, 0, sizeof(servaddr));
        servaddr.sin_family = AF_INET;
        servaddr.sin_addr.s_addr = htonl(INADDR_ANY);//IP地址设置成INADDR_ANY,让系统自动获取本机的IP地址。  
        servaddr.sin_port = htons(8080);
        if (bind(serverfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) == -1)
        {
                printf("bind socket error\n");
                return -1;
        }

        if (listen(serverfd, 10) == -1)
        {
                printf("listen socket error\n");
                return -1;
        }

        int ep_fd = epoll_create(10000);
        struct epoll_event listen_ep_event;
        struct epoll_event *ep_event;
        listen_ep_event.events = EPOLLIN;
        listen_ep_event.data.fd = serverfd;
        // [2] 将listen_fd和对应的结构体设置到树上
        epoll_ctl(ep_fd, EPOLL_CTL_ADD, serverfd, &listen_ep_event);

        while(1)
        {
                // [3] 为server阻塞(默认)监听事件,ep_event是数组,装满足条件后的所有事件结构体
                int n_ready = epoll_wait(ep_fd, ep_event, 10000, -1);
                for(int i=0; i<n_ready; i++)
                {
                        int temp_fd = ep_event[i].data.fd;

                        if(ep_event[i].events & EPOLLIN)
                        {
                                if(temp_fd == serverfd)
                                {  //说明有新连接到来
                                        socklen_t client_socket_len;
                                        struct epoll_event temp_ep_event;
                                        struct sockaddr_in client_socket_addr;
                                        int connect_fd = accept(serverfd, (struct sockaddr *)&client_socket_addr, &client_socket_len);
                                        int flag  = fcntl(connect_fd, F_GETFL);
                                        flag |= O_NONBLOCK;
                                        fcntl(connect_fd, F_SETFL, flag);
                                        // 给即将上树的结构体初始化
                                        temp_ep_event.events  = EPOLLIN;
                                        temp_ep_event.data.fd = connect_fd;
                                        // 上树
                                        epoll_ctl(ep_fd, EPOLL_CTL_ADD, connect_fd, &temp_ep_event);
                                }
                                else
                                {                      //cfd有数据到来
                                        char buf[1024] = {0};
                                        int n_data = read(temp_fd , buf, sizeof(buf));
                                        if(n_data == 0)
                                        {        //客户端关闭
                                                epoll_ctl(ep_fd, EPOLL_CTL_DEL, temp_fd, NULL); //下树
                                                close(temp_fd);
                                        }
                                        else if(n_data < 0)
                                        {
                                        }
                                        else
                                        {
                                                printf("buf:%s\n", buf);
                                        }

                                        do
                                        {
                                                //处理数据
                                        }
                                        while( (n_data = read(temp_fd , buf, sizeof(buf))) >0 ) ;
                                }
                        }
                        else if(ep_event[i].events & EPOLLOUT)
                        {
                                //处理写事件
                        }
                        else if(ep_event[i].events & EPOLLERR)
                        {
                                //处理异常事件
                        }
                }
        }
        close(serverfd);
        close(ep_fd);
}

客户端代码

/*******************************
client.cpp
********************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define PORT 8080

int main()
{
        /*1 创建socket*/
        int sockfd = socket(AF_INET, SOCK_STREAM, 0);
        if(sockfd == -1)
        {
                perror("socket failed");
                exit(-1);
        }
        /*2 准备通信地址*/
        struct sockaddr_in addr;
        addr.sin_family = AF_INET;
        /*设置为服务器进程的端口号*/
        addr.sin_port = htons(PORT);
        /*服务器所在主机IP地址*/
        inet_aton("192.168.30.30", &addr.sin_addr);

        /*3 连接服务器*/
        int res = connect(sockfd, (struct sockaddr *)&addr, sizeof(addr));
        if(res == -1)
        {
                perror("connect failed");
                exit(-1);
        }
        printf("connect success....\n");
        /*4 和服务器交换数据*/
        char buf[100] = {0};
        char *str = "client message ...";
        write(sockfd, str, strlen(str)) ;
        read(sockfd, buf, sizeof(buf));
        printf("server:%s\n", buf);

        /*关闭连接*/
        close(sockfd);
        return 0;
}

此外epoll还有反应堆的模型(libevent的核心),这里不描述