3
2;Z~ @ sX d Z ddlmZ ddlmZ ddlmZmZmZ ddl Z ddl
m Z ddl Z ddl
Z
ddl
mZ ddlZddlZddlZdZe
jd krejdCk ry ddlZW n ek
r ed Y nX e jeZd ZZd
Z Z d
Z Z ee e dZ e!e"e j# e j$ Z%ej Z&Z'dd Z(G dd de)Z*G dd de*Z+G dd de*Z,G dd de*Z-G dd de*Z.G dd d e*Z/G d d d e*Z0d d d!d"Z1d#d$d%d&d'd(d)d*d+Z2ed,e d,e d-iZ3dDd/d0Z4d1d2 Z5d3d4 Z6e6Z7d5d6 Z8e8Z9dEd7d8Z:G d9d: d:ejZ;d;d< Ze?e>ese;j@jAe eBe;ee 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
) b64encode)Callable)
EOPNOTSUPPEINVALEAGAINN)BytesIO)SEEK_CURz1.6.7nt z8To run PySocks on Windows you must install win_inet_pton )SOCKS4SOCKS5HTTPc s t j fdd}|S )Nc sp | d }zNy$|j }|dkr&|jd | |S tk
rR } z W Y d d }~X nX W d |dkrj|jd X d S )Nr TF)
gettimeout setblocking Exception)argskwargsselfZ _is_blockinge)function /usr/lib/python3.6/socks.pywrapper] s
z"set_self_blocking..wrapper) functoolswraps)r r r )r r set_self_blocking[ s r c @ s" e Zd ZdZdddZdd ZdS )
ProxyErrorz4Socket_err contains original socket.error exception.Nc C s( || _ || _|r$| j dj|7 _ d S )Nz: {0})msg
socket_errformat)r r r r r r __init__p s zProxyError.__init__c C s | j S )N)r )r r r r __str__w s zProxyError.__str__)N)__name__
__module__ __qualname____doc__r" r# r r r r r n s
r c @ s e Zd ZdS )GeneralProxyErrorN)r$ r% r& r r r r r( { s r( c @ s e Zd ZdS )ProxyConnectionErrorN)r$ r% r& r r r r r) s r) c @ s e Zd ZdS )SOCKS5AuthErrorN)r$ r% r& r r r r r* s r* c @ s e Zd ZdS ) SOCKS5ErrorN)r$ r% r& r r r r r+ s r+ c @ s e Zd ZdS ) SOCKS4ErrorN)r$ r% r& r r r r r, s r, c @ s e Zd ZdS ) HTTPErrorN)r$ r% r& r r r r r- s 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)[ \ ] z General 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
i8 i Tc C s. | ||||r|j nd|r"|j 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_typeaddrportrdnsusernamepasswordr r r set_default_proxy s r? c O s d|kr|j d|d< t| |S )N proxytyper9 )popr? )r r r r r setdefaultproxy s rB c C s t jS )z4Returns the default proxy, set by set_default_proxy.)r7 r8 r r r r get_default_proxy s rC c C s t jrt | j_ntddS )a Attempts 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)r7 r8 socketr( )moduler r r wrap_module s
rF c
C s8 | \}
} |
j dr |
jd}
|r4|j dr4|jd}d} xtj||dtjD ]}
|
\}}}}}d}yxt|||}| rx| D ]}|j| qzW t|tt fr|j
| |r|j |||||| |r|j | |j
|
| f |S tjtfk
r } z |} |r
|j d}W Y dd}~X qLX qLW | r*| tjddS )a create_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[]Nr zgai returned empty list.)
startswithstriprD getaddrinfo SOCK_STREAMr7 Z
setsockopt
isinstanceintfloat
settimeout set_proxybindconnecterrorr) close) dest_pairtimeoutZsource_addressr9
proxy_addr
proxy_portZ
proxy_rdnsZproxy_usernameZproxy_passwordZsocket_optionsZ remote_hostZ remote_porterrrfamilyZ socket_typeprotoZ canonnameZsaZsockoptr r r r create_connection s>
r^ c @ s e Zd ZdZdd Ze ZdS ) _BaseSocketzBAllows Python 2 delegated methods such as send() to be overridden.c O sH t j| f|| t | _x(| jD ] }t| || j|< t| | q"W d S )N) _orig_socketr" dict
_savedmethods
_savenamesgetattrdelattr)r poskwnamer r r r"
s
z_BaseSocket.__init__N)r$ r% r& r' r" listrc r r r r r_ s r_ c s fddS )Nc s | j ||S )N)rb )r rf rg )rh r r s z _makemethod..r )rh r )rh r _makemethod s rk sendtosendrecvfromrecvc s. e Zd ZdZdZejejdf fdd Zdd Z fdd Z
d
d Z d d
Z d5ddZ
dd Z fddZ fddZd6 fdd Zd7 fdd Zdd Z fd d Zd d ZeZd!d" ZeZd#d$ ZeZd%d& Z fd'd(Z d)d* Z d+d, Z d-d. Z d/d0 Z e!e e"ee#e iZ$e% fd1d2Z&d3d4 Z' Z(S )8r7 a2 socksocket([family[, type[, proto]]]) -> 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.
Nr c sp |t jt jfkr"d}t|j|tt| j|||f|| d | _| j rT| j | _
nd| _
d | _ d | _ d | _
d S )Nz0Socket type must be stream or datagram, not {!r})NNNNNN)rD rK
SOCK_DGRAM
ValueErrorr! superr7 r"
_proxyconnr8 proxyproxy_socknameproxy_peername_timeout)r r[ typer\ r r r ) __class__r r r" 1 s
zsocksocket.__init__c C s@ d}x6t ||k r:|j|t | }|s0td||7 }qW |S )zReceive EXACTLY the number of bytes requested from the file object.
Blocks until the required number of bytes have been received. z Connection closed unexpectedly)lenreadr( )r filecountdatadr r r _readallC s zsocksocket._readallc s@ || _ y | j }tt| j| j W n tjk
r: Y nX d S )N)rw get_proxy_peernamerr r7 rO rD rS )r rV Zpeer)ry r r rO O s zsocksocket.settimeoutc C s | j S )N)rw )r r r r r X s zsocksocket.gettimeoutc C s |r| j d n
| j d d S )Ng )rO )r vr r r r [ s zsocksocket.setblockingTc C s. |||||r|j nd|r"|j 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)r6 rt )r r9 r: r; r< r= r> r r r rP a s zsocksocket.set_proxyc O s" d|kr|j d|d< | j||S )Nr@ r9 )rA rP )r r r r r r setproxyv s zsocksocket.setproxyc s | j \}}}}}}| s$| jtjkr6tj| f||S | jrHtjtd|t kr`d} tjt
| t t | j|| | j
\}
} d| f} t | _| j }
| jj|
d}| j| j|| \}
}|
\}}
|\}
} t t | j|| f t t | j| j d| _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.0r N)r r )rt rx rD rp r` rQ rs rS r r r rr r7 Z getsockname _proxy_addrrR _SOCKS5_requestrO rw ru )r rf rg r9 rW rX r< r= r> r _r; dstrt Z
UDP_ASSOCIATEZrelayhost)ry r r rQ { s* zsocksocket.bindc
s | j tjkr$tt| j|f||S | js4| jd |d }|d d }t }d}|j | d}|j | | j
|| tt| j |j | f||} | |j
S ) N r r s )r r r )rx rD rp rr r7 rl rs rQ r write_write_SOCKS5_addressrm getvaluetell)
r bytesr r ZaddressflagsheaderZRSVZ
STANDALONEZsent)ry r r rl s
zsocksocket.sendtoc s: | j tjkr | j||| jf|S tt| j||f|S d S )N)rx rD rp rl rv rr r7 rm )r r r r )ry r r rm s zsocksocket.sendc s | j tjkr tt| j||S | js.| jd ttt| j |d |}|j
dt |j d}t
|rntd| j|\}}| jr| j\}}||ks|d|fkrtjtd|j |||ffS ) Nr r i r r z Received UDP packet fragmentzPacket filtered)r r )rx rD rp rr r7 rn rs rQ r ro seekr r| ordNotImplementedError_read_SOCKS5_addressrv rS r ) r bufsizer ZbufZfragZfromhostZfromportZpeerhostZpeerport)ry r r rn s
zsocksocket.recvfromc O s | j ||\}}|S )N)rn )r rf rg r r r r r ro s zsocksocket.recvc s | j r| j j tt| j S )N)rs rT rr r7 )r )ry r r rT s
zsocksocket.closec C s | j S )z:Returns the bound IP address and port number at the proxy.)ru )r r r r get_proxy_sockname s z socksocket.get_proxy_socknamec C s | j S )z>
Returns the IP and port number of the proxy.
) getpeername)r r r r r s z socksocket.get_proxy_peernamec C s | j S )zwReturns the IP address and port number of the destination machine.
Note: get_proxy_peername returns the proxy.)rv )r r r r get_peername s zsocksocket.get_peernamec G s d}| j | ||\| _| _dS )z7Negotiates a stream connection through a SOCKS5 server. N)r rv ru )r dest_addrZCONNECTr r r _negotiate_SOCKS5 s z socksocket._negotiate_SOCKS5c s | j \}}}}}} |jd}
|jdd} z|r@| r@|
jd n
|
jd |
j | j| d} | dd dkrvtd | dd d
kr|
jd tt|j | tt| j | |
j | j| d}
|
dd d krtd |
dd d krt d
n6| dd d kr4| dd dkr,t dntd |
jd| d | j
||
}|
j | j| d}|dd dkrtd t |dd }|dkrt j
|d}tdj||| j| }tt| j| j ||fS | j |
j X dS )z
Send SOCKS5 request with given command (CMD field) and
address (DST field). Returns resolved DST address that was used.
wbrbr s s r r z%SOCKS5 proxy server sent invalid data r r z SOCKS5 authentication failed z7All offered SOCKS5 authentication methods were rejectedr
z
Unknown errorz
{0:#04x}: {1}N)rt makefiler flushr r( chrr{ r6 r* r r
SOCKS5_ERRORSgetr+ r! r rr r7 rO rw rT )r Zconncmdr r9 r: r; r< r= r> writerreaderZ chosen_authZ auth_statusZresolvedrespstatusrS Zbnd)ry r r r sX
.
zsocksocket._SOCKS5_requestc C sH |\}}| j \}}}}}} tjdtjdi}
xrtjtjfD ]b} yDtj| |} |j|
| | tj| | }|jtjd| ||fS tj k
r w8Y q8X q8W |r|j
d}
|jdt t |
j
|
nbtj
||tjtjtjtj}|d }|d } |d d }tj| |} |j|
| | tj| | }|jtjd| ||fS )z~
Return the host and port packed for the SOCKS5 protocol,
and the resolved address as a tuple object.
r z>Hidnar r r1 )rt rD AF_INETAF_INET6Z inet_ptonr inet_ntopstructpackrS r6 r r{ rJ Z AF_UNSPECrK Z IPPROTO_TCPZ
AI_ADDRCONFIG)r r: r} r r; r9 r r< r= r> Zfamily_to_byter[
addr_bytesZ
host_bytesZ addressesZ target_addrr r r r M s6
z socksocket._write_SOCKS5_addressc C s | j |d}|dkr(tj| j |d}nN|dkrN| j |d}| j |t|}n(|dkrntjtj| j |d}ntdtjd| j |d d
}||fS ) Nr r r1 r r z%SOCKS5 proxy server sent invalid dataz>Hr r ) r rD inet_ntoar r r r( r unpack)r r} Zatypr: Zlengthr; r r r r z s z socksocket._read_SOCKS5_addressc C s | j \}}}}}}| jd} | jdd}
zLd} ytj|} W n4 tjk
rr |r^d} d} ntjtj|} Y nX | jtjddd | | j| |r| j| | jd
| r| j|j d d
| j
| j |
d }
|
dd d
krt d
t
|
d d }|dkr&tj|d}tdj||tj|
dd tjd|
dd d f| _| rjtj| |f| _n
||f| _W d|
j | j X dS )z0Negotiates a connection through a SOCKS4 server.r r r Fs Tz>BBHr1 r r r r5 z%SOCKS4 proxy server sent invalid datar Z z
Unknown errorz
{0:#04x}: {1}Nz>H)rt r rD Z inet_atonrS
gethostbynamer r r r6 r r r( r
SOCKS4_ERRORSr r, r! r r ru rv rT )r r dest_portr9 r: r; r< r= r> r r Zremote_resolver r r rS r r r _negotiate_SOCKS4 sH
z socksocket._negotiate_SOCKS4c C sj | j \}}}}}}|r|ntj|}d|jd d t|j d d|jd g} |rv|rv| jdt|d | | jd | jdj| | j }
|
j
} |
j | st dy| j
d d
\} }
}W n tk
r t d Y nX | jd st d
y t|
}
W n tk
r$ tdY nX |
dkrVdj|
|}|
dkrN|d7 }t|d| _||f| _dS )zwNegotiates a connection through an HTTP server.
NOTE: This currently only supports HTTP CONNECT-style proxies.s CONNECT r :s HTTP/1.1s Host: s Proxy-Authorization: basic s
z Connection closed unexpectedly 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 status z{0}: {1} za
[*] Note: The HTTP proxy server may not be supported by PySocks (must be a CONNECT tunnel proxy) 0.0.0.0r N)r r r )r r )rt rD r r6 strappendr Zsendalljoinr readlinerT r( splitrq rH rM r- r! ru rv )r r r r9 r: r; r< r= r> Z http_headersZfobjZ status_liner\ Z status_codeZ
status_msgrS r r r _negotiate_HTTP sB
zsocksocket._negotiate_HTTPc s t |dks|d jdr,tjdt| |\}}| jtjkr~| jsP| jd
tj |}|dkrp| rpd| _
n
||f| _
dS | j \}}}}}} t |t
tf st |dks| st |t rtdtt| j| j |dkr|| _
tt| j| j tt| j||f dS | j }ytt| j| W nn tjk
r }
zN| j |\}}d j||} t| } d
j| | }
tjd |
|
t |
|
W Y dd}
~
X ntX y| j | }|| || W nV tjk
r }
z| j td |
W Y dd}
~
X n t k
r | j Y nX dS )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 r rG z PySocks doesn't support IPv6: %sr z0.0.0.0Nz0Invalid destination-connection (host, port) pairz{0}:{1}z!Error connecting to {0} proxy {1}z
%s due to: %sz Socket error)r r ) r{ rH rD rS r rx rp rs rQ r rv rt rL ri tuplerM r( rr r7 rO rw rR r rT r! PRINTABLE_PROXY_TYPESlogdebugr) _proxy_negotiatorsr )r rU r r r9 rW rX r< r= r> rS Z proxy_serverZprintable_typer Z negotiate)ry r r rR s\
zsocksocket.connectc C s4 | j \}}}}}}|p tj|}|s,td||fS )zD
Return proxy address to connect to as tuple object
zInvalid proxy type)rt
DEFAULT_PORTSr r( )r r9 rW rX r< r= r> r r r r ] s
zsocksocket._proxy_addr)NNNTNN)r )r ))r$ r% r& r' r8 rD r rK r" r rO r r rP r rQ rl rm rn ro rT r Zgetproxysocknamer Zgetproxypeernamer r r r r r r r r
r r r r rR r
__classcell__r r )ry r r7 & sD
% Z-<<Wr7 )r
r )NNNTNN) NNNNNTNNN)rl rm rn ro )Dr' base64r collectionsr errnor r r r ior Zloggingosr rD r sys __version__rh version_infoZ
win_inet_pton ImportErrorZ getLoggerr$ r ZPROXY_TYPE_SOCKS4r
ZPROXY_TYPE_SOCKS5r ZPROXY_TYPE_HTTPr Z PROXY_TYPESra zipvalueskeysr Z
_orgsocketr` r IOErrorr r( r) r* r+ r, r- r r r r? rB rC ZgetdefaultproxyrF Z
wrapmoduler^ r_ rk rd methodrL rc r setattrr7 r r r r 7 s
8