B
Žp¾ õV»ã @ 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ÚstopÚwin32Ú 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 )Únew_sock© úD/opt/alt/python37/lib64/python3.7/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 s* t | j¡}| ¡ }t |¡S Q R X d S )N)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 )N)r Z send_handle)r
r )Únew_fdr
r r 1 s z DupFd.__init__..sendc s t ˆ ¡ d S )N)Ú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 |¡S Q R X d S )N)r r r r Z recv_handle)r r
r
r
r r 7 s z DupFd.detachN)r r r r r 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 | _g | _t ¡ | _d | _d | _d | _t
| t j ¡ d S )Nr )
Ú_keyÚ_cacheÚ
_old_locksÚ threadingÚLockÚ_lockÚ _listenerÚ_addressÚ_threadr Zregister_after_forkr! Ú
_afterfork)r r
r
r r ? s
z_ResourceSharer.__init__c C sN | j > | jd kr| ¡ | jd7 _||f| j| j< | j| jfS Q R X d S )Nr )r' r) Ú_startr" r# )r r r r
r
r r I 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 R s
z _ResourceSharer.get_connectionNc C sº ddl m} | jž | jd k r¬|| jt ¡ jd}| d ¡ | ¡ | j
|¡ | j ¡ rdt
d¡ | j ¡ d | _ d | _d | _x | j ¡ D ]\}\}}|ƒ qŒW | j ¡ W d Q R X d S )Nr )r- )r. z._ResourceSharer thread did not stop when asked)r/ r- r' r) r r0 r. r r r* ÚjoinZis_aliver Z sub_warningr( r# ÚitemsÚclear)r Ztimeoutr- r4 r3 r r r
r
r r [ s"
z_ResourceSharer.stopc C sn x | j ¡ D ]\}\}}|ƒ q W | j ¡ | j | j¡ t ¡ | _| jd k rX| j ¡ d | _d | _
d | _ d S )N) r# r6 r7 r$ Úappendr' r% r& r( r r) r* )r r3 r r r
r
r r+ p 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 handles)r. )ÚtargetT)r/ r9 r Údebugr r0 r. r( r2 r) r% ZThreadÚ_serveZdaemonÚstartr* )r r9 Útr
r
r r, ~ s
z_ResourceSharer._startc C sª t tdƒr t tjtdtjƒ¡ x„y\| j ¡ H}| ¡ }|d krBP |\}}| j
|¡\}}z|||ƒ W d |ƒ X W d Q R X W q" t ¡ sœt
jt
¡ Ž Y q"X q"W d S )NÚpthread_sigmaskr )ÚhasattrÚsignalr? Ú SIG_BLOCKÚrangeÚNSIGr( ZacceptZrecvr# Úpopr Z
is_exitingÚsysÚ
excepthookÚexc_info)r r
Úmsgr3 Zdestination_pidr r r
r
r r< ‰ s
z_ResourceSharer._serve)N) r r r r r Ú staticmethodr r r+ r, r< r
r
r
r r! = s
r! )r rA r rF r% Ú r Úcontextr r Ú__all__ÚplatformÚobjectr r r! r r r
r
r
r Ú s
`