File: //usr/lib/python3/dist-packages/twisted/conch/client/__pycache__/agent.cpython-38.pyc
U
W[� � @ sv d Z ddlZddlmZmZmZ ddlmZmZ ddl m
Z
G dd� dej �Z G dd � d ej �Z
G d
d � d ej�ZdS ) zJ
Accesses the key agent for user authentication.
Maintainer: Paul Swartz
� N)�agent�channel�keys)�protocol�reactor)�logc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�SSHAgentClientc C s t j�| � g | _d S �N)r r �__init__�blobs��self� r �</usr/lib/python3/dist-packages/twisted/conch/client/agent.pyr
s zSSHAgentClient.__init__c C s | � � �| j�S r )ZrequestIdentities� addCallback�
_cbPublicKeysr r r r �
getPublicKeys s z SSHAgentClient.getPublicKeysc C s& t �dt|� � dd� |D �| _d S )Nzgot %i public keysc S s g | ] }|d �qS )r r )�.0�xr r r �
<listcomp> s z0SSHAgentClient._cbPublicKeys.<locals>.<listcomp>)r �msg�lenr )r
Zblobcommr r r r s z SSHAgentClient._cbPublicKeysc C s | j rtj�| j �d��S dS )zj
Return a L{Key} from the first blob in C{self.blobs}, if any, or
return L{None}.
r N)r r ZKeyZ
fromString�popr r r r � getPublicKey# s zSSHAgentClient.getPublicKeyN)�__name__�
__module__� __qualname__r
r r r r r r r r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�SSHAgentForwardingChannelc sD t �tt�}|�tjd �}|�� j� |� � fdd�� d� _
d S )NZ
SSH_AUTH_SOCKc s � � � S r )�loseConnection)r r r r �<lambda>4 � z7SSHAgentForwardingChannel.channelOpen.<locals>.<lambda>� ) r Z
ClientCreatorr �SSHAgentForwardingLocalZ connectUNIX�os�environr � _cbGotLocalZ
addErrback�buf)r
Z specificDataZcc�dr r r � channelOpen0 s
z%SSHAgentForwardingChannel.channelOpenc C s || _ | j jj| _| j| j _d S r )�localZ transport�write� dataReceived)r
r) r r r r% 8 s z%SSHAgentForwardingChannel._cbGotLocalc C s | j |7 _ d S r )r& )r
�datar r r r+ > s z&SSHAgentForwardingChannel.dataReceivedc C s | j r| j �� d | _ d S r )r) r r r r r �closedB s
z SSHAgentForwardingChannel.closedN)r r r r( r% r+ r- r r r r r . s r c @ s e Zd ZdS )r" N)r r r r r r r r" H s r" )�__doc__r# Ztwisted.conch.sshr r r Ztwisted.internetr r Ztwisted.pythonr r Z
SSHChannelr ZProtocolr" r r r r �<module> s