File: //usr/lib64/python3.9/site-packages/borg/crypto/__pycache__/nonces.cpython-39.opt-1.pyc
a
HZ�h � @ sv d dl Z d dlZd dlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
dd l m Z m
Z
d
Zd ZG d d
� d
�ZdS )� N)� unhexlify� )�get_security_dir)�
bin_to_hex)�SaveFile)�InvalidRPCMethod� )�
bytes_to_long�
long_to_bytesl ���� i c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� NonceManagerc C s. || _ d | _|| _tj�t| j j�d�| _d S )N�nonce) �
repository�end_of_nonce_reservation�manifest_nonce�os�path�joinr Zid_str�
nonce_file)�selfr
r � r �8/usr/lib64/python3.9/site-packages/borg/crypto/nonces.py�__init__ s zNonceManager.__init__c C sZ z@t | j��"}tt|�� ��W d � W S 1 s40 Y W n tyT Y d S 0 d S �N)�openr r r �read�FileNotFoundError)r �fdr r r �get_local_free_nonce s
4 z!NonceManager.get_local_free_noncec C sX | � � |krtd��t| jdd��"}|�tt|��� W d � n1 sJ0 Y 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 sF z | j �� W S ty@ tj�d� dd� | _dd� | _Y d S 0 d S )NzTPlease upgrade to borg version 1.1+ on the server for safer AES-CTR nonce handling.
c S s d S r r r r r r �<lambda>* � z2NonceManager.get_repo_free_nonce.<locals>.<lambda>c S s d S r r )r r! r r r r# + r$ )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 �||� d S r )r
Zcommit_nonce_reservation)r r r! r r r r( . s z*NonceManager.commit_repo_nonce_reservationc C sx | j r |}|| | j kr |S | �� }| �� }tdd� ||| j| j fD ��}|| t }| �||� | �||� || _ |S )a�
Call this before doing encryption, give current, yet unused, integer IV as <nonce>
and the amount of subsequent (counter-like) IVs needed as <nonce_space_needed>.
Return value is the IV (counter) integer you shall use for encryption.
Note: this method may return the <nonce> you gave, if a reservation for it exists or
can be established, so make sure you give a unused nonce.
c s s | ]}|d ur|V qd S r r )�.0�xr r r � <genexpr>Q r$ z2NonceManager.ensure_reservation.<locals>.<genexpr>)r r' r �maxr �NONCE_SPACE_RESERVATIONr( r" )r r Znonce_space_neededZ
next_nonceZrepo_free_nonceZlocal_free_nonceZfree_nonce_spaceZreservation_endr r r �ensure_reservation1 s z NonceManager.ensure_reservationN) �__name__�
__module__� __qualname__r r r" r' r( r. r r r r r s
r )r r% Zbinasciir Zhelpersr r �platformr Zremoter Z low_levelr r
ZMAX_REPRESENTABLE_NONCEr- r r r r r �<module> s