从主机名和端口创建套接字，但 ::select() 指示没有数据

Question

我正在尝试使用主机名和端口创建套接字。然而，

::select()

表示套接字没有数据。

有人可以帮忙吗？

下面是独立的示例（和 Godbolt 链接）。

https://godbolt.org/z/r7G3aczdK

#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/ioctl.h>
#include <net/if.h>
#include <netinet/tcp.h>

#include <string>
#include <iostream>

bool isSocketReady(int sockfd)
{
    fd_set rfds;
    struct timeval tv;

    FD_ZERO(&rfds);
    FD_SET(0, &rfds);

    tv.tv_sec = 2;
    tv.tv_usec = 0;

    const int retval = ::select(sockfd, &rfds, NULL, NULL, &tv);

    if (retval == -1)
    {
        printf("select() error");
        return false;
    }
    else if (retval)
    {
        printf("Data is available now.\n");
        return true;
    }
    else
    {
        printf("No data within timeout.\n");
        return false;
    }
}

int connectByHostName(const std::string& host, int port)
{
    int sockfd = -1;

    struct addrinfo hints;
    memset(&hints, 0, sizeof(hints));
    hints.ai_flags = AI_ADDRCONFIG | AI_NUMERICSERV;
    hints.ai_family = AF_UNSPEC;
    hints.ai_socktype = SOCK_STREAM;

    const std::string port_str = std::to_string(port);

    struct addrinfo* res{nullptr};
    const int r =  getaddrinfo(host.c_str(), port_str.c_str(), &hints, &res);
    
    if(r < 0)
    {
        return -1;
    }

    for (struct addrinfo* address = res; address != nullptr; address = address->ai_next)
    {
        sockfd = socket(address->ai_family, address->ai_socktype, address->ai_protocol);
        
        if (sockfd < 0)
        {
            continue;
        }

        int flag = 1;
        setsockopt(sockfd, IPPROTO_TCP, TCP_NODELAY, (char*) &flag, sizeof(flag));

        if(::connect(sockfd, address->ai_addr, address->ai_addrlen) < 0)
        {
            continue;
        }
        else if(isSocketReady(sockfd))
        {
            break;
        }            
    }

    freeaddrinfo(res);
    return sockfd;
}

int main()
{
    int sock = connectByHostName("example.com", 80);
    std::cout << sock << std::endl;
}

Answer 1

您拨打的电话

select()

不正确。

您需要更改此：

FD_SET(0, &rfds);

对此：

FD_SET(sockfd, &rfds);

并更改此：

::select(sockfd, ...);

对此：

::select(sockfd+1, ...); // or 0 on Windows

话虽如此，在您显示的代码中在

isSocketReady()

之后使用

connect()

是没有意义的。 HTTP 和 WebSocket 协议都要求您先发送请求，然后才能读取任何数据，但您没有这样做，因此套接字永远不会报告为可读。

如果您尝试使用

isSocketReady()

来检查

connect()

是否成功，那么这也是错误的，原因有两个：

• 您必须测试套接字的可写性而不是可读性。

• 您没有将套接字置于非阻塞模式（

  TCP_NODELAY

  不是用于此目的），因此

  connect()

  将阻塞调用线程，直到连接完全建立或失败。因此您不能使用

  select()

  来检查该条件。要将套接字置于非阻塞模式，您必须执行以下操作之一（取决于您的代码运行的平台）：
  • 在

    SOCK_NONBLOCK

    调用上使用

    socket()

    标志创建套接字。
  • 使用

    fctrl(F_SETFL)

    启用套接字上的

    O_NONBLOCK

    标志。
  • 使用

    ioctl()

    (posix) 或

    ioctlsocket()

    (Windows) 启用套接字上的

    FIONBIO

    选项。