File: //usr/local/CyberCP/lib64/python3.10/site-packages/asyncssh/__pycache__/agent_win32.cpython-310.pyc
o
�h � @ s� d Z ddlZddlZddlZddlZddlmZmZmZm Z ddl
m Z er.ddl m
Z
mZ zddlZddlZddlZddlZdZW n eyM dZY nw d Zd
Zd Zd Zddd�ZG dd� dej�ZG dd� d�Z G dd� d�Z de d
ed fdd�Z dS )z"SSH agent support code for Windows� N)�
TYPE_CHECKING�Tuple�Union�cast� )� open_file)� AgentReader� AgentWriterTFl �P� i �Pageantz\\.\pipe\openssh-ssh-agent�return�win32ui.PyCWndc C s@ t rzt�tt�W S tjy ttjd�d�w ttjd�d�)z"Find and return the Pageant windowzAgent not foundNzPyWin32 not installed)�_pywin32_available�win32ui�
FindWindow� _AGENT_NAME�error�OSError�errno�ENOENT� r r �G/usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/agent_win32.py�_find_agent_window9 s �r c @ s0 e Zd ZdZdejjfdejjfdejffZ dS )�_CopyDataStructz7Windows COPYDATASTRUCT argument for WM_COPYDATA message�dwData�cbData�lpDataN)
�__name__�
__module__� __qualname__�__doc__�ctypes�wintypes�LPARAM�DWORD�c_char_p�_fields_r r r r r E s
�r c @ sR e Zd ZdZddd�Zdeddfdd�Zd edefd
d �Zdd d
�Z ddd�Z
dS )�_PageantTransportz0Transport to connect to Pageant agent on Windowsr Nc
C s� t � t�� d��| _z
t�d| jtdd�| _W n tjy- } z tt j
t |��d �d }~ww t t
t| j�d | j�� �| _d| _d S )N�08x� r r F)r �win32api�GetCurrentThreadId�_mapname�mmapfile�_AGENT_MAX_MSGLEN�_mapfiler r r �EIO�strr �_AGENT_COPYDATA_ID�len�encode�_cds�_writing)�self�excr r r �__init__P s
����
z_PageantTransport.__init__�datac
C sV | j s | j�d� d| _ z | j�|� W dS ty* } z ttjt|��d�d}~ww )z#Write request data to Pageant agentr TN) r5 r. �seek�write�
ValueErrorr r r/ r0 )r6 r9 r7 r r r r; ^ s ��z_PageantTransport.write�nc � sl �| j r"t� }|�tjdtt| j��stt j
d��d| _ | j � d� | j �
|�}t|�|kr4t�||��|S )z%Read response data from Pageant agentr z Unable to send agent requestF)r5 r � SendMessage�win32con� WM_COPYDATAr �intr4 r r r/ r. r: �readr2 �asyncio�IncompleteReadError)r6 r= �cwnd�resultr r r � readexactlyj s �
� z _PageantTransport.readexactlyc C � | j r
| j �� dS dS )z Close the connection to PageantN)r. �close�r6 r r r rI ~ � �z_PageantTransport.closec � � �dS �z Wait for the transport to closeNr rJ r r r � wait_closed� � z _PageantTransport.wait_closed�r N) r r r r r8 �bytesr; rA rG rI rN r r r r r& M s
r& c @ sV e Zd ZdZdefdd�Zdedefdd�Zd edd
fd d �Z dd
d�Z
ddd�Z d
S )�_W10OpenSSHTransportz3Transport to connect to OpenSSH agent on Windows 10�
agent_pathc C s t |d�| _d S )Nzr+b)r �
_agentfile)r6 rS r r r r8 � s z _W10OpenSSHTransport.__init__r= r c � s* �| j �|�}t|�|krt�||��|S )z%Read response data from OpenSSH agent)rT rB r2 rC rD )r6 r= rF r r r rG � s
� z _W10OpenSSHTransport.readexactlyr9 Nc C s | j �|� dS )z#Write request data to OpenSSH agentN)rT r; )r6 r9 r r r r; � s z_W10OpenSSHTransport.writec C rH )z Close the connection to OpenSSHN)rT rI rJ r r r rI � rK z_W10OpenSSHTransport.closec � rL rM r rJ r r r rN � rO z _W10OpenSSHTransport.wait_closedrP ) r r r r r0 r8 rA rQ rG r; rI rN r r r r rR � s
rR rS c � sF �d}| szt � t� }W n ty t} Y nw |s t| �}||fS )z<Open a connection to the Pageant or Windows 10 OpenSSH agentN)r r& r �_DEFAULT_OPENSSH_PATHrR )rS � transportr r r �
open_agent� s �
�rW )r r ) r rC r �ctypes.wintypesr �typingr r r r �miscr �agentr r r, r) r? r r
� ImportErrorr1 r- r rU r � Structurer r& rR r0 rW r r r r �<module> s6 �
;