Python TLS Socket：如何提前检测证书问题导致的服务器关闭连接？

问题背景 #

当Python 3编写的TLS客户端未加载服务器要求的客户端证书（或证书无效）时，服务器会直接终止连接，但客户端要到调用SSLSocket.recv()时才会触发ssl.SSLError错误，无法在进入接收循环前提前检测该问题。

示例代码 #

客户端代码（证书加载被注释）

import ssl
import socket

context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
context.verify_mode = ssl.CERT_REQUIRED
context.check_hostname = False
# context.load_cert_chain(certfile='client.crt', keyfile='client.key')
context.load_verify_locations(cafile='ca.crt')
with socket.create_connection(('127.0.0.1', 12345)) as client:
    with context.wrap_socket(client, server_hostname='example.com') as ssock:
        ssock.sendall(b'Hello, world\n')
        while 1:
            data = ssock.recv(1024)
            if not data:
                ssock.close()
                exit()
            do_something(data)

服务器启动命令（要求客户端证书）

openssl s_server -port 12345 -CAfile ca.crt -cert server.crt -key server.key -Verify 1

问题现象 #

服务器会因客户端未提供证书断开连接：

ERROR
00A77D53F87F0000:error:0A0000C7:SSL routines:tls_process_client_certificate:peer did not return a certificate:ssl/statem/statem_srvr.c:3511:
shutting down SSL
CONNECTION CLOSED

但客户端需等到首次调用recv()时才会报错：

Traceback (most recent call last):
  File "client.py", line 13, in <module>
    data = ssock.recv(1024)
           ^^^^^^^^^^^^^^^^
  File "***/python3.11/ssl.py", line 1263, in recv
    return self.read(buflen)
           ^^^^^^^^^^^^^^^^^
  File "***/python3.11/ssl.py", line 1136, in read
    return self._sslobj.read(len)
           ^^^^^^^^^^^^^^^^^^^^^^
ssl.SSLError: [SSL: TLSV13_ALERT_CERTIFICATE_REQUIRED] tlsv13 alert certificate required (_ssl.c:2576)

解决方案 #

核心原因是Python的wrap_socket()默认会延迟TLS握手到第一次I/O操作时执行。要提前检测握手结果，只需手动触发TLS握手：

修改后的客户端代码 #

import ssl
import socket

def do_something(data):
    print(data.decode())

context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
context.verify_mode = ssl.CERT_REQUIRED
context.check_hostname = False
# context.load_cert_chain(certfile='client.crt', keyfile='client.key')
context.load_verify_locations(cafile='ca.crt')
with socket.create_connection(('127.0.0.1', 12345)) as client:
    with context.wrap_socket(client, server_hostname='example.com') as ssock:
        try:
            # 手动触发TLS握手，强制在发送数据前完成握手验证
            ssock.do_handshake()
        except ssl.SSLError as e:
            print(f"TLS握手失败: {e}")
            exit()
        ssock.sendall(b'Hello, world\n')
        while 1:
            data = ssock.recv(1024)
            if not data:
                ssock.close()
                exit()
            do_something(data)

原理说明 #

• do_handshake()会强制立即完成TLS握手流程，任何握手阶段的错误（包括服务器要求客户端证书但未提供的情况）都会直接抛出ssl.SSLError，无需等到后续的recv()操作。
• 原代码中sendall()虽然会触发后台握手，但握手失败的TLS alert可能不会被立即捕获，直到recv()尝试读取服务器返回的响应时才会暴露错误。

内容的提问来源于stack exchange，提问作者RichardLiu