U
    ÈZjQ–  ã                   @   sè   d dl Z d dlmZ d dlmZ d dlmZ d dlmZm	Z	 d dl
mZ d dlmZ d dlmZ d d	lmZ d
d„ ZG dd„ de jƒZG dd„ de jƒZG dd„ de jƒZG dd„ de jƒZi fdd„Zedkrädd„ Ze jdd� dS )é    N©Ú	unhexlify)Úlist_test_cases)Úload_test_vectors_wycheproof)ÚtobytesÚbchr)ÚAES)ÚSHAKE128)Ústrxor)ÚCCMMessageTooLongErrorc                 C   s   t jt| ƒd� |¡S )N)Údata)r	   Únewr   Úread)ÚtagÚlength© r   úb/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/Crypto/SelfTest/Cipher/test_CCM.pyÚget_tag_random-   s    r   c                   @   sê   e Zd ZeddƒZeddƒZeddƒZdd„ Zd	d
„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zd+d,„ Zd-d.„ Zd/d0„ Zd1d2„ Zd3d4„ Zd5d6„ Zd7S )8ÚCcmTestsÚkey_128é   Ú	nonce_128é   r   é€   c                 C   sZ   t j| jt j| jd�}tddƒ}| |¡}t j| jt j| jd�}| |¡}|  ||¡ d S )N©ÚnonceÚ	plaintexti@  )	r   r   r   ÚMODE_CCMÚnonce_96r   ÚencryptÚdecryptÚassertEqual)ÚselfÚcipherÚptÚctÚpt2r   r   r   Útest_loopback_1287   s    


zCcmTests.test_loopback_128c                 C   s–   t  | jt j¡}|j}t  | jt j¡}|j}|  t|ƒd¡ |  ||¡ t  | jt j| j¡}| 	| j
¡}t j| jt j| jd�}|  || 	| j
¡¡ d S )Né   r   )r   r   r   r   r   r!   ÚlenÚassertNotEqualr   r   r   )r"   r#   Únonce1Únonce2r%   r   r   r   Ú
test_nonce@   s    zCcmTests.test_noncec                 C   s   | j ttj| jtjdd� d S )NZtest12345678r   )ÚassertRaisesÚ	TypeErrorr   r   r   r   ©r"   r   r   r   Útest_nonce_must_be_bytesO   s    ÿz!CcmTests.test_nonce_must_be_bytesc                 C   sŽ   | j ttj| jtjdd� | j ttj| jtjtdƒd d� | j ttj| jtjtdƒd d� tddƒD ] }tj| jtjtdƒ| d� qhd S )Nó    r   é   é   é   é   )r.   Ú
ValueErrorr   r   r   r   r   Úrange)r"   Úxr   r   r   Útest_nonce_lengthS   s    ÿ
ÿ
ÿzCcmTests.test_nonce_lengthc                 C   s*   t j| jt j| jd�}|  |jt j¡ d S ©Nr   )r   r   r   r   r   r!   Ú
block_size©r"   r#   r   r   r   Útest_block_size]   s    zCcmTests.test_block_sizec                 C   sj   t j| jt j| jd�}|  |j| j¡ t  | jt j¡j}t  | jt j¡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_attributea   s    zCcmTests.test_nonce_attributec                 C   sV   |   ttj| jtj| jd¡ | j ttj| jtj| jdd� tj| jtj| jdd� d S )Nr6   )r   ÚunknownF)r   Z	use_aesni)r.   r/   r   r   r   r   r   r0   r   r   r   Útest_unknown_parametersk   s     ÿ ÿÿz CcmTests.test_unknown_parametersc                 C   s>   dD ]4}t j| jt j| jd�}t||ƒdƒ}|  |d¡ qd S )N©r   r    r   r2   )r   r   r   r   r   Úgetattrr!   )r"   Úfuncr#   Úresultr   r   r   Útest_null_encryption_decryptionv   s    z(CcmTests.test_null_encryption_decryptionc                 C   sd   t j| jt j| jd�}| d¡ |  t|jd¡ t j| jt j| jd�}| d¡ |  t|jd¡ d S )Nr   r2   )	r   r   r   r   r   r   r.   r/   r    r=   r   r   r   Útest_either_encrypt_or_decrypt|   s    

