网络编程day4

多进程

#include <myhead.h>

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

#define PORT 8888
#define IP "192.168.125.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[])
{
	//捕获１７）SINGCHLD信号
	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;  //必须填AF_INFT
	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("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 pid;
	
	//获取一个已经完成的客户端信息，生成一个新的文件描述符 accept
	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);

		//能运行到当前位置，则代表有客户端连接成功，子进程只负责交互
		pid = fork();
		if(0 == pid)
		{
			close(sfd);
			deal_cli_msg(newfd,cin);
			exit(0);
		}
		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);
		//发送
		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
#define IP "192.168.125.61"

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

void* deal_cli_msg(void* arg);

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;  //必须填AF_INFT
	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("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)
	{
		//获取一个已经完成的客户端信息，生成一个新的文件描述符 accept
		//父进程
		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);
	}

	//关闭文件描述符
	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");
			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 : %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);
}