From 66d450ae3639520ba5c926b5ebeeaf46a0520ed2 Mon Sep 17 00:00:00 2001
From: Franck Royer <franck@status.im>
Date: Fri, 9 Jul 2021 14:12:17 +1000
Subject: [PATCH] Fix bad comparison logic

---
 examples/eth-dm/src/waku.ts | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/examples/eth-dm/src/waku.ts b/examples/eth-dm/src/waku.ts
index 4ed5f957bd..900d8002bf 100644
--- a/examples/eth-dm/src/waku.ts
+++ b/examples/eth-dm/src/waku.ts
@@ -58,20 +58,24 @@ export async function handleDirectMessage(
   console.log('Direct Message received:', wakuMsg);
   if (!wakuMsg.payload) return;
   const directMessage: DirectMessage = decode(wakuMsg.payload);
-  // Do not return our own messages
-  if (directMessage.toAddress === address) return;
+  // Only decrypt messages for us
+  if (!equalByteArrays(directMessage.toAddress, address)) return;
 
-  const text = await decryptMessage(privateKey, directMessage);
+  try {
+    const text = await decryptMessage(privateKey, directMessage);
 
-  const timestamp = wakuMsg.timestamp ? wakuMsg.timestamp : new Date();
+    const timestamp = wakuMsg.timestamp ? wakuMsg.timestamp : new Date();
 
-  console.log('Message decrypted:', text);
-  setter((prevMsgs: Message[]) => {
-    const copy = prevMsgs.slice();
-    copy.push({
-      text: text,
-      timestamp: timestamp,
+    console.log('Message decrypted:', text);
+    setter((prevMsgs: Message[]) => {
+      const copy = prevMsgs.slice();
+      copy.push({
+        text: text,
+        timestamp: timestamp,
+      });
+      return copy;
     });
-    return copy;
-  });
+  } catch (e) {
+    console.log(' Failed to decrypt message', e);
+  }
 }