File: //usr/local/lib/python3.7/multiprocessing/__pycache__/reduction.cpython-37.opt-1.pyc
B
"��g�$ � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZddl m
Z
ddddd gZ ej d
kp�e
ed �o�e
ed �o�e
ejd
�ZG dd� dej�ZejZd5dd �Zej d
k�r e dddg7 Z d dlZd6dd�Zdd� Zdd� Zdd� ZG dd� de�ZnHe 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
k�r�d,d-� Z+d.d/� Z,eeje+� n d0d-� Z+d1d/� Z,eeje+� G d2d3� d3ed4�Z-dS )7� )�ABCMetaN� )�context� send_handle� recv_handle�ForkingPickler�register�dump�win32ZCMSG_LEN�
SCM_RIGHTS�sendmsgc sJ e Zd ZdZi ZejZ� fdd�Ze dd� �Z
e d dd��Z 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_table�copy�dispatch_table�update�_extra_reducers)�self�args)� __class__� �5/usr/local/lib/python3.7/multiprocessing/reduction.pyr & s zForkingPickler.__init__c C s || j |< dS )z&Register a reduce function for a type.N)r )�cls�type�reducer r r r + s zForkingPickler.registerNc C s t �� }| ||��|� |�� S )N)�io�BytesIOr � getbuffer)r �obj�protocol�bufr r r �dumps0 s zForkingPickler.dumps)N)�__name__�
__module__� __qualname__�__doc__r �copyregr r r � classmethodr r" �pickle�loads�
__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_handleFc C s* |dkrt �� }t �t �� | |d|t j�S )z<Duplicate a handle. (target_process is a handle not a pid!)Nr )�_winapi�GetCurrentProcess�DuplicateHandle�DUPLICATE_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 dS )z5Steal a handle from process identified by source_pid.Fr N)r0 � OpenProcess�PROCESS_DUP_HANDLEr2 r1 r3 �DUPLICATE_CLOSE_SOURCE� CloseHandle)Z
source_pidr4 Zsource_process_handler r r r/ O s
c C s t |tj|�}| �|� dS )z&Send a handle over a local connection.N)r- r0 r3 �send)�connr4 �destination_pidZdhr r r r [ s c C s | � � �� S )z)Receive a handle over a local connection.)�recv�detach)r: r r r r ` s c @ s" e Zd ZdZddd�Zdd� ZdS )r- z Picklable wrapper for a handle.Nc C s\ |d krt �� }t�tjd|�}z t�t�� |||dd�| _W d t�|� X || _ || _
d S )NFr ) �os�getpidr0 r5 r6 r2 r1 �_handler8 �_access�_pid)r r4 �access�pid�procr r r r f s zDupHandle.__init__c C sV | j t�� kr| jS t�tjd| j �}z t�|| jt�� | j dtj
�S t� |� X dS )z1Get the handle. This should only be called once.FN) 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 r- d s
�DupFd�sendfds�recvfds�darwinc C sV t � d|�}tt|�d g�}| �|gtjtj|fg� trR| �d�dkrRt d��dS )z,Send an array of fds over an AF_UNIX socket.�i� r � Az%did not receive acknowledgement of fdN)
�array�bytes�lenr �socket�
SOL_SOCKETr � ACKNOWLEDGEr< � RuntimeError)�sockZfds�msgr r r rG � s
c C s t � d�}|j| }| �dt�|��\}}}}|s:|s:t�y�trJ| �d� t|�dkrft dt|� ��|d \}} }
|tj
kr�| tj kr�t|
�|j dkr�t �|�
|
� t|�d |d kr�td�t|�|d ���t|�S W n t tfk
r� Y nX t d��d S )
z/Receive an array of fds over an AF_UNIX socket.rJ r rL z received %d items of ancdatar rK z Len is {0:n} but msg[0] is {1!r}zInvalid data receivedN)rM �itemsizeZrecvmsgrP Z
CMSG_SPACE�EOFErrorrR r9 rO rS rQ r �
ValueErrorZ frombytes�AssertionError�format�list�
IndexError) rT �size�aZ
bytes_sizerU Zancdata�flagsZaddrZ
cmsg_levelZ cmsg_typeZ cmsg_datar r r rH � s2
c C s2 t �| �� t jt j��}t||g� W dQ R X dS )z&Send a handle over a local connection.N)rP �fromfd�fileno�AF_UNIX� SOCK_STREAMrG )r: r4 r; �sr r r r � s c C s0 t �| �� t jt j��}t|d�d S Q R X dS )z)Receive a handle over a local connection.r r N)rP r` ra rb rc rH )r: rd r r r r � s c C sF t �� }|dk r |�|�| ��S tr:ddlm} |�| �S td��dS )zReturn a wrapper for an fd.Nr )�resource_sharerz&SCM_RIGHTS appears not to be available)r Zget_spawning_popenrF Zduplicate_for_child�HAVE_SEND_HANDLE� re rX )�fdZ popen_objre 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
rm c @ s e Zd Zdd� ZdS )�_Cc C s d S )Nr )r r r r �f� s z_C.fN)r# r$ r% ro r r r r rn � s rn c C s t | j| jffS )N)rj � __objclass__r# )rl r r r �_reduce_method_descriptor� s rq c C s t | j| j| jpi ffS )N)�_rebuild_partial�funcr �keywords)�pr r r �_reduce_partial� s rv c C s t j| f|�|�S )N)� functools�partial)rs r rt r r r rr � s rr c C s ddl m} t|| �ffS )Nr )� DupSocket)re ry �_rebuild_socket)rd ry r r r �_reduce_socket� s r{ c C s | � � S )N)r= )Zdsr r r rz � s rz c C s" t | �� �}t|| j| j| jffS )N)rF ra rz �familyr �proto)rd �dfr r r r{ � s c C s | � � }tj||||d�S )N)ra )r= rP )r~ r| r r} rh r r r rz � s c @ sd e Zd ZdZeZeZeZeZeZe j
dkr8e Z e Z e
Z
n eZeZeZeZeZeZeZeZdd� ZdS )�AbstractReducerz�Abstract 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� j�t� t ttj�t� t ttj �t� t t
j t � t t
j
t� d S )N)r r rn ro rm r[ �appendrq �int�__add__rw rx rv rP r{ )r r r r r r s
zAbstractReducer.__init__N)r# r$ r% r& r r r r r �sys�platformr/ r. r- rG rH rF rm rq rr r{ rz r r r r r r � s&
r )� metaclass)N)NF).�abcr r' rw r r> r) rP r� rg r �__all__r� �hasattrrf ZPicklerr r r r0 r. r/ r r �objectr- rM rR rG rH rF rm rn r ro rq r[ r� r� r� rv rr rx r{ rz r r r r r �<module>
sb
#