File: //usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/__pycache__/kex.cpython-310.pyc
o
�hn � @ s� U d Z ddlZddlmZ ddlmZmZmZmZm Z m
Z
ddl m Z ddl
mZ ddlmZ er7dd lmZ ee Zeee e
d
eef f Zg Zeed < g Zeed < i Zeed
< g Zeed< g Z eed< i Z eed< G dd
� d
e�Z dede
e dede de ddf dd�Z dede
e dede de ddf dd�Z!dee fd d �Z"dee fd d �Z#d ee d!ee d"e dee fd#d$�Z$d%d&dede fd'd(�Z%dS ))zSSH key exchange handlers� N)�md5)�
TYPE_CHECKING�Dict�List�Sequence�Tuple�Type� )� SSHLogger)�HashType)�SSHPacketHandler)�
SSHConnection�Kex� _kex_algs�_default_kex_algs�
_kex_handlers�
_gss_kex_algs�_default_gss_kex_algs�_gss_kex_handlersc @ sz e Zd ZdZdedddefdd�Zdd
d �Zd ed
edd fdd�Z e
de fdd��Z dedededededef dd�Z
d S )r z&Parent class for key exchange handlers�alg�connr
�hash_algc C s || _ || _|j| _|| _d S )N)� algorithm�_conn�logger�_logger� _hash_alg)�selfr r r � r �?/usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/kex.py�__init__5 s
z Kex.__init__�returnNc � s �t �)zStart key exchange)�NotImplementedError�r r r r �start= s �z Kex.start�pkttype�argsc G s | j j|g|�R d| i� dS )zSend a kex packet�handlerN)r � send_packet)r r% r& r r r r( B s zKex.send_packetc C s | j S )z(A logger associated with this connection)r r# r r r r G s z
Kex.logger�k�h�x�
session_id�keylenc C sf d}t |�|k r-| �� }|�|� |�|� |�|r |n|| � ||�� 7 }t |�|k s|d|� S )z(Compute keys from output of key exchange� N)�lenr �update�digest)r r) r* r+ r, r- �key�hash_objr r r � compute_keyM s
� zKex.compute_key)r! N)�__name__�
__module__� __qualname__�__doc__�bytesr r r$ �intr( �propertyr
r r4 r r r r r 2 s
���r r' r r& �defaultr! c C �* t �| � |r t�| � |||ft| < dS )z!Register a key exchange algorithmN)r �appendr r �r r' r r& r< r r r �register_kex_alg\ �
r@ c C r= )z(Register a GSSAPI key exchange algorithmN)r r> r r r? r r r �register_gss_kex_algh rA rB c C � t t S )z(Return supported key exchange algorithms)r r r r r r � get_kex_algst � rD c C rC )z&Return default key exchange algorithms)r r r r r r �get_default_kex_algsz rE rF �kex_algs�mechs�host_key_availablec C sb g }| D ]*}|� d�r'|D ]}dt�t|��� �dd� }|�|| � q
q|r.|�|� q|S )z<Add mechanisms to GSS entries in key exchange algorithm list� gss-� -N���)�
startswith�binascii�
b2a_base64r r1 r> )rG rH rI �expanded_kex_algsr �mech�suffixr r r �expand_kex_algs� s
�
�rS r r
c C sL |� d�r|�dd�d }t| \}}}nt| \}}}||| |g|�R � S )z�Return a key exchange handler
The function looks up a key exchange algorithm and returns a
handler which can perform that type of key exchange.
rJ rK r r )rM �rsplitr r )r r r' r r& r r r �get_kex� s
rU )&r8 rN �hashlibr �typingr r r r r r �loggingr
�miscr �packetr �
connectionr
r9 � _KexAlgList�object�
_KexAlgMapr �__annotations__r r r r r r �boolr@ rB rD rF rS rU r r r r �<module> sP *��
� ��
� �
