c - 在 2 个对等点之间传输文件时防止超时

Question

我正在使用非阻塞套接字将文件从服务器传输到客户端。有些文件可能很大(数百兆字节)。一个单独的线程处理每个客户端。

while (more_to_read)
{
    written = write(fd, ...);

    if (written == 0)
    {
        struct pollfd wait = {
            .fd = fd,
            .events = POLLOUT,
            .revents = 0
        };
        int status = poll(&wait, 1, TIMEOUT);
        // ...
    }

    // ...
}

有时 poll() 会超时，服务器会关闭连接(以防止恶意客户端耗尽服务器资源)。我发现即使客户端仍在尝试读取数据时也会发生这种情况。在这种情况下，客户端在调用 read() 时得到 ECONNRESET。

我发现服务器重复发送文件的一些 block ，并且在某些时候套接字可能在几秒钟内仍未准备好写入。我假设这是因为数据还没有发送，相应的内核缓冲区已满。然而，这会导致超时，导致服务器关闭连接。

当我增加 TIMEOUT 的值时，错误发生的频率就会降低。确保不会发生意外超时的最佳方法是什么？

服务器应该在多个平台上运行，所以我不能真正使用特定于操作系统的解决方案。

Answer 1

您永远无法完全保护自己免受 DOS 攻击和意外超时。

我建议您根据已发送的数据量调整允许的超时时间。从 shart 超时开始，并在您发送越来越多的数据时增加它。

DOS 攻击者和失效的客户端可能会打开许多​​连接，但不会费心从中读取数据。你想快速关闭这种连接。

收到数十或数百 MB 数据的客户端应允许更长的超时时间。

例子:

• 初始:500 毫秒
• 1 MB 之后:2 秒
• 100 MB 之后:10 秒