| Current File : //usr/lib64/python3.6/site-packages/borg/crypto/__pycache__/nonces.cpython-36.pyc |
3
y��b� � @ sv d dl Z d dlZd dlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
dd l m Z m
Z
dZdZG d d
� d
�ZdS )� N)� unhexlify� )�get_security_dir)�
bin_to_hex)�SaveFile)�InvalidRPCMethod� )�
bytes_to_long�
long_to_bytes�@ � c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� NonceManagerc C s4 || _ || _d | _|| _tjjt| j j�d�| _ d S )NZnonce)
�
repository�
enc_cipher�end_of_nonce_reservation�manifest_nonce�os�path�joinr Zid_str�
nonce_file)�selfr r r � r � /usr/lib64/python3.6/nonces.py�__init__ s
zNonceManager.__init__c C sB y(t | jd��}tt|j� ��S Q R X W n tk
r< d S X d S )N�r)�openr r r �read�FileNotFoundError)r �fdr r r �get_local_free_nonce s
z!NonceManager.get_local_free_noncec C sD | j � |krtd��t| jdd��}|jtt|��� W d Q R X d S )Nz6nonce space reservation with mismatched previous stateF)Zbinary)r � Exceptionr r �writer r
)r �next_unreserved� start_noncer r r r � commit_local_nonce_reservation s z+NonceManager.commit_local_nonce_reservationc C sD y
| j j� S tk
r> tjjd� dd� | _dd� | _d S X d S )NzTPlease upgrade to borg version 1.1+ on the server for safer AES-CTR nonce handling.
c S s d S )Nr r r r r �<lambda>+ s z2NonceManager.get_repo_free_nonce.<locals>.<lambda>c S s d S )Nr )r" r# r r r r% , s )r Zget_free_noncer �sys�stderrr! �get_repo_free_nonce� commit_repo_nonce_reservation)r r r r r( % s
z NonceManager.get_repo_free_noncec C s | j j||� d S )N)r Zcommit_nonce_reservation)r r" r# r r r r) / s z*NonceManager.commit_repo_nonce_reservationc C s� | j r8tj| jjdd�}|| j ks&t�|| | j kr8d S | j� }| j� }tdd� ||| j | j fD ��}|| t
}|t k s~t�| j d kr�| jj d |j
ddd�� n"|| j kr�| jj d |j
ddd�� | j||� | j||� || _ d S )NZbig)� byteorderc s s | ]}|d k r|V qd S )Nr )�.0�xr r r � <genexpr>J s z2NonceManager.ensure_reservation.<locals>.<genexpr>� )r �int�
from_bytesr Ziv�AssertionErrorr( r �maxr �NONCE_SPACE_RESERVATION�MAX_REPRESENTABLE_NONCE�reset�to_bytesr) r$ )r Znonce_space_neededZ
next_nonceZrepo_free_nonceZlocal_free_nonceZfree_nonce_spaceZreservation_endr r r �ensure_reservation2 s"
z NonceManager.ensure_reservationN) �__name__�
__module__� __qualname__r r r$ r( r) r7 r r r r r
s
r
l l ���� i )r r& Zbinasciir Zhelpersr r �platformr Zremoter Z low_levelr r
r4 r3 r
r r r r �<module> s