Initial commit after restructure

network/clean.go

@@ -0,0 +1,103 @@
+// Copyright Safing ICS Technologies GmbH. Use of this source code is governed by the AGPL license that can be found in the LICENSE file.
+
+package network
+
+import (
+	"time"
+
+	"github.com/Safing/safing-core/process"
+)
+
+func init() {
+	go cleaner()
+}
+
+func cleaner() {
+	time.Sleep(15 * time.Second)
+	for {
+		markDeadLinks()
+		purgeDeadFor(5 * time.Minute)
+		time.Sleep(15 * time.Second)
+	}
+}
+
+func markDeadLinks() {
+	activeIDs := process.GetActiveConnectionIDs()
+
+	allLinksLock.RLock()
+	defer allLinksLock.RUnlock()
+
+	now := time.Now().Unix()
+	var found bool
+	for key, link := range allLinks {
+
+		// skip dead links
+		if link.Ended > 0 {
+			continue
+		}
+
+		// check if link is dead
+		found = false
+		for _, activeID := range activeIDs {
+			if key == activeID {
+				found = true
+				break
+			}
+		}
+
+		// mark end time
+		if !found {
+			link.Ended = now
+			link.Save()
+		}
+
+	}
+}
+
+func purgeDeadFor(age time.Duration) {
+	connections := make(map[*Connection]bool)
+	processes := make(map[*process.Process]bool)
+
+	allLinksLock.Lock()
+	defer allLinksLock.Unlock()
+
+	// delete old dead links
+	// make a list of connections without links
+	ageAgo := time.Now().Add(-1 * age).Unix()
+	for key, link := range allLinks {
+		if link.Ended != 0 && link.Ended < ageAgo {
+			link.Delete()
+			delete(allLinks, key)
+			_, ok := connections[link.Connection()]
+			if !ok {
+				connections[link.Connection()] = false
+			}
+		} else {
+			connections[link.Connection()] = true
+		}
+	}
+
+	// delete connections without links
+	// make a list of processes without connections
+	for conn, active := range connections {
+		if conn != nil {
+			if !active {
+				conn.Delete()
+				_, ok := processes[conn.Process()]
+				if !ok {
+					processes[conn.Process()] = false
+				}
+			} else {
+				processes[conn.Process()] = true
+			}
+		}
+	}
+
+	// delete processes without connections
+	for proc, active := range processes {
+		if proc != nil && !active {
+			proc.Delete()
+		}
+	}
+
+}