From e5449f56eec2348b4fe6140465f963fc2ea31863 Mon Sep 17 00:00:00 2001
From: Beaudan <beau@loki.network>
Date: Thu, 14 Mar 2019 10:02:38 +1100
Subject: [PATCH] Added success and failure callbacks when queueing envelope,
 now responding with 404 if we fail to decrypt a p2p message

---
 libloki/modules/local_loki_server.js |  7 +++++--
 libtextsecure/http-resources.js      |  4 ++--
 libtextsecure/message_receiver.js    | 24 +++++++++++++++++++-----
 3 files changed, 26 insertions(+), 9 deletions(-)

diff --git a/libloki/modules/local_loki_server.js b/libloki/modules/local_loki_server.js
index 2b72c84ac..ed1393756 100644
--- a/libloki/modules/local_loki_server.js
+++ b/libloki/modules/local_loki_server.js
@@ -57,8 +57,11 @@ class LocalLokiServer extends EventEmitter {
                 sendResponse(STATUS.NOT_FOUND, 'Invalid endpoint!');
                 return;
               }
-              this.emit('message', bodyObject.params.data);
-              sendResponse(STATUS.OK);
+              this.emit('message', {
+                message: bodyObject.params.data,
+                onSuccess: () => sendResponse(STATUS.OK),
+                onFailure: () => sendResponse(STATUS.NOT_FOUND),
+              });
             } catch (e) {
               // Bad Request: Failed to decode json
               sendResponse(STATUS.BAD_REQUEST, 'Failed to decode JSON');
diff --git a/libtextsecure/http-resources.js b/libtextsecure/http-resources.js
index bbd36e2d3..8c3387963 100644
--- a/libtextsecure/http-resources.js
+++ b/libtextsecure/http-resources.js
@@ -77,7 +77,7 @@
       });
     };
 
-    this.handleMessage = (message, isP2p = false) => {
+    this.handleMessage = (message, options = {}) => {
       try {
         const dataPlaintext = stringToArrayBufferBase64(message);
         const messageBuf = textsecure.protobuf.WebSocketMessage.decode(
@@ -93,7 +93,7 @@
               body: messageBuf.request.body,
               id: messageBuf.request.id,
             }),
-            isP2p
+            options
           );
         }
       } catch (error) {
diff --git a/libtextsecure/message_receiver.js b/libtextsecure/message_receiver.js
index 83d75c497..d689c4416 100644
--- a/libtextsecure/message_receiver.js
+++ b/libtextsecure/message_receiver.js
@@ -137,8 +137,13 @@ MessageReceiver.prototype.extend({
       setTimeout(this.startLocalServer.bind(this), 30 * 1000);
     }
   },
-  handleP2pMessage(message) {
-    this.httpPollingResource.handleMessage(message, true);
+  handleP2pMessage({ message, onSuccess, onFailure }) {
+    const options = {
+      isP2p: true,
+      onSuccess,
+      onFailure,
+    };
+    this.httpPollingResource.handleMessage(message, options);
   },
   shutdown() {
     if (this.socket) {
@@ -215,7 +220,8 @@ MessageReceiver.prototype.extend({
     //     return this.dispatchAndWait(event);
     //   });
   },
-  handleRequest(request, isP2p = false) {
+  handleRequest(request, options) {
+    const { isP2p, onSuccess, onFailure } = options;
     this.incoming = this.incoming || [];
     const lastPromise = _.last(this.incoming);
 
@@ -258,7 +264,7 @@ MessageReceiver.prototype.extend({
 
             // To ensure that we queue in the same order we receive messages
             await lastPromise;
-            this.queueEnvelope(envelope);
+            this.queueEnvelope(envelope, onSuccess, onFailure);
           },
           error => {
             request.respond(500, 'Failed to cache message');
@@ -534,7 +540,7 @@ MessageReceiver.prototype.extend({
       );
     });
   },
-  queueEnvelope(envelope) {
+  queueEnvelope(envelope, onSuccess = null, onFailure = null) {
     const id = this.getEnvelopeId(envelope);
     window.log.info('queueing envelope', id);
 
@@ -544,6 +550,11 @@ MessageReceiver.prototype.extend({
       `queueEnvelope ${id}`
     );
     const promise = this.addToQueue(taskWithTimeout);
+    promise.then(() => {
+      if (onSuccess) {
+        onSuccess();
+      }
+    });
 
     return promise.catch(error => {
       window.log.error(
@@ -552,6 +563,9 @@ MessageReceiver.prototype.extend({
         ':',
         error && error.stack ? error.stack : error
       );
+      if (onFailure) {
+        onFailure();
+      }
     });
   },
   // Same as handleEnvelope, just without the decryption step. Necessary for handling