网络编程 8/15 基于UDP多人聊天室

news2024/11/15 5:46:13
//客户端代码
#include <myhead.h>
struct msgType
{
    char type; // 消息类型L:登录,Q:退出,C:聊天
    char usrName[20];
    char msgText[1024];
};
#define SER_PORT 6666          // 服务器端口
#define SER_IP "192.168.2.161" // 服务器IP
int main(int argc, char const *argv[])
{
    int cfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (cfd == -1)
    {
        perror("socket error");
        return -1;
    }
    // 端口号快速重用
    int reuse = 1;
    if (setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
    {
        perror("setsockopt error");
        return -1;
    }
    // 初始化服务器地址
    struct sockaddr_in sin;                  // 对端服务器地址结构体
    sin.sin_family = AF_INET;                // 协议族
    sin.sin_port = htons(SER_PORT);          // 端口号
    sin.sin_addr.s_addr = inet_addr(SER_IP); // IP地址
    socklen_t len = sizeof(sin);             // 地址长度

    struct msgType msg;
    msg.type = 'L';
    printf("请输入用户名:");
    fgets(msg.usrName, 20, stdin);
    msg.usrName[strlen(msg.usrName) - 1] = '\0';
    char temp[20]; // 将用户名保存起来
    strcpy(temp, msg.usrName);
    strcpy(msg.msgText, "");
    sendto(cfd, &msg, sizeof(msg), 0, (struct sockaddr *)&sin, len);

    // 创建子进程用于收消息
    pid_t pid = fork();
    if (pid == -1)
    {
        perror("fork error");
        return -1;
    }

    if (pid == 0)
    {
        // 子进程
        {
            while (1)
            {
                bzero(&msg, sizeof(msg));
                // 接收消息
                recvfrom(cfd, &msg, sizeof(msg), 0, NULL, NULL);
                switch (msg.type)
                {
                case 'L':
                    printf("%s\n", msg.msgText);
                    break;
                case 'Q':
                {
                    if (strcmp(temp, msg.usrName) == 0)
                    {
                        printf("退出成功\n");
                        exit(EXIT_SUCCESS);
                    }
                    printf("%s\n", msg.msgText);
                }
                break;
                case 'C':
                    printf("[%s]:%s\n", msg.usrName, msg.msgText);
                    break;
                }
            }
        }
    }
    else if (pid > 0)
    {
        // 父进程
        msg.type = 'C';
        while (1)
        {
            sleep(1);
            // 发送消息
            fgets(msg.msgText, 1024, stdin);
            msg.msgText[strlen(msg.msgText) - 1] = '\0';
            if (strcmp(msg.msgText, "q") == 0 || strcmp(msg.msgText, "Q") == 0)
            {
                msg.type = 'Q';
                sendto(cfd, &msg, sizeof(msg), 0, (struct sockaddr *)&sin, len);
                sleep(2);
                goto end;
            }

            sendto(cfd, &msg, sizeof(msg), 0, (struct sockaddr *)&sin, len);
        }
    }

end:
    kill(pid, SIGKILL);
    wait(NULL);
    close(cfd);

    return 0;
}
//服务器主代码
#include "head.h"
#define SER_PORT 6666          // 服务器端口
#define SER_IP "192.168.2.161" // 服务器IP
int main(int argc, char const *argv[])
{
    // 创建用于通信的服务器套接字文件描述符
    int sfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sfd == -1)
    {
        perror("socket error");
        return 1;
    }
    // 将端口号快速重启
    int reuse = 1;
    if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
    {
        perror("setsockopt error");
        return 1;
    }
    // 绑定服务器套接字
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_port = htons(SER_PORT);
    sin.sin_addr.s_addr = inet_addr(SER_IP);
    if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) == -1)
    {
        perror("bind error");
        return 1;
    }
    printf("服务器启动成功\n");
    // 创建链表
    NodePtr L = list_create();
    // 接收客户端发送的数据
    struct sockaddr_in cin;
    socklen_t len = sizeof(cin);
    struct msgType msg;
    bzero(&msg, sizeof(msg));
    while (1)
    {
        // 创建子进程
        pid_t pid = fork();
        // 接收客户端发送的数据
        if (pid == 0)
        {
            while (1)
            {
                // 接收信息
                recvfrom(sfd, &msg, sizeof(msg), 0, (struct sockaddr *)&cin, &len);
                switch (msg.type) // 判断信息类型
                {
                case 'L':
                {
                    list_insert(L, msg.usrName, &cin);
                    strcpy(msg.msgText, "欢迎您进入聊天室 开始聊天吧"); // 发送欢迎信息
                    sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr *)&cin, len);

                    strcpy(msg.msgText, "*******欢迎");
                    strcat(msg.msgText, msg.usrName);
                    strcat(msg.msgText, "进入聊天室*******");
                    NodePtr p = L->next;
                    while (p != NULL) // 发送所有人进入聊天室信息(除了刚进来的人 因为已经发了欢迎)
                    {

                        if (p->addr.sin_port == cin.sin_port)
                        {
                            p = p->next;
                            continue;
                        }
                        if (p == NULL)
                        {
                            break;
                        }
                        sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr *)&p->addr, len);
                        p = p->next;
                    }
                }
                break;
                case 'C':
                {
                    printf("[%s]:%s\n", msg.usrName, msg.msgText); // 先在终端展示
                    for (NodePtr p = L->next; p != NULL; p = p->next)
                    {
                        sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr *)&p->addr, len);
                    }
                    printf("发送成功\n");
                }
                break;
                case 'Q':
                {
                    strcpy(msg.msgText, "*********");
                    strcat(msg.msgText, msg.usrName);
                    strcat(msg.msgText, "退出聊天室*********");
                    for (NodePtr p = L->next; p != NULL; p = p->next)
                    {
                        sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr *)&p->addr, len);
                    }
                    sleep(1);
                    list_delete(L, msg.usrName);
                }
                break;
                }
            }
        }
        else if (pid > 0)
        {
            // 发送数据给客户端

            while (1)
            {
                msg.type = 'C';
                strcpy(msg.usrName, "GM");
                fgets(msg.msgText, sizeof(msg.msgText), stdin);
                msg.msgText[strlen(msg.msgText) - 1] = '\0';
                sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr *)&sin, len);
                if (strcmp(msg.msgText, "exit") == 0)
                {
                    goto end;
                }
            }
        }
        else if (pid < 0)
        {
            perror("fork error");
            return 1;
        }
    end:
        kill(pid, SIGKILL);
        list_destroy(L);
        close(sfd);
        wait(NULL);
        return 0;
    }
}
//头文件
#include <myhead.h>
#ifndef __HEAD_H__
#define __HEAD_H__
// 定义结构体

