] @ 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 threadingLock_lock_preload_modules)self r //opt/alt/python35/lib64/python3.5/forkserver.py__init__! s
zForkServer.__init__c C s5 t d d | j D s( t d | | _ d S)Nc 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 strings)allr TypeError)r Z
modules_namesr r r r ( s z!ForkServer.set_forkserver_preloadc C s | j S)N)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)N z too many fds)r lenMAXFDS_TO_SEND
ValueErrorsocketAF_UNIXconnectr ospiper r Zgetfdr sendfdsclose)r fdsclientZparent_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) 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 arbitrary_addressbindr* chmodlistenr+ filenoZget_executabler _args_from_interpreter_flagsspawnv_passfdsr- r ) r cmddatalisteneraddressalive_rZalive_wZ fds_to_passZexeargspidr )r7 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 sH | 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 | f} t j P} | j t _ | j | t j | j | t j xy x$ d d | j D }
|
r+Pq+W| |
k rat | 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 QRXWq%t% k
r1}
z |
j& t& j' k r WYd d }
~
Xq%Xq%WWd QRXWd QRXd S)
Nr Tc S s+ i | ]! \ } } t j | | | q Sr )signal)r sigvalr r r
<dictcomp> s zmain.<locals>.<dictcomp>r> c S s g | ] \ } } | j q Sr )fileobj)r keyeventsr r r
<listcomp> s zmain.<locals>.<listcomp>r r )(r current_processZ _inheritingr Zimport_main_path
__import__ ImportErrorr _close_stdinrK SIGCHLDSIG_IGNSIGINTSIG_DFLr9 r' r( selectorsDefaultSelector getsockname _forkserverr register
EVENT_READselect
SystemExitacceptr* fork
_serve_one Exceptionsys
excepthookexc_infostderrflush_exitOSErrorerrnoZ ECONNABORTED)Z listener_fdrE Zpreloadr2 r3 modnamehandlersZ old_handlersrC selectorZrfdsscodeer r r main sR
ru c C s | j t j | x* | j D] \ } } t j | | q$ Wt j | t d } | j | ^ } } 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* r9 rK r recvfdsr% r^ r r r Z_semaphore_trackerZ_fdwrite_unsignedgetpidrg modulesrv seedr _main) rr rC rE rp rL rM r. r1 r0 Zstfdrv rs r r r re s
re c C sq d } t j } xN t | | k r_ t j | | t | } | sR t d | | 7} q Wt j | d S)N zunexpected EOFr )UNSIGNED_STRUCTsizer$ r* readEOFErrorunpack)fdrB lengthrr r r r
read_unsigned s r c C sZ t j | } xD | rU t j | | } | d k rB t d | | d } q Wd S)Nr zshould not get here)r~ packr* write RuntimeError)r nmsgnbytesr r r rx s rx ) rn r* r[ rK r' structrg r r r r r r r __all__r% Structr~ objectr ru re r rx r^ r r
r r r r r r <module> s6 h= !