From 880365cab09dfe1378fc15eed550b383b71139e0 Mon Sep 17 00:00:00 2001
From: YinMo19 <me@yinmo19.top>
Date: Thu, 24 Apr 2025 21:49:39 +0800
Subject: [PATCH] [bug fix] add enc logic.

---
 src/websocket.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/websocket.rs b/src/websocket.rs
index 434849b..dd794e0 100644
--- a/src/websocket.rs
+++ b/src/websocket.rs
@@ -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
     }