Linux作业

2023-5-18作业

创建多线程服务器

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

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

int deal_cli_msg(int newfd,struct sockaddr_in cin);

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

int main(int argc,const char*argv[])
{
	//获取信号
	if (signal(SIGCHLD,zombie_handler)==SIG_ERR)
	{
		ERR_MSG("signal");
		return -1;
	}

	//创建套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d __%d__\n",sfd,__LINE__);

	//端口重用
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		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))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n",__LINE__);

	//监听
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success __%d__\n",__LINE__);
	
	//生成描述符
	struct sockaddr_in cin;
	socklen_t len=sizeof(cin);
	int newfd=-1;
	pid_t cpid=0;
	while(1)
	{
		//链接客户端
		newfd=accept(sfd,(struct sockaddr*)&cin,&len);
		if(newfd<0)
		{
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s:%d] cin link success! newfd=%d __%d__\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);

		//与客户端交互
		if(fork()==0)
		{
			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(0==res)
		{
			printf("[%s:%d]cin exit! newfd=%d __%d__\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);
			break;
		}
		printf("[%s:%d]newfd=%d:%s,__%d__\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf,__LINE__);

		//回发数据
		strcat(buf,"*_*");
		if(send(newfd,buf,sizeof(buf),0)<0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("Transmit success!\n");
	}
	if(close(newfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
}