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
\