当前位置: 首页 > article >正文

udp聊天室

创建一个简单的udp聊天室

服务器代码思路:

  1. 初始化

    • 创建UDP套接字。
    • 配置服务器的IP和端口号,并绑定套接字到这个地址。
  2. 数据接收和处理

    • 使用循环接收客户端发来的消息。recvfrom()
    • 解析消息类型(如登录、发送、下线)和内容。
  3. 广播消息

    • 对于聊天消息,将其广播给所有连接的客户端。
    • 对于登录和下线消息,通知其他客户端有新用户上线或某用户下线。
  4. 客户端管理

    • 维护一个客户端列表或字典,用于跟踪在线客户端的地址和状态。
    • 更新客户端列表以反映登录和下线事件。
  5. 清理和关闭

    • 在服务器关闭时,适当清理资源并关闭套接字。

服务器代码:

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

typedef struct linklist
{ //创建存储客户端ip、端口号的链表
    struct sockaddr_in addr;
    struct linklist *next;
} l_node, *l_pnode;

typedef struct sockt_data
{                   //创建收发的信息结构体
    char name[64];  //名字
    char type[8];   //消息类别
    char text[128]; //信息内容
} data_t;

typedef struct Sock
{ //创建存储服务器发送数据的信息结构体
    struct sockaddr_in caddr;
    data_t cdata; //发送消息结构体
    int sockfd;   //套接字
} sock_t;

l_pnode create();                                              //创建链表节点
void login(int sockfd, data_t data, struct sockaddr_in saddr); //登录
void line(int sockfd, data_t data, struct sockaddr_in saddr);  //发送和下线
void *func(void *arg);                                         //子线程

l_pnode S; //开辟一个全局变量的链表

int main(int argc, char *argv[])
{
//  1、创建套接字 -- socket 
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    // 2、绑定IP地址和端口号 -- bind
    struct sockaddr_in saddr, caddr;

    caddr.sin_family = AF_INET;
    caddr.sin_port = htons(8888); //将主机字节序转换为网络字节序再赋值
    caddr.sin_addr.s_addr = htons(INADDR_ANY);

    int s_len = sizeof(saddr);
    bind(sockfd, (struct sockaddr *)&caddr, s_len);
    // 3、发送接送数据 
    data_t sdata = {0, 0, 0}; //定义接送数据结构体
    int d_len = sizeof(sdata);
    int c_len = sizeof(caddr); //计算服务器地址结构的大小
    memset(&saddr, 0, c_len);  //清空接收地址结构

    S = (l_pnode)malloc(sizeof(l_node)); //初始化链表
    S->next = NULL;

    sock_t csock;          //创建子线程所需结构体
    csock.sockfd = sockfd; //传递套接字

    pthread_t thread;
    pthread_create(&thread, NULL, func, &csock); //创建子线程

    while (1)
    {
        memset(&sdata, 0, d_len);                                              //清空接送到的数据
        recvfrom(sockfd, &sdata, d_len, 0, (struct sockaddr *)&saddr, &c_len); //接送来自客户端的消息
        printf("%s:%s\n", sdata.name, sdata.text);                             //在服务器中显示接送到的消息
        if (strcmp(sdata.type, "login") == 0)                                  //判断接受消息内型
            login(sockfd, sdata, saddr);                                       //上线操作
        else
            line(sockfd, sdata, saddr); //聊天、下线操作
    }
//   4、关闭套接字 
    close(sockfd);
    return 0;
}
//  子线程 
void *func(void *arg) //线程处理函数
{
    sock_t csock = *(sock_t *)arg;     //获取主函数中传递过来的内容
    strcpy(csock.cdata.type, "login"); //设置要发送的结构图
    strcpy(csock.cdata.name, "服务器");
    while (1)
    {
        memset(&csock.cdata.text, 0, 128);                     //清空输入进来的内容
        fgets(csock.cdata.text, 64, stdin);                    //获取输入进来的内容
        csock.cdata.text[strlen(csock.cdata.text) - 1] = '\0'; //删除获取进来的回车'\n'
        line(csock.sockfd, csock.cdata, csock.caddr);          //向所有在线的客户端发送入进来的内容
    }
}
//  上线 
void login(int sockfd, data_t data, struct sockaddr_in saddr) //登录
{
    int d_len = sizeof(data);
    int s_len = sizeof(saddr);
    l_pnode new = create(); //创建新节点
    new->addr = saddr;      //传递接送到的客户端IP、端口号
    new->next = NULL;
    l_pnode N = create();
    N = S;
    while (N->next) //尾插并发送给其他客户端
    {
        N = N->next;
        sendto(sockfd, &data, d_len, 0, (struct sockaddr *)&N->addr, s_len); //发送给除该客户信息的其他客户
    }
    N->next = new;
}
//  聊天、下线 
void line(int sockfd, data_t data, struct sockaddr_in saddr) //聊天、下线
{
    int d_len = sizeof(data);
    int s_len = sizeof(saddr);
    l_pnode N = create();
    N = S;
    while (N->next)
    {
        if ((N->next->addr.sin_addr.s_addr == saddr.sin_addr.s_addr) && (N->next->addr.sin_port == saddr.sin_port)) //判断是否存在该客户端数据
        {
            if (strcmp(data.type, "line") == 0) //下线,删除用户信息
            {
                l_pnode Q = N->next;
                N->next = Q->next; //连线
                free(Q);           //释放空间
                Q = NULL;
            }
            if (strcmp(data.type, "send_t") == 0) //聊天,跳过用户
                N = N->next;
        }
        if (N->next != NULL) //判断是否到尾节点
        {
            N = N->next;
            sendto(sockfd, &data, d_len, 0, (struct sockaddr *)&N->addr, s_len); //发送给除该客户信息的其他客户
        }
        else
            break;
    }
}
//  创建链表新节点
l_pnode create()
{
    l_pnode S = (l_pnode)malloc(sizeof(l_node));
    S->next = NULL;
    return S;
}

