From cf06939a55f37ffa05b0bca1155981e1fa09f206 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Tue, 27 Dec 2022 22:23:05 +0100
Subject: [PATCH] Workaround proxy sockets not timing out properly

---
 pyrogram/connection/transport/tcp/tcp.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pyrogram/connection/transport/tcp/tcp.py b/pyrogram/connection/transport/tcp/tcp.py
index 6aff86af..22b367ae 100644
--- a/pyrogram/connection/transport/tcp/tcp.py
+++ b/pyrogram/connection/transport/tcp/tcp.py
@@ -20,6 +20,7 @@ import asyncio
 import ipaddress
 import logging
 import socket
+
 import socks
 
 log = logging.getLogger(__name__)
@@ -58,6 +59,8 @@ class TCP:
                 password=proxy.get("password", None)
             )
 
+            self.socket.settimeout(TCP.TIMEOUT)
+
             log.info("Using proxy %s", hostname)
         else:
             self.socket = socket.socket(
@@ -65,7 +68,7 @@ class TCP:
                 else socket.AF_INET
             )
 
-        self.socket.setblocking(False)
+            self.socket.setblocking(False)
 
     async def connect(self, address: tuple):
         try: