U ¯ªh!?ã@sÀddlZddlZddlZddlZz ddlZWnek rDdZYnXz ddlZWnek rjdZYnXddlm Z m Z ddl m Z Gdd„dej ƒZGdd„deƒZGd d „d ej ƒZdS) éNé)ÚInvalidChecksumErrorÚInvalidBufferError)Úhelpersc@s¤eZdZdZdZdddœdd„Zedd„ƒZed#d d „ƒZd$d d „Z d%d d„Z d&dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd „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|_ t d¡|_t d¡|_dS)NFé)Ú_ipÚ_portZ_dc_idÚ__name__Ú_logÚ_proxyÚ _local_addrÚ_readerÚ_writerÚ _connectedÚ _send_taskÚ _recv_taskÚ_codecÚ _obfuscationÚasyncioÚQueueÚ _send_queueÚ _recv_queue)ÚselfÚipÚportZdc_idÚloggersrr©rúJ/tmp/pip-unpacked-wheel-c81u5j2r/telethon/network/connection/connection.pyÚ__init__$s  zConnection.__init__cCs$tdkrtdƒ‚tj|dtjddS)NzICannot use proxy that requires SSL without the SSL module being availableTzADH-AES256-SHA)Údo_handshake_on_connectÚ ssl_versionÚciphers)Ússl_modÚ RuntimeErrorÚ wrap_socketÚPROTOCOL_SSLv23©Úsockrrr Ú_wrap_socket_ssl5sÿüzConnection._wrap_socket_sslTc Cst|tƒr| ¡}tr¦ddlm}||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)Ú ProxyTypeéZsocks5r Zsocks4rÚhttpzUnknown proxy protocol type: {})ÚSOCKS5ÚSOCKS4ÚHTTP) Ú isinstanceÚstrÚlowerÚ python_socksr,r/r0r1Ú ValueErrorÚformatÚsocks) Ú proxy_typeÚaddrrZrdnsÚusernameÚpasswordr,Úprotocolr/r0r1rrr Ú _parse_proxyCs*  zConnection._parse_proxyc Ãsút|jttfƒr|j|jŽ}n0t|jtƒr:|jf|jŽ}ntd t|jƒ¡ƒ‚t rPGdd„dt ƒ}|t j _ t t j _ t t j _ddlm}|j|Ž}|dkr¶|j|j|j|dIdH}n˜d|jkrÚtj|j|jddf}} ntj|j|jf}} t |tj¡}| d¡| |¡tjt  ¡j!|| d |d IdH|j|j|j||d IdH}n¦ddl"} d|d kr€tj|j|jddf}} ntj|j|jf}} |  #|tj¡}|j$|Ž| %|¡|dk rÊ| |¡tjt  ¡j!|| d |d IdH| d¡|S) NzProxy of unknown format: {}cseZdZd‡fdd„ Z‡ZS)z7Connection._proxy_connect..ConnectionErrorExtraNcstƒ |¡||_dS©N)Úsuperr!Ú error_code)rÚmessagerA©Ú __class__rr r!{s z@Connection._proxy_connect..ConnectionErrorExtra.__init__)N)r Ú __module__Ú __qualname__r!Ú __classcell__rrrCr ÚConnectionErrorExtrazsrHr)ÚProxy)Ú dest_hostÚ dest_portÚtimeoutú:F)r*Úaddress©rL)rJrKrLÚ_socketr )&r2rÚtupleÚlistr>ÚdictÚ TypeErrorr7Útyper5ÚConnectionErrorÚ_errorsÚ ProxyErrorZProxyConnectionErrorZProxyTimeoutErrorZpython_socks.async_.asynciorIÚcreateÚconnectr r Z proxy_hostÚsocketÚAF_INET6Z proxy_portÚAF_INETÚ SOCK_STREAMÚ setblockingÚbindrÚwait_forrÚget_running_loopZ sock_connectr8Z socksocketÚ set_proxyÚ settimeout) rrLrÚparsedrHrIrr*ÚmoderNr8rrr Ú_proxy_connectjsb   ý   þ ü    þ  zConnection._proxy_connectcÃsô|jdk rVt|jtƒr,t|jƒdkr,|j}qZt|jtƒrD|jdf}qZtd |j¡ƒ‚nd}|jstj tj |j |j ||d|dIdH\|_ |_n<|j||dIdH}|r²| |¡}tj |dIdH\|_ |_| |¡|_| ¡|j ¡IdHdS)Nr-rz Unknown local address format: {})ÚhostrÚsslrrO)rLrr))rr2rQÚlenr3r6r7rrraZopen_connectionr r rrrgr+Ú packet_codecrÚ _init_connÚdrain)rrLrirr*rrr Ú_connectÏs6   üú þ   zConnection._connectcÃsF|j||dIdHd|_t ¡}| | ¡¡|_| | ¡¡|_dS)z; Establishes a connection with the server. )rLriNT) rnrrrbZ create_taskÚ _send_looprÚ _recv_loopr)rrLriZlooprrr rZös zConnection.connectc ÃsÀ|js dSd|_tj|j|j|jdIdH|jr¼|j ¡tj dkr¼zt j |j  ¡ddIdHWnVt j k r†|j d¡Yn6tk rº}z|j dt|ƒ|¡W5d}~XYnXdS) zi Disconnects from the server, and clears pending outgoing and incoming messages. NF)Z send_taskZ recv_task)réé rOz;Graceful disconnection timed out, forcibly ignoring cleanupz%s during disconnect: %s)rrZ_cancelr rrrÚcloseÚsysÚ version_inforraZ wait_closedÚ TimeoutErrorÚwarningÚ ExceptionÚinforU©rÚerrr Ú disconnects"ý   zConnection.disconnectcCs|jstdƒ‚|j |¡S)zp Sends a packet of data through this connection mode. This method returns a coroutine. ú Not connected)rrVrÚput©rÚdatarrr Úsend"szConnection.sendcÃs8|jr,|j ¡IdH\}}|r"|‚|r|Sqtdƒ‚dS)zs Receives a packet of data through this connection mode. This method returns a coroutine. Nr})rrÚgetrV)rÚresultÚerrrrr Úrecv-szConnection.recvc Ãs z2|jr0| |j ¡IdH¡|j ¡IdHqWnhtjk rHYnTtk rš}z6t |t ƒrp|j   d¡n |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Ú_sendrr‚rrmrÚCancelledErrorrxr2ÚIOErrorr ryÚ exceptionr|rzrrr ro<s  zConnection._send_loopc Ãs¢zŠ|jrŠz| ¡IdH}WnPtjk r<YqŠYqttjfk r”}z6|j d|¡|j   d|f¡IdH| ¡IdHW5d}~XYqt k rØ}z(|j d|¡|j   d|f¡IdHW5d}~XYqt k r}z(|j d|¡|j   d|f¡IdHW5d}~XYqt k rp}z4|j d¡|j   d|f¡IdH| ¡IdHW5d}~XYqX|j   |df¡IdHqW5| ¡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)r|rÚ_recvrr‡rˆZIncompleteReadErrorr rwrr~rrrxr‰)rr€r{rrr rpNs, &&  zConnection._recv_loopcCs|jjr|j |jj¡dS)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)rÚtagrÚwrite©rrrr rlls zConnection._init_conncCs|j |j |¡¡dSr?)rrŒrÚ encode_packetrrrr r†xszConnection._sendcÃs|j |j¡IdHSr?)rÚ read_packetrrrrr rŠ{szConnection._recvcCsd |j|j|jj dd¡¡S)Nz{}:{}/{}rÚ)r7r r rDr Úreplacerrrr Ú__str__~s þzConnection.__str__)TNN)NN)NN)NN)r rErFÚ__doc__rkr!Ú staticmethodr+r>rgrnrZr|rr…rorprlr†rŠr’rrrr rs&   & e ' !  rc@s,eZdZdZdZdd„Zdd„Zdd„ZdS) ÚObfuscatedConnectionzR Base class for "obfuscated" connections ("obfuscated2", "mtproto proxy") NcCs | |¡|_|j |jj¡dSr?)Ú obfuscated_iorrrŒÚheaderrrrr rlŽs zObfuscatedConnection._init_conncCs|j |j |¡¡dSr?)rrŒrrŽrrrr r†’szObfuscatedConnection._sendcÃs|j |j¡IdHSr?)rrrrrrr rŠ•szObfuscatedConnection._recv)r rErFr“r–rlr†rŠrrrr r•…s r•c@s8eZdZdZdZdd„Zejdd„ƒZejdd„ƒZ 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__cCst‚dS)zB Encodes single packet and returns encoded bytes. N©ÚNotImplementedErrorrrrr rŽ«szPacketCodec.encode_packetcÃst‚dS)zl Reads single packet from `reader` object that should have `readexactly(n)` method. Nrš)rÚreaderrrr r²szPacketCodec.read_packet) r rErFr“r‹r!ÚabcÚabstractmethodrŽrrrrr r˜™s  r˜)rrr[rtrir%Ú ImportErrorr5ÚerrorsrrrrÚABCrr•r˜rrrr Ús$     s