网络编程day3

多线程并发任务
include <myhead.h>
 
#define PORT 8888
#define IP "192.168.114.153"
 
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__:",__LINE__);\
	perror(msg);\
}while(0)
 
void handler(int sig){
	while(waitpid(-1,NULL,WNOHANG) > 0);
}
 
int deal_cli_msg(int newfd,struct sockaddr_in cin);
 
int main(int argc, const char *argv[])
{
	//处理僵尸进程
	if(signal(SIGCHLD,handler) == SIG_ERR){
		ERR_MSG("signal");
		return -1;
	}
	//创建流式套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd == -1){
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success\n");
	//允许端口可以被快速复用
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) == -1){
		ERR_MSG("setsockopt");
		return -1;
	}
	//地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	//绑定
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
	//监听
	if(listen(sfd,128) == -1){
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
 
	//创建进程号
	pid_t pid;
	int newfd;
	//循环
	while(1){	
		//存储客户端的地址信息
		struct sockaddr_in cin;
		socklen_t addrlen = sizeof(cin);
		//连接，产生新的文件描述符
		newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
		if(newfd == -1){
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s:%d]客户端连接成功  newfd = %d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
		//创建进程
		pid = fork();
		if(pid < 0){
			ERR_MSG("fork");
			return -1;
		}else if(pid == 0){
			//关闭不必要的sfd
			close(sfd);
			//处理客户端信息
			deal_cli_msg(newfd,cin);
			//退出进程
			exit(0);
		}else{
			close(newfd);
		}
	}
	close(sfd);
	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 == -1){
			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 buf = %s\n",inet_ntoa(cin.sin_addr),htons(cin.sin_port),newfd,buf);
 
		strcat(buf," *-*");
		if(send(newfd,buf,sizeof(buf),0) == -1){
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");
	}
	close(newfd);
	return 0;

多线程并发任务
#include <myhead.h>
 
#define PORT 8888
#define IP "192.168.114.153"
 
#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 == -1){
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success\n");
	//允许端口可以被快速复用
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) == -1){
		ERR_MSG("setsockopt");
		return -1;
	}
	//地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);
	//绑定
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
	//监听
	if(listen(sfd,128) == -1){
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
	//循环
	//定义新的文件描述符和地址信息结构体
	int newfd;
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	pthread_t tid;
	while(1){
		newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
		if(newfd == -1){
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s:%d]客户端连接成功 newfd = %d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
		//定义线程结构体函数参数
		struct climsg info;
		info.newfd = newfd;
		info.cin = cin;
		//创建线程
		if(pthread_create(&tid,NULL,deal_cli_msg,(void *)&info)){
			printf("pthread_create error\n");
			return -1;
		}
		//将线程设置为分离态回收资源
		pthread_detach(tid);
	}
	//关闭sfd
	close(sfd);
	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;
	while(1){
		//接收
		res = recv(newfd,buf,sizeof(buf),0);
		if(res == -1){
			ERR_MSG("recv");
			break;
		}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 buf = %s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf);
		
		//发送
		strcat(buf," *-*");
		if(send(newfd,buf,sizeof(buf),0) == -1){
			ERR_MSG("send");
			break;
		}
		printf("send success\n");
	}
	//关闭newfd并退出线程
	close(newfd);
	pthread_exit(NULL);
}