epoll示例

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

#define MAX_CLIENT 3000 

int accept_count = 0;
int read_count= 0;
struct epoll_event events[MAX_CLIENT];

void sig_ctrl(int sig)
...{
        printf("accept:%d, read:%d ", accept_count, read_count);
        exit(0);
}

int main(void)
...{
        signal(SIGINT, sig_ctrl);

        int listener = socket(AF_INET, SOCK_STREAM, 0);
        int flag = 1;
        int ret = setsockopt(listener, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag));

        struct sockaddr_in addr;
        memset(&addr, 0, sizeof(struct sockaddr_in));
        addr.sin_family = AF_INET;
        addr.sin_port = htons(10000);
        addr.sin_addr.s_addr = htonl(INADDR_ANY);
        ret = bind(listener, (struct sockaddr*)&addr, sizeof(addr));
        if( ret == -1 ) ...{
                printf("bind port %d failed ", 10000);
                return -1;
        }

        if( listen(listener, 5) == -1 ) ...{
                printf("listen failed .. ");
                return -1;
        }

        int epfd = epoll_create(MAX_CLIENT);
        if( epfd == -1 ) ...{
                printf("create epoll failed ");
                return -1;
        }

        struct epoll_event event;
        event.data.fd = listener;
        event.events = EPOLLIN;
        ret = epoll_ctl(epfd, EPOLL_CTL_ADD, listener, &event);
        if( ret == - 1) ...{
                printf("epoll_ctl add failed ");
                close(epfd);
                return -1;
        }

        while( 1 ) ...{
                int n = epoll_wait(epfd, &events, MAX_CLIENT, -1);
                if( n == -1 ) ...{
                        continue;
                }

                int i = 0;
                for( i; i < n; i++ ) ...{
                        if( events[i].data.fd == listener ) ...{
                                int client = accept(listener, NULL, NULL);
                                if( ret == - 1) ...{
                                        printf("epoll_ctl add failed ");
                                        continue;
                                }else ...{
                                        accept_count++;
                                        event.data.fd = client;
                                        event.events = EPOLLIN | EPOLLOUT;
                                        ret = epoll_ctl(epfd, EPOLL_CTL_ADD, client, &event);
                                        printf("epoll_ctl add ok ");
                                }

                                continue;
                        }

                        dowith_socket(epfd, &events[i]);
                }
        }
}


int dowith_socket(int epfd, struct epoll_event* event)
...{
        int ret = 0;
        char buf[1024] = ...{0};
        if( event->events & EPOLLIN ) ...{
                ret = read(event->data.fd, buf, 1024);
                if( ret == 0 ) ...{
                        close(event->data.fd); /**//* epoll auto clean this fd */
                        return 0;
                }
                if( ret < 0 ) ...{
                        return -1;
                }
                read_count++;
                buf[ret] = 0;
                printf("Readed data:%s ", buf);

                write(event->data.fd, buf, ret);

        } else if( event->events & EPOLLOUT ) ...{
                ;
        }

        return 0;
}