网络编程作业 服务器

搭建一个服务器实现收发

代码

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

typedef void (*sighandler_t)(int);


void handler1(int sig)
{
	pid_t pid=waitpid(-1,NULL,0);
	printf("子进程退出，父进程结束\n");	
	exit(0);
}

#define PORT 8888                      //本机端口号
#define IP "192.168.31.18"           //本机IP号

//打印错误新的宏函数
#define ERR_MSG(msg) do{\
	fprintf(stderr,"line: %d ",__LINE__);\
	perror(msg);\
}while(0);

int main(int argc, const char *argv[])
{


	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("create socket success\n");

	//填充地址信息结构体，真实的地址信息结构体与协议族相关
	//AF_INET
	struct sockaddr_in sin;
	sin.sin_family      =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 successs\n");

	//将套接字设置为被动监听状态，让内核去监听是否有客户端连接;
	if(listen(sfd,10)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

	struct sockaddr_in cin;
	socklen_t addrlen=sizeof(cin);
	//从已完成连接的队列头中，取出一个客户端的信息，创建生成一个新的套接字文件描述符，
	//该文件描述符才是与客户端通信的文件描述符!!!	
	int newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
	if(newfd<0)
	{
		ERR_MSG("accept");
		return -1;
	}

	//网络字节序的IP-->点分十进制  网络字节序的port -->本机字节序
	printf("[%s :%d] newfd=%d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

	//创建父子进程
	pid_t pid=fork();
	if(pid>0)
	{   
		sighandler_t s=signal(SIGCHLD,handler1);
		if(s==SIG_ERR)
		{
			ERR_MSG("signal");
			return -1;
		}
		char buf[1218]="";
		ssize_t res=0;
		while(1)
		{
			//循环发送
			bzero(buf,sizeof(buf));
			printf("请输入-->>");
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]=0;

			res=send(newfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("[%s :%d] newfd=%d: 消息 %s已发送\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf);
		}
		close(sfd); 
		close(newfd);

	}
	if(pid==0)
	{
		char buf[1218]="";
		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("\r[%s :%d] newfd=%d: 客户端退出 \n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
				break;
			}
			printf("\r[%s :%d] newfd=%d: %s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf);
			printf("请输入-->>");
			fflush(stdout);
	
		}
		close(sfd); 
		close(newfd);

		exit(0);

	}
	return 0;
}