z'CcmTests.test_either_encrypt_or_decryptc                 C   sP   t j| jt j| jd�}|  t|jd¡ t j| jt j| jd�}|  t|jd¡ d S )Nr   ztest1234567890-*)	r   r   r   r   r   r.   r/   r   r    r=   r   r   r   Útest_data_must_be_bytes…   s    z CcmTests.test_data_must_be_bytesc              	   C   s´   t dddƒD ]"}| jttj| jtj| j|d� qt dddƒD ]<}tj| jtj| j|d�}| | j	¡\}}|  
t|ƒ|¡ q<tj| jtj| jd�}| | j	¡\}}|  
t|ƒd¡ d S )	Né   é   é   )r   Úmac_lené   é   r   r   )r8   r.   r7   r   r   r   r   r   Úencrypt_and_digestr   r!   r)   )r"   rL   r#   Ú_Úmacr   r   r   Útest_mac_lenŒ   s     ÿÿzCcmTests.test_mac_lenc                 C   sh   ddl m} tj| jtj| jd�}| | j¡\}}||dƒ}tj| jtj| jd�}|  	t
|j||¡ d S )Nr   )Ústrxor_cr   r3   )ÚCrypto.Util.strxorrS   r   r   r   r   r   rO   r   r.   r7   Údecrypt_and_verify)r"   rS   r#   r%   rQ   Zinvalid_macr   r   r   Útest_invalid_macž   s    
ÿzCcmTests.test_invalid_macc                 C   sV   t j| jt j| jd�}| ¡ }|  | ¡ t|ƒ¡ t j| jt j| jd�}| 	|¡ d S r;   )
r   r   r   r   r   Ú	hexdigestr!   Údigestr   Z	hexverify)r"   r#   Zmac_hexr   r   r   Útest_hex_mac©   s
    zCcmTests.test_hex_macc                 C   sV   t j| jt j| jdd�}|  t|jd¡ t j| jt j| jdd�}|  t|j| j¡ d S )Nr   ©r   Ú	assoc_lenó   1é   )	r   r   r   r   r   r.   r7   Úupdater   r=   r   r   r   Ú$test_longer_assoc_data_than_declared±   s    ÿÿz-CcmTests.test_longer_assoc_data_than_declaredc                 C   s  t | jƒ}tj| jtj| j|d d�}| | j¡ |  t	|j
| j¡ tj| jtj| j|d d�}| | j¡ |  t	|j¡ tj| jtj| j|d d�}| | j¡ |  t	|j| j¡ tj| jtj| jd�}| | j¡ | ¡ }tj| jtj| j|d d�}| | j¡ |  t	|j|¡ d S )Nr3   rZ   r   )r)   r   r   r   r   r   r   r^   r.   r7   r   rX   r    Úverify)r"   ÚDATA_LENr#   rQ   r   r   r   Ú%test_shorter_assoc_data_than_expected¼   s0    
ÿÿÿÿz.CcmTests.test_shorter_assoc_data_than_expectedc                 C   sr   t | jƒ}tj| jtj| j|d d�}| | j¡ |  t	|j
¡ tj| jtj| j|d d�}|  t	|j| j¡ d S )Nr3   ©r   Úmsg_len)r)   r   r   r   r   r   r   r   r.   r7   rX   )r"   ra   r#   r   r   r   Ú/test_shorter_and_longer_plaintext_than_declaredÛ   s    
ÿÿz8CcmTests.test_shorter_and_longer_plaintext_than_declaredc                 C   s–   t | jƒ}tj| jtj| jd�}| | j¡\}}tj| jtj| j|d d�}| |¡ |  	t
|j|¡ tj| jtj| j|d d�}|  	t
|j|¡ d S )Nr   r3   rc   )r)   r   r   r   r   r   r   rO   r    r.   r7   r`   )r"   ra   r#   r%   rQ   r   r   r   Ú%test_shorter_ciphertext_than_declaredç   s    
ÿ
ÿz.CcmTests.test_shorter_ciphertext_than_declaredc                 C   sB  t ddƒ}t ddƒ}tj| jtj| jd�}| |¡ | |¡\}}dd„ }dD ]p}tj| jtj| jddd�}|||ƒD ]}| |¡ qvd	}	|||ƒD ]}|	| |¡7 }	q”|  	||	¡ | 
|¡ qNdD ]x}tj| jtj| jddd�}|||ƒD ]}| |¡ qìd	}
|||ƒD ]}|
| |¡7 }
�q
|  	||
¡ |  	| ¡ |¡ qÄd S )
Nzauthenticated dataé   r   r   c                    s    ‡ ‡fdd„t dtˆƒˆ ƒD ƒS )Nc                    s   g | ]}ˆ||ˆ  … ‘qS r   r   )Ú.0Úi©Úchunk_lengthr   r   r   Ú
<listcomp>  s     zBCcmTests.test_message_chunks.<locals>.break_up.<locals>.<listcomp>r   )r8   r)   )r   rk   r   rj   r   Úbreak_up  s    ÿz.CcmTests.test_message_chunks.<locals>.break_up)
r3   rK   rI   r6   é
   é   r   é(   éP   r   ©r   rd   r[   r2   )r   r   r   r   r   r   r^   rO   r    r!   r`   r   rX   )r"   Ú	auth_datar   r#   Z
