day0830

最新推荐文章于 2024-07-12 14:44:00 发布

小嘉~230523

最新推荐文章于 2024-07-12 14:44:00 发布

阅读量239

点赞数

文章标签：开发语言

本文链接：https://blog.csdn.net/m0_73743794/article/details/132592060

版权

多进程

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)

#define PORT 8888 				//端口号，范围1024~49151
#define IP   "192.168.114.92" 	//本机IP，ifconfig

//子进程只负责与客户端交互
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;
	}
	
	//创建流式套接字，socket
	int sfd;
	if((sfd = socket(AF_INET, SOCK_STREAM,0)) < 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("setsocketopt");
		return -1;
	}
	printf("允许端口被快速的复用成功\n");

	//填充地址信息结构体给 bind函数绑定，
	//真实的地址信息根据地址族指定 AF_INET: man 7 IP
	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

	if(bind(sfd, (struct sockaddr *)&sin, sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//将套接字设置为被动监听状态 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;
	//获取一个已经完成的客户端信息，生成一个新的文件描述符 accept
	while(1)
	{
		newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
		if(newfd < 0)
		{
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s:%d]客户端连接成功 newfd=%d\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
		//子进程只负责交互
		cpid = fork();
		if(0 == cpid)
		{
			close(sfd); 				//对于子进程,sfd没用
			//子进程只负责与客户端交互
			deal_cli_msg(newfd, cin);   //退出子进程，因为子进程只负责交互
			exit(0); 	
		}
		close(newfd); 					//对于父进程，newfd没用
	}
	//关闭文件描述符
	if(close(sfd)<0)
	{
		ERR_MSG("close\n");
		return -1;
	}
	return 0;
}

int deal_cli_msg(int newfd, struct sockaddr_in cin)
{
	char buf[128] = "";
	ssize_t res;
	while(1)
	{
		//清空字符串
		bzero(buf, sizeof(buf));
		//接收
		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;
}

多线程

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)

#define PORT 8888 				//端口号，范围1024~49151
#define IP   "192.168.1.6" 	//本机IP，ifconfig

//需要传入到分支线程的参数
struct Climsg{
	int newfd;
	struct sockaddr_in cin;
};

void* deal_cli_msg(void* arg);

/******************主函数********************/
int main(int argc, const char *argv[])
{
	
	//创建流式套接字，socket
	int sfd;
	if((sfd = socket(AF_INET, SOCK_STREAM,0)) < 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("setsocketopt");
		return -1;
	}
	printf("允许端口被快速的复用成功\n");

	//填充地址信息结构体给 bind函数绑定，
	//真实的地址信息根据地址族指定 AF_INET: man 7 IP
	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

	if(bind(sfd, (struct sockaddr *)&sin, sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");

	//将套接字设置为被动监听状态 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("accept");
			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_creatr failed __%d__\n",__LINE__);
			return -1;
		}
		//分离线程，分离线程退出后，由内核自动回收
		pthread_detach(tid);
	}
	//关闭文件描述符
	if(close(sfd)<0)
	{
		ERR_MSG("close\n");
		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");
			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;
}