From 4bac2d579af8e599437c96332d00df3d808bb5ca Mon Sep 17 00:00:00 2001
From: Simon Grim <scary@tasty.sexy>
Date: Fri, 14 Jul 2017 03:12:35 +0500
Subject: [PATCH] add encryption of DMs's data cache

---
 home.html             |  1 +
 js/twister_newmsgs.js | 30 +++++++++++++++++++++---------
 2 files changed, 22 insertions(+), 9 deletions(-)

diff --git a/home.html b/home.html
index d83a2fe..d11f9f5 100644
--- a/home.html
+++ b/home.html
@@ -28,6 +28,7 @@
     <script src="js/interface_common.js"></script>
     <script src="js/interface_home.js"></script>
     <script src="js/jquery.textcomplete.min.js"></script>
+    <script src="js/twister-crypto-bundle.js"></script>
 
     <link rel="shortcut icon" type="image/png" href="img/twister_mini.png" />
   </head>
diff --git a/js/twister_newmsgs.js b/js/twister_newmsgs.js
index a648a7e..5d175a8 100644
--- a/js/twister_newmsgs.js
+++ b/js/twister_newmsgs.js
@@ -200,6 +200,8 @@ function saveDMsToStorage() {
         };
     }
 
+    pool = twister.var.key.pub.encrypt(JSON.stringify(pool));
+    delete pool.orig;  // WORKAROUND the decrypt function does .slice(0, orig) but something goes wrong in process of buffer decoding (if original string contains non-ASCII characters) and orig may be smaller than the actual size, if it is undefined .slice gets it whole
     $.initNamespaceStorage(defaultScreenName).localStorage.set('DMs', pool);
 }
 
@@ -208,6 +210,12 @@ function loadDMsFromStorage() {
 
     if (storage.isSet('DMs')) {
         var pool = storage.get('DMs');
+        if (pool.key && pool.body && pool.mac) {
+            if (pool = twister.var.key.decrypt(pool))
+                pool = JSON.parse(pool.toString());
+            else
+                console.warn('can\'t decrypt DMs\' data cache');
+        }
         if (typeof pool === 'object') {
             for (var peerAlias in pool) {
                 if (!twister.DMs[peerAlias])
@@ -454,16 +462,20 @@ function updateGroupList() {
 
 function initDMsCount() {
     twister.DMs = {};
-    loadDMsFromStorage();
-    $.MAL.updateNewDMsUI(getNewDMsCount());
-    $.MAL.updateNewGroupDMsUI(getNewGroupDMsCount());
-    //quick hack to obtain list of group chat aliases
-    updateGroupList();
-    setInterval(updateGroupList, 60000);
+    dumpPrivkey(defaultScreenName, function (req, res) {
+        twister.var.key = TwisterCrypto.PrivKey.fromWIF(res);
 
-    setTimeout(requestDMsCount, 200);
-    //polling not needed: processNewPostsConfirmation will call requestDMsCount.
-    //setInterval('requestDMsCount()', 5000);
+        loadDMsFromStorage();
+        $.MAL.updateNewDMsUI(getNewDMsCount());
+        $.MAL.updateNewGroupDMsUI(getNewGroupDMsCount());
+        //quick hack to obtain list of group chat aliases
+        updateGroupList();
+        setInterval(updateGroupList, 60000);
+
+        setTimeout(requestDMsCount, 200);
+        //polling not needed: processNewPostsConfirmation will call requestDMsCount.
+        //setInterval('requestDMsCount()', 5000);
+    });
 }
 
 function newmsgsChangedUser() {