U
e5d(% @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZddl m
Z
ddddd gZ ej d
kpe
ed oe
ed oe
ejd
ZG dd dejZejZd6dd Zej d
kre dddg7 Z d dlZd7ddddZdd Zdd Zdd ZG dd deZnHe ddd g7 Z d dlZej d kZd 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( ej d
krd-d. Z+d/d0 Z,eeje+ n d1d. Z+d2d0 Z,eeje+ G d3d4 d4ed5Z-dS )8 )ABCMetaN )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| | j | _| j| j d S N)super__init___copyreg_dispatch_tablecopydispatch_tableupdate_extra_reducersselfargs __class__ 1/usr/lib64/python3.8/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 }| ||| | S r
)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 ||| dS )z3Replacement for pickle.dump() using ForkingPickler.N)r r )r" filer# r r r r : s DupHandle duplicate steal_handleF)source_processc C s6 t }|dkr|}|dkr |}t || |d|t jS )z r r r r d s c @ s" e Zd ZdZdddZdd ZdS )r/ z Picklable wrapper for a handle.Nc C s\ |d krt }ttjd|}z tt |||dd| _W 5 t| X || _ || _
d S )NFr ) osgetpidr3 r8 r9 r: r5 r4 _handle_access_pid)r r7 accesspidprocr r r r j s
zDupHandle.__init__c C sZ | j t kr| jS ttjd| j }z"t|| jt | j
dtj W S t| X dS )z1Get the handle. This should only be called once.FN) rG rC rD rE r3 r8 r9 r: r5 r4 rF r; )r rJ r r r rB y s
zDupHandle.detach)N)r% r& r' r( r rB r r r r r/ h s
DupFdsendfdsrecvfdsdarwinc C sV t d|}tt|d g}| |gtjtj|fg trR| ddkrRt 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 ACKNOWLEDGErA RuntimeError)sockZfdsmsgr r r rL s
c C s t d}|j| }| dt|\}}}}|s:|s:tztrJ| d t|dkrft dt| |d \}} }
|tj
kr| tj krt|
|j dkrt |
|
t|d |d krtdt||d t|W S W n t tfk
r Y nX t dd S )
z/Receive an array of fds over an AF_UNIX socket.rO r rQ z received %d items of ancdatar rP z Len is {0:n} but msg[0] is {1!r}zInvalid data receivedN)rR itemsizeZrecvmsgrU Z
CMSG_SPACEEOFErrorrW r= rT rX rV r
ValueErrorZ frombytesAssertionErrorformatlist
IndexError) rY sizeaZ
bytes_sizerZ ZancdataflagsZaddrZ
cmsg_levelZ cmsg_typeZ cmsg_datar r r rM s<
c C s2 t | t jt j}t||g W 5 Q R X dS r< )rU fromfdfilenoAF_UNIX SOCK_STREAMrL )r> r7 r? sr r r r s c
C s< t | t jt j }t|dd W 5 Q R S Q R X dS )r@ r r N)rU re rf rg rh rM )r> ri r r r r s c C sF t }|dk r ||| S tr:ddlm} || S tddS )zReturn a wrapper for an fd.Nr )resource_sharerz&SCM_RIGHTS appears not to be available)r Zget_spawning_popenrK Zduplicate_for_childHAVE_SEND_HANDLE rj r] )fdZ popen_objrj r r r rK s
c C s2 | j d kr t| j| jjffS t| j | jjffS d S r
)__self__getattrr __func__r% mr r r _reduce_method s
rs c @ s e Zd Zdd ZdS )_Cc C s d S r
r )r r r r f s z_C.fN)r% r& r' ru r r r r rt s rt c C s t | j| jffS r
)ro __objclass__r% rq r r r _reduce_method_descriptor s rw c C s t | j| j| jpi ffS r
)_rebuild_partialfuncr keywords)pr r r _reduce_partial s r| c C s t j| f||S r
) functoolspartial)ry r rz r r r rx s rx c C s ddl m} t|| ffS )Nr ) DupSocket)rj r _rebuild_socket)ri r r r r _reduce_socket s r c C s | S r
)rB )Zdsr r r r s r c C s" t | }t|| j| j| jffS r
)rK rf r familyr proto)ri dfr r r r s c C s | }tj||||dS )N)rf )rB rU )r r r r rm r r r r 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 r
)r r rt ru rs r` appendrw int__add__r} r~ r| rU r r r r r r s
zAbstractReducer.__init__N)r% r& r' r( r r r r r sysplatformr1 r0 r/ rL rM rK rs rw rx r r r r r r r r s&
r ) metaclass)N)NF).abcr r) r} r rC r+ rU r rl r __all__r hasattrrk ZPicklerr r r r3 r0 r1 r r objectr/ rR rW rL rM rK rs rt r ru rw r` r r r r| rx r~ r r r r r r r
sj
#