o +ke@svddlZddlZddlmZddlmZddlmZddlmZddl m Z dd l m Z m Z d Zd ZGd d d ZdS)N) unhexlify)get_security_dir) bin_to_hex)SaveFile)InvalidRPCMethod) bytes_to_long long_to_byteslic@s<eZdZddZddZddZddZd d Zd d Zd S) NonceManagercCs.||_d|_||_tjt|jjd|_dS)Nnonce) repositoryend_of_nonce_reservationmanifest_nonceospathjoinrZid_str nonce_file)selfr rr6usr/lib/python3.10/site-packages/borg/crypto/nonces.py__init__szNonceManager.__init__cCsXz!t|j}tt|WdWS1swYWdSty+YdSwN)openrr rreadFileNotFoundError)rfdrrrget_local_free_nonces ( z!NonceManager.get_local_free_noncecCsZ||kr tdt|jdd}|tt|WddS1s&wYdS)Nz6nonce space reservation with mismatched previous stateF)binary)r Exceptionrrwriterr )rnext_unreserved start_noncerrrrcommit_local_nonce_reservations "z+NonceManager.commit_local_nonce_reservationcCsBz|jWSty tjddd|_dd|_YdSw)NzTPlease upgrade to borg version 1.1+ on the server for safer AES-CTR nonce handling. cSdSrrrrrr*z2NonceManager.get_repo_free_nonce..cSr$rr)r!r"rrrr%+r&)r Zget_free_noncersysstderrr get_repo_free_noncecommit_repo_nonce_reservation)rrrrr)$s     z NonceManager.get_repo_free_noncecCs|j||dSr)r Zcommit_nonce_reservation)rr!r"rrrr*.sz*NonceManager.commit_repo_nonce_reservationcCs|jr|}||jks J|||jkr|S|}|}tdd|||j|jfD}||t}|tks8J||||||||_|S)a Call this before doing encryption, give current, yet unused, integer IV as and the amount of subsequent (counter-like) IVs needed as . Return value is the IV (counter) integer you shall use for encryption. Note: this method may return the you gave, if a reservation for it exists or can be established, so make sure you give a unused nonce. css|] }|dur|VqdSrr).0xrrr Qsz2NonceManager.ensure_reservation..) rr)rmaxrNONCE_SPACE_RESERVATIONMAX_REPRESENTABLE_NONCEr*r#)rr Znonce_space_neededZ next_nonceZrepo_free_nonceZlocal_free_nonceZfree_nonce_spaceZreservation_endrrrensure_reservation1s    zNonceManager.ensure_reservationN) __name__ __module__ __qualname__rrr#r)r*r1rrrrr s r )rr'binasciirhelpersrrplatformrZremoterZ low_levelr r r0r/r rrrrs