Your IP : 13.59.76.213
3
\�! � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddlm Z ddlm
Z
ddl m Z ddlm
Z
ddlmZ ddlmZ d d
d d gZd
Zejd�ZG dd� de�Zddd�Zdd� Zdd� Zdd� Ze� ZejZej 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 | _tj� | _dg| _d S )N�__main__)�_forkserver_address�_forkserver_alive_fd�_forkserver_pid�_inherited_fds� threadingZLock�_lock�_preload_modules)�self� r �"/usr/lib64/python3.6/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
/ s z ForkServer.get_inherited_fdsc C s� | j � t|�d tkr td��tjtj���}|j| j� tj � \}}tj � \}}||| j
t j � g}||7 }z<yt
j||� ||fS tj|� tj|� � Y nX W dtj|� tj|� 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 7 s&
z!ForkServer.connect_to_new_processc s� | j ��n tj� | jdk rVtj| jtj�\}}|s8dS tj| j� d| _ d| _d| _d }| j
r�ddh� t j d�}t
� fdd�|j� D ��}ni }tjtj���}tjd �}|j|� tj|d
� |j� tj� \}}ztyV|j� |g} ||j� || j
|f; }t j� }
|
gtj� } | d |g7 } tj|
| | �}W n tj|� � Y nX W dtj|� 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.
Nz-from multiprocessing.forkserver import main; zmain(%d, %d, %r, **%r)� main_path�sys_path�ignorec 3 s" | ]\}}|� kr||fV qd S )Nr )r �x�y)� desired_keysr r r o s z,ForkServer.ensure_running.<locals>.<genexpr>r( i� z-czCfrom multiprocessing.forkserver import main; main(%d, %d, %r, **%r)) r r r r r) �waitpid�WNOHANGr+ r r r r Zget_preparation_data�dict�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 T sL
zForkServer.ensure_runningN)�__name__�
__module__� __qualname__r r r
r r r r r r r s
r c 4 C s� |rhd|kr8|dk r8dt j� _ztj|� W dt j� `X x.|D ]&}y t|� W q> tk
rb Y q>X q>W tj� t j
t j t j t j i}dd� |j
� D �}tjtj| d���,}tj� ��} |j� t_| j|tj� | j|tj� x�y�x dd� | j� D �}
|
r�P q�W ||
k�r t�|j� d �l} d
} tj� d k�r�zFyt | |||� W n. t k
�rn t j t j � � t j!j"� Y nX W dtj#| � X W dQ R X W q� t$k
�r� }
z|
j%t%j&k�r�� W Y dd}
~
X q�X q�W W dQ R X W dQ R X dS ) zRun forkserver.r NTc S s i | ]\}}t j ||�|�qS r )�signal)r �sig�valr r r �
<dictcomp>� s zmain.<locals>.<dictcomp>)r: c S s g | ]\}}|j �qS r )Zfileobj)r �keyZeventsr r r �
<listcomp>� s zmain.<locals>.<listcomp>r r )'r Zcurrent_processZ _inheritingr Zimport_main_path�
__import__� ImportErrorr Z _close_stdinrD �SIGCHLD�SIG_IGN�SIGINTr8 r' r( � selectorsZDefaultSelectorZ getsockname� _forkserverr �registerZ
EVENT_READZselect�
SystemExitZacceptr) �fork�
_serve_one� Exception�sys�
excepthook�exc_info�stderr�flush�_exit�OSError�errnoZ ECONNABORTED)Z listener_fdr? Zpreloadr/ r0 �modname�handlersZ old_handlersr> ZselectorZrfds�s�code�er r r �main� sR
rc c C s� |j � tj |� x |j� D ]\}}tj||� q W tj| td �}| j � |^}}t_} t_ | t
j _ t
|tj� � dtjkr�dd l}
|
j� tj|�} t
|| � d S )Nr �randomr )r+ r) r8 rD r Zrecvfdsr% rP r r r Z_semaphore_trackerZ_fd�write_unsigned�getpidrV �modulesrd Zseedr �_main) r` r> r? r_ rE rF r, r. r- Zstfdrd ra r r r rT � s
rT c C sR d}t j}x8t|�|k rBtj| |t|� �}|s8td��||7 }q W t j|�d S )N� zunexpected EOFr )�UNSIGNED_STRUCT�sizer$ r) �read�EOFError�unpack)�fdr= Zlengthr` r r r �
read_unsigned� s rp c C s@ t j|�}x0|r:tj| |�}|dkr,td��||d � }q W d S )Nr zshould not get here)rj �packr) �write� RuntimeError)ro �n�msg�nbytesr r r re � s
re )NN) r] r) rO rD r' �structrV r � r r �contextr r r r �__all__r% �Structrj �objectr rc rT rp re rP r r
r r r r r r �<module> s6
s
=!