File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/Cryptodome/SelfTest/Cipher/__pycache__/test_ChaCha20_Poly1305.cpython-310.pyc
Back
o ;jx � @ s d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d dl mZ d d � ZG d d � d e j�ZG d d� de j�ZG dd� de j�Zdd� ZG dd� de j�ZG dd� de j�ZG dd� de j�Zi fdd�Zedkr�d d � Ze jd d � dS dS ) � N)� unhexlify)�list_test_cases�� load_test_vectors_wycheproof)�tobytes)�ChaCha20_Poly1305)�SHAKE128)�strxorc C s t jt| �d��|�S )N)�data)r �newr �read)�tag�length� r �e/opt/saltstack/salt/lib/python3.10/site-packages/Cryptodome/SelfTest/Cipher/test_ChaCha20_Poly1305.py�get_tag_random+ s r c @ s� e Zd Zedd�Zedd�Zedd�Zdd� 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 )(�ChaCha20Poly1305Tests�key_256� �nonce_96� �data_128� c C sR t j| j| jd�}tdd�}|�|�}t j| j| jd�}|�|�}| �||� d S )N��key�nonce� plaintexti@ )r r r r r �encrypt�decrypt� assertEqual)�self�cipher�pt�ct�pt2r r r � test_loopback5 s � � z#ChaCha20Poly1305Tests.test_loopbackc C s� t j| jdd�}| �t|j�d� t j| jdd�}| �t|j�d� t j| jd�}|j}t j| jd�}|j}| �t|�d� | �||� t j| j| jd�}|�| j �}t j| j| jd�}| �||�| j �� d S )Ns HHHHHHHHr � s HHHHHHHHHHHHr �r ) r r r r �lenr �assertNotEqualr r r )r r! �nonce1�nonce2r# r r r � test_nonce@ s, �� � �z ChaCha20Poly1305Tests.test_noncec C � | j ttj| jdd� d S )NZ test12345678r �� assertRaises� TypeErrorr r r �r r r r �test_nonce_must_be_bytesY � �z.ChaCha20Poly1305Tests.test_nonce_must_be_bytesc C s0 | j ttj| jdd� | j ttj| jdd� d S )Ns 0000000r � )r/ � ValueErrorr r r r1 r r r �test_nonce_length_ s � �z'ChaCha20Poly1305Tests.test_nonce_lengthc C s& t j| j| jd�}| �t|d�� d S )Nr � block_size)r r r r Z assertFalse�hasattr�r r! r r r �test_block_sizej s �z%ChaCha20Poly1305Tests.test_block_sizec C sb t j| j| jd�}| �|j| j� t j| jd�j}t j| jd�j}| �t|�d� | �||� d S )Nr r' r )r r r r r r r( r) )r r! r* r+ r r r �test_nonce_attributep s �z*ChaCha20Poly1305Tests.test_nonce_attributec C r- )N� )r �paramr. r1 r r r �test_unknown_parameters{ r3 z-ChaCha20Poly1305Tests.test_unknown_parametersc C s: dD ]}t j| j| jd�}t||�d�}| �|d� qd S )N�r r r r4 )r r r r �getattrr )r �funcr! �resultr r r � test_null_encryption_decryption� s ��z5ChaCha20Poly1305Tests.test_null_encryption_decryptionc C s\ t j| j| jd�}|�d� | �t|jd� t j| j| jd�}|�d� | �t|jd� d S )Nr r4 )r r r r r r/ r0 r r9 r r r � test_either_encrypt_or_decrypt� s � � z4ChaCha20Poly1305Tests.test_either_encrypt_or_decryptc C sH t j| j| jd�}| �t|jd� t j| j| jd�}| �t|jd� d S )Nr ztest1234567890-*)r r r r r/ r0 r r r9 r r r �test_data_must_be_bytes� s ��z-ChaCha20Poly1305Tests.test_data_must_be_bytesc C s6 t j| j| jd�}|�| j�\}}| �t|�d� d S )Nr r )r r r r �encrypt_and_digestr r r( )r r! �_�macr r r � test_mac_len� s �z"ChaCha20Poly1305Tests.test_mac_lenc C s` ddl m} tj| j| jd�}|�| j�\}}||d�}tj| j| jd�}| �t |j ||� d S )Nr )�strxor_cr � ) �Cryptodome.Util.strxorrJ r r r r rF r r/ r5 �decrypt_and_verify)r rJ r! r# rH Z invalid_macr r r �test_invalid_mac� s � � �z&ChaCha20Poly1305Tests.test_invalid_macc C sN t j| j| jd�}|�� }| �|�� t|�� t j| j| jd�}|�|� d S �Nr ) r r r r � hexdigestr �digestr Z hexverify)r r! Zmac_hexr r r � test_hex_mac� s ��z"ChaCha20Poly1305Tests.test_hex_macc C s, t dd�}t dd�}tj| j| jd�}|�|� |�|�\}}dd� }dD ]4}tj| j| jd�}|||�D ]}|�|� q5d} |||�D ] }| |�|�7 } qD| �|| � |� |� q%dD ]7}tj| j| jd�}|||�D ]}|�|� qld} |||�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 zOChaCha20Poly1305Tests.test_message_chunks.<locals>.break_up.<locals>.<listcomp>r )�ranger( )r rW r rV r �break_up� s �z;ChaCha20Poly1305Tests.test_message_chunks.<locals>.break_up) rK � � � � � r �( �P � r4 ) r r r r r �updaterF r r �verifyr rQ ) r � auth_datar r! Z ciphertextZref_macrZ rW �chunkr$ �ct2r r r �test_message_chunks� s> � � � �z)ChaCha20Poly1305Tests.test_message_chunksc C s� t | j�}t | j�}t | j�}t | j�}tj| j| jd�}|�| j� |�| j�}|�� }tj| j| jd�}d|d d�< d|d d�< |�|� d|d d�< |�|�} d|d d�< |�� } | � || � | � || � | � |j |j � t | j�}t | j�}t | j�}t |�} t |�} ~tj| j| jd�} d|d d�< d|d d�< | �|� d|d d�< | � | �}d| d d�< | � | � | � || j� d S �Nr s ���r\ s ���) � bytearrayr r r r r rc r rQ r r r rd )r Zkey_baZnonce_baZ header_baZdata_ba�cipher1r# r �cipher2�ct_test�tag_testZct_baZtag_ba�cipher3�pt_testr r r �test_bytearray� sP � � � z$ChaCha20Poly1305Tests.test_bytearrayc C s� t t| j��}t t| j��}t t| j��}t t| j��}tj| j| jd�}|�| j� |�| j�}|� � }tj| j| jd�}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| j| jd�} d|d d�< d|d d�< | �|� d|d d�< | � | �}d| d d�< | � | � | � || j� d S ri )� memoryviewrj r r r r r rc r rQ r r r rd )r Zkey_mvZnonce_mvZ header_mvZdata_mvrk r# r rl rm rn Zct_mvZtag_mvro rp r r r �test_memoryview sP � � � z%ChaCha20Poly1305Tests.test_memoryviewN)�__name__� __module__� __qualname__r r r r r% r, r2 r6 r: r; r> rC rD rE rI rN rR rh rq rs r r r r r / s( , 0r c @ s e Zd Zdd� Zdd� ZdS )�XChaCha20Poly1305Testsc C s2 t jddd�}| �t|j�d� | �|jd� d S )Ns YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYs HHHHHHHHHHHHHHHHHHHHHHHHr � )r r r r( r r9 r r r r, H s �z!XChaCha20Poly1305Tests.test_noncec C s� d}t |�dd��dd��}t d�}t d�}t d�}d}t |�dd��dd��}t d �}tj||d �}|�|� |�|�\}} | �||� | �|| � tj||d �}|�|� |�||� d S ) Ns( 4c616469657320616e642047656e746c656d656e206f662074686520636c6173 73206f66202739393a204966204920636f756c64206f6666657220796f75206f 6e6c79206f6e652074697020666f7220746865206675747572652c2073756e73 637265656e20776f756c642062652069742e� r4 � s 50515253c0c1c2c3c4c5c6c7s@ 808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fs0 404142434445464748494a4b4c4d4e4f5051525354555657s( bd6d179d3e83d43b9576579493c0e939572a1700252bfaccbed2902c21396cbb 731c7f1b0b4aa6440bf3a82f4eda7e39ae64c6708c54c216cb96b72e1213b452 2f8c9ba40db5d945b11b69b982c1bb9e3f3fac2bc369488f76b2383565d3fff9 21f9664c97637da9768812f615c68b13b52es c0875924c1c7987947deafd8780acf49r )r �replacer r rc rF r rM ) r r" �aadr �ivr# r r! rm rn r r r � test_encryptO s z#XChaCha20Poly1305Tests.test_encryptN)rt ru rv r, r~ r r r r rw F s rw c @ s� e Zd Zedd�Zedd�Zedd�Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd S ) �ChaCha20Poly1305FSMTestsr r r r r r c C sP t j| j| jd�}|�| j�}|�� }t j| j| jd�}|�|� |�|� d S rO ) r r r r r r rQ r rd �r r! r# rH r r r �-test_valid_init_encrypt_decrypt_digest_verifyy s � � zFChaCha20Poly1305FSMTests.test_valid_init_encrypt_decrypt_digest_verifyc C sR t j| j| jd�}|�| j� |�� }t j| j| jd�}|�| j� |�|� d S rO )r r r r rc r rQ rd �r r! rH r r r �$test_valid_init_update_digest_verify� s � � z=ChaCha20Poly1305FSMTests.test_valid_init_update_digest_verifyc C sh t j| j| jd�}|�| j� |�| j�}|�� }t j| j| jd�}|�| j� |�|� |� |� d S rO ) r r r r rc r r rQ r rd r� r r r �test_valid_full_path� s � � z-ChaCha20Poly1305FSMTests.test_valid_full_pathc C s t j| j| jd�}|�� d S rO )r r r r rQ r9 r r r �test_valid_init_digest� s � z/ChaCha20Poly1305FSMTests.test_valid_init_digestc C s: t j| j| jd�}|�� }t j| j| jd�}|�|� d S rO )r r r r rQ rd r� r r r �test_valid_init_verify� s ��z/ChaCha20Poly1305FSMTests.test_valid_init_verifyc C s~ dD ]:}d d| j | j d fD ]-}tj| j| jd�}|d ur"|�|� t||�}|| j � || j � || j � || j � qqd S )Nr? s 333� 3r )r r r r r rc r@ )r � method_namere r! �methodr r r �&test_valid_multiple_encrypt_or_decrypt� s �� ��z?ChaCha20Poly1305FSMTests.test_valid_multiple_encrypt_or_decryptc C s~ t j| j| jd�}|�| j� |�� }td�D ] }| �||�� � qt j| j| jd�}|�| j� td�D ]}|� |� q5d S )Nr � � ) r r r r rc r rQ rY r rd )r r! Z first_mac�xr r r �$test_valid_multiple_digest_or_verify� s � � �z=ChaCha20Poly1305FSMTests.test_valid_multiple_digest_or_verifyc C sj t j| j| jd�}|�| j� |�| j�\}}t j| j| jd�}|�| j� |�||�}| �| j|� d S rO ) r r r r rc r rF rM r )r r! r# rH r" r r r �0test_valid_encrypt_and_digest_decrypt_and_verify� s � � zIChaCha20Poly1305FSMTests.test_valid_encrypt_and_digest_decrypt_and_verifyc C sd dD ]-\}}dD ]&}t j| j| jd�}|r|�| j� t||�| j� | �tt||�| j� qqd S )N)r? )r r )TFr ) r r r r rc r r@ r/ r0 )r Z method1_nameZ method2_nameZassoc_data_presentr! r r r �#test_invalid_mixing_encrypt_decrypt� s � ���z<ChaCha20Poly1305FSMTests.test_invalid_mixing_encrypt_decryptc C sh dD ]/}t j| j| jd�}|�| j� |�� | �tt ||�| j� t j| j| jd�}|� | j� qd S )N)r rc r ) r r r r r r rQ r/ r0 r@ rF )r r� r! r r r �+test_invalid_encrypt_or_update_after_digest� s � ���zDChaCha20Poly1305FSMTests.test_invalid_encrypt_or_update_after_digestc C s� t j| j| jd�}|�| j�}|�� }dD ]X}t j| j| jd�}|�|� |�|� | � t t ||�| j� t j| j| jd�}|�|� |�|� | � t t ||�| j� t j| j| jd�}|� ||� | � t t ||�| j� qd S )Nr )r rc ) r r r r r r rQ r rd r/ r0 r@ rM )r r! r# rH r� r r r �+test_invalid_decrypt_or_update_after_verify� s< � � �� �� ��zDChaCha20Poly1305FSMTests.test_invalid_decrypt_or_update_after_verifyN)rt ru rv r r r r r� r� r� r� r� r� r� r� r� r� r� r r r r r s s r c C s t | �dd��dd��S )N� � �:�r r{ )r� r r r �compact s r� c @ s. e Zd ZdZddgZdd� eD �Zdd� ZdS ) �TestVectorsRFCzTest cases from RFC7539)z#50 51 52 53 c0 c1 c2 c3 c4 c5 c6 c7aN 4c 61 64 69 65 73 20 61 6e 64 20 47 65 6e 74 6c65 6d 65 6e 20 6f 66 20 74 68 65 20 63 6c 61 7373 20 6f 66 20 27 39 39 3a 20 49 66 20 49 20 636f 75 6c 64 20 6f 66 66 65 72 20 79 6f 75 20 6f6e 6c 79 20 6f 6e 65 20 74 69 70 20 66 6f 72 2074 68 65 20 66 75 74 75 72 65 2c 20 73 75 6e 7363 72 65 65 6e 20 77 6f 75 6c 64 20 62 65 20 6974 2eaN d3 1a 8d 34 64 8e 60 db 7b 86 af bc 53 ef 7e c2a4 ad ed 51 29 6e 08 fe a9 e2 b5 a7 36 ee 62 d63d be a4 5e 8c a9 67 12 82 fa fb 69 da 92 72 8b1a 71 de 0a 9e 06 0b 29 05 d6 a5 b6 7e cd 3b 3692 dd bd 7f 2d 77 8b 8c 98 03 ae e3 28 09 1b 58fa b3 24 e4 fa d6 75 94 55 85 80 8b 48 31 d7 bc3f f4 de f0 8e 4b 7a 9d e5 76 d2 65 86 ce c6 4b61 16z/1a:e1:0b:59:4f:09:e2:6a:7e:90:2e:cb:d0:60:06:91z^80 81 82 83 84 85 86 87 88 89 8a 8b 8c 8d 8e 8f90 91 92 93 94 95 96 97 98 99 9a 9b 9c 9d 9e 9fz"07 00 00 0040 41 42 43 44 45 46 47)z#f3 33 88 86 00 00 00 00 00 00 4e 91a 49 6e 74 65 72 6e 65 74 2d 44 72 61 66 74 73 2061 72 65 20 64 72 61 66 74 20 64 6f 63 75 6d 656e 74 73 20 76 61 6c 69 64 20 66 6f 72 20 61 206d 61 78 69 6d 75 6d 20 6f 66 20 73 69 78 20 6d6f 6e 74 68 73 20 61 6e 64 20 6d 61 79 20 62 6520 75 70 64 61 74 65 64 2c 20 72 65 70 6c 61 6365 64 2c 20 6f 72 20 6f 62 73 6f 6c 65 74 65 6420 62 79 20 6f 74 68 65 72 20 64 6f 63 75 6d 656e 74 73 20 61 74 20 61 6e 79 20 74 69 6d 65 2e20 49 74 20 69 73 20 69 6e 61 70 70 72 6f 70 7269 61 74 65 20 74 6f 20 75 73 65 20 49 6e 74 6572 6e 65 74 2d 44 72 61 66 74 73 20 61 73 20 7265 66 65 72 65 6e 63 65 20 6d 61 74 65 72 69 616c 20 6f 72 20 74 6f 20 63 69 74 65 20 74 68 656d 20 6f 74 68 65 72 20 74 68 61 6e 20 61 73 202f e2 80 9c 77 6f 72 6b 20 69 6e 20 70 72 6f 6772 65 73 73 2e 2f e2 80 9da 64 a0 86 15 75 86 1a f4 60 f0 62 c7 9b e6 43 bd5e 80 5c fd 34 5c f3 89 f1 08 67 0a c7 6c 8c b24c 6c fc 18 75 5d 43 ee a0 9e e9 4e 38 2d 26 b0bd b7 b7 3c 32 1b 01 00 d4 f0 3b 7f 35 58 94 cf33 2f 83 0e 71 0b 97 ce 98 c8 a8 4a bd 0b 94 8114 ad 17 6e 00 8d 33 bd 60 f9 82 b1 ff 37 c8 5597 97 a0 6e f4 f0 ef 61 c1 86 32 4e 2b 35 06 3836 06 90 7b 6a 7c 02 b0 f9 f6 15 7b 53 c8 67 e4b9 16 6c 76 7b 80 4d 46 a5 9b 52 16 cd e7 a4 e990 40 c5 a4 04 33 22 5e e2 82 a1 b0 a0 6c 52 3eaf 45 34 d7 f8 3f a1 15 5b 00 47 71 8c bc 54 6a0d 07 2b 04 b3 56 4e ea 1b 42 22 73 f5 48 27 1a0b b2 31 60 53 fa 76 99 19 55 eb d6 31 59 43 4ece bb 4e 46 6d ae 5a 10 73 a6 72 76 27 09 7a 1049 e6 17 d9 1d 36 10 94 fa 68 f0 ff 77 98 71 3030 5b ea ba 2e da 04 df 99 7b 71 4d 6c 6f 2c 29a6 ad 5c b4 02 2b 02 70 9bz/ee ad 9d 67 89 0c bb 22 39 23 36 fe a1 85 1f 38z^1c 92 40 a5 eb 55 d3 8a f3 33 88 86 04 f6 b5 f047 39 17 c1 40 2b 80 09 9d ca 5c bc 20 70 75 c0z#00 00 00 00 01 02 03 04 05 06 07 08c C s g | ] }d d� |D ��qS )c S s$ g | ]}t |�d d��dd���qS )r� r� r� r� )rT r� r r r rX ` s $ z$TestVectorsRFC.<listcomp>.<listcomp>r )rT �tvr r r rX ` s zTestVectorsRFC.<listcomp>c C s� | j D ]?\}}}}}}tj||d�}|�|� |�|�\}} | �||� | �|| � tj||d�}|�|� |�||�} | �|| � qd S rO )� test_vectorsr r rc rF r rM ) r Z assoc_datar" r# rH r r r! rg Zmac2r$ r r r �runTestb s �zTestVectorsRFC.runTestN)rt ru rv �__doc__Ztest_vectors_hexr� r� r r r r r� s �A r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� 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� t s z TestVectorsWycheproof.__init__c C s. dd� }dd� }t d|dd|id|id �}|S ) Nc S s | d d S )NZtagSizer& r )�groupr r r � filter_tag{ s z4TestVectorsWycheproof.load_tests.<locals>.filter_tagc S s | d S )N� algorithmr )�rootr r r � filter_algo~ s z5TestVectorsWycheproof.load_tests.<locals>.filter_algo)ZCipherZ wycheproofz Wycheproof ChaCha20-Poly1305�algoZtag_size)Zroot_tagZ group_tagr )r �filenamer� r� rB r r r � load_testsy s �z TestVectorsWycheproof.load_testsc C s. g | _ | j �| �d�� | j �| �d�� d S )Nzchacha20_poly1305_test.jsonz xchacha20_poly1305_test.json)r� �extendr� r1 r r r �setUp� s zTestVectorsWycheproof.setUpc C s | j S �N)r� r1 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� d|j |jf | _z tj|j|jd�}W n" ty6 } zt|j�dvr)dt |�v s+J �W Y d }~d S d }~ww |� |j � |� |j �\}}|jr]| �||j� | �||j� | �|� d S d S )Nz Wycheproof Encrypt %s Test #%sr �r& r � Nonce must be)r� �idr� r r r r} r5 r( �strrc r| rF �msg�validr r# r r� )r r� r! �er# r r r r r~ � s �� �z"TestVectorsWycheproof.test_encryptc C s� d|j |jf | _z tj|j|jd�}W n" ty6 } zt|j�dvr)dt |�v s+J �W Y d }~d S d }~ww |� |j � z |� |j |j�}W n tyV |jrSJ �Y d S w |js\J �| �||j� | �|� d S )Nz Wycheproof Decrypt %s Test #%sr r� r� )r� r� r� r r r r} r5 r( r� rc r| rM 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|j|jd�}|� |j � t |jdt|j�d d �}| � t |j||j� d S )Nz3Wycheproof Corrupt Decrypt ChaCha20-Poly1305 Test #r rK r � � )r� r� r� r( r} r# r r r rc r| r r/ r5 rM 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) rt ru rv r� r� r� r� r� r~ r� r� r� r r r r r� r s r� c @ s e Zd Zdd� ZdS )� TestOutputc C s� d}d}t j||d�}d}|�|�}td�}t j||d�}|j||d�}| �||� | �|d � t j||d�}|j||d�}| �||� | �|d � ttd��}t j||d�}|j||d� | �||� t j||d�}|j||d� | �||� t j||d�}| jt|j|dd� t j||d�}| jt|j|dd� td�}t j||d�}| jt |j||d� t j||d�}| jt |j||d� d S ) Ns 44444444444444444444444444444444s 555555555555r s 5555555555555555r )�outputs 0000000000000000r] ) r r r rj r r rr r/ r0 r5 ) r r r r! r" r# r� �resZshorter_outputr r r r� � s< zTestOutput.runTestN)rt ru rv r� r r r r r� � s r� c C s\ | � d�}g }|tt�7 }|tt�7 }|tt�7 }|t� g7 }|t|�g7 }|t� g7 }|S )Nr� )�getr r rw r r� r� r� )Zconfigr� Ztestsr r r � get_tests� s r� �__main__c C s t �t� � d S r� )r� Z TestSuiter� r r r r �suite s r� )Z defaultTest) r� �binasciir Z Cryptodome.SelfTest.st_commonr ZCryptodome.SelfTest.loaderr ZCryptodome.Util.py3compatr ZCryptodome.Cipherr ZCryptodome.Hashr rL r r r� r rw r r� r� r� r� r� rt r� �mainr r r r �<module> s0 - %WW / �
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings