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 Z d d l Z d d l m Z d d d d d g Z e j
d k p e e d
o e e d o e e j d Z Gd
d d e j
Z e j Z d d d Z e j
d k rfe d d d g 7Z d d l Z d d d d Z d d Z d d Z d d Z Gd d d e Z nj e d d d g 7Z d d l Z e j
d k Z d d Z d d Z d d Z d d Z d d Z d! d" Z Gd# d$ d$ Z e e e j e d% d& Z! e e e" j# e! e e e$ j% e! d' d( Z& d) d* Z' e e j( e& e j
d k rd+ d, Z) d- d. Z* e e j e) n( d/ d, Z) d0 d. Z* e e j e) d S)1 N )context send_handle recv_handleForkingPicklerregisterdumpwin32CMSG_LEN
SCM_RIGHTSsendmsgc sg e Z d Z d Z i Z e j Z f d d Z e d d Z
e d d d Z e j
Z
S) r z)Pickler subclass used by multiprocessing.c s9 t j | | j j | _ | j j | j d S)N)super__init___copyreg_dispatch_tablecopydispatch_tableupdate_extra_reducers)selfargs) __class__ >/opt/alt/python35/lib64/python3.5/multiprocessing/reduction.pyr % s zForkingPickler.__init__c C s | | j | <d S)z&Register a reduce function for a type.N)r )clstypereducer r r r * s zForkingPickler.registerNc C s, t j } | | | j | | j S)N)ioBytesIOr getbuffer)r objprotocolZbufr r r dumps/ s zForkingPickler.dumps)__name__
__module__ __qualname____doc__r copyregr r r classmethodr r! pickleloadsr r )r r r s c C s t | | j | d S)z3Replacement for pickle.dump() using ForkingPickler.N)r r )r filer r r r r 9 s DupHandle duplicate steal_handleFc C s= | d k r t j } t j t j | | d | t j S)z<Duplicate a handle. (target_process is a handle not a pid!)Nr )_winapiGetCurrentProcessDuplicateHandleDUPLICATE_SAME_ACCESS)handleZtarget_processZ inheritabler r r r, F s
c C s] t j t j d | } z0 t j | | t j d d t j t j B SWd t j | Xd S)z5Steal a handle from process identified by source_pid.Fr N)r. OpenProcessPROCESS_DUP_HANDLEr0 r/ r1 DUPLICATE_CLOSE_SOURCE CloseHandle)Z
source_pidr2 Zsource_process_handler r r r- N s c C s& t | t j | } | j | d S)z&Send a handle over a local connection.N)r+ r. r1 send)connr2 destination_pidZdhr r r r Z s c C s | j j S)z)Receive a handle over a local connection.)recvdetach)r8 r r r r _ s c @ s1 e Z d Z d Z d d d Z d d Z d S)r+ z Picklable wrapper for a handle.Nc
C s | d k r t j } t j t j d | } z+ t j t j | | | d d | _ Wd t j | X| | _ | | _
d S)NFr ) osgetpidr. r3 r4 r0 r/ _handler6 _access_pid)r r2 accesspidprocr r r r e s zDupHandle.__init__c
C s{ | j t j k r | j St j t j d | j } z/ t j | | j t j | j d t j
SWd t j | Xd S)z1Get the handle. This should only be called once.FN) r@ r< r= r> r. r3 r4 r0 r/ r? r5 r6 )r rC r r r r; t s zDupHandle.detach)r" r# r$ r% r r; r r r r r+ c s DupFdsendfdsrecvfdsdarwinc C s{ t j d | } t t | d g } | j | g t j t j | f g t rw | j d d k rw t d d S)z,Send an array of fds over an AF_UNIX socket.i r Az%did not receive acknowledgement of fdN)
arraybyteslenr socket
SOL_SOCKETr ACKNOWLEDGEr: RuntimeError)sockZfdsmsgr r r rE s
%c C sH t j d } | j | } | j d t j | \ } } } } | rW | rW t y t rm | j d t | d k r t d t | | d \ } } }
| t j
k r | t j k r t |
| j d k r t | j
|
t | d | d k st t | SWn t t f k
r7Yn Xt d d S) z/Receive an array of fds over an AF_UNIX socket.rH r rJ z received %d items of ancdatar rI zInvalid data receivedN)rK itemsizeZrecvmsgrN r
EOFErrorrP r7 rM rQ rO r
ValueErrorZ frombytesAssertionErrorlist
IndexError) rR sizeaZ
bytes_sizerS ZancdataflagsZaddrZ
cmsg_levelZ cmsg_typeZ cmsg_datar r r rF s,
'
c
C s? t j | j t j t j } t | | g Wd QRXd S)z&Send a handle over a local connection.N)rN fromfdfilenoAF_UNIX SOCK_STREAMrE )r8 r2 r9 sr r r r s $c
C s@ t j | j t j t j } t | d d SWd QRXd S)z)Receive a handle over a local connection.r r N)rN r] r^ r_ r` rF )r8 ra r r r r s $c C sa t j } | d k r. | j | j | St rQ d d l m } | j | St d d S)zReturn a wrapper for an fd.Nr )resource_sharerz&SCM_RIGHTS appears not to be available)r Zget_spawning_popenrD Zduplicate_for_childHAVE_SEND_HANDLE rb rV )fdZ popen_objrb r r r rD s
c C sE | j d k r( t | j | j j f f St | j | j j f f Sd S)N)__self__getattrr __func__r" )mr r r _reduce_method s rj c @ s e Z d Z d d Z d S)_Cc C s d S)Nr )r r r r f s z_C.fN)r" r# r$ rl r r r r rk s rk c C s t | j | j f f S)N)rg __objclass__r" )ri r r r _reduce_method_descriptor s rn c C s" t | j | j | j p i f f S)N)_rebuild_partialfuncr keywords)pr r r _reduce_partial s rs c C s t j | | | S)N) functoolspartial)rp r rq r r r ro s ro c C s# d d l m } t | | f f S)Nr ) DupSocket)rb rv _rebuild_socket)ra rv r r r _reduce_socket s rx c C s
| j S)N)r; )Zdsr r r rw s rw c C s1 t | j } t | | j | j | j f f S)N)rD r^ rw familyr proto)ra dfr r r rx s c C s% | j } t j | | | d | S)Nr^ )r; rN )r{ ry r rz re r r r rw s )+r& rt r r< r( rN sysrd r __all__platformhasattrrc ZPicklerr r r r. r, r- r r objectr+ rK rP rE rF rD rj rk r rl rn rX appendint__add__rs ro ru rx rw r r r r <module>
s^ #