3
\
$ @ s d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z ddl
m Z ddddd gZ e j
d
kpeed oeed oeejd
ZG dd dejZejZd5dd Ze j
d
kre dddg7 Z d dlZd6ddZdd Zdd Zdd ZG dd deZnHe dddg7 Z d dlZe j
d kZ d d Z d d Z d d Zd d Zd!d Z d"d# Z G d$d% d%Z!ee"e! j#e d&d' Z$ee"e%j&e$ ee"e'j(e$ d(d) Z)d*d+ Z*eej+e) e j
d
krd,d- Z,d.d/ Z-eeje, n d0d- Z,d1d/ Z-eeje, G d2d3 d3ed4Z.dS )7 )ABCMetaabstractmethodN )context send_handle recv_handleForkingPicklerregisterdumpwin32ZCMSG_LEN
SCM_RIGHTSsendmsgc sJ e Zd ZdZi ZejZ fddZe dd Z
e d ddZ e j
Z
ZS )
r z)Pickler subclass used by multiprocessing.c s* t j| | jj | _| jj| j d S )N)super__init___copyreg_dispatch_tablecopydispatch_tableupdate_extra_reducers)selfargs) __class__ 1/usr/lib64/python3.6/multiprocessing/reduction.pyr & s zForkingPickler.__init__c C s || j |< dS )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 dumps0 s zForkingPickler.dumps)N)__name__
__module__ __qualname____doc__r copyregr r r classmethodr r" pickleloads
__classcell__r r )r r r ! s c C s t ||j| dS )z3Replacement for pickle.dump() using ForkingPickler.N)r r
)r filer! r r r r
: s DupHandle duplicate steal_handleFc C s* |dkrt j }t jt j | |d|t jS )z r? r@ r0 r5 r6 r2 r1 rA r7 r8 )r rE r r r r= u s
zDupHandle.detach)N)r# r$ r% r& r r= r r r r r- d s
DupFdsendfdsrecvfdsdarwinc C sV t j d|}tt|d g}| j|gtjtj|fg trR| jddkrRt ddS )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 rG s
c C s t j d}|j| }| jdtj|\}}}}| r>| r>tytrN| jd t|dkrjt dt| |d \}} }
|tj
kr| tj krt|
|j dkrt |j
|
t|d |d kstt|S W n t tfk
r Y nX t ddS ) z/Receive an array of fds over an AF_UNIX socket.rJ r rL z received %d items of ancdatar rK zInvalid data receivedN)rM itemsizeZrecvmsgrP Z
CMSG_SPACEEOFErrorrR r9 rO rS rQ r
ValueErrorZ frombytesAssertionErrorlist
IndexError) rT sizeaZ
bytes_sizerU ZancdataflagsZaddrZ
cmsg_levelZ cmsg_typeZ cmsg_datar r r rH s,
c
C s2 t j| j t jt j}t||g W dQ R X dS )z&Send a handle over a local connection.N)rP fromfdfilenoAF_UNIX SOCK_STREAMrG )r: r4 r; sr r r r s c
C s0 t j| j t jt j}t|dd S Q R X dS )z)Receive a handle over a local connection.r r N)rP r_ r` ra rb rH )r: rc r r r r s c C sF t j }|dk r |j|j| S tr:ddlm} |j| S tddS )zReturn a wrapper for an fd.Nr )resource_sharerz&SCM_RIGHTS appears not to be available)r Zget_spawning_popenrF Zduplicate_for_childHAVE_SEND_HANDLE rd rX )fdZ popen_objrd r r r rF s
c C s2 | j d kr t| j| jjffS t| j | jjffS d S )N)__self__getattrr __func__r# )mr r r _reduce_method s
rl c @ s e Zd Zdd ZdS )_Cc C s d S )Nr )r r r r f s z_C.fN)r# r$ r% rn r r r r rm s rm c C s t | j| jffS )N)ri __objclass__r# )rk r r r _reduce_method_descriptor s rp c C s t | j| j| jpi ffS )N)_rebuild_partialfuncr keywords)pr r r _reduce_partial s ru c C s t j| f||S )N) functoolspartial)rr r rs r r r rq s rq c C s ddl m} t|| ffS )Nr ) DupSocket)rd rx _rebuild_socket)rc rx r r r _reduce_socket s rz c C s | j S )N)r= )Zdsr r r ry s ry c C s" t | j }t|| j| j| jffS )N)rF r` ry familyr proto)rc dfr r r rz s c C s | j }tj||||dS )N)r` )r= rP )r} r{ r r| rg r r r ry s c @ sd e Zd ZdZeZeZeZeZeZe j
dkr8e Z e Z e
Z
n eZeZeZeZeZeZeZeZdd ZdS )AbstractReducerzAbstract base class for use in implementing a Reduction class
suitable for use in replacing the standard reduction mechanism
used in multiprocessing.r c G sN t tt jt t ttjt t ttj t t t
j t t t
j
t d S )N)r r rm rn rl rZ appendrp int__add__rv rw ru rP rz )r r r r r r
s
zAbstractReducer.__init__N)r# r$ r% r& r r r
r r sysplatformr/ r. r- rG rH rF rl rp rq rz ry r r r r r r~ s&
r~ ) metaclass)N)NF)/abcr r r' rv r r> r) rP r rf r __all__r hasattrre ZPicklerr r r
r0 r. r/ r r objectr- rM rR rG rH rF rl rm r rn rp rZ r r r ru rq rw rz ry r~ r r r r
sb
#