mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2025-01-19 08:19:15 +08:00
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package firewall
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/buserr"
|
|
"github.com/1Panel-dev/1Panel/backend/constant"
|
|
"github.com/1Panel-dev/1Panel/backend/utils/firewall/client"
|
|
)
|
|
|
|
type FirewallClient interface {
|
|
Name() string // ufw firewalld
|
|
Start() error
|
|
Stop() error
|
|
Restart() error
|
|
Reload() error
|
|
Status() (string, error) // running not running
|
|
Version() (string, error)
|
|
|
|
ListPort() ([]client.FireInfo, error)
|
|
ListForward() ([]client.FireInfo, error)
|
|
ListAddress() ([]client.FireInfo, error)
|
|
|
|
Port(port client.FireInfo, operation string) error
|
|
RichRules(rule client.FireInfo, operation string) error
|
|
PortForward(info client.Forward, operation string) error
|
|
|
|
EnableForward() error
|
|
}
|
|
|
|
func NewFirewallClient() (FirewallClient, error) {
|
|
_, firewalldErr := os.Stat("/usr/sbin/firewalld")
|
|
_, ufwErr := os.Stat("/usr/sbin/ufw")
|
|
|
|
if firewalldErr == nil && ufwErr == nil {
|
|
return nil, buserr.New("firewalld and ufw both found, only one firewall should be active")
|
|
}
|
|
|
|
if firewalldErr == nil {
|
|
return client.NewFirewalld()
|
|
}
|
|
if ufwErr == nil {
|
|
return client.NewUfw()
|
|
}
|
|
return nil, buserr.New(constant.ErrFirewall)
|
|
}
|