socket编程（多进程）

利用多进程实现多个客户端和一个服务端的CS模型

【上一篇】：一个客户端和一个服务器模型

利用父子进程实现，父进程监听接受新的连接，子进程处理新的连接。父进程还负责回收子进程
accept和read是阻塞函数，会被信号打断，此时不应该视为一个错误，需要加一个判断：errno=EINTER

服务端代码如下，客户端代码不变

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<ctype.h>
#include<sys/types.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<netinet/in.h>
#include <sys/wait.h>
#include <errno.h>

//信号处理函数
void waitchild()
{
    
	pid_t wpid;
	while(1)
	{
    
		wpid = waitpid(-1, NULL, WNOHANG);
		if(wpid > 0)
		{
    
			printf("child exit\n");
		}
		else if(wpid == 0 || wpid == -1)
		{
    
			break;
		}
	}
}

int main()
{
    
	int lfd = socket(AF_INET, SOCK_STREAM, 0);
	if(lfd < 0)
	{
    
		perror("socket error");
		return -1;
	}
	
	struct sockaddr_in serv;
	bzero(&serv, sizeof(serv));
	serv.sin_family = AF_INET;
	serv.sin_port = htons(8888);
	serv.sin_addr.s_addr = htonl(INADDR_ANY);
	int ret = bind(lfd, (struct sockaddr*)&serv, sizeof(serv));
	if(ret < 0)
	{
    
		perror("bind error");
		return -1;
	}
	
	listen(lfd, 128);

	//阻塞SIGCHLD信号，避免处理函数还未注册就产生信号
	sigset_t mask;
	sigemptyset(&mask);
	sigaddset(&mask, SIGCHLD);
	sigprocmask(SIG_BLOCK, &mask, NULL);

	struct sockaddr_in client;
	socklen_t len;
	int cfd;
	char ip[16];
	pid_t pid;
       	while(1)
	{
    
		len = sizeof(client);
		cfd = accept(lfd, (struct sockaddr*)&client, &len);
		if(errno == EINTR)
		{
    
			cfd = accept(lfd, (struct sockaddr*)&client, &len);
		}
		memset(ip, 0, sizeof(ip));
		printf("client: ip == [%s], port == [%d]\n", inet_ntop(AF_INET, &client.sin_addr.s_addr, ip, sizeof(ip)), ntohs(client.sin_port));
		printf("lfd == [%d], cfd == [%d]\n", lfd, cfd);
		
		pid = fork();
		if(pid < 0)
		{
    
			perror("fork error\n");
			exit(-1);
		}
		else if(pid > 0)
		{
    
			close(cfd);
			//注册回调函数
			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
		{
    
			close(lfd);
			int n = 0;
			char buf[1024];
			while(1)
			{
    
				memset(buf, 0, sizeof(buf));
				n = read(cfd, buf, sizeof(buf));
				if(n <= 0)
				{
    
					printf("read error or client close\n");
					break;
				}
				//父进程和创建出来的子进程client都互不影响（读共享写复制）
				//printf("client: ip == [%s], port == [%d]\n", inet_ntop(AF_INET, &client.sin_addr.s_addr, ip, sizeof(ip)), ntohs(client.sin_port));
				printf("port == [%d]: n == [%d], buf == [%s]\n", ntohs(client.sin_port), n, buf);

				for(int i=0; i<n; i++)
				{
    
					buf[i] = toupper(buf[i]);
				}

				write(cfd, buf, n);
			}
			close(cfd);
			//避免通信结束后子进程循环创建子进程
			exit(0);
		}
	}
	close(lfd);
	return 0;
}

【下一篇】:多线程版本