Merge pull request #8 from Amnesic-Systems/toggle-nat

Toggle NAT when starting.
Amnesic-Systems · Nov 16, 2024 · a76261b · a76261b
2 parents 29f0303 + 3014acc
commit a76261b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -106,6 +106,16 @@ func main() {
 		l.Fatalf("Flag -port must be in interval [1, %d].", math.MaxUint32)
 	}
 
+	l.Print("Enabling NAT.")
+	if err := proxy.ToggleNAT(proxy.On); err != nil {
+		l.Fatalf("Error enabling NAT: %v", err)
+	}
+	defer func() {
+		if err := proxy.ToggleNAT(proxy.Off); err != nil {
+			l.Printf("Error disabling NAT: %v", err)
+		}
+	}()
+
 	ln = listenVSOCK(uint32(port))
 	defer ln.Close()