File: //var/opt/nydus/ops/oscrypto/__pycache__/kdf.cpython-39.pyc
a
�,�h� � @ s* d dl mZmZmZmZ d dlZd dlZd dlmZ ddlm Z ddl
m Z ddl m
Z
mZmZ ddlmZ dd lmZmZ e � Zed
kr�dd lmZmZ nBed ks�ed
kr�dd lmZmZ ddlmZm Z ndd l mZmZ g d�Z ej dk�rdd� Z dd� Z!ndd� Z dd� Z!d dd�Z"dd � Z#dS ) � )�unicode_literals�division�absolute_import�print_functionN)�datetime� )�backend)�
rand_bytes)� type_name�byte_cls� int_types)�pretty_message)�new�deref�mac)�pbkdf2�
pkcs12_kdf�winZ winlegacy)�kernel32� handle_error)�pbkdf1r �pbkdf2_iteration_calculatorr �win32c C s$ t td�} t�| �}t|� t| �S )NzLARGE_INTEGER *)r r ZQueryPerformanceCounterr r )�number�res� r �=/opt/nydus/tmp/pip-target-wkfpz8uv/lib/python/oscrypto/kdf.py�
_get_start$ s
r c C s t � | }t|d �S )Ng @�@)r �int)�start�lengthr r r � _get_elapsed* s
r! c C s t �� S )N)r �nowr r r r r / s c C s: t �� | }|j|jd d }|jd }t||d �S )N� i i� )r r" �seconds�days� microsecondsr )r r r$ Z millisecondsr r r r! 2 s
�d Fc s* � t g d��vr"ttdt� ����t�t�s>ttdt������dk rXttdt�����t�t�stttdt������dk r�ttdt�����tj r�t
td���d �d
� d ��t ���� ������fd d
�}|� }t
�| d ��|� }�| ��dk r�dnd}t
t�|��}|dk�r&|d d }|S )a�
Runs pbkdf2() twice to determine the approximate number of iterations to
use to hit a desired time per run. Use this on a production machine to
dynamically adjust the number of iterations as high as you can.
:param hash_algorithm:
The string name of the hash algorithm to use: "md5", "sha1", "sha224",
"sha256", "sha384", "sha512"
:param key_length:
The length of the desired key in bytes
:param target_ms:
The number of milliseconds the derivation should take
:param quiet:
If no output should be printed as attempts are made
:return:
An integer number of iterations of PBKDF2 using the specified hash
that will take at least target_ms
)�sha1�sha224�sha256�sha384�sha512zz
hash_algorithm must be one of "sha1", "sha224", "sha256", "sha384",
"sha512", not %s
�?
key_length must be an integer, not %s
r �C
key_length must be greater than 0 - is %s
z>
target_ms must be an integer, not %s
zB
target_ms must be greater than 0 - is %s
z�
Only a very slow, pure-python version of PBKDF2 is available,
making this function useless
i' zthis is a testzutf-8c s> t � } t� ����� t| �}�s2td�|f � d� | S )Nz%s iterations in %smsg �?)r r r! �print)r Z observed_ms��hash_algorithm�
iterations�
key_length�password�quiet�salt� target_msr r �_measure� s z-pbkdf2_iteration_calculator.<locals>._measureg @i�� ������i N i� )�set�
ValueErrorr
�repr�
isinstancer � TypeErrorr
r Z pure_python�OSError�encoder r �round)r1 r3 r7 r5 r8 �fractionZ round_factor�resultr r0 r r 9 sT �
��
���
r c C sb t |t�s ttdt|����t |t�s8ttdt|����t |t�sTttdt|����|dk rnttdt|����t |t�s�ttdt|����|dk r�ttdt|����| tg d��vr�ttd t| ����|d
kr�| td d g�v r�ttd
| t|�f���|dk�r| dk�rttdt|����t t
| �}||| �� � }t d|d �D ]}||�� � }�qB|d|� S )aF
An implementation of PBKDF1 - should only be used for interop with legacy
systems, not new architectures
:param hash_algorithm:
The string name of the hash algorithm to use: "md2", "md5", "sha1"
:param password:
A byte string of the password to use an input to the KDF
:param salt:
A cryptographic random byte string
:param iterations:
The numbers of iterations to use when deriving the key
:param key_length:
The length of the desired key in bytes
:return:
The derived key as a byte string
z@
password must be a byte string, not %s
z<
salt must be a byte string, not %s
z?
iterations must be an integer, not %s
r zC
iterations must be greater than 0 - is %s
r- r. )�md2�md5r( zT
hash_algorithm must be one of "md2", "md5", "sha1", not %s
� rE rF zM
key_length can not be longer than 16 for %s - is %s
� r( zO
key_length can not be longer than 20 for sha1 - is %s
� N)
r> r r? r
r
r r< r= r; �getattr�hashlib�digest�range)r1 r4 r6 r2 r3 �algo�output�_r r r r � sd
�
�
��
���
��
r )r' F)$�
__future__r r r r �sysrK r � r �utilr �_typesr
r r �_errorsr
Z_ffir r Z_backendZ _mac.utilr r Z _win.utilZ_win._kernel32r r Z
_openssl.util�__all__�platformr r! r r r r r r �<module> s.
g