From dd9cbcd284aec3184cf84eb171f1f66cdde616c6 Mon Sep 17 00:00:00 2001
From: Sergey Ivanov <ivanovs-4@yandex.ru>
Date: Thu, 13 Jul 2023 00:57:38 +0400
Subject: [PATCH] Change encryption handshake brains methods

---
 hbs2-peer/app/Brains.hs   | 70 ++++++++++++++-------------------------
 hbs2-peer/app/PeerMain.hs | 11 +++---
 2 files changed, 32 insertions(+), 49 deletions(-)

diff --git a/hbs2-peer/app/Brains.hs b/hbs2-peer/app/Brains.hs
index 6df0c074..6a710dfc 100644
--- a/hbs2-peer/app/Brains.hs
+++ b/hbs2-peer/app/Brains.hs
@@ -605,63 +605,51 @@ transactional brains action = do
 
 ---
 
-insertPeerAsymmKey :: forall e m .
-    ( e ~ L4Proto
-    , MonadIO m
-    )
+insertPeerAsymmKey :: forall e m . (e ~ L4Proto, MonadIO m)
     => BasicBrains e
     -> Peer e
     -> Encrypt.PublicKey
-    -> m ()
-
-insertPeerAsymmKey br peer hAsymmKey = do
-  let conn = view brainsDb br
-  void $ liftIO $ execute conn [qc|
-        INSERT INTO peer_asymmkey (peer,asymmkey)
-        VALUES (?,?)
-        ON CONFLICT (peer)
-        DO UPDATE SET
-          asymmkey = excluded.asymmkey
-
-    |] (show $ pretty peer, show hAsymmKey)
-
-insertPeerSymmKey :: forall e m .
-    ( e ~ L4Proto
-    , MonadIO m
-    )
-    => BasicBrains e
-    -> Peer e
     -> Encrypt.CombinedKey
     -> m ()
 
-insertPeerSymmKey br peer hSymmKey = do
+insertPeerAsymmKey br peer hAsymmKey hSymmKey = do
+    insertPeerAsymmKey br peer hAsymmKey hSymmKey
+    insertPeerAsymmKey' br (show $ pretty peer) hAsymmKey hSymmKey
+
+insertPeerAsymmKey' :: forall e m . (e ~ L4Proto, MonadIO m)
+    => BasicBrains e
+    -> String
+    -> Encrypt.PublicKey
+    -> Encrypt.CombinedKey
+    -> m ()
+
+insertPeerAsymmKey' br key hAsymmKey hSymmKey = do
   let conn = view brainsDb br
   void $ liftIO $ execute conn [qc|
-        INSERT INTO peer_symmkey (peer,symmkey)
-        VALUES (?,?)
+        INSERT INTO peer_asymmkey (peer,asymmkey,symmkey)
+        VALUES (?,?,?)
         ON CONFLICT (peer)
         DO UPDATE SET
-          symmkey = excluded.symmkey
+          asymmkey = excluded.asymmkey
+        , symmkey = excluded.symmkey
+    |] (key, show hAsymmKey, show hSymmKey)
 
-    |] (show $ pretty peer, show hSymmKey)
+---
 
 deletePeerAsymmKey :: forall e m . (e ~ L4Proto, MonadIO m)
     => BasicBrains e -> Peer e -> m ()
 
 deletePeerAsymmKey br peer =
-    void $ liftIO $ execute (view brainsDb br) [qc|
-        DELETE FROM peer_symmkey
-        WHERE peer = ?
-    |] (Only (show $ pretty peer))
+    deletePeerAsymmKey' br (show $ pretty peer)
 
-deletePeerSymmKey :: forall e m . (e ~ L4Proto, MonadIO m)
-    => BasicBrains e -> Peer e -> m ()
+deletePeerAsymmKey' :: forall e m . (e ~ L4Proto, MonadIO m)
+    => BasicBrains e -> String -> m ()
 
-deletePeerSymmKey br peer =
+deletePeerAsymmKey' br key =
     void $ liftIO $ execute (view brainsDb br) [qc|
-        DELETE FROM peer_symmkey
+        DELETE FROM peer_asymmkey
         WHERE peer = ?
-    |] (Only (show $ pretty peer))
+    |] (Only key)
 
 ---
 
@@ -757,14 +745,6 @@ newBasicBrains cfg = liftIO do
     create table if not exists peer_asymmkey
     ( peer text not null
     , asymmkey text not null
-    , ts DATE DEFAULT (datetime('now','localtime'))
-    , primary key (peer)
-    )
-  |]
-
-  execute_ conn [qc|
-    create table if not exists peer_symmkey
-    ( peer text not null
     , symmkey text not null
     , ts DATE DEFAULT (datetime('now','localtime'))
     , primary key (peer)
diff --git a/hbs2-peer/app/PeerMain.hs b/hbs2-peer/app/PeerMain.hs
index 0aca3331..fcb46d06 100644
--- a/hbs2-peer/app/PeerMain.hs
+++ b/hbs2-peer/app/PeerMain.hs
@@ -654,14 +654,17 @@ runPeer opts = U.handle (\e -> myException e
                     { encHandshake_considerPeerAsymmKey = \peer mpeerData -> \case
                         Nothing -> do
                             deletePeerAsymmKey brains peer
-                            deletePeerSymmKey brains peer
+                            forM_ mpeerData \peerData ->
+                                deletePeerAsymmKey' brains (show peerData)
                         Just pk -> do
                             -- emit PeerAsymmInfoKey (PeerAsymmPubKey peer pk)
-                            insertPeerAsymmKey brains peer pk
-                            insertPeerSymmKey brains peer $
-                                genCommonSecret @s
+                            let symmk = genCommonSecret @s
                                     (privKeyFromKeypair @s (view envAsymmetricKeyPair penv))
                                     pk
+                            case mpeerData of
+                                Nothing -> insertPeerAsymmKey brains peer pk symmk
+                                Just peerData ->
+                                    insertPeerAsymmKey' brains (show peerData) pk symmk
                     }
 
               env <- ask