feat(wip): outbound handle connection response

Signed-off-by: Martichou <m@rtin.fyi>
Martichou · Feb 27, 2024 · 21c7dd8 · 21c7dd8
1 parent d4f2da5
commit 21c7dd8
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 0 deletions.
diff --git a/core_lib/src/hdl/mod.rs b/core_lib/src/hdl/mod.rs
@@ -28,6 +28,7 @@ pub enum State {
     SentUkeyServerInit,
     SentUkeyClientInit,
     SentUkeyClientFinish,
+    SentPairedKeyEncryption,
     ReceivedUkeyClientFinish,
     SentConnectionResponse,
     SentPairedKeyResult,

diff --git a/core_lib/src/hdl/outbound.rs b/core_lib/src/hdl/outbound.rs
@@ -14,6 +14,7 @@ use tokio::sync::broadcast::{Receiver, Sender};
 use super::{InnerState, State};
 use crate::channel::{ChannelAction, ChannelDirection, ChannelMessage};
 use crate::location_nearby_connections::bandwidth_upgrade_negotiation_frame::upgrade_path_info::Medium;
+use crate::location_nearby_connections::connection_response_frame::ResponseStatus;
 use crate::location_nearby_connections::payload_transfer_frame::{
     payload_header, PacketType, PayloadChunk, PayloadHeader,
 };
@@ -155,6 +156,18 @@ impl OutboundRequest {
             }
             State::SentUkeyClientFinish => {
                 debug!("Handling State::SentUkeyClientFinish frame");
+                let frame = location_nearby_connections::OfflineFrame::decode(&*frame_data)?;
+                self.process_connection_response(&frame).await?;
+
+                // Advance current state
+                self.update_state(
+                    |e: &mut InnerState| {
+                        e.state = State::SentPairedKeyEncryption;
+                        e.server_init_data = Some(frame_data);
+                        e.encryption_done = true;
+                    },
+                    false,
+                );
             }
             _ => {
                 debug!("Handling SecureMessage frame");
@@ -318,6 +331,49 @@ impl OutboundRequest {
         Ok(())
     }
 
+    async fn process_connection_response(
+        &mut self,
+        frame: &location_nearby_connections::OfflineFrame,
+    ) -> Result<(), anyhow::Error> {
+        let v1_frame = frame
+            .v1
+            .as_ref()
+            .ok_or_else(|| anyhow!("Missing required fields"))?;
+
+        if v1_frame.r#type() != location_nearby_connections::v1_frame::FrameType::ConnectionResponse
+        {
+            return Err(anyhow!(format!(
+                "Unexpected frame type: {:?}",
+                v1_frame.r#type()
+            )));
+        }
+
+        if v1_frame.connection_response.is_none() {
+            return Err(anyhow!(format!("Unexpected None connection_response",)));
+        }
+
+        if v1_frame.connection_response.as_ref().unwrap().response() != ResponseStatus::Accept {
+            return Err(anyhow!(format!("Connection rejected by third party",)));
+        }
+
+        let paired_encryption = sharing_nearby::Frame {
+            version: Some(sharing_nearby::frame::Version::V1.into()),
+            v1: Some(sharing_nearby::V1Frame {
+                r#type: Some(sharing_nearby::v1_frame::FrameType::PairedKeyEncryption.into()),
+                paired_key_encryption: Some(sharing_nearby::PairedKeyEncryptionFrame {
+                    secret_id_hash: Some(gen_random(6)),
+                    signed_data: Some(gen_random(72)),
+                    ..Default::default()
+                }),
+                ..Default::default()
+            }),
+        };
+
+        self.send_encrypted_frame(&paired_encryption).await?;
+
+        Ok(())
+    }
+
     async fn decrypt_and_process_secure_message(
         &mut self,
         smsg: &SecureMessage,