U
f} @ s> d Z ddlmZ z ddlZW n6 ek
rR ddlZddlmZ ede Y nX ddl Z ddl
m Z ddl
mZmZ dd lmZmZ dd
lmZmZ dd lmZ dd lmZ z ddlZW n ek
r dZY nX G d
d de jZG dd deZ G dd de eZ G dd deZ G dd deZ G dd deZ dS )a
This module contains provisional support for SOCKS proxies from within
urllib3. This module supports SOCKS4, SOCKS4A (an extension of SOCKS4), and
SOCKS5. To enable its functionality, either install PySocks or install this
module with the ``socks`` extra.
The SOCKS implementation supports the full range of urllib3 features. It also
supports the following SOCKS features:
- SOCKS4A (``proxy_url='socks4a://...``)
- SOCKS4 (``proxy_url='socks4://...``)
- SOCKS5 with remote DNS (``proxy_url='socks5h://...``)
- SOCKS5 with local DNS (``proxy_url='socks5://...``)
- Usernames and passwords for the SOCKS proxy
.. note::
It is recommended to use ``socks5h://`` or ``socks4a://`` schemes in
your ``proxy_url`` to ensure that DNS resolution is done from the remote
server instead of client-side when connecting to a domain name.
SOCKS4 supports IPv4 and domain names with the SOCKS4A extension. SOCKS5
supports IPv4, IPv6, and domain names.
When connecting to a SOCKS4 proxy the ``username`` portion of the ``proxy_url``
will be sent as the ``userid`` section of the SOCKS request:
.. code-block:: python
proxy_url="socks4a://@proxy-host"
When connecting to a SOCKS5 proxy the ``username`` and ``password`` portion
of the ``proxy_url`` will be sent as the username/password to authenticate
with the proxy:
.. code-block:: python
proxy_url="socks5h://:@proxy-host"
) annotationsN )DependencyWarningzSOCKS support in urllib3 requires the installation of optional dependencies: specifically, PySocks. For more information, see https://urllib3.readthedocs.io/en/latest/advanced-usage.html#socks-proxies)timeout)HTTPConnectionHTTPSConnection)HTTPConnectionPoolHTTPSConnectionPool)ConnectTimeoutErrorNewConnectionError) PoolManager) parse_urlc @ s> e Zd ZU ded< ded< ded< ded< ded< ded < d
S ) _TYPE_SOCKS_OPTIONSint
socks_version
str | None
proxy_host
proxy_portusernamepasswordboolrdnsN)__name__
__module__ __qualname____annotations__ r r F/opt/alt/python38/lib/python3.8/site-packages/urllib3/contrib/socks.pyr K s
r c s: e Zd ZdZddddd fdd Zdd d
d Z ZS ) SOCKSConnectionzG
A plain-text HTTP connection that connects via a SOCKS proxy.
r
typing.AnyNone)_socks_optionsargskwargsreturnc s || _ t j|| d S )N)r! super__init__)selfr! r" r# __class__r r r& Y s zSOCKSConnection.__init__zsocks.socksocket)r$ c
C st i }| j r| j |d< | jr$| j|d< zTtj| j| jff| jd | jd | jd | jd | jd | jd | jd |}W n tk
r } z"t | d
| j d | j d |W 5 d
}~X Y n tj
k
r: } zb|j r|j }t |trt | d
| j d | j d |nt
| d| nt
| d| |W 5 d
}~X Y n6 tk
rn } zt
| d| |W 5 d
}~X Y nX |S )zA
Establish a new connection via the SOCKS proxy.
source_addresssocket_optionsr r r r r r )
proxy_typeZ
proxy_addrr Zproxy_usernameZproxy_passwordZ
proxy_rdnsr zConnection to z timed out. (connect timeout=)Nz&Failed to establish a new connection: )r* r+ sockscreate_connectionhostportr! r
SocketTimeoutr
ProxyErrorZ
socket_err
isinstancer OSError)r' extra_kwconneerrorr r r _new_connb sl
zSOCKSConnection._new_conn)r r r __doc__r& r:
__classcell__r r r( r r T s r c @ s e Zd ZdS )SOCKSHTTPSConnectionN)r r r r r r r r= s r= c @ s e Zd ZeZdS )SOCKSHTTPConnectionPoolN)r r r r
ConnectionClsr r r r r> s r> c @ s e Zd ZeZdS )SOCKSHTTPSConnectionPoolN)r r r r= r? r r r r r@ s r@ c s< e Zd ZdZeedZd
dddddd d
fd d
Z ZS )SOCKSProxyManagerzh
A version of the urllib3 ProxyManager that routes connections via the
defined SOCKS proxy.
)httphttpsN
strr r z typing.Mapping[str, str] | Noner ) proxy_urlr r num_poolsheadersconnection_pool_kwc s t |}|d krB|d krB|jd k rB|jd}t|dkrB|\}}|jdkrXtj} d}
nP|jdkrntj} d}
n:|jdkrtj} d}
n$|jdkrtj} d}
ntd | || _ | |j
|j |||
d
} | |d < t j
||f| tj| _d S ) N:r Zsocks5FZsocks5hTZsocks4Zsocks4az'Unable to determine SOCKS version from )r r r r r r r! )r
authsplitlenschemer. ZPROXY_TYPE_SOCKS5ZPROXY_TYPE_SOCKS4
ValueErrorrF r0 r1 r% r& rA pool_classes_by_scheme) r' rF r r rG rH rI parsedrL r r Z
socks_optionsr( r r r& s:
zSOCKSProxyManager.__init__)NNrD N) r r r r; r> r@ rP r& r< r r r( r rA s rA )!r;
__future__r r. ImportErrorwarnings
exceptionsr warntypingsocketr r2
connectionr r connectionpoolr r r
r poolmanagerr Zutil.urlr
ssl TypedDictr r r= r> r@ rA r r r r s8 (
M