diff --git a/oml/ssl_request.py b/oml/ssl_request.py index f783e04..c48087e 100644 --- a/oml/ssl_request.py +++ b/oml/ssl_request.py @@ -1,7 +1,8 @@ +# -*- coding: utf-8 -*- +# vi:si:et:sw=4:sts=4:ts=4 + import http.client -import socket import urllib.request, urllib.error, urllib.parse -import ssl import hashlib import logging logger = logging.getLogger('oml.ssl_request') @@ -9,75 +10,59 @@ logger = logging.getLogger('oml.ssl_request') class InvalidCertificateException(http.client.HTTPException, urllib.error.URLError): def __init__(self, fingerprint, cert, reason): http.client.HTTPException.__init__(self) - self.fingerprint = fingerprint - self.cert_fingerprint = hashlib.sha1(cert).hexdigest() + self._fingerprint = fingerprint + self._cert_fingerprint = hashlib.sha1(cert).hexdigest() self.reason = reason def __str__(self): return ('%s (local) != %s (remote) (%s)\n' % - (self.fingerprint, self.cert_fingerprint, self.reason)) + (self._fingerprint, self._cert_fingerprint, self.reason)) -class CertValidatingHTTPSConnection(http.client.HTTPConnection): - default_port = http.client.HTTPS_PORT +class FingerprintHTTPSConnection(http.client.HTTPSConnection): - def __init__(self, host, port=None, fingerprint=None, strict=None, **kwargs): - http.client.HTTPConnection.__init__(self, host, port, strict, **kwargs) - self.fingerprint = fingerprint - if self.fingerprint: - self.cert_reqs = ssl.CERT_REQUIRED - else: - self.cert_reqs = ssl.CERT_NONE - self.cert_reqs = ssl.CERT_NONE + def __init__(self, host, port=None, fingerprint=None, check_hostname=None, **kwargs): + self._fingerprint = fingerprint + if self._fingerprint: + check_hostname = None + http.client.HTTPSConnection.__init__(self, host, port, + check_hostname=check_hostname, **kwargs) - def _ValidateCertificateFingerprint(self, cert): - if len(self.fingerprint) == 40: + def _check_fingerprint(self, cert): + if len(self._fingerprint) == 40: fingerprint = hashlib.sha1(cert).hexdigest() - elif len(self.fingerprint) == 64: + elif len(self._fingerprint) == 64: fingerprint = hashlib.sha256(cert).hexdigest() - elif len(self.fingerprint) == 128: + elif len(self._fingerprint) == 128: fingerprint = hashlib.sha512(cert).hexdigest() else: - logging.error('unkown fingerprint length %s (%s)', self.fingerprint, len(self.fingerprint)) + logging.error('unkown _fingerprint length %s (%s)', + self._fingerprint, len(self._fingerprint)) return False - return fingerprint == self.fingerprint + return fingerprint == self._fingerprint def connect(self): - sock = socket.create_connection((self.host, self.port)) - self.sock = ssl.wrap_socket(sock, cert_reqs=self.cert_reqs) - #if self.cert_reqs & ssl.CERT_REQUIRED: - if self.fingerprint: + http.client.HTTPSConnection.connect(self) + if self._fingerprint: cert = self.sock.getpeercert(binary_form=True) - if not self._ValidateCertificateFingerprint(cert): - raise InvalidCertificateException(self.fingerprint, cert, + if not self._check_fingerprint(cert): + raise InvalidCertificateException(self._fingerprint, cert, 'fingerprint mismatch') #logger.debug('CIPHER %s VERSION %s', self.sock.cipher(), self.sock.ssl_version) -class VerifiedHTTPSHandler(urllib.request.HTTPSHandler): - def __init__(self, **kwargs): - urllib.request.AbstractHTTPHandler.__init__(self) - self._connection_args = kwargs +class FingerprintHTTPSHandler(urllib.request.HTTPSHandler): + + def __init__(self, debuglevel=0, context=None, check_hostname=None, fingerprint=None): + urllib.request.AbstractHTTPHandler.__init__(self, debuglevel) + self._context = context + self._check_hostname = check_hostname + self._fingerprint = fingerprint def https_open(self, req): - def http_class_wrapper(host, **kwargs): - full_kwargs = dict(self._connection_args) - full_kwargs.update(kwargs) - print(self._connection_args) - print(kwargs) - if 'timeout' in full_kwargs: - del full_kwargs['timeout'] - return CertValidatingHTTPSConnection(host, **full_kwargs) - - try: - return self.do_open(http_class_wrapper, req) - except urllib.error.URLError as e: - if type(e.reason) == ssl.SSLError and e.reason.args[0] == 1: - raise InvalidCertificateException(self.fingerprint, '', - e.reason.args[1]) - raise - - https_request = urllib.request.HTTPSHandler.do_request_ + return self.do_open(FingerprintHTTPSConnection, req, + context=self._context, check_hostname=self._check_hostname, + fingerprint=self._fingerprint) def get_opener(fingerprint): - handler = VerifiedHTTPSHandler(fingerprint=fingerprint) + handler = FingerprintHTTPSHandler(fingerprint=fingerprint) opener = urllib.request.build_opener(handler) return opener