File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/Cryptodome/SelfTest/Protocol/__pycache__/test_HPKE.cpython-310.pyc
Back
o ;jE � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d dlm Z d dlmZmZmZ G d d � d ej�ZG d d � d ej�Zi fd d�Zedkrbdd� Zejdd� dS dS )� N)� unhexlify)�HPKE)�DeserializeError)�ECC)�list_test_cases)�DH)�SHA256�SHA384�SHA512c @ s� e Zd Zejdd�Zejdd�Zdddddd�Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'S )(� HPKE_Tests�p256��curve�A �a � � �8 )r �p384�p521Z curve25519Zcurve448c C s� t j|d�}|}tj|�� |d�}| �t|j�| j| � |j ddd�}tj|||jd�}|j |dd�}| �d|� |� d�}|� |�} | �d| � d S )Nr �� receiver_key�aead_id� ABC� DEF�Z auth_data�r r �encs GHI) r �generater �new� public_key� assertEqual�lenr �curves�seal�unseal) �selfr r �key1� encryptor�ct� decryptor�ptZct2Zpt2� r, �Z/opt/saltstack/salt/lib/python3.10/site-packages/Cryptodome/SelfTest/Protocol/test_HPKE.py� round_trip s �� zHPKE_Tests.round_tripc C s, | j �� D ]}tjD ]}| �||� q qd S �N)r# �keysr �AEADr. )r& r r r, r, r- �test_round_trip5 s ��zHPKE_Tests.test_round_tripc C �" t jj}t j| j�� |dd� d S )N)� as cccccccccccccccccccccccccccccccc)r r �psk�r r1 � AES128_GCMr r' r �r& r r, r, r- �test_psk: � �zHPKE_Tests.test_pskc C r3 )Ns baba)r r �infor6 r8 r, r, r- � test_info@ r: zHPKE_Tests.test_infoc C sb t jdd�}| �t��}tj|�� tjjd� W d � n1 s!w Y | � dt |j �� d S )NZp224r r zUnsupported curve) r r � assertRaises� ValueErrorr r r r1 r7 �assertIn�str� exception�r& Zkey3�cmr, r, r- �test_neg_unsupported_curveF s � �z%HPKE_Tests.test_neg_unsupported_curvec C sX | � t��}tj| j| jtjjd� W d � n1 s w Y | �dt |j �� d S )N�r � sender_keyr zExactly 1 private key) r= r> r r r' �key2r1 r7 r? r@ rA �r& rC r, r, r- � test_neg_too_many_private_keysM s � �z)HPKE_Tests.test_neg_too_many_private_keysc C sf t jdd�}| �t��}tj| j�� |tjj d� W d � n1 s#w Y | � dt |j �� d S )Nr r rE zbut recipient key) r r r= r> r r r' r r1 r7 r? r@ rA rB r, r, r- �test_neg_curve_mismatchT s � �z"HPKE_Tests.test_neg_curve_mismatchc C s� | � t��}tj| j�� dtjjd� W d � n1 s w Y | � t��}tj| j�� dtjjd� W d � n1 s?w Y | � t��}tj| j�� dtjjd� W d � n1 saw Y | �dt |j �� d S )N)� s GGGGGGGGGGGGGGGGGGGGGGGGGGGGGGGG)r r5 r )� JJJrK )rL s YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYz at least 32) r= r> r r r' r r1 r7 r? r@ rA rH r, r, r- � test_neg_psk\ s& � � � � � �zHPKE_Tests.test_neg_pskc C s� d}| � t�� tj| jtjj|d� W d � n1 s w Y | � t�� }tj| j�� | j�� j dd�tjjd� W d � n1 sFw Y | � dt |j �� | � t��}tj| jtjjd� W d � n1 snw Y | � dt |j �� d S ) NsA �8888888888888888888888888888888888888888888888888888888888888888r �raw��format)r r r z'enc' cannot be an inputr z'enc' required) r= r r r r' r1 r7 r> r � export_keyr? r@ rA )r& Z wrong_encrC r, r, r- �test_neg_wrong_encm s( � � � � � �z HPKE_Tests.test_neg_wrong_encc C s^ t j| jt jj| j�� jdd�d�}| �t �� |� d� W d � d S 1 s(w Y d S )NrN rO r s< XYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZXYZ) r r r' r1 �CHACHA20_POLY1305rG r rQ r= r> r% )r& r* r, r, r- �test_neg_unseal_wrong_ct s � "�z#HPKE_Tests.test_neg_unseal_wrong_ctc C sx t jj}t j| j�� |d�}|jddd�}t j| j||jd�}| �t �� |� |� W d � d S 1 s5w Y d S )Nr r r r r ) r r1 rS r r' r r$ r r= r> r% )r& r r( r) r* r, r, r- � test_neg_unseal_no_auth_data� s �� "�z'HPKE_Tests.test_neg_unseal_no_auth_datac C s� d}t �t�|��}d}t�|�}d}t�|�}d}t�|�}d} t�| �} d} t�| �} d} t�| �}d}t�|�}tjj}tj||||d �}|�|| �}| � ||� |�|| �}| � ||� d S ) NZ@4612c550263fc8ad58375df3f557aac531d26850903e55a9f23f21d8534e8ac8�:4265617574792069732074727574682c20747275746820626561757479ZZf938558b5d72f1a23810b4be2ab4f84331acc02fc97babc53a52ae8218a355a96d8770ac83d07bea87e13c512aZ@37fda3567bdbd628e88668c3c8d7e97d1d1253b6d4ea6d44c150f741f1bf4431�436f756e742d30�436f756e742d31�(4f6465206f6e2061204772656369616e2055726eZZaf2d7e9ac9ae7e270f46ba1f975be53c09f8d875bdc8535458c2494e8a6eab251c03d0c22a56b8ca42c2063b84)r r r; r � r �import_x25519_private_key�bytes�fromhexr r1 r7 r r% r! )r& �keyR_hex�keyR�pt_hexr+ �ct0_hex�ct0�enc_hexr �aad0_hex�aad0�aad1_hex�aad1�info_hexr; �ct1_hex�ct1r r* �pt_X0�pt_X1r, r, r- �test_x25519_mode_0� s4 � z HPKE_Tests.test_x25519_mode_0c C s� d}t �t�|��}d}t�|�}d}t�|�}d}t�|�}d} t�| �} d} t�| �} d} t�| �}d}t�|�}d }t�|�}d }t�|�}tjj}tj|||||f| d �}|�| |�}| � ||� |�||�}| � ||� d S ) NZ@c5eb01eb457fe6c6f57577c5413b931550a162c71a03ac8d196babbd4e5ce0fd�,456e6e796e20447572696e206172616e204d6f726961�@0247fd33b913760fa1fa51e1892d9f307fbe65eb171e8132c2af18555a738b82rV ZZe52c6fed7f758d0cf7145689f21bc1be6ec9ea097fef4e959440012f4feb73fb611b946199e681f4cfc34db8eaZ@0ad0950d9fb9588e59690b74f1237ecdf1d775cd60be2eca57af5a4b0471c91brW rX rY ZZ49f3b19b28a9ea9f43e8c71204c00d4a490ee7f61387b6719db765e948123b45b61633ef059ba22cd62437c8ba)r r r; r5 r rZ )r& r^ r_ � psk_id_hex�psk_id�psk_hexr5 r` r+ ra rb rc r rd re rf rg rh r; ri rj r r* rk rl r, r, r- �test_x25519_mode_1� s> � z HPKE_Tests.test_x25519_mode_1c C s� d}t �t�|��}d}t �t�|��}d}t�|�}d}t�|�}d} t�| �} d} t�| �} d} t�| �}d}t�|�}d }t�|�}tjj}tj||�� ||| d �}|� || �}| � ||� |� ||�}| � ||� d S ) NZ@fdea67cf831f1ca98d8e27b1f6abeb5b7745e9d35348b80fa407ff6958f9137eZ@dc4a146313cce60a278a5323d321f051c5707e9c45ba21a3479fecdf76fc69ddrV ZZ5fd92cc9d46dbf8943e72a07e42f363ed5f721212cd90bcfd072bfd9f44e06b80fd17824947496e21b680c141bZ@23fb952571a14a25e3d678140cd0e5eb47a0961bb18afcf85896e5453c312e76rW rX rY ZZd3736bb256c19bfa93d79e8f80b7971262cb7c887e35c26370cfed62254369a1b52e3d505b79dd699f002bc8ed)r rF r r; r � r r[ r\ r] r r1 r7 r r r% r! )r& r^ r_ �keyS_hex�keySr` r+ ra rb rc r rd re rf rg rh r; ri rj r r* rk rl r, r, r- �test_x25519_mode_2� s: � z HPKE_Tests.test_x25519_mode_2c C s d}t �t�|��}d}t �t�|��}d}t�|�}d}t�|�}d} t�| �} d} t�| �} d} t�| �}d}t�|�}d }t�|�}d }t�|�}d }t�|�}tjj}tj||�� |||f||d �}|� | |�}| � || � |� ||�}| � || � d S ) NZ@cb29a95649dc5656c2d054c1aa0d3df0493155e9d5da6d7e344ed8b6a64a9423Z@fc1c87d2f3832adb178b431fce2ac77c7ca2fd680f3406c77b5ecdf818b119f4rn ro rV ZZa84c64df1e11d8fd11450039d4fe64ff0c8a99fca0bd72c2d4c3e0400bc14a40f27e45e141a24001697737533eZ@820818d3c23993492cc5623ab437a48a0a7ca3e9639c140fe1e33811eb844b7crW rX rY ZZ4d19303b848f424fc3c3beca249b2c6de0a34083b8e909b6aa4c3688505c05ffe0c8f57a0a4c5ab9da127435d9)r rF r r5 r; r rt )r& r^ r_ ru rv rp rq rr r5 r` r+ ra rb rc r rd re rf rg rh r; ri rj r r* rk rl r, r, r- �test_x25519_mode_3 sD � z HPKE_Tests.test_x25519_mode_3N)�__name__� __module__� __qualname__r r r' rG r# r. r2 r9 r< rD rI rJ rM rR rT rU rm rs rw rx r, r, r, r- r s0 �(/ ,r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �HPKE_TestVectorsc C s� g | _ z4dd l}tj�|j�}tj�|ddd�}t|d��}t� |�| _ W d � W d S 1 s0w Y W d S t t fyG t d� Y d S w )Nr �ProtocolZ wycheproofzHPKE-test-vectors.json�rzN Warning: skipping extended tests for HPKE (install pycryptodome-test-vectors)) �vectors�pycryptodome_test_vectors�os�path�dirname�__file__�join�open�json�load�FileNotFoundError� ImportError�print)r& r� Zinit_dirZfull_file_name�fr, r, r- �setUpO s &��zHPKE_TestVectors.setUpc C s� t |�}|dkrtjdtj|dd�d�S |dkr$tjdtj|dd�d�S |dkr4tjd tj|dd�d�S |d kr=t�|�S |d krFt�|�S d S ) N� r �big)� byteorder)r �d� r � r r �! )r r Z construct�int� from_bytesr r[ Zimport_x448_private_key)r& Zkey_hex�kem_idZkey_binr, r, r- �import_private_keyZ s$ � � � �z#HPKE_TestVectors.import_private_keyc C s0 | j s| �d� t| j �D ]�\}}|d }|d }|d }|dkr"q tttttd�}|�||f�}|du r6q | j|||d��H | �|d |�� � }d} d |v rW| �|d |�} | �|d |�} t j � |||| | �\} } | � | �� |d � | � | t|d �� W d � n1 s�w Y tdddd� q dS )z+Test HPKE encapsulation using test vectors.�No test vectors availabler� �kdf_idr � �)r� � )r� � )r� � )r r� )r� r� N��idxr� r �skRm�skSmZskEmr � shared_secret�.� T��end�flush)r �skipTest� enumerater r r �get�subTestr� r r Z HPKE_CipherZ_encapr! �hexr r� ) r& r� �vectorr� r� r �supported_combiZhashmodZ receiver_pub� sender_priv� encap_keyr� r r, r, r- �test_hpke_encapj sT � �� �� � ��z HPKE_TestVectors.test_hpke_encapc C sn | j s| �d� t| j �D ]�\}}|d }|d }|d }|dkr"q d}||f|vr+q | j|||d��r | �|d |�}d }d |v rN| �|d |�} | �� }t|d �} d } d |v rft|d �t|d �f} tj|t� |�| || t|d �d�} |d D ]!} t| d �}t| d �}t| d �}| � ||�}| � ||d� q|W d � n1 s�w Y t dddd� q d S )z7Test HPKE encryption and decryption using test vectors.r� r� r� r r� r� r� r� Nr� r rq r5 r; )r r r rF r5 r; Z encryptionsr+ r) �aadzDecryption failedr� r� Tr� ) r r� r� r� r� r r r r r1 r% r! r� )r& r� r� r� r� r r� Z receiver_privZ sender_pubr� r� r5 Z receiver_hpkeZ encryptionZ plaintextZ ciphertextr� Z decryptedr, r, r- �test_hpke_unseal� sT � � � � �"�z!HPKE_TestVectors.test_hpke_unsealN)ry rz r{ r� r� r� r� r, r, r, r- r| M s 3r| c C s* g }|t t�7 }| �d�r|t t�7 }|S )NZ slow_tests)r r r� r| )ZconfigZtestsr, r, r- � get_tests� s r� �__main__c C s t �t� �S r/ )�unittestZ TestSuiter� r, r, r, r- �suite� s r� )Z defaultTest)r� r� r� �binasciir ZCryptodome.Protocolr ZCryptodome.Protocol.HPKEr ZCryptodome.PublicKeyr Z Cryptodome.SelfTest.st_commonr r ZCryptodome.Hashr r r ZTestCaser r| r� ry r� �mainr, r, r, r- �<module> s( ? �
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings