Current File : //usr/local/lib/python3.8/multiprocessing/__pycache__/reduction.cpython-38.pyc
U
qa(% @ 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_handleForkingPicklerregisterdumpwin32CMSG_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__ 5/usr/local/lib/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 objprotocolbufr 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<Duplicate a handle. (target_process is a handle not a pid!)Nr )_winapiGetCurrentProcessDuplicateHandleDUPLICATE_SAME_ACCESS)handleZtarget_process inheritabler4 current_processr r r r2 G s c C sF t t jd| }z$t ||t ddt jt jB W S t | X dS )z5Steal a handle from process identified by source_pid.Fr N)r5 OpenProcessPROCESS_DUP_HANDLE CloseHandler7 r6 r8 DUPLICATE_CLOSE_SOURCE)Z
source_pidr9 Zsource_process_handler r r r3 S s
c C s t |tj|}| | dS z&Send a handle over a local connection.N)r1 r5 r8 send)connr9 destination_pidZdhr r r r _ s c C s | S ))Receive a handle over a local connection.)recvdetach)rB r r r r d s c @ s" e Zd ZdZdddZdd ZdS )r1 z Picklable wrapper for a handle.Nc C s\ |d krt }ttjd|}z tt |||dd| _W 5 t| X || _ || _
d S )NFr ) osgetpidr5 r< r= r> r7 r6 _handle_access_pid)r r9 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) rK rG rH rI r5 r< r= r> r7 r6 rJ r? )r rN r r r rF y s
zDupHandle.detach)N)r' r( r) r* r rF r r r r r1 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 ACKNOWLEDGErE RuntimeError)sockfdsmsgr r r rP 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.rS r rU z received %d items of ancdatar rT z Len is {0:n} but msg[0] is {1!r}zInvalid data receivedN)rV itemsizerecvmsgrY
CMSG_SPACEEOFErrorr[ rA rX r\ rZ r
ValueError frombytesAssertionErrorformatlist
IndexError) r] sizea
bytes_sizer_ ancdataflagsaddr
cmsg_level cmsg_type cmsg_datar r r rQ s<
c C s2 t | t jt j}t||g W 5 Q R X dS r@ )rY fromfdfilenoAF_UNIX SOCK_STREAMrP )rB r9 rC sr r r r s c
C s< t | t jt j }t|dd W 5 Q R S Q R X dS )rD r r N)rY rs rt ru rv rQ )rB rw 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 get_spawning_popenrO duplicate_for_childHAVE_SEND_HANDLE rx rd )fd popen_objrx r r r rO 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
r 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) r r r r r r s r c C s t | j| jffS r )r __objclass__r' r r r r _reduce_method_descriptor s r 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)r r r r r r r s r c C s ddl m} t|| ffS )Nr ) DupSocket)rx r _rebuild_socket)rw r r r r _reduce_socket s r c C s | S r )rF )Zdsr r r r s r c C s" t | }t|| j| j| jffS r )rO rt r familyr proto)rw dfr r r r s c C s | }tj||||dS )N)rt )rF rY )r r r r r} 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 r r r rh appendr int__add__r r r rY r r r r r r s
zAbstractReducer.__init__N)r' r( r) r* r r r r r sysplatformr3 r2 r1 rP rQ rO r r r r r r r r r r r s&
r ) metaclass)N)NF).abcr r+ r r rG r- rY r r| r __all__r hasattrr{ Picklerr r r r5 r2 r3 r r objectr1 rV r[ rP rQ rO r r r r r rh r r r r r r r r r r r r r <module>
sj
#
Mini Shell