UDP聊天室 - 技术栈

服务器

cpp 复制代码

#include<myhead.h>
 
struct sockaddr_in serveraddr,caddr;
enum type_t//枚举
{
	Login,
	Chat,
	Quit,
};
typedef struct MSG
{
	char type;//L C Q
	char name[32];//
	char text[128];//
}msg_t;
 
typedef struct NODE//链表
{
	struct sockaddr_in caddr;
	struct NODE *next;
}node_t;
 
node_t *create_node(void)//建头节点
{
	node_t *p=(node_t *)malloc(sizeof(node_t));
	if(p==NULL)
	{
		perror("malloc err");
		return NULL;
	}
	p->next=NULL;
	return p;
 
}
void do_login(int ,msg_t ,node_t *,struct sockaddr_in);//登录的函数
void do_chat(int ,msg_t ,node_t *,struct sockaddr_in);//群聊的函数
void do_quit(int ,msg_t ,node_t *,struct sockaddr_in);//退出函数
int main(int argc, char const *argv[])
{
	if(argc !=3)
	{
		printf("Usage:./a.out <port>\n");
		return -1;
	}
	//创建UDP套接字
	int sockfd = socket(AF_INET,SOCK_DGRAM,0);
	if(sockfd<0)
	{
		perror("socket err");
		exit(-1);
	}
	//填充服务器网络信息结构体
	serveraddr.sin_family=AF_INET;
	serveraddr.sin_port=htons(atoi(argv[2]));
 
	serveraddr.sin_addr.s_addr=inet_addr(argv[1]);
	socklen_t len = sizeof(caddr);
	//定义保存客户端网络信息的结构体
	//绑定套接字和服务器网络信息的结构体
	bind(sockfd,(struct sockaddr *)&serveraddr,sizeof(serveraddr));
	printf("bind ok!\n");
	msg_t msg;
	node_t *p=create_node();
	while(1)
	{
		if(recvfrom(sockfd,&msg,sizeof(msg),0,(struct sockaddr *)&caddr,&len)<0)
		{
			perror("recvfrom err");
			return -1;
		}
		if(msg.type==Login)
		{
			strcpy(msg.text,"已上线");
			printf("ip:%s pord:%d name:%s\n",inet_ntoa(caddr.sin_addr),ntohs(caddr.sin_port),msg.name);
			printf("状态:%s\n",msg.text);
			do_login(sockfd,msg,p,caddr);
		}
		else if(msg.type==Chat)
		{
			do_chat(sockfd,msg,p,caddr);    
		}
		else if(msg.type==Quit)
		{
			strcpy(msg.text,"已下线");
			printf("ip:%s pord:%d name:%s\n",inet_ntoa(caddr.sin_addr),ntohs(caddr.sin_port),msg.name);
			printf("状态:%s\n",msg.text);
			do_quit(sockfd,msg,p,caddr);        
		}
	}
	close(sockfd);
	return 0;
}
//登录的函数
//功能：
//1》将新登录的用户转发给所有已经登录的用户（遍历链表发送谁登录的消息）
//2》创建新节点来保存新登录用户的信息，链接到链表尾就可已
void do_login(int sockfd,msg_t msg,node_t *p,struct sockaddr_in caddr)
{
 
	sprintf(msg.text,"%s 已上线",msg.name);
	while(p->next != NULL)
	{
		p= p->next;
		sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr *)&(p->caddr),sizeof(p->caddr));
		//printf("%s\n",msg.text);
	}
	node_t *new=(node_t *)malloc(sizeof(node_t));
	//初始化
	new->caddr=caddr;
	new->next=NULL;
	//链接到链表尾
	p->next=new;
	return;
}
//群聊的函数
//功能：将客户端发来的聊天内容转发给所有已登录的用户，除了发送聊天内容的用户已外
void do_chat(int sockfd,msg_t msg,node_t *p,struct sockaddr_in caddr)
{
	//遍历链表
	while(p->next != NULL)
	{
		p=p->next;
 
		if(memcmp(&(p->caddr),&caddr,sizeof(caddr)) != 0)
		{
			sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr *)&(p->caddr),sizeof(p->caddr));
		}
	}
	return;
}
//退出函数
//功能：
//1》将谁退出的消息转发给i所有用户
//2》将链表中保存这个推出的用户信息的节点删除
void do_quit(int sockfd,msg_t msg,node_t *p,struct sockaddr_in caddr)
{
	sprintf(msg.text,"%s 已下线",msg.name);
	while(p->next != NULL)
	{
		if((memcmp(&(p->next->caddr),&caddr,sizeof(caddr)))==0)
		{ 
			node_t *dele=NULL;
			dele = p->next;
			p->next=dele->next;
			free(dele);
			dele=NULL;
		}
		else
		{
			p=p->next;
			sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr *)&(p->caddr),sizeof(p->caddr));
		}  
	}
	return;
}

客户端

cpp 复制代码

#include <myhead.h>
 
enum type_t
{
    Login,
    Chat,
    Quit,
};
typedef struct 
{
    char type;//L C Q
    char name[32];//
    char text[128];//
}msg_t;
 
int main(int argc, char const *argv[])
{
     if(argc !=3)
    {
        printf("Usage ./a.out <ip> <port>\n");
        return -1;
    }
   
    int sockfd = socket(AF_INET,SOCK_DGRAM,0);
    if(sockfd<0)
    {
        perror("socket err");
        exit(-1);
    }
    struct sockaddr_in serveraddr;
    serveraddr.sin_family=AF_INET;
    serveraddr.sin_port=htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr=inet_addr(argv[1]);
    socklen_t len = sizeof(serveraddr);
    msg_t msg;
    //先执行登录操作 
    printf("请登录:\n");
    msg.type=Login;
    printf("请输入用户名:");
    fgets(msg.name,32,stdin);
    if(msg.name[strlen(msg.name)-1]=='\n')
       msg.name[strlen(msg.name)-1]='\0';
    //发送登录消息
    if(sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr *)&serveraddr,len)<0)
       {
         perror("sendto err");
         exit(-1);
       }
    pid_t pid=fork();
     if(pid<0)
    {
        perror("fork err");
        exit(-1);
    }
    else if(pid==0)
    {
        while(1)
        {
            if(recvfrom(sockfd,&msg,sizeof(msg),0,NULL,NULL)<0)
            {
                perror("recvfrom err");
                return -1;
            }
            printf("[%s]:%s\n",msg.name,msg.text);
        } 
    }    
    else 
    {
        while(1)
        {
            fgets(msg.text,sizeof(msg.text),stdin);
            if(msg.text[strlen(msg.text)-1]=='\n')
               msg.text[strlen(msg.text)-1]='\0';
            if(strcmp(msg.text,"quit")==0)
            {
                msg.type=Quit; 
                sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr *)&serveraddr,len);
                kill(pid,SIGKILL);
                wait(NULL);
                exit(-1);
            }else
        {
            msg.type=Chat;
        }
        //发送消息
        sendto(sockfd,&msg,sizeof(msg),0,(struct sockaddr *)&serveraddr,len);
        }
    }
    close(sockfd);
    return 0;
}