o ja@sddlZddlZddlZddlmZddlmZmZddlm Z m Z m Z ddl m Z mZmZddlmZmZmZddlmZmZmZmZmZmZmZmZmZmZddlm Z Gd d d e!Z"Gd d d Z#dS) N)logger)SocksServerVersionSocksCommsMode) HTTPResponseHTTPProxyAuthRequiredExceptionHTTPProxyAuthFailed) SOCKS4Request SOCKS4Reply SOCKS4CDCode)SOCKS4ARequest SOCKS4AReply SOCKS4ACDCode) SOCKS5Method SOCKS5NegoSOCKS5NegoReply SOCKS5Request SOCKS5ReplySOCKS5ReplyTypeSOCKS5PlainAuthSOCKS5PlainAuthReplySOCKS5ServerErrorReplySOCKS5AuthFailed) SSLWrappercseZdZdfdd ZZS)SocksTunnelErrorKSomething failed setting up the tunnel! See innerexception for more detailscs||_||_t|jdSN)innerexceptionmessagesuper__init__)selfrr __class__/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/asysocks/client.pyrszSocksTunnelError.__init__)r)__name__ __module__ __qualname__r __classcell__r#r#r!r$rsrc@seZdZdddZddZddZedd d Zed d d Zedd dZ dddZ d!ddZ d!ddZ d!ddZ ddZddZd"ddZdS)# SOCKSClientNcCsb||_t|tdur|g|_||_d|_d|_d|_d|_d|_d|_ ||_ d|_ d|_ ||_ dS)NF)proxies isinstancelistcomms server_task proxytask_in proxytask_outproxy_stopped_evtproxy_running_evt http_auth_ctxbind_progress_evt bind_port ssl_wrapperchannel_open_evt)r r-r*bind_evtr7r#r#r$rs zSOCKSClient.__init__csF|jdur |j|jdur|j|jdur!|jdSdSr)r1setr/cancelr0r r#r#r$ terminate%s     zSOCKSClient.terminatecCs|jdur |jSdSr)r6get_server_certr;r#r#r$ get_peercert-s  zSOCKSClient.get_peercertc szz ttj|||d}t|}tj||gtjdIdH\}}|D]q} | IdH} d} | duroz|D] } tj| ddIdH} q9WnYn| dkr]|| |IdHt dWW| | dS| dkrt dWW| | dS|| |IdH|D]} | qq)qtj yYW| | dStjyt dt|Ynty} zt d t| WYd} ~ nd} ~ wwW| | dSW| | dS| | w) NTtimeout) return_wheng?zother side disconnected!zStream channel broken!z"proxy_stream timeout! (timeout=%s)zproxy_stream err: %s)asyncio create_taskwait_forreadwaitFIRST_COMPLETEDwritedrainrdebugcloser9r:CancelledError TimeoutErrorstr Exception) in_stream out_streamr1 buffer_sizerA read_task wait_taskfinished_tasks pending_taskstaskresult last_datapter#r#r$ proxy_stream1sn             zSOCKSClient.proxy_streamc s,zz:|s;|IdH}|dur+tdWW|z|WdSYdS|||IdH|rWn6tj yXYW|z|WdSYdSt yr}ztdt |WYd}~nd}~wwW|z|WdSYdS|z|WwYw)Nz"proxy_queue_in client disconncted!zproxy_queue_in err: %s) is_setgetrrLr9rMrJrKrDrNrQrP)in_queuewriterr1rTdatar]r#r#r$proxy_queue_in^sL   zSOCKSClient.proxy_queue_inc s4zzFt|d} |r||IdH}n tj|||dIdH}|dkr>td|dtdfIdHWW|dS||dfIdHq tj yd|dtdfIdHYW|dSty}ztd|z |d|fIdHWnYWYd}~nd}~wwW|dS|w) Nwsnet_reader_typeTr@rCz%proxy_queue_out endpoint disconncted!z0proxy_queue_out endpoint disconncted gracefully!zproxy_queue_out got cancelled!zproxy_queue_out exception %s) hasattrrGrDrFrrLputrQr9rN) out_queuereaderr1rTrAis_wsnetrcr]r#r#r$proxy_queue_outss>      zSOCKSClient.proxy_queue_outc sztd|j|jfd|j|j|j|jf}|}|dur||dtj||dIdHt j ||dIdH\}}|durF||j dkrVtdd|dfWS|j dkrtd |j durit d |jd d} | durxt d | d d\} } td| | }d|tfWSWdS|dkrtd|j j|j jf} |d| d} || dtj||dIdHt j ||dIdH\}}|dur||j dkrtdd|dfWStt d|t y } ztddd| fWYd} ~ Sd} ~ ww)Nz[HTTP] Opening channel to %s:%sz%CONNECT %s:%s HTTP/1.1 Host: %s:%s s r@z&[HTTP] Server succsessfully connected!Tiz"[HTTP] Server proxy auth required!z/HTTP proxy auth required but no credential set!zPROXY-AUTHENTICATEzSHTTP proxy requires authentication, but requested auth type could not be determined zHTTP proxy requires %s authFBASICz%s:%ssProxy-Authorization: Basic z"[HTTP] Server proxy auth succsess!z?HTTP proxy requires %s authentication, but it's not implementedrun_http)rrL endpoint_ip endpoint_portencoderJrDrFrKrfrom_streamreaderstatus credentialrQ headers_upperr`splitupperrbase64 b64encodeusernamepasswordr)r proxy remote_reader remote_writerrAr3 connect_cmdresperr auth_type_ auth_data auth_connectr]r#r#r$rpsX           zSOCKSClient.run_httpc zjtd|j|j|j|jft|}|jdur&|jj dur&|jj |_ | | t j||dIdHt jt||dIdH\}}|durN||durVtd|jtjkrdtd|jjtdWdSty}ztdd |fWYd}~Sd}~ww) h Does the intial "handshake" instructing the remote server to set up the connection to the endpoint z0[SOCKS4 %s:%s] Opening channel to %s:%s (target)Nr@0Socks server failed to reply to CONNECT request!*Socks server returned error on CONNECT! %sz[SOCKS4] Channel openedTN run_socks4F)rrL server_ip server_portrqrrr from_targetrvr|USERIDrJto_bytesrDrFrKr rtrQCDr REP_GRANTEDvalue r r~rrrAreqreprr]r#r#r$r,     zSOCKSClient.run_socks4c r) rz1[SOCKS4A %s:%s] Opening channel to %s:%s (target)Nr@rrz[SOCKS4A] Channel openedr run_socks4aF)rrLrrrqrrr rrvr|rrJrrDrFrKr rtrQrr rrrr#r#r$rrzSOCKSClient.run_socks4ac s|}|}tjg}|jdur$|jjdur$|jjdur$|tjzEt |}t d|| |tj||dIdHtjt||dIdH} t d|| jjf| jtjkr|jdust|jjdust|jjdurztd|t d|| t|jj|jjtj|ddIdHtj|d|dIdH} | dkrtt| } | jtjkrtn | jtj krtd |t d ||f|j!d urWd S| t"#|tj||dIdHtjt$||dIdH} | j%tjkrt &d || j%jft'| j%|j(dur(t d||fWd St d|| j)f| j)|_*|j+,|j-d urDWd Stjt$||dIdH} | j%tjkrht &d || j%jft'| j%Wd Sty} zt d|| fd| fWYd} ~ Sd} ~ ww)rNz8[SOCKS5 %s][SETUP] Sending negotiation command to serverr@z?[SOCKS5 %s] Got negotiation reply! Server choosen auth type: %szRSOCKS5 %s] server requires PLAIN authentication, but no credentials were supplied!z$[SOCKS5 %s]Preforming plaintext authrCzW[SOCKS5 %s] server requires GSSAPI authentication, but it's not supported at the momentz![SOCKS5 %s] Opening channel to %sTrz<[SOCKS5 %s] remote end failed to connect to proxy! Reson: %sFzB[SOCKS5 %s] Successfully set up the connection to the endpoint %s z8[SOCKS5 %s] BIND first set completed, port %s available!z"[SOCKS5 %s] Error in run_socks5 %s). get_sname get_tnamerNOAUTHrvr|r}appendPLAINr from_methodsrrLrJrrDrFrKrrtMETHODnamerQr constructrGrr from_bytesSTATUSr SUCCEEDEDGSSAPI only_authrrrREPinforis_bind BIND_PORTr5r4r9 only_bind) r r~rrrAsnametnamemethodsnegorep_negorep_datarr]r#r#r$ run_socks5s"     "                zSOCKSClient.run_socks5c sd}d}zztdt|jdkrtdtdddD] }|dur+|z|jdjtjkrTddl m }| |jdj |jdj |jdjIdH\}}nl|jdjtjtjfvrddlm}| |jdj |jdj |jdj|jdj|jdj|jdj|jdjIdH\}}|dur|n'tjtj |jdj|jdj|jdjd |jdjd IdH\}}td Wn td zht|jD]^\}} | jtjtjfvr&ztj|| |||jdjd IdH} Wn tjytd w| \}} | dur%t|jdkr#|t|jdkr#t | | q| jtj!tj"fvruztj|#| |||jdjd IdH} Wn tjyRtd w| \}} | durtt|jdkrr|t|jdkrrt | | q| jtj$tj%fvrztj|&| |||jdjd IdH} Wn tjytd w| \}} | durt|jdkr|t|jdkrt | | q| jtj'tj(fvrztj|j)| |||j*d|jdjd IdH} Wn tjytd w| \}|_*} | durt|jdkr|t|jdkrt | | q| jtjtjtjfvr.|dkr-tdqtdWn t+y>Yq td|j,-t.|_/t0t1j2|||j/|jdj3|jdj4d|_5t0t1j2|||j/|jdj3|jdj4d|_6td|j/7IdHtd|j58|j68Wnty} z t9dWYd} ~ nd} ~ wwW|dur|dSdS|dur|ww)Nz![handle_client] Client connected!rnStart chaining...r WSNETNetworkWSNETNetworkWS)sslr@zConnected to socks server!"Failed to connect to SOCKS server!z&Proxy Connection establishment timeoutr31WSNET only supported as the first proxy in chain!Unknown SOCKS version!z![handle_client] Starting proxy...rTrAz[handle_client] Proxy started!z[handle_client] Proxy stopped!z[handle_client]):rrLlenr*rangerMversionrWSNETasysocks.network.wsnetropen_connectionrqrr wsnet_reuseWSNETWSWSNETWSSasysocks.network.wsnetwsrrragentidrArDrFssl_ctx enumerateSOCKS4SOCKS4SrrOrQrSOCKS4ASOCKS4ASrSOCKS5SOCKS5SrHTTPHTTPSrpr3rr7r9Eventr1rEr)r^rTendpoint_timeoutr/r0rHr: exception) r rirbrrrrrir~xrr]r#r#r$ handle_clientms                     * $* $* $0  $               zSOCKSClient.handle_clientc sZtdd}d}|jdurt|_zzFt|jdkr$tdtdddD]}|dur5|z|jdj t j kr^ddl m }||jdj|jdj|jdjIdH\}}na|jdj t jt jfvrddlm}||jdj|jdj|jdj|jdj|jdj |jdj|jdjIdH\}}|dur|ntjt|jdj|jdj|jdjd IdH\}}Wn td td zht|jD]^\}}|j t jt jfvr*ztj|||||jdjd IdH}Wn tjyt d w|\}} | dur)t|jdkr'|t|jdkr't!| | q|j t j"t jfvryztj|#||||jdjd IdH}Wn tjyVt d w|\}} | durxt|jdkrv|t|jdkrvt!| | q|j t j$t j%fvrztj|&||||jdjd IdH}Wn tjyt dw|\}} | durt|jdkr|t|jdkrt!| | q|j t j't j(fvrztj|j)||||j*d|jdjd IdH}Wn tjyt dw|\}|_*} | durt|jdkr|t|jdkrt!| | q|j t j t jt jfvr2|dkr1t dqt dWn t+yBYq*|jdj,durcWW|j-.|dura|dSdS|jdj/dusu|jdj0durWW|j-.|dur|dSdS|j1dur|j1.tdt|_2|j3j4durt5t6j7|j3j8||j2|jdj9d|_:t5t6j;|j3j<||j2|jdj9|jdj=d|_>nVt?} t?} t5t6j7| ||j2|jdj9d|_:t5t6j;| ||j2|jdj9|jdj=d|_>t@A|j3| | IdH\|_B} | dur| tj|jBjCD|j3jEd IdH|j-.td|j2DIdHtd|j:F|j>FWW|j-.|dur\|dSdSt y} z2|j3j8Gd| fIdH|j3j staticmethodr^rdrkrprrrrrrr#r#r#r$r)s&  ,    <  l?r))$rDrz tracebackasysocksrasysocks.common.constantsrrasysocks.protocol.httprrrasysocks.protocol.socks4rr r asysocks.protocol.socks4ar r r asysocks.protocol.socks5rrrrrrrrrrasysocks.utils.sslwrapperrrQrr)r#r#r#r$s 0