관리-도구
편집 파일: pyopenssl.cpython-38.pyc
U )�e�J � @ sP U d Z ddlmZ ddlZddlmZ zddlmZ W n$ e k rZ G dd� de �ZY nX ddl Z ddl Z ddl Z ddlmZ dd lmZ dd lmZ d d lmZ e jr�dd lmZ ddgZejjejj ejj ejj e j ejj iZ e!e d��r e!ejd��r ejj"e e j#< e!e d��r4e!ejd��r4ejj$e e j%< e j&ejj'e j(ejj)e j*ejj)ejj+ iZ,dd� e,�-� D �Z.e/ejdd�e/ejdd�B Z0de1d< e/ejdd�Z2de1d< e/ejd d�Z3de1d < e/ejd d�Z4de1d < e/ejd d�Z5de1d!< e j6j7e0e j6j8e0e j6j9e0e2B e j6j:e0e2B e3B e j6j;e0e2B e3B e4B e j6j<e0e2B e3B e4B iZ=d"e1d#< e j6j7e0e2B e3B e4B e5B e j6j8e0e3B e4B e5B e j6j9e0e4B e5B e j6j:e0e5B e j6j;e0e j6j<e0iZ>d"e1d$< d%Z?ejj@ZAe �BeC�ZDd&d'�d(d�ZEd&d'�d)d�ZFd&d'�d*d+�ZGd,d-d.�d/d0�ZHd1d2d3�d4d5�ZIG d6d7� d7�ZJejKeJ_KG d8d9� d9�ZLd:d1dddd;d<�d=d>�ZMdS )?a� Module for using pyOpenSSL as a TLS backend. This module was relevant before the standard library ``ssl`` module supported SNI, but now that we've dropped support for Python 2.7 all relevant Python versions support SNI so **this module is no longer recommended**. This needs the following packages installed: * `pyOpenSSL`_ (tested with 16.0.0) * `cryptography`_ (minimum 1.3.4, from pyopenssl) * `idna`_ (minimum 2.0) However, pyOpenSSL depends on cryptography, so while we use all three directly here we end up having relatively few packages required. You can install them with the following command: .. code-block:: bash $ python -m pip install pyopenssl cryptography idna To activate certificate checking, call :func:`~urllib3.contrib.pyopenssl.inject_into_urllib3` from your Python code before you begin making HTTP requests. This can be done in a ``sitecustomize`` module, or at any other time before your application begins using ``urllib3``, like this: .. code-block:: python try: import urllib3.contrib.pyopenssl urllib3.contrib.pyopenssl.inject_into_urllib3() except ImportError: pass .. _pyopenssl: https://www.pyopenssl.org .. _cryptography: https://cryptography.io .. _idna: https://github.com/kjd/idna � )� annotationsN)�x509)�UnsupportedExtensionc @ s e Zd ZdS )r N)�__name__� __module__� __qualname__� r r �G/opt/hc_python/lib/python3.8/site-packages/urllib3/contrib/pyopenssl.pyr 2 s r )�BytesIO)�socket)�timeout� )�util��X509�inject_into_urllib3�extract_from_urllib3�PROTOCOL_TLSv1_1�TLSv1_1_METHOD�PROTOCOL_TLSv1_2�TLSv1_2_METHODc C s i | ]\}}||�qS r r )�.0�k�vr r r � <dictcomp>Y s r � OP_NO_SSLv2� OP_NO_SSLv3�int�_OP_NO_SSLv2_OR_SSLv3� OP_NO_TLSv1� _OP_NO_TLSv1� OP_NO_TLSv1_1�_OP_NO_TLSv1_1� OP_NO_TLSv1_2�_OP_NO_TLSv1_2� OP_NO_TLSv1_3�_OP_NO_TLSv1_3zdict[int, int]� _openssl_to_ssl_minimum_version� _openssl_to_ssl_maximum_versioni @ �None��returnc C s& t � tt_ttj_dt_dtj_dS )z7Monkey-patch urllib3 with PyOpenSSL-backed SSL-support.TN)�_validate_dependencies_met�PyOpenSSLContextr � SSLContext�ssl_� IS_PYOPENSSLr r r r r � s c C s t t_t tj_dt_dtj_dS )z4Undo monkey-patching by :func:`inject_into_urllib3`.FN)�orig_util_SSLContextr r. r/ r0 r r r r r � s c C sR ddl m} t| dd�dkr$td��ddlm} |� }t|dd�dkrNtd��dS ) z{ Verifies that PyOpenSSL's package-level dependencies have been met. Throws `ImportError` if they are not met. r )� Extensions�get_extension_for_classNzX'cryptography' module missing required functionality. Try upgrading to v1.3.4 or newer.r Z_x509zS'pyOpenSSL' module missing required functionality. Try upgrading to v0.14 or newer.)Z cryptography.x509.extensionsr2 �getattr� ImportError�OpenSSL.cryptor )r2 r r r r r r, � s � �r, �str� str | None��namer+ c C s: ddd�dd�}d| kr | S || �}|dkr0dS |� d�S ) a% Converts a dNSName SubjectAlternativeName field to the form used by the standard library on the given Python version. Cryptography produces a dNSName as a unicode string that was idna-decoded from ASCII bytes. We need to idna-encode that string to get it back, and then on Python 3 we also need to convert to unicode via UTF-8 (the stdlib uses PyUnicode_FromStringAndSize on it, which decodes via UTF-8). If the name cannot be idna-encoded then we return None signalling that the name given should be skipped. r7 � bytes | Noner9 c S sr ddl }zJdD ]8}| �|�r| t|�d� } |�d�|�| � W S q|�| �W S |jjk rl Y dS X dS )z� Borrowed wholesale from the Python Cryptography Project. It turns out that we can't just safely call `idna.encode`: it can explode for wildcard names. This avoids that problem. r N)z*.�.�ascii)�idna� startswith�len�encode�core� IDNAError)r: r> �prefixr r r � idna_encode� s z'_dnsname_to_stdlib.<locals>.idna_encode�:N�utf-8)�decode)r: rE � encoded_namer r r �_dnsname_to_stdlib� s rJ r zlist[tuple[str, str]])� peer_certr+ c C s� | � � }z|j�tj�j}W nZ tjk r6 g Y S tjttj t fk rv } zt � d|� g W Y �S d}~X Y nX dd� t t|�tj��D �}|�dd� |�tj�D �� |S )zU Given an PyOpenSSL certificate, provides all the subject alternative names. z�A problem was encountered with the certificate that prevented urllib3 from finding the SubjectAlternativeName field. This can affect certificate validation. The error was %sNc S s g | ]}|d k rd|f�qS )N�DNSr �r r: r r r � <listcomp> s �z%get_subj_alt_name.<locals>.<listcomp>c s s | ]}d t |�fV qdS )z IP AddressN)r7 rM r r r � <genexpr> s z$get_subj_alt_name.<locals>.<genexpr>)Zto_cryptography� extensionsr3 r ZSubjectAlternativeName�valueZExtensionNotFoundZDuplicateExtensionr ZUnsupportedGeneralNameType� UnicodeError�log�warning�maprJ Zget_values_for_typeZDNSName�extendZ IPAddress)rK �cert�ext�e�namesr r r �get_subj_alt_name� s. �� � �r[ c @ s� e Zd ZdZd/ddddd�dd �Zd d �d d �Zdd �dd�Zdddd�dd�Zddd d�dd�Zddd�dd�Z dd d�d d �Z ddd�d d �Z dd �d d!�Z dd �d"d#�Z dd �d$d%�Zd0dd'd(�d)d*�Zd+d �d,d-�Zd.S )1� WrappedSocketz@API-compatibility wrapper for Python OpenSSL's Connection-class.T�OpenSSL.SSL.Connection� socket_cls�boolr) )� connectionr �suppress_ragged_eofsr+ c C s" || _ || _|| _d| _d| _d S �Nr F)r` r ra �_io_refs�_closed)�selfr` r ra r r r �__init__ s zWrappedSocket.__init__r r* c C s | j �� S �N)r �fileno�re r r r rh ! s zWrappedSocket.filenoc C s* | j dkr| j d8 _ | jr&| �� d S )Nr � )rc rd �closeri r r r �_decref_socketios% s z WrappedSocket._decref_socketiosz typing.Any�bytes)�args�kwargsr+ c O s4 z| j j||�}W �n tjjk rj } z4| jrD|jdkrDW Y � dS t|jd t|��|�W 5 d }~X Y n� tjj k r� | j � � tjj kr�Y dS � Y n� tjj k r� } z8t �| j| j�� �s�td�|�n| j||� W Y �S W 5 d }~X Y n> tjjk �r* } zt�d|���|�W 5 d }~X Y nX |S d S )N����zUnexpected EOF� r � The read operation timed out� read error: )r` �recv�OpenSSL�SSL� SysCallErrorra rn �OSErrorr7 �ZeroReturnError� get_shutdown�RECEIVED_SHUTDOWN� WantReadErrorr � wait_for_readr � gettimeoutr �Error�ssl�SSLError)re rn ro �datarY r r r ru + s" & $$zWrappedSocket.recvc O s, z| j j||�W S tjjk rf } z4| jr@|jdkr@W Y � dS t|jd t|��|�W 5 d }~X Y n� tjj k r� | j � � tjj kr�Y dS � Y n� tjj k r� } z8t �| j| j�� �s�td�|�n| j||� W Y �S W 5 d }~X Y n: tjjk �r& } zt�d|���|�W 5 d }~X Y nX d S )Nrp r rs rt )r` � recv_intorv rw rx ra rn ry r7 rz r{ r| r} r r~ r r r r� r� r� )re rn ro rY r r r r� D s & $zWrappedSocket.recv_into�float)r r+ c C s | j �|�S rg )r � settimeout)re r r r r r� [ s zWrappedSocket.settimeout)r� r+ c C s� z| j �|�W S tjjk rX } z(t�| j| j�� �s@t � |�W Y �q W 5 d }~X Y q tjj k r� } zt |j d t |��|�W 5 d }~X Y q X q d S �Nr )r` �sendrv rw ZWantWriteErrorr �wait_for_writer r r rx ry rn r7 )re r� rY r r r �_send_until_done^ s z WrappedSocket._send_until_donec C s4 d}|t |�k r0| �|||t � �}||7 }qd S r� )r@ r� �SSL_WRITE_BLOCKSIZE)re r� � total_sent�sentr r r �sendalli s �zWrappedSocket.sendallc C s | j �� d S rg )r` �shutdownri r r r r� q s zWrappedSocket.shutdownc C s d| _ | jdkr| �� d S )NTr )rd rc � _real_closeri r r r rk u s zWrappedSocket.closec C s, z | j �� W S tjjk r& Y d S X d S rg )r` rk rv rw r� ri r r r r� z s zWrappedSocket._real_closeFz"dict[str, list[typing.Any]] | None)� binary_formr+ c C sD | j �� }|s|S |r(tj�tjj|�S d|�� jffft|�d�S )N� commonName)�subject�subjectAltName) r` Zget_peer_certificaterv ZcryptoZdump_certificateZ FILETYPE_ASN1Z get_subjectZCNr[ )re r� r r r r � getpeercert� s �zWrappedSocket.getpeercertr7 c C s | j �� S rg )r` Zget_protocol_version_nameri r r r �version� s zWrappedSocket.versionN)T)F)r r r �__doc__rf rh rl ru r� r� r� r� r� rk r� r� r� r r r r r\ s � �r\ c @ sB e Zd ZdZddd�dd�Zedd�dd ��Zejddd �d d ��Zedd�d d ��Zejddd �dd ��Zdd�dd�Z ddd�dd�Z d7ddddd�dd�Z d8d dddd �d d �Z d dd!�d"d#�Z d9d&d'd'd'd(d)d*�d+d,�Zdd�d-d.�Zedd�d/d0��Zejddd1�d2d0��Zedd�d3d4��Zejddd5�d6d4��ZdS ):r- z� I am a wrapper class for the PyOpenSSL ``Context`` object. I am responsible for translating the interface of the standard library ``SSLContext`` object to calls into PyOpenSSL. r r) )�protocolr+ c C s> t | | _tj�| j�| _d| _d| _tj j | _ tj j | _ d S rb )�_openssl_versionsr� rv rw �Context�_ctx�_options�check_hostnamer� � TLSVersion�MINIMUM_SUPPORTED�_minimum_version�MAXIMUM_SUPPORTED�_maximum_version)re r� r r r rf � s zPyOpenSSLContext.__init__r* c C s | j S rg )r� ri r r r �options� s zPyOpenSSLContext.options)rQ r+ c C s || _ | �� d S rg )r� �_set_ctx_options�re rQ r r r r� � s c C s t | j�� S rg )�_openssl_to_stdlib_verifyr� Zget_verify_moderi r r r � verify_mode� s z PyOpenSSLContext.verify_modezssl.VerifyModec C s | j �t| t� d S rg )r� Z set_verify�_stdlib_to_openssl_verify�_verify_callbackr� r r r r� � s c C s | j �� d S rg )r� �set_default_verify_pathsri r r r r� � s z)PyOpenSSLContext.set_default_verify_pathsz bytes | str)�ciphersr+ c C s$ t |t�r|�d�}| j�|� d S )NrG )� isinstancer7 rA r� Zset_cipher_list)re r� r r r � set_ciphers� s z PyOpenSSLContext.set_ciphersNr8 r; )�cafile�capath�cadatar+ c C s� |d k r|� d�}|d k r$|� d�}z*| j�||� |d k rL| j�t|�� W n8 tjjk r� } zt�d|���|�W 5 d }~X Y nX d S )NrG z%unable to load trusted certificates: ) rA r� �load_verify_locationsr rv rw r� r� r� )re r� r� r� rY r r r r� � s z&PyOpenSSLContext.load_verify_locationsr7 )�certfile�keyfile�passwordr+ c s� zP| j �|� � d k r>t� t�s*� �d�� | j �� fdd�� | j �|pJ|� W n8 tjj k r� } zt � d|���|�W 5 d }~X Y nX d S )NrG c s � S rg r )�_�r� r r �<lambda>� rr z2PyOpenSSLContext.load_cert_chain.<locals>.<lambda>z"Unable to load certificate chain: ) r� Zuse_certificate_chain_filer� rm rA Z set_passwd_cbZuse_privatekey_filerv rw r� r� r� )re r� r� r� rY r r� r �load_cert_chain� s z PyOpenSSLContext.load_cert_chainzlist[bytes | str])� protocolsr+ c C s dd� |D �}| j �|�S )Nc S s g | ]}t j �|d ��qS )r= )r �to_bytes)r �pr r r rN � s z7PyOpenSSLContext.set_alpn_protocols.<locals>.<listcomp>)r� Zset_alpn_protos)re r� r r r �set_alpn_protocols� s z#PyOpenSSLContext.set_alpn_protocolsFTr^ r_ zbytes | str | Noner\ )�sock� server_side�do_handshake_on_connectra �server_hostnamer+ c C s� t j�| j|�}|r>tj�|�s>t|t�r4|� d�}|� |� |� � z |� � W q� t jj k r� } z&t�||�� �s�td�|�W Y �qFW 5 d }~X Y q� t jjk r� } zt�d|���|�W 5 d }~X Y q�X q�qFt||�S )NrG zselect timed outzbad handshake: )rv rw � Connectionr� r r/ � is_ipaddressr� r7 rA Zset_tlsext_host_nameZset_connect_state� do_handshaker} r~ r r r� r� r� r\ )re r� r� r� ra r� �cnxrY r r r � wrap_socket� s $z PyOpenSSLContext.wrap_socketc C s& | j �| jt| j B t| j B � d S rg )r� Z set_optionsr� r' r� r( r� ri r r r r� s ���z!PyOpenSSLContext._set_ctx_optionsc C s | j S rg )r� ri r r r �minimum_version s z PyOpenSSLContext.minimum_version)r� r+ c C s || _ | �� d S rg )r� r� )re r� r r r r� s c C s | j S rg )r� ri r r r �maximum_version s z PyOpenSSLContext.maximum_version)r� r+ c C s || _ | �� d S rg )r� r� )re r� r r r r� s )NNN)NN)FTTN)r r r r� rf �propertyr� �setterr� r� r� r� r� r� r� r� r� r� r r r r r- � sD � � � r- r] r_ )r� r �err_no� err_depth� return_coder+ c C s |dkS r� r )r� r r� r� r� r r r r� s r� )Nr� � __future__r Z OpenSSL.SSLrv Z cryptographyr Zcryptography.x509r r5 � Exception�loggingr� �typing�ior r r^ r � r � TYPE_CHECKINGr6 r �__all__r/ � PROTOCOL_TLSrw Z SSLv23_METHOD�PROTOCOL_TLS_CLIENT�PROTOCOL_TLSv1Z TLSv1_METHODr� �hasattrr r r r � CERT_NONEZ VERIFY_NONE� CERT_OPTIONALZ VERIFY_PEER� CERT_REQUIREDZVERIFY_FAIL_IF_NO_PEER_CERTr� �itemsr� r4 r �__annotations__r r" r$ r& r� r� �TLSv1�TLSv1_1�TLSv1_2�TLSv1_3r� r' r( r� r. r1 � getLoggerr rS r r r, rJ r[ r\ �makefiler- r� r r r r �<module> s� ( � �� � � ���� � )0