mirror of
https://github.com/Control-D-Inc/ctrld.git
synced 2026-05-27 12:52:27 +02:00
Cuong Manh Le
97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
package pfsense
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
|
|
"github.com/Control-D-Inc/ctrld"
|
|
"github.com/kardianos/service"
|
|
)
|
|
|
|
const (
|
|
Name = "pfsens"
|
|
|
|
rcPath = "/usr/local/etc/rc.d"
|
|
unboundRcPath = rcPath + "/unbound"
|
|
dnsmasqRcPath = rcPath + "/dnsmasq"
|
|
)
|
|
|
|
const pfsenseInitScript = `#!/bin/sh
|
|
|
|
# PROVIDE: {{.Name}}
|
|
# REQUIRE: SERVERS
|
|
# REQUIRE: unbound dnsmasq securelevel
|
|
# KEYWORD: shutdown
|
|
|
|
. /etc/rc.subr
|
|
|
|
name="{{.Name}}"
|
|
{{.Name}}_env="IS_DAEMON=1"
|
|
pidfile="/var/run/${name}.pid"
|
|
command="/usr/sbin/daemon"
|
|
daemon_args="-P ${pidfile} -r -t \"${name}: daemon\"{{if .WorkingDirectory}} -c {{.WorkingDirectory}}{{end}}"
|
|
command_args="${daemon_args} {{.Path}}{{range .Arguments}} {{.}}{{end}}"
|
|
|
|
run_rc_command "$1"
|
|
`
|
|
|
|
type Pfsense struct {
|
|
cfg *ctrld.Config
|
|
svcName string
|
|
}
|
|
|
|
// New returns a router.Router for configuring/setup/run ctrld on Pfsense routers.
|
|
func New(cfg *ctrld.Config) *Pfsense {
|
|
return &Pfsense{cfg: cfg}
|
|
}
|
|
|
|
func (p *Pfsense) ConfigureService(svc *service.Config) error {
|
|
svc.Option["SysvScript"] = pfsenseInitScript
|
|
p.svcName = svc.Name
|
|
return nil
|
|
}
|
|
|
|
func (p *Pfsense) Install(config *service.Config) error {
|
|
// pfsense need ".sh" extension for script to be run at boot.
|
|
// See: https://docs.netgate.com/pfsense/en/latest/development/boot-commands.html#shell-script-option
|
|
oldname := filepath.Join(rcPath, p.svcName)
|
|
newname := filepath.Join(rcPath, p.svcName+".sh")
|
|
_ = os.Remove(newname)
|
|
if err := os.Symlink(oldname, newname); err != nil {
|
|
return fmt.Errorf("os.Symlink: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Pfsense) Uninstall(config *service.Config) error {
|
|
if err := os.Remove(filepath.Join(rcPath, p.svcName+".sh")); err != nil {
|
|
return fmt.Errorf("os.Remove: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Pfsense) PreRun() error {
|
|
// TODO: remove this hacky solution.
|
|
// If Pfsense is in DNS Resolver mode, ensure no unbound processes running.
|
|
_ = exec.Command("killall", "unbound").Run()
|
|
|
|
// If Pfsense is in DNS Forwarder mode, ensure no dnsmasq processes running.
|
|
_ = exec.Command("killall", "dnsmasq").Run()
|
|
return nil
|
|
}
|
|
|
|
func (p *Pfsense) Setup() error {
|
|
return nil
|
|
}
|
|
|
|
func (p *Pfsense) Cleanup() error {
|
|
if p.cfg.FirstListener().IsDirectDnsListener() {
|
|
_ = exec.Command(unboundRcPath, "onerestart").Run()
|
|
_ = exec.Command(dnsmasqRcPath, "onerestart").Run()
|
|
}
|
|
|
|
return nil
|
|
}
|