[omemo] Better handle key exchange messages

1 files changed, 6 insertions, 3 deletions

diff --git a/omemo/omemo/state.py b/omemo/omemo/state.py
index 4086635..050db17 100644
--- a/omemo/omemo/state.py
+++ b/omemo/omemo/state.py
@@ -211,9 +211,12 @@ class OmemoState:
             log.warning('Duplicate message found ' + str(e.args))
             return
 
-        result = decrypt(key, iv, payload)
-
-        log.debug("Decrypted Message => " + result)
+        if payload is None:
+            result = None
+            log.debug("Decrypted Key Exchange Message")
+        else:
+            result = decrypt(key, iv, payload)
+            log.debug("Decrypted Message => " + result)
         return result
 
     def create_msg(self, from_jid, jid, plaintext):