e f ! @ 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/python34/lib64/python3.4/multiprocessing/forkserver.py__init__! s
zForkServer.__init__c C s8 t d d | j D s+ t d n | | _ 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 * s z4ForkServer.set_forkserver_preload..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 n 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 QXd 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 d 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 QXWd QXd 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..r' i d 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
< C s | r d | k rQ | d k rQ d t j _ z t j | Wd t j ` Xn x4 | D]) } y t | WqX t k
r YqX XqX Wn t j d k r y# t j j t
t j t _ Wq t
t f k
r Yq Xn t j t j t j } t j t j d | } t j } | j t _ | j | t j | j | t j x]yx' d d | j D } | rXPqXqXW| | k rt j | d d k st t n | | k st | j d }
d } t j d k rSzP y t! |
| | | Wn2 t" k
r=t j# t j$ t j% j& Yn XWd t j' | Xn Wd QXWqRt( k
rnYqRt
k
r} z | j) t) j* k r n WYd d } ~ XqRXqRWWd QXWd QXd S)
zRun forkserver.r NTr8 c S s g | ] \ } } | j q Sr )Zfileobj)r keyZeventsr r r
s zmain..r r )+r Zcurrent_processZ _inheritingr Zimport_main_path
__import__ ImportErrorsysstdinr* openr( devnullOSErrorr% signalSIGCHLDSIG_IGNr& r' selectorsZDefaultSelectorZ getsockname _forkserverr registerZ
EVENT_READZselectreadAssertionError
SystemExitZacceptfork
_serve_one Exception
excepthookexc_infostderrflush_exitInterruptedErrorerrnoZ ECONNABORTED)
Z listener_fdr< Zpreloadr. r/ modnamehandlerr; ZselectorZrfdsscodeer r r main s^
rd c
C s | j t j | t j t j | t j | t d } | j t | t k sb t | ^ } } t _
} t _ | t j
_ t | t j d t j k r d d l } | j n t j | } t | | d S)Nr randomr )r* r( rL rM r Zrecvfdsr$ r# rS rP r r r Z_semaphore_trackerZ_fdwrite_unsignedgetpidrG modulesre Zseedr _main)
ra r; r< r` r+ r- r, Zstfdre rb r r r rV s
rV c C s d } t j } xr t | | k r x: y t j | | t | } Wn t k
r[ Yq' XPq' W| sv t d n | | 7} q Wt j | d S)NrD zunexpected EOFr )UNSIGNED_STRUCTsizer# r( rR r] EOFErrorZunpack)fdr: Zlengthra r r r
read_unsigned s
rn c C s~ t j | } xh | ry x0 y t j | | } Wn t k
rE Yq XPq W| d k rf t d n | | d } q Wd S)Nr zshould not get here)rj Zpackr( writer] RuntimeError)rm nmsgnbytesr r r rf s
rf ) r^ r( rO rL r& ZstructrG r r r r r r r __all__r$ ZStructrj objectr rd rV rn rf rP r r
r r r r r r s6 h>