File: //opt/saltstack/salt/lib/python3.10/site-packages/zmq/auth/__pycache__/thread.cpython-310.pyc
o
;j � @ sv d Z ddlZddlmZmZ ddlmZmZmZ ddl Z ddl
Z ddl m Z G dd� de�Z
G d d
� d
e �Zd
gZdS ) z>ZAP Authenticator in a Python Thread.
.. versionadded:: 14.1
� N)�Event�Thread)�Any�List�Optional� )�
Authenticatorc s� e Zd ZU dZejed< ejed< e ed< dZ
e ejj ed< de dejddf� fdd � Z
dd
d �Zd d
� Zdee defdd�Z� ZS )�AuthenticationThreadzgA Thread for running a zmq Authenticator
This is run in the background by ThreadAuthenticator
�pipe�loop�
authenticatorN�poller�returnc s. t � jdd� || _|j| _|| _t� | _d S )NT)�daemon)�super�__init__r �logr
r �started)�selfr r
�� __class__� �C/opt/saltstack/salt/lib/python3.10/site-packages/zmq/auth/thread.pyr s
z AuthenticationThread.__init__c C s\ t �� }z|�| �� � W | jr| j�� d| _|�� dS | jr)| j�� d| _|�� w )z*Start the Authentication Agent thread taskN)�asyncioZnew_event_loopZrun_until_complete�_runr
�close)r r r r r �run* s
�
zAuthenticationThread.runc � s� �t j�� | _| j�| jt j� | j�| jjt j� | j �
� t | j� � I d H �}| j|v r;| j�
� }| �|�r;d S | jj|v rP| jj�
� }| j�|�I d H q �N)�zmqr �Pollerr
�registerr
�POLLINr Z
zap_socketr �set�dict�pollZrecv_multipart�_handle_pipe_messageZhandle_zap_message)r �events�msgr r r r 7 s �
�zAuthenticationThread._runr' c C s@ |d }| j �d|� |dkrdS | j �d|� | j�d� dS )Nr z auth received API command %r� TERMINATETz!Invalid auth command from API: %rs ERRORF)r �debug�errorr
�send)r r' Zcommandr r r r% G s z)AuthenticationThread._handle_pipe_message�r N)�__name__�
__module__� __qualname__�__doc__r �Socket�__annotations__r ZAbstractEventLoopr r
r r r r r r �bytes�boolr% �
__classcell__r r r r r s
���
r c s� e Zd ZU dZded< dZeed< eed< dd ed
d ed e f� fd
d�
Z
d� fdd� Z d� fdd� Z de
fdd�Zddd�Z� ZS )�ThreadAuthenticatorz-Run ZAP authentication in a background threadz
zmq.Socketr
� �
pipe_endpoint�threadN�utf-8�contextz zmq.Context�encodingr c s4 t � j|||d� d | _dt| �� d�| _d | _d S )N)r; r< r z inproc://z.inproc)r r r
�idr8 r9 )r r; r< r r r r r \ s
z ThreadAuthenticator.__init__r c s� t � �� | jjtjtjd�| _d| j_| j� | j
� | jjtjtjd�}d|_|� | j
� t | |d�| _
| j
�� | j
jjdd�sFtd��dS )z Start the authentication thread)Z socket_classr )r r
�
)�timeoutz$Authenticator thread failed to startN)r �startr; �socketr ZPAIRr1 r
Zlinger�bindr8 �connectr r9 r �wait� RuntimeError)r Z thread_piper r r r@ g s
�zThreadAuthenticator.startc sH | j r | j �d� | �� r| j�� d| _| j �� d| _ t� �� dS )z Stop the authentication threadr( N)r
r+ �is_aliver9 �joinr r �stop�r r r r rH x s
zThreadAuthenticator.stopc C s t | jo| j�� �S )z$Is the ZAP thread currently running?)r4 r9 rF rI r r r rF � s z ThreadAuthenticator.is_alivec C s | � � d S r )rH rI r r r �__del__� s zThreadAuthenticator.__del__)Nr: Nr, )r- r. r/ r0 r2 r8 �strr r r r r@ rH r4 rF rJ r5 r r r r r6 U s&
���� r6 )r0 r � threadingr r �typingr r r r Z zmq.asyncio�baser r r6 �__all__r r r r �<module> s B
6