a
�M�e� � @ s� U d dl Z d dlZd dlZd dlZd dlZd dlmZmZ G dd� d�Ze� Z ee e
d< z e� � Z W n e
y| dZ Y n0 ed�dd�Zeeed �d
d �ZdS ) � N)�Any�Optionalc @ s� e Zd Zdee d�dd�Zedd�dd�Zedd�dd �Zdd
�d d �Ze d
�d
d�Z
e d
�dd�Z e d
�dd�Z e e e d�dd�Z
dS )� EntropyPoolN)�seedc C sp d| _ d | _d| _t�� | _t�� | _d| _ t
d| j �| _ |d ur`| � |� d| _
t�� | _n d| _
d| _d S )Nr � � TF)�
pool_index�digest� next_byte� threading�Lock�lock�hashlibZsha1�hash�hash_len� bytearray�pool�_stir�seeded�os�getpid�seed_pid)�selfr � r �//usr/lib/python3.9/site-packages/dns/entropy.py�__init__ s
zEntropyPool.__init__)�entropy�returnc C sJ |D ]@}| j | jkrd| _ |d@ }| j| j |N < | j d7 _ qd S )Nr � � )r r r )r r �c�br r r r 0 s zEntropyPool._stirc C s4 | j � | �|� W d � n1 s&0 Y d S )N)r
r )r r r r r �stir8 s zEntropyPool.stir�r c C s� | j r| jt�� kr�zt�d�}W np ty� z:tddd��}|�d�}W d � n1 s^0 Y W n" ty� tt � � ��
� }Y n0 Y n0 d| _ t�� | _d | _ t |�}| �
|� d S )N� z /dev/urandom�rbr T)r r r r �urandom� Exception�open�read�str�time�encoder r r )r r �rr r r � _maybe_seed< s ,
zEntropyPool._maybe_seedc C s� | j �| | �� | jd u s&| j| jkrZ| j�t| j�� t | j�� �| _| �
| j� d| _| j| j }| jd7 _W d � n1 s�0 Y |S )Nr r ) r
r. r r
r r �update�bytesr r r )r �valuer r r �random_8L s ,zEntropyPool.random_8c C s | � � d | � � S )N� )r2 �r r r r � random_16X s zEntropyPool.random_16c C s | � � d | � � S )N� )r5 r4 r r r � random_32[ s zEntropyPool.random_32��first�lastr c C sd || d }|dkr t d��|dkr0| j}d}n |dkrD| j}d}n
| j}d}|||� |d S ) Nr l ztoo bigr6 l �� r3 i�� r )�
ValueErrorr7 r5 r2 )r r9 r: �sizeZrand�maxr r r �random_between^ s zEntropyPool.random_between)N)�__name__�
__module__� __qualname__r r0 r r r" r. �intr2 r5 r7 r> r r r r r s r �
system_randomr# c C s t d urt �dd�S t�� S d S )Nr r6 )rC � randranger r5 r r r r r5 w s r5 r8 c C s( t d urt �| |d �S t�| |�S d S )Nr )rC rD r r> )r9 r: r r r �between~ s rE )r r Zrandomr r+ �typingr r r r �__annotations__Z SystemRandomrC r' rB r5 rE r r r r �<module> s
T