【基于UDP的网络聊天室】

sept_94

已于 2022-09-25 14:58:30 修改

阅读量2k

点赞数 6

分类专栏：牛刀小试文章标签： c语言学习网络 udp

于 2022-09-24 16:14:47 首次发布

本文链接：https://blog.csdn.net/sept_94/article/details/127026681

版权

牛刀小试专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文介绍了一个使用UDP协议实现的网络聊天室项目，具备用户登录通知、群聊消息广播、用户退出通知及服务器发送系统消息等功能。通过链表存储在线用户网络信息，消息结构体包含了操作码、用户名和消息内容。代码分别展示了服务器端和客户端的实现，实现了多进程通信，确保了聊天室的实时性和稳定性。

摘要由CSDN通过智能技术生成

总结下近期写的小项目，在学习中同时积累解决问题的经验，以及真正的项目中解决问题的思路，如有不合理地方，请多指教！

一、项目名称

基于UDP的网络聊天室

二、功能

1.当有新用户登录时，其他在线用户可以收到该用户登录的消息。

2.当用户发送聊天信息时，其他所有在线用户均能收到群聊信息。

3.当有用户退出时，其他所有在线用户均能收到退出消息。

4.服务器可以发送系统消息。

三、初步构思

1.服务器需要给多个用户发送信息，需要创建链表保存各用户的网络信息。

2.消息类型有四种，上线、群聊、退出、系统消息，需要定义消息结构体，结构体中需要包含信息操作码、用户名、消息数据。

四、流程图

客户端

服务器端：

五、代码实现

服务器端：

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

//宏定义高频率使用的判断出错
#define PRINERR(msg)                                        \
    do {                                                    \
        printf("%s:%s:%d\n", __FILE__, __func__, __LINE__); \
        perror(msg);                                        \
        exit(-1);                                           \
    } while (0)

typedef struct _MSG {
    char code; //操作码：'L'表示登录 'C'表示群聊 'Q'表示退出 'S'表示系统消息
               //定义操作码为1个字符，免去考虑网络字节序问题
    char name[45];
    char txt[256];
} msg_t; //定义消息结构体类型

typedef struct _ADDR {
    struct sockaddr_in clientaddr;
    struct _ADDR* rear;
} addrlist_t; //定义保存客户端网络信息结构体的链表类型

//登录操作的函数
void do_login(int sockfd,msg_t msg,addrlist_t*addr,struct sockaddr_in clientaddr){
    //先遍历链表 将新用户加入群聊的消息发给所有人
    addrlist_t* tmp = addr;//记录链表头节点
    while(tmp->rear != NULL){
        tmp = tmp->rear;
        if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(tmp->clientaddr), sizeof(tmp->clientaddr))){
            PRINERR("sendto error");
        }
    }
    //将新用户的网络信息结构体 头插入链表
    addrlist_t* pnew = NULL;
    if (NULL == (pnew = (addrlist_t*)malloc(sizeof(addrlist_t)))) {
            printf("malloc error\n");
            return;
        }
    pnew->clientaddr = clientaddr;
    pnew->rear = addr->rear;
    addr->rear = pnew;
    return;
}

//群聊操作的函数
void do_chat(int sockfd,msg_t msg,addrlist_t*addr,struct sockaddr_in clientaddr){
    //遍历链表，将群聊消息发给除了自己之外的所有人
    addrlist_t* ptmp = addr;
    while(ptmp->rear != NULL){
        ptmp = ptmp->rear;
        if(memcmp(&clientaddr, &(ptmp->clientaddr), sizeof(clientaddr))){
            //说明不是自己 就发送数据
            if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(ptmp->clientaddr), sizeof(ptmp->clientaddr))){
                PRINERR("sendto error");
            }
        }
    }
    return;
}

//退出操作的函数
void do_quit(int sockfd,msg_t msg,addrlist_t*addr,struct sockaddr_in clientaddr){
    //遍历链表 是自己就将自己在链表中删除
    //不是自己 就发送 退出群聊的数据
    addrlist_t* ptmp = addr;
    addrlist_t* del = NULL;
    while(ptmp->rear != NULL){
        if(memcmp(&(ptmp->rear->clientaddr), &clientaddr, sizeof(clientaddr))){
            //不是自己
            ptmp = ptmp->rear;
            if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(ptmp->clientaddr), sizeof(ptmp->clientaddr))){
                PRINERR("sendto error");
            }
        }else{
            //是自己
            del = ptmp->rear;
            ptmp->rear = del->rear;
            free(del);
            del = NULL;
        }
    }
    return;
}

