gny ddlmZ ddlmZn#e$r ddlmZYnwxYwddlmZmZm Z ddl Z ddl m Z ddl Z ddlZddlmZddlZddlZddlZdZejdkr%ejd kr ddlZn#e$r ed wxYwe jeZd xZZd xZZd xZZeeedZ e!e"e #e $Z%ejxZ&Z'dZ(Gdde)Z*Gdde*Z+Gdde*Z,Gdde*Z-Gdde*Z.Gdde*Z/Gdde*Z0ddd d!Z1d"d#d$d%d&d'd(d)d*Z2ed+ed+ed,iZ3 d9d.Z4d/Z5d0Z6e6Z7d1Z8e8Z9 d:d2Z:Gd3d4ejZ;d5Ze?e>es0e;j@AeeBe;ee.wrapper(sAw (??,,Lq    &&&8T,V,, q    ''''!      q    ''''!s#6A A.(A))A..A11B) functoolswraps)rrs` rset_self_blockingr &s8_X ( ( ( ( ( Nc eZdZdZddZdZdS) ProxyErrorz4Socket_err contains original socket.error exception.Ncp||_||_|r%|xjd|z c_dSdS)Nz: {})msg socket_errformat)rr%r&s r__init__zProxyError.__init__;sB$  2 HH j11 1HHHH 2 2r!c|jSN)r%rs r__str__zProxyError.__str__Bs xr!r*)__name__ __module__ __qualname____doc__r(r,r!rr#r#9s=>>2222r!r#ceZdZdS)GeneralProxyErrorNr-r.r/r1r!rr3r3FDr!r3ceZdZdS)ProxyConnectionErrorNr4r1r!rr7r7Jr5r!r7ceZdZdS)SOCKS5AuthErrorNr4r1r!rr9r9Nr5r!r9ceZdZdS) SOCKS5ErrorNr4r1r!rr;r;Rr5r!r;ceZdZdS) SOCKS4ErrorNr4r1r!rr=r=Vr5r!r=ceZdZdS) HTTPErrorNr4r1r!rr?r?Zr5r!r?zRequest rejected or failedzLRequest rejected because SOCKS server cannot connect to identd on the clientzPRequest rejected because the client program and identd report different user-ids)[\]zGeneral SOCKS server failurez!Connection not allowed by rulesetzNetwork unreachablezHost unreachablezConnection refusedz TTL expiredz(Command not supported, or protocol errorzAddress type not supported)r r r i8iTc|||||r|nd|r|ndft_dS)zSets a default proxy. All further socksocket objects will use the default unless explicitly changed. All parameters are as for socket.set_proxy().N)encode socksocket default_proxy) proxy_typeaddrportrdnsusernamepasswords rset_default_proxyrRssL !+D$5= G 1 1 145= G 1 1 14 IJr!cRd|vr|d|d<t|i|SN proxytyperL)poprR)rrs rsetdefaultproxyrW~s6f%zz+66| d -f - --r!ctjS)z4Returns the default proxy, set by set_default_proxy.)rJrKr1r!rget_default_proxyrYs  ##r!c^tjrt|j_dStd)aAttempts to replace a module's socket library with a SOCKS socket. Must set a default proxy using set_default_proxy(...) first. This will only work on modules that import socket directly into the namespace; most of the Python Standard Library falls into this category.zNo default proxy specifiedN)rJrKsocketr3)modules r wrap_moduler]s. >)  <===r!c |\} } | dr| d} |r*|dr|d}d} tj||dtjD]} | \}}}}}d} t |||}| r| D] }|j| t|ttfr| ||r| |||||||r| || | | f|cS#tjtf$r$}|} |r|d}Yd}~d}~wwxYw| r| tjd)acreate_connection(dest_pair, *[, timeout], **proxy_args) -> socket object Like socket.create_connection(), but connects to proxy before returning the socket object. dest_pair - 2-tuple of (IP/hostname, port). **proxy_args - Same args passed to socksocket.set_proxy() if present. timeout - Optional socket timeout value, in seconds. source_address - tuple (host, port) for the socket to bind to as its source address before connecting (only for compatibility) [z[]Nrzgai returned empty list.) startswithstripr[ getaddrinfo SOCK_STREAMrJ setsockopt isinstanceintfloat settimeout set_proxybindconnecterrorr#close) dest_pairtimeoutsource_addressrL proxy_addr proxy_port proxy_rdnsproxy_usernameproxy_passwordsocket_options remote_host remote_porterrrfamily socket_typeproto canonnamesasockoptrs rcreate_connectionrs$ )Kc"".!''-- ,j++C00,%%d++ C  J6;M N N451 UIr fk599D *)**C#DOS)))'C<00 )((( ?z:z:-~??? * .))) LL+{3 4 4 4KKK j)   C       ,1 2 22s BD--E'E""E'c,eZdZdZdZeZdS) _BaseSocketzBAllows Python 2 delegated methods such as send() to be overridden.ctj|g|Ri|t|_|jD]*}t |||j|<t ||+dSr*) _orig_socketr(dict _savedmethods _savenamesgetattrdelattrrposkwnames rr(z_BaseSocket.__init__std/S///B///!VVO  D'.tT':':D t $ D$      r!N)r-r.r/r0r(listrr1r!rrrs0LL   JJJr!rcfdS)Nc*|j|i|Sr*)rrs rz_makemethod..s$ socket object Open a SOCKS enabled socket. The parameters are the same as those of the standard socket init. In order for SOCKS to work, you must specify family=AF_INET and proto=0. The "type" argument must be either SOCK_STREAM or SOCK_DGRAM. Nrc>|tjtjfvr$d}t||t t |j|||g|Ri|d|_|j r |j |_ nd|_ d|_ d|_ d|_ dS)Nz0Socket type must be stream or datagram, not {!r})NNNNNN)r[rc SOCK_DGRAM ValueErrorr'superrJr( _proxyconnrKproxyproxy_socknameproxy_peername_timeout)rr{typer}rrr% __class__s rr(zsocksocket.__init__s *F,=> > >DCSZZ--.. .(j$(uNtNNNvNNN   >+DJJ=DJ"" r!cd}t||krN||t|z }|std||z }t||kN|S)zReceive EXACTLY the number of bytes requested from the file object. Blocks until the required number of bytes have been received.r!Connection closed unexpectedly)lenreadr3)rfilecountdatads r_readallzsocksocket._readallsq$ii% %#d))+,,A J'(HIII AID $ii%  r!c||_ |}tt||jdS#t j$rYdSwxYwr*)rget_proxy_peernamerrJrhr[rl)rropeerrs rrhzsocksocket.settimeoutsj  **,,D *d # # . .t} = = = = =|    DD sAA A A c|jSr*)rr+s rrzsocksocket.gettimeout#s }r!cb|r|ddS|ddS)Ng)rh)rvs rrzsocksocket.setblocking&s:  ! OOD ! ! ! ! ! OOC r!Tcx|||||r|nd|r|ndf|_dS)a Sets the proxy to be used. proxy_type - The type of the proxy to be used. Three types are supported: PROXY_TYPE_SOCKS4 (including socks4a), PROXY_TYPE_SOCKS5 and PROXY_TYPE_HTTP addr - The address of the server (IP or DNS). port - The port of the server. Defaults to 1080 for SOCKS servers and 8080 for HTTP proxy servers. rdns - Should DNS queries be performed on the remote side (rather than the local side). The default is True. Note: This has no effect with SOCKS4 servers. username - Username to authenticate with to the server. The default is no authentication. password - Password to authenticate with to the server. Only relevant when username is also provided.N)rIr)rrLrMrNrOrPrQs rrizsocksocket.set_proxy,sH"!$d+3=hoo'''+3=hoo'''? r!cTd|vr|d|d<|j|i|SrT)rVri)rrrs rsetproxyzsocksocket.setproxyAs: & #)::k#:#:F< t~t.v...r!c|j\}}}}}}|r|jtjkrt j|g|Ri|S|jrtjtd|tkrd} tjt| tt|j|i|| \} } d| f} t |_|} |j| d}||j|| \} }| \}} |\} } tt||| ftt||jd|_dS)zVImplements proxy connection for UDP sockets. Happens during the bind() phase.z"Socket already bound to an addressz'UDP only supported by SOCKS5 proxy type0)0.0.0.0rN)rrr[rrrjrrlrrrrrJ getsockname _proxy_addrrk_SOCKS5_requestrhrr)rrrrLrqrrrOrPrQr%_rNdstr UDP_ASSOCIATErelayhostrs rrjzsocksocket.bindFs Z ZT8  7TY&*;;;$T6C666266 6 ? M,v'KLL L   ;C,z3// /$j$$c0R000 ""$$4Dk&..  "" &&& ''LL5a4 j$''t 555 j$**4=999,r!c|jtjkr$tt|j|g|Ri|S|js|d|d}|dd}t}d}| |d}| || ||tt|j | |zg|Ri|} | | z S)Nrs)rr[rrrJrrrjrwrite_write_SOCKS5_addressrgetvaluetell) rbytesrraddressflagsheaderRSV STANDALONEsentrs rrzsocksocket.sendtoks 9) ) )15T**1%I$III&II I  IIg   r(SbS  S  Z    ""7F333+uZ&&+FOO,=,=,E6666.466fkkmm##r!c |jtjkr|j|||jfi|St t |j||fi|Sr*)rr[rrrrrJr)rrrrrs rrzsocksocket.sends[ 9) ) )4;ueT-@KKFKK K/5T**/uGGGG Gr!c|jtjkr)tt|||S|js|dttt| |dz|}| dt| d}t|rtd||\}}|jr0|j\}}||ks|d|fvrtjt$d| |||ffS)Nrir r zReceived UDP packet fragmentrzPacket filtered)rr[rrrJrrrjrrseekr rordNotImplementedError_read_SOCKS5_addressrrlr) rbufsizerbuffragfromhostfromportpeerhostpeerportrs rrzsocksocket.recvfroms3 9) ) )T**33GUCC C  IIg   eJ--227T>5IIJJ Hxx{{ t99 F%&DEE E!66s;;(   >!%!4 Hh8##x8}'D'Dl6+<===!!Hh#788r!c&|j|i|\}}|Sr*)r)rrrrrs rrzsocksocket.recvs! 4=#,,,q r!c|jr|jtt|Sr*)rrmrrJ)rrs rrmzsocksocket.closes= ? $ O ! ! # # #Z&&,,...r!c|jS)z:Returns the bound IP address and port number at the proxy.)rr+s rget_proxy_socknamezsocksocket.get_proxy_socknames ""r!c*|S)z> Returns the IP and port number of the proxy. ) getpeernamer+s rrzsocksocket.get_proxy_peernames!!!r!c|jS)zwReturns the IP address and port number of the destination machine. Note: get_proxy_peername returns the proxy.)rr+s r get_peernamezsocksocket.get_peernames ""r!cRd}||||\|_|_dS)z7Negotiates a stream connection through a SOCKS5 server.N)rrr)r dest_addrCONNECTs r_negotiate_SOCKS5zsocksocket._negotiate_SOCKS5s4373G3G '94&4&0T000r!c|j\}}}}}} |d} |dd} |r| r| dn| d| || d} | dddkrt d | ddd kr|r| st d | d tt| z|ztt|  z| z| || d} | ddd krt d | ddd krt dn:| ddd kr,| dddkrt dt d | d|zd z| || }| || d}|dddkrt d t|dd}|dkr>t |d}td|||| }t#t$||j||f| | S#| | wxYw)z Send SOCKS5 request with given command (CMD field) and address (DST field). Returns resolved DST address that was used. wbrbrssr r %SOCKS5 proxy server sent invalid datazPNo username/password supplied. Server requested username/password authenticationrrzSOCKS5 authentication failedz7All offered SOCKS5 authentication methods were rejectedr Unknown error {:#04x}: {})rmakefilerflushrr3r9chrrrIrr SOCKS5_ERRORSgetr;r'rrrJrhrrm)rconncmdrrLrMrNrOrPrQwriterreader chosen_auth auth_statusresolvedrespstatusrlbndrs rrzsocksocket._SOCKS5_requests <@:8 D$ht$$tQ''W  .H . 01111 _--- LLNNN--22K1Q37**(;=== 1Q37**!=X=*+<=== Ws3x=='9'9'@'@'B'BB'("3x==1188::; (())) "mmFA66 qs#w..+?AAAqs#w..)*HIII/QqS!W,,qs#w..)$%%%,?AAA LL30 1 1 111#v>>H LLNNN==++DAaCyG##';===ac^^F~~%))&/BB!-"6"6vu"E"EFFF++F33C *d # # . .t} = = =c? LLNNN LLNNNN LLNNN LLNNNNs KL,,*Mc |\}}|j\}}}}}} tjdtjdi} tjtjfD]} tj| |} || | | ztj| | }|tjd|||fcS#tj $rYwxYw|r]| d} |dtt|  z| zntj ||tjtjtjtj}|d}|d} |dd}tj| |} || | | ztj| | }|tjd|||fS)z~ Return the host and port packed for the SOCKS5 protocol, and the resolved address as a tuple object. r>HidnarrrC)rr[AF_INETAF_INET6 inet_ptonr inet_ntopstructpackrlrIrrrb AF_UNSPECrc IPPROTO_TCP AI_ADDRCONFIG)rrMrrrNrLrrOrPrQfamily_to_byter{ addr_bytes host_bytes addresses target_addrs rrz socksocket._write_SOCKS5_address s  d59Z2 Aq$( .'6?GL ~v7  F #-fd;;  >&1J>???' ;; 6;tT22333Tz!!!<      8V,,J JJwS__!5!5!!>>K L L L L*4v7G+1+=+1+=+1+?AAI $A,K ^Fq>!$D)&$77J JJ~f- : ; ; ;#FJ77D 6;tT**+++Tzs A3B??CCc||d}|dkr)tj||d}n|dkr:||d}||t|}nI|dkr4tjtj||d}nt dtjd||d d }||fS) Nr rrCrrrrr r) rr[ inet_ntoarr r r3r unpack)rratyprMlengthrNs rrzsocksocket._read_SOCKS5_addressMs}}T1%% 7??#DMM$$:$:;;DD W__]]4++F==s6{{33DD W__#FOT]]45L5LMMDD#$KLL L}T4==q#9#9::1=Tzr!c"|j\}}}}}}|d} |dd} d} tj|} nB#tj$r0|rd} d} n&tjtj|} YnwxYw| tjddd || | |r| || d | r+| | d d z| | | d } | dd d krtd t| d d}|dkr>t|d}t!d||tj| ddtjd| dddf|_| rtj| |f|_n ||f|_| | dS#| | wxYw)z0Negotiates a connection through a SOCKS4 server.rrrFsTz>BBHrCr rrrGz%SOCKS4 proxy server sent invalid datar ZrrNr)rrr[ inet_atonrl gethostbynamerr rrIrrr3r SOCKS4_ERRORSrr=r'rrrrrm)rr dest_portrLrMrNrOrPrQrrremote_resolverrrrls r_negotiate_SOCKS4zsocksocket._negotiate_SOCKS4\s;?:8 D$ht$$tQ''4 "N 9#-i88 < 9 9 99!4J%)NN!'!1,Y77"9"9J  9 LLVT4CC D D D LL $ $ $ ' X&&& LL ! ! !  A Y--f55?@@@ LLNNN==++DAaCyG##';===ac^^F~~%))&/BB!-"6"6vu"E"EFFF$*#3DH#=#=#)=tAaCy#A#A!#D#FD  ;&,&6z&B&BI&M##&/&:# LLNNN LLNNNNN LLNNN LLNNNNs/I$AI$"+HtOh,F"G"G!H I I I G$$$ W\\,//000}}mmoo   F#$DEE E O-8->->sA-F-F *E;  O O O#$MNN N O(( D#BDD D Hk**KK H H HFHH H H #  OOK<>Q  )A,"9"9#">"> ,A!$Y 011 1 ) 9 9) ) )? # '""",Y77II%%i%&*##'0)&<# FZ ZT8 9tUm44 Dy>>Q&& '!)S11'$BDD D j$**4=999  "+D  *d # # . .t} = = = *d # # + +Y ,B C C C F%%'' #  *d # # + +J 7 7 7&  3J?  $ 955555<   # JJLLL+NEBBBK     9|    JJLLL )3& J&~~j*EE !6z!B7>>~DPRR /3666*3666  s1(I")HI(H<<#I"K81BK33K8c ||ddS#t$r}|jr |jcYd}~Sd}~wwxYw)a& https://docs.python.org/3/library/socket.html#socket.socket.connect_ex Like connect(address), but return an error indicator instead of raising an exception for errors returned by the C-level connect() call (other problems, such as "host not found" can still raise exceptions). T)r;rN)rkOSErrorerrno)rrnrs r connect_exzsocksocket.connect_ex6sa  LLL 6 6 61   w w  s > 9>9>c|j\}}}}}}|pt|}|std||fS)zD Return proxy address to connect to as tuple object zInvalid proxy type)r DEFAULT_PORTSrr3)rrLrqrrrOrPrQs rrzsocksocket._proxy_addrFsU Z ZT8 @=#4#4Z#@#@  :#$899 9:%%r!NNNTNN)rr*)*r-r.r/r0rKr[r rcr(rrhrrrirrjrrrrrmrgetproxysocknamergetproxypeernamerrrrrrr$r4rrrr7r rkrBr __classcell__)rs@rrJrJsM$n63E$   !!! EI*.????*/// #-#-#-#-#-J$$$$$(HHHHHH 999999(///// ###*""" *### K&&& `````D+++Z   :::x9.9.9.x"#4!#4 ZZZZZZx    & & & & & & &r!rJrE) NNNNNTNNN)Dbase64rcollections.abcr ImportError collectionsrArrrriorloggingosr r[r sys __version__r version_info win_inet_pton getLoggerr-r9PROXY_TYPE_SOCKS4rPROXY_TYPE_SOCKS5rPROXY_TYPE_HTTPr PROXY_TYPESrzipvalueskeysr8 _orgsocketrr IOErrorr#r3r7r9r;r=r?r!rrDrRrWrYgetdefaultproxyr] wrapmodulerrrrmethodrerr*setattrrJr1r!rrbs%(((((((%%%$$$$$$$$%,,,,,,,,,,   7d??s'&00H HHHk FHH HHg!!FF$64@@ SS!3!3!5!5{7G7G7I7IJJKK"M) \&                     :        j        *        *            '  !   ) -     4 &   vtT48 CG.2IIII... $$$$ > > > 482626:>%) :3:3:3:3z     &-   III 2 6 6D W[$ - -F :fh ' '6%%d+++ T;;t#4#4555^ &^ &^ &^ &^ &^ &^ &^ &^ &^ &s 'A,,A=