Add timeout when connecting

pyrogram/connection/connection.py

@@ -25,6 +25,7 @@ log = logging.getLogger(__name__)
 
 
 class Connection:
+    TIMEOUT = 10
     MAX_RETRIES = 3
 
     MODES = {
@@ -49,8 +50,8 @@ class Connection:
 
             try:
                 log.info("Connecting...")
-                await self.protocol.connect(self.address)
-            except OSError:
+                await asyncio.wait_for(self.protocol.connect(self.address), Connection.TIMEOUT)
+            except (OSError, asyncio.TimeoutError):
                 self.protocol.close()
                 await asyncio.sleep(1)
             else: