84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package controllers
|
||
|
||
import (
|
||
"github.com/Mmx233/BitSrunLoginGo/global"
|
||
srunModels "github.com/Mmx233/BitSrunLoginGo/models"
|
||
"github.com/Mmx233/BitSrunLoginGo/util"
|
||
"os"
|
||
"os/exec"
|
||
"time"
|
||
)
|
||
|
||
// Guardian 守护模式逻辑
|
||
func Guardian(output bool) {
|
||
util.Log.OutPut = output
|
||
GuardianDuration := time.Duration(global.Config.Settings.Guardian.Duration) * time.Second
|
||
|
||
if global.Config.Settings.Daemon.Enable {
|
||
go Daemon.DaemonChan()
|
||
|
||
if e := Daemon.MarkDaemon(); e != nil {
|
||
util.Log.Warn("写入daemon标记文件失败: ", e)
|
||
}
|
||
}
|
||
|
||
var c = make(chan bool)
|
||
for {
|
||
util.Log.OutPut = output
|
||
go func() {
|
||
defer func() {
|
||
_ = recover()
|
||
}()
|
||
if global.Config.Settings.Basic.Interfaces == "" { //单网卡
|
||
if !util.Checker.NetOk(global.Transports(nil)) {
|
||
util.Log.Info("检测到掉线, trying to login")
|
||
e := Login(output, true, nil)
|
||
if e != nil {
|
||
util.Log.Warn("登陆失败: ", e)
|
||
}
|
||
} else {
|
||
if global.Config.Settings.Debug.Enable {
|
||
util.Log.Debug("Network ok")
|
||
}
|
||
}
|
||
} else { //多网卡
|
||
interfaces, e := util.GetInterfaceAddr()
|
||
if e == nil {
|
||
var down []srunModels.Eth
|
||
for _, eth := range interfaces {
|
||
if !util.Checker.NetOk(global.Transports(eth.Addr)) {
|
||
util.Log.Info("检测到掉线网口 ", eth.Name)
|
||
down = append(down, eth)
|
||
}
|
||
}
|
||
|
||
for _, eth := range down {
|
||
util.Log.Info(eth.Name)
|
||
e := Login(output, true, eth.Addr)
|
||
if e != nil {
|
||
util.Log.Warn("网口 ", eth.Name+" 登录失败: ", e)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
c <- false
|
||
}()
|
||
<-c
|
||
time.Sleep(GuardianDuration)
|
||
}
|
||
}
|
||
|
||
// EnterGuardian 守护模式入口，控制是否进入daemon
|
||
func EnterGuardian() {
|
||
util.Log.Info("[Guardian mode]")
|
||
if global.Config.Settings.Daemon.Enable || global.Flags.Daemon {
|
||
if err := exec.Command(os.Args[0], append(os.Args[1:], "--running-daemon")...).Start(); err != nil {
|
||
util.Log.Fatal("启动守护失败: ", err)
|
||
}
|
||
util.Log.Info("[Daemon mode entered]")
|
||
return
|
||
}
|
||
Guardian(true)
|
||
}
|