File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/Cryptodome/SelfTest/Cipher/__pycache__/test_EAX.cpython-310.pyc
Back
o ;j�p � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z d dl m Z m Z d dl mZ d dlmZ d d � ZG d d � d e j�ZG d d� de j�ZG dd� de j�ZG dd� de j�ZG dd� de j�Zd dl mZm Z mZmZmZ e� de ej � eej � e j D ] Z e� de e � e e � q�ej D ] Z e� de e � ee � q�ej D ] Z e� de e � ee � q�ej D ] Z e� de e � ee � q�i fdd �Z e!d kr�d d � Z"e j#d d!� dS dS )"� N�� unhexlify)�list_test_cases)� load_test_vectors_wycheproof)�tobytes�bchr)�AES�DES3)�SHAKE128)�strxorc C s t jt| �d��|�S )N)�data)r �newr �read)�tag�length� r �W/opt/saltstack/salt/lib/python3.10/site-packages/Cryptodome/SelfTest/Cipher/test_EAX.py�get_tag_random+ s r c @ s� e Zd Zedd�Zedd�Zedd�Zedd�Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1S )2�EaxTests�key_128� �key_192� nonce_128� �data_128c C �Z t j| jt j| jd�}tdd�}|�|�}t j| jt j| jd�}|�|�}| �||� d S )N��nonce� plaintexti@ ) r r r �MODE_EAX�nonce_96r �encrypt�decrypt� assertEqual��self�cipher�pt�ct�pt2r r r �test_loopback_1286 � zEaxTests.test_loopback_128c C r )Nr r i ) r r r r r r r! r" r# r$ r r r �test_loopback_64? r+ zEaxTests.test_loopback_64c C s� t �| jt j�}|j}t �| jt j�}|j}| �t|�d� | �||� t �| jt j| j�}|� | j �}t j| jt j| jd�}| �||� | j �� d S )Nr r ) r r r r r r# �len�assertNotEqualr r! r )r% r& �nonce1�nonce2r( r r r � test_nonceH s zEaxTests.test_noncec C s | j ttj| jtjdd� d S )NZ test12345678r )� assertRaises� TypeErrorr r r r �r% r r r �test_nonce_must_be_bytesW s �z!EaxTests.test_nonce_must_be_bytesc C sX | j ttj| jtjdd� tdd�D ]}tj| jtjtd�| d�}|�td�� qd S )N� r � � ) r2 � ValueErrorr r r r �ranger r! )r% �xr& r r r �test_nonce_length[ s � �zEaxTests.test_nonce_lengthc C s* t j| jt j| jd�}| �|jt j� d S �Nr )r r r r r r# � block_size�r% r& r r r �test_block_size_128d � z EaxTests.test_block_size_128c C s* t j| jtj| jd�}| �|jt j� d S r= )r r r r r r r# r> r? r r r �test_block_size_64h rA zEaxTests.test_block_size_64c C sj t j| jt j| jd�}| �|j| j� t �| jt j�j}t �| jt j�j}| �t|�d� | �||� d S )Nr r ) r r r r r r# r r- r. )r% r& r/ r0 r r r �test_nonce_attributel s z EaxTests.test_nonce_attributec C sV | � ttj| jtj| jd� | j ttj| jtj| jdd� tj| jtj| jdd� d S )N� )r �unknownF)r Z use_aesni)r2 r3 r r r r r r4 r r r �test_unknown_parametersv s �� �z EaxTests.test_unknown_parametersc C s> dD ]}t j| jt j| jd�}t||�d�}| �|d� qd S )N�r! r" r r6 )r r r r r �getattrr# )r% �funcr& �resultr r r � test_null_encryption_decryption� s �z(EaxTests.test_null_encryption_decryptionc C sd t j| jt j| jd�}|�d� | �t|jd� t j| jt j| jd�}|�d� | �t|jd� d S )Nr r6 ) r r r r r r! r2 r3 r" r? r r r � test_either_encrypt_or_decrypt� s z'EaxTests.test_either_encrypt_or_decryptc C sP t j| jt j| jd�}| �t|jd� t j| jt j| jd�}| �t|jd� d S )Nr ztest1234567890-*) r r r r r r2 r3 r! r" r? r r r �test_data_must_be_bytes� s z EaxTests.test_data_must_be_bytesc C s� | j ttj| jtj| jdd� | j ttj| jtj| jdd� tdd�D ] }tj| jtj| j|d�}|�| j �\}}| � t |�|� q#tj| jtj| jd�}|�| j �\}}| � t |�d� d S )Nr7 )r �mac_len� � r r ) r2 r9 r r r r r r: �encrypt_and_digestr r# r- )r% rN r& �_�macr r r � test_mac_len� s ���zEaxTests.test_mac_lenc C sh ddl m} tj| jtj| jd�}|�| j�\}}||d�}tj| jtj| jd�}| � t |j ||� d S )Nr )�strxor_cr r7 ) �Cryptodome.Util.strxorrU r r r r r rQ r r2 r9 �decrypt_and_verify)r% rU r& r( rS Z invalid_macr r r �test_invalid_mac� s �zEaxTests.test_invalid_macc C sV t j| jt j| jd�}|�� }| �|�� t|�� t j| jt j| jd�}|� |� d S r= ) r r r r r � hexdigestr# �digestr Z hexverify)r% r& Zmac_hexr r r � test_hex_mac� s zEaxTests.test_hex_macc C s8 t dd�}t dd�}tj| jtj| jd�}|�|� |�|�\}}dd� }dD ]6}tj| jtj| jd�}|||�D ]}|�|� q9d} |||�D ] }| |�|�7 } qH| � || � |� |� q'dD ]9}tj| jtj| jd�}|||�D ]}|�|� qrd} |||�D ] }| |� |�7 } q�| � || � | � |� � |� q`d S ) Nzauthenticated data� r r c s � �fdd�t dt��� �D �S )Nc s g | ] }�||� � �qS r r )�.0�i�� chunk_lengthr r r � <listcomp>� s zBEaxTests.test_message_chunks.<locals>.break_up.<locals>.<listcomp>r )r: r- )r r` r r_ r �break_up� s �z.EaxTests.test_message_chunks.<locals>.break_up) r7 rP � rD � � r �( �P r8 r6 ) r r r r r r �updaterQ r" r# �verifyr! rZ ) r% � auth_datar r& Z ciphertextZref_macrb r` �chunkr) �ct2r r r �test_message_chunks� s2 �z EaxTests.test_message_chunksc C s� t | j�}t | j�}t | j�}t | j�}tj| jtj| jd�}|�| j� |�| j�}|� � }tj|tj|d�}d|d d�< d|d d�< |�|� d|d d�< |�|�} d|d d�< |� � } | � || � | � || � | � |j |j � t | j�}t | j�}t | j�}t |�} t |�} ~tj|tj|d�} d|d d�< d|d d�< | �|� d|d d�< | � | �}d| d d�< | � | � | � || j� d S �Nr s ���rc s ���)� bytearrayr r r r r r rh r! rZ r# r r" ri )r% Zkey_baZnonce_baZ header_baZdata_ba�cipher1r( r �cipher2�ct_test�tag_testZct_baZtag_ba�cipher3�pt_testr r r �test_bytearray� sV � � � zEaxTests.test_bytearrayc C s� t t| j��}t t| j��}t t| j��}t t| j��}tj| jtj| jd�}|�| j� |� | j�}|� � }tj|tj|d�}d|d d�< d|d d�< |�|� d|d d�< |� |�} d|d d�< |� � } | � || � | � || � | � |j |j � t t| j��}t t| j��}t t| j��}t t|��} t t|��} ~tj|tj|d�} d|d d�< d|d d�< | �|� d|d d�< | � | �}d| d d�< | �| � | � || j� d S rn )� memoryviewro r r r r r r rh r! rZ r# r r" ri )r% Zkey_mvZnonce_mvZ header_mvZdata_mvrp r( r rq rr rs Zct_mvZtag_mvrt ru r r r �test_memoryview sV � � � zEaxTests.test_memoryviewc C s: d}t j| jt j| jd�}|�|�}|�� }td�}t j| jt j| jd�}|j||d�}| �||� | �|d � t j| jt j| jd�}|j ||d�}| �||� | �|d � t j| jt j| jd�}|j ||d�\}}| �||� | �|d � | �||� t j| jt j| jd�}|j |||d�}| �||� | �|d � d S �Ns� 55555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555r r8 ��output) r r r r r r! rZ ro r# r" rQ rW )r% r' r& r( r r{ �resZtag_outr r r �test_output_paramL s, zEaxTests.test_output_paramc C s� d}t j| jt j| jd�}|�|�}ttd��}t j| jt j| jd�}|j||d� | �||� t j| jt j| jd�}|j ||d� | �||� d S ry ) r r r r r r! rw ro r# r" )r% r' r& r( r{ r r r � test_output_param_memoryviewi s z%EaxTests.test_output_param_memoryviewc C s� d}d| }t j| jt j| jd�}|�|�}t j| jt j| jd�}| jt|j|d| d� t j| jt j| jd�}| jt|j|d| d� t |d �}t j| jt j| jd�}| jt |j||d� t j| jt j| jd�}| jt |j||d� d S )Nr � 5r � 0rz r7 ) r r r r r r! r2 r3 r" ro r9 )r% ZLEN_PTr' r& r( Zshorter_outputr r r �test_output_param_negx s z EaxTests.test_output_param_negN) �__name__� __module__� __qualname__r r r r r r* r, r1 r5 r< r@ rB rC rF rK rL rM rT rX r[ rm rv rx r} r~ r� r r r r r / s4 )33 r c @ s� e Zd Zedd�Zedd�Zedd�Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zd S ) � EaxFSMTestsr r r r r c C sX t j| jt j| jd�}|�| j�}|�� }t j| jt j| jd�}|�|� |� |� d S r= ) r r r r r r! r rZ r" ri �r% r& r( rS r r r �-test_valid_init_encrypt_decrypt_digest_verify� s � � z9EaxFSMTests.test_valid_init_encrypt_decrypt_digest_verifyc C sZ t j| jt j| jd�}|�| j� |�� }t j| jt j| jd�}|�| j� |�|� d S r= ) r r r r r rh r rZ ri �r% r& rS r r r �$test_valid_init_update_digest_verify� s � � z0EaxFSMTests.test_valid_init_update_digest_verifyc C sp t j| jt j| jd�}|�| j� |�| j�}|�� }t j| jt j| jd�}|�| j� |� |� |� |� d S r= ) r r r r r rh r r! rZ r" ri r� r r r �test_valid_full_path� s � � z EaxFSMTests.test_valid_full_pathc C s" t j| jt j| jd�}|�� d S r= )r r r r r rZ r? r r r �test_valid_init_digest� s z"EaxFSMTests.test_valid_init_digestc C sB t j| jt j| jd�}|�� }t j| jt j| jd�}|�|� d S r= )r r r r r rZ ri r� r r r �test_valid_init_verify� s z"EaxFSMTests.test_valid_init_verifyc C s� dD ]G}d d| j | j d fD ]:}|d u rd }nt|�}tj| jtj| jd�}|d ur/|�|� t||�}|| j � || j � || j � || j � qqd S )NrG s 333� 3r ) r r- r r r r r rh rH )r% � method_namerj Z assoc_lenr& �methodr r r �&test_valid_multiple_encrypt_or_decrypt� s& � � ��z2EaxFSMTests.test_valid_multiple_encrypt_or_decryptc C s� t j| jt j| jd�}|�| j� |�� }td�D ] }| � ||�� � qt j| jt j| jd�}|�| j� td�D ]}|� |� q9d S )Nr � � ) r r r r r rh r rZ r: r# ri )r% r& Z first_macr; r r r �$test_valid_multiple_digest_or_verify� s �z0EaxFSMTests.test_valid_multiple_digest_or_verifyc C sr t j| jt j| jd�}|�| j� |�| j�\}}t j| jt j| jd�}|�| j� |�||�}| � | j|� d S r= ) r r r r r rh r rQ rW r# )r% r& r( rS r' r r r �0test_valid_encrypt_and_digest_decrypt_and_verify� s z<EaxFSMTests.test_valid_encrypt_and_digest_decrypt_and_verifyc C sh dD ]/\}}dD ](}t j| jt j| jd�}|r |�| j� t||�| j� | �t t||�| j� qqd S )N)rG )r" r! )TFr ) r r r r r rh r rH r2 r3 )r% Z method1_nameZ method2_nameZassoc_data_presentr& r r r �#test_invalid_mixing_encrypt_decrypt� s � ���z/EaxFSMTests.test_invalid_mixing_encrypt_decryptc C sp dD ]3}t j| jt j| jd�}|�| j� |�� | �t t ||�| j� t j| jt j| jd�}|� | j� qd S )N)r! rh r ) r r r r r r! r rZ r2 r3 rH rQ )r% r� r& r r r �+test_invalid_encrypt_or_update_after_digest s ��z7EaxFSMTests.test_invalid_encrypt_or_update_after_digestc C s� t j| jt j| jd�}|�| j�}|�� }dD ]>}t j| jt j| jd�}|�|� |� |� | � t t ||�| j� t j| jt j| jd�}|� ||� | � t t ||�| j� qd S )Nr )r" rh )r r r r r r! r rZ r" ri r2 r3 rH rW )r% r& r( rS r� r r r �+test_invalid_decrypt_or_update_after_verify s � ��z7EaxFSMTests.test_invalid_decrypt_or_update_after_verifyN)r� r� r� r r r r r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s. e Zd ZdZg d�Zdd� eD �Zdd� ZdS )�TestVectorsPaperzgClass exercising the EAX test vectors found in http://www.cs.ucdavis.edu/~rogaway/papers/eax.pdf) )Z6bfb914fd07eae6b� r� Z e037830e8389f27b025a2d6527e79d01Z 233952dee4d5ed5f9b9c6d6ff80ff478Z 62EC67F9C3A4A407FCB2A8C49031A8B3)Zfa3bfd4806eb53faZf7fbZ19ddZ 5c4c9331049d0bdab0277408f67967e5Z 91945d3f4dcbee0bf45ef52255f095a4Z BECAF043B0A23D843194BA972C66DEBD)Z234a3463c1264ac6Z 1a47cb4933Z d851d5bae0Z 3a59f238a23e39199dc9266626c40f80Z 01f74ad64077f2e704c0f60ada3dd523Z 70C3DB4F0D26368400A10ED05D2BFF5E)Z33cce2eabff5a79dZ 481c9e39b1Z 632a9d131aZ d4c168a4225d8e1ff755939974a7bedeZ d07cf6cbb7f313bdde66b727afd3c5e8Z 8408DFFF3C1A2B1292DC199E46B7D617)Zaeb96eaebe2970e9Z 40d0c07da5e4Z 071dfe16c675Z cb0677e536f73afe6a14b74ee49844ddZ 35b6d0580005bbc12b0587124557d2c2Z FDB6B06676EEDC5C61D74276E1F8E816)Zd4482d1ca78dce0fZ4de3b35c3fc039245bd1fb7dZ835bb4f15d743e350e728414Z abb8644fd6ccb86947c5e10590210a4fZ bd8e6e11475e60b268784c38c62feb22Z 6EAC5C93072D8E8513F750935E46DA1B)Z65d2017990d62528Z"8b0a79306c9ce7ed99dae4f87f8dd61636Z"02083e3979da014812f59f11d52630da30Z 137327d10649b0aa6e1c181db617d7f2Z 7c77d6e813bed5ac98baa417477a2e7dZ 1A8C98DCD73D38393B2BF1569DEEFC19)Z54b9f04e6a09189aZ$1bda122bce8a8dbaf1877d962b8592dd2d56Z$2ec47b2c4954a489afc7ba4897edcdae8cc3Z 3b60450599bd02c96382902aef7f832aZ 5fff20cafab119ca2fc73549e20f5b0dZ DDE59B97D722156D4D9AFF2BC7559826)Z899a175897561d7eZ$6cf36720872b8513f6eab1a8a44438d5ef11Z$0de18fd0fdd91e7af19f1d8ee8733938b1e8Z e7f6d2231618102fdb7fe55ff1991700Z a4a4782bcffd3ec5e7ef6d8c34a56123Z B781FCF2F75FA5A8DE97A9CA48E522EC)Z126735fcc320d25aZ*ca40d7446e545ffaed3bd12a740a659ffbbb3ceab7Z*cb8920f87a6c75cff39627b56e3ed197c552d295a7Z cfc46afc253b4652b1af3795b124ab6eZ 8395fcf1e95bebd697bd010bc766aac3Z 22E7ADD93CFC6393C57EC0B3C17D6B44c C s g | ] }d d� |D ��qS )c S s g | ]}t |��qS r r )r] r; r r r ra v s z&TestVectorsPaper.<listcomp>.<listcomp>r )r] �tvr r r ra v s zTestVectorsPaper.<listcomp>c C s� | j D ]I\}}}}}}tj|tj|t|�d�}|�|� |�|�\}} | �||� | �|| � tj|tj|t|�d�}|�|� |�||�} | �|| � qd S )N�rN ) � test_vectorsr r r r- rh rQ r# rW ) r% Z assoc_datar' r( rS �keyr r& rl Zmac2r) r r r �runTestx s �zTestVectorsPaper.runTestN)r� r� r� �__doc__Ztest_vectors_hexr� r� r r r r r� ! s Q r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestVectorsWycheproofc C s t j�| � || _d| _d S )N�None)�unittest�TestCase�__init__�_wycheproof_warnings�_id)r% �wycheproof_warningsr r r r� � s z TestVectorsWycheproof.__init__c C s" dd� }t dddd|id�| _d S )Nc S s | d d S )NZtagSize� r )�groupr r r � filter_tag� s z/TestVectorsWycheproof.setUp.<locals>.filter_tag)ZCipherZ wycheproofzaes_eax_test.jsonzWycheproof EAX�tag_size)Z group_tag)r r� )r% r� r r r �setUp� s �zTestVectorsWycheproof.setUpc C s | j S �N)r� r4 r r r �shortDescription� s z&TestVectorsWycheproof.shortDescriptionc C s6 |j r| jrdd l}|�d| j|jf � d S d S d S )Nr zWycheproof warning: %s (%s))�warningr� �warnings�warnr� Zcomment)r% r� r� r r r r� � s �zTestVectorsWycheproof.warnc C s� dt |j� | _ztj|jtj|j|jd�}W n" t y9 } zt |j�dkr,dt |�v s.J �W Y d }~d S d }~ww |� |j � |� |j�\}}|jr`| �||j� | �||j� | �|� d S d S )Nz Wycheproof Encrypt EAX Test #r� r �Nonce cannot be empty)�str�idr� r r r� r �ivr� r9 r- rh �aadrQ �msg�validr# r( r r� )r% r� r& �er( r r r r � test_encrypt� s �� �z"TestVectorsWycheproof.test_encryptc C s� dt |j� | _ztj|jtj|j|jd�}W n" t y9 } zt |j�dkr,dt |�v s.J �W Y d }~d S d }~ww |� |j � z |� |j|j�}W n t yY |jrVJ �Y d S w |js_J �| �||j� | �|� d S )Nz Wycheproof Decrypt EAX Test #r� r r� )r� r� r� r r r� r r� r� r9 r- rh r� rW r( r r� r# r� r� )r% r� r& r� r' r r r � test_decrypt� s" �� � z"TestVectorsWycheproof.test_decryptc C s� dt |j� | _t|j�dkst|j�dk rd S tj|jtj |j|j d�}|� |j � t |jdt|j�d d �}| �t|j||j� d S )Nz%Wycheproof Corrupt Decrypt EAX Test #r r7 r� � � )r� r� r� r- r� r( r r r� r r� rh r� r r2 r9 rW r )r% r� r& Z ct_corruptr r r �test_corrupt_decrypt� s z*TestVectorsWycheproof.test_corrupt_decryptc C s. | j D ]}| �|� | �|� | �|� qd S r� )r� r� r� r� )r% r� r r r r� � s �z TestVectorsWycheproof.runTestN) r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zedd� �ZdS )�TestOtherCiphersc C s" ||fdd�}t | d| |� d S )Nc S sZ |j td|�|jdd�}|�d�\}}|j td|�|jdd�}|�||�}| �d|� d S )Nr& s noncer s plaintext)r r r rQ rW r# )r% �factory�key_sizer& r( rS r) r r r � test_template� s � � z3TestOtherCiphers.create_test.<locals>.test_templateZtest_)�setattr)�cls�namer� r� r� r r r � create_test� s z TestOtherCiphers.create_testN)r� r� r� � classmethodr� r r r r r� � s r� )�DESr �ARC2�CAST�BlowfishZDES_ZDES3_ZARC2_ZCAST_Z Blowfish_c C sP | � d�}g }|tt�7 }|tt�7 }|t� g7 }|t|�g7 }|tt�7 }|S )Nr� )�getr r r� r� r� r� )Zconfigr� Ztestsr r r � get_tests� s r� �__main__c C s t �t� �S r� )r� Z TestSuiter� r r r r �<lambda> s r� �suite)Z defaultTest)$r� �binasciir Z Cryptodome.SelfTest.st_commonr ZCryptodome.SelfTest.loaderr ZCryptodome.Util.py3compatr r ZCryptodome.Cipherr r ZCryptodome.Hashr rV r r r� r r� r� r� r� r� r� r� r� r� r� r� �ksr� r� r� �mainr r r r �<module> s@ _ gM �
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings