python 中的 select 和 ssl

Question

我有一个使用 select.select() 的服务器应用程序，现在我尝试向其中添加 SSL，但是在侦听“原始”套接字时出现以下错误：

ValueError: file descriptor cannot be a negative integer (-1)

所以我想我应该使用 ssl而是由 ssl.wrap_socket 在 select 中返回的流。这样做，它不会返回任何错误，但也不起作用 - 我不太确定问题是什么，我做了很多研究并遇到了类似问题的帖子，但我发现没有解决这个问题还没有。

非常感谢任何帮助。

Answer 1

将 SSL 套接字与 select() 一起使用并不像乍看起来那么简单。虽然它们工作得很好，因为当你给它一个错误时它不会抛出错误，但如果你只是像普通套接字一样使用它们，你迟早会遇到一些奇怪的情况。

由于 select() 需要文件描述符，因此它将获取原始套接字。但即使原始套接字变得可读，也不意味着您将从 SSL 套接字获取数据。您需要使用非阻塞套接字（无论如何，在使用 select() 时这是一个好主意），并且如果它抛出 SSL_ERROR_WANT_READ （SSL 相当于EWOULDBLOCK）。

另一个问题是，如果您向另一端的连接写入 2048 字节，则您一端的 select() 返回。但是，如果您只从 SSL 套接字读取 1024 字节，则 SSL 套接字内部可能会读取更多数据，并且下一个 select() 将不会返回，即使会有更多数据要读取。读取，可能会导致连接死锁。这是因为 select() 使用的原始套接字没有任何数据，因为它已经在 SSL 套接字的缓冲区中。

我想到的第一个解决方案是读取更多数据，直到读取抛出 SSL_ERROR_WANT_READ ，从而清空缓冲区。但是，如果另一端生成数据的速度比您处理数据的速度快，则最终会导致所有其他连接处于饥饿状态，直到这一端完成数据生成。

您可以通过调用 sslsock.pending() 来查看 SSL 套接字持有多少缓冲数据。那么，更好的方法是首先对一定数量的数据进行一次读取，检查待处理的数据量，然后针对该数量的数据发出第二次读取，从而清空缓冲区，而不会导致更多的读取。

SSL_pending()（幕后的 C 函数）的手册页也这样说：

SSL_pending() 仅考虑当前正在处理的 TLS/SSL 记录中的字节（如果有）。如果设置了 SSL 对象的 read_ahead 标志，则可能已读取包含更多 TLS/SSL 记录的附加协议字节；这些会被 SSL_pending() 忽略

根据我的理解，这意味着如果设置了 read_ahead，您需要重复第二步，直到 SSL_pending() 返回 0。我很确定 python 没有设置 read_ahead，但安全总比后悔好，所以我在示例代码中包含了循环。

我对此不太熟悉，但这样的事情应该有效：

# Put the SSL socket to non-blocking mode
sslsock.setblocking(0)

while True:
    r, w, e = select.select([sslsock], [], [])
    if sslsock in r:
        try:
            data = sslsock.recv(1024)
        except ssl.SSLError as e:
            # Ignore the SSL equivalent of EWOULDBLOCK, but re-raise other errors
            if e.errno != ssl.SSL_ERROR_WANT_READ:
                raise
            continue
        # No data means end of file
        if not data:
            break
        # Drain the SSL socket's internal buffer.
        # If you want to remove the loop, make sure you don't call recv()
        # with a 0 length, since that could cause a read to the raw socket.
        data_left = sslsock.pending()
        while data_left:
            data += sslsock.recv(data_left)
            data_left = sslsock.pending()
        # Process the data
        process(data)

Answer 2

正如 Marius 指出的那样， select.select() 适用于 SSL 套接字，我仍然不知道是什么导致了我的无声错误，但当我认为它是 SSL + select() 时，我就急了。这样这个问题就得到了解答。