From 2b0746a140fcc7b2ea152ae5b5fe377b1f6f44b0 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Sun, 17 Jun 2018 18:33:23 +0200
Subject: [PATCH] Add timeout on recv loop

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

diff --git a/pyrogram/session/session.py b/pyrogram/session/session.py
index c3dec02e..60276947 100644
--- a/pyrogram/session/session.py
+++ b/pyrogram/session/session.py
@@ -334,7 +334,10 @@ class Session:
         log.info("RecvTask started")
 
         while True:
-            packet = await self.connection.recv()
+            try:
+                packet = await asyncio.wait_for(self.connection.recv(), self.connection.TIMEOUT)
+            except asyncio.TimeoutError:
+                packet = None
 
             if packet is None or len(packet) == 4:
                 if packet: