mirror of
https://github.com/gravitl/netmaker.git
synced 2024-09-21 15:56:07 +08:00
134 lines
3.6 KiB
Go
134 lines
3.6 KiB
Go
package daemon
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/gravitl/netmaker/netclient/ncutils"
|
|
)
|
|
|
|
// SetupFreebsdDaemon -- sets up daemon for freebsd
|
|
func SetupFreebsdDaemon() error {
|
|
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
binarypath := dir + "/netclient"
|
|
|
|
_, err = os.Stat("/etc/netclient/config")
|
|
if os.IsNotExist(err) {
|
|
os.MkdirAll("/etc/netclient/config", 0744)
|
|
} else if err != nil {
|
|
log.Println("couldnt find or create /etc/netclient")
|
|
return err
|
|
}
|
|
//install binary
|
|
//should check if the existing binary is the corect version -- for now only copy if file doesn't exist
|
|
if !ncutils.FileExists(EXEC_DIR + "netclient") {
|
|
err = ncutils.Copy(binarypath, EXEC_DIR+"netclient")
|
|
if err != nil {
|
|
log.Println(err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
rcFile := `#!/bin/sh
|
|
#
|
|
# PROVIDE: netclient
|
|
# REQUIRE: LOGIN
|
|
# KEYWORD: shutdown
|
|
|
|
# Description:
|
|
# This script runs netclient as a service as root on boot
|
|
|
|
# How to use:
|
|
# Place this file in /usr/local/etc/rc.d/
|
|
# Add netclient="YES" to /etc/rc.config.d/netclient
|
|
# To pass args, add netclient_args="daemon" to /etc/rc.config.d/netclient
|
|
|
|
# Freebsd rc library
|
|
. /etc/rc.subr
|
|
|
|
# General Info
|
|
name="netclient" # Safe name of program
|
|
program_name="netclient" # Name of exec
|
|
title="netclient" # Title to display in top/htop
|
|
|
|
# RC.config vars
|
|
load_rc_config $name # Loading rc config vars
|
|
: ${netclient_enable="YES"} # Default: enable netclient
|
|
: ${netclient_runAs="root"} # Default: Run Node-RED as root
|
|
|
|
# Freebsd Setup
|
|
rcvar=netclient_enable # Enables the rc.conf YES/NO flag
|
|
pidfile="/var/run/${program_name}.pid" # File that allows the system to keep track of node-red status
|
|
|
|
# Env Setup
|
|
#export HOME=$( getent passwd "$netclient_runAs" | cut -d: -f6 ) # Gets the home directory of the runAs user
|
|
|
|
# Command Setup
|
|
exec_path="/sbin/${program_name}" # Path to the netclient exec
|
|
output_file="/var/log/${program_name}.log" # Path to netclient logs
|
|
|
|
# Command
|
|
command="/usr/sbin/daemon"
|
|
command_args="-r -t ${title} -u ${netclient_runAs} -o ${output_file} -P ${pidfile} ${exec_path} ${netclient_args}"
|
|
|
|
# Loading Config
|
|
load_rc_config ${name}
|
|
run_rc_command "$1"
|
|
`
|
|
|
|
rcConfig := `netclient="YES"
|
|
netclient_args="daemon"`
|
|
|
|
rcbytes := []byte(rcFile)
|
|
if !ncutils.FileExists("/etc/rc.d/netclient") {
|
|
err := os.WriteFile("/etc/rc.d/netclient", rcbytes, 0744)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
rcConfigbytes := []byte(rcConfig)
|
|
if !ncutils.FileExists("/etc/rc.conf.d/netclient") {
|
|
err := os.WriteFile("/etc/rc.conf.d/netclient", rcConfigbytes, 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
FreebsdDaemon("start")
|
|
return nil
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func FreebsdDaemon(command string) {
|
|
out, err := ncutils.RunCmdFormatted("service netclient "+command, true)
|
|
ncutils.Log("output from RunCmd " + out)
|
|
ncutils.Log("error from RunCmd " + err.Error())
|
|
}
|
|
|
|
func CleanupFreebsd() {
|
|
if err := os.RemoveAll(ncutils.GetNetclientPath()); err != nil {
|
|
ncutils.PrintLog("Removing netclient configs: "+err.Error(), 1)
|
|
}
|
|
if err := os.Remove(EXEC_DIR + "netclient"); err != nil {
|
|
ncutils.PrintLog("Removing netclient binary: "+err.Error(), 1)
|
|
}
|
|
}
|
|
|
|
func RemoveFreebsdDaemon() {
|
|
if ncutils.FileExists("/etc/rc.d/netclient") {
|
|
err := os.Remove("/etc/rc.d/netclient")
|
|
if err != nil {
|
|
ncutils.Log("Error removing /etc/rc.d/netclient. Please investigate.")
|
|
}
|
|
}
|
|
if ncutils.FileExists("/etc/rc.conf.d/netclient") {
|
|
err := os.Remove("/etc/rc.conf.d/netclient")
|
|
if err != nil {
|
|
ncutils.Log("Error removing /etc/rc.conf.d/netclient. Please investigate.")
|
|
}
|
|
}
|
|
}
|