U :e{@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)timeoutr?/usr/local/python38/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) localhostrrz listener-)prefixdirrz\\.\pipe\pyc-%d-%d-zunrecognized familyN) tempfilemktempr 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   r0cCsTt|tkrdSt|tkr*|dr*dSt|tks@t|rDdStd|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 startswithris_abstract_socket_namespacer()addressrrr address_type_s r7c@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|dSrr:_closer=rrr__del__s z_ConnectionBase.__del__cCs|jdkrtddS)Nzhandle is closed)r:OSErrorrDrrr _check_closeds z_ConnectionBase._check_closedcCs|jstddS)Nzconnection is write-only)r;rFrDrrr_check_readablesz_ConnectionBase._check_readablecCs|jstddS)Nzconnection is read-only)r<rFrDrrr_check_writablesz_ConnectionBase._check_writablecCs"|jrd|_n|tddS)NFzbad message length)r<r;closerFrDrrr_bad_message_lengthsz#_ConnectionBase._bad_message_lengthcCs |jdkS)z True if the connection is closedNr:rDrrrclosedsz_ConnectionBase.closedcCs|jS)z"True if the connection is readable)r;rDrrrr?sz_ConnectionBase.readablecCs|jS)z"True if the connection is writable)r<rDrrrr@sz_ConnectionBase.writablecCs||jS)z+File descriptor or handle of the connection)rGr:rDrrrfilenosz_ConnectionBase.filenocCs$|jdk r z |W5d|_XdS)zClose the connectionNrBrDrrrrJs  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)rGrI memoryviewitemsizebyteslenr( _send_bytes)r=bufoffsetsizemnrrr send_bytess"     z_ConnectionBase.send_bytescCs$|||t|dS)zSend a (picklable) objectN)rGrIrS_ForkingPicklerdumpsr=objrrrsendsz_ConnectionBase.sendcCsJ|||dk r(|dkr(td||}|dkrB||S)z7 Receive bytes data as a bytes object. Nrznegative maxlength)rGrHr( _recv_bytesrKgetvalue)r= maxlengthrTrrr 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) rGrHrOrPrRr(r_tellr r`seekreadinto)r=rTrUrWrPbytesizeresultrVrrrrecv_bytes_intos$        z_ConnectionBase.recv_bytes_intocCs&|||}t|S)zReceive a (picklable) object)rGrHr_rZloads getbuffer)r=rTrrrrecvsz_ConnectionBase.recvcCs||||S)z/Whether there is any input available to be read)rGrH_pollr=rrrrpollsz_ConnectionBase.pollcCs|SrrrDrrr __enter__sz_ConnectionBase.__enter__cCs |dSrrJr=exc_type exc_valueexc_tbrrr__exit__sz_ConnectionBase.__exit__)TT)rN)N)r)rl)__name__ __module__ __qualname__r:rArErGrHrIrKpropertyrMr?r@rNrJrYr^rbrhrkrorprvrrrrr8rs.       r8c@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||jdSrrL)r=Z _CloseHandlerrrrCszPipeConnection._closec Cshtj|j|dd\}}z      zPipeConnection._recv_bytescCs.|jst|jddkrdStt|g|S)NrT)rr~ PeekNamedPiper:boolrrnrrrrmFs zPipeConnection._pollcCs|}t}||t|jd}|dk rJt|||krJ|tj |j|dd\}}| d\}}|||S)NrTr|) rjrrrr~rr:rRrKrr)r=rrrTrleftrZrbytesrrrrLs zPipeConnection._get_more_data)N) rwrxry__doc__rr~ CloseHandlerCrSr_rmrrrrrr{ s r{c@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||jdSrrLr=rCrrrrCcszConnection._closecCs||jdSrrLrrrrrChscCs8t|}||j|}||8}|dkr&q4||d}qdSNr)rRr:)r=rTr remainingrXrrr_sendms  zConnection._sendcCsbt}|j}|}|dkr^|||}t|}|dkrJ||krBtntd||||8}q|S)Nrzgot end of file during message)rrr:rRrrFr)r=rVreadrTr>rchunkrXrrr_recvvs   zConnection._recvcCst|}|dkrHtdd}td|}||||||n8td|}|dkrr||||n|||dS)Ni!i!Qi@)rRstructpackr)r=rTrX pre_headerheaderrrrrSs        zConnection._send_bytesNcCs^|d}td|\}|dkr@|d}td|\}|dk rT||krTdS||S)Nrrr)rrunpackr`)r=rrTrVrrrr_s  zConnection._recv_bytescCst|g|}t|Sr)rr)r=rrrrrrms zConnection._poll)N)rwrxryrr~_multiprocessingZ closesocketrCr^_writerk_readr$rJrrrrrSr_rmrrrrr\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) r7default_familyr+r0 PipeListener _listenerSocketListener isinstancerQ TypeError_authkey)r=r6r*backlogauthkeyrrrrAs 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)rrFacceptrdeliver_challengeanswer_challenge)r=crrrrs    zListener.acceptcCs |j}|dk rd|_|dS)zA Close the bound socket or named pipe of `self`. N)rrJ)r=listenerrrrrJszListener.closecCs|jjSr)r_addressrDrrrr6szListener.addresscCs|jjSr)r_last_acceptedrDrrr last_acceptedszListener.last_acceptedcCs|SrrrDrrrrpszListener.__enter__cCs |dSrrqrrrrrrvszListener.__exit__)NNrN) rwrxryrrArrJrzr6rrprvrrrrrs    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) r7r0 PipeClient SocketClientrrQrrr)r6r*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/ socketpair setblockingrdetachr$pipe)duplexs1s2c1c2fd1fd2rrrrs       c Cstd}|r*tj}tjtjB}tt}}ntj}tj}dt}}t||tjBtj Btj tj Btj Bd||tj tj}t||dtjtjtjtj}t|tj ddtj|dd}|d\} } t||d} t||d} | | fS) rrrrNTr|rr)r+r~PIPE_ACCESS_DUPLEX GENERIC_READ GENERIC_WRITEBUFSIZEZPIPE_ACCESS_INBOUNDCreateNamedPipeFILE_FLAG_OVERLAPPEDFILE_FLAG_FIRST_PIPE_INSTANCEPIPE_TYPE_MESSAGEPIPE_READMODE_MESSAGE PIPE_WAITNMPWAIT_WAIT_FOREVERNULL CreateFile OPEN_EXISTINGSetNamedPipeHandleStateConnectNamedPiperr{) rr6ZopenmodeaccessZobsizeZibsizeZh1Zh2r}_rrrrrrrsT       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|_|dkrt|stj|tj|fdd|_nd|_dS)NposixrTrrargs exitpriority)r/getattr_socketr$name setsockopt SOL_SOCKET SO_REUSEADDRrbindlisten getsocknamerrFrJ_familyrrr5Finalizeunlink_unlink)r=r6r*rrrrrAGs0       zSocketListener.__init__cCs&|j\}|_|dt|S)NT)rrrrrrr=srrrr`s zSocketListener.acceptcCs0z|jW5|j}|dk r*d|_|XdSr)rrrJ)r=rrrrrJes zSocketListener.closeN)r)rwrxryrrArrJrrrrrCs rc CsPt|}ttt|.}|d||t|W5QRSQRXdS)zO Return a connection object connected to the socket given by `address` TN)r7r/rrconnectrr)r6r*rrrrros   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_listenerrJ)r=r6rrrrrAs zPipeListener.__init__Fc CsHtjtjB}|r|tjO}t|j|tjtjBtjBtj t t tj tj Sr) r~rrrrrrrrZPIPE_UNLIMITED_INSTANCESrrr)r=rflagsrrrrs   zPipeListener._new_handlec Cs|j||jd}ztj|dd}Wn0tk r^}z|jtjkrNW5d}~XYnPXzrrrrresrrrrs(    zPipeListener.acceptcCs$td||D]}t|qdS)Nz closing listener with address=%r)rrr~r)queuer6r>rrrrs z$PipeListener._finalize_pipe_listener)N)F) rwrxryrrArr staticmethodrrrrrrs  rc Cst}z6t|dt|tjtjBdtjtjtjtj}Wqt k rz}z |j tj tj fksht |rjW5d}~XYqXqqt|tjddt|S)zU Return a connection object connected to the pipe given by `address` rN)rr~Z WaitNamedPiperrrrrrrFrZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYrrrr{)r6rhrrrrrs8  rs #CHALLENGE#s #WELCOME#s #FAILURE#cCsddl}t|ts$tdt|tt}| t || ||d }| d}||krl| tn| ttddS)Nr Authkey must be bytes, not {0!s}md5zdigest received was wrong)hmacrrQr(formatr1r$urandomMESSAGE_LENGTHrY CHALLENGEnewdigestrbWELCOMEFAILUREr connectionrr messagerresponserrrrs      rcCsxddl}t|ts$tdt||d}|ttd}| ||d }| ||d}|t krtt ddS)Nrrr rzdigest sent was rejected)r rrQr(r r1rbrRrrrrYrrrrrrrs     rc@s$eZdZddZddZddZdS)ConnectionWrappercCs6||_||_||_dD]}t||}t|||qdS)N)rNrJrorbrY)_conn_dumps_loadsrsetattr)r=connr[riattrr]rrrrAs  zConnectionWrapper.__init__cCs||}|j|dSr)rrrY)r=r]rrrrr^s zConnectionWrapper.sendcCs|j}||Sr)rrbrrrrrrks zConnectionWrapper.recvN)rwrxryrAr^rkrrrrrsrcCst|fdddddS)Nrutf-8) xmlrpclibr[encode)r]rrr _xml_dumps sr!cCst|d\\}}|S)Nr)rridecode)rr]methodrrr _xml_loadssr$c@seZdZddZdS) XmlListenercCs"ddlmat|}t|ttSr) xmlrpc.clientclientrrrrr!r$r\rrrrs  zXmlListener.acceptN)rwrxryrrrrrr%sr%cOsddlmatt||ttSr)r&r'rrrr!r$)rkwdsrrr XmlClients r)cCst|}g}|rt|d|}|tkr*qn\t|krFtt|krTnn |t8}n2t|krptt|kr~nn |t8}ntd|||||dd}d}q |S)NFzShould not get hererr) listr~rr r rRr rr)ZhandlesrLreadyrrrr_exhaustive_wait%s    r-c s^|dkrt}n|dkrd}nt|dd}t|}ig}tt}z@|D]&}zt|d} Wn tk r||<YqPXzt | dd\}}Wn8tk r}zd|j}}|tkrƂW5d}~XYnX|t jkr||||j <qP|rjtddd krjz|d \}}Wn*tk rP}z |j}W5d}~XYnX|sjt |drjd|_ |d}qPt|}W5|D]}|q|D]}z|d\}}Wn6tk r}z|j}|tkrW5d}~XYnX|t j kr|j } ||dkrt |drd|_qXfd d |Dfd d|DS) Wait till an object in object_list is ready/readable. Returns list of those objects in object_list which are ready/readable. Nrrg?TrrN)r/Fc3s|]}|VqdSrr).0r)waithandle_to_objrr szwait..csg|]}|kr|qSrr)r1o) ready_objectsrr swait..)rintr*setrrrFr _ready_errorsr~ZERROR_OPERATION_ABORTEDraddr.rrAttributeErrorr9rrrr,Zgetwindowsversionr-keysupdate) object_listrZov_listZ ready_handlesrrrrr4rNr)r5r2rr;sh               PollSelectorc Cst}|D]}||tjq |dk r4t|}||}|r\dd|DW5QRS|dk r4|t}|dkr4|W5QRSq4W5QRXdS)r.NcSsg|]\}}|jqSr)fileobj)r1keyeventsrrrr6sr7r) _WaitSelectorregister selectors EVENT_READrrselect)r?rselectorr]deadliner,rrrrs   c CsZ|}t|tjtj6}ddlm}||}t||j |j ffW5QRSQRXdS)Nr)resource_sharer) rNr/fromfdr SOCK_STREAMr rKZ DupSocketrebuild_connectionr?r@)rr>rrKdsrrrreduce_connections   rPcCs|}t|||Srrr)rOr?r@sockrrrrNsrNcCsB|jr tjnd|jrtjndB}t||}t||j|jffSr) r?r~ZFILE_GENERIC_READr@ZFILE_GENERIC_WRITEr Z DupHandlerNrebuild_pipe_connection)rrdhrrrreduce_pipe_connections rUcCs|}t|||Sr)rr{)rTr?r@r>rrrrSsrScCs t|}t||j|jffSr)r DupFdrNrNr?r@)rdfrrrrPscCs|}t|||SrrQ)rWr?r@fdrrrrNs)NN)T)T)N)N)I__all__rr$r,r/rrr! itertoolsrr rrr contextr ForkingPicklerrZr~r r r r ImportErrorr-rCONNECTION_TIMEOUTcountr'rfamiliesr.rrr+r0r7r8r{robjectrrrrrrrr rrrrrrr!r$r%r)r-rZERROR_NETNAME_DELETEDr:rrFr@rDSelectSelectorrPrNrErUrSrrrr s           PT=    ,, 8   P