o g+ @sUddlmZmZmZmZddlmZddlm Z m Z ddl m Z e Zeed<e Zeed<e Zeed<e Zeed<e Zeed <e Zeed <e Zeed <e Zeed <e Z eed <de!fddZ"GdddZ#de#de!de!fddZ$defde#de!dee!dede!f ddZ%de#de!de!ddfddZ& d%de#d e!dee!dee!effd!d"Z'de#ddfd#d$Z(dS)&) ByteStringOptionalTuplecast) exceptions)ffilib)ensure,crypto_secretstream_xchacha20poly1305_ABYTES1crypto_secretstream_xchacha20poly1305_HEADERBYTES.crypto_secretstream_xchacha20poly1305_KEYBYTES6crypto_secretstream_xchacha20poly1305_MESSAGEBYTES_MAX0crypto_secretstream_xchacha20poly1305_STATEBYTES1crypto_secretstream_xchacha20poly1305_TAG_MESSAGE.crypto_secretstream_xchacha20poly1305_TAG_PUSH/crypto_secretstream_xchacha20poly1305_TAG_REKEY/crypto_secretstream_xchacha20poly1305_TAG_FINALreturncCs(tdt}t|t|ddS)zd Generate a key for use with :func:`.crypto_secretstream_xchacha20poly1305_init_push`. unsigned char[]N)rnewr r,crypto_secretstream_xchacha20poly1305_keygenbuffer)keybufrL/usr/local/lib/python3.10/dist-packages/nacl/bindings/crypto_secretstream.pyr4s  rc@s"eZdZdZgdZdddZdS)+crypto_secretstream_xchacha20poly1305_statezN An object wrapping the crypto_secretstream_xchacha20poly1305 state. )statebufrawbuftagbufrNcCstdt|_d|_d|_dS)z Initialize a clean state object.rN)rrrrrr)selfrrr__init__Js  z4crypto_secretstream_xchacha20poly1305_state.__init__)rN)__name__ __module__ __qualname____doc__ __slots__r rrrrrBsrstatekeycCstt|tdtjdtt|tdtjdtt|tkdtjdt dt }t |j||}t|dkdtjdt |ddS) a\ Initialize a crypto_secretstream_xchacha20poly1305 encryption buffer. :param state: a secretstream state object :type state: crypto_secretstream_xchacha20poly1305_state :param key: must be :data:`.crypto_secretstream_xchacha20poly1305_KEYBYTES` long :type key: bytes :return: header :rtype: bytes BState must be a crypto_secretstream_xchacha20poly1305_state objectraisingKey must be a bytes sequenceInvalid key lengthzunsigned char []rUnexpected failureN)r isinstancerexc TypeErrorbyteslenr ValueErrorrrr r/crypto_secretstream_xchacha20poly1305_init_pushr RuntimeErrorr)r&r' headerbufrcrrrr4Us0 r4Nmadtagc Cstt|tdtjdtt|tdtjdtt|tkdtjdt|dup+t|tdtjdt|t }|j dusCt|j |krJt d||_ |durTt j }d}nt|}t|j|j t j |t||||}t|dkd tjdt |j |ddS) a Add an encrypted message to the secret stream. :param state: a secretstream state object :type state: crypto_secretstream_xchacha20poly1305_state :param m: the message to encrypt, the maximum length of an individual message is :data:`.crypto_secretstream_xchacha20poly1305_MESSAGEBYTES_MAX`. :type m: bytes :param ad: additional data to include in the authentication tag :type ad: bytes or None :param tag: the message tag, usually :data:`.crypto_secretstream_xchacha20poly1305_TAG_MESSAGE` or :data:`.crypto_secretstream_xchacha20poly1305_TAG_FINAL`. :type tag: int :return: ciphertext :rtype: bytes r(r)zMessage is not byteszMessage is too longN%Additional data must be bytes or Nonerrr-)r r.rr/r0r1r2r r3r rrrNULLr*crypto_secretstream_xchacha20poly1305_pushrr5r)r&r8r9r:clenadlenr7rrrr=sF   r=headercCstt|tdtjdtt|tdtjdtt|tkdtjdtt|tdtjdtt|t kdtjd|j durDt d|_ t |j||}t|d kd tjddS) a Initialize a crypto_secretstream_xchacha20poly1305 decryption buffer. :param state: a secretstream state object :type state: crypto_secretstream_xchacha20poly1305_state :param header: must be :data:`.crypto_secretstream_xchacha20poly1305_HEADERBYTES` long :type header: bytes :param key: must be :data:`.crypto_secretstream_xchacha20poly1305_KEYBYTES` long :type key: bytes r(r)zHeader must be a bytes sequencezInvalid header lengthr+r,Nzunsigned char *rr-)r r.rr/r0r1r2r r3r rrrr/crypto_secretstream_xchacha20poly1305_init_pullrr5)r&r@r'r7rrrrAs>    rAcc CsDtt|tdtjdt|jdudtjdtt|tdtjdtt|t kdtjdtt|t t kdtjdt|dupDt|tdtjdt|t }|j dus\t|j |krct d ||_ |durmt j}d }nt|}t|j|j t j|j|t|||}t|d kd tjdt |j |ddttt|jd fS) aM Read a decrypted message from the secret stream. :param state: a secretstream state object :type state: crypto_secretstream_xchacha20poly1305_state :param c: the ciphertext to decrypt, the maximum length of an individual ciphertext is :data:`.crypto_secretstream_xchacha20poly1305_MESSAGEBYTES_MAX` + :data:`.crypto_secretstream_xchacha20poly1305_ABYTES`. :type c: bytes :param ad: additional data to include in the authentication tag :type ad: bytes or None :return: (message, tag) :rtype: (bytes, int) r(r)NzOState must be initialized using crypto_secretstream_xchacha20poly1305_init_pullzCiphertext is not byteszCiphertext is too shortzCiphertext is too longr;rrr-)r r.rr/r0rr3r1r2r r rrrr<r*crypto_secretstream_xchacha20poly1305_pullrr5rintr)r&rBr9mlenr?r7rrrrCsn    rCcCs&tt|tdtjdt|jdS)a Explicitly change the encryption key in the stream. Normally the stream is re-keyed as needed or an explicit ``tag`` of :data:`.crypto_secretstream_xchacha20poly1305_TAG_REKEY` is added to a message to ensure forward secrecy, but this method can be used instead if the re-keying is controlled without adding the tag. :param state: a secretstream state object :type state: crypto_secretstream_xchacha20poly1305_state r(r)N)r r.rr/r0r+crypto_secretstream_xchacha20poly1305_rekeyr)r&rrrrFQs rF)N))typingrrrrnaclrr/ nacl._sodiumrrnacl.exceptionsr ,crypto_secretstream_xchacha20poly1305_abytesr rD__annotations__1crypto_secretstream_xchacha20poly1305_headerbytesr .crypto_secretstream_xchacha20poly1305_keybytesr 6crypto_secretstream_xchacha20poly1305_messagebytes_maxr 0crypto_secretstream_xchacha20poly1305_statebytesr1crypto_secretstream_xchacha20poly1305_tag_messager.crypto_secretstream_xchacha20poly1305_tag_pushr/crypto_secretstream_xchacha20poly1305_tag_rekeyr/crypto_secretstream_xchacha20poly1305_tag_finalrr1rrr4r=rArCrFrrrrs            / C 8  X