多路文件IO-select(8.21）

运行1个服务器和2个客户端
实现效果：
    服务器和2个客户端互相聊天，服务器和客户端都需要使用select模型去实现
    服务器要监视2个客户端是否连接，2个客户端是否发来消息以及服务器自己的标准输入流
    客户端要监视服务器是否发来消息以及客户端自己的标准输入流
    在不开线程的情况下，实现互相聊天

服务器端：

#include <myhead.h>
typedef struct sockaddr_in addr_in_t;
typedef struct sockaddr addr_t;
typedef struct sockaddr_un addr_un_t;

// 向客户端数组中插入一个新的客户端句柄
void insert_client(int *client_arr, int *len, int client)
{
    client_arr[*len] = client;
    (*len)++;
}

// 查找客户端在数组中的位置
int find_client(int *client_arr, int len, int client)
{
    for (int i = 0; i < len; i++)
    {
        if (client_arr[i] == client)
        {
            return i;
        }
    }
    return -1;
}

// 从客户端数组中移除一个客户端
void remove_client(int *client_arr, int *len, int client)
{
    int tar = find_client(client_arr, *len, client);
    if (tar == -1)
    {
        return;
    }
    int i = -1;
    for (i = tar; i < *len - 1; i++)
    {
        client_arr[i] = client_arr[i + 1];
    }
    client_arr[i] = 0;
    (*len)--;
}

int main(int argc, char const *argv[])
{
    if (argc != 2)
    {
        printf("请输入正确的端口号\n");
        return 1;
    }
    int port = atoi(argv[1]);
    fd_set readfds;
    FD_ZERO(&readfds);                            // 初始化文件描述符集
    int client_arr[100] = {0};                    // 客户端数组
    int client_count = 0;                         // 当前客户端数量
    int server = socket(AF_INET, SOCK_STREAM, 0); // 创建服务器套接字

    addr_in_t addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    addr.sin_addr.s_addr = inet_addr("192.168.123.165");

    if (bind(server, (addr_t *)&addr, sizeof(addr)) == -1) // 绑定地址到套接字
    {
        perror("bind");
        return 1;
    }
    listen(server, 100);      // 开始监听连接请求
    FD_SET(server, &readfds); // 将服务器套接字加入文件描述符集中

    while (1)
    {
        fd_set temp = readfds;              // 复制文件描述符集
        select(FD_SETSIZE, &temp, 0, 0, 0); // 等待读事件

        // 如果服务器套接字可读，则表示有新的客户端连接请求
        if (FD_ISSET(server, &temp))
        {
            int client = accept(server, 0, 0); // 接受连接
            printf("有新客户端连接\n");
            FD_SET(client, &readfds);                         // 将客户端套接字加入文件描述符集中
            insert_client(client_arr, &client_count, client); // 添加到客户端数组
        }
        else
        {
            // 遍历所有已连接的客户端
            for (int i = 0; i < client_count; i++)
            {
                int client = client_arr[i];
                int new_client = client;
                if (FD_ISSET(client, &temp)) // 如果客户端套接字可读
                {
                    char rbuf[128] = {0}, sbuf[128] = {0};
                    int res = read(client, rbuf, 128); // 读取客户端数据
                    if (res == 0)                      // 如果客户端关闭连接
                    {
                        printf("有客户端断开连接\n");
                        // 从文件描述符集中移除客户端
                        FD_CLR(client, &readfds);
                        // 从客户端数组中移除客户端
                        remove_client(client_arr, &client_count, client);
                        // 关闭客户端套接字
                        close(client);
                        break;
                    }
                    printf("客户端发来消息:%s\n", rbuf); // 打印客户端发送的消息
                    addr_in_t client_addr;
                    socklen_t client_addr_len = sizeof(client_addr);
                    if (getpeername(client, (struct sockaddr *)&client_addr, &client_addr_len) == -1)
                    {
                        perror("getpeername");
                        continue;
                    }
                    // 遍历所有已连接的客户端，发送消息
                    for (int j = 0; j < client_count; j++)
                    {
                        int client = client_arr[j];
                        if (client != server && client != new_client ) // 排除服务器和发送消息的客户端
                        {
                            // 将发送消息的IP地址添加到消息中
                            strcpy(sbuf, inet_ntoa(client_addr.sin_addr));
                            strcat(sbuf, ":");
                            strcat(sbuf, rbuf);
                            write(client, sbuf, strlen(sbuf)); // 发送消息
                        }
                    }
                }
            }
        }
    }
    return 0;
}

客户端： 

#include <myhead.h>

#define SER_PORT 12345
#define SER_IP "192.168.123.165"
#define CLI_PORT 8888
#define CLI_IP "192.168.123.165"

int main(int argc, const char *argv[])
{
    //1、创建用于通信的套接字文件描述符
    int cfd = socket(AF_INET, SOCK_STREAM, 0);
    if (cfd == -1)
    {
        perror("socket error");
        return -1;
    }
    printf("cfd = %d\n", cfd); //3

    //2、连接到服务器
    //2.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地址

    //2.2 连接服务器
    if (connect(cfd, (struct sockaddr *)&sin, sizeof(sin)) == -1)
    {
        perror("connect error");
        return -1;
    }
    printf("连接服务器成功\n");
    printf("可以开始聊天了\n");
    //4、数据收发
    char buf[128] = "";
    int max_fd = cfd; // 最大文件描述符
    fd_set readfds;   // 文件描述符集合

    while (1)
    {
        FD_ZERO(&readfds);              // 清空文件描述符集合
        FD_SET(cfd, &readfds);          // 添加客户端套接字到文件描述符集合
        FD_SET(STDIN_FILENO, &readfds); // 添加标准输入流到文件描述符集合

        // 使用 select 监控客户端套接字和标准输入流
        int activity = select(max_fd + 1, &readfds, 0, 0, 0);
        if (activity < 0)
        {
            perror("select error");
            break;
        }

        // 检查标准输入流是否有数据
        if (FD_ISSET(STDIN_FILENO, &readfds))
        {
            fgets(buf, sizeof(buf), stdin); // 从终端获取一个字符串
            buf[strlen(buf) - 1] = 0;       // 去掉换行符

            // 将数据发送给服务器
            send(cfd, buf, strlen(buf), 0);
        }

        // 检查客户端套接字是否有数据
        if (FD_ISSET(cfd, &readfds))
        {
            bzero(buf, sizeof(buf)); // 清空容器
            int bytes_received = recv(cfd, buf, sizeof(buf), 0);
            if (bytes_received <= 0)
            {
                printf("连接已断开\n");
                break;
            }
            printf("%s\n", buf);
        }
    }

    //5、关闭套接字
    close(cfd);

    return 0;
}

客户端1：

客户端2：