File: //usr/lib/python3/dist-packages/twisted/conch/scripts/__pycache__/tkconch.cpython-38.pyc
U
��W[�Z � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZm Z m
Z
m Z m Z ddlm
Z
mZmZ ddlmZ ddlmZmZmZmZ dd lmZmZ dd
lmZ dd l Z dd l Z dd l Z dd l Z dd l Z dd l!Z!er�dd l"Z#dd l$m%Z& dd l'm(Z) ndd l#Z#dd l&Z&dd l)Z)G d d
� d
e#j*�Z+G dd� dej,�Z-d a.d a/da0d a1dd� Z2dd� Z3dd� Z4G dd� dej5�Z6G dd� dej7�Z7G dd� de j8�Z8G d d � d e
j9�Z9G d d � d ej:�Z;e<d k�r�e3� d S )!z2
Implementation module for the `tkconch` command.
� )�print_function)�error)�tkvt100)� transport�userauth�
connection�common�keys)�session�
forwarding�channel)�isInKnownHosts)�reactor�defer�protocol� tksupport)�usage�log)�_PY3Nc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � TkConchMenuc O s� t jj| f|�|� | j�d� t �� | _| j�d� t j| dddd�j dddd� t �
| �| _ | j j d d dd
d � t j| ddd d�j dd dd� t �
| �| _ | j j d d d d
d � t j| ddd
d�j dddd� t �
| �| _
| j
j d d dd
d � t j| dddd�j dddd� t �
| �| _| jj d d dd
d � t j| dddd�j dddd� t �
| �| _| jj d dd
d� t j| | jdd�j ddd
d� t j| dd�j dddd� t j| ddd�| _| jj d d dd
d � t j| d| jd�j dd d � t j| d | jd�j dd d � t �
| �| _| jj d d d
d� t j| d d�j dd d
d� t �
| �| _| jj d d d
d� t j| d d�j dd d
d� t j| d!| jdd"�| _| jj d d#d � t j| d$| jd%d"�| _| jj dd#d � t j| d&d�j ddd'd
d � t j| ddd(d�j dd)dd� t j
| d*d+�| _| jj d d d)d
d � t j| ddd,d�j dd-dd� t j
| d.d+�| _ | j j d d d-d
d � t j| ddd/d�j dd0dd� t j
| d1d+�| _ | j j d d d0d
d � t j| d2| j d�j ddd3d
d � | j ddd4d5� | j d dd d5� | j�!d6t"j#� d S )7N�TkConch�local�w�leftZHostname)ZanchorZjustify�text� )�column�row�sticky� Znesw)r Z
columnspanr r ZPortZUsername� ZCommand� ZIdentity� ZBrowse)�commandr zPort Forwarding)r � r )�height�widthZAdd)r r# � )r r ZRemove� ZHostZLocal)r Zvariable�value� ZRemoteZremotezAdvanced Options�
ZCipher� �cipher)�nameZMAC� �macz Escape Char�
�escapezConnect!� �@ )ZweightZminsizeZWM_DELETE_WINDOW)$�Tkinter�Frame�__init__�master�titleZ StringVar�localRemoteVar�setZLabelZgridZEntry�host�port�userr# �identityZButton�getIdentityFileZListbox�forwards�
addForward�
removeForward� forwardPort� forwardHostZ RadiobuttonZ localForwardZ
remoteForwardr- r0 r2 � doConnectZgrid_rowconfigureZgrid_columnconfigurer �sys�exit)�self�argsZparams� rK �?/usr/lib/python3/dist-packages/twisted/conch/scripts/tkconch.pyr7 sb
zTkConchMenu.__init__c C s0 t �� }|r,| j�dtj� | j�tj|� d S �Nr )� tkFileDialogZaskopenfilenamer? �deleter5 �END�insert)rI �rrK rK rL r@ a s zTkConchMenu.getIdentityFilec C sx | j �� }| j �dtj� | j�� }| j�dtj� | j�� dkr\| j�tjd||f � n| j�tjd||f � d S )Nr r zL:%s:%szR:%s:%s) rD �getrO r5 rP rE r: rA rQ )rI r= r< rK rK rL rB g s
zTkConchMenu.addForwardc C s" | j �� }|r | j �|d � d S rM )rA Z curselection�remove)rI ZcurrK rK rL rC q s
zTkConchMenu.removeForwardc C s� d}| j �� td< | j�� td< | j�� td< | j�� td< | j�� }| j�� }| j�� }|r�|t j
krr|gt _
nt � dd� d}|r�|t j
kr�|gt _
n|r�t � dd � d}|�r*|d
kr�d td < nd|d d kr�t|�d
kr�tt|d �d �td < n2t|�dk�r|td < n|�r*t � dd| � d}| j�� �rHtj�| j�� � | j�dtj�D ]:}|d dk�r~t�|d
d � � nt�|d
d � � �qXdtd k�r�td �dd�\td< td< td �r�td �s�|�r�t � dd� d}|�r�| j�� | j� � td �r&t j }t � t j!� |t _ nt �"� t#t _$tj�sFddgt_td }t%td �p\d�}t �&||f� t'�(||t)� � t*j�+� t*j�,dtd td f � n| �-� d S )Nr r< r= r> r# r z Bad cipher.r zBad MAC.Znoner2 �^r r4 zBad escape character '%s'.�L�@zMissing host or username.r z
~/.ssh/id_rsaz
~/.ssh/id_dsa� z%s@%s - TkConch).r< rS �optionsr= r> r# r- r0 r2 �SSHClientTransport�supportedCiphers� tkMessageBoxZ showerror�
supportedMACs�len�chr�ordr? � identitys�appendrA r5 rP �opt_localforward�opt_remoteforward�splitr8 �quitZdestroyrG �stdoutr Z startLogging�stderrZ discardLogs� handleError�deferr�int�msgr Z
connectTCP�SSHClientFactory�frameZ deiconifyr9 Zfocus) rI Zfinishedr- r0 r2 �lineZrealoutr< r= rK rK rL rF v st
zTkConchMenu.doConnectN)�__name__�
__module__� __qualname__r7 r@ rB rC rF rK rK rK rL r s
C
r c @ s@ e Zd ZdZddddgdddd gd
d d d
gddddgddddgddddgddddgdddd ggZd d d gd d!d"gd#d$d%gd&d'd(gd)d*d+gd,d-d.gd/d0d1gd2d3d4ggZejjZ ejj
Z e j
d5ge �e �e �e �e jd6d7�e jd6d7�d8�e �� e jd9d7�e jd:d;d<�gd=�Zg Zg Zg Zd>d?� Zd@dA� ZdBdC� ZdDdE� ZdFdG� ZdS )H�GeneralOptionsz,Usage: tkconch [options] host [command]
r> �lNz Log in using this user name.r? �iz~/.ssh/identityz&Identity for public key authenticationr2 �e�~z(Set escape character; ``none'' = disabler- �cz Select encryption algorithm.�macs�mz.Specify MAC algorithms for protocol version 2.r= �pz7Connect to this port. Server must be on the same port.� localforwardrV z<listen-port:host:port Forward local port to remote address�
remoteforward�Rz<listen-port:host:port Forward remote port to local address�tty�tz-Tty; allocate a tty even if command is given.�notty�TzDo not allocate a tty.�version�Vz Display version number only.�compress�CzEnable compression.�noshell�Nz"Do not execute a shell or command.� subsystem�sz-Invoke command (mandatory) as SSH2 subsystem.r �vz
Log to stderr�ansilog�az!Print the received data to stdout)r r� zlisten-port:host:port)�descr)r- ry r| r} r# ZargumentT)r� �repeat)ZmutuallyExclusiveZ
optActionsZ extraActionsc C s | j �|� d S �N)ra rb )rI ru rK rK rL � opt_identity� s zGeneralOptions.opt_identityc C s8 |� d�\}}}t|�}t|�}| j�|||ff� d S �N�:)re rk �
localForwardsrb )rI �f� localPortZ
remoteHost�
remotePortrK rK rL rc � s z GeneralOptions.opt_localforwardc C s8 |� d�\}}}t|�}t|�}| j�|||ff� d S r� )re rk �remoteForwardsrb )rI r� r� ZconnHostZconnPortrK rK rL rd � s z GeneralOptions.opt_remoteforwardc C s dgt jdd�< d S )N�zlibr r )rZ ZsupportedCompressions�rI rK rK rL � opt_compress� s zGeneralOptions.opt_compressc G s< |r(|d | d< d� |dd � �| d< nd| d< d| d< d S )Nr r< � r r# � )�join)rI rJ rK rK rL � parseArgs� s
zGeneralOptions.parseArgs)rp rq rr ZsynopsisZ
optParametersZoptFlagsr rZ r[ Z_ciphersr] Z_macsr Z CompletionsZ CompleteListZ CompleterZCompleteUserAtHostZcompDatara r� r� r� rc rd r� r� rK rK rK rL rs � sR
�
�
�
�� rs c sJ t jrtd��t�� � g }|f� �fdd� }|t _t �| � t j�� � S )Nz can't ask 2 questions at once!c s� | sd S | dkrt �� | dkrJt�d� d�|�}~d t_� �|� d S dt| � krbdk r�n n|�| � �r�t�| � n&t| �dkr�|r��r�t�d� |�� d S ) N��
z
r� � � r( z ) r �stoprn �writer� �callbackr` rb �pop)Zch�respZstresp��d�echorK rL �gotChar s&
z!deferredAskFrame.<locals>.gotChar)rn r� �
ValueErrorr ZDeferredr� �canvas� focus_force)Zquestionr� r� r� rK r� rL �deferredAskFrame s
r� c
C s� 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�� }|�� t�� }t|�a t j
tj tj dd� t
� azt�| � W nD tjk
�r< } z"td | � t�� t �d� W 5 d }~X Y nX t�� D ]0\}}|�rFtt |��rFtt |��tj|� �qFtjD ](\}\} }
t j�tjd
|| |
f � �q~g t_tj D ](\}\} }
t j�tjd || |
f � �q�g t_ t j |d d �a |� d
t j!t j" d t j#t j$ d f � t j
tj d� t%�&|� |�� td �rPtd �s^dtd k�rht �'� n|�(� t)�*� t �t+� d S )Nr z-lr r! z-or �-)�sideZfill�expandz ERROR: %sz
L:%s:%s:%sz
R:%s:%s:%s)r� z%dx%dr )r� r< r> rW ),rG �argv�indexr� r5 ZTkZwithdrawZToplevelr �menuZpackZTOPZBOTHrs rY Z parseOptionsr Z
UsageError�printZopt_helprH �items�hasattr�getattrrQ rP r� rA r� r Z
VT100Framern ZgeometryZ fontWidthr& Z
fontHeightr% r �installrF Zmainloopr �run�
exitStatus) rJ ru �arg�root�top�u�kr� r{ ZrhZrprK rK rL r� " sV
$ *
"
r� c C s, ddl m} dat�| �� � t�� � d S )Nr ��failurer )�twisted.pythonr� r� r �errZFailurer r� r� rK rK rL ri Q s
ri c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) rm r c C s t �� d S r� )r r� r� rK rK rL � stopFactory\ s z SSHClientFactory.stopFactoryc C s t � S r� )rZ )rI ZaddrrK rK rL �
buildProtocol_ s z SSHClientFactory.buildProtocolc C s t �dd|j|jf � d S )Nr z"Connection Failed, Reason:
%s: %s)r\ � showwarning�typer) )rI Z connector�reasonrK rK rL �clientConnectionFailedb s z'SSHClientFactory.clientConnectionFailedN)rp rq rr Znoisyr� r� r� rK rK rK rL rm Y s rm c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )rZ c C s d||f a d S )NzDconch: Remote side disconnected with error code %i
conch: reason: %s)r� )rI �codeZdescrK rK rL � receiveErrorg s z SSHClientTransport.receiveErrorc C s d||f a tj�| ||� d S )Nz>conch: Sending disconnect with error code %i
conch: reason: %s)r� r rZ �sendDisconnect)rI r� r� rK rK rL r� k s z!SSHClientTransport.sendDisconnectc C s |s t d rt�d| � d S )Nr zReceived Debug Message: %s)rY r rl )rI Z
alwaysDisplay�messageZlangrK rK rL � receiveDebugp s z SSHClientTransport.receiveDebugc C s� t td |dd i�}|dkr&t�d�S |dkr>t�t�d��S td | j�� j krdtd }td }n0dtd | j�� j f }dtd | j�� j f }t
� |�d }d |d
d d �| |f }|d
7 }t |d��
| j|||�S d S )Nr< z known-hostsr r � bad host keyz%s (%s)z%s,%sr zNThe authenticity of host '%s' can't be established.
%s key fingerprint is %s.�DSA�RSA�s ssh-dsss ssh-rsaz9
Are you sure you want to continue connecting (yes/no)? )r
rY r �succeedZfailr �
ConchErrorr ZgetPeerr< r ZgetNSr� � addCallback�_cbVerifyHostKey)rI �pubKeyZ fingerprintZgoodKeyr< �khHost�keyTypeZquesrK rK rL �
verifyHostKeyu s.
�
� ��z SSHClientTransport.verifyHostKeyc C s� |� � dkr$tdd��| j|||�S |� � dkrDt�d� t�d��zdt�d|dd d
�| f � tt j
� d �d ��,}t �
|��d
d�}|�d|||f � W 5 Q R X W n t�� tj�Y nX d S )N)Zyes�noz Please type 'yes' or 'no': r r� z Host key verification failed.
r� zBWarning: Permanently added '%s' (%s) to the list of known hosts.
r� r� r� z~/.ssh/known_hostsr� �
� z
%s %s %s)�lowerr� r� r� rn r� r r� �open�os�path�
expanduser�base64Z encodestring�replacer rj )rI �ansr� r� r� Z known_hostsZ
encodedKeyrK rK rL r� � s"
��"z#SSHClientTransport._cbVerifyHostKeyc C s0 t d rt d }nt�� }| �t|t� �� d S )Nr> )rY �getpassZgetuserZrequestService�SSHUserAuthClient�
SSHConnection)rI r> rK rK rL �connectionSecure� s
z#SSHClientTransport.connectionSecureN) rp rq rr r� r� r� r� r� r� rK rK rK rL rZ e s rZ c @ s2 e Zd Zg Zd
dd�Zdd� Zdd� Zdd � ZdS ) r� Nc C s |sd| j td f }t|d�S )Nz%s@%s's password: r< r )r> rY r� )rI �promptrK rK rL � getPassword� s z SSHUserAuthClient.getPasswordc s� � fdd�t jD �}|s d S |d }t�|� � j�|� tj�|�}|d7 }tj� |�s^d S zt
j � |��
� W S � �� Y S X d S )Nc s g | ]}|� j kr|�qS rK )� usedFiles)�.0�xr� rK rL �
<listcomp>� s
z2SSHUserAuthClient.getPublicKey.<locals>.<listcomp>r z.pub)rY ra r rl r� rb r� r� r� �existsr �Key�fromFileZblob� getPublicKey)rI �files�filerK r� rL r� � s
z SSHUserAuthClient.getPublicKeyc
C s� t j�| jd �}t j�|�s"d S zt�tj� |�j
�W S tj k
r� } z<|j d dkr�d| jd }t
|d��| jd� W Y �S W 5 d }~X Y nX d S )N���r z encrypted key with no password� Enter passphrase for key '%s': )r� r� r� r� r� r r� r r� r� � keyObject� BadKeyErrorrJ r� r� �_cbGetPrivateKey)rI r� rv r� rK rK rL �
getPrivateKey� s �z SSHUserAuthClient.getPrivateKeyc C sr t j�| jd �}ztjj||d�jW S tjk
rl |dkrB� d| jd }t |d��
| j |d � Y S X d S )Nr� )Zpasswordr r� r r ) r� r� r� r� r r� r� r� r� r� r� r� )rI r� �countr� r� rK rK rL r� � s �z"SSHUserAuthClient._cbGetPrivateKey)N)rp rq rr r� r� r� r� r� rK rK rK rL r� � s
r� c @ s e Zd Zdd� ZdS )r� c C s� t d s| �t� � t jrBt jD ] \}}t�|t�| |tj�� q t j r�t j D ]>\}}t
� d||f � t� d|f�}| �
d|� || j |< qNd S )Nr� z&asking for remote forwarding for %s:%sz0.0.0.0z
tcpip-forward)rY Z openChannel�
SSHSessionr� r Z listenTCPr ZSSHListenForwardingFactoryZ SSHListenClientForwardingChannelr� r rl ZpackGlobal_tcpip_forwardZsendGlobalRequest)rI r� Zhostportr� �datarK rK rL �serviceStarted� s( ���� z SSHConnection.serviceStartedN)rp rq rr r rK rK rK rL r� � s r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )r s sessionc C s d| _ t�� }td r | j|_n| j|_| j|_|jt _
t j � � td rf| j
�| dt�td �� n�td r�td r�tj�dd�}d }t�||d
�}| j
�| d |� | j
�| d t�td �� nHtd
s�tj�dd�}d }t�||d
�}| j
�| d |� | j
�| dd� | j
jj�d� d S )Nr r2 r� s subsystemr# r ZTERMZxterm)� �P r r r� s pty-req�execr� s shellr� )�
escapeModer
ZSSHSessionClientrY � handleInput� dataReceivedr� �sendEOFZconnectionLostrn r� r� r� �connZ sendRequestr ZNSr� �environrS ZpackRequest_pty_reqr Z
setTcpNoDelay)rI Zfoorx ZtermZwinSizeZ
ptyReqDatarK rK rL � channelOpen� s8
�
�zSSHSession.channelOpenc C s� |dkrd| _ | �|� n�| j dkr8|td kr8d| _ n�| j dkr�d| _ |dkrft�d� t�� d S |dkr�t�t� � t
j � d S |dkr�t�d � | j j
�� d S | �d
| � nd | _ | �|� d S ) N)�
r� r r2 r �.zdisconnecting from escape�r~ zrekeying connectionrw r )r r� rY r rl r r� r� �kill�getpid�signal�SIGSTOPr
r Z sendKexInit)rI �charrK rK rL r s*
zSSHSession.handleInputc C s: t rt|t�r|�d�}td r,tt|�� t�|� d S )Nzutf-8r� ) r �
isinstance�bytes�decoderY r� �reprrn r� �rI r rK rK rL r + s
zSSHSession.dataReceivedc C s6 |t jkr2t�dt|� � tj�|� tj�� d S )Nzgot %s stderr data) r ZEXTENDED_DATA_STDERRr rl r^ rG rh r� �flush)rI r� r rK rK rL � extReceived2 s
zSSHSession.extReceivedc C s t �d� tj�� d S )Nzgot eof)r rl rG �stdin�closer� rK rK rL � eofReceived8 s
zSSHSession.eofReceivedc C s* t �d| � t| jj�dkr&t�� d S )Nz closed %sr )r rl r^ r
Zchannelsr r� r� rK rK rL �closed< s zSSHSession.closedc C s& t t�d|�d �at�dt � d S )Nz>Lr zexit status: %s)rk �structZunpackr� r rl r rK rK rL �request_exit_statusA s z SSHSession.request_exit_statusc C s | j �| � d S r� )r
r r� rK rK rL r F s zSSHSession.sendEOFN) rp rq rr r. r r r r r r r! r rK rK rK rL r � s #r �__main__)=�__doc__Z
__future__r Z
twisted.conchr Ztwisted.conch.uir Ztwisted.conch.sshr r r r r r
r r Z twisted.conch.client.defaultr
Ztwisted.internetr r r r r� r r Ztwisted.python.compatr r� rG r� r r� r Ztkinterr5 Ztkinter.filedialogZ
filedialogrN Ztkinter.messageboxZ
messageboxr\ r6 r ZOptionsrs rY r� r� rn r� r� ri Z
ClientFactoryrm rZ r� r� Z
SSHChannelr rp rK rK rK rL �<module> s@ 0 E / D._