o svXj@srddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl m Z ddl mZmZddlmZddlmZmZdd lmZdd l mZzdd lmZWneyqejd krmdZYnwejd krejZejZejZejZgd Ze edZ!dZ"dZ#e $Z%dZ&dgZ'e edrdZ&e'dg7Z'ejd krdZ&e'dg7Z'e#fddZ(ddZ)ddZ*ddZ+ddZ,Gddde-Z.Gd d!d!e-Z/erGd"d#d#e/Z0Gd$d%d%e/Z1Gd&d'd'e-Z2d^d(d)Z3d*d+Z4ejd krd_d.d/Z5nd_d0d/Z5Gd1d2d2e-Z6d3d4Z7ejd kr/Gd5d6d6e-Z8ej9ej:ffd7d8Z;d9ZdZ@d?d@ZAGdAdBdBe-ZBdCdDZCdEdFZDGdGdHdHe2ZEdIdJZFejd krqdKdLZGejHejIhZJd`dMdNZKne edOr|dPdQZLndRdQZLd`dSdNZKejd krdTdUZMdVdWZNe Oe1eMdXdYZPdZd[ZQe Oe0ePdSd\dUZMd]dWZNe Oe1eMdS)a)absolute_importN) reduction)util)AuthenticationErrorBufferTooShort) _billiard) setblocking send_offset monotonic)ForkingPickler)_winapiwin32)ClientListenerPipewaitpypy_version_infoi g4@AF_INETAF_UNIXAF_PIPEcCs t|SNr )timeoutrP/var/www/html/myproject/venv/lib/python3.10/site-packages/billiard/connection.py _init_timeoutH rcCs t|kSrr )trrr_check_timeoutLrrcCsT|dkrdS|dkrtjdtdS|dkr&tjdtttfddStd ) z? Return an arbitrary free address for the given family r) localhostrrz listener-)prefixdirrz\\.\pipe\pyc-%d-%d-zunrecognized family) tempfilemktempr get_temp_dirosgetpidnext _mmap_counter ValueErrorfamilyrrrarbitrary_addressTsr.cCsRtjdkr|dkrtd|tjdkr#|dkr%tt|s'td|dSdSdS)zD Checks if the family is valid for the current environment. rrzFamily %s is not recognized.rN)sysplatformr+hasattrsocketr,rrr_validate_familycs   r3cCsFt|tkrdSt|tur|drdSt|turdStd|)z] Return the types of the address This can be 'AF_INET', 'AF_UNIX', or 'AF_PIPE' rz\\rrzaddress type of %r unrecognized)typetuplestr startswithr+)addressrrr address_typeps   r9c@eZdZddZdS)_SocketContainercCs ||_dSrsock)selfr=rrr__init__rz_SocketContainer.__init__N)__name__ __module__ __qualname__r?rrrrr; r;c@seZdZdZd/ddZddZddZd d Zd d Zd dZ e ddZ e ddZ e ddZ ddZddZd0ddZddZd1ddZd2d d!Zd"d#Zd3d%d&Zd'd(Zd)d*Zd+d,Zd-d.ZdS)4_ConnectionBaseNTcCsZt|tr|j|_|j}|}|dkrtd|s"|s"td||_||_||_ dS)Nrzinvalid handlez6at least one of `readable` and `writable` must be True) isinstancer;r=_socketfileno __index__r+_handle _readable _writable)r>handlereadablewritablerrrr?s   z_ConnectionBase.__init__cCs|jdur |dSdSrrI_closer>rrr__del__s  z_ConnectionBase.__del__cCs|jdur tddS)Nzhandle is closed)rIOSErrorrQrrr _check_closeds z_ConnectionBase._check_closedcC|jstddS)Nzconnection is write-only)rJrSrQrrr_check_readablez_ConnectionBase._check_readablecCrU)Nzconnection is read-only)rKrSrQrrr_check_writablerWz_ConnectionBase._check_writablecCs$|jr d|_td|td)NFzbad message length)rKrJcloserSrQrrr_bad_message_lengths z#_ConnectionBase._bad_message_lengthcCs |jduS)z True if the connection is closedNrIrQrrrcloseds z_ConnectionBase.closedcC|jS)z"True if the connection is readable)rJrQrrrrMz_ConnectionBase.readablecCr])z"True if the connection is writable)rKrQrrrrNr^z_ConnectionBase.writablecCs||jS)z+File descriptor or handle of the connection)rTrIrQrrrrGsz_ConnectionBase.filenocCs,|jdurz |Wd|_dSd|_wdS)zClose the connectionNrOrQrrrrYs  z_ConnectionBase.closercCs||t|}|jdkrtt|}t|}|dkr#td||kr+td|dur4||}n|dkrbufoffsetsizemnrrr send_bytess"    z_ConnectionBase.send_bytescCs$|||t|dS)zSend a (picklable) objectN)rTrXrcr dumpsr>objrrrsendsz_ConnectionBase.sendcCsJ|||dur|dkrtd||}|dur!||S)z7 Receive bytes data as a bytes object. Nrznegative maxlength)rTrVr+ _recv_bytesrZgetvalue)r> maxlengthrdrrr recv_bytess z_ConnectionBase.recv_bytescCs||t|K}|j}|t|}|dkrtd||kr&td|}|}|||kr:t| | d| |||||||WdS1sZwYdS)zq Receive bytes data into a writeable bytes-like object. Return the number of bytes read. rznegative offsetzoffset too largeN) rTrVr_r`rbr+rntellrroseekreadinto)r>rdrergr`bytesizeresultrfrrrrecv_bytes_intos&     $z_ConnectionBase.recv_bytes_intocCs"|||}t|S)zReceive a (picklable) object)rTrVrnr loadbuf)r>rdrrrrecvs z_ConnectionBase.recvcCs||||S)z/Whether there is any input available to be read)rTrV_pollr>rrrrpolls z_ConnectionBase.pollcC|SrrrQrrr __enter__z_ConnectionBase.__enter__cC |dSrrYr>exc_type exc_valueexc_tbrrr__exit__" z_ConnectionBase.__exit__cCst|||Sr)r rG)r>rdrerrrr %z_ConnectionBase.send_offsetcCst||dSr)r rG)r>blockingrrrr (sz_ConnectionBase.setblocking)TT)rNr)r)rz)r@rArBrIr?rRrTrVrXrZpropertyr\rMrNrGrYrirmrqrwryr}rrr r rrrrrDs4         rDc@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. FcC||jdSrr[)r> _CloseHandlerrrrP6zPipeConnection._closec Cstj|j|dd\}}z(z|tjkr!t|jgdt}|tks!JWn|W| d\}}n| d\}}w|dksBJ|t |ksJJdS)NT overlappedFr) r WriteFilerIERROR_IO_PENDINGWaitForMultipleObjectseventINFINITE WAIT_OBJECT_0cancelGetOverlappedResultrb)r>rdoverrwaitresnwrittenrrrrc9s     zPipeConnection._send_bytesNc Csh|jr d|_tS|durdnt|d}ztj|j|dd\}}zFz|tjkr7t|j gdt }|t ks7JWn| W| d\}}|dkr[t}|||WS|tjkrg|||WSn2| d\}}|dkrt}|||YWS|tjkr|||YWSwWtdty}z |jtjkrtd}~ww)NFTrrz.shouldn't get here; expected KeyboardInterrupt)_got_empty_messageioBytesIOminrReadFilerIrrrrrrrwrite getbufferERROR_MORE_DATA_get_more_datarSwinerrorERROR_BROKEN_PIPEEOFError RuntimeError) r>maxsizebsizerrrnreadferrrrnIsX       zPipeConnection._recv_bytescCs.|js t|jddkrdStt|g|S)NrT)rr PeekNamedPiperIboolrr|rrrr{kszPipeConnection._pollcCs|}t}||t|jd}|dksJ|dur+t|||kr+|tj |j|dd\}}| d\}}|dksCJ||ksIJ|||S)NrrTr) rrrrrrrIrbrZrr)r>rrrdrleftrrbytesrrrrqs    zPipeConnection._get_more_datar) r@rArB__doc__rr CloseHandlerPrcrnr{rrrrrr.s " rc@seZdZdZerejfddZejZ ej Z n e j fddZe jZ e jZ e fddZe fddZefd d Zdd d ZddZd S) Connectionzo Connection class based on an arbitrary file descriptor (Unix only), or a socket handle (Windows). cCrrr[r>rPrrrrPrzConnection._closecCrrr[rrrrrPrc Cs~t|} z||j|}Wn tttjfy-}zt|ddtjkr#WYd}~nd}~ww||8}|dkr8dS||d}q)NTerrnor) rbrIrSIOErrorr2errorgetattrrEINTR)r>rdr remainingrhexcrrr_sends  zConnection._sendc Cst}|j}|}|dkrTz|||}Wn tttjfy4}zt|ddtj kr*WYd}~n d}~wwt |}|dkrG||krCt td| |||8}|dks |S)Nrrzgot end of file during message) rrrIrSrr2rrrrrbrr) r>rfreadrdrLrchunkrrhrrr_recvs*  zConnection._recvcCsXt|}td|}|dkr||||dSt||r#|}|||dS)N!ii@)rbstructpackrrEtobytes)r>rdr_rhheaderrrrrcs   zConnection._send_bytesNcCs:|d}td|\}|dur||krdS||S)Nr)rrunpackro)r>rrdrfrrrrns  zConnection._recv_bytescCst|g|}t|Sr)rr)r>rrrrrr{s zConnection._pollr)r@rArBrrr closesocketrPrm_writery_readr'rYrrrrr_rcrnr{rrrrrs     rc@sReZdZdZdddZddZdd Zed d Zed d Z d dZ 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 |ot|p t}|pt|}t||dkrt|||_nt||||_|dur3t|ts3t d||_ dS)Nrauthkey should be a byte string) r9default_familyr.r3 PipeListener _listenerSocketListenerrEra TypeError_authkey)r>r8r-backlogauthkeyrrrr?s   zListener.__init__cCs>|jdur td|j}|jrt||jt||j|S)zz Accept a connection on the bound socket or named pipe of `self`. Returns a `Connection` object. Nzlistener is closed)rrSacceptrdeliver_challengeanswer_challenge)r>crrrrs    zListener.acceptcCs$|j}|durd|_|dSdS)zA Close the bound socket or named pipe of `self`. N)rrY)r>listenerrrrrYs  zListener.closecC|jjSr)r_addressrQrrrzListener.cCrr)r_last_acceptedrQrrrrrcCr~rrrQrrrrrzListener.__enter__cCrrrrrrrr rzListener.__exit__)NNrN) r@rArBrr?rrYrr8 last_acceptedrrrrrrrs   rcCsh|pt|}t||dkrt|}nt|}|dur$t|ts$td|dur2t||t|||S)z= Returns a connection to the address of a `Listener` rNr) r9r3 PipeClient SocketClientrErarrr)r8r-rrrrrrs    rcCst|dr |St|S)Ndetach)r1rr;r<rrrr$s rTFc Cs|r$t\}}|| || tt|}tt|}||fSt\}}|r1t|d|r8t|dt|dd}t|dd}||fS)L Returns pair of connection objects at either end of a pipe rFrNrM)r2 socketpairr rrr'pipe) duplex rnonblock wnonblocks1s2c1c2fd1fd2rrrr.s          rc Cs|rJd|r Jdtd}|r!tj}tjtjB}tt}}n tj}tj}dt}}t||tjBtj Btj tj Btj Bd||tj tj}t||dtjtjtjtj} t| tj ddtj|dd} | d\} } | dkssJt||d } t| |d }| |fS) rz"rnonblock not supported on windowsz"wnonblock not supported on windowsrrrNTrrr)r.rPIPE_ACCESS_DUPLEX GENERIC_READ GENERIC_WRITEBUFSIZEPIPE_ACCESS_INBOUNDCreateNamedPipeFILE_FLAG_OVERLAPPEDFILE_FLAG_FIRST_PIPE_INSTANCEPIPE_TYPE_MESSAGEPIPE_READMODE_MESSAGE PIPE_WAITNMPWAIT_WAIT_FOREVERNULL CreateFile OPEN_EXISTINGSetNamedPipeHandleStateConnectNamedPiperr)rrrr8openmodeaccessobsizeibsizeh1h2r_rrrrrrrEsB             c@s*eZdZdZd ddZddZddZd S) rzO Representation of a socket which is bound to an address and listening rcCsttt||_z)tjdkr|jtjtjd|jd|j ||j ||j |_ Wn t y?|jw||_d|_|dkrXtj|tj|fdd|_dSd|_dS)NposixrTrrargs exitpriority)r2rrFr'name setsockopt SOL_SOCKET SO_REUSEADDRr bindlisten getsocknamerrSrY_familyrrFinalizeunlink_unlink)r>r8r-rrrrr?xs*         zSocketListener.__init__c Csr z |j\}|_Wn tttjfy+}zt|ddtj kr!WYd}~nd}~wwnq| dt t |S)NTr) rFrrrSrr2rrrrr rr)r>srrrrrs   zSocketListener.acceptcCsNz|jW|j}|durd|_|dSdS|j}|dur&d|_|wwr)rFrYr)r>rrrrrYs  zSocketListener.closeN)r)r@rArBrr?rrYrrrrrts   rcCs8t|}ttt|}|d||tt|S)zO Return a connection object connected to the socket given by `address` T)r9r2rr connectrr)r8r-rrrrrs    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_listenerrY)r>r8rrrrr?s  zPipeListener.__init__Fc CsHtjtjB}|r |tjO}t|j|tjtjBtjBtj t t tj tj Sr) rrrrrrrrrPIPE_UNLIMITED_INSTANCESrrr)r>rflagsrrrrs   zPipeListener._new_handlec Cs|j||jd}z tj|dd}Wnty3}z|jtjkr&WYd}~t|Sd}~wwz+z t |j gdt Wn | t |W|d\}}|dks\Jt|S|d\}}|dksmJw)NrTrF)rappendrpoprrrSr ERROR_NO_DATArrrrrrr)r>rLrrr rrrrrs2      zPipeListener.acceptcCs$td||D]}t|qdS)Nz closing listener with address=%r)rr rr)queuer8rLrrrr!s  z$PipeListener._finalize_pipe_listenerr)F) r@rArBrr?rr staticmethodr!rrrrrs   rc Cst} zt|dt|tjtjBdtjtjtjtj}Wnt y;}z|j |vs0t |r1WYd}~nd}~wwnqt |tj ddt|S)zU Return a connection object connected to the pipe given by `address` rrN)rr WaitNamedPiperrrrrrrSrrrrr)r8_ignorerhrrrrrs*     rs #CHALLENGE#s #WELCOME#s #FAILURE#cCsrddl}t|ts Jtt}|t||||d }| d}||kr0|t dS|t t d)Nrmd5zdigest received was wrong)hmacrErar'urandomMESSAGE_LENGTHri CHALLENGEnewdigestrqWELCOMEFAILUREr connectionrr0messager5responserrrrs   rcCsddl}t|ts J|d}|dtttks Jd||ttd}|||d}|||d}|t krCt ddS)Nrr/z message = %rr.zdigest sent was rejected) r0rErarqrbr3r4r5rir6rr8rrrr"s    rc@s$eZdZddZddZddZdS)ConnectionWrappercCs6||_||_||_dD] }t||}t|||q dS)N)rGrYr}rqri)_conn_dumps_loadsrsetattr)r>connrjloadsattrrlrrrr?5s zConnectionWrapper.__init__cCs||}|j|dSr)r>r=ri)r>rlrrrrrm=s zConnectionWrapper.sendcCs|j}||Sr)r=rqr?)r>rrrrryAs  zConnectionWrapper.recvN)r@rArBr?rmryrrrrr<3s r<cCst|fdddd}|dS)Nrutf-8) xmlrpclibrjencode)rlorrr _xml_dumpsFs rHcCst|d\\}}|S)NrD)rErBdecode)rrlmethodrrr _xml_loadsKsrKc@r:) XmlListenercCs"ddlmat|}t|ttSNr) xmlrpc.clientclientrErrr<rHrKrkrrrrRs   zXmlListener.acceptN)r@rArBrrrrrrLPrCrLcOs"ddlmatt|i|ttSrM)rNrOrEr<rrHrK)rkwdsrrr XmlClientYs rQcCst|}g}|rZt|d|}|tkr |St|kr$tt|kr+nn|t8}nt|kr9tt|krCntd|t8}ntd|||||dd}d}|s|S)NFzShould not get hererr) listrr WAIT_TIMEOUTrrbWAIT_ABANDONED_0rr$)handlesrLreadyresrrr_exhaustive_waitds$   rYc sx|durt}n|dkrd}nt|dd}t|}ig}tt}z|D]W}zt|d}Wnty?||<Yq'wz t|dd\}}Wnt yf}z|j }|t vr\WYd}~nd}~ww|tj krw| |||j<q'|d}q't|}W|D]}|q|D]A}z |d\} }Wnt y}z|j }|t vrWYd}~nd}~ww|tjkrӈ|j}||dkrt|drd|_qnR|D]}|q|D]E}z |d\} }Wnt y}z|j }|t vrWYd}~nd}~ww|tjkr%|j}||dkr%t|dr%d|_qwfdd |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. Nrr)g?rGTrc3s|]}|VqdSrr).0r,)waithandle_to_objrr szwait..csg|]}|vr|qSrr)r[p) ready_objectsrr szwait..)rintrRsetrAttributeErrorrHrrrSr _ready_errorsrr$raddrYkeysrrERROR_OPERATION_ABORTEDr1rupdate) object_listrov_list ready_handlesrGrGrrrr r)r_r\rrzs                   rr}cCs|dur t|d}i}t}|D]}||tjt|dr'|||<q|||<qg}||D]\}}|tj@rBtd|| ||q3|S)Nr)rGzinvalid file descriptor %i) raselectr}registerPOLLINr1rGPOLLNVALr+r$)fdsrfd_mappollsterfdlsrrrrr{s     r{cCst|gg|dSrM)rl)rprrrrr{sc Cs|dur|dkr t|dSt|} zt||WStttjfy6}z |jtjkr,WYd}~nd}~ww|dur@|t}q)rZNr)r{r rSrr2rrr)rirdeadlinerrrrrs       cCsf|}t|tjtj}ddlm}||}t||j |j ffWdS1s,wYdS)Nr)resource_sharer) rGr2fromfdr SOCK_STREAMr#rv DupSocketrebuild_connectionrMrN)rArLrrvdsrrrreduce_connections   $r|cCs|}tt|||Sr)rr)r{rMrNr=rrrrzsrzcCsB|jrtjnd|jr tjndB}t||}t||j|jffSrM) rMrFILE_GENERIC_READrNFILE_GENERIC_WRITEr DupHandlerGrebuild_pipe_connection)rArdhrrrreduce_pipe_connections rcCtt|||Sr)rr)rrMrNrrrrrrcCs t|}t||j|jffSr)rDupFdrGrzrMrN)rAdfrrrr|scCrr)rr)rrMrNrrrrz r)NN)TFFr)R __future__rrrr'r/r2rlrr$ itertoolsr#rrrr_extrcompatr r fiver r r ImportErrorr0rrTrSr__all__r1is_pypyrCONNECTION_TIMEOUTcountr*rfamiliesrrr.r3r9objectr;rDrrrrrrrrrERROR_SEM_TIMEOUTERROR_PIPE_BUSYrr2r3r6r7rrr<rHrKrLrQrYrERROR_NETNAME_DELETEDrdrr{r|rzrmrrrrrrs                #SV 8   /2 8     G