143 lines
3.0 KiB
Go
143 lines
3.0 KiB
Go
package warewulfd
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"strconv"
|
|
"syscall"
|
|
|
|
"github.com/hpcng/warewulf/internal/pkg/util"
|
|
"github.com/hpcng/warewulf/internal/pkg/version"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
WAREWULFD_PIDFILE = "/var/run/warewulfd.pid"
|
|
WAREWULFD_LOGFILE = "/var/log/warewulfd.log"
|
|
)
|
|
|
|
func DaemonStart() error {
|
|
if os.Getenv("WAREWULFD_BACKGROUND") == "1" {
|
|
err := RunServer()
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to run server")
|
|
}
|
|
|
|
} else {
|
|
if util.IsFile(WAREWULFD_PIDFILE) {
|
|
return errors.New("process is already running")
|
|
}
|
|
|
|
os.Setenv("WAREWULFD_BACKGROUND", "1")
|
|
|
|
f, err := os.OpenFile(WAREWULFD_LOGFILE, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
p, err := os.OpenFile(WAREWULFD_PIDFILE, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer p.Close()
|
|
|
|
cmd := exec.Command(os.Args[0], "server", "start")
|
|
cmd.Stdout = f
|
|
cmd.Stderr = f
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to start command")
|
|
}
|
|
pid := cmd.Process.Pid
|
|
|
|
fmt.Fprintf(p, "%d", pid)
|
|
|
|
fmt.Printf("Started Warewulf (%s) server at PID: %d\n", version.GetVersion(), pid)
|
|
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func DaemonStatus() error {
|
|
if !util.IsFile(WAREWULFD_PIDFILE) {
|
|
return errors.New("Warewulf server is not running")
|
|
}
|
|
|
|
dat, err := ioutil.ReadFile(WAREWULFD_PIDFILE)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not read Warewulfd PID file")
|
|
}
|
|
|
|
pid, _ := strconv.Atoi(string(dat))
|
|
process, err := os.FindProcess(pid)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to find running PID")
|
|
} else {
|
|
err := process.Signal(syscall.Signal(0))
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to send process SIGCONT")
|
|
} else {
|
|
fmt.Printf("Warewulf server is running at PID: %d\n", pid)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func DaemonReload() error {
|
|
if !util.IsFile(WAREWULFD_PIDFILE) {
|
|
return errors.New("Warewulf server is not running")
|
|
}
|
|
|
|
dat, err := ioutil.ReadFile(WAREWULFD_PIDFILE)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not read Warewulfd PID file")
|
|
}
|
|
|
|
pid, _ := strconv.Atoi(string(dat))
|
|
process, err := os.FindProcess(pid)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to find running PID")
|
|
} else {
|
|
err := process.Signal(syscall.Signal(syscall.SIGHUP))
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to send process SIGHUP")
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func DaemonStop() error {
|
|
if !util.IsFile(WAREWULFD_PIDFILE) {
|
|
fmt.Printf("Warewulf daemon process not running\n")
|
|
return nil
|
|
}
|
|
|
|
dat, err := ioutil.ReadFile(WAREWULFD_PIDFILE)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_ = os.Remove(WAREWULFD_PIDFILE)
|
|
|
|
pid, _ := strconv.Atoi(string(dat))
|
|
process, err := os.FindProcess(pid)
|
|
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to find running PID")
|
|
} else {
|
|
err := process.Signal(syscall.Signal(15))
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to send process SIGTERM")
|
|
} else {
|
|
fmt.Printf("Terminated Warewulf server at PID: %d\n", pid)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|