From 47f3ff1679bb1ffe8484f46b1580dba899de9e77 Mon Sep 17 00:00:00 2001 From: dlorenc Date: Thu, 19 Mar 2015 11:43:05 -0700 Subject: [PATCH] Add assert_fingerprint option. Signed-off-by: dlorenc --- docker/ssladapter/ssladapter.py | 5 ++++- docker/tls.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/docker/ssladapter/ssladapter.py b/docker/ssladapter/ssladapter.py index 9727d373..3a70a916 100644 --- a/docker/ssladapter/ssladapter.py +++ b/docker/ssladapter/ssladapter.py @@ -25,10 +25,12 @@ def get_max_tls_protocol(): class SSLAdapter(HTTPAdapter): '''An HTTPS Transport Adapter that uses an arbitrary SSL version.''' - def __init__(self, ssl_version=None, assert_hostname=None, **kwargs): + def __init__(self, ssl_version=None, assert_hostname=None, + assert_fingerprint=None, **kwargs): ssl_version = ssl_version or get_max_tls_protocol() self.ssl_version = ssl_version self.assert_hostname = assert_hostname + self.assert_fingerprint = assert_fingerprint super(SSLAdapter, self).__init__(**kwargs) def init_poolmanager(self, connections, maxsize, block=False): @@ -37,6 +39,7 @@ class SSLAdapter(HTTPAdapter): 'maxsize': maxsize, 'block': block, 'assert_hostname': self.assert_hostname, + 'assert_fingerprint': self.assert_fingerprint, } if self.can_override_ssl_version(): kwargs['ssl_version'] = self.ssl_version diff --git a/docker/tls.py b/docker/tls.py index c007d0bc..d888b7d4 100644 --- a/docker/tls.py +++ b/docker/tls.py @@ -10,7 +10,8 @@ class TLSConfig(object): ssl_version = None def __init__(self, client_cert=None, ca_cert=None, verify=None, - ssl_version=None, assert_hostname=None): + ssl_version=None, assert_hostname=None, + assert_fingerprint=None): # Argument compatibility/mapping with # http://docs.docker.com/examples/https/ # This diverges from the Docker CLI in that users can specify 'tls' @@ -24,6 +25,7 @@ class TLSConfig(object): ssl_version = ssl_version or ssladapter.get_max_tls_protocol() self.ssl_version = ssl_version self.assert_hostname = assert_hostname + self.assert_fingerprint = assert_fingerprint # "tls" and "tls_verify" must have both or neither cert/key files # In either case, Alert the user when both are expected, but any are @@ -72,4 +74,5 @@ class TLSConfig(object): client.mount('https://', ssladapter.SSLAdapter( ssl_version=self.ssl_version, assert_hostname=self.assert_hostname, + assert_fingerprint=self.assert_fingerprint, ))