用epoll实现并发

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

#define INFO printf

#define LISTENPORT  7777    //监听端口
#define BACKLOG 5           //最大监听数
#define MAXLEN  2048        //recv缓冲区长度
#define EPOLL_SIZE    1024    


int listen_socket(int listen_port);
int processing_client_requests(int listenfd);

int main(int argc, char const *argv[])
{
    int listenfd = listen_socket(LISTENPORT);    
    if (listenfd == -1)
    {
        INFO("listen socket error: %s(errno: %d)\n", strerror(errno), errno);
        return -1;          
    }
    processing_client_requests(listenfd);

    close(listenfd);

    return 0;
}


int listen_socket(int listen_port)
{
    int connfd = 0; //客户端的fd
    struct sockaddr_in serv_addr;
    int listenfd = socket(AF_INET, SOCK_STREAM, 0);    //创建套接字

    if (listenfd == -1)
    {
        INFO("create socket error: %s(errno: %d)\n", strerror(errno), errno);
        return -1;
    }

    memset(&serv_addr, 0, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);  //监听服务器所有IP
    serv_addr.sin_port = htons(listen_port);

    if (bind(listenfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) == -1)  //绑定地址结构体和socket
    {
        INFO("bind socket error: %s(errno: %d)\n", strerror(errno), errno);
        close(listenfd);
        return -1;
    }

    if (listen(listenfd, BACKLOG) == -1)      //开启监听 ，第二个参数是最大监听数
    {
        INFO("listen socket error: %s(errno: %d)\n", strerror(errno), errno);
        close(listenfd);
        return -1;        
    }

    return listenfd;
}

int processing_client_requests(int listenfd)
{
    int connfd = 0; //客户端的fd
    int n = 0;  //recv的返回值
    char buff[MAXLEN];  //存储客户端发送过来的信息    
    
    int epfd = epoll_create(1);  //epoll_create的参数要求大于0
    struct epoll_event events[EPOLL_SIZE] = {0};
    struct epoll_event ev;

    ev.events = EPOLLIN;
    ev.data.fd = listenfd;

    epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev);

    while (1)
    {
        int nready = epoll_wait(epfd, events, EPOLL_SIZE, 5);
        if (nready == -1) continue;

        int i = 0;
        for (i = 0; i < nready; i++)
        {
            int clientfd = events[i].data.fd;
            if (clientfd == listenfd)
            {
                struct sockaddr_in client;
                socklen_t len = sizeof(client);
                if ((connfd = accept(listenfd, (struct sockaddr*)&client, &len)) == -1)
                {
                    INFO("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
                    return -1;
                }
                
                ev.events = EPOLLIN;
                ev.data.fd = connfd;
                epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev);
            }
            else if (events[i].events & EPOLLIN)
            {
                n = recv(clientfd, buff, MAXLEN, 0);
                if (n > 0)
                {
                    buff[n] = '\0';
                    INFO("recv msg from client: %s\n", buff);
                    send(clientfd, buff, n, 0);
                }
                else if (n == 0)
                {
                    INFO("client close------------\n");
                    ev.events = EPOLLIN;
                    ev.data.fd = clientfd;
                    epoll_ctl(epfd, EPOLL_CTL_DEL, clientfd, &ev);
                    close(clientfd);
                }
            }
        }
    }
    return 0;
}