2024/4/21 UDP网络聊天室day7

#include "head.h"
//服务器端
void server_char(int sockfd,list_t *p,MSG_t msg,struct sockaddr_in caddr);
void server_login(int sockfd,list_t *p,MSG_t msg,struct sockaddr_in caddr);
void server_quit(int sockfd,list_t *p,MSG_t msg,struct sockaddr_in caddr);
list_t *Createlist();
int main(int argc, const char *argv[])
{
	if(argc!=3){
		printf("请输入IP地址 端口号\n");
		return -1;
	}
	int sockfd=socket(AF_INET,SOCK_DGRAM,0);
	if(sockfd<0)
	{
		perror("socket error");
		return -1;
	}
	struct sockaddr_in saddr,caddr;
	saddr.sin_family=AF_INET;
	saddr.sin_port=htons(atoi(argv[2]));
	saddr.sin_addr.s_addr=inet_addr(argv[1]);
	int len=sizeof(caddr);
	if(bind(sockfd,(struct sockaddr*)&saddr,sizeof(saddr))==-1){
		perror("bind error");
		return -1;
	}
	MSG_t msg;
	list_t *p=Createlist();
	while(1){//接收客户端发来的结构体，判断类型，执行相应函数
		recvfrom(sockfd,&msg,sizeof(msg),0,(struct sockaddr*)&caddr,&len);
		switch(msg.type)
		{
			case login:
				server_login(sockfd,p,msg,caddr);
				break;
			case chat:
				server_char(sockfd,p,msg,caddr);
				break;
			case quit:
				server_quit(sockfd,p,msg,caddr);
				break;
		}
	}
	close(sockfd);
	return 0;
}

void server_char(int sockfd,list_t *p,MSG_t msg,struct sockaddr_in caddr)
{
	while(p->next!=NULL)
	{
		p=p->next;
		//当前用户结点跳过不发送信息
		if(memcmp(&(p->addr),&caddr,sizeof(caddr))==0){
			continue;
		}
		//其他用户发送信息
		sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr*)&(p->addr),sizeof(p->addr));
	}
}
void server_login(int sockfd,list_t *p,MSG_t msg,struct sockaddr_in caddr)
{
	sprintf(msg.text,"%s加入chat\n",msg.name);
	printf("%s\n",msg.text);
	while(p->next!=NULL)
	{//向所有链表内用户发送上线信息
		p=p->next;
		sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr*)&(p->addr),sizeof(p->addr));
	}
	//向链表添加新用户
	list_t *pnew=(list_t*)malloc(sizeof(list_t));
	pnew->next=NULL;
	p->next=pnew;
	pnew->addr=caddr;

}
void server_quit(int sockfd,list_t *p,MSG_t msg,struct sockaddr_in caddr)
{
	list_t *pdel=NULL;
	//将离线信息写入text
	sprintf(msg.text,"%s已下线\n",msg.name);
	while(p->next!=NULL)
	{
		//找到当前用户的位置删除结点
		if(memcmp(&(p->next->addr),&caddr,sizeof(caddr))==0){
			pdel=p->next;
			p->next=pdel->next;
			free(pdel);
			pdel=NULL;
		}else{//非当前用户发送离线信息
			p=p->next;
			sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr*)&(p->addr),sizeof(p->addr));
			}
	}


}

list_t *Createlist()
{
	list_t *p=(list_t*)malloc(sizeof(list_t));
	if(p==NULL)
	{
		perror("malloc error");
		return NULL;
	}
	p->next=NULL;
	return p;
}

#include "head.h"
//客户端
int main(int argc, const char *argv[])
{
	if(argc!=3){
		printf("请输入IP地址 端口号\n");
		return -1;
	}
	int sockfd=socket(AF_INET,SOCK_DGRAM,0);
	if(sockfd==-1){
		perror("socket error");
		return -1;
	}
	struct sockaddr_in caddr;
	caddr.sin_family=AF_INET;
	caddr.sin_port=htons(atoi(argv[2]));
	caddr.sin_addr.s_addr=inet_addr(argv[1]);

	MSG_t msg;
	printf("请输入用户名:");
	fgets(msg.name,sizeof(msg.name),stdin);
	if(msg.name[strlen(msg.name)-1]=='\n'){
		msg.name[strlen(msg.name)-1]='\0';
	}
	msg.type=login;//发送用户名和登录类型的结构体到服务器
	sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr*)&caddr,sizeof(caddr));

	pid_t pid=fork();
	if(pid<0){
		perror("fork error");
		return -1;
	}else if(pid==0){//子进程接收服务器返回的结构体消息
		while(1){
			recvfrom(sockfd,&msg,sizeof(msg),0,NULL,NULL);
			printf("%s:%s\n",msg.name,msg.text);
		}
	}else{//父进程向服务器发送结构体类型消息
		while(1){
			//向结构text写内容
			fgets(msg.text,sizeof(msg.text),stdin);
			
		 	if(msg.text[strlen(msg.text)-1]=='\n'){
				msg.text[strlen(msg.text)-1]='\0';
				}//在输入内容后添加\0
			
			//输入quit跳出循环下线
		 	if(strncmp(msg.text,"quit",4)==0){
				msg.type=quit;
				sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr*)&caddr,sizeof(caddr));
				break;
			}else{//非quit正常聊天
				msg.type=chat;
				
				sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr*)&caddr,sizeof(caddr));
				}
		}
		kill(pid,SIGKILL);//循环结束说明用户下线杀死子进程
		wait(NULL);
	}
	close(sockfd);
	return 0;
}

#include <myhead.h>
//重定向list_t用户地址端口链表
typedef struct node
{
	struct sockaddr_in addr;
	struct node *next;
}list_t;
//重定向MSG_t信息结构体
typedef struct msg_t
{
	int type;
	char name[32];
	char text[128];
}MSG_t;
//定义枚举变量
enum type_t
{
	login,
	chat,
	quit,
};