File: //usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/crypto/__pycache__/ec.cpython-310.pyc
o
�h� � @ s� U d Z ddlmZmZmZmZ ddlmZ ddlm Z ddl
m Z ddl
m Z ddl
mZmZmZ e je je je jd �Zeeee j f ed
< G d d � d e�ZG d
d� de�ZG dd� de�ZG dd� d�Z dS )z;A shim around PyCA for elliptic curve keys and key exchange� )�Mapping�Optional�Type�cast)�InvalidSignature)�ec)�Encoding)� PublicFormat� )� CryptoKey�PyCAKey�hashes)s 1.3.132.0.10s nistp256s nistp384s nistp521�_curvesc s� e Zd ZdZ ddededejdedeej f
� fdd �
Z
e ded
e ej
fd d ��Zed
efd
d��Zed
efdd��Zed
efdd��Zed
ee fdd��Zed
efdd��Zed
ee fdd��Z� ZS )�_ECKeyz+Base class for shim around PyCA for EC keysN�pyca_key�curve_id�pub�point�privc s( t � �|� || _|| _|| _|| _d S �N)�super�__init__� _curve_id�_pub�_point�_priv)�selfr r r r r �� __class__� �E/usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/crypto/ec.pyr / s
z_ECKey.__init__�returnc C s. zt | W S ty td|�� � ��d�w )z Look up curve and hash algorithm�Unknown EC curve N)r �KeyError�
ValueError�decode)�clsr r r r � lookup_curve9 s
�z_ECKey.lookup_curvec C � | j S )zReturn the EC curve name)r �r r r r r B � z_ECKey.curve_idc C � | j jS )z!Return the EC public x coordinate)r �xr) r r r r, H � z_ECKey.xc C r+ )z!Return the EC public y coordinate)r �yr) r r r r. N r- z_ECKey.yc C s | j r| j jS dS )z)Return the EC private value as an integerN)r �
private_valuer) r r r �dT s z_ECKey.dc C r( )z9Return the EC public point value encoded as a byte string)r r) r r r � public_valueZ r* z_ECKey.public_valuec C s, | j r| jjjd d }| j j�|d�S dS )z4Return the EC private value encoded as a byte string� � �bigN)r r �curve�key_sizer/ �to_bytes)r �keylenr r r r/ ` s z_ECKey.private_valuer )�__name__�
__module__� __qualname__�__doc__r �bytesr �EllipticCurvePublicNumbersr �EllipticCurvePrivateNumbersr � classmethodr �
EllipticCurver' �propertyr �intr, r. r0 r1 r/ �
__classcell__r r r r r , s0 ����
r c @ s\ e Zd ZdZededededd fdd��Zededd fdd ��Zdd ed e defd
d�Z
dS )�ECDSAPrivateKeyz)A shim around PyCA for ECDSA private keysr r1 r/ r! c C s6 | � |�}t�||� �}|�� }|j}| |||||�S )z Construct an ECDSA private key)r' r �derive_private_key�private_numbers�public_numbers)r& r r1 r/ r5 �priv_keyr r r r r � constructn s
zECDSAPrivateKey.constructc C sN | � |�}t�|� �}|�� }|�� }|�� }|�tjt j
�}| |||||�S )z Generate a new ECDSA private key) r' r �generate_private_keyrG �
public_keyrH � public_bytesr �X962r �UncompressedPoint)r& r r5 rI r �pub_keyr r1 r r r �generate{ s
�zECDSAPrivateKey.generate� �data� hash_namec C s$ t d| j�}|�|t�t| � ��S )zSign a block of datazec.EllipticCurvePrivateKey)r r �signr �ECDSAr
)r rS rT rI r r r rU � s zECDSAPrivateKey.signN�rR ) r9 r: r; r< r@ r= rC rJ rQ �strrU r r r r rE k s
� � rE c @ sF e Zd ZdZedededd fdd��Zdded ed
edefd d �Z d
S )�ECDSAPublicKeyz(A shim around PyCA for ECDSA public keysr r1 r! c C s0 | � |�}tj�|� |�}|�� }| ||||�S )z Construct an ECDSA public key)r' r �EllipticCurvePublicKey�from_encoded_pointrH )r& r r1 r5 rP r r r r rJ � s
�zECDSAPublicKey.constructrR rS �sigrT c C sB zt d| j�}|�||t�t| � �� W dS ty Y dS w )z'Verify the signature on a block of datazec.EllipticCurvePublicKeyTF)r r �verifyr rV r
r )r rS r\ rT rP r r r r] � s �zECDSAPublicKey.verifyNrW )
r9 r: r; r< r@ r= rJ rX �boolr] r r r r rY � s � � rY c @ sP e Zd ZdZdefdd�Zdefdd�Zdedefd d
�Zdedefd d �Z d
S )�ECDHz(A shim around PyCA for ECDH key exchanger c C sB zt | }W n ty td|�� � ��d �w t�|� �| _d S )Nr" )r r# r$ r% r rK � _priv_key)r r r5 r r r r � s �z
ECDH.__init__r! c C s | j �� }|�tjtj�S )z.Return the public key to send in the handshake)r` rL rM r rN r rO )r rP r r r �
get_public� s
�zECDH.get_public� peer_publicc C s$ t j�| jj|�}| j�t �� |�S )z9Return the shared key from the peer's public key as bytes)r rZ r[ r` r5 �exchanger_ )r rb �peer_keyr r r �get_shared_bytes� s �zECDH.get_shared_bytesc C s t �| �|�d�S )z0Return the shared key from the peer's public keyr4 )rC �
from_bytesre )r rb r r r �
get_shared� s zECDH.get_sharedN)
r9 r: r; r< r= r ra re rC rg r r r r r_ � s r_ N) r< �typingr r r r �cryptography.exceptionsr �)cryptography.hazmat.primitives.asymmetricr �,cryptography.hazmat.primitives.serializationr r �miscr r r
� SECP256K1� SECP256R1� SECP384R1� SECP521R1r r= rA �__annotations__r rE rY r_ r r r r �<module> s �?*