|
|
|
|
@@ -9,6 +9,7 @@ and maintain connections.
|
|
|
|
|
import os.path
|
|
|
|
|
import socket # noqa: F401
|
|
|
|
|
import typing
|
|
|
|
|
import warnings
|
|
|
|
|
|
|
|
|
|
from urllib3.exceptions import ClosedPoolError, ConnectTimeoutError
|
|
|
|
|
from urllib3.exceptions import HTTPError as _HTTPError
|
|
|
|
|
@@ -72,26 +73,44 @@ DEFAULT_POOLSIZE = 10
|
|
|
|
|
DEFAULT_RETRIES = 0
|
|
|
|
|
DEFAULT_POOL_TIMEOUT = None
|
|
|
|
|
|
|
|
|
|
_preloaded_ssl_context = create_urllib3_context()
|
|
|
|
|
_preloaded_ssl_context.load_verify_locations(
|
|
|
|
|
extract_zipped_paths(DEFAULT_CA_BUNDLE_PATH)
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
import ssl # noqa: F401
|
|
|
|
|
|
|
|
|
|
_preloaded_ssl_context = create_urllib3_context()
|
|
|
|
|
_preloaded_ssl_context.load_verify_locations(
|
|
|
|
|
extract_zipped_paths(DEFAULT_CA_BUNDLE_PATH)
|
|
|
|
|
)
|
|
|
|
|
except ImportError:
|
|
|
|
|
# Bypass default SSLContext creation when Python
|
|
|
|
|
# interpreter isn't built with the ssl module.
|
|
|
|
|
_preloaded_ssl_context = None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _urllib3_request_context(
|
|
|
|
|
request: "PreparedRequest",
|
|
|
|
|
verify: "bool | str | None",
|
|
|
|
|
client_cert: "typing.Tuple[str, str] | str | None",
|
|
|
|
|
poolmanager: "PoolManager",
|
|
|
|
|
) -> "(typing.Dict[str, typing.Any], typing.Dict[str, typing.Any])":
|
|
|
|
|
host_params = {}
|
|
|
|
|
pool_kwargs = {}
|
|
|
|
|
parsed_request_url = urlparse(request.url)
|
|
|
|
|
scheme = parsed_request_url.scheme.lower()
|
|
|
|
|
port = parsed_request_url.port
|
|
|
|
|
|
|
|
|
|
# Determine if we have and should use our default SSLContext
|
|
|
|
|
# to optimize performance on standard requests.
|
|
|
|
|
poolmanager_kwargs = getattr(poolmanager, "connection_pool_kw", {})
|
|
|
|
|
has_poolmanager_ssl_context = poolmanager_kwargs.get("ssl_context")
|
|
|
|
|
should_use_default_ssl_context = (
|
|
|
|
|
_preloaded_ssl_context is not None and not has_poolmanager_ssl_context
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
cert_reqs = "CERT_REQUIRED"
|
|
|
|
|
if verify is False:
|
|
|
|
|
cert_reqs = "CERT_NONE"
|
|
|
|
|
elif verify is True:
|
|
|
|
|
elif verify is True and should_use_default_ssl_context:
|
|
|
|
|
pool_kwargs["ssl_context"] = _preloaded_ssl_context
|
|
|
|
|
elif isinstance(verify, str):
|
|
|
|
|
if not os.path.isdir(verify):
|
|
|
|
|
@@ -374,13 +393,83 @@ class HTTPAdapter(BaseAdapter):
|
|
|
|
|
|
|
|
|
|
return response
|
|
|
|
|
|
|
|
|
|
def _get_connection(self, request, verify, proxies=None, cert=None):
|
|
|
|
|
# Replace the existing get_connection without breaking things and
|
|
|
|
|
# ensure that TLS settings are considered when we interact with
|
|
|
|
|
# urllib3 HTTP Pools
|
|
|
|
|
def build_connection_pool_key_attributes(self, request, verify, cert=None):
|
|
|
|
|
"""Build the PoolKey attributes used by urllib3 to return a connection.
|
|
|
|
|
|
|
|
|
|
This looks at the PreparedRequest, the user-specified verify value,
|
|
|
|
|
and the value of the cert parameter to determine what PoolKey values
|
|
|
|
|
to use to select a connection from a given urllib3 Connection Pool.
|
|
|
|
|
|
|
|
|
|
The SSL related pool key arguments are not consistently set. As of
|
|
|
|
|
this writing, use the following to determine what keys may be in that
|
|
|
|
|
dictionary:
|
|
|
|
|
|
|
|
|
|
* If ``verify`` is ``True``, ``"ssl_context"`` will be set and will be the
|
|
|
|
|
default Requests SSL Context
|
|
|
|
|
* If ``verify`` is ``False``, ``"ssl_context"`` will not be set but
|
|
|
|
|
``"cert_reqs"`` will be set
|
|
|
|
|
* If ``verify`` is a string, (i.e., it is a user-specified trust bundle)
|
|
|
|
|
``"ca_certs"`` will be set if the string is not a directory recognized
|
|
|
|
|
by :py:func:`os.path.isdir`, otherwise ``"ca_certs_dir"`` will be
|
|
|
|
|
set.
|
|
|
|
|
* If ``"cert"`` is specified, ``"cert_file"`` will always be set. If
|
|
|
|
|
``"cert"`` is a tuple with a second item, ``"key_file"`` will also
|
|
|
|
|
be present
|
|
|
|
|
|
|
|
|
|
To override these settings, one may subclass this class, call this
|
|
|
|
|
method and use the above logic to change parameters as desired. For
|
|
|
|
|
example, if one wishes to use a custom :py:class:`ssl.SSLContext` one
|
|
|
|
|
must both set ``"ssl_context"`` and based on what else they require,
|
|
|
|
|
alter the other keys to ensure the desired behaviour.
|
|
|
|
|
|
|
|
|
|
:param request:
|
|
|
|
|
The PreparedReqest being sent over the connection.
|
|
|
|
|
:type request:
|
|
|
|
|
:class:`~requests.models.PreparedRequest`
|
|
|
|
|
:param verify:
|
|
|
|
|
Either a boolean, in which case it controls whether
|
|
|
|
|
we verify the server's TLS certificate, or a string, in which case it
|
|
|
|
|
must be a path to a CA bundle to use.
|
|
|
|
|
:param cert:
|
|
|
|
|
(optional) Any user-provided SSL certificate for client
|
|
|
|
|
authentication (a.k.a., mTLS). This may be a string (i.e., just
|
|
|
|
|
the path to a file which holds both certificate and key) or a
|
|
|
|
|
tuple of length 2 with the certificate file path and key file
|
|
|
|
|
path.
|
|
|
|
|
:returns:
|
|
|
|
|
A tuple of two dictionaries. The first is the "host parameters"
|
|
|
|
|
portion of the Pool Key including scheme, hostname, and port. The
|
|
|
|
|
second is a dictionary of SSLContext related parameters.
|
|
|
|
|
"""
|
|
|
|
|
return _urllib3_request_context(request, verify, cert, self.poolmanager)
|
|
|
|
|
|
|
|
|
|
def get_connection_with_tls_context(self, request, verify, proxies=None, cert=None):
|
|
|
|
|
"""Returns a urllib3 connection for the given request and TLS settings.
|
|
|
|
|
This should not be called from user code, and is only exposed for use
|
|
|
|
|
when subclassing the :class:`HTTPAdapter <requests.adapters.HTTPAdapter>`.
|
|
|
|
|
|
|
|
|
|
:param request:
|
|
|
|
|
The :class:`PreparedRequest <PreparedRequest>` object to be sent
|
|
|
|
|
over the connection.
|
|
|
|
|
:param verify:
|
|
|
|
|
Either a boolean, in which case it controls whether we verify the
|
|
|
|
|
server's TLS certificate, or a string, in which case it must be a
|
|
|
|
|
path to a CA bundle to use.
|
|
|
|
|
:param proxies:
|
|
|
|
|
(optional) The proxies dictionary to apply to the request.
|
|
|
|
|
:param cert:
|
|
|
|
|
(optional) Any user-provided SSL certificate to be used for client
|
|
|
|
|
authentication (a.k.a., mTLS).
|
|
|
|
|
:rtype:
|
|
|
|
|
urllib3.ConnectionPool
|
|
|
|
|
"""
|
|
|
|
|
proxy = select_proxy(request.url, proxies)
|
|
|
|
|
try:
|
|
|
|
|
host_params, pool_kwargs = _urllib3_request_context(request, verify, cert)
|
|
|
|
|
host_params, pool_kwargs = self.build_connection_pool_key_attributes(
|
|
|
|
|
request,
|
|
|
|
|
verify,
|
|
|
|
|
cert,
|
|
|
|
|
)
|
|
|
|
|
except ValueError as e:
|
|
|
|
|
raise InvalidURL(e, request=request)
|
|
|
|
|
if proxy:
|
|
|
|
|
@@ -404,7 +493,10 @@ class HTTPAdapter(BaseAdapter):
|
|
|
|
|
return conn
|
|
|
|
|
|
|
|
|
|
def get_connection(self, url, proxies=None):
|
|
|
|
|
"""Returns a urllib3 connection for the given URL. This should not be
|
|
|
|
|
"""DEPRECATED: Users should move to `get_connection_with_tls_context`
|
|
|
|
|
for all subclasses of HTTPAdapter using Requests>=2.32.2.
|
|
|
|
|
|
|
|
|
|
Returns a urllib3 connection for the given URL. This should not be
|
|
|
|
|
called from user code, and is only exposed for use when subclassing the
|
|
|
|
|
:class:`HTTPAdapter <requests.adapters.HTTPAdapter>`.
|
|
|
|
|
|
|
|
|
|
@@ -412,6 +504,15 @@ class HTTPAdapter(BaseAdapter):
|
|
|
|
|
:param proxies: (optional) A Requests-style dictionary of proxies used on this request.
|
|
|
|
|
:rtype: urllib3.ConnectionPool
|
|
|
|
|
"""
|
|
|
|
|
warnings.warn(
|
|
|
|
|
(
|
|
|
|
|
"`get_connection` has been deprecated in favor of "
|
|
|
|
|
"`get_connection_with_tls_context`. Custom HTTPAdapter subclasses "
|
|
|
|
|
"will need to migrate for Requests>=2.32.2. Please see "
|
|
|
|
|
"https://github.com/psf/requests/pull/6710 for more details."
|
|
|
|
|
),
|
|
|
|
|
DeprecationWarning,
|
|
|
|
|
)
|
|
|
|
|
proxy = select_proxy(url, proxies)
|
|
|
|
|
|
|
|
|
|
if proxy:
|
|
|
|
|
@@ -529,7 +630,9 @@ class HTTPAdapter(BaseAdapter):
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
conn = self._get_connection(request, verify, proxies=proxies, cert=cert)
|
|
|
|
|
conn = self.get_connection_with_tls_context(
|
|
|
|
|
request, verify, proxies=proxies, cert=cert
|
|
|
|
|
)
|
|
|
|
|
except LocationValueError as e:
|
|
|
|
|
raise InvalidURL(e, request=request)
|
|
|
|
|
|
|
|
|
|
|