代理中的非阻塞套接字和 poll() 怪癖 - 新手

Question

我是一个涉足 C 的新手，我的小项目是编写一个简单的 SOCKS4 代理。多亏了这里的帮助，我才得以在日常工作中使用非阻塞套接字和 poll()。但是在这一点上我似乎有两个问题：

1. 如果传入的 Socket rcvSocket 关闭，则传出的 Socket dstSocket 不会关闭，反之亦然。我不在循环中检查这个，但我不知道如何。我尝试将 POLLHUP 作为 revents，但这似乎不起作用。正常检查似乎是 recv() 是否返回 0，但这对非阻塞套接字也有效吗？如果是这样，它如何与 revents 一起工作，我似乎无法弄清楚我会把它放在哪里，因为如果 POLLIN | POLLPRI 已设置，在我看来，recv() 永远不应该返回 0？另外我不明白 POLLIN 和 POLLPRI 之间的确切区别是什么，在我看来，这两种情况都只是检查“数据可供阅读”？

2. 代理似乎适用于我用 netcat 测试过的连接。但是，如果我使用浏览器，它会说（当我定位一个网站时）我是否要保存“二进制数据”。我检查了wireshark中的数据，从服务器接收到的数据似乎被正确地逐字节转发给客户端。如果有人可能知道为什么这个程序会发生这种情况，那就太好了:)

附上相关代码（小心编程新手）：

 fds[1].fd = dstSocket;
 fds[0].fd = rcvSocket;
 fds[1].events = POLLIN | POLLPRI | POLLHUP;
 fds[0].events = POLLIN | POLLPRI | POLLHUP;

 timer = poll(fds, 2, timeout_msecs); /* i dont use this yet */

 fcntl(rcvSocket, F_SETFL, O_NONBLOCK);
 fcntl(dstSocket, F_SETFL, O_NONBLOCK);

 while (1 == 1)
 {
     if (fds[0].revents & POLLIN | POLLPRI)
     {
        recvMsgSize = recv(rcvSocket, rcvBuffer, RCVBUFSIZE, 0);
        if (recvMsgSize > 0) {send(dstSocket, rcvBuffer, recvMsgSize, 0);}
     }
     if (fds[1].revents & POLLIN | POLLPRI)
     { 
        sndMsgSize = recv(dstSocket, sndBuffer, RCVBUFSIZE, 0);
        if (sndMsgSize > 0) { send(rcvSocket, sndBuffer, sndMsgSize, 0);}
     }         

     if ((fds[0].revents & POLLHUP) || (fds[1].revents & POLLHUP))
     {
        close(rcvSocket);
        close(dstSocket);
     }
} 

Answer 1

recv() 在干净的远程关闭时返回 0 - 即使对于非阻塞套接字也是如此。在这种情况下，POLLIN 将被返回 - 远程端已关闭套接字的通知被视为“可读”事件。

您不需要将POLLPRI 用于 SOCKS / HTTP 连接 - 它表示 TCP“紧急数据”，这些协议不使用（或者实际上，根本不使用）。

---

除了你的直接问题，你还需要做更多的事情来实现一个可靠的代理：

1. 您需要在每个循环中调用poll()，而不仅仅是一次。您编写它的方式是忙循环，这通常被认为是不可接受的做法。
2. 您应该使用signal(SIGPIPE, SIG_IGN); 将SIGPIPE 的处置设置为忽略。这使您可以优雅地处理写入失败。
3. 您应该检查send() 的结果。请注意，它可以写入少于您请求的数量 - 在这种情况下，您必须保持未发送的数据缓冲，返回到 poll() 并尝试再次发送剩余的数据，如果 POLLOUT 在套接字上引发。您只想在剩余未发送数据的情况下请求POLLOUT，因此您需要确保在每次poll() 调用之前正确设置.events。
4. 如果recv() 或send() 返回的值小于0，您应该检查errno。EINTR 和EWOULDBLOCK 应该被忽略；任何其他错误都应视为连接失败。
5. 当一个套接字关闭时，您不应立即关闭两个方向 - 您必须支持非对称关闭。这意味着当你看到fds[0]已经被远端关闭时，你应该调用shutdown(fds[1], SHUT_WR);，反之亦然；只有当 两个 都已关闭（或发生连接失败）时，您才应该在两个文件描述符上调用 close() 并结束。