U
vf+ @ sj U d dl mZmZmZmZ d dlmZ d dlm Z m
Z
d dl m Z e
Zeed< e
Zeed< e
Zeed< e
Zeed< e
Zeed < e
Zeed
< e
Z eed < e
Z eed < e
Z eed
< e!dddZ"G dd dZ#e#e!e!dddZ$defe#e!ee! ee!dddZ%e#e!e!dddd Z&d#e#e!ee! ee!ef d d d Z'e#dd d!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_FINALreturnc C s( t dt} t| t | dd S )zd
Generate a key for use with
:func:`.crypto_secretstream_xchacha20poly1305_init_push`.
unsigned char[]N)r newr r ,crypto_secretstream_xchacha20poly1305_keygenbuffer)Zkeybuf r T/opt/alt/python38/lib64/python3.8/site-packages/nacl/bindings/crypto_secretstream.pyr 4 s
r c @ s( e Zd ZdZdddgZddddZdS ) +crypto_secretstream_xchacha20poly1305_statezN
An object wrapping the crypto_secretstream_xchacha20poly1305 state.
statebufrawbuftagbufNr c C s t dt| _d| _d| _dS )z Initialize a clean state object.r N)r r r r r r )selfr r r __init__J s z4crypto_secretstream_xchacha20poly1305_state.__init__)__name__
__module__ __qualname____doc__ __slots__r r r r r r B s
r )statekeyr c C s t t| tdtjd t t|tdtjd t t|tkdtjd t
dt }t
| j||}t |dkdtjd t |dd S ) 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 objectZraising Key must be a bytes sequenceInvalid key lengthzunsigned char []r Unexpected failureN)r
isinstancer exc TypeErrorbyteslenr
ValueErrorr r r r /crypto_secretstream_xchacha20poly1305_init_pushr RuntimeErrorr )r& r' Z headerbufrcr r r r3 U s4
r3 N)r& madtagr c
C s t t| tdtjd t t|tdtjd t t|tkdtjd t |dkpVt|tdtjd t|t }| j
dkst| j
|k rt d|| _
|dkrt j
}d}nt|}t| j| j
t j
|t||||}t |dkd tjd t | j
|dd S )
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 Noner r r, )r r- r r. r/ r0 r1 r
r2 r
r r r NULLr *crypto_secretstream_xchacha20poly1305_pushr r4 r )r& r6 r7 r8 Zclenadlenr5 r r r r; sF
r; )r& headerr' r c C s t t| tdtjd t t|tdtjd t t|tkdtjd t t|tdtjd t t|t kdtjd | j
dkrt d| _
t
| j||}t |d kd
tjd dS ) 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) z Header must be a bytes sequencezInvalid header lengthr* r+ Nzunsigned char *r r, )r r- r r. r/ r0 r1 r r2 r r r r r /crypto_secretstream_xchacha20poly1305_init_pullr r4 )r& r= r' r5 r r r r> sB
r> )r& cr7 r c
C sD t t| tdtjd t | jdk dtjd t t|tdtjd t t|t kdtjd t t|t
t kdtjd t |dkpt|tdtjd t|t }| j dkst| j |k rt
d || _ |dkrt j}d
}nt|}t| j| j t j| j|t|||}t |d
kd tjd t | j |dd ttt| 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 longr9 r r r, )r r- r r. r/ r r2 r0 r1 r
r
r r r r: r *crypto_secretstream_xchacha20poly1305_pullr r4 r intr )r& r? r7 mlenr< r5 r r r r@ sn
r@ )r& r c C s& t t| tdtjd t| j dS )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- r r. r/ r +crypto_secretstream_xchacha20poly1305_rekeyr )r& r r r rC Q s rC )N))typingr r r r Znaclr r. Z nacl._sodiumr r Znacl.exceptionsr Z,crypto_secretstream_xchacha20poly1305_abytesr
rA __annotations__Z1crypto_secretstream_xchacha20poly1305_headerbytesr Z.crypto_secretstream_xchacha20poly1305_keybytesr Z6crypto_secretstream_xchacha20poly1305_messagebytes_maxr
Z0crypto_secretstream_xchacha20poly1305_statebytesr Z1crypto_secretstream_xchacha20poly1305_tag_messager Z.crypto_secretstream_xchacha20poly1305_tag_pushr Z/crypto_secretstream_xchacha20poly1305_tag_rekeyr Z/crypto_secretstream_xchacha20poly1305_tag_finalr r0 r r r3 r; r> r@ rC r r r r sh
/ D 8
Y