Edit File: forkserver.cpython-36.pyc
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 �2/usr/lib64/python3.6/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 / 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���R}tj� ��<} |j� t_| j|tj� | j|tj� �x y�x dd� | j� D �} | r�P q�W || k�r$tj|d �d k�s t�t�|| k�s2t�|j � d �l} d } tj � d k�r�zFyt | |||� W n. t k �r� 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_READZselectr) �read�AssertionError� SystemExitZaccept�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� sV rf c C s� |j � tj |� x |j� D ]\}}tj||� q W tj| td �}| j � t|�tks\t�|^}}t _ } t _ | t j _t|tj� � dtjkr�dd l} | j� tj|�} t|| � d S )Nr �randomr )r+ r) r8 rD r Zrecvfdsr% r$ rT rQ r r r Z_semaphore_trackerZ_fd�write_unsigned�getpidrY �modulesrg Zseedr �_main) rc r> r? rb rE rF r, r. r- Zstfdrg rd r r r rW � s rW c C sR d}t j}x8t|�|k rBtj| |t|� �}|s8td��||7 }q W t j|�d S )NrJ zunexpected EOFr )�UNSIGNED_STRUCT�sizer$ r) rS �EOFErrorZunpack)�fdr= Zlengthrc 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)rl Zpackr) �write� RuntimeError)ro �n�msg�nbytesr r r rh � s rh )NN) r` r) rP rD r' ZstructrY r � r r �contextr r r r �__all__r% ZStructrl �objectr rf rW rp rh rQ r r r r r r r r �<module> s6 s =!