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| d,}tj } |j t_| j|tj | j|tj xyx dd | j D }
|
rP qW ||
kr t|j d l} d
} tj d krzFyt | ||| W n. t k
rn 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_READZselect
SystemExitZacceptr) fork
_serve_one Exceptionsys
excepthookexc_infostderrflush_exitOSErrorerrnoZ ECONNABORTED)Z listener_fdr? Zpreloadr/ r0 modnamehandlersZ old_handlersr> ZselectorZrfdsscodeer r r main sR
rc c C s |j tj | x |j D ]\}}tj|| q W tj| td }| j |^}}t_} t_ | t
j _ t
|tj dtjkrdd l}
|
j tj|} t
|| d S )Nr randomr )r+ r) r8 rD r Zrecvfdsr% rP r r r Z_semaphore_trackerZ_fdwrite_unsignedgetpidrV modulesrd Zseedr _main) r` r> r? r_ rE rF r, r. r- Zstfdrd ra r r r rT s
rT c C sR d}t j}x8t||k rBtj| |t| }|s8td||7 }q W t j|d S )N zunexpected EOFr )UNSIGNED_STRUCTsizer$ r) readEOFErrorZunpack)fdr= Zlengthr` r r r
read_unsigned s ro c C s@ t j|}x0|r:tj| |}|dkr,td||d }q W d S )Nr zshould not get here)rj Zpackr) write RuntimeError)rn nmsgnbytesr r r re s
re )NN) r] r) rO rD r' ZstructrV r r r contextr r r r __all__r% ZStructrj objectr rc rT ro re rP r r
r r r r r r s6
s
=!