Skip to content

Commit 7232508

Browse files
committed
fix: 修复 Socket 漏洞
1 parent c304b07 commit 7232508

File tree

1 file changed

+4
-15
lines changed

1 file changed

+4
-15
lines changed

core/cluster.py

Lines changed: 4 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -437,6 +437,10 @@ async def connect(self) -> None:
437437
@self.socket.on("connect")
438438
async def _() -> None:
439439
logger.tsuccess("client.success.connected")
440+
if self.wantEnable:
441+
await self.enable()
442+
if self.scheduler:
443+
self.scheduler.resume()
440444

441445
@self.socket.on("disconnect")
442446
async def _() -> None:
@@ -449,21 +453,6 @@ async def _() -> None:
449453
async def _(message: str) -> None:
450454
logger.tinfo("client.info.message", message=message)
451455

452-
@self.socket.on("exception")
453-
async def _(error: str) -> None:
454-
logger.tinfo("client.error.exception", error=error)
455-
456-
@self.socket.on("reconnect")
457-
async def _() -> None:
458-
if self.wantEnable:
459-
await self.enable()
460-
if self.scheduler:
461-
self.scheduler.resume()
462-
463-
@self.socket.on("reconnect_error")
464-
async def _(error: str) -> None:
465-
logger.terror("client.error.reconnect", e=error)
466-
467456
await self.socket.connect(
468457
self.base_url, transports=["websocket"], auth={"token": str(self.token.token)}
469458
)

0 commit comments

Comments
 (0)