U
    g>                     @   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mZmZ d dlmZ d dlmZ d	d
 ZG dd deZG dd de jZdZdZeD ]8ZdD ].ZedeZeefddZeede e qqeD ]8ZdD ].ZedeZeefddZeede e qq[[dZeD ]<ZdD ] ZedeZeefddZq*eede e q"G dd de jZi fddZ e!dkrd d! Z"e j#d"d# dS )$    N)	unhexlify)load_test_vectors)list_test_cases)tobytes	is_string)AESDES3DES)SHAKE128)BlockChainingTestsc                 C   s   t jt| d|S )N)data)r
   newr   read)taglength r   G/tmp/pip-unpacked-wheel-_q8s9isk/Cryptodome/SelfTest/Cipher/test_CFB.pyget_tag_random+   s    r   c                   @   s8   e Zd ZejZejZdd Zdd Z	dd Z
dd Zd	S )
CfbTestsc                    s   dgd }t j| jt j| jdd  fdd|D }t j| jt j| jdd | d| d| t j| jt j| jdd  fd	d|D }t j| jt j| jdd | d| d| d S )
N   7777777d      segment_sizec                    s   g | ]}  |qS r   encrypt.0xcipherr   r   
<listcomp>:   s     z4CfbTests.test_unaligned_data_128.<locals>.<listcomp>       c                    s   g | ]}  |qS r   r   r   r   r   r   r!   ?   s     )r   r   key_128MODE_CFBiv_128assertEqualjoinr   selfZ
plaintextsZciphertextsr   r   r   test_unaligned_data_1286   s    
z CfbTests.test_unaligned_data_128c                    s   dgd }t j| jt j| jdd  fdd|D }t j| jt j| jdd | d| d| t j| jt j| jdd  fd	d|D }t j| jt j| jdd | d| d| d S )
Nr   r   r   r   c                    s   g | ]}  |qS r   r   r   r   r   r   r!   F   s     z3CfbTests.test_unaligned_data_64.<locals>.<listcomp>r"   @   c                    s   g | ]}  |qS r   r   r   r   r   r   r!   K   s     )r   r   key_192r%   iv_64r'   r(   r   r)   r   r   r   test_unaligned_data_64C   s    
zCfbTests.test_unaligned_data_64c              	   C   sV   t dddD ]}tj| jtj| j|d}qdD ]"}| jttj| jtj| j|d q.d S )Nr      r   )r      	      r0   )ranger   r   r$   r%   r&   assertRaises
ValueErrorr*   bitsr    r   r   r   test_segment_size_128Q   s    zCfbTests.test_segment_size_128c              	   C   sV   t dddD ]}tj| jtj| j|d}qdD ]"}| jttj| jtj| j|d q.d S )Nr   A   r   )r   r1   r2   ?   r:   )	r4   r   r   r-   r%   r.   r5   r6   r   r7   r   r   r   test_segment_size_64[   s    zCfbTests.test_segment_size_64N)__name__
__module____qualname__r   r%   Zaes_moder   Z	des3_moder+   r/   r9   r<   r   r   r   r   r   /   s   
r   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )NistCfbVectorsc                 C   s   t d|d| ddd i}|d kr&d S d }|D ]z}t|r@|}q.|j| _tj|jtj|j|d}|dkr| 	|
|j|j q.|dkr| 	||j|j q.d	s.tq.d S )
NCipherr   zAES CFB%d KATcountc                 S   s   t | S Nintr   r   r   r   <lambda>m   r"   z1NistCfbVectors._do_kat_aes_test.<locals>.<lambda>r   	[ENCRYPT]	[DECRYPT]F)r   r   descdescriptionr   r   keyr%   ivr'   r   	plaintext
ciphertextdecryptAssertionError)r*   	file_namer   test_vectors	directiontvr    r   r   r   _do_kat_aes_testh   s*    
zNistCfbVectors._do_kat_aes_testc                    s"  t d|d  ddd i}|d kr&d S  dks2td }|D ]trL}q:j| _tjjtjj	 d} fdd	}|d
krg }t
dD ]"}|j||}	|||	 q| |d j q:|dkrg }
t
dD ]"}|j|
|}|
|| q| |
d j q:ds:tq:d S )NrA   zAES CFB%d MontecarlorC   c                 S   s   t | S rD   rE   rG   r   r   r   rH      r"   z1NistCfbVectors._do_mct_aes_test.<locals>.<lambda>)r   r#   r   c                    sX    dkr&|dkr|d S | j g| S |dkr2| S |dkrLj |d | S ||d  S )Nr#      r            )rN   )Z
input_textZ
output_seqjr   rV   r   r   	get_input   s    z2NistCfbVectors._do_mct_aes_test.<locals>.get_inputrI   i  rJ   F)r   rR   r   rK   rL   r   r   rM   r%   rN   r4   rO   appendr   r'   rP   rQ   )r*   rS   r   rT   rU   r    r_   Zctsr]   rO   ZptsrP   r   r^   r   _do_mct_aes_test   s>    

