mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-25 09:50:24 +08:00
began refactoring of client
This commit is contained in:
worker-9
@@ -1,82 +1,121 @@
|
||||
package local
|
||||
|
||||
import (
|
||||
//"github.com/davecgh/go-spew/spew"
|
||||
//"github.com/davecgh/go-spew/spew"
|
||||
"errors"
|
||||
"github.com/gravitl/netmaker/netclient/config"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"io"
|
||||
"runtime"
|
||||
"strings"
|
||||
"os"
|
||||
"os/exec"
|
||||
|
||||
"github.com/gravitl/netmaker/netclient/config"
|
||||
)
|
||||
|
||||
func SetIPForwarding() error {
|
||||
os := runtime.GOOS
|
||||
var err error
|
||||
switch os {
|
||||
case "linux":
|
||||
err = SetIPForwardingLinux()
|
||||
default:
|
||||
err = errors.New("This OS is not supported")
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func SetIPForwardingLinux() error {
|
||||
out, err := RunCmd("sysctl net.ipv4.ip_forward")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
log.Println("WARNING: Error encountered setting ip forwarding. This can break functionality.")
|
||||
return err
|
||||
} else {
|
||||
s := strings.Fields(string(out))
|
||||
if s[2] != "1" {
|
||||
_, err = RunCmd("sysctl -w net.ipv4.ip_forward=1")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
log.Println("WARNING: Error encountered setting ip forwarding. You may want to investigate this.")
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func RunCmd(command string) (string, error) {
|
||||
args := strings.Fields(command)
|
||||
out, err := exec.Command(args[0], args[1:]...).Output()
|
||||
return string(out), err
|
||||
}
|
||||
|
||||
func RunCmds(commands []string) error {
|
||||
var err error
|
||||
for _, command := range commands {
|
||||
args := strings.Fields(command)
|
||||
out, err := exec.Command(args[0], args[1:]...).Output()
|
||||
if string(out) != "" {
|
||||
var err error
|
||||
for _, command := range commands {
|
||||
args := strings.Fields(command)
|
||||
out, err := exec.Command(args[0], args[1:]...).Output()
|
||||
if string(out) != "" {
|
||||
log.Println(string(out))
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return err
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func FileExists(f string) bool {
|
||||
info, err := os.Stat(f)
|
||||
if os.IsNotExist(err) {
|
||||
return false
|
||||
}
|
||||
return !info.IsDir()
|
||||
info, err := os.Stat(f)
|
||||
if os.IsNotExist(err) {
|
||||
return false
|
||||
}
|
||||
return !info.IsDir()
|
||||
}
|
||||
|
||||
func ConfigureSystemD(network string) error {
|
||||
/*
|
||||
path, err := os.Getwd()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
path, err := os.Getwd()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
*/
|
||||
//binarypath := path + "/netclient"
|
||||
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
binarypath := dir + "/netclient"
|
||||
|
||||
_, err = os.Stat("/etc/netclient")
|
||||
if os.IsNotExist(err) {
|
||||
os.Mkdir("/etc/netclient", 744)
|
||||
} else if err != nil {
|
||||
log.Println("couldnt find or create /etc/netclient")
|
||||
return err
|
||||
}
|
||||
|
||||
if !FileExists("/usr/local/bin/netclient") {
|
||||
os.Symlink("/etc/netclient/netclient","/usr/local/bin/netclient")
|
||||
/*
|
||||
_, err = copy(binarypath, "/usr/local/bin/netclient")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
*/
|
||||
}
|
||||
if !FileExists("/etc/netclient/netclient") {
|
||||
_, err = copy(binarypath, "/etc/netclient/netclient")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
binarypath := dir + "/netclient"
|
||||
|
||||
_, err = os.Stat("/etc/netclient")
|
||||
if os.IsNotExist(err) {
|
||||
os.Mkdir("/etc/netclient", 744)
|
||||
} else if err != nil {
|
||||
log.Println("couldnt find or create /etc/netclient")
|
||||
return err
|
||||
}
|
||||
|
||||
if !FileExists("/usr/local/bin/netclient") {
|
||||
os.Symlink("/etc/netclient/netclient", "/usr/local/bin/netclient")
|
||||
/*
|
||||
_, err = copy(binarypath, "/usr/local/bin/netclient")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
*/
|
||||
}
|
||||
if !FileExists("/etc/netclient/netclient") {
|
||||
_, err = copy(binarypath, "/etc/netclient/netclient")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
systemservice := `[Unit]
|
||||
Description=network check for remote peers and local config
|
||||
@@ -94,18 +133,18 @@ WantedBy=multi-user.target
|
||||
Description=Calls the Netmaker Mesh Client Service
|
||||
|
||||
`
|
||||
systemtimer = systemtimer + "Requires=netclient@"+network+".service"
|
||||
systemtimer = systemtimer + "Requires=netclient@" + network + ".service"
|
||||
|
||||
systemtimer = systemtimer +
|
||||
`
|
||||
systemtimer = systemtimer +
|
||||
`
|
||||
|
||||
[Timer]
|
||||
|
||||
`
|
||||
systemtimer = systemtimer + "Unit=netclient@"+network+".service"
|
||||
systemtimer = systemtimer + "Unit=netclient@" + network + ".service"
|
||||
|
||||
systemtimer = systemtimer +
|
||||
`
|
||||
systemtimer = systemtimer +
|
||||
`
|
||||
|
||||
OnCalendar=*:*:0/30
|
||||
|
||||
@@ -117,67 +156,41 @@ WantedBy=timers.target
|
||||
timerbytes := []byte(systemtimer)
|
||||
|
||||
if !FileExists("/etc/systemd/system/netclient@.service") {
|
||||
err = ioutil.WriteFile("/etc/systemd/system/netclient@.service", servicebytes, 0644)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
err = ioutil.WriteFile("/etc/systemd/system/netclient@.service", servicebytes, 0644)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if !FileExists("/etc/systemd/system/netclient-"+network+".timer") {
|
||||
err = ioutil.WriteFile("/etc/systemd/system/netclient-"+network+".timer", timerbytes, 0644)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
if !FileExists("/etc/systemd/system/netclient-" + network + ".timer") {
|
||||
err = ioutil.WriteFile("/etc/systemd/system/netclient-"+network+".timer", timerbytes, 0644)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return err
|
||||
}
|
||||
}
|
||||
//sysExec, err := exec.LookPath("systemctl")
|
||||
|
||||
cmdSysEnableService := exec.Command("systemctl", "enable", "netclient@.service")/*&exec.Cmd {
|
||||
Path: sysExec,
|
||||
Args: []string{ sysExec, "enable", "netclient@.service" },
|
||||
Stdout: os.Stdout,
|
||||
Stderr: os.Stdout,
|
||||
}*/
|
||||
cmdSysDaemonReload := exec.Command("systemctl", "daemon-reload")/*&exec.Cmd {
|
||||
Path: sysExec,
|
||||
Args: []string{ sysExec, "daemon-reload"},
|
||||
Stdout: os.Stdout,
|
||||
Stderr: os.Stdout,
|
||||
}*/
|
||||
cmdSysEnableTimer := exec.Command("systemctl", "enable", "netclient-"+network+".timer")/*&exec.Cmd {
|
||||
Path: sysExec,
|
||||
Args: []string{ sysExec, "enable", "netclient-"+network+".timer" },
|
||||
Stdout: os.Stdout,
|
||||
Stderr: os.Stdout,
|
||||
}*/
|
||||
cmdSysStartTimer := exec.Command("systemctl", "start", "netclient-"+network+".timer")/*&exec.Cmd {
|
||||
Path: sysExec,
|
||||
Args: []string{ sysExec, "start", "netclient-"+network+".timer"},
|
||||
Stdout: os.Stdout,
|
||||
Stderr: os.Stdout,
|
||||
}*/
|
||||
|
||||
err = cmdSysEnableService.Run()
|
||||
if err != nil {
|
||||
log.Println("Error enabling netclient@.service. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
err = cmdSysDaemonReload.Run()
|
||||
if err != nil {
|
||||
log.Println("Error reloading system daemons. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
err = cmdSysEnableTimer.Run()
|
||||
if err != nil {
|
||||
log.Println("Error enabling netclient.timer. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
err = cmdSysStartTimer.Run()
|
||||
if err != nil {
|
||||
log.Println("Error starting netclient-"+network+".timer. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
_, err = RunCmd("systemctl enable netclient@.service")
|
||||
if err != nil {
|
||||
log.Println("Error enabling netclient@.service. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
_, err = RunCmd("systemctl daemon-reload")
|
||||
if err != nil {
|
||||
log.Println("Error reloading system daemons. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
_, err = RunCmd("systemctl enable netclient-" + network + ".timer")
|
||||
if err != nil {
|
||||
log.Println("Error enabling netclient.timer. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
_, err = RunCmd("systemctl start netclient-" + network + ".timer")
|
||||
if err != nil {
|
||||
log.Println("Error starting netclient-" + network + ".timer. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -188,7 +201,7 @@ func isOnlyService(network string) (bool, error) {
|
||||
return isonly, err
|
||||
}
|
||||
count := len(files)
|
||||
if count == 0 {
|
||||
if count == 0 {
|
||||
isonly = true
|
||||
}
|
||||
return isonly, err
|
||||
@@ -196,78 +209,67 @@ func isOnlyService(network string) (bool, error) {
|
||||
}
|
||||
|
||||
func RemoveSystemDServices(network string) error {
|
||||
//sysExec, err := exec.LookPath("systemctl")
|
||||
|
||||
//sysExec, err := exec.LookPath("systemctl")
|
||||
|
||||
fullremove, err := isOnlyService(network)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
cmdSysDisableService := exec.Command("systemctl","disable","netclient@.service")
|
||||
cmdSysDaemonReload := exec.Command("systemctl","daemon-reload")
|
||||
cmdSysResetFailed := exec.Command("systemctl","reset-failed")
|
||||
cmdSysStopTimer := exec.Command("systemctl", "stop", "netclient-"+network+".timer")
|
||||
cmdSysDisableTimer := exec.Command("systemctl", "disable", "netclient-"+network+".timer")
|
||||
if err != nil {
|
||||
log.Println("Error stopping netclient@.service. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
if fullremove {
|
||||
err = cmdSysDisableService.Run()
|
||||
if err != nil {
|
||||
log.Println("Error disabling netclient@.service. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
_, err = RunCmd("systemctl disable netclient@.service")
|
||||
if err != nil {
|
||||
log.Println("Error disabling netclient@.service. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
_, err = RunCmd("systemctl daemon-reload")
|
||||
if err != nil {
|
||||
log.Println("Error stopping netclient-" + network + ".timer. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
_, err = RunCmd("systemctl disable netclient-" + network + ".timer")
|
||||
if err != nil {
|
||||
log.Println("Error disabling netclient-" + network + ".timer. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
err = cmdSysStopTimer.Run()
|
||||
if err != nil {
|
||||
log.Println("Error stopping netclient-"+network+".timer. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
err = cmdSysDisableTimer.Run()
|
||||
if err != nil {
|
||||
log.Println("Error disabling netclient-"+network+".timer. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
if fullremove {
|
||||
if FileExists("/etc/systemd/system/netclient@.service") {
|
||||
err = os.Remove("/etc/systemd/system/netclient@.service")
|
||||
}
|
||||
}
|
||||
if FileExists("/etc/systemd/system/netclient-"+network+".timer") {
|
||||
err = os.Remove("/etc/systemd/system/netclient-"+network+".timer")
|
||||
if FileExists("/etc/systemd/system/netclient-" + network + ".timer") {
|
||||
err = os.Remove("/etc/systemd/system/netclient-" + network + ".timer")
|
||||
}
|
||||
if err != nil {
|
||||
log.Println("Error removing file. Please investigate.")
|
||||
log.Println(err)
|
||||
log.Println("Error removing file. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
_, err = RunCmd("systemctl daemon-reload")
|
||||
if err != nil {
|
||||
log.Println("Error reloading system daemons. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
_, err = RunCmd("systemctl reset-failed")
|
||||
if err != nil {
|
||||
log.Println("Error reseting failed system services. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
err = cmdSysDaemonReload.Run()
|
||||
if err != nil {
|
||||
log.Println("Error reloading system daemons. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
err = cmdSysResetFailed.Run()
|
||||
if err != nil {
|
||||
log.Println("Error reseting failed system services. Please investigate.")
|
||||
log.Println(err)
|
||||
}
|
||||
return err
|
||||
|
||||
}
|
||||
|
||||
func WipeLocal(network string) error{
|
||||
cfg, err := config.ReadConfig(network)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
nodecfg := cfg.Node
|
||||
ifacename := nodecfg.Interface
|
||||
func WipeLocal(network string) error {
|
||||
cfg, err := config.ReadConfig(network)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
nodecfg := cfg.Node
|
||||
ifacename := nodecfg.Interface
|
||||
|
||||
//home, err := homedir.Dir()
|
||||
home := "/etc/netclient"
|
||||
//home, err := homedir.Dir()
|
||||
home := "/etc/netclient"
|
||||
if FileExists(home + "/netconfig-" + network) {
|
||||
_ = os.Remove(home + "/netconfig-" + network)
|
||||
_ = os.Remove(home + "/netconfig-" + network)
|
||||
}
|
||||
if FileExists(home + "/nettoken-" + network) {
|
||||
_ = os.Remove(home + "/nettoken-" + network)
|
||||
@@ -276,79 +278,79 @@ func WipeLocal(network string) error{
|
||||
_ = os.Remove(home + "/wgkey-" + network)
|
||||
}
|
||||
|
||||
ipExec, err := exec.LookPath("ip")
|
||||
ipExec, err := exec.LookPath("ip")
|
||||
|
||||
if ifacename != "" {
|
||||
cmdIPLinkDel := &exec.Cmd {
|
||||
Path: ipExec,
|
||||
Args: []string{ ipExec, "link", "del", ifacename },
|
||||
Stdout: os.Stdout,
|
||||
Stderr: os.Stdout,
|
||||
}
|
||||
err = cmdIPLinkDel.Run()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
if nodecfg.PostDown != "" {
|
||||
runcmds := strings.Split(nodecfg.PostDown, "; ")
|
||||
err = RunCmds(runcmds)
|
||||
if err != nil {
|
||||
log.Println("Error encountered running PostDown: " + err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
return err
|
||||
if ifacename != "" {
|
||||
cmdIPLinkDel := &exec.Cmd{
|
||||
Path: ipExec,
|
||||
Args: []string{ipExec, "link", "del", ifacename},
|
||||
Stdout: os.Stdout,
|
||||
Stderr: os.Stdout,
|
||||
}
|
||||
err = cmdIPLinkDel.Run()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
if nodecfg.PostDown != "" {
|
||||
runcmds := strings.Split(nodecfg.PostDown, "; ")
|
||||
err = RunCmds(runcmds)
|
||||
if err != nil {
|
||||
log.Println("Error encountered running PostDown: " + err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
return err
|
||||
|
||||
}
|
||||
|
||||
func WipeGRPCClient() error{
|
||||
home := "/etc/netclient"
|
||||
_ = os.Remove(home + "/netconfig-global-001")
|
||||
func WipeGRPCClient() error {
|
||||
home := "/etc/netclient"
|
||||
_ = os.Remove(home + "/netconfig-global-001")
|
||||
|
||||
ipExec, err := exec.LookPath("ip")
|
||||
|
||||
cmdIPLinkDel := &exec.Cmd {
|
||||
Path: ipExec,
|
||||
Args: []string{ ipExec, "link", "del", "grpc-wg-001" },
|
||||
Stdout: os.Stdout,
|
||||
Stderr: os.Stdout,
|
||||
}
|
||||
err = cmdIPLinkDel.Run()
|
||||
return err
|
||||
cmdIPLinkDel := &exec.Cmd{
|
||||
Path: ipExec,
|
||||
Args: []string{ipExec, "link", "del", "grpc-wg-001"},
|
||||
Stdout: os.Stdout,
|
||||
Stderr: os.Stdout,
|
||||
}
|
||||
err = cmdIPLinkDel.Run()
|
||||
return err
|
||||
}
|
||||
|
||||
func HasNetwork(network string) bool{
|
||||
func HasNetwork(network string) bool {
|
||||
|
||||
return FileExists("/etc/systemd/system/netclient-"+network+".timer") ||
|
||||
FileExists("/etc/netclient/netconfig-"+network)
|
||||
return FileExists("/etc/systemd/system/netclient-"+network+".timer") ||
|
||||
FileExists("/etc/netclient/netconfig-"+network)
|
||||
|
||||
}
|
||||
|
||||
func copy(src, dst string) (int64, error) {
|
||||
sourceFileStat, err := os.Stat(src)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
sourceFileStat, err := os.Stat(src)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
if !sourceFileStat.Mode().IsRegular() {
|
||||
return 0, errors.New(src + " is not a regular file")
|
||||
}
|
||||
if !sourceFileStat.Mode().IsRegular() {
|
||||
return 0, errors.New(src + " is not a regular file")
|
||||
}
|
||||
|
||||
source, err := os.Open(src)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer source.Close()
|
||||
source, err := os.Open(src)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer source.Close()
|
||||
|
||||
destination, err := os.Create(dst)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer destination.Close()
|
||||
nBytes, err := io.Copy(destination, source)
|
||||
destination, err := os.Create(dst)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer destination.Close()
|
||||
nBytes, err := io.Copy(destination, source)
|
||||
err = os.Chmod(dst, 0755)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
return nBytes, err
|
||||
return nBytes, err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user