Current File : //lib/python3.9/site-packages/passlib/crypto/scrypt/__pycache__/_builtin.cpython-39.pyc
a
f�Wc�" � @ sR d Z ddlZddlZddlmZ ddlmZ ddlmZ dgZ G dd� de
�Z dS )z<passlib.utils.scrypt._builtin -- scrypt() kdf in pure-python� N)�izip)� pbkdf2_hmac)�salsa20� ScryptEnginec @ sh e Zd ZdZdZdZdZdZdZdZ dZ
dZ dZ e
dd� �Zdd� Zdd � Zd
d � Zd d
� Zdd� ZdS )r a
helper class used to run scrypt kdf, see scrypt() for frontend
.. warning::
this class does NO validation of the input ranges or types.
it's not intended to be used directly,
but only as a backend for :func:`passlib.utils.scrypt.scrypt()`.
r Nc C s | |||�� |||�S )z-create engine & run scrypt() hash calculation)�run)�cls�secret�salt�n�r�p�keylen� r �B/usr/lib/python3.9/site-packages/passlib/crypto/scrypt/_builtin.py�execute2 s zScryptEngine.executec s� || _ || _|| _|d> | _| j| | _|d> | _}|d> | _t�d�dksRJ �t� dt
|� d �| _ |dkrz| j | _
|dkr�t�d�}n.|d ks�J �t�d�� t�d
��� �fd d �}|| _d S )
N� � � �I�<� l �� ��l ���� i��c s � | ��| �d> B S )N� r )�X�Zig1Zig2r r �
integerifyU s z)ScryptEngine.__init__.<locals>.integerify)r
r r �
smix_bytes�iv_bytes�bmix_len�
bmix_half_len�struct�calcsize�Struct�str� bmix_struct�_bmix_1�bmix�operator�
itemgetterr )�selfr
r r r r r r r �__init__: s$
zScryptEngine.__init__c sn | j }td||d|d�� | j�| jdkr2�� �}n*| j�d�� ��fdd�td|��D ��}td||d|d�S )z�
run scrypt kdf for specified secret, salt, and keylen
.. note::
* time cost is ``O(n * r * p)``
* mem cost is ``O(n * r)``
Zsha256r )�roundsr
� c 3 s" | ]}�� ||� � �V qd S �Nr )�.0�offset��input�smixr r r � <genexpr>r s �z#ScryptEngine.run.<locals>.<genexpr>r )r r r2 r r �join�range)r) r r r
r �outputr r0 r r \ s
�zScryptEngine.runc s� | j � | j}| j}| j�t|�|���� ��fdd�}t|� �}|j}�d }d}|�k r�|��|@ } tdd� t�|| ��D ��}
� |
�� |d7 }qR|j �� S )as run SCrypt smix function on a single input block
:arg input:
byte string containing input data.
interpreted as 32*r little endian 4 byte integers.
:returns:
byte string containing output data
derived by mixing input using n & r parameters.
.. note:: time & mem cost are both ``O(n * r)``
c 3 s2 d} | �k r.t ��}|V � |�� | d7 } qd S )Nr r )�tuple)�iZlast�r& �bufferr
r r �vgen� s
z ScryptEngine.smix.<locals>.vgenr r c s s | ]\}}||A V qd S r- r �r. �a�br r r r3 � r, z$ScryptEngine.smix.<locals>.<genexpr>)
r& r$ r r
�list�unpack� __getitem__r7 r �pack) r) r1 r$ r r; �VZ
get_v_elemZn_maskr8 �j�resultr r9 r r2 } s
zScryptEngine.smixc C s� | j }|dd� }t|�}d}||k r�|d }tdd� t||�D �� |||�< }tdd� t||�D �� ||| || �< }|}q dS )a�
block mixing function used by smix()
uses salsa20/8 core to mix block contents.
:arg source:
source to read from.
should be list of 32*r 4-byte integers
(2*r salsa20 blocks).
:arg target:
target to write to.
should be list with same size as source.
the existing value of this buffer is ignored.
.. warning::
this operates *in place* on target,
so source & target should NOT be same list.
.. note::
* time cost is ``O(r)`` -- loops 16*r times, salsa20() has ``O(1)`` cost.
* memory cost is ``O(1)`` -- salsa20() uses 16 x uint4,
all other operations done in-place.
r Nr � c s s | ]\}}||A V qd S r- r r< r r r r3 � r, z$ScryptEngine.bmix.<locals>.<genexpr>c s s | ]\}}||A V qd S r- r r< r r r r3 � r, )r �iterr r )r) �source�targetZhalf�tmpZsiterrD Zjnr r r r& � s $,zScryptEngine.bmixc C sX |dd� }t dd� t|t|��D �� |dd�< }t dd� t||�D ��|dd�< dS )z0special bmix() method optimized for ``r=1`` caserF Nc s s | ]\}}||A V qd S r- r r< r r r r3 � r, z'ScryptEngine._bmix_1.<locals>.<genexpr>c s s | ]\}}||A V qd S r- r r< r r r r3 � r, )r r rG )r) rH rI �BrJ r r r r% � s (zScryptEngine._bmix_1)�__name__�
__module__� __qualname__�__doc__r
r r r r r r r$ r � classmethodr r* r r2 r& r% r r r r r s"
"!C() rO r' r Zpasslib.utils.compatr Zpasslib.crypto.digestr Z passlib.crypto.scrypt._salsar �__all__�objectr r r r r �<module> s �
Mini Shell