File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/Cryptodome/SelfTest/Cipher/__pycache__/test_OCB.cpython-310.pyc
Back
o ;j� � @ s� d dl Z d dlmZ d dlmZmZmZ d dlmZ d dl m Z d dl m Z d dl mZ d dlmZ d d � ZG d d � d e j�ZG d d� de j�Zdd� ZG dd� de j�ZG dd� de j�Zi fdd�Zedkrtdd� Ze jdd� dS dS )� N)� unhexlify)�b�tobytes�bchr)� long_to_bytes)�load_test_vectors)�list_test_cases)�AES)�SHAKE128c 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_OCB.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 )(�OcbTests�key_128� � nonce_128� r � c C s` t j| jt j| jd�}tdd�}|�|�\}}t j| jt j| jd�}|�||�}| �||� d S )N��nonce� plaintexti@ ) r r r �MODE_OCB�nonce_96r �encrypt_and_digest�decrypt_and_verify� assertEqual)�self�cipher�pt�ct�mac�pt2r r r �test_loopback_1285 s zOcbTests.test_loopback_128c C s^ t �| jt j� t �| jt j| j�}|�| j�}t j| jt j| jd�}| �||�| j�� d S �Nr )r r r r r �encryptr r �r! r" r$ r r r � test_nonce> s zOcbTests.test_noncec C s | j ttj| jtjdd� d S )NZ test12345678r )� assertRaises� TypeErrorr r r r �r! r r r �test_nonce_must_be_bytesH s �z!OcbTests.test_nonce_must_be_bytesc C sl | j ttj| jtjtd�d� tdd�D ]}tj| jtj| jd |� d� q| j ttj| jtj| jd� d S )N� r � r ) r, � ValueErrorr r r r r �ranger )r! r r r r �test_nonce_lengthL s � �zOcbTests.test_nonce_lengthc C sj t j| jt j| jd�}| �|jt j� t �| jt j�j}t �| jt j�j}| �t|�d� | � ||� d S �Nr � ) r r r r r r � block_sizer �len�assertNotEqual�r! r" Znonce1Znonce2r r r �test_block_size_128X � z OcbTests.test_block_size_128c C sj t j| jt j| jd�}| �|j| j� t �| jt j�j}t �| jt j�j}| �t|�d� | �||� d S r5 ) r r r r r r r r8 r9 r: r r r �test_nonce_attributeb r< z OcbTests.test_nonce_attributec C sV | � ttj| jtj| jd� | j ttj| jtj| jdd� tj| jtj| jdd� d S )N� )r �unknownF)r Z use_aesni)r, r- r r r r r r. r r r �test_unknown_parametersl s �� �z OcbTests.test_unknown_parametersc C sF dD ] }t j| jt j| jd�}t||�td��}| �|td�� qd S )N�r) �decryptr r0 )r r r r r �getattrr r )r! �funcr" �resultr r r � test_null_encryption_decryptionw s �z(OcbTests.test_null_encryption_decryptionc C st t j| jt j| jd�}|�td�� | �t|j td�� t j| jt j| jd�}|� td�� | �t|jtd�� d S )Nr Zxyz) r r r r r r) r r, r- rB �r! r" r r r � test_either_encrypt_or_decrypt} s z'OcbTests.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 r, r- r) rB rG r r r �test_data_must_be_bytes� s z OcbTests.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 )Nr> �r �mac_len� � r r ) r, r2 r r r r r r3 r r r r8 )r! rK r" �_r% r r r � test_mac_len� s ���zOcbTests.test_mac_lenc C sh ddl m} tj| jtj| jd�}|�| j�\}}||d�}tj| jtj| jd�}| � t |j ||� d S )Nr )�strxor_cr r1 ) ZCryptodome.Util.strxorrP r r r r r r r r, r2 r )r! rP r" r$ r% Z invalid_macr r r �test_invalid_mac� s �zOcbTests.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 zOcbTests.test_hex_macc C sX t dd�}t dd�}tj| jtj| jd�}|�|� |�|�\}}dd� }dD ]>}tj| jtj| jd�}|||�D ]}|�|� q9td�} |||�D ] }| |� |�7 } qJ| |� � 7 } | � || � |� |� q'dD ]A}tj| jtj| jd�}|||�D ]}|�|� qztd�} |||�D ] }| |� |�7 } q�| |� � 7 } | � || � | � |� � |� qhd 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 zBOcbTests.test_message_chunks.<locals>.break_up.<locals>.<listcomp>r )r3 r8 )r rY r rX r �break_up� s �z.OcbTests.test_message_chunks.<locals>.break_up) r1 � � r> � � r �( �P r r0 )r r r r r r �updater r rB r �verifyr) rS ) r! � auth_datar r" Z ciphertextZref_macr[ rY �chunkr& �ct2r r r �test_message_chunks� s6 �z OcbTests.test_message_chunksc C sz 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�}~tj|tj|d�} d|d d�< d|d d�< | �|� d|d d�< | � t | �t | ��} | � | j| � d S �Nr s ���r] ) � bytearrayr r r r r r rb r) rS r r r ) r! Zkey_baZnonce_baZ header_baZdata_ba�cipher1r$ r �cipher2�ct_test�tag_test�cipher4�pt_testr r r �test_bytearray� sN � � � zOcbTests.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��}~tj|tj|d�} d|d d�< d|d d�< | �|� d|d d�< | � t | �t | ��} | � | j| � d S rh )� memoryviewri r r r r r r rb r) rS r r r ) r! Zkey_mvZnonce_mvZ header_mvZdata_mvrj r$ r rk rl rm rn ro r r r �test_memoryview sN � � � zOcbTests.test_memoryviewN)�__name__� __module__� __qualname__r r r r r' r+ r/ r4 r; r= r@ rF rH rI rO rQ rT rg rp rr r r r r r / 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#S )$� OcbFSMTestsr r r r r r c C sl t j| jt j| jd�}|�| j�}||�� 7 }|�� }t j| jt j| jd�}|�|� |�� |� |� d S r( ) r r r r r r) r rS rB rc )r! r" r$ r% r r r �-test_valid_init_encrypt_decrypt_digest_verifyC s � � z9OcbFSMTests.test_valid_init_encrypt_decrypt_digest_verifyc C sb t j| jt j| jd�}|�| j�}| �t|j � t j| jt j| jd�}|� |� | �t|j � d S r( ) r r r r r r) r r, r- rS rB rc r* r r r �/test_invalid_init_encrypt_decrypt_digest_verifyS s � � z;OcbFSMTests.test_invalid_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 rb r rS rc �r! r" r% r r r �$test_valid_init_update_digest_verifya s � � z0OcbFSMTests.test_valid_init_update_digest_verifyc C s t j| jt j| jd�}|�| j� |�| j�}||�� 7 }|�� }t j| jt j| jd�}|�| j� |� |� |� � |� |� t j| jt j| jd�}|�| j� |�| jd d� �}|� | jdd � �\}}t j| jt j| jd�}|�| j� |� |� |� ||� d S )Nr r\ ) r r r r r rb r r) rS rB rc r r )r! r" r$ r% Zct1rf r r r �test_valid_full_patho s4 � � � � z OcbFSMTests.test_valid_full_pathc C �L t j| jt j| jd�}|�| j� |�| j� |�� | �t |j| j� d S r( ) r r r r r rb r r) r, r- rG r r r � test_invalid_encrypt_after_final� � � z,OcbFSMTests.test_invalid_encrypt_after_finalc C r| r( ) r r r r r rb r rB r, r- rG r r r � test_invalid_decrypt_after_final� r~ z,OcbFSMTests.test_invalid_decrypt_after_finalc C s" t j| jt j| jd�}|�� d S r( )r r r r r rS rG r r r �test_valid_init_digest� s z"OcbFSMTests.test_valid_init_digestc C sB t j| jt j| jd�}|�� }t j| jt j| jd�}|�|� d S r( )r r r r r rS rc ry r r r �test_valid_init_verify� s z"OcbFSMTests.test_valid_init_verifyc C s� dD ]C}d t d�| j| jt d� fD ]2}tj| jtj| jd�}|d ur(|�|� t||�}|| j� || j� || j� || j� |� qqd S )NrA Z333�3r ) r r r r r r r rb rC )r! � method_namerd r" �methodr r r �&test_valid_multiple_encrypt_or_decrypt� s" � � ��z2OcbFSMTests.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 rb r rS r3 r rc )r! r" Z first_mac�xr r r �$test_valid_multiple_digest_or_verify� s �z0OcbFSMTests.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 rb r r r r )r! r" r$ r% r# r r r �0test_valid_encrypt_and_digest_decrypt_and_verify� s z<OcbFSMTests.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)rA )rB r) )TFr ) r r r r r rb r rC r, r- )r! Z method1_nameZ method2_nameZassoc_data_presentr" r r r �#test_invalid_mixing_encrypt_decrypt� s � ���z/OcbFSMTests.test_invalid_mixing_encrypt_decryptc C sx dD ]7}t j| jt j| jd�}|�| j� |�� |�� | �t t ||�| j� t j| jt j| jd�}|� | j� qd S )N)r) rb r ) r r r r r r) r rS r, r- rC r )r! r� r" r r r �+test_invalid_encrypt_or_update_after_digest� s ��z7OcbFSMTests.test_invalid_encrypt_or_update_after_digestc C s� t j| jt j| jd�}|�| j�}||�� 7 }|�� }dD ]B}t j| jt j| jd�}|�|� |�� |� |� | � t t ||�| j� t j| jt j| jd�}|� ||� | � t t ||�| j� q d S )Nr )rB rb )r r r r r r) r rS rB rc r, r- rC r )r! r" r$ r% r� r r r �+test_invalid_decrypt_or_update_after_verify� s$ � ��z7OcbFSMTests.test_invalid_decrypt_or_update_after_verifyN)rs rt ru r r r r rw rx rz r{ r} r r� r� r� r� r� r� r� r� r r r r rv = s$ rv c C sh t d�| d d t |� }d}td�D ]}t d�| }td| d |d �}tj|tj||d d�}|�|� ||�|�|�� |�� 7 }td| d |d �}tj|tj||d d�}||�|�|�� |�� 7 }td| d |d �}tj|tj||d d�}|�|� ||�� |�� 7 }qtd |d �}tj|tj||d d�}|�|� |�� |�� S ) z.Implement the algorithm at page 18 of RFC 7253r rM r1 � r r] rJ r\ i� ) r r3 r r r r rb r) rS ) �keylen�taglen�noncelen�key�CrW �S�Nr" r r r � algo_rfc7253 s& r� c @ s4 e Zd ZdZdZdZdZdd� Zdd� Zd d � Z d S ) �OcbRfc7253Test� 000102030405060708090A0B0C0D0E0F))ZBBAA99887766554433221100r0 r0 Z 785407BFFFC8AD9EDCC5520AC9111EE6)ZBBAA99887766554433221101�0001020304050607r� Z06820B3657B6F615A5725BDA0D3B4EB3A257C9AF1F8F03009)ZBBAA99887766554433221102r� r0 Z 81017F8203F081277152FADE694A0A00)ZBBAA99887766554433221103r0 r� Z045DD69F8F5AAE72414054CD1F35D82760B2CD00D2F99BFA9)ZBBAA99887766554433221104r� r� Z@571D535B60B277188BE5147170A9A22C3AD7A4FF3835B8C5701C1CCEC8FC3358)ZBBAA99887766554433221105r� r0 Z 8CF761B6902EF764462AD86498CA6B97)ZBBAA99887766554433221106r0 r� Z@5CE88EC2E0692706A915C00AEB8B2396F40E1C743F52436BDF06D8FA1ECA343D)ZBBAA99887766554433221107�0000102030405060708090A0B0C0D0E0F1011121314151617r� ZP1CA2207308C87C010756104D8840CE1952F09673A448A122C92C62241051F57356D7F3C90BB0E07F)ZBBAA99887766554433221108r� r0 Z 6DC225A071FC1B9F7C69F93B0F1E10DE)ZBBAA99887766554433221109r0 r� ZP221BD0DE7FA6FE993ECCD769460A0AF2D6CDED0C395B1C3CE725F32494B9F914D85C0B1EB38357FF)ZBBAA9988776655443322110A�@000102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1Fr� Z`BD6F6C496201C69296C11EFD138A467ABD3C707924B964DEAFFC40319AF5A48540FBBA186C5553C68AD9F592A79A4240)ZBBAA9988776655443322110Br� r0 Z FE80690BEE8A485D11F32965BC9D2A32)ZBBAA9988776655443322110Cr0 r� Z`2942BFC773BDA23CABC6ACFD9BFD5835BD300F0973792EF46040C53F1432BCDFB5E1DDE3BC18A5F840B52E653444D5DF)�BBAA9988776655443322110D�P000102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1F2021222324252627r� ZpD5CA91748410C1751FF8A2F618255B68A0A12E093FF454606E59F9C1D0DDC54B65E8628E568BAD7AED07BA06A4A69483A7035490C5769E60)ZBBAA9988776655443322110Er� r0 Z C5CD9D1850C141E358649994EE701B68)ZBBAA9988776655443322110Fr0 r� Zp4412923493C57D5DE0D700F753CCE0D1D2D95060122E9F15A5DDBFC5787E50B5CC55EE507BCB084E479AD363AC366B95A98CA5F3000B1479)� 0F0E0D0C0B0A09080706050403020100r� r� r� Zh1792A4E31E0755FB03E31B22116E6C2DDF9EFD6E33D536F1A0124B0A55BAE884ED93481529C76B6AD0C515F4D1CDD4FDAC4F02AA) )r r Z 67E944D23256C5E0B6C61FA22FDF1EA2)� r Z F673F2C3E7174AAE7BAE986CA9F29E17)� r Z D90EB8E9C977C88B79DD793D7FFA161C)r �` Z77A3D8E73589158D25D01209)r� r� Z05D56EAD2752C86BE6932C5E)r� r� Z5458359AC23B0CBA9E6330DD)r �@ Z192C9B7BD90BA06A)r� r� Z0066BC6E0EF34E24)r� r� Z7D4EA5D445501CBEc C s� t t| j��}| jD ]a}dd� |D �\}}}}|d d� |dd � }}tj|tj|d�}|�|� |�|�|�� } | � || � | � ||� � � tj|tj|d�}|�|� |� |�|� � } | � || � |� |� q d S )Nc S � g | ]}t t|���qS r �r r �rV r� r r r rZ � � z(OcbRfc7253Test.test1.<locals>.<listcomp>i�r ) r r �tv1_key�tv1r r r rb r) r rS rB rc ) r! r� �tvr �aadr# r$ �mac_tagr" rf r& r r r �test1� s �zOcbRfc7253Test.test1c C s� dd� | j D �\}}}}}|d d� |dd � }}tj|tj|dd�}|�|� |�|�|�� }| �||� | �||�� � tj|tj|dd�}|�|� |�|�|�� } | �|| � |� |� d S )Nc S r� r r� r� r r r rZ � r� z(OcbRfc7253Test.test2.<locals>.<listcomp>i�r rJ ) �tv2r r r rb r) r rS rB rc ) r! r� r r� r# r$ r� r" rf r& r r r �test2� s zOcbRfc7253Test.test2c C s6 | j D ]\}}}t||d�}| �tt|��|� qd S )Nr� )�tv3r� r r r )r! r� r� rE Zresult2r r r �test3� s �zOcbRfc7253Test.test3N) rs rt ru r� r� r� r� r� r� r� r r r r r� s { r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � OcbDkgTestz9Test vectors from https://gitlab.com/dkg/ocb-test-vectorsc C s� g }dD ]2}dD ]-}t dd||f d||f i �}|d u r n|d j}|dd � D ] }||_|�|� q*qq|D ]9}|j|j|j|j|jf\}}} } } t| �t| � } tj |tj || d�} | � | � | � | �\}}| � | || � q9d S ) N)r1 r\ )�h �p �x )ZCipherr ztest-vector-%d-nonce%d.txtzDKG tests, %d, %d bitsr r1 rJ )r �k�append�n�a�p�cr8 r r r rb r r )r! Ztvs�fi�nbZtv_filer� r� r� r� r� r� r� rK r" Zc_outZtag_outr r r �test_1_2� s0 � ��" �zOcbDkgTest.test_1_2c s� � fdd�}|dddd� |dddd� |dddd � |dd dd � |dd dd � |dd dd � |dddd� |dddd� |dddd� |dddd� |dddd� |dddd� |dd dd� |dd dd� |dd dd� |dddd� |dddd� |dddd� |ddd d � |ddd d � |ddd d � |dd d d � |dd d d!� |dd d d"� |ddd d#� |ddd d$� |ddd d%� d S )&Nc s t | ||�}� �|t|�� d S �N)r� r r )r� r� r� �exprE r. r r �check s z OcbDkgTest.test_3.<locals>.checkr r� Z C47F5F0341E15326D4D1C46F47F05062r� Z 95B9167A38EB80495DFC561A8486E109r� Z AFE1CDDB97028FD92F8FB3C8CFBA7D83r� ZF471B4983BA80946DF217A54Z5AE828BC51C24D85FA5CC7B2Z8C8335982E2B734616CAD14Cr� ZB553F74B85FD1E5BZ3B49D20E513531F9ZED6DA5B1216BF8BBr� Z CA8AFCA031BAC3F480A583BD6C50A547Z D170C1DF356308079DA9A3F619147148Z 57F94381F2F9231EFB04AECD323757C3Z3A618B2531ED39F260C750DCZ9071EB89FEDBADDA88FD286EZFDF0EFB97F21A39AC4BAB5ACZFAB2FF3A8DD82A13ZAC01D912BD0737D3Z9D1FD0B500EA4ECFr� Z 9E043A7140A25FB91F43BCC9DD7E0F46Z 680000E53908323A7F396B955B8EC641Z 8304B97FAACDA56E676602E1878A7E6FZ81F978AC9867E825D339847DZEFCF2D60B24926ADA48CF5B1Z84961DC56E917B165E58C174Z227AEE6C9D905A61Z541DE691B9E1A2F9ZB0E761381C7129FCr )r! r� r r. r �test_3 s8 zOcbDkgTest.test_3c C s� t d�}t d�}t d�}t d�}t d�}t|�t|� }t d�}tj|tj|d d� |d�}|�|� |�|�\} } | �|| | � d S )NZ EEDDCCBBAA9988776655443322110Dr� r� Zh07E903BFC49552411ABC865F5ECE60F6FAD1F5A9F14D3070FA2F1308A563207FFE14C1EEA44B22059C7484319D8A2C53C236A7B3ZhBA015C4E5AE54D76C890AE81BD40DC5703EDC30E8AC2A58BC5D8FA4D61C5BAE6C39BEAC435B2FD56A2A5085C1B135D770C8264B7���rJ )r r8 r r r rb r r ) r! r r� �A�Pr� rK Z buggy_resultr" ZC_out2Ztag_out2r r r � test_2_bugfix( s zOcbDkgTest.test_2_bugfixN)rs rt ru �__doc__r� r� r� r r r r r� � s 'r� c C s8 g }|t t�7 }|t t�7 }|t t�7 }|t t�7 }|S r� )r r rv r� r� )ZconfigZtestsr r r � get_testsA s r� �__main__c C s t �t� �S r� )�unittestZ TestSuiter� r r r r �suiteK s r� )Z defaultTest) r� �binasciir ZCryptodome.Util.py3compatr r r ZCryptodome.Util.numberr ZCryptodome.SelfTest.loaderr Z Cryptodome.SelfTest.st_commonr ZCryptodome.Cipherr ZCryptodome.Hashr r ZTestCaser rv r� r� r� r� rs r� �mainr r r r �<module> s. F I Z �
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings