Avoid protocol errors in ASGI and RSGI on ws pong recv (#321)

emmett-framework · Jun 7, 2024 · 8378b36 · 8378b36
1 parent 2a006a0
commit 8378b36
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/asgi/io.rs b/src/asgi/io.rs
@@ -428,7 +428,7 @@ impl ASGIWebsocketProtocol {
             if let Some(ws) = &mut *(transport.lock().await) {
                 while let Some(recv) = ws.next().await {
                     match recv {
-                        Ok(Message::Ping(_)) => continue,
+                        Ok(Message::Ping(_) | Message::Pong(_)) => continue,
                         Ok(message @ Message::Close(_)) => {
                             closed.store(true, atomic::Ordering::Relaxed);
                             return ws_message_into_py(message);

diff --git a/src/rsgi/io.rs b/src/rsgi/io.rs
@@ -218,7 +218,7 @@ impl RSGIWebsocketTransport {
             if let Ok(mut stream) = transport.try_lock() {
                 while let Some(recv) = stream.next().await {
                     match recv {
-                        Ok(Message::Ping(_)) => continue,
+                        Ok(Message::Ping(_) | Message::Pong(_)) => continue,
                         Ok(message) => return message_into_py(message),
                         _ => break,
                     }