o g< @sUddlmZddlmZddlmZmZddlm Z  e Z e e d<eZe e d<eZe e d<eZe e d<eZe e d <eeZeZe e d <eZe e d <eZe e d <eZe e d <eZ e e d<e eZ!e"Z#e e d<e$Z%e e d<e&Z'e e d<e(Z)e e d<e*Z+e e d<e+e)Z,de-dee-de-de-de-f ddZ.de-dee-de-de-de-f ddZ/de-dee-de-de-de-f ddZ0de-dee-de-de-de-f d d!Z1de-dee-de-de-de-f d"d#Z2de-dee-de-de-de-f d$d%Z3d&S)')Optional) exceptions)ffilib)ensure*crypto_aead_chacha20poly1305_ietf_KEYBYTES+crypto_aead_chacha20poly1305_ietf_NSECBYTES+crypto_aead_chacha20poly1305_ietf_NPUBBYTES(crypto_aead_chacha20poly1305_ietf_ABYTES2crypto_aead_chacha20poly1305_ietf_MESSAGEBYTES_MAX%crypto_aead_chacha20poly1305_KEYBYTES&crypto_aead_chacha20poly1305_NSECBYTES&crypto_aead_chacha20poly1305_NPUBBYTES#crypto_aead_chacha20poly1305_ABYTES-crypto_aead_chacha20poly1305_MESSAGEBYTES_MAX+crypto_aead_xchacha20poly1305_ietf_KEYBYTES,crypto_aead_xchacha20poly1305_ietf_NSECBYTES,crypto_aead_xchacha20poly1305_ietf_NPUBBYTES)crypto_aead_xchacha20poly1305_ietf_ABYTES3crypto_aead_xchacha20poly1305_ietf_MESSAGEBYTES_MAXmessageaadnoncekeyreturnc Cs(tt|tdtjdt|}t|tkdttjdtt|tp%|dudtjdtt|to6t|t kdt tjdtt|toJt|t kdt tjd|r\|}t|}nt j }d}|t }t d }t d |} t| |||||t j || } t| dkd tjdt | |dddS) aZ Encrypt the given ``message`` using the IETF ratified chacha20poly1305 construction described in RFC7539. :param message: :type message: bytes :param aad: :type aad: Optional[bytes] :param nonce: :type nonce: bytes :param key: :type key: bytes :return: authenticated ciphertext :rtype: bytes Input message type must be bytesraising%Message must be at most {} bytes longN%Additional data must be bytes or None,Nonce must be a {} bytes long bytes sequence*Key must be a {} bytes long bytes sequencerunsigned long long *unsigned char[]Encryption failed.)r isinstancebytesexc TypeErrorlenr format ValueErrorr rrNULLr newr)crypto_aead_chacha20poly1305_ietf_encrypt CryptoErrorbuffer rrrrmlen_aadaalenmxoutclen ciphertextresr9D/usr/local/lib/python3.10/dist-packages/nacl/bindings/crypto_aead.pyr.Ws^        r.r7c C(tt|tdtjdt|}t|tkdttjdtt|tp%|dudtjdtt|to6t|t kdt tjdtt|toJt|t kdt tjd|t }t d}t d |}|rk|}t|} nt j}d } t||t j|||| || } t| d kd tjdt ||d ddS) aR Decrypt the given ``ciphertext`` using the IETF ratified chacha20poly1305 construction described in RFC7539. :param ciphertext: :type ciphertext: bytes :param aad: :type aad: Optional[bytes] :param nonce: :type nonce: bytes :param key: :type key: bytes :return: message :rtype: bytes #Input ciphertext type must be bytesr(Ciphertext must be at most {} bytes longNrr r!r"r#rDecryption failed.)rr%r&r'r(r)*_aead_chacha20poly1305_ietf_CRYPTBYTES_MAXr*r+r rr rr-r,r)crypto_aead_chacha20poly1305_ietf_decryptr/r0 r7rrrr6r5r2rr3r4r8r9r9r:r@^       r@c C0tt|tdtjdt|}t|tkdttjdtt|tp%|dudtjdtt|to6t|t kdt tjdtt|toJt|t kdt tjd|r\|}t|}nt j }d}t|}|t }t d }t d |} t| |||||t j || } t| dkd tjdt | |dddS) a[ Encrypt the given ``message`` using the "legacy" construction described in draft-agl-tls-chacha20poly1305. :param message: :type message: bytes :param aad: :type aad: Optional[bytes] :param nonce: :type nonce: bytes :param key: :type key: bytes :return: authenticated ciphertext :rtype: bytes rrrNrr r!rr"r#r$)rr%r&r'r(r)rr*r+rr rr,r r-r$crypto_aead_chacha20poly1305_encryptr/r0r1r9r9r:rD`        rDc Cr;) al Decrypt the given ``ciphertext`` using the "legacy" construction described in draft-agl-tls-chacha20poly1305. :param ciphertext: authenticated ciphertext :type ciphertext: bytes :param aad: :type aad: Optional[bytes] :param nonce: :type nonce: bytes :param key: :type key: bytes :return: message :rtype: bytes r<rr=Nrr r!r"r#rr>)rr%r&r'r(r)%_aead_chacha20poly1305_CRYPTBYTES_MAXr*r+rr rrr-r,r$crypto_aead_chacha20poly1305_decryptr/r0rAr9r9r:rGErBrGc CrC) aD Encrypt the given ``message`` using the long-nonces xchacha20poly1305 construction. :param message: :type message: bytes :param aad: :type aad: Optional[bytes] :param nonce: :type nonce: bytes :param key: :type key: bytes :return: authenticated ciphertext :rtype: bytes rrrNrr r!rr"r#r$)rr%r&r'r(r)rr*r+rrrr,rr-r*crypto_aead_xchacha20poly1305_ietf_encryptr/r0r1r9r9r:rHrErHc Cr;) aU Decrypt the given ``ciphertext`` using the long-nonces xchacha20poly1305 construction. :param ciphertext: authenticated ciphertext :type ciphertext: bytes :param aad: :type aad: Optional[bytes] :param nonce: :type nonce: bytes :param key: :type key: bytes :return: message :rtype: bytes r<rr=Nrr r!r"r#rr>)rr%r&r'r(r)+_aead_xchacha20poly1305_ietf_CRYPTBYTES_MAXr*r+rrrrr-r,r*crypto_aead_xchacha20poly1305_ietf_decryptr/r0rAr9r9r:rJs^       rJN)4typingrnaclrr' nacl._sodiumrrnacl.exceptionsr*crypto_aead_chacha20poly1305_ietf_keybytesrint__annotations__+crypto_aead_chacha20poly1305_ietf_nsecbytesr+crypto_aead_chacha20poly1305_ietf_npubbytesr (crypto_aead_chacha20poly1305_ietf_abytesr 2crypto_aead_chacha20poly1305_ietf_messagebytes_maxr r?%crypto_aead_chacha20poly1305_keybytesr &crypto_aead_chacha20poly1305_nsecbytesr &crypto_aead_chacha20poly1305_npubbytesr#crypto_aead_chacha20poly1305_abytesr-crypto_aead_chacha20poly1305_messagebytes_maxrrF+crypto_aead_xchacha20poly1305_ietf_keybytesr,crypto_aead_xchacha20poly1305_ietf_nsecbytesr,crypto_aead_xchacha20poly1305_ietf_npubbytesr)crypto_aead_xchacha20poly1305_ietf_abytesr3crypto_aead_xchacha20poly1305_ietf_messagebytes_maxrrIr&r.r@rDrGrHrJr9r9r9r:s                  O O P O P