Current File : //usr/local/lib/python3.8/multiprocessing/__pycache__/connection.cpython-38.pyc
U
qa{ @ s d dddgZ ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl
m Z m
Z
dd lmZ ejZz$ddlZdd
lmZmZmZmZ W n$ ek
r ejd kr dZY nX d Zd
Ze Z dZ dgZ e edrdZ e dg7 Z ejd krdZ e dg7 Z efddZ dd Z!dd Z"dd Z#dd Z$G dd d Z%erhG d d d e%Z&G d d d e%Z'G d!d de(Z)dOd"d Z*ejd krdPd$dZ+n
dQd%dZ+G d&d' d'e(Z,d(d) Z-ejd krG d*d+ d+e(Z.d,d- Z/d.Z0d/Z1d0Z2d1Z3d2d3 Z4d4d5 Z5G d6d7 d7e(Z6d8d9 Z7d:d; Z8G d<d= d=e)Z9d>d? Z:ejd krtd@dA Z;ej<ej=hZ>dRdBdZ?n,ddl@Z@e e@dCre@jAZBne@jCZBdSdDdZ?ejd krdEdF ZDdGdH ZEeFe'eD dIdJ ZGdKdL ZHeFe&eG n dMdF ZDdNdH ZEeFe'eD dS )TClientListenerPipewait N )util)AuthenticationErrorBufferTooShort) reduction)
WAIT_OBJECT_0WAIT_ABANDONED_0 WAIT_TIMEOUTINFINITEwin32i g 4@AF_INETAF_UNIXAF_PIPEc C s t | S Ntime monotonic)timeout r 6/usr/local/lib/python3.8/multiprocessing/connection.py
_init_timeout; s r c C s t | kS r r )tr r r _check_timeout> s r c C sX | dkr dS | dkr&t jdt dS | dkrLt jdt ttf ddS td d
S ) z?
Return an arbitrary free address for the given family
r ) localhostr r z listener-)prefixdirr z\\.\pipe\pyc-%d-%d- zunrecognized familyN) tempfilemktempr get_temp_dirosgetpidnext
_mmap_counter
ValueErrorfamilyr r r arbitrary_addressE s r+ c C sJ t jdkr | dkr td| t jdkrF| dkrFtt| sFtd| dS )zD
Checks if the family is valid for the current environment.
r r z Family %s is not recognized.r N)sysplatformr( hasattrsocketr) r r r _validate_familyS s
r0 c C sT t | tkrdS t | tkr*| dr*dS t | tks@t| rDdS td| dS )z]
Return the types of the address
This can be 'AF_INET', 'AF_UNIX', or 'AF_PIPE'
r z\\r r z address type of %r unrecognizedN)typetuplestr
startswithr is_abstract_socket_namespacer( )addressr r r address_type_ s r7 c @ s e Zd ZdZd+ddZdd Zdd Zd d
Zd d Zd
d Z e
dd Z e
dd Z e
dd Z
dd Zdd Zd,ddZd d Zd-d d Zd.d d!Zd"d# Zd/d%d&Zd'd( Zd)d* ZdS )0_ConnectionBaseNTc C s> | }|dk rtd|s(|s(td|| _|| _|| _d S )Nr zinvalid handlez6at least one of `readable` and `writable` must be True) __index__r( _handle _readable _writable)selfhandlereadablewritabler r r __init__u s z_ConnectionBase.__init__c C s | j d k r| d S r r: _closer= r r r __del__ s
z_ConnectionBase.__del__c C s | j d krtdd S )Nzhandle is closed)r: OSErrorrD r r r
_check_closed s
z _ConnectionBase._check_closedc C s | j stdd S )Nzconnection is write-only)r; rF rD r r r _check_readable s z _ConnectionBase._check_readablec C s | j stdd S )Nzconnection is read-only)r< rF rD r r r _check_writable s z _ConnectionBase._check_writablec C s" | j rd| _n| tdd S )NFzbad message length)r< r; closerF rD r r r _bad_message_length s z#_ConnectionBase._bad_message_lengthc C s
| j dkS )z True if the connection is closedNr: rD r r r closed s z_ConnectionBase.closedc C s | j S )z"True if the connection is readable)r; rD r r r r? s z_ConnectionBase.readablec C s | j S )z"True if the connection is writable)r< rD r r r r@ s z_ConnectionBase.writablec C s | | jS )z+File descriptor or handle of the connection)rG r: rD r r r fileno s z_ConnectionBase.filenoc C s$ | j dk r z | W 5 d| _ X dS )zClose the connectionNrB rD r r r rJ s
z_ConnectionBase.closer c C s | | t|}|jdkr.tt|}t|}|dk rFtd||k rVtd|dkrh|| }n&|dk rztdn|| |krtd| |||| dS )z,Send the bytes data from a bytes-like objectr r zoffset is negativezbuffer length < offsetNzsize is negativez buffer length < offset + size)rG rI
memoryviewitemsizebyteslenr( _send_bytes)r= bufoffsetsizemnr r r
send_bytes s"
z_ConnectionBase.send_bytesc C s$ | | | t| dS )zSend a (picklable) objectN)rG rI rS _ForkingPicklerdumpsr= objr r r send s z_ConnectionBase.sendc C sJ | | |dk r(|dk r(td| |}|dkrB| | S )z7
Receive bytes data as a bytes object.
Nr znegative maxlength)rG rH r( _recv_bytesrK getvalue)r= maxlengthrT r r r
recv_bytes s
z_ConnectionBase.recv_bytesc
C s | | t|}|j}|t| }|dk r>tdn||krNtd| }| }||| k rvt| |
d | ||| || | |W 5 Q R S Q R X dS )zq
Receive bytes data into a writeable bytes-like object.
Return the number of bytes read.
r znegative offsetzoffset too largeN) rG rH rO rP rR r( r_ tellr r` seekreadinto)r= rT rU rW rP bytesizeresultrV r r r recv_bytes_into s$
z _ConnectionBase.recv_bytes_intoc C s& | | | }t| S )z Receive a (picklable) object)rG rH r_ rZ loads getbuffer)r= rT r r r recv s z_ConnectionBase.recv c C s | | | |S )z/Whether there is any input available to be read)rG rH _pollr= r r r r poll s z_ConnectionBase.pollc C s | S r r rD r r r __enter__ s z_ConnectionBase.__enter__c C s | d S r rJ r= exc_type exc_valueexc_tbr r r __exit__ s z_ConnectionBase.__exit__)TT)r N)N)r )rl )__name__
__module__ __qualname__r: rA rE rG rH rI rK propertyrM r? r@ rN rJ rY r^ rb rh rk ro rp rv r r r r r8 r s.
r8 c @ sD e Zd ZdZdZejfddZdd Zddd Z d
d Z
d d
Z dS )PipeConnectionz
Connection class based on a Windows named pipe.
Overlapped I/O is used, so the handles must have been created
with FILE_FLAG_OVERLAPPED.
Fc C s || j d S r rL )r= Z _CloseHandler r r rC s zPipeConnection._closec C s t j| j|dd\}}zHz,|t jkrBt |jgdt}|tksBt W n |
Y nX W 5 |d\}}X |dks|t |t |kst d S )NT
overlappedFr ) _winapiZ WriteFiler: GetOverlappedResultERROR_IO_PENDINGWaitForMultipleObjectseventr r AssertionErrorcancelrR )r= rT overrZnwrittenwaitresr r r rS s
zPipeConnection._send_bytesNc C s2 | j rd| _ t S |d kr dnt|d}ztj| j|dd\}}d zHz,|tj krpt
|jgdt}|tksptW n | Y nX W 5 |d\}}|dkrt }|| | W S |tj
kr| || W S X W n: tk
r$ } z|jtjkrtn W 5 d }~X Y nX tdd S )NF Tr| r z.shouldn't get here; expected KeyboardInterrupt)_got_empty_messageioBytesIOminr~ ReadFiler: r writerj ZERROR_MORE_DATA_get_more_datar r r r r r r rF winerrorERROR_BROKEN_PIPEEOFError RuntimeError) r= maxsizeZbsizer r Znreadfr er r r r_ & s>
zPipeConnection._recv_bytesc C s. | j st| jd dkr dS tt| g|S )Nr T)r r~
PeekNamedPiper: boolr rn r r r rm F s
zPipeConnection._pollc C s | }t }|| t| jd }|dks6t|d k rVt|| |krV| tj
| j|dd\}}| d\}}|dkst||kst|| |S )Nr r Tr| ) rj r r r r~ r r: r rR rK r r )r= r r rT r leftr Zrbytesr r r r L s
z PipeConnection._get_more_data)N) rw rx ry __doc__r r~ CloseHandlerC rS r_ rm r r r r r r{ s
r{ c @ s| e Zd ZdZer,ejfddZejZ ej
Z ne j
fddZe jZ e jZ e fddZe fddZd d
Zdd d
Zdd Zd S )
Connectionzo
Connection class based on an arbitrary file descriptor (Unix only), or
a socket handle (Windows).
c C s || j d S r rL r= rC r r r rC c s zConnection._closec C s || j d S r rL r r r r rC h s c C s8 t |}|| j|}||8 }|dkr&q4||d }qd S Nr )rR r: )r= rT r remainingrX r r r _sendm s zConnection._sendc C sb t }| j}|}|dkr^|||}t|}|dkrJ||krBtntd|| ||8 }q|S )Nr z got end of file during message)r r r: rR r rF r )r= rV readrT r> r chunkrX r r r _recvv s
zConnection._recvc C s t |}|dkrHtdd}td|}| | | | | | n8td|}|dkrr| | | | n| || d S )Ni!i!Qi @ )rR structpackr )r= rT rX
pre_headerheaderr r r rS s
zConnection._send_bytesNc C s^ | d}td| \}|dkr@| d}td| \}|d k rT||krTd S | |S )N r r r )r r unpackr` )r= r rT rV r r r r_ s
zConnection._recv_bytesc C s t | g|}t|S r )r r )r= r rr r r rm s zConnection._poll)N)rw rx ry r r~ _multiprocessingZ closesocketrC r^ _writerk _readr$ rJ r r r r rS r_ rm r r r r r \ s
r c @ sR e Zd ZdZdddZdd Zdd Zed
d Zed d
Z dd Z
dd Z dS )r z
Returns a listener object.
This is a wrapper for a bound socket which is 'listening' for
connections, or for a Windows named pipe.
Nr c C sp |p|rt |pt}|p t|}t| |dkr>t||| _nt|||| _|d k rft|tsft d|| _
d S )Nr authkey should be a byte string) r7 default_familyr+ r0 PipeListener _listenerSocketListener
isinstancerQ TypeError_authkey)r= r6 r* backlogauthkeyr r r rA s zListener.__init__c C s> | j dkrtd| j }| jr:t|| j t|| j |S )zz
Accept a connection on the bound socket or named pipe of `self`.
Returns a `Connection` object.
Nzlistener is closed)r rF acceptr deliver_challengeanswer_challenge)r= cr r r r s
zListener.acceptc C s | j }|dk r d| _ | dS )zA
Close the bound socket or named pipe of `self`.
N)r rJ )r= listenerr r r rJ s zListener.closec C s | j jS r )r _addressrD r r r r6 s zListener.addressc C s | j jS r )r _last_acceptedrD r r r
last_accepted s zListener.last_acceptedc C s | S r r rD r r r rp s zListener.__enter__c C s | d S r rq rr r r r rv s zListener.__exit__)NNr N) rw rx ry r rA r rJ rz r6 r rp rv r r r r r s
c C sh |p
t | }t| |dkr&t| }nt| }|dk rHt|tsHtd|dk rdt|| t|| |S )z=
Returns a connection to the address of a `Listener`
r Nr ) r7 r0
PipeClient SocketClientr rQ r r r )r6 r* r r r r r r s
Tc C sj | r>t \}}|d |d t| }t| }n$t \}}t|dd}t|dd}||fS )L
Returns pair of connection objects at either end of a pipe
TFr@ r? )r/
socketpair setblockingr detachr$ pipe)duplexs1s2c1c2fd1fd2r r r r s
c
C s t d}| r*tj}tjtjB }tt }}ntj}tj}dt }}t||tjB tj B tj
tj B tj B d||tj
tj}t||dtjtjtjtj}t|tj dd tj|dd}|d\} }
|
dkstt|| d} t|| d} | | fS ) r r r r NTr| r r )r+ r~ PIPE_ACCESS_DUPLEX GENERIC_READ
GENERIC_WRITEBUFSIZEZPIPE_ACCESS_INBOUNDCreateNamedPipeFILE_FLAG_OVERLAPPED FILE_FLAG_FIRST_PIPE_INSTANCEPIPE_TYPE_MESSAGEPIPE_READMODE_MESSAGE PIPE_WAITNMPWAIT_WAIT_FOREVERNULL
CreateFile
OPEN_EXISTINGSetNamedPipeHandleStateConnectNamedPiper r r{ )
r r6 ZopenmodeaccessZobsizeZibsizeZh1Zh2r} _r r r r r r r sV
c @ s* e Zd ZdZd
ddZdd Zdd Zd S ) r zO
Representation of a socket which is bound to an address and listening
r c C s t tt || _zRtjdkr2| jt jt jd | jd | j | | j
| | j | _ W n t
k
r | j Y nX || _d | _|dkrt|stj| tj|fdd| _nd | _d S )Nposixr Tr r args exitpriority)r/ getattr_socketr$ name
setsockopt
SOL_SOCKET SO_REUSEADDRr bindlisten getsocknamer rF rJ _familyr r r5 Finalizeunlink_unlink)r= r6 r* r r r r rA G s0
zSocketListener.__init__c C s& | j \}| _|d t| S )NT)r r r r r r r= sr r r r ` s
zSocketListener.acceptc C s0 z| j W 5 | j }|d k r*d | _ | X d S r )r r rJ )r= r r r r rJ e s zSocketListener.closeN)r )rw rx ry r rA r rJ r r r r r C s
r c
C sP t | }ttt|.}|d || t| W 5 Q R S Q R X dS )zO
Return a connection object connected to the socket given by `address`
TN)r7 r/ r r connectr r )r6 r* r r r r r o s
r c @ s8 e Zd ZdZd ddZd
ddZdd Zed
d ZdS )r z0
Representation of a named pipe
Nc C sL || _ | jddg| _d | _td| j tj| tj| j| j fdd| _ d S )NT)firstz listener created with address=%rr r )
r _new_handle
_handle_queuer r sub_debugr r _finalize_pipe_listenerrJ )r= r6 r r r r rA s
zPipeListener.__init__Fc
C sH t jt jB }|r|t jO }t | j|t jt jB t jB t j t
t
t j t j S r )
r~ r r r r r r r r ZPIPE_UNLIMITED_INSTANCESr r r )r= r flagsr r r r s
zPipeListener._new_handlec
C s | j | | j d}ztj|dd}W n0 tk
r^ } z|jtjkrN W 5 d }~X Y n\X z<zt |j gdt
}W n | t| Y nX W 5 | d\}}|dkst
X t|S )Nr Tr| F)r appendr popr~ r rF r Z
ERROR_NO_DATAr r r r r r r r{ )r= r> r r r r resr r r r s(
zPipeListener.acceptc C s$ t d| | D ]}t| qd S )Nz closing listener with address=%r)r r r~ r )queuer6 r> r r r r s z$PipeListener._finalize_pipe_listener)N)F) rw rx ry r rA r r staticmethodr r r r r r s
r c
C s t }z6t| d t| tjtjB dtjtjtjtj}W q t k
rz } z |j
tj tj fksht
|rj W 5 d}~X Y qX qq t|tjdd t|S )zU
Return a connection object connected to the pipe given by `address`
r N)r r~ Z
WaitNamedPiper r r r r r rF r ZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYr r r r{ )r6 r hr r r r r s8
r s #CHALLENGE#s #WELCOME#s #FAILURE#c C s dd l }t|ts$tdt|tt}| t
| | ||d }|
d}||krl| t n| t tdd S )Nr Authkey must be bytes, not {0!s}md5 zdigest received was wrong)hmacr rQ r( formatr1 r$ urandomMESSAGE_LENGTHrY CHALLENGEnewdigestrb WELCOMEFAILUREr
connectionr r messager responser r r r s
r c C s dd l }t|ts$tdt|| d}|d tt tksNt d| |ttd }|
||d }| | | d}|t
krtdd S )Nr r r
z message = %rr zdigest sent was rejected)r r rQ r( r r1 rb rR r r r r rY r r r r r r r s
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ConnectionWrapperc C s6 || _ || _|| _dD ]}t||}t| || qd S )N)rN rJ ro rb rY )_conn_dumps_loadsr setattr)r= connr[ ri attrr] r r r rA s
zConnectionWrapper.__init__c C s | |}| j| d S r )r r rY )r= r] r r r r r^ s
zConnectionWrapper.sendc C s | j }| |S r )r rb r r r r r rk s
zConnectionWrapper.recvN)rw rx ry rA r^ rk r r r r r s r c C s t | fd d d ddS )Nr utf-8) xmlrpclibr[ encode)r] r r r
_xml_dumps s r" c C s t | d\\}}|S )Nr )r ri decode)r r] methodr r r
_xml_loads s r% c @ s e Zd Zdd ZdS ) XmlListenerc C s" dd l ma t| }t|ttS r )
xmlrpc.clientclientr r r r r" r% r\ r r r r s
zXmlListener.acceptN)rw rx ry r r r r r r& s r& c O s dd l ma tt| |ttS r )r' r( r r r r" r% )r kwdsr r r XmlClient s r* c C s t | }g }|rt|d|}|tkr*qn\t| krFtt| k rTn n
|t8 }n2t| krptt| k r~n n
|t8 }ntd||| ||d d }d}q |S )NFzShould not get herer r ) listr~ r r
r rR r r r )Zhandlesr Lreadyr r r r _exhaustive_wait% s
r. c
s^ |dkrt }n |dk r d}nt|d d }t| } i g }t t }z@| D ]&}zt|d} W n tk
r || < Y qPX zt | dd\}}W n8 tk
r } zd|j }}|tkrƂ W 5 d}~X Y nX |t jkr|| ||j < qP|rjt dd d krjz|d
\}}W n* tk
rP } z
|j}W 5 d}~X Y nX |sjt
|drjd|_ | d}qPt |}W 5 |D ]}| q|D ]}z|d\}}W n6 tk
r } z|j}|tkr W 5 d}~X Y nX |t j
kr|j } | |dkrt
|drd|_qX fd d |D fd
d| D S )
Wait till an object in object_list is ready/readable.
Returns list of those objects in object_list which are ready/readable.
Nr r g ?Tr rN ) r0 Fc 3 s | ]} | V qd S r r ).0r )waithandle_to_objr r <genexpr> s zwait.<locals>.<genexpr>c s g | ]}| kr|qS r r )r2 o)
ready_objectsr r
<listcomp> s wait.<locals>.<listcomp>)r intr+ setr r rF r
_ready_errorsr~ ZERROR_OPERATION_ABORTEDr addr. r r AttributeErrorr9 r r r r, Zgetwindowsversionr. keysupdate)
object_listr Zov_listZ
ready_handlesr r r r r5 rN r )r6 r3 r r ; sh
PollSelectorc
C s t }| D ]}||tj q |dk r4t | }||}|r\dd |D W 5 Q R S |dk r4|t }|dk r4|W 5 Q R S q4W 5 Q R X dS )r/ Nc S s g | ]\}}|j qS r )fileobj)r2 keyeventsr r r r7 s r8 r )
_WaitSelectorregister selectors
EVENT_READr r select)r@ r selectorr] deadliner- r r r r s
c
C sZ | }t|tjtj6}ddlm} ||}t|| j | j
ffW 5 Q R S Q R X d S )Nr )resource_sharer) rN r/ fromfdr SOCK_STREAMr rL Z DupSocketrebuild_connectionr? r@ )r r> r rL dsr r r reduce_connection s
rQ c C s | }t| ||S r r r )rP r? r@ sockr r r rO s rO c C sB | j r tjnd| jrtjndB }t| |}t|| j | jffS r ) r? r~ ZFILE_GENERIC_READr@ ZFILE_GENERIC_WRITEr
Z DupHandlerN rebuild_pipe_connection)r r dhr r r reduce_pipe_connection s
rV c C s | }t|||S r )r r{ )rU r? r@ r> r r r rT s rT c C s t | }t|| j| jffS r )r
DupFdrN rO r? r@ )r dfr r r rQ s c C s | }t|||S r rR )rX r? r@ fdr r r rO s )NN)T)T)N)N)I__all__r r$ r, r/ r r r! itertoolsr r r r r contextr
ForkingPicklerrZ r~ r r r
r ImportErrorr- r CONNECTION_TIMEOUTcountr' r familiesr. r r r+ r0 r7 r8 r{ r objectr r r r r r r r r r r r r r r" r% r& r* r. r ZERROR_NETNAME_DELETEDr; r rG rA rE SelectSelectorrQ rO rF rV rT r r r r <module>
s
PT=
,, 8 P
Mini Shell