From de6c1e660192126f70bdc71214bfcd2288c862f2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C4=99drzej=20Stuczy=C5=84ski?=
 <jedrzej.stuczynski@gmail.com>
Date: Wed, 19 Aug 2020 11:35:01 +0100
Subject: [PATCH] Assigning connection shared key post registration (#308)

---
 .../src/node/client_handling/websocket/connection_handler.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/gateway/src/node/client_handling/websocket/connection_handler.rs b/gateway/src/node/client_handling/websocket/connection_handler.rs
index 2047a902ea..ad29df86a7 100644
--- a/gateway/src/node/client_handling/websocket/connection_handler.rs
+++ b/gateway/src/node/client_handling/websocket/connection_handler.rs
@@ -327,7 +327,7 @@ impl<S> Handle<S> {
         let (res_sender, res_receiver) = oneshot::channel();
         let clients_handler_request = ClientsHandlerRequest::Register(
             remote_address.clone(),
-            derived_shared_key,
+            derived_shared_key.clone(),
             mix_sender,
             res_sender,
         );
@@ -341,6 +341,9 @@ impl<S> Handle<S> {
             // managed to complete registration handshake)
             ClientsHandlerResponse::Register(status) => {
                 self.remote_address = Some(remote_address);
+                if status {
+                    self.shared_key = Some(derived_shared_key);
+                }
                 ServerResponse::Register { status }
             }
             ClientsHandlerResponse::Error(e) => {