mg* dZddlmZmZmZmZddlmZmZmZm Z ddl m Z ddl m Z ddlmZerddlmZmZegd fZGd d eeZGd d eeZGddeeZGdd eeZGdde eZy)zSSH subprocess handlers) TYPE_CHECKINGAnyAnyStrCallable)DictGenericIterableOptional)EXTENDED_DATA_STDERR)SSHClientProcess)DataType) SSHChannelSSHClientChannelSSHSubprocessProtocolcBeZdZdZ d dddefdZd dZd d ed edefd Z y)SSHSubprocessPipezSSH subprocess pipeNchanSSHClientChannel[AnyStr]datatypec ||_||_yN)_chan _datatype)selfrrs 3lib/python3.12/site-packages/asyncssh/subprocess.py__init__zSSHSubprocessPipe.__init__*s15 !returnc8|jjy)zShut down the remote processN)rclosers rr!zSSHSubprocessPipe.close/ rnamedefaultc:|jj||S)a>Return additional information about the remote process This method returns extra information about the channel associated with this subprocess. See :meth:`get_extra_info() ` on :class:`SSHClientChannel` for additional information. )rget_extra_info)rr$r%s rr'z SSHSubprocessPipe.get_extra_info4szz((w77rrrN) __name__ __module__ __qualname____doc__rrr!strrr'rrrr's>'+"7"#"  83 8 8 8rrc eZdZdZddZddZy)SSHSubprocessReadPipezSSH subprocess pipe readerNc8|jjy)z7Pause delivery of incoming data from the remote processN)r pause_readingr"s rr2z#SSHSubprocessReadPipe.pause_readingDs   "rc8|jjy)z8Resume delivery of incoming data from the remote processN)rresume_readingr"s rr4z$SSHSubprocessReadPipe.resume_readingIs !!#rr()r)r*r+r,r2r4r.rrr0r0As$# $rr0ceZdZdZddZdefdZdefdZ dde ede eddfd Z d e ddfd Z d e e ddfd ZddZy)SSHSubprocessWritePipezSSH subprocess pipe writerrNc8|jjy)z0Forcibly close the channel to the remote processN)rabortr"s rr8zSSHSubprocessWritePipe.abortRr#rc6|jjS)z2Return whether the pipe supports :meth:`write_eof`)r can_write_eofr"s rr:z$SSHSubprocessWritePipe.can_write_eofWszz''))rc6|jjS)z3Return the current size of the pipe's output buffer)rget_write_buffer_sizer"s rr<z,SSHSubprocessWritePipe.get_write_buffer_size\szz//11rhighlowc<|jj||y)z9Set the high- and low-water limits for write flow controlN)rset_write_buffer_limits)rr=r>s rr@z.SSHSubprocessWritePipe.set_write_buffer_limitsas **45rdatacP|jj||jy)zWrite data on this pipeN)rwriter)rrAs rrCzSSHSubprocessWritePipe.writegs t~~.r list_of_datacP|jj||jy)z'Write a list of data bytes on this pipeN)r writelinesr)rrDs rrFz!SSHSubprocessWritePipe.writelinesls lDNN;rc8|jjy)zWrite EOF on this pipeN)r write_eofr"s rrHz SSHSubprocessWritePipe.write_eofqs rr()NN)r)r*r+r,r8boolr:intr<r r@rrCr rFrHr.rrr6r6Os$ *t* 2s2 =A596HSM6%-c]6>B6 /&/T/ ` on the transport provided in :meth:`connection_made`. Nr.r"s rprocess_exitedz$SSHSubprocessProtocol.process_exitedrOr)rMzSSHSubprocessTransport[AnyStr]rNr() r)r*r+r,rNrJrrRr ExceptionrUrWr.rrrrwsZ   #C  HL   S  4 & s )1D  & rc eZdZUdZded<deffd ZdeefdZ dfd Z dd Z d e e ddffd Zd ededdfdZdeddffd Zdededededdf fd Zde efdZdede eefdZde efdZxZS)SSHSubprocessTransportaSSH subprocess transport This class conforms to :class:`asyncio.SubprocessTransport`, but with the following enhancements: * All functionality available through :class:`SSHClientProcess` is also available here, such as the ability to dynamically redirect stdin, stdout, and stderr at any time during the lifetime of the process. * If encoding is set when the subprocess is created, all data written to the transports created by :meth:`get_pipe_transport` should be strings containing Unicode data. The encoding defaults to `None`, though, to preserve compatibility with :class:`asyncio.SubprocessTransport`, which expects data to be written as bytes. rrprotocol_factorycHt|i|_||_yr)superr_pipes _protocol)rr[ __class__s rrzSSHSubprocessTransport.__init__s <> 8H8Jrrc|jS)z=Return the subprocess protocol associated with this transport)r_r"s r get_protocolz#SSHSubprocessTransport.get_protocols~~rNct|||jj|t|jt |jt |jt d|_y)zHandle a newly opened channel)rr N)r]rNr_r6rr0r r^)rrr`s rrNz&SSHSubprocessTransport.connection_madesV % &&t,&djj1$TZZ0$TZZ1EF  rcy)zOverride SSHClientProcess to avoid creating SSHReader/SSHWriter streams, since this class uses read/write pipe objects insteadNr.r"s rsession_startedz&SSHSubprocessTransport.session_startedrOrrSc|jjd||jjd|t| |y)z Handle an incoming channel closer rdN)r_rUr]connection_lost)rrSr`s rrhz&SSHSubprocessTransport.connection_losts: ++As3 ++As3 $rrArc|jj|}|r|j|y|tk(rdnd}|jj ||y)z,Handle incoming data from the remote processrdr N)_writersgetrCr r_rR)rrArwriterrPs r data_receivedz$SSHSubprocessTransport.data_receivedsJ""8,  LL "66AB NN - -b$ 7rstatuscXt|||jjy)z)Handle exit status for the remote processN)r]exit_status_receivedr_rW)rrnr`s rrpz+SSHSubprocessTransport.exit_status_receiveds" $V, %%'rsignal core_dumpedmsglangc^t||||||jjy)z)Handle exit signal for the remote processN)r]exit_signal_receivedr_rW)rrqrrrsrtr`s rrvz+SSHSubprocessTransport.exit_signal_receiveds( $V[#tD %%'rcy)zReturn the PID of the remote process This method always returns `None`, since SSH doesn't report remote PIDs. Nr.r"s rget_pidzSSHSubprocessTransport.get_pidsrrPc8|jj|S)amReturn a transport for the requested stream :param fd: The integer file descriptor (0-2) to return the transport for, where 0 means stdin, 1 means stdout, and 2 means stderr. :type fd: `int` :returns: an :class:`SSHSubprocessReadPipe` or :class:`SSHSubprocessWritePipe` )r^rk)rrPs rget_pipe_transportz)SSHSubprocessTransport.get_pipe_transport!s{{r""rc|jS)aReturn the exit status or signal for the remote process This method returns the exit status of the session if one has been sent. If an exit signal was sent, this method returns the negative of the numeric value of that signal, matching the behavior of :meth:`asyncio.SubprocessTransport.get_returncode`. If neither has been sent, this method returns `None`. :returns: `int` or `None` ) returncoder"s rget_returncodez%SSHSubprocessTransport.get_returncode1sr)rzSSHChannel[AnyStr]rNr()r)r*r+r,__annotations__SubprocessFactoryrrrrbrNrfr rXrhrrmrJrpr-rIrvrxrrzr} __classcell__)r`s@rrZrZs& &%K):K 3F;  M%8I#6%4% 8& 8H 8 8(3(4( (3(T("%(-0(59( # #S# &v. /#  rrZN)r,typingrrrrrrr r constantsr processr sessionrchannelrrrrr0r6rrZr.rrrs*7744+%5R!889884 $-f5 $%.v6%PJ GFOJ Zz-f5zr