a XC?hn/�@s�ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl mZddlm Z ddl mZddl mZddl mZgd �Zd Ze�d�ZGdd �d e�Zddd�Zdd�Zdd�Zdd�Ze�ZejZejZejZejZdS)�N�)� connection)�process)� reduction)�resource_tracker)�spawn)�util)�ensure_running�get_inherited_fds�connect_to_new_process�set_forkserver_preload��qc@sDeZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd d�Z dS)� ForkServercCs.d|_d|_d|_d|_t��|_dg|_dS)N�__main__)�_forkserver_address�_forkserver_alive_fd�_forkserver_pid�_inherited_fds� threadingZLock�_lock�_preload_modules��self�r�?/opt/alt/python39/lib64/python3.9/multiprocessing/forkserver.py�__init__"s zForkServer.__init__cCs2|j�|��Wd�n1s$0YdS�N)r�_stop_unlockedrrrr�_stop*szForkServer._stopcCsV|jdurdSt�|j�d|_t�|jd�d|_t�|j�sLt�|j�d|_dS)Nr) r�os�closer�waitpidr�is_abstract_socket_namespacer�unlinkrrrrr/s zForkServer._stop_unlockedcCs&tdd�|jD��std��||_dS)Ncss|]}t|�tuVqdSr)�type�str)�.0�modrrr� @�z4ForkServer.set_forkserver_preload..z&module_names must be a list of strings)�allr� TypeError)rZ modules_namesrrrr>sz!ForkServer.set_forkserver_preloadcCs|jSr)rrrrrr DszForkServer.get_inherited_fdsc Cs|��t|�dtkr td��t�tj��}|�|j�t� �\}}t� �\}}|||j t��g}||7}zvz:t �||�||fWWt�|�t�|�Wd�St�|�t�|��Yn0Wt�|�t�|�nt�|�t�|�0Wd�n1�s0YdS)N�ztoo many fds)r �len�MAXFDS_TO_SEND� ValueError�socket�AF_UNIXZconnectrr �piperrZgetfdrZsendfdsr!)r�fdsZclientZparent_r�child_w�child_rZparent_wZallfdsrrrrLs2� � � z!ForkServer.connect_to_new_processcs�|j��t��|jdurbt�|jtj�\}}|sDWd�dSt�|j�d|_ d|_d|_d}|j r�ddh�t�d�}�fdd�|� �D�}ni}t�tj��}t�d�}|�|�t�|�s�t�|d�|��t��\}}z~zV|��|g} ||��||j |f;}t��} | gt��}|d |g7}t�| || �}Wnt�|��Yn0Wt�|�nt�|�0||_ ||_||_Wd�n1�s�0YWd�n1�s�0YdS) NzCfrom multiprocessing.forkserver import main; main(%d, %d, %r, **%r)� main_path�sys_path�ignorecsi|]\}}|�vr||�qSrr)r'�x�y�Zdesired_keysrr� �r*z-ForkServer.ensure_running..r2i�z-c)rrr rr r"�WNOHANGr!rrrrZget_preparation_data�itemsr1r2rZarbitrary_addressZbindrr#�chmodZlistenr3�filenoZget_executableZ_args_from_interpreter_flagsZspawnv_passfds)r�pidZstatus�cmd�data�listenerZaddress�alive_rZalive_wZfds_to_passZexe�argsrr<rr isN � zForkServer.ensure_runningN) �__name__� __module__�__qualname__rrrrr rr rrrrr srcCs�|rjd|vr@|dur@dt��_zt�|�Wt��`n t��`0|D]$}zt|�WqDtyfYqD0qDt��t � �\}}t �|d�t �|d�dd�}tj |tjtji} dd�| ��D�} t�|�i}tjtj|d��}t��} |��t_| �|tj�| �|tj�| �|tj��zPd d �| ��D�}|�r(�qH�q(||v�rVt�||v�rt �|d�zt �dt j�\}}Wnt �y�Y�qYn0|d k�r��q|�!|d�}|du�r�t �"|�}zt#||�Wnt$�y�Yn0t �%|�nt&�'d|��ql||v�rt|�(�d ��<}t)�*|t+d�}t,|�t+k�rXt-d�.t,|��|^}}}|�%�t �/�}|d k�rd}zxz<|�%�| �%�||||g}|�0|�1��t2|||| �}Wn,t3�y�t4j5t4�6��t4j7�8�Yn0Wt �9|�nt �9|�0nLzt#||�Wnt$�y*Yn0|||<t �%|�|D]}t �%|��qBWd�n1�sj0YWn6t:�y�}z|j;t;j.sigchld_handlercSsi|]\}}|t�||��qSr)�signal)r'�sig�valrrrr=�s�zmain..)rAcSsg|]\}}|j�qSr)Zfileobj)r'�keyZeventsrrr� �r*zmain..i��rz.forkserver: waitpid returned unexpected pid %drzToo many ({0:n}) fds to send)=rZcurrent_processZ_inheritingrZimport_main_path� __import__�ImportErrorrZ_close_stdinr r3�set_blockingrL�SIGCHLD�SIGINT�SIG_IGNr?� set_wakeup_fdr1r2� selectorsZDefaultSelectorZgetsockname�_forkserverr�registerZ EVENT_READZselect� SystemExit�readr"r>�ChildProcessError�pop�waitstatus_to_exitcode�write_signed�BrokenPipeErrorr!�warnings�warnZacceptrZrecvfdsr/r.�RuntimeError�format�fork�extend�values� _serve_one� Exception�sys� excepthook�exc_info�stderr�flush�_exit�OSError�errnoZECONNABORTED)Zlistener_fdrFZpreloadr7r8�modnameZsig_rZsig_wrK�handlersZold_handlersZ pid_to_fdrEZselectorZrfdsrB�stsr5� returncode�sr4r6�code� unused_fds�fd�errr�main�s� �� 2r}c Csht�d�|��D]\}}t�||�q|D]}t�|�q,|^t_tj_ t_ t�|�}t� ||�}|S)NrQ)rLrXr?r r!rZrrZ_resource_trackerZ_fdr�duprZ_main) r6r4rzrurMrNr{Zparent_sentinelryrrrrj+s � rjcCsNd}tj}t|�|kr@t�||t|��}|s6td��||7}q t�|�dS)Nr*zunexpected EOFr)� SIGNED_STRUCT�sizer.r r]�EOFErrorZunpack)r{rDZlengthrxrrr�read_signedBs r�cCs<t�|�}|r8t�||�}|dkr*td��||d�}q dS)Nrzshould not get here)rZpackr �writere)r{�n�msg�nbytesrrrraLs ra)NN) rsr rYrLr1Zstructrlrrc�rr�contextrrrr�__all__r/ZStructr�objectrr}rjr�rarZr r rrrrrr�s: