利用UDP协议，实现一套聊天室软件。服务器端记录客户端的地址，客户端发送消息后，服务器群发给各个客户端软件。

最新推荐文章于 2024-05-03 21:43:52 发布

威猛的佩奇

最新推荐文章于 2024-05-03 21:43:52 发布

阅读量305

点赞数

文章标签： udp 网络协议 tcp/ip Powered by 金山文档

本文链接：https://blog.csdn.net/cxh0105/article/details/128943660

版权

UDP（User Datagram Protocol）用户数据报协议，是不可靠的无连接的协议。在数据发送前，因为不需要进行连接，所以可以进行高效率的数据传输。

服务器端

#include <stdio.h>
#include "linklist.h"
#define LINE printf("%d\n", __LINE__)

link_node_t *Linklist_Create();
struct sockaddr_in serveraddr, caddr;
int Log_in_Insert_linked_list(struct sockaddr_in caddr, int sockfd, MSG_t msg, link_list_t p);
int do_chat(struct sockaddr_in caddr, int sockfd, MSG_t msg, link_list_t head);
int quit(struct sockaddr_in caddr, int sockfd, MSG_t msg, link_list_t head);

void *pthread(void *arg)
{
    MSG_t msg;
    int sockfd = (*(int *)arg);
    msg.type = 'C';
    strcpy(msg.name, "server");
    while (1)
    {
        fgets(msg.text, sizeof(msg.text), stdin);
        if (msg.text[strlen(msg.text) - 1] == '\n')
            msg.text[strlen(msg.text) - 1] = '\0';
        sendto(sockfd, &msg, sizeof(msg), 0,
               (struct sockaddr *)&serveraddr, sizeof(serveraddr));
    }
    pthread_exit(NULL);
}

int main(int argc, char const *argv[])
{
    int sockfd = socket(AF_INET, SOCK_DGRAM, 0); //创建套接子
    if (sockfd < 0)
    {
        perror("socket err");
        return -1;
    }
    //多路端口复用
    int optval = 1;
    setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
    //填充ipv4结构体

    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(8888);
    serveraddr.sin_addr.s_addr = inet_addr("0.0.0.0");
    socklen_t len = sizeof(caddr);

    //绑定
    if (bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
    {
        perror("bind err");
        return -1;
    }

    pthread_t tid;
    pthread_create(&tid, NULL, pthread, &sockfd);
    pthread_detach(tid);

    MSG_t msg;
    link_list_t head = Linklist_Create();

    while (1)
    {

        if (recvfrom(sockfd, (char *)&msg, sizeof(msg), 0, (struct sockaddr *)&caddr, &len) < 0)
        {
            perror("recvfrom err");
        }

        switch (msg.type)
        {
        case 'L':
            Log_in_Insert_linked_list(caddr, sockfd, msg, head);
            break;

        case 'C':
            do_chat(caddr, sockfd, msg, head);
            break;
        case 'Q':
            quit(caddr, sockfd, msg, head);
            break;
        }
    }
    close(sockfd);
    return 0;
}

link_node_t *Linklist_Create() //创建链表
{
    link_list_t p = (link_list_t)malloc(sizeof(link_node_t));
    if (NULL == p)
    {
        perror("Linklist_Create err");
        return NULL;
    }
    p->next = NULL;
    return p;
}

int Log_in_Insert_linked_list(struct sockaddr_in caddr, int sockfd, MSG_t msg, link_list_t head) //插入新节点
{
    link_list_t ptemp = head;
    //发送上线信息
    sprintf(msg.text, "%s %s", msg.name, "上线了");
    //printf("%s\n", msg.text);
    while (ptemp->next != NULL)
    {
        ptemp = ptemp->next;
        sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&ptemp->addr, sizeof(ptemp->addr));
    }

    //将上线的客户端插入链表 头插
    link_list_t pnew = Linklist_Create();
    pnew->addr = caddr;
    pnew->next = head->next;
    head->next = pnew;

    return 0;
}
int do_chat(struct sockaddr_in caddr, int sockfd, MSG_t msg, link_list_t head)
{
    link_list_t ptemp = head;
    while (ptemp->next != NULL)
    {
        ptemp = ptemp->next;
        if (memcmp(&(ptemp->addr), &caddr, sizeof(caddr)))
        {

            if (sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(ptemp->addr), sizeof(ptemp->addr)) < 0)
            {
                perror("sendto error");
            }
        }
    }
    return 0;
}
int quit(struct sockaddr_in caddr, int sockfd, MSG_t msg, link_list_t head)
{
    link_list_t ptemp = head;
    sprintf(msg.text, "%s %s", msg.name, "下线了");
    //printf("%s\n", msg.text);
    while (ptemp->next != NULL)
    {

        if (memcmp(&(ptemp->next->addr), &caddr, sizeof(caddr)))
        {
            LINE;
            ptemp = ptemp->next;
            if (sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&(ptemp->addr), sizeof(caddr)) < 0)
            {
                perror("sendto err");
            }
            LINE;
        }
        else
        {
            link_list_t pdel = ptemp->next;
            ptemp->next = pdel->next;
            free(pdel);
            pdel = NULL;
        }
    }
    return 0;
}

