小项目-UDP聊天室

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

关键数据结构:

"[code:c]"
//不论发什么类型的消息，服务器和客户端都统一为以下结构体，服务器将处理好的回复消息放到text字段中
typedef struct
{
    int type; // 消息类型
    char name[N];  //客户名
    char text[N]; // 消息正文
} MSG;

//链表的节点结构参考
typedef struct node
{
    struct sockaddr_in addr; // 存储客户端的地址
    struct node *next;       // 指向下一个用户结点
} linknode, *linklist;

程序流程图

服务器

客户端

服务器代码：

"[code:c]"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>

#define N 256

enum MSG_TYPE
{
    L = 1,
    C,
    Q
};

typedef struct
{
    int type; // 消息类型
    char name[N];
    char text[N]; // 消息正文
} MSG;

typedef struct node
{                            // 用户结点
    struct sockaddr_in addr; // 地址
    struct node *next;       // 指向下一个用户结点
} linknode, *linklist;

static int sockfd;
static linknode head;
#define LEN sizeof(MSG)
static socklen_t len = sizeof(struct sockaddr);

void init_head()
{
    head.next = NULL;
}

/*
    1.把新注册用户登陆消息告诉其它用户
    2.把新用户插入到用户链表中
    3.服务器打印一下
*/
void do_login(struct sockaddr_in clientaddr, MSG msg)
{
    linklist p, q;

    // 把新注册用户登陆消息告诉其它用户
    p = head.next;
    sprintf(msg.text, "%s login", msg.name); // 拼串函数
    while (p)
    {
        sendto(sockfd, &msg, LEN, 0, (struct sockaddr *)&p->addr, len);
        p = p->next;
    }

    // 插入新的地址
    q = (linklist)malloc(sizeof(linknode));
    q->addr = clientaddr;

    q->next = head.next;
    head.next = q;

    // 服务器打印
    puts(msg.text);
}

/*
    1.把聊天信息发给其它用户
    2.发给自己一份
*/
void do_chat(struct sockaddr_in clientaddr, MSG msg)
{
    linklist p;
    char buf[N] = {0};

    // 把聊天信息发给其它用户
    p = head.next;

    // sprintf不允许一个变量既当参数又当目的
    sprintf(buf, "%s said %s", msg.name, msg.text);
    strcpy(msg.text, buf);

    while (p)
    {
        if (memcmp(&p->addr, &clientaddr, len) != 0)
        {
            sendto(sockfd, &msg, LEN, 0, (struct sockaddr *)&p->addr, len);
        }
        p = p->next;
    }

    // 服务器打印
    puts(msg.text);
}

void do_quit(struct sockaddr_in clientaddr, MSG msg) // 退出处理
{
    linklist p = &head;
    linklist q;
    sprintf(msg.text, "%s offline", msg.name);
    socklen_t len = sizeof(struct sockaddr_in);

    while (p->next)
    {
        if (memcmp(&clientaddr, &p->next->addr, sizeof(clientaddr)) == 0) // 删除节点操作
        {
            q = p->next;
            p->next = q->next;
            free(q);
            q = NULL;
        }
        else
        {
            sendto(sockfd, &msg, LEN, 0, (struct sockaddr *)&p->next->addr, len);
            p = p->next;
        }
    }
    puts(msg.text);
}

int main(int argc, char *argv[])
{
    // 0定义变量
    int nbytes; // 获取接收到数据字节数
    MSG msg;
    struct sockaddr_in serveraddr, clientaddr;
    int addrlen = sizeof(struct sockaddr_in);

    if (argc < 3)
    {
        puts("server <addr> <port>");
        return -1;
    }

    // 1创建UDP套接字--socket
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd < 0)
    {
        perror("socket err");
        return -1;
    }

    // 2定义套接字地址--sockaddr_in
    bzero(&serveraddr, addrlen);
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = INADDR_ANY;

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

    init_head();

    while (1)
    {
        recvfrom(sockfd, &msg, LEN, 0, (struct sockaddr *)&clientaddr, &len);
        switch (msg.type)
        {
        case L:
            do_login(clientaddr, msg);
            break;
        case C:
            do_chat(clientaddr, msg);
            break;
        case Q:
            do_quit(clientaddr, msg);
            break;
        default:
            break;
        }
    }
}

客户端代码：

"[code:c]"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <signal.h>

#define N 256

enum MSG_TYPE
{
    L = 1,
    C,
    Q
};

typedef struct
{
    int type; // 消息类型
    char name[N];
    char text[N]; // 消息正文
} MSG;

#define LEN sizeof(MSG)

int main(int argc, const char *argv[])
{
    int sockfd;
    struct sockaddr_in serveraddr;
    MSG msg;
    pid_t pid;
    socklen_t len = sizeof(struct sockaddr);

    if (argc != 3)
    {
        printf("user:%s ip port\n", argv[0]);
        return -1;
    }

    // 创建socket
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sockfd < 0)
    {
        perror("socket err");
        exit(-1);
    }

    // 指定服务器地址
    bzero(&serveraddr, len);
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);

    // 输入用户名，并发给服务器登录信息
    puts("====================client login=================");
    printf("input your name:");
    gets(msg.name);
    msg.type = L;
    sendto(sockfd, &msg, LEN, 0, (struct sockaddr *)&serveraddr, len);

    // fork--子进程发送，父进程接收
    pid = fork();
    if (pid < 0)
    {
        perror("fork err");
        exit(0);
    }
    else if (pid > 0)
    { // 父进程只负责接收消息
        while (1)
        {
            recvfrom(sockfd, &msg, LEN, 0, NULL, NULL);
            puts(msg.text);
        }
    }
    else
    { 
        // 子进程负责循环发送消息
        while (1)
        {
            // frok共享父子进程空间，name前面已经赋值了
            gets(msg.text);
            if (strncmp(msg.text, "quit", 4) == 0) // 退出
            {
                msg.type = Q;
                sendto(sockfd, &msg, LEN, 0, (struct sockaddr *)&serveraddr, len);
                close(sockfd);
                kill(getppid(), SIGKILL); // 杀死父亲
                exit(0);                  // 自杀
            }
            else // 聊天
            {
                msg.type = C;
                sendto(sockfd, &msg, LEN, 0, (struct sockaddr *)&serveraddr, len);
            }
        }
    }
}