U h{@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_secretZ_dc_id packet_codecheader_encrypt_decrypt)self connectionrJ/tmp/pip-unpacked-wheel-c81u5j2r/telethon/network/connection/tcpmtproxy.py__init__szMTProxyIO.__init__cCst|dko|ddk}t|t}|r2|s2td|rB|ddn|}t|dkrZtdd}td }|dd kr^|dd |kr^|d d d kr^qq^t|}|d dd}tt |dd| }t |dd} tt |dd| } t |dd} t || } t | | } |j |dd<|j dddd}|dd||dd}| t |dd |dd <|| | fS)Nrz7Only RandomizedIntermediate can be used with dd-secretsrz9MTProxy secret must be a hex-string representing 16 bytes)sPVrGsGET sPOSTs@s7(8 0<littleT)signed>)len issubclassr ValueErrorosurandom bytearrayhashlibsha256bytesdigestrZ obfuscate_tagto_bytesencrypt)secretdc_idrZis_ddZ is_rand_codeckeywordsrandomZrandom_reversedZ encrypt_keyZ encrypt_ivZ decrypt_keyZ decrypt_ivZ encryptorZ decryptorZ dc_id_bytesrrrr sN     zMTProxyIO.init_headercs|j|j|IdHSr )rr6r readexactly)rnrrrr;LszMTProxyIO.readexactlycCs|j|j|dSr )r writerr6)rdatarrrr=OszMTProxyIO.write) __name__ __module__ __qualname____doc__rr staticmethodr r;r=rrrrr 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) ripportr8rGrErFZ proxy_hostZ proxy_port __class__rrrcszTcpMTProxy.__init__cstj||dIdHzt|jddIdHWn8tjk rJYn$tk rltdIdHYnX|j r| IdHt ddS)N)timeoutsslrEr'z9Proxy closed the connection after sending initial payload) rJ_connectasynciowait_forr Z_wait_for_data TimeoutError ExceptionsleepZat_eofZ disconnectConnectionError)rrOrPrMrrrQjs  zTcpMTProxy._connectcCs|dkrtd|ddS)Nz.No proxy info specified for MTProxy connectionr')r-)Z proxy_inforrrrHszTcpMTProxy.address_infocCsp|dddkr|dd}zt|}Wn8tk rb|dt| d}t|}YnX|ddS)Nr')eedd=rr)r3fromhexr-r+base64 b64decodeencode)r7Z secret_bytesrrrrIs zTcpMTProxy.normalize_secret)NN) r?r@rArBrr Z obfuscated_iorrQrCrHrI __classcell__rrrMrrDSs  rDc@seZdZdZeZdS)ConnectionTcpMTProxyAbridgedz2 Connect to proxy using abridged protocol N)r?r@rArBrrrrrrr`sr`c@seZdZdZeZdS) ConnectionTcpMTProxyIntermediatez6 Connect to proxy using intermediate protocol N)r?r@rArBrrrrrrrasrac@seZdZdZeZdS)*ConnectionTcpMTProxyRandomizedIntermediatezN Connect to proxy using randomized intermediate protocol (dd-secrets) N)r?r@rArBrrrrrrrbsrb)rRr1r\r.rrZ tcpabridgedrZtcpintermediaterrZcryptorr rDr`rarbrrrrs   C@