ciphertextZref_macrm   rk   Úchunkr&   Úct2r   r   r   Útest_message_chunksö   s<    


 ÿ ÿzCcmTests.test_message_chunksc                 C   sj  t | jƒ}t | jƒ}t | jƒ}t | jƒ}tj| jtj| jd�}| | j¡ | | j¡}| 	¡ }tj|tj|d�}d|d d…< d|d d…< | |¡ d|d d…< | |¡}	d|d d…< | 	¡ }
|  
||	¡ |  
||
¡ |  
|j|j¡ t | jƒ}t | jƒ}t | jƒ}~tj|tj|d�}d|d d…< d|d d…< | |¡ d|d d…< | t |	ƒt |
ƒ¡}|  
| j|¡ d S ©Nr   s   ÿÿÿrI   )Ú	bytearrayr   r   r   r   r   r   r^   r   rX   r!   r   rU   )r"   Zkey_baZnonce_baZ	header_baZdata_baÚcipher1r%   r   Úcipher2Úct_testÚtag_testÚcipher4Úpt_testr   r   r   Útest_bytearray!  sN    



þþ




þ
zCcmTests.test_bytearrayc                 C   s†  t t| jƒƒ}t t| jƒƒ}t t| jƒƒ}t t| jƒƒ}tj| jtj| jd�}| | j¡ | 	| j¡}| 
¡ }tj|tj|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j|tj|d�}d|d d…< d|d d…< | |¡ d|d d…< | t |	ƒt |
ƒ¡}|  | j|¡ d S rw   )Ú
memoryviewrx   r   r   r   r   r   r   r^   r   rX   r!   r   rU   )r"   Zkey_mvZnonce_mvZ	header_mvZdata_mvry   r%   r   rz   r{   r|   r}   r~   r   r   r   Útest_memoryviewP  sN    þþ

