MINI MINI MANI MO
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 <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 tkr tdtjtj}|j| j tj \}}tj \}}||| j
t j g}||7 }z<yt
j|| ||fS tj| tj| Y nX W dtj| tj| X W dQ R X dS )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 7 s&
z!ForkServer.connect_to_new_processc s | j n tj | jdk rVtj| jtj\}}|s8dS tj| j d| _ d| _d| _d }| j
rddh t j d}t
fdd|j D }ni }tjtj}tjd }|j| tj|d
|j tj \}}ztyV|j |g} ||j || j
|f; }t j }
|
gtj } | d |g7 } tj|
| | }W n tj| Y nX W dtj| X || _ || _|| _W dQ R X W dQ R X dS )
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" | ]\}}| kr||fV qd S )Nr )r xy) desired_keysr r r o s z,ForkServer.ensure_running.<locals>.<genexpr>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
<dictcomp> s zmain.<locals>.<dictcomp>)r: c S s g | ]\}}|j qS r )Zfileobj)r keyZeventsr r r
<listcomp> s zmain.<locals>.<listcomp>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 <module> s6
s
=!
OHA YOOOO