��Yf �@sdddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlmZddlmZddlm Z ddlmZd d ddgZd Zejd�ZGdd�de�Zdddd�Zdd�Zdd�Zdd�Ze�ZejZejZejZejZdS)�N�)� connection)�process)� reduction)�semaphore_tracker)�spawn)�util�ensure_running�get_inherited_fds�connect_to_new_process�set_forkserver_preload��Qc@sLeZdZdd�Zdd�Zdd�Zdd�Zd d �ZdS)� ForkServercCs:d|_d|_d|_tj�|_dg|_dS)N�__main__)�_forkserver_address�_forkserver_alive_fd�_inherited_fds� threadingZLock�_lock�_preload_modules)�self�r�?/opt/alt/python35/lib64/python3.5/multiprocessing/forkserver.py�__init__!s zForkServer.__init__cCs5tdd�|jD��s(td��||_dS)z>Set list of module names to try to load in forkserver process.css!|]}t|�tkVqdS)N)�type�str)�.0�modrrr� *sz4ForkServer.set_forkserver_preload..z&module_names must be a list of stringsN)�allr� TypeError)rZ modules_namesrrrr(sz!ForkServer.set_forkserver_preloadcCs|jS)z�Return list of fds inherited from parent process. This returns None if the current process was not started by fork server. )r)rrrrr .szForkServer.get_inherited_fdscCs |j�t|�dtkr,td��tjtj��}|j|j�tj �\}}tj �\}}|||j tj�g}||7}zJyt j||�||fSWn%tj|�tj|��YnXWdtj|�tj|�XWdQRXdS)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. �ztoo many fdsN)r �len�MAXFDS_TO_SEND� ValueError�socket�AF_UNIXZconnectr�os�piperrZgetfdrZsendfds�close)r�fdsZclientZparent_r�child_w�child_rZparent_wZallfdsrrrr6s& z!ForkServer.connect_to_new_processcs�|j��tj�|jdk r'dSdd}|jr}ddh�tjd�}t�fdd�|j�D��}ni}t j t j �� }tjd �}|j |�tj|d �|j�tj�\}}z�yy|j�|g}||j�||j|f;}tj�}|gtj�} | d|g7} tj|| |�} Wntj|��YnXWdtj|�X||_||_WdQRXWdQRXdS)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�ignorec3s-|]#\}}|�kr||fVqdS)Nr)r�x�y)�desired_keysrrres z,ForkServer.ensure_running..r'i�z-c)rrr rrrZget_preparation_data�dict�itemsr&r'rZarbitrary_addressZbindr(�chmodZlistenr)�filenoZget_executablerZ_args_from_interpreter_flagsZspawnv_passfdsr*r)r�cmd�data�listenerZaddress�alive_rZalive_wZfds_to_passZexe�args�pidr)r3rr Ss> ( zForkServer.ensure_runningN)�__name__� __module__�__qualname__rrr rr rrrrrs rc4Csx|r�d|krN|dk rNdtj�_ztj|�Wdtj�`Xx1|D])}yt|�WqUtk r}YqUXqUWtj�t j t jt jt j i}dd�|j�D�}tjtjd|��}tj��} |j�t_| j|tj�| j|tj�xAyx$dd�| j�D�} | r+Pq+W|| krtj|d �d ksyt�t�|| ks�t�|j�d��}d }tj�dkrzPyt||||�Wn2t k r t!j"t!j#��t!j$j%�YnXWdtj&|�XWdQRXWq%t'k ra} z| j(t(j)krO�WYdd} ~ Xq%Xq%WWdQRXWdQRXdS)zRun forkserver.rNTcSs+i|]!\}}tj||�|�qSr)�signal)r�sig�valrrr� �s zmain..r7cSsg|]\}}|j�qSr)Zfileobj)r�keyZeventsrrr� �s zmain..r�r)*rZcurrent_processZ_inheritingrZimport_main_path� __import__�ImportErrorrZ_close_stdinrA�SIGCHLD�SIG_IGN�SIGINT�SIG_DFLr5r&r'� selectorsZDefaultSelectorZgetsockname�_forkserverr�registerZ EVENT_READZselectr(�read�AssertionError� SystemExitZaccept�fork� _serve_one� Exception�sys� excepthook�exc_info�stderr�flush�_exit�OSError�errnoZECONNABORTED)Zlistener_fdr;Zpreloadr.r/�modname�handlersZold_handlersr:ZselectorZrfds�s�code�errr�main�sV rdcCs�|j�tj|�x*|j�D]\}}tj||�q$Wtj|td�}|j�t|�tks|t�|^}}t _ } t _| tj _t|tj��dtjkr�ddl} | j�tj|�}t||�dS)Nr�randomr)r*r(r5rArZrecvfdsr$r#rRrOrrrZ_semaphore_trackerZ_fd�write_unsigned�getpidrW�modulesreZseedr�_main)rar:r;r`rBrCr+r-r,ZstfdrerbrrrrU�s rUcCsqd}tj}xNt|�|kr_tj||t|��}|sRtd��||7}qWtj|�dS)NrGzunexpected EOFr)�UNSIGNED_STRUCT�sizer#r(rQ�EOFErrorZunpack)�fdr9Zlengthrarrr� read_unsigned�s rncCsZtj|�}xD|rUtj||�}|dkrBtd��||d�}qWdS)Nrzshould not get here)rjZpackr(�write�RuntimeError)rm�n�msg�nbytesrrrrf�s rf)r^r(rNrAr&ZstructrWr�rrrrrr�__all__r$ZStructrj�objectrrdrUrnrfrOr r rrrrrr�s6 h=!