File: //usr/lib/python3.5/multiprocessing/__pycache__/forkserver.cpython-35.pyc
Fa @ 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 0/usr/lib/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 sY | 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 t j
d | } t j } | j t _ | j | t j | j | t j xAy x$ d d | j D } | r Pq W| | k r`t j | d d k sZt t | | k srt | j d }
d } t j d k rzP y t |
| | | Wn2 t k
rt j t j t j! j" Yn XWd t j# | XWd QRXWqt$ k
rB} z | j% t% j& k r0 WYd d } ~ XqXqWWd QRXWd QRXd S)
zRun forkserver.r NTr7 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_stdinsignalSIGCHLDSIG_IGNr& r' selectorsZDefaultSelectorZ getsockname _forkserverr registerZ
EVENT_READZselectr( readAssertionError
SystemExitZacceptfork
_serve_one Exceptionsys
excepthookexc_infostderrflush_exitOSErrorerrnoZ ECONNABORTED)
Z listener_fdr; Zpreloadr. r/ modnamehandlerr: ZselectorZrfdsscodeer r r main sP
r_ 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 t j | } t | | d S)Nr randomr )r* r( rF rG r Zrecvfdsr$ r# rM rJ r r r Z_semaphore_trackerZ_fdwrite_unsignedgetpidrR modulesr` Zseedr _main)
r\ r: r; r[ r+ r- r, Zstfdr` r] r r r rP s
rP c C sq d } t j } xN t | | k r_ t j | | t | } | sR t d | | 7} q Wt j | d S)NrC zunexpected EOFr )UNSIGNED_STRUCTsizer# r( rL EOFErrorZunpack)fdr9 Zlengthr\ r r r
read_unsigned s ri c C sZ t j | } xD | rU t j | | } | d k rB t d | | d } q Wd S)Nr zshould not get here)re Zpackr( write RuntimeError)rh nmsgnbytesr r r ra s ra ) rY r( rI rF r& ZstructrR r r r r r r r __all__r$ ZStructre objectr r_ rP ri ra rJ r r
r r r r r r <module> s6 h6