File: //usr/local/lib/python3.9/site-packages/billiard/__pycache__/reduction.cpython-39.pyc
a
X>h�$ � @ s d dl Z d dlZd dlZd dlZd dlZd dlZddlmZ g d�Zej d dkZ
ej dkpze ed�oze ed�oze ejd �Z
e
r�d dlZG d
d � d ej�ZnG d d � d ej�ZejZd6d
d�Zej dk�reg d�7 ZddlmZ d7dd�Zdd� Zdd� Zdd� ZG dd� d�ZnFeg d �7 Zd 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)ee j*e(� ej dk�r�d0d1� Z+d2d3� Z,eeje+� n d4d1� Z+d5d3� Z,eeje+� dS )8� N� )�context)� send_handle� recv_handle�ForkingPickler�register�dump� �win32�CMSG_LEN�
SCM_RIGHTS�sendmsgc sX e Zd ZdZi ZejZ� fdd�Ze dd� �Z
e d dd��Z e d d d
��Z e
jZ� ZS )
r �)Pickler subclass used by multiprocessing.c s. t t| �j|� | j�� | _| j�| j� d S �N)�superr �__init__�_copyreg_dispatch_table�copy�dispatch_table�update�_extra_reducers)�self�args�� __class__� �</usr/local/lib/python3.9/site-packages/billiard/reduction.pyr * s zForkingPickler.__init__c C s || j |< dS )�&Register a reduce function for a type.N)r )�cls�type�reducer r r r / s �ForkingPickler.registerNc C s t �� }| ||��|� |�� S r )�io�BytesIOr � getbuffer�r �obj�protocol�bufr r r �dumps4 s �ForkingPickler.dumpsc C s | � |�� �S r )�loadsr$ �r r( r' r r r �loadbuf: s �ForkingPickler.loadbuf)N)N)�__name__�
__module__� __qualname__�__doc__r �copyregr r r � classmethodr r) r- �pickler+ �
__classcell__r r r r r % s
r c @ sV e Zd ZdZejj�� Zedd� �Z ed dd��Z
ed dd��Z eej fd d
��Z dS )
r r c s � fdd�}|| j |< dS )r c s � |�}| j |d|i� d S )Nr& )� save_reduce)r r& �rv�r r r �
dispatcherI s z+ForkingPickler.register.<locals>.dispatcherN)�dispatch)r r r r: r r9 r r F s r! Nc C s t �� }| ||��|� |�� S r )r" r# r �getvaluer% r r r r) N s r* c C s | � |�� �S r )r+ r< r, r r r r- T s r. c C s t |tj�r|�� }||�S r )�
isinstancer" r# r< )r r( r+ r r r r+ X s zForkingPickler.loads)N)N)
r/ r0 r1 r2 r5 �Picklerr; r r4 r r) r- r+ r r r r r B s
c C s t ||��| � dS )z3Replacement for pickle.dump() using ForkingPickler.N)r r )r& �filer' r r r r ` s r )� DupHandle� duplicate� steal_handle)�_winapiFc C s* |du rt �� }t �t �� | |d|t j�S )z<Duplicate a handle. (target_process is a handle not a pid!)Nr )rC �GetCurrentProcess�DuplicateHandle�DUPLICATE_SAME_ACCESS)�handleZtarget_process� inheritabler r r rA m s
�rA c C sN t �t jd| �}z,t �||t �� ddt jt jB �W t �|� S t �|� 0 dS )z5Steal a handle from process identified by source_pid.Fr N)rC � OpenProcess�PROCESS_DUP_HANDLErE rD rF �DUPLICATE_CLOSE_SOURCE� CloseHandle)Z
source_pidrG Zsource_process_handler r r rB u s �
�
�rB c C s t |tj|�}| �|� dS �z&Send a handle over a local connection.N)r@ rC rF �send)�connrG �destination_pidZdhr r r r � s r c C s | � � �� S )�)Receive a handle over a local connection.)�recv�detach)rO r r r r � s r c @ s" e Zd ZdZddd�Zdd� ZdS )r@ z Picklable wrapper for a handle.Nc C sf |d u rt �� }t�tjd|�}z(t�t�� |||dd�| _W t�|� n t�|� 0 || _ || _
d S )NFr ) �os�getpidrC rI rJ rE rD �_handlerL �_access�_pid)r rG �access�pid�procr r r r � s
�zDupHandle.__init__c C sb | j t�� kr| jS t�tjd| j �}z*t�|| jt�� | j dtj
�W t� |� S t� |� 0 dS )z1Get the handle. This should only be called once.FN) rX rT rU rV rC rI rJ rE rD rW rK rL )r r[ r r r rS � s
�
�
�zDupHandle.detach)N)r/ r0 r1 r2 r rS r r r r r@ � s
r@ )�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 � ACKNOWLEDGErR � RuntimeError)�sock�fds�msgr r r r] � s
r] 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 ks�J �t|�W S W n t tfy� Y n0 t d��dS ) z/Receive an array of fds over an AF_UNIX socket.r` r rb z received %d items of ancdatar ra zInvalid data receivedN)rc �itemsize�recvmsgrf r �EOFErrorrh rN re ri rg r �
ValueError� frombytes�list�
IndexError) rj �size�aZ
bytes_sizerl �ancdata�flags�addr�
cmsg_level� cmsg_type� cmsg_datar r r r^ � s4
�
�
�
r^ c C sJ | � � }t�|tjtj�� }t||g� W d � n1 s<0 Y dS rM )�filenorf �fromfd�AF_UNIX� SOCK_STREAMr] )rO rG rP �fd�sr r r r � s c C sL | � � }t�|tjtj�� }t|d�d W d � S 1 s>0 Y dS )rQ r r N)r| rf r} r~ r r^ )rO r� r� r r r r � s c C sF t �� }|dur |�|�| ��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_popenr\ Zduplicate_for_child�HAVE_SEND_HANDLE� r� rp )r� Z popen_objr� r r r r\ � s
r\ 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
r� c @ s e Zd Zdd� ZdS )�_Cc C s d S r r )r r r r �f� s z_C.fN)r/ r0 r1 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_partial�funcr �keywords)�pr r r �_reduce_partial s r� c C s t j| g|�R i |��S r )� functools�partial)r� r r� r r r r�
s r� c C s ddl m} t|| �ffS )Nr )� DupSocket)r� r� �_rebuild_socket)r� r� r r r �_reduce_socket s r� c C s | � � S r )rS )Zdsr r r r� s r� c C s" t | �� �}t|| j| j| jffS r )r\ r| r� �familyr �proto)r� �dfr r r r� s c C s | � � }tj||||d�S )N)r| )rS rf )r� r� r r� r� r r r r� " s )N)NF)-r� r" rT r5 rf �sysr� r �__all__� version_info�PY3�platform�hasattrr� r3 r> r r r �compatrC rA rB r r r@ rc rh r] r^ r\ r� r� r r� r� rr �append�int�__add__r� r� r� r� r� r r r r �<module>
sh
�
�
#