Granian项目中WebSocket Pong消息处理异常问题分析

在Python Web服务框架Granian的最新版本中，发现了一个与WebSocket协议处理相关的重要问题。该问题主要影响使用ASGI接口的WebSocket通信场景，特别是当客户端发送单向Pong消息时的异常处理机制。

问题现象

当客户端在WebSocket连接中发送空Pong消息作为心跳保活机制时，Granian服务端会出现两种不同表现：

1. 如果Pong消息在正常通信过程中到达，服务端仅记录警告日志"Unsupported websocket message received Pong([])"
2. 如果Pong消息恰好在服务端等待客户端输入时到达，则会引发ASGI流错误(ASGI flow error)，导致连接异常终止

技术背景

WebSocket协议中，Ping/Pong机制用于连接保活：

• Ping消息由一端发送，另一端必须回应Pong消息
• Pong消息也可以主动发送作为心跳指示
• 协议允许空内容的Ping/Pong消息

在标准实现中(如Uvicorn)，单向Pong消息会被正常处理而不影响通信流程。但Granian当前版本对此场景的处理存在缺陷。

问题根源分析

通过代码分析，问题主要源于Granian对WebSocket协议中Pong消息的处理逻辑不够完善：

1. 服务端未正确区分请求-响应式Pong(回应Ping)和主动发送的Pong消息
2. 在等待接收消息的状态下，对意外到达的Pong消息处理不当，导致ASGI状态机异常
3. 错误传播机制过于严格，将可安全忽略的协议行为视为致命错误

影响范围

该问题影响以下环境组合：

• Granian 1.4.1版本
• Python 3.10/3.11环境
• 使用ASGI接口
• 任何发送单向Pong消息的客户端(如.NET WebSocket实现)

临时解决方案

在官方修复发布前，开发者可以采用以下临时解决方案：

try:
    data = await websocket.receive_bytes()
except RuntimeError as e:
    if str(e) == "ASGI flow error" and \
       websocket.client_state == "connected" and \
       websocket.application_state == "connected":
        continue  # 忽略可恢复的错误
    raise  # 重新抛出其他异常

技术建议

对于WebSocket服务实现，建议：

1. 完整实现WebSocket协议的所有消息类型处理
2. 对协议允许的非致命性异常情况应有容错处理
3. 保持与主流实现(Uvicorn等)的行为一致性
4. 区分日志级别，避免将预期内行为记录为警告

该问题预计将在Granian的下个修补版本中得到修复，届时服务端将正确忽略单向Pong消息而不会中断连接。