31 lines
654 B
Go
31 lines
654 B
Go
package blocker
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/d3m0k1d/BanForge/internal/logger"
|
|
)
|
|
|
|
type BlockerEngine interface {
|
|
Ban(ip string) error
|
|
Unban(ip string) error
|
|
Setup(config string) error
|
|
PortOpen(port int, protocol string) error
|
|
PortClose(port int, protocol string) error
|
|
}
|
|
|
|
func GetBlocker(fw string, config string) BlockerEngine {
|
|
switch fw {
|
|
case "ufw":
|
|
return NewUfw(logger.New(false))
|
|
case "iptables":
|
|
return NewIptables(logger.New(false), config)
|
|
case "nftables":
|
|
return NewNftables(logger.New(false), config)
|
|
case "firewalld":
|
|
return NewFirewalld(logger.New(false))
|
|
default:
|
|
panic(fmt.Sprintf("Unknown firewall: %s", fw))
|
|
}
|
|
}
|