Merge branch 'master' of http://gitea.stonelan.cn/admin/membank
This commit is contained in:
@@ -25,7 +25,7 @@ type Config struct {
|
||||
// 分类配置项
|
||||
// ============================
|
||||
|
||||
// ScanConfig 网段扫描
|
||||
// ScanConfig 目标扫描配置(支持 CIDR 和 IP 范围格式)
|
||||
type ScanConfig struct {
|
||||
CIDR string `yaml:"cidr"`
|
||||
Timeout time.Duration `yaml:"timeout"`
|
||||
|
||||
@@ -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 个回包。
|
||||
|
||||
@@ -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))
|
||||
|
||||
@@ -5,7 +5,7 @@
|
||||
总耗时: {{ .Duration }}
|
||||
{{ "" }}
|
||||
── 配置 ──
|
||||
网段: {{ .Config.CIDR }}
|
||||
目标: {{ .Config.CIDR }}
|
||||
SSH: {{ .Config.User }}@:{{ .Config.Port }}
|
||||
压测: {{ .Config.StressTypes }}
|
||||
并发: {{ .Config.Concurrency }}
|
||||
|
||||
Reference in New Issue
Block a user