o
    5hQ                     @   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rtdd Ze jdd dS 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   s/var/www/html/construction_image-detection-poc/venv/lib/python3.10/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 )N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 q4d 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 )Nr7   )r   unknownF)r   	use_aesni)r/   r0   r   r   r   r   r   r1   r   r   r   test_unknown_parametersk   s   
z CcmTests.test_unknown_parametersc                 C   s>   dD ]}t j| jt j| jd}t||d}| |d qd S )Nr   r    r   r3   )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   r3   )	r   r   r   r   r   r   r/   r0   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/   r0   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   )r9   r/   r8   r   r   r   r   r   encrypt_and_digestr   r!   r)   )r"   rN   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   r4   )Crypto.Util.strxorrU   r   r   r   r   r   rQ   r   r/   r8   decrypt_and_verify)r"   rU   r#   r%   rS   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   	hexverify)r"   r#   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/   r8   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 )Nr4   r_   r   )r)   r   r   r   r   r   r   rc   r/   r8   r   r[   r    verify)r"   DATA_LENr#   rS   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 )Nr4   r   msg_len)r)   r   r   r   r   r   r   r   r/   r8   r[   )r"   rf   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   r4   rh   )r)   r   r   r   r   r   r   rQ   r    r/   r8   re   )r"   rf   r#   r%   rS   r   r   r   %test_shorter_ciphertext_than_declared   s   

z.CcmTests.test_shorter_ciphertext_than_declaredc                 C   s@  t dd}t dd}tj| jtj| jd}|| ||\}}dd }dD ]8}tj| jtj| jddd}|||D ]}|| q;d	}	|||D ]	}|	||7 }	qJ| 	||	 |
| q'dD ];}tj| jtj| jddd}|||D ]}|| qvd	}
|||D ]	}|
||7 }
q| 	||
 | 	| | qbd 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   )r9   r)   )r   rp   r   ro   r   break_up  s   z.CcmTests.test_message_chunks.<locals>.break_up)
