a  ze}@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)timeoutrN/opt/bitninja-python-dojo/embedded/lib/python3.9/multiprocessing/connection.py _init_timeout;srcCs t|kSrr)trrr_check_timeout>srcCsv|dkr dS|dkrDtjr2dtdttStjdtdS|dkrjtjd tttfd dSt d d S) z? Return an arbitrary free address for the given family r) localhostrrz listener--z listener-)prefixdirrz\\.\pipe\pyc-%d-%d-zunrecognized familyN) rabstract_sockets_supportedosgetpidnext _mmap_countertempfilemktemp get_temp_dir 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_familyXs   r2cCsTt|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_typeds r9c@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__zsz_ConnectionBase.__init__cCs|jdur|dSrr<_closer?rrr__del__s z_ConnectionBase.__del__cCs|jdurtddS)Nzhandle is closed)r<OSErrorrFrrr _check_closeds z_ConnectionBase._check_closedcCs|jstddS)Nzconnection is write-only)r=rHrFrrr_check_readablesz_ConnectionBase._check_readablecCs|jstddS)Nzconnection is read-only)r>rHrFrrr_check_writablesz_ConnectionBase._check_writablecCs"|jrd|_n|tddS)NFzbad message length)r>r=closerHrFrrr_bad_message_lengthsz#_ConnectionBase._bad_message_lengthcCs |jduS)z True if the connection is closedNr<rFrrrclosedsz_ConnectionBase.closedcCs|jS)z"True if the connection is readable)r=rFrrrrAsz_ConnectionBase.readablecCs|jS)z"True if the connection is writable)r>rFrrrrBsz_ConnectionBase.writablecCs||jS)z+File descriptor or handle of the connection)rIr<rFrrrfilenosz_ConnectionBase.filenocCs*|jdur&z|Wd|_nd|_0dS)zClose the connectionNrDrFrrrrLs  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)rIrK memoryviewitemsizebyteslenr* _send_bytes)r?bufoffsetsizemnrrr send_bytess"     z_ConnectionBase.send_bytescCs$|||t|dS)zSend a (picklable) objectN)rIrKrU_ForkingPicklerdumpsr?objrrrsendsz_ConnectionBase.sendcCsJ|||dur(|dkr(td||}|durB||S)z7 Receive bytes data as a bytes object. Nrznegative maxlength)rIrJr* _recv_bytesrMgetvalue)r? maxlengthrVrrr 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) rIrJrQrRrTr*ratellr rbseekreadinto)r?rVrWrYrRbytesizeresultrXrrrrecv_bytes_intos$        z_ConnectionBase.recv_bytes_intocCs&|||}t|S)zReceive a (picklable) object)rIrJrar\loads getbuffer)r?rVrrrrecvsz_ConnectionBase.recvcCs||||S)z/Whether there is any input available to be read)rIrJ_pollr?rrrrpollsz_ConnectionBase.pollcCs|SrrrFrrr __enter__sz_ConnectionBase.__enter__cCs |dSrrLr?exc_type exc_valueexc_tbrrr__exit__ sz_ConnectionBase.__exit__)TT)rN)N)r)rn)__name__ __module__ __qualname__r<rCrGrIrJrKrMpropertyrOrArBrPrLr[r`rdrjrmrqrrrxrrrrr:ws.       r:c@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||jdSrrN)r?Z _CloseHandlerrrrEszPipeConnection._closec Csvtj|j|dd\}}zJz |tjkr6t|jgdt}Wn|Yn0W|d\}}n|d\}}0dS)NT overlappedF) _winapiZ WriteFiler<ERROR_IO_PENDINGWaitForMultipleObjectseventrcancelGetOverlappedResult)r?rVoverrwaitresZnwrittenrrrrUs    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)NFTr~rz.shouldn't get here; expected KeyboardInterrupt)_got_empty_messageioBytesIOminrReadFiler<rrrrrrwriterlZERROR_MORE_DATA_get_more_datarHwinerrorERROR_BROKEN_PIPEEOFError RuntimeError) r?maxsizeZbsizerrrZnreadferrrra+sH         zPipeConnection._recv_bytescCs.|jst|jddkrdStt|g|S)NrT)rr PeekNamedPiper<boolrrprrrroKs zPipeConnection._pollcCs|}t}||t|jd}|durJt|||krJ|tj |j|dd\}}| d\}}|||S)NrTr~) rlrrrrrr<rTrMrr)r?rrrVrleftrZrbytesrrrrQs zPipeConnection._get_more_data)N) ryrzr{__doc__rr CloseHandlerErUrarorrrrrr}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||jdSrrNr?rErrrrEhszConnection._closecCs||jdSrrNrrrrrEmscCs8t|}||j|}||8}|dkr&q4||d}qdSNr)rTr<)r?rVr remainingrZrrr_sendrs  zConnection._sendcCsbt}|j}|}|dkr^|||}t|}|dkrJ||krBtntd||||8}q|S)Nrzgot end of file during message)rrr<rTrrHr)r?rXreadrVr@rchunkrZrrr_recv{s   zConnection._recvcCst|}|dkrHtdd}td|}||||||n8td|}|dkrr||||n|||dS)Ni!i!Qi@)rTstructpackr)r?rVrZ pre_headerheaderrrrrUs        zConnection._send_bytesNcCs^|d}td|\}|dkr@|d}td|\}|durT||krTdS||S)Nrrr)rrunpackrb)r?rrVrXrrrras  zConnection._recv_bytescCst|g|}t|Sr)rr)r?rrrrrros zConnection._poll)N)ryrzr{rr_multiprocessingZ closesocketrEr`_writerm_readr#rLrrrrrUrarorrrrras  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) r9default_familyr-r2 PipeListener _listenerSocketListener isinstancerS TypeError_authkey)r?r8r,backlogauthkeyrrrrCs 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)rrHacceptrdeliver_challengeanswer_challenge)r?crrrrs    zListener.acceptcCs |j}|durd|_|dS)zA Close the bound socket or named pipe of `self`. N)rrL)r?listenerrrrrLszListener.closecCs|jjSr)r_addressrFrrrr8szListener.addresscCs|jjSr)r_last_acceptedrFrrr last_acceptedszListener.last_acceptedcCs|SrrrFrrrrrszListener.__enter__cCs |dSrrsrtrrrrxszListener.__exit__)NNrN) ryrzr{rrCrrLr|r8rrrrxrrrrrs    rcCsh|p t|}t||dkr&t|}nt|}|durHt|tsHtd|durdt||t|||S)z= Returns a connection to the address of a `Listener` rNr) r9r2 PipeClient SocketClientrrSrrr)r8r,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 TFrBrA)r1 socketpair setblockingrdetachr#pipe)duplexs1s2c1c2fd1fd2rrrr s       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) rrrrNTr~rr)r-rPIPE_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}) rr8ZopenmodeaccessZobsizeZibsizeZh1Zh2r_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)NposixrTrrargs exitpriority)r1getattr_socketr#name setsockopt SOL_SOCKET SO_REUSEADDRrbindlisten getsocknamerrHrL_familyrrr7Finalizeunlink_unlink)r?r8r,rrrrrCLs(        zSocketListener.__init__cCs&|j\}|_|dt|S)NT)rrrrrrr?srrrres zSocketListener.acceptcCsJz(|jW|j}|durFd|_|n|j}|durDd|_|0dSr)rrLr)r?rrrrrLjs zSocketListener.closeN)r)ryrzr{rrCrrLrrrrrHs rcCs\t|}ttt|0}|d||t|WdS1sN0YdS)zO Return a connection object connected to the socket given by `address` TN)r9r1rrconnectrr)r8r,rrrrrts   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_listenerrL)r?r8rrrrrCs zPipeListener.__init__Fc CsHtjtjB}|r|tjO}t|j|tjtjBtjBtj t t tj tj Sr) rrrrrrrrrZPIPE_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)NrTr~F)rappendrpoprrrHrZ ERROR_NO_DATArrrrrrr})r?r@rrresrrrrrrs(     zPipeListener.acceptcCs$td||D]}t|qdS)Nz closing listener with address=%r)rrrr)queuer8r@rrrrs z$PipeListener._finalize_pipe_listener)N)F) ryrzr{rrCrr 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)rrZ WaitNamedPiperrrrrrrHrZERROR_SEM_TIMEOUTZERROR_PIPE_BUSYrrrr})r8rhrrrrrs*    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)hmacrrSr*formatr3r#urandomMESSAGE_LENGTHr[ CHALLENGEnewdigestrdWELCOMEFAILUREr connectionrr messagerresponserrrrs      rcCsxddl}t|ts$tdt||d}|ttd}| ||d }| ||d}|t krtt ddS)Nrr r r zdigest sent was rejected)r rrSr*r r3rdrTrrrr[rrrrrrrs     rc@s$eZdZddZddZddZdS)ConnectionWrappercCs6||_||_||_dD]}t||}t|||qdS)N)rPrLrqrdr[)_conn_dumps_loadsrsetattr)r?connr]rkattrr_rrrrCs  zConnectionWrapper.__init__cCs||}|j|dSr)rrr[)r?r_rrrrr` s zConnectionWrapper.sendcCs|j}||Sr)rrdrrrrrrm s zConnectionWrapper.recvN)ryrzr{rCr`rmrrrrrsrcCst|fdddddS)Nrutf-8) xmlrpclibr]encode)r_rrr _xml_dumpssr#cCst|d\\}}|S)Nr )r!rkdecode)rr_methodrrr _xml_loadssr&c@seZdZddZdS) XmlListenercCs"ddlmat|}t|ttSr) xmlrpc.clientclientr!rrrr#r&r^rrrrs  zXmlListener.acceptN)ryrzr{rrrrrr'sr'cOs"ddlmatt|i|ttSr)r(r)r!rrr#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) listrrr r rTr 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?rPT)r1Frc3s|]}|VqdSrr).0r)waithandle_to_objrr zwait..csg|]}|vr|qSrr)r3o) ready_objectsrr r6wait..)rintr,setrAttributeErrorr;rrrHr _ready_errorsrrrr.Zgetwindowsversionrr0raddr/keysrZERROR_OPERATION_ABORTEDupdate) object_listrZov_listZ ready_handlesr7rPrrrrr)r8r4rr@s                       r PollSelectorcCst}|D]}||tjq |dur4t|}||}|r^dd|DWdS|dur4|t}|dkr4|WdSq4Wdn1s0YdS)r0NcSsg|]\}}|jqSr)fileobj)r3keyeventsrrrr9r6r:r) _WaitSelectorregister selectors EVENT_READrrselect)rBrselectorr_deadliner.rrrrs   cCsf|}t|tjtj8}ddlm}||}t||j |j ffWdS1sX0YdS)Nr)resource_sharer) rPr1fromfdr SOCK_STREAMr!rNZ DupSocketrebuild_connectionrArB)rr@rrNdsrrrreduce_connections   rScCs|}t|||Srrr)rRrArBsockrrrrQsrQcCsB|jr tjnd|jrtjndB}t||}t||j|jffSr) rArZFILE_GENERIC_READrBZFILE_GENERIC_WRITEr Z DupHandlerPrebuild_pipe_connection)rrdhrrrreduce_pipe_connections rXcCs|}t|||Sr)rr})rWrArBr@rrrrVsrVcCs t|}t||j|jffSr)r DupFdrPrQrArB)rdfrrrrSscCs|}t|||SrrT)rZrArBfdrrrrQs)NN)T)T)N)N)I__all__rr#r.r1rrr' itertoolsrr!rrr contextr ForkingPicklerr\rr r r r ImportErrorr/rCONNECTION_TIMEOUTcountr&rfamiliesr0rrr-r2r9r:r}robjectrrrrrrrrrrrrrrr#r&r'r+r/rZERROR_NETNAME_DELETEDr>rrIrCrGSelectSelectorrSrQrHrXrVrrrr s           PT=    ,, 8   P