简单进程池实现多TCP客户服务

原创 2012年03月28日 09:31:40

根据预定义进程数创建进程池。

父子进程通信使用的IPC方式为:UNIX域套接字

父进程listen,aeecpt,并将连接套接字发送到子进程,交由子进程处理该连接。

子进程处理完毕,与父进程通信,实现资源回收,并在下一连接到来交由该完毕的子进程。

#include "unp.h"
#include <assert.h>
#define IDLE		0
#define BUSY		1
#define END		2


typedef struct pro {
	pid_t	id;		/*进程的进程ID*/
	int 		fd;		/*与父进程通信的描述符*/
	int		flags;	/*进程当前状态*/
	int		cnt;		/*进程服务的次数*/
}process;
process	*mypid;		/*进程池*/


ssize_t write__the_fd(int fd, void *ptr, size_t nbytes, int sendfd);
pid_t make_child( int i, int listenfd );
ssize_t read_the_fd(int fd, void *ptr, size_t nbytes, int *recvfd);
void child_func( void);
void clr(int signo);
int	main( int argc, char **argv ) {
	int 		nchild, fdmax, listenfd, connfd, i, len, nselect;
	char		c;
	struct sockaddr_in		addr, clientaddr;
	signal(SIGCHLD, clr);
	signal(SIGINT, clr);
	fd_set	rset, oset;
	if(argc != 2)
		err_quit("args");
	listenfd = socket( AF_INET, SOCK_STREAM, 0 );/*服务器监听套接字*/
	if( listenfd < 0 ) {
		err_sys("socket");
	}
	nchild = atoi(argv[1]);
	mypid = (struct pro *)malloc( nchild * sizeof( struct pro) );
	bzero( &addr, sizeof(addr));
	addr.sin_family = AF_INET;
	addr.sin_port	 = htons(SERV_PORT);
	addr.sin_addr.s_addr = htonl(INADDR_ANY);
	bind(listenfd, (SA *)&addr, sizeof(struct sockaddr));
	fdmax = listenfd; 						/*用于select */
	for( i = 0; i < nchild; i++ ) {
		make_child(i, listenfd);				/*创建子进程并等待给客户服务*/
	}									/*传入listenfd子进程关闭监听套接字*/
	int idle = nchild;						/*当前空闲的子进程*/
	len = sizeof(struct sockaddr);
	FD_ZERO(&oset);
	FD_ZERO(&rset);
	listen(listenfd,5);
	FD_SET(listenfd, &oset);
	for( ; ;) {
		if(idle == 0) 
			FD_CLR(listenfd, &oset);		/*若没有空闲子进程,则禁止accept新连接*/
		else 
			FD_SET(listenfd, &oset);	
		rset = oset;
		nselect = select(fdmax+1, &rset, NULL, NULL, NULL);
		if( FD_ISSET(listenfd, &rset) ) {		/*收到客户端连接请求*/
			connfd = accept(listenfd, (SA *)&clientaddr, &len);
			for( i = 0; i < nchild; i++) {
				if(mypid[i].flags == IDLE) {	/*查找可供服务的子进程*/
					break;
				}
			}
			/*  */
			assert(i < nchild);
			/*  */
			//connfd = accept(listenfd, (SA *)&clientaddr, &len);
			printf("the id is %d \n",i);
			mypid[i].flags = BUSY;  				/*改变子进程状态*/
			mypid[i].cnt ++;						/*子进程服务次数*/
			FD_SET(mypid[i].fd, &oset);				/*加入描述集,等待子进程*/
			fdmax = max(fdmax ,mypid[i].fd);
			idle --;								/*空闲进程减少*/
			/*通知子进程*/
			write_the_fd(mypid[i].fd, " ", 1,connfd);/*把连接套接字传送给子进程*/
			close(connfd);						/*因为有有一个confd在飞,所以这个可以关闭*/
			if(--nselect  ==0)
				continue;
		}
		else {
			for(i = 0; i < nchild; i++ ) {
				if(FD_ISSET(mypid[i].fd , &rset)) {
					read(mypid[i].fd, &c, 1);
					mypid[i].flags = IDLE;
					FD_CLR(mypid[i].fd, &oset);
					idle ++;
				}
				if(--nselect  == 0)
					break;
			}

		}
		
		
	}
}
void clr(int signo) {
	pid_t pid;
	while( (pid = waitpid(-1, NULL, WNOHANG)) > 0 ) {
		printf("The child %d is done\n", pid);
	}
	sleep(2);
	kill(getpid(), SIGTERM);
	return;
}
pid_t make_child( int i, int listenfd ) {
	pid_t 	child;
	int		fd[2];
	socketpair( AF_LOCAL, SOCK_STREAM,0, fd);
	if( (child = fork( ) ) < 0 ) {
		err_quit("fork ");
	}
	if( child > 0 ) {  /*父进程*/
		//printf("the  i is  %d\n", i);
		close( fd[1] );
		mypid[i].id = child;
		mypid[i].fd= fd[0];
		mypid[i].flags = IDLE;
		mypid[i].cnt = 0;
		return child;
	}
	/*子进程*/
	close( fd[0] );
	close(listenfd);
	dup2( fd[1], STDERR_FILENO);
	close( fd[1]);
	child_func();
}

