3
nf\ev ) @ s d Z ddlmZ ddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddlmZmZmZ ddlmZmZmZmZ ydd l mZ W n$ ek
r dZdd
lmZ Y nX d d gZ d
Z ej Z ej j Z!e j" Z#e j$ Z%dZ&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1ej2ej3ej4ej5ej6ej7ej8ej9ej:ej;ejej?ej@ejAejBejCejDejEejFejGg!ZHe
jIejJejKfiZLeMe
drejNejNfeLe
jO< eMe
drejPejPfeLe
jQ< eMe
drejJejJfeLe
jR< eMe
drejSejSfeLe
jT< eMe
dr$ejKejKfeLe
jU< eMe
dr@eLe
jI eLe
jV< dd ZWdd ZXdd ZYdd ZZej[eYZ\ej]eZZ^G dd d e_Z`erd$d d Zan
d%d d Zaeae`_aG d!d" d"e_ZbdS )&aU
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!
)absolute_importN )util )Security
SecurityConstCoreFoundation)_assert_no_error_cert_array_from_pem_temporary_keychain_load_client_cert_chain) _fileobject)backport_makefileinject_into_urllib3extract_from_urllib3Ti @ PROTOCOL_SSLv2PROTOCOL_SSLv3PROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSc C s( t tj_tt_ttj_dt_dtj_dS )zG
Monkey-patch urllib3 with SecureTransport-backed SSL-support.
TN)SecureTransportContextr ssl_
SSLContextHAS_SNIIS_SECURETRANSPORT r r %/usr/lib/python3.6/securetransport.pyr s
c C s( t tj_tt_ttj_dt_dtj_dS )z>
Undo monkey-patching by :func:`inject_into_urllib3`.
FN)orig_util_SSLContextr r r orig_util_HAS_SNIr r r r r r r s
c
C sx d}y8t j| }|dkr tjS |j}|d }|j }d}d}y|xv||k r|dksZ|dkrttj||sttjt j
d|| } t j | j
|| }
|j|
| } || 7 }| sB|stjS P qBW W nh tjk
r" } zH| j }|dk o|t j
kr||d< |t jks
|t jkrtjS W Y dd} ~ X nX ||d< ||kr r r r sendallC s zWrappedSocket.sendallc C s$ | j tj| j W d Q R X d S )N)rL r ZSSLCloserA )rI r r r shutdownI s
zWrappedSocket.shutdownc C s | j dk rd| _| jr(tj| j d | _| jr@tj| j d | _| jrvtj| j tj| j t j
| j d | _| _ | j j
S | j d8 _ d S )Nr T)rB rC rA r r\ rF rD r ZSecKeychainDeleteshutilZrmtreerE r# rK )rI r r r rK M s
zWrappedSocket.closeFc
C s |s t dtj }d }d }zptj| jtj|}t| |sBd S tj|}|sTd S tj |d}tj
|}t j |}t j
|} tj| |}W d |rt j| |rt j| X |S )Nz2SecureTransport only supports dumping binary certsr )
ValueErrorr rX rY rA r( rZ r ZSecTrustGetCertificateCountZ SecTrustGetCertificateAtIndexZSecCertificateCopyDatar ZCFDataGetLengthZCFDataGetBytePtrr; r\ )
rI Z binary_formra ZcertdataZ der_bytesrP Z
cert_countZleafZ data_lengthr3 r r r getpeercert` s2
zWrappedSocket.getpeercertc C s | j d7 _ d S )Nr )rB )rI r r r _reuse s zWrappedSocket._reusec C s& | j dk r| j n| j d8 _ d S )Nr )rB rK )rI r r r _drop s
zWrappedSocket._drop)N)F)__name__
__module__ __qualname____doc__rJ
contextlibcontextmanagerrL rQ rb rk rl rm rn r* rH r$ r< rs rt rK rw rx ry r r r r r@ ! s&
>Z
(
>r@ c C s | j d7 _ t| ||ddS )Nr T)rK )rB r
)rI modebufsizer r r makefile s r rc O s d}t | ||f||S )Nr )r )rI r bufferingargskwargsr r r r s c @ s e Zd ZdZdd Zedd Zejdd Zedd Zejd d Zed
d Z e jd d Z d
d Z
dd Z dd Z d ddZ
d ddZd ddZdS ) r z
I am a wrapper class for the SecureTransport library, to translate the
interface of the standard library ``SSLContext`` object to calls into
SecureTransport.
c C s8 t | \| _| _d| _d| _d | _d | _d | _d | _d S )Nr F) _protocol_to_min_max _min_version _max_version_options_verify
_trust_bundle _client_cert _client_key_client_key_passphrase)rI Zprotocolr r r rJ s z SecureTransportContext.__init__c C s dS )z
SecureTransport cannot have its hostname checking disabled. For more,
see the comment on getpeercert() in this file.
Tr )rI r r r check_hostname s z%SecureTransportContext.check_hostnamec C s dS )z
SecureTransport cannot have its hostname checking disabled. For more,
see the comment on getpeercert() in this file.
Nr )rI r] r r r r s c C s | j S )N)r )rI r r r options s z SecureTransportContext.optionsc C s
|| _ d S )N)r )rI r] r r r r s c C s | j r tjS tjS )N)r r[
CERT_REQUIREDZ CERT_NONE)rI r r r verify_mode s z"SecureTransportContext.verify_modec C s |t jkrdnd| _d S )NTF)r[ r r )rI r] r r r r s c C s d S )Nr )rI r r r set_default_verify_paths s
z/SecureTransportContext.set_default_verify_pathsc C s | j S )N)r )rI r r r load_default_certs s z)SecureTransportContext.load_default_certsc C s |t jjkrtdd S )Nz5SecureTransport doesn't support custom cipher strings)r r ZDEFAULT_CIPHERSrv )rI rO r r r set_ciphers s z"SecureTransportContext.set_ciphersNc C s |d k rt d|p|| _d S )Nz1SecureTransport does not support cert directories)rv r )rI ZcafileZcapathZcadatar r r load_verify_locations s z,SecureTransportContext.load_verify_locationsc C s || _ || _|| _d S )N)r r Z_client_cert_passphrase)rI ZcertfileZkeyfileZpasswordr r r load_cert_chain s z&SecureTransportContext.load_cert_chainFTc C s2 t |}|j|| j| j| j| j| j| j| j |S )N) r@ rk r r r r r r r )rI ZsockZ server_sideZdo_handshake_on_connectZsuppress_ragged_eofsrj r5 r r r wrap_socket s
z"SecureTransportContext.wrap_socket)NNN)NN)FTTN)rz r{ r| r} rJ propertyr setterr r r r r r r r r r r r r s
r )r )r N)cr} Z
__future__r r~ r( r& Zos.pathrS ru r# r[ Z threadingweakref r Z_securetransport.bindingsr r r Z_securetransport.low_levelr r
r r r
ImportErrorZpackages.backports.makefiler __all__r r r r r WeakValueDictionaryr ZLockre rr ZTLS_AES_256_GCM_SHA384Z TLS_CHACHA20_POLY1305_SHA256ZTLS_AES_128_GCM_SHA256Z'TLS_ECDHE_ECDSA_WITH_AES_256_GCM_SHA384Z%TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384Z'TLS_ECDHE_ECDSA_WITH_AES_128_GCM_SHA256Z%TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256Z#TLS_DHE_DSS_WITH_AES_256_GCM_SHA384Z#TLS_DHE_RSA_WITH_AES_256_GCM_SHA384Z#TLS_DHE_DSS_WITH_AES_128_GCM_SHA256Z#TLS_DHE_RSA_WITH_AES_128_GCM_SHA256Z'TLS_ECDHE_ECDSA_WITH_AES_256_CBC_SHA384Z%TLS_ECDHE_RSA_WITH_AES_256_CBC_SHA384Z$TLS_ECDHE_ECDSA_WITH_AES_256_CBC_SHAZ"TLS_ECDHE_RSA_WITH_AES_256_CBC_SHAZ#TLS_DHE_RSA_WITH_AES_256_CBC_SHA256Z#TLS_DHE_DSS_WITH_AES_256_CBC_SHA256Z TLS_DHE_RSA_WITH_AES_256_CBC_SHAZ TLS_DHE_DSS_WITH_AES_256_CBC_SHAZ'TLS_ECDHE_ECDSA_WITH_AES_128_CBC_SHA256Z%TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA256Z$TLS_ECDHE_ECDSA_WITH_AES_128_CBC_SHAZ"TLS_ECDHE_RSA_WITH_AES_128_CBC_SHAZ#TLS_DHE_RSA_WITH_AES_128_CBC_SHA256Z#TLS_DHE_DSS_WITH_AES_128_CBC_SHA256Z TLS_DHE_RSA_WITH_AES_128_CBC_SHAZ TLS_DHE_DSS_WITH_AES_128_CBC_SHAZ TLS_RSA_WITH_AES_256_GCM_SHA384Z TLS_RSA_WITH_AES_128_GCM_SHA256Z TLS_RSA_WITH_AES_256_CBC_SHA256Z TLS_RSA_WITH_AES_128_CBC_SHA256Z TLS_RSA_WITH_AES_256_CBC_SHAZ TLS_RSA_WITH_AES_128_CBC_SHArN ZPROTOCOL_SSLv23Z
kTLSProtocol1ZkTLSProtocol12r hasattrZ
kSSLProtocol2r Z
kSSLProtocol3r r ZkTLSProtocol11r r r r r r: r? Z SSLReadFuncrc Z SSLWriteFuncrd objectr@ r r r r r r s 76