int main(int argc, const char* argv[])
{
    if (3 != argc) { //考虑用命令行传参方式输入ip地址及端口号，先进行参数判断
        printf("input error!\n");
        printf("usage: %s <IP> <port>\n",argv[0]);
        return -1;
    }

    int sockfd;
    if (-1 == (sockfd = socket(AF_INET, SOCK_DGRAM, 0))) {
        PRINERR("socket error");
    }
    //填充服务器网络信息结构体
    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 serveraddr_len = sizeof(serveraddr);
    //将套接字绑定网络信息
    if (-1 == bind(sockfd, (struct sockaddr*)&serveraddr, serveraddr_len)) {
        PRINERR("bind error");
    }

    //定义客户端网络信息结构体
    struct sockaddr_in clientaddr;
    socklen_t clientaddr_len = sizeof(clientaddr);
    msg_t msg; //定义接收信息的变量msg
    pid_t pid; //进程号
    if (-1 == (pid = fork())) { //创建多进程
        PRINERR("fock error");
    }
    if (0 == pid) { //子进程，用来收发数据
        //创建保存客户端信息的链表头节点
        addrlist_t* addr;
        if (NULL == (addr = (addrlist_t*)malloc(sizeof(addrlist_t)))) {
            printf("malloc error\n");
            return -1;
        }
        memset(addr, 0, sizeof(addr));
        addr->rear = NULL;

        while (1) { //循环收发数据
            //每次接收新数据及新用户信息前清空，确保信息准确
            memset(&msg, 0, sizeof(msg));
            memset(&clientaddr, 0, sizeof(clientaddr));
            //接收客户端发送的消息,存放在msg中
            if (-1 == recvfrom(sockfd, &msg, sizeof(msg), 0, (struct sockaddr*)&clientaddr, &clientaddr_len)) {
                PRINERR("recvfrom error");
            }
            switch (msg.code) { //判断消息中的操作码，根据操作码执行对应操作
                case 'L': //登录操作
                    do_login(sockfd,msg,addr,clientaddr);
                    break;
                case 'C': //群聊操作
                    do_chat(sockfd,msg,addr,clientaddr);
                    break;
                case 'Q': //退出操作
                    do_quit(sockfd,msg,addr,clientaddr);
                    break;
            }
        }

    } else {
        //父进程，用来发送系统消息
        //将父进程当做一个客户端，向子进程发送群聊消息
        //msg在fock 前定义的，所以父进程中同样有msg
        strcpy(msg.name, "系统消息");
        msg.code = 'C';
         while(1){
            memset(msg.txt,0,sizeof(msg.txt));
            fgets(msg.txt, 256, stdin);//循环在终端接收消息
            msg.txt[strlen(msg.txt)-1] = '\0';
            if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, serveraddr_len)){
                PRINERR("sendto error");
            }
        }
    }
    close(sockfd);

    return 0;
}

客户端：

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

//宏定义高频率使用的判断出错
#define PRINERR(msg)                                        \
    do {                                                    \
        printf("%s:%s:%d\n", __FILE__, __func__, __LINE__); \
        perror(msg);                                        \
        exit(-1);                                           \
    } while (0)

typedef struct _MSG {
    char code; //操作码：'L'表示登录 'C'表示群聊 'Q'表示退出 'S'表示系统消息
               //定义操作码为1个字符，免去考虑网络字节序问题
    char name[45];
    char txt[256];
} msg_t; //定义消息结构体类型

int main(int argc, const char *argv[])
{
    if (3 != argc) { //考虑用命令行传参方式输入ip地址及端口号，先进行参数判断
        printf("input error!\n");
        printf("usage: %s <IP> <port>\n",argv[0]);
        return -1;
    }

    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(-1 == sockfd){
        PRINERR("socket error");
    }

    struct sockaddr_in serveraddr;
    memset(&serveraddr, 0, sizeof(serveraddr));
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
    socklen_t serveraddr_len = sizeof(serveraddr);

    msg_t msg;
    memset(&msg, 0, sizeof(msg));
    //输入用户名
    printf("请输入用户名:>>");
    fgets(msg.name, 45, stdin);
    msg.name[strlen(msg.name)-1] = '\0';
    msg.code = 'L';
    strcpy(msg.txt, "加入群聊");
    //给服务器发送登录的信息
    if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, serveraddr_len)){
        PRINERR("sendto error");
    }

    pid_t pid = 0;
    if(-1 == (pid = fork())){
        PRINERR("fork error");
    }
    if(0 == pid){//子进程，循环接收并打印收到的数据
        while(1){
            if(-1 == recvfrom(sockfd, &msg, sizeof(msg), 0, NULL, NULL)){
                PRINERR("recvfrom error");
            }
            //打印收到的数据
            printf("[%s]: %s\n", msg.name, msg.txt);
        }
    }else if(0 < pid){//父进程,循环在终端接收数据并发送给客户端
        while(1){
            memset(msg.txt,0,sizeof(msg.txt));
            fgets(msg.txt, 128, stdin);//在终端获取聊天信息
            msg.txt[strlen(msg.txt)-1] = '\0';
            if(!strcmp(msg.txt, "quit")){
                msg.code = 'Q';
                strcpy(msg.txt, "退出群聊");
            }else{
                msg.code = 'C';
            }
            if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, serveraddr_len)){
                PRINERR("sendto error");
            }
            if(!strcmp(msg.txt, "退出群聊")){
                break;
            }
        }
        //杀死子进程
        kill(pid, SIGKILL);
        wait(NULL);//等待回收子进程资源
    }
    close(sockfd);

    return 0;
}

六、最终效果