File: //usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/crypto/__pycache__/pq.cpython-310.pyc
o
�h�
� @ s� U d Z ddlZddlZddlmZmZ dddd�Zeeeeeeee f f e
d< d Z d Z d
D ] Z
ej�e
�Zer: nq.dZerTej�e�Zeed �pNeed �Z eed
�Z G dd� d�ZdS )zGA shim around liboqs for Streamlined NTRU Prime post-quantum encryption� N)�Mapping�Tuple)i� i` i@ � �KEM_ml_kem_768)� i` r r �KEM_ml_kem_1024)i� i� i r �KEM_ntruprime_sntrup761)s mlkem768s mlkem1024s sntrup761�_pq_algsF)�oqs�liboqs�OQS_KEM_ml_kem_768_keypair� OQS_KEM_ml_kem_768_ipd_keypair�#OQS_KEM_ntruprime_sntrup761_keypairc @ sd e Zd ZdZdefdd�Zdeeef fdd�Zdedeeef fd d
�Zd ed edefd
d�Z dS )�PQDHz=A shim around liboqs for post-quantum key exchange algorithms�alg_namec C s� zt | \| _| _| _| _}W n ty td|� ��d �w ttd| d �s,|d7 }t td| d �| _
t td| d �| _ t td| d �| _ d S )NzUnknown PQ algorithm �OQS_�_keypair�_ipd�_encaps�_decaps)
r � pubkey_bytes�
privkey_bytes�ciphertext_bytes� secret_bytes�KeyError�
ValueError�hasattr�_oqs�getattrr r r )�selfr �oqs_name� r! �E/usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/crypto/pq.py�__init__8 s
� �z
PQDH.__init__�returnc C s0 t �| j�}t �| j�}| �||� |j|jfS )zMake a new key pair)�ctypes�create_string_bufferr r r �raw)r �pubkey�privkeyr! r! r" �keypairG s z PQDH.keypairr( c C sH t |�| jkr td��t�| j�}t�| j�}| �|||� |j|jfS )z9Generate a random secret and encrypt it with a public keyzInvalid public key) �lenr r r% r& r r r r' )r r( �
ciphertext�secretr! r! r" �encapsP s z PQDH.encapsr, r) c C s6 t |�| jkr td��t�| j�}| �|||� |jS )z/Decrypt an encrypted secret using a private keyzInvalid ciphertext)r+ r r r% r& r r r' )r r, r) r- r! r! r" �decaps] s
z PQDH.decapsN)
�__name__�
__module__� __qualname__�__doc__�bytesr# r r* r. r/ r! r! r! r" r 5 s
r )r3 r% � ctypes.util�typingr r r r4 �int�str�__annotations__�mlkem_available�sntrup_available�lib�util� find_library�_oqs_lib�cdll� LoadLibraryr r r r! r! r! r" �<module> s. $� �
