fergalmoran/ladybird
1
0
Fork 0
mirror of https://github.com/fergalmoran/ladybird.git synced 2025-12-22 09:19:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

LibWebSocket: Send closing frame when server begins a closing handshake

Browse Source
This commit is contained in:
Tim Ledbetter
2024-10-12 20:51:31 +01:00
committed by Andreas Kling
parent 0b365061d1
commit ebe1dc6dcd
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
Userland/Libraries/LibWebSocket/WebSocket.cpp
View File

@@ -483,11 +483,24 @@ void WebSocket::read_frame()
}
if (op_code == WebSocket::OpCode::ConnectionClose) {
dbgln("Close connection with status: {}", to_underlying(m_state));
if (m_state == InternalState::Closing) {
dbgln("Close connection lmao");
set_state(WebSocket::InternalState::Closed);
notify_close(m_last_close_code, m_last_close_message, true);
discard_connection();
return;
}
send_frame(WebSocket::OpCode::ConnectionClose, {}, true);
set_state(WebSocket::InternalState::Closing);
if (payload.size() > 1) {
m_last_close_code = (((u16)(payload[0] & 0xff) << 8) | ((u16)(payload[1] & 0xff)));
m_last_close_message = ByteString(ReadonlyBytes(payload.offset_pointer(2), payload.size() - 2));
} else {
m_last_close_code = 1000;
m_last_close_message = {};
}
set_state(WebSocket::InternalState::Closing);
return;
}
if (op_code == WebSocket::OpCode::Ping) {