epoll 模式

#include "socket_includes.h"
#include <unistd.h>
#include <fcntl.h>

#include <sys/epoll.h>
#include <errno.h>


int mz_ipv4_tcp_create_socket(void)
{
    int listenfd, sockfd, opt = 1;
    struct sockaddr_in server, client;
    socklen_t len;
    int timep;
    int ret;

    listenfd = socket(AF_INET, SOCK_STREAM, 0);
    if(listenfd < 0){
        perror("Create socket fail.");
        return -1;
    } 

    if((ret = setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt))) < 0){
        perror("Error, set socket reuse addr failed");  
        return -1;
    }

    bzero(&server, sizeof(server));
    server.sin_family = AF_INET;
    server.sin_port   = htons(SERV_PORT);
    server.sin_addr.s_addr  = htonl(INADDR_ANY);
    
    len = sizeof(struct sockaddr);
    if(bind(listenfd, (struct sockaddr *)&server, len)<0){
              perror("bind error.");
        return -1;
    }
    
      
    listen(listenfd, MAX_LISTEN_QUE);

    return listenfd;
}


int mz_process_data(int sockfd)
{
    int bytes;
    char buf[100];
    char *s = buf;
    char flag = 1;
    int len;

    while(flag){
        bytes = recv(sockfd, s, 5, 0);
        if(bytes < 0){
            if(errno == EAGAIN){
                printf("no data.\n");
                break;
            }
            perror("recv err:");
            return -1;
        }
        if(bytes == 0){
            return -2;
        }

        if(bytes == 5){
            flag = 1;
        }else {
            flag = 0;
        }
        
        s += bytes;
        len += bytes;
        printf("bytes:%d\n", bytes);
        //sleep(2);
    }
    printf("buf:%s\n", buf);
    send(sockfd, buf, len, 0);
    
    return 0;
}


int main(int argc, char *argv[])
{
    int listenfd, sockfd;
    int epollfd, fds;
    struct epoll_event ev, events[MAX_EVENTS];
    int i, rv;
    struct sockaddr_in client;
    int len;

    len = sizeof(struct sockaddr_in);
    epollfd = epoll_create(MAX_EVENTS);
    if(epollfd < 0){
        perror("epoll_create err:");
        return -1;
    }

    listenfd = mz_ipv4_tcp_create_socket();

    fcntl(listenfd, F_SETFL, O_NONBLOCK);

    ev.data.fd = listenfd;
    ev.events = EPOLLIN;
    rv = epoll_ctl(epollfd, EPOLL_CTL_ADD, listenfd, &ev);
    if(rv < 0){
        perror("epoll_ctl err:");
        return -1;
    }


    while(1){
        //time_out
        fds = epoll_wait(epollfd, events, MAX_EVENTS, -1);
        if(fds < 0){
            perror("epoll_wait err:");
            return -1;
        }

        for(i = 0; i < fds; i++){
            if(events[i].data.fd == listenfd){
                sockfd = accept(listenfd, (struct sockaddr *)&client, &len);
                if(sockfd < 0){
                    perror("accept err:");
                    continue;
                }
                ev.data.fd = sockfd;
                ev.events = EPOLLIN | EPOLLET;
                epoll_ctl(epollfd, EPOLL_CTL_ADD, sockfd, &ev);
                continue;
            }else{
                rv = mz_process_data(events[i].data.fd);
                if(rv == -2){
                    epoll_ctl(epollfd, EPOLL_CTL_DEL, events[i].data.fd, &ev);
                    close(events[i].data.fd);
                    continue;
                }
            }

        }
        


    }

}





#include <stdio.h>
#include <stdlib.h>
#include <string.h>


#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>


#include <unistd.h>
#include <fcntl.h>


#define SERV_PORT 8888
#define MAX_LISTEN_QUE 5

#define MAX_BUFFER_SIZE 1024

#define RT_ERR (-1)
#define RT_OK  0

#define MAX_EVENTS 500