Yf @sdddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZd d d d gZd ZejdZGdddeZddddZddZddZddZeZejZejZejZejZdS)N) connection)process) reduction)semaphore_tracker)spawn)utilensure_runningget_inherited_fdsconnect_to_new_processset_forkserver_preloadQc@sLeZdZddZddZddZddZd d Zd S) ForkServercCs:d|_d|_d|_tj|_dg|_dS)N__main__)_forkserver_address_forkserver_alive_fd_inherited_fds threadingZLock_lock_preload_modules)selfr?/opt/alt/python35/lib64/python3.5/multiprocessing/forkserver.py__init__!s    zForkServer.__init__cCs5tdd|jDs(td||_dS)z>Set list of module names to try to load in forkserver process.css!|]}t|tkVqdS)N)typestr).0modrrr *sz4ForkServer.set_forkserver_preload..z&module_names must be a list of stringsN)allr TypeError)rZ modules_namesrrrr (s z!ForkServer.set_forkserver_preloadcCs|jS)zReturn 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 |jt|dtkr,tdtjtj}|j|jtj \}}tj \}}|||j t j g}||7}zJyt j||||fSWn%tj|tj|YnXWdtj|tj|XWdQRXdS)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_UNIXZconnectrospiperrZgetfdrZsendfdsclose)rfdsZclientZparent_rchild_wchild_rZparent_wZallfdsrrrr 6s&        z!ForkServer.connect_to_new_processc s|jtj|jdk r'dSdd}|jr}ddhtjd}tfdd|jD}ni}t j t j  }t j d }|j |tj|d |jtj\}}zyy|j|g}||j||j|f;}tj}|gtj} | d |g7} tj|| |} Wntj|YnXWdtj|X||_||_WdQRXWdQRXdS) 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_pathignorec3s-|]#\}}|kr||fVqdS)Nr)rxy) desired_keysrrres z,ForkServer.ensure_running..r'iz-c)rrr rrrZget_preparation_datadictitemsr&r'rZarbitrary_addressZbindr(chmodZlistenr)filenoZget_executablerZ_args_from_interpreter_flagsZspawnv_passfdsr*r) rcmddatalistenerZaddressalive_rZalive_wZ fds_to_passZexeargspidr)r3rr Ss>    (       zForkServer.ensure_runningN)__name__ __module__ __qualname__rr r r r rrrrrs     rc4CsH|rd|krN|dk rNdtj_ztj|Wdtj`Xx1|D])}yt|WqUtk r}YqUXqUWtjt j t j t j t j i}dd|jD}tjtjd|f}tjP} |jt_| j|tj| j|tjxyx$dd| jD} | r+Pq+W|| krat|jd } d } tjd krzPyt| |||Wn2tk rtj tj!tj"j#YnXWdtj$| XWdQRXWq%t%k r1} z| j&t&j'krWYdd} ~ Xq%Xq%WWdQRXWdQRXdS) zRun forkserver.rNTcSs+i|]!\}}tj|||qSr)signal)rsigvalrrr s zmain..r7cSsg|]\}}|jqSr)Zfileobj)rkeyZeventsrrr s zmain..rr)(rZcurrent_processZ _inheritingrZimport_main_path __import__ ImportErrorrZ _close_stdinrASIGCHLDSIG_IGNSIGINTSIG_DFLr5r&r' selectorsZDefaultSelectorZ getsockname _forkserverrregisterZ EVENT_READZselect SystemExitZacceptr(fork _serve_one Exceptionsys excepthookexc_infostderrflush_exitOSErrorerrnoZ ECONNABORTED)Z listener_fdr;Zpreloadr.r/modnamehandlersZ old_handlersr:ZselectorZrfdsscodeerrrmainsR         rac Cs|jtj|x*|jD]\}}tj||q$Wtj|td}|j|^}}t_} t_ | t j _ t |tjdtjkrddl} | jtj|} t || dS)Nrrandomr)r*r(r5rArZrecvfdsr$rNrrrZ_semaphore_trackerZ_fdwrite_unsignedgetpidrTmodulesrbZseedr_main) r^r:r;r]rBrCr+r-r,Zstfdrbr_rrrrRs      rRcCsqd}tj}xNt||kr_tj||t|}|sRtd||7}qWtj|dS)Nzunexpected EOFr)UNSIGNED_STRUCTsizer#r(readEOFErrorZunpack)fdr9Zlengthr^rrr read_unsigneds  rmcCsZtj|}xD|rUtj||}|dkrBtd||d}qWdS)Nrzshould not get here)rhZpackr(write RuntimeError)rlnmsgnbytesrrrrcs    rc)r[r(rMrAr&ZstructrTrrrrrrr__all__r$ZStructrhobjectrrarRrmrcrNr r r r rrrrs6          h= !