mercuryworkshop-mirror/epoxy-tls
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/epoxy-tls.git synced 2025-05-12 14:00:01 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix pong read

Browse source
This commit is contained in:
Toshit Chawda 2024-09-23 17:42:00 -07:00
parent 09b15e3c43
commit 2dfcb0021f
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
2 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
server/src/route.rs
View file

@ -111,6 +111,7 @@ pub async fn route(
ws_upgrade(req, |fut, wsproxy, udp, path| async move {
let mut ws = fut.await.context("failed to await upgrade future")?;
ws.set_max_message_size(CONFIG.server.max_message_size);
ws.set_auto_pong(false);
if wsproxy {
let ws = WebSocketStreamWrapper(FragmentCollector::new(ws));

4
wisp/src/inner.rs
View file

@ -27,6 +27,7 @@ pub(crate) enum WsEvent {
SendPong(Payload<'static>),
WispMessage(Option<Packet<'static>>, Option<Frame<'static>>),
EndFut(Option<CloseReason>),
Noop
}
struct MuxMapValue {
@ -238,6 +239,8 @@ impl<R: WebSocketRead + Send> MuxInner<R> {
return Ok(None);
} else if frame.opcode == OpCode::Ping {
return Ok(Some(WsEvent::SendPong(frame.payload)));
} else if frame.opcode == OpCode::Pong {
return Ok(Some(WsEvent::Noop));
}
if let Some(ref extra_frame) = optional_frame {
@ -342,6 +345,7 @@ impl<R: WebSocketRead + Send> MuxInner<R> {
}
}
}
WsEvent::Noop => {}
}
}