r4   rM   rK   r7   
      r   (   P   r   r   ri   r`   r3   )r   r   r   r   r   r   rc   rQ   r    r!   re   r   r[   )r"   	auth_datar   r#   
ciphertextref_macrr   rp   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   rK   )	bytearrayr   r   r   r   r   r   rc   r   r[   r!   r   rW   )r"   key_banonce_ba	header_ba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 r~   )
memoryviewr   r   r   r   r   r   r   rc   r   r[   r!   r   rW   )r"   key_mvnonce_mv	header_mvdata_mvr   r%   r   r   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   r[   r   r!   r    rQ   rW   )r"   r$   r#   r%   r   r   res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   r   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   rb   )r   r   r   r   r   r   r/   r0   r    r   r8   )r"   r$   r#   r%   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`   ri   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-   r2   r;   r?   r@   rC   rH   rI   rJ   rT   rY   r^   rd   rg   rj   rk   r}   r   r   r   r   r   r   r   r   r   r   r   1   s8    


	

	+//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 ]:}d t | jfD ]0}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   r[   r    re   r"   r`   ri   r#   r%   rS   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 ]6}dD ]1}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   rc   r[   re   )r"   r`   ri   r#   rS   r   r   r   $test_valid_init_update_digest_verify  s$   z0CcmFSMTests.test_valid_init_update_digest_verifyc              	   C   s   d t | jfD ]F}d t | jfD ]<}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   rc   r   r[   r    re   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   r[   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   r[   re   )r"   r#   rS   r   r   r   test_valid_init_verify'  s   z"CcmFSMTests.test_valid_init_verifyc              	   C   s   dD ]I}d d| j | j d fD ]<}|d u rd }nt|}tj| jtj| jd|d}|d ur1|| t||}|| j  || j  || j  || j  qqd S )NrD   s   333   3@   rw   )	r   r)   r   r   r   r   r   rc   rE   )r"   method_namerx   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 ]
}| 	||  qt j| jt j| jd}|| j tdD ]}|
| q9d S )Nr   rO      )r   r   r   r   r   rc   r   r[   r9   r!   re   )r"   r#   	first_macr:   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   rc   r   rQ   rW   r!   )r"   r#   r%   rS   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 ],}dD ]'}t j| jt j| jd}|r|| j t||}|| j | t	|| j qqd S )NrD   TFr   
r   r   r   r   r   rc   r   rE   r/   r0   )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 ]0\}}dD ])}t j| jt j| jdd}|r|| j t||| j | t	t||| j qqd S )N)rD   )r    r   r       rh   r   )r"   method1_namemethod2_namer   r#   r   r   r   #test_invalid_mixing_encrypt_decryptj  s   z/CcmFSMTests.test_invalid_mixing_encrypt_decryptc                 C   sp   dD ]3}t j| jt j| jd}|| j |  | t	t
||| j t j| jt j| jd}|| j qd S )N)r   rc   r   )r   r   r   r   r   r   r   r[   r/   r0   rE   rQ   )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    rc   )r   r   r   r   r   r   r   r[   r    re   r/   r0   rE   rW   )r"   r#   r%   rS   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)0001020304050607202122237162015b4dac255d 404142434445464748494a4b4c4d4e4f10111213141516) 000102030405060708090a0b0c0d0e0f 202122232425262728292a2b2c2d2e2f d2a1f0e051ea5f62081a7792073d593d1fc64fbfaccdr   1011121314151617)(000102030405060708090a0b0c0d0e0f101112130202122232425262728292a2b2c2d2e2f30313233343536370e3b201a9f5b71a7a9b1ceaeccd97e70b6176aad9a4428aa5484392fbc1b09951r   101112131415161718191a1b c                 C   s,   g | ]}t d dD ]
}d|d |  q	qS )r   r   z%02X)r9   )rm   r:   yr   r   r   rq     s   , zTestVectors.<listcomp>r   r      @202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f@69915dad1e84c6376a68c2967e4dab615ae0fd1faec44cc484828529463ccf72b4ac6bec93e8598e7f0dadbcea5br   101112131415161718191a1b1c)r   .08090a0b0c0d0e0f101112131415161718191a1b1c1d1e.588c979a61c663d2f066d0c2c0f989806d5f6b61dac38417e8d12cfdf926e0 c0c1c2c3c4c5c6c7c8c9cacbcccdcecf00000003020100a0a1a2a3a4a5)r   008090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f072c91a36e135f8cf291ca894085c87e3cc15c439c9e43a3ba091d56e10400916r   00000004030201a0a1a2a3a4a5)r   208090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f20251b1e5f44a197d1da46b0f8e2d282ae871e838bb64da8596574adaa76fbd9fb0c5r   00000005040302A0A1A2A3A4A5)000102030405060708090a0b&0c0d0e0f101112131415161718191a1b1c1d1e&a28c6865939a9a79faaa5c4c2a9d4a91cdac8c96c861b9c9e61ef1r   00000006050403a0a1a2a3a4a5)r   (0c0d0e0f101112131415161718191a1b1c1d1e1f(dcf1fb7b5d9e23fb9d4e131253658ad86ebdca3e51e83f077d9c2d93r   00000007060504a0a1a2a3a4a5)r   *0c0d0e0f101112131415161718191a1b1c1d1e1f20*6fc1b011f006568b5171a42d953d469b2570a4bd87405a0443ac91cb94r   00000008070605a0a1a2a3a4a5)r   r   .0135d1b2c95f41d5d1d4fec185d166b8094e999dfed96c048c56602c97acbb7490r   00000009080706a0a1a2a3a4a5)r   r   07b75399ac0831dd2f0bbd75879a2fd8f6cae6b6cd9b7db24c17b4433f434963f34b4r   0000000a090807a0a1a2a3a4a5)r   r   282531a60cc24945a4b8279181ab5c84df21ce7f9b73f42e197ea9c07e56b5eb17e5f4er   0000000b0a0908a0a1a2a3a4a5)r   r   &07342594157785152b074098330abb141b947b566aa9406b4d999988ddr   0000000c0b0a09a0a1a2a3a4a5)r   r   (676bb20380b0e301e8ab79590a396da78b834934f53aa2e9107a8b6c022cr   0000000d0c0b0aa0a1a2a3a4a5)r   r   *c0ffa0d6f05bdb67f24d43a4338d2aa4bed7b20e43cd1aa31662e7ad65d6dbr   0000000e0d0c0ba0a1a2a3a4a5)0be1a88bace018b1.08e8cf97d820ea258460e96ad9cf5289054d895ceac47c.4cb97f86a2a4689a877947ab8091ef5386a6ffbdd080f8e78cf7cb0cddd7b3 d7828d13b2b0bdc325a76236df93cc6b00412b4ea9cdbe3c9696766cfa)63018f76dc8a1bcb09020ea6f91bdd85afa0039ba4baff9bfb79c7028949cd0ec04ccb1e7ca981befaa0726c55d378061298c85c92814abc33c52ee81d7d77c08ar  0033568ef7b2633c9696766cfa)aa6cfa36cae86b402b916e0eacc1c00d7dcec68ec0b3bbb1a02de8a2d1aa346132e2b1d23a2220ddc0ac900d9aa03c61fcf4a559a4417767089708a776796edb723506r  00103fe41336713c9696766cfa)d0d0735c531e1becf049c244&12daac5630efa5396f770ce1a66b21f7b2101c&14d253c3967b70609b7cbb7c499160283245269a6f49975bcadeafr  00764c63b8058e3c9696766cfa)77b60f011c03e1525899bcae(e88b6a46c78d63e52eb8c546efb5de6f75e9cc0d(5545ff1a085ee2efbf52b2e04bee1e2336c73e3f762c0c7744fe7e3cr  00f8b678094e3b3c9696766cfa)cd9044d2b71fdb8120ea60c0*6435acbafb11a82e2f071d7ca4a5ebd93a803ba87f*009769ecabdf48625594c59251e6035722675e04c847099e5ae0704551r  00d560912d3f703c9696766cfa)d85bc7e69f944fb8.8a19b950bcf71a018e5e6701c91787659809d67dbedd18.bc218daa947427b6db386a99ac1aef23ade0b52939cb6a637cf9bec2408897c6bar  0042fff8f1951c3c9696766cfa)74a0ebc9069f5b3701761433c37c5a35fc1f39f406302eb907c6163be38c9843705810e6fd25874022e80361a478e3e9cf484ab04f447efff6f0a477cc2fc9bf548944r  00920f40e56cdc3c9696766cfa)44a3aa3aae6475ca2a434a8e58500c6e41530538862d686ea9e81301b5ae4226bfa2f2beed7bc5098e83feb5b31608f8e29c38819a89c8e776f1544d4151a4ed3a8b87b9cer  0027ca0c7120bc3c9696766cfa)ec46bb63b02520c33c49fd70&b96b49e21d621741632875db7f6c9243d2d7c2&31d750a09da3ed7fddd49a2032aabf17ec8ebf7d22c8088c666be5c197r  005b8ccbcd9af83c9696766cfa)47a65ac78b3d594227e85e71(e2fcfbb880442c731bf95167c8ffd7895e337076(e882f1dbd38ce3eda7c23f04dd65071eb41342acdf7e00dccec7ae52987dr  003ebe94044b9a3c9696766cfa)6e37a6ef546d955d34ab6059*abf21c0b02feb88f856df4a37381bce3cc128517d4*f32905b88a641b04b9c9ffb58cc390900f3da12ab16dce9e82efa16da62059r  008d493b30ae8b3c9696766cfac                 C   s   g | ]	}d d |D qS )c                 S   s   g | ]}t |qS r   r   )rm   r:   r   r   r   rq   N  s    z!TestVectors.<listcomp>.<listcomp>r   )rm   tvr   r   r   rq   N  s    c                 C   s   | j D ]I\}}}}}}tj|tj|t|d}|| ||\}}	| || | ||	 tj|tj|t|d}|| |||}
| ||
 qd S )N)rN   )	test_vectorsr   r   r   r)   rc   rQ   r!   rW   )r"   
