a
XC?h � @ 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( |� � � � fdd�}t�|� j�| _d S )Nc s � � |�}| �|� d S �N)�shareZ
send_bytes)�conn�pidr �Znew_sock� �D/opt/alt/python39/lib64/python3.9/multiprocessing/resource_sharer.py�send s
z DupSocket.__init__..send)�dup�_resource_sharer�register�close�_id)�selfZsockr r
r r �__init__ s zDupSocket.__init__c C sB t �| j��"}|�� }t�|�W d � S 1 s40 Y d S r )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 �|�� � fdd�}� fdd�}t�||�| _d S )Nc s t �| � |� d S r )r Z send_handle)r
r �Znew_fdr
r r 1 s z DupFd.__init__..sendc s t �� � d S r )�osr r
r r
r r 3 s z DupFd.__init__..close)r r r r r )r �fdr r r
r r r / s
zDupFd.__init__c C s: t �| j��}t�|�W d � S 1 s,0 Y d S r )r r r r Z 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 | _t�� | _d | _d | _d | _t� | t
j � d S )Nr ) �_key�_cache� threadingZLock�_lock� _listener�_address�_threadr Zregister_after_forkr" �
_afterfork)r r
r
r r ? s
z_ResourceSharer.__init__c C sf | j �L | jd u r| �� | jd7 _||f| j| j< | j| jfW d � S 1 sX0 Y d S )Nr )r& r( �_startr# r$ )r r r r
r
r r H s
z_ResourceSharer.registerc C s< ddl m} | \}}||t�� jd�}|�|t�� f� |S )Nr ��Client��authkey)�
connectionr- r �current_processr/ r r �getpid)Zidentr- �address�key�cr
r
r r Q s
z _ResourceSharer.get_connectionNc C s� ddl m} | j�� | jd ur�|| jt�� jd�}|�d � |�� | j �
|� | j � � rdt �
d� | j�� d | _ d | _d | _| j�� D ]\}\}}|� q�| j�� W d � n1 s�0 Y d S )Nr r, r. z._ResourceSharer thread did not stop when asked)r0 r- r& r( r r1 r/ r r r) �joinZis_aliver Z sub_warningr' r$ �items�clear)r Ztimeoutr- r5 r4 r r r
r
r r Z s$
�
z_ResourceSharer.stopc C s\ | j �� D ]\}\}}|� q
| j �� | j�� | jd urF| j�� d | _d | _d | _d S r ) r$ r7 r8 r& �_at_fork_reinitr' r r( r) )r r4 r r r
r
r r* o 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 handlesr. )�targetT)r0 r: r �debugr r1 r/ r' r3 r( r% ZThread�_serveZdaemon�startr) )r r: �tr
r
r r+ z s
z_ResourceSharer._startc C s� t td�r t�tjt�� � z�| j�� �f}|�� }|d u rJW d � W q�|\}}| j� |�\}}z|||� W |� n|� 0 W d � n1 s�0 Y W q t
� � s�t j
t �� � Y q 0 q d S )N�pthread_sigmask)�hasattr�signalr@ � SIG_BLOCK�
valid_signalsr' ZacceptZrecvr$ �popr Z
is_exiting�sys�
excepthook�exc_info)r r
�msgr4 Zdestination_pidr r r
r
r r= � s
2z_ResourceSharer._serve)N) r r r r r � staticmethodr r r* r+ r= r
r
r
r r" = s
r" )r rB r rF r% � r �contextr r �__all__�platform�objectr r r" r r r
r
r
r � s
\