o i@s,dZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddl m Z dd l mZdd l mZdd l mZd Zd Ze djZeeddZeZeddejjejjfDZeddfddZeddfddZddZddZ ddZ!ddZ"d&d d!Z#d"d#Z$d$d%Z%dS)'z werkzeug.security ~~~~~~~~~~~~~~~~~ Security related helpers such as secure password hashing tools. :copyright: 2007 Pallets :license: BSD-3-Clause N) SystemRandom)Struct)izip)PY2) range_type) text_type)to_bytes) to_native>abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789iIz>Icompare_digestccs|] }|dvr|VqdS))N/N.0seprrL/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/werkzeug/security.py srcCs t|||||}tt|dS)a)Like :func:`pbkdf2_bin`, but returns a hex-encoded string. .. versionadded:: 0.9 :param data: the data to derive. :param salt: the salt for the derivation. :param iterations: the number of iterations. :param keylen: the length of the resulting key. If not provided, the digest size will be used. :param hashfunc: the hash function to use. This can either be the string name of a known hash function, or a function from the hashlib module. Defaults to sha256. hex_codec) pbkdf2_binr codecsencode)datasalt iterationskeylenhashfuncrvrrr pbkdf2_hex%srcCsJ|sd}t|}t|}t|r|}t|dd}n|}t|||||S)aReturns a binary digest for the PBKDF2 hash algorithm of `data` with the given `salt`. It iterates `iterations` times and produces a key of `keylen` bytes. By default, SHA-256 is used as hash function; a different hashlib `hashfunc` can be provided. .. versionadded:: 0.9 :param data: the data to derive. :param salt: the salt for the derivation. :param iterations: the number of iterations. :param keylen: the length of the resulting key. If not provided the digest size will be used. :param hashfunc: the hash function to use. This can either be the string name of a known hash function or a function from the hashlib module. Defaults to sha256. sha256nameN)r callablegetattrhashlib pbkdf2_hmac)rrrrr _test_hash hash_namerrrr9srcCst|tr |d}t|tr|d}tdurt||St|t|kr'dSd}trCt||D]\}}|t|t|AO}q0|dkSt||D] \}}|||AO}qH|dkS)zThis function compares strings in somewhat constant time. This requires that the length of at least one string is known in advance. Returns `True` if the two strings are equal, or `False` if they are not. .. versionadded:: 0.7 utf-8NFr) isinstancerr_builtin_safe_str_cmplenrrord)abrxyrrr safe_str_cmpZs     r0cCs(|dkrtddddt|DS)zAGenerate a random string of SALT_CHARS with specified ``length``.rzSalt length must be positivecss|]}ttVqdSN)_sys_rngchoice SALT_CHARS)r_rrrr|szgen_salt..) ValueErrorjoinr)lengthrrrgen_saltxsr:c Cs|dkr||fSt|tr|d}|drD|ddd}t|dvr*td|d }|r9t|d p7d p:t }d }d ||f}nd }|}|r\|sPtd t ||||d}||fS|rvt|trh|d}t |||}| }||fSt || }||fS)zInternal password hash helper. Supports plaintext without salt, unsalted and salted passwords. In case salted passwords are used hmac is used. plainr'zpbkdf2:N:)rz&Invalid number of arguments for PBKDF2rTz pbkdf2:%s:%dFzSalt is required for PBKDF2)r)r(rr startswithsplitr*r7popintDEFAULT_PBKDF2_ITERATIONSr _create_mac hexdigestr#new) methodrpasswordargsr is_pbkdf2 actual_methodrmacrrr_hash_internals6        rMcs8tr t||Sdfdd }||_t|||S)Ncs t|Sr2)r#rF)drGrrrs z_create_mac..hashfunc)rN)r!hmacHMAC__call__)keymsgrGrrrPrrDs rD pbkdf2:sha256cCs2|dkrt|nd}t|||\}}d|||fS)aHash a password with the given method and salt with a string of the given length. The format of the string returned includes the method that was used so that :func:`check_password_hash` can check the hash. The format for the hashed string looks like this:: method$salt$hash This method can **not** generate unsalted passwords but it is possible to set param method='plain' in order to enforce plaintext passwords. If a salt is used, hmac is used internally to salt the password. If PBKDF2 is wanted it can be enabled by setting the method to ``pbkdf2:method:iterations`` where iterations is optional:: pbkdf2:sha256:80000$salt$hash pbkdf2:sha256$salt$hash :param password: the password to hash. :param method: the hash method to use (one that hashlib supports). Can optionally be in the format ``pbkdf2:[:iterations]`` to enable PBKDF2. :param salt_length: the length of the salt in letters. r;r1z%s$%s$%s)r:rM)rHrG salt_lengthrhrKrrrgenerate_password_hashsrZcCs:|ddkr dS|dd\}}}tt|||d|S)acheck a password against a given salted and hashed password value. In order to support unsalted legacy passwords this method supports plain text passwords, md5 and sha1 hashes (both salted and unsalted). Returns `True` if the password matched, `False` otherwise. :param pwhash: a hashed string like returned by :func:`generate_password_hash`. :param password: the plaintext password to compare against the hash. $r>Fr)countr@r0rM)pwhashrHrGrhashvalrrrcheck_password_hashs r_csp|g}|D]-dkrttfddtDs*tjs*dks*dr-dS|qtj |S)a2Safely 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``. r1c3s|]}|vVqdSr2rrfilenamerrrszsafe_join..z..z../N) posixpathnormpathany _os_alt_sepsospathisabsr?appendr8) directory pathnamespartsrr`r safe_joins     rm)rVrW)&__doc__rr#rQrfrbrandomrstructr_compatrrrrr r r5rCpack _pack_intr"r)r3listrgraltseprerrr0r:rMrDrZr_rmrrrrsB             !%