[omemo] Add auth tag to key instead of payload

1 files changed, 2 insertions, 6 deletions

diff --git a/omemo/omemo/state.py b/omemo/omemo/state.py
index 2a899d1..49dd7cd 100644
--- a/omemo/omemo/state.py
+++ b/omemo/omemo/state.py
@@ -228,9 +228,7 @@ class OmemoState:
 

         payload, tag = encrypt(key, iv, plaintext)

 

-        # for XEP-384 Compliance uncomment

-        # key += tag

-        payload += tag

+        key += tag

 

         # Encrypt the message key with for each of receivers devices

         for device in devices_list:

@@ -289,9 +287,7 @@ class OmemoState:
 

         payload, tag = encrypt(key, iv, plaintext)

 

-        # for XEP-384 Compliance uncomment

-        # key += tag

-        payload += tag

+        key += tag

 

         for tup in devices_list:

             self.get_session_cipher(tup[0], tup[1])