o õ±jfFã@sþddlZddlZddlZddlZddlZddlmZddlZddlm Z m Z ddl m Z ddl mZmZmZddlmZmZmZmZmZmZmZmZmZmZmZmZddlmZe  d¡Z!e "¡Z#e $d ¡Z%e# &e%¡e! 'e#¡e! (ej)¡Gd d „d ƒZ*dS) éN)Úurlparse)ÚSocksServerVersionÚSocksCommsMode)Ú HTTPRequest)Ú SOCKS4RequestÚ SOCKS4ReplyÚ SOCKS4CDCode) Ú SOCKS5CommandÚSOCKS5AddressTypeÚ SOCKS5MethodÚ SOCKS5NegoÚSOCKS5NegoReplyÚ SOCKS5RequestÚ SOCKS5ReplyÚSOCKS5ReplyTypeÚSOCKS5PlainAuthÚSOCKS5PlainAuthReplyÚSOCKS5ServerErrorReplyÚSOCKS5AuthFailed)Ú BaseMonitorzasysocks.proxyserverz3%(asctime)s %(name)-12s %(levelname)-8s %(message)sc@sfeZdZdddgd¢dfdd„Zdd„Zdd d „Zd d „Zd d„Zdd„Zdd„Z dd„Z dd„Z dS)Ú SOCKSServerNé i()ÚSOCKS4ÚSOCKS5ÚHTTPcCs4||_||_||_||_||_||_||_d|_dS)Nr)Ú listen_ipÚ listen_portÚssl_ctxÚclient_timeoutÚ buffer_sizeÚsupported_protocolsÚmonitor_dispatch_qÚ session_id)Úselfrrrrrr r!©r$ú‚/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/asysocks/server.pyÚ__init__s zSOCKSServer.__init__cCs|j}|jd7_|S)Né)r")r#Útr$r$r%Úget_session_id$szSOCKSServer.get_session_idc ƒs‡‡fdd„}z¶zq|durHt |ˆ||jƒ¡ˆ ¡sG| ¡IdH}|dks+|dur7WW| ¡ˆ ¡dS| |¡| ¡IdHˆ ¡rn1ˆ ¡syˆ  |j¡IdH}|dks]|duriWW| ¡ˆ ¡dS| |¡| ¡IdHˆ ¡rLWn.tj yŽYW| ¡ˆ ¡dSt y¨} zt   d|| f¡WYd} ~ nd} ~ wwW| ¡ˆ ¡dSW| ¡ˆ ¡dS| ¡ˆ ¡w)Nc“s˜zEz2ˆ ¡s3ˆ |¡IdH}|dks|dur'| |¡IdHWWˆ ¡dS| |¡IdHˆ ¡rWn YWˆ ¡dSWˆ ¡dSˆ ¡w)Nó)Úis_setÚreadÚputÚset)ÚreaderÚin_qrÚdata©Úreader_aÚstop_evtr$r%Úpi*s € üû€ €zSOCKSServer.__proxy..pir*z+[%s][TCPPROXY] Connection ended. Reason: %s)ÚasyncioÚ create_taskrr+ÚgetÚcloser.ÚwriteÚdrainr,ÚCancelledErrorÚ ExceptionÚ srvloggerÚdebug) r#r3Úwriter_br4Ú src_moduleÚ monitor_in_qÚ monitor_out_qr5r1Úer$r2r%Ú__proxy)sR€   ïù€   øú€ ü€ÿ€ ý ÿ zSOCKSServer.__proxyc ƒs~zš|jtjkr t dt|jƒ|jf¡zt  t|jƒ|j¡IdH\}}Wn?t yg}z3t dt|jƒ|j|f¡t ƒ}tj |_|j|_|j|_ˆ  | ¡¡ˆ ¡IdHWYd}~WdSd}~wwt dt|jƒ|jf¡t ƒ}tj|_|j|_|j|_ˆ  | ¡¡ˆ ¡IdHt ¡}d} d} d} d} ˆjdurÑ| d¡\} }ˆ d¡\}}t||| |dˆ ¡ƒ}|j} |j} |j} |j} ˆj |¡IdHt ˆ ˆ||d| | ¡¡}t ˆ |ˆ|d| | ¡¡}| ¡IdHt dt|jƒ|jf¡| ¡| ¡WdS|jtjkr–‡‡‡fdd„}zJt   t j!t j"¡}| #t|jƒd f¡| $d ¡| %¡\}}t ƒ}tj|_||_t& '|¡|_ˆ  | ¡¡ˆ ¡IdHtj(||d d IdHWWdSt y•}z%t ƒ}tj |_|j|_|j|_ˆ  | ¡¡ˆ ¡IdHWYd}~WdSd}~wwt d |jƒ‚t y¾}zd dl)}| *¡t+d |ƒWYd}~dSd}~ww)Nz*[SOCKS4] Client wants to connect to: %s:%sz/[SOCKS4] Could not connect to: %s:%s Reason: %sz<[SOCKS4] Sucsessfully connected to: %s:%s Starting TCP proxyÚpeernamerz[SOCKS4] Connection ended %s:%sc “sàzU| d¡\}}tƒ}tj|_||_t |¡|_ˆ  |  ¡¡ˆ  ¡IdHt   ¡}t  ˆ ˆ||d¡¡}t  ˆ |ˆ|d¡¡}| ¡IdHtdƒ| ¡| ¡WdStyo}z td|ƒWYd}~dSd}~ww)NrFrzConnection terminatedz__handle_remote %s)Úget_extra_inforrÚ REP_GRANTEDÚCDÚDSTPORTÚ ipaddressÚ IPv4AddressÚDSTIPr:Úto_bytesr;r6ÚEventr7Ú_SOCKSServer__proxyÚwaitÚprintÚcancelr=© Ú dst_readerÚ dst_writerÚrem_ipÚrem_portÚreplyr4Útask_1Útask_2rD©r/r#Úwriterr$r%Ú__handle_remoteƒs(€ €ÿz2SOCKSServer.handle_socks4..__handle_remoterr'©ÚsockÚbacklogzUnknown client request! %szhandle_socks4 %s),rIrÚ REQ_CONNECTr>r?ÚstrrMrJr6Úopen_connectionr=rÚ REP_FAILEDr:rNr;rHrOr!rGrr)Úc2d_inÚc2d_outÚd2c_inÚd2c_outr-r7rPrQrSÚREQ_BINDÚsocketÚAF_INETÚ SOCK_STREAMÚbindÚlistenÚ getsocknamerKrLÚ start_serverÚ tracebackÚ print_excrR)r#Úinit_cmdr/r]rUrVrDrYr4rfrgrhrirWrXÚ client_ipÚ client_portÚmonitorrZr[Ú_SOCKSServer__handle_remoteÚ serversocketÚserversocket_ipÚserversocket_portrrr$r\r%Ú handle_socks4Ss˜€"€ø     €ù €ýzSOCKSServer.handle_socks4c ƒsØzNt dd dd„|jDƒ¡¡tj|jvr0t tj¡}ˆ  |  ¡¡ˆ  ¡IdHWdSt tj¡}ˆ  |  ¡¡ˆ  ¡IdHt j t ˆ¡ˆjdIdH}|jtjkr_t dt|jƒ|jf¡zt  t|jƒ|j¡IdH\}}Wn:ty±}z.t dt|jƒ|j|f¡t tj|j|j¡}ˆ  |  ¡¡ˆ  ¡IdHWYd}~WdSd}~wwt dt|jƒ|jf¡t tj|j|j¡}ˆ  |  ¡¡ˆ  ¡IdHt  ¡} d} d} d} d} ˆjdur&| d ¡\}}ˆ d ¡\}}t ||||d ˆ !¡ƒ}|j"t#j$kr| %t|jƒ¡ˆj &|¡IdH|j'} |j(} |j)} |j*} t  +ˆ ,ˆ|| d | | ¡¡}t  +ˆ ,|ˆ| d | | ¡¡}|  -¡IdHt d t|jƒ|jf¡| .¡| .¡WdS|jtj/kr:t d t|jƒ|jf¡‡‡‡fd d„}zƒ|j"t#j0kr—t1 1t1j2t1j3¡}| 4t|jƒ|jf¡n5|j"t#j5kr²t1 1t1j6t1j3¡}| 4t|jƒ|jf¡n|j"t#j$krÌt1 1t1j2t1j3¡}| 4t|jƒ|jf¡| 7d¡| 8¡\}}t tjt9 :|¡|¡}ˆ  |  ¡¡ˆ  ¡IdHt j;||ddIdHWWdSty9}z.t dt|jƒ|j|f¡t tj|j|j¡}ˆ  |  ¡¡ˆ  ¡IdHWYd}~WdSd}~ww|jtj¸sz-SOCKSServer.handle_socks5..©Útimeoutz*[SOCKS5] Client wants to connect to: %s:%sz/[SOCKS5] Could not connect to: %s:%s Reason: %sz<[SOCKS5] Sucsessfully connected to: %s:%s Starting TCP proxyrFrz[SOCKS5] Connection ended %s:%sz'[SOCKS5] Client wants to BIND to: %s:%sc “szm| d¡\}}t dt|ƒ|f¡t tjt  |¡|¡}ˆ  |  ¡¡ˆ  ¡IdHt dt|ƒ|f¡t  ¡}t  ˆ ˆ||d¡¡}t  ˆ |ˆ|d¡¡}| ¡IdHt dt|ƒ|f¡| ¡| ¡WdSty†}z t d¡WYd}~dSd}~ww)NrFzL[SOCKS5] Client BIND server get a connection from: %s:%s Notifying client...z-[SOCKS5] Client BIND starting TCP proxy %s:%srz+[SOCKS5] Client BIND connection ended %s:%sz[SOCKS5] Client BIND error)rGr>r?rcrÚ constructrÚ SUCCEEDEDrKÚ ip_addressr:rNr;r6rOr7rPrQrSr=Ú exceptionrTr\r$r%r^òs&€€ÿz2SOCKSServer.handle_socks5..__handle_remoter'r_z3[SOCKS5] Error! Could not bind to: %s:%s Reason: %szJ[SOCKS5] Error! Client requested UDP Associate, but it is not implemented!z,UDP assoc requested, but is not implemented!z6Client requested something that was not understood. %sz[SOCKS5] Generic error: %s)>r>r?ÚjoinÚMETHODSr ÚNOAUTHr r„Ú NOTACCEPTABLEr:rNr;r6Úwait_forrÚfrom_streamreaderrÚCMDr ÚCONNECTrcÚDST_ADDRÚDST_PORTrdr=rrÚFAILUREr…rOr!rGrr)ÚATYPr Ú DOMAINNAMEÚ set_hostnamer-rfrgrhrir7rPrQrSÚBINDÚIP_V4rkrlrmrnÚIP_V6ÚAF_INET6rorprKr†rqÚ UDP_ASSOCIATEr‡)r#rtr/r]rYÚreqrUrVrDr4rfrgrhrirWrXrurvrwrZr[rxryrzr{r$r\r%Ú handle_socks5µs¬€    "€û   €û €ÿzSOCKSServer.handle_socks5c Ãszí|j ¡dkrtd|jƒ‚|j d¡\}}t|ƒ}t d||f¡z t  ||¡IdH\}}Wn,ty^}z t d|||f¡d} |  | ¡|  ¡IdHWYd}~WdSd}~wwt d||f¡d} |  | ¡|  ¡IdHt  ¡} d} d} d}d}|j durº| d ¡\}}| d ¡\}}t||||d | ¡ƒ}| |¡|j  |¡IdH|j} |j} |j}|j}t | ||| d | | ¡¡}t | ||| d ||¡¡}|  ¡IdHt d ||f¡| ¡| ¡WdSty }zt d |¡WYd}~dSd}~ww) NrzUnknown request type! %sú:z/[HTTPCONNECT] Client wants to connect to: %s:%sz4[HTTPCONNECT] Could not connect to %s:%s. Reason: %sóHTTP/1.1 500 Error zB[HTTPCONNECT] Sucsessfully connected to %s:%s. Starting tcp proxy.sHTTP/1.1 200 OK rFÚ HTTPCONNECTz'[HTTPCONNECT] Connection ended @ %s:%s.ú[HTTP] Generic error: %s)ÚmethodÚupperr=ÚuriÚsplitÚintr>r?r6rdr:r;rOr!rGrr)r•r-rfrgrhrir7rPrQrSr‡)r#rtr/r]ÚhostÚportrUrVrDÚfailÚokr4rfrgrhrirWrXrurvrwrZr[r$r$r%Úhandle_http_connect,s^€ €û   €ÿzSOCKSServer.handle_http_connectc Ãs*zøt|jƒ}|j}|j}|dus|dkr |j ¡dkrdnd}z t ||¡IdH\}}Wn,tyY} z t   d||| f¡d} |  | ¡|  ¡IdHWYd} ~ WdSd} ~ wwt  |¡} |j|j| _d} d} |jdurª| d¡\}}| d¡\}}t||||d| ¡ƒ} |  |¡|j | ¡IdH| j |  ¡¡IdH| j ¡IdH} n|  ¡} |  | ¡|  ¡IdHd} tj||jd IdH\}}|durÏ|‚| durç| j | ¡¡IdH| j ¡IdH} n| ¡} |  | ¡|  ¡IdHWdSty} zt   d | ¡WYd} ~ dSd} ~ ww) NÚÚhttpéPi»z-[HTTP] Could not connect to %s:%s. Reason: %sržrFrŸr‚r )!rr£Úhostnamer§ÚschemeÚlowerr6rdr=r>r?r:r;ÚcopyÚdeepcopyÚpathÚqueryr!rGrr)r•r-rfrNrgr8rrrrhrir‡)r#rtr/r]Úor¦r§rUrVrDr¨Úc_cmdr1rwrWrXrurvrYÚerrr$r$r%Úhandle_http_genericbs\€  €û     €ÿzSOCKSServer.handle_http_genericc Ãsdzztj| d¡|jdIdH}Wntjjy.}z t d¡WYd}~WdSd}~ww|dkrnd|jvrClient terminated the socket before socks/http proxy handshakeórzr?r r=Ú readuntilrÚ from_bytesr|r¥r rœrrrªr¸r‡) r#r/r]Útempr·Útemp2ÚrestrtÚnmethodsÚ t_nmethodsÚmethodsrDr$r$r%rÉ™s\€  €þ       €ÿzSOCKSServer.handle_clientcÃs4tj|j|j|j|jdIdH}| ¡IdHdS)N)Ússl)r6rqrÉrrrÚ wait_closed)r#Úserverr$r$r%ÚrunÒs€ zSOCKSServer.run)NN) Ú__name__Ú __module__Ú __qualname__r&r)rPr|rœrªr¸rÉrØr$r$r$r%rs *bw67 9r)+Úloggingr6rkr±rKÚ urllib.parserrÕÚasysocks.common.constantsrrÚasysocks.protocol.httprÚasysocks.protocol.socks4rrrÚasysocks.protocol.socks5r r r r r rrrrrrrÚ#asysocks.intercepting.monitors.baserÚ getLoggerr>Ú StreamHandlerÚhandlerÚ FormatterÚ formatterÚ setFormatterÚ addHandlerÚsetLevelÚINFOrr$r$r$r%Ús*  8  ÿ