관리-도구
편집 파일: low_level.cpython-38.pyc
U ��.e�. � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z e� dej �Zdd� Zdd � Zd d � Zdd d �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )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� )�Security�CoreFoundation�CFConsts; -----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 �Z/usr/lib/python3.8/site-packages/pip/_vendor/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>, s z-_cf_dictionary_from_tuples.<locals>.<genexpr>c s s | ]}|d V qdS )r Nr r r r r r - s )r r � CFTypeRefZCFDictionaryCreater Z kCFTypeDictionaryKeyCallBacksZ kCFTypeDictionaryValueCallBacks)ZtuplesZdictionary_size�keys�valuesZcf_keysZ cf_valuesr r r �_cf_dictionary_from_tuples% s �r c C sn t �| t �t j��}t�|tj�}|dkrXt �d�}t� ||dtj�}|sRt d��|j }|dk rj|� 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) �ctypes�castZPOINTERZc_void_pr ZCFStringGetCStringPtrr ZkCFStringEncodingUTF8Zcreate_string_bufferZCFStringGetCString�OSError�value�decode)r Zvalue_as_void_p�string�buffer�resultr r r �_cf_string_to_unicode; s& � � r c C s\ | dkr dS t �| d�}t|�}t�|� |dks:|dkrBd| }|dkrPtj}||��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 � CFRelease�ssl�SSLError)�errorZexception_classZcf_error_string�outputr r r �_assert_no_errorT s r$ 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 tk r� t�|� Y nX |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>q s z(_cert_array_from_pem.<locals>.<listcomp>z No root certificates specifiedr zUnable to allocate memory!z Unable to build cert object!)�replace� _PEM_CERTS_RE�finditerr r! r �CFArrayCreateMutabler r �byref�kCFTypeArrayCallBacksr r Z SecCertificateCreateWithDatar �CFArrayAppendValue� Exception)Z pem_bundleZ der_certsZ cert_arrayZ der_bytesZcertdataZcertr r r �_cert_array_from_pemi s< � � � r2 c C s t �� }t�| �|kS )z= Returns True if a given CFTypeRef is a certificate. )r ZSecCertificateGetTypeIDr � CFGetTypeID��itemZexpectedr r r �_is_cert� s r6 c C s t �� }t�| �|kS )z; Returns True if a given CFTypeRef is an identity. )r ZSecIdentityGetTypeIDr r3 r4 r r r � _is_identity� s r7 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�urandomr& Z b16encoder �tempfileZmkdtemp�path�join�encoder ZSecKeychainRefZSecKeychainCreater r r. r$ )Z random_bytes�filenameZpasswordZ tempdirectoryZ keychain_path�keychain�statusr r r �_temporary_keychain� s �rC c C s g }g }d}t |d��}|�� }W 5 Q R X 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 5 |r�t�|� t�|� X ||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 r Z CFArrayRefr Z SecItemImportr r. r$ ZCFArrayGetCount�rangeZCFArrayGetValueAtIndexr r r6 ZCFRetain�appendr7 ) rA r= � certificates� identitiesZ result_array�fZ raw_filedataZfiledatar Z result_count�indexr5 r r r �_load_items_from_file� sJ �� rM c G s� g }g }dd� |D �}z�|D ]&}t| |�\}}|�|� |�|� q |s�t�� }t�| |d t � |��} t | � |� |� t�|� d�� t�tjdt � tj��} t �||�D ]} t�| | � q�| W �S t �||�D ]}t�|� q�X 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 r= r r r r $ s z*_load_client_cert_chain.<locals>.<genexpr>r N)� itertools�chainr r rM �extendr ZSecIdentityRefZ SecIdentityCreateWithCertificater r. r$ rH �popr- r r/ r0 ) rA �pathsrI rJ �objZ file_pathZnew_identitiesZ new_certsZ new_identityrB Z trust_chainr5 r r r �_load_client_cert_chain s8 � �rT )N)�__doc__r& r rN �rer: r r< Zbindingsr r r �compile�DOTALLr+ r r r r$ r2 r6 r7 rC rM rT r r r r �<module> s, � -#7