o
0i @ s d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z dgZ
ej dkr.send)dup_resource_sharerregisterclose_id)selfZsockr r r
r __init__ s zDupSocket.__init__c C sD t | j}| }t|W d S 1 sw Y d S r )r get_connectionr Z
recv_bytessocketZ fromshare)r r r
r r r detach$ s
$zDupSocket.detachN__name__
__module__ __qualname__r r r r r r r s DupFdc @ r )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 sw 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 @ sL e Zd Z dd 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 sh | j & | jd u r| | jd7 _||f| j| j< | j| jfW d S 1 s-w 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 Clientauthkey)
connectionr. r current_processr0 r r! getpid)Zidentr. addresskeycr r r r Q s z _ResourceSharer.get_connectionNc C s ddl m} | j[ | jd ur]|| jt jd}|d | | j
| | j r3t
d | j d | _ d | _d | _| j D ] \}\}}| qF| j W d d S W d d S 1 shw Y d S )Nr r- r/ z._ResourceSharer thread did not stop when asked)r1 r. r' r) r r2 r0 r r r* joinZis_aliver Z sub_warningr( r% itemsclear)r Ztimeoutr. r6 r5 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% r8 r9 r' _at_fork_reinitr( r r) r* )r r5 r r r r r r+ o s
z_ResourceSharer._afterforkc C sX ddl m} td |t jd| _| jj| _ t
j | j d}d|_
| || _d S )Nr )Listenerz0starting listener and thread for sending handlesr/ )targetT)r1 r; r debugr r2 r0 r( r4 r) r& ZThread_serveZdaemonstartr* )r r; tr r r r, z s
z_ResourceSharer._startc C s t tdrttjt 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)Npthread_sigmask)hasattrsignalrA SIG_BLOCK
valid_signalsr( ZacceptZrecvr% popr Z
is_exitingsys
excepthookexc_info)r r msgr5 Zdestination_pidr r r r r r> s*
z_ResourceSharer._server ) r r r r r staticmethodr r r+ r, r> r r r r r# = s
r# )r! rC r rG r& r contextr r __all__platformobjectr r r# r r r r r r s"
\