o
    Hb                     @   s   d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm	Z	 dgZ
ejdkr<e
dg7 Z
G d	d deZne
d
g7 Z
G dd
 d
eZG dd deZe ZejZdS )    N   )process)	reduction)utilstopwin32	DupSocketc                   @       e Zd ZdZdd Zdd ZdS )r   zPicklable wrapper for a socket.c                    s(   |    fdd}t| j| _d S )Nc                    s     |}| | d S N)shareZ
send_bytes)connpidr   Znew_sock 6/usr/lib/python3.10/multiprocessing/resource_sharer.pysend   s   
z DupSocket.__init__.<locals>.send)dup_resource_sharerregisterclose_id)selfZsockr   r   r   r   __init__   s   zDupSocket.__init__c                 C   sB   t | j}| }t|W  d   S 1 sw   Y  dS )z1Get the socket.  This should only be called once.N)r   get_connectionr   Z
recv_bytessocketZ	fromshare)r   r   r   r   r   r   detach$   s   $zDupSocket.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 |  fdd} fdd}t||| _d S )Nc                    s   t |  | d S r
   )r   Zsend_handle)r   r   Z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   
zDupFd.__init__c                 C   s:   t | j}t|W  d   S 1 sw   Y  dS )z-Get the fd.  This should only be called once.N)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                   @   sN   e Zd ZdZdd Zdd Zedd Zdd	d
Zdd Z	dd Z
dd ZdS )_ResourceSharerz.Manager for resources using background thread.c                 C   s:   d| _ i | _t | _d | _d | _d | _t	| t
j d S )Nr   )_key_cache	threadingZLock_lock	_listener_address_threadr   Zregister_after_forkr%   
_afterfork)r   r   r   r   r   ?   s   
z_ResourceSharer.__init__c                 C   sf   | j & | jdu r|   |  jd7  _||f| j| j< | j| jfW  d   S 1 s,w   Y  dS )z+Register resource, returning an identifier.Nr   )r)   r+   _startr&   r'   )r   r   r   r   r   r   r   H   s   

$z_ResourceSharer.registerc                 C   s<   ddl m} | \}}||t jd}||t f |S )z<Return connection from which to receive identified resource.r   Clientauthkey)
connectionr0   r   current_processr2   r   r#   getpid)Zidentr0   addresskeycr   r   r   r   Q   s
   z_ResourceSharer.get_connectionNc                 C   s   ddl m} | j[ | jdur\|| jt jd}|d |  | j	
| | j	 r2td | j  d| _	d| _d| _| j D ]	\}\}}|  qE| j  W d   dS W d   dS 1 sgw   Y  dS )z:Stop the background thread and clear registered resources.r   r/   Nr1   z._ResourceSharer thread did not stop when asked)r3   r0   r)   r+   r   r4   r2   r   r   r,   joinZis_aliver   Zsub_warningr*   r'   itemsclear)r   Ztimeoutr0   r8   r7   r   r   r   r   r   r   Z   s*   




"z_ResourceSharer.stopc                 C   s\   | j  D ]	\}\}}|  q| j   | j  | jd ur#| j  d | _d | _d | _d S r
   )	r'   r:   r;   r)   _at_fork_reinitr*   r   r+   r,   )r   r7   r   r   r   r   r   r-   o   s   




z_ResourceSharer._afterforkc                 C   sj   ddl m} | jd u sJ dtd |t jd| _| jj| _	t
j| jd}d|_|  || _d S )Nr   )ListenerzAlready have Listenerz0starting listener and thread for sending handlesr1   )targetT)r3   r=   r*   r   debugr   r4   r2   r6   r+   r(   ZThread_serveZdaemonstartr,   )r   r=   tr   r   r   r.   z   s   


z_ResourceSharer._startc                 C   s   t tdrttjt  	 zD| j 5}| }|d u r(	 W d    W d S |\}}| j	|\}}z
||| W |  n|  w W d    n1 sMw   Y  W n   t
 sbtjt   Y q)Npthread_sigmask)hasattrsignalrC   	SIG_BLOCKvalid_signalsr*   ZacceptZrecvr'   popr   Z
is_exitingsys
excepthookexc_info)r   r   msgr7   Zdestination_pidr   r   r   r   r   r@      s*   

z_ResourceSharer._server
   )r   r   r   r    r   r   staticmethodr   r   r-   r.   r@   r   r   r   r   r%   =   s    		

r%   )r#   rE   r   rI   r(    r   contextr   r   __all__platformobjectr   r!   r%   r   r   r   r   r   r   <module>   s"   



\
