2019独角兽企业重金招聘Python工程师标准>>>

//重新装填epoll内fd的监听事件int i;for(i = 0; i < EVENTS_SIZE; ++i){if(connsets[i] != NULL){int fd = i; //fdtcp_connection_t *pt = connsets[i]; //tcp connuint32_t event = 0;if(buffer_is_readable(&pt->buffer_))event |= kWriteEvent;if(buffer_is_writeable(&pt->buffer_))event |= kReadEvent;//重置监听事件epoll_mod_fd(epollfd, fd, event);}}

//建立连接
int peerfd = accept4(listenfd, NULL, NULL, SOCK_NONBLOCK | SOCK_CLOEXEC);
if(peerfd == -1)ERR_EXIT("accept4");
//新建tcp连接
tcp_connection_t *pt = (tcp_connection_t*)malloc(sizeof(tcp_connection_t));
buffer_init(&pt->buffer_);
//将该tcp连接放入connsets
connsets[peerfd] = pt;
epoll_add_fd(epollfd, peerfd, 0);

//close
epoll_del_fd(epollfd, fd);
close(fd);
free(pt);
connsets[fd] = NULL;

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 */
};

#define _GNU_SOURCE             /* See feature_test_macros(7) */
#include <sys/socket.h>
#include "sysutil.h"
#include "buffer.h"
#include <assert.h>
#include <sys/epoll.h>#define EVENTS_SIZE 1024typedef struct{buffer_t buffer_;
} tcp_connection_t; //表示一条TCP连接

tcp_connection_t *connsets[EVENTS_SIZE]; //提供从fd到TCP连接的映射int main(int argc, char const *argv[])
{//获取监听fdint listenfd = tcp_server("localhost", 9981);//将监听fd设置为非阻塞
    activate_nonblock(listenfd);//初始化connsetsint ix;for(ix = 0; ix < EVENTS_SIZE; ++ix){connsets[ix] = NULL;}//初始化epollint epollfd = epoll_create1(0);epoll_add_fd(epollfd, listenfd, kReadEvent);struct epoll_event events[1024];while(1){//重新装填epoll内fd的监听事件int i;for(i = 0; i < EVENTS_SIZE; ++i){if(connsets[i] != NULL){int fd = i; //fdtcp_connection_t *pt = connsets[i]; //tcp connuint32_t event = 0;if(buffer_is_readable(&pt->buffer_))event |= kWriteEvent;if(buffer_is_writeable(&pt->buffer_))event |= kReadEvent;//重置监听事件epoll_mod_fd(epollfd, fd, event);}}//epoll监听fdint nready = epoll_wait(epollfd, events, 1024, 5000);if(nready == -1)ERR_EXIT("epoll wait");else if(nready == 0){printf("epoll timeout.\n");continue;}//处理fdfor(i = 0; i < nready; ++i){int fd = events[i].data.fd;uint32_t revents = events[i].events;if(fd == listenfd) //处理listen fd
            {if(revents & kReadREvent){//建立连接int peerfd = accept4(listenfd, NULL, NULL, SOCK_NONBLOCK | SOCK_CLOEXEC);if(peerfd == -1)ERR_EXIT("accept4");//新建tcp连接tcp_connection_t *pt = (tcp_connection_t*)malloc(sizeof(tcp_connection_t));buffer_init(&pt->buffer_);//将该tcp连接放入connsetsconnsets[peerfd] = pt;epoll_add_fd(epollfd, peerfd, 0);}}else //处理普通客户的fd
            {//取出指针tcp_connection_t *pt = connsets[fd];assert(pt != NULL);if(revents & kReadREvent){if(buffer_read(&pt->buffer_, fd) == 0){//close
                        epoll_del_fd(epollfd, fd);close(fd);free(pt);connsets[fd] = NULL;continue; //继续下一次循环
                    } }if(revents & kWriteREvent){buffer_write(&pt->buffer_, fd);}}}}close(listenfd);return 0;
}