3
\! @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddlm Z ddlm
Z
ddl m Z ddlm
Z
ddlmZ ddlmZ d d
d d gZd
ZejdZG dd deZdddZdd Zdd Zdd Ze ZejZej Z ej Z ej Z dS ) N )
connection)process) reduction)semaphore_tracker)spawn)utilensure_runningget_inherited_fdsconnect_to_new_processset_forkserver_preload Qc @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S )
ForkServerc C s. d | _ d | _d | _d | _tj | _dg| _d S )N__main__)_forkserver_address_forkserver_alive_fd_forkserver_pid_inherited_fds threadingZLock_lock_preload_modules)self r 2/usr/lib64/python3.6/multiprocessing/forkserver.py__init__! s
zForkServer.__init__c C s& t dd | jD s td|| _dS )z>Set list of module names to try to load in forkserver process.c s s | ]}t |tkV qd 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 tkr tdtjtj}|j| j tj \}}tj \}}||| j
t j g}||7 }z.r( i z-czCfrom multiprocessing.forkserver import main; main(%d, %d, %r, **%r)) r r r r r) waitpidWNOHANGr+ r r r r Zget_preparation_datadictitemsr' r( r Zarbitrary_addressZbindchmodZlistenr* filenoZget_executabler Z _args_from_interpreter_flagsZspawnv_passfds) r pidZstatuscmddatalistenerZaddressalive_rZalive_wZ fds_to_passZexeargsr )r4 r r T sL
zForkServer.ensure_runningN)__name__
__module__ __qualname__r r r
r r r r r r r s
r c 4 C s |rhd|kr8|dk r8dt j _ztj| W dt j `X x.|D ]&}y t| W q> tk
rb Y q>X q>W tj t j
t j t j t j i}dd |j
D }tjtj| dR}tj <} |j t_| j|tj | j|tj x yx dd | j D }
|
rP qW ||
kr$tj|d d
ks tt||
ks2t|j d l} d } tj d krzFyt | ||| W n. t k
r t j!t j" t j#j$ Y nX W dtj%| X W dQ R X W q t&k
r }
z|
j't'j(krւ W Y dd}
~
X qX qW W dQ R X W dQ R X dS ) zRun forkserver.r NTc S s i | ]\}}t j |||qS r )signal)r sigvalr r r
s zmain..)r: c S s g | ]\}}|j qS r )Zfileobj)r keyZeventsr r r
s zmain..r r ))r Zcurrent_processZ _inheritingr Zimport_main_path
__import__ ImportErrorr Z _close_stdinrD SIGCHLDSIG_IGNSIGINTr8 r' r( selectorsZDefaultSelectorZ getsockname _forkserverr registerZ
EVENT_READZselectr) readAssertionError
SystemExitZacceptfork
_serve_one Exceptionsys
excepthookexc_infostderrflush_exitOSErrorerrnoZ ECONNABORTED)Z listener_fdr? Zpreloadr/ r0 modnamehandlersZ old_handlersr> ZselectorZrfdsscodeer r r main sV
rf c C s |j tj | x |j D ]\}}tj|| q W tj| td }| j t|tks\t|^}}t _
} t _ | t j
_t|tj dtjkrdd l}
|
j tj|} t|| d S )Nr randomr )r+ r) r8 rD r Zrecvfdsr% r$ rT rQ r r r Z_semaphore_trackerZ_fdwrite_unsignedgetpidrY modulesrg Zseedr _main) rc r> r? rb rE rF r, r. r- Zstfdrg rd r r r rW s
rW c C sR d}t j}x8t||k rBtj| |t| }|s8td||7 }q W t j|d S )NrJ zunexpected EOFr )UNSIGNED_STRUCTsizer$ r) rS EOFErrorZunpack)fdr= Zlengthrc r r r
read_unsigned s rp c C s@ t j|}x0|r:tj| |}|dkr,td||d }q W d S )Nr zshould not get here)rl Zpackr) write RuntimeError)ro nmsgnbytesr r r rh s
rh )NN) r` r) rP rD r' ZstructrY r r r contextr r r r __all__r% ZStructrl objectr rf rW rp rh rQ r r
r r r r r r s6
s
=!