a DOg{@sgdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z m Z ddlmZejZz$ddlZddlmZmZmZmZWn"eyejdkrdZYn0d Zd ZeZd Zd gZeed rd Zed g7Zejdkrd Zed g7ZefddZ ddZ!ddZ"ddZ#ddZ$GdddZ%erbGddde%Z&Gddde%Z'Gddde(Z)dPd d!Z*ejdkrdQd#d$Z+n dRd%d$Z+Gd&d'd'e(Z,d(d)Z-ejdkrGd*d+d+e(Z.d,d-Z/d.Z0d/Z1d0Z2d1Z3d2d3Z4d4d5Z5Gd6d7d7e(Z6d8d9Z7d:d;Z8Gdd?Z:ejdkrnd@dAZ;ejdSdBdCZ?n,ddl@Z@ee@dDre@jAZBne@jCZBdTdEdCZ?ejdkrdFdGZDdHdIZEeFe'eDdJdKZGdLdMZHeFe&eGndNdGZDdOdIZEeFe'eDdS)U)ClientListenerPipewaitN)util)AuthenticationErrorBufferTooShort) reduction) WAIT_OBJECT_0WAIT_ABANDONED_0 WAIT_TIMEOUTINFINITEwin32i g4@AF_INETAF_UNIXAF_PIPEcCs t|SNtime monotonic)timeoutr?/opt/alt/python39/lib64/python3.9/multiprocessing/connection.py _init_timeout;srcCs t|kSrr)trrr_check_timeout>srcCsX|dkr dS|dkr&tjdtdS|dkrLtjdtttfddStd dS) Nr)Z localhostrrz listener-)prefixdirrz\\.\pipe\pyc-%d-%d-zunrecognized family) tempfileZmktemprZ get_temp_dirosgetpidnext _mmap_counter ValueErrorfamilyrrrarbitrary_addressEsr(cCsJtjdkr|dkrtd|tjdkrF|dkrFtt|sFtd|dS)NrrzFamily %s is not recognized.r)sysplatformr%hasattrsocketr&rrr_validate_familySs   r-cCsTt|tkrdSt|tur*|dr*dSt|tus@t|rDdStd|dS)Nrz\\rrzaddress type of %r unrecognized)typetuplestr startswithris_abstract_socket_namespacer%)addressrrr address_type_s r4c@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|jdur|dSrr7_closer:rrr__del__s z_ConnectionBase.__del__cCs|jdurtddS)Nzhandle is closed)r7OSErrorrArrr _check_closeds z_ConnectionBase._check_closedcCs|jstddS)Nzconnection is write-only)r8rCrArrr_check_readablesz_ConnectionBase._check_readablecCs|jstddS)Nzconnection is read-only)r9rCrArrr_check_writablesz_ConnectionBase._check_writablecCs"|jrd|_n|tddS)NFzbad message length)r9r8closerCrArrr_bad_message_lengthsz#_ConnectionBase._bad_message_lengthcCs |jduSrr7rArrrclosedsz_ConnectionBase.closedcCs|jSr)r8rArrrr<sz_ConnectionBase.readablecCs|jSr)r9rArrrr=sz_ConnectionBase.writablecCs||jSr)rDr7rArrrfilenosz_ConnectionBase.filenocCs*|jdur&z|Wd|_nd|_0dSrr?rArrrrGs  z_ConnectionBase.closercCs||t|}|jdkr.tt|}t|}|dkrFtd||krVtd|durh||}n&|dkrztdn|||krtd|||||dS)Nrrzoffset is negativezbuffer length < offsetzsize is negativezbuffer length < offset + size)rDrF memoryviewitemsizebyteslenr% _send_bytes)r:bufoffsetsizemnrrr send_bytess"     z_ConnectionBase.send_bytescCs$|||t|dSr)rDrFrP_ForkingPicklerdumpsr:objrrrsendsz_ConnectionBase.sendcCsJ|||dur(|dkr(td||}|durB||S)Nrznegative maxlength)rDrEr% _recv_bytesrHgetvalue)r:Z maxlengthrQrrr recv_bytess z_ConnectionBase.recv_bytescCs||t|}|j}|t|}|dkr>tdn||krNtd|}|}|||krvt| | d| |||||||WdS1s0YdS)Nrznegative offsetzoffset too large) rDrErLrMrOr%r\tellr r]seekreadinto)r:rQrRrTrMZbytesizeresultrSrrrrecv_bytes_intos$        z_ConnectionBase.recv_bytes_intocCs&|||}t|Sr)rDrEr\rWloads getbuffer)r:rQrrrrecvsz_ConnectionBase.recvcCs||||Sr)rDrE_pollr:rrrrpollsz_ConnectionBase.pollcCs|SrrrArrr __enter__sz_ConnectionBase.__enter__cCs |dSrrGr:exc_type exc_valueZexc_tbrrr__exit__sz_ConnectionBase.__exit__)TT)rN)N)r)rg)__name__ __module__ __qualname__r7r>rBrDrErFrHpropertyrJr<r=rKrGrVr[r^rcrfrjrkrprrrrr5rs.       r5c@s@eZdZdZejfddZddZd ddZd d Z d d Z dS)PipeConnectionFcCs||jdSrrI)r:Z _CloseHandlerrrr@szPipeConnection._closec Csvtj|j|dd\}}zJz |tjkr6t|jgdt}Wn|Yn0W|d\}}n|d\}}0dS)NT overlappedF) _winapiZ WriteFiler7ERROR_IO_PENDINGWaitForMultipleObjectseventrcancelGetOverlappedResult)r:rQoverrwaitresZnwrittenrrrrPs    zPipeConnection._send_bytesNc Cs|jrd|_tS|dur dnt|d}z tj|j|dd\}}zz |tjkrdt|j gdt }Wn| Yn0W| d\}}|dkrt}| ||WS|tjkr|||WSnh| d\}}|dkr t}| ||YWS|tjkr2|||YWS0Wn<tyr}z"|jtjkr\tnWYd}~n d}~00tddS)NFTrvrz.shouldn't get here; expected KeyboardInterrupt)_got_empty_messageioBytesIOminrxReadFiler7ryrzr{rr|r}writereZERROR_MORE_DATA_get_more_datarCwinerrorERROR_BROKEN_PIPEEOFError RuntimeError) r:maxsizeZbsizer~rrZnreadferrrr\&sH         zPipeConnection._recv_bytescCs.|jst|jddkrdStt|g|S)NrT)rrx PeekNamedPiper7boolrrirrrrhFs zPipeConnection._pollcCs|}t}||t|jd}|durJt|||krJ|tj |j|dd\}}| d\}}|||S)NrTrv) rerrrrxrr7rOrHrr})r:r~rrQrleftrZrbytesrrrrLs zPipeConnection._get_more_data)N) rqrrrsrrx CloseHandler@rPr\rhrrrrrru s  ruc@sxeZdZer(ejfddZejZej Z ne j fddZe j Ze jZ efddZe fddZdd Zdd d Zd dZd S) ConnectioncCs||jdSrrIr:r@rrrr@cszConnection._closecCs||jdSrrIrrrrr@hscCs8t|}||j|}||8}|dkr&q4||d}qdSNr)rOr7)r:rQr remainingrUrrr_sendms  zConnection._sendcCsbt}|j}|}|dkr^|||}t|}|dkrJ||krBtntd||||8}q|S)Nrzgot end of file during message)rrr7rOrrCr)r:rSreadrQr;rchunkrUrrr_recvvs   zConnection._recvcCst|}|dkrHtdd}td|}||||||n8td|}|dkrr||||n|||dS)Ni!i!Qi@)rOstructZpackr)r:rQrUZ pre_headerheaderrrrrPs        zConnection._send_bytesNcCs^|d}td|\}|dkr@|d}td|\}|durT||krTdS||S)Nrrr)rrZunpackr])r:rrQrSrrrr\s  zConnection._recv_bytescCst|g|}t|Sr)rr)r:rrrrrrhs zConnection._poll)N)rqrrrsrx_multiprocessingZ closesocketr@r[Z_writerfZ_readr!rGrrrrrPr\rhrrrrr\s  rc@sNeZdZdddZddZddZed d Zed d Zd dZ ddZ dS)rNrcCsp|p|rt|pt}|pt|}t||dkr>t|||_nt||||_|durft|tsft d||_ dSNrzauthkey should be a byte string) r4default_familyr(r- PipeListener _listenerSocketListener isinstancerN TypeError_authkey)r:r3r'backlogauthkeyrrrr>s zListener.__init__cCs>|jdurtd|j}|jr:t||jt||j|S)Nzlistener is closed)rrCacceptrdeliver_challengeanswer_challenge)r:crrrrs    zListener.acceptcCs |j}|durd|_|dSr)rrG)r:ZlistenerrrrrGszListener.closecCs|jjSr)r_addressrArrrr3szListener.addresscCs|jjSr)r_last_acceptedrArrr last_acceptedszListener.last_acceptedcCs|SrrrArrrrkszListener.__enter__cCs |dSrrlrmrrrrpszListener.__exit__)NNrN) rqrrrsr>rrGrtr3rrkrprrrrrs    rcCsh|p t|}t||dkr&t|}nt|}|durHt|tsHtd|durdt||t|||Sr) r4r- PipeClient SocketClientrrNrrr)r3r'rrrrrrs    rTcCsj|r>t\}}|d|dt|}t|}n$t\}}t|dd}t|dd}||fS)NTFr=r<)r,Z socketpair setblockingrdetachr!pipe)duplexs1s2c1c2Zfd1Zfd2rrrrs       rc 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)NrrrTrvrr)r(rxPIPE_ACCESS_DUPLEX GENERIC_READ GENERIC_WRITEBUFSIZEZPIPE_ACCESS_INBOUNDCreateNamedPipeFILE_FLAG_OVERLAPPEDFILE_FLAG_FIRST_PIPE_INSTANCEPIPE_TYPE_MESSAGEPIPE_READMODE_MESSAGE PIPE_WAITNMPWAIT_WAIT_FOREVERNULL CreateFile OPEN_EXISTINGSetNamedPipeHandleStateConnectNamedPiper}ru) rr3ZopenmodeaccessZobsizeZibsizeZh1Zh2rw_rrrrrrrs<          c@s&eZdZd ddZddZddZdS) rrcCsttt||_zRtjdkr2|jtjtjd|jd|j ||j ||j |_ Wnt y|jYn0||_d|_|dkrt|stj|tj|fdd|_nd|_dS)NposixrTrrargsZ exitpriority)r,getattr_socketr!nameZ setsockoptZ SOL_SOCKETZ SO_REUSEADDRrZbindZlistenZ getsocknamerrCrGZ_familyrrr2Finalizeunlink_unlink)r:r3r'rrrrr>Gs(        zSocketListener.__init__cCs&|j\}|_|dt|SNT)rrrrrrr:srrrr`s zSocketListener.acceptcCsJz(|jW|j}|durFd|_|n|j}|durDd|_|0dSr)rrGr)r:rrrrrGes zSocketListener.closeN)r)rqrrrsr>rrGrrrrrCs rcCs\t|}ttt|0}|d||t|WdS1sN0YdSr)r4r,rrZconnectrr)r3r'rrrrros   rc@s4eZdZd ddZd ddZddZed d ZdS) rNcCsL||_|jddg|_d|_td|jtj|tj|j|jfdd|_ dS)NT)firstz listener created with address=%rrr) r _new_handle _handle_queuerr sub_debugrr_finalize_pipe_listenerrG)r:r3rrrrr>s zPipeListener.__init__Fc CsHtjtjB}|r|tjO}t|j|tjtjBtjBtj t t tj tj Sr) rxrrrrrrrrZPIPE_UNLIMITED_INSTANCESrrr)r:rflagsrrrrs   zPipeListener._new_handlec Cs|j||jd}ztj|dd}Wn2ty`}z|jtjkrLWYd}~nfd}~00zJzt |j gdt }Wn | t |Yn0W|d\}}n|d\}}0t|S)NrTrvF)rappendrpoprxrrCrZ ERROR_NO_DATArzr{rr|rr}ru)r:r;r~rresrrrrrrs(     zPipeListener.acceptcCs$td||D]}t|qdS)Nz closing listener with address=%r)rrrxr)Zqueuer3r;rrrrs z$PipeListener._finalize_pipe_listener)N)F)rqrrrsr>rr staticmethodrrrrrrs  rc Cst}z6t|dt|tjtjBdtjtjtjtj}Wqt y|}z(|j tj tj fvsft |rhWYd}~qd}~00qqt|tjddt|S)Nr)rrxZ WaitNamedPiperrrrrrrCrZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYrrrru)r3rhrrrrrs*    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)hmacrrNr%formatr.r!urandomMESSAGE_LENGTHrV CHALLENGEnewdigestr^WELCOMEFAILURErZ connectionrrmessagerZresponserrrrs      rcCsxddl}t|ts$tdt||d}|ttd}| ||d }| ||d}|t krtt ddS)Nrrrrzdigest sent was rejected)rrrNr%rr.r^rOrrrrVrrrrrrrs     rc@s$eZdZddZddZddZdS)ConnectionWrappercCs6||_||_||_dD]}t||}t|||qdS)N)rKrGrjr^rV)_conn_dumps_loadsrsetattr)r:connrXrdattrrZrrrr>s  zConnectionWrapper.__init__cCs||}|j|dSr)rrrV)r:rZrrrrr[s zConnectionWrapper.sendcCs|j}||Sr)rr^rrrrrrfs zConnectionWrapper.recvN)rqrrrsr>r[rfrrrrrsrcCst|fdddddS)Nrutf-8) xmlrpclibrXencode)rZrrr _xml_dumps srcCst|d\\}}|S)Nr)rrddecode)rrZmethodrrr _xml_loadssrc@seZdZddZdS) XmlListenercCs"ddlmat|}t|ttSr) xmlrpc.clientclientrrrrrrrYrrrrs  zXmlListener.acceptN)rqrrrsrrrrrr sr cOs"ddlmatt|i|ttSr)r r 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) listrxrzr r rOr rr)ZhandlesrLreadyrrrr_exhaustive_wait%s    rc 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)Nrrg?rKT)rFrc3s|]}|VqdSrr).0r)waithandle_to_objrr zwait..csg|]}|vr|qSrr)ro) ready_objectsrr rwait..)rintrsetrAttributeErrorr6rxrrCr _ready_errorsryrr{r)Zgetwindowsversionr}r+raddrkeysr|ZERROR_OPERATION_ABORTEDupdate) object_listrZov_listZ ready_handlesrrKr~rrrr)rrrr;s                       r PollSelectorcCst}|D]}||tjq |dur4t|}||}|r^dd|DWdS|dur4|t}|dkr4|WdSq4Wdn1s0YdS)NcSsg|]\}}|jqSr)Zfileobj)rkeyZeventsrrrrrrr) _WaitSelectorregister selectorsZ EVENT_READrrZselect)r#rZselectorrZZdeadlinerrrrrs   cCsf|}t|tjtj8}ddlm}||}t||j |j ffWdS1sX0YdS)Nr)resource_sharer) rKr,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<rxZFILE_GENERIC_READr=ZFILE_GENERIC_WRITEr Z DupHandlerKrebuild_pipe_connection)rrdhrrrreduce_pipe_connections r0cCs|}t|||Sr)rru)r/r<r=r;rrrr.sr.cCs t|}t||j|jffSr)r ZDupFdrKr*r<r=)rdfrrrr,scCs|}t|||Srr-)r1r<r=fdrrrr*s)NN)T)T)N)N)I__all__rr!r)r,rrr itertoolsrrrrr contextr ZForkingPicklerrWrxr r r r ImportErrorr*rZCONNECTION_TIMEOUTcountr$rZfamiliesr+rrr(r-r4r5rurobjectrrrrrrrrrrrrrrrrr r rrZERROR_NETNAME_DELETEDrrr(r$r&ZSelectSelectorr,r*r'r0r.rrrr s           PT=    ,, 8   P