linux-cpp-chatroom/Experiment_5/client.cpp

#include "client.h"

client::client(int port, string ip) : server_port(port), server_ip(ip) {} // 初始化服务器 ip 和端口号
client::~client()
{
    close(sock); // 关闭套接字描述符
}

void client::run() // run 函数负责建立与服务器的连接并且启动发送线程和接收线程
{
    // 定义 sockfd 套接字描述符
    sock = socket(AF_INET, SOCK_STREAM, 0);

    // 定义套接字信息
    struct sockaddr_in servaddr; // 定义套接字地址结构
    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(server_port);                  // 服务器端口
    servaddr.sin_addr.s_addr = inet_addr(server_ip.c_str()); // 服务器 ip

    // 连接服务器，成功返回 0，错误返回 -1
    if (connect(sock, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0)
    {
        perror("connect");
        exit(1);
    }
    cout << "连接服务器成功\n";

    HandleClient(sock); // 处理客户端请求

    return;
}

void client::HandleClient(int conn)
{
    int choice;                              // 用户选项
    string name, password, password_confirm; // 用户名，密码，确认密码
    bool if_login = false;                   // 记录是否登录成功
    string login_name;                       // 记录成功登录的用户名

    cout << " ---------------------\n"; // 功能选择界面
    cout << "|                     |\n";
    cout << "|  欢迎来到聊天室~    |\n";
    cout << "|  请输入您的选择:    |\n";
    cout << "|     0:退出          |\n";
    cout << "|     1:登录          |\n";
    cout << "|     2:注册          |\n";
    cout << "|                     |\n";
    cout << " ---------------------\n\n";

    while (1)
    {
        cin >> choice;
        if (choice == 0)
            break; // 退出
        else if (choice == 3 && if_login)
        {
            if_login = false;                   // 退出登录
            system("clear");                    // 清屏
            cout << " ---------------------\n"; // 功能选择界面
            cout << "|                     |\n";
            cout << "|  欢迎来到聊天室~    |\n";
            cout << "|  请输入您的选择:    |\n";
            cout << "|     0:退出          |\n";
            cout << "|     1:登录          |\n";
            cout << "|     2:注册          |\n";
            cout << "|                     |\n";
            cout << " ---------------------\n\n";
        }
        // 登录
        else if (choice == 1 && !if_login)
        {
            while (1)
            {
                cout << "用户名:";
                cin >> name;
                cout << "密码:";
                cin >> password;
                // 格式化字符串
                string str = "login:" + name;
                str += "password:";
                str += password;
                send(sock, str.c_str(), str.length(), 0); // 发送登录信息:用户名和密码:格式为name:xxxpassword:xxx
                char buffer[1000];
                memset(buffer, 0, sizeof(buffer));
                recv(sock, buffer, sizeof(buffer), 0); // 接收服务器返回的登录结果
                string recv_str(buffer);
                // 登录成功
                if (recv_str.substr(0, 2) == "ok")
                {
                    if_login = true;
                    login_name = name;
                    cout << "登录成功！" << endl;
                    break;
                }
                // 登录失败
                else if (recv_str.substr(0) == "password_wrong")
                    cout << "密码错误！\n\n";
                else if (recv_str.substr(0) == "not_have_this_user")
                    cout << "用户名不存在！\n\n";
            }
        }

        // 注册
        else if (choice == 2)
        {
            cout << "请输入注册用户名：";
            cin >> name;
            while (1)
            { // 循环输入密码，直到两次密码一致
                cout << "密码：";
                cin >> password;
                cout << "确认密码：";
                cin >> password_confirm;
                if (password == password_confirm)
                    break; // 两次密码一致,跳出循环
                else
                    cout << "两次密码不一致，请重新输入\n\n"
                         << endl;
            }
            name = "name:" + name;
            password = "password:" + password;
            string str = name + password;
            send(conn, str.c_str(), str.length(), 0); // 发送注册信息:用户名和密码:格式为name:xxxpassword:xxx
            cout << "注册成功！" << endl;
            cout << "\n请继续选择:" << endl;
        }

        // 登录成功选择功能
        if (if_login)
        {
            system("clear"); // 清屏
            cout << "欢迎回来," << login_name << "!" << endl;
            cout << " ---------------------\n"; // 功能选择界面
            cout << "|                     |\n";
            cout << "|  请输入您的选择:    |\n";
            cout << "|     0:退出系统      |\n";
            cout << "|     1:发起单独聊天  |\n";
            cout << "|     2:发起群聊      |\n";
            cout << "|     3:退出登录      |\n";
            cout << "|                     |\n";
            cout << " ---------------------\n\n";
        }
    }
}

// 向服务器发送消息
// 注意：前面不用再加static
void client::SendMsg(int conn)
{
    char sendbuf[100]; // 声明一个大小为100的字符数组sendbuf
    while (1)
    {
        memset(sendbuf, 0, sizeof(sendbuf));               // 每次循环中，它使用memset函数将sendbuf清零
        cin >> sendbuf;                                    // 从cin输入流中读取用户输入的信息，将其存放到sendbuf中。
        int ret = send(conn, sendbuf, strlen(sendbuf), 0); // send函数将sendbuf中的内容发送给服务器，其中conn参数表示连接套接字。如果发送成功，send函数返回发送的字节数
        if (strcmp(sendbuf, "exit") == 0 || ret <= 0)      // ret <= 0表示发送失败
            break;
    }
}

// 从服务器接收消息
void client::RecvMsg(int conn)
{
    // 接收缓冲区
    char buffer[1000];
    // 不断接收数据
    while (1)
    {
        memset(buffer, 0, sizeof(buffer));
        int len = recv(conn, buffer, sizeof(buffer), 0);
        // recv 返回值小于等于 0，退出
        if (len <= 0)
            break;
        cout << "收到服务器发来的信息：" << buffer << endl;
    }
}