U
    /v¶hg  ã                   @  sÄ   U d dl mZ d dlZd dlZd dlZd dlZd dlZdZdZe	dd„ ej
ejjfD ƒƒZded< d	d
dœdd„Zd
d
d
ddœdd„Zdd
d
d	d
dœdd„Zd
d
ddœdd„Zd
d
ddœdd„ZdS ) é    )ÚannotationsNZ>abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789iÀ'	 c                 c  s"   | ]}|d k	r|dkr|V  qd S )Nú/© ©Ú.0Úsepr   r   ú5/tmp/pip-unpacked-wheel-c3xdgso7/werkzeug/security.pyÚ	<genexpr>   s      r	   z	list[str]Ú_os_alt_sepsÚintÚstr)ÚlengthÚreturnc                 C  s(   | dkrt dƒ‚d dd„ t| ƒD ƒ¡S )zAGenerate a random string of SALT_CHARS with specified ``length``.r   zSalt length must be at least 1.Ú c                 s  s   | ]}t  t¡V  qd S ©N)ÚsecretsÚchoiceÚ
SALT_CHARS)r   Ú_r   r   r   r	      s     zgen_salt.<locals>.<genexpr>)Ú
ValueErrorÚjoinÚrange)r   r   r   r   Úgen_salt   s    r   ztuple[str, str])ÚmethodÚsaltÚpasswordr   c                 C  sF  |   d¡^} }| ¡ }| ¡ }| dkr¬|s8d}d}d}n4ztt|ƒ\}}}W n tk
rj   tdƒd ‚Y nX d| | | }	tj||||||	d� ¡ d	|› d|› d|› �fS | d
k�r2t|ƒ}
|
dkrÐd}t	}n>|
dkræ|d }t	}n(|
dk�r|d }t|d ƒ}ntdƒ‚t 
||||¡ ¡ d|› d|› �fS td| › d�ƒ‚d S )Nú:Úscrypti €  é   é   z'scrypt' takes 3 arguments.é„   )r   ÚnÚrÚpÚmaxmemzscrypt:Zpbkdf2r   Úsha256é   z'pbkdf2' takes 2 arguments.zpbkdf2:zInvalid hash method 'z'.)ÚsplitÚencodeÚmapr   r   Úhashlibr   ÚhexÚlenÚDEFAULT_PBKDF2_ITERATIONSÚpbkdf2_hmac)r   r   r   ÚargsZ
salt_bytesZpassword_bytesr!   r"   r#   r$   Zlen_argsÚ	hash_nameZ
iterationsr   r   r   Ú_hash_internal   sX         ÿü

   ÿür1   r   é   )r   r   Úsalt_lengthr   c                 C  s,   t |ƒ}t||| ƒ\}}|› d|› d|› �S )aá  Securely hash a password for storage. A password can be compared to a stored hash
    using :func:`check_password_hash`.

    The following methods are supported:

    -   ``scrypt``, the default. The parameters are ``n``, ``r``, and ``p``, the default
        is ``scrypt:32768:8:1``. See :func:`hashlib.scrypt`.
    -   ``pbkdf2``, less secure. The parameters are ``hash_method`` and ``iterations``,
        the default is ``pbkdf2:sha256:600000``. See :func:`hashlib.pbkdf2_hmac`.

    Default parameters may be updated to reflect current guidelines, and methods may be
    deprecated and removed if they are no longer considered secure. To migrate old
    hashes, you may generate a new hash when checking an old hash, or you may contact
    users with a link to reset their password.

    :param password: The plaintext password.
    :param method: The key derivation function and parameters.
    :param salt_length: The number of characters to generate for the salt.

    .. versionchanged:: 2.3
        Scrypt support was added.

    .. versionchanged:: 2.3
        The default iterations for pbkdf2 was increased to 600,000.

    .. versionchanged:: 2.3
        All plain hashes are deprecated and will not be supported in Werkzeug 3.0.
    ú$)r   r1   )r   r   r3   r   ÚhZactual_methodr   r   r   Úgenerate_password_hashI   s    r6   Úbool)Úpwhashr   r   c                 C  sF   z|   dd¡\}}}W n tk
r,   Y dS X t t|||ƒd |¡S )aA  Securely check that the given stored password hash, previously generated using
    :func:`generate_password_hash`, matches the given password.

    Methods may be deprecated and removed if they are no longer considered secure. To
    migrate old hashes, you may generate a new hash when checking an old hash, or you
    may contact users with a link to reset their password.

    :param pwhash: The hashed password.
    :param password: The plaintext password.

    .. versionchanged:: 2.3
        All plain hashes are deprecated and will not be supported in Werkzeug 3.0.
    r4   r&   Fr   )r'   r   ÚhmacÚcompare_digestr1   )r8   r   r   r   Zhashvalr   r   r   Úcheck_password_hashm   s
    r;   z
str | None)Ú	directoryÚ	pathnamesr   c                   s‚   | sd} | g}|D ]d‰ ˆ dkr(t  ˆ ¡‰ t‡ fdd„tD ƒƒsftj ˆ ¡sfˆ  d¡sfˆ dksfˆ  d¡rl dS | ˆ ¡ qt j	|Ž S )	a2  Safely join zero or more untrusted path components to a base
    directory to avoid escaping the base directory.

    :param directory: The trusted base directory.
    :param pathnames: The untrusted path components relative to the
        base directory.
    :return: A safe path, otherwise ``None``.
    Ú.r   c                 3  s   | ]}|ˆ kV  qd S r   r   r   ©Úfilenamer   r   r	   ˜   s     zsafe_join.<locals>.<genexpr>r   z..z../N)
Ú	posixpathÚnormpathÚanyr
   ÚosÚpathÚisabsÚ
startswithÚappendr   )r<   r=   Úpartsr   r?   r   Ú	safe_joinƒ   s&    	
ÿ
þüûúrJ   )r   r2   )Ú
__future__r   r*   r9   rD   rA   r   r   r-   Úlistr   rE   Úaltsepr
   Ú__annotations__r   r1   r6   r;   rJ   r   r   r   r   Ú<module>   s$    ÿ1   ÿ$