관리-도구
편집 파일: low_level.cpython-39.pyc
a t~�`T6 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z e� dej�Zdd� Zdd � Zd d � Zd d � Zdd� Zd&dd�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d d d!d"d#�Z d$d%� Z dS )'a� Low-level helpers for the SecureTransport bindings. These are Python functions that are not directly related to the high-level APIs but are necessary to get them to work. They include a whole bunch of low-level CoreFoundation messing about and memory management. The concerns in this module are almost entirely about trying to avoid memory leaks and providing appropriate and useful assistance to the higher-level code. � N� )�CFConst�CoreFoundation�Securitys; -----BEGIN CERTIFICATE----- (.*?) -----END CERTIFICATE-----c C s t �t j| t| ��S )zv Given a bytestring, create a CFData object from it. This CFData object must be CFReleased by the caller. )r � CFDataCreate�kCFAllocatorDefault�len)Z bytestring� r �N/usr/lib/python3.9/site-packages/urllib3/contrib/_securetransport/low_level.py�_cf_data_from_bytes s �r c C sZ t | �}dd� | D �}dd� | D �}tj| |� }tj| |� }t�tj|||tjtj�S )zK Given a list of Python tuples, create an associated CFDictionary. c s s | ]}|d V qdS )r Nr ��.0�tr r r � <genexpr>, � z-_cf_dictionary_from_tuples.<locals>.<genexpr>c s s | ]}|d V qdS )r Nr r r r r r - r )r r � CFTypeRefZCFDictionaryCreater Z kCFTypeDictionaryKeyCallBacksZ kCFTypeDictionaryValueCallBacks)ZtuplesZdictionary_size�keys�valuesZcf_keysZ cf_valuesr r r �_cf_dictionary_from_tuples% s �r c C s t �| �}t�tj|tj�}|S )zi Given a Python binary data, create a CFString. The string must be CFReleased by the caller. )�ctypesZc_char_pr ZCFStringCreateWithCStringr r �kCFStringEncodingUTF8)Zpy_bstrZc_str�cf_strr r r �_cfstr; s �r c C s� d}znt �t jdt�t j��}|s*td��| D ]@}t|�}|sFtd��zt �||� W t � |� q.t � |� 0 q.W nB t y� } z*|r�t � |� t � d|f ��W Y d}~n d}~0 0 |S )z� Given a list of Python binary data, create an associated CFMutableArray. The array must be CFReleased by the caller. Raises an ssl.SSLError on failure. Nr �Unable to allocate memory!z Unable to allocate array: %s) r �CFArrayCreateMutabler r �byref�kCFTypeArrayCallBacks� MemoryErrorr �CFArrayAppendValue� CFRelease� BaseException�ssl�SSLError)ZlstZcf_arr�itemr �er r r �_create_cfstring_arrayI s* � &r% c C sn t �| t �t j��}t�|tj�}|du rXt �d�}t� ||dtj�}|sRt d��|j }|durj|� d�}|S )z� Creates a Unicode string from a CFString object. Used entirely for error reporting. Yes, it annoys me quite a lot that this function is this complex. Ni z'Error copying C string from CFStringRef�utf-8) r �castZPOINTERZc_void_pr ZCFStringGetCStringPtrr r Zcreate_string_bufferZCFStringGetCString�OSError�value�decode)r) Zvalue_as_void_p�string�buffer�resultr r r �_cf_string_to_unicodeh s � � r. c C s\ | dkr dS t �| d�}t|�}t�|� |du s:|dkrBd| }|du rPtj}||��dS )z[ Checks the return code and throws an exception if there is an error to report r N� z OSStatus %s)r ZSecCopyErrorMessageStringr. r r r! r" )�errorZexception_classZcf_error_string�outputr r r �_assert_no_error� s r2 c C s� | � dd�} dd� t�| �D �}|s.t�d��t�tjdt� tj ��}|sTt�d��z`|D ]V}t |�}|stt�d��t � tj|�}t�|� |s�t�d��t�||� t�|� qZW n ty� t�|� Y n0 |S ) z� Given a bundle of certs in PEM format, turns them into a CFArray of certs that can be used to validate a cert chain. s � c S s g | ]}t �|�d ���qS )r )�base64Z b64decode�group)r �matchr r r � <listcomp>� s z(_cert_array_from_pem.<locals>.<listcomp>z No root certificates specifiedr r z Unable to build cert object!)�replace� _PEM_CERTS_RE�finditerr! r" r r r r r r r r Z SecCertificateCreateWithDatar r � Exception)Z pem_bundleZ der_certsZ cert_arrayZ der_bytesZcertdataZcertr r r �_cert_array_from_pem� s: � � � r< c C s t �� }t�| �|kS )z= Returns True if a given CFTypeRef is a certificate. )r ZSecCertificateGetTypeIDr � CFGetTypeID�r# Zexpectedr r r �_is_cert� s r? c C s t �� }t�| �|kS )z; Returns True if a given CFTypeRef is an identity. )r ZSecIdentityGetTypeIDr r= r> r r r � _is_identity� s r@ c C s� t �d�} t�| dd� ��d�}t�| dd� �}t�� }t j�||�� d�}t � � }t � |t |�|ddt�|��}t|� ||fS )a� This function creates a temporary Mac keychain that we can use to work with credentials. This keychain uses a one-time password and a temporary file to store the data. We expect to have one keychain per socket. The returned SecKeychainRef must be freed by the caller, including calling SecKeychainDelete. Returns a tuple of the SecKeychainRef and the path to the temporary directory that contains it. �( N� r& F)�os�urandomr4 Z b16encoder* �tempfileZmkdtemp�path�join�encoder ZSecKeychainRefZSecKeychainCreater r r r2 )Z random_bytes�filename�passwordZ tempdirectoryZ keychain_path�keychain�statusr r r �_temporary_keychain� s �rM c C s. g }g }d}t |d��}|�� }W d � n1 s40 Y z�t�tj|t|��}t�� }t�|ddddd| t � |��}t |� t� |�} t | �D ]X} t�|| �} t �| tj�} t| �r�t�| � |�| � q�t| �r�t�| � |�| � q�W |r�t�|� t�|� n |�rt�|� t�|� 0 ||fS )z� Given a single file, loads all the trust objects from it into arrays and the keychain. Returns a tuple of lists: the first list is a list of identities, the second a list of certs. N�rbr )�open�readr r r r Z CFArrayRefr Z SecItemImportr r r2 ZCFArrayGetCount�rangeZCFArrayGetValueAtIndexr' r r? ZCFRetain�appendr@ r ) rK rF � certificates� identitiesZ result_array�fZ raw_filedataZfiledatar- Z result_count�indexr# r r r �_load_items_from_file� sL & �� � rW c G s g }g }dd� |D �}z�|D ]&}t | |�\}}|�|� |�|� q |s�t�� }t�| |d t�|��}t|� |�|� t � |� d�� t � t j dt�t j��} t�||�D ]} t �| | � q�| W t�||�D ]} t � | � q�S ]} t � | � q�n t�||�D ]} t � | � �q 0 dS )z� Load certificates and maybe keys from a number of files. Has the end goal of returning a CFArray containing one SecIdentityRef, and then zero or more SecCertificateRef objects, suitable for use as a client certificate trust chain. c s s | ]}|r|V qd S )Nr )r rF r r r r Q r z*_load_client_cert_chain.<locals>.<genexpr>r N)rW �extendr ZSecIdentityRefZ SecIdentityCreateWithCertificater r r2 rR r r �popr r r � itertools�chainr ) rK �pathsrS rT Z file_pathZnew_identitiesZ new_certsZ new_identityrL Z trust_chainr# �objr r r �_load_client_cert_chain- s> � � ��r^ )r � )� r )r` r )r` r_ )r` r` )ZSSLv2ZSSLv3ZTLSv1zTLSv1.1zTLSv1.2c C sH t | \}}d}d}t�d||�}t|�}d}t�d||||�| }|S )z6 Builds a TLS alert record for an unknown CA. r_ �0 z>BB� z>BBBH)�TLS_PROTOCOL_VERSIONS�struct�packr ) �versionZver_majZver_minZseverity_fatalZdescription_unknown_ca�msgZmsg_lenZrecord_type_alert�recordr r r �_build_tls_unknown_ca_alert� s ri )N) �__doc__r4 r rZ rC �rer! rd rE Zbindingsr r r �compile�DOTALLr9 r r r r% r. r2 r< r? r@ rM rW r^ rc ri r r r r �<module> s> � -#7L