3 $cp&@sdZddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZddlmZddlmZmZmZdd lmZmZmZmZmZmZydd l mZWn$ek rdZdd l m!Z!YnXd d gZ"dZ#ej#Z$ej%j&Z'e j(Z)e j*Z+dZ,ej-ej.ej/ej0ej1ej2ej3ej4ej5ej6ej7ej8ej9ej:ej;ejej?ej@ejAejBejCejDejEejFejGejHejIejJgZKejLejMejNfeejMejNfiZOePe drejQejQfeOe jR<ePe drejSejSfeOe jT<ePe dr ejMejMfeOe jU<ePe dr*ejVejVfeOe jW<ePe drHejNejNfeOe jX<dd ZYdd ZZddZ[ddZ\ej]e[Z^ej_e\Z`GdddeaZberd$ddZcn d%d dZceceb_cGd!d"d"eaZddS)&a SecureTranport support for urllib3 via ctypes. This makes platform-native TLS available to urllib3 users on macOS without the use of a compiler. This is an important feature because the Python Package Index is moving to become a TLSv1.2-or-higher server, and the default OpenSSL that ships with macOS is not capable of doing TLSv1.2. The only way to resolve this is to give macOS users an alternative solution to the problem, and that solution is to use SecureTransport. We use ctypes here because this solution must not require a compiler. That's because pip is not allowed to require a compiler either. This is not intended to be a seriously long-term solution to this problem. The hope is that PEP 543 will eventually solve this issue for us, at which point we can retire this contrib module. But in the short term, we need to solve the impending tire fire that is Python on Mac without this kind of contrib module. So...here we are. To use this module, simply import and inject it:: import urllib3.contrib.securetransport urllib3.contrib.securetransport.inject_into_urllib3() Happy TLSing! This code is a bastardised version of the code found in Will Bond's oscrypto library. An enormous debt is owed to him for blazing this trail for us. For that reason, this code should be considered to be covered both by urllib3's license and by oscrypto's: .. code-block:: Copyright (c) 2015-2016 Will Bond Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. )absolute_importN)util)PROTOCOL_TLS_CLIENT)CoreFoundationSecurity SecurityConst)_assert_no_error_build_tls_unknown_ca_alert_cert_array_from_pem_create_cfstring_array_load_client_cert_chain_temporary_keychain) _fileobject)backport_makefileinject_into_urllib3extract_from_urllib3Ti@PROTOCOL_SSLv2PROTOCOL_SSLv3PROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2cCs.tt_ttj_tt_ttj_dt_dtj_dS)zG Monkey-patch urllib3 with SecureTransport-backed SSL-support. TN)SecureTransportContextr SSLContextssl_HAS_SNIIS_SECURETRANSPORTrrF/tmp/pip-target-ezim9f7s/lib/python/urllib3/contrib/securetransport.pyrs cCs.tt_ttj_tt_ttj_dt_dtj_dS)z> Undo monkey-patching by :func:`inject_into_urllib3`. FN)orig_util_SSLContextrrrorig_util_HAS_SNIrrrrrrrs c Csxd}y8tj|}|dkr tjS|j}|d}|j}d}d}y|xv||kr|dksZ|dkrttj||sttjt j d||} t j | j ||} |j| | } || 7}| sB|stjSPqBWWnhtjk r"} zH| j }|dk o|t j kr||d<|t jks |t jkrtjSWYdd} ~ XnX||d<||krt|\|_|_d|_d|_d|_d|_d|_d|_d|_ dS)NrF) _protocol_to_min_max _min_version _max_version_options_verify _trust_bundle _client_cert _client_key_client_key_passphrase_alpn_protocols)rPrrrrrQszSecureTransportContext.__init__cCsdS)z SecureTransport cannot have its hostname checking disabled. For more, see the comment on getpeercert() in this file. Tr)rPrrrcheck_hostname"sz%SecureTransportContext.check_hostnamecCsdS)z SecureTransport cannot have its hostname checking disabled. For more, see the comment on getpeercert() in this file. Nr)rPrxrrrr*scCs|jS)N)r)rPrrroptions2szSecureTransportContext.optionscCs ||_dS)N)r)rPrxrrrr<scCs|jr tjStjS)N)rrg CERT_REQUIRED CERT_NONE)rPrrr verify_modeAsz"SecureTransportContext.verify_modecCs|tjkrdnd|_dS)NTF)rgrr)rPrxrrrrEscCsdS)Nr)rPrrrset_default_verify_pathsIs z/SecureTransportContext.set_default_verify_pathscCs|jS)N)r)rPrrrload_default_certsUsz)SecureTransportContext.load_default_certscCs|tjjkrtddS)Nz5SecureTransport doesn't support custom cipher strings)rrDEFAULT_CIPHERSr)rPrWrrr set_ciphersXs z"SecureTransportContext.set_ciphersNc Cs:|dk rtd|dk r,t|WdQRX|p2||_dS)Nz1SecureTransport does not support cert directories)rrsr)rPcafilecapathcadatarrrload_verify_locations]s   z,SecureTransportContext.load_verify_locationscCs||_||_||_dS)N)rrZ_client_cert_passphrase)rPcertfilekeyfilepasswordrrrload_cert_chainisz&SecureTransportContext.load_cert_chaincCs&ttdstddd|D|_dS)z Sets the ALPN protocols that will later be set on the context. Raises a NotImplementedError if ALPN is not supported. rZz2SecureTransport supports ALPN only in macOS 10.12+cSsg|]}tj|qSr)six ensure_binary).0prrr xsz=SecureTransportContext.set_alpn_protocols..N)hasattrrNotImplementedErrorr)rPr\rrrset_alpn_protocolsns z)SecureTransportContext.set_alpn_protocolsFTc CsP| s t|st|stt|}|j||j|j|j|j|j|j|j |j |S)N) rrGrrrrrrrrr)rPsock server_sidedo_handshake_on_connectsuppress_ragged_eofsrr9rrr wrap_socketzs z"SecureTransportContext.wrap_socket)NNN)NN)FTTN)rrrrrQpropertyrsetterrrrrrrrrrrrrrrs$     r)r)rN)er __future__rrr+r)os.pathrprr%rgrb threadingweakrefrrZ util.ssl_rZ_securetransport.bindingsrrr Z_securetransport.low_levelr r r r rrr ImportErrorZpackages.backports.makefiler__all__rr!rrr WeakValueDictionaryr"Lockr}rZ'TLS_ECDHE_ECDSA_WITH_AES_256_GCM_SHA384Z'TLS_ECDHE_ECDSA_WITH_AES_128_GCM_SHA256Z%TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384Z%TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256Z-TLS_ECDHE_ECDSA_WITH_CHACHA20_POLY1305_SHA256Z+TLS_ECDHE_RSA_WITH_CHACHA20_POLY1305_SHA256Z#TLS_DHE_RSA_WITH_AES_256_GCM_SHA384Z#TLS_DHE_RSA_WITH_AES_128_GCM_SHA256Z'TLS_ECDHE_ECDSA_WITH_AES_256_CBC_SHA384Z$TLS_ECDHE_ECDSA_WITH_AES_256_CBC_SHAZ'TLS_ECDHE_ECDSA_WITH_AES_128_CBC_SHA256Z$TLS_ECDHE_ECDSA_WITH_AES_128_CBC_SHAZ%TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA384Z"TLS_ECDHE_RSA_WITH_AES_256_CBC_SHAZ%TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA256Z"TLS_ECDHE_RSA_WITH_AES_128_CBC_SHAZ#TLS_DHE_RSA_WITH_AES_256_CBC_SHA256Z TLS_DHE_RSA_WITH_AES_256_CBC_SHAZ#TLS_DHE_RSA_WITH_AES_128_CBC_SHA256Z TLS_DHE_RSA_WITH_AES_128_CBC_SHAZTLS_AES_256_GCM_SHA384ZTLS_AES_128_GCM_SHA256ZTLS_RSA_WITH_AES_256_GCM_SHA384ZTLS_RSA_WITH_AES_128_GCM_SHA256ZTLS_AES_128_CCM_8_SHA256ZTLS_AES_128_CCM_SHA256ZTLS_RSA_WITH_AES_256_CBC_SHA256ZTLS_RSA_WITH_AES_128_CBC_SHA256ZTLS_RSA_WITH_AES_256_CBC_SHAZTLS_RSA_WITH_AES_128_CBC_SHArV PROTOCOL_TLSrrrrrrrrrrrrrrr@rFZ SSLReadFuncr{Z SSLWriteFuncr|objectrGrrrrrr5s           76  <