o Xzj{@sddlZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z Gdd d Z Gd d d eZGd d d eZGdddeZGdddeZdS)N)ObfuscatedConnection)AbridgedPacketCodec)IntermediatePacketCodec!RandomizedIntermediatePacketCodec) AESModeCTRc@s8eZdZdZdZddZeddZddZd d Z dS) MTProxyIOz It's very similar to tcpobfuscated.ObfuscatedIO, but the way encryption keys, protocol tag and dc_id are encoded is different. NcCs4|j|_|j|_||j|j|j\|_|_|_dSN) _reader_writer init_header_secret_dc_id packet_codecheader_encrypt_decrypt)self connectionra/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telethon/network/connection/tcpmtproxy.py__init__s zMTProxyIO.__init__cCst|dko |ddk}t|t}|r|std|r!|ddn|}t|dkr-tdd} td }|dd krL|dd |vrL|d d d krLnq0t|}|ddd}tt |dd| }t |dd} tt |dd| } t |dd} t || } t | | } |j |dd<|j ddd d}|dd||dd}| t |dd |dd <|| | fS)Nrz7Only RandomizedIntermediate can be used with dd-secretsrz9MTProxy secret must be a hex-string representing 16 bytes)sPVrGsGET sPOSTsT@s7(8 0<little)signed>)len issubclassr ValueErrorosurandom bytearrayhashlibsha256bytesdigestr obfuscate_tagto_bytesencrypt)secretdc_idris_dd is_rand_codeckeywordsrandomrandom_reversed encrypt_key encrypt_iv decrypt_key decrypt_iv encryptor decryptor dc_id_bytesrrrr sT      zMTProxyIO.init_headercs|j|j|IdHSr )rr8r readexactly)rnrrrrGLszMTProxyIO.readexactlycCs|j|j|dSr )r writerr8)rdatarrrrIOszMTProxyIO.write) __name__ __module__ __qualname____doc__rr staticmethodr rGrIrrrrr s + r csVeZdZdZdZeZdddfdd Zd fdd Ze dd Z e d d Z Z S) TcpMTProxya Connector which allows user to connect to the Telegram via proxy servers commonly known as MTProxy. Implemented very ugly due to the leaky abstractions in Telethon networking classes that should be refactored later (TODO). .. warning:: The support for TcpMTProxy classes is **EXPERIMENTAL** and prone to be changed. You shouldn't be using this class yet. N)proxy local_addrc s6||\}}||d|_tj||||ddS)Nr()loggers) address_infonormalize_secretrsuperr) ripportr:rSrQrR proxy_host proxy_port __class__rrrcs  zTcpMTProxy.__init__cstj||dIdHzt|jddIdHWntjy%Ynty5tdIdHYnw|j rF| IdHt ddS)N)timeoutsslrQr(z9Proxy closed the connection after sending initial payload) rV_connectasynciowait_forr _wait_for_data TimeoutError Exceptionsleepat_eof disconnectConnectionError)rr]r^r[rrr_js   zTcpMTProxy._connectcCs|durtd|ddS)Nz.No proxy info specified for MTProxy connectionr()r.) proxy_inforrrrTs zTcpMTProxy.address_infocCsn|dddvr|dd}zt|}Wnty0|dt| d}t|}Ynw|ddS)Nr()eedd=rr)r4fromhexr.r,base64 b64decodeencode)r9 secret_bytesrrrrUs   zTcpMTProxy.normalize_secret)NN) rKrLrMrNrr obfuscated_iorr_rOrTrU __classcell__rrr[rrPSs  rPc@eZdZdZeZdS)ConnectionTcpMTProxyAbridgedz2 Connect to proxy using abridged protocol N)rKrLrMrNrrrrrrruruc@rt) ConnectionTcpMTProxyIntermediatez6 Connect to proxy using intermediate protocol N)rKrLrMrNrrrrrrrwrvrwc@rt)*ConnectionTcpMTProxyRandomizedIntermediatezN Connect to proxy using randomized intermediate protocol (dd-secrets) N)rKrLrMrNrrrrrrrxrvrx)r`r2rnr/rr tcpabridgedrtcpintermediaterrcryptorr rPrurwrxrrrrs   C@