o tBh'@sUdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ejdkrQddlZddlZeZejjZejZded <ejZded <nddlZejZGd d d eZGd ddZGdddeZGdddeZdS)zCross platform abstractions for inter-process communication On Unix, this uses AF_UNIX sockets. On Windows, this uses NamedPipes. ) annotationsN) TracebackType)Callable)Finalwin32zCallable[[_IPCHandle], int]DisconnectNamedPipeFlushFileBuffersc@seZdZdZdS) IPCExceptionzException for IPC issues.N)__name__ __module__ __qualname____doc__rr_/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/mypy/ipc.pyr #sr c@sDeZdZUdZded<dd d ZddddZdddZdddZdS)IPCBasezBase class for communication between the dmypy client and server. This contains logic shared between the client and server, such as reading and writing. _IPCHandle connectionnamestrtimeout float | NonereturnNonecCs||_||_dSN)rr)selfrrrrr__init__0s zIPCBase.__init__順sizeintbytesc Cst}tjdkrs tj|j|dd\}}z'|tjkr:|jr$t|jdntj }t |j |}|tj kr:t d|Wn tyG|w|d\}}|}|rZ|||dkrc t|S|tjkriq|tjkrrt dq |j|}|s t|S||qt)z2Read bytes from an IPC connection until its empty.rT overlappedBad result from I/O wait: rzReadFile operation aborted.) bytearraysysplatform_winapiReadFilerERROR_IO_PENDINGrrINFINITEWaitForSingleObjectevent WAIT_OBJECT_0r BaseExceptioncancelGetOverlappedResult getbufferextendERROR_MORE_DATAERROR_OPERATION_ABORTEDrecvr) rrbdataoverrrres_morerrrread4sD         z IPCBase.readdatac Cs<tjdkrzttj|j|dd\}}t|tjsJt|ts Jz3|tjkrG|j r0t|j dntj }t |j |}|tj krFtd|n |dkrRtd|Wn ty_|w|d\}}|dksoJ||t|kswJWd Sty}z td|j|d }~ww|j||jtjd S) z!Write bytes to an IPC connection.rTr r"r#rz#Failed writing to pipe with error: zFailed to write with error: N)r%r&r' WriteFiler isinstance Overlappedrr)rr*r+r,r-r r.r/r0lenOSErrorwinerrorsendallshutdownsocketSHUT_WR)rr=r7r8rr9 bytes_writtenerrrwriteWs8     z IPCBase.writecCs8tjdkr|jtjkrt|jdSdS|jdSNr)r%r&rr'NULL CloseHandlecloserrrrrNss  z IPCBase.closeNrrrrrr)r)rrrr)r=rrrrr) r r r r __annotations__rr<rJrNrrrrr's    #rcs>eZdZdZdfdd Zdd d Z ddddZZS) IPCClientz%The client side of an IPC connection.rrrrrrc sBt||tjdkr|jrt|jdntj}z t|j |Wn-t y8}z t d|j d|d}~wt yO}z |j tjkrJt d|d}~wwzt|j tjtjBdtjtjtjtj|_Wnt y}z |j tjkrzt d|d}~wwt|jtjdddSttj|_|j||j|dS)Nrr"zThe NamedPipe at z was not found.z!Timed out waiting for connection.rzThe connection is busy.)superrr%r&rrr'NMPWAIT_WAIT_FOREVER WaitNamedPiperFileNotFoundErrorr rBrCERROR_SEM_TIMEOUT CreateFile GENERIC_READ GENERIC_WRITErL OPEN_EXISTINGFILE_FLAG_OVERLAPPEDrERROR_PIPE_BUSYSetNamedPipeHandleStatePIPE_READMODE_MESSAGErFAF_UNIX settimeoutconnect)rrrrI __class__rrr~sH         zIPCClient.__init__cCs|SrrrOrrr __enter__szIPCClient.__enter__Nexc_tytype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NonecCs |dSr)rNrrgrirkrrr__exit__s zIPCClient.__exit__rP)rrSNNNrgrhrirjrkrlrr)r r r r rrfrn __classcell__rrrdrrS{s $rScsbeZdZUdZded<ddfd d Zdd dZ   ddddZd ddZe d!ddZ Z S)" IPCServerir BUFFER_SIZENrrrrrrc stjdkrd|ttd}n|d}t ||tjdkrXt |j t j t jBt jBt jt jBt jBdBd|j|jt jd|_|jdkrVt }td |dSt|_tj|j|j }ttj|_|j ||j!d|dur|j"|dSdS) Nrz\\.\pipe\{}-{}.pipez.sockrzInvalid handle to pipe: )#r%r&formatbase64urlsafe_b64encodeosurandomdecoderTrr'CreateNamedPiperPIPE_ACCESS_DUPLEXFILE_FLAG_FIRST_PIPE_INSTANCEr]r`PIPE_TYPE_MESSAGE PIPE_WAITrsrUr GetLastErrorr tempfilemkdtempsock_directorypathjoinrFrasockbindlistenrb)rrrr8sockfilerdrrrsN       zIPCServer.__init__c Cs$tjdkrtztj|jdd}t|tjsJWnty4}z|jtj tj fvr)WYd}~|Sd}~wwz|j r@t |j dntj }t|j|}|tjksQJWntyd|t|jw|d\}}|dksrJ|Sz |j\|_}W|Stj y}ztd|d}~ww)NrTr r"rzThe socket timed out)r%r&r'ConnectNamedPiperr?r@rBrCERROR_PIPE_CONNECTED ERROR_NO_DATArrr*r+r,r-r.r/rMr0racceptrFr )rr7rIrr9r:r8rrrrfs<      zIPCServer.__enter__rgrhrirjrkrlcCsFtjdkrzt|jstdWt|jdSt|jw|dS)Nrz:Failed to flush NamedPipe buffer,maybe the client hung up?)r%r&rrr rrNrmrrrrns   zIPCServer.__exit__cCs&tjdkr |dSt|jdSrK)r%r&rNshutilrmtreerrOrrrcleanups  zIPCServer.cleanupcCstjdkr|jS|jSrK)r%r&rr getsocknamerOrrrconnection_name s  zIPCServer.connection_namerrP)rrrrorprQ)rr) r r r rsrRrrfrnrpropertyrrqrrrdrrrs  $  rr)r __future__rryr{rr%rtypesrtypingrtyping_extensionsrr&ctypesr'rrwindllkernel32rrRrrF Exceptionr rrSrrrrrrs.     T3