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 sF e Zd Zi ZejZ fddZedd Z edddZ
e j Z Z
S ) r 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 |< d S r
)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__r copyregr r r classmethodr r$ pickleloads
__classcell__r r r r r ! s
c C s t ||| d S r
)r r )r" filer# r r r r : s DupHandle duplicate steal_handleF)source_processc C s6 t }|d kr|}|d kr |}t || |d|t jS )Nr )_winapiGetCurrentProcessDuplicateHandleDUPLICATE_SAME_ACCESS)handleZtarget_processZ inheritabler1 Zcurrent_processr r r r/ G s c C sF t t jd| }z$t ||t ddt jt jB W S t | X d S NFr )r2 OpenProcessPROCESS_DUP_HANDLE CloseHandler4 r3 r5 DUPLICATE_CLOSE_SOURCE)Z
source_pidr6 Zsource_process_handler r r r0 S s
c C s t |tj|}| | d S r
)r. r2 r5 send)connr6 destination_pidZdhr r r r _ s c C s | S r
)recvdetach)r= r r r r d s c @ s e Zd ZdddZdd ZdS )r. Nc C s\ |d krt }ttjd|}z tt |||dd| _W 5 t| X || _ || _
d S r7 ) osgetpidr2 r8 r9 r: r4 r3 _handle_access_pid)r r6 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 d S )NF) rE rA rB rC r2 r8 r9 r: r4 r3 rD r; )r rH r r r r@ y s
zDupHandle.detach)N)r% r& r' r r@ r r r r r. h s
DupFdsendfdsrecvfdsdarwinc C sV t d|}tt|d g}| |gtjtj|fg trR| ddkrRt dd S )Ni r Az%did not receive acknowledgement of fd)
arraybyteslenr socket
SOL_SOCKETr ACKNOWLEDGEr? RuntimeError)sockZfdsmsgr r r rJ 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 ) NrM r rO z received %d items of ancdatar rN z Len is {0:n} but msg[0] is {1!r}zInvalid data received)rP itemsizeZrecvmsgrS Z
CMSG_SPACEEOFErrorrU r< rR rV rT r
ValueErrorZ frombytesAssertionErrorformatlist
IndexError) rW sizeaZ
bytes_sizerX ZancdataflagsZaddrZ
cmsg_levelZ cmsg_typeZ cmsg_datar r r rK s<
c C s2 t | t jt j}t||g W 5 Q R X d S r
)rS fromfdfilenoAF_UNIX SOCK_STREAMrJ )r= r6 r> sr r r r s c
C s< t | t jt j }t|dd W 5 Q R S Q R X d S )Nr r )rS rc rd re rf rK )r= rg r r r r s c C sF t }|d k r ||| S tr:ddlm} || S tdd S )Nr )resource_sharerz&SCM_RIGHTS appears not to be available)r Zget_spawning_popenrI Zduplicate_for_childHAVE_SEND_HANDLE rh r[ )fdZ popen_objrh r r r rI 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
rq 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' rs r r r r rr s rr c C s t | j| jffS r
)rm __objclass__r% ro r r r _reduce_method_descriptor s ru c C s t | j| j| jpi ffS r
)_rebuild_partialfuncr keywords)pr r r _reduce_partial s rz c C s t j| f||S r
) functoolspartial)rw r rx r r r rv s rv c C s ddl m} t|| ffS )Nr ) DupSocket)rh r} _rebuild_socket)rg r} r r r _reduce_socket s r c C s | S r
)r@ )Zdsr r r r~ s r~ c C s" t | }t|| j| j| jffS r
)rI rd r~ familyr proto)rg dfr r r r s c C s | }tj||||dS )N)rd )r@ rS )r r r r rk r r r r~ s c @ s` e Zd ZeZeZeZeZeZej dkr4e
Z
e Z e Z n e
Z
eZeZeZeZeZeZeZdd ZdS )AbstractReducerr
c G sN t tt jt t ttjt t ttj t t t
j t t t
j
t d S r
)r r rr rs rq r^ appendru int__add__r{ r| rz rS r r r r r r s
zAbstractReducer.__init__N)r% r& r' r r r r r sysplatformr0 r/ r. rJ rK rI rq ru rv r r~ r r r r r r s$
r ) metaclass)N)NF).abcr r( r{ r rA r* rS r rj r __all__r hasattrri ZPicklerr r r r2 r/ r0 r r objectr. rP rU rJ rK rI rq rr r rs ru r^ r r r rz rv r| r r~ r r r r r
sj
#