­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 yb @sddlZddlZddlZddlmZmZmZddlmZddlm Z m Z m Z m Z m Z mZddlmZddlmZmZmZmZmZGd d d e ZGd d d e ZGd dde ZGddde ZddZGdddZdS)N) unhexlify a2b_base64 b2a_base64)sha256)ManifestNoManifestErrorErroryes bin_to_hex dash_open) Repository) KeyfileKeyKeyfileNotFoundErrorRepoKeyNotFoundErrorKeyBlobStorage identify_keyc@seZdZdZdS)UnencryptedRepoz9Keymanagement not available for unencrypted repositories.N)__name__ __module__ __qualname____doc__rr"/usr/lib64/python3.6/keymanager.pyr src@seZdZdZdS)UnknownKeyTypezKeytype {0} is unknown.N)rrrrrrrrrsrc@seZdZdZdS)RepoIdMismatchzHThis key backup seems to be for a different backup repository, aborting.N)rrrrrrrrrsrc@seZdZdZdS)NotABorgKeyFilez-This file is not a borg key backup, aborting.N)rrrrrrrrrsrcCs t}|j||jd|S)N)rupdateZ hexdigest)dataZnumhrrrsha256_truncateds r!c@s\eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ dS) KeyManagerc Csf||_d|_d|_y|jjtj}Wntjk r>tYnXt |}|j |_|jt j krbt dS)N) repositorykeyblobkeyblob_storagegetrZ MANIFEST_IDr ZObjectNotFoundrrZSTORAGErZ NO_STORAGEr)selfr#Z manifest_datakeyrrr__init__$s  zKeyManager.__init__c Cs|jtjkrNt|j}|j}t|d}dj|jdd|_ WdQRXn:|jtj kr|jj j }|s|jj j}t|d||_ dS)Nrr)r%rKEYFILErr#find_keyopenjoin readlinesr$REPOZload_keydecodeZ _locationZcanonical_pathr)r'ktargetfdkey_datalocrrr load_keyblob3s   $   zKeyManager.load_keyblobc Csr|jtjkrNt|j}y |j}Wntk r@|j|}YnX|j|n |jtj krn|jj |j j ddS)Nzutf-8) r%rr,rr#r-rZget_new_target store_keyfiler1Zsave_keyr$encode)r'argsr3r4rrr store_keyblobBs     zKeyManager.store_keyblobcCs8dtjt|jjf}||j7}|jjds4|d7}|S)Nz%s %s  )rFILE_IDr r#idr$endswith)r'rrrrget_keyfile_dataNs   zKeyManager.get_keyfile_datac Cs(t|d}|j|jWdQRXdS)Nw)r writerA)r'r4r5rrrr9Us zKeyManager.store_keyfilecCs|dkr d}|j|dS)N-)r9)r'pathrrrexportYszKeyManager.exportc CsX|dkr d}t|d8}|j}tjdd}|jd|jd}|j|WdQRXdS)NrDwbZborgz paperkey.htmls )r rApkgutilget_datareplacer:rC)r'rEr5r6Zhtmlrrr export_qr_s  zKeyManager.export_qrc Cs|dkr d}dd}d}t|j}|d7}t|dd}t|jjdd}t|d}|d j|||||tt|d |d |j d d 7}d }x^t|r|d7}|dd} t|j d dd| d } |dj||t| | 7}|dd}qWt |d} | j |WdQRXdS)NrDcSsBd}d}x4|D],}|r*|ddkr*|d7}||7}|d7}qW|S)Nr+r rr)sretiZchrrrgroupedms  z+KeyManager.export_paperkey..groupedz:To restore key use borg key import --paper /path/to/repo zBORG PAPER KEY v1  zid: {0:d} / {1} / {2} - {3} /asciirrrbig) byteorderz{0:2d}: {1} - {2} rB) rr$lenr r#r?r!formatstrr:to_bytesr rC) r'rErQrFZbinarylinesrepoidZcomplete_checksumidxZbinlinechecksumr5rrrexport_paperkeyis,   (   zKeyManager.export_paperkeyc Cs|tj}|dt|jjd}t|jd>}|jt|}||krZ|j |sTt nt |j|_ WdQRX|j |dS)NrMr=r*)rr>r r#r?r rEreadrY startswithrrr$r<)r'r;Zfile_idZ first_liner5Zfile_first_linerrrimport_keyfiles zKeyManager.import_keyfilec)Csy ddl}Wntk r(tdYnXt|jjdd}y<x4xtdjdd}|dkrntdrnt y|j d\}}Wnt k rtd wHYnXy|j d \}}} Wnt k rtd wHYnXt |j jd d |krtdqHy t|} Wnt k r tdYnXPqHW||kr8td} d} xtdj| } | jdd} | dkrvtdrvt y| j d\}}Wn"t k rtd wBYnXy t|}Wn$tjk rtdwBYnXt | jd dd|d |krtdj| qB| |7} | | kr"P| d7} qBWt | d| krJtdqFdjtjt| jd d|_|j|PqFWWnt k rtddSXdS)Nrz?Note: No line editing available due to missing readline supportrSzid: rMr+zAbort import? [yN]:rDz1each line must contain exactly one '-', try againrUz5the id line must contain exactly three '/', try againrVrz0line checksum did not match, try same line againz#internal error while parsing lengthrz{0:2d}: z8only characters 0-9 and a-f and '-' are valid, try againrW)rXz/line checksum did not match, try line {0} againrTzIThe overall checksum did not match, retry or enter a blank line to abort.r=z - aborted)readline ImportErrorprintr r#r?inputrJr EOFErrorsplit ValueErrorr!lowerr:intrrZrbinasciir r\r/textwrapZwraprr2r$r<)r'r;rfr^Zidlinerr`Zid_linesZ id_repoidZid_complete_checksumr]resultr_Zinlinepartrrrimport_paperkeys             zKeyManager.import_paperkeyN) rrrr)r8r<rAr9rFrKrardrsrrrrr"#s  $r")rorHrprrrZhashlibrZhelpersrrr r r r r#r r(rrrrrrrrrr!r"rrrrs