From f54b085dded5098d7b080d7d0c0409d6cd2e7cae Mon Sep 17 00:00:00 2001 From: nex Date: Wed, 29 Nov 2023 15:50:13 +0000 Subject: [PATCH] Add timeout to ytdl proxy check --- cogs/other.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/cogs/other.py b/cogs/other.py index 63c34f9..9919ea2 100644 --- a/cogs/other.py +++ b/cogs/other.py @@ -1001,7 +1001,10 @@ class OtherCog(commands.Cog): description = "" proxy_url = "socks5://localhost:1090" try: - proxy_down = await self.check_proxy("socks5://localhost:1090") + proxy_down = await asyncio.wait_for( + self.check_proxy("socks5://localhost:1090"), + timeout=10 + ) if proxy_down > 0: if proxy_down == 1: description += ":warning: (SHRoNK) Proxy check leaked IP - trying backup proxy.\n" @@ -1010,7 +1013,10 @@ class OtherCog(commands.Cog): else: description += ":warning: (SHRoNK) Unknown proxy error - trying backup proxy.\n" - proxy_down = await self.check_proxy("socks5://localhost:1080") + proxy_down = await asyncio.wait_for( + self.check_proxy("socks5://localhost:1080"), + timeout=10 + ) if proxy_down > 0: if proxy_down == 1: description += ":warning: (NexBox) Proxy check leaked IP..\n"