From 918206389203d981a192fabd68a60d21bd8aef9b Mon Sep 17 00:00:00 2001
From: Daniel <dhaavi@users.noreply.github.com>
Date: Fri, 11 Aug 2023 11:57:24 +0200
Subject: [PATCH] Stop firewall handler earlier

---
 network/clean.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/network/clean.go b/network/clean.go
index cd17f392..1a690757 100644
--- a/network/clean.go
+++ b/network/clean.go
@@ -81,6 +81,11 @@ func cleanConnections() (activePIDs map[int]struct{}) {
 				// Step 2: mark as ended
 				if !exists {
 					conn.Ended = nowUnix
+
+					// Stop the firewall handler, in case one is running.
+					conn.StopFirewallHandler()
+
+					// Save to database.
 					conn.Save()
 				}
 
@@ -93,8 +98,6 @@ func cleanConnections() (activePIDs map[int]struct{}) {
 				// DEBUG:
 				// log.Tracef("network.clean: deleted %s (ended at %s)", conn.DatabaseKey(), time.Unix(conn.Ended, 0))
 
-				// Stop the firewall handler, in case one is running.
-				conn.StopFirewallHandler()
 				// Remove connection from state.
 				conn.delete()
 			}