U x0h!?@sddlZddlZddlZddlZz ddlZWnek rDdZYnXz ddlZWnek rjdZYnXddlm Z m Z ddl m Z Gdddej ZGdddeZGd d d ej ZdS) N)InvalidChecksumErrorInvalidBufferError)helpersc@seZdZdZdZdddddZeddZed#d d Zd$d d Z d%d dZ d&ddZ ddZ ddZ ddZddZddZddZddZdd Zd!d"ZdS)' Connectiona The `Connection` class is a wrapper around ``asyncio.open_connection``. Subclasses will implement different transport modes as atomic operations, which this class eases doing since the exposed interface simply puts and gets complete data payloads to and from queues. The only error that will raise from send and receive methods is ``ConnectionError``, which will raise when attempting to send if the client is disconnected (includes remote disconnections). N)proxy local_addrcCsn||_||_||_|t|_||_||_d|_d|_d|_ d|_ d|_ d|_ d|_ td|_td|_dS)NF)_ip_portZ_dc_id__name___log_proxy _local_addr_reader_writer _connected _send_task _recv_task_codec _obfuscationasyncioQueue _send_queue _recv_queue)selfipportZdc_idloggersrrrJ/tmp/pip-unpacked-wheel-_zbkcapf/telethon/network/connection/connection.py__init__$s  zConnection.__init__cCs$tdkrtdtj|dtjddS)NzICannot use proxy that requires SSL without the SSL module being availableTzADH-AES256-SHA)do_handshake_on_connect ssl_versionciphers)ssl_mod RuntimeError wrap_socketPROTOCOL_SSLv23sockrrr _wrap_socket_ssl5szConnection._wrap_socket_sslTc Cst|tr|}trddlm}||jks<|dks<|dkrD|j}nR||jks^|dks^|dkrf|j}n0||jks|dks|dkr|j}ntd |||||||fSdd l m}m} m} |dks|dkr|}n:|dks|dkr| }n$|dks|dkr| }ntd |||||||fSdS) Nr) ProxyTypesocks5r socks4rhttpzUnknown proxy protocol type: {})SOCKS5SOCKS4HTTP) isinstancestrlower python_socksr,r1r2r3 ValueErrorformatsocks) proxy_typeaddrrrdnsusernamepasswordr,protocolr1r2r3rrr _parse_proxyCs*  zConnection._parse_proxyc st|jttfr|j|j}n0t|jtr:|jf|j}ntdt|jt rPGdddt }|t j _ t t j _ t t j _ddlm}|j|}|dkr|j|j|j|dIdH}nd|jkrtj|j|jddf}} ntj|j|jf}} t|tj}|d||tjt j!|| d |d IdH|j|j|j||d IdH}nddl"} d|d krtj|j|jddf}} ntj|j|jf}} | #|tj}|j$||%||dk r||tjt j!|| d |d IdH|d|S) NzProxy of unknown format: {}cseZdZdfdd ZZS)z7Connection._proxy_connect..ConnectionErrorExtraNcst|||_dSN)superr! error_code)rmessagerD __class__rr r!{s z@Connection._proxy_connect..ConnectionErrorExtra.__init__)N)r __module__ __qualname__r! __classcell__rrrFr ConnectionErrorExtrazsrKr)Proxy) dest_host dest_porttimeout:F)r*addressrO)rMrNrO_socketr )&r4rtuplelistrAdict TypeErrorr9typer7ConnectionError_errors ProxyErrorProxyConnectionErrorZProxyTimeoutErrorZpython_socks.async_.asynciorLcreateconnectr r proxy_hostsocketAF_INET6 proxy_portAF_INET SOCK_STREAM setblockingbindrwait_forrget_running_loopZ sock_connectr: socksocket set_proxy settimeout) rrOrparsedrKrLrr*moderQr:rrr _proxy_connectjsb             zConnection._proxy_connectcs|jdk rVt|jtr,t|jdkr,|j}qZt|jtrD|jdf}qZtd|jnd}|jstj tj |j |j ||d|dIdH\|_ |_n<|j||dIdH}|r||}tj |dIdH\|_ |_|||_||jIdHdS)Nr-rz Unknown local address format: {})hostrsslrrR)rOrr))rr4rTlenr5r8r9rrrgZopen_connectionr r rrrnr+ packet_codecr _init_conndrain)rrOrprr*rrr _connects6       zConnection._connectcsF|j||dIdHd|_t}|||_|||_dS)z; Establishes a connection with the server. )rOrpNT) rurrrhZ create_task _send_loopr _recv_loopr)rrOrpZlooprrr r^s zConnection.connectc s|js dSd|_tj|j|j|jdIdH|jr|jtj dkrzt j |j ddIdHWnVt j k r|jdYn6tk r}z|jdt||W5d}~XYnXdS) zi Disconnects from the server, and clears pending outgoing and incoming messages. NF)Z send_taskZ recv_task)r rRz;Graceful disconnection timed out, forcibly ignoring cleanupz%s during disconnect: %s)rrZ_cancelr rrrclosesys version_inforrgZ wait_closed TimeoutErrorwarning ExceptioninforXrerrr disconnects"   zConnection.disconnectcCs|jstd|j|S)zp Sends a packet of data through this connection mode. This method returns a coroutine. Not connected)rrYrputrdatarrr send"szConnection.sendcs8|jr,|jIdH\}}|r"||r|SqtddS)zs Receives a packet of data through this connection mode. This method returns a coroutine. Nr)rrgetrY)rresulterrrrr recv-szConnection.recvc sz2|jr0||jIdH|jIdHqWnhtjk rHYnTtk r}z6t |t rp|j dn |j d|IdHW5d}~XYnXdS)zS This loop is constantly popping items off the queue to send them. Nz.The server closed the connection while sendingz%Unexpected exception in the send loop)r_sendrrrrtrCancelledErrorrr4IOErrorr r exceptionrrrrr rv<s  zConnection._send_loopc sz|jrz|IdH}WnPtjk r<YqYqttjfk r}z6|jd||j d|fIdH|IdHW5d}~XYqt k r}z(|jd||j d|fIdHW5d}~XYqt k r}z(|jd||j d|fIdHW5d}~XYqt k rp}z4|jd|j d|fIdH|IdHW5d}~XYqX|j |dfIdHqW5|IdHXdS)zU This loop is constantly putting items on the queue as they're read. Nz Server closed the connection: %sz(Server response had invalid checksum: %sz&Server response had invalid buffer: %sz(Unexpected exception in the receive loop)rr_recvrrrZIncompleteReadErrorr r~rrrrrr)rrrrrr rwNs, &&  zConnection._recv_loopcCs|jjr|j|jjdS)a This method will be called after `connect` is called. After this method finishes, the writer will be drained. Subclasses should make use of this if they need to send data to Telegram to indicate which connection mode will be used. N)rtagrwriterrrr rsls zConnection._init_conncCs|j|j|dSrB)rrr encode_packetrrrr rxszConnection._sendcs|j|jIdHSrB)r read_packetrrrrr r{szConnection._recvcCsd|j|j|jjddS)Nz{}:{}/{}r)r9r r rGr replacerrrr __str__~s zConnection.__str__)TNN)NN)NN)NN)r rHrI__doc__rrr! staticmethodr+rArnrur^rrrrvrwrsrrrrrrr rs&   & e ' !  rc@s,eZdZdZdZddZddZddZdS) ObfuscatedConnectionzR Base class for "obfuscated" connections ("obfuscated2", "mtproto proxy") NcCs |||_|j|jjdSrB) obfuscated_iorrrheaderrrrr rss zObfuscatedConnection._init_conncCs|j|j|dSrB)rrrrrrrr rszObfuscatedConnection._sendcs|j|jIdHSrB)rrrrrrr rszObfuscatedConnection._recv)r rHrIrrrsrrrrrr rs rc@s8eZdZdZdZddZejddZejddZ dS) PacketCodecz& Base class for packet codecs NcCs ||_dS)z@ Codec is created when connection is just made. N)Z_conn)r connectionrrr r!szPacketCodec.__init__cCstdS)zB Encodes single packet and returns encoded bytes. NNotImplementedErrorrrrr rszPacketCodec.encode_packetcstdS)zl Reads single packet from `reader` object that should have `readexactly(n)` method. Nr)rreaderrrr rszPacketCodec.read_packet) r rHrIrrr!abcabstractmethodrrrrrr rs  r)rrr`r{rpr% ImportErrorr7errorsrrrrABCrrrrrrr s$     s