← Back
Editing: 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
Save File
Cancel