U \k^{@sddddgZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z m Z dd lmZejZz$ddlZdd lmZmZmZmZWn$ek rejd kr‚dZYnXd Zd ZeZdZdgZeedrdZedg7Zejd krdZedg7ZefddZ ddZ!ddZ"ddZ#ddZ$GdddZ%erhGddde%Z&Gdd d e%Z'Gd!dde(Z)dOd"dZ*ejd krdPd$dZ+n dQd%dZ+Gd&d'd'e(Z,d(d)Z-ejd krGd*d+d+e(Z.d,d-Z/d.Z0d/Z1d0Z2d1Z3d2d3Z4d4d5Z5Gd6d7d7e(Z6d8d9Z7d:d;Z8Gdd?Z:ejd krtd@dAZ;ejdRdBdZ?n,ddl@Z@ee@dCre@jAZBne@jCZBdSdDdZ?ejd krdEdFZDdGdHZEeFe'eDdIdJZGdKdLZHeFe&eGndMdFZDdNdHZEeFe'eDdS)TClientListenerPipewaitN)util)AuthenticationErrorBufferTooShort) reduction) WAIT_OBJECT_0WAIT_ABANDONED_0 WAIT_TIMEOUTINFINITEwin32i g4@AF_INETAF_UNIXAF_PIPEcCs t|SNtime monotonic)timeoutr0/usr/lib/python3.8/multiprocessing/connection.py _init_timeout;srcCs t|kSrr)trrr_check_timeout>srcCsX|dkr dS|dkr&tjdtdS|dkrLtjdtttfddStd d S) z? Return an arbitrary free address for the given family r)Z localhostrrz listener-)prefixdirrz\\.\pipe\pyc-%d-%d-zunrecognized familyN) tempfileZmktemprZ get_temp_dirosgetpidnext _mmap_counter ValueErrorfamilyrrrarbitrary_addressEsr(cCsJtjdkr|dkrtd|tjdkrF|dkrFtt|sFtd|dS)zD Checks if the family is valid for the current environment. rrzFamily %s is not recognized.rN)sysplatformr%hasattrsocketr&rrr_validate_familySs   r-cCsJt|tkrdSt|tkr*|dr*dSt|tkr:dStd|dS)z] Return the types of the address This can be 'AF_INET', 'AF_UNIX', or 'AF_PIPE' rz\\rrzaddress type of %r unrecognizedN)typetuplestr startswithr%)addressrrr address_type_s  r3c@seZdZdZd+ddZddZddZd d Zd d Zd dZ e ddZ e ddZ e ddZ ddZddZd,ddZddZd-ddZd.d d!Zd"d#Zd/d%d&Zd'd(Zd)d*ZdS)0_ConnectionBaseNTcCs>|}|dkrtd|s(|s(td||_||_||_dS)Nrzinvalid handlez6at least one of `readable` and `writable` must be True) __index__r%_handle _readable _writable)selfhandlereadablewritablerrr__init__usz_ConnectionBase.__init__cCs|jdk r|dSrr6_closer9rrr__del__s z_ConnectionBase.__del__cCs|jdkrtddS)Nzhandle is closed)r6OSErrorr@rrr _check_closeds z_ConnectionBase._check_closedcCs|jstddS)Nzconnection is write-only)r7rBr@rrr_check_readablesz_ConnectionBase._check_readablecCs|jstddS)Nzconnection is read-only)r8rBr@rrr_check_writablesz_ConnectionBase._check_writablecCs"|jrd|_n|tddS)NFzbad message length)r8r7closerBr@rrr_bad_message_lengthsz#_ConnectionBase._bad_message_lengthcCs |jdkS)z True if the connection is closedNr6r@rrrclosedsz_ConnectionBase.closedcCs|jS)z"True if the connection is readable)r7r@rrrr;sz_ConnectionBase.readablecCs|jS)z"True if the connection is writable)r8r@rrrr<sz_ConnectionBase.writablecCs||jS)z+File descriptor or handle of the connection)rCr6r@rrrfilenosz_ConnectionBase.filenocCs$|jdk r z |W5d|_XdS)zClose the connectionNr>r@rrrrFs  z_ConnectionBase.closercCs||t|}|jdkr.tt|}t|}|dkrFtd||krVtd|dkrh||}n&|dkrztdn|||krtd|||||dS)z,Send the bytes data from a bytes-like objectrrzoffset is negativezbuffer length < offsetNzsize is negativezbuffer length < offset + size)rCrE memoryviewitemsizebyteslenr% _send_bytes)r9bufoffsetsizemnrrr send_bytess"     z_ConnectionBase.send_bytescCs$|||t|dS)zSend a (picklable) objectN)rCrErO_ForkingPicklerdumpsr9objrrrsendsz_ConnectionBase.sendcCsJ|||dk r(|dkr(td||}|dkrB||S)z7 Receive bytes data as a bytes object. Nrznegative maxlength)rCrDr% _recv_bytesrGgetvalue)r9Z maxlengthrPrrr recv_bytess z_ConnectionBase.recv_bytesc Cs||t|}|j}|t|}|dkr>tdn||krNtd|}|}|||krvt| | d| |||||||W5QRSQRXdS)zq Receive bytes data into a writeable bytes-like object. Return the number of bytes read. rznegative offsetzoffset too largeN) rCrDrKrLrNr%r[tellr r\seekreadinto)r9rPrQrSrLZbytesizeresultrRrrrrecv_bytes_intos$        z_ConnectionBase.recv_bytes_intocCs&|||}t|S)zReceive a (picklable) object)rCrDr[rVloads getbuffer)r9rPrrrrecvsz_ConnectionBase.recvcCs||||S)z/Whether there is any input available to be read)rCrD_pollr9rrrrpollsz_ConnectionBase.pollcCs|Srrr@rrr __enter__sz_ConnectionBase.__enter__cCs |dSrrFr9exc_type exc_valueZexc_tbrrr__exit__sz_ConnectionBase.__exit__)TT)rN)N)r)rf)__name__ __module__ __qualname__r6r=rArCrDrErGpropertyrIr;r<rJrFrUrZr]rbrerirjrorrrrr4rs.       r4c@sDeZdZdZdZejfddZddZddd 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. FcCs||jdSrrH)r9Z _CloseHandlerrrr?szPipeConnection._closec Cstj|j|dd\}}zHz,|tjkrBt|jgdt}|tksBt Wn| YnXW5|d\}}X|dks|t |t |kst dS)NT overlappedFr) _winapiZ WriteFiler6GetOverlappedResultERROR_IO_PENDINGWaitForMultipleObjectseventrr AssertionErrorcancelrN)r9rPoverrZnwrittenwaitresrrrrOs   zPipeConnection._send_bytesNc Cs2|jrd|_tS|dkr dnt|d}ztj|j|dd\}}dzHz,|tj krpt |jgdt}|tksptWn|YnXW5|d\}}|dkrt}|| |WS|tj kr| ||WSXWn:tk r$}z|jtjkrtnW5d}~XYnXtddS)NFTrurz.shouldn't get here; expected KeyboardInterrupt)_got_empty_messageioBytesIOminrwReadFiler6rxwriterdZERROR_MORE_DATA_get_more_dataryrzr{rr r|r}rBwinerrorERROR_BROKEN_PIPEEOFError RuntimeError) r9maxsizeZbsizer~rZnreadfrerrrr[&s>      zPipeConnection._recv_bytescCs.|jst|jddkrdStt|g|S)NrT)rrw PeekNamedPiper6boolrrhrrrrgFs zPipeConnection._pollcCs|}t}||t|jd}|dks6t|dk rVt|||krV| tj |j|dd\}}| d\}}|dkst||kst|||S)NrrTru) rdrrrrwrr6r|rNrGrrx)r9r~rrPrleftrZrbytesrrrrLs    zPipeConnection._get_more_data)N) rprqrr__doc__rrw CloseHandler?rOr[rgrrrrrrt s rtc@s|eZdZdZer,ejfddZejZ ej Z ne j fddZe jZ e jZ e fddZe fddZd d Zdd d ZddZd S) Connectionzo Connection class based on an arbitrary file descriptor (Unix only), or a socket handle (Windows). cCs||jdSrrHr9r?rrrr?cszConnection._closecCs||jdSrrHrrrrr?hscCs8t|}||j|}||8}|dkr&q4||d}qdSNr)rNr6)r9rPr remainingrTrrr_sendms  zConnection._sendcCsbt}|j}|}|dkr^|||}t|}|dkrJ||krBtntd||||8}q|S)Nrzgot end of file during message)rrr6rNrrBr)r9rRreadrPr:rchunkrTrrr_recvvs   zConnection._recvcCst|}|dkrHtdd}td|}||||||n8td|}|dkrr||||n|||dS)Ni!i!Qi@)rNstructZpackr)r9rPrTZ pre_headerheaderrrrrOs        zConnection._send_bytesNcCs^|d}td|\}|dkr@|d}td|\}|dk rT||krTdS||S)Nrrr)rrZunpackr\)r9rrPrRrrrr[s  zConnection._recv_bytescCst|g|}t|Sr)rr)r9rrrrrrgs zConnection._poll)N)rprqrrrrw_multiprocessingZ closesocketr?rZZ_writereZ_readr!rFrrrrrOr[rgrrrrr\s  rc@sReZdZdZdddZddZdd Zed d Zed d Z ddZ ddZ dS)rz Returns a listener object. This is a wrapper for a bound socket which is 'listening' for connections, or for a Windows named pipe. NrcCsp|p|rt|pt}|pt|}t||dkr>t|||_nt||||_|dk rft|tsft d||_ dS)Nrauthkey should be a byte string) r3default_familyr(r- PipeListener _listenerSocketListener isinstancerM TypeError_authkey)r9r2r'backlogauthkeyrrrr=s zListener.__init__cCs>|jdkrtd|j}|jr:t||jt||j|S)zz Accept a connection on the bound socket or named pipe of `self`. Returns a `Connection` object. Nzlistener is closed)rrBacceptrdeliver_challengeanswer_challenge)r9crrrrs    zListener.acceptcCs |j}|dk rd|_|dS)zA Close the bound socket or named pipe of `self`. N)rrF)r9ZlistenerrrrrFszListener.closecCs|jjSr)r_addressr@rrrr2szListener.addresscCs|jjSr)r_last_acceptedr@rrr last_acceptedszListener.last_acceptedcCs|Srrr@rrrrjszListener.__enter__cCs |dSrrkrlrrrroszListener.__exit__)NNrN) rprqrrrr=rrFrsr2rrjrorrrrrs    cCsh|p t|}t||dkr&t|}nt|}|dk rHt|tsHtd|dk rdt||t|||S)z= Returns a connection to the address of a `Listener` rNr) r3r- PipeClient SocketClientrrMrrr)r2r'rrrrrrs    TcCsj|r>t\}}|d|dt|}t|}n$t\}}t|dd}t|dd}||fS)L Returns pair of connection objects at either end of a pipe TFr<r;)r,Z socketpair setblockingrdetachr!pipe)duplexs1s2c1c2Zfd1Zfd2rrrrs       c Cstd}|r*tj}tjtjB}tt}}ntj}tj}dt}}t||tjBtj Btj tj Btj Bd||tj tj}t||dtjtjtjtj}t|tj ddtj|dd}|d\} } | dkstt||d} t||d} | | fS) rrrrNTrurr)r(rwPIPE_ACCESS_DUPLEX GENERIC_READ GENERIC_WRITEBUFSIZEZPIPE_ACCESS_INBOUNDCreateNamedPipeFILE_FLAG_OVERLAPPEDFILE_FLAG_FIRST_PIPE_INSTANCEPIPE_TYPE_MESSAGEPIPE_READMODE_MESSAGE PIPE_WAITNMPWAIT_WAIT_FOREVERNULL CreateFile OPEN_EXISTINGSetNamedPipeHandleStateConnectNamedPiperxr|rt) rr2ZopenmodeaccessZobsizeZibsizeZh1Zh2rv_rrrrrrrsV        c@s*eZdZdZd ddZddZddZd S) rzO Representation of a socket which is bound to an address and listening rcCsttt||_zRtjdkr2|jtjtjd|jd|j ||j ||j |_ Wn t k r|jYnX||_d|_|dkrtj|tj|fdd|_nd|_dS)NposixrTrrargsZ exitpriority)r,getattr_socketr!nameZ setsockoptZ SOL_SOCKETZ SO_REUSEADDRrZbindZlistenZ getsocknamerrBrFZ_familyrrFinalizeunlink_unlink)r9r2r'rrrrr=Gs0       zSocketListener.__init__cCs&|j\}|_|dt|S)NT)rrrrrrr9srrrr_s zSocketListener.acceptcCs0z|jW5|j}|dk r*d|_|XdSr)rrrF)r9rrrrrFds zSocketListener.closeN)r)rprqrrrr=rrFrrrrrCs rc CsPt|}ttt|.}|d||t|W5QRSQRXdS)zO Return a connection object connected to the socket given by `address` TN)r3r,rrZconnectrr)r2r'rrrrrns   rc@s8eZdZdZd ddZd ddZdd Zed d ZdS)rz0 Representation of a named pipe NcCsL||_|jddg|_d|_td|jtj|tj|j|jfdd|_ dS)NT)firstz listener created with address=%rrr) r _new_handle _handle_queuerr sub_debugrr_finalize_pipe_listenerrF)r9r2rrrrr=s zPipeListener.__init__Fc CsHtjtjB}|r|tjO}t|j|tjtjBtjBtj t t tj tj Sr) rwrrrrrrrrZPIPE_UNLIMITED_INSTANCESrrr)r9rflagsrrrrs   zPipeListener._new_handlec Cs|j||jd}ztj|dd}Wn0tk r^}z|jtjkrNW5d}~XYn\Xzszwait..csg|]}|kr|qSrr)ro) ready_objectsrr swait..)rintrsetr}rxrBr _ready_errorsrwZERROR_OPERATION_ABORTEDr{addr+rrAttributeErrorr5rryrr)Zgetwindowsversionrkeysupdate) object_listrZov_listZ ready_handlesr~rrrrrJr)rrrr:sh               PollSelectorc Cst}|D]}||tjq |dk r4t|}||}|r\dd|DW5QRS|dk r4|t}|dkr4|W5QRSq4W5QRXdS)rNcSsg|]\}}|jqSr)Zfileobj)rkeyZeventsrrrrsrr) _WaitSelectorregister selectorsZ EVENT_READrrZselect)r$rZselectorrYZdeadlinerrrrrs   c CsZ|}t|tjtj6}ddlm}||}t||j |j ffW5QRSQRXdS)Nr)resource_sharer) rJr,ZfromfdrZ SOCK_STREAMrr*Z DupSocketrebuild_connectionr;r<)rr:rr*dsrrrreduce_connections   r-cCs|}t|||Srrr)r,r;r<Zsockrrrr+sr+cCsB|jr tjnd|jrtjndB}t||}t||j|jffSr) r;rwZFILE_GENERIC_READr<ZFILE_GENERIC_WRITEr Z DupHandlerJrebuild_pipe_connection)rrdhrrrreduce_pipe_connections r1cCs|}t|||Sr)rrt)r0r;r<r:rrrr/sr/cCs t|}t||j|jffSr)r ZDupFdrJr+r;r<)rdfrrrr-scCs|}t|||Srr.)r2r;r<fdrrrr+s)NN)T)T)N)N)I__all__rr!r)r,rrr itertoolsrrrrr contextr ZForkingPicklerrVrwr r r r ImportErrorr*rZCONNECTION_TIMEOUTcountr$rZfamiliesr+rrr(r-r3r4rtrobjectrrrrrrrrrrrrrrrr r rrrZERROR_NETNAME_DELETEDrrr)r%r'ZSelectSelectorr-r+r(r1r/rrrr s           PT=    ,+ 8   P