From fe4de09f536b1fb04d02b474d081b5ffa3b9da75 Mon Sep 17 00:00:00 2001
From: evilsocket <evilsocket@gmail.com>
Date: Thu, 8 Feb 2018 02:06:37 +0100
Subject: [PATCH] balls

---
 firewall/firewall_windows.go | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/firewall/firewall_windows.go b/firewall/firewall_windows.go
index 183ffb4e..865b0773 100644
--- a/firewall/firewall_windows.go
+++ b/firewall/firewall_windows.go
@@ -52,14 +52,17 @@ func (f WindowsFirewall) EnableForwarding(enabled bool) error {
 	return nil
 }
 
-func (f WindowsFirewall) generateRule(r *Redirection) []string {
+func (f WindowsFirewall) generateRule(r *Redirection, enabled bool) []string {
 	rule := []string{
-		fmt.Sprintf("connectport=%d", r.SrcPort),
 		fmt.Sprintf("listenport=%d", r.DstPort),
 	}
 
-	if r.SrcAddress != "" {
-		rule = append(rule, fmt.Sprintf("connectaddress=%s", r.SrcAddress))
+	if enabled == true {
+		rule = append(rule, fmt.Sprintf("connectport=%d", r.SrcPort))
+
+		if r.SrcAddress != "" {
+			rule = append(rule, fmt.Sprintf("connectaddress=%s", r.SrcAddress))
+		}
 	}
 
 	if r.DstAddress != "" {