From b45ace4d58514d19ace00f6f55393f54efff0347 Mon Sep 17 00:00:00 2001
From: lilia <liliakai@gmail.com>
Date: Wed, 24 May 2017 16:14:13 -0700
Subject: [PATCH] Clear old sessions for a device when its identity changes

---
 dist/libsignal-protocol.js | 23 +++++++++++++++++++++--
 src/SessionBuilder.js      | 12 ++++++++++--
 src/SessionRecord.js       | 11 +++++++++++
 3 files changed, 42 insertions(+), 4 deletions(-)

diff --git a/dist/libsignal-protocol.js b/dist/libsignal-protocol.js
index 1f39859..c4f16d5 100644
--- a/dist/libsignal-protocol.js
+++ b/dist/libsignal-protocol.js
@@ -35762,6 +35762,17 @@ Internal.SessionRecord = function() {
                 delete sessions[util.toString(oldestBaseKey)];
             }
         },
+        removePreviousSessions: function() {
+            var sessions = this.sessions;
+            for (var key in sessions) {
+                if (sessions[key].indexInfo.closed === -1) {
+                    var newSessions = {};
+                    newSessions[key] = sessions[key];
+                    this.sessions = newSessions;
+                    return;
+                }
+            }
+        },
     };
 
     return SessionRecord;
@@ -35857,7 +35868,11 @@ SessionBuilder.prototype = {
           record.updateSessionState(session);
           return Promise.all([
             this.storage.storeSession(address, record.serialize()),
-            this.storage.saveIdentity(this.remoteAddress.getName(), session.indexInfo.remoteIdentityKey)
+            this.storage.saveIdentity(this.remoteAddress.getName(), session.indexInfo.remoteIdentityKey).then(function(changed) {
+              if (changed) {
+                return record.removePreviousSessions();
+              }
+            }.bind(this))
           ]);
         }.bind(this));
       }.bind(this));
@@ -35914,7 +35929,11 @@ SessionBuilder.prototype = {
             // end of decryptWhisperMessage ... to ensure that the sender
             // actually holds the private keys for all reported pubkeys
             record.updateSessionState(new_session);
-            return this.storage.saveIdentity(this.remoteAddress.getName(), message.identityKey.toArrayBuffer()).then(function() {
+            return this.storage.saveIdentity(this.remoteAddress.getName(), message.identityKey.toArrayBuffer()).then(function(changed) {
+              if (changed) {
+                return record.removePreviousSessions();
+              }
+            }).then(function() {
               return message.preKeyId;
             });
         }.bind(this));
diff --git a/src/SessionBuilder.js b/src/SessionBuilder.js
index 35583c6..04cdecd 100644
--- a/src/SessionBuilder.js
+++ b/src/SessionBuilder.js
@@ -51,7 +51,11 @@ SessionBuilder.prototype = {
           record.updateSessionState(session);
           return Promise.all([
             this.storage.storeSession(address, record.serialize()),
-            this.storage.saveIdentity(this.remoteAddress.getName(), session.indexInfo.remoteIdentityKey)
+            this.storage.saveIdentity(this.remoteAddress.getName(), session.indexInfo.remoteIdentityKey).then(function(changed) {
+              if (changed) {
+                return record.removePreviousSessions();
+              }
+            }.bind(this))
           ]);
         }.bind(this));
       }.bind(this));
@@ -108,7 +112,11 @@ SessionBuilder.prototype = {
             // end of decryptWhisperMessage ... to ensure that the sender
             // actually holds the private keys for all reported pubkeys
             record.updateSessionState(new_session);
-            return this.storage.saveIdentity(this.remoteAddress.getName(), message.identityKey.toArrayBuffer()).then(function() {
+            return this.storage.saveIdentity(this.remoteAddress.getName(), message.identityKey.toArrayBuffer()).then(function(changed) {
+              if (changed) {
+                return record.removePreviousSessions();
+              }
+            }).then(function() {
               return message.preKeyId;
             });
         }.bind(this));
diff --git a/src/SessionRecord.js b/src/SessionRecord.js
index f56cef5..49fbabc 100644
--- a/src/SessionRecord.js
+++ b/src/SessionRecord.js
@@ -263,6 +263,17 @@ Internal.SessionRecord = function() {
                 delete sessions[util.toString(oldestBaseKey)];
             }
         },
+        removePreviousSessions: function() {
+            var sessions = this.sessions;
+            for (var key in sessions) {
+                if (sessions[key].indexInfo.closed === -1) {
+                    var newSessions = {};
+                    newSessions[key] = sessions[key];
+                    this.sessions = newSessions;
+                    return;
+                }
+            }
+        },
     };
 
     return SessionRecord;