Файловый менеджер - Редактировать - /opt/imh-python/lib/python3.9/site-packages/zmq/tests/__pycache__/test_auth.cpython-39.pyc
Ðазад
a a�h�P � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl m Z d dl m Z mZmZ G dd� de �ZG dd� de�Zdd d �Zd d � Zdd� ZG dd� de�ZdS )� N)�ThreadAuthenticator)�u)�BaseZMQTestCase�SkipTest� skip_pypyc sH e Zd Z� fdd�Zdd� Z� fdd�Zdd� Zd d � Zd d � Z� Z S ) �BaseAuthTestCasec s� t �� dk rtd��z t �� W n t jy< td��Y n0 tt| ��� t� d�� tj � | � � | _ | j �� | �� \| _| _| _d S )N)� r z security is new in libzmq 4.0z.security requires libzmq to have curve supportzzmq.auth)�zmqZzmq_version_infor Z curve_keypairZZMQError�superr �setUp�logging� getLogger�setLevel�DEBUG� make_auth�auth�start� create_certs�base_dir�public_keys_dir�secret_keys_dir��self�� __class__� ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/python3.9/site-packages/zmq/tests/test_auth.pyr s zBaseAuthTestCase.setUpc C s t � �d S �N)�NotImplementedErrorr r r r r $ s zBaseAuthTestCase.make_authc s4 | j r| j �� d | _ | �| j� tt| ��� d S r )r �stop� remove_certsr r r �tearDownr r r r r! ' s zBaseAuthTestCase.tearDownc C s� t �� }tj�|d�}tj�|d�}tj�|d�}t�|� t�|� t�|� tj�|d�\}}tj�|d�\}}t� |�D ].} | � d�r~t � tj�|| �tj�|d�� q~t� |�D ].} | � d�r�t � tj�|| �tj�|d�� q�|||fS ) z$Create CURVE certificates for a testZ certificatesZ public_keysZ private_keys�server�clientz.key�.z .key_secret) �tempfile�mkdtemp�os�path�join�mkdirr r Zcreate_certificates�listdir�endswith�shutil�move) r r Zkeys_dirr r Zserver_public_file�server_secret_fileZclient_public_file�client_secret_file�key_filer r r r . s( � �z BaseAuthTestCase.create_certsc C s t �|� dS )z Remove certificates for a testN)r- �rmtree)r r r r r r M s z BaseAuthTestCase.remove_certsc C sH t j�|d�}t j�|d�}tj�|�\}}tj�|�\}}||||fS )z)Return server and client certificate keyszserver.key_secretzclient.key_secret)r' r( r) r r Zload_certificate)r r r/ r0 � server_public� server_secret� client_public� client_secretr r r � load_certsQ s zBaseAuthTestCase.load_certs) �__name__� __module__� __qualname__r r r! r r r7 � __classcell__r r r r r s r c @ s\ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z e dd� �Z dS )�TestThreadAuthenticationz'Test authentication running in a threadc C s t | j�S r )r �contextr r r r r _ s z"TestThreadAuthentication.make_authc C sh d}d}|� |�}|�d||f � dg}|�dtj�rB|�|� |�d�rd|�� }| �||� d}|S )�9Check if client can connect to server using tcp transportF�tcp://127.0.0.1�%s:%i� Hello Worldi� T)�bind_to_random_port�connect�pollr �POLLOUT�send_multipartZrecv_multipartZ assertEqual)r r" r# �result�iface�port�msgZrcvd_msgr r r � can_connectb s z$TestThreadAuthentication.can_connectc C sx | j �� d| _ | �� | _| �tj�}| �tj�}| �| � ||�� | �tj�}d|_ | �tj�}| �| � ||�� dS )zthreaded auth - NULLN� global) r r �Contextr= �socketr �PUSH�PULL� assertTruerK � zap_domain�r r" r# r r r � test_nullq s z"TestThreadAuthentication.test_nullc C s@ | j �d� | �tj�}d|_| �tj�}| �| �||�� dS )zthreaded auth - Blacklist� 127.0.0.1rL N) r �denyrN r rO rR rP � assertFalserK rS r r r �test_blacklist� s z'TestThreadAuthentication.test_blacklistc C s@ | j �d� | �tj�}d|_| �tj�}| �| �||�� dS )zthreaded auth - WhitelistrU rL N) r �allowrN r rO rR rP rQ rK rS r r r �test_whitelist� s z'TestThreadAuthentication.test_whitelistc C s | � tj�}d|_| � tj�}d|_d|_| �| �||�� | � tj�}d|_| � tj�}d|_d|_| j j dddid� | � | �||�� | � tj�}d|_| � tj�}d|_d|_| �| �||�� | j � � d | _ | � tj�}| � tj�}| � | �||�� |� � |� � d S ) zthreaded auth - PLAINT� admin� Password�*�admin�Password��domain� passwords� BogusN)rN r rO � plain_serverrP �plain_username�plain_passwordrW rK r �configure_plainrQ r �closerS r r r � test_plain� s4 z#TestThreadAuthentication.test_plainc C sr | j �d� | �| j�}|\}}}}| �tj�}||_||_d|_ | �tj �}||_||_||_ | � | � ||�� | j jdtj jd� | �tj�}||_||_d|_ | �tj �}||_||_||_ | �| � ||�� | j jd| jd� | �tj �}||_||_d|_ | �tj�}||_||_||_ | � ||��s4J �| j �� d| _ | �tj�}| �tj �}| �| � ||�� dS )�threaded auth - CURVErU Tr] �ra �locationN)r rY r7 r rN r rO �curve_publickey�curve_secretkey� curve_serverrP �curve_serverkeyrW rK �configure_curve�CURVE_ALLOW_ANYrQ r r )r �certsr3 r4 r5 r6 r" r# r r r � test_curve� sJ z#TestThreadAuthentication.test_curvec s^ | j �d� | �| j�}|\}}� }| �tj�}||_||_d|_ | �tj �}� |_||_||_ | � | � ||�� G � fdd�dt�}|� }| j j|d� | �tj�}||_||_d|_ | �tj �}� |_||_||_ | �| � ||�� G dd� dt�} | � }| j j|d� | �tj�}||_||_d|_ | �tj �}� |_||_||_ | � | � ||�� dS ) z2threaded auth - CURVE with callback authenticationrU Tc s e Zd Z� fdd�Zdd� ZdS )zITestThreadAuthentication.test_curve_callback.<locals>.CredentialsProviderc s � | _ d S r �r# r �r5 r r �__init__ s zRTestThreadAuthentication.test_curve_callback.<locals>.CredentialsProvider.__init__c S s || j krdS dS d S �NTFru �r ra �keyr r r �callback s zRTestThreadAuthentication.test_curve_callback.<locals>.CredentialsProvider.callbackN�r8 r9 r: rw r{ r rv r r �CredentialsProvider s r} )Zcredentials_providerc @ s e Zd Zdd� Zdd� ZdS )zNTestThreadAuthentication.test_curve_callback.<locals>.WrongCredentialsProviderc S s d| _ d S )NZWrongCredentialsru r r r r rw s zWTestThreadAuthentication.test_curve_callback.<locals>.WrongCredentialsProvider.__init__c S s || j krdS dS d S rx ru ry r r r r{ # s zWTestThreadAuthentication.test_curve_callback.<locals>.WrongCredentialsProvider.callbackNr| r r r r �WrongCredentialsProvider s r~ N)r rY r7 r rN r rO rm rn ro rP rp rW rK �objectZconfigure_curve_callbackrQ ) r rs r3 r4 r6 r" r# r} �providerr~ r rv r �test_curve_callback� sH z,TestThreadAuthentication.test_curve_callbackc C sx | j �d� | �| j�}|\}}}}| j jd| jd� | �tj�}||_ ||_ d|_ | �tj �}||_ ||_ ||_ | �||�s�J �|�d� | j|dd�}|jdks�J �z|�d�} W n tjy� Y n0 | t|�ks�J �d d � | j _| �tj �} || _ || _ || _ | �| |��sJ �| �d � | j|dd�}|jd k�s@J �z|�d�} W n tj�yd Y n0 | d k�stJ �d S )rj rU r] rk Ts testF)�copyzUser-Idc S s dS )N�customr )Z client_keyr r r �<lambda>U � z=TestThreadAuthentication.test_curve_user_id.<locals>.<lambda>s test2r� N)r rY r7 r rq r rN r rP rm rn ro rO rp rK �send�recv�bytes�getZZMQVersionErrorr Z curve_user_id) r rs r3 r4 r5 r6 r" r# rJ Zuser_idZclient2r r r �test_curve_user_id7 sF z+TestThreadAuthentication.test_curve_user_idN)r8 r9 r: �__doc__r rK rT rX rZ ri rt r� r r� r r r r r< \ s &2Cr< Tc s � �fdd�}|S )z*decorator for running tests with an IOLoopc s� �| �}| j }� r"| j�| j� n| j�| j� |�d| j� |�d| j� � r`|�d| j� n|�d| j � |� � | j r�| � | j � |S )N� g333333�?� ) �io_loop� pullstreamZon_recv�on_message_succeed�on_message_failZ call_later�attempt_connection�send_msg�on_test_timeout_fail�on_test_timeout_succeedr �fail_msgZfail)r �r�loop��expect_success�methodr r � test_methodj s z with_ioloop.<locals>.test_methodr )r� r� r� r r� r � with_iolooph s r� c C s t | d�S )NT�r� �r� r r r � should_auth� s r� c C s t | d�S )NFr� r� r r r �should_not_auth� s r� c s� e Zd ZdZ� fdd�Zdd� Z� fdd�Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z e dd� �Ze dd� �Zedd� �Ze dd� �Zed d � �Ze d d � �Zed d!� �Zed"d#� �Ze d$d%� �Ze d&d'� �Z� ZS )(�TestIOLoopAuthenticationz%Test authentication running in ioloopc s� zddl m} W n ty, t�d� Y n0 ddlm} d | _|�� | _ t t | �� � | � tj�| _| � tj�| _|�| j| j �| _|�| j| j �| _d S )Nr )�ioloopzRequires tornado)� zmqstream)�tornador� � ImportError�pytest�skipZ zmq.eventloopr� r� ZIOLoopr� r r� r rN r rO r" rP r# Z ZMQStream� pushstreamr� )r r� r� r r r r � s z TestIOLoopAuthentication.setUpc C s ddl m} || j| jd�S )Nr )�IOLoopAuthenticator)r� )Zzmq.auth.ioloopr� r= r� )r r� r r r r � s z"TestIOLoopAuthentication.make_authc s6 | j r| j �� d | _ | jjdd� tt| ��� d S )NT)Zall_fds)r r r� rh r r� r! r r r r r! � s z!TestIOLoopAuthentication.tearDownc C s( d}| j �|�}| j�d||f � dS )r> r? r@ N)r" rB r# rC )r rH rI r r r r� � s z+TestIOLoopAuthentication.attempt_connectionc C s dg}| j �|� dS )z&Send a message from server to a clientrA N)r� rF )r rJ r r r r� � s z!TestIOLoopAuthentication.send_msgc C s |dgkrd| _ | j�� dS )z$A message was received, as expected.rA zUnexpected message receivedN�r� r� r �r �framesr r r r� � s z+TestIOLoopAuthentication.on_message_succeedc C s d| _ | j�� dS )z%A message was received, unexpectedly.z/Received messaged unexpectedly, security failedNr� r� r r r r� � s z(TestIOLoopAuthentication.on_message_failc C s | j �� dS )z*Test timer expired, indicates test successN)r� r r r r r r� � s z0TestIOLoopAuthentication.on_test_timeout_succeedc C s d| _ | j�� dS )z*Test timer expired, indicates test failurezTest timed outNr� r r r r r� � s z-TestIOLoopAuthentication.on_test_timeout_failc C s | j �� d| _ dS )zioloop auth - NONEN)r r r r r r � test_none� s z"TestIOLoopAuthentication.test_nonec C s d| j _dS )zioloop auth - NULLrL N)r" rR r r r r rT � s z"TestIOLoopAuthentication.test_nullc C s | j �d� d| j_dS )zioloop auth - BlacklistrU rL N)r rV r" rR r r r r rX � s z'TestIOLoopAuthentication.test_blacklistc C s | j �d� | j�tjd� dS )zioloop auth - WhitelistrU rL N)r rY r" � setsockoptr Z ZAP_DOMAINr r r r rZ � s z'TestIOLoopAuthentication.test_whitelistc C s d| j _d| j _d| j_dS )z(ioloop auth - PLAIN, unconfigured serverr[ r\ TN)r# re rf r" rd r r r r � test_plain_unconfigured_server� s z7TestIOLoopAuthentication.test_plain_unconfigured_serverc C s0 d| j _d| j _d| j_| jjdddid� dS ) z&ioloop auth - PLAIN, configured serverr[ r\ Tr] r^ r_ r` N�r# re rf r" rd r rg r r r r � test_plain_configured_server� s z5TestIOLoopAuthentication.test_plain_configured_serverc C s0 d| j _d| j _d| j_| jjdddid� dS ) z&ioloop auth - PLAIN, bogus credentialsr[ rc Tr] r^ r_ r` Nr� r r r r � test_plain_bogus_credentials� s z5TestIOLoopAuthentication.test_plain_bogus_credentialsc C sX | � | j�}|\}}}}| j�d� || j_|| j_d| j_|| j_|| j_|| j_ dS )z(ioloop auth - CURVE, unconfigured serverrU TN) r7 r r rY r" rm rn ro r# rp �r rs r3 r4 r5 r6 r r r � test_curve_unconfigured_server� s z7TestIOLoopAuthentication.test_curve_unconfigured_serverc C sl | � | j�}|\}}}}| j�d� | jjdtjjd� || j_|| j_ d| j_ || j _|| j _ || j _ dS )z$ioloop auth - CURVE, CURVE_ALLOW_ANYrU r] rk TN) r7 r r rY rq r rr r" rm rn ro r# rp r� r r r �test_curve_allow_any s z-TestIOLoopAuthentication.test_curve_allow_anyc C sj | j �d� | �| j�}|\}}}}| j jd| jd� || j_|| j_d| j_ || j _|| j _|| j _ dS )z&ioloop auth - CURVE, configured serverrU r] rk TN) r rY r7 r rq r r" rm rn ro r# rp r� r r r � test_curve_configured_server s z5TestIOLoopAuthentication.test_curve_configured_server)r8 r9 r: r� r r r! r� r� r� r� r� r� r� r� rT r� rX rZ r� r� r� r� r� r� r; r r r r r� � s< r� )T)r r' r- �sysr% r� Zzmq.authr Zzmq.auth.threadr Zzmq.utils.strtypesr Z zmq.testsr r r r r< r� r� r� r� r r r r �<module> s" G
| ver. 1.1 | |
.
| PHP 8.3.30 | Ð“ÐµÐ½ÐµÑ€Ð°Ñ†Ð¸Ñ Ñтраницы: 0 |
proxy
|
phpinfo
|
ÐаÑтройка