þ
zCcmTests.test_memoryviewc                 C   s:  d}t j| jt j| jd�}| |¡}| ¡ }tdƒ}t j| jt j| jd�}|j||d�}|  ||¡ |  |d ¡ t j| jt j| jd�}|j	||d�}|  ||¡ |  |d ¡ t j| jt j| jd�}|j
||d�\}}|  ||¡ |  |d ¡ |  ||¡ t j| jt j| jd�}|j|||d�}|  ||¡ |  |d ¡ d S ©Ns€   55555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555r   r   ©Úoutput)r   r   r   r   r   r   rX   rx   r!   r    rO   rU   )r"   r$   r#   r%   r   r„   ÚresZtag_outr   r   r   Útest_output_param  s,    
zCcmTests.test_output_paramc                 C   s”   d}t j| jt j| jd�}| |¡}ttdƒƒ}t j| jt j| jd�}|j||d� |  ||¡ t j| jt j| jd�}|j	||d� |  ||¡ d S r‚   )
r   r   r   r   r   r   r€   rx   r!   r    )r"   r$   r#   r%   r„   r   r   r   Útest_output_param_memoryviewœ  s    
z%CcmTests.test_output_param_memoryviewc                 C   sØ   d}t j| jt j| jd�}| |¡}t j| jt j| jd�}| jt|j|dd� t j| jt j| jd�}| jt|j|dd� t	dƒ}t j| jt j| jd�}| jt
|j||d� t j| jt j| jd�}| jt
|j||d� d S )Ns   5555555555555555r   s   0000000000000000rƒ   r]   )r   r   r   r   r   r   r.   r/   r    rx   r7   )r"   r$   r#   r%   Zshorter_outputr   r   r   Útest_output_param_neg«  s    
zCcmTests.test_output_param_negc              	   C   s    d}| j ttj| jtj|ddd� d}| j ttj| jtj|ddd� d}tj| jtj|d�}|   t|jdd ¡ d}tj| jtj|d�}|   t|jdd ¡ d S )	Ns   NNNNNNNNNNNNNé   i   ©r   r[   rd   s   NNNNNNNl            r   ó   C)r.   r   r   r   r   r   r   r    )r"   r   r#   r   r   r   Útest_message_too_long½  s<    úúþþzCcmTests.test_message_too_longN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r'   r-   r1   r:   r>   r?   rA   rF   rG   rH   rR   rV   rY   r_   rb   re   rf   rv   r   r�   r†   r‡   rˆ   rŒ   r   r   r   r   r   1   s6   


	

	+//r   c                   @   sŠ   e Zd ZeddƒZeddƒZeddƒZdd„ Zdd	„ Zd
d„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )ÚCcmFSMTestsr   r   r   r   r   c              	   C   s~   dD ]t}d t | jƒfD ]`}tj| jtj| j||d�}| | j¡}| ¡ }tj| jtj| j||d�}| 	|¡ | 
|¡ qqd S ©N)Nr   rŠ   )r)   r   r   r   r   r   r   r   rX   r    r`   ©r"   r[   rd   r#   r%   rQ   r   r   r   Ú-test_valid_init_encrypt_decrypt_digest_verifyä  s     ýý
z9CcmFSMTests.test_valid_init_encrypt_decrypt_digest_verifyc              	   C   s€   d t | jƒfD ]l}dD ]b}tj| jtj| j||d�}| | j¡ | ¡ }tj| jtj| j||d�}| | j¡ | 	|¡ qqd S r‘   )
r)   r   r   r   r   r   r   r^   rX   r`   )r"   r[   rd   r#   rQ   r   r   r   Ú$test_valid_init_update_digest_verifyø  s     ýýz0CcmFSMTests.test_valid_init_update_digest_verifyc              	   C   s    d t | jƒfD ]Œ}d t | jƒfD ]x}tj| jtj| j||d�}| | j¡ | | j¡}| 	¡ }tj| jtj| j||d�}| | j¡ | 
|¡ | |¡ q qd S )NrŠ   )r)   r   r   r   r   r   r   r^   r   rX   r    r`   r’   r   r   r   Útest_valid_full_path  s$    ýý
z CcmFSMTests.test_valid_full_pathc                 C   s"   t j| jt j| jd�}| ¡  d S r;   )r   r   r   r   r   rX   r=   r   r   r   Útest_valid_init_digest"  s    z"CcmFSMTests.test_valid_init_digestc                 C   sB   t j| jt j| jd�}| ¡ }t j| jt j| jd�}| |¡ d S r;   )r   r   r   r   r   rX   r`   )r"   r#   rQ   r   r   r   Útest_valid_init_verify'  s    z"CcmFSMTests.test_valid_init_verifyc              	   C   sœ   dD ]’}d d| j | j d fD ]x}|d kr.d }nt|ƒ}tj| jtj| jd|d�}|d k	rb| |¡ t||ƒ}|| j ƒ || j ƒ || j ƒ || j ƒ qqd S )NrB   s   333ó   3é@   rr   )	r   r)   r   r   r   r   r   r^   rC   )r"   Úmethod_namers   r[   r#   Úmethodr   r   r   Ú&test_valid_multiple_encrypt_or_decrypt/  s&    ÿý




