File: //usr/lib/python3/dist-packages/twisted/conch/ssh/__pycache__/session.cpython-38.pyc
U
��W[�+ �
@ s> d Z ddlmZmZ ddlZddlZddlZddlZddlm Z ddl
m Z m Z ddl
mZ ddlmZmZ ddlmZ dd lmZmZmZ G d
d � d ej�ZG d d
� d
e j�Z G dd� d�Z dd� Z dd� Z ddddddddd d d d d g
Z e e j!�G d!d"� d"e j��Z"G d#d$� d$e j#�Z$d%d&� Z%d'd(� Z&d)d*� Z'd+d,� Z(dS )-z�
This module contains the implementation of SSHSession, which (by default)
allows access to a shell and a python interpreter over SSH.
Maintainer: Paul Swartz
� )�division�absolute_importN)� implementer)�
interfaces�protocol)�log)� _bytesChr�
networkString)�ISession)�common�channel�
connectionc @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
SSHSessions sessionc O s* t jj| f|�|� d| _d | _d | _d S �N� )r �
SSHChannel�__init__�buf�client�session)�self�args�kw� r �;/usr/lib/python3/dist-packages/twisted/conch/ssh/session.pyr s zSSHSession.__init__c C sr t �|�\}}t�d| � | j�||�}|r`t| �}t|�}|�|� |�t |�� || _
dS t�d� dS d S )Nzasking for subsystem "%s"� zfailed to get subsystemr ) r �getNSr �msg�avatarZlookupSubsystem�SSHSessionProcessProtocol�wrapProcessProtocolZmakeConnection� wrapProtocolr )r �dataZ subsystem�ignoredr �pp�protor r r �request_subsystem% s
z SSHSession.request_subsystemc C sZ t �d� | js t| j�| _zt| �}| j�|� W n t �� Y dS X || _dS d S )Nz
getting shellr r ) r r r r
r r Z openShell�deferrr )r r" r$ r r r �
request_shell4 s
zSSHSession.request_shellc C sn | j st| j�| _ t�|�\}}t�d| � zt| �}| j �||� W n t� � Y dS X || _
dS d S )Nzexecuting command "%s"r r ) r r
r r r r r r Z execCommandr' r )r r" �fr$ r r r � request_execB s zSSHSession.request_execc C sf | j st| j�| _ t|�\}}}t�d||f � z| j �|||� W n t�� Y dS X dS d S )Nzpty request: %r %rr r )r r
r �parseRequest_pty_reqr r ZgetPty�err)r r" �termZ
windowSize�modesr r r �request_pty_reqQ s zSSHSession.request_pty_reqc C sT | j st| j�| _ t|�}z| j �|� W n t�d� t�� Y dS X dS d S )Nzerror changing window sizer r )r r
r �parseRequest_window_changeZ
windowChangedr r r, )r r" �winSizer r r �request_window_change^ s
z SSHSession.request_window_changec C s* | j s| j|7 _d S | j j�|� d S �N)r r � transport�write�r r" r r r � dataReceivedk s zSSHSession.dataReceivedc C s@ |t jkr.| jr<t| jjd�r<| jj�|� nt�d| � d S )N�writeErrzweird extended data: %s)r
�EXTENDED_DATA_STDERRr �hasattrr4 r8 r r )r ZdataTyper" r r r � extReceivedr s
zSSHSession.extReceivedc C s( | j r| j �� n| jr$| j�| � d S r3 )r � eofReceivedr �connZ sendClose�r r r r r< y s zSSHSession.eofReceivedc C s( | j r| j �� n| jr$| jj�� d S r3 )r �closedr r4 �loseConnectionr> r r r r? s zSSHSession.closedc C s" | j r| j j�� tj�| � d S r3 )r r4 r@ r r r> r r r r@ � s zSSHSession.loseConnectionN)�__name__�
__module__� __qualname__�namer r& r( r* r/ r2 r7 r; r< r? r@ r r r r r s
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �_ProtocolWrapperzS
This class wraps a L{Protocol} instance in a L{ProcessProtocol} instance.
c C s
|| _ d S r3 �r% �r r% r r r r � s z_ProtocolWrapper.__init__c C s | j �� d S r3 )r% �connectionMader> r r r rH � r z _ProtocolWrapper.connectionMadec C s | j �|� d S r3 �r% r7 r6 r r r � outReceived� r z _ProtocolWrapper.outReceivedc C s | j �|� d S r3 )r% �connectionLost�r �reasonr r r � processEnded� r z _ProtocolWrapper.processEndedN)rA rB rC �__doc__r rH rJ rN r r r r rE � s
rE c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �_DummyTransportc C s
|| _ d S r3 rF rG r r r r � s z_DummyTransport.__init__c C s | j j�|� d S r3 )r% r4 r5 r6 r r r r7 � s z _DummyTransport.dataReceivedc C s | j �|� d S r3 rI r6 r r r r5 � s z_DummyTransport.writec C s | � d�|�� d S r )r5 �join�r �seqr r r �
writeSequence� s z _DummyTransport.writeSequencec C s | j �tj� d S r3 )r% rK r ZconnectionDoner> r r r r@ � s z _DummyTransport.loseConnectionN)rA rB rC r r7 r5 rT r@ r r r r rP � s
rP c C s t | tj�rt| �S | S d S r3 )�
isinstancer �ProtocolrE )�instr r r r � s r c C s t | �S r3 )rP rF r r r r! � s r! ZABRTZALRMZFPEZHUPZILLZINTZKILL�PIPEZQUITZSEGVZTERMZUSR1ZUSR2c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
d dd�Z dd� Z d dd�Z
dd� Zdd� Zdd� Zdd � Zd d � ZdS )!r z�I am both an L{IProcessProtocol} and an L{ITransport}.
I am a transport to the remote endpoint and a process protocol to the
local subsystem.
Nc C s || _ d| _d S )NF)r �lostOutOrErrFlag)r r r r r r � s z"SSHSessionProcessProtocol.__init__c C s$ | j jr | j�| j j� d | j _d S r3 )r r r4 r5 r> r r r rH � s z(SSHSessionProcessProtocol.connectionMadec C s | j �|� d S r3 �r r5 r6 r r r rJ � s z%SSHSessionProcessProtocol.outReceivedc C s | j �tj|� d S r3 )r Z
writeExtendedr
r9 )r r, r r r � errReceived� s z%SSHSessionProcessProtocol.errReceivedc C s" | j r| jj�| j� nd| _ dS )zW
EOF should only be sent when both STDOUT and STDERR have been closed.
TN)rY r r= ZsendEOFr> r r r �outConnectionLost� s z+SSHSessionProcessProtocol.outConnectionLostc C s | � � dS )z*
See outConnectionLost().
N)r\ r> r r r �errConnectionLost� s z+SSHSessionProcessProtocol.errConnectionLostc C s | j �� d S r3 �r r@ rL r r r rK � s z(SSHSessionProcessProtocol.connectionLostc C s� | j dkr�i | _ tD ]*}d| }tt|d�}|dk r|| j |< qtj�� D ]:\}}|�d�rJ|�d�sJ|| j krJ|d tj | j |< qJ| j | S )z:
Get a signal name given a signal number.
NZSIGZSIG_�@) �_signalValuesToNames�SUPPORTED_SIGNALS�getattr�signal�__dict__�items�
startswith�sys�platform)r Zsignum�signameZsigvalue�k�vr r r �_getSignalName� s
z(SSHSessionProcessProtocol._getSignalNamec
C s� |dk r�|j }|jdk r�| �|j�}ttdd�dk rVt�|j�rVt�d|f � d}nt�d|f � d}| j j
� | j dt �
t|dd� ��t|� t �
d � t �
d � � n:|jdk r�t�d
|jf � | j j
� | j d t�d |j�� | j �� dS )
z�
When we are told the process ended, try to notify the other side about
how the process ended using the exit-signal or exit-status requests.
Also, close the channel.
N� WCOREDUMPz exitSignal: %s (core dumped)r zexitSignal: %sr s exit-signal� r z exitCode: %rs exit-status�>L)�valuerc rl rb �osrm Zstatusr r r r= Z sendRequestr �NSr �chrZexitCode�struct�packr@ )r rM r, ri Z
coreDumpedr r r rN s4
� ���
�z&SSHSessionProcessProtocol.processEndedc C s | j jj�� S )z>
Return the host from my session's transport.
)r r= r4 �getHostr> r r r rv s z!SSHSessionProcessProtocol.getHostc C s | j jj�� S )z>
Return the peer from my session's transport.
)r r= r4 �getPeerr> r r r rw # s z!SSHSessionProcessProtocol.getPeerc C s | j �|� d S r3 rZ r6 r r r r5 * s z SSHSessionProcessProtocol.writec C s | j �d�|�� d S r )r r5 rQ rR r r r rT . s z'SSHSessionProcessProtocol.writeSequencec C s | j �� d S r3 r^ r> r r r r@ 2 s z(SSHSessionProcessProtocol.loseConnection)N)N)rA rB rC rO r` r rH rJ r[ r\ r] rK rl rN rv rw r5 rT r@ r r r r r � s
r c @ s e Zd Zdd� ZdS )�SSHSessionClientc C s | j r| j �|� d S r3 )r4 r5 r6 r r r r7 9 s z SSHSessionClient.dataReceivedN)rA rB rC r7 r r r r rx 7 s rx c sx t �| �\}}t�d|dd� �\}}}}t �|dd� �\� }||||f}� fdd�tdt� �d d�D �� ||� fS ) z�Parse the data from a pty-req request into usable data.
@returns: a tuple of (terminal type, (rows, cols, xpixel, ypixel), modes)
�>4LN� c
s@ g | ]8}t � ||d � �t�d� |d |d � �d f�qS )r ro � r )�ordrt �unpack)�.0�i�r. r r �
<listcomp>G s �z(parseRequest_pty_req.<locals>.<listcomp>r r r{ )r r rt r} �range�len) r" r- �rest�cols�rows�xpixel�ypixelr# r1 r r� r r+ > s
�r+ c
C s> |\}}}}t �| �}t�d||||�}t �|�} || | S )z�
Pack a pty-req request so that it is suitable for sending.
NOTE: modes must be packed before being sent here.
@type geometry: L{tuple}
@param geometry: A tuple of (rows, columns, xpixel, ypixel)
ry )r rr rt ru )
r- �geometryr. r� r� r� r� Z
termPackedZ
winSizePackedZ modesPackedr r r �packRequest_pty_reqK s
r� c C s t �d| �\}}}}||||fS )zzParse the data from a window-change request into usuable data.
@returns: a tuple of (rows, cols, xpixel, ypixel)
ry )rt r} )r" r� r� r� r� r r r r0 Z s r0 c C s | \}}}}t �d||||�S )z�
Pack a window-change request so that it is suitable for sending.
@type geometry: L{tuple}
@param geometry: A tuple of (rows, columns, xpixel, ypixel)
ry )rt ru )r� r� r� r� r� r r r �packRequest_window_changeb s r� ))rO Z
__future__r r rt rc rg rq Zzope.interfacer Ztwisted.internetr r Ztwisted.pythonr Ztwisted.python.compatr rs r Ztwisted.conch.interfacesr
Ztwisted.conch.sshr r r
r r ZProcessProtocolrE rP r r! ra Z
ITransportr rV rx r+ r� r0 r� r r r r �<module> s> q
�y