← Back
Editing: reduction.cpython-37.pyc
B � f�$ � @ 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__� �>/opt/alt/python37/lib64/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�protocolZbufr 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)r/ � OpenProcess�PROCESS_DUP_HANDLEr1 r0 r2 �DUPLICATE_CLOSE_SOURCE� CloseHandle)Z source_pidr3 Zsource_process_handler r r r. O s c C s t |tj|�}| �|� dS )z&Send a handle over a local connection.N)r, r/ r2 �send)�connr3 �destination_pidZdhr r r r [ s c C s | � � �� S )z)Receive a handle over a local connection.)�recv�detach)r9 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�getpidr/ r4 r5 r1 r0 �_handler7 �_access�_pid)r r3 �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) 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 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 rF � 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.rI r rK z received %d items of ancdatar rJ z Len is {0:n} but msg[0] is {1!r}zInvalid data receivedN)rL �itemsizeZrecvmsgrO Z CMSG_SPACE�EOFErrorrQ r8 rN rR rP r � ValueErrorZ frombytes�AssertionError�format�list� IndexError) rS �size�aZ bytes_sizerT Zancdata�flagsZaddrZ cmsg_levelZ cmsg_typeZ cmsg_datar r r rG � s2 c C s2 t �| �� t jt j��}t||g� W dQ R X dS )z&Send a handle over a local connection.N)rO �fromfd�fileno�AF_UNIX� SOCK_STREAMrF )r9 r3 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)rO r_ r` ra rb rG )r9 rc 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_popenrE Zduplicate_for_child�HAVE_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_partial�funcr �keywords)�pr r r �_reduce_partial� s ru c C s t j| f|�|�S )N)� functools�partial)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||||d�S )N)r` )r< rO )r} r{ r r| rg r r r ry � 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 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 r �sys�platformr. 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 �<module> sb #
Save File
Cancel