z2CcmFSMTests.test_valid_multiple_encrypt_or_decryptc                 C   s†   t j| jt j| jd�}| | j¡ | ¡ }tdƒD ]}|  	|| ¡ ¡ q2t j| jt j| jd�}| | j¡ tdƒD ]}| 
|¡ qrd S )Nr   rM   é   )r   r   r   r   r   r^   r   rX   r8   r!   r`   )r"   r#   Z	first_macr9   r   r   r   Ú$test_valid_multiple_digest_or_verifyD  s    z0CcmFSMTests.test_valid_multiple_digest_or_verifyc                 C   sr   t j| jt j| jd�}| | j¡ | | j¡\}}t j| jt j| jd�}| | j¡ | ||¡}|  	| j|¡ d S r;   )
r   r   r   r   r   r^   r   rO   rU   r!   )r"   r#   r%   rQ   r$   r   r   r   Ú0test_valid_encrypt_and_digest_decrypt_and_verifyR  s    z<CcmFSMTests.test_valid_encrypt_and_digest_decrypt_and_verifyc                 C   sb   dD ]X}dD ]N}t j| jt j| jd�}|r6| | j¡ t||ƒ}|| jƒ |  t	|| j¡ qqd S )NrB   ©TFr   ©
r   r   r   r   r   r^   r   rC   r.   r/   )r"   rš   Úassoc_data_presentr#   r›   r   r   r   Ú5test_invalid_multiple_encrypt_decrypt_without_msg_len^  s    ÿ

zACcmFSMTests.test_invalid_multiple_encrypt_decrypt_without_msg_lenc                 C   sj   dD ]`\}}dD ]R}t j| jt j| jdd�}|r<| | j¡ t||ƒ| jƒ |  t	t||ƒ| j¡ qqd S )N)rB   )r    r   r    é    rc   r¡   )r"   Zmethod1_nameZmethod2_namer¢   r#   r   r   r   Ú#test_invalid_mixing_encrypt_decryptj  s    þÿz/CcmFSMTests.test_invalid_mixing_encrypt_decryptc                 C   sp   dD ]f}t j| jt j| jd�}| | j¡ | ¡  |  t	t
||ƒ| j¡ t j| jt j| jd�}| | j¡ qd S )N)r   r^   r   )r   r   r   r   r   r   r   rX   r.   r/   rC   rO   )r"   rš   r#   r   r   r   Ú+test_invalid_encrypt_or_update_after_digestx  s    ÿz7CcmFSMTests.test_invalid_encrypt_or_update_after_digestc                 C   s°   t j| jt j| jd�}| | j¡}| ¡ }dD ]|}t j| jt j| jd�}| |¡ | 	|¡ |  
tt||ƒ| j¡ t j| jt j| jd�}| ||¡ |  
tt||ƒ| j¡ q.d S )Nr   )r    r^   )r   r   r   r   r   r   r   rX   r    r`   r.   r/   rC   rU   )r"   r#   r%   rQ   rš   r   r   r   Ú+test_invalid_decrypt_or_update_after_verifyƒ  s    

ÿÿz7CcmFSMTests.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   r�   Þ  s   


r�   c                   @   sˆ   e Zd ZdZdddd dd„ edd	ƒD ƒ¡d
 dddddfddddddddddddddddd d!d"d#d$d%d&d'gZd(d„ eD ƒZd)d*„ Zd+S ),ÚTestVectorsz`Class exercising the CCM test vectors found in Appendix C
    of NIST SP 800-38C and in RFC 3610)Ú0001020304050607Z20212223Z7162015bZ4dac255dÚ 404142434445464748494a4b4c4d4e4fZ10111213141516)Z 000102030405060708090a0b0c0d0e0fZ 202122232425262728292a2b2c2d2e2fZ d2a1f0e051ea5f62081a7792073d593dZ1fc64fbfaccdrª   Z1011121314151617)Z(000102030405060708090a0b0c0d0e0f10111213Z0202122232425262728292a2b2c2d2e2f3031323334353637Z0e3b201a9f5b71a7a9b1ceaeccd97e70b6176aad9a4428aa5Z484392fbc1b09951rª   Z101112131415161718191a1bÚ c                 C   s,   g | ]$}t d dƒD ]}d|d |  ‘qqS )r   r   z%02X)r8   )rh   r9   Úyr   r   r   rl   ´  s       zTestVectors.<listcomp>r   r   é   Z@202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3fZ@69915dad1e84c6376a68c2967e4dab615ae0fd1faec44cc484828529463ccf72Zb4ac6bec93e8598e7f0dadbcea5brª   Z101112131415161718191a1b1c)r©   Ú.08090a0b0c0d0e0f101112131415161718191a1b1c1d1eZ.588c979a61c663d2f066d0c2c0f989806d5f6b61dac384Z17e8d12cfdf926e0Ú c0c1c2c3c4c5c6c7c8c9cacbcccdcecfZ00000003020100a0a1a2a3a4a5)r©   Ú008090a0b0c0d0e0f101112131415161718191a1b1c1d1e1fZ072c91a36e135f8cf291ca894085c87e3cc15c439c9e43a3bZa091d56e10400916r¯   Z00000004030201a0a1a2a3a4a5)r©   Ú208090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f20Z251b1e5f44a197d1da46b0f8e2d282ae871e838bb64da859657Z4adaa76fbd9fb0c5r¯   Z00000005040302A0A1A2A3A4A5)Ú000102030405060708090a0bÚ&0c0d0e0f101112131415161718191a1b1c1d1eZ&a28c6865939a9a79faaa5c4c2a9d4a91cdac8cZ96c861b9c9e61ef1r¯   Z00000006050403a0a1a2a3a4a5)r²   Ú(0c0d0e0f101112131415161718191a1b1c1d1e1fZ(dcf1fb7b5d9e23fb9d4e131253658ad86ebdca3eZ51e83f077d9c2d93r¯   Z00000007060504a0a1a2a3a4a5)r²   Ú*0c0d0e0f101112131415161718191a1b1c1d1e1f20Z*6fc1b011f006568b5171a42d953d469b2570a4bd87Z405a0443ac91cb94r¯   Z00000008070605a0a1a2a3a4a5)r©   r®   Z.0135d1b2c95f41d5d1d4fec185d166b8094e999dfed96cZ048c56602c97acbb7490r¯   Z00000009080706a0a1a2a3a4a5)r©   r°   Z07b75399ac0831dd2f0bbd75879a2fd8f6cae6b6cd9b7db24Zc17b4433f434963f34b4r¯   Z0000000a090807a0a1a2a3a4a5)r©   r±   Z282531a60cc24945a4b8279181ab5c84df21ce7f9b73f42e197Zea9c07e56b5eb17e5f4er¯   Z0000000b0a0908a0a1a2a3a4a5)r²   r³   Z&07342594157785152b074098330abb141b947bZ566aa9406b4d999988ddr¯   Z0000000c0b0a09a0a1a2a3a4a5)r²   r´   Z(676bb20380b0e301e8ab79590a396da78b834934Zf53aa2e9107a8b6c022cr¯   Z0000000d0c0b0aa0a1a2a3a4a5)r²   rµ   Z*c0ffa0d6f05bdb67f24d43a4338d2aa4bed7b20e43Zcd1aa31662e7ad65d6dbr¯   Z0000000e0d0c0ba0a1a2a3a4a5)Z0be1a88bace018b1Z.08e8cf97d820ea258460e96ad9cf5289054d895ceac47cZ.4cb97f86a2a4689a877947ab8091ef5386a6ffbdd080f8Ze78cf7cb0cddd7b3Ú d7828d13b2b0bdc325a76236df93cc6bZ00412b4ea9cdbe3c9696766cfa)Z63018f76dc8a1bcbZ09020ea6f91bdd85afa0039ba4baff9bfb79c7028949cd0ecZ04ccb1e7ca981befaa0726c55d378061298c85c92814abc33Zc52ee81d7d77c08ar¶   Z0033568ef7b2633c9696766cfa)Zaa6cfa36cae86b40Z2b916e0eacc1c00d7dcec68ec0b3bbb1a02de8a2d1aa346132eZ2b1d23a2220ddc0ac900d9aa03c61fcf4a559a4417767089708Za776796edb723506r¶   Z00103fe41336713c9696766cfa)Zd0d0735c531e1becf049c244Z&12daac5630efa5396f770ce1a66b21f7b2101cZ&14d253c3967b70609b7cbb7c49916028324526Z9a6f49975bcadeafr¶   Z00764c63b8058e3c9696766cfa)Z77b60f011c03e1525899bcaeZ(e88b6a46c78d63e52eb8c546efb5de6f75e9cc0dZ(5545ff1a085ee2efbf52b2e04bee1e2336c73e3fZ762c0c7744fe7e3cr¶   Z00f8b678094e3b3c9696766cfa)Zcd9044d2b71fdb8120ea60c0Z*6435acbafb11a82e2f071d7ca4a5ebd93a803ba87fZ*009769ecabdf48625594c59251e6035722675e04c8Z47099e5ae0704551r¶   Z00d560912d3f703c9696766cfa)Zd85bc7e69f944fb8Z.8a19b950bcf71a018e5e6701c91787659809d67dbedd18Z.bc218daa947427b6db386a99ac1aef23ade0b52939cb6aZ637cf9bec2408897c6bar¶   Z0042fff8f1951c3c9696766cfa)Z74a0ebc9069f5b37Z01761433c37c5a35fc1f39f406302eb907c6163be38c98437Z05810e6fd25874022e80361a478e3e9cf484ab04f447efff6Zf0a477cc2fc9bf548944r¶   Z00920f40e56cdc3c9696766cfa)Z44a3aa3aae6475caZ2a434a8e58500c6e41530538862d686ea9e81301b5ae4226bfaZ2f2beed7bc5098e83feb5b31608f8e29c38819a89c8e776f154Z4d4151a4ed3a8b87b9cer¶   Z0027ca0c7120bc3c9696766cfa)Zec46bb63b02520c33c49fd70Z&b96b49e21d621741632875db7f6c9243d2d7c2Z&31d750a09da3ed7fddd49a2032aabf17ec8ebfZ7d22c8088c666be5c197r¶   Z005b8ccbcd9af83c9696766cfa)Z47a65ac78b3d594227e85e71Z(e2fcfbb880442c731bf95167c8ffd7895e337076Z(e882f1dbd38ce3eda7c23f04dd65071eb41342acZdf7e00dccec7ae52987dr¶   Z003ebe94044b9a3c9696766cfa)Z6e37a6ef546d955d34ab6059Z*abf21c0b02feb88f856df4a37381bce3cc128517d4Z*f32905b88a641b04b9c9ffb58cc390900f3da12ab1Z6dce9e82efa16da62059r¶   Z008d493b30ae8b3c9696766cfac                 C   s   g | ]}d d„ |D ƒ‘qS )c                 S   s   g | ]}t |ƒ‘qS r   r   )rh   r9   r   r   r   rl   N  s     z!TestVectors.<listcomp>.<listcomp>r   )rh   Útvr   r   r   rl   N  s     c                 C   sž   | j D ]’\}}}}}}tj|tj|t|ƒd�}| |¡ | |¡\}}	|  ||¡ |  ||	¡ tj|tj|t|ƒd�}| |¡ | ||¡}
