Add closing if raw socket to handle re-attach

2 files changed, 18 insertions, 5 deletions

diff --git a/embassy-net-nrf91/src/context.rs b/embassy-net-nrf91/src/context.rs
index f73719224..511468316 100644
--- a/embassy-net-nrf91/src/context.rs
+++ b/embassy-net-nrf91/src/context.rs
@@ -4,8 +4,8 @@ use core::str::FromStr;
 
 use at_commands::builder::CommandBuilder;
 use at_commands::parser::CommandParser;
+use embassy_time::{Duration, Timer};
 use heapless::Vec;
-use embassy_time::{Timer, Duration};
 
 /// Provides a higher level API for controlling a given context.
 pub struct Control<'a> {
@@ -129,8 +129,6 @@ impl<'a> Control<'a> {
         let n = self.control.at_command(op, &mut buf).await;
         CommandParser::parse(&buf[..n]).expect_identifier(b"OK").finish()?;
 
-
-
         Ok(())
     }
 
@@ -289,13 +287,13 @@ impl<'a> Control<'a> {
 
         loop {
             if !self.attached().await? {
-                // TODO: self.control.close_raw_socket(fd).await;
+                self.control.close_raw_socket(fd).await;
                 self.attach().await?;
                 while !self.attached().await? {
                     Timer::after(Duration::from_secs(1)).await;
                 }
                 let status = self.status().await?;
-                // TODO: let mut fd = self.control.open_raw_socket().await;
+                fd = self.control.open_raw_socket().await;
                 reattach(&status);
             }
             Timer::after(Duration::from_secs(10)).await;
diff --git a/embassy-net-nrf91/src/lib.rs b/embassy-net-nrf91/src/lib.rs
index ab3c6f327..d8cbe47fc 100644
--- a/embassy-net-nrf91/src/lib.rs
+++ b/embassy-net-nrf91/src/lib.rs
@@ -870,6 +870,21 @@ impl<'a> Control<'a> {
         trace!("got FD: {}", fd);
         fd
     }
+
+    async fn close_raw_socket(&self, fd: u32) {
+        let mut msg: Message = unsafe { mem::zeroed() };
+        msg.channel = 2; // data
+        msg.id = 0x7009_0004; // close socket
+        msg.param_len = 8;
+        msg.param[4..8].copy_from_slice(&fd.to_le_bytes());
+
+        self.request(&mut msg, &[], &mut []).await;
+
+        assert_eq!(msg.id, 0x80090004);
+        assert!(msg.param_len >= 12);
+        let status = u32::from_le_bytes(msg.param[8..12].try_into().unwrap());
+        assert_eq!(status, 0);
+    }
 }
 
 /// Background runner for the driver.