From cc04bee38a8b8dff857ccb6a3c00d43b0052fff0 Mon Sep 17 00:00:00 2001
From: sachaaaaa <sacha@loki.network>
Date: Mon, 30 Sep 2019 12:22:21 +1000
Subject: [PATCH] Start all the receivers only after the secondary registration
 is finished

---
 js/background.js                         | 26 +++++++++++++++++++++---
 js/modules/loki_app_dot_net_api.js       | 11 ++++++++++
 js/views/standalone_registration_view.js |  2 ++
 libtextsecure/message_receiver.js        | 23 +++++++++++++++------
 4 files changed, 53 insertions(+), 9 deletions(-)

diff --git a/js/background.js b/js/background.js
index b243a91b7..5c786fff8 100644
--- a/js/background.js
+++ b/js/background.js
@@ -875,14 +875,14 @@
     );
   }
 
-  function disconnect() {
+  async function disconnect() {
     window.log.info('disconnect');
 
     // Clear timer, since we're only called when the timer is expired
     disconnectTimer = null;
 
     if (messageReceiver) {
-      messageReceiver.close();
+      await messageReceiver.close();
     }
     window.Signal.AttachmentDownloads.stop();
   }
@@ -912,7 +912,7 @@
     }
 
     if (messageReceiver) {
-      messageReceiver.close();
+      await messageReceiver.close();
     }
 
     const USERNAME = storage.get('number_id');
@@ -927,6 +927,26 @@
 
     Whisper.Notifications.disable(); // avoid notification flood until empty
 
+    if (Whisper.Registration.ongoingSecondaryDeviceRegistration()) {
+      const ourKey = textsecure.storage.user.getNumber();
+      window.lokiMessageAPI = new window.LokiMessageAPI(ourKey);
+      window.localLokiServer = null;
+      window.lokiPublicChatAPI = null;
+      window.feeds = [];
+      messageReceiver = new textsecure.MessageReceiver(
+        USERNAME,
+        PASSWORD,
+        mySignalingKey,
+        options
+      );
+      messageReceiver.addEventListener('message', onMessageReceived);
+      window.textsecure.messaging = new textsecure.MessageSender(
+        USERNAME,
+        PASSWORD
+      );
+      return;
+    }
+
     // initialize the socket and start listening for messages
     startLocalLokiServer();
     await initAPIs();
diff --git a/js/modules/loki_app_dot_net_api.js b/js/modules/loki_app_dot_net_api.js
index ca5bb3ade..3461f6503 100644
--- a/js/modules/loki_app_dot_net_api.js
+++ b/js/modules/loki_app_dot_net_api.js
@@ -19,6 +19,10 @@ class LokiAppDotNetAPI extends EventEmitter {
     this.myPrivateKey = false;
   }
 
+  async close() {
+    await Promise.all(this.servers.map(server => server.close()));
+  }
+
   async getPrivateKey() {
     if (!this.myPrivateKey) {
       const myKeyPair = await textsecure.storage.protocol.getIdentityKeyPair();
@@ -89,6 +93,13 @@ class LokiAppDotNetServerAPI {
     })();
   }
 
+  async close() {
+    this.channels.forEach(channel => channel.stop());
+    if (this.tokenPromise) {
+      await this.tokenPromise;
+    }
+  }
+
   // channel getter/factory
   findOrCreateChannel(channelId, conversationId) {
     let thisChannel = this.channels.find(
diff --git a/js/views/standalone_registration_view.js b/js/views/standalone_registration_view.js
index 6cf43b501..977ca45c5 100644
--- a/js/views/standalone_registration_view.js
+++ b/js/views/standalone_registration_view.js
@@ -141,6 +141,8 @@
       clearInterval(this.pairingInterval);
       // Ensure the left menu is updated
       Whisper.events.trigger('userChanged', { isSecondaryDevice: true });
+      // will re-run the background initialisation
+      Whisper.events.trigger('registration_done');
       this.$el.trigger('openInbox');
     },
     async resetRegistration() {
diff --git a/libtextsecure/message_receiver.js b/libtextsecure/message_receiver.js
index 8dbb648d4..2641cb692 100644
--- a/libtextsecure/message_receiver.js
+++ b/libtextsecure/message_receiver.js
@@ -77,11 +77,15 @@ MessageReceiver.prototype.extend({
       handleRequest: this.handleRequest.bind(this),
     });
     this.httpPollingResource.pollServer();
-    localLokiServer.on('message', this.handleP2pMessage.bind(this));
-    lokiPublicChatAPI.on(
-      'publicMessage',
-      this.handleUnencryptedMessage.bind(this)
-    );
+    if (localLokiServer) {
+      localLokiServer.on('message', this.handleP2pMessage.bind(this));
+    }
+    if (lokiPublicChatAPI) {
+      lokiPublicChatAPI.on(
+        'publicMessage',
+        this.handleUnencryptedMessage.bind(this)
+      );
+    }
     // set up pollers for any RSS feeds
     feeds.forEach(feed => {
       feed.on('rssMessage', this.handleUnencryptedMessage.bind(this));
@@ -118,6 +122,9 @@ MessageReceiver.prototype.extend({
     this.incoming = [this.pending];
   },
   async startLocalServer() {
+    if (!localLokiServer) {
+      return;
+    }
     try {
       // clearnet change: getMyLokiIp -> getMyClearIp
       // const myLokiIp = await window.lokiSnodeAPI.getMyLokiIp();
@@ -184,7 +191,7 @@ MessageReceiver.prototype.extend({
       );
     }
   },
-  close() {
+  async close() {
     window.log.info('MessageReceiver.close()');
     this.calledClose = true;
 
@@ -198,6 +205,10 @@ MessageReceiver.prototype.extend({
       localLokiServer.close();
     }
 
+    if (lokiPublicChatAPI) {
+      await lokiPublicChatAPI.close();
+    }
+
     if (this.httpPollingResource) {
       this.httpPollingResource.close();
     }