3
2a
$ @ 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 sF e Zd Zi ZejZ fddZedd Z edddZ
e j Z Z
S ) r c s* t j| | jj | _| jj| j d S )N)super__init___copyreg_dispatch_tablecopydispatch_tableupdate_extra_reducers)selfargs) __class__ ./opt/alt/python36/lib64/python3.6/reduction.pyr & s zForkingPickler.__init__c C s || j |< d S )N)r )clstypereducer r r r + s zForkingPickler.registerNc C s t j }| ||j| |j 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 ||j| d S )N)r r
)r filer! r r r r
: s DupHandle duplicate steal_handleFc C s* |d krt j }t jt j | |d|t jS )Nr )_winapiGetCurrentProcessDuplicateHandleDUPLICATE_SAME_ACCESS)handleZtarget_processZ inheritabler r r r. G s
c C sB t jt jd| }z t j||t j ddt jt jB S t j| X d S )NFr )r0 OpenProcessPROCESS_DUP_HANDLEr2 r1 r3 DUPLICATE_CLOSE_SOURCE CloseHandle)Z
source_pidr4 Zsource_process_handler r r r/ O s
c C s t |tj|}| j| d S )N)r- r0 r3 send)connr4 destination_pidZdhr r r r [ s c C s | j j S )N)recvdetach)r: r r r r ` s c @ s e Zd ZdddZdd ZdS )r- Nc
C s\ |d krt j }tjtjd|}z tjtj |||dd| _W d tj| X || _ || _
d S )NFr ) osgetpidr0 r5 r6 r2 r1 _handler8 _access_pid)r r4 accesspidprocr r r r f s zDupHandle.__init__c
C sV | j tj kr| jS tjtjd| j }z tj|| jtj | j dtj
S tj | X d S )NF) rB r> 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- d s
DupFdsendfdsrecvfdsdarwinc C sV t j d|}tt|d g}| j|gtjtj|fg trR| jddkrRt dd S )Ni r Az%did not receive acknowledgement of fd)
arraybyteslenr
socket
SOL_SOCKETr ACKNOWLEDGEr< RuntimeError)sockZfdsmsgr r r rG s
c C s t j d}|j| }| jdtj|\}}}}| r>| r>tyxtrN| jd t|dkrjt dt| |d \}} }
|tj
kr| tj krt|
|j dkrt |j
|
t|S W n t tfk
r Y nX t dd S )NrJ r rL z received %d items of ancdatar zInvalid data received)rM itemsizeZrecvmsgrP Z
CMSG_SPACEEOFErrorrR r9 rO rS rQ r
ValueErrorZ frombyteslist
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 d Q R X d S )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 d S )Nr r )rP r^ r_ r` ra rH )r: rb r r r r s c C sF t j }|d k r |j|j| S tr:ddlm} |j| S tdd S )Nr )resource_sharerz&SCM_RIGHTS appears not to be available)r Zget_spawning_popenrF Zduplicate_for_childHAVE_SEND_HANDLE rc rX )fdZ popen_objrc 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
rk c @ s e Zd Zdd ZdS )_Cc C s d S )Nr )r r r r f s z_C.fN)r$ r% r& rm r r r r rl s rl c C s t | j| jffS )N)rh __objclass__r$ )rj r r r _reduce_method_descriptor s ro c C s t | j| j| jpi ffS )N)_rebuild_partialfuncr keywords)pr r r _reduce_partial s rt c C s t j| f||S )N) functoolspartial)rq r rr r r r rp s rp c C s ddl m} t|| ffS )Nr ) DupSocket)rc rw _rebuild_socket)rb rw r r r _reduce_socket s ry c C s | j S )N)r= )Zdsr r r rx s rx c C s" t | j }t|| j| j| jffS )N)rF r_ rx familyr proto)rb dfr r r ry s c C s | j }tj||||dS )N)r_ )r= rP )r| rz r r{ rf r r r rx 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 rl rm rk rY appendro int__add__ru rv rt rP ry )r r r r r r
s
zAbstractReducer.__init__N)r$ r% r& r r r
r r sysplatformr/ r. r- rG rH rF rk ro rp ry rx r r r r r r} s$
r} ) metaclass)N)NF)/abcr r r' ru r r> r) rP r re r __all__r hasattrrd ZPicklerr r r
r0 r. r/ r r objectr- rM rR rG rH rF rk rl r rm ro rY r~ r r rt rp rv ry rx r} r r r r
sb
#