epoll实现并发服务器

#include <arpa/inet.h>
#include <netinet/in.h>
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/epoll.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

#define ERR_MSG(msg)                           \
    do                                         \
    {                                          \
        fprintf(stderr, "line:%d ", __LINE__); \
        perror(msg);                           \
    } while (0)

#define PORT 8888         // 1024~49151
#define IP "192.168.1.27" // IP地址，本机IP ifconfig

struct cli_msg
{
    int newfd;
    struct sockaddr_in cin;
};

int epfd, ret;
char buf[128] = {0};

int main(int argc, const char *argv[])
{
    // 创建套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("socket create success  sfd = %d\n", sfd);
    // 设置允许端口快速被重用
    int resue = 1;
    if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &resue, sizeof(resue)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("setsockopt success __%d__\n", __LINE__);
    // 填充服务器的地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_port = htons(PORT);
    sin.sin_addr.s_addr = inet_addr(IP);
    // 绑定套接字到服务器地址
    if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success __%d__\n", __LINE__);
    // 将套接字设置为被动监听状态
    if (listen(sfd, 128) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success __%d__\n", __LINE__);
    struct epoll_event event;
    struct epoll_event events[10];
    // 创建一个新epoll
    epfd = epoll_create(1);
    if (epfd < 0)
    {
        printf("epoll_create err\n");
        exit(-1);
    }
    printf("epoll create success __%d__\n", __LINE__);
    // 添加要监听套接字的事件文件描述符到epoll
    event.events = EPOLLIN;
    event.data.fd = sfd;
    ret = epoll_ctl(epfd, EPOLL_CTL_ADD, sfd, &event);
    if (ret)
    {
        printf("epoll_ctl err\n");
        exit(-1);
    }
    printf("epoll ctl success __%d__\n", __LINE__);
    struct sockaddr_in saddr, caddr;
    socklen_t addrlen;
    while (1)
    {
        ret = epoll_wait(epfd, events, 10, -1);
        if (ret < 0)
        {
            perror("epoll_wait err\n");
            exit(-1);
        }
        for (int i = 0; i < ret; i++)
        {
            if (events[i].data.fd == sfd)
            {
                addrlen = sizeof(struct sockaddr_in);
                int cfd = accept(sfd, (struct sockaddr *)&caddr, &addrlen);
                if (-1 == cfd)
                {
                    perror("accept");
                    exit(-1);
                }
                event.events = EPOLLIN;
                event.data.fd = cfd;
                ret = epoll_ctl(epfd, EPOLL_CTL_ADD, cfd, &event);
                if (-1 == ret)
                {
                    perror("epoll_ctl");
                    exit(-1);
                }
                else
                {
                    cfd = events[i].data.fd;
                    memset(buf, 0, sizeof(buf));
                    ret = read(cfd, buf, sizeof(buf));
                    if (-1 == ret)
                    {
                        perror("read");
                        exit(-1);
                    }
                    else if (0 == ret)
                    {
                        printf("client quit\n");
                        ret = epoll_ctl(epfd, EPOLL_CTL_DEL, cfd, &event);
                        if (-1 == ret)
                        {
                            perror("epoll_ctl");
                            return -1;
                        }
                        close(cfd);
                        continue;
                    }
                    printf("cfd : %s\n", buf);
                }
            }
        }
        close(sfd);
        return 0;
    }
}