Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 8ce1e6531a | |||
| b00f95b7fb |
191
API.md
Normal file
191
API.md
Normal file
@@ -0,0 +1,191 @@
|
|||||||
|
# API Reference — Smart Network Shutdown Monitor
|
||||||
|
|
||||||
|
完整的 CLI 命令说明与参数参考文档。
|
||||||
|
|
||||||
|
> **注意:** 服务管理相关指令(`start`, `stop`, `restart`, `install`, `uninstall`)均需在**管理员 (Windows) / root (Linux)** 权限下执行。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 服务生命周期
|
||||||
|
|
||||||
|
### `start`
|
||||||
|
唤起并注册后台监控守护进程,服务将随系统开机自动启动。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown start
|
||||||
|
```
|
||||||
|
|
||||||
|
### `stop`
|
||||||
|
停止当前运行的后台网络探测服务。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown stop
|
||||||
|
```
|
||||||
|
|
||||||
|
### `restart`
|
||||||
|
重启后台服务(配置变更后需执行此命令以应用新配置)。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown restart
|
||||||
|
```
|
||||||
|
|
||||||
|
### `status`
|
||||||
|
输出当前服务存活状态、载入的配置参数、日志存放路径及最近 10 条日志。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown status
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 安装与卸载
|
||||||
|
|
||||||
|
### `install`
|
||||||
|
将可执行文件复制至系统目录并写入全局环境变量,同时注册为开机自启服务。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown install
|
||||||
|
```
|
||||||
|
|
||||||
|
### `uninstall`
|
||||||
|
停止并注销后台服务,清理系统目录中的程序副本与环境变量。执行时将询问是否同时清除配置文件与日志。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown uninstall
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 配置管理
|
||||||
|
|
||||||
|
### `config set <Key> <Value>`
|
||||||
|
直接修改指定配置项,并执行合法性校验。改写后需执行 `restart` 使其生效。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 修改目标监控 IP 地址
|
||||||
|
smart-shutdown config set TargetIP 192.168.0.1
|
||||||
|
|
||||||
|
# 修改断网容忍时长(秒)
|
||||||
|
smart-shutdown config set MonitorWindowSeconds 300
|
||||||
|
|
||||||
|
# 修改关机倒计时缓冲(秒)
|
||||||
|
smart-shutdown config set ShutdownCountdown 60
|
||||||
|
|
||||||
|
# 修改探测发包间隔(秒)
|
||||||
|
smart-shutdown config set NormalPingInterval 15
|
||||||
|
```
|
||||||
|
|
||||||
|
**可用配置项:**
|
||||||
|
|
||||||
|
| 参数项 | 含义 | 默认值 |
|
||||||
|
|:---|:---|:---|
|
||||||
|
| `TargetIP` | 目标 IPv4 地址,程序向其发送 ICMP 包检测联通性。 | `192.168.3.1` |
|
||||||
|
| `MonitorWindowSeconds` | 网络中断超出此时长(秒)则触发关机流程。 | `180` |
|
||||||
|
| `ShutdownCountdown` | 正式关机前的倒计时缓冲时间(秒)。 | `60` |
|
||||||
|
| `NormalPingInterval` | 网络正常时的探测发包间隔(秒)。 | `15` |
|
||||||
|
|
||||||
|
**配置文件路径:**
|
||||||
|
- Windows: `C:\ProgramData\SmartNetworkMonitor\config.json`
|
||||||
|
- Linux: `/etc/smart-network-monitor/config.json`
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 后台休眠控制
|
||||||
|
|
||||||
|
### `pause`
|
||||||
|
向正在运行的后台守护进程发送休眠指令,后台停止发包但**不启动前台监控**。必须指定 `--stop-after` 或 `--stop-at` 之一(或同时指定,以先到达的时间为准)。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 休眠 2 小时后自动唤醒
|
||||||
|
smart-shutdown pause --stop-after 2h
|
||||||
|
|
||||||
|
# 休眠至明早 8 点自动唤醒
|
||||||
|
smart-shutdown pause --stop-at "2026-03-25 08:00:00"
|
||||||
|
|
||||||
|
# 同时指定两个时间,取较早者
|
||||||
|
smart-shutdown pause --stop-after 3h --stop-at "2026-03-25 08:00:00"
|
||||||
|
```
|
||||||
|
|
||||||
|
| Flag | 说明 | 格式示例 |
|
||||||
|
|:---|:---|:---|
|
||||||
|
| `--stop-after` | 休眠时长,到期自动唤醒。 | `30m`, `2h`, `1h30m` |
|
||||||
|
| `--stop-at` | 休眠至指定绝对时间自动唤醒。 | `"2026-03-25 08:00:00"` |
|
||||||
|
|
||||||
|
### `resume`
|
||||||
|
立即撤销休眠指令,后台将在下一轮探测周期(约 15 秒内)重新激活。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown resume
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 前台临时监控模式
|
||||||
|
|
||||||
|
直接执行 `smart-shutdown`(不带子命令)可进入前台临时监控模式,日志将写入 `network_monitor_front.log` 与原后台日志隔离。
|
||||||
|
|
||||||
|
- 监控参数(`--target-ip` 等)未指定时,使用配置文件中的值。
|
||||||
|
- **运行时长参数**(`--stop-after` / `--stop-at`)均未指定时,前台监控将**一直运行**,直到手动按 `Ctrl+C` 终止。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown [flags]
|
||||||
|
```
|
||||||
|
|
||||||
|
| Flag | 说明 | 示例 |
|
||||||
|
|:---|:---|:---|
|
||||||
|
| `--target-ip` | 临时覆盖目标监控 IP | `--target-ip 8.8.8.8` |
|
||||||
|
| `--window-sec` | 临时覆盖断网容忍时长(秒) | `--window-sec 60` |
|
||||||
|
| `--shutdown-cnt` | 临时覆盖关机倒计时(秒) | `--shutdown-cnt 10` |
|
||||||
|
| `--ping-interval` | 临时覆盖探测发包间隔(秒) | `--ping-interval 5` |
|
||||||
|
| `--override-bg` | 挂起后台守护进程,由前台全面接管,退出时自动恢复后台 | `--override-bg` |
|
||||||
|
| `--stop-after` | 前台运行指定时长后自动退出 | `--stop-after 2h30m` |
|
||||||
|
| `--stop-at` | 前台运行至指定时间自动退出 | `--stop-at "2026-03-25 08:00:00"` |
|
||||||
|
|
||||||
|
**使用示例:**
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 临时将目标 IP 改为 8.8.8.8,60 秒无响应触发
|
||||||
|
smart-shutdown --target-ip 8.8.8.8 --window-sec 60
|
||||||
|
|
||||||
|
# 接管后台,30 分钟后自动退出并恢复后台
|
||||||
|
smart-shutdown --override-bg --stop-after 30m
|
||||||
|
|
||||||
|
# 接管后台,运行至明早 8 点自动退出并恢复后台
|
||||||
|
smart-shutdown --override-bg --stop-at "2026-03-25 08:00:00"
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 其他
|
||||||
|
|
||||||
|
### `update`
|
||||||
|
联网拉取最新版本并热部署更新。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown update
|
||||||
|
```
|
||||||
|
|
||||||
|
### `--version` / `-V`
|
||||||
|
查看当前版本号并拉取最新发布状态。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown --version
|
||||||
|
```
|
||||||
|
|
||||||
|
### `--verbose` / `-v`
|
||||||
|
打印底层部署及环境追溯 Debug 信息(对所有子命令生效)。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
smart-shutdown status --verbose
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 日志文件位置
|
||||||
|
|
||||||
|
| 场景 | Windows | Linux |
|
||||||
|
|:---|:---|:---|
|
||||||
|
| 后台服务日志 | `C:\ProgramData\SmartNetworkMonitor\logs\network_monitor.log` | `/var/log/smart-network-monitor/network_monitor.log` |
|
||||||
|
| 前台临时监控日志 | `C:\ProgramData\SmartNetworkMonitor\logs\network_monitor_front.log` | `/var/log/smart-network-monitor/network_monitor_front.log` |
|
||||||
|
|
||||||
|
程序按日自动切割日志,默认保留最近 30 天。
|
||||||
110
README.md
110
README.md
@@ -18,12 +18,6 @@ irm https://raw.githubusercontent.com/UnbalancedCat/smart-shutdown/main/install.
|
|||||||
|
|
||||||
> **说明:** 该命令会自动下载最新版本,执行 `install` 将程序复制到 `C:\Program Files\SmartShutdown\` 并写入系统 `PATH` 环境变量,同时注册为 Windows 开机自启服务。此后您可在任意管理员终端直接使用 `smart-shutdown` 命令。
|
> **说明:** 该命令会自动下载最新版本,执行 `install` 将程序复制到 `C:\Program Files\SmartShutdown\` 并写入系统 `PATH` 环境变量,同时注册为 Windows 开机自启服务。此后您可在任意管理员终端直接使用 `smart-shutdown` 命令。
|
||||||
|
|
||||||
如需手动安装,也可前往 [Releases](https://github.com/UnbalancedCat/smart-shutdown/releases) 页面下载对应架构版本,然后在管理员终端中执行:
|
|
||||||
```powershell
|
|
||||||
.\smart-shutdown_windows_amd64.exe install
|
|
||||||
smart-shutdown start
|
|
||||||
```
|
|
||||||
|
|
||||||
### Linux (Ubuntu/Debian/CentOS 等)
|
### Linux (Ubuntu/Debian/CentOS 等)
|
||||||
|
|
||||||
在终端中执行以下一键安装命令:
|
在终端中执行以下一键安装命令:
|
||||||
@@ -32,24 +26,9 @@ smart-shutdown start
|
|||||||
curl -sSL https://raw.githubusercontent.com/UnbalancedCat/smart-shutdown/main/install.sh | sudo sh
|
curl -sSL https://raw.githubusercontent.com/UnbalancedCat/smart-shutdown/main/install.sh | sudo sh
|
||||||
```
|
```
|
||||||
|
|
||||||
如需手动安装,可执行以下命令:
|
|
||||||
|
|
||||||
```bash
|
|
||||||
sudo curl -sSL https://github.com/UnbalancedCat/smart-shutdown/releases/latest/download/smart-shutdown_linux_amd64 -o /usr/local/bin/smart-shutdown && \
|
|
||||||
sudo chmod +x /usr/local/bin/smart-shutdown && \
|
|
||||||
sudo smart-shutdown install && \
|
|
||||||
sudo smart-shutdown start
|
|
||||||
```
|
|
||||||
|
|
||||||
## 快速开始
|
## 快速开始
|
||||||
|
|
||||||
安装完成后,在管理员 / root 终端中执行首次启动:
|
安装完成后,默认会自动运行如下引导,进行初始设置:
|
||||||
|
|
||||||
```bash
|
|
||||||
smart-shutdown start
|
|
||||||
```
|
|
||||||
|
|
||||||
如果程序未检测到配置文件,将自动引导您完成初始设置:
|
|
||||||
|
|
||||||
```
|
```
|
||||||
========== 首次配置 ==========
|
========== 首次配置 ==========
|
||||||
@@ -76,60 +55,49 @@ Windows `sudo` 默认运行在 **ForceNewWindow(强制新窗口)** 模式下
|
|||||||
1. **切换 sudo 为 Inline 模式**:打开 **系统设置 → 开发者选项 → 启用 sudo**,将模式改为 **内联 (Inline)**。此后 `sudo smart-shutdown status` 的输出将正常回显至当前终端。
|
1. **切换 sudo 为 Inline 模式**:打开 **系统设置 → 开发者选项 → 启用 sudo**,将模式改为 **内联 (Inline)**。此后 `sudo smart-shutdown status` 的输出将正常回显至当前终端。
|
||||||
2. **直接使用管理员终端**:右键点击终端图标,选择 **以管理员身份运行**,随后无需 `sudo` 前缀即可执行所有指令。
|
2. **直接使用管理员终端**:右键点击终端图标,选择 **以管理员身份运行**,随后无需 `sudo` 前缀即可执行所有指令。
|
||||||
|
|
||||||
## CLI 终端管理指令 (Commands)
|
## 命令参考
|
||||||
|
|
||||||
完成服务注册后,可通过以下指令直接管理守护进程状态流。
|
完整的命令说明、参数列表与使用示例,请查阅 [API.md](./API.md)。
|
||||||
*(注: 启停服务及修改配置的指令需在 **管理员级别 / root 权限** 终端内执行)*
|
|
||||||
|
常用快捷参考:
|
||||||
|
|
||||||
|
| 命令 | 说明 |
|
||||||
|
|:---|:---|
|
||||||
|
| `smart-shutdown start` | 启动后台监控服务 |
|
||||||
|
| `smart-shutdown stop` | 停止后台监控服务 |
|
||||||
|
| `smart-shutdown restart` | 重启服务 |
|
||||||
|
| `smart-shutdown status` | 查看服务状态与最近日志 |
|
||||||
|
| `smart-shutdown config set <Key> <Value>` | 修改配置项 |
|
||||||
|
| `smart-shutdown pause --stop-after 2h` | 临时挂起后台 2 小时 |
|
||||||
|
| `smart-shutdown resume` | 立即恢复挂起的后台 |
|
||||||
|
| `smart-shutdown uninstall` | 卸载服务及环境变量 |
|
||||||
|
|
||||||
|
## 免安装便携使用与前台监控模式
|
||||||
|
|
||||||
|
如果您只是想临时使用网络检测自动关机功能,而**不想将程序安装为系统常驻服务**,您可以直接下载可执行文件进行免安装的便携式运行(基于前台模式)。
|
||||||
|
|
||||||
|
前往 [Releases](https://github.com/UnbalancedCat/smart-shutdown/releases) 页面下载对应您系统的可执行文件后,在同一目录下的终端中,直接带参数执行即可。例如:
|
||||||
|
|
||||||
### 运行状态查询
|
|
||||||
输出当前服务存活状态、载入的配置参数、日志存放路径及最近 10 条日志:
|
|
||||||
```bash
|
```bash
|
||||||
smart-shutdown status
|
# 临时在前台启动监控(非后台服务),检测目标 IP 8.8.8.8,脱机容忍设为 60 秒
|
||||||
|
.\smart-shutdown --target-ip 8.8.8.8 --window-sec 60
|
||||||
```
|
```
|
||||||
|
|
||||||
### 修改系统配置
|
> **提示:**
|
||||||
直接修改配置参数并执行合法性校验。(注: 改写配置后必须执行 `smart-shutdown restart` 重启服务才能应用生效)
|
> 1. 前台模式下产生的日志将单独写入 `network_monitor_front.log`,不会影响原有数据。关闭当前终端即可停止监控进程。
|
||||||
```bash
|
> 2. 关于前台模式的 `--stop-after` (设定运行多少时间后退出)、`--override-bg` (接管正在运行的后台服务) 等高阶参数详细说明,请参阅 [API.md](./API.md)。
|
||||||
# 修改目标监控 IP 地址
|
|
||||||
smart-shutdown config set TargetIP 192.168.0.1
|
|
||||||
|
|
||||||
# 修改断网响应的容忍监控窗口
|
**扩展:手动安装为服务**
|
||||||
smart-shutdown config set MonitorWindowSeconds 300
|
|
||||||
```
|
|
||||||
|
|
||||||
### 服务生命周期控制
|
如果您下载了二进制文件后改变主意,想手动将其注册为系统服务,也可以直接使用 `install` 命令:
|
||||||
```bash
|
|
||||||
# 停止当前运行的后台网络探测服务
|
|
||||||
smart-shutdown stop
|
|
||||||
|
|
||||||
# 启动后台服务并恢复网络探测
|
- **Windows**:
|
||||||
smart-shutdown start
|
```powershell
|
||||||
|
.\smart-shutdown install
|
||||||
# 重启后台服务
|
```
|
||||||
smart-shutdown restart
|
- **Linux**:
|
||||||
```
|
```bash
|
||||||
|
sudo mv ./smart-shutdown_linux_amd64 /usr/local/bin/smart-shutdown
|
||||||
### 服务与环境卸载
|
sudo chmod +x /usr/local/bin/smart-shutdown
|
||||||
停止并注销后台服务,清理系统目录中的程序副本与对应的系统环境变量:
|
sudo smart-shutdown install
|
||||||
```bash
|
```
|
||||||
smart-shutdown uninstall
|
|
||||||
```
|
|
||||||
|
|
||||||
*(如需查看完整的帮助说明,可执行 `smart-shutdown help`)*
|
|
||||||
|
|
||||||
## 默认配置参数详情
|
|
||||||
程序会在对应操作系统的后台数据存放区建立或读取 JSON 配置文件:
|
|
||||||
- Windows: `C:\ProgramData\SmartNetworkMonitor\config.json`
|
|
||||||
- Linux: `/etc/smart-network-monitor/config.json`
|
|
||||||
|
|
||||||
| 参数项 | 含义与功能说明 | 默认值 |
|
|
||||||
|:---|:---|:---|
|
|
||||||
| `TargetIP` | 需发送 ICMP 包验证联通性的目标 IPv4 地址。 | `192.168.3.1` |
|
|
||||||
| `MonitorWindowSeconds` | 网络中断被判定为异常并触发系统关机前,所能容忍的最长超时时长 (秒)。 | `180` |
|
|
||||||
| `ShutdownCountdown` | 容忍超限后,执行正式关机系统指令的警告倒计时缓冲时间 (秒)。 | `60` |
|
|
||||||
| `NormalPingInterval` | 网络连通性正常时,每次静默发包探测的间隔时间 (秒)。 | `15` |
|
|
||||||
|
|
||||||
## 本地日志存放位置
|
|
||||||
程序按日切割保存网络探测及中断记录,默认留存最近 30 天的文件:
|
|
||||||
- **Windows**: `C:\ProgramData\SmartNetworkMonitor\logs\network_monitor.log`
|
|
||||||
- **Linux**: `/var/log/smart-network-monitor/network_monitor.log`
|
|
||||||
|
|||||||
@@ -1,24 +1,29 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
|
"net"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/kardianos/service"
|
"github.com/kardianos/service"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"smart-shutdown/pkg/config"
|
"smart-shutdown/pkg/config"
|
||||||
"smart-shutdown/pkg/daemon"
|
"smart-shutdown/pkg/daemon"
|
||||||
"smart-shutdown/pkg/logger"
|
"smart-shutdown/pkg/logger"
|
||||||
|
"smart-shutdown/pkg/monitor"
|
||||||
"smart-shutdown/pkg/updater"
|
"smart-shutdown/pkg/updater"
|
||||||
)
|
)
|
||||||
|
|
||||||
var AppVersion = "dev"
|
var AppVersion = "dev"
|
||||||
var verbose bool
|
var verbose bool
|
||||||
|
var configPath string
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
if err := logger.InitLogger(); err != nil {
|
if err := logger.InitLogger(); err != nil {
|
||||||
@@ -26,7 +31,22 @@ func main() {
|
|||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
cfg, err := config.LoadConfig()
|
// Pre-parse -c / --config before cobra so we can load correct config early
|
||||||
|
for i, arg := range os.Args[1:] {
|
||||||
|
if (arg == "-c" || arg == "--config") && i+1 < len(os.Args)-1 {
|
||||||
|
configPath = os.Args[i+2]
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var cfg *config.Config
|
||||||
|
var err error
|
||||||
|
if configPath != "" {
|
||||||
|
logger.Debug("使用自定义配置文件: %s", configPath)
|
||||||
|
cfg, err = config.LoadConfigFrom(configPath)
|
||||||
|
} else {
|
||||||
|
cfg, err = config.LoadConfig()
|
||||||
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Crit("读取配置文件失败: %v", err)
|
logger.Crit("读取配置文件失败: %v", err)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
@@ -38,7 +58,16 @@ func main() {
|
|||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
var showVersion bool
|
var (
|
||||||
|
showVersion bool
|
||||||
|
tmpTargetIP string
|
||||||
|
tmpWindowSec int
|
||||||
|
tmpShutdownCnt int
|
||||||
|
tmpPingInt int
|
||||||
|
tmpOverrideBg bool
|
||||||
|
tmpStopAfterStr string
|
||||||
|
tmpStopAtStr string
|
||||||
|
)
|
||||||
|
|
||||||
var rootCommand = &cobra.Command{
|
var rootCommand = &cobra.Command{
|
||||||
Use: "smart-shutdown",
|
Use: "smart-shutdown",
|
||||||
@@ -56,16 +85,94 @@ func main() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
err := svc.Run()
|
if !service.Interactive() {
|
||||||
if err != nil {
|
err := svc.Run()
|
||||||
logger.Fail("后台监控流崩溃: %v", err)
|
if err != nil {
|
||||||
|
logger.Fail("后台监控流崩溃: %v", err)
|
||||||
|
}
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Interactive foreground mode
|
||||||
|
logger.SwitchToFrontLog()
|
||||||
|
|
||||||
|
if tmpTargetIP != "" {
|
||||||
|
if net.ParseIP(tmpTargetIP) != nil {
|
||||||
|
cfg.TargetIP = tmpTargetIP
|
||||||
|
} else {
|
||||||
|
logger.Fail("参数解析错误: 非法的 IP 地址 %s", tmpTargetIP)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if tmpWindowSec > 0 { cfg.MonitorWindowSeconds = tmpWindowSec }
|
||||||
|
if tmpShutdownCnt > 0 { cfg.ShutdownCountdown = tmpShutdownCnt }
|
||||||
|
if tmpPingInt > 0 { cfg.NormalPingInterval = tmpPingInt }
|
||||||
|
|
||||||
|
if tmpOverrideBg {
|
||||||
|
status, _ := svc.Status()
|
||||||
|
if status == service.StatusRunning {
|
||||||
|
logger.Info("检测到后台服务运行中,正在挂起以接管前台监控...")
|
||||||
|
service.Control(svc, "stop")
|
||||||
|
defer func() {
|
||||||
|
logger.Info("前台监控结束,恢复后台服务...")
|
||||||
|
service.Control(svc, "start")
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
importTime := false
|
||||||
|
var earliest = time.Now().Add(100 * 365 * 24 * time.Hour) // very far
|
||||||
|
if tmpStopAfterStr != "" {
|
||||||
|
d, err := time.ParseDuration(tmpStopAfterStr)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fail("解析持续时间失败 (如 120m, 2h): %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
t := time.Now().Add(d)
|
||||||
|
if t.Before(earliest) { earliest = t; importTime = true }
|
||||||
|
}
|
||||||
|
if tmpStopAtStr != "" {
|
||||||
|
layout := "2006-01-02 15:04:05"
|
||||||
|
t, err := time.Parse(layout, tmpStopAtStr)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fail("解析绝对时间失败 (格式需为 2006-01-02 15:04:05): %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if t.Before(earliest) { earliest = t; importTime = true }
|
||||||
|
}
|
||||||
|
|
||||||
|
if importTime {
|
||||||
|
if earliest.Before(time.Now()) {
|
||||||
|
logger.Fail("指定的结束时间不能早于当前时间")
|
||||||
|
cancel()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var deadlineCancel context.CancelFunc
|
||||||
|
ctx, deadlineCancel = context.WithDeadline(ctx, earliest)
|
||||||
|
defer deadlineCancel()
|
||||||
|
logger.Info("已设定前台监控截止退出时间: %s", earliest.Format("2006-01-02 15:04:05"))
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Info("开始执行前台监控进程...")
|
||||||
|
monitor.Run(ctx, cfg)
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
rootCommand.CompletionOptions.DisableDefaultCmd = true
|
rootCommand.CompletionOptions.DisableDefaultCmd = true
|
||||||
rootCommand.Flags().BoolVarP(&showVersion, "version", "V", false, "输出当前二进制内核版本并联网拉取发布树状态")
|
rootCommand.Flags().BoolVarP(&showVersion, "version", "V", false, "输出当前二进制内核版本并联网拉取发布树状态")
|
||||||
rootCommand.PersistentFlags().BoolVarP(&verbose, "verbose", "v", false, "打印底层部署及环境追溯 Debug 信息")
|
rootCommand.PersistentFlags().BoolVarP(&verbose, "verbose", "v", false, "打印底层部署及环境追溯 Debug 信息")
|
||||||
|
rootCommand.PersistentFlags().StringVarP(&configPath, "config", "c", "", "指定自定义配置文件路径 (如 ./example_config.json)")
|
||||||
|
|
||||||
|
rootCommand.Flags().StringVar(&tmpTargetIP, "target-ip", "", "临时覆盖: 目标监控 IP")
|
||||||
|
rootCommand.Flags().IntVar(&tmpWindowSec, "window-sec", 0, "临时覆盖: 容忍超时时长 (秒)")
|
||||||
|
rootCommand.Flags().IntVar(&tmpShutdownCnt, "shutdown-cnt", 0, "临时覆盖: 关机倒计时 (秒)")
|
||||||
|
rootCommand.Flags().IntVar(&tmpPingInt, "ping-interval", 0, "临时覆盖: 发包间隔 (秒)")
|
||||||
|
rootCommand.Flags().BoolVar(&tmpOverrideBg, "override-bg", false, "挂起后台运行的守护进程,由前台全面接管")
|
||||||
|
rootCommand.Flags().StringVar(&tmpStopAfterStr, "stop-after", "", "前台运行多少时长后自动退出 (如 120m, 2h)")
|
||||||
|
rootCommand.Flags().StringVar(&tmpStopAtStr, "stop-at", "", "前台运行至指定时间自动退出 (如 '2026-03-24 23:59:00')")
|
||||||
|
|
||||||
cmds := []*cobra.Command{
|
cmds := []*cobra.Command{
|
||||||
{
|
{
|
||||||
@@ -112,30 +219,39 @@ func main() {
|
|||||||
Short: "废除并移除在册的后台服务、扫除环境关联残留",
|
Short: "废除并移除在册的后台服务、扫除环境关联残留",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
service.Control(svc, "stop")
|
service.Control(svc, "stop")
|
||||||
handleServiceControl(svc, "uninstall")
|
if !handleServiceControl(svc, "uninstall") {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
targetDir, targetExe := getTargetSystemPath()
|
targetDir, targetExe := getTargetSystemPath()
|
||||||
currentExe, _ := os.Executable()
|
currentExe, _ := os.Executable()
|
||||||
|
isSelfExe := strings.EqualFold(filepath.Clean(currentExe), filepath.Clean(targetExe))
|
||||||
|
|
||||||
if !strings.EqualFold(filepath.Clean(currentExe), filepath.Clean(targetExe)) {
|
// Clean up PATH (Windows)
|
||||||
if _, err := os.Stat(targetExe); err == nil {
|
if runtime.GOOS == "windows" {
|
||||||
logger.Debug("查明环境曾记录过全局部署逻辑,现已启动强力移除可执行源地址流: %s", targetExe)
|
envClearCmd := fmt.Sprintf(`$p=[Environment]::GetEnvironmentVariable("Path","Machine");$np=($p -split ';' | Where-Object {$_ -ne "%s" -and $_ -ne ""}) -join ';';[Environment]::SetEnvironmentVariable("Path",$np,"Machine")`, targetDir)
|
||||||
|
exec.Command("powershell", "-Command", envClearCmd).Run()
|
||||||
|
logger.Debug("环境变量系统清理:Windows Global Path 中的指向挂载项业已剥除卸载完成。")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Delete executable
|
||||||
|
if _, err := os.Stat(targetExe); err == nil {
|
||||||
|
if isSelfExe && runtime.GOOS == "windows" {
|
||||||
|
// Windows: cannot delete a running exe, use delayed self-deletion
|
||||||
|
delCmd := fmt.Sprintf(`ping 127.0.0.1 -n 2 > nul & del "%s" & rmdir "%s"`, targetExe, targetDir)
|
||||||
|
exec.Command("cmd", "/C", "start", "/min", "cmd", "/C", delCmd).Start()
|
||||||
|
logger.Debug("已调度延迟自删除任务: %s", targetExe)
|
||||||
|
} else {
|
||||||
os.Remove(targetExe)
|
os.Remove(targetExe)
|
||||||
|
|
||||||
if runtime.GOOS == "windows" {
|
if runtime.GOOS == "windows" {
|
||||||
os.Remove(targetDir)
|
os.Remove(targetDir)
|
||||||
envClearCmd := fmt.Sprintf(`$p=[Environment]::GetEnvironmentVariable("Path","Machine");$np=($p -split ';' | Where-Object {$_ -ne "%s" -and $_ -ne ""}) -join ';';[Environment]::SetEnvironmentVariable("Path",$np,"Machine")`, targetDir)
|
|
||||||
exec.Command("powershell", "-Command", envClearCmd).Run()
|
|
||||||
logger.Debug("环境变量系统清理:Windows Global Path 中的指向挂载项业已剥除卸载完成。")
|
|
||||||
}
|
}
|
||||||
|
logger.Debug("已移除可执行文件: %s", targetExe)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if config.ConfirmConfigCleanup() {
|
// Prompt for config and log cleanup
|
||||||
configDir := config.GetConfigDir()
|
config.ConfirmAndCleanup()
|
||||||
os.RemoveAll(configDir)
|
|
||||||
fmt.Println("配置文件与日志已清除。")
|
|
||||||
}
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@@ -268,8 +384,63 @@ func main() {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var pauseCmd = &cobra.Command{
|
||||||
|
Use: "pause",
|
||||||
|
Short: "临时休眠后台运行的守护进程(不开启前台监控)",
|
||||||
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
if tmpStopAfterStr == "" && tmpStopAtStr == "" {
|
||||||
|
logger.Fail("必须指定 --stop-after 或 --stop-at 参数才能执行休眠指令")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var until = time.Now().Add(100 * 365 * 24 * time.Hour)
|
||||||
|
if tmpStopAfterStr != "" {
|
||||||
|
d, err := time.ParseDuration(tmpStopAfterStr)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fail("解析持续时间失败: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
t := time.Now().Add(d)
|
||||||
|
if t.Before(until) { until = t }
|
||||||
|
}
|
||||||
|
if tmpStopAtStr != "" {
|
||||||
|
layout := "2006-01-02 15:04:05"
|
||||||
|
t, err := time.Parse(layout, tmpStopAtStr)
|
||||||
|
if err != nil {
|
||||||
|
logger.Fail("解析绝对时间失败: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if t.Before(until) { until = t }
|
||||||
|
}
|
||||||
|
|
||||||
|
if until.Before(time.Now()) {
|
||||||
|
logger.Fail("指定的唤醒时间不能早于当前时间")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := config.SetPause(until); err != nil {
|
||||||
|
logger.Fail("写入休眠指令失败: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
logger.Succ("指令下达成功!后台守护进程将挂机休息至 %s", until.Format("2006-01-02 15:04:05"))
|
||||||
|
},
|
||||||
|
}
|
||||||
|
pauseCmd.Flags().StringVar(&tmpStopAfterStr, "stop-after", "", "休眠多少时长后自动唤醒 (如 120m, 2h)")
|
||||||
|
pauseCmd.Flags().StringVar(&tmpStopAtStr, "stop-at", "", "休眠至指定时间自动唤醒 (如 '2026-03-24 23:59:00')")
|
||||||
|
|
||||||
|
var resumeCmd = &cobra.Command{
|
||||||
|
Use: "resume",
|
||||||
|
Short: "撤销休眠指令,立即唤醒后台网络探测",
|
||||||
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
config.ClearPause()
|
||||||
|
logger.Succ("休眠指令已撤销,后台将在下一轮探测周期中重新激活!")
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
configCmd.AddCommand(configSetCmd)
|
configCmd.AddCommand(configSetCmd)
|
||||||
rootCommand.AddCommand(configCmd)
|
rootCommand.AddCommand(configCmd)
|
||||||
|
rootCommand.AddCommand(pauseCmd)
|
||||||
|
rootCommand.AddCommand(resumeCmd)
|
||||||
|
|
||||||
for _, c := range cmds {
|
for _, c := range cmds {
|
||||||
rootCommand.AddCommand(c)
|
rootCommand.AddCommand(c)
|
||||||
@@ -287,18 +458,19 @@ func main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleServiceControl(s service.Service, action string) {
|
func handleServiceControl(s service.Service, action string) bool {
|
||||||
err := service.Control(s, action)
|
err := service.Control(s, action)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
errStr := err.Error()
|
errStr := err.Error()
|
||||||
if strings.Contains(errStr, "Access is denied") || strings.Contains(errStr, "permission denied") || strings.Contains(errStr, "拒绝访问") {
|
if strings.Contains(errStr, "Access is denied") || strings.Contains(errStr, "permission denied") || strings.Contains(errStr, "拒绝访问") {
|
||||||
logger.Fail("管控流程 [%s] 触发越级防卫!无核准身份特设记录。请开具含有全 Root 及高配权限窗格承接口径。", action)
|
logger.Fail("管控流程 [%s] 触发越级防卫!无核准身份特设记录。请开具含有全 Root 及高配权限窗格承接口径。", action)
|
||||||
return
|
return false
|
||||||
}
|
}
|
||||||
logger.Crit("后台执行流程组块阻断报错 [%s] : %v", action, err)
|
logger.Crit("后台执行流程组块阻断报错 [%s] : %v", action, err)
|
||||||
return
|
return false
|
||||||
}
|
}
|
||||||
logger.Succ("指令动作 [%s] 解析下发执行完毕,无阻断警告。", action)
|
logger.Succ("指令动作 [%s] 解析下发执行完毕,无阻断警告。", action)
|
||||||
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
func printLastLogLines(n int) {
|
func printLastLogLines(n int) {
|
||||||
|
|||||||
6
example_config.json
Normal file
6
example_config.json
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
{
|
||||||
|
"TargetIP": "192.168.3.1",
|
||||||
|
"MonitorWindowSeconds": 180,
|
||||||
|
"ShutdownCountdown": 60,
|
||||||
|
"NormalPingInterval": 15
|
||||||
|
}
|
||||||
@@ -43,7 +43,14 @@ func GetLogDir() string {
|
|||||||
func LoadConfig() (*Config, error) {
|
func LoadConfig() (*Config, error) {
|
||||||
configDir := GetConfigDir()
|
configDir := GetConfigDir()
|
||||||
configPath := filepath.Join(configDir, "config.json")
|
configPath := filepath.Join(configDir, "config.json")
|
||||||
|
return loadConfigFile(configPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoadConfigFrom(path string) (*Config, error) {
|
||||||
|
return loadConfigFile(path)
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadConfigFile(configPath string) (*Config, error) {
|
||||||
data, err := os.ReadFile(configPath)
|
data, err := os.ReadFile(configPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if os.IsNotExist(err) {
|
if os.IsNotExist(err) {
|
||||||
|
|||||||
@@ -63,25 +63,39 @@ func InteractiveSetup() (*Config, error) {
|
|||||||
return cfg, nil
|
return cfg, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// ConfirmConfigCleanup prompts the user to confirm deletion of config and log files.
|
// ConfirmAndCleanup prompts the user to confirm deletion of config and log files separately.
|
||||||
// Returns true if the user confirms. Non-terminal stdin defaults to no.
|
// Non-terminal stdin skips all prompts.
|
||||||
func ConfirmConfigCleanup() bool {
|
func ConfirmAndCleanup() {
|
||||||
if !isTerminal() {
|
if !isTerminal() {
|
||||||
return false
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
scanner := bufio.NewScanner(os.Stdin)
|
||||||
configDir := GetConfigDir()
|
configDir := GetConfigDir()
|
||||||
configPath := filepath.Join(configDir, "config.json")
|
configPath := filepath.Join(configDir, "config.json")
|
||||||
if _, err := os.Stat(configPath); err != nil {
|
logDir := GetLogDir()
|
||||||
return false
|
|
||||||
|
if _, err := os.Stat(configPath); err == nil {
|
||||||
|
fmt.Printf("检测到配置文件: %s\n是否清除配置文件?[y/N]: ", configPath)
|
||||||
|
if scanner.Scan() && strings.EqualFold(strings.TrimSpace(scanner.Text()), "y") {
|
||||||
|
os.Remove(configPath)
|
||||||
|
fmt.Println("配置文件已清除。")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Printf("检测到配置文件: %s\n是否一并清除配置文件与日志?[y/N]: ", configDir)
|
if _, err := os.Stat(logDir); err == nil {
|
||||||
scanner := bufio.NewScanner(os.Stdin)
|
fmt.Printf("检测到日志目录: %s\n是否清除日志文件?[y/N]: ", logDir)
|
||||||
if scanner.Scan() {
|
if scanner.Scan() && strings.EqualFold(strings.TrimSpace(scanner.Text()), "y") {
|
||||||
return strings.EqualFold(strings.TrimSpace(scanner.Text()), "y")
|
os.RemoveAll(logDir)
|
||||||
|
fmt.Println("日志文件已清除。")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If configDir is now empty, remove it too
|
||||||
|
entries, err := os.ReadDir(configDir)
|
||||||
|
if err == nil && len(entries) == 0 {
|
||||||
|
os.Remove(configDir)
|
||||||
}
|
}
|
||||||
return false
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func promptString(scanner *bufio.Scanner, label, defaultVal string, validate func(string) error) string {
|
func promptString(scanner *bufio.Scanner, label, defaultVal string, validate func(string) error) string {
|
||||||
|
|||||||
52
pkg/config/pause.go
Normal file
52
pkg/config/pause.go
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// SetPause writes a timestamp into pause_until.txt
|
||||||
|
func SetPause(until time.Time) error {
|
||||||
|
dir := GetConfigDir()
|
||||||
|
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
path := filepath.Join(dir, "pause_until.txt")
|
||||||
|
ts := strconv.FormatInt(until.Unix(), 10)
|
||||||
|
return os.WriteFile(path, []byte(ts), 0644)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ClearPause removes the pause_until.txt file
|
||||||
|
func ClearPause() error {
|
||||||
|
path := filepath.Join(GetConfigDir(), "pause_until.txt")
|
||||||
|
return os.Remove(path)
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsPaused checks if the service is currently paused via IPC file
|
||||||
|
func IsPaused() (bool, int64) {
|
||||||
|
path := filepath.Join(GetConfigDir(), "pause_until.txt")
|
||||||
|
data, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return false, 0
|
||||||
|
}
|
||||||
|
|
||||||
|
tsStr := strings.TrimSpace(string(data))
|
||||||
|
untilUnix, err := strconv.ParseInt(tsStr, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
// Invalid file, clean it up
|
||||||
|
os.Remove(path)
|
||||||
|
return false, 0
|
||||||
|
}
|
||||||
|
|
||||||
|
now := time.Now().Unix()
|
||||||
|
if now < untilUnix {
|
||||||
|
return true, untilUnix - now
|
||||||
|
}
|
||||||
|
|
||||||
|
// Expired
|
||||||
|
os.Remove(path)
|
||||||
|
return false, 0
|
||||||
|
}
|
||||||
@@ -34,6 +34,24 @@ func InitLogger() error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func SwitchToFrontLog() {
|
||||||
|
if fileLogger != nil {
|
||||||
|
fileLogger.Close()
|
||||||
|
}
|
||||||
|
logDir := config.GetLogDir()
|
||||||
|
if err := os.MkdirAll(logDir, 0755); err != nil {
|
||||||
|
logDir = "logs"
|
||||||
|
}
|
||||||
|
logFile := filepath.Join(logDir, "network_monitor_front.log")
|
||||||
|
fileLogger = &lumberjack.Logger{
|
||||||
|
Filename: logFile,
|
||||||
|
MaxSize: 10,
|
||||||
|
MaxBackups: 30,
|
||||||
|
MaxAge: 30,
|
||||||
|
Compress: false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func EnableDebug() {
|
func EnableDebug() {
|
||||||
debugEnabled = true
|
debugEnabled = true
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -18,6 +18,7 @@ func Run(ctx context.Context, cfg *config.Config) {
|
|||||||
|
|
||||||
var failureStartTime time.Time
|
var failureStartTime time.Time
|
||||||
var inFailureWindow bool = false
|
var inFailureWindow bool = false
|
||||||
|
var wasPausedBefore bool = false
|
||||||
|
|
||||||
normalStatusCounter := 0
|
normalStatusCounter := 0
|
||||||
const normalStatusLogInterval = 24
|
const normalStatusLogInterval = 24
|
||||||
@@ -30,6 +31,27 @@ func Run(ctx context.Context, cfg *config.Config) {
|
|||||||
default:
|
default:
|
||||||
}
|
}
|
||||||
|
|
||||||
|
isPaused, remainingSec := config.IsPaused()
|
||||||
|
if isPaused {
|
||||||
|
if !wasPausedBefore {
|
||||||
|
mins := remainingSec / 60
|
||||||
|
if mins < 1 {
|
||||||
|
mins = 1
|
||||||
|
}
|
||||||
|
logger.Info("进入管理员特设休眠挂起状态, 预计将在 %d 分钟后自动唤醒", mins)
|
||||||
|
wasPausedBefore = true
|
||||||
|
if inFailureWindow {
|
||||||
|
inFailureWindow = false
|
||||||
|
normalStatusCounter = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
time.Sleep(15 * time.Second)
|
||||||
|
continue
|
||||||
|
} else if wasPausedBefore {
|
||||||
|
logger.Info("休眠指令逾期或已被撤消, 唤醒并恢复网络探测")
|
||||||
|
wasPausedBefore = false
|
||||||
|
}
|
||||||
|
|
||||||
isOnline := pinger.Ping(cfg.TargetIP, 3)
|
isOnline := pinger.Ping(cfg.TargetIP, 3)
|
||||||
|
|
||||||
if isOnline {
|
if isOnline {
|
||||||
|
|||||||
Reference in New Issue
Block a user