网络编程 Day4

目录

1. 多进程并发服务器根据流程图重新编写

2. 多线程并发服务器重新写一遍

3.思维导图

---

1. 多进程并发服务器根据流程图重新编写

#include <myhead.h>
#define PORT 8888   //端口号，范围1024-49151
#define IP "192.168.114.61"

int deal_cli_msg(int newfd,struct sockaddr_in cin);

//回收僵尸进程
void handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0);
}

int main(int argc, const char *argv[])
{
	//捕获17)SIGCHID信号，为信号注册新的处理函数
	if(signal(17,handler)==SIG_ERR)
	{
		ERR_MSG("signal");
		return -1;
	}

	//创建流式套接字  socket
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d\n",sfd);

	//允许端口快速的被复用
	int reuse =1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口被快速复用成功\n");

	//填充地址信息结构体给bind函数绑定，
	//真实的地址信息结构体根据地址族指定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;   //必须填这个
	sin.sin_port=htons(PORT);  //端口号的网络字节序
	sin.sin_addr.s_addr=inet_addr(IP);   //本机IP

	//绑定服务器的地址信息--->必须绑定   bind
	if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("绑定成功\n");

	//将套接字设置为被动监听状态   listen
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("监听状态设置成功\n");

	//获取一个已经完成的客户端信息，生产一个新的文件描述符   accept
	int newfd=-1;
	struct sockaddr_in cin;     //存储客户端的地址信息
	socklen_t addrlen=sizeof(cin);    //真实的地址信息结构体的大小
	//newfd=accept(sfd,NULL,NULL);
	int cpid;
	while(1)
	{
		newfd=accept(sfd,(struct sockaddr *)&cin,&addrlen);
		if(newfd<0)
		{
			ERR_MSG("newfd");
			return -1;
		}
		printf("[%s:%d]客户端连接成功 newfd=%d\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
		cpid=fork();
		if(cpid==0)
		{
			//子进程
			close(sfd);  //对于子进程，sfd没用
			deal_cli_msg(newfd,cin);
			exit(0);
		}else if(cpid>0)
		{
			//父进程
			close(newfd);  //对于父进程而言，newfd没用
		}
		else
		{
			ERR_MSG("fork");
			return -1;
		}
	}
	//关闭文件描述符
	close(newfd);
	if(close(sfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}

int deal_cli_msg(int newfd,struct sockaddr_in cin)
{
	while(1)
	{
	char buf[128]="";
	ssize_t res=0;
		//清空字符串
		bzero(buf,sizeof(buf));   //memset
		//接收
		res=recv(newfd,buf,sizeof(buf),0);
		if(res<0)
		{
			ERR_MSG("recv");
			return -1;
		}
		else if(0==res)
		{
			printf("[%s:%d]客户端下线 newfd=%d\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
			break;
		}
		printf("[%s:%d]收到消息 newfd=%d:%s\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf);

		//发送
		strcat(buf,">_<");
		if(send(newfd,buf,sizeof(buf),0)<0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");

	}
	close(newfd);
	return 0;
}

2. 多线程并发服务器重新写一遍

#include <myhead.h>
#define PORT 8888   //端口号，范围1024-49151
#define IP "192.168.114.83"

void *deal_cli_msg(void *arg);

struct Tcp
{
	int newfd;
	struct sockaddr_in cin;
};

int main(int argc, const char *argv[])
{
	//创建流式套接字  socket
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d\n",sfd);

	//允许端口快速的被复用
	int reuse =1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口被快速复用成功\n");

	//填充地址信息结构体给bind函数绑定，
	//真实的地址信息结构体根据地址族指定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;   //必须填这个
	sin.sin_port=htons(PORT);  //端口号的网络字节序
	sin.sin_addr.s_addr=inet_addr(IP);   //本机IP

	//绑定服务器的地址信息--->必须绑定   bind
	if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("绑定成功\n");

	//将套接字设置为被动监听状态   listen
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("监听状态设置成功\n");

	//获取一个已经完成的客户端信息，生产一个新的文件描述符   accept
	int newfd=-1;
	struct sockaddr_in cin;     //存储客户端的地址信息
	socklen_t addrlen=sizeof(cin);    //真实的地址信息结构体的大小
	//newfd=accept(sfd,NULL,NULL);
	pthread_t cpid;
	struct Tcp info;
	while(1)
	{
		newfd=accept(sfd,(struct sockaddr *)&cin,&addrlen);
		if(newfd<0)
		{
			ERR_MSG("newfd");
			return -1;
		}
		printf("[%s:%d]客户端连接成功 newfd=%d\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
		

		//创建子线程
		info.newfd=newfd;
		info.cin=cin;
		if(pthread_create(&cpid,NULL,deal_cli_msg,&info)!=0)
		{
			printf("pthread_create error\n");
			return -1;
		}
		pthread_detach(cpid);
	}
	//关闭文件描述符
	close(newfd);
	if(close(sfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}

void *deal_cli_msg(void *arg)
{
	int newfd=((struct Tcp *)arg)->newfd;
	struct sockaddr_in cin=((struct Tcp *)arg)->cin;
	while(1)
	{
	char buf[128]="";
	ssize_t res=0;
		//清空字符串
		bzero(buf,sizeof(buf));   //memset
		//接收
		res=recv(newfd,buf,sizeof(buf),0);
		if(res<0)
		{
			ERR_MSG("recv");
			break;	
		}
		else if(0==res)
		{
			printf("[%s:%d]客户端下线 newfd=%d\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
			break;
		}
		printf("[%s:%d]收到消息 newfd=%d:%s\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf);

		//发送
		strcat(buf,">_<");
		if(send(newfd,buf,sizeof(buf),0)<0)
		{
			ERR_MSG("send");
			break;
		}
		printf("send success\n");

	}
	close(newfd);
	pthread_exit(NULL);
	return 0;
}

3.思维导图