o `šjy(ã@sxddlZddlZddlZddlZddlZddlmZddlmZddl m Z e  e ¡Z Gdd„dƒZGdd„de ƒZdS) éN)Úurlparse)ÚKafkaConnectionError)ÚKafkaNetSocketc@s0eZdZdZdZdZdZdZdZdZ dZ d Z d S) ÚProxyConnectionStateszz zz zzz zz N) Ú__name__Ú __module__Ú __qualname__Ú DISCONNECTEDÚ CONNECTINGÚNEGOTIATE_PROPOSEÚ NEGOTIATINGÚAUTHENTICATINGÚREQUEST_SUBMITÚ REQUESTINGÚ READ_ADDRESSÚCOMPLETE©rrúN/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/net/socks5.pyrsrcsteZdZdZdZdd„Zdd„Zdd„Zd‡fd d „ Ze j e j e j fd d „Z dd„Z dd„Zdd„Zdd„Z‡ZS)Ú Socks5ProxyzuSocks5 proxy Manages connection through socks5 proxy with support for username/password authentication. )Úsocks5Úsocks5hcCsZd|_d|_t|ƒ|_|jj|jvrtd|jjfƒ‚d|_tj |_ t j |_ | ¡|_dS)NózUnsupported proxy scheme: %s)Ú _buffer_inÚ _buffer_outrÚ _proxy_urlÚschemeÚSCHEMESÚ ValueErrorÚ_sockrr Ú_stateÚsocketÚ AF_UNSPECÚ _target_afiÚ_get_proxy_addrÚ _proxy_addr)ÚselfÚ proxy_urlrrrÚ__init__$s zSocks5Proxy.__init__cCs.|j|jj|jjdd}|stdƒ‚t |¡S)NT)Úproxyz#Unable to resolve proxy_url via dns)Ú dns_lookuprÚhostnameÚportrÚrandomÚchoice)r%Ú proxy_addrsrrrr#/s zSocks5Proxy._get_proxy_addrcCs |jjdkS)Nr)rr)r%rrrÚ_use_remote_lookup5s zSocks5Proxy._use_remote_lookupFcsF|r tƒj||ddS| ¡rtjtjtjd||ffgStƒ ||¡S)NT)Ú raise_errorÚ)Úsuperr)r/r r!Ú SOCK_STREAMÚ IPPROTO_TCP)r%Úhostr+r(©Ú __class__rrr)8s zSocks5Proxy.dns_lookupcCs,||_|j\}}}}}t |||¡|_|jS)zšOpen and record a socket. Returns the actual underlying socket object to ensure e.g. selects and ssl wrapping works as expected. )r"r$r r)r%ÚfamilyÚ sock_typeÚprotoÚ proxy_familyÚ_rrrr @szSocks5Proxy.socketcCs2|jr|j |j¡}|j|d…|_|jsdSdS)zÊSend out all data that is stored in the outgoing buffer. It is expected that the caller handles error handling, including non-blocking as well as connection failure exceptions. N)rrÚsend)r%Ú sent_bytesrrrÚ _flush_bufKsþzSocks5Proxy._flush_bufcCsH |t|jƒ}|dkr n|j |¡}|sn|j||_q|jd|…S)z´Ensure local inbound buffer has enough data, and return that data without consuming the local buffer It's expected that the caller handles e.g. blocking exceptionsTrN)ÚlenrrÚrecv)r%ÚdatalenÚbytes_remainingÚdatarrrÚ _peek_bufUs  ù zSocks5Proxy._peek_bufcCs&| |¡}|r|jt|ƒd…|_|S)zuRead and consume bytes from socket connection It's expected that the caller handles e.g. blocking exceptionsN)rErr@)r%rBÚbufrrrÚ _read_bufes zSocks5Proxy._read_bufc Cs||jusJ‚|jtjkrtj|_|jtjkr3|j\}}}}}|j |¡}|r,|tjkr1tj |_n|S|jtj krL|j j rE|j j rEd|_ nd|_ tj|_|jtjkrÂ| ¡| d¡}|dd…dkrtt d¡tj|_|j ¡tjS|dd…dkrtj|_nA|dd…d kr±t|j j ƒ}t|j j ƒ}t d  ||¡d||j j  ¡||j j  ¡¡|_ tj|_nt d ¡tj|_|j ¡tjS|jtjkrë| ¡| d¡}|d krÚtj|_nt d ¡tj|_|j ¡tjS|jtjkrt| ¡rÿd} t|dƒ} n+|jtjkr d} d} n|jtj krd} d} nt d|j¡tj|_|j ¡tjSt dddd| ¡|_ | dkrN|j t d | ¡| |d d¡¡7_ n|j t d | ¡t !|j|d¡¡7_ |j t d|d¡7_ tj"|_|jtj"kr¨| ¡| d¡}|dd…dkr’tj#|_nt d|dd…¡tj|_|j ¡tjS|jtj#krì| $d¡}|dd…dkrÃ| d¡}n%|dd…dkrÒ| d¡}nt d|dd…¡tj|_|j ¡tjStj%|_|jtj%krõdSt d|j¡tj|_|jr |j ¡tjS) a Runs a state machine through connection to authentication to proxy connection request. The somewhat strange setup is to facilitate non-intrusive use from BrokerConnection state machine. This function is called with a socket in non-blocking mode. Both send and receive calls can return in EWOULDBLOCK/EAGAIN which we specifically avoid handling here. These are handled in main BrokerConnection connection loop, which then would retry calls to this function.sséréózUnrecognized SOCKS versionóóz !bb{}sb{}sz(Unrecognized SOCKS authentication methodsz#Socks5 proxy authentication failureééézUnknown address family, %rz!bbbbéz!b{}sÚasciiz!{}sz!HszProxy request failed: %rséséz#Unrecognized remote address type %rz.Internal error, state %r not handled correctly)&rrrr r r$Ú connect_exÚerrnoÚEISCONNr rÚusernameÚpasswordrr r?rGÚlogÚerrorÚcloseÚ ECONNREFUSEDrr@ÚstructÚpackÚformatÚencoder r/r"r ÚAF_INETÚAF_INET6Ú inet_ptonrrrEr) r%ÚsockÚaddrr<ÚsockaddrÚretrFÚuserlenÚpasslenÚ addr_typeÚaddr_lenrrrrTnsÚ               ú         û    ý þ        zSocks5Proxy.connect_ex)F)rrrÚ__doc__rr'r#r/r)r r!r3r4r?rErGrTÚ __classcell__rrr6rrs    r)rUÚloggingr,r r]Ú urllib.parserÚ kafka.errorsrÚkafka.net.inetrÚ getLoggerrrYrrrrrrÚs