From 003b6aebb86f9c7ff131b067f555a45c3103f248 Mon Sep 17 00:00:00 2001
From: habcawa <212562656+habcawa@users.noreply.github.com>
Date: Wed, 21 May 2025 19:23:09 +0200
Subject: [PATCH] fix: resolve flagged issues in review by Lonami

---
 telethon/client/telegrambaseclient.py | 2 +-
 telethon/network/mtprotosender.py     | 2 +-
 telethon/network/mtprotostate.py      | 4 +++-
 3 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/telethon/client/telegrambaseclient.py b/telethon/client/telegrambaseclient.py
index 62160857..31ad135f 100644
--- a/telethon/client/telegrambaseclient.py
+++ b/telethon/client/telegrambaseclient.py
@@ -776,7 +776,7 @@ class TelegramBaseClient(abc.ABC):
     def _tmp_auth_key_callback(self: 'TelegramClient', tmp_auth_key):
         """
         Callback from the sender whenever it needed to generate a
-        new authorization key. This means we are not authorized.
+        new temporary authorization key. This means we are not authorized.
         """
         self.session.tmp_auth_key = tmp_auth_key
         self.session.save()
diff --git a/telethon/network/mtprotosender.py b/telethon/network/mtprotosender.py
index f813af63..a9f9a170 100644
--- a/telethon/network/mtprotosender.py
+++ b/telethon/network/mtprotosender.py
@@ -260,7 +260,7 @@ class MTProtoSender:
                 # establish tmp_auth_key here, but make sure to bind to the auth_key later
                 try:
                     if not await self._try_gen_tmp_auth_key(attempt):
-                            continue  # keep retrying until we have the tmp auth key
+                        continue  # keep retrying until we have the tmp auth key
                 except (IOError, asyncio.TimeoutError) as e:
                     # Sometimes, specially during user-DC migrations,
                     # Telegram may close the connection during auth_key
diff --git a/telethon/network/mtprotostate.py b/telethon/network/mtprotostate.py
index 692cb7eb..b3d966f1 100644
--- a/telethon/network/mtprotostate.py
+++ b/telethon/network/mtprotostate.py
@@ -171,7 +171,9 @@ class MTProtoState:
         bind = bytes(bind)
         assert len(bind) == 40
 
-        payload = os.urandom(int(128/8)) + struct.pack('<qii', msg_id, seq_no, len(bind)) + bind
+        # 128 bit random + msg_id + seq_no + len(bind) + bind
+        payload = os.urandom(128//8) + struct.pack('<qii', msg_id, seq_no, len(bind)) + bind
+        # 16-byte alignment
         padding = os.urandom(len(payload) % 16)
 
         msg_key = sha1(payload).digest()[4:20]