] @ 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 7Z Gd d d e Z n# e d
g 7Z Gd d
d
e Z
Gd d
d
e Z e Z e j Z d S) N )process) reduction)utilstopwin32 DupSocketc @ s( e Z d Z d d Z d d Z d S)r c s: | j f d d } t j | j | _ d S)Nc s j | } | j | d S)N)share
send_bytes)connpidr )new_sock 4/opt/alt/python35/lib64/python3.5/resource_sharer.pysend s z DupSocket.__init__..send)dup_resource_sharerregisterclose_id)selfsockr r )r
r __init__ s zDupSocket.__init__c C s9 t j | j } | j } t j | SWd QRXd S)N)r get_connectionr
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 @ s( e Z d Z d d Z d d Z d S)r c sL t j | f d d } f d d } t j | | | _ d S)Nc s t j | | d S)N)r send_handle)r r )new_fdr r r 1 s z DupFd.__init__..sendc s t j 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 | j } t j | SWd QRXd S)N)r r r r 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 @ sm e 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)_ResourceSharerc C s\ d | _ i | _ g | _ t j | _ d | _ d | _ d | _ t j
| t j d S)Nr )
_key_cache
_old_locks threadingLock_lock _listener_address_threadr register_after_forkr&
_afterfork)r r r r r ? s z_ResourceSharer.__init__c
C sc | j S | j d k r# | j | j d 7_ | | f | j | j <| j | j f SWd QRXd S)Nr )r, r. _startr' r( )r r r r r r r I s
z_ResourceSharer.registerc C sT d d l m } | \ } } | | d t j j } | j | t j f | S)Nr )Clientauthkey)
connectionr3 r current_processr4 r r# getpid)identr3 addresskeycr r r r R s
z _ResourceSharer.get_connectionNc C s d d l m } | j | j d k r | | j d t j j } | j d | j | j j
| | j j r t j
d | j j d | _ d | _ d | _ x* | j j D] \ } \ } } | q W| j j Wd QRXd S)Nr )r3 r4 z._ResourceSharer thread did not stop when asked)r5 r3 r, r. r r6 r4 r r r/ joinis_aliver sub_warningr- r( itemsclear)r timeoutr3 r; r: r r r r r r [ s"
" z_ResourceSharer.stopc C s x* | j j D] \ } \ } } | q W| j j | j j | j t j | _ | j d k rx | j j d | _ d | _
d | _ d S)N) r( r? r@ r) appendr, r* r+ r- r r. r/ )r r: r r r r r r1 p s "
z_ResourceSharer._afterforkc C s| d d l m } t j d | d t j j | _ | j j | _ t
j d | j } d | _
| j | | _ d S)Nr )Listenerz0starting listener and thread for sending handlesr4 targetT)r5 rC r debugr r6 r4 r- r9 r. r* Thread_servedaemonstartr/ )r rC tr r r r2 ~ s
z_ResourceSharer._startc C s t t d r. t j t j t d t j x yv | j j ` } | j } | d k r_ P| \ } } | j j
| \ } } z | | | Wd | XWd QRXWq1 t j s t
j t
j Yq1 Xq1 Wd S)Npthread_sigmaskr )hasattrsignalrK SIG_BLOCKrangeNSIGr- acceptrecvr( popr
is_exitingsys
excepthookexc_info)r r msgr: destination_pidr r r r r rG s z_ResourceSharer._serve) r r r r r staticmethodr r r1 r2 rG r r r r r& = s
r& )r# rM r rU r* r r r __all__platformobjectr r r& r r r r r r s
`