File: //usr/local/CyberCP/lib64/python3.10/site-packages/asyncssh/__pycache__/kex_rsa.cpython-310.pyc
o
�hU � @ s d Z ddlmZmZ ddlmZmZmZ ddlm Z m
Z
ddl m Z m
Z
mZ ddl mZmZ ddlmZmZmZ dd lmZmZ dd
lmZmZ dd lmZ er^dd l m Z m Z dd
l m Z dZ dZ!dZ"G dd� de �Z#dedddfdedddffD ]\Z$Z%Z&Z'Z(e
e$e#e%e&e'fe(� qzdS ) zRSA key exchange handler� )�sha1�sha256)�
TYPE_CHECKING�Optional�cast� )�Kex�register_kex_alg)�HashType�KeyExchangeFailed�
ProtocolError)�get_symbol_names� randrange)�MPInt�String� SSHPacket)�KeyImportError�SSHKey)�decode_ssh_public_key�generate_private_key)�RSAKey)�
SSHConnection�SSHClientConnection)�SSHServerConnection� � � c
s� e Zd ZdZee� d�Zdedddede de f
� fd d
� Z
dd
d�Z d efdd�Z de de de
d d fdd�Zde de de
d d fdd�Zde de de
d d fdd�ZeeeeeeiZ� ZS )�_KexRSAz Handler for RSA key exchange�
MSG_KEXRSA�alg�connr �hash_alg�key_size� hash_sizec sN t � �|||� || _d|d| d > | _d| _d | _d| _d| _d| _d S )Nr � �1 � r ) �super�__init__� _key_size�_k_limit�_host_key_data�
_trans_key�_trans_key_data�_k� _encrypted_k)�selfr r r! r" r# �� __class__� �C/usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/kex_rsa.pyr( 4 s
z_KexRSA.__init__�returnNc � sp �| j �� r6td| j �}|�� }|dusJ �|j| _td| jd�| _| jj| _ | �
t t | j�t | j �� dS dS )zStart RSA key exchanger Nzssh-rsa)r" )
�_conn� is_serverr �get_server_host_key� public_datar+ r r) r, r- � send_packet�MSG_KEXRSA_PUBKEYr )r0 � server_conn�host_keyr3 r3 r4 �startC s �
�
��z
_KexRSA.startc C s` | � � }|�| j�� � |�t| j�� |�t| j�� |�t| j�� |�t| j �� |�
� S )z@Compute a hash of key information associated with the connection) � _hash_alg�updater6 �get_hash_prefixr r+ r- r/ r r. �digest)r0 �hash_objr3 r3 r4 �
_compute_hashS s z_KexRSA._compute_hash�_pkttype�_pktid�packetc C s� | j �� r td��|�� | _|�� | _|�� zt| j�}W n ty* td�d�w t t
|�}t | j �| _
t t|�t| j
�| j��| _| �tt| j�� dS )z Process a KEXRSA pubkey messagez Unexpected KEXRSA pubkey msgzInvalid KEXRSA pubkey msgN)r6 r7 r �
get_stringr+ r- � check_endr r r r r r* r. �bytes�encryptr � algorithmr/ r: �MSG_KEXRSA_SECRETr )r0 rE rF rG �pubkey� trans_keyr3 r3 r4 �_process_pubkey^ s
�
�z_KexRSA._process_pubkeyc
C s� | j �� r td��|�� | _|�� tt| j�}|� | j| j
�}|s&t d��t |�}|�
� | _|�� td| j �}|�� }|dusCJ �| �� }|�|�} | �tt| �� | j �t| j�|� dS )z Process a KEXRSA secret messagez Unexpected KEXRSA secret msgz Key exchange decryption failedr N)r6 � is_clientr rH r/ rI r r r, �decryptrL r r � get_mpintr. r8 rD �signr: �MSG_KEXRSA_DONEr � send_newkeysr )
r0 rE rF rG rO � decrypted_kr< r= �h�sigr3 r3 r4 �_process_secretu s$
z_KexRSA._process_secretc C sn | j �� r td��|�� }|�� td| j �}|�| j�}| �� }|� ||�s+t
d��| j � t | j
�|� dS )z Process a KEXRSA done messagezUnexpected KEXRSA done msgr zKey exchange hash mismatchN)r6 r7 r rH rI r �validate_server_host_keyr+ rD �verifyr rV r r. )r0 rE rF rG rY � client_connr= rX r3 r3 r4 �
_process_done� s
z_KexRSA._process_done)r5 N)�__name__�
__module__� __qualname__�__doc__r
�globals�_handler_namesrJ r
�intr( r> rD r rP rZ r^ r; rM rU �_packet_handlers�
__classcell__r3 r3 r1 r4 r / s: ��
�
�
�
�
�
� �r s rsa2048-sha256i � Ts rsa1024-sha1i � FN))rb �hashlibr r �typingr r r �kexr r �miscr
r r r
r rG r r r �
public_keyr r r r �rsar �
connectionr r r r; rM rU r �_namer? r) �
_hash_size�_defaultr3 r3 r3 r4 �<module> s0 ���