optimize

ipset.go

@@ -80,7 +80,7 @@ func IsIpset(setName string) int {
 	return 0
 }
 
-func iptables(setName string) error {
+func iptables_add(setName string) error {
 
 	cmd := exec.Command("sh", "-c", fmt.Sprintf("iptables -A INPUT -p tcp -m set --match-set %s src -j DROP", setName))
 
@@ -90,8 +90,25 @@ func iptables(setName string) error {
 
 	err := cmd.Run()
 	if err != nil {
-		log.Printf("cmd.Run() failed with %v, stderr: %s\n", err, stderr.String())
-		err = fmt.Errorf("failed to execute command: %w, stderr: %s", err, stderr.String())
+		//log.Printf("cmd.Run() failed with %v, stderr: %s\n", err, stderr.String())
+		//err = fmt.Errorf("failed to execute command: %w, stderr: %s", err, stderr.String())
+	}
+
+	return err
+}
+
+func iptables_del(setName string) error {
+
+	cmd := exec.Command("sh", "-c", fmt.Sprintf("iptables -D INPUT -p tcp -m set --match-set %s src -j DROP", setName))
+
+	var stdout, stderr bytes.Buffer
+	cmd.Stdout = &stdout
+	cmd.Stderr = &stderr
+
+	err := cmd.Run()
+	if err != nil {
+		//log.Printf("cmd.Run() failed with %v, stderr: %s\n", err, stderr.String())
+		//err = fmt.Errorf("failed to execute command: %w, stderr: %s", err, stderr.String())
 	}
 
 	return err