From baefb7728fedafb4611974b4a69dec89a18d557f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20H=C3=B6rist?= <philipp@hoerist.com>
Date: Sat, 15 Dec 2018 16:15:31 +0100
Subject: [PATCH] [omemo] Better handle key exchange messages

---
 omemo/omemo/state.py | 9 ++++++---
 1 file 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):