网络聊天室

代码大魔王ㅤ2023-08-23 15:03

一、项目要求

利用UDP协议，实现一套聊天室软件。服务器端记录客户端的地址，客户端发送消息后，服务器群发给各个客户端软件。

问题思考

  • 客户端会不会知道其它客户端地址？

UDP客户端不会直接互连，所以不会获知其它客户端地址，所有客户端地址存储在服务器端。

  • 有几种消息类型？
  • 登录 ：服务器存储新的客户端的地址。把某个客户端登录的消息发给其它客户端。
  • 聊天 ：服务器只需要把某个客户端的聊天消息转发给所有其它客户端。
  • 退出 ：服务器删除退出客户端的地址，并把退出消息发送给其它客户端。
  • 服务器如何存储客户端的地址？

数据结构可以选择线性数据结构

链表节点结构体：

struct node{

struct sockaddr_in addr;//data memcmp

struct node *next;

};

消息对应的结构体(同一个协议)

typedef struct msg_t

{

int type;//'L' C Q enum un{login,chat,quit};

char name[32];//用户名

char text[128];//消息正文

}MSG_t;

int memcmp(void *s1,void *s2,int size)

  • 客户端如何同时处理发送和接收？

客户端不仅需要读取服务器消息，而且需要发送消息。读取需要调用recvfrom，发送需要先调用gets，两个都是阻塞函数。所以必须使用多任务来同时处理，可以使用多进程或者多线程来处理。

二、程序流程图

服务器端

​

客户端

​三、代码实现

server.c代码部分：

cpp 复制代码 
#include<stdio.h>
#include <sys/types.h>      
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h> 
#include<unistd.h>
#include<arpa/inet.h>
#include<string.h>
#include<stdlib.h>
#include <sys/wait.h>
#include<dirent.h>
#include<sys/stat.h>
#include<signal.h>
#include <pthread.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;
}

client.c代码部分：

cpp 复制代码 
#include<stdio.h>
#include <sys/types.h>      
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h> 
#include<unistd.h>
#include<arpa/inet.h>
#include<string.h>
#include<stdlib.h>
#include <sys/wait.h>
#include<dirent.h>
#include<sys/stat.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;
}
相关推荐
sunfove7 小时前
光网络的立交桥：光开关 (Optical Switch) 原理与主流技术解析
网络
乌萨奇也要立志学C++8 小时前
【洛谷】递归初阶 三道经典递归算法题（汉诺塔 / 占卜 DIY/FBI 树）详解
数据结构·c++·算法
鱼跃鹰飞8 小时前
Leetcode1891:割绳子
数据结构·算法
Felven9 小时前
A. Helmets in Night Light
c语言
Kevin Wang7279 小时前
欧拉系统服务部署注意事项
网络·windows
min1811234569 小时前
深度伪造内容的检测与溯源技术
大数据·网络·人工智能
汤愈韬10 小时前
NAT策略
网络协议·网络安全·security·huawei
汤愈韬10 小时前
Full Cone Nat
网络·网络协议·网络安全·security·huawei
zbtlink10 小时前
现在还需要带电池的路由器吗？是用来干嘛的？
网络·智能路由器
桌面运维家11 小时前
vDisk配置漂移怎么办？VOI/IDV架构故障快速修复
网络·架构
热门推荐
01GitHub 镜像站点02Labelme从安装到标注：零基础完整指南03Linux下V2Ray安装配置指南04安娜的档案(Anna’s Archive) 镜像网站/国内最新可访问入口（持续更新）05jdk21下载、安装（Windows、Linux、macOS）06Claude Code 2.1.2 升级报错？别折腾了，一行命令搞定07KGG转MP3工具|非KGM文件|解密音频08【踩坑笔记】50系显卡适配的 PyTorch 安装092025-04-03 Latex学习1——本地配置Latex + VScode环境10UV安装并设置国内源