天天看點

Linux網絡程式設計多程序模型

#include <string.h>
#include <errno.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <stdio.h>

#define SERV_PORT 8080
#define MAXLINE 100
#define MAXSIZE 4096

void sys_error(const char *fun_str,int exitno)
{
	perror(fun_str);
	exit(exitno);
}

void sig_handle(int signo)
{
	int status;
	pid_t pid;
	
	while( (pid = waitpid(-1,&status,WNOHANG))>0 )
	{
		printf("child %d terminated\n",pid);
	}
	
	return;
}

//使用多程序版本中容易忽視的問題 ,我們隻是負責建立了子程序去處理去用戶端的互動 ,卻沒有考慮到子程序結束時 ,父程序并沒有回收子程序的資源
//子程序在結束時會發送SIGCHLD信号 是以我們可以在信号處理函數中 去回收子程序的資源(PCB等)
int main(void)
{
	int ret=0;
	int n;
	int sockfd,connfd;
	int cli_len;
	struct sockaddr_in serv_addr,cli_addr;
	char buf[MAXLINE];
	char info[MAXSIZE];
	pid_t pid;

	sockfd = socket(AF_INET,SOCK_STREAM,0);
	if(-1 == sockfd)
	{
		sys_error("socket",-1);
	}

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

	ret = bind(sockfd,(struct sockaddr*)&serv_addr,sizeof(serv_addr));
	if(-1 == ret)
	{
		sys_error("bind",-2);
	}

	//listen 中backlog的值為兩個連接配接隊列的和 已完成連接配接隊列 和正處于三次握手的待連接配接隊列
	ret = listen(sockfd,20);
	if(-1 == ret)
	{
		sys_error("listen",-3);
	}

	printf("waiting for connnect\n");
	
	//SIGCHLD的信号處理函數用于回收子程序的資源
	signal(SIGCHLD,sig_handle);
	
	while(1)
	{
		cli_len = sizeof(cli_addr);
		connfd =accept(sockfd,(struct sockaddr*)&cli_addr,&cli_len);
		if(-1 == connfd)
		{
			if(EINTR == errno)
			{
				connfd = accept(sockfd,(struct sockaddr*)&cli_addr,&cli_len);
			}
			sys_error("accept",-4);
		}
		
		printf("recived from ip %s port %d\n",
			inet_ntop(AF_INET,&cli_addr.sin_addr.s_addr,buf,sizeof(buf)),ntohs(cli_addr.sin_port));
		
		/*accept阻塞于等待用戶端的連接配接 如果此時服務端需要讀取資料又會阻塞 這時服務端便不能與新的用戶端進行連接配接 如果将read設定為非阻塞模式,也隻是以輪詢的模式與
			新的用戶端進行連接配接 如果服務端有資料需要讀取 這時新的連接配接又會失敗
		這時采取多程序的方式處理 父程序等待用戶端的連接配接 而子程序負責處理與用戶端具體的互動
		*/
		pid = fork();
		if(0==pid)
		{
			//建立父子程序時 子程序是父程序的完全拷貝 關閉不必要的檔案描述符
			close(sockfd);
			while(1)
			{
				n = read(connfd, info, sizeof(info));
				if(n<0)
				{
					sys_error("read",-6);
				}
				
				ret = write(connfd, info, n); 
				if(ret<0)
				{
					sys_error("write",-7);
				}
			}
			ret = close(connfd);
			if(-1 == ret)
			{
				sys_error("close",-5);
			}
		}
		else if(pid >0)
		{
			//父程序隻負責監聽來自用戶端的連接配接
			close(connfd);
		}
		else
		{
			sys_error("fork",-8);
		}
	}

	return 0;
}
           

繼續閱讀