typedef struct Node
{

    union
    {
        int len;       // 节点长度
        char name[20]; // 节点名称
    };
    struct sockaddr_in addr; // 地址信息结构体
    struct Node *next;       // 指向下一个节点
} Node, *NodePtr;

struct msgType
{
    char type;          // 消息类型 L:登录,C:聊天,Q;退出
    char usrName[20];   // 姓名
    char msgText[1024]; // 消息
};
// 创建链表
NodePtr list_create();
// 申请封装节点函数
NodePtr apply_node(char *name, struct sockaddr_in *addr);
// 链表判空
int list_empty(NodePtr L);
// 链表插入
int list_insert(NodePtr L, char *name, struct sockaddr_in *addr);


// 按名字找位置函数
int list_search_name(NodePtr L, char *name);
// 任意位置查找节点
NodePtr list_search_pos(NodePtr L, int pos);
// 任意位置删
int list_delete(NodePtr L, char *name);

// 链表销毁
void list_destroy(NodePtr L);

#endif

//函数文件
#include "head.h"
// 创建链表
NodePtr list_create()
{
    // 申请头结点
    NodePtr L = (NodePtr)malloc(sizeof(Node));
    if (NULL == L)
    {
        printf("创建失败\n");
        return NULL;
    }
    // 赋值
    L->len = 0;
    L->next = NULL;
    printf("聊天室创建成功\n");
    return L;
}
// 申请封装节点函数
NodePtr apply_node(char *name, struct sockaddr_in *addr)
{
    // 申请空间
    NodePtr p = (NodePtr)malloc(sizeof(Node));
    if (NULL == p)
    {
        printf("申请失败\n");
        return NULL;
    }
    // 赋值
    strcpy(p->name, name);
    p->addr.sin_family = addr->sin_family;
    p->addr.sin_port = addr->sin_port;
    p->addr.sin_addr.s_addr = addr->sin_addr.s_addr;
    p->next = NULL;
    
    return p;
}
// 链表判空
int list_empty(NodePtr L)
{   
    return L->next == NULL;
}
// 链表插入
int list_insert(NodePtr L, char *name, struct sockaddr_in *addr)
{
    // 判断逻辑
    if (NULL == L)
    {
        return -1;
    }
    
    // 申请节点
    NodePtr p = apply_node(name, addr);
    if (NULL == p)
    {
        return -1;
    }
    // 插入逻辑
    p->next = L->next;
    L->next = p;
    printf("*********%s进入聊天室*********\n", p->name);
    L->len++;
    return 0;
}
// 按名字找位置函数
int list_search_name(NodePtr L, char *name)
{
    // 判断逻辑
    if (NULL == L || list_empty(L))
    {
        return -1;
    }
    // 遍历逻辑
    NodePtr q = L->next;
    for (int index = 1;index<=L->len; index++)
    {
        if(strcmp(q->name, name) == 0)
        {
            return index;
            
        }
        q = q->next;
    }
    printf("未找到该用户\n");
    return -1;
}
// 任意位置查找节点
NodePtr list_search_pos(NodePtr L, int pos)
{
    if (NULL == L || list_empty(L) || pos < 0 || pos > L->len)
    {
        return NULL;
    }
    //查找逻辑
    //定义指针从头结点出发
    NodePtr q = L;
    for (int i = 0; i < pos; i++)
    {
        q = q->next;
    }
    return q; //返回找到的节点
}
// 任意明治删
int list_delete(NodePtr L, char *name)
{
    // 判断逻辑)
    if (NULL == L || list_empty(L))
    {
        return -1;
    }
    // 找到前一个节点
    int pos = list_search_name(L, name);
    NodePtr q = list_search_pos(L, pos-1);
    // 删除逻辑
    NodePtr p = q->next;
    q->next = p->next;
    // 释放空间
    free(p);
    p = NULL;
    printf("*********%s退出聊天室*********\n", name);
    return 0;
}