assoc_datar$   r%   rS   keyr   r#   r|   mac2r&   r   r   r   runTestP  s   

zTestVectors.runTestN)	r   r   r   __doc__joinr9   test_vectors_hexr>  rB  r   r   r   r   r     sR      /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_warningsextra_paramsr   r   r   rJ  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 )NtagSize   r   )groupr   r   r   
filter_tagj  s   z/TestVectorsWycheproof.setUp.<locals>.filter_tag)Cipher
wycheproofzaes_ccm_test.jsonzWycheproof AES CCMtag_size)	group_tag)r   r=  )r"   rS  r   r   r   setUph  s   zTestVectorsWycheproof.setUpc                 C   s   | j S N)rM  r1   r   r   r   shortDescriptionr  s   z&TestVectorsWycheproof.shortDescriptionc                 C   s6   |j r| jrdd l}|d| j|jf  d S d S d S )Nr   zWycheproof warning: %s (%s))warningrK  warningswarnrM  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 nF t
yb } z:t|jtdddvrAdt |v rA|jr:J W Y d }~d S |jtdddvr\d	t |v r\|jrUJ W Y d }~d S |d }~ww ||j ||j\}}|jr| ||j | ||j | | d S d S )
NzWycheproof Encrypt CCM Test #rN   r7   r6   rM   Length of parameter 'nonce'rO   rP   Parameter 'mac_len')stridrM  r   r   r@  r   ivrV  rL  r8   r)   r9   validrc   aadrQ   msgr!   r%   r   r]  )r"   r=  r#   er%   r   r   r   r   test_encryptz  s,   
"

	z"TestVectorsWycheproof.test_encryptc              
   C   s,  dt |j | _ztj|jtj|jfd|ji| j	}W nF t
