网络编程-day3 0518

多线程并发服务器

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <pthread.h>

#define ERR_MSG(msg) do{\
	printf("line=%d\n",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 1234
#define IP "192.168.10.224"

struct climsg
{
	int nfd;
	struct sockaddr_in cin;
};
void* callback2(void *arg)
{	
	char buf[128]="";
	int nfd=((struct climsg *)arg)->nfd;
	struct sockaddr_in cin=((struct climsg *)arg)->cin;
	while(1)
	{
		bzero(buf,sizeof(buf));
		ssize_t res = recv(nfd,buf,sizeof(buf),0);
		if(res<0)
		{
			ERR_MSG("recv");
			return NULL;
		}
		else if(0==res)
		{
			printf("newfd=%d 客户端下线\n",nfd);
			close(nfd);
			break;
		}
		printf("[%s:%d] :%s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),buf);
	}
	pthread_exit(NULL);
}

int main(int argc, const char *argv[])
{
	int nfd;	
	struct sockaddr_in cin;

	ssize_t res=0;


	//创建流式套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket creat success line = %d\n",__LINE__);
	//填充服务器自身的地址信息结构体
	//真实的地址信息结构体根据地址族制定 man 7 IP
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;//必须填充AF_INET
	sin.sin_port=htons(PORT);//端口号的网络字节序 1024 ~ 49151
	sin.sin_addr.s_addr=inet_addr(IP);// 填充本机IP
	//将IP地址和端口号绑定到指定的套接字文件描述符上
	if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind creat success line = %d\n",__LINE__);
	//将通讯套接字设置为被动监听状态，用于监听是否有客户端连接，128为未完成连接的队列容量，从此
	//后sfd这个文件描述符不用于通讯，用于监听
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen creat success line = %d\n",__LINE__);	
	pthread_t tid2;
	struct climsg info;
	socklen_t len=sizeof(cin);
	while(1)
	{
		//从已完成连接的队列中，获取一个客户端信息，并生成一个新的文件描述符
		//这个文件符才是服务器与客户端通信的文件描述符
		nfd=accept(sfd,(struct sockaddr *)&cin,&len);
		if(nfd<0)
		{
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s:%d] 客户端连接成功 line = %d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),__LINE__);
		info.nfd=nfd;
		info.cin=cin;
		if(pthread_create(&tid2,NULL,callback2,&info)<0)
		{
			printf("error\n");
			return -1;
		}
		pthread_detach(tid2);
	}
	pthread_join(tid2,NULL);
	return 0;
}

多进程并发服务器

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <stdlib.h>
#define ERR_MSG(msg) do{\
	printf("line=%d\n",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 1234
#define IP "192.168.10.224"
int deal_cli_msg(int nfd,struct sockaddr_in cin);
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 creat success line = %d\n",__LINE__);
	//填充服务器自身的地址信息结构体
	//真实的地址信息结构体根据地址族制定 man 7 IP
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;//必须填充AF_INET
	sin.sin_port=htons(PORT);//端口号的网络字节序 1024 ~ 49151
	sin.sin_addr.s_addr=inet_addr(IP);// 填充本机IP
	//将IP地址和端口号绑定到指定的套接字文件描述符上
	if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind creat success line = %d\n",__LINE__);
	//将通讯套接字设置为被动监听状态，用于监听是否有客户端连接，128为未完成连接的队列容量，从此
	//后sfd这个文件描述符不用于通讯，用于监听
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen creat success line = %d\n",__LINE__);
	//从已完成连接的队列中，获取一个客户端信息，并生成一个新的文件描述符
	//这个文件符才是服务器与客户端通信的文件描述符
	struct sockaddr_in cin;
	socklen_t len=sizeof(cin);
	int nfd=-1;
	if(signal(17,zombie_handler)==SIG_ERR)
	{
		ERR_MSG("signal");
		return -1;
	}
	while(1)
	{
		//父进程进行监听
		nfd=accept(sfd,(struct sockaddr *)&cin,&len);
		if(nfd<0)
		{
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s:%d] 客户端连接成功 line = %d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),__LINE__);
		//子进程进行写入和输出
		if(fork()==0)
		{
			close(sfd);
			deal_cli_msg(nfd,cin);
			exit(0);
		}
		close(nfd);
	}
	if(close(sfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}

void zombie_handler(int sig)
{
	while(waitpid(-1,NULL,0)>0);
}

int deal_cli_msg(int nfd,struct sockaddr_in cin)
{
	char buf[128]="";
	ssize_t res;
	while(1)
	{
		bzero(buf,sizeof(buf));
		res = recvfrom(nfd,buf,sizeof(buf),0,NULL,NULL);
		if(res<0)
		{
			ERR_MSG("recv");
			return -1;
		}
		else if(0==res)
		{
			printf("newfd=%d 客户端下线\n",nfd);
			break;
		}
		printf("[%s:%d] :%s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),buf);
	}
	close(nfd);
}