rustdesk/hbb_common
1
0
Fork 0
mirror of https://github.com/rustdesk/hbb_common.git synced 2026-04-02 22:16:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

[bug fix] add enc logic.

Browse Source
This commit is contained in:
YinMo19
2025-04-24 21:49:39 +08:00
parent 608eb5983f
commit 880365cab0
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/websocket.rs
View File

@@ -95,7 +95,9 @@ impl WsFramedStream {
// }
}
pub fn set_raw(&mut self) {}
pub fn set_raw(&mut self) {
self.encrypt = None;
}
pub async fn from_tcp_stream(stream: TcpStream, addr: SocketAddr) -> ResultType<Self> {
let ws_stream =
@@ -133,6 +135,10 @@ impl WsFramedStream {
#[inline]
pub async fn send_raw(&mut self, msg: Vec<u8>) -> ResultType<()> {
let mut msg = msg;
if let Some(key) = self.encrypt.as_mut() {
msg = key.enc(&msg);
}
self.send_bytes(Bytes::from(msg)).await
}