File: //usr/lib/python3/dist-packages/twisted/conch/ssh/__pycache__/agent.cpython-38.pyc
U
W[�% � @ s d Z ddlmZmZ ddlZddlmZmZmZ ddl m
Z
m Z ddl m
Z
ddlmZmZ ddlmZ G d d
� d
ej�ZG d d � d ej�Zd
ZdZdZdZdZdZdZ dZ dZ dZ dZ dZ!dZ"i Z#e$� �%� �&� D ](\Z'Z(e'dd� dkr�e'dd� e#e(< q�dS ) z�
Implements the SSH v2 key agent protocol. This protocol is documented in the
SSH source code, in the file
U{PROTOCOL.agent<http://www.openbsd.org/cgi-bin/cvsweb/src/usr.bin/ssh/PROTOCOL.agent>}.
Maintainer: Paul Swartz
� )�absolute_import�divisionN)�NS�getNS�getMP)�
ConchError�MissingKeyStoreError)�keys)�defer�protocol)�
itervaluesc @ sb e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zdd
d�Z dd� Z
dd� Z dd� Z dd� Z
dS )�SSHAgentClientz�
The client side of the SSH agent protocol. This is equivalent to
ssh-add(1) and can be used with either ssh-agent(1) or the SSHAgentServer
protocol, also in this package.
c C s d| _ g | _d S �N� )�buf� deferreds��self� r �9/usr/lib/python3/dist-packages/twisted/conch/ssh/agent.py�__init__ s zSSHAgentClient.__init__c C s� | j |7 _ t| j �dkr d S t�d| j d d� �d }t| j �d| k rPd S | j dd| � | j d| d � }| _ t|dd� �}| j�d�}|tkr�|�t d�� q|t
kr�|� d� q|� |� qd S )N� �!Lr � z
agent failurer ) r �len�struct�unpack�ordr �pop�
AGENT_FAILUREZerrbackr �
AGENT_SUCCESS�callback)r �data�packLen�packet�reqType�dr r r � dataReceived$ s ( zSSHAgentClient.dataReceivedc C s> t �dt|�d |�| }| j�|� t�� }| j�|� |S �Nz!LBr ) r �packr � transport�writer
ZDeferredr �append)r r% r" r) r& r r r � sendRequest7 s
zSSHAgentClient.sendRequestc C s | � td�}|�| j� |S )z�
@return: A L{Deferred} which will fire with a list of all keys found in
the SSH agent. The list of keys is comprised of (public key blob,
comment) tuples.
r )r- �AGENTC_REQUEST_IDENTITIES� addCallback�_cbRequestIdentities)r r& r r r �requestIdentities? s z SSHAgentClient.requestIdentitiesc C s� t |dd� �tkr,tdt |dd� � ��t�d|dd� �d }g }|dd� }t|�D ]*}t|�\}}t|�\}}|�||f� q\|S )z}
Unpack a collection of identities into a list of tuples comprised of
public key blobs and comments.
r r zunexpected response: %ir � N)r �AGENT_IDENTITIES_ANSWERr r r �ranger r, )r r" �numKeys�result�i�blob�commentr r r r0 J s z#SSHAgentClient._cbRequestIdentitiesr c C s |}|t |�7 }| �t|�S )zK
Add a private key blob to the agent's collection of keys.
)r r- �AGENTC_ADD_IDENTITY)r r8 r9 �reqr r r � addIdentity[ s zSSHAgentClient.addIdentityc C s0 t |�}|t |�7 }|d7 }| �t|��| j�S )a�
Request that the agent sign the given C{data} with the private key
which corresponds to the public key given by C{blob}. The private
key should have been added to the agent already.
@type blob: L{bytes}
@type data: L{bytes}
@return: A L{Deferred} which fires with a signature for given data
created with the given key.
� )r r- �AGENTC_SIGN_REQUESTr/ � _cbSignData)r r8 r" r; r r r �signDatad s zSSHAgentClient.signDatac C sD t |dd� �tkr,tdt |dd� � ��t|dd � �d }|S )Nr r zunexpected data: %i)r �AGENT_SIGN_RESPONSEr r )r r" Z signaturer r r r? u s zSSHAgentClient._cbSignDatac C s t |�}| �t|�S )zp
Remove the private key corresponding to the public key in blob from the
running agent.
)r r- �AGENTC_REMOVE_IDENTITY)r r8 r; r r r �removeIdentity| s z SSHAgentClient.removeIdentityc C s | � td�S )z9
Remove all keys from the running agent.
r )r- � AGENTC_REMOVE_ALL_IDENTITIESr r r r �removeAllIdentities� s z"SSHAgentClient.removeAllIdentitiesN)r )�__name__�
__module__� __qualname__�__doc__r r' r- r1 r0 r< r@ r? rC rE r r r r r
s
r
c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�SSHAgentServerz�
The server side of the SSH agent protocol. This is equivalent to
ssh-agent(1) and can be used with either ssh-add(1) or the SSHAgentClient
protocol, also in this package.
c C s
d| _ d S r )r r r r r r � s zSSHAgentServer.__init__c C s� | j |7 _ t| j �dkr d S t�d| j d d� �d }t| j �d| k rPd S | j dd| � | j d| d � }| _ t|dd� �}t�|d �}|s�| �td� qt | d| �}t | j
dd �d kr�| �td� t � �||dd � � qd S )Nr r r r r z agentc_%sr ) r r r r r �messages�get� sendResponser �getattr�factoryr )r r" r# r$ r% ZreqName�fr r r r' � s ( zSSHAgentServer.dataReceivedc C s* t �dt|�d |�| }| j�|� d S r( )r r) r r* r+ )r r% r" r) r r r rM � s zSSHAgentServer.sendResponsec C sz |dks t �t| jj�}g }|�t�d|�� t| jj�D ](\}}|�t|� � �� |�t|�� q:| �
t d� |�� dS )zQ
Return all of the identities that have been added to the server
r r N)
�AssertionErrorr rO r r, r r) r r r8 rM r3 �join)r r" r5 Zresp�keyr9 r r r �agentc_REQUEST_IDENTITIES� s z(SSHAgentServer.agentc_REQUEST_IDENTITIESc C sb t |�\}}|| jjkr$| �td�S t |�\}}|dks<t�| �tt| jj| d �|��� dS )z�
Data is a structure with a reference to an already added key object and
some data that the clients wants signed with that key. If the key
object wasn't loaded, return AGENT_FAILURE, else return the signature.
r r= r N) r rO r rM r rQ rA r Zsign)r r" r8 r@ r r r �agentc_SIGN_REQUEST� s z"SSHAgentServer.agentc_SIGN_REQUESTc C s� t |�\}}|dkrd}n |dkr(d}nt�d| ��t||�d }t |�\}}tjj|dd�}||f| jj|�� < | �t d � d
S ) z�
Adds a private key to the agent's collection of identities. On
subsequent interactions, the private key can be accessed using only the
corresponding public key.
s ssh-rsa� s ssh-dssr2 zunknown blob type: %s���Z private_blob��typer N)
r r Z BadKeyErrorr �Key�
fromStringrO r8 rM r )r r" ZkeyType�restZnmpr9 �kr r r �agentc_ADD_IDENTITY� s z"SSHAgentServer.agentc_ADD_IDENTITYc C s: t |�\}}tjj|dd�}| jj|�� = | �td� dS )zR
Remove a specific key from the agent's collection of identities.
r8 rX r N)r r rZ r[ rO r8 rM r )r r" r8 �_r] r r r �agentc_REMOVE_IDENTITY� s z%SSHAgentServer.agentc_REMOVE_IDENTITYc C s$ |dks t �i | j_| �td� dS )zL
Remove all keys from the agent's collection of identities.
r N)rQ rO r rM r �r r" r r r � agentc_REMOVE_ALL_IDENTITIES� s z+SSHAgentServer.agentc_REMOVE_ALL_IDENTITIESc C s | � tt�dd�� dS )z�
v1 message for listing RSA1 keys; superseded by
agentc_REQUEST_IDENTITIES, which handles different key types.
r r N)rM �AGENT_RSA_IDENTITIES_ANSWERr r) ra r r r � agentc_REQUEST_RSA_IDENTITIES� s z,SSHAgentServer.agentc_REQUEST_RSA_IDENTITIESc C s | � td� dS )z�
v1 message for removing RSA1 keys; superseded by
agentc_REMOVE_IDENTITY, which handles different key types.
r N�rM r ra r r r �agentc_REMOVE_RSA_IDENTITY s z)SSHAgentServer.agentc_REMOVE_RSA_IDENTITYc C s | � td� dS )z�
v1 message for removing all RSA1 keys; superseded by
agentc_REMOVE_ALL_IDENTITIES, which handles different key types.
r Nre ra r r r � agentc_REMOVE_ALL_RSA_IDENTITIES
s z/SSHAgentServer.agentc_REMOVE_ALL_RSA_IDENTITIESN)rF rG rH rI r r' rM rT rU r^ r` rb rd rf rg r r r r rJ � s
rJ r � r2 rV � � � � �
� � � � � ZAGENTC_))rI Z
__future__r r r Ztwisted.conch.ssh.commonr r r Ztwisted.conch.errorr r Ztwisted.conch.sshr Ztwisted.internetr
r Ztwisted.python.compatr ZProtocolr
rJ Z AGENTC_REQUEST_RSA_IDENTITIESrc r r ZAGENTC_REMOVE_RSA_IDENTITYZ AGENTC_REMOVE_ALL_RSA_IDENTITIESr. r3 r> rA r: rB rD rK �locals�copy�items�name�valuer r r r �<module> s6 u