feat: add new interface method to firewals
All checks were successful
build / build (push) Successful in 3m9s

This commit is contained in:
d3m0k1d
2026-02-09 19:50:06 +03:00
parent b8b9b227a9
commit 9519eedf4f
7 changed files with 70 additions and 1 deletions

View File

@@ -2,6 +2,7 @@ package blocker
import (
"os/exec"
"strconv"
"github.com/d3m0k1d/BanForge/internal/logger"
)
@@ -58,6 +59,31 @@ func (f *Firewalld) Unban(ip string) error {
return nil
}
func (f *Firewalld) PortOpen(port int) error {
// #nosec G204 - handle is extracted from nftables output and validated
if port >= 0 && port <= 65535 {
s := strconv.Itoa(port)
cmd := exec.Command("firewall-cmd", "--zone=public", "--add-port="+s+"/tcp", "--permanent")
output, err := cmd.CombinedOutput()
if err != nil {
f.logger.Error(err.Error())
return err
}
f.logger.Info("Add port " + s + " " + string(output))
output, err = exec.Command("firewall-cmd", "--reload").CombinedOutput()
if err != nil {
f.logger.Error(err.Error())
return err
}
f.logger.Info("Reload " + string(output))
}
return nil
}
func (f *Firewalld) PortClose(port int) error {
return nil
}
func (f *Firewalld) Setup(config string) error {
return nil
}

View File

@@ -10,6 +10,8 @@ type BlockerEngine interface {
Ban(ip string) error
Unban(ip string) error
Setup(config string) error
PortOpen(port int) error
PortClose(port int) error
}
func GetBlocker(fw string, config string) BlockerEngine {

View File

@@ -102,6 +102,14 @@ func (f *Iptables) Unban(ip string) error {
return nil
}
func (f *Iptables) PortOpen(port int) error {
return nil
}
func (f *Iptables) PortClose(port int) error {
return nil
}
func (f *Iptables) Setup(config string) error {
return nil
}

View File

@@ -166,6 +166,14 @@ func (n *Nftables) findRuleHandle(ip string) (string, error) {
return "", nil
}
func (n *Nftables) PortOpen(port int) error {
return nil
}
func (n *Nftables) PortClose(port int) error {
return nil
}
func saveNftablesConfig(configPath string) error {
err := validateConfigPath(configPath)
if err != nil {

View File

@@ -56,6 +56,14 @@ func (u *Ufw) Unban(ip string) error {
return nil
}
func (u *Ufw) PortOpen(port int) error {
return nil
}
func (u *Ufw) PortClose(port int) error {
return nil
}
func (u *Ufw) Setup(config string) error {
if config != "" {
fmt.Printf("Ufw dont support config file\n")