【网络编程笔记三】多进程并发服务器简单例子

实现一个多进程并发服务器，可以用多个客户端同时连接该服务端。服务端每连接也给客户端就创建一个子进程来处理，父进程则继续负责接收客户端的连接。

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <ctype.h>
#include <strings.h>
#include <sys/wait.h>

#define SERV_PORT 8888

/**************************************************
父进程收到子进程结束的信号时，等待while循环等待子进程结束
waitpid参数二:pid=0：等待进程组识别码与目前进程相同的任何子进程。
waitpid参数三：WNOHANG：若pid指定的子进程没有结束，则waitpid()函数返回0，不予以等待。若结束，则返回该子进程的ID
**************************************************/
void wait_child(int signo)
{
 	while (waitpid(0, NULL, WNOHANG) > 0);
	return;
}

int main(void)
{
	pid_t pid;
    int sfd, cfd;
   	struct sockaddr_in serv_addr, clie_addr;
    socklen_t clie_addr_len;
   	char buf[BUFSIZ], clie_IP[BUFSIZ];
    int n,i;
	
	sfd = socket(AF_INET, SOCK_STREAM, 0);
	bzero(&serv_addr, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(SERV_PORT);
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	bind(sfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr));
    listen(sfd, 128);
	
	/*父进程不断的给自己“揽活”，即循环等待子进程的连接*/
	while(1)
	{
		clie_addr_len = sizeof(clie_addr);
        	cfd = accept(sfd, (struct sockaddr *)&clie_addr, &clie_addr_len);
        	printf("client IP:%s, port:%d\n", inet_ntop(AF_INET, &clie_addr.sin_addr.s_addr, clie_IP, 			sizeof(clie_IP)), ntohs(clie_addr.sin_port));

		/*连接成功后，创建子进程来处理和刚连接的客户端通信的事情，然后父进程继续负责接收其他子进程的连接*/
		pid = fork();
		if (pid < 0)
		{
            		perror("fork error");
            		exit(1);
		}
		else if(pid == 0)//子进程关闭当前服务端套接字，然后跳出去循环去处理和客户端之间工作
		{
			close(sfd);
            		break;
		}
		else//父进程关闭连接套接字，等待子进程结束，回收僵尸进程
		{
			close(cfd);
            		signal(SIGCHLD, wait_child);//SIGCHLD：在一个进程终止或者停止时，将SIGCHLD信号发送给其父进程
		}
	}
	
	/*子进程负责通信的内容部分*/
	if (pid == 0)
	{
        	while (1)
			{
            	n = read(cfd, buf, sizeof(buf));//read读套接字时，读到返回值为0则说明对端（客户端）关闭
            	if (n == 0)
				{
                	close(cfd);
                	return 0;
            	}
    			else if (n == -1)
    			{
                    	perror("read error");
                    	exit(1);
           		}
    			else
    			{
    	        	for(i = 0; i < n; i++)
                    buf[i] = toupper(buf[i]);
                    write(cfd, buf, n);
                    write(STDOUT_FILENO, buf, n);
                }            	
            }
    }
	return 0;
}

执行结果：