CasperSecurity
o
`�wi � @ s� d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z dgZ
ej dkr<e
dg7 Z
G d d� de �Z
n
e
d
g7 Z
G d d
� d
e �ZG d d
� d
e �Ze� ZejZdS )� N� )�process)� reduction)�util�stop�win32� DupSocketc @ � e Zd ZdZdd� Zdd� ZdS )r z Picklable wrapper for a socket.c s( |� � � � fdd�}t�|� j�| _d S )Nc s � � |�}| �|� d S �N)�share�
send_bytes)�conn�pidr ��new_sock� �6/usr/lib/python3.10/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 sB t �| j��}|�� }t�|�W d � S 1 sw Y dS )z1Get the socket. This should only be called once.N)r �get_connectionr �
recv_bytes�socket� fromshare)r r
r r r r �detach$ s $�zDupSocket.detachN��__name__�
__module__� __qualname__�__doc__r r r r r r r s �DupFdc @ r )r& z-Wrapper for fd which can be used at any time.c s4 t �|�� � fdd�}� fdd�}t�||�| _d S )Nc s t �| � |� d S r
)r � send_handle)r
r ��new_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 d � S 1 sw Y dS )z-Get the fd. This should only be called once.N)r r r r � recv_handle)r r
r r r r 7 s $�z DupFd.detachNr! r r r r r& - s c @ sN e Zd ZdZdd� Zdd� Zedd� �Zdd d
�Zd d � Z d
d� Z
dd� Z dS )�_ResourceSharerz.Manager for resources using background thread.c C s: d| _ i | _t�� | _d | _d | _d | _t� | t
j � d S )Nr ) �_key�_cache� threading�Lock�_lock� _listener�_address�_threadr �register_after_forkr- �
_afterfork)r r r r r ? s
z_ResourceSharer.__init__c C sf | j �& | jdu r
| �� | jd7 _||f| j| j< | j| jfW d � S 1 s,w Y dS )z+Register resource, returning an identifier.Nr )r2 r4 �_startr. r/ )r r r r r r r H s
$�z_ResourceSharer.registerc C s< ddl m} | \}}||t�� jd�}|�|t�� f� |S )z<Return connection from which to receive identified resource.r ��Client��authkey)�
connectionr: r �current_processr< r r* �getpid)�identr: �address�key�cr r r r Q s
z _ResourceSharer.get_connectionNc C s� ddl m} | j�[ | jdur\|| jt�� jd�}|�d� |�� | j �
|� | j � � r2t �
d� | j�� d| _ d| _d| _| j�� D ] \}\}}|� qE| j�� W d � dS W d � dS 1 sgw Y dS )z:Stop the background thread and clear registered resources.r r9 Nr; z._ResourceSharer thread did not stop when asked)r= r: r2 r4 r r> r< r r r5 �join�is_aliver � sub_warningr3 r/ �items�clear)r �timeoutr: rC rB r r r r r r Z s*
�
�"�z_ResourceSharer.stopc C s\ | j �� D ] \}\}}|� q| j �� | j�� | jd ur#| j�� d | _d | _d | _d S r
) r/ rG rH r2 �_at_fork_reinitr3 r r4 r5 )r rB r r r r r r7 o s
z_ResourceSharer._afterforkc C sj ddl m} | jd u sJ d��t�d� |t�� jd�| _| jj| _ t
j | j d�}d|_
|�� || _d S )Nr )�ListenerzAlready have Listenerz0starting listener and thread for sending handlesr; )�targetT)r= rK r3 r �debugr r> r< rA r4 r0 �Thread�_serve�daemon�startr5 )r rK �tr r r r8 z s
z_ResourceSharer._startc C s� t td�rt�tjt�� � zD| j�� �5}|�� }|d u r( W d � W d S |\}}| j� |�\}}z
|||� W |� n|� w W d � n1 sMw Y W n t
� � sbt j
t �� � Y q)N�pthread_sigmask)�hasattr�signalrS � SIG_BLOCK�
valid_signalsr3 �accept�recvr/ �popr �
is_exiting�sys�
excepthook�exc_info)r r
�msgrB �destination_pidr r r r r rO � s*
� ��
��z_ResourceSharer._server
) r" r# r$ r% r r � staticmethodr r r7 r8 rO r r r r r- = s
r- )r* rU r r\ r0 � r �contextr r �__all__�platform�objectr r& r- r r r r r r �<module> s"
\