File: //lib/python3/dist-packages/twisted/conch/scripts/__pycache__/conch.cpython-310.pyc
o
�bG � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z d dl
m Z m Z d dl
mZ d dlmZ d dlmZmZmZmZmZ d dlmZmZmZ d dlm Z m Z d d l m Z m Z G d
d � d e�Z!da"da#d a$da%d a&da'd d
� Z(dd� Z)dd� Z*dd� Z+dd� Z,dd� Z-dd� Z.dd� Z/d d � Z0G d d � d �Z1G d d!� d!ej2�Z2G d"d#� d#ej3�Z4G d$d%� d%ej5�Z5G d&d'� d'ej6�Z6d(d)� Z7d*d+� Z8e9d,kr�e(� dS dS )-� N)�List�Tuple)�connect�default)� ConchOptions)�
ConchError)�channel�common�
connection�
forwarding�session)�reactor�stdio�task)�log�usage)�ioType�
networkStringc @ s� e Zd ZU dZdZg d�g d�g d�gZg d�g d�g d�g d �g d
�g d �gZejd gej d
d�ej d
d�d�e�
� ej dd�ej ddd�gd�Z g Z e
eeeeef f ed< g Ze
eeeeef f ed< dd� Zdd� Zdd � Zd d � Zd S ) �
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�RNz<listen-port:host:port Forward remote port to local address)�null�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 �command�argumentT)r( �repeat)�mutuallyExclusive�
optActions� extraActions�
localForwards�remoteForwardsc C sv |dkr
d| d< dS |d dkr$t |�dkr$tt|d �d �| d< dS t |�dkr0|| d< dS t�d |� d
�� dS ) z:
Set escape character; ``none'' = disable
�noneNr r �^� � �@ zBad escape character 'z'.)�len�chr�ord�sys�exit)�self�esc� r= �=/usr/lib/python3/dist-packages/twisted/conch/scripts/conch.py�
opt_escapeP s zClientOptions.opt_escapec C �8 |� d�\}}}t|�}t|�}| j�|||ff� dS )zH
Forward local port to remote address (lport:host:port)
�:N)�split�intr/ �append)r; r � localPort�
remoteHost�
remotePortr= r= r> �opt_localforward] � z ClientOptions.opt_localforwardc C r@ )zH
Forward remote port to local address (rport:host:port)
rA N)rB rC r0 rD )r; r rG �connHost�connPortr= r= r> �opt_remoteforwardf rI z ClientOptions.opt_remoteforwardc G s || d< d� |�| d< d S )N�host� r) )�join)r; rM r) r= r= r> � parseArgso s zClientOptions.parseArgsN)�__name__�
__module__� __qualname__�synopsis�longdesc�
optParameters�optFlagsr � Completions� Completer�CompleteUserAtHost�compDatar/ r r rC �__annotations__r0 r? rH rL rP r= r= r= r> r s>
���
�
��
r c C s� t jdd � } d| v r#| �d�}| ||d � | } | |d |d �= | d d � D ],}z!| �|�}|d d� dkrK| |d d dkrKg | ||d �< W q) tyU Y q)w t� azt�| � W n# tjy� } zt d|� �� t�
� t � d� W Y d }~nd }~ww td r�td
r�td
dkr�t j }n t
td
d �}nt j}t j }t�|� |t _ nt�� t� t j�� }zt�|�aW n ty� d aY nw z t�tjd d
� �}W n ty� d }Y nw z0t� � W tr�t� |tj t� |r�t�tj|� td �rtd �s td �st�tj tj � n.t�r t� |tj t� |�r)t�tj|� td �r3td �s8td �sAt�tj tj � w w t j �!� �rVtd �sVt d�"td �� t � t#� d S )Nr4 z-lr3 � z-or �-zERROR: r �logfileza+c W s t �dt�S �Nr )r
� callLater� reConnect)�ar= r= r> �<lambda>� s zrun.<locals>.<lambda>r) r r" zConnection to {} closed.rM )$r9 �argv�index�
ValueErrorr �options� parseOptionsr �
UsageError�print�opt_helpr: �stdout�open�stderrr � startLogging� discardLogs� doConnect�stdin�filenor � tcgetattr�old�
BaseException�signal�SIGUSR1r
�run� tcsetattr�TCSANOW�SIGWINCH�SIG_DFL�isatty�format�
exitStatus)�args�i�arg�ur �realout�fd�oldUSR1r= r= r> rz } s�
$� ���
�
� �
�� �rz c C s, ddl m} dat�dt� t�| �� � � )Nr ��failurer3 g{�G�z�?) �twisted.pythonr� r� r
ra � _stopReactorr �err�Failurer� r= r= r> � handleError� s
r� c C s$ zt �� W d S ty Y d S w �N)r
�stoprw r= r= r= r> r� � s
�r� c C s� dt d v rt d �dd�\t d< t d< t js ddgt _t d } t d s*t�� t d< t d s3dt d< ntt d �t d< t d } t d }tj}t d sRt�| t �t d < t� t d t t
� �}t � | |t ||�� t
� d S )
N�@rM r4 �userz
~/.ssh/id_rsaz
~/.ssh/id_dsa�port� zhost-key-algorithms)rh rB � identitys�getpass�getuserrC r �
verifyHostKey�getHostKeyAlgorithms�SSHUserAuthClient�
SSHConnectionr �
addErrback�_ebExit)rM r� �vhk�uaor= r= r> rr � s"
rr c C s d| � �a t�dt� d S )Nzconch: exiting with error 皙�����?)r� r
ra r� )r r= r= r> r� � s
r� c C s. t tjd�r
tt� tjr&tjD ]\} }t�| t� t|t
��}tj� |� qtj rItj D ]\}}t
�d|� d|� �� t�||� q,t�ddt� td rQtd rWt�t� � td r�t�� rdt�d � t�� td
�D ](}zt�|� W ql ty� } zd d l}|j|j kr�� W Y d }~qld }~ww d S d S ) N�
sendIgnorez!asking for remote forwarding for rA �before�shutdownr$ �agentr r � ) �hasattr�conn� transport�
_KeepAliverh r/ r
� listenTCPr �SSHListenForwardingFactory� SSHListenClientForwardingChannelrD r0 r �msg�requestRemoteForwarding�addSystemEventTrigger�beforeShutdown� openChannel�
SSHSession�osr �_exit�setsid�range�close�OSError�errno�EBADF)rE �hostportr' rG r� r r� r= r= r> � onConnect� sD ��
����r� c C s t � tjj�� d S r� )r� r� r� �loseConnectionr= r= r= r> rb s rb c C s8 t j} | D ]\}}t�d|� d|� �� t�|� qd S )Nz cancelling rA )rh r0 r r� r� �cancelRemoteForwarding)r0 rG r� r= r= r> r� s
�r� c C s t d s t�dt� d S d S )N� reconnectr� )rh r
ra r� r= r= r= r> �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 � LoopingCall�
sendGlobal�lc�start)r; r� r= r= r> �__init__ s z_KeepAlive.__init__c C s2 | j jdddd�}|�| j� t�d| j�| _d S )Ns" conch-keep-alive@twistedmatrix.com� r4 �� wantReply� )r� �sendGlobalRequest�addBoth� _cbGlobalr
ra � _ebGlobalr� )r; �dr= r= r> r� ! s
� z_KeepAlive.sendGlobalc C s | j r
| j �� d | _ d S d S r� )r� �cancel)r; �resr= r= r> r� ( s
�z_KeepAlive._cbGlobalc C s | j rd | _ | jj�� d S d S r� )r� r� r� r� �r; r= r= r> r� - s �z_KeepAlive._ebGlobalN)rQ rR rS r� r� r� r� r= r= r= r> r� s
r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )r� c C s | a g | _i | _t� d S r� )r� r/ r0 r� r� r= r= r> �serviceStarted4 s
z SSHConnection.serviceStartedc C s( | j }g | _ |D ]}|�� qt� d S r� )r/ r� r� )r; �lfr' r= r= r> �serviceStopped; s
z SSHConnection.serviceStoppedc C sX t �d|f�}| jd|dd�}t�d|� d|� �� |�| j||� |�| j||� d S )N�0.0.0.0s
tcpip-forwardr4 r� z requesting remote forwarding rA ) r �packGlobal_tcpip_forwardr� r r� � addCallback�_cbRemoteForwardingr� �_ebRemoteForwarding)r; rG r� �datar� r= r= r> r� B s
z%SSHConnection.requestRemoteForwardingc C s4 t �d|� d|� �� || j|< t �t| j�� d S )Nzaccepted remote forwarding rA )r r� r0 �repr)r; �resultrG r� r= r= r> r� I s
z!SSHConnection._cbRemoteForwardingc C s& t �d|� d|� d�� t �|� d S )Nzremote forwarding rA z failed�r r� )r; r rG r� r= r= r> r� N s z!SSHConnection._ebRemoteForwardingc C s^ t �d|f�}| �d|� t�d|� �� z| j|= W n ty$ Y nw t�t| j�� d S )Nr� s cancel-tcpip-forwardz cancelling remote forwarding )r r� r� r r� r0 � Exceptionr� )r; rG r� r= r= r> r� R s �z$SSHConnection.cancelRemoteForwardingc C s| t �d|��� t�|�\}}t �| j� t �|� |d | jv r8| j|d }t �d|� �� t|||| d�S ttjd��)NzFTCP r4 zconnect forwarding )� remoteWindow�remoteMaxPacketr� zdon't know about that port) r r� r �unpackOpen_forwarded_tcpipr0 �SSHConnectForwardingChannelr r
�OPEN_CONNECT_FAILED)r; �
windowSize� maxPacketr� �remoteHP�origHP� connectHPr= r= r> �channel_forwarded_tcpip\ s
��z%SSHConnection.channel_forwarded_tcpipc C sV t �d|� �� t �| j� t| j�dkr t �d� t� d S | jjd �| |� d S )Nzconnection closing r4 zstopping connectionr )r r� �channelsr6 r� � __class__� __bases__�
channelClosed)r; r r= r= r> r� l s
zSSHConnection.channelClosedN) rQ rR rS r� r� r� r� r� r� r� r� r= r= r= r> r� 3 s
r� 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� d�� td r � jj� dddd�}|�dd � � td
r%d S td r-td s1td
s4t� t� � }td rHtd
sHd� _
� j |_ n� j
|_ � fdd �|_t�|�� _d}td rn� j�� dt�td �� d S td r�td r�tjd }t�|tjd�}t�d|�}t�||d�}� j�� d|� t � t j � j � � j�� dt�td �� d S td
s�tjd }t�|tjd�}t�d|�}t�||d�}� j�� d|� t � t j � j � � j�� dd� d S )Nzsession z openr� s auth-agent-req@openssh.comr� r4 r� c S s
t �| �S r� r� ��xr= r= r> rd � s
z(SSHSession.channelOpen.<locals>.<lambda>r$ r) r r" r c s � � � S r� )�sendEOFr� r� r= r> rd � s r r& s subsystem�TERM�12345678�4H� s pty-reqs execs shell) r r� �idrh r� � sendRequestr� �
_enterRawModer �SSHSessionClient�
escapeMode� handleInput� dataReceived�write�connectionLostr �
StandardIOr �NSr� �environ�fcntl�ioctlr �
TIOCGWINSZ�struct�unpack�packRequest_pty_reqrx r} �_windowResized) r; �foor� �cr� �term�winsz�winSize�
ptyReqDatar= r� r> � channelOpen{ sJ �
zSSHSession.channelOpenc C s& |dv rd| _ | �|� d S | j dkr |td kr d| _ d S | j dkr�d| _ |dkr4t�d� t� d S |dkrDdd � }t�d
|� d S |d krUt�d � | jj �
� d S |d
kr�| j �d� | jj �
� }|�� |D ]}| j �td�|| jj | ��� qkd S | �d| � d S d
| _ | �|� d S )N)�
�
r4 r r3 � .zdisconnecting from escape� c S s6 t � tj�� tj�� t�t�� tj � t
� d S r� ) �
_leaveRawModer9 rm �flushrs r� �kill�getpidrx �SIGTSTPr� r= r= r= r> �_� s
z!SSHSession.handleInput.<locals>._r � Rzrekeying connection� #s'
The following connections are open:
z
#{} {}
� ~)r� r� rh r r� r� r
ra r� r� � sendKexInitr r� �keys�sortr r� )r; �charr r� � channelIdr= r= r> r� � sF
���zSSHSession.handleInputc C s | j �|� d S r� )r r� �r; r� r= r= r> r� � � zSSHSession.dataReceivedc C sT |t jkr(t�dt|�� d�� ttj�tkr tjj �
|� d S tj�
|� d S d S )Nzgot z stderr data) r
�EXTENDED_DATA_STDERRr r� r6 r r9 ro �str�bufferr� )r; r! r� r= r= r> � extReceived� s
�zSSHSession.extReceivedc C s t �d� | j�� d S )Nzgot eof)r r� r �loseWriteConnectionr� r= r= r> � eofReceived� s
zSSHSession.eofReceivedc C s t �d| � �� | j�| � d S )Nzremote side closed )r r� r� � sendCloser� r= r= r> �
closeReceived� s zSSHSession.closeReceivedc C s& t �d| � �� t �t| jj�� d S )Nzclosed )r r� r� r� r� r� r= r= r> �closed� s zSSHSession.closedc C s( t t�d|�d �at�dt� �� d S )Nz>Lr z
exit status: )rC r r r� r r� r# r= r= r> �request_exit_status� s z SSHSession.request_exit_statusc C s | j �| � d S r� )r� r� r� r= r= r> r� � r$ zSSHSession.sendEOFc C � | j �� d S r� )r �pauseProducingr� r= r= r> � stopWriting� � zSSHSession.stopWritingc C r/ r� )r �resumeProducingr� r= r= r> � startWriting� r2 zSSHSession.startWritingc G sZ t �dtjd�}t�d|�}|d |d |d |d f}| j�| dtjdg|�R � � d S ) Nr r� r� r4 r3 r� s
window-changez!4L) r r r r r r r� r� �pack)r; r� r
r �newSizer= r= r> r � s "zSSHSession._windowResizedN)rQ rR rS �namer r� r� r( r* r, r- r. r� r1 r4 r r= r= r= r> r� w s +- r� c @ � e Zd ZdS )r� N�rQ rR rS r= r= r= r> r� � r� c @ r8 )r� Nr9 r= r= r= r> r� r: r� c C s* t sd S tj�� } t�| tjt� da d S r` )�
_inRawModer9 rs rt r r{ r| �
_savedRawMode)r� r= r= r> r s
r c C sV t rd S tj�� } z
t�| �}|d d � }W n ty% t�d� Y d S w |d tj B |d< |d tj
tj B tj B tj
B tjB tjB tjB @ |d< ttd�rZ|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� �IEXTENr4 � ) r; r9 rs rt r ru rw r r� �IGNPAR�ISTRIP�INLCR�IGNCR�ICRNL�IXON�IXANY�IXOFFr� r= �ISIG�ICANON�ECHO�ECHOE�ECHOK�ECHONLr> �OPOST�VMIN�VTIMEr< r{ r| )r� rv �newr= r= r> r� sd
�������
�
������
�
r� �__main__):r r� r� rx r r9 r �typingr r �twisted.conch.clientr r � twisted.conch.client.optionsr �twisted.conch.errorr �twisted.conch.sshr r r
r r �twisted.internetr
r r r� r r �twisted.python.compatr r r rh r� r� rv r; r< rz r� r� rr r� r� rb r� r� r� r� �
SSHChannelr� r� r� r r� rQ r= r= r= r> �<module> sT W<
$D
2
