CasperSecurity
o
�bK � @ s� d Z ddlmZ ddlmZ ddlmZmZm Z
ddl m Z ddl
mZmZ ddlmZ G dd � d �ZG d
d � d �Zee
j�G d d
� d
ej��ZG dd� dejej�ZG dd� d�ZG dd� dej�ZdS )z9
insults/SSH integration support.
@author: Jp Calderone
� )�Dict)� implementer)�avatar�error�
interfaces)�insults)�factory�session)�
componentsc @ s e Zd ZdZdd� Zdd� ZdS )�_Gluez�
A feeble class for making one attribute look like another.
This should be replaced with a real class at some point, probably.
Try not to write new code that uses it.
c K s | j �|� d S �N)�__dict__�update)�self�kw� r �;/usr/lib/python3/dist-packages/twisted/conch/manhole_ssh.py�__init__ s z_Glue.__init__c C s t | jd|��)Nzhas no attribute)�AttributeError�name)r r r r r � __getattr__ � z_Glue.__getattr__N)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ � e Zd Zdd� ZdS )�TerminalSessionTransportc sx |�_ � �_|�_�j j��j �t�jj� �fdd�dd�� �fdd�}�j�t�j j|dd�� �jj� ||� d S )Nc s � j ���S r )�conn� sendCloser )r � protoSessionr r �<lambda>/ s z3TerminalSessionTransport.__init__.<locals>.<lambda>zSSH Proto Transport)�write�loseConnectionr c s � j �� d S r )�protor# r �r r r r# 4 r z9TerminalSessionTransport.__init__.<locals>.loseConnectionzChained Proto Transport)
r$ r �chainedProtocolr �makeConnectionr � dataReceivedr" �terminalProtocol� terminalSize)r r$ r& r �width�heightr# r )r r r r r % s( �� ��z!TerminalSessionTransport.__init__N�r r r r r r r r r $ � r c @ sF e Zd ZeZejZdd� Zdd� Z dd� Z
dd� Z d d
� Z d d � Z
d
S )�TerminalSessionc C s |d d� \| _ | _d S )N� )r, r+ )r �term�
windowSize�attrsr r r �getPtyM s zTerminalSession.getPtyc C s& | � || �� t�| j�| j| j� d S r )�transportFactory�chainedProtocolFactory�iconch�
IConchUser�originalr+ r, )r r$ r r r � openShellP s
�zTerminalSession.openShellc C s
t �d��)NzCannot execute commands)�econch�
ConchError)r r$ �cmdr r r � execCommandY �
zTerminalSession.execCommandc C � t d��)Nz,Unimplemented: TerminalSession.windowChanged��NotImplementedError)r �
newWindowSizer r r �
windowChanged\ � z TerminalSession.windowChangedc C r@ )Nz*Unimplemented: TerminalSession.eofReceivedrA r% r r r � eofReceived` rE zTerminalSession.eofReceivedc C � d S r r r% r r r �closedd s zTerminalSession.closedN)r r r r r5 r �ServerProtocolr6 r4 r: r> rD rF rH r r r r r/ H s r/ c @ r )� TerminalUserc C s* t j�| |� tj�| � tj| jd< d S )Ns session)r
�Adapterr r � ConchUserr �
SSHSession�
channelLookup)r r9 �avatarIdr r r r j s zTerminalUser.__init__Nr- r r r r rJ i r. rJ c @ s8 e Zd ZeZeZeZe j
Z dd� Z ddd�Z
dd� ZdS ) �
TerminalRealmc C sN t �� }| �||�}| �|�}| j|_| j|_|�tj|� |�tj |� |S r )
r
�
Componentized� userFactory�sessionFactoryr5 r6 � setComponentr7 r8 �ISession)r rO �comp�user�sessr r r �
_getAvatarw s
zTerminalRealm._getAvatarNc C s |d ur || _ d S d S r )r5 )r r5 r r r r � s
�zTerminalRealm.__init__c G s4 |D ]}|t ju rt j| �|�dd� f S qt� �)Nc S rG r r r r r r r! � s z-TerminalRealm.requestAvatar.<locals>.<lambda>)r7 r8 rY rB )r rO �mindr �ir r r �
requestAvatar� s
�zTerminalRealm.requestAvatarr )r r r rJ rR r/ rS r r5 r rI r6 rY r r\ r r r r rP p s
rP c @ s> e Zd ZU i Zeeef ed< i Zeeef ed< dd� ZdS )� ConchFactory�
publicKeys� privateKeysc C s
|| _ d S r )�portal)r r` r r r r � r? zConchFactory.__init__N) r r r r^ r �bytes�__annotations__r_ r r r r r r] � s
r] N) r �typingr �zope.interfacer �
twisted.conchr r r; r r7 �twisted.conch.insultsr �twisted.conch.sshr r �twisted.pythonr
r r rU rK r/ rL rJ rP �
SSHFactoryr] r r r r �<module> s $