基于UDP的网络聊天室

于 2023-02-22 09:25:17 发布
阅读量55 收藏
点赞数
文章标签： 网络 udp 网络协议 Powered by 金山文档
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_59039307/article/details/129155365
版权

项目需求：

  1. 如果有用户登录，其他用户可以收到这个人的登录信息

  1. 如果有人发送信息，其他用户可以收到这个人的群聊信息

  1. 如果有人下线，其他用户可以收到这个人的下线信息

  1. 服务器可以发送系统信息

chatSer.c

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

#define N 128
//协议
#define LOGIN     1
#define CHAT     2
#define QUIT     3

#define ERR_LOG(msg) do{\
    printf("%d %s %s \n", __LINE__, __func__, __FILE__);\
    perror(msg);\
}while(0)

//收发信息的结构体
typedef struct
{
    int type;
    char name[20];
    char text[N] ;

}MSG;


typedef struct node
{
    struct sockaddr_in cin;
    struct node* next;
}__linklist;

int do_recv(int sfd, __linklist* head);
int do_login(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin);
int do_quit(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin);
int do_chat(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin);
int do_system(int sfd, struct sockaddr_in sin);

int main(int argc, const char *argv[])
{
    if(argc < 3)
    {
        printf("请输入 ip 和端口号\n");
        return -1;
    }

    //1.创建报式套接字
    int sfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(sfd < 0)
    {
        perror("socket");
        return -1;
    }

    //2.绑定服务器ip和端口号
    struct sockaddr_in sin;
    sin.sin_family         = AF_INET;
    sin.sin_port         = htons(atoi(argv[2]));
    sin.sin_addr.s_addr = inet_addr(argv[1]);

    if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        perror("bind");
        return -1;
    }
    
    //创建进程或线程
    int pid = 0;
    pid = fork();
    if(pid > 0)
    {
        //父进程, 接收消息
        __linklist* head = (__linklist*)malloc(sizeof(__linklist));
        head->next = NULL;
        do_recv(sfd , head);

    }
    else if(pid==0)
    {
        //子进程,发送系统消息
        do_system(sfd, sin);
    }


    //4.关闭套接字
    close(sfd);
    return 0;
}

int do_system(int sfd, struct sockaddr_in sin)
{
    MSG sys_msg = {htonl(CHAT), "**system**"};

    while(1)
    {
        bzero(sys_msg.text, N);
        fgets(sys_msg.text, N, stdin);
        sys_msg.text[strlen(sys_msg.text)-1] = 0;

        //将当前进程当做客户端,父进程当做服务器,发送信息;
        if(sendto(sfd, &sys_msg, sizeof(sys_msg), 0, (void*)&sin, sizeof(sin)) < 0)
        {
            ERR_LOG("sendto");
            return -1;
        }
    }
    printf("系统消息发送成功");
    return 0;
}

int do_recv(int sfd, __linklist* head)
{
    MSG rcv_msg;
    int recv_len = 0;
    struct sockaddr_in cin;
    socklen_t clen = sizeof(cin);

    while(1)
    {
        //接收消息
        recv_len = recvfrom(sfd, &rcv_msg, sizeof(rcv_msg), 0, (void*)&cin, &clen);
        if(recv_len < 0)
        {
            ERR_LOG("recvfrom");
            return -1;
        }
        //提取出协议
        //将网络字节序转换为主机字节序
        int type = ntohl(rcv_msg.type);

        switch(type)
        {
        case LOGIN:
            do_login(sfd, head, rcv_msg, cin);
            break;
        case CHAT:
            do_chat(sfd, head, rcv_msg, cin);
            break;
        case QUIT:
            do_quit(sfd, head, rcv_msg, cin);
            break;
        }

    }
}

int do_chat(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin)
{
    
    printf("%s [%s:%d]chat成功\n", rcv_msg.name,\
            (char*)inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
    //重新拼接群聊消息: 名字+消息
    char buf[258] = "";
    sprintf(buf, "%s:%s", rcv_msg.name, rcv_msg.text);
    strcpy(rcv_msg.text, buf);

    //循环发送,除了自己以外的ip地址
    while(head->next != NULL)
    {
        head = head->next;
        if(memcmp(&cin, &head->cin, sizeof(cin)) != 0)
        {    
            if(sendto(sfd, &rcv_msg, sizeof(rcv_msg),0, (void*)&(head->cin), sizeof(head->cin))<0)
            {
                ERR_LOG("sendto");
                return -1;
            }
        }
    }
    return 0;    
}



int do_quit(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin)
{
    sprintf(rcv_msg.text, "-----%s 已下线-----\n", rcv_msg.name);
    //循环遍历链表,发送"xxx一下线"的信息
    //发送给当前客户端外的其余客户端
    while(head->next != NULL)
    {
        if(memcmp(&cin, &head->next->cin, sizeof(cin)) == 0)
        {
            //从链表中删除该客户端信息
            //free
            __linklist* temp = head->next;
            head->next = temp->next;
            free(temp);
        }
        else
        {
            head = head->next;
            if(sendto(sfd, &rcv_msg, sizeof(rcv_msg),0, (void*)&head->cin, sizeof(head->cin))<0)
            {
                ERR_LOG("sendto");
                return -1;
            }
        }
    }
    return 0;
}


int do_login(int sfd, __linklist* head, MSG rcv_msg, struct sockaddr_in cin)
{
    //打印登录成功
    printf("%s [%s:%d]登录成功\n", rcv_msg.name,\
            (char*)inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));

    sprintf(rcv_msg.text, "-----%s登录成功-----", rcv_msg.name);


    while(head->next != NULL)
    {
        head = head->next;
        if(sendto(sfd, &rcv_msg, sizeof(rcv_msg), 0, (void*)&(head->cin), sizeof(head->cin))<0)
        {
            ERR_LOG("sendto");
            return -1;
        }
    }

    //将登录成功的客户端信息添加到链表中;
    __linklist *temp = (__linklist*)malloc(sizeof(__linklist));
    temp->cin = cin;
    temp->next= NULL;
    
    head->next = temp;
    return 0;

}

