使用 poll 在 C 中进行套接字连接超时

Question

我有一个现有的多线程应用程序，它使用阻塞 connect() 调用。

但是，我想为应用程序引入连接超时，如果服务器在 x 毫秒内没有响应我们的查询，应用程序将停止尝试并给出错误。

但是，我无法弄清楚如何使用 poll 来做到这一点。

@caf 使用 select 的非阻塞连接有很大帮助。但我读到 select 比 poll 慢，因此我想使用 poll。你能告诉我这是真的吗？

我将他的代码从帖子粘贴到这里

int main(int argc, char **argv) {
u_short port;                /* user specified port number */
char *addr;                  /* will be a pointer to the address */
struct sockaddr_in address;  /* the libc network address data structure */
short int sock = -1;         /* file descriptor for the network socket */
fd_set fdset;
struct timeval tv;

if (argc != 3) {
    fprintf(stderr, "Usage %s <port_num> <address>\n", argv[0]);
    return EXIT_FAILURE;
}

port = atoi(argv[1]);
addr = argv[2];

address.sin_family = AF_INET;
address.sin_addr.s_addr = inet_addr(addr); /* assign the address */
address.sin_port = htons(port);            /* translate int2port num */

sock = socket(AF_INET, SOCK_STREAM, 0);
fcntl(sock, F_SETFL, O_NONBLOCK);

connect(sock, (struct sockaddr *)&address, sizeof(address));

FD_ZERO(&fdset);
FD_SET(sock, &fdset);
tv.tv_sec = 10;             /* 10 second timeout */
tv.tv_usec = 0;

if (select(sock + 1, NULL, &fdset, NULL, &tv) == 1)
{
    int so_error;
    socklen_t len = sizeof so_error;

    getsockopt(sock, SOL_SOCKET, SO_ERROR, &so_error, &len);

    if (so_error == 0) {
        printf("%s:%d is open\n", addr, port);
    }
}

close(sock);
return 0;

}

您能帮我使用 poll.js 编写类似的功能吗？

我在 RHEL 上并使用 gcc 4.5.x 版本。

更新：对于当前代码，一旦应用程序创建与服务器的连接，如何将套接字更改为阻塞模式。我无法找到取消设置此 O_NONBLOCK 的方法。 更新 2：fcntl(sockfd, F_SETFL, fcntl(sockfd, F_GETFL) & ~O_NONBLOCK); 一篇文章指出我们可以使用上面的命令来做到这一点。虽然没有得到登录。

Answer 1

如果只有少数文件描述符被轮询，则 select() 和 poll() 相对于其他依赖于系统的轮询 API 的性能是相当的。它们的扩展性都很差，但是当程序只监听几个 fd 时，它们的性能就很好。

其他特定于系统的轮询 API 的优点往往是它们将关联的状态推送到内核中，以便每个轮询系统调用不需要从用户空间到内核空间的状态副本。对于只有少数事件的情况，这种复制和扫描开销是微不足道的，而且这种好处并不存在。

只要坚持使用 select() 除非您需要扩展以处理数百或更多的 fd。

如果您确实要处理许多 fd，并且必须同时支持 Linux 和 Solaris，请考虑使用libevent或libev等库来抽象出特定于内核的高效等待 API。