From 51de021fb736f4ef5921d94cd0406f768915af54 Mon Sep 17 00:00:00 2001
From: Christian Schneppe <christian@pix-art.de>
Date: Sat, 14 Jan 2017 20:30:47 +0100
Subject: treat omemo keys >= 32 bytes as containing auth tag. add config flag
 to put auth tag in key

---
 .../messenger/crypto/axolotl/AxolotlService.java   | 46 +++++++++++-----------
 1 file changed, 24 insertions(+), 22 deletions(-)

(limited to 'src/main/java/de/pixart/messenger/crypto/axolotl/AxolotlService.java')

diff --git a/src/main/java/de/pixart/messenger/crypto/axolotl/AxolotlService.java b/src/main/java/de/pixart/messenger/crypto/axolotl/AxolotlService.java
index 16a4919fc..11f51eaa2 100644
--- a/src/main/java/de/pixart/messenger/crypto/axolotl/AxolotlService.java
+++ b/src/main/java/de/pixart/messenger/crypto/axolotl/AxolotlService.java
@@ -1003,14 +1003,12 @@ public class AxolotlService implements OnAdvancedStreamFeaturesLoaded {
     }
 
     @Nullable
-    private XmppAxolotlMessage buildHeader(Conversation conversation) {
-        final XmppAxolotlMessage axolotlMessage = new XmppAxolotlMessage(
-                account.getJid().toBareJid(), getOwnDeviceId());
+    private boolean buildHeader(XmppAxolotlMessage axolotlMessage, Conversation conversation) {
 
         Set<XmppAxolotlSession> remoteSessions = findSessionsForConversation(conversation);
         Collection<XmppAxolotlSession> ownSessions = findOwnSessions();
         if (remoteSessions.isEmpty()) {
-            return null;
+            return false;
         }
         for (XmppAxolotlSession session : remoteSessions) {
             axolotlMessage.addDevice(session);
@@ -1019,26 +1017,26 @@ public class AxolotlService implements OnAdvancedStreamFeaturesLoaded {
             axolotlMessage.addDevice(session);
         }
 
-        return axolotlMessage;
+        return true;
     }
 
     @Nullable
     public XmppAxolotlMessage encrypt(Message message) {
-        XmppAxolotlMessage axolotlMessage = buildHeader(message.getConversation());
-
-        if (axolotlMessage != null) {
-            final String content;
-            if (message.hasFileOnRemoteHost()) {
-                content = message.getFileParams().url.toString();
-            } else {
-                content = message.getBody();
-            }
-            try {
-                axolotlMessage.encrypt(content);
-            } catch (CryptoFailedException e) {
-                Log.w(Config.LOGTAG, getLogprefix(account) + "Failed to encrypt message: " + e.getMessage());
-                return null;
-            }
+        final XmppAxolotlMessage axolotlMessage = new XmppAxolotlMessage(account.getJid().toBareJid(), getOwnDeviceId());
+        final String content;
+        if (message.hasFileOnRemoteHost()) {
+            content = message.getFileParams().url.toString();
+        } else {
+            content = message.getBody();
+        }
+        try {
+            axolotlMessage.encrypt(content);
+        } catch (CryptoFailedException e) {
+            Log.w(Config.LOGTAG, getLogprefix(account) + "Failed to encrypt message: " + e.getMessage());
+            return null;
+        }
+        if (!buildHeader(axolotlMessage, message.getConversation())) {
+            return null;
         }
 
         return axolotlMessage;
@@ -1065,8 +1063,12 @@ public class AxolotlService implements OnAdvancedStreamFeaturesLoaded {
         executor.execute(new Runnable() {
             @Override
             public void run() {
-                XmppAxolotlMessage axolotlMessage = buildHeader(conversation);
-                onMessageCreatedCallback.run(axolotlMessage);
+                final XmppAxolotlMessage axolotlMessage = new XmppAxolotlMessage(account.getJid().toBareJid(), getOwnDeviceId());
+                if (buildHeader(axolotlMessage, conversation)) {
+                    onMessageCreatedCallback.run(axolotlMessage);
+                } else {
+                    onMessageCreatedCallback.run(null);
+                }
             }
         });
     }
-- 
cgit v1.2.3