客户端

#include <stdio.h>
#include "linklist.h"

#define N 128

#define LINE printf("%d\n",__LINE__)

struct sockaddr_in serveraddr,caddr;
int sockfd;
MSG_t msg;
void handler(int sig)
{
     msg.type='Q';
     sendto(sockfd,&msg, sizeof(msg), 0,
            (struct sockaddr *)&serveraddr,sizeof(serveraddr)); 
      exit(-1);
}

int main(int argc, char const *argv[])
{
    
    //1.创建套接字 socket
    int sockfd;
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd < 0)
    {
        perror("socket err.");
        return -1;
    }
    printf("socket ok %d\n", sockfd);

    //填充ipv4的通信结构体  服务器端ip和端口
    
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(8888);
    serveraddr.sin_addr.s_addr = inet_addr("192.168.30.230");

    //&serveraddr -->struct sockaddr_in *
    socklen_t len = sizeof(serveraddr);

    signal(SIGINT,handler);
    //3.循环收发消息
    char buf[128];
    //int recvbyte;
    MSG_t msg;
    printf("请输入用户名:");
    fgets(msg.name,N , stdin); //换行符
    msg.name[strlen(msg.name) - 1] = '\0';
    msg.type = 'L';
    sendto(sockfd, (char*)&msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
    

    pid_t pid = fork(); //子进程实现全双工通信
    if (pid < 0)
    {
        perror("fork err");
        return -1;
    }
    else if (pid == 0)//子 接收
    {
        while (1)
        {
            //printf("aaaaaa\n");
            if ((recvfrom(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, &len)) < 0)
            {
                perror("recvfrom err.");
                return -1;
            }
            //printf("message:%s\n", msg.text);
            printf("%s:%s\n",msg.name,msg.text);
        }
    }
    else
    {
        while (1)
        {
            msg.type = 'C';
            fgets(msg.text,N,stdin);
            if (msg.text[strlen(msg.text) - 1] == '\n')
                msg.text[strlen(msg.text) - 1] = '\0';
            if(0 == strcmp(msg.text, "quit"))
            {
                msg.type = 'Q';
                //strcpy(msg.text, "退出群聊");
                if(-1 == sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, sizeof(serveraddr)))
                    perror("sendto err");
                break;
            }

            //发送消息
            sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
        }
        kill(pid, SIGKILL);
        wait(NULL);
    }

    close(sockfd);
    return 0;
}

头文件

#ifndef _LINKLIST_H_
#define _LINKLIST_H_


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

typedef struct node//存放客户端
{
    struct sockaddr_in addr;
    struct node *next;
}link_node_t,*link_list_t; 

typedef struct msg_t
{
    char type;//L  C  Q  
    char name[32];//用户名
    char text[128];//消息正文
}MSG_t;

#endif