관리-도구
편집 파일: test_utils_handlers.cpython-39.pyc
a f�Wc�} � @ s* d Z ddlmZ ddlZddlZddlmZ ddlZddlm Z m Z ddl m Z m Z ddlmZmZmZ ddlm mZ ddlmZmZ dd lmZ ee�Zd d � Z G d d � d e�Z G dd� de �Z G dd� de�Z G dd� dej!�Z"G dd� dej#ej$�Z%ed�Z&G dd� de�Z'G dd� de�Z(dS )z>tests for passlib.hash -- (c) Assurance Technologies 2003-2009� )�with_statementN)� getLogger)�ldap_md5� sha256_crypt)�MissingBackendError�PasslibHashWarning)� str_to_uascii� uascii_to_str�unicode)� HandlerCase�TestCase)�uc s � �fdd��t �|��S )z1generate all strings of given size using alphabetc 3 sD | dk r� D ] }|V q n&� D ] }�| d �D ]}|| V q.q d S )N� � � )�size�char�tail��alphabet�helperr �E/usr/lib/python3.9/site-packages/passlib/tests/test_utils_handlers.pyr s z_makelang.<locals>.helper)�set)r r r r r � _makelang s r c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS )� SkeletonTestztest hash support classesc s� G � fdd�dt j�� | �� �td��� | �� �d�� | �� �td��� | �� �td��� | �� �d�� | �� �td��� | �� �td ��� | �� �td ��� | �t� jd � | �t� jd � | �� � � � � d �� | �� � dd�� | �� � dtd��� | �� � dd�� | �� � dtd��� | �� j dddd�� | �t � j dd� | �t � j dtd�� | �� � d�d� | �� j ddd�d� d S )ztest StaticHandler classc sD e Zd ZdZdZed�Zed�ZdZd � �fdd� Z d d � Z � Z S ) z/SkeletonTest.test_00_static_handler.<locals>.d1�d1��flag�_�abr Fc s t �| �jf i |�� || _d S �N)�super�__init__r )�selfr �kwds)� __class__r r r r" 9 s z8SkeletonTest.test_00_static_handler.<locals>.d1.__init__c S s | j rtd�S td�S )N�b�a)r r �r# �secretr r r �_calc_checksum= s z>SkeletonTest.test_00_static_handler.<locals>.d1._calc_checksum)F) �__name__� __module__� __qualname__�nameZ context_kwdsr Z _hash_prefix�checksum_chars� checksum_sizer" r* � __classcell__r �r )r% r r 2 s r Z_as _aZ_bZ_cs _cr' r& �cNr � �ss _bTr ) �uh� StaticHandler� assertTrue�identifyr � assertFalse� assertRaises� TypeError� assertEqual� genconfig�hash�verify� ValueError�r# r r2 r �test_00_static_handler/ s* z#SkeletonTest.test_00_static_handlerc C s� G dd� dt j�}| �d�� |�d�}W d � n1 s<0 Y | �|d� | �|�d|�� | �|�d|�� |`| � t |jd� dS )z test StaticHandler legacy attrc @ s4 e Zd ZdZedd� �Zedd� �Zedd� �ZdS ) z3SkeletonTest.test_01_calc_checksum_hack.<locals>.d1r c S s< |rt |�dkrdS zt|d� W n ty6 Y dS 0 dS )N�( F� T)�len�intrA ��clsr? r r r r9 f s z<SkeletonTest.test_01_calc_checksum_hack.<locals>.d1.identifyc S sP |d u rt d��t|t�r$|�d�}|d ur>| �|�s>td��t�d| ��� S )Nzno secret provided�utf-8z invalid hashs xyz) r< � isinstancer �encoder9 rA �hashlib�sha1� hexdigest�rI r) r? r r r �genhashp s z;SkeletonTest.test_01_calc_checksum_hack.<locals>.d1.genhashc S s$ |d u rt d��| �||�|�� kS )Nzno hash specified)rA rQ �lowerrP r r r r@ { s z:SkeletonTest.test_01_calc_checksum_hack.<locals>.d1.verifyN)r+ r, r- r. � classmethodr9 rQ r@ r r r r r c s r z%d1.*should be updated.*_calc_checksum�testNZ(7c622762588a0e5cc786ad0a143156f9fd38eea3Zxtest) r6 r7 �assertWarningListr? r= r8 r@ r: rQ r; �NotImplementedError)r# r r? r r r �test_01_calc_checksum_hack] s ( z'SkeletonTest.test_01_calc_checksum_hackc C s G dd� dt j�}| �t|jd� | �t|jd� | �|�d�� | �|�d�� | �|�d�� t�t d��|_ | �t|jd� | �t|jd� | �|�d �� | �|�d�� |` t d �|_ | �t|jd� | �t|jd� | �|�d �� | �|�d�� |` dS ) z test GenericHandler.identify()c @ s e Zd Zedd� �ZdS )z)SkeletonTest.test_10_identify.<locals>.d1c S s2 t |t�r|�d�}|td�kr*| |d�S t�d S )N�asciir' ��checksum)rK �bytes�decoder rA rH r r r � from_string� s z5SkeletonTest.test_10_identify.<locals>.d1.from_stringN)r+ r, r- rS r] r r r r r � s r Nr r4 r' r& z@.z@a�!z!a) r6 �GenericHandlerr; r< r9 r: r8 �re�compiler � _hash_regex�ident)r# r r r r �test_10_identify� s$ z SkeletonTest.test_10_identifyc s� G dd� dt j�� d � fdd� }| �t|td�� | �|td��td�� | �|td��td�� | �t|td �� | �t|td �� | �t|d � | �� � jtd�� dS ) z%test GenericHandler checksum handlingc @ s e Zd ZdZdZed�ZdS )z.SkeletonTest.test_11_norm_checksum.<locals>.d1r � �xzN)r+ r, r- r. r0 r r/ r r r r r � s r Nc s � f d| i|��j S )NrZ rY )rZ �kr2 r r � norm_checksum� s z9SkeletonTest.test_11_norm_checksum.<locals>.norm_checksum�xxxZxxxxZxzxzZxxxxx�xxyxs xxyx)N)r6 r_ r; rA r r= r< �_stub_checksum�r# rh r r2 r �test_11_norm_checksum� s z"SkeletonTest.test_11_norm_checksumc sX G dd� dt jt j�� � fdd�}| �|d�d� | �t|td�� | �� � jd� dS ) z*test GenericHandler + HasRawChecksum mixinc @ s e Zd ZdZdZdS )z2SkeletonTest.test_12_norm_checksum_raw.<locals>.d1r re N)r+ r, r- r. r0 r r r r r � s r c s � | i |��j S r rY )r' rg r2 r r rh � s z=SkeletonTest.test_12_norm_checksum_raw.<locals>.norm_checksums 1234rj s N)r6 ZHasRawChecksumr_ r= r; r< r rk rl r r2 r �test_12_norm_checksum_raw� s z&SkeletonTest.test_12_norm_checksum_rawc sP G dd� dt jt j�� � fdd�}� fdd�}tdd�}tdd �}tdd �}| �t|� | jt|d d � | �|d d�|� tjd d���}| jt |dd � | jt |dd � | � |� | � |dd �d� | � |dd �d� | � |dd �d� | � |� | jt |dd � | � |� W d � n1 �s"0 Y tjd d���}| �t |d� | �t |d� | � |� | �|d�|� | �|d �|� | �|d �|� | � |� | �t |d� | � |� | �|dd d�|� | � |dg� W d � n1 �s�0 Y � ` | � g ��8 | � t|d ��d � | � t|d��d� W d � n1 �sB0 Y d S )z#test GenericHandler + HasSalt mixinc @ s$ e Zd ZdZdZdZdZdZdZdS )z*SkeletonTest.test_20_norm_salt.<locals>.d1r ��saltr re � r N) r+ r, r- r. � setting_kwds� min_salt_size� max_salt_sizeZdefault_salt_size� salt_charsr r r r r � s r c s � f i | ��j S r ro �rg r2 r r � norm_salt� s z1SkeletonTest.test_20_norm_salt.<locals>.norm_saltc s � j f d| i|��dd�jS )NZ salt_sizeT�Z use_defaults)Zusingrp )Zszrg r2 r r �gen_salt� s z0SkeletonTest.test_20_norm_salt.<locals>.gen_saltr r rq re Nro Trx ��recordr4 r' ZabaZabbaZaaaabbr r � )Zrelaxedz salt_size.*above max_salt_size)r6 �HasSaltr_ r r; r< ZassertIn�warnings�catch_warningsrA �consumeWarningListr= rt rU rF )r# rw ry Zsalts2Zsalts3Zsalts4�wlogr r2 r �test_20_norm_salt� sF * . z SkeletonTest.test_20_norm_saltc s G dd� dt jt j�� � fdd�}| �t|� | jt|dd� | �|dd�d � | jt|d d� tjdd ���}| jt|d d� | � |� | �|d d�d � | �|d d�d � | �|dd�d� | � |� | jt|dd� | � |� W d � n1 s�0 Y d� _ | jt|dd� dS )z%test GenericHandler + HasRounds mixinc @ s e Zd ZdZdZdZdZdZdS )z,SkeletonTest.test_30_init_rounds.<locals>.d1r ��roundsr rq r N)r+ r, r- r. rr Z min_rounds� max_rounds�default_roundsr r r r r 8 s r c s � f i | ��j S r r� rv r2 r r � norm_rounds@ s z5SkeletonTest.test_30_init_rounds.<locals>.norm_roundsNr� Trx r g �?rz r r rq re ) r6 Z HasRoundsr_ r; r< r= r~ r rA r� r� )r# r� r� r r2 r �test_30_init_rounds5 s" (z SkeletonTest.test_30_init_roundsc C sR G dd� dt jt j�}| �t|j� | �t|j� | �t|jd� | �t|jd� | �|�� � d|_ |� }| � |� d�d� |�d� |�d� | � |� d�d� | �t|jd� | � |�d�� | �|�d�� d|_ | � |�� � |�d� | � |� d�d� | �t|jd � | �t|jd � |�d� G d d � d |�}| �t|jd� d S ) z+test GenericHandler + HasManyBackends mixinc @ sH e Zd ZdZdZdZdZdZedd� �Z edd� �Z d d � Z d d � Z d S )z)SkeletonTest.test_40_backends.<locals>.d1r r �r' r& Fc S s | j r| �| j� dS dS d S �NTF)� _enable_a�_set_calc_checksum_backend�_calc_checksum_a�rI r r r �_load_backend_aj s z9SkeletonTest.test_40_backends.<locals>.d1._load_backend_ac S s | j r| �| j� dS dS d S r� )� _enable_br� �_calc_checksum_br� r r r �_load_backend_br s z9SkeletonTest.test_40_backends.<locals>.d1._load_backend_bc S s dS �Nr' r r( r r r r� z s z:SkeletonTest.test_40_backends.<locals>.d1._calc_checksum_ac S s dS �Nr& r r( r r r r� } s z:SkeletonTest.test_40_backends.<locals>.d1._calc_checksum_bN) r+ r, r- r. rr �backendsr� r� rS r� r� r� r� r r r r r a s r �any�defaultTr5 r& r' r3 c @ s e Zd ZdZdS )z)SkeletonTest.test_40_backends.<locals>.d2TN)r+ r, r- �_has_backend_ar r r r �d2� s r� N)r6 �HasManyBackendsr_ r; r � get_backend� set_backendr: � has_backendr� r= r* r8 r� rA �AssertionError)r# r �objr� r r r �test_40_backends_ s0 z SkeletonTest.test_40_backendsc C s8 t jdtdd� G dd� dtjtj�}| �t|j� | �t|j � | �t|j d� | �t|j d� | � |� � � d|_ |� }| � |�d �d � |� d � |� d� | � |�d �d � | �t|j d � | �|� d �� | � |� d �� d|_| �|� � � |� d � | � |�d �d � | �t|j d � | �t|j d � d S )z<test GenericHandler + HasManyBackends mixin (deprecated api)�ignorez0.* support for \._has_backend_.* is deprecated.*)�category�messagec @ s0 e Zd ZdZdZdZdZdZdd� Zdd� Z d S ) z)SkeletonTest.test_41_backends.<locals>.d1r r r� Fc S s dS r� r r( r r r r� � s z:SkeletonTest.test_41_backends.<locals>.d1._calc_checksum_ac S s dS r� r r( r r r r� � s z:SkeletonTest.test_41_backends.<locals>.d1._calc_checksum_bN) r+ r, r- r. rr r� r� �_has_backend_br� r� r r r r r � s r r� r� Tr5 r& r' r3 N)r~ �filterwarnings�DeprecationWarningr6 r� r_ r; r r� r� r: r� r� r= r* r8 r� rA )r# r r� r r r �test_41_backends� s2 � z SkeletonTest.test_41_backendsc sh G dd� dt jt j�� � fdd�}| �t|� | jt|dd� | �|dd�td �� | �|td �d�td �� | �|td �d�td �� | jt|td �d� | �|td �d�td �� | jt|td �d� | �� � td��� | �� � td��� | � � � td��� | � � � td ��� | � � � td��� | �t� j d� | �t� j d� d� _ | jt |dd� dS )z#test GenericHandler + HasManyIdentsc @ s< e Zd ZdZdZed�Zed�ed�fZed�ed�iZdS )z+SkeletonTest.test_50_norm_ident.<locals>.d1r �rc �!A�!B�AN) r+ r, r- r. rr r � default_ident� ident_valuesZ ident_aliasesr r r r r � s r c s � f i | ��j S r r� rv r2 r r � norm_ident� s z3SkeletonTest.test_50_norm_ident.<locals>.norm_identNr� Trx r� r� z!Cr� �Bz!Axxxz!Bxxxz!Cxxxr4 r ) r6 Z HasManyIdentsr_ r; r< r= r rA r8 r9 r: r� r� )r# r� r r2 r �test_50_norm_ident� s& z SkeletonTest.test_50_norm_identc C sB ddl m} |j�d�}| �|td�td�d�� d}td�}td �}|j�|�}| �|t|d d |d �� |jj|d d�}| �|t|d d d�� |jj|d d�}| �|td d td�td�d�� |j�d�}| �|ttd�td�dd�� d}|j�|�}| �|tddd d �� |jj|d d�}| �|ttd �dtd �d �� d S )!ztest parsehash()r �r? Z OgAwTx2l6NADIZ AwTx2l6NADIZOg�rZ rp z?$5$LKO/Ute40T3FNF95$U0prpBQd4PloSGU0pnpM4z9wKn4vZ1.jsrzQfPqxph9zLKO/Ute40T3FNF95z+U0prpBQd4PloSGU0pnpM4z9wKn4vZ1.jsrzQfPqxph9i� T)rp r� �implicit_roundsrZ FrY )rp r� r� )ZsanitizezLK**************z+U0pr***************************************)r� r� rp rZ zL$5$rounds=10428$uy/jIAhCetNCTtb0$YWvUOXbkqlqhyoPMpN8BMe.ZGsGx2aBvxTvDFI613c3z+YWvUOXbkqlqhyoPMpN8BMe.ZGsGx2aBvxTvDFI613c3zuy/jIAhCetNCTtb0i�( )rZ rp r� z@$pbkdf2$60000$DoEwpvQeA8B4T.k951yLUQ$O26Y3/NJEiLCVaOVPxGXshyjW8ks ;n���I"�U��?�� �[�i`� s �0�� �xO�=�\�Q�rZ r� rp zO26************************zDo********************N) �passlibr? � des_cryptZ parsehashr= r r �dict� pbkdf2_sha1)r# r? �result�hr5 r3 �h1r r r �test_91_parsehash sH � � � ��z SkeletonTest.test_91_parsehashc C s� ddl m} | �|j�� ddd�� | �|j�� ddd�� | �|jdd � | �|j�� d d d d �� | �|jdd� | �|j�� dddd �� dS )ztest bitsize()r r� �B � r� � � r� i i � �` r� i � � � N) r� r? r= r� ZbitsizeZbcryptZ patchAttrr r� �r# r? r r r �test_92_bitsizeJ s � � � �z SkeletonTest.test_92_bitsizeN)r+ r, r- �__doc__rC rW rd rm rn r� r� r� r� r� r� r� r r r r r ) s .3#$I*K:.8r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �dummy_handler_in_registryz6context manager that inserts dummy handler in registryc C s( || _ td| tjft|dd��| _d S )NZdummy_r )r. rr )r. �typer6 r_ r� �dummy)r# r. r r r r" p s �z"dummy_handler_in_registry.__init__c C sD ddl m} |j| jdd� |�| j� |�| j�| ju s>J �| jS �Nr )�registryF)Z locations)r� r� �_unload_handler_namer. Zregister_crypt_handlerr� Zget_crypt_handler)r# r� r r r � __enter__w s z#dummy_handler_in_registry.__enter__c G s ddl m} |j| jdd� d S r� )r� r� r� r. )r# �exc_infor� r r r �__exit__~ s z"dummy_handler_in_registry.__exit__N)r+ r, r- r� r" r� r� r r r r r� n s r� c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�PrefixWrapperTestztest PrefixWrapper classc C s� t jdddddd�}| �|jd� | �|jd� | �|jt� | �|jt� td�� }| �|jt� W d � n1 sx0 Y dS )z*test PrefixWrapper lazy loading of handlerr r �{XXX}�{MD5}T)ZlazyN) r6 � PrefixWrapperr= � _wrapped_name�assertIs�_wrapped_handler�wrappedr r� �r# r r� r r r �test_00_lazy_loading� s z&PrefixWrapperTest.test_00_lazy_loadingc C st t �dddd�}| �|jd� | �|jt� | �|jt� td�� }| �|jt� W d � n1 sf0 Y dS )z,test PrefixWrapper active loading of handlerr r r� r� N) r6 r� r= r� r� r� r r� r� r� r r r �test_01_active_loading� s z(PrefixWrapperTest.test_01_active_loadingc C st t �dtdd�}| �|jd� | �|jt� | �|jt� td�� }| �|jt� W d � n1 sf0 Y dS )z4test PrefixWrapper with explicitly specified handlerr r� r� Nr ) r6 r� r r= r� r� r� r� r� r� r r r �test_02_explicit� s z"PrefixWrapperTest.test_02_explicitc C st t �dddd�}| �|jd� | �|jtj� | �dt|�v � t �ddd�}| �|jt j� | � dt|�v � d S )Nr r r� r� r� r� r ) r6 r� r= r. r� rr r r: �dirr r8 )r# r r� r r r �test_10_wrapped_attributes� s z,PrefixWrapperTest.test_10_wrapped_attributesc C s� t �dddd�}d}d}| �|�� d� | �t|jdd � | �|�d|�|� | �t|jd|� | �|�d�|� | � |� |�� | � |� |�� | �t|j d|� | � |� d|�� d S ) Nr r r� r� z {XXX}X03MO1qnZdYdgyfeuILPmQ==z {MD5}X03MO1qnZdYdgyfeuILPmQ==z {XXX}1B2M2Y8AsgTpgAmY7PhCfg==�password) r6 r� r= r> r; r< rQ rA r? r8 r9 r: r@ )r# r ZdphZlphr r r �test_11_wrapped_methods� s z)PrefixWrapperTest.test_11_wrapped_methodsc C s� t �ddd�}| �|jtd�� | �|jd � t �ddd�}| �|jd � | �|jd � t �dddd�}| �|jd � | �|jd � t jdddd d �}| �|jtd �� | �|jd � t jdddd d �}| jtt jdddd d � | jtt jdddd d � t �ddd�}| �|jd � | �|jtd�td�f� t jddddd �}| �|jtd�� | �|jd � | jtt jdddd � | � d��" t jddddd�}W d � n1 �s�0 Y | �|jd � | �|jd � d S )N�h2r r� z {XXX}{MD5}r� r� r� Zh3z{Xr� z{XXX}Az{XYz{XXXXZh4Zphpassz{XXX}$P$z{XXX}$H$Zh5TZh6z#orig_prefix.*may not work correctlyZh7�$�?)� orig_prefix�prefix) r6 r� r= rc r r� r� r; rA rU �r# r� r r r � test_12_ident� s< �� 2z PrefixWrapperTest.test_12_identc C s& t jddddd�}| �t|�d� dS )z test repr()r� � md5_cryptr� z$1$�r� z�(?x)^PrefixWrapper\( ['"]h2['"],\s+ ['"]md5_crypt['"],\s+ prefix=u?["']{XXX}['"],\s+ orig_prefix=u?["']\$1\$['"] \)$N)r6 r� Z assertRegex�reprr� r r r � test_13_repr s �z PrefixWrapperTest.test_13_reprc C s$ t jdddd�}| �t|jd� dS )z test orig_prefix sanity checkr� r� z$6$r� rT N)r6 r� r; rA r? r� r r r �test_14_bad_hash s z"PrefixWrapperTest.test_14_bad_hashN) r+ r, r- r� r� r� r� r� r� r� r� r� r r r r r� � s / r� c @ s& e Zd ZdZdZejZdZdd� Z dS )� UnsaltedHashz!test algorithm which lacks a saltZunsalted_test_hashrD c C s. t |t�r|�d�}d| }tt�|��� �S )NrJ s boblious)rK r rL r rM rN rO �r# r) �datar r r r* s zUnsaltedHash._calc_checksumN) r+ r, r- r� r. r6 �LOWER_HEX_CHARSr/ r0 r* r r r r r� s r� c @ sX e Zd ZdZdZdZdZdZdZe j Z Z e �ed��Zedd � �Zd d � Zd d � ZdS )� SaltedHashztest algorithm with a saltZsalted_test_hashro r re rD z^@salt[0-9a-f]{42,44}$c C sF | � |�stj�| ��t|t�r*|�d�}| |dd� |dd � d�S )NrX r| i���)rp rZ )r9 r6 �excZInvalidHashErrorrK r[ r\ rH r r r r] 0 s zSaltedHash.from_stringc C s t d�| j| jf }t|�S )Nz @salt%s%s)r rp rZ r r� r r r � to_string8 s zSaltedHash.to_stringc C sB t |t�r|�d�}| j�d�| | j�d� }tt�|��� �S )NrJ rX )rK r rL rp r rM rN rO r� r r r r* < s zSaltedHash._calc_checksumN)r+ r, r- r� r. rr rs rt r0 r6 r� ru r/ r` ra r rb rS r] r� r* r r r r r� $ s r� u Ιωαννηςc @ s$ e Zd ZeZdedfgZdd� ZdS )�UnsaltedHashTest)r� Z(61cfd32684c47de231f1f982c214e884133762c0Z(96b329d120b97ff81ada770042e44ba87343ad2bc C s& | j ttdd� | j ttjdd� d S )N�xro r r� )r; r< r� r> rB r r r � test_bad_kwdsS s z UnsaltedHashTest.test_bad_kwdsN)r+ r, r- r� �handler� UPASS_TEMP�known_correct_hashesr� r r r r r� K s �r� c @ s$ e Zd ZeZdedfgZdd� ZdS )�SaltedHashTest)r� z/@salt77d71f8fe74f314dac946766c1ac4a2a58365482c0z1@salt9f978a9bfe360d069b0c13f2afecd570447407fa7e48c C s4 t dd�j}| jtt |d d� | jtt |dd� d S )NTrx r� ri )r� rk r; r< rA )r# Zstubr r r r� _ s z SaltedHashTest.test_bad_kwdsN)r+ r, r- r� r� r� r� r� r r r r r� W s �r� ))r� Z __future__r r` rM Zloggingr r~ Z passlib.hashr r Z passlib.excr r Zpasslib.utils.compatr r r Zpasslib.utils.handlersZutils�handlersr6 Zpasslib.tests.utilsr r r r+ �logr r �objectr� r� r7 r� r} r_ r� r� r� r� r r r r �<module> s4 I %