File manager

�
e f!�@sd
dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
lZdd
lZdd
lZddlm	Z	
ddlm
Z

ddlmZ
ddlmZ
ddlm
Z

ddlmZ
d	d
ddgZd
Zejd�
ZGdd�de�Zd
d
dd�Zdd�Zdd�Zdd�Ze�ZejZejZejZejZd
S)�N�
)
�
connection)
�process)
�	reduction)
�semaphore_tracker)
�spawn)
�util�ensure_running�get_inherited_fds�connect_to_new_process�set_forkserver_preload�
�
Qc@sLeZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�ZdS)�
ForkServerc

Cs:d|_d|_
d|_tj�|_d
g
|_dS)N�__main__)�_forkserver_address�_forkserver_alive_fd�_inherited_fds�	threadingZLock�_lock�_preload_modules)
�self�r�?/opt/alt/python34/lib64/python3.4/multiprocessing/forkserver.py�__init__!s

	
	
	

zForkServer.__init__cCs8td
d�|j
D�
�
s+td�
�
n|
|_
dS)z>Set list of module names to try to load in forkserver process.c
ss!|]}
t|
�
t
kV
qdS)
N)�type�str)�.0�modrrr�	<genexpr>*sz4ForkServer.set_forkserver_preload.<locals>.<genexpr>z&module_names must be a list of stringsN)�allr�	TypeError)rZ
modules_namesrrrr(s

z!ForkServer.set_forkserver_preloadc


Cs|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
|
�
d
tkr/td�
�
ntjtj�
��}|j|j�

tj	�\}}tj	�\}}|||j
tj�g}||
7}zJyt
j||�
||fSWn%


tj|�

tj|�

�Yn
XWdtj|�

tj|�

XWdQXdS)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�piperrZgetfdrZsendfds�close)r�fdsZclientZparent_r�child_w�child_rZparent_wZallfdsrrrr6s&





















z!ForkServer.connect_to_new_processc
s�
|j��

t
j�
|jd
k	r'd
Sdd}
|jr}ddh�tjd�
}t�f
dd�|j�D�
�
}ni}t	j	t	j
�
�

}tjd	�
}|j
|�

tj|d
�
|jd�

tj�\}}z�yy|j�|g}|
|j�||j|f;}
tj�}|g
tj�}	|	d|
g7}	tj||	|�}
Wn


tj|�

�Yn
XWd
tj|�

X||_||_Wd
QXWd
QXd
S)
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
3s-|]#\}
}|
�kr|
|fV
qdS)
Nr)r�
x�
y)
�desired_keysrrres	
z,ForkServer.ensure_running.<locals>.<genexpr>r'i�
�dz-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__rrr
rr	rrrrrs
rc
<Cs�|r�d
|krQ|dk	rQdtj
�_ztj|�

Wdtj
�`Xnx4|D])}yt|�

WqXtk
r�


YqXXqXWntjdk	r�y#tjj	�
t
tj�
t_Wq�t
tfk
r�


Yq�Xntjtjtj�}tjtjd|�

��
}tj��l
}|j�t_|j|tj�
|j|
tj�
x-
y�x'dd�|j�D�
}	|	rX
PqX
qX
W|
|	kr�
t�
n|j�d��}
d}tj�dkr#zPyt|
||
|�
Wn2t k
r



tj!tj"��
tj#j$�
Yn
XWdtj%|�

XnWdQXWqR
t&k
r>


YqR
t
k
rz
}
z|j't'j(krh�nWYdd}~XqR
XqR
WWdQXWdQXdS)	zRun forkserver.rNTr8c
Ssg|]\}
}|
j�qSr)
Zfileobj)r�keyZeventsrrr�
<listcomp>�s	zmain.<locals>.<listcomp>r
r))rZcurrent_processZ_inheritingrZimport_main_path�
__import__�ImportError�sys�stdinr*�openr(�devnull�OSErrorr%�signal�SIGCHLD�SIG_IGNr&r'�	selectorsZDefaultSelectorZgetsockname�_forkserverr�registerZ
EVENT_READZselect�
SystemExitZaccept�fork�
_serve_one�	Exception�
excepthook�exc_info�stderr�flush�_exit�InterruptedError�errnoZECONNABORTED)
Zlistener_fdr<Zpreloadr.r/�modname�handlerr;ZselectorZrfds�
s�code�
errr�main�sZ

























	












rac
Cs�|
j�
t
j|�

tjtj|�
tj|td
�}|j�
|^}}t_}t_	|t
j_t
|t
j��
dtjkr�ddl}|j�
ntj|�
}	t
||	�
dS)Nr�randomr
)r*r(rKrLrZrecvfdsr$rOrrrZ_semaphore_trackerZ_fd�write_unsigned�getpidrF�modulesrbZseedr�_main)
r^r;r<r]r+r-r,Zstfdrbr_rrrrS�s









rSc
Cs�d
}
tj
}xrt|
�
|kr�x:y tj||t|
�
�}Wntk
r[


Yq'XPq'W|svtd�
�
n|
|7}
qWtj|
�
dS)N�zunexpected EOFr
)�UNSIGNED_STRUCT�sizer#r(�readrZ�EOFErrorZunpack)�fdr:Zlengthr^rrr�
read_unsigned�s

	



 






rmcCs~tj
|
�
}xh|ryx0ytj||�}Wntk
rE


YqXPqW|d
krftd�
�
n||d�}qWdS)Nr
zshould not get here)rhZpackr(�writerZ�RuntimeError)rl�
n�msg�nbytesrrrrc�s

	








rc)r[r(rNrKr&ZstructrFr�rrrrrr�__all__r$ZStructrh�objectrrarSrmrcrOr	r
rrrrrr�<module>
s6











	
	
h>