// 链表销毁
void list_destroy(NodePtr L)
{
    // 判断逻辑
    if (NULL == L)
    {
        return;
    }
    // 遍历逻辑
    while (!list_empty(L))
    {
        list_delete(L, L->next->name);
    }
    // 释放头结点
    free(L);
    L = NULL;
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2041859.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

SpringBoot解决创建项目无法选择JDK8和JDK11

文章目录 解决方案1解决方案2 在创建SpringBoot项目的时候&#xff0c;我们发现只能勾选JDK17以上的。并且官方没有提供2.X版本&#xff0c;但是目前大多数企业使用的还是 springboot 初始化的网址&#xff0c;我们一般使用的是官方的网址。 解决方案1 就选择jdk17和spring…

BIM+GIS在管廊机电监控与运维管控系统中的应用

研究背景 根据《GB50838-2015城市综合管廊工程技术规范》及《GBT51274-2017城镇综合管廊监控与报警系统工程技术标准》的相关条款要求&#xff0c;城市综合管廊监控报警系统用于对综合管廊内的设备运行状态及参数、实时环境信息、出入口状态等进行全方位在线监控&#xff0c;保…

java基础概念17-static

一、 static的作用 static修饰的变量、方法被类的所有实例共享。 示例&#xff1a; static用于声明属于类本身的变量、方法&#xff0c;而不是类的某个特定对象的。 二、static内存图 静态区中的成员变量&#xff0c;对象共享&#xff0c;内存中只有一份&#xff0c;谁要用&am…

u2net 和u2netp 的具体区别

U2Net和U2NetP是两种基于深度学习的图像分割模型&#xff0c;它们都使用了编码器-解码器架构和跳跃连接来提高分割的精度。然而&#xff0c;它们在网络结构和参数配置上存在一些差异。 初始化阶段的中间通道数 (mid_ch): U2Net: self.stage1 RSU7(in_ch, 32, 64)U2NetP: self.…

RHEL8 配置epel源

** RHEL8 配置epel源 ** 此次环境为最小化安装&#xff0c;版本信息如下&#xff1a;redhat8 一、安装epel源&#xff0c;执行如下命令&#xff1a; #yum install https://dl.fedoraproject.org/pub/epel/epel-release-latest-8.noarch.rpm 之后执行#yum repolis 此时我们…

山海关古城测试--房产房屋

1.树状图搜索房产 1.1实现思路 因为树状图搜索要显示在界面中&#xff0c;所以需要在html文件中进行添加树状结构 而树状结构怎么来的&#xff0c;所以需要在controller中新写一个方法来传输一个树状结构&#xff0c;让html直接访问它即可树状结构在framework->web->Zt…

ES主分片和副本分片

在 Elasticsearch 中&#xff0c;主分片的数量在创建索引时设置&#xff0c;并且不能在索引创建后更改。主分片的数量因索引而异&#xff0c;对于每个索引&#xff0c;可以根据实际需要进行调整。 主分片数量的设置 默认值&#xff1a; 在 Elasticsearch 中&#xff0c;默认的主…

C++入门基础(太干了,别噎住)

1.C了解&#xff08;可跳过&#xff09; 1.1 C发展历史 C的起源可以追溯到1979年&#xff0c;当时Bjarne Stroustrup(本贾尼斯特劳斯特卢普&#xff0c;这个翻译的名字不同的地方可能有差异)在贝尔实验室从事计算机科学和软件工程的研究工作。面对项目中复杂的软件开发任务&am…

maven配置aspose镜像

问题 安装aspose一直失败&#xff0c;原因是aspose文件不在公共仓库中 <dependency><groupId>com.aspose</groupId><artifactId>aspose-pdf</artifactId><version>21.11</version> </dependency>配置文件如下 <mirror&g…

苍穹外卖-知识点

搭建环境 前端 使用nginx&#xff08;文件路径带中文 会启动不成功&#xff09; 后端

有哪些文档翻译在线免费工具?安利5款文档翻译利器给你

想要翻译文档但却不知道该用什么工具好&#xff1f;这其实是不少学生党和打工人共同的烦恼~ 实际上你既可以借助支持全文翻译的文档翻译工具来获取译文&#xff0c;也可以利用自带划词翻译功能的文档翻译工具来操作。借此机会便给安排上了两种不同翻译类型的文档翻译工具&…

谐波抑制方法分析

一、谐波抑制概述 抑制谐波方法可分为两大类&#xff1a;一是主动型&#xff0c;即从装置本身出发&#xff0c;使其不产生谐波或者降低谐波的产生&#xff1b;被动型&#xff0c;即外加滤波器&#xff0c;吸收或者抵消谐波产生设备注入电网的谐波&#xff0c;比如在电网和设备的…

干货分享|如何使用Stable Diffusion打造会说话的数字人?

数字人已不是什么新鲜名词了。在许多领域&#xff0c;尤其是媒体和娱乐领域&#xff0c;经常可以看到卡通形象的人物或逼真的虚拟主持人。在Stable Diffusion中&#xff0c;我们可以上传一段录制好的音频文件&#xff0c;然后使用SadTalker插件&#xff0c;将音频和图片相结合&…

怎么用打印功能导出为文件为pdf格式

概述 有时候我们再打印的时候需要把文件打印出来的同时&#xff0c;但又想下载下来该文件为pdf格式&#xff01;该怎么办接下来我来教大家解决办法&#xff01; 方法一 安装WPS PDF虚拟打印机&#xff1a;‌首先&#xff0c;‌从可靠的来源下载WPS PDF虚拟打印机的安装包。‌这…

HAProxy负载均衡详细解释

目录 1、HAProxy的负载均衡 1.1socat工具的使用 1.1.1对于单进程 1.1.2对于多进程处理方法(对haproxy做热处理) 2、Haproxy的算法 2.1静态算法 <1>static-rr <2>first 2.2动态算法 <1>roundrobin <2>leastconn <3>random 2.3其他算…

基于SSM的考试管理系统---附源码16535

目录 摘要 Abstract 1 绪论 1.1课题目的及意义 1.2研究背景 1.3研究方法 1.4论文结构与章节安排 2 考试管理系统的设计与实现系统分析 2.1 可行性分析 2.2 系统功能分析 2.3 系统用例分析 2.4本章小结 3 考试管理系统的设计与实现总体设计 3.1 系统架构设计 3.2 …

C语言:基于单链表实现通讯录项目

前言 之前我们用顺序表实现了通讯录&#xff0c;这次我们使用单链表实现。我们定义五个文件&#xff0c; SingleLinkedList.h SingleLinkedList.c Contact.h Contact.c test.c SingleLinkedList.h 是包含了单链表的结构&#xff0c;及各类库文件声明&#xff0c;各个单链表接…

优思学院|六西格玛绿带重不重要?绿带的报考条件是什么?

在最原始的六西格玛管理之中&#xff0c;六西格玛的最主要角色就只有六西格玛黑带&#xff0c;这是Mikel Harry所命名的&#xff0c;意思是一个对六西格玛方法十分熟练的人。 然而&#xff0c;六西格玛黑带也要靠团队才能完成工作&#xff0c;如果团队中只有黑带了解六西格玛流…

AcWing算法提高课笔记——数字三角形

本文给出了几道数字三角形模型的题解。 数字三角形模型通常给定一个矩阵&#xff0c;从矩阵的左上角走到右下角&#xff0c;每次只能向右走或者向下走&#xff0c;求最大值或者最小值。 这种问题可以用f[i, j]来表示从[1, 1]走到[i, j]处的Max\Min&#xff0c;有状态转移方程…

minikube 实践练习4 - 滚动更新

1. 应用版本更新 #查看image kubectl describe pods#设置应用的image为新版本 v1 -> v2 kubectl set image deployments/kubernetes-bootcamp kubernetes-bootcampdocker.io/jocatalin/kubernetes-bootcamp:v2#验证更新结果 export NODE_PORT"$(kubectl get services/…