File: //lib/python3.7/multiprocessing/__pycache__/forkserver.cpython-37.pyc
B
U��e. � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZddl m
Z
ddl m Z ddl m
Z
ddl mZ ddl mZ ddl mZ d d
d d gZd
Ze�d�ZG dd� de�Zddd�Zdd� Zdd� Zdd� Ze� Zej Z ej Z ej Z ej Z dS )� N� )�
connection)�process)� reduction)�semaphore_tracker)�spawn)�util�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 | _d | _t�� | _dg| _d S )N�__main__)�_forkserver_address�_forkserver_alive_fd�_forkserver_pid�_inherited_fds� threadingZLock�_lock�_preload_modules)�self� r �0/usr/lib/python3.7/multiprocessing/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 |�tkV qd S )N)�type�str)�.0�modr r r � <genexpr>, s 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
0 s z ForkServer.get_inherited_fdsc
C s� | � � t|�d tkr td��t�tj���}|�| j� t� � \}}t� � \}}||| j
t � � g}||7 }z<yt
�||� ||fS t�|� t�|� � Y nX W dt�|� t�|� X W dQ R X 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_UNIXZconnectr �os�piper r Zgetfdr Zsendfds�close)r �fdsZclientZparent_r�child_w�child_rZparent_wZallfdsr r r r 8 s&
z!ForkServer.connect_to_new_processc s| | j ��j t�� | jdk rVt�| jtj�\}}|s8dS t�| j� d| _ d| _d| _d}| j
r�ddh� t � d�}� fdd�|�
� D �}ni }t�tj���}t�d�}|�|� t�|d � |�� t�� \}}ztyV|�� |g} ||�� || j
|f; }t �� }
|
gt�� } | d
|g7 } t�|
| | �}W n t�|� � Y nX W dt�|� X || _ || _|| _W dQ R X W dQ R X 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.
NzCfrom multiprocessing.forkserver import main; main(%d, %d, %r, **%r)� main_path�sys_path�ignorec s i | ]\}}|� kr||�qS r r )r �x�y)� desired_keysr r �
<dictcomp>p s z-ForkServer.ensure_running.<locals>.<dictcomp>r( i� z-c)r r r r r) �waitpid�WNOHANGr+ r r r r Zget_preparation_data�itemsr' r( r Zarbitrary_addressZbind�chmodZlistenr* �filenoZget_executabler Z _args_from_interpreter_flagsZspawnv_passfds) r �pidZstatus�cmd�data�listenerZaddress�alive_rZalive_wZ fds_to_passZexe�argsr )r4 r r U sJ
zForkServer.ensure_runningN)�__name__�
__module__� __qualname__r r r
r r r r r r r s
r c C s |rhd|kr8|dk r8dt �� _zt�|� W dt �� `X x.|D ]&}y t|� W q> tk
rb Y q>X q>W t�� t �
� \}}t � |d� t � |d� dd� }t j
|t jt ji} dd� | �� D �}
t �|� i } tjtj| d ��� } t�� ��}
| �� t_|
�| tj� |
�|tj� |
�|tj� �xʐy�x d
d � |
�� D �}|�r,P �q,W ||k�rtt � |d �d
k�spt d��t �||k�rdt � |d� x�yt � dt j �\}}W n t!k
�r� P Y nX |dk�r�P | �"|d�}|dk �rPt �#|��r�t �$|� }n&t �%|��st d�&||���t �'|�}yt(||� W n t)k
�rB Y nX t �*|� nt+�,d| � �q�W | |k�r�| �-� d ��0}t.�/|t0d �}t1|�t0k�r�t2d�&t1|����|^}}}|�*� t �3� }|dk�rVd }zpy<| �*� |
�*� ||||g}|�4| �5� � t6||||
�}W n. t7k
�rB t8j9t8�:� � t8j;�<� Y nX W dt �=|� X nRyt(||� W n t)k
�rz Y nX || |< t �*|� x|D ]}t �*|� �q�W W dQ R X W n4 t>k
�r� } z|j?t?j@k�r W dd}~X Y nX �q&W W dQ R X W dQ R X dS )zRun forkserver.r NTFc W s d S )Nr )Z_unusedr r r �sigchld_handler� s z main.<locals>.sigchld_handlerc S s i | ]\}}t � ||�|�qS r )�signal)r �sig�valr r r r5 � s zmain.<locals>.<dictcomp>)r: c S s g | ]\}}|j �qS r )Zfileobj)r �keyZeventsr r r �
<listcomp>� s zmain.<locals>.<listcomp>r � z Not at EOF?i ���r zChild {0:n} status is {1:n}z.forkserver: waitpid returned unexpected pid %dz Too many ({0:n}) fds to send)Ar Zcurrent_processZ _inheritingr Zimport_main_path�
__import__� ImportErrorr Z _close_stdinr) r* � set_blockingrE �SIGCHLD�SIGINT�SIG_IGNr8 �
set_wakeup_fdr' r( � selectorsZDefaultSelectorZ getsockname� _forkserverr �registerZ
EVENT_READZselect�read�AssertionError�
SystemExitr6 r7 �ChildProcessError�pop� WIFSIGNALED�WTERMSIG� WIFEXITED�format� WEXITSTATUS� write_signed�BrokenPipeErrorr+ �warnings�warnZacceptr Zrecvfdsr% r$ � RuntimeError�fork�extend�values�
_serve_one� Exception�sys�
excepthook�exc_info�stderr�flush�_exit�OSError�errnoZ ECONNABORTED)Z listener_fdr? Zpreloadr/ r0 �modnameZsig_rZsig_wrD �handlersZ old_handlersZ pid_to_fdr> ZselectorZrfdsr; �stsr- �
returncode�sr, r. �code�
unused_fds�fd�er r r �main� s�
r{ c C sd t �d� x |�� D ]\}}t � ||� qW x|D ]}t�|� q2W |^t_tj_ t_
t � | �}|S )NrK )
rE rR r8 r) r+ rT r r Z_semaphore_trackerZ_fdr r �_main)r. r, rx rs rF rG ry rw r r r rh s
rh c C sR d}t j}x8t|�|k rBt�| |t|� �}|s8td��||7 }q W t �|�d S )NrJ zunexpected EOFr )�
SIGNED_STRUCT�sizer$ r) rV �EOFErrorZunpack)ry r= �lengthrv r r r � read_signed2 s r� c C s@ t �|�}x0|r:t�| |�}|dkr,td��||d � }q W d S )Nr zshould not get here)r} Zpackr) �writerd )ry �n�msg�nbytesr r r r` < s
r` )NN) rq r) rS rE r' Zstructrj r rb � r r �contextr r r r �__all__r% ZStructr} �objectr r{ rh r� r` rT r r
r r r r r r �<module> s:
r