网络编程day4

多进程并发服务器

#include<myhead.h>

#define ERR_MSG(msg) do{printf("__%d__:\n",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 8888    //端口号
#define IP "192.168.114.44" //本机IP
void handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0);
}
int main(int argc, const char *argv[])
{
	if(signal(17,handler)==SIG_ERR)
	{
		perror("signal error");
		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
	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
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
	//将套接字设置为被动监听状态
	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 fd;
	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);
		fd=fork();
		if(fd==0)
		{
			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(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);
				//发送
				if(send(newfd,buf,sizeof(buf),0)<0)
				{
					ERR_MSG("send");
					return -1;
				}
			}
			close(newfd);
			exit(0);
		}
		close(newfd);
	}
	//关闭文件描述符
	if(close(sfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}

多线程并发服务器

#include<myhead.h>
#define dk 8888
#define IP "192.168.114.44"
struct climsg
{
	int newfd;
	struct sockaddr_in cin;
};
//线程体
void* deal_cli_msg(void *arg)
{
	
	int newfd=((struct climsg *)arg)->newfd;
	struct sockaddr_in cin=((struct climsg *)arg)->cin;
	char buf[256];
	int ret;
	while(1)
	{
		bzero(buf,sizeof(buf));
		ret=recv(newfd,buf,sizeof(buf),0);
		if(ret<0)
		{
			perror("recv error");
			return NULL;
		}
		else if(ret==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,ret,0)<0)
		{
			perror("send error");
			return NULL;
		}
		printf("send success\n");
	}
	close(newfd);
	pthread_exit(NULL);
}
int main(int argc, const char *argv[])
{
	//创建套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		perror("socket perror");
		return -1;
	}
	printf("socket success\n");
	//填充真实的地址信息
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(dk);
	sin.sin_addr.s_addr=inet_addr(IP);
	//地址信息绑定
	if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin))<0)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");
	//装换成监听状态
	if(listen(sfd,128)<0)
	{
		perror("listen error");
		return -1;
	}
	printf("listen success\n");
	//存储发送方的地址信息
	int newfd;
	struct sockaddr_in cin;
	struct climsg info;
	socklen_t len=sizeof(cin);
	pthread_t tid;
	while(1)
	{
		//获取发送方的地址信息,并生成一个新的文件描述符
		newfd=accept(sfd,(struct sockaddr *)&cin,&len);
		if(newfd<0)
		{
			perror("accept error");
			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,&info))
		{
			printf("pthread_create error");
			return -1;
		}
		printf("create success\n");
		pthread_detach(tid);
	}
	if(close(sfd)<0)
	{
		perror("close error");
		return -1;
	}
	return 0;
}