Tag messages with originating session

This can be used later in order to display trust status of messages, as
well as for potential resending of messages in case of preKey conflicts.

2 files changed, 9 insertions, 0 deletions

diff --git a/src/main/java/eu/siacs/conversations/crypto/axolotl/XmppAxolotlMessage.java b/src/main/java/eu/siacs/conversations/crypto/axolotl/XmppAxolotlMessage.java
index e1b95650..01b06c22 100644
--- a/src/main/java/eu/siacs/conversations/crypto/axolotl/XmppAxolotlMessage.java
+++ b/src/main/java/eu/siacs/conversations/crypto/axolotl/XmppAxolotlMessage.java
@@ -75,6 +75,11 @@ public class XmppAxolotlMessage {
 		public String getPlaintext() {
 			return plaintext;
 		}
+
+		public AxolotlService.XmppAxolotlSession getSession() {
+			return session;
+		}
+
 	}
 
 	public XmppAxolotlMessage(Contact contact, Element axolotlMessage) {
diff --git a/src/main/java/eu/siacs/conversations/entities/Message.java b/src/main/java/eu/siacs/conversations/entities/Message.java
index 336af972..b1dffc82 100644
--- a/src/main/java/eu/siacs/conversations/entities/Message.java
+++ b/src/main/java/eu/siacs/conversations/entities/Message.java
@@ -670,4 +670,8 @@ public class Message extends AbstractEntity {
 	public boolean isTrusted() {
 		return this.axolotlSession != null && this.axolotlSession.isTrusted();
 	}
+
+	public void setAxolotlSession(AxolotlService.XmppAxolotlSession session) {
+		this.axolotlSession = session;
+	}
 }