Current File : //lib64/python3.6/multiprocessing/__pycache__/resource_sharer.cpython-36.opt-2.pyc
3
\� � @ s� d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z dgZ
ej dkrxe
dg7 Z
G d d� de �Z
ne
d
g7 Z
G d d
� d
e �ZG d d
� d
e �Ze� ZejZdS )� N� )�process)� reduction)�util�stopZwin32� DupSocketc @ s e Zd Zdd� Zdd� ZdS )r c s( |j � � � fdd�}tj|� j�| _d S )Nc s � j |�}| j|� d S )N)�shareZ
send_bytes)�conn�pidr )�new_sock� �7/usr/lib64/python3.6/multiprocessing/resource_sharer.py�send s
z DupSocket.__init__.<locals>.send)�dup�_resource_sharer�register�close�_id)�selfZsockr r )r r
�__init__ s zDupSocket.__init__c C s* t j| j��}|j� }tj|�S Q R X d S )N)r �get_connectionr Z
recv_bytes�socketZ fromshare)r r r r r r
�detach$ s zDupSocket.detachN)�__name__�
__module__� __qualname__r r r r r r
r s �DupFdc @ s e Zd Zdd� Zdd� ZdS )r c s4 t j|�� � fdd�}� fdd�}tj||�| _d S )Nc s t j| � |� d S )N)r Z send_handle)r r
)�new_fdr r
r 1 s z DupFd.__init__.<locals>.sendc s t j� � d S )N)�osr r )r r r
r 3 s z DupFd.__init__.<locals>.close)r r r r r )r �fdr r r )r r
r / s
zDupFd.__init__c C s" t j| j�� }tj|�S Q R X d S )N)r r r r Z recv_handle)r r r r r
r 7 s z DupFd.detachN)r r r r r r r r r
r - s c @ sJ e Zd Zdd� Zdd� Zedd� �Zddd �Zd
d � Zd d
� Z dd� Z
dS )�_ResourceSharerc C s@ d| _ i | _g | _tj� | _d | _d | _d | _t j
| t j � d S )Nr )
�_key�_cache�
_old_locks� threading�Lock�_lock� _listener�_address�_threadr Zregister_after_forkr �
_afterfork)r r r r
r ? s
z_ResourceSharer.__init__c
C sN | j �> | jd kr| j� | jd7 _||f| j| j< | j| jfS Q R X d S )Nr )r& r( �_startr! r" )r r r r r r
r I s
z_ResourceSharer.registerc C s< ddl m} | \}}||tj� jd�}|j|tj� f� |S )Nr )�Client)�authkey)�
connectionr, r �current_processr- r r �getpid)Zidentr, �address�key�cr r r
r R s
z _ResourceSharer.get_connectionNc C s� ddl m} | j�� | jd k r�|| jtj� jd�}|jd � |j� | j j
|� | j j � rdt j
d� | jj� d | _ d | _d | _x | jj� D ]\}\}}|� q�W | jj� W d Q R X d S )Nr )r, )r- z._ResourceSharer thread did not stop when asked)r. r, r& r( r r/ r- r r r) �joinZis_aliver Z sub_warningr' r" �items�clear)r Ztimeoutr, r3 r2 r r r r r
r [ s"
z_ResourceSharer.stopc C sn x | j j� D ]\}\}}|� q W | j j� | jj| j� tj� | _| jd k rX| jj � d | _d | _
d | _ d S )N) r" r5 r6 r# �appendr& r$ r% r' r r( r) )r r2 r r r r r
r* p s
z_ResourceSharer._afterforkc C sX ddl m} tjd� |tj� jd�| _| jj| _ t
j | j d�}d|_
|j� || _d S )Nr )�Listenerz0starting listener and thread for sending handles)r- )�targetT)r. r8 r �debugr r/ r- r' r1 r( r$ ZThread�_serveZdaemon�startr) )r r8 �tr r r
r+ ~ s
z_ResourceSharer._startc C s� t td�r tjtjtdtj�� x�y\| jj� �H}|j� }|d krBP |\}}| j j
|�\}}z|||� W d |� X W d Q R X W q" t j � s�t
jt
j� � Y q"X q"W d S )N�pthread_sigmaskr )�hasattr�signalr> � SIG_BLOCK�range�NSIGr' ZacceptZrecvr" �popr Z
is_exiting�sys�
excepthook�exc_info)r r �msgr2 Zdestination_pidr r r r r
r; � s
z_ResourceSharer._serve)N) r r r r r � staticmethodr r r* r+ r; r r r r
r = s
r )r r@ r rE r$ � r �contextr r �__all__�platform�objectr r r r r r r r r
�<module> s
`
Mini Shell