hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-22 04:56:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactor error handling in deserialize method to use specific exception classes

Browse Source
This commit is contained in:
Jahongir Qurbonov 2025-06-03 09:41:06 +05:00
parent 9c5a6af608
commit 4315ddbb57
No known key found for this signature in database
GPG Key ID: 256976CED13D5F2D
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
client/src/telethon/_impl/mtproto/mtp/plain.py
View File

@ -2,7 +2,16 @@ import struct
from typing import Optional from typing import Optional
from ..utils import check_message_buffer from ..utils import check_message_buffer
from .types import Deserialization, MsgId, Mtp, RpcResult from .types import (
BadAuthKeyError,
BadMsgIdError,
Deserialization,
MsgId,
Mtp,
NegativeLengthError,
RpcResult,
TooLongMsgError,
)
class Plain(Mtp): class Plain(Mtp):
@ -38,17 +47,17 @@ class Plain(Mtp):
auth_key_id, msg_id, length = struct.unpack_from("<qqi", payload) auth_key_id, msg_id, length = struct.unpack_from("<qqi", payload)
if auth_key_id != 0: if auth_key_id != 0:
raise ValueError(f"bad auth key, expected: 0, got: {auth_key_id}") raise BadAuthKeyError(got=auth_key_id, expected=0)
# https://core.telegram.org/mtproto/description#message-identifier-msg-id # https://core.telegram.org/mtproto/description#message-identifier-msg-id
if msg_id <= 0 or (msg_id % 4) != 1: if msg_id <= 0 or (msg_id % 4) != 1:
raise ValueError(f"bad msg id, got: {msg_id}") raise BadMsgIdError(got=msg_id)
if length < 0: if length < 0:
raise ValueError(f"bad length: expected >= 0, got: {length}") raise NegativeLengthError(got=length)
if 20 + length > (lp := len(payload)): if 20 + length > (lp := len(payload)):
raise ValueError(f"message too short, expected: {20 + length}, got {lp}") raise TooLongMsgError(got=length, max_length=lp - 20)
return [RpcResult(MsgId(0), bytes(payload[20 : 20 + length]))] return [RpcResult(MsgId(0), bytes(payload[20 : 20 + length]))]