|  ||
¡ qd S )N)rL   )	Útest_vectorsr   r   r   r)   r^   rO   r!   rU   )r"   Z
assoc_datar$   r%   rQ   Úkeyr   r#   ru   Zmac2r&   r   r   r   ÚrunTestP  s    

zTestVectors.runTestN)	r�   rŽ   r�   Ú__doc__Újoinr8   Ztest_vectors_hexr¸   rº   r   r   r   r   r¨   •  sP   û€ Ú /r¨   c                   @   sL   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dS )ÚTestVectorsWycheproofc                 K   s"   t j | ¡ || _|| _d| _d S )NÚNone)ÚunittestÚTestCaseÚ__init__Ú_wycheproof_warningsÚ_extra_paramsÚ_id)r"   Úwycheproof_warningsZextra_paramsr   r   r   rÁ   b  s    zTestVectorsWycheproof.__init__c                 C   s"   dd„ }t dddd|id�| _d S )Nc                 S   s   | d d S )NZtagSizeé   r   )Úgroupr   r   r   Ú
filter_tagj  s    z/TestVectorsWycheproof.setUp.<locals>.filter_tag)ZCipherZ
wycheproofzaes_ccm_test.jsonzWycheproof AES CCMÚtag_size)Z	group_tag)r   r·   )r"   rÈ   r   r   r   ÚsetUph  s    ýzTestVectorsWycheproof.setUpc                 C   s   | j S ©N)rÄ   r0   r   r   r   ÚshortDescriptionr  s    z&TestVectorsWycheproof.shortDescriptionc                 C   s.   |j r*| jr*dd l}| d| j|jf ¡ d S )Nr   zWycheproof warning: %s (%s))ÚwarningrÂ   ÚwarningsÚwarnrÄ   Úcomment)r"   r·   rÎ   r   r   r   rÏ   u  s    zTestVectorsWycheproof.warnc              
   C   s  dt |jƒ | _z(tj|jtj|jfd|ji| j	—Ž}W nŽ t
