File: //usr/lib64/python3.9/site-packages/borg/crypto/__pycache__/keymanager.cpython-39.pyc
a
HZ�h@ � @ s� d dl Z d dlZd dlZd dl mZmZmZ d dlmZ ddlm Z m
Z
m Z m Z m
Z
mZ ddlmZ ddlmZmZmZmZmZ G d d
� d
e �ZG d d � d e �ZG d
d� de �ZG dd� de �Zdd� ZG dd� d�Z dS )� N)� unhexlify�
a2b_base64�
b2a_base64)�sha256� )�Manifest�NoManifestError�Error�yes�
bin_to_hex� dash_open)�
Repository� )�
KeyfileKey�KeyfileNotFoundError�RepoKeyNotFoundError�KeyBlobStorage� identify_keyc @ s e Zd ZdZdS )�UnencryptedRepoz9Keymanagement not available for unencrypted repositories.N��__name__�
__module__� __qualname__�__doc__� r r �</usr/lib64/python3.9/site-packages/borg/crypto/keymanager.pyr
s r c @ s e Zd ZdZdS )�UnknownKeyTypezKeytype {0} is unknown.Nr r r r r r s r c @ s e Zd ZdZdS )�RepoIdMismatchzHThis key backup seems to be for a different backup repository, aborting.Nr r r r r r s r c @ s e Zd ZdZdS )�NotABorgKeyFilez-This file is not a borg key backup, aborting.Nr r r r r r s r c C s t � }|�| � |�� d |� S �N)r �updateZ hexdigest)�dataZnum�hr r r �sha256_truncated s
r# c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�
KeyManagerc C sd || _ d | _d | _z| j �tj�}W n tjy< t�Y n0 t |�}|j
| _| jt j kr`t
� �d S r )�
repository�keyblob�keyblob_storage�getr Z MANIFEST_IDr
ZObjectNotFoundr r ZSTORAGEr Z
NO_STORAGEr )�selfr% Z
manifest_data�keyr r r �__init__$ s
zKeyManager.__init__c C s� | j tjkr`t| j�}|�� }t|��(}d�|�� dd � �| _ W d � q�1 sT0 Y n:| j tj
kr�| j� � � � }|s�| jj
�� }t|�d �|| _ d S )N� r )r' r �KEYFILEr r% Zfind_key�open�join� readlinesr& �REPOZload_key�decodeZ _locationZcanonical_pathr )r) �k�target�fd�key_data�locr r r � load_keyblob3 s
8
zKeyManager.load_keyblobc C sP | j tjkr,t| j�}|�|�}| �|� n | j tjkrL| j�| j �
d�� d S )Nzutf-8) r' r r- r r% Zget_existing_or_new_target�
store_keyfiler1 Zsave_keyr& �encode)r) �argsr3 r4 r r r �
store_keyblobB s
zKeyManager.store_keyblobc C s< t j� dt| jj�� d�}|| j7 }| j�d�s8|d7 }|S )N� �
)r �FILE_IDr r% �idr& �endswith)r) r! r r r �get_keyfile_dataK s
zKeyManager.get_keyfile_datac C s< t |d�� }|�| �� � W d � n1 s.0 Y d S )N�w)r �writerB )r) r4 r5 r r r r9 R s zKeyManager.store_keyfilec C s |d u r d}| � |� d S )N�-)r9 )r) �pathr r r �exportV s zKeyManager.exportc C sl |d u r d}t |d��B}| �� }t�dd�}|�d|�� d �}|�|� W d � n1 s^0 Y d S )NrE �wbZborgz
paperkey.htmls </textarea>)r rB �pkgutil�get_data�replacer: rD )r) rF r5 r6 Zhtmlr r r � export_qr\ s zKeyManager.export_qrc
C s8 |d u r d}dd� }d}t | j�}|d7 }t|�d d }t| jj�d d� }t|d�}|d �|||�||�tt|�d
| d
| � d �d ��7 }d
}t|�r�|d7 }|d d� } t|j
d dd�| d �}
||d�d|t| ��� d|
� d�7 }|dd � }q�t |d��} | � |� W d � n1 �s*0 Y d S )NrE c S s> d}d}| D ],}|r(|d dkr(|d7 }||7 }|d7 }q |S )Nr, r � r= r r )�s�ret�iZchr r r �groupedj s
z+KeyManager.export_paperkey.<locals>.groupedz:To restore key use borg key import --paper /path/to/repo
zBORG PAPER KEY v1
� � � zid: {:d} / {} / {} - {}
�/�asciir r r �big�� byteorder�2d�: z - r> rC )
r r& �lenr r% r@ r# �format�strr: �to_bytesr rD ) r) rF rQ rG Zbinary�lines�repoidZcomplete_checksum�idxZbinline�checksumr5 r r r �export_paperkeyf s.
"� $ zKeyManager.export_paperkeyc C s� t j}|d t| jj� d }t|jd��H}|�t|��}||krZ|� |�sTt
� �nt � �|�� | _ W d � n1 sx0 Y | �
|� d S )Nr= r> �r)r r? r r% r@ r rF �readr\ �
startswithr r r& r< )r) r; Zfile_idZ
first_liner5 Zfile_first_liner r r �import_keyfile� s
(zKeyManager.import_keyfilec C s� z dd l }W n ty& td� Y n0 t| jj�d d� }�z6td��dd�}|dkrftd�rft � �z|�
d�\}}W n t y� td � Y q@Y n0 z|�
d
�\}}} W n t y� td � Y q@Y n0 t |�
� �d �d
�|kr�td� q@z t|�}
W n t �y td� Y n0 �q q@||k�r.t� �d} d} t| d�d��}
|
�dd�}
|
dk�rltd��rlt � �z|
�
d�\}}W n" t �y� td � Y �q6Y n0 z t|�}W n$ tj�y� td� Y �q6Y n0 t | jd
dd�| d
�|k�rtd| � d�� �q6| |7 } | |
k�r �q(| d7 } �q6t | d�| k�rBtd� q@d�t�t| ��d ���d | _| �|� �qrq@W n t �y� td � Y d S 0 d S ) Nr z?Note: No line editing available due to missing readline supportrS zid: r= r, zAbort import? [yN]:rE z1each line must contain exactly one '-', try againrU z3the id line must contain exactly two '/', try againrV r z0line checksum did not match, try same line againz#internal error while parsing length� r rZ r[ z8only characters 0-9 and a-f and '-' are valid, try againrW rX z&line checksum did not match, try line z againrT zIThe overall checksum did not match, retry or enter a blank line to abort.r> z
- aborted) �readline� ImportError�printr r% r@ �inputrK r
�EOFError�split�
ValueErrorr# �lowerr: �intr r �binasciir r_ r/ �textwrap�wrapr r2 r& r< )r) r; rj ra Zidliner! rc Zid_linesZ id_repoidZid_complete_checksumr` �resultrb Zinline�partr r r �import_paperkey� s~
zKeyManager.import_paperkeyN)
r r r r+ r8 r< rB r9 rG rL rd rh rx r r r r r$ # s
$r$ ) rs rI rt r r r Zhashlibr Zhelpersr r r r
r r r% r
r* r r r r r r r r r r# r$ r r r r �<module> s