File: //usr/local/lib/python3.9/site-packages/billiard/__pycache__/forkserver.cpython-39.pyc
a
X>hZ � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddlm Z ddlm
Z
ddlm Z ddlm Z ddlm
Z
ddlmZ dd lmZ g d
�Zd Ze�d �ZG d
d� d�Zddd�Zdd� Zdd� Zdd� Zdd� Ze� Zej Z ej Z ej Z ej Z dS )� N� )�
connection)�process)� reduction)�semaphore_tracker)�spawn)�util)�spawnv_passfds)�ensure_running�get_inherited_fds�connect_to_new_process�set_forkserver_preload� �Qc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
ForkServerc C s( d | _ d | _d | _t�� | _dg| _d S )N�__main__)�_forkserver_address�_forkserver_alive_fd�_inherited_fds� threading�Lock�_lock�_preload_modules��self� r �=/usr/local/lib/python3.9/site-packages/billiard/forkserver.py�__init__$ s
zForkServer.__init__c C s& t dd� | jD ��s td��|| _dS )z>Set list of module names to try to load in forkserver process.c s s | ]}t |�tu V qd S �N)�type�str)�.0�modr r r � <genexpr>- � z4ForkServer.set_forkserver_preload.<locals>.<genexpr>z&module_names must be a list of stringsN)�allr � TypeError)r Z
modules_namesr r r r
+ s z!ForkServer.set_forkserver_preloadc C s | j S )z�Return list of fds inherited from parent process.
This returns None if the current process was not started by fork
server.
)r r r r r r 1 s z ForkServer.get_inherited_fdsc
C s | � � t|�d tkr td��t�tj���}|�| j� t� � \}}t� � \}}||| j
t � � g}||7 }zvz:t
�||� ||fW W t�|� t�|� W d � S t�|� t�|� � Y n0 W t�|� t�|� nt�|� t�|� 0 W d � n1 �s0 Y dS )a; Request forkserver to create a child process.
Returns a pair of fds (status_r, data_w). The calling process can read
the child process's pid and (eventually) its returncode from status_r.
The calling process should write to data_w the pickled preparation and
process data.
� z too many fdsN)r
�len�MAXFDS_TO_SEND�
ValueError�socket�AF_UNIX�connectr �os�piper r Zgetfdr Zsendfds�close)r �fds�clientZparent_r�child_w�child_rZparent_wZallfdsr r r r 9 s2 �
�
�
z!ForkServer.connect_to_new_processc
s� | j ��b t�� | jdur,W d � dS d}| jr`ddh� t�d�}� fdd�|�� D �}ni }t�tj ���}t
� d�}|� |� t
�|d � |�� t
�� \}}z|zT|�� |g}||�� || j|f; }t�� }|gt�� } | d
|g7 } t|| |� W n t
�|� � Y n0 W t
�|� n t
�|� 0 || _|| _W d � n1 �sR0 Y W d � n1 �sr0 Y dS ) z�Make sure that a fork server is running.
This can be called from any process. Note that usually a child
process will just reuse the forkserver started by its parent, so
ensure_running() will do nothing.
Nz<from billiard.forkserver import main; main(%d, %d, %r, **%r)� main_path�sys_path�ignorec s i | ]\}}|� v r||�qS r r )r! �x�y�Z desired_keysr r �
<dictcomp>h s z-ForkServer.ensure_running.<locals>.<dictcomp>r, i� z-c)r r r
r r r Zget_preparation_data�itemsr+ r, r Zarbitrary_address�bindr. �chmod�listenr/ �fileno�get_executabler � _args_from_interpreter_flagsr r0 r )
r �cmd�data�listener�address�alive_rZalive_wZ fds_to_passZexe�argsr r: r r
V sB
�
�
zForkServer.ensure_runningN)�__name__�
__module__� __qualname__r r
r r r
r r r r r " s
r c
C sh |rjd|v r@|dur@dt �� _zt�|� W t �� `n
t �� `0 |D ]$}z t|� W qD tyf Y qD0 qDtjdur�ztj� � t
t j �t_W n t
tfy� Y n0 t�tjtj�}tjtj| d����}t�� ��^}|�� t_|�|tj� |�|tj� z�dd� |�� D �} | r��qq�|| v �r>t � |d�dk�s:J �t �|| v �sLJ �|� � d �~}
d} t � � d k�r�zNzt |
|||� W n, t!�y� tj"t�#� � tj$�%� Y n0 W t �&| � n t �&| � 0 W d � n1 �s�0 Y W q� t
�y } z | j't'j(k�r � W Y d} ~ q�d} ~ 0 0 q�W d � n1 �s:0 Y W d � n1 �sZ0 Y dS )
zRun forkserver.r NT)r@ c S s g | ]\}}|j �qS r )�fileobj)r! �key�eventsr r r �
<listcomp>� r$ zmain.<locals>.<listcomp>r r$ r ))r �current_processZ _inheritingr Zimport_main_path�
__import__� ImportError�sys�stdinr0 �openr. �devnull�OSErrorr* �signal�SIGCHLD�SIG_IGNr+ r, � selectors�DefaultSelector� getsockname� _forkserverr �register�
EVENT_READ�select�read�
SystemExit�accept�fork�
_serve_one� Exception�
excepthook�exc_info�stderr�flush�_exit�errno� ECONNABORTED)
Z listener_fdrG Zpreloadr5 r6 �modname�handlerrE �selectorZrfds�s�code�er r r �main� sZ
�
<ru c G s | ||||fS r r )r4 r3 �alive�stfdZ inheritedr r r � __unpack_fds� s rx c
C s� |� � t� |� t�tj|� t�| td �}| � � t|�tksHJ �t|� \}}t _
}t _ |t j
_t|t�� � dtjv r�dd l}|�� t�|�} t|| � d S )Nr �randomr )r0 r. rX rY r Zrecvfdsr) r( rx r^ r r r Z_semaphore_trackerZ_fd�write_unsigned�getpidrS �modulesry �seedr �_main)
rr rE rG rp r1 r4 r3 rw ry rs r r r rf � s
�
rf c C sN d}t j}t|�|k r@t�| |t|� �}|s6td��||7 }q
t �|�d S )Nr$ zunexpected EOFr )�UNSIGNED_STRUCT�sizer( r. rb �EOFError�unpack)�fdrD �lengthrr r r r �
read_unsigned� s
r� c C s< t �|�}|r8t�| |�}|dkr*td��||d � }q
d S )Nr zshould not get here)r �packr. �write� RuntimeError)r� �n�msg�nbytesr r r rz � s
rz )NN) rm r. r[ rX r+ �structrS r � r r r r r r �compatr �__all__r) �Structr r ru rx rf r� rz r^ r
r r r
r r r r �<module> s8
j
;"