LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-15 02:22:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added the control of unknown exception handling times. If not controlled, I think it is very dangerous and may fall into an infinite loop

Browse Source
This commit is contained in:
super_rocky 2023-08-04 00:19:02 +08:00
parent 6ded164b85
commit caf79da177
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
telethon/network/mtprotosender.py
View File

@ -51,6 +51,7 @@ class MTProtoSender:
self._loggers = loggers self._loggers = loggers
self._log = loggers[__name__] self._log = loggers[__name__]
self._retries = retries self._retries = retries
self._retry_unknown_error = retries
self._delay = delay self._delay = delay
self._auto_reconnect = auto_reconnect self._auto_reconnect = auto_reconnect
self._connect_timeout = connect_timeout self._connect_timeout = connect_timeout
@ -519,6 +520,9 @@ class MTProtoSender:
return return
except Exception as e: except Exception as e:
self._log.exception('Unhandled error while receiving data') self._log.exception('Unhandled error while receiving data')
self._retry_unknown_error = self._retry_unknown_error - 1
if self._retry_unknown_error < 0:
raise
self._start_reconnect(e) self._start_reconnect(e)
return return