Assigning connection shared key post registration (#308)

2020-08-19 11:35:01 +01:00
parent 7d2d512405
commit de6c1e6601
1 changed files with 4 additions and 1 deletions
@@ -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) => {