File: //lib64/python3.6/site-packages/zmq/auth/__pycache__/thread.cpython-36.pyc
3
VS�_� � @ s� d Z ddlZddlmZmZ ddlZddlmZ ddlm Z m
Z
ddl Z ddl m
Z
G dd � d e�Zd
d � ZeG d d
� d
e��Zd
gZdS )z>ZAP Authenticator in a Python Thread.
.. versionadded:: 14.1
� N)�Thread�Event)�jsonapi)�b�u� )�
Authenticatorc s: e Zd ZdZd � fdd� Zdd� Zdd � Zd
d � Z� ZS )
�AuthenticationThreadziA Thread for running a zmq Authenticator
This is run in the background by ThreadedAuthenticator
�utf-8Nc sz t t| �j� |ptjj� | _|| _|p0tj d� | _
}t � | _ |pPt
|||d�| _|jtj�| _d| j_| jj|� d S )Nzzmq.auth)�encoding�logr )�superr �__init__�zmq�Context�instance�contextr �loggingZ getLoggerr r �startedr �
authenticator�socket�PAIR�pipe�lingerZconnect)�selfr Zendpointr r r )� __class__� � /usr/lib64/python3.6/thread.pyr s z AuthenticationThread.__init__c C s | j j� | jj� | j j}tj� }|j| jtj � |j|tj � x�yt
|j � �}W n tj k
rl P Y nX | j|kr�|| j tj kr�xBy| jj
tjd�}W n tjk
r� P Y nX | j|�}|r�P q�W |r�P ||krD|| tj krD| j� qDW | jj� | j j� dS )z*Start the Authentication Agent thread task)�flagsN)r �startr �set�
zap_socketr ZPoller�registerr ZPOLLIN�dictZpollZZMQError�recv_multipartZNOBLOCKZAgain� _handle_pipe� _handle_zap�close�stop)r ZzapZpollerZsocks�msg� terminater r r �run& s4
zAuthenticationThread.runc C s$ | j jj� }|sdS | j j|� dS )z7
Handle a message from the ZAP socket.
N)r r! r$ Zhandle_zap_message)r r) r r r r& I s z AuthenticationThread._handle_zapc s� d}|dkrd}|S |d }� j jd|� |dkr�� fdd�|d d� D �}y� jj|� W n0 tk
r� } z� j jd
|� W Y dd}~X nX n�|d kr�� fd d�|d d� D �}y� jj|� W n0 tk
r� } z� j jd
|� W Y dd}~X nX n�|dk�r.t|d � j�}|d }� jj |t
j |�� nX|dk�rht|d � j�}t|d � j�}� jj ||� n |dk�rxd}n� j j
d|� |S )z6
Handle a message from front-end API.
FNTr z auth received API command %rs ALLOWc s g | ]}t |� j��qS r )r r )�.0�m)r r r �
<listcomp>_ s z5AuthenticationThread._handle_pipe.<locals>.<listcomp>r zFailed to allow %ss DENYc s g | ]}t |� j��qS r )r r )r, r- )r r r r. f s zFailed to deny %ss PLAIN� s CURVEs TERMINATEz!Invalid auth command from API: %r)r �debugr �allow� ExceptionZ exception�denyr r �configure_plainr �loads�configure_curve�error) r r) r* Zcommand� addresses�e�domainZjson_passwords�locationr )r r r% Q s< ""
z!AuthenticationThread._handle_pipe)r
NN) �__name__�
__module__� __qualname__�__doc__r r+ r& r% �
__classcell__r r )r r r s
#r c C sL xF| j j� D ]8\}}|jd�s t|� r*q tt|d�}|js |j|_q W | S )zAinherit docstrings from Authenticator, so we don't duplicate them�_N)�__dict__�items�
startswith�callable�getattrr r? )�cls�name�methodZupstream_methodr r r �_inherit_docstrings� s rJ c @ s� e Zd ZdZdZdZdZdZdZdZ dZ
d dd�Z dd� Z d d
� Z
d d � Zd
d� Zd dd�Zd dd�Zd!dd�Zdd� Zdd� Zdd� Zd d � ZdS )"�ThreadAuthenticatorz-Run ZAP authentication in a background threadN� �utf-8c C s< |p t jj� | _|| _|| _d | _djt| ��| _ d | _
d S )Nzinproc://{0}.inproc) r r r r r r r �format�id�
pipe_endpoint�thread)r r r r r r r r � s z ThreadAuthenticator.__init__c C sJ x4| g| j j� D ] }||jkrtj| ||� d S qW t| jj||� d S )N)r �mrorB �object� __setattr__�setattrrQ r )r �key�value�objr r r rT � s
z ThreadAuthenticator.__setattr__c C s4 yt j| |� W n tk
r. t| jj|�S X d S )N)rS � __getattr__�AttributeErrorrF rQ r )r rV r r r rY � s z ThreadAuthenticator.__getattr__c s$ � j jdg� fdd�|D � � d S )Ns ALLOWc s g | ]}t |� j��qS r )r r )r, �a)r r r r. � s z-ThreadAuthenticator.allow.<locals>.<listcomp>)r �send_multipart)r r8 r )r r r1 � s zThreadAuthenticator.allowc s$ � j jdg� fdd�|D � � d S )Ns DENYc s g | ]}t |� j��qS r )r r )r, r[ )r r r r. � s z,ThreadAuthenticator.deny.<locals>.<listcomp>)r r\ )r r8 r )r r r3 � s zThreadAuthenticator.deny�*c C s( | j jdt|| j�tj|p i �g� d S )Ns PLAIN)r r\ r r r �dumps)r r: Z passwordsr r r r4 � s z#ThreadAuthenticator.configure_plainc C s. t || j�}t || j�}| jjd||g� d S )Ns CURVE)r r r r\ )r r: r; r r r r6 � s z#ThreadAuthenticator.configure_curvec C s | j jj||d� d S )N)�credentials_provider)rQ r �configure_curve_callback)r r: r_ r r r r` � s z,ThreadAuthenticator.configure_curve_callbackc C s� | j jtj�| _d| j_| jj| j� t| j | j| j | j
d�| _ | j j � t
jd k rf| j jjdd� n| j jjdd�s~td��dS )
z Start the authentication threadr )r r r/ � �
)Ztimeoutz$Authenticator thread failed to startN)r/ ra )r r r r r r ZbindrP r r r rQ r �sys� version_infor �wait� RuntimeError)r r r r r � s
zThreadAuthenticator.startc C s> | j r:| j jd� | j� r$| jj� d| _| j j� d| _ dS )z Stop the authentication threads TERMINATEN)r �send�is_aliverQ �joinr' )r r r r r( � s
zThreadAuthenticator.stopc C s | j r| j j� rdS dS )z$Is the ZAP thread currently running?TF)rQ rh )r r r r rh � s z ThreadAuthenticator.is_alivec C s | j � d S )N)r( )r r r r �__del__� s zThreadAuthenticator.__del__)NrM N)r] N)r] rL )r] N)r< r= r> r? r r r r rP rQ Zauthr rT rY r1 r3 r4 r6 r` r r( rh rj r r r r rK � s(
rK )r? r Z threadingr r r Z zmq.utilsr Zzmq.utils.strtypesr r rc �baser r rJ rS rK �__all__r r r r �<module> s o
T