e dZddlmZddlZejrddlZnddlmZddl m Z m Z m Z ddl Z ddlmZddlZddlZddlmZddlZddlZddlZdZej.dk(rej0d kr ddlZej6eZd xZZd xZ Z!d xZ"Z#ee!e#dZ$e%e&e$jOe$jQZ)ej&xZ*Z+dZ,Gdde-Z.Gdde.Z/Gdde.Z0Gdde.Z1Gdde.Z2Gdde.Z3Gdde.Z4ddd d!Z5d"d#d$d%d&d'd(d)d*Z6ed+e!d+e#d,iZ7 d8d-Z8d.Z9d/Z:e:Z;d0ZGd2d3ej&Z?d4Z@d5D]KZeAe?edZBeCeBejr!e?jjeeGe?ee@eMGd6d7e?ZHy#e$r ed wxYw):a SocksiPy - Python SOCKS module. Copyright 2006 Dan-Haim. All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. 3. Neither the name of Dan Haim nor the names of his contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY DAN HAIM "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL DAN HAIM OR HIS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. This module provides a standard socket-like interface for Python for tunneling connections through SOCKS proxies. =============================================================================== Minor modifications made by Christopher Gilbert (http://motomastyle.com/) for use in PyLoris (http://pyloris.sourceforge.net/) Minor modifications made by Mario Vilas (http://breakingcode.wordpress.com/) mainly to merge bug fixes found in Sourceforge Modifications made by Anorov (https://github.com/Anorov) -Forked and renamed to PySocks -Fixed issue with HTTP proxy failure checking (same bug that was in the old ___recvall() method) -Included SocksiPyHandler (sockshandler.py), to be used as a urllib2 handler, courtesy of e000 (https://github.com/e000): https://gist.github.com/869791#file_socksipyhandler.py -Re-styled code to make it readable -Aliased PROXY_TYPE_SOCKS5 -> SOCKS5 etc. -Improved exception handling and output -Removed irritating use of sequence indexes, replaced with tuple unpacked variables -Fixed up Python 3 bytestring handling - chr(0x03).encode() -> b"" -Other general fixes -Added clarification that the HTTP proxy connection method only supports CONNECT-style tunneling HTTP proxies -Various small bug fixes ) b64encodeN) EOPNOTSUPPEINVALEAGAIN)BytesIO)SEEK_CURz1.6.7nt)rz8To run PySocks on Windows you must install win_inet_ptonr )SOCKS4SOCKS5HTTPcBtjfd}|S)Nc|d} |j}|dk(r|jd|i||dk(r|jdSS#t$r}d}~wwxYw#dk(r|jdwwxYw)NrTF) gettimeout setblocking Exception)argskwargsself _is_blockingefunctions )platform/bq/third_party/socks/__init__.pywrapperz"set_self_blocking..wrapperfsAw (??,Lq   &T,V, q   '!    q   '!s#-A AAAAA8) functoolswraps)rrs` rset_self_blockingrds%__X ( ( NceZdZdZddZdZy) ProxyErrorz4Socket_err contains original socket.error exception.Ncn||_||_|r%|xjdj|z c_yy)Nz: {0})msg socket_errformat)rr$r%s r__init__zProxyError.__init__ys/$  HHz2 2H r c|jSN)r$rs r__str__zProxyError.__str__s xxr r))__name__ __module__ __qualname____doc__r'r+r rr"r"ws>3r r"c eZdZy)GeneralProxyErrorNr,r-r.r0r rr2r2r r2c eZdZy)ProxyConnectionErrorNr3r0r rr6r6r4r r6c eZdZy)SOCKS5AuthErrorNr3r0r rr8r8r4r r8c eZdZy) SOCKS5ErrorNr3r0r rr:r:r4r r:c eZdZy) SOCKS4ErrorNr3r0r rr<r<r4r r<c eZdZy) HTTPErrorNr3r0r rr>r>r4r 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 i8ict|dr|j}t|dr|j}|||||r|nd|r |ft_ydft_y)zSets a default proxy. All further socksocket objects will use the default unless explicitly changed. All parameters are as for socket.set_proxy().encodeN)hasattrrH socksocket default_proxy) proxy_typeaddrportrdnsusernamepasswords rset_default_proxyrRs^ x""hx""h *D$,4$,4 @J:> @Jr cHd|vr|jd|d<t|i|SN proxytyperL)poprR)rrs rsetdefaultproxyrWs-f%zz+6| d -f --r c"tjS)z4Returns the default proxy, set by set_default_proxy.)rJrKr0r rget_default_proxyrYs  # ##r cdtjrt|j_ytd)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)rJrKsocketr2)modules r wrap_moduler]s% )  <==r c |\} } | jdr| jd} |r"|jdr|jd}d} tj||dtjD]} | \}}}}}d} t |||}| r| D]}|j |t|ttfr|j||r|j|||||||r|j||j| | f|cS| r| tjd#tjtf$r }|} |r|j!d}Yd}~d}~wwxYw)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_proxybindconnecterrorr6close) dest_pairtimeoutsource_addressrL proxy_addr proxy_port proxy_rdnsproxy_usernameproxy_passwordsocket_options remote_host remote_porterrrfamily socket_typeproto canonnamesasockoptrs rcreate_connectionrsb$ )Kc"!''- j++C0%%d+ C   J6;M;M N451 UIr fk59D)C#DOOS)*'C<0(z:z:-~? .) LL+{3 4K)O8  ,,1 22 23 C   sB D**E#EE#c$eZdZdZdZeZy) _BaseSocketzBAllows Python 2 delegated methods such as send() to be overridden.ctj|g|i|t|_|jD]'}t |||j|<t ||)yr)) _orig_socketr'dict _savedmethods _savenamesgetattrdelattrrposkwnames rr'z_BaseSocket.__init__sSd/S/B/!VOOD'.tT':D  t $ D$ $r N)r,r-r.r/r'listrr0r rrrsL Jr 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. Nrc6|tjtjfvrd}t|j |t t ||||g|i|d|_|jr|j|_ nd|_ d|_ d|_ d|_ y)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$(uNtNvN   ++DJ=DJ"" r cd}t||kr>|j|t|z }|s td||z }t||kr>|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)lenreadr2)rfilecountdatads r_readallzsocksocket._readallPsY$i% %#d)+,A'(HII AID $i%  r c||_ |j}tt||jy#t j $rYywxYwr))rget_proxy_peernamerrJrhr[rl)rropeerrs rrhzsocksocket.settimeout\sF  **,D *d .t}} =||   s-8A Ac|jSr))rr*s rrzsocksocket.gettimeoutes }}r cN|r|jdy|jdy)Ng)rh)rvs rrzsocksocket.setblockinghs OOD ! OOC r ct|dr|j}t|dr|j}|||||r|nd|r |f|_ydf|_y)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.rHN)rIrHr)rrLrMrNrOrPrQs rrizsocksocket.set_proxynsZ" 8X &__&( 8X &__&( $d"*h"*h6 046 r cVd|vr|jd|d<|j|i|SrT)rVri)rrrs rsetproxyzsocksocket.setproxys3 & #)::k#:F< t~~t.v..r c|j\}}}}}}|r|jtjk7rt j |g|i|S|j rtjtd|tk7rd} tjt| tt||i||j\} } d| f} t |_|j} |j j| d}|j!|j || \} }| \}} |\} } tt|?|| ftt|G|j$d|_y)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.bindsA ZZ ZT8 TYY&*;*;;$$T6C626 6 ??,,v'KL L  ;C,,z3/ / j$$c0R0 ""$4Dk&.  " & ''L5a4 j$'t 5 j$*4==9,r c|jtjk7rtt||g|i|S|j s|jd|d}|dd}t}d}|j|d}|j||j||tt|.|j|zg|i|} | |jz S)Nrs)rr[rrrJrrrjrwrite_write_SOCKS5_addressrgetvaluetell) rbytesrraddressflagsheaderRSV STANDALONEsentrs rrzsocksocket.sendtos 99)) )T1%I$I&I I IIg r(Sb  S  Z  ""7F3Z+FOO,=,E66.46fkkm##r c |jtjk(r|j|||jfi|St t |||fi|Sr))rr[rrrrrJr)rrrrrs rrzsocksocket.sendsN 99)) )4;;ueT-@-@KFK KT/uGG Gr c,|jtjk7rtt|||S|j s|jdttt|'|dz|}|jdt|jd}t|r td|j|\}}|j r4|j \}}||k7s|d|fvrtj"t$d|j|||ffS)Nrir r zReceived UDP packet fragmentrzPacket filtered)rr[rrrJrrrjrrseekrrordNotImplementedError_read_SOCKS5_addressrrlr) rbufsizerbuffragfromhostfromportpeerhostpeerportrs rrzsocksocket.recvfroms 99)) )T3GUC C IIg eJ27T>5IJ Hxx{ t9%&DE E!66s;(   !%!4!4 Hh8#x8}'Dll6+<==!Hh#788r c0|j|i|\}}|Sr))r)rrrrrs rrzsocksocket.recvs  4==#,,q r ct|jr|jjtt|Sr))rrmrrJ)rrs rrmzsocksocket.closes* ?? OO ! ! #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"|jS)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 cHd}|j|||\|_|_y)z7Negotiates a stream connection through a SOCKS5 server.N)rrr)r dest_addrCONNECTs r_negotiate_SOCKS5zsocksocket._negotiate_SOCKS5s)373G3G '94&0T0r c|j\}}}}}} |jd} |jdd} |r| r| jdn| jd| j|j | d} | dddk7r t d | ddd k(r| jd t t|jz|zt t| jz| z| j|j | d} | ddd k7r t d | ddd k7r1td | ddd k7r| dddk(r tdt d | jd|zd z|j|| }| j|j | d}|dddk7r t d t|dd}|dk7r1tj|d}tdj|||j!| }t"t$|O|j(||f| j+| j+S#| j+| j+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 datarrzSOCKS5 authentication failedz7All offered SOCKS5 authentication methods were rejectedr Unknown error {0:#04x}: {1})rmakefilerflushrr2chrrrHr8rr SOCKS5_ERRORSgetr:r&rrrJrhrrm)rconncmdrrLrMrNrOrPrQwriterreader chosen_auth auth_statusresolvedrespstatusrlbndrs rrzsocksocket._SOCKS5_requests <@::8 D$ht$tQ'O H 01 _- LLN--2K1Q7*(;== 1Q7* Ws3x='9'@'@'BB'("3x=188:; (() "mmFA6 q#w.+?AAq#w.)*HII Qq!W,q#w.)$%%,?AA LL30 111#v>H LLN==+DAayG#';==a^F~%))&/B!/"8"8"GHH++F3C *d .t}} =c? LLN LLN LLN LLNs HI"I>c|\}}|j\}}}}}} tjdtjdi} tjtjfD]p} tj| |} |j | | | ztj | | }|j tjd|||fcS|rI|jd} |j dtt| jz| zntj||tjtjtj tj"}|d}|d} |dd}tj| |} |j | | | ztj | | }|j tjd|||fS#tj$rYwxYw)z~ Return the host and port packed for the SOCKS5 protocol, and the resolved address as a tuple object. r>HidnarrrB)rr[AF_INETAF_INET6 inet_ptonr inet_ntopstructpackrlrHrrrb AF_UNSPECrc IPPROTO_TCP AI_ADDRCONFIG)rrMrrrNrLrrOrPrQfamily_to_byter{ addr_bytes host_bytes addresses target_addrs rrz socksocket._write_SOCKS5_address^s  d59ZZ2 Aq$( ..'6??GL ~~v7F #--fd;  >&1J>?'' ; 6;;tT23Tz! 8 V,J JJwS_!5!>K L**4v7G7G+1+=+=+1+=+=+1+?+?AI $A,K ^Fq>!$D))&$7J JJ~f- : ;##FJ7D 6;;tT*+Tz1<<  sA+G))H?Hc|j|d}|dk(r&tj|j|d}nx|dk(r.|j|d}|j|t|}nE|dk(r5tjtj |j|d}n t dtjd|j|d d }||fS) Nr rrBrrrrr r) rr[ inet_ntoarr r r2r unpack)rratyprMlengthrNs rrzsocksocket._read_SOCKS5_addresss}}T1% 7?##DMM$$:;D W_]]4+F==s6{3D W_##FOOT]]45LMD#$KL L}}T4==q#9:1=Tzr ct|j\}}}}}}|jd} |jdd} d} tj|} | j tjddd || j | |r| j || j d | r#| j |jd d z| j|j| d } | dd d k7r td t| d d}|dk7r1tj|d}t!dj#||tj$| ddtj&d| dddf|_| rtj$| |f|_n ||f|_| j-| j-y#tj$r3|rd} d} n(tjtj |} YwxYw#| j-| j-wxYw)z0Negotiates a connection through a SOCKS4 server.rrrFsTz>BBHrBr rrrFz%SOCKS4 proxy server sent invalid datar ZrrNr)rrr[ inet_atonrl gethostbynamerr rrHrrr2r SOCKS4_ERRORSrr<r&rrrrrm)rr dest_portrLrMrNrOrPrQrrremote_resolverrrrls r_negotiate_SOCKS4zsocksocket._negotiate_SOCKS4s;?::8 D$ht$tQ'4 "N 9#--i8  LLVT4C D LL $ X& LL !  Y--f5?@ LLN==+DAayG#';==a^F~%))&/B!/"8"8"GHH$*#3#3DH#=#)==tAay#A!#D#FD &,&6&6z&BI&M#&/&:# LLN LLN_<< 9!4J%)N!'!1!1,,Y7"9J  9\ LLN LLNs0HG EH AHHHH"H7cj|j\}}}}}}|r|ntj|}d|jdzdzt |jzdzd|jdzg} |r%|r#| j dt |dz|zz| j d|jdj| |j} | j} | j| s td | jd d \} } }| jd s td  t!| } | dk7r&dj%| |}| dvr|dz }t#|d|_||f|_y#t$r td wxYw#t$r t#dwxYw)zwNegotiates a connection through an HTTP server. NOTE: This currently only supports HTTP CONNECT-style proxies.sCONNECT r:s HTTP/1.1sHost: sProxy-Authorization: basic s r r z'HTTP proxy server sent invalid responsezHTTP/z0Proxy server does not appear to be an HTTP proxyz4HTTP proxy server did not return a valid HTTP statusz{0}: {1})iiiza [*] Note: The HTTP proxy server may not be supported by PySocks (must be a CONNECT tunnel proxy))s0.0.0.0rN)rr[r rHstrappendrsendalljoinrreadlinermr2splitrr`rfr>r&rr)rrr"rLrMrNrOrPrQ http_headersfobj status_liner} status_code status_msgrls r_negotiate_HTTPzsocksocket._negotiate_HTTPs<@::8 D$h!yf&:&:9&E4;;v. . 59~$$&')56  ((0 0     >"+HtOh,F"G!H I G$ W\\,/0}}mmo  #$DE E O-8->->sA-F *E; (#BD D Hk*K # %%k:>Eo-$%E" "-"Io1 O#$MN N O HFH H Hs F> FFF2ct|dk7s|djdr!tjdt |z|\}}|j tj k(rK|js|jdtj|}|dk(r |sd|_ y||f|_ y|j\}}}}}} t|ttfr t|dk7s|rt|ts t!dt"t$|O|j(|:||_ t"t$|O|j(t"t$|W||fy|j-} t"t$|W| |j.|} | |||y#tj$r!} |j1t!d | d} ~ wt2$r|j1wxYw#tj$rj} |j1|\}}d j5||} t6|} d j5| | }t8j;d || t=|| d} ~ wwxYw) z Connects to the specified destination through a proxy. Uses the same API as socket's connect(). To select the proxy server, use set_proxy(). dest_pair - 2-tuple of (IP/hostname, port). r rr_z PySocks doesn't support IPv6: %srrNz0Invalid destination-connection (host, port) pairz Socket errorz{0}:{1}z!Error connecting to {0} proxy {1}z %s due to: %s)rr`r[rlr)rrrrjr rrrertuplerfr2rrJrhrrkr_proxy_negotiatorsrmr"r&PRINTABLE_PROXY_TYPESlogdebugr6)rrnrr"rLrqrrrOrPrQ negotiaterl proxy_serverprintable_typer$rs rrkzsocksocket.connectsK y>Q )A,"9"9#">,,A!$Y 01 1 ) 9 99)) )?? '",,Y7II%i&*# (1)&<# ZZ ZT8 9tUm4y>Q& !)S1#BD D j$*4==9  "+D  *d .t}} = *d +Y ,B C %%'   *d +J 7  33J? $ 95<< ? '>>   -|| 3 JJL%/ "J $++J CL2:>N5<<^=IKC IIosE 2&sE2 2 3s10G.FG+1G  G+.I+A%I&&I+c||j\}}}}}}|xstj|}|s td||fS)zD Return proxy address to connect to as tuple object zInvalid proxy type)r DEFAULT_PORTSrr2)rrLrqrrrOrPrQs rrzsocksocket._proxy_addrnsL ZZ ZT8 @=#4#4Z#@ #$89 9:%%r NNNTNN)r))r,r-r.r/rKr[r rcr'rrhrrrirrjrrrrrmrgetproxysocknamergetproxypeernamerrrrrrr$r4r rrr7rrkr __classcell__)rs@rrJrJ3sM$nn63E3E$ ! EI*.62/ #-J$(H 9(/ #*" *# K& Xt+Z :x9.x"#4!#4 TTl &r rJr@) NNNNNTNNN)Ir/base64rsixPY2 collectionscollections_abccollections.abcabcerrnorrrriorloggingosrr[r sys __version__r version_info win_inet_pton ImportError getLoggerr,r9PROXY_TYPE_SOCKS4r PROXY_TYPE_SOCKS5rPROXY_TYPE_HTTPr PROXY_TYPESrzipvalueskeysr8 _orgsocketrrIOErrorr"r2r6r8r:r<r>r!rr?rRrWrYgetdefaultproxyr] wrapmodulerrrrmethodreCallablerr*setattrrJr0r rrcs[6p 77',,,   77d?s''&0H g!FF$64@ S!3!3!5{7G7G7IJK"MM) \&     :  j  *  *    '  !   ) -     4 &  vtT48 CG.2 @. $$ > 482626:>%) :3z &-- I 2D [$ -F fo66 7%%d+ T;t#45 3D &D & H FH HHs 3G G