简单实现多进程和多线程的并发服务器

多进程版本的并发服务器

//多进程版本的网络服务器
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<sys/types.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include<ctype.h>
#include<signal.h>
#include"wrap.h"
#include<sys/wait.h>


//信号处理函数
void waitchild(int signo){

	pid_t wpid;
	//情况:在SIGCHLD执行期间,另外的2个以上的进程同时退出,此时会出现僵尸进程
	//循环回收子进程,
	//当waitpid==-1时表示已经回收所有子进程
	while(1){

		wpid = waitpid(-1 , NULL , WNOHANG);
		if(wpid>0){

			printf("child exit , wpid==[%d]\n", wpid);
		}
		else if(wpid==0 || wpid==-1){

			break;
		}
	}

}
int main()
{
	//创建socket
	

	int lfd=Socket(AF_INET,SOCK_STREAM,0);


	//设置端口复用 
	//解决"bind Address already in use "
	int opt = 1;
	int ret=setsockopt(lfd, SOL_SOCKET ,SO_REUSEADDR, &opt ,sizeof(int));
	if(ret==-1){

		printf("sockopt");
		exit(1);
	}
	//绑定
	struct sockaddr_in serv;
	bzero(&serv,sizeof(serv));
	serv.sin_family=AF_INET;
	serv.sin_addr.s_addr=htonl(INADDR_ANY);
	serv.sin_port = htons(8888);
	Bind(lfd,(struct sockaddr *)&serv, sizeof(serv));

	//设置监听
	listen(lfd,128);


	//阻塞SIGCHLD信号
	//防止情况发生:还未完成信号处理函数的注册,子进程都推出了
	sigset_t mask;
	sigemptyset(&mask);
	sigaddset(&mask , SIGCHLD);
	sigprocmask(SIG_BLOCK, &mask ,NULL);


	pid_t pid;
	int cfd;
	char sIP[16];
	socklen_t len;
	struct sockaddr_in client;

	while(1){

		//等待客户端连接
		memset(sIP , 0x00, sizeof(sIP));
		len = sizeof(client);

		cfd=Accept(lfd, (struct sockaddr *)&client , &len);
		printf("client-->[%s]:[%d]\n",inet_ntop(AF_INET, &client.sin_addr.s_addr ,sIP, sizeof(sIP)), ntohs(client.sin_port));

		pid = fork();
		if(pid<0){

			perror("fork error\n");
			exit(-1);

		}
		else if(pid>0){

			//关闭通信文件描述符
			close(cfd);

			//注册SIGCHLD信号处理函数
			//sa_handler:指定信号捕捉后的处理函数名
			//sa_mask:用来指定在信号处理函数执行期间所要屏蔽(阻塞)的信号
			struct sigaction act;
			act.sa_handler = waitchild;
			act.sa_flags = 0;
			sigemptyset(&act.sa_mask);   //清空信号集
			sigaction(SIGCHLD , &act , NULL);

			//解除对SIGCHLD信号的阻塞
			sigprocmask(SIG_UNBLOCK , &mask , NULL);
		}
		else if(pid==0){

			//关闭监听文件描述符
			close(lfd);

			//读数据
			int n;
			char buf[1024];

			while(1){

				n=Read(cfd, buf ,sizeof(buf));
			
				if(n<=0){

				printf("read error or client closed,n=[%d]\n",n);
				break;
				}

				for(int i=0;i<n;++i){

					buf[i] = toupper(buf[i]);
				}
			//发送数据
				Write( cfd, buf, n );
			}
		
		}
			
		//关闭cfd;
		
		close(cfd);
		exit(0);

	}

	close(lfd);
	return 0;
}

注意情况发生:
1.还未完成信号处理函数的注册,子进程都退出的问题
2.端口复用问题

