관리-도구
편집 파일: _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 