Current File : //lib/python3.9/site-packages/passlib/crypto/scrypt/__pycache__/_builtin.cpython-39.opt-1.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�dt |� d �| _
|dkrh| j | _ |dkr|t
�d�}n"t
�d�� t
�d ��� �fd
d �}|| _d S ) N� � � �<�I� 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�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 r1 r r �join�range)r( r r r
r �outputr r/ 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 r2 � r+ z$ScryptEngine.smix.<locals>.<genexpr>)
r% r# r r
�list�unpack� __getitem__r6 r �pack) r( r0 r# r r: �VZ
get_v_elemZn_maskr7 �j�resultr r8 r r1 } 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 r2 � r+ z$ScryptEngine.bmix.<locals>.<genexpr>c s s | ]\}}||A V qd S r, r r; r r r r2 � r+ )r �iterr r )r( �source�targetZhalf�tmpZsiterrC 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`` caserE Nc s s | ]\}}||A V qd S r, r r; r r r r2 � r+ z'ScryptEngine._bmix_1.<locals>.<genexpr>c s s | ]\}}||A V qd S r, r r; r r r r2 � r+ )r r rF )r( rG rH �BrI r r r r$ � s (zScryptEngine._bmix_1)�__name__�
__module__� __qualname__�__doc__r
r r r r r r r# r � classmethodr r) r r1 r% r$ r r r r r s"
"!C() rN r& r Zpasslib.utils.compatr Zpasslib.crypto.digestr Z passlib.crypto.scrypt._salsar �__all__�objectr r r r r �<module> s �
Mini Shell