File: //usr/lib/python3.5/multiprocessing/__pycache__/connection.cpython-35.pyc
Fax @ sI d d d d g Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
m Z d d l
m Z d d l
m
Z
m Z d d
l m Z y2 d d l Z d d l m Z m Z m Z m Z Wn* e k
r"e j d k r d Z Yn Xd
Z d Z e j Z d Z d g Z e e d rld Z e d g 7Z e j d k rd Z e d g 7Z e d d Z d d Z d d Z d d Z! d d Z" Gd d d Z# e rGd d d e# Z$ Gd d! d! e# Z% Gd" d d e& Z' d d d# d Z( e j d k r[d$ d% d Z) n d$ d& d Z) Gd' d( d( e& Z* d) d* Z+ e j d k rGd+ d, d, e& Z, d- d. Z- d/ Z. d0 Z/ d1 Z0 d2 Z1 d3 d4 Z2 d5 d6 Z3 Gd7 d8 d8 e& Z4 d9 d: Z5 d; d< Z6 Gd= d> d> e' Z7 d? d@ Z8 e j d k r|dA dB Z9 e j: e j; h Z< d dC d Z= n? d d l> Z> e e> dD re> j? Z@ n e> jA Z@ d dE d Z= e j d k r dF dG ZB dH dI ZC e jD e% eB dJ dK ZE dL dM ZF e jD e$ eE n( dN dG ZB dO dI ZC e jD e% eB d S)PClientListenerPipewait N ) reduction)util)AuthenticationErrorBufferTooShort)ForkingPickler)
WAIT_OBJECT_0WAIT_ABANDONED_0 WAIT_TIMEOUTINFINITEwin32i g 4@AF_INETAF_UNIXAF_PIPEc C s t j | S)N)time)timeout r 0/usr/lib/python3.5/multiprocessing/connection.py
_init_timeout; s r c C s t j | k S)N)r )tr r r _check_timeout> s r c C s | d k r d
S| d k r8 t j d d d t j S| d k rp t j d d t j t t f d d
St d d S)z?
Return an arbitrary free address for the given family
r localhostr r prefixz listener-dirr z\\.\pipe\pyc-%d-%d- zunrecognized familyN)r r ) tempfileZmktempr Z get_temp_dirosgetpidnext
_mmap_counter
ValueError)familyr r r arbitrary_addressE s r& c C si t j d k r+ | d k r+ t d | t j d k re | d k re t t | se t d | d S)zD
Checks if the family is valid for the current environment.
r r z Family %s is not recognized.r N)sysplatformr$ hasattrsocket)r% r r r _validate_familyS s
r+ c C se t | t k r d St | t k r; | j d r; d St | t k rQ d St d | d S)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$ )addressr r r address_type_ s !r1 c @ s! e Z d Z d Z d d d d Z d d Z d d Z d d
Z d d Z d
d Z e
d d Z e
d d Z e
d d Z
d d Z d d Z d d d d Z d d Z d d d Z d d d! Z d" d# Z d$ d% d& Z d' d( Z d) d* Z d S)+_ConnectionBaseNTc C s] | j } | d k r$ t d | r> | r> t d | | _ | | _ | | _ 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 | j d S)N)r4 _close)r7 r r r __del__ s z_ConnectionBase.__del__c C s | j d k r t d d S)Nzhandle is closed)r4 OSError)r7 r r r
_check_closed s z _ConnectionBase._check_closedc C s | j s t d d S)Nzconnection is write-only)r5 r> )r7 r r r _check_readable s z _ConnectionBase._check_readablec C s | j s t d d S)Nzconnection is read-only)r6 r> )r7 r r r _check_writable s z _ConnectionBase._check_writablec C s/ | j r d | _ n
| j t d d S)NFzbad message length)r6 r5 closer> )r7 r r r _bad_message_length s
z#_ConnectionBase._bad_message_lengthc C s
| j d k S)z True if the connection is closedN)r4 )r7 r r r closed s z_ConnectionBase.closedc C s | j S)z"True if the connection is readable)r5 )r7 r r r r9 s z_ConnectionBase.readablec C s | j S)z"True if the connection is writable)r6 )r7 r r r r: s z_ConnectionBase.writablec C s | j | j S)z+File descriptor or handle of the connection)r? r4 )r7 r r r fileno s
z_ConnectionBase.filenoc C s. | j d k r* z | j Wd d | _ Xd S)zClose the connectionN)r4 r< )r7 r r r rB s z_ConnectionBase.closer c C s | j | j t | } | j d k rA t t | } t | } | d k re t d | | k r} t d | d k r | | } n7 | d k r t d n | | | k r t d | j | | | | d S)z,Send the bytes data from a bytes-like objectr r zoffset is negativezbuffer length < offsetNzsize is negativez buffer length < offset + size)r? rA
memoryviewitemsizebyteslenr$ _send_bytes)r7 bufoffsetsizemnr r r
send_bytes s"
z_ConnectionBase.send_bytesc C s. | j | j | j t j | d S)zSend a (picklable) objectN)r? rA rJ r dumps)r7 objr r r send s
z_ConnectionBase.sendc C sg | j | j | d k r8 | d k r8 t d | j | } | d k r] | j | j S)z7
Receive bytes data as a bytes object.
Nr znegative maxlength)r? r@ r$ _recv_bytesrC getvalue)r7 Z maxlengthrK r r r
recv_bytes s
z_ConnectionBase.recv_bytesc C s | j | j t | } | j } | t | } | d k rW t d n | | k ro t d | j } | j } | | | k r t | j | j
d | j | | | | | | | SWd QRXd S)zq
Receive bytes data into a writeable bytes-like object.
Return the number of bytes read.
r znegative offsetzoffset too largeN) r? r@ rF rG rI r$ rT tellr
rU seekreadinto)r7 rK rL rN rG ZbytesizeresultrM r r r recv_bytes_into s"
z _ConnectionBase.recv_bytes_intoc C s3 | j | j | j } t j | j S)z Receive a (picklable) object)r? r@ rT r loads getbuffer)r7 rK r r r recv s
z_ConnectionBase.recvg c C s! | j | j | j | S)z/Whether there is any input available to be read)r? r@ _poll)r7 r r r r poll s
z_ConnectionBase.pollc C s | S)Nr )r7 r r r __enter__ s z_ConnectionBase.__enter__c C s | j d S)N)rB )r7 exc_type exc_valueexc_tbr r r __exit__ s z_ConnectionBase.__exit__)__name__
__module__ __qualname__r4 r; r= r? r@ rA rC propertyrD r9 r: rE rB rP rS rV r[ r^ r` ra re r r r r r2 r s(
r2 c @ sa e Z d Z d Z d Z e j d d Z d d Z d d d Z d
d Z
d d
Z d S)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)N)r4 )r7 Z _CloseHandler r r r< s zPipeConnection._closec C s t j | j | d d \ } } z\ y@ | t j k rc t j | j g d t } | t k sc t Wn | j Yn XWd | j
d \ } } X| d k s t | t | k s t d S)N
overlappedTFr ) _winapiZ WriteFiler4 ERROR_IO_PENDINGWaitForMultipleObjectseventr r AssertionErrorcancelGetOverlappedResultrI )r7 rK overrwaitresZnwrittenr r r rJ s !
zPipeConnection._send_bytesNc C s{ | j r d | _ t j S| d k r. d n t | d } y t j | j | d d \ } } z\ y@ | t j k r t j | j g d t
} | t k s t Wn | j
Yn XWd | j d \ } } | d k rt j } | j | j | S| t j k r#| j | | SXWnC t k
rj} z# | j t j k rUt n WYd d } ~ Xn Xt d d S)NF rk Tr z.shouldn't get here; expected KeyboardInterrupt)_got_empty_messageioBytesIOminrl ReadFiler4 rm rn ro r r rp rq rr writer] ZERROR_MORE_DATA_get_more_datar> winerrorERROR_BROKEN_PIPEEOFError RuntimeError) r7 maxsizeZbsizers rt ru Znreadfer r r rT & s6
!
zPipeConnection._recv_bytesc C s? | j s% t j | j d d k r) d St t | g | S)Nr T)rw rl
PeekNamedPiper4 boolr )r7 r r r r r_ F s zPipeConnection._pollc C s | j } t j } | j | t j | j d } | d k sM t | d k ry t | | | k ry | j t j
| j | d d \ } } | j d \ } } | d k s t | | k s t | j | j | S)Nr r rk T) r] rx ry r| rl r r4 rp rI rC r{ rr )r7 rs r rK r leftrt Zrbytesr r r r} L s
"
!z PipeConnection._get_more_data) rf rg rh __doc__rw rl CloseHandler< rJ rT r_ r} r r r r rj s rj c @ s e Z d Z d Z e r? e j d d Z e j Z e j
Z n$ e j
d d Z e j Z e j Z e d d Z e d d Z d d
Z d d d
Z d d Z d S)
Connectionzo
Connection class based on an arbitrary file descriptor (Unix only), or
a socket handle (Windows).
c C s | | j d S)N)r4 )r7 r< r r r r< c s zConnection._closec C s | | j d S)N)r4 )r7 r< r r r r< h s c C sP t | } x= | | j | } | | 8} | d k r8 P| | d } q Wd S)Nr )rI r4 )r7 rK r| remainingrO r r r _sendm s
zConnection._sendc C s t j } | j } | } xo | d k r | | | } t | } | d k rr | | k rf t n t d | j | | | 8} q W| S)Nr z got end of file during message)rx ry r4 rI r r> r| )r7 rM readrK r8 r chunkrO r r r _recvv s
zConnection._recvc C s\ t | } t j d | } | d k rG | j | | j | n | j | | d S)Nz!ii @ )rI structZpackr )r7 rK rO headerr r r rJ s
zConnection._send_bytesNc C sS | j d } t j d | j \ } | d k rF | | k rF d S| j | S)N z!i)r r ZunpackrU )r7 r rK rM r r r rT s
zConnection._recv_bytesc C s t | g | } t | S)N)r r )r7 r rr r r r_ s zConnection._poll)rf rg rh r rl _multiprocessingZ closesocketr< rS Z_writer^ Z_readr rB r| r r r rJ rT r_ r r r r r \ s r c @ s e Z d Z d Z d d d d d d Z d d Z d d Z e d
d Z e d d Z d
d Z
d d Z d S)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 s | p | r t | p t } | p- t | } t | | d k r[ t | | | _ n t | | | | _ | d k r t | t r t d | | _
d S)Nr z authkey should be a byte string) r1 default_familyr& r+ PipeListener _listenerSocketListener
isinstancerH TypeError_authkey)r7 r0 r% backlogauthkeyr r r r; s
zListener.__init__c C sW | j d k r t d | j j } | j rS 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 r> acceptr deliver_challengeanswer_challenge)r7 cr r r r s zListener.acceptc C s, | j } | d k r( d | _ | j d S)zA
Close the bound socket or named pipe of `self`.
N)r rB )r7 Zlistenerr r r rB s zListener.closec C s
| j j S)N)r _address)r7 r r r <lambda> s zListener.<lambda>c C s
| j j S)N)r _last_accepted)r7 r r r r s c C s | S)Nr )r7 r r r ra s zListener.__enter__c C s | j d S)N)rB )r7 rb rc rd r r r re s zListener.__exit__) rf rg rh r r; r rB ri r0 Z
last_acceptedra re r r r r r s c C s | p t | } t | | d k r7 t | } n t | } | d k rk t | t rk t d | d k r t | | t | | | S)z=
Returns a connection to the address of a `Listener`
r Nz authkey should be a byte string) r1 r+
PipeClient SocketClientr rH r r r )r0 r% r r r r r r s
Tc C s | rY t j \ } } | j d | j d t | j } t | j } n6 t j \ } } t | d d } t | d d } | | f S)zL
Returns pair of connection objects at either end of a pipe
Tr: Fr9 )r* Z
socketpair setblockingr detachr pipe)duplexs1s2c1c2Zfd1Zfd2r r r r s
c
C sS t d } | r; t j } t j t j B} t t } } n t j } t j } d t } } t j | | t j Bt j Bt j
t j Bt j Bd | | t j
t j } t j | | d t j t j t j t j } t j | t j d d t j | d d } | j d \ } }
|
d k s%t t | d | } t | d | } | | f S) zL
Returns pair of connection objects at either end of a pipe
r r r Nrk Tr: r9 )r& rl 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_EXISTINGSetNamedPipeHandleStateConnectNamedPiperr rp rj )
r r0 ZopenmodeaccessZobsizeZibsizeZh1Zh2rk _rt r r r r r r s4
c @ s= e Z d Z d Z d d d Z d d Z d d Z d S)
r zO
Representation of a socket which is bound to an address and listening
r c C s t j t t | | _ yq t j d k rI | j j t j t j d | j j d | j j | | j j
| | j j | _ Wn" t
k
r | j j Yn X| | _ d | _ | d k r t j | t j d | f d d | _ n d | _ d S)Nposixr Tr args exitpriorityr )r* getattr_socketr nameZ
setsockoptZ
SOL_SOCKETZ SO_REUSEADDRr ZbindZlistenZ getsocknamer r> rB Z_familyr r Finalizeunlink_unlink)r7 r0 r% r r r r r; 8 s$
$zSocketListener.__init__c C s5 | j j \ } | _ | j d t | j S)NT)r r r r r r )r7 sr r r r P s
zSocketListener.acceptc C s> z | j j Wd | j } | d k r9 d | _ | Xd S)N)r rB r )r7 r r r r rB U s zSocketListener.closeN)rf rg rh r r; r rB r r r r r 4 s r c C s\ t | } t j t t | 1 } | j d | j | t | j SWd QRXd S)zO
Return a connection object connected to the socket given by `address`
TN)r1 r* r r Zconnectr r )r0 r% r r r r r _ s
r c @ sR e Z d Z d Z d d d Z d d d Z d d Z e d
d Z d S) r z0
Representation of a named pipe
Nc C sq | | _ | j d d g | _ d | _ t j d | j t j | t j d | j | j f d d | _ d S)NfirstTz listener created with address=%rr r r )
r _new_handle
_handle_queuer r sub_debugr r _finalize_pipe_listenerrB )r7 r0 r r r r r; s s zPipeListener.__init__Fc C sb t j t j B} | r# | t j O} t j | j | t j t j Bt j Bt j t
t
t j t j S)N)
rl r r r r r r r r ZPIPE_UNLIMITED_INSTANCESr r r )r7 r flagsr r r r ~ s
zPipeListener._new_handlec C s | j j | j | j j d } y t j | d d } Wn: t k
r} } z | j t j k rk WYd d } ~ Xnt XzH y t j | j
g d t } Wn" | j t j
| Yn XWd | j d \ } } | d k s t Xt | S)Nr rk TF)r appendr poprl r r> r~ Z
ERROR_NO_DATArn ro r rq r rr rp rj )r7 r8 rs r resr rt r r r r s"
zPipeListener.acceptc C s2 t j d | x | D] } t j | q Wd S)Nz closing listener with address=%r)r r rl r )Zqueuer0 r8 r r r r s
z$PipeListener._finalize_pipe_listener) rf rg rh r r; r r staticmethodr r r r r r o s
r c C s t } x yK t j | d t j | t j t j Bd t j t j t j t j } WnO t k
r } z/ | j
t j t j f k s t
| r WYd d } ~ Xq XPq W t j | t j d d t | S)zU
Return a connection object connected to the pipe given by `address`
i r N)r rl Z
WaitNamedPiper r r r r r r> r~ ZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYr r r rj )r0 r hr r r r r s" r s #CHALLENGE#s #WELCOME#s #FAILURE#c C s d d l } t | t s! t t j t } | j t | | j | | d j
} | j d } | | k r | j t n | j t
t d d S)Nr md5 zdigest received was wrong)hmacr rH rp r urandomMESSAGE_LENGTHrP CHALLENGEnewdigestrV WELCOMEFAILUREr )
connectionr r messager responser r r r s
r c C s d d l } t | t s! t | j d } | d t t t k s\ t d | | t t d } | j | | d j } | j | | j d } | t
k r t d d S)Nr r z message = %rr zdigest sent was rejected) r r rH rp rV rI r r r rP r r )r r r r r r r r r r s ,
r c @ s4 e Z d Z d d Z d d Z d d Z d S)ConnectionWrapperc C sO | | _ | | _ | | _ x- d D]% } t | | } t | | | q" Wd S)NrE rB r` rV rP )rE rB r` rV rP )_conn_dumps_loadsr setattr)r7 connrQ r\ attrrR r r r r; s
zConnectionWrapper.__init__c C s# | j | } | j j | d S)N)r r rP )r7 rR r r r r rS s zConnectionWrapper.sendc C s | j j } | j | S)N)r rV r )r7 r r r r r^ s zConnectionWrapper.recvN)rf rg rh r; rS r^ r r r r r s r c C s% t j | f d d d d j d S)Nr zutf-8) xmlrpclibrQ encode)rR r r r
_xml_dumps s r c C s% t j | j d \ \ } } | S)Nzutf-8)r r\ decode)r rR methodr r r
_xml_loads s !r c @ s e Z d Z d d Z d S) XmlListenerc C s. d d l j a t j | } t | t t S)Nr )
xmlrpc.clientclientr r r r r r )r7 rR r r r r s zXmlListener.acceptN)rf rg rh r r r r r r s r c O s( d d l j a t t | | t t S)Nr )r r r r r r r )r kwdsr r r XmlClient s r c C s t | } g } x | r t j | d | } | t k r@ Pnr t | k oa t t | k n rs | t 8} n? t | k o t t | k n r | t 8} n t d | j | | | | d d } d } q W| S)NFzShould not get herer r ) listrl rn r r rI r
r r )Zhandlesr Lreadyr r r r _exhaustive_wait s &
&
r c
. s
| d k r t } n) | d k r* d } n t | d d } t | } i g } t t } zxy| D]q} y t | d } Wn" t k
r | | j <Yqr Xy" t j | d d \ } } WnD t k
r} z$ d | j
} } | t k r WYd d } ~ Xn X| t j k rE| j
| | | j <qr | rt j d d d k ry | j d \ } } Wn. t k
r} z | j
} WYd d } ~ Xn X| rt | d
rd | _ j | d } qr Wt j | } Wd x | D] } | j qWx | D] } y | j d \ } } Wn= t k
r} z | j
} | t k rn WYd d } ~ Xn X| t j k r" | j } j | | d k r"t | d
r"d | _ q"WX j f d d | D f d
d | D S)z
Wait till an object in object_list is ready/readable.
Returns list of those objects in object_list which are ready/readable.
Nr i g ?rE T Frw c 3 s | ] } | Vq d S)Nr ).0r )waithandle_to_objr r <genexpr>r s zwait.<locals>.<genexpr>c s" g | ] } | k r | q Sr r )r o)
ready_objectsr r
<listcomp>s s zwait.<locals>.<listcomp>)r r )r intr setr AttributeErrorr3 rl r{ r> r~
_ready_errorsrm r ro r' Zgetwindowsversionrr r) rw addr keysrq ZERROR_OPERATION_ABORTEDupdate)
object_listr Zov_listZ
ready_handlesr rE rs rt r r r )r r r r ' sh
"
"
PollSelectorc C s t } x! | D] } | j | t j q W| d k rL t j | } xV | j | } | ru d d | D S| d k rO | t j } | d k rO | SqO WWd QRXd S)z
Wait till an object in object_list is ready/readable.
Returns list of those objects in object_list which are ready/readable.
Nc S s g | ] \ } } | j q Sr )Zfileobj)r keyZeventsr r r r s zwait.<locals>.<listcomp>r )
_WaitSelectorregister selectorsZ
EVENT_READr Zselect)r r ZselectorrR Zdeadliner r r r r s
c C sm | j } t j | t j t j ? } d d l m } | j | } t | | j | j
f f SWd QRXd S)Nr )resource_sharer) rE r* Zfromfdr Z SOCK_STREAMr r Z DupSocketrebuild_connectionr9 r: )r r8 r r dsr r r reduce_connection s
r c C s" | j } t | j | | S)N)r r )r r9 r: Zsockr r r r s r c C s_ | j r t j n d | j r' t j n d B} t j | j | } t | | j | j f f S)Nr ) r9 rl ZFILE_GENERIC_READr: ZFILE_GENERIC_WRITEr Z DupHandlerE rebuild_pipe_connection)r r dhr r r reduce_pipe_connection s r c C s | j } t | | | S)N)r rj )r r9 r: r8 r r r r s r c C s. t j | j } t | | j | j f f S)N)r ZDupFdrE r r9 r: )r dfr r r r s c C s | j } t | | | S)N)r r )r r9 r: fdr r r r s )G__all__rx r r' r* r r r itertoolsr r r r r r
r rl r r
r r ImportErrorr( r ZCONNECTION_TIMEOUTcountr# r Zfamiliesr) r r r& r+ r1 r2 rj 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 r r
r ZSelectSelectorr r r
r r r r r r <module>
s &
PJ8,+ 8
P