feat: Rename and set as method NftablesSetup -> Setup, fix template and types config, add create template config in system, update logic finds firewalls on system, add BurntSushi/toml as dependencies
This commit is contained in:
@@ -5,6 +5,8 @@ import (
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/BurntSushi/toml"
|
||||
)
|
||||
|
||||
var DetectedFirewall string
|
||||
@@ -39,31 +41,65 @@ func CreateConf() error {
|
||||
if err := os.Chmod(configPath, 0600); err != nil {
|
||||
return fmt.Errorf("failed to set permissions: %w", err)
|
||||
}
|
||||
|
||||
err = os.WriteFile(configPath, []byte(Base_config), 0600)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to write config file: %w", err)
|
||||
}
|
||||
fmt.Printf(" Config file created: %s\n", configPath)
|
||||
return nil
|
||||
}
|
||||
|
||||
func FindFirewall() error {
|
||||
|
||||
if os.Getegid() != 0 {
|
||||
fmt.Printf("Firewall settings needs sudo privileges\n")
|
||||
os.Exit(1)
|
||||
}
|
||||
firewalls := []string{"iptables", "nft", "firewall-cmd", "ufw"}
|
||||
|
||||
firewalls := []string{"nft", "firewall-cmd", "iptables", "ufw"}
|
||||
for _, firewall := range firewalls {
|
||||
_, err := exec.LookPath(firewall)
|
||||
if err == nil {
|
||||
if firewall == "firewall-cmd" {
|
||||
switch firewall {
|
||||
case "firewall-cmd":
|
||||
DetectedFirewall = "firewalld"
|
||||
}
|
||||
if firewall == "nft" {
|
||||
case "nft":
|
||||
DetectedFirewall = "nftables"
|
||||
default:
|
||||
DetectedFirewall = firewall
|
||||
}
|
||||
DetectedFirewall = firewall
|
||||
fmt.Printf("Detected firewall: %s\n", firewall)
|
||||
|
||||
fmt.Printf("Detected firewall: %s\n", DetectedFirewall)
|
||||
|
||||
cfg := &Config{}
|
||||
_, err := toml.DecodeFile("/etc/banforge/config.toml", cfg)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to decode config: %w", err)
|
||||
}
|
||||
|
||||
cfg.Firewall.Name = DetectedFirewall
|
||||
|
||||
file, err := os.Create("/etc/banforge/config.toml")
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to create config file: %w", err)
|
||||
}
|
||||
|
||||
encoder := toml.NewEncoder(file)
|
||||
if err := encoder.Encode(cfg); err != nil {
|
||||
err = file.Close()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to close file: %w", err)
|
||||
}
|
||||
return fmt.Errorf("failed to encode config: %w", err)
|
||||
}
|
||||
|
||||
if err := file.Close(); err != nil {
|
||||
return fmt.Errorf("failed to close file: %w", err)
|
||||
}
|
||||
|
||||
fmt.Printf("Config updated with firewall: %s\n", DetectedFirewall)
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return fmt.Errorf("no firewall found (checked ufw, firewall-cmd, iptables, nft) please install one of them")
|
||||
|
||||
return fmt.Errorf("firewall not found")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user