From f42ba56f51a35bee2d345ce7ab8a484c37c14e10 Mon Sep 17 00:00:00 2001
From: topquark12 <aw@m-labs.hk>
Date: Mon, 25 Jan 2021 10:53:53 +0800
Subject: [PATCH] tcp_rst: init commit

---
 src/main.rs | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 461f3fa..431baa6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -202,6 +202,7 @@ fn main() -> ! {
                         warn!("poll: {:?}", e);
                     });
 
+                let mut reset_flag = false;
                 // TCP protocol handling
                 server.for_each(|mut socket, session| {
                     if ! socket.is_active() {
@@ -421,8 +422,8 @@ fn main() -> ! {
                                     for i in 0..CHANNELS {
                                         channels.power_down(i);
                                     }
-
-                                    SCB::sys_reset();
+                                    reset_flag = true;
+                                    // SCB::sys_reset();
                                 }
                                 Command::Dfu => {
                                     for i in 0..CHANNELS {
@@ -431,8 +432,8 @@ fn main() -> ! {
                                     unsafe {
                                         dfu::set_dfu_trigger();
                                     }
-                                    
-                                    SCB::sys_reset();
+                                    reset_flag = true;
+                                    // SCB::sys_reset();
                                 }
                             }
                             Ok(SessionInput::Error(e)) => {
@@ -458,6 +459,13 @@ fn main() -> ! {
                     }
                 });
 
+                if reset_flag == true {
+                    server.for_each(|mut socket, _| {
+                        socket.close();
+                    });
+                    SCB::sys_reset();
+                }
+
                 // Apply new IPv4 address/gateway
                 new_ipv4_config.take()
                     .map(|config| {