我正在尝试为 connect() 提供超时。我四处搜寻,发现了几篇与此相关的文章。我已经编写了我认为应该有效的代码,但不幸的是我没有收到 getsockopt() 报告的错误。但是当我执行 write() 时,它失败了,错误号为 107 - ENOTCONN。
有几点。我在 Fedora 23 上运行。 connect() 的文档说,对于尚未完成的连接,它应该返回失败,错误号为 EINPROGRESS,但我遇到了 EAGAIN,所以我将其添加到我的检查中。目前,我的套接字服务器正在listen() 调用中将积压设置为零。许多调用都会成功,但失败的调用都会失败,并出现我在 write() 调用中提到的 107 - ENOTCONN。
我希望我只是错过了一些东西,但到目前为止还不知道是什么。
int domain_socket_send(const char* socket_name, unsigned char* buffer,
unsigned int length, unsigned int timeout)
{
struct sockaddr_un addr;
int fd = -1;
int result = 0;
// Create socket.
fd = socket(AF_UNIX, SOCK_STREAM, 0);
if (fd == -1)
{
result = -1;
goto done;
}
if (timeout != 0)
{
// Enabled non-blocking.
int flags;
flags = fcntl(fd, F_GETFL);
fcntl(fd, F_SETFL, flags | O_NONBLOCK);
}
// Set socket name.
memset(&addr, 0, sizeof(addr));
addr.sun_family = AF_UNIX;
strncpy(addr.sun_path, socket_name, sizeof(addr.sun_path) - 1);
// Connect.
result = connect(fd, (struct sockaddr*) &addr, sizeof(addr));
if (result == -1)
{
// If some error then we're done.
if ((errno != EINPROGRESS) && (errno != EAGAIN))
goto done;
fd_set write_set;
struct timeval tv;
// Set timeout.
tv.tv_sec = timeout / 1000000;
tv.tv_usec = timeout % 1000000;
unsigned int iterations = 0;
while (1)
{
FD_ZERO(&write_set);
FD_SET(fd, &write_set);
result = select(fd + 1, NULL, &write_set, NULL, &tv);
if (result == -1)
goto done;
else if (result == 0)
{
result = -1;
errno = ETIMEDOUT;
goto done;
}
else
{
if (FD_ISSET(fd, &write_set))
{
socklen_t len;
int socket_error;
len = sizeof(socket_error);
// Get the result of the connect() call.
result = getsockopt(fd, SOL_SOCKET, SO_ERROR,
&socket_error, &len);
if (result == -1)
goto done;
// I think SO_ERROR will be zero for a successful
// result and errno otherwise.
if (socket_error != 0)
{
result = -1;
errno = socket_error;
goto done;
}
// Now that the socket is writable issue another connect.
result = connect(fd, (struct sockaddr*) &addr,
sizeof(addr));
if (result == 0)
{
if (iterations > 1)
{
printf("connect() succeeded on iteration %d\n",
iterations);
}
break;
}
else
{
if ((errno != EAGAIN) && (errno != EINPROGRESS))
{
int err = errno;
printf("second connect() failed, errno = %d\n",
errno);
errno = err;
goto done;
}
iterations++;
}
}
}
}
}
// If we put the socket in non-blocking mode then put it back
// to blocking mode.
if (timeout != 0)
{
// Turn off non-blocking.
int flags;
flags = fcntl(fd, F_GETFL);
fcntl(fd, F_SETFL, flags & ~O_NONBLOCK);
}
// Write buffer.
result = write(fd, buffer, length);
if (result == -1)
{
int err = errno;
printf("write() failed, errno = %d\n", err);
errno = err;
goto done;
}
done:
if (result == -1)
result = errno;
else
result = 0;
if (fd != -1)
{
shutdown(fd, SHUT_RDWR);
close(fd);
}
return result;
}
2016 年 4 月 5 日更新:
我突然意识到,也许我需要多次调用 connect() 直到成功,毕竟这是非阻塞 io 而不是异步 io。就像在 read() 上遇到 EAGAIN 后有数据要读取时必须再次调用 read() 一样。另外,我发现了以下SO问题:
使用 select() 进行非阻塞套接字连接始终返回 1
其中 EJP 的回答说您需要发出多个 connect() 。另外,来自 EJP 参考文献:
它似乎表明您需要发出多个 connect() 。我已经修改了这个问题中的代码片段以调用 connect() 直到成功。我可能仍然需要对可能更新传递给 select() 的超时值进行更改,但这不是我直接的问题。
多次调用 connect() 似乎已经解决了我原来的问题,即我在调用 write() 时收到 ENOTCONN,我猜是因为套接字未连接。但是,您可以从代码中看到我正在跟踪 select 循环的次数,直到 connect() 成功。我见过这个数字达到数千。这让我担心我正处于繁忙的等待循环中。为什么套接字可写,即使它没有处于 connect() 会成功的状态?调用 connect() 是否清除了可写状态,并且由于某种原因它被操作系统再次设置,或者我真的处于繁忙的等待循环中?
谢谢,
缺口