B
NUny @ 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 dejZejZd5dd Zej d
kr e dddg7 Z d dlZd6ddZdd Zdd Zdd ZG dd deZnHe dddg7 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.d/ Z,eeje+ n d0d- Z+d1d/ Z,eeje+ G d2d3 d3ed4Z-dS )7 )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_reducers)selfargs) __class__ >/opt/alt/python37/lib64/python3.7/multiprocessing/reduction.pyr & s zForkingPickler.__init__c C s || j |< d S )N)r )clstypereducer r r r + s zForkingPickler.registerNc C s t }| ||| | S )N)ioBytesIOr getbuffer)r objprotocolbufr 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 )N)r r )r filer r r r r : s DupHandle duplicate steal_handleFc C s* |d krt }t t | |d|t jS )Nr )_winapiGetCurrentProcessDuplicateHandleDUPLICATE_SAME_ACCESS)handleZtarget_processZ inheritabler r r r- G s
c C sB t t jd| }z t ||t ddt jt jB S t | X d S )NFr )r/ OpenProcessPROCESS_DUP_HANDLEr1 r0 r2 DUPLICATE_CLOSE_SOURCE CloseHandle)Z
source_pidr3 Zsource_process_handler r r r. O s
c C s t |tj|}| | d S )N)r, r/ r2 send)connr3 destination_pidZdhr r r r [ s c C s | S )N)recvdetach)r9 r r r r ` s c @ s e Zd ZdddZdd ZdS )r, Nc C s\ |d krt }ttjd|}z tt |||dd| _W d t| X || _ || _
d S )NFr ) osgetpidr/ r4 r5 r1 r0 _handler7 _access_pid)r r3 accesspidprocr r r r f s zDupHandle.__init__c C sV | j t kr| jS ttjd| j }z t|| jt | j dtj
S t | X d S )NF) rA r= r> r? r/ r4 r5 r1 r0 r@ r6 r7 )r rD r r r r< u s
zDupHandle.detach)N)r# r$ r% r r< r r r r r, d 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 rF s
c C s t d}|j| }| dt|\}}}}|s:|s:tytrJ| d t|dkrft dt| |d \}} }
|tj
kr| tj krt|
|j dkrt |
|
t|d |d krtdt||d t|S W n t tfk
r Y nX t dd S ) NrI r rK z received %d items of ancdatar rJ z Len is {0:n} but msg[0] is {1!r}zInvalid data received)rL itemsizeZrecvmsgrO Z
CMSG_SPACEEOFErrorrQ r8 rN rR rP r
ValueErrorZ frombytesAssertionErrorformatlist
IndexError) rS sizeaZ
bytes_sizerT ZancdataflagsZaddrZ
cmsg_levelZ cmsg_typeZ cmsg_datar r r rG s2
c C s2 t | t jt j}t||g W d Q R X d S )N)rO fromfdfilenoAF_UNIX SOCK_STREAMrF )r9 r3 r: sr r r r s c C s0 t | t jt j}t|dd S Q R X d S )Nr r )rO r_ r` ra rb rG )r9 rc 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_popenrE Zduplicate_for_childHAVE_SEND_HANDLE rd rW )fdZ popen_objrd r r r rE 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 | S )N)r< )Zdsr r r ry s ry c C s" t | }t|| j| j| jffS )N)rE r` ry familyr proto)rc dfr r r rz s c C s | }tj||||dS )N)r` )r< rO )r} r{ r r| rg r r r ry 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 )N)r r rm rn rl rZ appendrp int__add__rv rw ru rO rz )r r r r r r s
zAbstractReducer.__init__N)r# r$ r% r r r r r sysplatformr. r- r, rF rG rE rl rp rq rz ry r r r r r r~ s$
r~ ) metaclass)N)NF).abcr r& rv r r= r( rO r rf r __all__r hasattrre ZPicklerr r r r/ r- r. r r objectr, rL rQ rF rG rE rl rm r rn rp rZ r r r ru rq rw rz ry r~ r r r r
sb
#