a
XC?h(% � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZddl m
Z
g d�Z ej dkp�e
ed�o�e
ed�o�e
ejd �ZG d
d � d ej�ZejZd8d d
�Zej dk� r�e g d�7 Z d dlZd9dd�dd�Zdd� Zdd� Zdd� ZG dd� de�ZnFe g d�7 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 dk�r�d/d0� Z+d1d2� Z,eeje+� n d3d0� Z+d4d2� Z,eeje+� G d5d6� d6ed7�Z-dS ):� )�ABCMetaN� )�context)� send_handle� recv_handle�ForkingPickler�register�dump�win32ZCMSG_LEN�
SCM_RIGHTS�sendmsgc sF e Zd Zi ZejZ� fdd�Zedd� �Z eddd��Z
e j Z � Z
S ) r 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/python39/lib64/python3.9/multiprocessing/reduction.pyr & s zForkingPickler.__init__c C s || j |< d S r
)r )�cls�type�reducer r r r + s zForkingPickler.registerNc C s t �� }| ||��|� |�� S r
)�io�BytesIOr � getbuffer)r �obj�protocolZbufr r r �dumps0 s zForkingPickler.dumps)N)�__name__�
__module__� __qualname__r �copyregr r r � classmethodr r$ �pickle�loads�
__classcell__r r r r r ! s
r c C s t ||��| � d S r
)r r )r" �filer# r r r r : s r )� DupHandle� duplicate� steal_handleF)�source_processc C s6 t �� }|d u r|}|d u r |}t �|| |d|t j�S )Nr )�_winapi�GetCurrentProcess�DuplicateHandle�DUPLICATE_SAME_ACCESS)�handleZtarget_processZ inheritabler1 Zcurrent_processr r r r/ G s �r/ c C sN t �t jd| �}z,t �||t �� ddt jt jB �W t �|� S t �|� 0 d S �NFr )r2 � OpenProcess�PROCESS_DUP_HANDLEr4 r3 r5 �DUPLICATE_CLOSE_SOURCE� CloseHandle)Z
source_pidr6 Zsource_process_handler r r r0 S s �
�
�r0 c C s t |tj|�}| �|� d S r
)r. r2 r5 �send)�connr6 �destination_pidZdhr r r r _ s r c C s | � � �� S r
)�recv�detach)r= r r r r d s r c @ s e Zd Zddd�Zdd� ZdS )r. Nc C sf |d u rt �� }t�tjd|�}z(t�t�� |||dd�| _W t�|� n t�|� 0 || _ || _
d S r7 ) �os�getpidr2 r8 r9 r4 r3 �_handler; �_access�_pid)r r6 �access�pid�procr r r r j s
�zDupHandle.__init__c C sb | j t�� kr| jS t�tjd| j �}z*t�|| jt�� | j dtj
�W t� |� S t� |� 0 d S )NF) rE rA rB rC r2 r8 r9 r4 r3 rD r: r; )r rH r r r r@ y s
�
�
�zDupHandle.detach)N)r% r& r' r r@ r r r r r. h s
r. )�DupFd�sendfds�recvfds�darwinc C sV t � d|�}tt|�d g�}| �|gtjtj|fg� trR| �d�dkrRt d��d S )N�i� r � Az%did not receive acknowledgement of fd)
�array�bytes�lenr �socket�
SOL_SOCKETr � ACKNOWLEDGEr? � RuntimeError)�sockZfds�msgr r r rJ � s
rJ c C s t � d�}|j| }| �dt�|��\}}}}|s:|s:t�z�trJ| �d� t|�dkrft dt|� ��|d \}} }
|tj
kr�| tj kr�t|
�|j dkr�t �|�
|
� t|�d |d kr�td�t|�|d ���t|�W S W n t tfy� Y n0 t d��d S ) NrM r rO z received %d items of ancdatar rN z Len is {0:n} but msg[0] is {1!r}zInvalid data received)rP �itemsizeZrecvmsgrS Z
CMSG_SPACE�EOFErrorrU r< rR rV rT r �
ValueErrorZ frombytes�AssertionError�format�list�
IndexError) rW �size�aZ
bytes_sizerX Zancdata�flagsZaddrZ
cmsg_levelZ cmsg_typeZ cmsg_datar r r rK � s:
�
�
��rK c C sF t �| �� t jt j�� }t||g� W d � n1 s80 Y d S r
)rS �fromfd�fileno�AF_UNIX� SOCK_STREAMrJ )r= r6 r> �sr r r r � s c C sH t �| �� t jt j�� }t|d�d W d � S 1 s:0 Y d S )Nr r )rS rc rd re rf rK )r= rg r r r r � s c C sF t �� }|d ur |�|�| ��S tr:ddlm} |�| �S td��d S )Nr )�resource_sharerz&SCM_RIGHTS appears not to be available)r Zget_spawning_popenrI Zduplicate_for_child�HAVE_SEND_HANDLE� rh r[ )�fdZ popen_objrh r r r rI � s
rI c C s2 | j d u r t| j| jjffS t| j | jjffS d S r
)�__self__�getattrr �__func__r% ��mr r r �_reduce_method� s
rq 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' rs r r r r rr � s rr c C s t | j| jffS r
)rm � __objclass__r% ro r r r �_reduce_method_descriptor� s ru c C s t | j| j| jpi ffS r
)�_rebuild_partial�funcr �keywords)�pr r r �_reduce_partial� s rz c C s t j| g|�R i |��S r
)� functools�partial)rw r rx r r r rv � s rv c C s ddl m} t|| �ffS )Nr )� DupSocket)rh r} �_rebuild_socket)rg r} r r r �_reduce_socket� s r c C s | � � S r
)r@ )Zdsr r r r~ � s r~ c C s" t | �� �}t|| j| j| jffS r
)rI rd r~ �familyr �proto)rg �dfr r r r � s c C s | � � }tj||||d�S )N)rd )r@ rS )r� r� r r� rk r r r r~ � s c @ s` e Zd ZeZeZeZeZeZej dkr4e
Z
e Z e Z n e
Z
eZeZeZeZeZeZeZdd� ZdS )�AbstractReducerr
c G sN t tt� j�t� t ttj�t� t ttj �t� t t
j t � t t
j
t� d S r
)r r rr rs rq r^ �appendru �int�__add__r{ r| rz rS r r r r r r s
zAbstractReducer.__init__N)r% r& r' r r r r r �sys�platformr0 r/ r. rJ rK rI rq ru rv r r~ r r r r r r� � s$
r� )� metaclass)N)NF).�abcr r( r{ r rA r* rS r� rj r �__all__r� �hasattrri ZPicklerr r r r2 r/ r0 r r �objectr. rP rU rJ rK rI rq rr r rs ru r^ r� r� r� rz rv r| r r~ r� r r r r �
sj
�
�
� #