a 5ghc|@s"gdZddlZddlZddlZddlZddlZddlZddlZddlZz ddl Z Wne ynddl Z Yn0ddl m Z ddl mZmZddlmZejZz$ddlZddlmZmZmZmZWn"e yejdkr܂dZYn0d Zd ZeZd Zd gZe ed rd Zed g7Zejdkr4d Zed g7ZefddZ!ddZ"ddZ#ddZ$ddZ%GdddZ&erGddde&Z'Gddde&Z(Gddde)Z*dPd d!Z+ejdkrdQd#d$Z,n dRd%d$Z,Gd&d'd'e)Z-d(d)Z.ejdkr Gd*d+d+e)Z/d,d-Z0d.Z1d/Z2d0Z3d1Z4d2d3Z5d4d5Z6Gd6d7d7e)Z7d8d9Z8d:d;Z9Gdd?Z;ejdkrd@dAZhZ?dSdBdCZ@n,ddlAZAe eAdDreAjBZCneAjDZCdTdEdCZ@ejdkrdFdGZEdHdIZFeGe(eEdJdKZHdLdMZIeGe'eHndNdGZEdOdIZFeGe(eEdS)U)ClientListenerPipewaitN)util)AuthenticationErrorBufferTooShort) reduction) WAIT_OBJECT_0WAIT_ABANDONED_0 WAIT_TIMEOUTINFINITEwin32i g4@AF_INETAF_UNIXAF_PIPEcCsttdtj|SN monotonicgetattrtime)timeoutrg/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/multiprocess/connection.py _init_timeout>srcCsttdtj|kSrr)trrr_check_timeoutAsrcCsX|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) tempfilemktemprZ get_temp_dirosgetpidnext _mmap_counter ValueErrorfamilyrrrarbitrary_addressHsr+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_familyVs   r0cCsTt|tkrdSt|tur*|dr*dSt|tus@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_typebs 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__xsz_ConnectionBase.__init__cCs|jdur|dSNr:_closer=rrr__del__s z_ConnectionBase.__del__cCs|jdurtddS)Nzhandle is closed)r:OSErrorrErrr _check_closeds z_ConnectionBase._check_closedcCs|jstddS)Nzconnection is write-only)r;rGrErrr_check_readablesz_ConnectionBase._check_readablecCs|jstddS)Nzconnection is read-only)r<rGrErrr_check_writablesz_ConnectionBase._check_writablecCs"|jrd|_n|tddS)NFzbad message length)r<r;closerGrErrr_bad_message_lengthsz#_ConnectionBase._bad_message_lengthcCs |jduS)z True if the connection is closedNr:rErrrclosedsz_ConnectionBase.closedcCs|jS)z"True if the connection is readable)r;rErrrr?sz_ConnectionBase.readablecCs|jS)z"True if the connection is writable)r<rErrrr@sz_ConnectionBase.writablecCs||jS)z+File descriptor or handle of the connection)rHr:rErrrfilenosz_ConnectionBase.filenocCs*|jdur&z|Wd|_nd|_0dS)zClose the connectionNrCrErrrrKs  z_ConnectionBase.closercCs||t|}|jdkr.tt|}t|}|dkrFtd||krVtd|durh||}n&|dkrztdn|||krtd|||||dS)z,Send the bytes data from a bytes-like objectrrzoffset is negativezbuffer length < offsetNzsize is negativezbuffer length < offset + size)rHrJ memoryviewitemsizebyteslenr( _send_bytes)r=bufoffsetsizemnrrr send_bytess"     z_ConnectionBase.send_bytescCs$|||t|dS)zSend a (picklable) objectN)rHrJrT_ForkingPicklerdumpsr=objrrrsendsz_ConnectionBase.sendcCsJ|||dur(|dkr(td||}|durB||S)z7 Receive bytes data as a bytes object. Nrznegative maxlength)rHrIr( _recv_bytesrLgetvalue)r=Z maxlengthrUrrr recv_bytess z_ConnectionBase.recv_bytescCs||t|}|j}|t|}|dkr>tdn||krNtd|}|}|||krvt| | d| |||||||WdS1s0YdS)zq Receive bytes data into a writeable bytes-like object. Return the number of bytes read. rznegative offsetzoffset too largeN) rHrIrPrQrSr(r`tellr raseekreadinto)r=rUrVrXrQZbytesizeresultrWrrrrecv_bytes_intos$        z_ConnectionBase.recv_bytes_intocCs&|||}t|S)zReceive a (picklable) object)rHrIr`r[loads getbuffer)r=rUrrrrecvsz_ConnectionBase.recvcCs||||S)z/Whether there is any input available to be read)rHrI_pollr=rrrrpollsz_ConnectionBase.pollcCs|SrBrrErrr __enter__sz_ConnectionBase.__enter__cCs |dSrBrKr=exc_type exc_valueexc_tbrrr__exit__ sz_ConnectionBase.__exit__)TT)rN)N)r)rk)__name__ __module__ __qualname__r:rArFrHrIrJrLpropertyrNr?r@rOrKrZr_rbrgrjrnrorurrrrr8us.       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||jdSrBrM)r=Z _CloseHandlerrrrDszPipeConnection._closec Cstj|j|dd\}}zVz,|tjkrBt|jgdt}|tksBJWn|Yn0W| d\}}n| d\}}0|dksJ|t |ksJdS)NT overlappedFr) _winapiZ WriteFiler:ERROR_IO_PENDINGWaitForMultipleObjectseventrr cancelGetOverlappedResultrS)r=rUoverrwaitresZnwrittenrrrrTs     zPipeConnection._send_bytesNc Cs|jrd|_tS|dur dnt|d}ztj|j|dd\}}zz,|tjkrpt|j gdt }|t kspJWn| Yn0W| d\}}|dkrt}|||WS|tjkr|||WSnh| d\}}|dkrt}|||YWS|tjkr>|||YWS0Wn<ty~}z"|jtjkrhtnWYd}~n d}~00tddS)NFTr{rz.shouldn't get here; expected KeyboardInterrupt)_got_empty_messageioBytesIOminr}ReadFiler:r~rrrr rrwriteriZERROR_MORE_DATA_get_more_datarGwinerrorERROR_BROKEN_PIPEEOFError RuntimeError) r=maxsizeZbsizerrrZnreadferrrr`)sH         zPipeConnection._recv_bytescCs.|jst|jddkrdStt|g|S)NrT)rr} PeekNamedPiper:boolrrmrrrrlIs zPipeConnection._pollcCs|}t}||t|jd}|dks6J|durVt|||krV|tj |j|dd\}}| d\}}|dksJ||ksJ|||S)NrrTr{) rirrrr}rr:rSrLrr)r=rrrUrleftrZrbytesrrrrOs    zPipeConnection._get_more_data)N) rvrwrx__doc__rr} CloseHandlerDrTr`rlrrrrrrzs rzc@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||jdSrBrMr=rDrrrrDfszConnection._closecCs||jdSrBrMrrrrrDkscCs8t|}||j|}||8}|dkr&q4||d}qdSNr)rSr:)r=rUr remainingrYrrr_sendps  zConnection._sendcCsbt}|j}|}|dkr^|||}t|}|dkrJ||krBtntd||||8}q|S)Nrzgot end of file during message)rrr:rSrrGr)r=rWreadrUr>rchunkrYrrr_recvys   zConnection._recvcCst|}|dkrHtdd}td|}||||||n8td|}|dkrr||||n|||dS)Ni!i!Qi@)rSstructpackr)r=rUrYZ pre_headerheaderrrrrTs        zConnection._send_bytesNcCs^|d}td|\}|dkr@|d}td|\}|durT||krTdS||S)Nrrr)rrunpackra)r=rrUrWrrrr`s  zConnection._recv_bytescCst|g|}t|SrB)rr)r=rrrrrrls zConnection._poll)N)rvrwrxrr}_multiprocessingZ closesocketrDr__writerj_readr$rKrrrrrTr`rlrrrrr_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||||_|durft|tsft d||_ dS)Nrauthkey should be a byte string) r7default_familyr+r0 PipeListener _listenerSocketListener isinstancerR TypeError_authkey)r=r6r*backlogauthkeyrrrrAs zListener.__init__cCs>|jdurtd|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)rrGacceptrdeliver_challengeanswer_challenge)r=crrrrs    zListener.acceptcCs |j}|durd|_|dS)zA Close the bound socket or named pipe of `self`. N)rrK)r=ZlistenerrrrrKszListener.closecCs|jjSrB)r_addressrErrrr6szListener.addresscCs|jjSrB)r_last_acceptedrErrr last_acceptedszListener.last_acceptedcCs|SrBrrErrrroszListener.__enter__cCs |dSrBrprqrrrruszListener.__exit__)NNrN) rvrwrxrrArrKryr6rrorurrrrrs    rcCsh|p t|}t||dkr&t|}nt|}|durHt|tsHtd|durdt||t|||S)z= Returns a connection to the address of a `Listener` rNr) r7r0 PipeClient SocketClientrrRrrr)r6r*rrrrrrs    rTcCsj|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)duplexs1s2c1c2Zfd1fd2rrrrs       rc Cstd}|r*tj}tjtjB}tt}}ntj}tj}dt}}t||tjBtj Btj tj Btj Bd||tj tj}t||dtjtjtjtj}t|tj ddtj|dd}|d\} } | dksJt||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_EXISTINGSetNamedPipeHandleStateConnectNamedPiperrz) rr6ZopenmodeaccessZobsizeZibsizeZh1Zh2r|_rrrrrrrs>           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 |_ Wnt y|jYn0||_d|_|dkrt|stj|tj|fdd|_nd|_dS)NposixrTrrargsZ exitpriority)r/r_socketr$name setsockopt SOL_SOCKET SO_REUSEADDRrbindlisten getsocknamerrGrKZ_familyrrr5Finalizeunlink_unlink)r=r6r*rrrrrAJs(        zSocketListener.__init__cCs&|j\}|_|dt|S)NT)rrrrrrr=srrrrcs zSocketListener.acceptcCsJz(|jW|j}|durFd|_|n|j}|durDd|_|0dSrB)rrKr)r=rrrrrKhs zSocketListener.closeN)r)rvrwrxrrArrKrrrrrFs rcCs\t|}ttt|0}|d||t|WdS1sN0YdS)zO Return a connection object connected to the socket given by `address` TN)r7r/rrconnectrr)r6r*rrrrrrs   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_listenerrK)r=r6rrrrrAs zPipeListener.__init__Fc CsHtjtjB}|r|tjO}t|j|tjtjBtjBtj t t tj tj SrB) r}rrrrrrrrZPIPE_UNLIMITED_INSTANCESrrr)r=rflagsrrrrs   zPipeListener._new_handlec Cs|j||jd}ztj|dd}Wn2ty`}z|jtjkrLWYd}~n~d}~00zVzt |j gdt }Wn | t |Yn0W|d\}}|dksJn|d\}}|dksJ0t|S)NrTr{F)rappendrpopr}rrGrZ ERROR_NO_DATArrrrrrrz)r=r>rrresrrrrrrs(     zPipeListener.acceptcCs$td||D]}t|qdS)Nz closing listener with address=%r)rrr}r)queuer6r>rrrrs z$PipeListener._finalize_pipe_listener)N)F) rvrwrxrrArr staticmethodrrrrrrs  rc Cst}z6t|dt|tjtjBdtjtjtjtj}Wqt y|}z(|j tj tj fvsft |rhWYd}~qd}~00qqt|tjddt|S)zU Return a connection object connected to the pipe given by `address` rN)rr}Z WaitNamedPiperrrrrrrGrZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYrrrrz)r6rhrrrrrs*    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)hmacrrRr(formatr1r$urandomMESSAGE_LENGTHrZ CHALLENGEnewdigestrbWELCOMEFAILUREr connectionrrmessager responserrrrs      rcCsddl}t|ts$tdt||d}|dtttksNJd||ttd}| ||d }| ||d}|t krt ddS)Nrrrz message = %rrzdigest sent was rejected)rrrRr(rr1rbrSrrr rZr rr rrrrs      rc@s$eZdZddZddZddZdS)ConnectionWrappercCs6||_||_||_dD]}t||}t|||qdS)N)rOrKrnrbrZ)_conn_dumps_loadsrsetattr)r=connr\rhattrr^rrrrAs  zConnectionWrapper.__init__cCs||}|j|dSrB)rrrZ)r=r^rrrrr_s zConnectionWrapper.sendcCs|j}||SrB)rrbrrrrrrj s zConnectionWrapper.recvN)rvrwrxrAr_rjrrrrrsrcCst|fdddddS)Nrutf-8) xmlrpclibr\encode)r^rrr _xml_dumpssrcCst|d\\}}|S)Nr)rrhdecode)rr^methodrrr _xml_loadssrc@seZdZddZdS) XmlListenercCs"ddlmat|}t|ttSr) xmlrpc.clientclientrrrrrrr]rrrrs  zXmlListener.acceptN)rvrwrxrrrrrrsrcOs"ddlmatt|i|ttSr)rr rrrrr)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 rSr rr)ZhandlesrLreadyrrrr_exhaustive_wait(s    r&c s |durt}n|dkrd}nt|dd}t|}ig}tt}z|D](}zt|d}Wnty||<YqP0zt|dd\}}Wn:t y}z"d|j }}|t vr‚WYd}~n d}~00|tj kr| |||j<qP|rltdddkrlz|d \} }Wn,t yR}z|j }WYd}~n d}~00|slt|d rld|_|d}qPt|}W|D]}|q|D]}z|d\} }Wn8t y}z|j }|t vrނWYd}~n d}~00|tjkr|j}||dkrt|d rd|_qn|D]}|q:|D]}z|d\} }Wn8t y}z|j }|t vrWYd}~n d}~00|tjkrN|j}||dkrNt|d rNd|_qN0fd 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?rOT)r(Frc3s|]}|VqdSrBr).0r)waithandle_to_objrr zwait..csg|]}|vr|qSrr)r*o) ready_objectsrr r-wait..)rintr#setrAttributeErrorr9r}rrGr _ready_errorsr~rrr,getwindowsversionrr.raddr&keysrZERROR_OPERATION_ABORTEDupdate) object_listrZov_listZ ready_handlesr.rOrrrrr)r/r+rr>s                       r PollSelectorcCst}|D]}||tjq |durrrFdsrrrreduce_connections   rKcCs|}t|||SrBrr)rJr?r@sockrrrrIsrIcCsB|jr tjnd|jrtjndB}t||}t||j|jffSr) r?r}ZFILE_GENERIC_READr@ZFILE_GENERIC_WRITEr Z DupHandlerOrebuild_pipe_connection)rrdhrrrreduce_pipe_connections rPcCs|}t|||SrB)rrz)rOr?r@r>rrrrNsrNcCs t|}t||j|jffSrB)r ZDupFdrOrIr?r@)rdfrrrrKscCs|}t|||SrBrL)rQr?r@fdrrrrIs)NN)T)T)N)N)J__all__rr$r,r/rrr" itertoolsZ _multiprocessr ImportErrorr!rrr contextr ZForkingPicklerr[r}r r r rr-rZCONNECTION_TIMEOUTcountr'rZfamiliesr.rrr+r0r7r8rzrobjectrrrrrrrrrr r rrrrrrr"r&rZERROR_NETNAME_DELETEDr5rrAr;r?SelectSelectorrKrIr@rPrNrrrr s             PT=    ,, 8   P