From a1ef2d42d320f075c75e8cef52a879114441c103 Mon Sep 17 00:00:00 2001 From: sunty <1172534965@qq.com> Date: 星期三, 30 十月 2019 15:54:07 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- extend/sys/system.go | 96 +++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 94 insertions(+), 2 deletions(-) diff --git a/extend/sys/system.go b/extend/sys/system.go index 563d85b..cccba7e 100644 --- a/extend/sys/system.go +++ b/extend/sys/system.go @@ -2,13 +2,24 @@ import ( "errors" + "os" "os/exec" + "strconv" + "webserver/extend/config" + "webserver/extend/util" "fmt" "net" "strings" "time" ) + +func execRootCommand(cmd string) ([]byte, error) { + pwd := config.Server.SudoPassword + cmdStr := fmt.Sprintf("echo %s | sudo -S %s", pwd, cmd) + + return exec.Command("/bin/sh", "-c", cmdStr).Output() +} // 妫�鏌� root鏉冮檺 func CheckRootPermissions() bool { @@ -18,6 +29,79 @@ } return true +} + +// 鑾峰彇鏈満缃戝崱IP +func GetLocalIP(networkName string) (ipv4 string, mask string, err error) { + interfaces, err := net.Interfaces() + if err != nil { + return "", "", err + } + + for _, i := range interfaces { + byName, err := net.InterfaceByName(i.Name) + if err != nil { + return "", "", err + } + addresses, err := byName.Addrs() + for _, v := range addresses { + if ipnet, ok := v.(*net.IPNet); ok && !ipnet.IP.IsLoopback() { + if ipnet.IP.To4() != nil { + if byName.Name == networkName { + maskStr := ipnet.Mask.String() + mask64, _ := strconv.ParseUint(maskStr, 16, 32) + return ipnet.IP.String(), util.IpIntToString(int(mask64)), nil + } + } + } + } + } + return "", "", errors.New("ipv4 not found") +} + +// 鑾峰彇鏈満缃戝叧ip +func GetDefaultRoute(networkName string) (route string, err error) { + cmdStr := fmt.Sprintf("ip route show | grep -P \"^default.*%s\" | awk '{printf $3}'", networkName) + cmd := exec.Command("/bin/sh", "-c", cmdStr) + b, err := cmd.Output() + if err != nil { + return "", err + } + + return string(b), nil +} + +func GetDnsServer() (dns string, err error) { + cmd := exec.Command("/bin/sh", "-c", "cat /etc/resolv.conf | grep nameserver | head -1 |awk '{printf $2}'") + b, err := cmd.Output() + if err != nil { + return "", err + } + + return string(b), nil +} + +// 閰嶇疆鏈嶅姟鍣ㄧ綉缁� +func SetNetworkConfig(ipv4, netmask, gateway, dns string) (bool, string) { + networkConfigScript := "/opt/vasystem/bin/netconfig" + ifname := config.Server.NetworkAdapter + localIPv4, localNetMask, _ := GetLocalIP(ifname) + localGW, _ := GetDefaultRoute(ifname) + localDNS, _ := GetDnsServer() + + if localIPv4 == ipv4 && localNetMask == netmask && localGW == gateway && localDNS == dns { + return true, "" + } + + // 妫�鏌ユ枃浠舵槸鍚﹀瓨鍦� + if _, err := os.Stat(networkConfigScript); err != nil { + return false, "network Config Script not exist" + } + // # netconfig enp59s0f0 192.168.1.2 255.255.255.0 192.168.1.1 192.168.100.1 + cmdStr := fmt.Sprintf("%s %s %s %s %s %s", networkConfigScript, ifname, ipv4, netmask, gateway, dns) + stdout, err := execRootCommand(cmdStr) + + return err == nil, string(stdout) } // 閰嶇疆鏃跺尯 @@ -76,8 +160,10 @@ return false } - args := []string{"-s", newTime} - exec.Command("date", args...).Run() + // args := []string{"-s", newTime} + // exec.Command("date", args...).Run() + dateCMD := fmt.Sprintf("date -s \"%s\"", newTime) + execRootCommand(dateCMD) stopNTPCron() return true @@ -111,3 +197,9 @@ ntpdate := fmt.Sprintf("/usr/sbin/ntpdate %s", server) return true, exec.Command("/bin/sh", "-c", ntpdate).Run() } + +func Reboot() (bool, string) { + stdout, err := execRootCommand("reboot") + + return err == nil, string(stdout) +} -- Gitblit v1.8.0