Your IP : 216.73.216.85
a
XC?h� �
@ 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)ZaddressZmemoryz.The crypt module SEGFAULTs on ASAN/MSAN buildsZwin32z 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 �4/opt/alt/python39/lib64/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�
mypasswordZab)�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_charsZident)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 r- 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 r. r r, 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 ) NZ4096r, g �@r r r# i� r+ Zsupporti ) r r/ r0 r( Z assertRaises� TypeErrorr �
ValueErrorr) Z
METHOD_MD5ZassertRaisesRegex)r r r- r r r �test_invalid_rounds^ s � , , 0 0z!CryptTestCase.test_invalid_roundsN)r r r
r r! r" r* �unittest�
skipUnlessr r/ r r0 r2 r( r4 r7 r r r r r s �
�
r �__main__)r% r8 Z test.supportr ZSkipTestr r � ImportErrorZexr& r1 r9 ZskipIfZTestCaser r r �mainr r r r �<module> s&
T