暂时停止收听套接字

时间:2018-06-10 17:44:16

标签: c++ c sockets network-programming

我使用POSIX套接字api在C ++中编写服务器。

这是GUI应用程序的一部分,需要能够运行 停止并启动服务器监听并向客户端发送数据。

服务器的主要部分看起来基本上都是这样的(我已经排除了很多代码,因为其中一些代码与此问题无关。)

if (listen(listener_fd, backlog) < 0) {
    std::perror("listen");
    exit(EXIT_FAILURE);
}

while (true) {
    /* This part sets up the FD set */
    FD_ZERO(&read_fds);
    FD_SET(0, &read_fds); // stdin (for server commands)
    FD_SET(listener_fd, &read_fds);
    FD_SET(read_pipe, &read_fds);
    for (auto it = client_socks.begin(); it != client_socks.end(); it++) {
        FD_SET(*it, &read_fds); // listen on each of the open client-server sockets
    }

    max_fd = 0;
    if (client_socks.size() > 0) {
        max_fd = *std::max_element(client_socks.begin(), client_socks.end());
    }
    if (listener_fd > max_fd) max_fd = listener_fd;
    if (read_pipe > max_fd) max_fd = read_pipe;

    std::fill_n(in_buf, IN_BUF_LENGTH, 0);

    // wait for input on stdin or any of the sockets, with a timeout of 20s
    sel_ret = select(max_fd + 1, &read_fds, NULL, NULL, &read_timeout);
    if (sel_ret == -1) {
        std::perror("select");
        exit(EXIT_FAILURE);
    } 
    else if (sel_ret) {
        if (FD_ISSET(0, &read_fds)) { /* stdin */
            // send stdin to all clients
            // excl. for brev.
        } 
        else if (FD_ISSET(listener_fd, &read_fds) && serving) { /* listen(...) can accept, but only bother if the server is listening */
            int newclient_sock = accept(listener_fd, (struct sockaddr*)&addr, &addrlen);
            std::cout << "New client: " << newclient_sock << std::endl;
            client_socks.push_back(newclient_sock);
        }
        else if (FD_ISSET(read_pipe, &read_fds)) { /* someone sent some data down the pipe - probably GUI info, like a button press */
            // here i've got a message from the GUI, which
            // is either starting or stopping the server
            // excluded here for brevity
        }
        else if (serving) { /* one of the sockets, but only if the server is listening */
            // here i've got data from a client socket
            // this handles the message that they send to me
            // again, excluded here for brevity
        }
    }
}

或换句话说:

  1. 让套接字侦听连接
  2. 为select()调用设置fds
  3. 使用stdin调用select(),所有连接客户端套接字,以及从GUI发送任何数据的管道
  4. 如果stdin处于活动状态,则将键入的数据发送给所有客户端
  5. 如果侦听器处于活动状态,请将此新客户端附加到客户端列表中。
  6. 如果数据管道处于活动状态,则根据服务器的数据启动或停止服务器。
  7. 如果客户端发送数据,将该数据发送给所有客户端,或者如果已断开连接,则将其从客户端列表中删除。
  8. 转到2。
  9. 我的问题

    我需要能够使服务器基本上停止,直到我告诉它重新开始。澄清一下,停止的意思是:

    • 不听任何客户
    • 无法从任何客户端接收数据
    • 任何已连接的客户端都已断开连接。

    我尝试了什么

    我的第一个想法是使用布尔值serving来跟踪它是否应该运行。

    在while(true)的每个循环开始时,我这样做:

    if (serving) {
        // listen(...), same as I did it originally
    } else {
        shutdown(listener_fd, SHUT_RD);
    }
    

    但那根本不起作用。它的工作非常糟糕,甚至很难说出做了做了什么,所以很抱歉。

    另一个想法是使用close(listener_fd)调用,但当然这只会使fd不再可用。

    我考虑的另一个选择是,在接受连接之前,首先检查是否设置了serving。如果是,则接受连接。接收和发送数据也是如此。这种有效,但没有告知客户我没有服务。

1 个答案:

答案 0 :(得分:3)

套接字不支持&#34;临时&#34;像你要求的那样停用。你需要:

  • close()侦听套接字,然后在准备重新启动时创建一个新的侦听套接字。

  • 保持侦听套接字打开并正常运行,但立即close()任何接受的新客户端。

对于已被接受的客户,您需要单独shutdown() + close()个,可选择先发送一条告别消息,具体取决于您的协议是否允许。