From 71c2a48a1fb0bffce9cc17c6f1f2e64ca787319b Mon Sep 17 00:00:00 2001
From: Marcin Kostrzewski <komar86311@gmail.com>
Date: Mon, 16 Nov 2020 12:22:26 +0100
Subject: [PATCH] Fixed incomplete agreement

---
 src/main/java/Main/DHExchange.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/main/java/Main/DHExchange.java b/src/main/java/Main/DHExchange.java
index d65cd1f..9beca4c 100644
--- a/src/main/java/Main/DHExchange.java
+++ b/src/main/java/Main/DHExchange.java
@@ -65,6 +65,9 @@ public class DHExchange {
             agreement.init(keyPair.getPrivate());
 
             thisPublicKey = keyPair.getPublic();
+            agreement.doPhase(otherPublicKey, true);
+
+            generateSecret();
 
         } catch (NoSuchAlgorithmException | InvalidKeySpecException | InvalidAlgorithmParameterException | InvalidKeyException e) {
             e.printStackTrace();
@@ -80,6 +83,7 @@ public class DHExchange {
             logger.log(Level.INFO, "Agreement phase...");
             agreement.doPhase(otherPublicKey, true);
 
+            generateSecret();
         } catch (NoSuchAlgorithmException | InvalidKeySpecException | InvalidKeyException e) {
             e.printStackTrace();
         }