void child_func( void) {
	int 		connfd, n;
	char		c;
	char		buf[MAXLINE];
	for( ; ; ) {
		n = read_fd(STDERR_FILENO, &c, 1, &connfd);
		if(n <0 ) 
			err_sys("read_fd");
		again:
		while( (n = read(connfd, buf, sizeof(buf))) > 0 ){
			writen(connfd, buf ,n);
		}
		if((n < 0)&&( errno == EINTR))
			goto again;
		else if(n < 0)
			err_sys("error");
		else if(n ==0) {
			close(connfd);
			write(STDERR_FILENO, " ", 1);
		}
	}
}
ssize_t write_the_fd(int fd, void *ptr, size_t nbytes, int sendfd) {
	struct msghdr	msg;
	struct iovec	iov[1];
	union {
	  struct cmsghdr	cm;
	  char				control[CMSG_SPACE(sizeof(int))];
	} control_un;
	struct cmsghdr	*cmptr;
	msg.msg_control = control_un.control;
	msg.msg_controllen = sizeof(control_un.control);
	cmptr = CMSG_FIRSTHDR(&msg);
	cmptr->cmsg_len = CMSG_LEN(sizeof(int));
	cmptr->cmsg_level = SOL_SOCKET;
	cmptr->cmsg_type = SCM_RIGHTS;
	*((int *) CMSG_DATA(cmptr)) = sendfd;
	msg.msg_name = NULL;
	msg.msg_namelen = 0;
	iov[0].iov_base = ptr;
	iov[0].iov_len = nbytes;
	msg.msg_iov = iov;
	msg.msg_iovlen = 1;
	return(sendmsg(fd, &msg, 0));
}
ssize_t read_the_fd(int fd, void *ptr, size_t nbytes, int *recvfd) {
	struct msghdr	msg;
	struct iovec	iov[1];
	ssize_t			n;
	int				newfd;
	union {
	  struct cmsghdr	cm;
	  char				control[CMSG_SPACE(sizeof(int))];
	} control_un;
	struct cmsghdr	*cmptr;

	msg.msg_control = control_un.control;
	msg.msg_controllen = sizeof(control_un.control);
	msg.msg_name = NULL;
	msg.msg_namelen = 0;

	iov[0].iov_base = ptr;
	iov[0].iov_len = nbytes;
	msg.msg_iov = iov;
	msg.msg_iovlen = 1;

	if ( (n = recvmsg(fd, &msg, 0)) <= 0)
		return(n);
	if ( (cmptr = CMSG_FIRSTHDR(&msg)) != NULL &&
	    cmptr->cmsg_len == CMSG_LEN(sizeof(int))) {
		if (cmptr->cmsg_level != SOL_SOCKET)
			err_quit("control level != SOL_SOCKET");
		if (cmptr->cmsg_type != SCM_RIGHTS)
			err_quit("control type != SCM_RIGHTS");
		*recvfd = *((int *) CMSG_DATA(cmptr));
	} else
		*recvfd = -1;		
	return(n);
}


Linux网络编程——tcp并发服务器(多进程)

一、tcp并发服务器概述 一个好的服务器,一般都是并发服务器(同一时刻可以响应多个客户端的请求)。并发服务器设计技术一般有:多进程服务器、多线程服务器、I/O复用服务器等。 二、多...
  • lianghe_work
  • lianghe_work
  • 2015年06月15日 15:02
  • 4707

Linux基于TCP/IP简单的客户端、服务器通信程序实例

服务器端代码: /************************************************************************* > File Name: ser...
  • fucangzxx
  • fucangzxx
  • 2016年07月26日 10:08
  • 9975

TCP实现ECHO程序(服务端同时处理多个客户端的响应)

一、ECHO意为应答,程序的功能是客户端向服务器发送一个字符串,服务器不做任何处理,直接把字符串返回给客户端,ECHO是自己笨的客户/服务器程序。 二、目前为止我们编写的程序中,服务器只能处理一个客...
  • u014756517
  • u014756517
  • 2016年07月27日 09:46
  • 1582

linux网络编程:使用进程池实现TCP多客户服务

进程池服务:让服务器在启动的某个阶段创建一个进程池,每个客户请求由当前可用子进程池中的某个(闲置)子进程处理。使用进程池的优点在于可以节省fork的开销直接就能处理新到的客户。缺点就是父进程必须父进程...
  • li_wen01
  • li_wen01
  • 2017年02月15日 14:14
  • 155

TCP与UDP客户服务端实现大小写及SOCKET编程实现聊天程序

  • 2015年01月23日 16:30
  • 344KB
  • 下载

采用select实现多客户服务端

服务端用fork来处理多个客户的思路是为每个客户创建一个新建的进程进行单独处理,也就是采取了多个服务器进程的方式,这在涉及到数据库的应用中不是最佳的解决方案。一般的解决方案是让单个服务器进程在不阻塞、...
  • gigglesun
  • gigglesun
  • 2013年10月20日 11:06
  • 1185

Linux网络编程(1):TCP客户服务程序设计

关于Linux的网络编程与普通程序的区别在于:网络编程涉及到了客户端与服务端。...
  • gzbaishabi
  • gzbaishabi
  • 2014年09月17日 11:16
  • 1053

09_TCP客户服务端

一服务端 二客户端一、服务端#include "TcpServer.h" TcpServer::TcpServer(QWidget *parent):QWidget(parent) { _Se...
  • a2604539133
  • a2604539133
  • 2017年07月04日 15:56
  • 132

tcp通信客户服务端

  • 2012年10月20日 01:24
  • 876KB
  • 下载

一个简单的Java客户服务端源代码

  • 2011年03月15日 23:01
  • 31KB
  • 下载
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:简单进程池实现多TCP客户服务
举报原因:
原因补充:

(最多只允许输入30个字)