File: //usr/local/lib/python3.8/multiprocessing/__pycache__/resource_sharer.cpython-38.opt-2.pyc
U
q��a� � @ 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�stop�win32� DupSocketc @ s e Zd Zdd� Zdd� ZdS )r c s( |� � � � fdd�}t�|� j�| _d S )Nc s � � |�}| �|� d S �N)�share�
send_bytes)�conn�pidr
�Znew_sock� �;/usr/local/lib/python3.8/multiprocessing/resource_sharer.py�send s
z DupSocket.__init__.<locals>.send)�dup�_resource_sharer�register�close�_id)�self�sockr r r r �__init__ s zDupSocket.__init__c
C s6 t �| j�� }|�� }t�|�W 5 Q R � S Q R X d S r )r �get_connectionr �
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 �|�� � fdd�}� fdd�}t�||�| _d S )Nc s t �| � |� d S r )r � send_handle)r r
�Znew_fdr r r 1 s z DupFd.__init__.<locals>.sendc s t �� � d S r )�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��}t�|�W 5 Q R � S Q R X d S r )r r r r � recv_handle)r r r r r r 7 s z DupFd.detachNr 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 | _t�� | _d | _d | _d | _t �
| t j � d S )Nr )
�_key�_cache�
_old_locks� threading�Lock�_lock� _listener�_address�_threadr �register_after_forkr( �
_afterfork)r r r r r ? s
z_ResourceSharer.__init__c
C sZ | j �J | jd kr| �� | jd7 _||f| j| j< | j| jfW 5 Q R � S Q R X d S )Nr )r. r0 �_startr) r* )r r r r r r r I s
z_ResourceSharer.registerc C s< ddl m} | \}}||t�� jd�}|�|t�� f� |S )Nr ��Client��authkey)�
connectionr6 r �current_processr8 r r% �getpid)�identr6 �address�key�cr r r r R s
z _ResourceSharer.get_connectionNc C s� ddl m} | j�� | jd k r�|| jt�� jd�}|�d � |�� | j �
|� | j � � rdt �
d� | j�� d | _ d | _d | _| j�� D ]\}\}}|� q�| j�� W 5 Q R X d S )Nr r5 r7 z._ResourceSharer thread did not stop when asked)r9 r6 r. r0 r r: r8 r r r1 �join�is_aliver � sub_warningr/ r* �items�clear)r �timeoutr6 r? r> r r r r r r [ s$
�
z_ResourceSharer.stopc C sj | j �� D ]\}\}}|� q
| j �� | j�| j� t�� | _| jd k rT| j� � d | _d | _
d | _ d S r ) r* rC rD r+ �appendr. r, r- r/ r r0 r1 )r r> r r r r r r3 p s
z_ResourceSharer._afterforkc C sX ddl m} t�d� |t�� jd�| _| jj| _ t
j | j d�}d|_
|�� || _d S )Nr )�Listenerz0starting listener and thread for sending handlesr7 )�targetT)r9 rG r �debugr r: r8 r/ r= r0 r, �Thread�_serve�daemon�startr1 )r rG �tr r r r4 ~ s
z_ResourceSharer._startc C s� t td�r t�tjt�� � zh| j�� �T}|�� }|d krHW 5 Q R � W q�|\}}| j� |�\}}z|||� W 5 |� X W 5 Q R X W q t
� � s�t j
t �� � Y q X q d S )N�pthread_sigmask)�hasattr�signalrO � SIG_BLOCK�
valid_signalsr/ �accept�recvr* �popr �
is_exiting�sys�
excepthook�exc_info)r r �msgr> �destination_pidr r r r r rK � s
z_ResourceSharer._serve)N) r r r! r r � staticmethodr r r3 r4 rK r r r r r( = s
r( )r% rQ r rX r, � r �contextr r �__all__�platform�objectr r" r( r r r r r r �<module> s
`