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|| _d S )Nc 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 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 tkr tdtjtj}|j| j tj \}}tj \}}||| j
t j g}||7 }z<yt
j|| ||fS tj| tj| Y nX W d tj| tj| X W d Q R X d S )N z too many fds)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 | jd k rVtj| jtj\}}|s8d S 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 d tj| X || _ || _|| _W d Q R X W d Q R X d S ) 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|d k r8dt j _ztj| W d t 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 dkrzFyt | ||| W n. t k
rn t j t j t j!j" Y nX W d tj#| X W d Q R X W q t$k
r }
z|
j%t%j&kr W Y d d }
~
X qX qW W d Q R X W d Q R X d S )
Nr Tc 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_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 <module> s6
s
=!
All system for education purposes only. For more tools: Telegram @jackleet