From ef1d6392a71f045adb5678b9027593c79e237adc Mon Sep 17 00:00:00 2001
From: Niels Andriesse <andriesseniels@gmail.com>
Date: Mon, 4 Jan 2021 10:53:08 +1100
Subject: [PATCH] Activate the Session protocol

---
 .../Sending & Receiving/MessageReceiver.swift      | 14 ++------------
 .../Sending & Receiving/MessageSender.swift        |  4 ++--
 2 files changed, 4 insertions(+), 14 deletions(-)

diff --git a/SessionMessagingKit/Sending & Receiving/MessageReceiver.swift b/SessionMessagingKit/Sending & Receiving/MessageReceiver.swift
index d46d9ae67..9f76f3ab0 100644
--- a/SessionMessagingKit/Sending & Receiving/MessageReceiver.swift	
+++ b/SessionMessagingKit/Sending & Receiving/MessageReceiver.swift	
@@ -66,19 +66,9 @@ public enum MessageReceiver {
         } else {
             switch envelope.type {
             case .unidentifiedSender:
-                do {
-                    (plaintext, sender) = try decryptWithSessionProtocol(envelope: envelope)
-                } catch {
-                    // Migration
-                    (plaintext, sender) = try decryptWithSignalProtocol(envelope: envelope, using: transaction)
-                }
+                (plaintext, sender) = try decryptWithSessionProtocol(envelope: envelope)
             case .closedGroupCiphertext:
-                do {
-                    (plaintext, sender) = try decryptWithSessionProtocol(envelope: envelope)
-                } catch {
-                    // Migration
-                    (plaintext, sender) = try decryptWithSharedSenderKeys(envelope: envelope, using: transaction)
-                }
+                (plaintext, sender) = try decryptWithSessionProtocol(envelope: envelope)
                 groupPublicKey = envelope.source
             default: throw Error.unknownEnvelopeType
             }
diff --git a/SessionMessagingKit/Sending & Receiving/MessageSender.swift b/SessionMessagingKit/Sending & Receiving/MessageSender.swift
index 998d7693c..b6a517776 100644
--- a/SessionMessagingKit/Sending & Receiving/MessageSender.swift	
+++ b/SessionMessagingKit/Sending & Receiving/MessageSender.swift	
@@ -174,8 +174,8 @@ public final class MessageSender : NSObject {
         let ciphertext: Data
         do {
             switch destination {
-            case .contact(let publicKey): ciphertext = try encryptWithSignalProtocol(plaintext, associatedWith: message, for: publicKey, using: transaction)
-            case .closedGroup(let groupPublicKey): ciphertext = try encryptWithSharedSenderKeys(plaintext, for: groupPublicKey, using: transaction)
+            case .contact(let publicKey): ciphertext = try encryptWithSessionProtocol(plaintext, for: publicKey)
+            case .closedGroup(let groupPublicKey): ciphertext = try encryptWithSessionProtocol(plaintext, for: groupPublicKey)
             case .openGroup(_, _): preconditionFailure()
             }
         } catch {