feat: clear public keys

DIY0R · Nov 28, 2024 · 4462228 · 4462228
1 parent 5fd607e
commit 4462228
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 5 deletions.
diff --git a/lib/constants.js b/lib/constants.js
@@ -14,6 +14,7 @@ const DECRYPT_ERROR = 'Unable to decrypt the message.';
 const SEND_ERROR = 'Failed to send the data.';
 const PUBLISH_ERROR = 'Failed to publish the data.';
 const DEFAULT_TIMEOUT = 10000;
+const TWO_DAYS = 2 * 24 * 60 * 60 * 1000;
 module.exports = {
   PUBLIC_KEY_REQUEST,
   PUBLIC_KEY_RESPONSE,
@@ -27,6 +28,7 @@ module.exports = {
   TTL,
   CRYPTO_DM,
   DM,
+  TWO_DAYS,
   ERROR,
   SEND_ERROR,
   DECRYPT_ERROR,

diff --git a/lib/cryptoRsa.js b/lib/cryptoRsa.js
@@ -12,6 +12,7 @@ const {
   ERROR,
   TIMEOUT_ERROR_REQUEST,
   DECRYPT_ERROR,
+  TWO_DAYS,
 } = require('./constants');
 
 class CryptoRSA extends Connection {
@@ -22,13 +23,13 @@ class CryptoRSA extends Connection {
 
   constructor(port, targetNode) {
     super(port, targetNode);
-    this.#generateKeys();
+    this.#init();
   }
 
   _requestPublicKey(targetNode) {
-    const publicKey = this.#getTargetNodeKey(targetNode);
+    const nodePublic = this.#getTargetNodeKey(targetNode);
     return new Promise((resolve, reject) => {
-      if (publicKey) return resolve(publicKey);
+      if (nodePublic) return resolve(nodePublic.publicKey);
       const messageId = uuid();
       const timerId = setTimeout(
         () => reject(new Error(TIMEOUT_ERROR_REQUEST)),
@@ -45,6 +46,7 @@ class CryptoRSA extends Connection {
         messageId,
         TTL,
       );
+      console.log(options);
       this._broadcast(options);
     });
   }
@@ -83,11 +85,23 @@ class CryptoRSA extends Connection {
     }
   }
 
+  clearPublicKeys(dayInMillis = TWO_DAYS) {
+    const now = Date.now();
+    this._nodePublicKeys.forEach(({ timestamp }, key) => {
+      const timestampMillis = new Date(timestamp).getTime();
+      if (now - timestampMillis > dayInMillis) this._deleteNodeKey(key);
+    });
+  }
+
   #getTargetNodeKey(nodeId) {
     const publicKey = this._nodePublicKeys.get(nodeId);
     if (publicKey) return publicKey;
   }
 
+  #timerClearKeys() {
+    setInterval(this.clearPublicKeys.bind(this), TWO_DAYS);
+  }
+
   #decryptMessage(message) {
     const { encryptedMessage, encryptedSymKey, iv } = message;
     const decryptedSymKey = crypto.privateDecrypt(
@@ -149,12 +163,20 @@ class CryptoRSA extends Connection {
       messageData,
       PUBLIC_KEY_RESPONSE,
     );
-    if (!processedMessage) {
+    if (processedMessage) {
       const { messageId, publicKey } = messageData.message;
-      this._nodePublicKeys.set(messageData.fromNode, publicKey);
+      this._nodePublicKeys.set(messageData.fromNode, {
+        publicKey,
+        timestamp: new Date().toISOString(),
+      });
       this.#responseEmitter.emit(messageId, publicKey);
     }
   }
+
+  #init() {
+    this.#generateKeys();
+    this.#timerClearKeys();
+  }
 }
 
 module.exports = CryptoRSA;