k
rÆ } zpt|jƒtdddƒkr€dt |ƒkr€|jrvt‚W Y ¢>d S |jtdddƒkr²d	t |ƒkr²|jr¨t‚W Y ¢d S |‚W 5 d }~X Y nX | |j¡ | |j¡\}}|j�r|  ||j¡ |  ||j¡ |  |¡ d S )
NzWycheproof Encrypt CCM Test #rL   r6   r5   rK   úLength of parameter 'nonce'rM   rN   úParameter 'mac_len')ÚstrÚidrÄ   r   r   r¹   r   ÚivrÉ   rÃ   r7   r)   r8   ÚvalidÚAssertionErrorr^   ÚaadrO   Úmsgr!   r%   r   rÏ   )r"   r·   r#   Úer%   r   r   r   r   Útest_encryptz  s&    ÿ
"



z"TestVectorsWycheproof.test_encryptc              
   C   s4  dt |jƒ | _z(tj|jtj|jfd|ji| j	—Ž}W nŽ t
k
rÆ } zpt|jƒtdddƒkr€dt |ƒkr€|jrvt‚W Y ¢>d S |jtdddƒkr²d	t |ƒkr²|jr¨t‚W Y ¢d S |‚W 5 d }~X Y nX | |j¡ z| |j|j¡}W n" t
k
�r
   |j�rt‚Y n&X |j�st‚|  ||j¡ |  |¡ d S )
