mirror of
https://github.com/morpheus65535/bazarr
synced 2024-12-23 00:03:33 +00:00
178 lines
7.2 KiB
Python
178 lines
7.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""A X509Adapter for use with the requests library.
|
|
|
|
This file contains an implementation of the X509Adapter that will
|
|
allow users to authenticate a request using an arbitrary
|
|
X.509 certificate without needing to convert it to a .pem file
|
|
|
|
"""
|
|
|
|
from OpenSSL.crypto import PKey, X509
|
|
from cryptography import x509
|
|
from cryptography.hazmat.primitives.serialization import (load_pem_private_key,
|
|
load_der_private_key)
|
|
from cryptography.hazmat.primitives.serialization import Encoding
|
|
from cryptography.hazmat.backends import default_backend
|
|
|
|
from datetime import datetime
|
|
from requests.adapters import HTTPAdapter
|
|
import requests
|
|
|
|
from .._compat import PyOpenSSLContext
|
|
from .. import exceptions as exc
|
|
|
|
"""
|
|
importing the protocol constants from _ssl instead of ssl because only the
|
|
constants are needed and to handle issues caused by importing from ssl on
|
|
the 2.7.x line.
|
|
"""
|
|
try:
|
|
from _ssl import PROTOCOL_TLS as PROTOCOL
|
|
except ImportError:
|
|
from _ssl import PROTOCOL_SSLv23 as PROTOCOL
|
|
|
|
|
|
class X509Adapter(HTTPAdapter):
|
|
r"""Adapter for use with X.509 certificates.
|
|
|
|
Provides an interface for Requests sessions to contact HTTPS urls and
|
|
authenticate with an X.509 cert by implementing the Transport Adapter
|
|
interface. This class will need to be manually instantiated and mounted
|
|
to the session
|
|
|
|
:param pool_connections: The number of urllib3 connection pools to
|
|
cache.
|
|
:param pool_maxsize: The maximum number of connections to save in the
|
|
pool.
|
|
:param max_retries: The maximum number of retries each connection
|
|
should attempt. Note, this applies only to failed DNS lookups,
|
|
socket connections and connection timeouts, never to requests where
|
|
data has made it to the server. By default, Requests does not retry
|
|
failed connections. If you need granular control over the
|
|
conditions under which we retry a request, import urllib3's
|
|
``Retry`` class and pass that instead.
|
|
:param pool_block: Whether the connection pool should block for
|
|
connections.
|
|
|
|
:param bytes cert_bytes:
|
|
bytes object containing contents of a cryptography.x509Certificate
|
|
object using the encoding specified by the ``encoding`` parameter.
|
|
:param bytes pk_bytes:
|
|
bytes object containing contents of a object that implements
|
|
``cryptography.hazmat.primitives.serialization.PrivateFormat``
|
|
using the encoding specified by the ``encoding`` parameter.
|
|
:param password:
|
|
string or utf8 encoded bytes containing the passphrase used for the
|
|
private key. None if unencrypted. Defaults to None.
|
|
:param encoding:
|
|
Enumeration detailing the encoding method used on the ``cert_bytes``
|
|
parameter. Can be either PEM or DER. Defaults to PEM.
|
|
:type encoding:
|
|
:class: `cryptography.hazmat.primitives.serialization.Encoding`
|
|
|
|
Usage::
|
|
|
|
>>> import requests
|
|
>>> from requests_toolbelt.adapters.x509 import X509Adapter
|
|
>>> s = requests.Session()
|
|
>>> a = X509Adapter(max_retries=3,
|
|
cert_bytes=b'...', pk_bytes=b'...', encoding='...'
|
|
>>> s.mount('https://', a)
|
|
"""
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self._check_version()
|
|
cert_bytes = kwargs.pop('cert_bytes', None)
|
|
pk_bytes = kwargs.pop('pk_bytes', None)
|
|
password = kwargs.pop('password', None)
|
|
encoding = kwargs.pop('encoding', Encoding.PEM)
|
|
|
|
password_bytes = None
|
|
|
|
if cert_bytes is None or not isinstance(cert_bytes, bytes):
|
|
raise ValueError('Invalid cert content provided. '
|
|
'You must provide an X.509 cert '
|
|
'formatted as a byte array.')
|
|
if pk_bytes is None or not isinstance(pk_bytes, bytes):
|
|
raise ValueError('Invalid private key content provided. '
|
|
'You must provide a private key '
|
|
'formatted as a byte array.')
|
|
|
|
if isinstance(password, bytes):
|
|
password_bytes = password
|
|
elif password:
|
|
password_bytes = password.encode('utf8')
|
|
|
|
self.ssl_context = create_ssl_context(cert_bytes, pk_bytes,
|
|
password_bytes, encoding)
|
|
|
|
super(X509Adapter, self).__init__(*args, **kwargs)
|
|
|
|
def init_poolmanager(self, *args, **kwargs):
|
|
if self.ssl_context:
|
|
kwargs['ssl_context'] = self.ssl_context
|
|
return super(X509Adapter, self).init_poolmanager(*args, **kwargs)
|
|
|
|
def proxy_manager_for(self, *args, **kwargs):
|
|
if self.ssl_context:
|
|
kwargs['ssl_context'] = self.ssl_context
|
|
return super(X509Adapter, self).proxy_manager_for(*args, **kwargs)
|
|
|
|
def _check_version(self):
|
|
if PyOpenSSLContext is None:
|
|
raise exc.VersionMismatchError(
|
|
"The X509Adapter requires at least Requests 2.12.0 to be "
|
|
"installed. Version {0} was found instead.".format(
|
|
requests.__version__
|
|
)
|
|
)
|
|
|
|
|
|
def check_cert_dates(cert):
|
|
"""Verify that the supplied client cert is not invalid."""
|
|
|
|
now = datetime.utcnow()
|
|
if cert.not_valid_after < now or cert.not_valid_before > now:
|
|
raise ValueError('Client certificate expired: Not After: '
|
|
'{0:%Y-%m-%d %H:%M:%SZ} '
|
|
'Not Before: {1:%Y-%m-%d %H:%M:%SZ}'
|
|
.format(cert.not_valid_after, cert.not_valid_before))
|
|
|
|
|
|
def create_ssl_context(cert_byes, pk_bytes, password=None,
|
|
encoding=Encoding.PEM):
|
|
"""Create an SSL Context with the supplied cert/password.
|
|
|
|
:param cert_bytes array of bytes containing the cert encoded
|
|
using the method supplied in the ``encoding`` parameter
|
|
:param pk_bytes array of bytes containing the private key encoded
|
|
using the method supplied in the ``encoding`` parameter
|
|
:param password array of bytes containing the passphrase to be used
|
|
with the supplied private key. None if unencrypted.
|
|
Defaults to None.
|
|
:param encoding ``cryptography.hazmat.primitives.serialization.Encoding``
|
|
details the encoding method used on the ``cert_bytes`` and
|
|
``pk_bytes`` parameters. Can be either PEM or DER.
|
|
Defaults to PEM.
|
|
"""
|
|
backend = default_backend()
|
|
|
|
cert = None
|
|
key = None
|
|
if encoding == Encoding.PEM:
|
|
cert = x509.load_pem_x509_certificate(cert_byes, backend)
|
|
key = load_pem_private_key(pk_bytes, password, backend)
|
|
elif encoding == Encoding.DER:
|
|
cert = x509.load_der_x509_certificate(cert_byes, backend)
|
|
key = load_der_private_key(pk_bytes, password, backend)
|
|
else:
|
|
raise ValueError('Invalid encoding provided: Must be PEM or DER')
|
|
|
|
if not (cert and key):
|
|
raise ValueError('Cert and key could not be parsed from '
|
|
'provided data')
|
|
check_cert_dates(cert)
|
|
ssl_context = PyOpenSSLContext(PROTOCOL)
|
|
ssl_context._ctx.use_certificate(X509.from_cryptography(cert))
|
|
ssl_context._ctx.use_privatekey(PKey.from_cryptography_key(key))
|
|
return ssl_context
|