blah/blahrs
1
0
Fork 0
mirror of https://github.com/Blah-IM/blahrs.git synced 2025-04-30 16:21:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(blahd/event): do not parse close frame as message

Browse source
This commit is contained in:
oxalica 2024-10-01 07:09:53 -04:00
parent 367f6d2a4b
commit 719c19dc64
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
blahd/src/event.rs
View file

@ -130,7 +130,11 @@ impl Stream for UserEventReceiver {
pub async fn handle_ws(st: Arc<AppState>, ws: &mut WebSocket) -> Result<Infallible> {
let config = &st.config.ws;
let (ws_tx, ws_rx) = ws.split();
let mut ws_rx = ws_rx.map(|ret| ret.and_then(|msg| msg.into_text()).map_err(|_| StreamEnded));
let mut ws_rx = ws_rx.map(|ret| match ret {
Ok(Message::Text(data)) => Ok(data),
Ok(Message::Close(_)) | Err(_) => Err(StreamEnded.into()),
_ => bail!("unexpected message type"),
});
let mut ws_tx = WsSenderWrapper {
inner: ws_tx,
config,