��Yf� � @ 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)�util�stop�win32� DupSocketc @ s. e Z d Z d Z d d � Z d d � Z d S)r z Picklable wrapper for a socket.c s: | j � � � f d d � } t j | � j � | _ d S)Nc s � j | � } | j | � d S)N)�shareZ
send_bytes)�conn�pidr )�new_sock� �D/opt/alt/python35/lib64/python3.5/multiprocessing/resource_sharer.py�send s z DupSocket.__init__.<locals>.send)�dup�_resource_sharer�register�close�_id)�selfZsockr r
)r r �__init__ s zDupSocket.__init__c C s9 t j | j � � } | j � } t j | � SWd QRXd 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 zDupSocket.detachN)�__name__�
__module__� __qualname__�__doc__r r r
r
r
r r s �DupFdc @ s. e Z d Z d Z d d � Z d d � Z d S)r z-Wrapper for fd which can be used at any time.c sL t j | � � � f d d � } � f d d � } t j | | � | _ d S)Nc s t j | � | � d S)N)r Z send_handle)r
r )�new_fdr
r r 1 s z DupFd.__init__.<locals>.sendc s t j � � d S)N)�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 zDupFd.__init__c C s- t j | j � � } t j | � SWd QRXd 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.detachN)r r r r r r r
r
r
r r - s c @ ss 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 resouces using background thread.c C s\ d | _ i | _ g | _ t j � | _ d | _ d | _ d | _ t j
| 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 sc | j �S | j d k r# | j � | j d 7_ | | f | j | j <| j | j f SWd QRXd S)z+Register resource, returning an identifier.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)z<Return connection from which to receive identified resource.r )�Client�authkey)�
connectionr. r �current_processr/ r r �getpid)Zidentr. �address�key�cr
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)z:Stop the background thread and clear registered resources.r )r. Nr/ 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 [ 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$ r7 r8 r% �appendr( r& r' r) r r* r+ )r r4 r r r
r
r r, p s "
z_ResourceSharer._afterforkc C s� d d l m } | j d k s% t � t j d � | d t j � j � | _ | j j | _
t j d | j
� } d | _ | j � | | _ d S)Nr )�Listenerz0starting listener and thread for sending handlesr/ �targetT)r0 r: r) �AssertionErrorr �debugr r1 r/ r3 r* r& ZThread�_serveZdaemon�startr+ )r r: �tr
r
r r- ~ 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)N�pthread_sigmaskr )�hasattr�signalrA � SIG_BLOCK�range�NSIGr) ZacceptZrecvr$ �popr Z
is_exiting�sys�
excepthook�exc_info)r r
�msgr4 Zdestination_pidr r r
r
r r> � s z_ResourceSharer._serve) r r r r r r � staticmethodr r r, r- r> r
r
r
r r" = s
r" )r rC r rH r& � r r r �__all__�platform�objectr r r" r r r
r
r
r �<module> s
`