chatCli.c

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

#define N 128
//协议
#define LOGIN     1
#define CHAT     2
#define QUIT     3

#define ERR_LOG(msg) do{\
    printf("%d %s %s \n", __LINE__, __func__, __FILE__);\
    perror(msg);\
}while(0)

//收发信息的结构体
typedef struct
{
    int type;
    char name[20];
    char text[N] ;

}MSG;

typedef void (*sighandler_t)(int);

int do_recv(int sfd);
int do_chat(int sfd, MSG msg, struct sockaddr_in sin);

void handler(int sig)
{
    //回收子进程资源并退出
    while(waitpid(-1,NULL, WNOHANG)>0);
    exit(0);
}

int main(int argc, const char *argv[])
{
    if(argc < 3)
    {   
        printf("请输入 ip 和端口号\n");
        return -1; 
    } 

    //注册信号处理函数,让子进程退出后,父进程回收子进程资源并退出
    sighandler_t s = signal(SIGCHLD, handler);
    if(s == SIG_ERR)
    {
        ERR_LOG("signal");
        return -1;
    }

    //1.创建报式套接字
    int sfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(sfd < 0)
    {
        perror("socket");
        return -1;
    }

    //2.填充服务器ip和端口号
    struct sockaddr_in sin;
    sin.sin_family         = AF_INET;
    sin.sin_port         = htons(atoi(argv[2]));
    sin.sin_addr.s_addr = inet_addr(argv[1]);
    socklen_t addrlen = sizeof(sin);


    //登录协议
    MSG msg ;
    msg.type = htonl(LOGIN);

    printf("请输入姓名>>>");
    fgets(msg.name, 20, stdin);
    msg.name[strlen(msg.name)-1] = 0;

    //发送登录信息sendto
    if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0)
    {
        ERR_LOG("sendto");    
        return -1;
    }

    pid_t pid = fork();

    if(pid > 0)
    {
        //父进程获取信息
        do_recv(sfd);
    }
    else if(0 == pid)
    {
        //子进程发送信息
        do_chat(sfd, msg, sin);
    }


    //4.关闭套接字
    close(sfd);
    return 0;
}


int do_chat(int sfd, MSG msg, struct sockaddr_in sin)
{
    while(1)
    {
        //从终端获取聊天信息
        bzero(msg.text, N);
        fgets(msg.text, N ,stdin);
        msg.text[strlen(msg.text)-1] = 0;

        //如果是聊天信息,则封装上 CHAT协议 
        //如果是退出信息,则封装上 QUIT协议
        if(strncasecmp(msg.text, "quit" , 4) == 0)
        {
            msg.type = htonl(QUIT);
        }
        else
        {
            msg.type = htonl(CHAT);
        }

        //发送 
        if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0)
        {
            ERR_LOG("sendto");
            return -1;
        }

        //如果是退出信息,则客户端要退出
        //子进程:即当前进程,需要退出
        //且父进程也要退出
        if(msg.type == htonl(QUIT))
        {
            exit(0);         //退出子进程;
        }
    }
}



int do_recv(int sfd)
{
    MSG rcv_msg ;
    while(1)
    {
        if(recvfrom(sfd, &rcv_msg, sizeof(rcv_msg), 0, NULL, NULL) < 0)
        {
            perror("recvfrom");
            return -1;
        }
        printf("%s\n",rcv_msg.text);
    }
}

怪笑的桀桀
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
基于UDP通信协议的网络聊天室
01-22
本资源是一个基于UDP协议，多线程，MFC网络聊天室的代码程序
项目：基于UDP的网络聊天室
weixin_55430573的博客
08-13 133
写项目的方法： 1. 画流程图 2. 根据流程图写框架 3. 将每个功能实现（一个一个实现，不要一起写）
网络 基于UDP网络聊天室
04-02
网络工程课程设计 里面有详细原代码和word 文档
TCP/UDP网络聊天室.zip
05-28
Java Swing开发的网络聊天室，模仿企鹅设计，实现一对一，一对多，多对多的聊天功能 （备注：使用idea开发出来，压缩的时候是刚买mac的时候，压缩程序有乱码，解决方案在文件夹的pdf中）
基于MFC的UDP简易聊天室
06-14
课程设计——简易网络聊天室。 C++基于MFC开发的UDP网络聊天室。
基于UDP协议的聊天程序
06-03
java环境下的基于UDP协议的聊天程序，udp协议聊天程序 具体功能： 1、实现多客户端之间的群聊功能； 2、客户端、服务器端均能显示在线用户列表； 3、服务器负责转发聊天消息；
基于MFC UDP的网络聊天室程序
05-25
C++编写的基于UDP，MFC的网络聊天程序，简单容易上手 输入IP和端口号，即可进入聊天
【基于UDP的网络聊天室】
sept_94的博客
09-24 1842
基于UDP的网络聊天室
基于UDP的简易网络聊天室
laoKdx的博客
08-12 1684
基于UDP的简易聊天室
c++基于udp的网络聊天室
最新发布
06-10
网络聊天室是一个多人在线聊天系统，基于UDP协议实现的话，需要建立一个UDP Socket来进行消息传输。以下是实现网络聊天室的大致步骤： 1. 创建UDP Socket，设置套接字选项以支持广播和多播。 2. 绑定Socket到本地IP...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值