Yf @ sd d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m
Z
d d l m Z d d l m Z d d l m
Z
d d l m Z d d
d d g Z d
Z e j d Z Gd d d e Z d d d d Z d d Z d d Z d d Z e Z e j Z e j Z e j Z e j Z d S) N )
connection)process) reduction)semaphore_tracker)spawn)utilensure_runningget_inherited_fdsconnect_to_new_processset_forkserver_preload Qc @ sL e Z d Z d d Z d d Z d d Z d d Z d d
Z d S)
ForkServerc C s: d | _ d | _ d | _ t j | _ d g | _ d S)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__c C s5 t d d | j D s( t d | | _ d S)z>Set list of module names to try to load in forkserver process.c s s! | ] } t | t k Vq d S)N)typestr).0modr 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)zReturn 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 t k r, t d t j t j } | j | j t j \ } } t j \ } } | | | j
t j g } | | 7} zJ y t
j | | | | f SWn% t j | t j | Yn XWd t j | t j | XWd QRXd S)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 lenMAXFDS_TO_SEND
ValueErrorsocketAF_UNIXZconnectr ospiper r Zgetfdr Zsendfdsclose)r fdsZclientZparent_rchild_wchild_rZparent_wZallfdsr r r r 6 s&
z!ForkServer.connect_to_new_processc s | j t j | j d k r' d Sd d } | j r} d d h t j d } t f d d | j D } n i } t j t j
} t j d } | j
| t j | d
| j t j \ } } z yy | j | g } | | j | | j | f ;} t j } | g t j } | d | g 7} t j | | | }
Wn t j | Yn XWd t j | X| | _ | | _ Wd QRXWd QRXd S) zMake 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_pathsys_pathignorec 3 s- | ]# \ } } | k r | | f Vq d S)Nr )r xy) desired_keysr r r e s z,ForkServer.ensure_running.<locals>.<genexpr>r' i z-c)r r r r r r Zget_preparation_datadictitemsr& r' r Zarbitrary_addressZbindr( chmodZlistenr) filenoZget_executabler Z _args_from_interpreter_flagsZspawnv_passfdsr* r ) r cmddatalistenerZaddressalive_rZalive_wZ fds_to_passZexeargspidr )r3 r r S s>
(
zForkServer.ensure_runningN)__name__
__module__ __qualname__r r r
r r r r r r r s
r c 4 C sx | r d | k rN | d k rN d t j _ z t j | Wd t j ` Xx1 | D]) } y t | WqU t k
r} YqU XqU Wt j t j
t j t j t j
i } d d | j D } t j t j d | } t j } | j t _ | j | t j | j | t j xAy x$ d d | j D }
|
r+Pq+W| |
k rt j | d d
k syt t | |
k st | j d } d } t j d k r zP y t | | | | Wn2 t k
r
t! j" t! j# t! j$ j% Yn XWd t j& | XWd QRXWq%t' k
ra}
z |
j( t( j) k rO WYd d }
~
Xq%Xq%WWd QRXWd QRXd S) zRun forkserver.r NTc S s+ i | ]! \ } } t j | | | q Sr )signal)r sigvalr r r
<dictcomp> s zmain.<locals>.<dictcomp>r7 c S s g | ] \ } } | j q Sr )Zfileobj)r keyZeventsr r r
<listcomp> s zmain.<locals>.<listcomp>r r )*r Zcurrent_processZ _inheritingr Zimport_main_path
__import__ ImportErrorr Z _close_stdinrA SIGCHLDSIG_IGNSIGINTSIG_DFLr5 r& r' selectorsZDefaultSelectorZ getsockname _forkserverr registerZ
EVENT_READZselectr( readAssertionError
SystemExitZacceptfork
_serve_one Exceptionsys
excepthookexc_infostderrflush_exitOSErrorerrnoZ ECONNABORTED)Z listener_fdr; Zpreloadr. r/ modnamehandlersZ old_handlersr: ZselectorZrfdsscodeer r r main sV
rd c C s | j t j | x* | j D] \ } } t j | | q$ Wt j | t d } | j t | t k s| t | ^ } } t _
} t _ | t j
_ t | t j d t j k r d d l }
|
j t j | } t | | d S)Nr randomr )r* r( r5 rA r Zrecvfdsr$ r# rR rO r r r Z_semaphore_trackerZ_fdwrite_unsignedgetpidrW modulesre Zseedr _main) ra r: r; r` rB rC r+ r- r, Zstfdre rb r r r rU s
rU c C sq d } t j } xN t | | k r_ t j | | t | } | sR t d | | 7} q Wt j | d S)NrG zunexpected EOFr )UNSIGNED_STRUCTsizer# r( rQ EOFErrorZunpack)fdr9 Zlengthra r r r
read_unsigned s rn c C sZ t j | } xD | rU t j | | } | d k rB t d | | d } q Wd S)Nr zshould not get here)rj Zpackr( write RuntimeError)rm nmsgnbytesr r r rf s rf ) r^ r( rN rA r& ZstructrW r r r r r r r __all__r$ ZStructrj objectr rd rU rn rf rO r r
r r r r r r <module> s6 h= !