a
G2]ny @ sr d dl mZ zd dlmZ W n ey: d dlmZ Y n0 d dlmZmZm Z d dl
Z
d dl m Z d dl
Z
d dlZd dlmZ d dlZd dlZd dlZdZejdkrejd k rz d dlZW n ey ed
Y n0 e
eZd ZZd Z Z d
Z Z ee e dZ e!e"e # e $ 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-iZ3d@d/d0Z4d1d2 Z5d3d4 Z6e6Z7d5d6 Z8e8Z9dAd7d8Z:G d9d: d:ejZ;d;d< Ze?e>es"e;j@Ae eBe;eed? d?e;ZCdS )B ) b64encode)Callable)
EOPNOTSUPPEINVALEAGAINN)BytesIO)SEEK_CURz1.7.1nt) r z8To run PySocks on Windows you must install win_inet_pton r
)SOCKS4SOCKS5HTTPc s t fdd}|S )Nc
s | d }z|z>| }|dkr&|d | i |W W |dkrH|d S tyn } z W Y d }~n
d }~0 0 W |dkr|d n|dkr|d 0 d S )Nr TF)
gettimeout setblocking Exception)argskwargsselfZ _is_blockingefunction )/usr/lib/python3.9/site-packages/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 d|7 _ d S )Nz: {})msg
socket_errformat)r r r! r r r __init__; s zProxyError.__init__c C s | j S N)r r r r r __str__B s zProxyError.__str__)N)__name__
__module__ __qualname____doc__r# r& r r r r r 9 s
r c @ s e Zd ZdS )GeneralProxyErrorNr' r( r) r r r r r+ F s r+ c @ s e Zd ZdS )ProxyConnectionErrorNr, r r r r r- J s r- c @ s e Zd ZdS )SOCKS5AuthErrorNr, r r r r r. N s r. c @ s e Zd ZdS ) SOCKS5ErrorNr, r r r r r/ R s r/ c @ s e Zd ZdS ) SOCKS4ErrorNr, r r r r r0 V s r0 c @ s e Zd ZdS ) HTTPErrorNr, r r r r r1 Z s r1 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| 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_typeaddrportrdnsusernamepasswordr r r set_default_proxys s rC c O s$ d|v r| d|d< t| i |S NZ proxytyper= )poprC )r r r r r setdefaultproxy~ s rF c C s t jS )z4Returns the default proxy, set by set_default_proxy.)r; r< r r r r get_default_proxy s rG 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)r; r< socketr+ )moduler r r wrap_module s
rJ c
C s8 | \}
} |
dr |
d}
|r4| dr4|d}d} t||dtjD ]}
|
\}}}}}d}zzt|||}| r| D ]}|j| qvt|tt fr|
| |r| |||||| |r| | |
|
| f |W S tjtfy } z"|} |r| d}W Y d}~qJd}~0 0 qJ| r*| tddS )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.)
startswithstriprH getaddrinfo SOCK_STREAMr; Z
setsockopt
isinstanceintfloat
settimeout set_proxybindconnecterrorr close) dest_pairtimeoutZsource_addressr=
proxy_addr
proxy_portZ
proxy_rdnsZproxy_usernameZproxy_passwordZsocket_optionsZ remote_hostZ remote_porterrrfamilyZ socket_typeprotoZ canonnameZsaZsockoptr r r r create_connection s@
rb c @ s e Zd ZdZdd Ze ZdS ) _BaseSocketzBAllows Python 2 delegated methods such as send() to be overridden.c O sJ t j| g|R i | t | _| jD ] }t| || j|< t| | q&d S r$ ) _orig_socketr# dict
_savedmethods
_savenamesgetattrdelattr)r poskwnamer r r r# s
z_BaseSocket.__init__N)r' r( r) r* r# listrg r r r r rc s rc c s fddS )Nc s | j |i |S r$ )rf )r rj rk rl r r z _makemethod..r rn r rn r _makemethod s rq )sendtosendrecvfromrecvc s< e Zd ZdZdZejejdf fdd Zdd Z fdd Z
d
d Z d d
Z d7ddZ
dd Z fddZ fddZd8 fdd Zd9 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%d: fd1d2 Z&e%d3d4 Z'd5d6 Z( Z)S );r; 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 sv |t jt jfvr"d}t||tt| j|||g|R i | d | _| j rZ| j | _
nd| _
d | _ d | _ d | _
d S )Nz0Socket type must be stream or datagram, not {!r})NNNNNN)rH rO
SOCK_DGRAM
ValueErrorr" superr; r#
_proxyconnr< proxyproxy_socknameproxy_peername_timeout)r r_ typer` r r r __class__r r r# s "
zsocksocket.__init__c C s< d}t ||k r8||t | }|s.td||7 }q|S )zReceive EXACTLY the number of bytes requested from the file object.
Blocks until the required number of bytes have been received.rp Connection closed unexpectedly)lenreadr+ )r filecountdatadr r r _readall s
zsocksocket._readallc s> || _ z | }tt| | j W n tjy8 Y n0 d S r$ )r} get_proxy_peernamerx r; rS rH rW )r rZ Zpeerr r r rS s zsocksocket.settimeoutc C s | j S r$ )r} r% r r r r # s zsocksocket.gettimeoutc C s |r| d n
| d d S )Ng )rS )r vr r r r & s zsocksocket.setblockingTc C s. |||||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)r: rz )r r= r> r? r@ rA rB r r r rT , s zsocksocket.set_proxyc O s& d|v r| d|d< | j|i |S rD )rE rT )r r r r r r setproxyA s zsocksocket.setproxyc s | j \}}}}}}|r"| jtjkr:tj| g|R i |S | jrLttd|t krdd} tt
| t t | j|i | |
\}
} d| f} t | _| }
| j|
d}| | j|| \}
}|
\}}
|\}
} t t | || f t t | | 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)rz r~ rH rv rd rU ry rW r r r rx r; Z getsockname _proxy_addrrV _SOCKS5_requestrS r} r{ )r rj rk r= r[ r\ r@ rA rB r _r? dstrz Z
UDP_ASSOCIATEZrelayhostr r r rU F s. zsocksocket.bindc
s | j tjkr*tt| j|g|R i |S | js:| d |d }|d d }t }d}| | d}| | |
|| tt| j | | g|R i |} | |
S )N r s )r~ rH rv rx r; rr ry rU r write_write_SOCKS5_addressrs getvaluetell)
r bytesr r ZaddressflagsheaderZRSVZ
STANDALONEsentr r r rr k s
zsocksocket.sendtoc sB | j tjkr$| j||| jfi |S tt| j||fi |S d S r$ )r~ rH rv rr r| rx r; rs )r r r r r r r rs s zsocksocket.sendc s | j tjkr tt| ||S | js.| d ttt| |d |}|
dt | d}t
|rntd| |\}}| jr| j\}}||ks|d|fvrttd| |||ffS )Nr i r r z Received UDP packet fragmentr zPacket filtered)r~ rH rv rx r; rt ry rU r ru seekr r ordNotImplementedError_read_SOCKS5_addressr| rW r ) r bufsizer buffragZfromhostZfromportZpeerhostZpeerportr r r rt s
zsocksocket.recvfromc O s | j |i |\}}|S r$ )rt )r rj rk r r r r r ru s zsocksocket.recvc s | j r| j tt| S r$ )ry rX rx r; r% r r r rX s
zsocksocket.closec C s | j S )z:Returns the bound IP address and port number at the proxy.)r{ r% r r r get_proxy_sockname s z socksocket.get_proxy_socknamec C s | S )z>
Returns the IP and port number of the proxy.
) getpeernamer% 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.)r| r% r r r get_peername s zsocksocket.get_peernamec G s d}| | ||\| _| _dS )z7Negotiates a stream connection through a SOCKS5 server. N)r r| r{ )r dest_addrZCONNECTr r r _negotiate_SOCKS5 s z socksocket._negotiate_SOCKS5c s | j \}}}}}} |d}
|dd} z|r@| r@|
d n
|
d |
| | d} | dd dkrvtd | dd d
kr|r| std |
d tt| | tt| | |
| | d}
|
dd d krtd |
dd d
krHtdn6| dd d
krH| dd dkr@tdntd |
d| d
|
||
}|
| | d}|dd dkrtd t |dd }|dkrt
|d}td||| | }tt| | j ||fW | |
S | |
0 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 %SOCKS5 proxy server sent invalid data zPNo username/password supplied. Server requested username/password authenticationr r z SOCKS5 authentication failed z7All offered SOCKS5 authentication methods were rejectedr
Unknown error {:#04x}: {}N)rz makefiler flushr r+ r. chrr r: r r
SOCKS5_ERRORSgetr/ r" r rx r; rS r} rX )r Zconncmdr r= r> r? r@ rA rB writerreaderZ chosen_authZ auth_statusZresolvedrespstatusrW Zbndr r r r sz
zsocksocket._SOCKS5_requestc C sJ |\}}| j \}}}}}} tjdtjdi}
tjtjfD ]h} zJt| |} ||
| | t| | }|td| ||fW S tj y Y q6Y q60 q6|r|
d}
|dt t |
|
nbt
||tjtjtjtj}|d }|d } |d d }t| |} ||
| | t| | }|td| ||fS )z~
Return the host and port packed for the SOCKS5 protocol,
and the resolved address as a tuple object.
r >Hidnar r r5 )rz rH AF_INETAF_INET6Z inet_ptonr inet_ntopstructpackrW r: r r rN Z AF_UNSPECrO Z IPPROTO_TCPZ
AI_ADDRCONFIG)r r> r r r? r= r r@ rA rB Zfamily_to_byter_
addr_bytesZ
host_bytesZ addressesZ target_addrr r r r s8
z socksocket._write_SOCKS5_addressc C s | |d}|dkr(t| |d}nN|dkrN| |d}| |t|}n(|dkrnttj| |d}ntdtd| |d d
}||fS ) Nr r r5 r r r r r r ) r rH inet_ntoar r r r+ r unpack)r r Zatypr> lengthr? r r r r M s z socksocket._read_SOCKS5_addressc C s | j \}}}}}}| d} | dd}
zZd} zt|} W n2 tjyp |r\d} d} ntt|} Y n0 | tddd | | | |r| | | d
| r| | d d
|
| |
d }
|
dd d
krt d
t
|
d d }|dkr$t|d}td||t|
dd td|
dd d f| _| rht| |f| _n
||f| _W |
| n|
| 0 dS )z0Negotiates a connection through a SOCKS4 server.r r r Fs Tz>BBHr5 r r r r9 z%SOCKS4 proxy server sent invalid datar Z r r Nr )rz r rH Z inet_atonrW
gethostbynamer r r r: r r r+ r
SOCKS4_ERRORSr r0 r" r r r{ r| rX )r r dest_portr= r> r? r@ rA rB r r Zremote_resolver r r rW r r r _negotiate_SOCKS4\ sR
z socksocket._negotiate_SOCKS4c C sf | j \}}}}}}|r|nt|}d|d d t| d d|d g} |rv|rv| dt|d | | d | d| | }
|
} |
| st dz|
d d
\} }
}W n ty t d Y n0 | d st d
z t|
}
W n ty tdY n0 |
dkrRd|
|}|
dv rJ|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
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 status z{}: {})i i i za
[*] Note: The HTTP proxy server may not be supported by PySocks (must be a CONNECT tunnel proxy))s 0.0.0.0r N)rz rH r r: strappendr Zsendalljoinr readlinerX r+ splitrw rL rQ r1 r" r{ r| )r r r r= r> r? r@ rA rB Z http_headersZfobjZ status_liner` Z status_codeZ
status_msgrW r r r _negotiate_HTTP sR
zsocksocket._negotiate_HTTPc
s& t |dks|d dr,tdt| |\}}| jtjkr|| jsP| d t |}|dkrn|snd| _
n
||f| _
dS | j \}}}}} }
t |t
tfrt |dks|rt |tstdtt| | j |du r
|| _
tt| | j tt| ||f dS | }ztt| | W n| tjy } z`| |s|\}}d ||} t| }
d
|
| }td || t || n| W Y d} ~ nd} ~ 0 0 z| j | }|| || W nb tjy } z*|s| td | n| W Y d} ~ n&d} ~ 0 t y | Y n0 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 rK z PySocks doesn't support IPv6: %sr r Nz0Invalid destination-connection (host, port) pairz{}:{}z Error connecting to {} proxy {}z
%s due to: %sz Socket error) r rL rH rW r r~ rv ry rU r r| rz rP rm tuplerQ r+ rx r; rS r} rV r rX r" PRINTABLE_PROXY_TYPESlogdebugr- _proxy_negotiatorsr )r rY catch_errorsr r r= r[ r\ r@ rA rB rW Z proxy_serverZprintable_typer Z negotiater r r rV st
zsocksocket.connectc
C sX z| j |dd W dS tyR } z&|jr<|jW Y d}~S W Y d}~n
d}~0 0 dS )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 r N)rV OSErrorerrno)r rY r r r r
connect_ex6 s zsocksocket.connect_exc C s4 | j \}}}}}}|p t|}|s,td||fS )zD
Return proxy address to connect to as tuple object
zInvalid proxy type)rz
DEFAULT_PORTSr r+ )r r= r[ r\ r@ rA rB r r r r F s zsocksocket._proxy_addr)NNNTNN)r )r )N)*r' r( r) r* r< rH r rO r# r rS r r rT r rU rr rs rt ru rX r Zgetproxysocknamer Zgetproxypeernamer r r r r r r r r
r r r r rV r r
__classcell__r r r r r; sP
% b-<<\
r; )NNNTNN) NNNNNTNNN)Dbase64r collections.abcr ImportError collectionsr r r r r ior Zloggingosr rH r sys __version__rl version_infoZ
win_inet_ptonZ getLoggerr' r ZPROXY_TYPE_SOCKS4r
ZPROXY_TYPE_SOCKS5r ZPROXY_TYPE_HTTPr Z PROXY_TYPESre zipvalueskeysr Z
_orgsocketrd r IOErrorr r+ r- r. r/ r0 r1 r r r rC rF rG ZgetdefaultproxyrJ Z
wrapmodulerb rc rq rh methodrP rg r setattrr; r r r r s
=