server.h
server.c
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);
}
</div>
bussiness.h
int write_sock(int sockfd, char *buff, size_t l