#ifndef SERVER_H #define SERVER_H #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <arpa/inet.h> #include <assert.h> #include <sys/epoll.h> #include <signal.h> #include <fcntl.h> #include "bussiness.h" #define LISTENTQ 5 #define DEF_PORT 6000 #define MAX_EPOLL_SIZE 10 #define DEF_PROCESS_NUM 5 int create_tcpsvr(char *ip, int port); void fill_sockaddr(struct sockaddr_in *addr,char *ip, int port); void comm_to_client(int sockfd); void epoll_business(int epollfd, int listenfd, struct epoll_event ev); int init_epoll(int listenfd); void create_process(int sockfd, int i); #endif

#include "server.h" /*  * Create a TCP Server  * Return socketfd  */ int create_tcpsvr(char *ip,int port) {     int sockfd;     struct sockaddr_in addr;     if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == EOF){         perror("create_tcpsvr(),socket()");         exit(EXIT_FAILURE);     }     fill_sockaddr(&addr,ip,port);     if (bind(sockfd, (struct sockaddr *)&addr, sizeof(addr)) == EOF){         perror("create_tcpsvr(),bind()");         exit(EXIT_FAILURE);     }     if (listen(sockfd,LISTENTQ) == EOF) {         perror("create_tcpsvr(),bind()");         exit(EXIT_FAILURE);     }     return sockfd; } /**  * Set TCP server's address  */ void fill_sockaddr(struct sockaddr_in *addr, char *ip, int port) {     assert(addr);     addr -> sin_family = AF_INET;     addr -> sin_port = htons(port?port:DEF_PORT);     if (ip == NULL) {         addr -> sin_addr.s_addr = htonl(INADDR_ANY);     } else if((addr -> sin_addr.s_addr = inet_addr(ip)) == EOF) {        perror("fill_sockaddr(),inet_addr()");        exit(EXIT_FAILURE);     } } /*  * Communicate to client  */ void comm_to_clt(int listenfd) {        printf("TCP SERVER IS WAITING!\n");     struct epoll_event events[MAX_EPOLL_SIZE];     int fd_num;     int epollfd = init_epoll(listenfd);     while (1) {         fd_num = epoll_wait(epollfd,events,MAX_EPOLL_SIZE,-1);         if (fd_num == EOF) {             perror("comm_to_clt(),epoll_wait()");             continue;         }         while (--fd_num >= 0) {             epoll_business(epollfd, listenfd, events[fd_num]);         }     }    } /*  * Initlization epoll  */ int init_epoll(int listenfd) {     struct epoll_event ev;     int epollfd;     if((epollfd = epoll_create(MAX_EPOLL_SIZE)) == EOF) {         perror("init_epoll(),epoll_create()");         exit(EXIT_FAILURE);     }     ev.events = EPOLLIN;     ev.data.fd = listenfd;     if (epoll_ctl(epollfd, EPOLL_CTL_ADD,listenfd, &ev)) {         perror("init_epoll(),epoll_ctl()");         exit(EXIT_FAILURE);     }     return epollfd; } /*  * Respond for io change  */ void epoll_business(int epollfd, int listenfd, struct epoll_event ev) {     struct epoll_event ev_n;     if (ev.data.fd == listenfd) {         int clt_fd = accept(listenfd, 0, 0);         if (clt_fd == EOF) {             perror("epoll_business(), accept()");             return;         }         fcntl(clt_fd, F_SETFL, O_NONBLOCK);         ev_n.events = EPOLLIN|EPOLLET;         ev_n.data.fd = clt_fd;         if (epoll_ctl(epollfd,EPOLL_CTL_ADD, clt_fd, &ev_n) == EOF) {             perror("epoll_business(), epoll_ctl()");             return;         }     }else {         main_service(ev.data.fd);         if (epoll_ctl(epollfd,EPOLL_CTL_DEL, ev.data.fd, &ev) == EOF) {             perror("epoll_business(), epoll_ctl()");         }         close(ev.data.fd);     } } /*  * Create some process  */ void create_process(int sockfd, int i) {     /*      * Avoid zombie process      */     signal(SIGCHLD,SIG_IGN);     while (i--) {         if (fork() == 0) {             comm_to_clt(sockfd);         }     }     while(1){         sleep(100);     } } int main(int argc, char *argv[]) {     int sockfd = create_tcpsvr(0, 0);     create_process(sockfd,DEF_PROCESS_NUM);     exit(EXIT_SUCCESS); }

#ifndef BUSSINESS_H #define BUSSINESS_H #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <assert.h> #define BUFF_SIZE 4096 void main_service(int sockfd); int write_sock(int sockfd, char *buff, size_t length); int read_sock(int sockfd, char *buff, size_t llength); #endif

#include "bussiness.h" void main_service(int sockfd) {     char buff[BUFF_SIZE];     size_t buff_len;     if ((buff_len = read_sock(sockfd, buff, BUFF_SIZE)) == EOF) {         perror("main_service(),read_sock()");         return;     } else {   //业务逻辑从这里开始         printf("%s", buff);     } } /*  * Recive messages  * Return the length  */ int read_sock(int sockfd, char *buff, size_t length) {     assert(buff);     return read(sockfd, buff, length); } /*  * Send messages  * Return the length  */ int write_sock(int sockfd, char *buff,size_t length) {     assert(buff);     return write(sockfd, buff, length); }