udt聊天室

创建一个简单的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);
}