File: //lib/python3/dist-packages/twisted/conch/scripts/__pycache__/conch.cpython-38.pyc
U
��W[ H � @ s� d dl mZ d dlmZmZma d dlmZ d dlm Z m
Z
d dlm Z m Z m
Z
d dlmZmZmZ d dlmZmZ d dlmZmZmZ d d lZd d lZd d lZd d l Z d d l Z d d l Z d d l Z G d
d � d tj �Z!d ad a"d a#d a$d a%d a&d d
� Z'dd� Z(dd� Z)dd� Z*dd� Z+dd� Z,dd� Z-dd� Z.d d � Z/G d d � d �Z0G d d!� d!e j1�Z1G d"d#� d#e
j2�Z3G d$d%� d%e j4�Z4G d&d'� d'e j5�Z5d(d)� Z6d*d+� Z7e8d,k�r�e'� d S )-� )�print_function)�connect�default�options)�
ConchError)�
connection�common)�session�
forwarding�channel)�reactor�stdio�task)�log�usage)�ioType�
networkString�unicodeNc @ s� e Zd ZdZdZdddgdddd gd
d dd ggZd
ddgdddgdddgdddgdddgd d d ggZejd gej d d!�ej d d!�d"�e�
� ej d#d!�ej d$d%d&�gd'�Z g Z g Z
d(d)� Zd*d+� Zd,d-� Zd.d/� ZdS )0�
ClientOptionsz(Usage: conch [options] host [command]
zYconch is a SSHv2 client that allows logging into a remote machine and executing commands.�escape�e�~� localforward�LNz<listen-port:host:port Forward local port to remote address�
remoteforward�Rz<listen-port:host:port Forward remote port to local addressZnull�nz Redirect input from /dev/null.�fork�fz(Fork to background after authentication.�tty�tz-Tty; allocate a tty even if command is given.�notty�TzDo not allocate a tty.�noshell�Nz"Do not execute a shell or command.� subsystem�sz-Invoke command (mandatory) as SSH2 subsystem.)r r! zlisten-port:host:port)�descr)r r �commandZargumentT)r' �repeat)ZmutuallyExclusiveZ
optActionsZ extraActionsc C sn |dkrd| d< nX|d dkrDt |�dkrDtt|d �d �| d< n&t |�dkrZ|| d< nt�d �|�� dS )
z:
Set escape character; ``none'' = disable
ZnoneNr r �^� � �@ zBad escape character '{}'.)�len�chr�ord�sys�exit�format)�selfZesc� r5 �=/usr/lib/python3/dist-packages/twisted/conch/scripts/conch.py�
opt_escape@ s
zClientOptions.opt_escapec C s8 |� d�\}}}t|�}t|�}| j�|||ff� dS )zH
Forward local port to remote address (lport:host:port)
�:N)�split�int�
localForwards�append)r4 r � localPortZ
remoteHost�
remotePortr5 r5 r6 �opt_localforwardN s z ClientOptions.opt_localforwardc C s8 |� d�\}}}t|�}t|�}| j�|||ff� dS )zH
Forward remote port to local address (rport:host:port)
r8 N)r9 r: �remoteForwardsr< )r4 r r> ZconnHostZconnPortr5 r5 r6 �opt_remoteforwardX s z ClientOptions.opt_remoteforwardc G s || d< d� |�| d< d S )N�host� r( )�join)r4 rB r( r5 r5 r6 � parseArgsb s zClientOptions.parseArgs)�__name__�
__module__� __qualname__ZsynopsisZlongdescZ
optParametersZoptFlagsr Z CompletionsZ CompleterZCompleteUserAtHostZcompDatar; r@ r7 r? rA rE r5 r5 r5 r6 r s:
��
�
��
r c C sb t jdd � } d| krF| �d�}| ||d � | } | |d |d �= | d d � D ]\}zB| �|�}|d d� dkr�| |d d dkr�g | ||d �< W qR tk
r� Y qRX qRt� azt�| � W nF tjk
�r
} z$t d�
|�� t� � t � d� W 5 d }~X Y nX td �rdtd
�rFtd
dk�r6t j
}nttd
d �}nt j}t j
}t�|� |t _
nt�� t� t j�� }zt�|�aW n d aY nX zt�tjd d
� �}W n d }Y nX z t � � W 5 t�r�t�|tj t� |�r�t�tj|� td �rtd �std �s*t�tj tj � X t j
�!� �rTtd �sTt d�
td �� t � t"� d S )Nr, z-lr+ � z-or �-z ERROR: {}r Zlogfileza+c W s t �dt�S �Nr )r � callLater� reConnect)�ar5 r5 r6 �<lambda>� � zrun.<locals>.<lambda>r( r r! zConnection to {} closed.rB )#r1 �argv�index�
ValueErrorr r Z parseOptionsr Z
UsageError�printr3 Zopt_helpr2 �stdout�open�stderrr Z startLoggingZ discardLogs� doConnect�stdin�filenor � tcgetattr�old�signal�SIGUSR1� tcsetattr�TCSANOW�SIGWINCH�SIG_DFLr �run�isatty�
exitStatus)�args�i�arg�ur Zrealout�fdZoldUSR1r5 r5 r6 rc r sd
$
rc c C s0 ddl m} dat�dt� t�| �� � � d S )Nr ��failurer+ g{�G�z�?) �twisted.pythonrl re r rL � _stopReactorr �errZFailurerk r5 r5 r6 � handleError� s
rp c C s z t �� W n Y nX d S �N)r �stopr5 r5 r5 r6 rn � s rn c C s� dt d kr(t d �dd�\t d< t d< t js8ddgt _t d } t d sTt�� t d< t d sfdt d< ntt d �t d< t d } t d }tj}t d s�t�| t �t d < t� t d t t
� �}t � | |t ||�� t
� d S )
N�@rB r, �userz
~/.ssh/id_rsaz
~/.ssh/id_dsa�port� zhost-key-algorithms)r r9 Z identitys�getpassZgetuserr: r Z
verifyHostKeyZgetHostKeyAlgorithmsZSSHUserAuthClient�
SSHConnectionr �
addErrback�_ebExit)rB ru ZvhkZuaor5 r5 r6 rX � s(
�rX c C s d� | �at�dt� d S )Nz conch: exiting with error {}皙�����?)r3 re r rL rn )r r5 r5 r6 rz � s
rz c C s, t tjd�rtt� tjrLtjD ]*\} }t�| t� t|t
��}tj� |� q tj r�tj D ]&\}}t
�d�||�� t�||� qXt�ddt� td r�td r�t�t� � td �r(t�� r�t�d� t�� td �D ]P}zt�|� W q� tk
�r$ } z dd l }|j |j k�r� W 5 d }~X Y q�X q�d S )
NZ
sendIgnorez&asking for remote forwarding for {}:{}ZbeforeZshutdownr# �agentr r � ) �hasattr�conn� transport�
_KeepAliver r; r Z listenTCPr
ZSSHListenForwardingFactory� SSHListenClientForwardingChannelr< r@ r �msgr3 �requestRemoteForwardingZaddSystemEventTrigger�beforeShutdownZ openChannel�
SSHSession�osr �_exit�setsid�range�close�OSError�errnoZEBADF)r= �hostportr& r> rg r r� r5 r5 r6 � onConnect� s@ �� �
r� c C s t � tjj�� d S rq )r� r r� �loseConnectionr5 r5 r5 r6 rM � s rM c C s4 t j} | D ]$\}}t�d�||�� t�|� q
d S )Nzcancelling {}:{})r r@ r r� r3 r �cancelRemoteForwarding)r@ r> r� r5 r5 r6 r� s r� c C s t d st�dt� d S )NZ reconnectr{ )r r rL rn r5 r5 r5 r6 �stopConnection
s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r� c C s* || _ d | _t�| j�| _| j�d� d S )Ni, )r �
globalTimeoutr Z LoopingCall�
sendGlobalZlc�start)r4 r r5 r5 r6 �__init__ s z_KeepAlive.__init__c C s2 | j jdddd�}|�| j� t�d| j�| _d S )Ns" conch-keep-alive@twistedmatrix.comrP r, �Z wantReply� )r �sendGlobalRequest�addBoth� _cbGlobalr rL � _ebGlobalr� )r4 �dr5 r5 r6 r� s � z_KeepAlive.sendGlobalc C s | j r| j �� d | _ d S rq )r� Zcancel)r4 �resr5 r5 r6 r� # s
z_KeepAlive._cbGlobalc C s | j rd | _ | jj�� d S rq )r� r r� r� �r4 r5 r5 r6 r� ) s z_KeepAlive._ebGlobalN)rF rG rH r� r� r� r� r5 r5 r5 r6 r� s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )rx c C s2 | a g | _i | _t| tj�s(| j`| j`t � d S rq )
r r; r@ �
isinstancer rx � __class__r� r� r� r� r5 r5 r6 �serviceStarted1 s z SSHConnection.serviceStartedc C s( | j }g | _ |D ] }|�� qt� d S rq )r; r� r� )r4 Zlfr& r5 r5 r6 �serviceStopped= s
z SSHConnection.serviceStoppedc C sT t �d|f�}| jd|dd�}t�d�||�� |�| j||� |�| j ||� d S )N�0.0.0.0s
tcpip-forwardr, r� z"requesting remote forwarding {}:{})
r
�packGlobal_tcpip_forwardr� r r� r3 Z addCallback�_cbRemoteForwardingry �_ebRemoteForwarding)r4 r> r� �datar� r5 r5 r6 r� E s � �z%SSHConnection.requestRemoteForwardingc C s0 t �d�||�� || j|< t �t| j�� d S )Nz accepted remote forwarding {}:{})r r� r3 r@ �repr)r4 �resultr> r� r5 r5 r6 r� O s �
z!SSHConnection._cbRemoteForwardingc C s t �d�||�� t �|� d S )Nz remote forwarding {}:{} failed)r r� r3 )r4 r r> r� r5 r5 r6 r� V s
�z!SSHConnection._ebRemoteForwardingc C s` t �d|f�}| �d|� t�d�|�� z | j|= W n tk
rJ Y nX t�t| j�� d S )Nr� s cancel-tcpip-forwardz cancelling remote forwarding {}) r
r� r� r r� r3 r@ � Exceptionr� )r4 r> r� r5 r5 r6 r� \ s z$SSHConnection.cancelRemoteForwardingc C s� t �d�|�� t�|�\}}t �| j� t �|� |d | jkrp| j|d }t �d�|�� t|||| d�S ttj d��d S )Nz FTCP {!r}r, zconnect forwarding {})Z remoteWindowZremoteMaxPacketr zdon't know about that port)
r r� r3 r
ZunpackOpen_forwarded_tcpipr@ �SSHConnectForwardingChannelr r ZOPEN_CONNECT_FAILED)r4 Z
windowSizeZ maxPacketr� ZremoteHPZorigHPZ connectHPr5 r5 r6 �channel_forwarded_tcpipg s
��z%SSHConnection.channel_forwarded_tcpipc C sT t �d�|�� t �| j� t| j�dkr<t �d� t� n| jjd �| |� d S )Nzconnection closing {}r, zstopping connectionr ) r r� r3 �channelsr. r� r� � __bases__�
channelClosed)r4 r r5 r5 r6 r� x s
zSSHConnection.channelClosedN) rF rG rH r� r� r� r� r� r� r� r� r5 r5 r5 r6 rx 0 s
rx c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )r� s sessionc s� t �d�� j�� td r<� jj� dddd�}|�dd� � td rHd S td
rXtd s`td sft� t �
� }td
r�td s�d� _ � j |_
n� j|_
� fdd�|_t�|�� _d}td r؈ j�� dt�td
�� n�td
�r\td �r@tjd }t�|tjd�}t�d|�}t � ||d�}� j�� d|� t � t j � j � � j�� dt�td
�� nntd �s�tjd }t�|tjd�}t�d|�}t � ||d�}� j�� d|� t � t j � j � � j�� dd� d S )Nzsession {} openr| s auth-agent-req@openssh.comrP r, r� c S s
t �| �S rq )r r� ��xr5 r5 r6 rO � rP z(SSHSession.channelOpen.<locals>.<lambda>r# r( r r! r c s � � � S rq )�sendEOFr� r� r5 r6 rO � rP r r% s subsystemZTERM�12345678�4H� s pty-reqs execs shell) r r� r3 �idr r � sendRequestr� �
_enterRawModer ZSSHSessionClient�
escapeMode� handleInput� dataReceived�writeZconnectionLostr
Z
StandardIOr ZNSr� �environ�fcntl�ioctlr �
TIOCGWINSZ�struct�unpackZpackRequest_pty_reqr] ra �_windowResized) r4 Zfoor� �crj Zterm�winsz�winSizeZ
ptyReqDatar5 r� r6 � channelOpen� sP
�
�
zSSHSession.channelOpenc C s& |dkr d| _ | �|� �n| j dkr:|td kr:d| _ n�| j dk�rd| _ |dkrht�d� t� d S |dkr�dd � }t�d
|� d S |d kr�t�d � | jj �
� d S |d
k�r| j �d� | jj �
� }|�� |D ]$}| j �td�|| jj | ��� q�d S | �d| � nd
| _ | �|� d S )N)�
�
r, r r+ � .zdisconnecting from escape� c S s6 t � tj�� tj�� t�t�� tj � t
� d S rq ) �
_leaveRawModer1 rU �flushrY r� �kill�getpidr] �SIGTSTPr� r5 r5 r5 r6 �_� s
z!SSHSession.handleInput.<locals>._r � Rzrekeying connection� #s'
The following connections are open:
z
#{} {}
� ~)r� r� r r r� r� r rL r r� Z sendKexInitr
r� �keys�sortr r3 )r4 �charr� r� Z channelIdr5 r5 r6 r� � sD
�
�
zSSHSession.handleInputc C s | j �|� d S rq )r
r� �r4 r� r5 r5 r6 r� � s zSSHSession.dataReceivedc C sL |t jkrHt�d�t|��� ttj�t kr<tjj
� |� n tj� |� d S )Nzgot {} stderr data) r ZEXTENDED_DATA_STDERRr r� r3 r. r r1 rW r �bufferr� )r4 r r� r5 r5 r6 � extReceived� s
zSSHSession.extReceivedc C s t �d� | j�� d S )Nzgot eof)r r� r
ZloseWriteConnectionr� r5 r5 r6 � eofReceived� s
zSSHSession.eofReceivedc C s t �d�| �� | j�| � d S )Nzremote side closed {})r r� r3 r Z sendCloser� r5 r5 r6 �
closeReceived� s zSSHSession.closeReceivedc C s& t �d�| �� t �t| jj�� d S )Nz closed {})r r� r3 r� r r� r� r5 r5 r6 �closed� s zSSHSession.closedc C s( t t�d|�d �at�d�t�� d S )Nz>Lr zexit status: {})r: r� r� re r r� r3 r� r5 r5 r6 �request_exit_status� s z SSHSession.request_exit_statusc C s | j �| � d S rq )r r� r� r5 r5 r6 r� s zSSHSession.sendEOFc C s | j �� d S rq )r
ZpauseProducingr� r5 r5 r6 � stopWriting s zSSHSession.stopWritingc C s | j �� d S rq )r
ZresumeProducingr� r5 r5 r6 � startWriting s zSSHSession.startWritingc G sV t �dtjd�}t�d|�}|d |d |d |d f}| j�| dtjd |�� � d S )
Nr r� r� r, r+ r} s
window-change�!4L)r� ) r� r� r r� r� r� r r� Zpack)r4 rf r� r� ZnewSizer5 r5 r6 r� s zSSHSession._windowResizedN)rF rG rH �namer� r� r� r� r� r� r� r� r� r� r� r� r5 r5 r5 r6 r� � s ,) r� c @ s e Zd ZdS )r� N�rF rG rH r5 r5 r5 r6 r� s r� c @ s e Zd ZdS )r� Nr� r5 r5 r5 r6 r� s r� c C s* t sd S tj�� } t�| tjt� da d S rK )�
_inRawModer1 rY rZ r r_ r` �
_savedRawMode)rj r5 r5 r6 r� s
r� c C sR t rd S tj�� } zt�| �}|d d � }W n t�d� Y �n
X |d tjB |d< |d tj tj
B tj B tj B tj
B tjB tjB @ |d< ttd�r�|d tj @ |d< |d tjtjB tjB tjB tjB tjB tjB @ |d< ttd��r|d tj @ |d< |d tj @ |d< d|d tj< d|d tj< |a t� | tj |� da d S )Nznot a typewriter!r �IUCLCr} �IEXTENr, � ) r� r1 rY rZ r r[ r r� ZIGNPARZISTRIPZINLCRZIGNCRZICRNLZIXONZIXANYZIXOFFr~ r� ZISIGZICANONZECHOZECHOEZECHOKZECHONLr� ZOPOSTZVMINZVTIMEr� r_ r` )rj r\ �newr5 r5 r6 r� # sB
���
��� r� �__main__)9Z
__future__r Ztwisted.conch.clientr r r Ztwisted.conch.errorr Ztwisted.conch.sshr r r r
r Ztwisted.internetr r
r rm r r Ztwisted.python.compatr r r r� r1 rw r� r r� r] Z ConchOptionsr r re r\ r� r� rc rp rn rX rz r� rM r� r� r� rx Z
SSHChannelr� r� r� r� r� rF r5 r5 r5 r6 �<module> sP J;
# T
%