File: //usr/libexec/oracle-cloud-agent/plugins/osms/multiprocessing/reduction.pyc
a
���d(% � @ 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�win32�CMSG_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__� �1/usr/lib64/python3.9/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 r )�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
r 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_handleF)�source_processc C s6 t �� }|du r|}|du r |}t �|| |d|t j�S )z<Duplicate a handle. (target_process is a handle not a pid!)Nr )�_winapi�GetCurrentProcess�DuplicateHandle�DUPLICATE_SAME_ACCESS)�handle�target_process� inheritabler4 �current_processr r r r2 G s �r2 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)r5 � OpenProcess�PROCESS_DUP_HANDLEr7 r6 r8 �DUPLICATE_CLOSE_SOURCE� CloseHandle)�
source_pidr9 �source_process_handler r r r3 S s �
�
�r3 c C s t |tj|�}| �|� dS �z&Send a handle over a local connection.N)r1 r5 r8 �send)�connr9 �destination_pid�dhr r r r _ s r c C s | � � �� S )�)Receive a handle over a local connection.)�recv�detach)rE r r r r d s r c @ s" e Zd ZdZddd�Zdd� ZdS )r1 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�getpidr5 r= r> r7 r6 �_handler@ �_access�_pid)r r9 �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 dS )z1Get the handle. This should only be called once.FN) rO rK rL rM r5 r= r> r7 r6 rN r? r@ )r rR r r r rJ y s
�
�
�zDupHandle.detach)N)r' r( r) r* r rJ r r r r r1 h s
r1 )�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 � ACKNOWLEDGErI � RuntimeError)�sock�fds�msgr r r rT � s
rT 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 )
z/Receive an array of fds over an AF_UNIX socket.rW r rY z received %d items of ancdatar rX z Len is {0:n} but msg[0] is {1!r}zInvalid data receivedN)rZ �itemsize�recvmsgr] �
CMSG_SPACE�EOFErrorr_ rD r\ r` r^ r �
ValueError� frombytes�AssertionError�format�list�
IndexError) ra �size�a�
bytes_sizerc �ancdata�flags�addr�
cmsg_level� cmsg_type� cmsg_datar r r rU � s:
�
�
��rU c C sF t �| �� t jt j�� }t||g� W d � n1 s80 Y dS rC )r] �fromfd�fileno�AF_UNIX� SOCK_STREAMrT )rE r9 rF �sr r r r � s c C sH t �| �� t jt j�� }t|d�d W d � S 1 s:0 Y dS )rH r r N)r] rw rx ry rz rU )rE 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 �get_spawning_popenrS �duplicate_for_child�HAVE_SEND_HANDLE� r| rh )�fd� popen_objr| r r r rS � s
rS 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' 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_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 )rJ )�dsr r r r� � s r� c C s" t | �� �}t|| j| j| jffS r )rS rx r� �familyr �proto)r{ �dfr r r r� � s c C s | � � }tj||||d�S )N)rx )rJ r] )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 )�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 r )r r r� r� r� rl �appendr� �int�__add__r� r� r� r] r� r r r r r s
zAbstractReducer.__init__N)r' r( r) r* r r r r r �sys�platformr3 r2 r1 rT rU rS r� r� r� r� r� r r r r r r� � s&
r� )� metaclass)N)NF).�abcr r+ r� r rK r- r] r� r� r �__all__r� �hasattrr �Picklerr r r r5 r2 r3 r r �objectr1 rZ r_ rT rU rS r� r� r r� r� rl r� r� r� r� r� r� r� r� r� r r r r �<module>
sj
�
�
� #