zNistCfbVectors._do_mct_aes_testc                 C   s   t d|d| ddd i}|d kr&d S d }|D ]}t|r@|}q.|j| _t|drltj|jtj|j	|d}nB|j
|jkr|j
|j |j }n|j
|j }tj|tj|j	|d}|dkr| ||j|j q.|d	kr| ||j|j q.d
s.tq.d S )N)rB   ZTDESzTDES CFB%d KATrC   c                 S   s   t | S rD   rE   rG   r   r   r   rH      r"   z.NistCfbVectors._do_tdes_test.<locals>.<lambda>keysr   rI   rJ   F)r   r   rK   rL   hasattrr	   r   rc   r%   rN   key1Zkey3key2r   r'   r   rO   rP   rQ   rR   )r*   rS   r   rT   rU   rV   r    rM   r   r   r   _do_tdes_test   s8    

zNistCfbVectors._do_tdes_testN)r=   r>   r?   rW   rb   rg   r   r   r   r   r@   f   s   3r@   )zCFB?GFSbox128.rspzCFB?GFSbox192.rspzCFB?GFSbox256.rspzCFB?KeySbox128.rspzCFB?KeySbox192.rspzCFB?KeySbox256.rspzCFB?VarKey128.rspzCFB?VarKey192.rspzCFB?VarKey256.rspzCFB?VarTxt128.rspzCFB?VarTxt192.rspzCFB?VarTxt256.rspzCFB?MMT128.rspzCFB?MMT192.rspzCFB?MMT256.rsp)zCFB?MCT128.rspzCFB?MCT192.rspzCFB?MCT256.rsp)8Z128?c                 C   s   |  |t| d S rD   )rW   rF   r*   rS   r8   r   r   r   new_func   s    rk   Z	test_AES_c                 C   s   |  |t| d S rD   )rb   rF   rj   r   r   r   rk      s    )zTCFB?MMT2.rspzTCFB?MMT3.rspzTCFB?invperm.rspzTCFB?permop.rspzTCFB?subtab.rspzTCFB?varkey.rspzTCFB?vartext.rsp)rh   Z64c                 C   s   |  |t| d S rD   )rg   rF   rj   r   r   r   rk     s    Z
test_TDES_c                   @   s@   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dS )SP800TestVectorszPClass exercising the CFB test vectors found in Section F.3
    of NIST SP 800-3Ac                 C   s   d}d}d}d}t |}t |}t |}t |}tj|tj|dd}| ||| tj|tj|dd}| ||| d S )N$6bc1bee22e409f96e93d7e117393172aae2dZ$3b79424c9c0dd436bace9e0ed4586a4f32b9 2b7e151628aed2a6abf7158809cf4f3c 000102030405060708090a0b0c0d0e0fr   r   r   r   r   r%   r'   r   rQ   r*   rO   rP   rM   rN   r    r   r   r   test_aes_128_cfb8  s    z"SP800TestVectors.test_aes_128_cfb8c                 C   s   d}d}d}d}t |}t |}t |}t |}tj|tj|dd}| ||| tj|tj|dd}| ||| d S )Nrm   Z$cda2521ef0a905ca44cd057cbf0d47a0678a08e73b0f7da0e6452c810f32b809079e562f8ead2522c6b7bro   r   r   rp   rq   r   r   r   test_aes_192_cfb8,  s    z"SP800TestVectors.test_aes_192_cfb8c                 C   s   d}d}d}d}t |}t |}t |}t |}tj|tj|dd}| ||| tj|tj|dd}| ||| d S )Nrm   Z$dc1f1a8520a64db55fcc8ac554844e889700@603deb1015ca71be2b73aef0857d77811f352c073b6108d72d9810a30914dff4ro   r   r   rp   rq   r   r   r   test_aes_256_cfb8<  s    z"SP800TestVectors.test_aes_256_cfb8c                 C   s   d}d}d}d}t |}t |}t |}t |}tj|tj|dd}| ||| tj|tj|dd}| ||| d S )Nڀ6bc1bee22e409f96e93d7e117393172aae2d8a571e03ac9c9eb76fac45af8e5130c81c46a35ce411e5fbc1191a0a52eff69f2445df4f9b17ad2b417be66c3710Z3b3fd92eb72dad20333449f8e83cfb4ac8a64537a0b3a93fcde3cdad9f1ce58b26751f67a3cbb140b1808cf187a4f4dfc04b05357c5d1c0eeac4c66f9ff7f2e6rn   ro   r#   r   rp   rq   r   r   r   test_aes_128_cfb128L  s    z$SP800TestVectors.test_aes_128_cfb128c                 C   s   d}d}d}d}t |}t |}t |}t |}tj|tj|dd}| ||| tj|tj|dd}| ||| d S )Nrw   Zcdc80d6fddf18cab34c25909c99a417467ce7f7f81173621961a2b70171d3d7a2e1e8a1dd59b88b1c8e60fed1efac4c9c05f9f9ca9834fa042ae8fba584b09ffrs   ro   r#   r   rp   rq   r   r   r   test_aes_192_cfb128b  s    z$SP800TestVectors.test_aes_192_cfb128c                 C   s   d}d}d}d}t |}t |}t |}t |}tj|tj|dd}| ||| tj|tj|dd}| ||| d S )Nrw   Zdc7e84bfda79164b7ecd8486985d386039ffed143b28b1c832113c6331e5407bdf10132415e54b92a13ed0a8267ae2f975a385741ab9cef82031623d55b1e471ru   ro   r#   r   rp   rq   r   r   r   test_aes_256_cfb128x  s    z$SP800TestVectors.test_aes_256_cfb128N)
r=   r>   r?   __doc__rr   rt   rv   rx   ry   rz   r   r   r   r   rl     s   rl   c                 C   s6   g }|t t7 }| dr&|t t7 }|t t7 }|S )NZ
slow_tests)r   r   getr@   rl   )configtestsr   r   r   	get_tests  s    
r   __main__c                   C   s   t t S rD   )unittestZ	TestSuiter   r   r   r   r   rH     r"   rH   suite)ZdefaultTest)$r   binasciir   ZCryptodome.SelfTest.loaderr   ZCryptodome.SelfTest.st_commonr   ZCryptodome.Util.py3compatr   r   ZCryptodome.Cipherr   r   r	   ZCryptodome.Hashr
   Z#Cryptodome.SelfTest.Cipher.test_CBCr   r   r   ZTestCaser@   Znist_aes_kat_mmt_filesZnist_aes_mct_filesZfile_gen_namer8   replacerS   rk   setattrZnist_tdes_filesrl   r   r=   r   mainr   r   r   r   <module>   sD   7v
x	
