Blame SOURCES/00234-PEP493-updated-implementation.patch

ae2451
@@ -, +, @@ 
ae2451
---
ae2451
 Lib/ssl.py | 53 ++++++++++++++++++++++++++++++++++++++---------------
ae2451
 1 file changed, 38 insertions(+), 15 deletions(-)
ae2451
--- a/Lib/ssl.py	
ae2451
+++ a/Lib/ssl.py	
ae2451
@@ -466,24 +466,47 @@ def _create_unverified_context(protocol=PROTOCOL_SSLv23, cert_reqs=None,
ae2451
 
ae2451
     return context
ae2451
 
ae2451
+_https_verify_envvar = 'PYTHONHTTPSVERIFY'
ae2451
 _cert_verification_config = '/etc/python/cert-verification.cfg'
ae2451
 
ae2451
-def _get_verify_status(protocol):
ae2451
-   context_factory = {
ae2451
-       'platform_default': _create_unverified_context,
ae2451
-       'enable': create_default_context,
ae2451
-       'disable': _create_unverified_context
ae2451
-   }
ae2451
-   import ConfigParser
ae2451
-   try:
ae2451
-       config = ConfigParser.RawConfigParser()
ae2451
-       config.read(_cert_verification_config)
ae2451
-       status = config.get(protocol, 'verify')
ae2451
-   except (ConfigParser.NoSectionError, ConfigParser.NoOptionError):
ae2451
-       status = 'platform_default'
ae2451
-   default = context_factory.get('platform_default')
ae2451
-   return context_factory.get(status, default)
ae2451
+# To provide same function name as specified in PEP493 with keeping
ae2451
+# the old name as defined in our previous patch
ae2451
+_get_https_context_factory = lambda: _get_verify_status('https')
ae2451
 
ae2451
+def _get_verify_status(protocol):
ae2451
+    # See https://www.python.org/dev/peps/pep-0493/#recommendation-for-combined-feature-backports
ae2451
+    # Check for an environmental override of the default behaviour
ae2451
+    if not sys.flags.ignore_environment:
ae2451
+        config_setting = os.environ.get(_https_verify_envvar)
ae2451
+        if config_setting is not None:
ae2451
+            if config_setting == '0':
ae2451
+                return _create_unverified_context
ae2451
+            return create_default_context
ae2451
+
ae2451
+    # Check for a system-wide override of the default behaviour
ae2451
+    context_factory = {
ae2451
+        'platform_default': create_default_context,
ae2451
+        'enable': create_default_context,
ae2451
+        'disable': _create_unverified_context
ae2451
+    }
ae2451
+    import ConfigParser
ae2451
+    try:
ae2451
+        config = ConfigParser.RawConfigParser()
ae2451
+        config.read(_cert_verification_config)
ae2451
+        status = config.get(protocol, 'verify')
ae2451
+    except (ConfigParser.NoSectionError, ConfigParser.NoOptionError):
ae2451
+        status = 'platform_default'
ae2451
+    default = context_factory.get('platform_default')
ae2451
+    return context_factory.get(status, default)
ae2451
+
ae2451
+# See https://www.python.org/dev/peps/pep-0493/#feature-configuration-api
ae2451
+def _https_verify_certificates(enable=True):
ae2451
+    """Verify server HTTPS certificates by default?"""
ae2451
+    global _create_default_https_context
ae2451
+    if enable:
ae2451
+        _create_default_https_context = create_default_context
ae2451
+    else:
ae2451
+        _create_default_https_context = _create_unverified_context
ae2451
 
ae2451
 # Used by http.client if no context is explicitly passed.
ae2451
 _create_default_https_context = _get_verify_status('https')
ae2451
--- a/Lib/test/test_ssl.py	Thu Jan 14 21:57:57 2016 -0800
ae2451
+++ a/Lib/test/test_ssl.py	Fri Jan 15 17:41:37 2016 +1000
ae2451
@@ -4,6 +4,7 @@
ae2451
 import sys
ae2451
 import unittest
ae2451
 from test import test_support as support
ae2451
+from test.script_helper import assert_python_ok
ae2451
 import asyncore
ae2451
 import socket
ae2451
 import select
ae2451
@@ -1149,6 +1149,57 @@
ae2451
         self.assertEqual(ctx.verify_mode, ssl.CERT_NONE)
ae2451
         self.assertEqual(ctx.options & ssl.OP_NO_SSLv2, ssl.OP_NO_SSLv2)
ae2451
 
ae2451
+    def test__https_verify_certificates(self):
ae2451
+        # Unit test to check the contect factory mapping
ae2451
+        # The factories themselves are tested above
ae2451
+        # This test will fail by design if run under PYTHONHTTPSVERIFY=0
ae2451
+        # (as will various test_httplib tests)
ae2451
+
ae2451
+        # Uses a fresh SSL module to avoid affecting the real one
ae2451
+        local_ssl = support.import_fresh_module("ssl")
ae2451
+        # Certificate verification is enabled by default
ae2451
+        self.assertIs(local_ssl._create_default_https_context,
ae2451
+                      local_ssl.create_default_context)
ae2451
+        # Turn default verification off
ae2451
+        local_ssl._https_verify_certificates(enable=False)
ae2451
+        self.assertIs(local_ssl._create_default_https_context,
ae2451
+                      local_ssl._create_unverified_context)
ae2451
+        # And back on
ae2451
+        local_ssl._https_verify_certificates(enable=True)
ae2451
+        self.assertIs(local_ssl._create_default_https_context,
ae2451
+                      local_ssl.create_default_context)
ae2451
+        # The default behaviour is to enable
ae2451
+        local_ssl._https_verify_certificates(enable=False)
ae2451
+        local_ssl._https_verify_certificates()
ae2451
+        self.assertIs(local_ssl._create_default_https_context,
ae2451
+                      local_ssl.create_default_context)
ae2451
+
ae2451
+    def test__https_verify_envvar(self):
ae2451
+        # Unit test to check the PYTHONHTTPSVERIFY handling
ae2451
+        # Need to use a subprocess so it can still be run under -E
ae2451
+        https_is_verified = """import ssl, sys; \
ae2451
+            status = "Error: _create_default_https_context does not verify certs" \
ae2451
+                       if ssl._create_default_https_context is \
ae2451
+                          ssl._create_unverified_context \
ae2451
+                     else None; \
ae2451
+            sys.exit(status)"""
ae2451
+        https_is_not_verified = """import ssl, sys; \
ae2451
+            status = "Error: _create_default_https_context verifies certs" \
ae2451
+                       if ssl._create_default_https_context is \
ae2451
+                          ssl.create_default_context \
ae2451
+                     else None; \
ae2451
+            sys.exit(status)"""
ae2451
+        extra_env = {}
ae2451
+        # Omitting it leaves verification on
ae2451
+        assert_python_ok("-c", https_is_verified, **extra_env)
ae2451
+        # Setting it to zero turns verification off
ae2451
+        extra_env[ssl._https_verify_envvar] = "0"
ae2451
+        assert_python_ok("-c", https_is_not_verified, **extra_env)
ae2451
+        # Any other value should also leave it on
ae2451
+        for setting in ("", "1", "enabled", "foo"):
ae2451
+            extra_env[ssl._https_verify_envvar] = setting
ae2451
+            assert_python_ok("-c", https_is_verified, **extra_env)
ae2451
+
ae2451
     def test_check_hostname(self):
ae2451
         ctx = ssl.SSLContext(ssl.PROTOCOL_TLSv1)
ae2451
         self.assertFalse(ctx.check_hostname)