客户端代码思路:

  1. 初始化和创建套接字:在函数中,创建一个UDP套接字,并配置客户端的IP地址和端口。main

  2. 进程创建:使用创建一个子进程。子进程负责读取用户输入的用户名和消息,并处理登录、发送和下线操作。fork()

  3. 父进程接收消息:父进程不断接收来自服务器的消息,并在控制台显示。

  4. 登录、发送和下线功能

    • login():向服务器发送登录消息。
    • send_t():发送聊天消息。
    • line():发送下线消息并退出子进程。
  5. 信号处理:使用处理子进程结束的信号,确保子进程资源得到清理。signal(SIGCHLD, mysignal)

客户端代码:


#include <stdio.h>
#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/wait.h>
typedef struct sockt_data
{
    char name[64];//用户户名
    char type[8];//消息类别
    char text[128];//消息内容
} data_t;

void mysignal(int arg) //回收子进程
{
    wait(NULL);
    exit(0);
}

void login(int sockfd, data_t cdata, struct sockaddr_in caddr); //登录
void send_t(int sockfd, data_t data, struct sockaddr_in saddr); //发送信息
void line(int sockfd, data_t cdata, struct sockaddr_in caddr);  //下线

int main(int argc, char *argv[])
{
    if (argc != 2)
        printf("请正确输入: ./client IP号\n");
    //  1、创建套接字
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    // 2、绑定IP地址和端口号 -- bind
    struct sockaddr_in saddr, caddr;
    caddr.sin_family = AF_INET;
    caddr.sin_port = htons(8888); //将主机字节序转换为网络字节序再赋值
    caddr.sin_addr.s_addr = inet_addr(argv[1]);

    int s_len = sizeof(saddr);
    //  3、发送接送数据
    data_t cdata = {0, 0, 0}, sdata = {0, 0, 0}; //定义收发数据结构体

    int d_len = sizeof(cdata); //计算数据结构体长度

    memset(&saddr, 0, s_len);  //清空接收地址结构
    memset(&cdata, 0, d_len);  //清空接收地址结构
    signal(SIGCHLD, mysignal); //捕获子进程释放信号

    pid_t pid = fork(); //创建线程
    if (0 == pid)       //子线程
    {
        printf("请输入你的用户名--->");
        fgets(cdata.name, 64, stdin);              //获取输入进来的名字
        cdata.name[strlen(cdata.name) - 1] = '\0'; //删除获取进来的回车'\n'
        login(sockfd, cdata, caddr);               //登录

        while (1)
        {
            fgets(cdata.text, 64, stdin);              //获取输入进来的内容
            cdata.text[strlen(cdata.text) - 1] = '\0'; //删除获取进来的回车'\n'
            if (strcmp(cdata.text, "line!!!") == 0)    //判断是否结束聊天
                line(sockfd, cdata, caddr);            //下线
            send_t(sockfd, cdata, caddr);              //聊天
        }
    }

    while (1)
    {
        memset(&sdata, 0, d_len); //清空接送到的数据
        recvfrom(sockfd, &sdata, d_len, 0, (struct sockaddr *)&saddr, &s_len);
        if (strcmp(sdata.type, "send_t") == 0) //在客户端中显示接送到的消息
            printf("%s:%s\n", sdata.name, sdata.text);
        else
            printf("%s-->%s\n", sdata.name, sdata.text);
    }
    //  4、关闭套接字
    close(sockfd);
    return 0;
}
// 登录
void login(int sockfd, data_t cdata, struct sockaddr_in caddr) //登录
{
    int d_len = sizeof(cdata);
    int s_len = sizeof(caddr);
    strcpy(cdata.type, "login");
    strcpy(cdata.text, "上线了");
    sendto(sockfd, &cdata, d_len, 0, (struct sockaddr *)&caddr, s_len);
}
//    下线
void line(int sockfd, data_t cdata, struct sockaddr_in caddr) //下线
{
    int d_len = sizeof(cdata);
    int s_len = sizeof(caddr);
    strcpy(cdata.type, "line");
    strcpy(cdata.text, "下线了");
    sendto(sockfd, &cdata, d_len, 0, (struct sockaddr *)&caddr, s_len);
    exit(0);
}
// 聊天
void send_t(int sockfd, data_t cdata, struct sockaddr_in caddr) //发送信息
{
    int ret = 0;
    int d_len = sizeof(cdata);
    int s_len = sizeof(caddr);
    strcpy(cdata.type, "send_t");
    ret = sendto(sockfd, &cdata, d_len, 0, (struct sockaddr *)&caddr, s_len);
}


