File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/Cryptodome/SelfTest/Cipher/__pycache__/test_CBC.cpython-310.pyc
Back
o ;j�N � @ sl d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z d dl m Z m Z m Z d dlmZ dd � ZG d d � d e j�ZG d d � d e�ZG dd� de j�ZG dd� de�ZdZdZeD ]Zefdd�Zeede e� qZeD ]Zefdd�Zeede e� qm[[dZeD ]Zefdd�Zeede e� q�G dd � d e j�Z i fd d �Z e d kr�d d!� Z e j d"d#� dS dS )$� N)� unhexlify)�load_test_vectors)�list_test_cases)�tobytes� is_string)�AES�DES3�DES)�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_CBC.py�get_tag_random) s r c @ s� e Zd Zedd�Zedd�Zedd�Zedd�Zedd�Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3S )4�BlockChainingTests�key_128� �key_192� �iv_128�iv_64� �data_128c C �V t �| j| j| j�}tdd�}|�|�}t �| j| j| j�}|�|�}| �||� d S )N� plaintext�@ ) r r r �aes_moder r �encrypt�decrypt� assertEqual��self�cipher�pt�ctZpt2r r r �test_loopback_1284 � z$BlockChainingTests.test_loopback_128c C r )Nr i ) r r r � des3_moder r r r! r" r# r r r �test_loopback_64= r) z#BlockChainingTests.test_loopback_64c C s� t �| j| j�}|j}t �| j| j�}|j}| �||� | �t|�d� t �| j| j| j�}|� | j �}t j| j| j| jd�}| �||� | j �� t j| j| j| jd�}| �||� | j �� d S )Nr ��iv)�IV) r r r r r- ZassertNotEqualr" �lenr r r )r$ r% Ziv1Ziv2r' r r r �test_ivF s zBlockChainingTests.test_ivc C s | j ttj| j| jdd� d S )N�test1234567890-*r, )� assertRaises� TypeErrorr r r r �r$ r r r �test_iv_must_be_bytesY s �z(BlockChainingTests.test_iv_must_be_bytesc C s$ | j ttj| j| j| j| jd� d S )N)r- r. �r2 r3 r r r r r r4 r r r �test_only_one_iv] s �z#BlockChainingTests.test_only_one_ivc C s\ | � ttj| j| jd� | � ttj| j| j| jd d� � | � ttj| j| j| jd � d S )N� � � 0)r2 � ValueErrorr r r r r r4 r r r � test_iv_with_matching_lengthb s � ��z/BlockChainingTests.test_iv_with_matching_lengthc C �( t �| j| j| j�}| �|jt j� d S �N)r r r r r r" � block_size�r$ r% r r r �test_block_size_128j � z&BlockChainingTests.test_block_size_128c C r= r> )r r r r* r r" r? r@ r r r �test_block_size_64n rB z%BlockChainingTests.test_block_size_64c C �t t �| j| j| j�}tdd�D ] }| �t|jd| � qt �| j| j| j�}tdd�D ] }| �t|j d| � q+d S )N� r � 5) r r r r r �ranger2 r; r r! �r$ r% Z wrong_lengthr r r �test_unaligned_data_128r � �z*BlockChainingTests.test_unaligned_data_128c C rD )NrE r rF ) r r r r* r rG r2 r; r r! rH r r r �test_unaligned_data_64{ rJ z)BlockChainingTests.test_unaligned_data_64c C sZ t dd�}dD ]#}t�| j| j| j�}t||�|� | �|j| j� | �|j | j� qd S )Nr r �r r! ) r r r r r r �getattrr" r- r. )r$ r �funcr% r r r �test_IV_iv_attributes� s �z(BlockChainingTests.test_IV_iv_attributesc C sV | � ttj| j| j| jd� | j ttj| j| j| jdd� tj| j| j| jdd� d S )N� )r- �unknownF)r- Z use_aesnir6 r4 r r r �test_unknown_parameters� s �� z*BlockChainingTests.test_unknown_parametersc C s< dD ]}t �| j| j| j�}t||�d�}| �|d� qd S )NrL r8 )r r r r r rM r" )r$ rN r% �resultr r r � test_null_encryption_decryption� s �z2BlockChainingTests.test_null_encryption_decryptionc C s` t �| j| j| j�}|�d� | �t|jd� t �| j| j| j�}|�d� | �t|jd� d S )Nr8 ) r r r r r r r2 r3 r! r@ r r r � test_either_encrypt_or_decrypt� s z1BlockChainingTests.test_either_encrypt_or_decryptc C sL t �| j| j| j�}| �t|jd� t �| j| j| j�}| �t|jd� d S )Nr1 ) r r r r r r2 r3 r r! r@ r r r �test_data_must_be_bytes� s z*BlockChainingTests.test_data_must_be_bytesc C s d}t |�}t | j�}t | j�}t�| j| j| j�}|�|�}t�|| j|�}d|d d�< d|d d�< |�|�}| �||� | �|j|j� t | j�}t | j�}t�| j| j| j�} | � |�} t�|| j|�} d|d d�< d|d d�< | � |�} | �| | � d S �Ns� 11111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111s ���� ) � bytearrayr r r r r r r" r- r! ) r$ r Zdata_baZkey_baZiv_ba�cipher1�ref1�cipher2�ref2�cipher3�ref3�cipher4�ref4r r r �test_bytearray� s* z!BlockChainingTests.test_bytearrayc C s d}t t|��}t t| j��}t t| j��}t�| j| j| j�}|�|�}t�|| j|�}d|d d�< d|d d�< |�|�}| �||� | �|j |j � t t| j��}t t| j��}t�| j| j| j�} | � |�} t�|| j|�} d|d d�< d|d d�< | � |�} | �| | � d S rW ) � memoryviewrY r r r r r r r" r- r! ) r$ r Zdata_mvZkey_mvZiv_mvrZ r[ r\ r] r^ r_ r` ra r r r �test_memoryview� s* z"BlockChainingTests.test_memoryviewc C s� d}t jd| j| jd�}|�|�}td�}t jd| j| jd�}|j||d�}| �||� | �|d � t jd| j| jd�}|j||d�}| �||� | �|d � d S �N� 55555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555555� 4444444444444444r, � ��output�r r r r r rY r" r! )r$ r& r% r' rj �resr r r �test_output_param� s z$BlockChainingTests.test_output_paramc C s� d}t jd| j| jd�}|�|�}t|�}t jd| j| jd�}|j||d�}| �||� | �|d � t|�}t jd| j| jd�}|j||d�}| �||� | �|d � d S )Nrf rg r, ri rk )r$ r& r% r' Zpt_barl Zct_bar r r � test_output_param_same_buffer� s z0BlockChainingTests.test_output_param_same_bufferc C s� d}t jd| j| jd�}|�|�}ttd��}t jd| j| jd�}|j||d� | �||� t jd| j| jd�}|j||d� | �||� d S re ) r r r r r rc rY r" r! )r$ r& r% r' rj r r r � test_output_param_memoryview s z/BlockChainingTests.test_output_param_memoryviewc C s� d}d| }t jd| j| jd�}|�|�}t jd| j| jd�}| jt|j|d| d� t jd| j| jd�}| jt|j|d| d� t|d �}t jd| j| jd�}| jt |j||d� t jd| j| jd�}| jt |j||d� d S )Nrh rF rg r, r: ri rE ) r r r r r r2 r3 r! rY r; )r$ ZLEN_PTr& r% r' Zshorter_outputr r r �test_output_param_neg s z(BlockChainingTests.test_output_param_negN) �__name__� __module__� __qualname__r r r r r r r( r+ r0 r5 r7 r< rA rC rI rK rO rR rT rU rV rb rd rm rn ro rp r r r r r , s6 !! r c @ s e Zd ZejZejZdS )�CbcTestsN)rq rr rs r �MODE_CBCr r r* r r r r rt 4 s rt c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�NistBlockChainingVectorsc C s� t d|dddd� i�}|d u rd S d }|D ]8}t|�r |}q|j| _t�|j| j|j�}|dkr<| � |� |j �|j � q|dkrL| � |� |j �|j � qJ �d S )N��Cipherr z AES CBC KAT�countc S � t | �S r> ��int��xr r r �<lambda>@ � z;NistBlockChainingVectors._do_kat_aes_test.<locals>.<lambda>� [ENCRYPT]� [DECRYPT])r r �desc� descriptionr r �keyr r- r" r r � ciphertextr! )r$ � file_name� test_vectors� direction�tvr% r r r �_do_kat_aes_test; s( ��z)NistBlockChainingVectors._do_kat_aes_testc C s� t d|dddd� i�}|d u rd S d }|D ]f}t|�r |}q|j| _t�|j| j|j�}|dkrS|jg}t d�D ]}|� |� |j �� |d |_ q8| � |d |j� q|d krz|jg}t d�D ]}|� |�|j�� |d |_q_| � |d |j � qJ �d S ) Nrw zAES CBC Montecarlory c S rz r> r{ r} r r r r \ r� z;NistBlockChainingVectors._do_mct_aes_test.<locals>.<lambda>r� i� ������r� )r r r� r� r r r� r r- rG �appendr r r" r� r! ) r$ r� r� r� r� r% Zctsry Zptsr r r �_do_mct_aes_testW s8 � �z)NistBlockChainingVectors._do_mct_aes_testc C s� t d|dddd� i�}|d u rd S d }|D ]]}t|�r |}q|j| _t|d�r2t�|j| j|j �}n |j |j krB|j |j |j }n|j |j }t �|| j|j �}|dkra| �|�|j�|j� q|dkrq| �|�|j�|j� qJ �d S ) N)rx ZTDESz TDES CBC KATry c S rz r> r{ r} r r r r r� z8NistBlockChainingVectors._do_tdes_test.<locals>.<lambda>�keysr� r� )r r r� r� �hasattrr r r� �des_moder- Zkey1Zkey3Zkey2r r* r" r r r� r! )r$ r� r� r� r� r% r� r r r � _do_tdes_testz s2 � �z&NistBlockChainingVectors._do_tdes_testN)rq rr rs r� r� r� r r r r rv 9 s #rv c @ s e Zd ZejZejZejZ dS )�NistCbcVectorsN) rq rr rs r ru r r r� r r* r r r r r� � s r� )zCBCGFSbox128.rspzCBCGFSbox192.rspzCBCGFSbox256.rspzCBCKeySbox128.rspzCBCKeySbox192.rspzCBCKeySbox256.rspzCBCVarKey128.rspzCBCVarKey192.rspzCBCVarKey256.rspzCBCVarTxt128.rspzCBCVarTxt192.rspzCBCVarTxt256.rspz CBCMMT128.rspz CBCMMT192.rspz CBCMMT256.rsp)z CBCMCT128.rspz CBCMCT192.rspz CBCMCT256.rspc C � | � |� d S r> )r� �r$ r� r r r �new_func� � r� Z test_AES_c C r� r> )r� r� r r r r� � r� )z TCBCMMT2.rspz TCBCMMT3.rspzTCBCinvperm.rspzTCBCpermop.rspzTCBCsubtab.rspzTCBCvarkey.rspzTCBCvartext.rspc C r� r> )r� r� r r r r� � r� Z test_TDES_c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �SP800TestVectorszPClass exercising the CBC test vectors found in Section F.2 of NIST SP 800-3Ac C �x d}d}d}d}t |�}t |�}t |�}t |�}t�|tj|�}| �|�|�|� t�|tj|�}| �|�|�|� d S )NZ 2b7e151628aed2a6abf7158809cf4f3c� 000102030405060708090a0b0c0d0e0fڀ6bc1bee22e409f96e93d7e117393172aae2d8a571e03ac9c9eb76fac45af8e5130c81c46a35ce411e5fbc1191a0a52eff69f2445df4f9b17ad2b417be66c3710Z�7649abac8119b246cee98e9b12e9197d5086cb9b507219ee95db113a917678b273bed6b8e3c1743b7116e69e222295163ff1caa1681fac09120eca307586e1a7�r r r ru r" r r! �r$ r� r- r r� r% r r r � test_aes_128� � z SP800TestVectors.test_aes_128c C r� )NZ08e73b0f7da0e6452c810f32b809079e562f8ead2522c6b7br� r� Z�4f021db243bc633d7178183a9fa071e8b4d9ada9ad7dedf4e5e738763f69145a571b242012fb7ae07fa9baac3df102e008b0e27988598881d920a9e64f5615cdr� r� r r r � test_aes_192� r� z SP800TestVectors.test_aes_192c C r� )NZ@603deb1015ca71be2b73aef0857d77811f352c073b6108d72d9810a30914dff4r� r� Z�f58c4c04d6e5f1ba779eabfb5f7bfbd69cfc4e967edb808d679f777bc6702c7d39f23369a9d9bacfa530e26304231461b2eb05e2c39be9fcda6c19078c6a9d1br� r� r r r � test_aes_256 r� z SP800TestVectors.test_aes_256N)rq rr rs �__doc__r� r� r� r r r r r� � s r� c C s6 g }|t t�7 }| �d�r|t t�7 }|t t�7 }|S )NZ slow_tests)r rt �getr� r� )ZconfigZtestsr r r � get_tests! s r� �__main__c C s t �t� �S r> )�unittestZ TestSuiter� r r r r r + s r �suite)Z defaultTest)!r� �binasciir ZCryptodome.SelfTest.loaderr Z Cryptodome.SelfTest.st_commonr ZCryptodome.Util.py3compatr r ZCryptodome.Cipherr r r ZCryptodome.Hashr r ZTestCaser rt rv r� Znist_aes_kat_mmt_filesZnist_aes_mct_filesr� r� �setattrZnist_tdes_filesr� r� rq r� �mainr r r r �<module> sB d G �
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings