华清远见第五课程作业day4

2. 多进程并发服务器。

#include<myhead.h>

#define IP "192.168.114.52"   //本机IP
#define PORT 25565           //服务器端口
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)          //输出错误信息和位置

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)SIGCHLD信号,为信号注册新的处理函数
	if(signal(17,handler)==SIG_ERR){
		ERR_MSG("signal");
		return -1;
	}

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

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

	//填充地址信息结构体给bind函数绑定,
	//真实的地址信息结构体根据地址族指定 AF_INET
	struct sockaddr_in sin;
	sin.sin_family = AF_INET; //必须填AF_INET
	sin.sin_port = htons(PORT);//将端口号转变为网络字节序
	sin.sin_addr.s_addr = inet_addr(IP);//本机IP
	
	//绑定服务器的地址信息 ——>必须绑定bind
	//bind 用来绑定服务器的地址信息，绑定到监听套接字文件对应的空间中
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//将套接字设置为被动监听状态 listen
	//将通信套接字转换成监听套接字
	//listen让内核维护量队列:未完成队列，已完成队列
	//其中未完成队列是正在连接的客户端信息；
	//已完成的是已经连接上的客户端信息
	if(listen(sfd,128)<0){    //数字为未完成连接队列的容量不能少于１
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	int newfd =-1;
	struct sockaddr_in cin; //用来储存客户端信息
	socklen_t addrlen = sizeof(cin); //真实地址的信息结构体的大小
	pid_t cpid= -1;

	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){
			//子进程只负责与客户端交互
			deal_cli_msg(newfd,cin);
			exit(0);
		}
		close(newfd);
	}

	//关闭文件名描述符
	close(newfd);
	if(close(sfd)<0){
		ERR_MSG("close");
		return -1;
	}

	return 0;
}

int deal_cli_msg(int newfd,struct sockaddr_in cin)
{
	char buf[128]="";
	ssize_t res = 0;
	while(1){
		//清空字符串
		bzero(buf,sizeof(buf));
		//接收
		res = recv(newfd,buf,sizeof(buf),0);
		if(res<0){
			ERR_MSG("recv");
			return -1;
		}
		else if(res == 0){
			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);
		//发送
		if(send(newfd,buf,sizeof(buf),0)<0){
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");
	}
	close(newfd);
	return 0;
}

3. 多线程并发服务器。

#include<myhead.h>

#define IP "192.168.114.52"   //本机IP
#define PORT 25565           //服务器端口
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)          //输出错误信息和位置

struct Climsg{
	int newfd;
	struct sockaddr_in cin;
};
void *deal_cli_msg(void* arg);


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

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

	//填充地址信息结构体给bind函数绑定,
	//真实的地址信息结构体根据地址族指定 AF_INET
	struct sockaddr_in sin;
	sin.sin_family = AF_INET; //必须填AF_INET
	sin.sin_port = htons(PORT);//将端口号转变为网络字节序
	sin.sin_addr.s_addr = inet_addr(IP);//本机IP
	
	//绑定服务器的地址信息 ——>必须绑定bind
	//bind 用来绑定服务器的地址信息，绑定到监听套接字文件对应的空间中
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//将套接字设置为被动监听状态 listen
	//将通信套接字转换成监听套接字
	//listen让内核维护量队列:未完成队列，已完成队列
	//其中未完成队列是正在连接的客户端信息；
	//已完成的是已经连接上的客户端信息
	if(listen(sfd,128)<0){    //数字为未完成连接队列的容量不能少于１
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	int newfd =-1;
	struct sockaddr_in cin; //用来储存客户端信息
	socklen_t addrlen = sizeof(cin); //真实地址的信息结构体的大小
	pthread_t tid;
	struct Climsg 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(&tid,NULL,deal_cli_msg,(void*)&info)!=0){
			fprintf(stderr,"pthread_create failed __%d__\n",__LINE__);
			return -1;
		}
		pthread_detach(tid);
	}

	//关闭文件描述符
	close(newfd);
	if(close(sfd)<0){
		ERR_MSG("close");
		return -1;
	}

	return 0;
}

void* deal_cli_msg(void* arg)
{
	int newfd = ((struct Climsg*)arg)->newfd;
	struct sockaddr_in cin = ((struct Climsg*)arg)->cin;
	char buf[128]="";
	ssize_t res = 0;
	while(1){
		//清空字符串
		bzero(buf,sizeof(buf));
		//接收
		res = recv(newfd,buf,sizeof(buf),0);
		if(res<0){
			ERR_MSG("recv");
			return NULL;
		}
		else if(res == 0){
			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);
		//发送
		if(send(newfd,buf,sizeof(buf),0)<0){
			ERR_MSG("send");
			return NULL;
		}
		printf("send success\n");
	}
	close(newfd);
	
	pthread_exit(NULL);
}