3 � f�!�@s�ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlmZddlm Z ddlmZddlmZd d ddgZd Zejd�ZGdd�de�Zddd�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@s4eZdZdd�Zdd�Zdd�Zdd�Zd d �ZdS)� ForkServercCs.d|_d|_d|_d|_tj�|_dg|_dS)N�__main__)�_forkserver_address�_forkserver_alive_fd�_forkserver_pid�_inherited_fds� threadingZLock�_lock�_preload_modules)�self�r�?/opt/alt/python36/lib64/python3.6/multiprocessing/forkserver.py�__init__!s zForkServer.__init__cCs&tdd�|jD��std��||_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}z.r(i�z-czCfrom multiprocessing.forkserver import main; main(%d, %d, %r, **%r))rrr rr)�waitpid�WNOHANGr+rrrrZget_preparation_data�dict�itemsr'r(rZarbitrary_addressZbind�chmodZlistenr*�filenoZget_executablerZ_args_from_interpreter_flagsZspawnv_passfds)r�pidZstatus�cmd�data�listenerZaddress�alive_rZalive_wZfds_to_passZexe�argsr)r4rr TsL zForkServer.ensure_runningN)�__name__� __module__�__qualname__rrr rr rrrrrs rc4Cs�|rhd|kr8|dk r8dtj�_ztj|�Wdtj�`Xx.|D]&}yt|�Wq>tk rbYq>Xq>Wtj�t j t jt jt ji}dd�|j �D�}tjtj|d��,}tj��} |j�t_| j|tj�| j|tj�x�y�xdd�| j�D�} | r�Pq�W|| k�rt�|j�d �l}d }tj�d k�r�zFyt||||�Wn.tk �rntjtj ��tj!j"�YnXWdtj#|�XWdQRXWq�t$k �r�} z| j%t%j&k�r��WYdd} ~ Xq�Xq�WWdQRXWdQRXdS)zRun forkserver.rNTcSsi|]\}}tj||�|�qSr)�signal)r�sig�valrrr� �szmain..)r:cSsg|]\}}|j�qSr)Zfileobj)r�keyZeventsrrr� �szmain..rr)'rZcurrent_processZ_inheritingrZimport_main_path� __import__�ImportErrorrZ_close_stdinrD�SIGCHLD�SIG_IGN�SIGINTr8r'r(� selectorsZDefaultSelectorZgetsockname�_forkserverr�registerZ EVENT_READZselect� SystemExitZacceptr)�fork� _serve_one� Exception�sys� excepthook�exc_info�stderr�flush�_exit�OSError�errnoZECONNABORTED)Zlistener_fdr?Zpreloadr/r0�modname�handlersZold_handlersr>ZselectorZrfds�s�code�errr�main�sR rccCs�|j�tj|�x |j�D]\}}tj||�qWtj|td�}|j�|^}}t_} t_ | t j_t |tj��dtjkr�ddl} | j�tj|�}t ||�dS)Nr�randomr)r+r)r8rDrZrecvfdsr%rPrrrZ_semaphore_trackerZ_fd�write_unsigned�getpidrV�modulesrdZseedr�_main)r`r>r?r_rErFr,r.r-ZstfdrdrarrrrT�s rTcCsRd}tj}x8t|�|krBtj||t|��}|s8td��||7}qWtj|�dS)N�zunexpected EOFr)�UNSIGNED_STRUCT�sizer$r)�read�EOFErrorZunpack)�fdr=Zlengthr`rrr� read_unsigned�srocCs@tj|�}x0|r:tj||�}|dkr,td��||d�}qWdS)Nrzshould not get here)rjZpackr)�write�RuntimeError)rn�n�msg�nbytesrrrre�s re)NN)r]r)rOrDr'ZstructrVr�rr�contextrrrr�__all__r%ZStructrj�objectrrcrTrorerPr r rrrrrr�s6 s =!