改进
//信号处理函数
void waitchild(int signo){

	pid_t wpid;
	//情况:在SIGCHLD执行期间,另外的2个以上的进程同时退出,此时会出现僵尸进程
	//循环回收子进程,
	//当waitpid==-1时表示已经回收所有子进程
	while(1){

		wpid = waitpid(-1 , NULL , WNOHANG);
		if(wpid>0){

			printf("child exit , wpid==[%d]\n", wpid);
		}
		else if(wpid==0 || wpid==-1){

			break;
		}
	}

}
int main()
{
	//创建socket
	

	int lfd=Socket(AF_INET,SOCK_STREAM,0);

	//绑定
	struct sockaddr_in serv;
	bzero(&serv,sizeof(serv));
	serv.sin_family=AF_INET;
	serv.sin_addr.s_addr=htonl(INADDR_ANY);
	serv.sin_port = htons(8888);
	Bind(lfd,(struct sockaddr *)&serv, sizeof(serv));

	//设置监听
	listen(lfd,128);


	//阻塞SIGCHLD信号
	//防止情况发生:还未完成信号处理函数的注册,子进程都推出了
	sigset_t mask;
	sigemptyset(&mask);
	sigaddset(&mask , SIGCHLD);
	sigprocmask(SIG_BLOCK, &mask ,NULL);


	pid_t pid;
	int cfd;
	char sIP[16];
	socklen_t len;
	struct sockaddr_in client;

	while(1){

		//等待客户端连接
		memset(sIP , 0x00, sizeof(sIP));
		len = sizeof(client);

		cfd=Accept(lfd, (struct sockaddr *)&client , &len);
		printf("client-->[%s]:[%d]\n",inet_ntop(AF_INET, &client.sin_addr.s_addr ,sIP, sizeof(sIP)), ntohs(client.sin_port));

		pid = fork();
		if(pid<0){

			perror("fork error\n");
			exit(-1);

		}
		else if(pid>0){

			//关闭通信文件描述符
			close(cfd);

		}
		else if(pid==0){

			//关闭监听文件描述符
			close(lfd);

			//读数据
			int n;
			char buf[1024];

			while(1){

				n=Read(cfd, buf ,sizeof(buf));
			
				if(n<=0){

				printf("read error or client closed,n=[%d]\n",n);
				break;
				}

				for(int i=0;i<n;++i){

					buf[i] = toupper(buf[i]);
				}
			//发送数据
				Write( cfd, buf, n );
			}
		
		}
			
		//关闭cfd;
		
		close(cfd);
		exit(0);

	}

	close(lfd);
	return 0;
}

多线程版本的高并发服务器

void *do_work(void *arg)
{
	
	int cfd=*(int *)arg;

	int n;
	char buf[1024];

	while(1){

		//read数据
		memset(buf, 0x00, sizeof(buf));

		n=Read(cfd , buf ,sizeof(buf));

		if(n<=0){

			printf("read error or client closed,n=[%d]\n",n);
			break;
		}
		printf("n==[%d],buf==[%s]\n",n,buf);

		for(int i=0;i<n;++i){

			buf[i]=toupper(buf[i]);
		}

		//发送数据给客户端
		Write(cfd, buf ,n);

	}

	//关闭通信文件描述符
	close(cfd);

}

int main()
{
	//创建socket
	int lfd = Socket(AF_INET, SOCK_STREAM, 0);

	//bind
	struct sockaddr_in serv;
	bzero(&serv,sizeof(serv));
	serv.sin_family = AF_INET;
	serv.sin_addr.s_addr = htonl(INADDR_ANY);
	serv.sin_port = htons(8888);

	bind(lfd, (struct sockaddr *)&serv , sizeof(serv));

	//监听
	Listen(lfd , 128);


	int cfd;
	pthread_t thread;
	while(1){

		//接受新的连接
		cfd = Accept(lfd ,NULL ,NULL);

		//创建线程
		pthread_create(&thread,NULL,&do_work,&cfd);


		//分离属性
		pthread_detach(thread);
	}

	//关闭监听文件描述符
	close(lfd);

	return 0;

}