yb } z:t|jtdddvrAdt |v rA|jr:J W Y d }~d S |jtdddvr\d	t |v r\|jrUJ W Y d }~d S |d }~ww ||j z
||j|j}W n t
y   |jrJ Y d S w |jsJ | ||j | | d S )
NzWycheproof Decrypt CCM Test #rN   r7   r6   rM   r_  rO   rP   r`  )ra  rb  rM  r   r   r@  r   rc  rV  rL  r8   r)   r9   rd  rc   re  rW   r%   r   r!   rf  r]  )r"   r=  r#   rg  r$   r   r   r   test_decrypt  s0   
"

	
z"TestVectorsWycheproof.test_decryptc                 C   s   dt |j | _t|jtdddvst|jdkrd 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 #r7   r6   rM   r   rN       r4      )ra  rb  rM  r)   rc  r9   r%   r   r   r@  r   rV  rL  rc   re  r
   r/   r8   rW   r   )r"   r=  r#   
ct_corruptr   r   r   test_corrupt_decrypt  s   $z*TestVectorsWycheproof.test_corrupt_decryptc                 C   s.   | j D ]}| | | | | | qd S rY  )r=  rh  ri  rm  )r"   r=  r   r   r   rB    s
   


zTestVectorsWycheproof.runTestN)r   r   r   rJ  rX  rZ  r]  rh  ri  rm  rB  r   r   r   r   rF  `  s    

rF  c                 C   sD   |  d}g }|tt7 }|tt7 }|t g7 }|t|g7 }|S )NrN  )getr   r   r   r   rF  )configrN  testsr   r   r   	get_tests  s   
rq  __main__c                   C   s   t t  d S rY  )rH  	TestSuiterq  r   r   r   r   suite  s   rt  )defaultTest)rH  binasciir   Crypto.SelfTest.st_commonr   Crypto.SelfTest.loaderr   Crypto.Util.py3compatr   r   Crypto.Cipherr   Crypto.Hashr	   rV   r
   Crypto.Cipher._mode_ccmr   r   rI  r   r   r   rF  rq  r   rt  mainr   r   r   r   <module>   s0      0 8 L[