前言

上一篇文章我们简单了解了一下什么是套接字编程，这篇文章我们利用UDP套接字来实现一个简单的网络聊天室。

编写UDP套接字服务器

成员变量

// 1. socket的id，相当于文件id
int _sock;
// 2. port
uint16_t _port;

// 3 一个线程负责收放消息；另一个线程负责发消息
Thread *c; // 收消息
Thread *p; // 发消息

// 4 环形队列来实现生产消费
RingQueue<std::string> rq;

// 5 信息要群发给所有人，要记录所有人的ip和端口号
std::unordered_map<std::string, sockaddr_in> _users;

// 6 一把锁保证在读取要发送给哪些人时是安全的
//  在读取_users时，另一个线程收到了消息，会修改这个map
pthread_mutex_t _mmtx;

成员函数

构造函数：

今天我们知道，对于服务器而言，需要指定端口号。我们在初始化服务器的时候，对锁和线程都进行初始化。

UdpServer(uint16_t port = default_port) : _port(port)
{
    pthread_mutex_init(&_mmtx,nullptr);

    p = new Thread(1,std::bind(&UdpServer::Recv,this));
    c = new Thread(1,std::bind(&UdpServer::BroadCast,this));
}

析构函数

析构函数完成资源释放，不要忘记等待进程。

~UdpServer()
{
    pthread_mutex_destroy(&_mmtx);

    p->join();
    c->join();

    delete p;
    delete c;
}

Start

void start()
{
    // 1 创建socket接口，打开网络文件
    _sock = socket(AF_INET, SOCK_DGRAM, 0);
    if (_sock < 0)
    {
        std::cerr << "create socket error" << std::endl;
        exit(SOCKED_ERR);
    }
    std::cout << "create socket success" << std::endl;

    // 2 给服务器指明IP地址和Port
    struct sockaddr_in local;
    memset(&local, sizeof(local), 0);

    local.sin_family = AF_INET;
    local.sin_port = htons(_port);      // host to network short
    local.sin_addr.s_addr = INADDR_ANY; // bind本主机任意ip
    if (bind(_sock, (sockaddr *)&local, sizeof(local)) < 0)
    {
        std::cerr << "bind socket error" << std::endl;
        exit(BIND_ERR);
    }
    std::cout << "bind socket success" << std::endl;

    c->run();
    p->run();
}

addUser

因为UDP不面向连接，要构建一个聊天室，就必须要记录下曾经所有给服务器发送过消息的主机，这样才能转发信息。因为是向map里插入数据，本身是线程不安全的，需要加锁。

void addUser(std::string &name, sockaddr_in &peer)
{
    LockGuard lg(&_mmtx);
    _users[name] = peer;
}

recv
接口细节我们已经在上一节着重说过，这里不做赘述。

void Recv()
{
    char buffer[1024];
    // 网络服务都是循环！
    while (true)
    {
        // recv里有两个输出型参数
        sockaddr_in peer;
        socklen_t len = sizeof(peer);
        int n = recvfrom(_sock, buffer, sizeof(buffer) - 1, 0, (sockaddr *)&peer, &len);
        if (n > 0)
        {
            buffer[n] = 0;
        }
        else
        {
            continue;
        }

        // 发消息人的ip和port
        std::string client_ip = inet_ntoa(peer.sin_addr); // net->ascii
        uint16_t client_port = ntohs(peer.sin_port);

        // 构造了发消息人的姓名
        std::string name = client_ip + "-" + to_string(client_port);

        addUser(name, peer);
        rq.push(name + buffer);
    }
}

broadcast
这里我们用了一个临时数组来存储要广播发送的主机，再向这个数组内的所有主机发送队列里的消息。那这样做有什么好处呢？

先把用户拷贝走，再让这个线程去独立发送数据，这样可以让使用map的其他线程更快得到锁，因为拷贝的过程要比发送数据快得多。

void BroadCast()
{
    while (true)
    {
        std::string send_string;
        rq.pop(&send_string);

        std::vector<sockaddr_in> v; // 临时数组，存放要发给那些人

        {
            LockGuard lg(&_mmtx);
            for (auto user : _users)
            {
                v.push_back(user.second);
            }
        }

        for (auto user : v)
        {
            sendto(_sock, send_string.c_str(), send_string.size(), 0, (sockaddr *)&user, sizeof(user));
        }
    }
}

编写UDP客户端

客户端的编写和服务器端编写大同小异。有一个地方需要我们注意一下，套接字编程需要绑定ip和端口号，我们在服务端也确实这样做了，那么客户端需不需要bind端口号和ip呢

答案是肯定需要，只是这个工作不是由我们来干了，而是操作系统替我们做了，在第一次发送的时候自动绑定ip和端口号。

原因也是很好想的，对于客户端来说，端口号都是随机的，自然应该让操作系统帮忙。

void *recv(void *args)
{
    int sock = *(static_cast<int *>(args));
    char buffer[1024];
    sockaddr_in temp;
    socklen_t len = sizeof(temp);

    while (true)
    {
        int n = recvfrom(sock, buffer, sizeof(buffer) - 1, 0, (sockaddr *)&temp, &len);
        if (n > 0)
        {
            buffer[n] = 0;
            std::cout << buffer << std::endl;
        }
    }
}

int main(int argc, char *argv[])
{
    if (argc != 3)
    {
        std::cout << "请输入程序名 + 对方ip + 对方端口号" << std::endl;
        exit(USAGE_ERR);
    }
    std::string serverip = argv[1];
    uint16_t serverport = atoi(argv[2]);

    int sock = socket(AF_INET, SOCK_DGRAM, 0);
    if (sock < 0)
    {
        std::cerr << "create socket error" << std::endl;
        exit(SOCKED_ERR);
    }
    // 系统在我们第一次调用系统调用发送的时候，会自动给我绑定ip和端口号

    // 填充server的信息
    struct sockaddr_in server;
    memset(&server, 0, sizeof(server));

    server.sin_family = AF_INET;
    server.sin_port = htons(serverport);
    server.sin_addr.s_addr = inet_addr(serverip.c_str());

    //
    pthread_t r_id;
    pthread_create(&r_id, nullptr, recv, (void*)&sock);

    while(true)
    {
        std::string message;
        std::cout << "请输入..." << std::endl;
        getline(std::cin,message);

        sendto(sock,message.c_str(),message.size(),0,(sockaddr*)&server,sizeof(server));
    }

    pthread_join(r_id,nullptr);

    return 0;
}