c语言tcp心跳程序,c语言多进程tcp服务器示例

#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);

}