diff --git a/auto-check/README.md b/auto-check/README.md index 4956a31..290cdb1 100644 --- a/auto-check/README.md +++ b/auto-check/README.md @@ -1,6 +1,6 @@ # auto-check -自动化网段扫描 + SSH 登录 + 远程压力测试工具。 +自动化目标扫描 + SSH 登录 + 远程压力测试工具（支持 CIDR 和 IP 范围格式）。 基于 [cobra](https://github.com/spf13/cobra) 命令行框架。 @@ -9,7 +9,7 @@ | 子命令 | 功能 | 包 | |--------|------|-----| | `run` | 完整流程：发现 → SSH → 压测 | 全部 | -| `scan` | 仅扫描网段，发现存活设备 | `pkg/discovery` | +| `scan` | 仅扫描目标，发现存活设备 | `pkg/discovery` | | `stress` | 对指定 IP 执行压力测试 | `pkg/stress` + `pkg/sshclient` | ## 安装 @@ -24,9 +24,12 @@ go build -o auto-check . # 查看帮助 ./auto-check --help -# 仅扫描网段 +# 仅扫描目标（CIDR 格式） ./auto-check scan --cidr 192.168.1.0/24 +# 仅扫描目标（IP 范围格式） +./auto-check scan --cidr 192.168.1.1-192.168.1.100 + # 对指定主机执行压力测试 ./auto-check stress --ips 192.168.1.100,192.168.1.101 --user root --password secret --stress cpu,memory,disk @@ -38,7 +41,7 @@ go build -o auto-check . | 标志 | 默认值 | 说明 | |------|--------|------| -| `--cidr` | `192.168.1.0/24` | 目标网段 (CIDR) | +| `--cidr` | `192.168.1.0/24` | 目标网段或 IP 范围 (CIDR 或 起始IP-结束IP) | | `--user` | `root` | SSH 用户名 | | `--password` | _(空)_ | SSH 密码 | | `--key` | _(空)_ | SSH 私钥路径 | diff --git a/auto-check/auto-check.yaml b/auto-check/auto-check.yaml index 1ae9c10..6a37bb6 100644 --- a/auto-check/auto-check.yaml +++ b/auto-check/auto-check.yaml @@ -2,14 +2,17 @@ # 所有参数仅从本文件加载，按分类组织 # --- 网段扫描 --- +# 支持两种格式： +# - CIDR: "10.0.3.0/24" +# - IP 范围: "10.0.3.1-10.0.3.100" scan: - cidr: "10.0.3.0/24" + cidr: "10.107.95.2-10.107.95.2" timeout: 3s concurrency: 10 # --- SSH 认证 --- ssh: - user: "admin" + user: "admin-pc" password: "admin123" key: "" port: 22 diff --git a/auto-check/cmd/root.go b/auto-check/cmd/root.go index 817320b..db88843 100644 --- a/auto-check/cmd/root.go +++ b/auto-check/cmd/root.go @@ -15,7 +15,7 @@ var rootCmd = &cobra.Command{ Use: "auto-check", Short: "自动化设备发现 + SSH 登录 + 远程压力测试工具", Long: `auto-check 是一个自动化运维工具，支持： - 1. 网段扫描发现存活设备 + 1. 目标扫描发现存活设备（支持 CIDR 和 IP 范围格式） 2. 自动 SSH 登录验证 3. 远程执行压力测试（CPU/内存/磁盘/网络） diff --git a/auto-check/cmd/run.go b/auto-check/cmd/run.go index 2196eeb..5baf735 100644 --- a/auto-check/cmd/run.go +++ b/auto-check/cmd/run.go @@ -18,7 +18,7 @@ var runCmd = &cobra.Command{ if err != nil { return err } - fmt.Printf(" [配置] 加载完成: 网段=%s 间隔=%v 压测=%s\n", + fmt.Printf(" [配置] 加载完成: 目标=%s 间隔=%v 压测=%s\n", cfg.Scan.CIDR, cfg.Workflow.Interval, cfg.Stress.Types) // 启动工作流（永久循环） diff --git a/auto-check/pkg/config/config.go b/auto-check/pkg/config/config.go index a6df400..d5092c8 100644 --- a/auto-check/pkg/config/config.go +++ b/auto-check/pkg/config/config.go @@ -25,7 +25,7 @@ type Config struct { // 分类配置项 // ============================ -// ScanConfig 网段扫描 +// ScanConfig 目标扫描配置（支持 CIDR 和 IP 范围格式） type ScanConfig struct { CIDR string `yaml:"cidr"` Timeout time.Duration `yaml:"timeout"` diff --git a/auto-check/pkg/discovery/scanner.go b/auto-check/pkg/discovery/scanner.go index ea01eaf..322eadb 100644 --- a/auto-check/pkg/discovery/scanner.go +++ b/auto-check/pkg/discovery/scanner.go @@ -15,7 +15,7 @@ import ( "golang.org/x/text/transform" ) -// Scanner 网段扫描器（并发 ping 探测） +// Scanner 网段扫描器（并发 ping 探测，支持 CIDR 和 IP 范围格式） type Scanner struct { CIDR string Timeout time.Duration // 单 IP 探测超时 @@ -40,7 +40,7 @@ func (s *Scanner) Scan() []model.Device { if !ok { return nil } - fmt.Printf("[扫描] 网段 %s，共 %d 个IP，%d 并发 ping 探测...\n", s.CIDR, len(ips), s.Concurrency) + fmt.Printf("[扫描] 目标 %s，共 %d 个IP，%d 并发 ping 探测...\n", s.CIDR, len(ips), s.Concurrency) jobs := make(chan string, len(ips)) results := make(chan model.Device, len(ips)) @@ -75,8 +75,21 @@ func (s *Scanner) Scan() []model.Device { return devices } -// candidateIPs 解析网段并生成候选 IP 列表（排除网络地址和广播地址） +// candidateIPs 解析网段并生成候选 IP 列表。 +// 支持两种格式： +// - CIDR: "10.0.3.0/24"（排除网络地址和广播地址） +// - IP 范围: "10.0.3.1-10.0.3.100"（包含起始和结束 IP） func (s *Scanner) candidateIPs() ([]string, bool) { + // 检查是否为 IP 范围格式 + if strings.Contains(s.CIDR, "-") { + return s.parseIPRange() + } + // 否则按 CIDR 解析 + return s.parseCIDR() +} + +// parseCIDR 解析 CIDR 格式并生成候选 IP 列表 +func (s *Scanner) parseCIDR() ([]string, bool) { _, ipnet, err := net.ParseCIDR(s.CIDR) if err != nil { fmt.Printf("[扫描] 网段解析失败: %v\n", err) @@ -93,6 +106,49 @@ func (s *Scanner) candidateIPs() ([]string, bool) { return ips, true } +// parseIPRange 解析 IP 范围格式（如 "10.0.3.1-10.0.3.100"） +func (s *Scanner) parseIPRange() ([]string, bool) { + parts := strings.SplitN(s.CIDR, "-", 2) + if len(parts) != 2 { + fmt.Printf("[扫描] IP 范围格式无效: %s\n", s.CIDR) + return nil, false + } + startIP := net.ParseIP(strings.TrimSpace(parts[0])) + endIP := net.ParseIP(strings.TrimSpace(parts[1])) + if startIP == nil || endIP == nil { + fmt.Printf("[扫描] IP 地址解析失败: %s\n", s.CIDR) + return nil, false + } + // 转换为 4 字节 IPv4 + start := startIP.To4() + end := endIP.To4() + if start == nil || end == nil { + fmt.Printf("[扫描] 仅支持 IPv4 范围: %s\n", s.CIDR) + return nil, false + } + // 确保起始 <= 结束 + if ipToUint32(start) > ipToUint32(end) { + fmt.Printf("[扫描] 起始 IP 大于结束 IP: %s\n", s.CIDR) + return nil, false + } + var ips []string + for ip := make(net.IP, len(start)); ; { + copy(ip, start) + ips = append(ips, ip.String()) + if start.Equal(end) { + break + } + inc(start) + } + return ips, true +} + +// ipToUint32 将 IPv4 地址转换为 uint32 用于比较 +func ipToUint32(ip net.IP) uint32 { + ip = ip.To4() + return uint32(ip[0])<<24 | uint32(ip[1])<<16 | uint32(ip[2])<<8 | uint32(ip[3]) +} + // ping 探测单个 IP 是否存活（ICMP，经系统 ping 命令）。 // 仅凭 exit code 不可靠（Windows 在某些代答/虚拟网卡场景会返回 0）， // 因此解析输出，要求至少收到 1 个回包。 diff --git a/auto-check/pkg/report/report.go b/auto-check/pkg/report/report.go index 59d08c9..dfab571 100644 --- a/auto-check/pkg/report/report.go +++ b/auto-check/pkg/report/report.go @@ -166,7 +166,7 @@ func (r *Report) ToText() string { // 配置摘要 sb.WriteString("── 配置摘要 ──\n") - sb.WriteString(fmt.Sprintf(" 网段: %s\n", r.Config.CIDR)) + sb.WriteString(fmt.Sprintf(" 目标: %s\n", r.Config.CIDR)) sb.WriteString(fmt.Sprintf(" SSH端口: %d\n", r.Config.Port)) sb.WriteString(fmt.Sprintf(" SSH用户: %s\n", r.Config.User)) sb.WriteString(fmt.Sprintf(" 并发数: %d\n", r.Config.Concurrency)) diff --git a/auto-check/pkg/report/templates/report_80mm.tmpl b/auto-check/pkg/report/templates/report_80mm.tmpl index 1a5d7b5..df47ebe 100644 --- a/auto-check/pkg/report/templates/report_80mm.tmpl +++ b/auto-check/pkg/report/templates/report_80mm.tmpl @@ -5,7 +5,7 @@ 总耗时: {{ .Duration }} {{ "" }} ── 配置 ── - 网段: {{ .Config.CIDR }} + 目标: {{ .Config.CIDR }} SSH: {{ .Config.User }}@:{{ .Config.Port }} 压测: {{ .Config.StressTypes }} 并发: {{ .Config.Concurrency }}