NzWycheproof Decrypt CCM Test #rL   r6   r5   rK   rÑ   rM   rN   rÒ   )rÓ   rÔ   rÄ   r   r   r¹   r   rÕ   rÉ   rÃ   r7   r)   r8   rÖ   r×   r^   rØ   rU   r%   r   r!   rÙ   rÏ   )r"   r·   r#   rÚ   r$   r   r   r   Útest_decrypt�  s*    ÿ
"



z"TestVectorsWycheproof.test_decryptc                 C   sž   dt |jƒ | _t|jƒtdddƒks4t|jƒdkr8d S tj|j	tj
|jfd|ji| j—Ž}| |j¡ t|jdt|jƒd  d	 ƒ}|  t|j||j¡ d S )
Nz%Wycheproof Corrupt Decrypt CCM Test #r6   r5   rK   r   rL   ó    r3   ó   )rÓ   rÔ   rÄ   r)   rÕ   r8   r%   r   r   r¹   r   rÉ   rÃ   r^   rØ   r
   r.   r7   rU   r   )r"   r·   r#   Z
ct_corruptr   r   r   Útest_corrupt_decrypt©  s    $ÿz*TestVectorsWycheproof.test_corrupt_decryptc                 C   s.   | j D ]"}|  |¡ |  |¡ |  |¡ qd 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                 C   sD   |   d¡}g }|ttƒ7 }|ttƒ7 }|tƒ g7 }|t|ƒg7 }|S )NrÅ   )Úgetr   r   r�   r¨   r½   )ÚconfigrÅ   Ztestsr   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Crypto.SelfTest.st_commonr   ZCrypto.SelfTest.loaderr   ZCrypto.Util.py3compatr   r   ZCrypto.Cipherr   ZCrypto.Hashr	   rT   r
   ZCrypto.Cipher._mode_ccmr   r   rÀ   r   r�   r¨   r½   râ   r�   rä   Úmainr   r   r   r   Ú<module>   s,      0 8 L[