File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/crypto.cpython-39.opt-1.pyc
a
HZ�h� � @ s` d dl mZ ddlmZmZmZ ddlmZmZmZ ddlm Z ddl
m Z G dd � d e �Z d
S ) � )�hexlify� )�AES256_CTR_HMAC_SHA256� UNENCRYPTED�IntegrityError)�
bytes_to_long� bytes_to_int�
long_to_bytes)�hkdf_hmac_sha512� )� BaseTestCasec @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�CryptoTestCasec C s | � td�d� d S )Ns r )� assert_equalr ��self� r �;/usr/lib64/python3.9/site-packages/borg/testsuite/crypto.py�test_bytes_to_int s z CryptoTestCase.test_bytes_to_intc C s$ | � td�d� | � td�d� d S )Ns r )r r r r r r r �test_bytes_to_long s z!CryptoTestCase.test_bytes_to_longc C sT d}d}d}t d d |dd�}|j||d�}| �||| � |�|�}| �||� d S )N� s datas header� )�
header_len��header)r �encryptr �decrypt)r �iv�datar �csZenvelopeZgot_datar r r �test_UNENCRYPTED s
z CryptoTestCase.test_UNENCRYPTEDc s8 d}d}d}d}d}t |||ddd�� � j||d�}|dd� }|dd � }|d d
� }|d
d � } | �t|�d � | �t|�d � | �t|�d
� | �t| �d� | �� �� d� t ||t|�dd�� � �|�}
| �||
� | �� �� d� t ||t|�dd�� |d d
� d |dd � �| �t� �fdd�� d S )N� YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYYY� XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXr � foofoofoofoofoofoofoofoofoofoo� Br �r Z
aad_offsetr �! �) s 42s@ af90b488b0cc4a8f768fe2d6814fa65aec66b148135e54f7d4d29a27f22f57a8� 0000000000000000�< c6efb702de12498f34a2c2bbc8149e759996d08bf6dc5c610aefc0c3a466r � �* c s
� � ��S �N�r r �r Zhdr_mac_iv_cdata_corruptedr r �<lambda>< r z<CryptoTestCase.test_AES256_CTR_HMAC_SHA256.<locals>.<lambda>� r r r r Znext_iv�lenr Z
assert_raisesr � r Zmac_keyZenc_keyr r r Zhdr_mac_iv_cdataZhdrZmacZcdataZpdatar r- r �test_AES256_CTR_HMAC_SHA256 s2
�z*CryptoTestCase.test_AES256_CTR_HMAC_SHA256c s8 d}d}d}d}d}t |||ddd�� � j||d �}|dd� }|dd
� }|d
d � }|d d � } | �t|�d � | �t|�d
� | �t|�d� | �t| �d� | �� �� d� t ||t|�dd�� � �|�}
| �||
� | �� �� d� t ||t|�dd�� |d d� d |dd � �| �t� �fdd�� d S )Nr r! r r" s 4V� r r$ r �# �+ s 123456s@ 7659a915d9927072ef130258052351a17ef882692893c3850dd798c03d2dd138r' r( r r) c s
� � ��S r+ r, r r- r r r. Y r z@CryptoTestCase.test_AES256_CTR_HMAC_SHA256_aad.<locals>.<lambda>r/ r1 r r- r � test_AES256_CTR_HMAC_SHA256_aad> s2
�z.CryptoTestCase.test_AES256_CTR_HMAC_SHA256_aadc C s. d}t �d�}t �d�}d}t||||�}d S )Ns �000102030405060708090a0b0c�f0f1f2f3f4f5f6f7f8f9r* ��bytes�fromhexr
�r ZikmZsalt�info�lZokmr r r �test_hkdf_hmac_sha512^ s
z$CryptoTestCase.test_hkdf_hmac_sha512c C s4 t �d�}t �d�}t �d�}d}t||||�}d S )NZ�000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f404142434445464748494a4b4c4d4e4fZ�606162636465666768696a6b6c6d6e6f707172737475767778797a7b7c7d7e7f808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a3a4a5a6a7a8a9aaabacadaeafZ�b0b1b2b3b4b5b6b7b8b9babbbcbdbebfc0c1c2c3c4c5c6c7c8c9cacbcccdcecfd0d1d2d3d4d5d6d7d8d9dadbdcdddedfe0e1e2e3e4e5e6e7e8e9eaebecedeeeff0f1f2f3f4f5f6f7f8f9fafbfcfdfeff�R r9 r< r r r �test_hkdf_hmac_sha512_2g s
z&CryptoTestCase.test_hkdf_hmac_sha512_2c C s( t �d�}d }d}d}t||||�}d S )NZ,0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0br r* r9 r< r r r �test_hkdf_hmac_sha512_3t s
z&CryptoTestCase.test_hkdf_hmac_sha512_3c C s4 t �d�}t �d�}t �d�}d}t||||�}d S )NZ0b0b0b0b0b0b0b0b0b0b0br7 r8 r* r9 r< r r r �test_hkdf_hmac_sha512_4} s
z&CryptoTestCase.test_hkdf_hmac_sha512_4c C s( t �d�}d }d}d}t||||�}d S )NZ,0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0cr r* r9 r< r r r �test_hkdf_hmac_sha512_5� s
z&CryptoTestCase.test_hkdf_hmac_sha512_5N)
�__name__�
__module__� __qualname__r r r r2 r6 r? rA rB rC rD r r r r r
s
r
N)
Zbinasciir Zcrypto.low_levelr r r r r r r
� r r
r r r r �<module> s