mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-23 01:46:35 +03:00
110 lines
3.2 KiB
Python
110 lines
3.2 KiB
Python
"""
|
|
Helper module around the system's libssl library if available for IGE mode.
|
|
"""
|
|
import ctypes
|
|
import ctypes.util
|
|
import logging
|
|
import os
|
|
|
|
__log__ = logging.getLogger(__name__)
|
|
|
|
|
|
lib = ctypes.util.find_library('ssl')
|
|
|
|
# This is a best-effort attempt at finding the full real path of lib.
|
|
#
|
|
# Unfortunately ctypes doesn't tell us *where* it finds the library,
|
|
# so we have to do that ourselves.
|
|
try:
|
|
# This is not documented, so it could fail. Be on the safe side.
|
|
import ctypes.macholib.dyld
|
|
paths = ctypes.macholib.dyld.DEFAULT_LIBRARY_FALLBACK
|
|
except (ImportError, AttributeError):
|
|
paths = [
|
|
os.path.expanduser("~/lib"),
|
|
"/usr/local/lib",
|
|
"/lib",
|
|
"/usr/lib",
|
|
]
|
|
|
|
for path in paths:
|
|
if os.path.isdir(path):
|
|
for root, _, files in os.walk(path):
|
|
if lib in files:
|
|
# Manually follow symbolic links on *nix systems.
|
|
# Fix for https://github.com/LonamiWebs/Telethon/issues/1167
|
|
lib = os.path.realpath(os.path.join(root, lib))
|
|
break
|
|
|
|
|
|
try:
|
|
if not lib:
|
|
raise OSError('no library called "ssl" found')
|
|
|
|
_libssl = ctypes.cdll.LoadLibrary(lib)
|
|
except OSError as e:
|
|
# See https://github.com/LonamiWebs/Telethon/issues/1167
|
|
# Sometimes `find_library` returns improper filenames.
|
|
__log__.info('Failed to load %s: %s (%s)', lib, type(e), e)
|
|
_libssl = None
|
|
|
|
if not _libssl:
|
|
decrypt_ige = None
|
|
encrypt_ige = None
|
|
else:
|
|
# https://github.com/openssl/openssl/blob/master/include/openssl/aes.h
|
|
AES_ENCRYPT = ctypes.c_int(1)
|
|
AES_DECRYPT = ctypes.c_int(0)
|
|
AES_MAXNR = 14
|
|
|
|
class AES_KEY(ctypes.Structure):
|
|
"""Helper class representing an AES key"""
|
|
_fields_ = [
|
|
('rd_key', ctypes.c_uint32 * (4 * (AES_MAXNR + 1))),
|
|
('rounds', ctypes.c_uint),
|
|
]
|
|
|
|
def decrypt_ige(cipher_text, key, iv):
|
|
aes_key = AES_KEY()
|
|
key_len = ctypes.c_int(8 * len(key))
|
|
key = (ctypes.c_ubyte * len(key))(*key)
|
|
iv = (ctypes.c_ubyte * len(iv))(*iv)
|
|
|
|
in_len = ctypes.c_size_t(len(cipher_text))
|
|
in_ptr = (ctypes.c_ubyte * len(cipher_text))(*cipher_text)
|
|
out_ptr = (ctypes.c_ubyte * len(cipher_text))()
|
|
|
|
_libssl.AES_set_decrypt_key(key, key_len, ctypes.byref(aes_key))
|
|
_libssl.AES_ige_encrypt(
|
|
ctypes.byref(in_ptr),
|
|
ctypes.byref(out_ptr),
|
|
in_len,
|
|
ctypes.byref(aes_key),
|
|
ctypes.byref(iv),
|
|
AES_DECRYPT
|
|
)
|
|
|
|
return bytes(out_ptr)
|
|
|
|
def encrypt_ige(plain_text, key, iv):
|
|
aes_key = AES_KEY()
|
|
key_len = ctypes.c_int(8 * len(key))
|
|
key = (ctypes.c_ubyte * len(key))(*key)
|
|
iv = (ctypes.c_ubyte * len(iv))(*iv)
|
|
|
|
in_len = ctypes.c_size_t(len(plain_text))
|
|
in_ptr = (ctypes.c_ubyte * len(plain_text))(*plain_text)
|
|
out_ptr = (ctypes.c_ubyte * len(plain_text))()
|
|
|
|
_libssl.AES_set_encrypt_key(key, key_len, ctypes.byref(aes_key))
|
|
_libssl.AES_ige_encrypt(
|
|
ctypes.byref(in_ptr),
|
|
ctypes.byref(out_ptr),
|
|
in_len,
|
|
ctypes.byref(aes_key),
|
|
ctypes.byref(iv),
|
|
AES_ENCRYPT
|
|
)
|
|
|
|
return bytes(out_ptr)
|