Sindbad~EG File Manager
a
��g� �
@ s� d dl Z d dlZd dlmZ z&eddd�r4e�d��d dlZdZW nH ey� Z z0e j dkrje�e
e���dZe
e�ZW Y dZ[n
dZ[0 0 e� e j dkd�e� ed�G d d
� d
ej
���Ze� ed �G d d
� d
ej
��Zedkr�e�� dS )� N)�check_sanitizerT)�addressZmemoryz.The crypt module SEGFAULTs on ASAN/MSAN builds�win32z This should only run on windowszimport succeededc @ s e Zd Zdd� ZdS )�TestWhyCryptDidNotImportc C s | � dt� d S )Nz
not supported)�assertIn� IMPORT_ERROR��self� r
�+/usr/local/lib/python3.9/test/test_crypt.py�test_import_failure_message s z4TestWhyCryptDidNotImport.test_import_failure_messageN)�__name__�
__module__� __qualname__r r
r
r
r r s r zcrypt module is requiredc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Ze�e oFe j
e j v pFe j e j v d �d
d � �Z
e�e ofe je j v d �d
d� �Zdd� ZdS )�
CryptTestCasec C sR t � d�}t � d|�}| �||� t � dd�}|d urNt � d|�}| �||� d S )N�
mypassword�ab)�crypt� assertEqual)r �cr�cr2r
r
r �
test_crypt s
zCryptTestCase.test_cryptc C sl | � ttj�d� tjD ]N}t�|�}| �t|�|j h d�� |jr| �|j|d t|�|j � � qd S )N�@ > r � � � � � ) r �lenr Z
_saltchars�methods�mksaltr �
salt_chars�ident)r �method�saltr
r
r � test_salt& s
zCryptTestCase.test_saltc C sj t jD ]^}t � d|�}| �t|�|j� t � d|�}| �||� t � dt �|��}| �t|�|j� qd S )NZassword)r r r r Z
total_sizer )r r# r r r
r
r �test_saltedcrypt. s
z CryptTestCase.test_saltedcryptc C sL | � ttj�dk� tj�d�r4| �tjtjg� n| �tjd tj � d S )Nr Zopenbsd���)
�
assertTruer r r �sys�platform�
startswithr �METHOD_BLOWFISH� METHOD_CRYPTr r
r
r � test_methods7 s zCryptTestCase.test_methodszrequires support of SHA-2c C s� t jt jfD ]|}dD ]r}t j||d�}| �d| |� | �t|�|j dtt|�� � t � d|�}| � |� t � d|�}| �||� qq d S )N)�� i' i�� ��roundsz $rounds=%d$� r )
r �
METHOD_SHA256�
METHOD_SHA512r r r r r! �strr( )r r# r1 r$ r r r
r
r �test_sha2_rounds> s �
z CryptTestCase.test_sha2_roundsz requires support of Blowfishc C s� t dd�D ]r}tjtjd|> d�}| �d| |� | �t|�tjj ddh� t�d|�}| �|� t�d|�}| �||� q
d S ) Nr r2 r r0 z$%02d$r r r ) �ranger r r, r r r! r( r )r Z
log_roundsr$ r r r
r
r �test_blowfish_roundsQ s
z"CryptTestCase.test_blowfish_roundsc
C sd t jt jt jfD ]�}| �t�� t j|dd� W d � n1 sB0 Y | �t�� t j|dd� W d � n1 sz0 Y ddddd> fD ]<}| �t�� t j||d� W d � q�1 s�0 Y q�q| �t�� t jt jdd� W d � n1 �s0 Y t jt j fD ]B}| �
td �� t j|d
d� W d � n1 �sR0 Y �q d S ) NZ4096r0 g �@r r r' i� r/ �supporti ) r r3 r4 r, Z assertRaises� TypeErrorr �
ValueErrorr- Z
METHOD_MD5ZassertRaisesRegex)r r# r1 r
r
r �test_invalid_rounds^ s � , , 0 0z!CryptTestCase.test_invalid_roundsN)r
r r r r% r& r. �unittest�
skipUnlessr r3 r r4 r6 r, r8 r< r
r
r
r r s �
�
r �__main__)r) r= Z test.supportr ZSkipTestr r � ImportError�exr* r5 r> ZskipIfZTestCaser r r
�mainr
r
r
r �<module> s&
T
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists