mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-09-10 11:15:27 +00:00
Code Issues Releases Wiki Activity

Merge IGE and CTR into a single class (AES)

Browse Source
This commit is contained in:
Dan
2018-01-28 01:44:38 +01:00
parent 0dd5843473
commit 2b7425019b
7 changed files with 98 additions and 112 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pyrogram/client/client.py
View File

@@ -46,7 +46,7 @@ from pyrogram.api.types import (
InputPeerEmpty, InputPeerSelf,
InputPeerUser, InputPeerChat, InputPeerChannel
)
from pyrogram.crypto import CTR
from pyrogram.crypto import AES
from pyrogram.session import Auth, Session
from .style import Markdown, HTML
@@ -1633,8 +1633,6 @@ class Client:
)
)
if isinstance(r, types.upload.FileCdnRedirect):
ctr = CTR(r.encryption_key, r.encryption_iv)
cdn_session = Session(
r.dc_id,
self.test_mode,
@@ -1673,7 +1671,7 @@ class Client:
break
# https://core.telegram.org/cdn#decrypting-files
decrypted_chunk = ctr.decrypt(chunk, offset)
decrypted_chunk = AES.ctr_decrypt(chunk, r.encryption_key, r.encryption_iv, offset)
# TODO: https://core.telegram.org/cdn#verifying-files
# TODO: Save to temp file, flush each chunk, rename to full if everything is ok