http://www.kler.cn/a/302438.html

相关文章:

  • Unity3D学习FPS游戏(12)敌人检测和攻击玩家
  • MyBatisPlus 用法详解
  • 2019年下半年试题二:论软件系统架构评估及其应用
  • 深度学习——权重初始化、评估指标、梯度消失和梯度爆炸
  • AI绘画经验(stable-diffusion)
  • 【Qt-ROS开发】使用 Qt Creator 构建和编译含 ROS 库的 Qt 项目
  • 【50 Pandas+Pyecharts | 暑期档电影票房数据分析可视化】
  • 深入探讨 Flannel 技术:性能、缺陷与优点
  • sql格式化工具
  • MongoDB 原子操作
  • 【计算机网络】UDP 协议详解及其网络编程应用
  • vue3中实现拖拽排序(vue-draggable-next的使用)
  • SVM 监督学习
  • Linux CentOS更换阿里云源解决Could not retrieve mirrorlist http://mirrorlist.centos.org
  • 电脑监控软件系统有哪些?|2024年干货整理,赶紧码住!
  • css父容器溢出隐藏 子容器溢出滚动
  • 使用TLS解决Docker API暴露2375端口的问题
  • C++-----STL简介(了解)
  • 【数据结构】第八节:链式二叉树
  • 为什么学习Python数据分析?学会能干什么?python数据分析有什么用
  • 研1日记10
  • 保证缓存一致性
  • Debian11.9镜像基于jre1.8的Dockerfile
  • ppt图标去哪找?自带素材库的在线PPT软件,免费获取!
  • Caffenie配合Redis做两级缓存,Redis发布订阅实现缓存一致更新
  • 关于支付宝小程序客户端退出不了的问题的认知和退出的解决方案