| Current File : //usr/lib64/python3.6/site-packages/borg/crypto/__pycache__/keymanager.cpython-36.pyc |
3
y��b� � @ 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.6/keymanager.pyr
s r c @ s e Zd ZdZdS )�UnknownKeyTypezKeytype {0} is unknown.N)r r r r r r r r r s r c @ s e Zd ZdZdS )�RepoIdMismatchzHThis key backup seems to be for a different backup repository, aborting.N)r r r r r r r r r s r c @ s e Zd ZdZdS )�NotABorgKeyFilez-This file is not a borg key backup, aborting.N)r r r r r r r r r s r c C s t � }|j| � |j� 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 sf || _ d | _d | _y| j jtj�}W n tjk
r> t�Y nX t |�}|j
| _| jt j krbt
� �d S )N)�
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 tjkrNt| j�}|j� }t|d�� }dj|j� dd � �| _ W d Q R X n:| j tj
kr�| jj � j � }|s�| jj
j� }t|�d �|| _ d S )N�r� r )r% r �KEYFILEr r# �find_key�open�join� readlinesr$ �REPOZload_key�decodeZ _locationZcanonical_pathr )r' �k�target�fd�key_data�locr r r � load_keyblob3 s
$
zKeyManager.load_keyblobc C sr | j tjkrNt| j�}y |j� }W n tk
r@ |j|�}Y nX | j|� n | j tj krn| jj
| j j d�� d S )Nzutf-8)
r% r r, r r# r- r Zget_new_target�
store_keyfiler1 Zsave_keyr$ �encode)r' �argsr3 r4 r r r �
store_keyblobB s
zKeyManager.store_keyblobc C s8 dt jt| jj�f }|| j7 }| jjd�s4|d7 }|S )Nz%s %s
�
)r �FILE_IDr r# �idr$ �endswith)r' r r r r �get_keyfile_dataN s
zKeyManager.get_keyfile_datac C s( t |d��}|j| j� � W d Q R X d S )N�w)r �writerA )r' r4 r5 r r r r9 U s zKeyManager.store_keyfilec C s |d kr d}| j |� d S )N�-)r9 )r' �pathr r r �exportY s zKeyManager.exportc C sX |d kr d}t |d��8}| j� }tjdd�}|jd|j� d �}|j|� W d Q R X d S )NrD �wbZborgz
paperkey.htmls </textarea>)r rA �pkgutil�get_data�replacer: rC )r' rE r5 r6 Zhtmlr r r � export_qr_ s zKeyManager.export_qrc C s |d kr d}dd� }d}t | j�}|d7 }t|�d d }t| jj�d d� }t|d�}|d j|||�||�tt|�d
| d
| j d �d ��7 }d
}x^t|�r�|d7 }|d d� } t|j
d dd�| d �}
|dj||t| ��|
�7 }|dd � }q�W t |d��} | j |� W d Q R X d S )NrD c S sB d}d}x4| D ],}|r*|d dkr*|d7 }||7 }|d7 }qW |S )Nr+ r � � r r )�s�ret�iZchr r r �groupedm s
z+KeyManager.export_paperkey.<locals>.groupedz:To restore key use borg key import --paper /path/to/repo
zBORG PAPER KEY v1
� � � z id: {0:d} / {1} / {2} - {3}
�/�asciir r r �big)� byteorderz{0:2d}: {1} - {2}
rB )
r r$ �lenr r# r? r! �format�strr: �to_bytesr rC ) r' rE rQ rF Zbinary�lines�repoidZcomplete_checksum�idxZbinline�checksumr5 r r r �export_paperkeyi s,
(
zKeyManager.export_paperkeyc
C s| t j}|d t| jj� d }t|jd��>}|jt|��}||krZ|j |�sTt
� �nt � �|j� | _ W d Q R X | j
|� d S )NrM r= r* )r r> r r# r? r rE �readrY �
startswithr r r$ r< )r' r; Zfile_idZ
first_liner5 Zfile_first_liner r r �import_keyfile� s
zKeyManager.import_keyfilec ) C s� y dd l }W n tk
r( td� Y nX t| jj�d d� }�y<�x4x�td�jdd�}|dkrntd�rnt � �y|j
d�\}}W n t k
r� td � wHY nX y|j
d
�\}}} W n t k
r� td � wHY nX t |j
� jd �d
�|kr�td� qHy t|�}
W n t k
�r td� Y nX P qHW ||k�r8t� �d} d} x�tdj| ��}
|
jdd�}
|
dk�rvtd��rvt � �y|
j
d�\}}W n" t k
�r� td � �wBY nX y t|�}W n$ tjk
�r� td� �wBY nX t | jd
dd�| d
�|k�rtdj| �� �qB| |7 } | |
k�r"P | d7 } �qBW t | d�| k�rJtd� qFdjtjt| �jd ���d | _| j |� P qFW W n t k
�r� td� d S X d S )Nr z?Note: No line editing available due to missing readline supportrS zid: rM r+ zAbort import? [yN]:rD z1each line must contain exactly one '-', try againrU z5the id line must contain exactly three '/', try againrV r z0line checksum did not match, try same line againz#internal error while parsing length� r z{0:2d}: z8only characters 0-9 and a-f and '-' are valid, try againrW )rX z/line checksum did not match, try line {0} againrT zIThe overall checksum did not match, retry or enter a blank line to abort.r= z
- aborted) �readline� ImportError�printr r# r? �inputrJ r
�EOFError�split�
ValueErrorr! �lowerr: �intr rZ r �binasciir r\ r/ �textwrapZwrapr r2 r$ r< )r' r; rf r^ Zidliner r` Zid_linesZ id_repoidZid_complete_checksumr] �resultr_ Zinline�partr r r �import_paperkey� s�
zKeyManager.import_paperkeyN)
r r r r) r8 r< rA r9 rF rK ra rd rs r r r r r" # s
$r" ) ro rH rp 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