­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 yb@svddlZddlZddlmZddlmZddlmZddlmZddl m Z dd l m Z m Z dZdZGd d d ZdS)N) unhexlify)get_security_dir) bin_to_hex)SaveFile)InvalidRPCMethod) bytes_to_long long_to_bytes@c@s<eZdZddZddZddZddZd d Zd d Zd S) NonceManagercCs4||_||_d|_||_tjjt|jjd|_ dS)NZnonce) repository enc_cipherend_of_nonce_reservationmanifest_nonceospathjoinrZid_str nonce_file)selfrrrr/usr/lib64/python3.6/nonces.py__init__s zNonceManager.__init__cCsBy(t|jd}tt|jSQRXWntk r<dSXdS)Nr)openrr rreadFileNotFoundError)rfdrrrget_local_free_nonces z!NonceManager.get_local_free_noncec CsD|j|krtdt|jdd}|jtt|WdQRXdS)Nz6nonce space reservation with mismatched previous stateF)Zbinary)r Exceptionrrwriterr )rnext_unreserved start_noncerrrrcommit_local_nonce_reservations z+NonceManager.commit_local_nonce_reservationc CsDy |jjStk r>tjjddd|_dd|_dSXdS)NzTPlease upgrade to borg version 1.1+ on the server for safer AES-CTR nonce handling. cSsdS)Nrrrrr+sz2NonceManager.get_repo_free_nonce..cSsdS)Nr)r"r#rrrr%,s)rZget_free_noncersysstderrr!get_repo_free_noncecommit_repo_nonce_reservation)rrrrr(%s    z NonceManager.get_repo_free_noncecCs|jj||dS)N)rZcommit_nonce_reservation)rr"r#rrrr)/sz*NonceManager.commit_repo_nonce_reservationcCs|jr8tj|jjdd}||jks&t|||jkr8dS|j}|j}tdd|||j |jfD}||t }|t ks~t|jdkr|jj d|j dddn"||jkr|jj d|j ddd|j|||j||||_dS)NZbig) byteordercss|]}|dk r|VqdS)Nr).0xrrr Jsz2NonceManager.ensure_reservation..)rint from_bytesrZivAssertionErrorr(rmaxrNONCE_SPACE_RESERVATIONMAX_REPRESENTABLE_NONCEresetto_bytesr)r$)rZnonce_space_neededZ next_nonceZrepo_free_nonceZlocal_free_nonceZfree_nonce_spaceZreservation_endrrrensure_reservation2s"      zNonceManager.ensure_reservationN) __name__ __module__ __qualname__rrr$r(r)r7rrrrr s  r lli)rr&ZbinasciirZhelpersrrplatformrZremoterZ low_levelr r r4r3r rrrrs